GenAnalysis.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef GENERATORMODULES_GENANALYSIS_H
#define GENERATORMODULES_GENANALYSIS_H
 
#include "GeneratorModules/GenBase.h"
#include "GaudiKernel/ITHistSvc.h"
#include "TH1D.h"
#include "TH2D.h"
#include "TProfile.h"
 

class GenAnalysis : public GenBase {
public:
 
  GenAnalysis(const std::string& name, ISvcLocator* pSvcLocator);
  virtual ~GenAnalysis() { }

  StatusCode initialize();
  StatusCode execute();


  virtual StatusCode init() { return StatusCode::SUCCESS; }
  virtual StatusCode analyze() = 0;
  virtual StatusCode finalize() { return StatusCode::SUCCESS; }
 
 
protected:



 
  TH1D* bookHisto1D(const std::string& name, const std::string& title, const std::vector<double>& binedges,
                    const std::string& xtitle="", const std::string& ytitle="");
  TH1D* bookHisto1D(const std::string& name, const std::string& title, size_t numbins, double low, double high,
                    const std::string& xtitle="", const std::string& ytitle="");
 
  TProfile* bookProfile1D(const std::string& name, const std::string& title, const std::vector<double>& binedges,
                          const std::string& xtitle="", const std::string& ytitle="");
  TProfile* bookProfile1D(const std::string& name, const std::string& title, size_t numbins, double low, double high,
                          const std::string& xtitle="", const std::string& ytitle="");
 
  TH2D* bookHisto2D(const std::string& name, const std::string& title,
                    const std::vector<double>& binedgesx, const std::vector<double>& binedgesy,
                    const std::string& xtitle="", const std::string& ytitle="", const std::string& ztitle="");
  TH2D* bookHisto2D(const std::string& name, const std::string& title,
                    size_t numbinsx, double xlow, double xhigh,
                    size_t numbinsy, double ylow, double yhigh,
                    const std::string& xtitle="", const std::string& ytitle="", const std::string& ztitle="");

  std::string fullhistopath(const std::string& localpath) {
    return "/" + m_histkey + "/" + name() + "/" + localpath;
  }

  TH1* histo(const std::string& key);

  TProfile* profile(const std::string& key);

  const ServiceHandle<ITHistSvc> histSvc() const {
    return m_histSvc;
  }

 
 
private:

  std::string m_histkey;
  ServiceHandle<ITHistSvc> m_histSvc;
 
};
 
 
#endif