CommonDiTauEfficiencyTool.h

Go to the documentation of this file.

 
 
#ifndef TAUANALYSISTOOLS_COMMONDITAUEFFICIENCYTOOL_H
#define TAUANALYSISTOOLS_COMMONDITAUEFFICIENCYTOOL_H
 
 
// Framework include(s):
#include "AsgTools/AsgTool.h"
#include "AsgTools/PropertyWrapper.h"
 
// EDM include(s):
#include "xAODTau/DiTauJet.h"
 
// Local include(s):
#include "TauAnalysisTools/Enums.h"
#include "TauAnalysisTools/IDiTauEfficiencyCorrectionsTool.h"
#include "TauAnalysisTools/HelperFunctions.h"
 
// ROOT include(s):
#include "TFile.h"
#include "TKey.h"
 
namespace TauAnalysisTools
{
double TruthLeadPt(const xAOD::DiTauJet& xDiTau);
double TruthSubleadPt(const xAOD::DiTauJet& xDiTau);
double TruthDeltaR(const xAOD::DiTauJet& xDiTau);
 
class CommonDiTauEfficiencyTool
  : public virtual IDiTauEfficiencyCorrectionsTool
  , public asg::AsgTool 
{
  ASG_TOOL_CLASS( CommonDiTauEfficiencyTool, TauAnalysisTools::IDiTauEfficiencyCorrectionsTool )
 
public:
 
  CommonDiTauEfficiencyTool(const std::string& sName);
 
  ~CommonDiTauEfficiencyTool();
 
  virtual StatusCode initialize();

  virtual CP::CorrectionCode getEfficiencyScaleFactor(const xAOD::DiTauJet& xDiTau, double& dEfficiencyScaleFactor);

  virtual CP::CorrectionCode applyEfficiencyScaleFactor(const xAOD::DiTauJet& xDiTau);

  virtual bool isAffectedBySystematic( const CP::SystematicVariation& systematic ) const;

  virtual CP::SystematicSet affectingSystematics() const;

  virtual CP::SystematicSet recommendedSystematics() const;

  virtual StatusCode applySystematicVariation ( const CP::SystematicSet& sSystematicSet);
 
  std::function<double(const xAOD::DiTauJet& xDiTau)> m_fXDiTau;
  std::function<double(const xAOD::DiTauJet& xDiTau)> m_fYDiTau;
  std::function<double(const xAOD::DiTauJet& xDiTau)> m_fZDiTau; 
 
protected:
 
  void ReadInputs(std::unique_ptr<TFile> &fFile);
  void addHistogramToSFMap(TKey* kKey, const std::string& sKeyName);
 
  typedef std::tuple<TObject*,
          CP::CorrectionCode (*)(const TObject* oObject,
                                 double& dEfficiencyScaleFactor,
                                 double dVars[] ) > tTupleObjectFunc;
  typedef std::map<std::string, tTupleObjectFunc > tSFMAP; 
  // In gcc10 builds, cling gets confused by the type of m_mSF and produces
  // an ugly warning message.  Hide this from cling to suppress that
  // (substitute another unique_ptr so that the class layout remains the same).
#ifdef __CLING__
  std::unique_ptr<int> m_dummy;
#else
  std::unique_ptr< tSFMAP > m_mSF;
#endif
 
  std::unordered_map < CP::SystematicSet, std::string > m_mSystematicSets;
  const CP::SystematicSet* m_sSystematicSet;
  std::map<std::string, std::string> m_mSystematicsHistNames;

  virtual CP::CorrectionCode getValue(const std::string& sHistName,
                              const xAOD::DiTauJet& xDiTau,
                              double& dEfficiencyScaleFactor) const;
 
  static CP::CorrectionCode getValueTH1(const TObject* oObject,
                                        double& dEfficiencyScaleFactor,
                                        double dVars[]
                                        );
  static CP::CorrectionCode getValueTH2(const TObject* oObject,
                                        double& dEfficiencyScaleFactor,
                                        double dVars[]
                                        );

  void generateSystematicSets(); 
  bool m_bSFIsAvailableDiTau;
  bool m_bSFIsAvailableCheckedDiTau;
 
  CP::SystematicSet m_sAffectingSystematics;
  CP::SystematicSet m_sRecommendedSystematics;
 
  Gaudi::Property<std::string> m_sInputFilePath{this, "InputFilePath", ""};
  Gaudi::Property<std::string> m_sWP{this, "WP", ""};
  Gaudi::Property<std::string> m_sVarName{this, "VarName", ""};
 
  std::string m_sInputFileName;
  std::string m_sSFHistName;
 
  TruthMatchedParticleType m_eCheckTruth;
 
};
} // namespace TauAnalysisTools
 
#endif // TAUANALYSISTOOLS_COMMONDITAUEFFICIENCYTOOL_H