KVSimDirAnalyser.cpp

//Created by KVClassFactory on Tue Feb 14 11:13:53 2017
//Author: John Frankland,,,
 
#include "KVSimDirAnalyser.h"
#include "KVDataAnalysisTask.h"
#include <KVSimFile.h>
#include <KVSimDir.h>
#include <KVClassFactory.h>
#include <TStopwatch.h>
#include "TSystem.h"
 
ClassImp(KVSimDirAnalyser)
 
 
 
KVSimDirAnalyser::KVSimDirAnalyser()
   : KVDataAnalyser(), fAnalysisChain(nullptr), fSimDir(nullptr), fCopyFilesToWorkDir(false)
{
   // Default constructor
}
 
 
 
 
void KVSimDirAnalyser::SubmitTask()
{
   // Set up and run the analysis
 
   BuildChain();
 
   Bool_t read_all_events = (GetNbEventToRead() == 0);
   Long64_t nevents = (!read_all_events ? GetNbEventToRead() : fAnalysisChain->GetEntries());
   Long64_t update_interval = (nevents > 10 ? nevents / 10 : 1);
   TString results_file_name;
   KVSimFile* first_file = (KVSimFile*)fListOfSimFiles.First();
   results_file_name.Form("%s_%s", GetUserClass().name.Data(), first_file->GetName());
   TString options;
   options.Form("SimFileName=%s,SimTitle=%s,OutputDir=%s,EventsReadInterval=%lld,BranchName=%s,CombinedOutputFile=%s,SimulationInfos=%s",
                first_file->GetName(), fAnalysisChain->GetTitle(),
                (is_filter_job ? (first_file->GetSimDir()->HasAuxDirectory() ? first_file->GetSimDir()->GetAuxDirectory().Data() : ".") : "."),// make sure filtered files are always placed in same directory
                update_interval, first_file->GetBranchName(),
                results_file_name.Data(), first_file->GetTitle());
   if (first_file->IsFiltered()) {
      options += Form(",DataSet=%s", first_file->GetDataSet());
      if (first_file->HasRun())
         options += Form(",Run=%d", first_file->GetRun());
      if (first_file->HasSystem())
         options += Form(",System=%s", first_file->GetSystem());
   }
   if (!fListOfAuxFiles.IsEmpty()) {
      // Set up list of auxiliary files
      KVString auxfiles;
      TIter ifi(&fListOfAuxFiles);
      KVSimFile* auxf;
      while ((auxf = (KVSimFile*)ifi())) {
         if (auxfiles != "") auxfiles += "|";
         auxfiles += auxf->GetName();
      }
      options += ",AuxFiles=";
      options += auxfiles;
      options += ",AuxDir=";
      KVSimFile* ffaux = (KVSimFile*)fListOfAuxFiles.First();
      options += ffaux->GetSimDir()->GetDirectory();
      options += ",AuxTreeName=";
      options += ffaux->GetTreeName();
      options += ",AuxBranchName=";
      options += ffaux->GetBranchName();
   }
   // Add any user-defined options - notably, all options for filter are passed here
   if (GetUserClassOptions() != "") {
      options += ",";
      options += GetUserClassOptions();
   }
 
   // Check compilation of user class & run
   if (CheckIfUserClassIsValid()) {
      Info("SubmitTask", "Beginning TChain::Process...");
#ifdef WITH_PROOF
#ifdef WITH_CPP11
      if (GetProofMode() != KVDataAnalyser::EProofMode::None) fAnalysisChain->SetProof(kTRUE);
#else
      if (GetProofMode() != KVDataAnalyser::None) fAnalysisChain->SetProof(kTRUE);
#endif
#endif
      TString analysis_class;
      if (GetAnalysisTask()->WithUserClass())
         analysis_class.Form("%s%s", GetUserClass().full_path_imp().Data(), GetACliCMode());
      else analysis_class = GetUserClassName();
      if (read_all_events) {
         fAnalysisChain->Process(analysis_class, options.Data());
      }
      else {
         fAnalysisChain->Process(analysis_class, options.Data(), GetNbEventToRead());
      }
   }
   delete fAnalysisChain;
   fAnalysisChain = nullptr;
}
 
 
 
 
KVString KVSimDirAnalyser::GetRootDirectoryOfDataToAnalyse() const
{
   // Returns path to data to be analysed
   if (!NeedToChooseWhatToAnalyse()) {
      KVSimFile* first_file = (KVSimFile*)fListOfSimFiles.First();
      return first_file->GetSimDir()->GetDirectory();
   }
   return "";
}
 
 
 
 
void KVSimDirAnalyser::WriteBatchEnvFile(const TString& jobname, Bool_t sav)
{
   //Save (in the TEnv fBatchEnv) all necessary information on analysis task which can be used to execute it later
   //(i.e. when batch processing system executes the job).
   //If save=kTRUE (default), write the information in a file whose name is given by ".jobname"
   //where 'jobname' is the name of the job as given to the batch system.
 
   KVDataAnalyser::WriteBatchEnvFile(jobname, kFALSE);
   KVSimFile* simF = (KVSimFile*)fListOfSimFiles.First();
   KVSimDir* simD = simF->GetSimDir();
   GetBatchInfoFile()->SetValue("SimDir", simD->GetDirectory());
   if (simD->HasAuxDirectory())
      GetBatchInfoFile()->SetValue("AuxDir", simD->GetAuxDirectory());
   if (simF->IsFiltered()) GetBatchInfoFile()->SetValue("SimFile.Type", "filtered");
   else GetBatchInfoFile()->SetValue("SimFile.Type", "simulated");
   GetBatchInfoFile()->SetValue("SimFiles", simF->GetName());
   if (fListOfSimFiles.GetEntries() > 1) {
      TIter next(&fListOfSimFiles);
      next();
      while ((simF = (KVSimFile*)next())) GetBatchInfoFile()->SetValue("+SimFiles", simF->GetName());
   }
   if (!fListOfAuxFiles.IsEmpty()) {
      // auxiliary files for batch job
      GetBatchInfoFile()->SetValue("AuxFiles", fListOfAuxFiles.First()->GetName());
      if (fListOfAuxFiles.GetEntries() > 1) {
         TIter next(&fListOfAuxFiles);
         next();
         while ((simF = (KVSimFile*)next())) GetBatchInfoFile()->SetValue("+AuxFiles", simF->GetName());
      }
   }
   if (is_filter_job) GetBatchInfoFile()->SetValue("FilterJob", 1);
   if (sav) GetBatchInfoFile()->SaveLevel(kEnvUser);
}
 
 
 
 
Bool_t KVSimDirAnalyser::ReadBatchEnvFile(const TString& filename)
{
   //Read the batch env file "filename" and initialise the analysis task using the
   //informations in the file
   //Returns kTRUE if all goes well
 
   Bool_t ok = kFALSE;
 
   if (!KVDataAnalyser::ReadBatchEnvFile(filename)) return ok;
 
   KVString simdir = GetBatchInfoFile()->GetValue("SimDir", "");
   if (simdir == "") return ok;
 
   KVString auxdir = GetBatchInfoFile()->GetValue("AuxDir", "");
 
   is_filter_job = (Bool_t)GetBatchInfoFile()->GetValue("FilterJob", 0);
 
   fSimDir = new KVSimDir("SIMDIR", simdir);
   if (!auxdir.IsNull()) fSimDir->SetAuxDirectory(auxdir);
   fSimDir->AnalyseDirectory();
 
   KVString filetype = GetBatchInfoFile()->GetValue("SimFile.Type", "");
   if (filetype == "") return ok;
 
   KVString simfiles = GetBatchInfoFile()->GetValue("SimFiles", "");
   if (simfiles == "") return ok;
 
   fListOfSimFiles.Clear();
 
   simfiles.Begin(" ");
   while (!simfiles.End()) {
      if (filetype == "simulated") fListOfSimFiles.Add(fSimDir->GetSimDataList()->FindObject(simfiles.Next()));
      else if (filetype == "filtered") fListOfSimFiles.Add(fSimDir->GetFiltDataList()->FindObject(simfiles.Next()));
   }
 
   KVString auxfiles = GetBatchInfoFile()->GetValue("AuxFiles", "");
   if (auxfiles == "") return (ok = kTRUE);
 
   fListOfAuxFiles.Clear();
 
   auxfiles.Begin(" ");
   while (!auxfiles.End()) {
      fListOfAuxFiles.Add(fSimDir->GetSimDataList()->FindObject(auxfiles.Next()));
   }
 
   ok = kTRUE;
 
   return ok;
}
 
 
 
 
void KVSimDirAnalyser::BuildChain()
{
   // Build a TChain with all files/trees to be analysed
   //
   // If fCopyFilesToWorkDir==true, files are first copied to working directory
 
   TIter next(&fListOfSimFiles);
   KVSimFile* file;
   while ((file = (KVSimFile*)next())) {
      if (!fAnalysisChain) {
         fAnalysisChain = new TChain(file->GetTreeName());
      }
      fAnalysisChain->Add(file->GetFullPathToFile());
   }
}
 
 
 
 
void KVSimDirAnalyser::GetInfosForJobNameFromFiles()
{
   // Use first file in list to analyse to obtain information for generating
   // automatic batch job names (name of model, eventually info on simulated reaction,
   // etc.)
 
   KVSimFile* sf = dynamic_cast<KVSimFile*>(fListOfSimFiles.First());
   // name of TTree is (usually) name of model
   fModel = sf->GetTreeName();
   if (fModel == "dit_events") fModel = "DIT"; // except for DIT
}
 
 
 
 
 
void KVSimDirAnalyser::Make(const Char_t* kvsname)
{
   // Generate a new simulated analysis selector class
 
#ifndef USING_ROOT6
   KVClassFactory cf(kvsname, "Analysis of simulated events", "",
                     kTRUE, "SimulatedEventAnalysisTemplate");
#else
   KVClassFactory cf(kvsname, "Analysis of simulated events", "",
                     kTRUE, "ROOT6SimulatedEventAnalysisTemplate");
#endif
   cf.AddImplIncludeFile("KVSimNucleus.h");
   cf.AddImplIncludeFile("KVBatchSystem.h");
 
   cf.GenerateCode();
}
 
 
 
 
TString KVSimDirAnalyser::ExpandAutoBatchName(const Char_t* format) const
{
   // Replace any special symbols in auto-format jobname with current values
 
   TString tmp = KVDataAnalyser::ExpandAutoBatchName(format);
   tmp.ReplaceAll("KVEventFiltering",
                  Form("%s_%s_FILTER_%s", fModel.Data(), fFilterSystem.Data(), fFilterDataSet.Data()));
   tmp.ReplaceAll(" ", "");
   tmp.ReplaceAll("@", "_");
   tmp.ReplaceAll("+", "_");
   tmp.ReplaceAll("/", "_");
   tmp.ReplaceAll("(", "_");
   tmp.ReplaceAll(")", "_");
   tmp.ReplaceAll("[", "_");
   tmp.ReplaceAll("]", "_");
   return tmp;
}