AsgPhotonEfficiencyCorrectionTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2021 CERN for the benefit of the ATLAS collaboration
*/
 
// Dear emacs, this is -*-c++-*-
 
#ifndef PHOTONEFFICIENCYCORRECTION_ASGPHOTONEFFICIENCYCORRECTIONTOOL
#define PHOTONEFFICIENCYCORRECTION_ASGPHOTONEFFICIENCYCORRECTIONTOOL
 
// STL includes
#include <vector>
#include <string>
#include <fstream>
#include <unordered_map>
 
//xAOD includes
#include "AsgTools/AsgTool.h"
#include "PATInterfaces/ISystematicsTool.h"
#include "PATInterfaces/SystematicRegistry.h"
#include "PATInterfaces/CorrectionCode.h"
#include "ElectronEfficiencyCorrection/TElectronEfficiencyCorrectionTool.h"
#include "EgammaAnalysisInterfaces/IAsgPhotonEfficiencyCorrectionTool.h"
 
#include "xAODEgamma/Egamma.h"
 
class AsgPhotonEfficiencyCorrectionTool
  : virtual public IAsgPhotonEfficiencyCorrectionTool,
    virtual public CP::ISystematicsTool,
            public asg::AsgTool
{
  ASG_TOOL_CLASS3(AsgPhotonEfficiencyCorrectionTool, IAsgPhotonEfficiencyCorrectionTool, CP::ISystematicsTool, CP::IReentrantSystematicsTool )
 
public:
  AsgPhotonEfficiencyCorrectionTool ( const std::string& myname );
 
  virtual ~AsgPhotonEfficiencyCorrectionTool();
 
  virtual StatusCode initialize() override;
 
 
public:
  virtual CP::CorrectionCode getEfficiencyScaleFactor(const xAOD::Egamma& inputObject, double& efficiencyScaleFactor) const override;
  virtual CP::CorrectionCode getEfficiencyScaleFactorError(const xAOD::Egamma& inputObject, double& efficiencyScaleFactorError) const override;
  virtual CP::CorrectionCode applyEfficiencyScaleFactor(xAOD::Egamma& inputObject) const override;
 
 
  virtual bool isAffectedBySystematic( const CP::SystematicVariation& systematic ) const override;
  
  virtual CP::SystematicSet affectingSystematics() const override;
  
  virtual CP::SystematicSet recommendedSystematics() const override;
  
  const CP::SystematicSet& appliedSystematics() const {
    assert (m_appliedSystematics != nullptr);
    return *m_appliedSystematics;
  }
  
  virtual StatusCode applySystematicVariation ( const CP::SystematicSet& systConfig ) override;
 
  StatusCode registerSystematics();
 
  // Private member variables
private:
  typedef Root::TElectronEfficiencyCorrectionTool::Result Result;
  CP::CorrectionCode calculate( const xAOD::Egamma* egam, Result& result ) const;
 
  Root::TElectronEfficiencyCorrectionTool* m_rootTool_unc;
  Root::TElectronEfficiencyCorrectionTool* m_rootTool_con;
  
  std::unordered_map<CP::SystematicSet, CP::SystematicSet> m_systFilter;
  
  CP::SystematicSet* m_appliedSystematics = nullptr;
  
  // The prefix for the systematic name
  std::string m_sysSubstring;
  
  // Get the correction filename from the map
  std::string getFileName(const std::string& isoWP, const std::string& trigWP, bool isConv);
  
  // Set prefix of the corresponding calibration filenames:
  std::string m_file_prefix_ID="efficiencySF.offline.";
  std::string m_file_prefix_ISO="efficiencySF.Isolation.";
  std::string m_file_prefix_Trig="efficiencySF.";
  std::string m_file_prefix_TrigEff="efficiency.";
  
  // Properties
  
  std::string m_corrFileNameConv;
  std::string m_corrFileNameUnconv;
 
  std::string m_resultPrefix;
 
  std::string m_resultName;
 
  int m_dataTypeOverwrite;
  
  std::string m_isoWP;
  
  std::string m_trigger;
  
  std::string m_mapFile;  
 
  // bin boundaries of correction files
  std::map<float, std::vector<float>> m_pteta_bins;
  
  //use RandomRun Number
  bool m_useRandomRunNumber;
  int m_defaultRandomRunNumber;
 
  // remove TRT converted photon for Run-3
  bool m_removeTRTConversion;
 
 
}; // End: class definition
 
 
#endif