CbmEventStore.cxx

Go to the documentation of this file.

 
#include "CbmEventStore.h"
 
#include "CbmLink.h"        // for CbmLink
#include "CbmMatch.h"       // for CbmMatch
#include "CbmModuleList.h"  // for CbmModuleList
 
#include <TString.h>  // for operator<<
 
#include <iostream>  // for operator<<, basic_ostream, stringstream
 
 
// -----   Constructor   ---------------------------------------------------
CbmEventStore::CbmEventStore(UInt_t eventId, Bool_t hasMatches)
  : fEventId(eventId), fHasMatches(hasMatches) {
  //  fDigis = new TObjArray(ToIntegralType(ECbmModuleId::kNofSystems));
}
// -------------------------------------------------------------------------
 
 
// -----   Copy constructor   ----------------------------------------------
CbmEventStore::CbmEventStore(const CbmEventStore& other) : TObject(other) {
  fEventId    = other.fEventId;
  fHasMatches = other.fHasMatches;
  // TODO: Create new map with a copy of the original map
  // Don't know how to do that yet
  //  fDigis = (TObjArray*)other.fDigis->Clone();
}
// -------------------------------------------------------------------------
 
 
// -----   Destructor   ----------------------------------------------------
CbmEventStore::~CbmEventStore() {
  for (auto system : fDigis) {
    delete system.second;
  }
  //  fDigis->Delete();
}
// -------------------------------------------------------------------------
 
 
// -----   Test for being empty   ------------------------------------------
Bool_t CbmEventStore::IsEmpty() const {
  UInt_t nDigis = 0;
  for (auto system : fDigis) {
    auto* digis = dynamic_cast<CbmDigiContainer*>(system.second);
    nDigis += digis->GetNofDigis();
  }
 
  /*
for ( Int_t system = 0; system < fDigis->GetEntriesFast(); system++) {
    auto* digis = dynamic_cast<CbmDigiContainer*>(fDigis->At(system));
    if ( digis ) nDigis += digis->GetNofDigis();
  }
*/
  return (nDigis > 0 ? kFALSE : kTRUE);
}
// -------------------------------------------------------------------------
 
 
// -----   Get number of data for a given system   -------------------------
UInt_t CbmEventStore::GetNofDigis(ECbmModuleId system) const {
  if (system >= ECbmModuleId::kNofSystems) return 0;
  auto* digis = dynamic_cast<CbmDigiContainer*>(fDigis.at(system));
  if (!digis) return 0;
  return digis->GetNofDigis();
}
// -------------------------------------------------------------------------
 
 
// -----   Match to MC event   ---------------------------------------------
void CbmEventStore::MatchToMC(CbmMatch& result) const {
  result.ClearLinks();
  if (!fHasMatches) return;
  for (auto system : fDigis) {
    auto* digis = dynamic_cast<CbmDigiContainer*>(system.second);
    if (!digis) continue;
    for (UInt_t index = 0; index < digis->GetNofDigis(); index++) {
      const CbmMatch* match = digis->GetDigiMatch(index);
      assert(match);
      for (Int_t iLink = 0; iLink < match->GetNofLinks(); iLink++) {
        const CbmLink& link = match->GetLink(iLink);
        result.AddLink(link.GetWeight(), 0, link.GetEntry(), link.GetFile());
      }  //# Links in match
    }    //# Matches in system
  }      //# Systems
 
 
  /*
  for (Int_t system = 0; system < fDigis->GetEntriesFast(); system++) {
    auto* digis = dynamic_cast<CbmDigiContainer*>(fDigis->At(system));
    if ( ! digis ) continue;
    for (UInt_t index = 0; index < digis->GetNofDigis(); index++) {
      const CbmMatch* match = digis->GetDigiMatch(index);
      assert(match);
      for (Int_t iLink = 0; iLink < match->GetNofLinks(); iLink++) {
        const CbmLink& link = match->GetLink(iLink);
        result.AddLink(link.GetWeight(), 0, link.GetEntry(), link.GetFile());
      } //# Links in match
    } //# Matches in system
  } //# Systems
*/
}
// -------------------------------------------------------------------------
 
 
// -----   String output   -------------------------------------------------
std::string CbmEventStore::ToString() const {
  std::stringstream ss;
  ss << "Event " << fEventId;
  if (IsEmpty())
    ss << " empty";
  else {
    ss << " Data: ";
    for (auto system : fDigis) {
      auto* vec = dynamic_cast<CbmDigiContainer*>(system.second);
      //     for ( Int_t system = 0; system < fDigis->GetEntriesFast(); system++) {
      //       if ( fDigis->At(system) ) {
      //         auto vec = static_cast<CbmDigiContainer*>(fDigis->At(system));
      assert(vec);
      ss << CbmModuleList::GetModuleNameCaps(system.first) << " "
         << vec->GetNofDigis() << " ";
      //       } //? Digi vector present
    }  //# Systems
  }    //? Not empty
 
  if (fHasMatches) ss << ", matches present";
  return ss.str();
}
// -------------------------------------------------------------------------
 
 
ClassImp(CbmEventStore)