GGSTMCTruthReader.h

Go to the documentation of this file.

/*
 * GGSTMCTruthReader.h
 *
 *  Created on: 25 Aug 2011
 *      Author: Nicola Mori
 */

#ifndef GGSTMCTRUTHREADER_H_
#define GGSTMCTRUTHREADER_H_

// GGS headers
#include "montecarlo/dataobjs/GGSTMCTruthInfo.h"
#include "montecarlo/readers/GGSTChainReader.h"

// ROOT headers
class TBranch;

class GGSTMCTruthReader : public GGSTChainReader {

public:
  GGSTMCTruthReader();

  ~GGSTMCTruthReader();

  bool SetChain(TChain *mcTruthChain);

  void GetEntry(Long64_t entry);

  GGSTParticle *GetPrimaryParticle(UInt_t particleNumber) {
    if (_mcTruthInfo)
      return _mcTruthInfo->GetPrimaryParticle(particleNumber);
    else
      return NULL;
  }

  UInt_t GetNPrimaries() {
    if (_mcTruthInfo)
      return _mcTruthInfo->GetNPrimaries();
    else
      return 0;
  }

  UInt_t GetNDiscarded() {
    if (_mcTruthInfo)
      return _mcTruthInfo->nDiscarded;
    else
      return 0;
  }

private:
  GGSTMCTruthInfo *_mcTruthInfo;
  TChain *_mcTruthChain;
};

#endif /* GGSTMCTRUTHREADER_H_ */