/cvmfs/atlas.cern.ch/repo/sw/ASG/AnalysisBase/2.4.31/ElectronEfficiencyCorrection/ElectronEfficiencyCorrection/TElectronEfficiencyCorrectionTool.h

#ifndef __TELECTRONEFFICIENCYCORRECTIONTOOL__
#define __TELECTRONEFFICIENCYCORRECTIONTOOL__

#include <sstream>

// STL includes
#include <vector>
#include <string>

// ROOT includes
#include <TString.h>
#include "TKey.h"
#include "TObjArray.h"
#include "TH1.h"
#include "TH2.h"
#include "TRandom3.h"
// Include the return object and the base class
#include "PATCore/TResult.h"
#include "PATCore/TCalculatorToolBase.h"
#include "PATCore/PATCoreEnums.h"
#include "AsgTools/AsgMessaging.h"


namespace Root {
  class TElectronEfficiencyCorrectionTool : public Root::TCalculatorToolBase,public asg::AsgMessaging
    {

    public: 
      TElectronEfficiencyCorrectionTool( const char* name="TElectronEfficiencyCorrectionTool" );

      ~TElectronEfficiencyCorrectionTool();

      // Main methods
      int initialize();

      inline int finalize() { return 1; }

      const Root::TResult& calculate( const PATCore::ParticleDataType::DataType dataType,
                      const unsigned int runnumber,
                      const double cluster_eta,
                      const double et /* in MeV */
                      );


      inline void addFileName ( const std::string& val ) 
      { m_corrFileNameList.push_back(val); }
        
      inline void setResultPrefix ( const std::string& val ) { m_resultPrefix = val; }

      inline void setResultName ( const std::string& val ) { m_resultName = val; }

      inline void bookToyMCScaleFactors(const int nToyMC) {
    m_doToyMC = kTRUE;
    m_nToyMC = nToyMC;
      }

      inline void bookCombToyMCScaleFactors(const int nToyMC) {
    m_doCombToyMC = kTRUE;
    m_nToyMC = nToyMC;
      }
  
      int getFirstToyMCPosition();
      int getLastToyMCPosition();
      int getFirstCorrSysPosition();
      int getLastCorrSysPosition();
      int getGlobalBinNumberPosition();
      int getNbins(std::map<float, std::vector<float> >&); 
      int getNSyst() {return m_nSysMax;};

      void printResultMap();

      enum detailLevel{simple,medium,detailed};

      inline void setDetailLevel ( const int input_detailLevel ) { m_detailLevel = input_detailLevel; }

      inline void setSeed( const int seed) { m_seed = seed; }

      inline const char* getName() const { return m_name; }

      // Private methods
    private:
      int getHistograms();
      int getHistogramInDirectory( TKey *key );
      int setupHistogramsInFolder( const TObjArray& dirNameArray, int lastIdx );

      void calcDetailLevels(TH1D *eig) ;

      std::vector<TObjArray> buildToyMCTable (const TObjArray &sf, const TObjArray &eig, 
                          const TObjArray& stat, const TObjArray& uncorr, const std::vector<TObjArray> &corr);

      std::vector<TH2D*> buildSingleToyMC(TH2D *sf, TH2D* stat, TH2D* uncorr, const TObjArray& corr);

      TH2D* buildSingleCombToyMC(TH2D *sf, TH2D* stat, TH2D* uncorr, const TObjArray& corr);

      int setup( const TObjArray& hist,
         std::vector< TObjArray >& histList,
         std::vector< unsigned int >& beginRunNumberList,
         std::vector< unsigned int >& endRunNumberList );
    
      int setupSys( std::vector<TObjArray> & hist,
            std::vector< std::vector< TObjArray>> & histList);

      void printDefaultReturnMessage(TString reason, int line);

      TRandom3 m_Rndm;
      int m_randomCounter;
      bool m_isInitialized;
      int m_detailLevel;
      int m_toyMCSF;
      //
      int m_seed;
      bool m_doToyMC;
      bool m_doCombToyMC;
      int m_nToyMC;
      int m_sLevel[3];
      int m_nSys;
      int m_nSysMax;
      int m_runNumBegin;
      int m_runNumEnd;

      std::string m_resultPrefix;

      std::string m_resultName;

      int m_position_eff; 

      int m_position_err; 

      int m_position_statErr; 

      int m_position_nSys; 

      int m_position_uncorrSys; 

      int m_nbins; 

      int m_nSimpleUncorrSyst;

      int m_position_globalBinNumber; 


      //description here ?
      std::vector< std::vector<TObjArray>> m_uncorrToyMCSystFull;
      std::vector< std::vector<TObjArray>> m_uncorrToyMCSystFast;
      std::vector< std::string > m_corrFileNameList;
      std::vector< unsigned int > m_begRunNumberList;
      std::vector< unsigned int > m_endRunNumberList;
      std::vector< unsigned int > m_begRunNumberListFastSim;
      std::vector< unsigned int > m_endRunNumberListFastSim;    
      std::map<int, std::vector< TObjArray > > m_histList;
      std::vector< std::vector< TObjArray > > m_sysList;
      std::map<int, std::vector< TObjArray > > m_fastHistList;
      std::vector< std::vector< TObjArray > > m_fastSysList;
      std::vector<int> m_position_corrSys; 
      std::vector<int> m_position_uncorrToyMCSF; 
      std::vector<int> m_keys;
    
    }; // End: class definition
  
} // End: namespace Root

#endif

---