CbmMCInputSet.h

Go to the documentation of this file.

/* Copyright (C) 2018-2020 GSI Helmholtzzentrum fuer Schwerionenforschung, Darmstadt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Volker Friese [committer] */
 
#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>
 
#include "Defs.h"
 
 
class CbmMCInputSet : public TObject {
 
public:
  CbmMCInputSet();
 
 
  CbmMCInputSet(cbm::sim::TimeDist dist, 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 GetEventRate() const { return fEventRate; }
 
 
  void RegisterChains();
 
 
private:
  cbm::sim::TimeDist fTimeDist = cbm::sim::TimeDist::Poisson;  // Event time distribution model
  Double_t fEventRate;                                         // 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 */