KVDataSetAnalyser.h

 
#ifndef __KVDATASETANALYSER_H
#define __KVDATASETANALYSER_H
 
#include "KVDataAnalyser.h"
#include "KVDataSet.h"
 
class KVDataSetAnalyser : public KVDataAnalyser {
private:
    KVDBSystem* fSystem;
    KVDBRun* fRun;
    KVNumberList fRunList;
    KVNumberList fFullRunList;
    KVDataSet* fDataSet;
 
    Bool_t fChoozDataSet;
    Bool_t fChoozTask;
    Bool_t fChoozSystem;
    Bool_t fChoozRuns;
 
protected:
   void SetCurrentRun(KVDBRun* r)
   {
      fRun = r;
      fSystem = r->GetSystem();
   }
   virtual KVNumberList PrintAvailableRuns(KVString& datatype);
   virtual void set_dataset_pointer(KVDataSet* ds);
   virtual void set_dataset_name(const Char_t* name);
   void _set_dataset_pointer(KVDataSet* ds)
   {
      fDataSet = ds;
   }
   virtual Bool_t PreSubmitCheck();
   virtual void PostRunReset();
 
   Bool_t NeedToChooseWhatToDo() const
   {
      return fChoozDataSet || IsChooseTask();
   }
   Bool_t NeedToChooseWhatToAnalyse() const
   {
      return fChoozSystem || fChoozRuns;
   }
   void ChooseWhatToAnalyse();
   void ChooseWhatToDo();
   Bool_t CheckWhatToAnalyseAndHow();
   virtual void set_up_analyser_for_task(KVDataAnalyser* the_analyser);
 
public:
   TString SystemBatchName() const;
   KVDataSetAnalyser();
   virtual ~KVDataSetAnalyser();
 
   void ChooseDataSet();
   void ChooseDataType();
   virtual void ChooseAnalysisTask();
   void ChooseSystem(const Char_t* data_type = "");
   void ChooseRuns(KVDBSystem* system = nullptr, const Char_t* data_type = "");
   void Reset();
 
   const KVDBSystem* GetSystem() const
   {
      return fSystem;
   }
   const KVNumberList& GetRunList() const
   {
      return fRunList;
   }
   Int_t GetNumberOfFilesToAnalyse() const
   {
      return fRunList.GetNValues();
   }
   const KVNumberList& GetFullRunList() const
   {
      return fFullRunList;
   }
   const KVDataSet* GetDataSet() const
   {
      return fDataSet;
   }
   void SetDataSet(KVDataSet* ds)
   {
      set_dataset_pointer(ds);
   }
   void SetDataSet(const Char_t* name)
   {
      set_dataset_name(name);
   }
   void SetSystem(KVDBSystem* syst);
   void SetRuns(const KVNumberList& nl, Bool_t check = kTRUE);
   void SetFullRunList(const KVNumberList& nl)
   {
      fFullRunList = nl;
   }
 
   void ClearRunList()
   {
      fRunList.Clear();
   }
   Int_t GetRunNumberFromFileName(const Char_t*);
   void SetChooseDataSet(Bool_t yes = kTRUE)
   {
      fChoozDataSet = yes;
   }
   Bool_t IsChooseDataSet() const
   {
      return fChoozDataSet;
   }
   void SetChooseTask(Bool_t yes = kTRUE)
   {
      fChoozTask = yes;
   }
   Bool_t IsChooseTask() const
   {
      return fChoozTask;
   }
   void SetChooseSystem(Bool_t yes = kTRUE)
   {
      fChoozSystem = yes;
   }
   Bool_t IsChooseSystem() const
   {
      return fChoozSystem;
   }
   void SetChooseRuns(Bool_t yes = kTRUE)
   {
      fChoozRuns = yes;
   }
   Bool_t IsChooseRuns() const
   {
      return fChoozRuns;
   }
   void WriteBatchEnvFile(const TString& jobname, Bool_t save = kTRUE);
   Bool_t ReadBatchEnvFile(const TString& filename);
   TString ExpandAutoBatchName(const Char_t* format) const;
   virtual void AddJobDescriptionList(TList*);
   KVString GetRootDirectoryOfDataToAnalyse() const;
   const Char_t* GetRecognisedAutoBatchNameKeywords() const;
 
   const KV2Body* GetKinematics() const;
   KVDBSystem* GetAnalysedSystem() const
   {
      return fSystem;
   }
   KVDBRun* GetAnalysedRun() const
   {
      return fRun;
   }
 
   ClassDef(KVDataSetAnalyser, 1) //Analysis of data in datasets
};
 
#endif