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