KVSimDirAnalyser.h

 
#ifndef __KVSIMDIRANALYSER_H
#define __KVSIMDIRANALYSER_H
 
#include "KVDataAnalyser.h"
#include "TChain.h"
#include "KVUnownedList.h"
#include <KVSimDir.h>
 
class KVSimDirAnalyser : public KVDataAnalyser {
 
    KVUnownedList fListOfSimFiles;
    KVUnownedList fListOfAuxFiles;
    TChain* fAnalysisChain;
    KVSimDir* fSimDir;
    Bool_t fCopyFilesToWorkDir;
 
    TString fModel;
    TString fFilterSystem;
    TString fFilterDataSet;
 
   Bool_t is_filter_job = false;
 
protected:
   void BuildChain();
   Bool_t NeedToChooseWhatToAnalyse() const override
   {
      return fListOfSimFiles.IsEmpty();
   }
   void GetInfosForJobNameFromFiles();
 
public:
   KVSimDirAnalyser();
   ~KVSimDirAnalyser()
   {
      SafeDelete(fSimDir);
   }
 
   void SetFileList(const KVUnownedList& l) override
   {
      fListOfSimFiles = l;
      GetInfosForJobNameFromFiles();
   }
   void SetAuxFileList(const KVUnownedList& l) override
   {
      fListOfAuxFiles = l;
   }
   auto& GetFileList() const
   {
      return fListOfSimFiles;
   }
   Int_t GetNumberOfFilesToAnalyse() const override
   {
      return fListOfSimFiles.GetEntries();
   }
 
   void SubmitTask() override;
   KVString GetRootDirectoryOfDataToAnalyse() const override;
 
   void WriteBatchEnvFile(const TString&, Bool_t sav = kTRUE) override;
   Bool_t ReadBatchEnvFile(const TString&) override;
 
   static void Make(const Char_t* kvsname = "MySimulatedAnalysis");
 
   TString ExpandAutoBatchName(const Char_t* format) const override;
   void SetDataSetForFilter(const TString& f)
   {
      fFilterDataSet = f;
      is_filter_job = true;
   }
   void SetSystemForFilter(const TString& f)
   {
      fFilterSystem = f;
      is_filter_job = true;
   }
   TTree* GetAnalysedTree() const override
   {
      return fAnalysisChain;
   }
   TString GetAnalyserClassNameForBatchJob() override
   {
      return "SimDirAnalyser";
   }
 
   ClassDefOverride(KVSimDirAnalyser, 1) //Analysis of trees containing simulated events
};
 
#endif