KVSimDirAnalyser.h

 
 
#ifndef __KVSIMDIRANALYSER_H
#define __KVSIMDIRANALYSER_H
 
#include "KVDataAnalyser.h"
#include "TChain.h"
 
#include <KVSimDir.h>
 
class KVSimDirAnalyser : public KVDataAnalyser {
 
    TList* fListOfSimFiles;
    TList* fListOfAuxFiles;
    TChain* fAnalysisChain;
    KVSimDir* fSimDir;
    Bool_t fCopyFilesToWorkDir;
 
    TString fModel;
    TString fFilterSystem;
    TString fFilterDataSet;
 
private:
   void DeleteSimFilesListIfOurs();
   void DeleteAuxFilesListIfOurs();
 
protected:
   void BuildChain();
   Bool_t NeedToChooseWhatToAnalyse() const override
   {
      return !(fListOfSimFiles && fListOfSimFiles->GetEntries());
   }
   void GetInfosForJobNameFromFiles();
 
public:
   KVSimDirAnalyser();
   ~KVSimDirAnalyser()
   {
      SafeDelete(fSimDir);
      DeleteSimFilesListIfOurs();
      DeleteAuxFilesListIfOurs();
   }
 
   void SetFileList(TList* l) override
   {
      fListOfSimFiles = l;
      GetInfosForJobNameFromFiles();
   }
   void SetAuxFileList(TList* l) override
   {
      fListOfAuxFiles = l;
   }
   TList* GetFileList() const
   {
      return fListOfSimFiles;
   }
   Int_t GetNumberOfFilesToAnalyse() const override
   {
      return fListOfSimFiles ? fListOfSimFiles->GetEntries() : 0;
   }
 
   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");
 
   void SetCopyFilesToWorkDir(Bool_t on = kTRUE)
   {
      fCopyFilesToWorkDir = on;
   }
   Bool_t IsCopyFilesToWorkDir() const
   {
      return fCopyFilesToWorkDir;
   }
   TString ExpandAutoBatchName(const Char_t* format) const override;
   void SetDataSetForFilter(const TString& f)
   {
      fFilterDataSet = f;
   }
   void SetSystemForFilter(const TString& f)
   {
      fFilterSystem = f;
   }
   TTree* GetAnalysedTree() const override
   {
      return fAnalysisChain;
   }
 
   ClassDefOverride(KVSimDirAnalyser, 1) //Analysis of trees containing simulated events
};
 
#endif