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