GGSTPrimaryDisReader.cpp

Go to the documentation of this file.

/*
 * GGSTPrimaryDisReader.cpp
 *
 *  Created on: 25 Aug 2011
 *      Author: Nicola Mori
 */

#include "montecarlo/readers/GGSTPrimaryDisReader.h"

GGSTPrimaryDisReader::GGSTPrimaryDisReader() : _disInfo(nullptr), _disInfoChain(nullptr) {}

GGSTPrimaryDisReader::~GGSTPrimaryDisReader() { delete _disInfo; }

bool GGSTPrimaryDisReader::SetChain(TChain *disInfoChain) {

  _disInfoChain = disInfoChain;
  if (_disInfoChain->SetBranchAddress("primaryDisInfo", &_disInfo) == 0) {
    return true;
  } else {
    return false;
  }
}

void GGSTPrimaryDisReader::GetEntry(Long64_t entry) {
  if (entry != _disInfoChain->GetEntries()) {
    _disInfoChain->GetEntry(entry);
  }
}

GGSTPrimaryDisInfo *GGSTPrimaryDisReader::GetDisInfo(Int_t trackID) {

  GGSTPrimaryDisInfo *interaction;
  for (int i = 0; i < _disInfo->GetEntries(); i++) {
    interaction = (GGSTPrimaryDisInfo *)(_disInfo->At(i));
    if (interaction->primary.trackID == trackID) {
      return interaction;
    }
  }

  return nullptr;
}