KVSimReader.cpp

//Created by KVClassFactory on Wed Jun 30 17:45:01 2010
//Author: bonnet
 
#include "KVSimReader.h"
#include "TDirectory.h"
#include "TStopwatch.h"
 
ClassImp(KVSimReader)
 
 
 
KVSimReader::KVSimReader()
   : evt{new KVSimEvent}
{
   linked_info.SetName("Simulation info");
   linked_objects.SetName("List of objects");
}
 
 
 
 
void KVSimReader::ConvertEventsInFile(KVString filename)
{
   if (!OpenFileToRead(filename)) return;
   Run();
   CloseFile();
}
 
 
 
 
void KVSimReader::CleanAll()
{
   nv.Clear();
   GetSimuInfo()->Clear();
   GetLinkedObjects()->Clear();
}
 
 
 
 
void KVSimReader::DeclareTree(Option_t* option)
{
   // Use root_file_name as basename for file
   //
   // If multiple files are being read, they will be named
   //~~~
   //    [root_file_name].1
   //    [root_file_name].2
   //    [root_file_name].etc
   //~~~
   //
   // Files are written in fOutputDirectory (default: "./")
 
   TString filename;
   filename = root_file_name;
   if (fMultiFiles) filename += Form(".%d", fFileIndex);
   if (!fOutputDirectory.EndsWith("/")) fOutputDirectory += "/";
   filename.Prepend(fOutputDirectory);
   KVError::Info(this, "DeclareTree", "Opened output file: %s", filename.Data());
   file = new TFile(filename, option);
 
   tree = new TTree(tree_name, tree_title);
   KVEvent::MakeEventBranch(tree, branch_name, evt);
}
 
 
 
 
void KVSimReader::ReadFile()
{
   AbstractMethod("ReadFile");
}
 
 
 
 
Bool_t KVSimReader::ReadHeader()
{
 
   AbstractMethod("ReadHeader");
   return kTRUE;
}
 
 
 
 
Bool_t KVSimReader::ReadEvent()
{
   AbstractMethod("ReadEvent");
   return kTRUE;
}
 
 
 
 
Bool_t KVSimReader::ReadNucleus()
{
   AbstractMethod("ReadNucleus");
   return kTRUE;
}
 
 
 
 
void KVSimReader::ConvertAndSaveEventsInFile(KVString filename)
{
   // Read events, convert and save in ROOT file
   ConvertEventsInFile(filename);
   SaveTree();
}
 
 
 
 
KVSimReader* KVSimReader::MakeSimReader(const char* model_uri)
{
   // Returns an instance of a child class specific to given model.
 
   TPluginHandler* ph = LoadPlugin("KVSimReader", model_uri);
   if (!ph) {
      ::Error("KVSimReader::MakeSimReader", "No plugin defined with name %s", model_uri);
      return nullptr;
   }
   return (KVSimReader*)ph->ExecPlugin(0);
}
 
 
 
 
void KVSimReader::Run(Option_t* option)
{
 
   AddInfo({"ascii file read", GetFileName()});
 
   TStopwatch chrono;
   chrono.Start();
 
   if (HasToFill()) DeclareTree(option);
   nuc = nullptr;
   nevt = 0;
 
   ReadFile();
 
   if (HasToFill())
      GetTree()->ResetBranchAddress(GetTree()->GetBranch(branch_name.Data()));
 
   chrono.Stop();
 
   //Info("Run","%d evts lus en %lf seconds",GetNumberOfEvents(),chrono.RealTime());
 
   AddInfo({"number of events read", nevt});
   AddInfo({"date", GetDate()});
   AddInfo({"user", gSystem->GetUserInfo()->fUser});
 
}
 
 
 
 
void KVSimReader::SaveTree()
{
   WriteObjects();
   WriteInfo();
 
   GetTree()->Write();
   CleanAll();
   file->Close();
}
 
 
 
 
KVSeqCollection* KVSimReader::GetSimuInfo()
{
   return &linked_info;
}
 
 
 
 
void KVSimReader::AddInfo(const KVNamedParameter& info)
{
   // Add information on simulation using KVNamedParameter, can be int, double or string:
   //~~~{.cpp}
   // AddInfo({"model", "AMD"});
   // AddInfo({"sigma_NN_screen_factor", 0.85});
   // AddInfo({"N_part_test", 150});
 
   linked_info.Add(new KVNamedParameter(info));
}
 
 
 
 
void KVSimReader::WriteInfo()
{
   TList* list = GetTree()->GetUserInfo();
   Int_t nentries = linked_info.GetEntries();
   for (Int_t ii = 0; ii < nentries; ii += 1) {
      list->Add(linked_info.RemoveAt(0));
   }
}
 
 
 
 
KVSeqCollection* KVSimReader::GetLinkedObjects()
{
   return &linked_objects;
}
 
 
 
 
void KVSimReader::AddObject(TObject* obj)
{
   GetLinkedObjects()->Add(obj);
}
 
 
 
 
void KVSimReader::WriteObjects()
{
 
   TList* list = GetTree()->GetUserInfo();
   Int_t nentries = linked_objects.GetEntries();
   for (Int_t ii = 0; ii < nentries; ii += 1) {
      list->Add(linked_objects.RemoveAt(0));
   }
 
}