SimSource.cxx

Go to the documentation of this file.

/* Copyright (C) 2025 Jagiellonian University, Krakow
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Bartosz Sobol [committer] */
 
#include "SimSource.h"
 
#include <fmt/format.h>
 
SimSource::SimSource(const std::string& locator)
{
  fFile = std::unique_ptr<TFile>(TFile::Open(locator.c_str()));
  if (fFile == nullptr || fFile->IsZombie()) {
    throw std::runtime_error(fmt::format("[SimSource] Error opening file: {}", locator));
  }
 
  fTree = std::unique_ptr<TTree>((TTree*) fFile->Get("cbmsim"));
  if (fTree == nullptr) {
    throw std::runtime_error(fmt::format("[SimSource] Error getting cbmsim tree: {}", locator));
  }
 
  fEntries = fTree->GetEntriesFast();
 
  fTree->SetBranchAddress(CbmBmonDigi::GetBranchName(), &fBmonDigis);
  fTree->SetBranchAddress(CbmStsDigi::GetBranchName(), &fStsDigis);
  fTree->SetBranchAddress(CbmFsdDigi::GetBranchName(), &fFsdDigis);
  fTree->SetBranchAddress(CbmMvdDigi::GetBranchName(), &fMvdDigis);
  fTree->SetBranchAddress(CbmTrdDigi::GetBranchName(), &fTrdDigis);
  fTree->SetBranchAddress(CbmTofDigi::GetBranchName(), &fTofDigis);
  fTree->SetBranchAddress(CbmRichDigi::GetBranchName(), &fRichDigis);
}
 
SimSource::~SimSource()
{
  if (fBmonDigis) delete fBmonDigis;
  if (fStsDigis) delete fStsDigis;
  if (fFsdDigis) delete fFsdDigis;
  if (fMvdDigis) delete fMvdDigis;
  if (fTrdDigis) delete fTrdDigis;
  if (fTofDigis) delete fTofDigis;
  if (fRichDigis) delete fRichDigis;
}
 
bool SimSource::eos() const { return fCurrent >= fEntries; }
 
fles::StorableTimeslice* SimSource::do_get()
{
  if (eos()) {
    return nullptr;
  }
 
  fTsBuilder.init(1, fCurrent);
  fTree->GetEntry(fCurrent);
 
  fMvdRawDigis.clear();
  fMvdRawDigis.reserve(fMvdDigis->size());
  std::transform(fMvdDigis->cbegin(), fMvdDigis->cend(), std::back_inserter(fMvdRawDigis),
                 CbmMvdDigi::ConvertToRawDigi);
 
  fTsBuilder.addDigisToTs(*fBmonDigis, fles::Subsystem::BMON);
  fTsBuilder.addDigisToTs(*fStsDigis, fles::Subsystem::STS);
  fTsBuilder.addDigisToTs(*fFsdDigis, fles::Subsystem::FSD);
  fTsBuilder.addDigisToTs(fMvdRawDigis, fles::Subsystem::MVD);
  fTsBuilder.addDigisToTs(*fTrdDigis, fles::Subsystem::TRD);
  fTsBuilder.addDigisToTs(*fTofDigis, fles::Subsystem::TOF);
  fTsBuilder.addDigisToTs(*fRichDigis, fles::Subsystem::RICH);
 
  ++fCurrent;
  return fTsBuilder.getRelease();
}