CbmMCDataObject.cxx

/* Copyright (C) 2017-2020 GSI Helmholtzzentrum fuer Schwerionenforschung, Darmstadt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Mikhail Prokudin, Volker Friese [committer], Florian Uhlig */

/** @file  CbmMCDataObject.cxx
 ** @author //Dr.Sys <mikhail.prokudin@cern.ch>
 **/
#include "CbmMCDataObject.h"

#include <FairRootManager.h>  // for FairRootManager
#include <Logger.h>           // for Logger, LOG

#include <TChain.h>   // for TChain
#include <TObject.h>  // for TObject

#include <utility>  // for pair

using namespace std;

// --- Standard constructor
CbmMCDataObject::CbmMCDataObject(const char* branchname, const std::vector<std::list<TString>>& fileList)
  : fLegacy(0)
  , fLegacyObject(nullptr)
  , fBranchName(branchname)
  , fSize(-1111)
  , fChains()
  , fTArr()
  , fN()
  , fArrays()
{
  list<TString>::const_iterator p;
  Int_t i;
  Int_t s = fileList.size();

  fSize = s;
  fChains.resize(s);
  fTArr.resize(s);
  fN.resize(s);

  for (i = 0; i < s; i++) {
    fN[i]      = 0;
    fTArr[i]   = nullptr;
    fChains[i] = nullptr;
    if (fileList[i].size() == 0) continue;
    fChains[i] = new TChain("cbmsim");
    for (p = fileList[i].begin(); p != fileList[i].end(); ++p)
      fChains[i]->AddFile(*p);
    fChains[i]->SetBranchAddress(branchname, &(fTArr[i]));
    fN[i] = fChains[i]->GetEntries();
  }

  fArrays.resize(s);
  for (i = 0; i < s; i++)
    fArrays[i].clear();
}

// --- Make TChain number chainNum2 friend of TChain number chainNum2
void CbmMCDataObject::AddFriend(Int_t chainNum1, Int_t chainNum2)
{
  if (fLegacy == 1) {
    LOG(error) << "AddFriend method should not be called in legacy mode";
    return;
  }
  if (chainNum1 < 0 || chainNum1 >= static_cast<Int_t>(fChains.size()) || fChains[chainNum1] == nullptr) {
    LOG(error) << "chainNum1=" << chainNum1 << " is not a correct chain number.";
    return;
  }
  if (chainNum2 < 0 || chainNum2 >= static_cast<Int_t>(fChains.size()) || fChains[chainNum2] == nullptr) {
    LOG(error) << "chainNum2=" << chainNum2 << " is not a correct chain number.";
    return;
  }
  fChains[chainNum1]->AddFriend(fChains[chainNum2]);
}

// --- Legacy constructor
CbmMCDataObject::CbmMCDataObject(const char* branchname)
  : fLegacy(1)
  , fLegacyObject(nullptr)
  , fBranchName(branchname)
  , fSize(-1111)
  , fChains()
  , fTArr()
  , fN()
  , fArrays()
{
  FairRootManager* fManager = FairRootManager::Instance();
  if (!fManager) {
    LOG(fatal) << "CbmMCDataObject():	Can't find a Root Manager.";
    return;
  }
  fLegacyObject = (TObject*) fManager->GetObject(branchname);
  if (!fLegacyObject) {
    LOG(fatal) << "CbmMCDataObject(): Can't find " << fBranchName << " in the system.";
    return;
  }
}

// --- Legacy Get
TObject* CbmMCDataObject::LegacyGet(Int_t fileNumber, Int_t eventNumber)
{
  if (fileNumber >= 0 || eventNumber >= 0)
    LOG(debug1) << "LegacyGet:	Trying to get object with fileNum=" << fileNumber << ", entryNum=" << eventNumber
                << " in legacy mode.";

  return fLegacyObject;
}


// --- Get an object
TObject* CbmMCDataObject::Get(Int_t fileNumber, Int_t eventNumber)
{
  if (fLegacy == 1) return LegacyGet(fileNumber, eventNumber);
  if (fileNumber < 0 || fileNumber >= fSize) return nullptr;
  if (eventNumber < 0 || eventNumber >= fN[fileNumber]) return nullptr;

  // --- Cached objects
  map<Int_t, TObject*>& arr = fArrays[fileNumber];

  // --- If the object for this event is already in the cache, use it.
  if (arr.find(eventNumber) != arr.end()) return arr[eventNumber];

  // --- If not, copy the object from the chain into the cache
  TChain* ch = fChains[fileNumber];
  ch->GetEntry(eventNumber);
  //  arr[eventNumber]=(TObject*)(fTArr[fileNumber]->Clone());
  arr[eventNumber] = fTArr[fileNumber]->Clone();

  return arr[eventNumber];
}

// --- At end of one event: clear the cache to free the memory
void CbmMCDataObject::FinishEvent()
{
  if (fLegacy == 1) return;

  Int_t i;
  map<Int_t, TObject*>::const_iterator p;

  for (i = 0; i < fSize; i++) {
    for (p = fArrays[i].begin(); p != fArrays[i].end(); ++p)
      delete (p->second);
    fArrays[i].clear();
  }
}


// --- Clean up
void CbmMCDataObject::Done()
{
  if (fLegacy == 1) return;
  Int_t i;

  FinishEvent();
  for (i = 0; i < fSize; i++)
    delete fChains[i];
}

ClassImp(CbmMCDataObject)