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/readers/GGSTChainReader.h"
#include "montecarlo/dataobjs/GGSTMCTruthInfo.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_ */