CbmMCInput.h

Go to the documentation of this file.

 
#ifndef CBMMCINPUT_H
#define CBMMCINPUT_H 1
 
#include "CbmDefs.h"
#include "TChain.h"
#include "TF1.h"
#include "TObject.h"
#include <set>
 
 
class CbmMCInput : public TObject {
 
public:
  CbmMCInput();
 
 
  CbmMCInput(TChain* chain, ECbmTreeAccess mode = ECbmTreeAccess::kRegular);
 
 
  virtual ~CbmMCInput();
 
 
  std::set<TString>& GetBranchList();
 
 
  TChain* GetChain() const { return fChain; }
 
 
  Int_t GetNextEntry();
 
 
  Int_t GetMaxNofEvents() const {
    return (fMode == ECbmTreeAccess::kRegular ? fChain->GetEntries() : -1);
  }
 
 
  ECbmTreeAccess GetMode() const { return fMode; }
 
 
  Long64_t GetNofEntries() const { return fChain->GetEntries(); }
 
 
  UInt_t GetNofUsedEntries() const { return fNofUsedEntries; }
 
 
private:
  TChain* fChain;               
  ECbmTreeAccess fMode;         // Access mode to tree
  std::set<TString> fBranches;  // List of branch names
  UInt_t fLastUsedEntry;        // Index of last used entry
  UInt_t fNofUsedEntries;       // Number of used entries
 
 
  UInt_t ReadBranches();
 
 
  ClassDef(CbmMCInput, 1);
};
 
#endif /* CBMMCINPUT_H */