KVRawDataAnalyser.h

 
#ifndef __KVRAWDATAANALYSER_H
#define __KVRAWDATAANALYSER_H
 
#include "KVDataSetAnalyser.h"
#include "KVDataAnalysisTreeHistoHandler.h"
#include "KVTreeVectorBranchHandler.h"
#include "KVHashList.h"
#include "KVRawDataReader.h"
#include "KVDBRun.h"
#include "KVRawDataPatchList.h"
#include <KVUserAnalysisOptionList.h>
#include <TH2.h>
#include <TH3.h>
#include <TProfile.h>
#include <TProfile2D.h>
 
class KVDetectorEvent;
 
class KVRawDataAnalyser : public KVDataSetAnalyser,
                          protected KVTreeVectorBranchHandler,
                          public KVDataAnalysisTreeHistoHandler<KVRawDataAnalyser>
{
   friend class KVDataAnalysisTreeHistoHandler<KVRawDataAnalyser>;
protected:
 
   KVRawDataReader* fRunFile = nullptr;    //currently analysed run file
    run_index_t fRunNumber;
    Long64_t fEventNumber;
   TFile* fTreeFile = nullptr;
   KVDBRun* fCurrentRun = nullptr;         //poiner to current run
   Long64_t TotalEntriesToRead;
    KVString fCombinedOutputFile;
    KVUserAnalysisOptionList fOptionList;
 
   KVRawDataPatchList fRustines;
 
   void ProcessRun();
 
   void AbortDuringRunProcessing();
 
   void add_histo(TH1* histo)
   {
      GetHistoList().Add(histo);
   }
   Bool_t CreateTreeFile(const Char_t* filename = "");
   void add_tree(TTree* tree)
   {
      CreateTreeFile();
      tree->SetDirectory(fTreeFile);
      tree->AutoSave();
      GetTreeList().Add(tree);
   }
 
   void SetCombinedOutputFile(const KVString& filename)
   {
      fCombinedOutputFile = filename;
   }
 
public:
   Long64_t GetTotalEntriesToRead() const override
   {
      return TotalEntriesToRead;
   }
 
   KVRawDataAnalyser();
   virtual ~KVRawDataAnalyser();
 
   virtual void InitAnalysis() = 0;
   virtual void InitRun() = 0;
   virtual Bool_t Analysis() = 0;
   virtual void EndRun() = 0;
   virtual void EndAnalysis() = 0;
 
   Int_t GetRunNumber() const
   {
      return fRunNumber.run();
   }
   const run_index_t& GetRunIndexNumber() const
   {
      return fRunNumber;
   }
   Long64_t GetEventNumber() const
   {
      return fEventNumber;
   }
   KVDBRun* GetCurrentRun() const
   {
      return fCurrentRun;
   }
   KVDBRunFile* GetCurrentRunFile() const
   {
      return &gExpDB->GetDBRunFile(GetRunIndexNumber());
   }
 
   void SubmitTask() override;
   static void Make(const Char_t* kvsname = "MyOwnRawDataAnalyser");
 
   void CalculateTotalEventsToRead();
 
   const KVRawDataReader& GetRunFileReader() const
   {
      return *fRunFile;
   }
 
   virtual void SaveHistos(const Char_t* filename = "", Option_t* option = "recreate", Bool_t onlyfilled = kFALSE);
 
   void SetJobOutputFileName(const TString& filename)
   {
      if (IsRunningBatchAnalysis())
         SetCombinedOutputFile(Form("%s.root", GetBatchSystem()->GetJobName()));
      else
         SetCombinedOutputFile(filename);
   }
 
   void SetOpt(const Char_t* option, const Char_t* value)
   {
      fOptionList.SetOpt(option, value);
   }
   Bool_t IsOptGiven(const Char_t* option)
   {
      return fOptionList.IsOptGiven(option);
   }
   TString GetOpt(const Char_t* option) const
   {
      return fOptionList.GetOpt(option);
   }
   void UnsetOpt(const Char_t* opt)
   {
      fOptionList.UnsetOpt(opt);
   }
 
   ClassDefOverride(KVRawDataAnalyser, 1) //Abstract base class for user analysis of raw data
};
 
#endif