CbmEventStore.h

Go to the documentation of this file.

/* Copyright (C) 2020 GSI Helmholtzzentrum fuer Schwerionenforschung, Darmstadt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Volker Friese [committer], Florian Uhlig */
 
#ifndef CBMEVENTSTORE_H
#define CBMEVENTSTORE_H 1
 
#include "CbmDefs.h"           // for kNofSystems
#include "CbmDigiContainer.h"  // for CbmDigiContainer
#include "CbmDigiVector.h"     // for CbmDigiVector
 
#include <Logger.h>  // for LOG
 
#include <Rtypes.h>      // for THashConsistencyHolder, ClassDef
#include <TObjArray.h>   // for TObjArray
#include <TObject.h>     // for TObject
 
#include <boost/any.hpp>  // for any_cast
 
#include <cassert>  // for assert
#include <cstdint>
#include <string>  // for string
 
class CbmMatch;
 
class CbmEventStore : public TObject {
 
public:
  CbmEventStore(uint32_t eventId = 0, bool hasMatches = false);
 
 
  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>(false);
    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>(true);
    auto digis = static_cast<CbmDigiContainer*>(fDigis.at(system));
    assert(digis);
    digis->AddDigi(digi, match);
  }
 
 
  template<class Digi>
  const Digi* GetDigi(uint32_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));
  }
 
 
  uint32_t GetEventId() const { return fEventId; }
 
 
  uint32_t GetNofDigis(ECbmModuleId system) const;
 
 
  bool HasMatches() const { return fHasMatches; }
 
 
  bool IsEmpty() const;
 
 
  void MatchToMC(CbmMatch& result) const;
 
 
  CbmEventStore& operator=(const CbmEventStore&) = delete;
 
 
  std::string ToString() const;
 
 
private:
  uint32_t fEventId = -1;     
  bool fHasMatches  = false;  
  //            TObjArray* fDigis = nullptr;   ///< Array of CbmDigiVector
  std::map<ECbmModuleId, TObject*> fDigis;  
 
  ClassDef(CbmEventStore, 2);
};
 
#endif /* CBMEVENTSTORE_H */