IOManager.h

/*
 * IOManager.h
 *
 *  Created on: 10 maj 2022
 *      Author: Daniel Wielanek
 *      E-mail: daniel.wielanek@gmail.com
 *      Warsaw University of Technology, Faculty of Physics
 */
#ifndef HAL_FEATURES_IO_STEER_IOMANAGER_H_
#define HAL_FEATURES_IO_STEER_IOMANAGER_H_
 
#include <TList.h>
#include <TObject.h>
#include <TString.h>
 
class TList;
class TChain;
class TFile;
namespace Hal {
  class InputDataInfo;
}
namespace Hal {
  class Package;
  class MagField;
  class BranchInfo : public TObject {
  public:
    enum class EFlag { kInActive, kInPassive, kOut, kVirtual, kNull };
 
  private:
    TString fBrName   = {""};
    TObject* fPointer = {nullptr};  
    EFlag fFlag       = {EFlag::kNull};
 
  public:
    BranchInfo(TString name = "", TObject* pointer = nullptr, EFlag used = EFlag::kNull) :
      fBrName(name), fPointer(pointer), fFlag(used) {}
    BranchInfo(const BranchInfo& other) : TObject(other), fBrName(other.fBrName), fPointer(other.fPointer), fFlag(other.fFlag) {}
    EFlag GetFlag() const { return fFlag; }
    TString GetBranchName() const { return fBrName; }
    void SetFlag(EFlag Flag = EFlag::kNull) { fFlag = Flag; }
    void SetBranchName(const TString name) { fBrName = name; }
    TObject* GetPointer() const { return fPointer; }
    void SetPointer(TObject* pointer = nullptr) { fPointer = pointer; }
    virtual ~BranchInfo() {};
    ClassDef(BranchInfo, 1)
  };
 
  class IOManager : public TObject {
  public:
  private:
    MagField* fField = {nullptr};
    std::vector<TString> fBranchNameList;
 
  protected:
    InputDataInfo* fDataInfo = {nullptr};
    std::vector<Hal::BranchInfo> fBranches;
    void AddBranch(TString name, TObject* object, BranchInfo::EFlag flag);
    BranchInfo FindBranch(TString name) const;
    virtual void RegisterInternal(const char* name, const char* folderName, TNamed* obj, Bool_t toFile) = 0;
    virtual void RegisterInternal(const char* name, const char* Foldername, TCollection* obj, Bool_t toFile) = 0;
    virtual Bool_t InitInternal()                                                                            = 0;
 
  public:
    IOManager() : fField(nullptr), fBranchNameList() {};
    IOManager(InputDataInfo* info);
    virtual Int_t GetEntries() const = 0;
    virtual Int_t GetEntry(Int_t i, Int_t flag) = 0;
    Bool_t Init();
    virtual void SetField(MagField* field) { fField = field; };
    virtual void SetOutput(TString /*name*/) {};
    virtual MagField* GetField() const { return fField; };
    void Register(const char* name, const char* folderName, TNamed* obj, Bool_t toFile);
    void Register(const char* name, const char* Foldername, TCollection* obj, Bool_t toFile);
    virtual void SetInChain(TChain* tempChain, Int_t ident = -1) = 0;
    virtual void UpdateBranches() {};
    BranchInfo::EFlag GetBranchStatus(const char* BrName);
    virtual TObject* GetObject(const char* BrName);
    void ActivateBranch(TString brName);
    TString GetSourceName() const;
    std::vector<TString> GetBranchNameList() {
      UpdateBranches();
      return fBranchNameList;
    };
    void PrintInfo();
    virtual void FillTree() = 0;
    virtual void CloseManager() {};
    virtual TList* GetBranchesList() const;
    Int_t GetNFiles() const;
    std::vector<TString> GetFilesNames(Int_t entry = 0) const;
    Int_t GetFriendsLevel() const;
    void AddFile(TString name);
    void AddFriend(TString friendName, Int_t level);
    virtual void LockUnusedBranches() {};
    TString GetFirstDataFileName() const;
    TString GetFirstFriendFileName(Int_t level) const;
    std::vector<TString> GetFileNameList(Int_t level) const;
    virtual ~IOManager();
    ClassDef(IOManager, 1)
  };
 
} /* namespace Hal */
 
#endif /* HAL_FEATURES_IO_STEER_IOMANAGER_H_ */