BmnFileSource.h

Go to the documentation of this file.

// -------------------------------------------------------------------------
// -----                       BmnFileSource header file               -----
// -------------------------------------------------------------------------
#ifndef BMNFILESOURCE_H
#define BMNFILESOURCE_H
 
#include "FairEventHeader.h"
#include "FairFileHeader.h"
#include "FairFileSourceBase.h"
#include "FairMCEventHeader.h"
 
#include <TArrayI.h>
#include <TChain.h>
#include <TF1.h>
#include <TFile.h>
#include <TFolder.h>
#include <TString.h>
#include <TTree.h>
#include <list>   // for _List_iterator, list, etc
#include <map>    // multimap
#include <memory>
 
class BmnFileSource : public FairFileSourceBase
{
  public:
    BmnFileSource(TFile* f, const char* Title = "InputRootFile", UInt_t identifier = 0);
    BmnFileSource(const TString* RootFileName, const char* Title = "InputRootFile", UInt_t identifier = 0);
    BmnFileSource(const TString RootFileName, const char* Title = "InputRootFile", UInt_t identifier = 0);
    BmnFileSource(const TString RootFileName, int& period_number, int& run_number);
    // BmnFileSource(const BmnFileSource& file);
    ~BmnFileSource() override;
 
    Bool_t Init() override;
    Int_t ReadEvent(UInt_t i = 0) override;
    void Close() override;
 
    Int_t CheckMaxEventNo(Int_t EvtEnd = 0) override;
    void ReadBranchEvent(const char* BrName) override;
    void ReadBranchEvent(const char* BrName, Int_t Entry) override;
    void FillEventHeader(FairEventHeader* feh) override;
 
    const TFile* GetRootFile() { return fRootFile; }
    void AddFriend(TString FileName);
    void AddFile(TString FileName);
    void AddFriendsToChain();
    void PrintFriendList();
    void CheckFriendChains();
    void CreateNewFriendChain(TString inputFile, TString inputLevel);
    TTree* GetInTree() { return fInChain->GetTree(); }
    TChain* GetInChain() { return fInChain; }
    TFile* GetInFile() { return fRootFile; }
    void CloseInFile()
    {
        if (fRootFile) {
            fRootFile->Close();
        }
    }
    void SetInTree(TTree* tempTree);
    TObjArray* GetListOfFoldersBmn() { return &fListFolderBmn; }
    TFolder* GetBranchDescriptionFolder() { return fCbmroot; }
    UInt_t GetEntries() { return fNoOfEntries; }
 
    //    TList*              GetBranchNameList() {return fBranchNameList;}
 
    void SetInputFile(TString name);
 
    void SetBeamTime(Double_t beamTime, Double_t gapTime);
    void SetEventTimeInterval(Double_t min, Double_t max);
    void SetEventMeanTime(Double_t mean);
    void SetEventTime();
    Double_t GetDeltaEventTime();
    void SetFileHeader(FairFileHeader* f) { fFileHeader = f; }
    Double_t GetEventTime();
 
    Bool_t ActivateObject(TObject** obj, const char* BrName) override;
    Bool_t ActivateObjectAny(void**, const std::type_info&, const char*) override;
 
    void SetEvtHeaderNew(Bool_t Status) { fEvtHeaderIsNew = Status; }
    Bool_t IsEvtHeaderNew() { return fEvtHeaderIsNew; }
 
    void SetCheckFileLayout(Bool_t enable) { fCheckFileLayout = enable; }
 
    Bool_t SpecifyRunId() override;
 
    void GetRunNumber(int& period_number, int& run_number);
    Int_t GetFileType() { return fFileType; }
 
  private:
    TString fInputTitle;
    TFile* fRootFile;
    std::list<TString> fFriendFileList;                                 
    std::list<TString> fInputChainList;                                 
    std::map<TString, TChain*> fFriendTypeList;                         
    std::list<TString> fInputLevel;                                     
    std::map<TString, std::multimap<TString, TArrayI>> fRunIdInfoAll;   
    TChain* fInChain;
    TTree* fInTree;
    TObjArray fListFolderBmn{16};   
    TFolder* fCbmout;
    TFolder* fCbmroot;
    /***/
    UInt_t fSourceIdentifier;
    UInt_t fNoOfEntries;
    Bool_t IsInitialized;
 
    BmnFileSource(const BmnFileSource&);
    BmnFileSource operator=(const BmnFileSource&);
 
    FairMCEventHeader* fMCHeader;   
 
    FairEventHeader* fEvtHeader;   
 
    FairFileHeader* fFileHeader;   
 
    Bool_t fEventTimeInMCHeader;   
    Bool_t fEvtHeaderIsNew;   
 
    UInt_t fCurrentEntryNo;   
    UInt_t fTimeforEntryNo;   
 
    Double_t fEventTimeMin;   
    Double_t fEventTimeMax;   
    Double_t fEventTime;   
    Double_t fBeamTime;   
    Double_t fGapTime;   
    Double_t fEventMeanTime;   
    std::unique_ptr<TF1> fTimeProb;   
    Bool_t fCheckFileLayout;   
 
    int ExtractRunNumber();
    Int_t fPeriodNumber = -1;   
    Int_t fRunNumber = -1;   
    Int_t fFileType = -1;
 
    ClassDefOverride(BmnFileSource, 1);
};
 
#endif /* defined(BMNFILESOURCE_H) */