EgammaObjectCollectionMaker.h

Go to the documentation of this file.

/*
   Copyright (C) 2002-2021 CERN for the benefit of the ATLAS collaboration
 */
 
// $Id: EgammaObjectCollectionMaker.h 811374 2017-10-24 13:04:52Z iconnell $
#ifndef ANALYSISTOP_TOPSYSTEMATICOBJECTMAKER_EGAMMAOBJECTCOLLECTIONMAKER_H
#define ANALYSISTOP_TOPSYSTEMATICOBJECTMAKER_EGAMMAOBJECTCOLLECTIONMAKER_H
 
// system include(s):
#include <memory>
#include <set>
#include <list>
 
// Framework include(s):
#include "AsgTools/AsgTool.h"
#include "AsgTools/ToolHandle.h"
 
// Systematic include(s):
#include "PATInterfaces/SystematicSet.h"
 
// CP Tool include(s):
#include "EgammaAnalysisInterfaces/IEgammaCalibrationAndSmearingTool.h"
#include "EgammaAnalysisInterfaces/IElectronPhotonShowerShapeFudgeTool.h"
#include "IsolationSelection/IIsolationSelectionTool.h"
//#include "IsolationSelection/IIsolationLowPtPLVTool.h"
#include "IsolationCorrections/IIsolationCorrectionTool.h"
 
// Forward declaration(s):
namespace top {
  class TopConfig;
}
 
namespace top {
  class EgammaObjectCollectionMaker final: public asg::AsgTool {
  public:
    explicit EgammaObjectCollectionMaker(const std::string& name);
    virtual ~EgammaObjectCollectionMaker() {}
 
    // Delete Standard constructors
    EgammaObjectCollectionMaker(const EgammaObjectCollectionMaker& rhs) = delete;
    EgammaObjectCollectionMaker(EgammaObjectCollectionMaker&& rhs) = delete;
    EgammaObjectCollectionMaker& operator = (const EgammaObjectCollectionMaker& rhs) = delete;
 
    StatusCode initialize();
 
    StatusCode executePhotons(bool);
    StatusCode executeElectrons(bool);
    StatusCode executeFwdElectrons(bool);
 
    StatusCode printoutPhotons();
    StatusCode printoutElectrons();
    StatusCode printoutFwdElectrons();
 
    // return specific Systematic
    inline virtual const std::list<CP::SystematicSet>& specifiedSystematicsPhotons() const {
      return m_specifiedSystematicsPhotons;
    }
    inline virtual const std::list<CP::SystematicSet>& specifiedSystematicsElectrons() const {
      return m_specifiedSystematicsElectrons;
    }
    inline virtual const std::list<CP::SystematicSet>& specifiedSystematicsFwdElectrons() const {
      return m_specifiedSystematicsFwdElectrons;
    }
 
    // return all recommendedSystematics
    inline const std::list<CP::SystematicSet>& recommendedSystematicsPhotons() const {
                                                                                      return m_recommendedSystematicsPhotons;
                                                                                                                             }
    inline const std::list<CP::SystematicSet>& recommendedSystematicsElectrons() const {return m_recommendedSystematicsElectrons;}
    inline const std::list<CP::SystematicSet>& recommendedSystematicsFwdElectrons() const {return m_recommendedSystematicsFwdElectrons;}
  protected:
    // specify Systematic
    virtual void specifiedSystematicsPhotons(const std::set<std::string>& specifiedSystematics);
    virtual void specifiedSystematicsElectrons(const std::set<std::string>& specifiedSystematics);
    virtual void specifiedSystematicsFwdElectrons(const std::set<std::string>& specifiedSystematics);
  private:
    std::shared_ptr<top::TopConfig> m_config;
 
    std::list<CP::SystematicSet> m_specifiedSystematicsPhotons;
    std::list<CP::SystematicSet> m_specifiedSystematicsElectrons;
    std::list<CP::SystematicSet> m_specifiedSystematicsFwdElectrons;
 
    std::list<CP::SystematicSet> m_recommendedSystematicsPhotons;
    std::list<CP::SystematicSet> m_recommendedSystematicsElectrons;
    std::list<CP::SystematicSet> m_recommendedSystematicsFwdElectrons;
 
    ToolHandle<CP::IEgammaCalibrationAndSmearingTool> m_calibrationTool;
    ToolHandle<IElectronPhotonShowerShapeFudgeTool> m_photonFudgeTool;
 
    std::unordered_map<std::string, ToolHandle<CP::IIsolationSelectionTool>> m_electronIsolationTools;
    std::unordered_map<std::string, ToolHandle<CP::IIsolationSelectionTool>> m_photonIsolationTools;
 
    ToolHandle<CP::IIsolationCorrectionTool> m_isolationCorr; // TODO not used at the moment -- to test in R22
 
    // Flag for applying calibration to objects
    bool calibrateElectrons;
    bool calibrateFwdElectrons;
    bool calibratePhotons;
 
    // Flag for recomputing CP vars
    bool m_recomputePhotonFudge;
  };
} // namespace
#endif