PIDDetectorWeights Class Reference

Class for handling the PID weights per detector, used to calculate the track helix isolation score per particle. More...

#include <PIDDetectorWeights.h>

Inheritance diagram for PIDDetectorWeights:

Collaboration diagram for PIDDetectorWeights:

Classes
class	WeightsTable
	Nested class acting as a container the per-detector weights. More...

Public Member Functions
	PIDDetectorWeights ()
	Default constructor, necessary for ROOT to stream the object.
	~PIDDetectorWeights ()
	Destructor.
	PIDDetectorWeights (const std::string &weightsCSVFileName)
	Constructor from CSV file of weights. More...
	PIDDetectorWeights (const std::string &treeName, const std::string &weightsROOTFileName)
	Constructor from ROOT file w/ TTree of weights. More...
ROOT::RDataFrame	getWeightsRDF () const
	Get the RDataFrame of detector weights. More...
double	getWeight (Const::ChargedStable hypo, Const::EDetector det, double p, double theta) const
	Lookup the weight from the internal map structures. More...

Private Member Functions
void	fillWeightsTablePerHypoFromRDF ()
	Fill the internal weights container class per particle hypo, based on the content of the RDataFrame.
	ClassDef (PIDDetectorWeights, 4)
	ClassDef as this is a TObject.

Private Attributes
std::map< int, WeightsTable >	m_weightsTablePerHypo
	Map containing a WeightsTable object per particle hypo. More...
ROOT::RDataFrame	m_weightsRDataFrame = ROOT::RDataFrame(1)
	The RDataFrame containing the detector weights per particle hypo, per phase space bin. More...
std::map< std::string, std::string >	m_weightNames
	The names of the per-detector weight columns in the RDataFrame. More...

Detailed Description

Class for handling the PID weights per detector, used to calculate the track helix isolation score per particle.

Definition at line 31 of file PIDDetectorWeights.h.

Constructor & Destructor Documentation

PIDDetectorWeights() [1/2]

PIDDetectorWeights ( const std::string &  weightsCSVFileName )

inline

Constructor from CSV file of weights.

NB: please ensure all numeric types are stored as double in the CSV!

Parameters

weightsCSVFileName

the path to the CSV file conatining the detector weights per std charged particle hypothesis, in bins of p and theta.

Definition at line 88 of file PIDDetectorWeights.h.

    {
      m_weightsRDataFrame = ROOT::RDF::MakeCsvDataFrame(weightsCSVFileName);
      fillWeightsTablePerHypoFromRDF();
    };

PIDDetectorWeights() [2/2]

PIDDetectorWeights ( const std::string &  treeName, const std::string &  weightsROOTFileName  )

inline

Constructor from ROOT file w/ TTree of weights.

Parameters

treeName	the name of the TTree with detector weights per std charged particle hypothesis, in bins of p and theta.
weightsROOTFileName	the path to the ROOT file containing the TTree.

Definition at line 99 of file PIDDetectorWeights.h.

Member Function Documentation

getWeight()

double getWeight	(	Const::ChargedStable	hypo,
		Const::EDetector	det,
		double	p,
		double	theta
	)		const

Lookup the weight from the internal map structures.

Parameters

hypo	the input std charged particle.
det	the input PID detector.
p	the particle momentum in [GeV/c].
theta	the particle polar angle in [rad].

Definition at line 112 of file PIDDetectorWeights.cc.

{
 
  const auto weightsTable = &m_weightsTablePerHypo.at(hypo.getPDGCode());
 
  if (weightsTable->m_isEmpty) {
    // Weights not available for this pdgId.
    return std::numeric_limits<float>::quiet_NaN();
  }
 
  // Get the bin index of the input p value.
  auto pBinUpperEdge_it = std::lower_bound(std::begin(weightsTable->m_pBinEdges),
                                           std::end(weightsTable->m_pBinEdges),
                                           p);
  auto pBinIdx = std::distance(std::begin(weightsTable->m_pBinEdges), pBinUpperEdge_it);
  // Get the bin index of the input theta value.
  auto thetaBinUpperEdge_it = std::lower_bound(std::begin(weightsTable->m_thetaBinEdges),
                                               std::end(weightsTable->m_thetaBinEdges),
                                               theta);
  auto thetaBinIdx = std::distance(std::begin(weightsTable->m_thetaBinEdges), thetaBinUpperEdge_it);
 
  // Get the linearised bin index to look up.
  double linBinIdx = (pBinIdx - 1.0) + (thetaBinIdx - 1.0) * weightsTable->m_nPBins;
 
  if (!weightsTable->m_linBinIdxsToRowIdxs.count(linBinIdx)) {
    // Out-of-bin range p or theta
    B2DEBUG(11, "\n"
            << "p = " << p << " [GeV/c], theta = " << theta << " [rad].\n"
            << "Bin indexes: (" <<  pBinIdx << ", " << thetaBinIdx << ").\n"
            << "Either input value is outside of bin range for PID detector weights:\n"
            << "p : [" << *weightsTable->m_pBinEdges.begin() << ", " << *weightsTable->m_pBinEdges.rbegin() << "],\n"
            << "theta: [" << *weightsTable->m_thetaBinEdges.begin() << ", " << *weightsTable->m_thetaBinEdges.rbegin() << "]");
    return std::numeric_limits<float>::quiet_NaN();
  }
 
  auto rowIdx = weightsTable->m_linBinIdxsToRowIdxs.at(linBinIdx);
 
  return weightsTable->m_weightsPerDet.at(Const::parseDetectors(det)).at(rowIdx);
 
};

getWeightsRDF()

ROOT::RDataFrame getWeightsRDF ( ) const

inline

Get the RDataFrame of detector weights.

To be used for testing/debugging only when creating the payload.

Definition at line 109 of file PIDDetectorWeights.h.

Member Data Documentation

m_weightNames

std::map<std::string, std::string> m_weightNames

private

Initial value:

= {
      {Const::parseDetectors(Const::CDC), "ablat_s_CDC"},
      {Const::parseDetectors(Const::TOP), "ablat_s_TOP"},
      {Const::parseDetectors(Const::ARICH), "ablat_s_ARICH"},
      {Const::parseDetectors(Const::ECL), "ablat_s_ECL"},
      {Const::parseDetectors(Const::KLM), "ablat_s_KLM"}
    }

The names of the per-detector weight columns in the RDataFrame.

Definition at line 141 of file PIDDetectorWeights.h.

m_weightsRDataFrame

ROOT::RDataFrame m_weightsRDataFrame = ROOT::RDataFrame(1)

private

The RDataFrame containing the detector weights per particle hypo, per phase space bin.

Using an RDataFrame is convenient when creating the payload from an input CSV file. However, an RDataFrame cannot be streamed in the persistent ROOT file, thus we must prevent this from happening.

Definition at line 136 of file PIDDetectorWeights.h.

m_weightsTablePerHypo

std::map<int, WeightsTable> m_weightsTablePerHypo

private

Map containing a WeightsTable object per particle hypo.

Use to lookup the index corresponding to a (p, theta) pair.

Definition at line 129 of file PIDDetectorWeights.h.

---

The documentation for this class was generated from the following files:

• analysis/dbobjects/include/PIDDetectorWeights.h
• analysis/dbobjects/src/PIDDetectorWeights.cc