/** CbmEventStore.cxx
 ** @author V.Friese <v.friese@gsi.de>
 ** @since 11.03.2020
 ** @version 1.0
 **/
#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)