HalCbmSource.h

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] */
#ifndef CBMROOT_2023_ANALYSIS_PWGC2F_FEMTOSCOPY_HAL_HELPERS_HALCBMSOURCE_H_
#define CBMROOT_2023_ANALYSIS_PWGC2F_FEMTOSCOPY_HAL_HELPERS_HALCBMSOURCE_H_
 
#include <RtypesCore.h>
#include <TChain.h>
#include <TFile.h>
#include <TString.h>
 
#include <vector>
 
#include <Hal/IOManager.h>
#include <Hal/RootIOManager.h>
#include <Hal/Source.h>

class CbmAnaTreeContainers;
 
class HalCbmATIOManager : public Hal::RootIOManager {
 protected:
  CbmAnaTreeContainers* fContainer = {nullptr};
  Bool_t fAtBranchesSet            = {kFALSE};
 
 public:
  HalCbmATIOManager(TString name = "") : Hal::RootIOManager(name){};
  HalCbmATIOManager(std::initializer_list<TString> lists);
  virtual void UpdateBranches();
  virtual ~HalCbmATIOManager();
  ClassDef(HalCbmATIOManager, 1)
};
 
class HalCbmSource : public Hal::Source {
 protected:
  HalCbmATIOManager* fManager;
 
 public:
  HalCbmSource(TString filename = "");
  HalCbmSource(std::initializer_list<TString> lists);
  Bool_t Init();
  virtual void AddFile(TString fileName = "");
  virtual void AddFriend(TString friendName = "", Int_t level = 0);
  Hal::IOManager* GetIOManager() const { return fManager; };
  virtual ~HalCbmSource();
  ClassDef(HalCbmSource, 1)
};
 
#endif /* CBMROOT_2023_ANALYSIS_PWGC2F_FEMTOSCOPY_HAL_HELPERS_HALCBMSOURCE_H_ */