KVPROOFSelector.h

 
#ifndef KVPROOFSelector_h
#define KVPROOFSelector_h
 
#include <KVHashList.h>
#include <KVNameValueList.h>
#include <KVString.h>
#include <TH1.h>
#include <TH2.h>
#include <TH3.h>
#include <TSelector.h>
#include "TProofOutputFile.h"
 
class KVPROOFSelector : public TSelector {
 
protected :
    Long64_t fEventsRead;
    Long64_t fEventsReadInterval;
 
    KVHashList* lhisto;
    KVHashList* ltree;
 
    KVString fCombinedOutputFile;
 
   KVNameValueList fOptionList;
 
    Bool_t fDisableCreateTreeFile;
 
   void FillTH1(TH1* h1, Double_t one, Double_t two);
   void FillTProfile(TProfile* h1, Double_t one, Double_t two, Double_t three);
   void FillTH2(TH2* h2, Double_t one, Double_t two, Double_t three);
   void FillTProfile2D(TProfile2D* h2, Double_t one, Double_t two, Double_t three, Double_t four);
   void FillTH3(TH3* h3, Double_t one, Double_t two, Double_t three, Double_t four);
 
public:
    TFile* writeFile;
    TProofOutputFile* mergeFile;
   TString tree_file_name;
   Bool_t CreateTreeFile(const Char_t* filename = "");
 
   virtual void ParseOptions();
 
   KVPROOFSelector(TTree* /*tree*/ = 0) :
      fEventsRead(0), fEventsReadInterval(50000), fDisableCreateTreeFile(kFALSE)
   {
      lhisto = new KVHashList();
      ltree = new KVHashList();
   }
   virtual ~KVPROOFSelector()
   {
      lhisto->Clear();
      delete lhisto;
      lhisto = 0;
      ltree->Clear();
      delete ltree;
      ltree = 0;
   }
   void SetEventsReadInterval(Long64_t N)
   {
      fEventsReadInterval = N;
   }
   Int_t   Version() const override
   {
      return 3;
   }
   void    Begin(TTree* tree) override;
   void    SlaveBegin(TTree* tree) override;
   Bool_t  Process(Long64_t entry) override;
   void    SlaveTerminate() override;
   void    Terminate() override;
 
   /* user entry points */
   virtual void InitAnalysis() = 0;
   virtual Bool_t Analysis() = 0;
   virtual void EndAnalysis() = 0;
 
   void AddHisto(TH1* histo);
   void AddTree(TTree* tree);
 
   void FillHisto(const Char_t* sname, Double_t one, Double_t two = 1, Double_t three = 1, Double_t four = 1);
   void FillTree(const Char_t* sname = "");
 
   KVHashList* GetHistoList() const;
   KVHashList* GetTreeList() const;
 
   TH1* GetHisto(const Char_t* name) const;
   TTree* GetTree(const Char_t* name) const;
 
   virtual void SaveHistos(const Char_t* filename = "", Option_t* option = "recreate", Bool_t onlyfilled = kFALSE);
 
   virtual void SetOpt(const Char_t* option, const Char_t* value);
   virtual Bool_t IsOptGiven(const Char_t* option);
   virtual TString GetOpt(const Char_t* option) const;
   virtual void UnsetOpt(const Char_t* opt);
 
   void SetCombinedOutputFile(const TString& filename)
   {
      fCombinedOutputFile = filename;
   }
   Long64_t GetCycleNumber() const
   {
      return fEventsRead;
   }
   static void Make(const Char_t* classname);
 
   ClassDefOverride(KVPROOFSelector, 1)//General purpose class for performing parallel tasks with PROOF
};
 
#endif