InputDataInfo.h

/*
 * ChainBuilder.h
 *
 *  Created on: 19 cze 2024
 *      Author: daniel
 */
 
#ifndef HAL_FEATURES_IO_INPUTDATAINFO_H_
#define HAL_FEATURES_IO_INPUTDATAINFO_H_
 
#include <TObject.h>
#include <TString.h>
 
#include <vector>
 
class TChain;
namespace Hal {
 
  class InputDataInfo : public TObject {
  protected:
    std::vector<std::vector<TString>> fFileNames;
    TString fListName;
 
  public:
    InputDataInfo(TString file = "data.root");
    InputDataInfo(std::initializer_list<TString> fileLists);
    InputDataInfo(std::vector<std::vector<TString>> files);
    TString GetSafeFile(Int_t level, Int_t entry) const;
    std::vector<TString> GetSafeFiles(Int_t level) const;
    void AddFile(TString file);
    void AddFriend(TString file, Int_t level);
    void OverwriteSourceName(TString name) { fListName = name; }
    virtual void Print(Option_t* option = "") const;
    Int_t GetNFiles() const;
    Int_t GetFriendsLevel() const;
    TString GetSourceName() const { return fListName; }
    std::vector<std::vector<TString>> GetListOfFiles() const { return fFileNames; }
    virtual ~InputDataInfo() {};
    ClassDef(InputDataInfo, 1);
  };
 
  class InputRootDataInfo : public InputDataInfo {
    TString GetChainName(TString file) const;
    void GuessTrees();
    std::vector<TString> fTreeNames;
    TChain* fChain = {nullptr};  
 
  public:
    InputRootDataInfo(TString file = "data.root", TString treename = "");
    InputRootDataInfo(std::initializer_list<TString> fileLists, std::initializer_list<TString> treeNames = {});
    InputRootDataInfo(std::vector<std::vector<TString>> files);
    TChain* GetChain();
    virtual void Print(Option_t* option = "") const;
    ClassDef(InputRootDataInfo, 1)
  };
 
 
} /* namespace Hal */
 
#endif /* HAL_FEATURES_IO_INPUTDATAINFO_H_ */