METSystematicsTool.h

Go to the documentation of this file.

 
/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
// METSystematicsTool.h
// Header file for class METSystematicsTool
// Author: Russell Smith <rsmith@cern.ch>
#ifndef METUTILITIES_METSYSTEMATICSTOOL_H
#define METUTILITIES_METSYSTEMATICSTOOL_H
 
#include "METInterface/IMETSystematicsTool.h"
#include "AsgDataHandles/ReadHandleKey.h"
#include "AsgTools/PropertyWrapper.h"
#include "AsgTools/AsgTool.h"
 
#include "ColumnarMode/ColumnarMode.h"
 
#include "PATInterfaces/SystematicsTool.h"
 
#include "TH1.h"
#include "TH2.h"
#include "TH3.h"
#include "TRandom3.h"
 
#include "xAODTracking/VertexContainer.h"
 
#include "xAODMissingET/MissingETContainer.h"
#include "xAODMissingET/MissingETAssociationMap.h"
#include "xAODMissingET/MissingETAssociationHelper.h"
 
#include "boost/thread/tss.hpp"
#include "gtest/gtest_prod.h"
 
namespace ADD_CMODE(met) {
 
  typedef ElementLink<xAOD::IParticleContainer> obj_link_t;
 
  //we create this to avoid calling the xAOD::MissingET constructor
  //since it creates a private store which is slow
  //please forgive the code duplication :)
  struct missingEt {
    missingEt() : mpx(0), mpy(0), sumet(0), name(""), source(static_cast<MissingETBase::Types::bitmask_t>(MissingETBase::Source::Type::UnknownType)){}
    missingEt(double mpx_in, double mpy_in, double sumet_in) :
      mpx(mpx_in),
      mpy(mpy_in),
      sumet(sumet_in),
      name(""),
      source(static_cast<MissingETBase::Types::bitmask_t>(MissingETBase::Source::Type::UnknownType))
    {}
    missingEt(double mpx_in, double mpy_in, double sumet_in, std::string const & iname, MissingETBase::Types::bitmask_t const & isource ) :
      mpx(mpx_in),
      mpy(mpy_in),
      sumet(sumet_in),
      name(iname),
      source(isource)
    {}
 
    double mpx;
    double mpy;
    double sumet;
    std::string name;
    MissingETBase::Types::bitmask_t  source;
  };
 
  //this is to speed up applyCorrection lookup.
  //we still store the full list in appliedSystematics()/m_appliedSystematics
  enum SystApplied {
    NONE              ,
    MET_SOFTTRK_SCALEUP   ,
    MET_SOFTTRK_SCALEDOWN ,
    MET_SOFTTRK_RESOPARA  ,
    MET_SOFTTRK_RESOPERP  ,
    MET_SOFTTRK_RESOCORR  ,
    MET_SOFTCALO_SCALEUP  ,
    MET_SOFTCALO_SCALEDOWN,
    MET_SOFTCALO_RESOPARA  ,
    MET_SOFTCALO_RESOPERP  ,
    MET_SOFTCALO_RESOCORR  ,
    MET_JETTRK_SCALEUP    ,
    MET_JETTRK_SCALEDOWN
  };
 
  class METSystematicsTool : public virtual IMETSystematicsTool,
                 public asg::AsgTool,
                 public CP::SystematicsTool
  {
    // This macro defines the constructor with the interface declaration
    ASG_TOOL_CLASS(METSystematicsTool, IMETSystematicsTool)
 
    // FRIEND_TEST( METSystTest, TestGetEventInfo );
    FRIEND_TEST( METSystTest, TestDefaultHistosFilled );
    FRIEND_TEST( METSystTest, TestJetTrkHistosFilled );
    FRIEND_TEST( METSystTest, TestAddAffectingSystematic );
    FRIEND_TEST( METSystTest, TestProjectST );
    FRIEND_TEST( METSystTest, TestProjectST2 );
 
    public:
 
    //Constructor
    METSystematicsTool(const std::string& name);
 
    //Destructor
    // virtual ~METSystematicsTool();
 
    StatusCode softTrkSystInitialize(); //initialize softTrk scale/reo histos from config file
    StatusCode softCaloSystInitialize();//initialize softCalo scale/reso histos from config file
    StatusCode jetTrkSystInitialize();  //initialize jetTrk scale/reso histos from config file
 
    //required  asg tool interface functions
    StatusCode  initialize();
 
    //required correction tool functions
    //we don't inherit from CorrectionTool directly, since we don't want to implement applyContainerCorrection
    CP::CorrectionCode applyCorrection(xAOD::MissingET& inputMet,
                       const xAOD::MissingETAssociationHelper& helper) const;
    CP::CorrectionCode correctedCopy(const xAOD::MissingET& met, xAOD::MissingET*& outputmet,
                     const xAOD::MissingETAssociationHelper& helper) const;
    //We don't want these for MET since we only apply systematics to the soft term, and this may be unclear
    //virtual CP::CorrectionCode applyContainerCorrection(xAOD::MissingETContainer& inputs, const CP::xAOD::EventInfo& eInfo) const;
    //virtual CP::CorrectionCode applyContainerCorrection(xAOD::MissingETContainer& inputs, const CP::xAOD::EventInfo& eInfo) const;
 
    //required ISystematicTool functions
    bool               isAffectedBySystematic  (const CP::SystematicVariation& var) const{return CP::SystematicsTool::isAffectedBySystematic(var)   ;}
    CP::SystematicSet  affectingSystematics    () const{                      return CP::SystematicsTool::affectingSystematics  ()      ;}
    CP::SystematicSet  recommendedSystematics  () const{                      return CP::SystematicsTool::recommendedSystematics()      ;}
    StatusCode applySystematicVariation(const CP::SystematicSet& set){        return CP::SystematicsTool::applySystematicVariation(set) ;}
    StatusCode sysApplySystematicVariation(const CP::SystematicSet&); //when inheriting from SystematicsTool, we should only have to implement this one
 
    void setRandomSeed(unsigned long seed) const;
 
  private:
 
    //default constructor
    METSystematicsTool();
 
    //this saves the the currently applied systematic as an enum for faster lookup
    SystApplied m_appliedSystEnum{NONE};
 
    //these are the internal computation functions
    CP::CorrectionCode internalSoftTermApplyCorrection(xAOD::MissingET& softMet,
                               xAOD::MissingETContainer const * METcont,
                               xAOD::EventInfo          const & eInfo
                               ) const;
    CP::CorrectionCode calcJetTrackMETWithSyst(xAOD::MissingET& jettrkmet, const xAOD::MissingETAssociationHelper& helper) const;
    CP::CorrectionCode calcJetTrackMETWithSyst(xAOD::MissingET& jettrkmet, const xAOD::MissingETAssociationHelper& helper, const xAOD::Jet* jet) const;
    CP::CorrectionCode getCorrectedJetTrackMET(xAOD::MissingET& jettrkmet, const xAOD::MissingETAssociationHelper& helper) const;
    TRandom3* getTLSRandomGen() const;
 
    //declared properties
    Gaudi::Property<std::string> m_configPrefix{this, "ConfigPrefix", "METUtilities/R22_PreRecs", ""};
    Gaudi::Property<std::string> m_configSoftTrkFile{this, "ConfigSoftTrkFile", "TrackSoftTerms-pflow_Dec24.config", ""};
    Gaudi::Property<std::string> m_configJetTrkFile{this, "ConfigJetTrkFile", "", ""};
    Gaudi::Property<std::string> m_configSoftCaloFile{this, "ConfigSoftCaloFile", "", ""};
    Gaudi::Property<bool> m_useDevArea{this, "UseDevArea", false, ""};
 
    std::unique_ptr<TH3D> m_trk_shiftpara_pthard_njet_mu{};
    std::unique_ptr<TH3D> m_trk_resopara_pthard_njet_mu{};
    std::unique_ptr<TH3D> m_trk_resoperp_pthard_njet_mu{};
    std::unique_ptr<TH3D> m_calo_shiftpara_pthard_njet_mu{};
    std::unique_ptr<TH3D> m_calo_resopara_pthard_njet_mu{};
    std::unique_ptr<TH3D> m_calo_resoperp_pthard_njet_mu{};
    std::unique_ptr<TH2D> m_jet_systRpt_pt_eta{};
 
    mutable boost::thread_specific_ptr<TRandom3> m_rand_tls; // thread-specific random number generator
 
    //internal property
    int m_units{-1};//by default = -1, not set.  Set to 1 (MeV) if not value found in config file
 
    int getNPV() const;
    xAOD::EventInfo const * getDefaultEventInfo() const;
 
 
    //Read/write handles
    SG::ReadHandleKey<xAOD::VertexContainer> m_VertexContKey{this, "VertexContainer", "PrimaryVertices", ""};
    SG::ReadHandleKey<xAOD::MissingETContainer> m_TruthContKey{this, "TruthContainer", "MET_Truth", ""};
    SG::ReadHandleKey<xAOD::EventInfo> m_EventInfoKey{this, "EventInfo", "EventInfo", ""};
 
    StatusCode addMETAffectingSystematics();
    StatusCode extractHistoPath(std::string & histfile, std::string & systpath, std::string & configdir, std::string & suffix, met::SystType const & type);
 
    missingEt calcPtHard(xAOD::MissingETContainer const * const cont) const;
    missingEt variedSoftTerm (missingEt const & softTerms, missingEt const & ptHard, double varPara, double varPerp) const;
    missingEt projectST     (missingEt const & softTerms, missingEt const & ptHard) const;
 
 
 
  };//METSystematicsTool
 
}
 
#endif //METUTILIES_METSYSTEMATICSTOOL_H
 
//  LocalWords:  METSystematicsTool