CbmEventStore.h

Go to the documentation of this file.

 
#ifndef CBMEVENTSTORE_H
#define CBMEVENTSTORE_H 1
 
#include <Rtypes.h>      // for THashConsistencyHolder, ClassDef
#include <RtypesCore.h>  // for UInt_t, Bool_t, kFALSE, kTRUE
#include <TObjArray.h>   // for TObjArray
#include <TObject.h>     // for TObject
 
#include <FairLogger.h>  // for LOG
 
#include <boost/any.hpp>  // for any_cast
 
#include <assert.h>  // for assert
#include <string>    // for string
 
#include "CbmDefs.h"           // for kNofSystems
#include "CbmDigiContainer.h"  // for CbmDigiContainer
#include "CbmDigiVector.h"     // for CbmDigiVector
 
class CbmMatch;
 
class CbmEventStore : public TObject {
 
public:
  CbmEventStore(UInt_t eventId = 0, Bool_t hasMatches = kFALSE);
 
 
  CbmEventStore(const CbmEventStore&);
 
 
  CbmEventStore(CbmEventStore&&) = default;
 
 
  virtual ~CbmEventStore();
 
 
  template<class Digi>
  void AddDigi(const Digi* digi) {
    if (fHasMatches) {
      LOG(fatal) << "CbmEventStore: Trying to add digi without match!";
      return;
    }
    assert(digi);
    ECbmModuleId system = Digi::GetSystem();
    assert(system < ECbmModuleId::kNofSystems);
    if (!fDigis[system]) fDigis[system] = new CbmDigiVector<Digi>(kFALSE);
    auto digis = static_cast<CbmDigiContainer*>(fDigis.at(system));
    assert(digis);
    digis->AddDigi(digi, nullptr);
  }
 
 
  template<class Digi>
  void AddDigi(const Digi* digi, const CbmMatch* match) {
    if (!fHasMatches) {
      LOG(fatal) << "CbmEventStore: Trying to add digi without match!";
      return;
    }
    assert(digi);
    assert(match);
    ECbmModuleId system = Digi::GetSystem();
    assert(system < ECbmModuleId::kNofSystems);
    if (!fDigis[system]) fDigis[system] = new CbmDigiVector<Digi>(kTRUE);
    auto digis = static_cast<CbmDigiContainer*>(fDigis.at(system));
    assert(digis);
    digis->AddDigi(digi, match);
  }
 
 
  template<class Digi>
  const Digi* GetDigi(UInt_t index) const {
    ECbmModuleId system = Digi::GetSystem();
    assert(system < ECbmModuleId::kNofSystems);
    auto digis = static_cast<CbmDigiContainer*>(fDigis.at(system));
    assert(digis);
    return boost::any_cast<const Digi*>(digis->GetDigi(index));
  }
 
 
  UInt_t GetEventId() const { return fEventId; }
 
 
  UInt_t GetNofDigis(ECbmModuleId system) const;
 
 
  Bool_t HasMatches() const { return fHasMatches; }
 
 
  Bool_t IsEmpty() const;
 
 
  void MatchToMC(CbmMatch& result) const;
 
 
  CbmEventStore& operator=(const CbmEventStore&) = delete;
 
 
  std::string ToString() const;
 
 
private:
  UInt_t fEventId    = -1;      
  Bool_t fHasMatches = kFALSE;  
  //            TObjArray* fDigis = nullptr;   ///< Array of CbmDigiVector
  std::map<ECbmModuleId, TObject*> fDigis;  
 
  ClassDef(CbmEventStore, 2);
};
 
#endif /* CBMEVENTSTORE_H */