CbmMCInputSet.h

Go to the documentation of this file.

 
#ifndef CBMMCINPUTSET_H
#define CBMMCINPUTSET_H 1
 
#include "CbmDefs.h"
#include "CbmMCInput.h"
#include "TF1.h"
#include "TObject.h"
#include <map>
#include <set>
#include <utility>
 
 
class CbmMCInputSet : public TObject {
 
public:
  CbmMCInputSet();
 
 
  CbmMCInputSet(Double_t rate);
 
 
  virtual ~CbmMCInputSet();
 
 
  virtual Bool_t ActivateObject(TObject** object, const char* branchName);
 
 
  void AddInput(UInt_t inputId,
                TChain* chain,
                ECbmTreeAccess mode = ECbmTreeAccess::kRegular);
 
 
  const std::set<TString>& GetBranchList() const { return fBranches; }
 
 
  Double_t GetDeltaT();
 
 
  std::pair<UInt_t, CbmMCInput*> GetFirstInput() {
    return (fInputs.empty() ? std::make_pair(-1, nullptr)
                            : std::make_pair(fInputs.begin()->first,
                                             fInputs.begin()->second));
  }
 
 
  CbmMCInput* GetInput(UInt_t id) {
    return (fInputs.find(id) == fInputs.end() ? nullptr : fInputs[id]);
  }
 
 
  Int_t GetMaxNofEvents() const;
 
 
  std::tuple<Bool_t, UInt_t, Int_t> GetNextEntry();
 
 
  Double_t GetRate() const { return fRate; }
 
 
  void RegisterChains();
 
 
private:
  Double_t fRate;                                        // Event rate [1/s]
  std::map<UInt_t, CbmMCInput*> fInputs;                 // Key is input ID
  std::map<UInt_t, CbmMCInput*>::iterator fInputHandle;  // Handle for inputs
  std::set<TString> fBranches;                           // List of branch names
  TF1* fDeltaDist;  // Probability distribution for delta(t)
 
 
  Bool_t CheckBranchList(CbmMCInput* input);
 
 
  ClassDef(CbmMCInputSet, 1);
};
 
#endif /* CBMMCINPUT_H */