KVSimReader.h

 
#ifndef __KVSIMREADER_H
#define __KVSIMREADER_H
 
#include "KVSimEvent.h"
#include "KVSimNucleus.h"
#include "KVString.h"
#include "TTree.h"
#include "TFile.h"
#include "KVUnownedList.h"
#include "TObject.h"
#include "TDatime.h"
#include "KVFileReader.h"
 
class KVSimReader : public KVFileReader
{
   void init();
   TTree* tree = nullptr;
   TFile* file = nullptr;
   KVSimEvent* evt = nullptr;
 
   Int_t total_nevents,total_nevents_current_output_file;
   KVString tree_name{"SIMULATION_NAME"}, tree_title{"SIMULATION"},
            branch_name{"Simulated_evts"}, root_file_name{"Output.root"};
   Bool_t fMultiFiles = false;
   Int_t fFileIndex = 1;
   TString fOutputDirectory = "./";
 
    KVUnownedList linked_info;
    KVUnownedList linked_objects;
 
   std::optional<int> max_number_events_per_output_file;
   int current_output_file_index=0;
 
   bool OutputFileIsFull() const
   {
      return max_number_events_per_output_file
            && total_nevents_current_output_file==*max_number_events_per_output_file;
   }
   bool NeedNewOutputFile() const
   {
      return !(file && file->IsOpen());
   }
   KVString GetDate()
   {
      TDatime now;
      KVString stime;
      stime.Form("%d_%02d_%02d_%02d:%02d:%02d",
                 now.GetYear(),
                 now.GetMonth(),
                 now.GetDay(),
                 now.GetHour(),
                 now.GetMinute(),
                 now.GetSecond()
                );
      return stime;
   }
   void InitializeOutputFileAndTree();
   void FillTree()
   {
      GetTree()->Fill();
   }
 
   void SaveTree();
   void Run();
 
   void WriteObjects();
   void CreateObjectList();
 
   TTree* GetTree()
   {
      return tree;
   }
   void WriteInfo();
   void ReadEventsInFile();
   TString GetCurrentRootFileName();
 
   protected:
   KVSeqCollection* GetLinkedObjects();
   void AddObject(TObject* obj);
   KVSeqCollection* GetSimuInfo();
   void AddInfo(const KVNamedParameter&);
 
   public:
   KVSimReader();
 
   void ConvertEventsInFile(KVString filename);
 
   int GetEventNumber() const { return total_nevents; }
 
   void SetMaxNumEventsPerOutputFile(int nev)
   {
      max_number_events_per_output_file=nev;
   }
 
   void SetMultiFiles(Bool_t on = kTRUE)
   {
      fMultiFiles = on;
   }
   void SetFileIndex(Int_t i)
   {
      fFileIndex = i;
   }
   void SetOutputDirectory(const TString& where)
   {
      fOutputDirectory = where;
   }
 
   virtual Bool_t ReadFileHeader();
   virtual Bool_t ReadEventHeader(KVSimEvent*);
 
   virtual Bool_t ReadEvent(KVSimEvent*evt) = 0;
 
   virtual Bool_t ReadNucleus(KVSimNucleus*);
   virtual void NotifyChangeOfOutputFile();
   virtual bool NeedToIterateReadEvent();
 
   void SetTreeName(KVString n)
   {
      tree_name = n;
   }
   void SetTreeTitle(KVString n)
   {
      tree_title = n;
   }
   KVString GetTreeTitle() const
   {
      return tree_title ;
   }
   void SetROOTFileName(KVString name)
   {
      root_file_name = name;
      if(root_file_name.Index(".root")>0)
         root_file_name.Remove(root_file_name.Index(".root"));
   }
   void SetBranchName(KVString n)
   {
      branch_name = n;
   }
 
   void ConvertAndSaveEventsInFile(KVString filename);
 
   static KVSimReader* MakeSimReader(const char* model_uri);
 
   ClassDef(KVSimReader, 0) //Base class to read output files for simulation and create tree using KVEvent type class
};
 
#endif