PMGHFProductionFractionTool.h

Go to the documentation of this file.

/*
   Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef PMGTOOLS_PMGHFPRODUCTIONFRACTIONTOOL_H
#define PMGTOOLS_PMGHFPRODUCTIONFRACTIONTOOL_H
 
// EDM include(s):
#include "AsgTools/AsgMetadataTool.h"
#include "PATInterfaces/SystematicsCache.h"
 
// Interface include(s):
#include "PMGAnalysisInterfaces/ISysTruthWeightTool.h"
 
#include <map>
 
namespace PMGTools
{
class PMGHFProductionFractionTool : public virtual ISysTruthWeightTool, public asg::AsgMetadataTool
{
  ASG_TOOL_CLASS2(PMGHFProductionFractionTool, PMGTools::ISysTruthWeightTool, IReentrantSystematicsTool)
 
public:
  PMGHFProductionFractionTool(const std::string &name);
 
 
  virtual StatusCode initialize() override;
 
 
 
  virtual float getSysWeight(const xAOD::TruthParticleContainer *truthParticles, const CP::SystematicSet &sys) const override;
 
  virtual CP::SystematicSet affectingSystematics() const override;
 
  virtual CP::SystematicSet recommendedSystematics() const override { return affectingSystematics(); };
 
 
private:
  struct ParameterSet {
      std::map<unsigned int, float> charmWeights;
      std::map<unsigned int, float> bottomWeights;
  };
 
  StatusCode setupProductionFractions();
 
  StatusCode readProductionFractionsFile(const std::string&, std::map<CP::SystematicVariation, std::map<unsigned int, float>> &);
 
  bool fromBdecay(const xAOD::TruthParticle *particle) const;
 
  const xAOD::TruthParticle *getInitialParticle(const xAOD::TruthParticle *tp) const;
 
  void printCurrentProdFractions() const;
 
  float getWeight(const xAOD::TruthParticleContainer *truthParticles, const ParameterSet &prod_fracs) const;
 
  std::string m_calibrationAreaPath;
 
  std::map<std::string, std::string> m_showerGeneratorMap;
 
  std::string m_showerGeneratorMapFile;
 
  std::string m_showerGenerator;
 
  std::string m_charmFilename;
 
  std::string m_bottomFilename;
 
  float m_fiducialPtCut;
 
  float m_fiducialEtaCut;
 
  std::map<CP::SystematicVariation, std::map<unsigned int, float>> m_charmProdFractionWeights;
 
  std::map<CP::SystematicVariation, std::map<unsigned int, float>> m_bottomProdFractionWeights;
 
  StatusCode setSystematicVariation(const CP::SystematicSet& systConfig, ParameterSet& param) const;
 
  CP::SystematicsCache<ParameterSet> m_Parameters{this};
};
}  // namespace PMGTools
 
#endif  // PMGTOOLS_PMGHFProductionFractionTool_H