HalCbmSource.cxx

Go to the documentation of this file.

/* Copyright (C) 2023-2023 Warsaw University of Technology, Warsaw
    SPDX-License-Identifier: GPL-3.0-only
    Authors: Daniel Wielanek [committer] */
#include "HalCbmSource.h"
 
#include "CbmAnaTreeContainer.h"
#include "HalCbmDetectorID.h"
 
#include <TBranch.h>
#include <TNamed.h>
#include <TObjArray.h>
 
#include <vector>
 
#include <Hal/Cout.h>
#include <Hal/InputDataInfo.h>
#include <Hal/RootIOManager.h>
 
HalCbmSource::HalCbmSource(TString filename) : Source(filename) { fManager = new HalCbmATIOManager(filename); }
 
void HalCbmSource::AddFile(TString fileName) { fManager->AddFile(fileName); }
 
void HalCbmSource::AddFriend(TString friendName, Int_t level) { fManager->AddFriend(friendName, level); }
 
HalCbmSource::~HalCbmSource() { delete fManager; }
 
void HalCbmATIOManager::UpdateBranches()
{
  if (!fAtBranchesSet) {
    fContainer   = new CbmAnaTreeContainers();
    fContainer->LoadConf(fDataInfo);
 
    auto names = CbmAnaTreeContainers::GetNames();
    for (int i = 0; i < fContainer->GetSize(); i++) {
      bool con = fContainer->fContainers[i]->ConnectToTree(GetInChain());
      if (con) {
        TString name = HalCbm::GetContainerName(names[i]);
        AddBranch(name, fContainer->fContainers[i], Hal::BranchInfo::EFlag::kInActive);
        TObject** obj = new TObject*();
        obj[0]        = fContainer->fContainers[i];
        PushTObject(obj);
      }
    }
    fAtBranchesSet = kTRUE;
  }
  Hal::RootIOManager::UpdateBranches();
}
 
HalCbmATIOManager::HalCbmATIOManager(std::initializer_list<TString> lists)
  : Hal::RootIOManager(new Hal::InputRootDataInfo(lists))
{
}
 
HalCbmATIOManager::~HalCbmATIOManager()
{
  if (fContainer) delete fContainer;
}
 
Bool_t HalCbmSource::Init() { return kTRUE; }
 
HalCbmSource::HalCbmSource(std::initializer_list<TString> lists) { fManager = new HalCbmATIOManager(lists); }