GGSTHadrIntReader.cpp

Go to the documentation of this file.

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

#include "montecarlo/readers/GGSTHadrIntReader.h"

//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.....

GGSTHadrIntReader::GGSTHadrIntReader() : _inelastic(NULL), _quasiElastic(NULL), _hadrIntChain(NULL) {}

//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.....

GGSTHadrIntReader::~GGSTHadrIntReader() {
  delete _quasiElastic;
  delete _inelastic;
}

//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.....

bool GGSTHadrIntReader::SetChain(TChain *hadrIntChain) {

  _hadrIntChain = hadrIntChain;
  if (_hadrIntChain->SetBranchAddress("inelastic", &_inelastic) == 0 &&
      _hadrIntChain->SetBranchAddress("quasiElastic", &_quasiElastic) == 0) {
    return true;
  } else
    return false;
}

//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.....

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

//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.....
GGSTHadrIntInfo *GGSTHadrIntReader::GetInelastic(Int_t trackID) {

  GGSTHadrIntInfo *interaction;
  for (int i = 0; i < _inelastic->GetEntries(); i++) {
    interaction = (GGSTHadrIntInfo *)(_inelastic->At(i));
    if (interaction->originalTrackID == trackID)
      return interaction;
  }

  return NULL;
}

//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.....

Int_t GGSTHadrIntReader::GetNQuasiElastic(Int_t trackID) {

  Int_t nInt = 0;
  for (int i = 0; i < _quasiElastic->GetEntries(); i++) {
    if (((GGSTHadrIntInfo *)(_quasiElastic->At(i)))->originalTrackID == trackID)
      nInt++;
  }

  return nInt;
}

//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.....

GGSTHadrIntInfo *GGSTHadrIntReader::GetQuasiElastic(Int_t nQE, Int_t trackID) {

  Int_t nInt = 0;
  GGSTHadrIntInfo *interaction;
  for (int i = 0; i < _quasiElastic->GetEntries(); i++) {
    interaction = (GGSTHadrIntInfo *)(_quasiElastic->At(i));
    if (interaction->originalTrackID == trackID) {
      if (nInt == nQE)
        return interaction;
      else
        nInt++;
    }
  }

  return NULL;
}

//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.....