RootIOManager.h

/*
 * RootIoManager.h
 *
 *  Created on: 6 maj 2022
 *      Author: Daniel Wielanek
 *      E-mail: daniel.wielanek@gmail.com
 *      Warsaw University of Technology, Faculty of Physics
 */
#ifndef HAL_FEATURES_IO_ROOTIOMANAGER_H_
#define HAL_FEATURES_IO_ROOTIOMANAGER_H_
 
#include "IOManager.h"
#include <vector>
 
#include <RtypesCore.h>
#include <TList.h>
#include <TString.h>
 
class TTree;
class TBranch;
namespace Hal {
 
  class RootIOManager : public IOManager {
    Int_t fEntries = {0};
    TString fOutFileName;
    TString fOutTreeName;
    TFile* fOutFile  = {nullptr};
    TTree* fOutTree  = {nullptr};
    TChain* fInChain = {nullptr};
    std::vector<TObject**> fObjects;  // just to call delete?
 
  protected:
    void PushTObject(TObject** obj);
    virtual void RegisterInternal(const char* name, const char* folderName, TNamed* obj, Bool_t toFile);
    virtual void RegisterInternal(const char* name, const char* Foldername, TCollection* obj, Bool_t toFile);
    TChain* GetInChain() { return fInChain; }
    virtual Bool_t InitInternal();
 
  public:
    RootIOManager(TString name);
    RootIOManager(InputDataInfo* info);
    void SetOutput(TString name) { fOutFileName = name; }
    void SetOutTreeName(TString name) { fOutTreeName = name; }
    Int_t GetEntries() const;
    Int_t GetEntry(Int_t i, Int_t flag = 1);
    virtual void UpdateBranches();
    void LockUnusedBranches();
    void SetInChain(TChain* tempChain, Int_t ident = -1);
    void FillTree();
    static std::vector<TString> GetListOfBranches(TChain* chain, Bool_t friends);
    virtual void CloseManager();
    virtual ~RootIOManager();
    ClassDef(RootIOManager, 1)
  };
 
} /* namespace Hal */
 
#endif /* HAL_FEATURES_IO_ROOTIOMANAGER_H_ */