PMGTruthWeightTool.h

Go to the documentation of this file.

/*
   Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef PMGTOOLS_PMGTRUTHWEIGHTTOOL_H
#define PMGTOOLS_PMGTRUTHWEIGHTTOOL_H
 
// EDM include(s):
#include "AsgTools/AsgMetadataTool.h"
#include "xAODTruth/TruthEventContainer.h"
#include "xAODTruth/TruthMetaData.h"
#include "xAODTruth/TruthMetaDataContainer.h"
 
// Interface include(s):
#include "PMGAnalysisInterfaces/IPMGTruthWeightTool.h"
#include "PATInterfaces/SystematicsCache.h"
 
 
namespace PMGTools
{
  class PMGTruthWeightTool : public virtual IPMGTruthWeightTool, public asg::AsgMetadataTool
  {
    ASG_TOOL_CLASS2(PMGTruthWeightTool, PMGTools::IPMGTruthWeightTool, IReentrantSystematicsTool)
 
  public:
    PMGTruthWeightTool(const std::string& name);
 
 
    virtual StatusCode initialize() override;
 
 
 
    virtual const std::vector<std::string>& getWeightNames() const override;
 
    virtual float getWeight(const xAOD::EventInfo* evtInfo, const std::string& weightName) const override;
 
    virtual bool hasWeight(const std::string& weightName) const override;
 
    virtual float getSysWeight(const xAOD::EventInfo* evtInfo, const CP::SystematicSet& sys) const override;
 
    virtual size_t getSysWeightIndex(const CP::SystematicSet& sys) const override;
 
 
 
    virtual CP::SystematicSet affectingSystematics() const override;
 
    virtual CP::SystematicSet recommendedSystematics() const override;
 
 
  protected:
 
    virtual StatusCode beginInputFile() override;
 
 
    StatusCode getMCChannelNumber(uint32_t &mcChannelNumber);
 
    StatusCode loadMetaData();
 
    StatusCode loadPOOLMetaData();
 
    StatusCode validateWeightLocationCaches();
 
    void clearWeightLocationCaches();
 
    std::string m_metaName;
 
    uint32_t m_mcChannelNumber{};
 
    const xAOD::TruthMetaDataContainer* m_metaDataContainer{nullptr};
 
    bool m_useChannelZeroInMetaData{false};
 
    std::vector<std::string> m_weightNames;
 
    std::unordered_map<std::string, size_t> m_weightIndices;
 
    CP::SystematicsCache<std::size_t> m_calibCache {this};
  };
} // namespace PMGTools
 
#endif // PMGTOOLS_PMGTRUTHWEIGHTTOOL_H