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();
 
protected:
   TTree* tree = nullptr;
   TFile* file = nullptr;
   KVSimEvent* evt = nullptr;
   KVSimNucleus* nuc = nullptr;
 
   Int_t nevt;
   KVString tree_name{"SIMULATION_NAME"}, tree_title{"SIMULATION"},
            branch_name{"Simulated_evts"}, root_file_name{"Output.root"};
   Bool_t kmode = true;
   Bool_t fMultiFiles = false;
   Int_t fFileIndex = 1;
   TString fOutputDirectory = "./";
 
    KVUnownedList linked_info;
    KVUnownedList linked_objects;
    KVNameValueList nv;
 
public:
   KVSimReader();
   virtual void ConvertEventsInFile(KVString filename);
 
   void CleanAll();
 
   void SetMultiFiles(Bool_t on = kTRUE)
   {
      fMultiFiles = on;
   }
   void SetFileIndex(Int_t i)
   {
      fFileIndex = i;
   }
   void SetOutputDirectory(const TString& where)
   {
      fOutputDirectory = where;
   }
 
   virtual 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 SetFillingMode(Bool_t mode = kTRUE)
   {
      kmode = mode;
   }
   void DeclareTree(Option_t* option);
 
   TTree* GetTree()
   {
      return tree;
   }
   virtual void FillTree()
   {
      GetTree()->Fill();
   }
   Bool_t HasToFill()
   {
      return kmode;
   }
   virtual void SaveTree();
   void Run(Option_t* option = "recreate");
 
   KVSeqCollection* GetLinkedObjects();
   void AddObject(TObject* obj);
   void WriteObjects();
   void CreateObjectList();
 
   KVSeqCollection* GetSimuInfo();
 
   void AddInfo(const KVNamedParameter&);
 
   void WriteInfo();
   void CreateInfoList();
 
   virtual void ReadFile();
   virtual Bool_t ReadHeader();
   virtual Bool_t ReadEvent();
   virtual Bool_t ReadNucleus();
 
   Int_t GetNumberOfEvents()
   {
      return nevt;
   }
 
   void SetTreeName(const Char_t* n)
   {
      tree_name = n;
   }
   void SetTreeTitle(const Char_t* n)
   {
      tree_title = n;
   }
   void SetROOTFileName(const Char_t* n)
   {
      root_file_name = n;
   }
   void SetBranchName(const Char_t* n)
   {
      branch_name = n;
   }
 
   const char* GetROOTFileName()
   {
      return root_file_name.Data();
   }
 
   virtual void ConvertAndSaveEventsInFile(KVString filename);
 
   static KVSimReader* MakeSimReader(const char* model_uri);
 
   ClassDef(KVSimReader, 1) //Base class to read output files for simulation and create tree using KVEvent type class
};
 
#endif