JetUncertaintiesTool.cxx File Reference

#include "JetUncertainties/JetUncertaintiesTool.h"
#include "JetUncertainties/Helpers.h"
#include "JetUncertainties/UncertaintyEnum.h"
#include "JetUncertainties/ConfigHelper.h"
#include "JetUncertainties/ResolutionHelper.h"
#include "JetUncertainties/CorrelationMatrix.h"
#include "JetUncertainties/UncertaintyHistogram.h"
#include "JetUncertainties/ValidityHistogram.h"
#include "JetUncertainties/UncertaintyComponent.h"
#include "JetUncertainties/UncertaintyGroup.h"
#include "JetUncertainties/UncertaintySet.h"
#include "JetUncertainties/PtUncertaintyComponent.h"
#include "JetUncertainties/PtEtaUncertaintyComponent.h"
#include "JetUncertainties/PtLogPtMassForTagSFUncertaintyComponent.h"
#include "JetUncertainties/PtAbsMassUncertaintyComponent.h"
#include "JetUncertainties/PtMassUncertaintyComponent.h"
#include "JetUncertainties/PtAbsMassEtaUncertaintyComponent.h"
#include "JetUncertainties/PtMassEtaUncertaintyComponent.h"
#include "JetUncertainties/ELogMassUncertaintyComponent.h"
#include "JetUncertainties/ELogMassEtaUncertaintyComponent.h"
#include "JetUncertainties/PileupUncertaintyComponent.h"
#include "JetUncertainties/FlavourUncertaintyComponent.h"
#include "JetUncertainties/PerJetFlavourUncertaintyComponent.h"
#include "JetUncertainties/PunchthroughUncertaintyComponent.h"
#include "JetUncertainties/ClosebyUncertaintyComponent.h"
#include "JetUncertainties/CombinedMassUncertaintyComponent.h"
#include "JetUncertainties/LargeRTopologyUncertaintyComponent.h"
#include "xAODTracking/VertexContainer.h"
#include "PATInterfaces/SystematicVariation.h"
#include "PATInterfaces/SystematicRegistry.h"
#include "TString.h"
#include "TFile.h"
#include "TDirectory.h"
#include "TROOT.h"
#include "TEnv.h"
#include "TH2D.h"
#include <unordered_set>

Go to the source code of this file.

Functions
bool	checkScalesSingleVar (const std::set< CompScaleVar::TypeEnum > &varSet, const CompScaleVar::TypeEnum var)
bool JetUncertaintiesTool::getValidity	ATLAS_NOT_THREAD_SAFE (size_t index, const xAOD::Jet &jet, const CompScaleVar::TypeEnum scaleVar) const
bool JetUncertaintiesTool::getValidUncertainty	ATLAS_NOT_THREAD_SAFE (size_t index, double &unc, const xAOD::Jet &jet, const CompScaleVar::TypeEnum scaleVar) const
TH2D *JetUncertaintiesTool::getPtCorrelationMatrix	ATLAS_NOT_THREAD_SAFE (const int numBins, const double minPt, const double maxPt, const double valEta)
TH2D *JetUncertaintiesTool::getPtCorrelationMatrix	ATLAS_NOT_THREAD_SAFE (const int numBins, const double minPt, const double maxPt, const double valEta1, const double valEta2)
CP::CorrectionCode JetUncertaintiesTool::applyCorrection	ATLAS_NOT_THREAD_SAFE (xAOD::Jet &jet) const
CP::CorrectionCode JetUncertaintiesTool::correctedCopy	ATLAS_NOT_THREAD_SAFE (const xAOD::Jet &input, xAOD::Jet *&output) const
CP::CorrectionCode JetUncertaintiesTool::applyContainerCorrection	ATLAS_NOT_THREAD_SAFE (xAOD::JetContainer &inputs) const
const xAOD::EventInfo *JetUncertaintiesTool::getDefaultEventInfo	ATLAS_NOT_THREAD_SAFE () const

Function Documentation

ATLAS_NOT_THREAD_SAFE() [1/8]

const xAOD::EventInfo *JetUncertaintiesTool::getDefaultEventInfo ATLAS_NOT_THREAD_SAFE

(

void

)

const

Definition at line 2328 of file JetUncertaintiesTool.cxx.

{
    // NPV decorator(s)
    static const SG::AuxElement::ConstAccessor<float> accNPV("NPV");
    static const SG::AuxElement::Decorator<float> decNPV("NPV");
 
    // Retrieve the EventInfo object
    static const std::string eiName = "EventInfo";
    const xAOD::EventInfo* ei = nullptr;
    if (evtStore()->retrieve(ei, eiName).isFailure())
    {
        ATH_MSG_ERROR("Failed to retrieve default EventInfo object");
        return nullptr;
    }
 
    // Check if NPV already exists on const EventInfo object, return if so
    if (accNPV.isAvailable(*ei)) {
        return ei;
    }
 
    // NPV doesn't already exist, so calculate it
    const xAOD::VertexContainer* vertices = nullptr;
    if (evtStore()->retrieve(vertices,"PrimaryVertices").isFailure())
    {
        ATH_MSG_ERROR("Failed to retrieve default NPV value from PrimaryVertices");
        return nullptr;
    }
 
    unsigned NPV = 0;
    xAOD::VertexContainer::const_iterator itr;
    for (itr = vertices->begin(); itr != vertices->end(); ++itr)
        if ( (*itr)->nTrackParticles() > 1)
            NPV++;
 
    // Add NPV to the EventInfo object as a decoration.
    decNPV(*ei) = NPV;
 
    // Return the decorated object.
    return ei;
}

ATLAS_NOT_THREAD_SAFE() [2/8]

TH2D *JetUncertaintiesTool::getEtaCorrelationMatrix ATLAS_NOT_THREAD_SAFE	(	const int	numBins,
		const double	minPt,
		const double	maxPt,
		const double	valEta )

Definition at line 2038 of file JetUncertaintiesTool.cxx.

{
    return getPtCorrelationMatrix(numBins,minPt,maxPt,valEta,valEta);
}

ATLAS_NOT_THREAD_SAFE() [3/8]

TH2D *JetUncertaintiesTool::getEtaCorrelationMatrix ATLAS_NOT_THREAD_SAFE	(	const int	numBins,
		const double	minPt,
		const double	maxPt,
		const double	valEta1,
		const double	valEta2 )

Definition at line 2043 of file JetUncertaintiesTool.cxx.

{
    if (!m_isInit)
    {
        ATH_MSG_FATAL("Tool must be initialized before calling getCorrelationMatrix");
        return nullptr;
    }
 
    std::cout << "Creating with max values " << valEta1 << " " << valEta2 << std::endl;
    CorrelationMatrix corrMat(Form("%s_varpt_eta%.2f_eta%.2f",m_name.c_str(),valEta1,valEta2),numBins,minPt*m_energyScale,maxPt*m_energyScale,valEta1,valEta2);
    if (corrMat.initializeForPt(*this).isFailure())
        return nullptr;
    return new TH2D(*corrMat.getMatrix());
}

ATLAS_NOT_THREAD_SAFE() [4/8]

CP::CorrectionCode JetUncertaintiesTool::correctedCopy ATLAS_NOT_THREAD_SAFE	(	const xAOD::Jet &	input,
		xAOD::Jet *&	output ) const

Definition at line 2270 of file JetUncertaintiesTool.cxx.

{
    const xAOD::EventInfo* eInfo = getDefaultEventInfo();
    if (!eInfo) return CP::CorrectionCode::Error;
    return correctedCopy(input,output,*eInfo);
}

ATLAS_NOT_THREAD_SAFE() [5/8]

double JetUncertaintiesTool::getUncertainty ATLAS_NOT_THREAD_SAFE	(	size_t	index,
		const xAOD::Jet &	jet,
		const CompScaleVar::TypeEnum	scaleVar ) const

Definition at line 1666 of file JetUncertaintiesTool.cxx.

{
    const xAOD::EventInfo* eInfo = getDefaultEventInfo();
    if (!eInfo) return false;
    return getValidity(index,jet,*eInfo,scaleVar);
}

ATLAS_NOT_THREAD_SAFE() [6/8]

bool JetUncertaintiesTool::getValidUncertainty ATLAS_NOT_THREAD_SAFE	(	size_t	index,
		double &	unc,
		const xAOD::Jet &	jet,
		const CompScaleVar::TypeEnum	scaleVar ) const

Definition at line 1770 of file JetUncertaintiesTool.cxx.

{
    const xAOD::EventInfo* eInfo = getDefaultEventInfo();
    if (!eInfo) return false;
    return getValidUncertainty(index,unc,jet,*eInfo,scaleVar);
}

ATLAS_NOT_THREAD_SAFE() [7/8]

CP::CorrectionCode JetUncertaintiesTool::applyCorrection ATLAS_NOT_THREAD_SAFE

(

xAOD::Jet &

jet

)

const

Definition at line 2084 of file JetUncertaintiesTool.cxx.

{
    const xAOD::EventInfo* eInfo = getDefaultEventInfo();
    if (!eInfo) return CP::CorrectionCode::Error;
    return applyCorrection(jet,*eInfo);
}

ATLAS_NOT_THREAD_SAFE() [8/8]

CP::CorrectionCode JetUncertaintiesTool::applyContainerCorrection ATLAS_NOT_THREAD_SAFE

(

xAOD::JetContainer &

inputs

)

const

Definition at line 2291 of file JetUncertaintiesTool.cxx.

{
    const xAOD::EventInfo* eInfo = getDefaultEventInfo();
    if (!eInfo) return CP::CorrectionCode::Error;
    return applyContainerCorrection(inputs,*eInfo);
}

checkScalesSingleVar()

bool checkScalesSingleVar	(	const std::set< CompScaleVar::TypeEnum > &	varSet,
		const CompScaleVar::TypeEnum	var )

Definition at line 1571 of file JetUncertaintiesTool.cxx.

{
    return varSet.size() == 1 && *(varSet.begin()) == var;
}