Class to contain the payload of MVA weightfiles needed for charged particle identification. More...

#include <ChargedPidMVAWeights.h>

Inheritance diagram for ChargedPidMVAWeights:

Public Types
enum class	ChargedPidMVATrainingMode : unsigned int { c_Classification = 0 , c_Multiclass = 1 , c_ECL_Classification = 2 , c_ECL_Multiclass = 3 , c_PSD_Classification = 4 , c_PSD_Multiclass = 5 , c_ECL_PSD_Classification = 6 , c_ECL_PSD_Multiclass = 7 }
	A (strongly-typed) enumerator identifier for each valid MVA training mode. More...

Public Member Functions
	ChargedPidMVAWeights ()
	Default constructor, necessary for ROOT to stream the object.

	ChargedPidMVAWeights (const double &energyUnit, const double &angUnit, const std::string &thetaVarName="clusterTheta", bool implictNaNmasking=false)
	Specialized constructor.

	~ChargedPidMVAWeights ()
	Destructor.

void	setEnergyUnit (const double &unit)
	Set the energy unit to ensure consistency w/ the one used to define the bins grid.

void	setAngularUnit (const double &unit)
	Set the angular unit to ensure consistency w/ the one used to define the bins grid.

void	setWeightCategories (const double clusterThetaBins, const int nClusterThetaBins, const double pBins, const int nPBins, const double *chargeBins, const int nChargeBins)
	Set the 3D (clusterTheta, p, charge) grid representing the categories for which weightfiles are defined.

void	storeMVAWeights (const int pdg, const std::vector< std::string > &filepaths, const std::vector< std::tuple< double, double, double > > &categoryBinCentres)
	Given a particle mass hypothesis' pdgId, store the list of MVA weight files (one for each category) into the payload.

void	storeMVAWeightsMultiClass (const std::vector< std::string > &filepaths, const std::vector< std::tuple< double, double, double > > &categoryBinCentres)
	For the multi-class mode, store the list of MVA weight files (one for each category) into the payload.

void	storeCuts (const int pdg, const std::vector< std::string > &cutfiles, const std::vector< std::tuple< double, double, double > > &categoryBinCentres)
	Given a particle mass hypothesis' pdgId, store the list of selection cuts (one for each category) into the payload.

void	storeCutsMultiClass (const std::vector< std::string > &cutfiles, const std::vector< std::tuple< double, double, double > > &categoryBinCentres)
	For the multi-class mode, store the list of selection cuts (one for each category) into the payload.

void	storeAliases (const VariablesByAlias &aliases)
	Store the map associating variable aliases to variable names knowm to VariableManager.

const TH3F *	getWeightCategories () const
	Get the raw pointer to the 3D grid representing the categories for which weightfiles are defined.

const std::vector< std::string > *	getMVAWeights (const int pdg) const
	Given a particle mass hypothesis' pdgId, get the list of (serialized) MVA weightfiles stored in the payload, one for each category.

const std::vector< std::string > *	getMVAWeightsMulticlass () const
	For the multi-class mode, get the list of (serialized) MVA weightfiles stored in the payload, one for each category.

const std::vector< std::string > *	getCuts (const int pdg) const
	Given a particle mass hypothesis' pdgId, get the list of selection cuts stored in the payload, one for each category.

const std::vector< std::string > *	getCutsMulticlass () const
	For the multi-class mode, get the list of selection cuts stored in the payload, one for each category.

const VariablesByAlias *	getAliases () const
	Get the map of unique aliases.

unsigned int	getMVAWeightIdx (const double &theta, const double &p, const double &charge, int &idx_theta, int &idx_p, int &idx_charge) const
	Get the index of the XML weight file, for a given reconstructed triplet (clusterTheta(theta), p, charge).

unsigned int	getMVAWeightIdx (const double &theta, const double &p, const double &charge) const
	Overloaded method, to be used if not interested in knowing the 3D bin coordinates.

void	dumpPayload (const double &theta, const double &p, const double &charge, const int pdg, bool dump_all=false) const
	Read and dump the payload content from the internal 'matrioska' maps into an XML weightfile for the given set of inputs.

void	dumpPayloadMulticlass (const double &theta, const double &p, const double &charge) const
	Special version for multi-class mode.

bool	isValidPdg (const int pdg) const
	Check if the input pdgId is that of a valid charged particle.

std::string	getThetaVarName () const
	Get the name of the polar angle variable.

bool	hasImplicitNaNmasking () const
	Check flag for implicit NaN masking.

Private Types
typedef std::unordered_map< int, std::vector< std::string > >	WeightfilesByParticle
	Typedef.

typedef std::map< std::string, std::string >	VariablesByAlias
	Typedef.

Private Member Functions
int	findBin (const double &x, const double &y, const double &z) const
	Find global bin index of the 3D categories histogram for the given (x, y, z) values.

	ClassDef (ChargedPidMVAWeights, 10)
	2: add energy/angular units.

Private Attributes
TParameter< double >	m_energy_unit
	The energy unit used for defining the bins grid.

TParameter< double >	m_ang_unit
	The angular unit used for defining the bins grid.

std::string	m_thetaVarName
	The name of the polar angle variable used in the MVA categorisation.

bool	m_implicitNaNmasking
	Flag to indicate whether the MVA variables have been NaN-masked directly in the weightfiles.

std::unique_ptr< TH3F >	m_categories
	A 3D histogram whose bins represent the categories for which XML weight files are defined.

WeightfilesByParticle	m_weightfiles
	For each charged particle mass hypothesis' pdgId, this map contains a list of (serialized) Weightfile objects to be stored in the payload.

WeightfilesByParticle	m_cuts
	For each charged particle mass hypothesis' pdgId, this map contains a list of selection cuts to be stored in the payload.

VariablesByAlias	m_aliases
	A map that associates variable aliases used in the MVA training to variable names known to the VariableManager.

Detailed Description

Class to contain the payload of MVA weightfiles needed for charged particle identification.

Definition at line 34 of file ChargedPidMVAWeights.h.

Member Typedef Documentation

◆ VariablesByAlias

typedef std::map<std::string, std::string> VariablesByAlias

private

Typedef.

Definition at line 37 of file ChargedPidMVAWeights.h.

◆ WeightfilesByParticle

typedef std::unordered_map<int, std::vector<std::string> > WeightfilesByParticle

private

Typedef.

Definition at line 36 of file ChargedPidMVAWeights.h.

Member Enumeration Documentation

◆ ChargedPidMVATrainingMode

enum class ChargedPidMVATrainingMode : unsigned int

strong

A (strongly-typed) enumerator identifier for each valid MVA training mode.

Enumerator
c_Classification	Binary classification.
c_Multiclass	Multi-class classification.
c_ECL_Classification	Binary classification, ECL only.
c_ECL_Multiclass	Multi-class classification, ECL only.
c_PSD_Classification	Binary classification, including PSD.
c_PSD_Multiclass	Multi-class classification, including PSD.
c_ECL_PSD_Classification	Binary classification, ECL only, including PSD.
c_ECL_PSD_Multiclass	Multi-class classification, ECL only, including PSD.

Definition at line 73 of file ChargedPidMVAWeights.h.

                                         : unsigned int {
      c_Classification = 0,
      c_Multiclass = 1,
      c_ECL_Classification = 2,
      c_ECL_Multiclass = 3,
      c_PSD_Classification = 4,
      c_PSD_Multiclass = 5,
      c_ECL_PSD_Classification = 6,
      c_ECL_PSD_Multiclass = 7
    };

Constructor & Destructor Documentation

◆ ChargedPidMVAWeights() [1/2]

ChargedPidMVAWeights ( )

inline

Default constructor, necessary for ROOT to stream the object.

Definition at line 44 of file ChargedPidMVAWeights.h.

                           :
      m_energy_unit("energyUnit", Unit::GeV),
      m_ang_unit("angularUnit", Unit::rad),
      m_thetaVarName("clusterTheta"),
      m_implicitNaNmasking(false)
    {};

◆ ChargedPidMVAWeights() [2/2]

ChargedPidMVAWeights	(	const double &	energyUnit,
		const double &	angUnit,
		const std::string &	thetaVarName = "clusterTheta",
		bool	implictNaNmasking = false )

inline

Specialized constructor.

Definition at line 55 of file ChargedPidMVAWeights.h.

    {
      setEnergyUnit(energyUnit);
      setAngularUnit(angUnit);
      m_thetaVarName = thetaVarName;
      m_implicitNaNmasking = implictNaNmasking;
    }

◆ ~ChargedPidMVAWeights()

~ChargedPidMVAWeights ( )

inline

Destructor.

Definition at line 68 of file ChargedPidMVAWeights.h.

68{};

Member Function Documentation

◆ ClassDef()

ClassDef	(	ChargedPidMVAWeights	,
		10	)

private

2: add energy/angular units.

Add name of polar angle variable used for categorisation, and a boolean flag to check if implicit NaN masking is set in the input data.9. Add map of variable aliases and original basf2 vars.8. Use unique_ptr for m_categories.7. Use double instead of float in tuple.6. Introduce charge bin in the parametrisation.5. remove 2D grid dependence on pdgId, add multi-class support, define enum for valid training modes4. add cuts map.3. add overloaded getMVAWeightIdx. 1: first class implementation.

◆ dumpPayload()

void dumpPayload	(	const double &	theta,
		const double &	p,
		const double &	charge,
		const int	pdg,
		bool	dump_all = false ) const

inline

Read and dump the payload content from the internal 'matrioska' maps into an XML weightfile for the given set of inputs.

Useful for debugging.

Parameters

theta	the particle polar angle (from the cluster, or from the track if no cluster match) in [rad].
p	the particle momentum (from the track) in [GeV/c].
charge	the particle charge (from the track).
pdg	the particle mass hypothesis' pdgId.
dump_all	dump all information.

Definition at line 391 of file ChargedPidMVAWeights.h.

    {
 
      B2INFO("Dumping payload content for:");
      B2INFO("clusterTheta(theta) = " << theta << " [rad], p = " << p << " [GeV/c], charge = " << charge);
 
      if (m_categories) {
        std::string filename = "db_payload_chargedpidmva__theta_p_charge_categories.root";
        B2INFO("\tWriting ROOT file w/ TH3F grid that defines categories:" << filename);
        auto f = std::make_unique<TFile>(filename.c_str(), "RECREATE");
        m_categories->Write();
        f->Close();
      } else {
        B2WARNING("\tThe TH3F object that defines categories is a nullptr!");
      }
 
      for (const auto& [pdgId, weights] : m_weightfiles) {
 
        if (!dump_all && pdg != pdgId) continue;
 
        auto idx = getMVAWeightIdx(theta, p, charge);
 
        auto serialized_weightfile = weights.at(idx);
 
        std::string filename = "db_payload_chargedpidmva__weightfile_pdg_" + std::to_string(pdgId) +
                               "_glob_bin_" + std::to_string(idx + 1) + ".xml";
 
        auto cutstr = getCuts(pdgId)->at(idx);
 
        B2INFO("\tpdgId = " << pdgId);
        B2INFO("\tCut: " << cutstr);
        B2INFO("\tWriting weight file: " << filename);
 
        std::ofstream weightfile;
        weightfile.open(filename.c_str(), std::ios::out);
        weightfile << serialized_weightfile << std::endl;
        weightfile.close();
 
      }
 
    };

◆ dumpPayloadMulticlass()

void dumpPayloadMulticlass	(	const double &	theta,
		const double &	p,
		const double &	charge ) const

inline

Special version for multi-class mode.

Uses the special value of pdg=0 reserved for multi-class mode.

Definition at line 438 of file ChargedPidMVAWeights.h.

    {
      dumpPayload(theta, p, charge, 0);
    }

◆ findBin()

int findBin	(	const double &	x,
		const double &	y,
		const double &	z ) const

inlineprivate

Find global bin index of the 3D categories histogram for the given (x, y, z) values.

This method had to be re-implemented b/c ROOT has no const version of TH1::FindBin() :(

Parameters

x	value along the x axis.
y	value along the y axis.
z	value along the z axis.

Returns: the global linearised bin index.

Definition at line 483 of file ChargedPidMVAWeights.h.

    {
 
      int nbinsx_vis = m_categories->GetXaxis()->GetNbins();
      int nbinsy_vis = m_categories->GetYaxis()->GetNbins();
      int nbinsz_vis = m_categories->GetZaxis()->GetNbins();
 
      double xx = x;
      double yy = y;
      double zz = z;
 
      // If x, y, z are outside of the 3D grid (visible) range, set their value to
      // fall in the last (first) bin before (after) overflow (underflow).
      if (x < m_categories->GetXaxis()->GetBinLowEdge(1)) { xx = m_categories->GetXaxis()->GetBinCenter(1); }
      if (x >= m_categories->GetXaxis()->GetBinLowEdge(nbinsx_vis + 1)) { xx = m_categories->GetXaxis()->GetBinCenter(nbinsx_vis); }
      if (y < m_categories->GetYaxis()->GetBinLowEdge(1)) { yy = m_categories->GetYaxis()->GetBinCenter(1); }
      if (y >= m_categories->GetYaxis()->GetBinLowEdge(nbinsy_vis + 1)) { yy = m_categories->GetYaxis()->GetBinCenter(nbinsy_vis); }
      if (z < m_categories->GetZaxis()->GetBinLowEdge(1)) { zz = m_categories->GetZaxis()->GetBinCenter(1); }
      if (z >= m_categories->GetZaxis()->GetBinLowEdge(nbinsz_vis + 1)) { zz = m_categories->GetZaxis()->GetBinCenter(nbinsz_vis); }
 
      int nbinsx = m_categories->GetXaxis()->GetNbins() + 2;
      int nbinsy = m_categories->GetYaxis()->GetNbins() + 2;
 
      int j = m_categories->GetXaxis()->FindBin(xx);
      int i = m_categories->GetYaxis()->FindBin(yy);
      int k = m_categories->GetZaxis()->FindBin(zz);
 
      return j + nbinsx * (i + nbinsy * k);
    }

◆ getAliases()

const VariablesByAlias * getAliases ( ) const

inline

Get the map of unique aliases.

Definition at line 333 of file ChargedPidMVAWeights.h.

    {
      return &m_aliases;
    }

◆ getCuts()

const std::vector< std::string > * getCuts ( const int pdg ) const

inline

Given a particle mass hypothesis' pdgId, get the list of selection cuts stored in the payload, one for each category.

Parameters

pdg	the particle mass hypothesis' pdgId.

Definition at line 313 of file ChargedPidMVAWeights.h.

    {
      return &(m_cuts.at(pdg));
    }

◆ getCutsMulticlass()

const std::vector< std::string > * getCutsMulticlass ( ) const

inline

For the multi-class mode, get the list of selection cuts stored in the payload, one for each category.

Uses the special value of pdg=0 reserved for multi-class mode.

Definition at line 324 of file ChargedPidMVAWeights.h.

    {
      return getCuts(0);
    }

◆ getMVAWeightIdx() [1/2]

unsigned int getMVAWeightIdx	(	const double &	theta,
		const double &	p,
		const double &	charge ) const

inline

Overloaded method, to be used if not interested in knowing the 3D bin coordinates.

Definition at line 375 of file ChargedPidMVAWeights.h.

    {
      int idx_theta, idx_p, idx_charge;
      return getMVAWeightIdx(theta, p, charge, idx_theta, idx_p, idx_charge);
    }

◆ getMVAWeightIdx() [2/2]

unsigned int getMVAWeightIdx	(	const double &	theta,
		const double &	p,
		const double &	charge,
		int &	idx_theta,
		int &	idx_p,
		int &	idx_charge ) const

inline

Get the index of the XML weight file, for a given reconstructed triplet (clusterTheta(theta), p, charge).

The index is obtained by linearising the 3D m_categories histogram. The same index can be used to look up the correct MVAExpert, Dataset and Cut in the application module, hence we believe it's more useful to return the index rather than a pointer to the weightfile itself. The function also retrieves the 3D bin coordinates.

Parameters

	theta	the particle polar angle (from the cluster, or from the track if no cluster match) in [rad].
	p	the particle momentum (from the track) in [GeV/c].
	charge	the particle charge (from the track).
[out]	idx_theta	the index of the 3D bin along the theta (X) axis.
[out]	idx_p	the index of the 3D bin along the p (Y) axis.
[out]	idx_charge	the index of the 3D bin along the charge (Z) axis.

Returns: the index of the weightfile of interest from the array of weightfiles.

Definition at line 353 of file ChargedPidMVAWeights.h.

    {
 
      if (!m_categories) {
        B2FATAL("No (clusterTheta, p, charge) TH3 grid was found in the DB payload. Most likely, you are using a GT w/ an old payload which is no longer compatible with the DB object class implementation. This should not happen! Abort...");
      }
 
      int nbins_th = m_categories->GetXaxis()->GetNbins(); // nr. of theta (visible) bins, along X.
      int nbins_p = m_categories->GetYaxis()->GetNbins(); // nr. of p (visible) bins, along Y.
 
      int glob_bin_idx = findBin(theta / m_ang_unit.GetVal(), p / m_energy_unit.GetVal(), charge);
      m_categories->GetBinXYZ(glob_bin_idx, idx_theta, idx_p, idx_charge);
 
      // The index of the linearised 3D m_categories.
      // The unit offset is b/c ROOT sets global bin idx also for overflows and underflows.
      return (idx_theta - 1) + nbins_th * ((idx_p - 1) + nbins_p * (idx_charge - 1));
    }

◆ getMVAWeights()

const std::vector< std::string > * getMVAWeights ( const int pdg ) const

inline

Given a particle mass hypothesis' pdgId, get the list of (serialized) MVA weightfiles stored in the payload, one for each category.

Parameters

pdg	the particle mass hypothesis' pdgId.

Definition at line 291 of file ChargedPidMVAWeights.h.

    {
      return &(m_weightfiles.at(pdg));
    }

◆ getMVAWeightsMulticlass()

const std::vector< std::string > * getMVAWeightsMulticlass ( ) const

inline

For the multi-class mode, get the list of (serialized) MVA weightfiles stored in the payload, one for each category.

Uses the special value of pdg=0 reserved for multi-class mode.

Definition at line 302 of file ChargedPidMVAWeights.h.

    {
      return getMVAWeights(0);
    }

◆ getThetaVarName()

std::string getThetaVarName ( ) const

inline

Get the name of the polar angle variable.

Definition at line 457 of file ChargedPidMVAWeights.h.

    {
      return m_thetaVarName;
    }

◆ getWeightCategories()

const TH3F * getWeightCategories ( ) const

inline

Get the raw pointer to the 3D grid representing the categories for which weightfiles are defined.

Used just to view the stored data.

Definition at line 280 of file ChargedPidMVAWeights.h.

    {
      return m_categories.get();
    }

◆ hasImplicitNaNmasking()

bool hasImplicitNaNmasking ( ) const

inline

Check flag for implicit NaN masking.

Definition at line 466 of file ChargedPidMVAWeights.h.

    {
      return m_implicitNaNmasking;
    }

◆ isValidPdg()

bool isValidPdg ( const int pdg ) const

inline

Check if the input pdgId is that of a valid charged particle.

An input value of pdg=0 is considered valid, since it's reserved for multi-class mode.

Definition at line 448 of file ChargedPidMVAWeights.h.

    {
      bool isValid = (Const::chargedStableSet.find(pdg) != Const::invalidParticle) || (pdg == 0);
      return isValid;
    }

◆ setAngularUnit()

void setAngularUnit ( const double & unit )

inline

Set the angular unit to ensure consistency w/ the one used to define the bins grid.

Definition at line 102 of file ChargedPidMVAWeights.h.

102{ m_ang_unit.SetVal(unit); }

◆ setEnergyUnit()

void setEnergyUnit ( const double & unit )

inline

Set the energy unit to ensure consistency w/ the one used to define the bins grid.

Definition at line 96 of file ChargedPidMVAWeights.h.

96{ m_energy_unit.SetVal(unit); }

◆ setWeightCategories()

void setWeightCategories	(	const double *	clusterThetaBins,
		const int	nClusterThetaBins,
		const double *	pBins,
		const int	nPBins,
		const double *	chargeBins,
		const int	nChargeBins )

inline

Set the 3D (clusterTheta, p, charge) grid representing the categories for which weightfiles are defined.

Parameters

clusterThetaBins	array of clusterTheta bin edges
nClusterThetaBins	number of clusterTheta bins
pBins	array of p bin edges
nPBins	number of p bins
chargeBins	array of charge bin edges
nChargeBins	number of charge bins

Definition at line 113 of file ChargedPidMVAWeights.h.

    {
 
      m_categories = std::make_unique<TH3F>("clustertheta_p_charge_binsgrid",
                                            ";ECL cluster #theta;p_{lab};Q",
                                            nClusterThetaBins, clusterThetaBins,
                                            nPBins, pBins,
                                            nChargeBins, chargeBins);
    }

◆ storeAliases()

void storeAliases ( const VariablesByAlias & aliases )

inline

Store the map associating variable aliases to variable names knowm to VariableManager.

Parameters

aliases a map of (alias, VM variable) pairs. NB: it is supposed to contain all the aliases for every category.

Definition at line 270 of file ChargedPidMVAWeights.h.

    {
      m_aliases = VariablesByAlias(aliases);
    }

◆ storeCuts()

void storeCuts	(	const int	pdg,
		const std::vector< std::string > &	cutfiles,
		const std::vector< std::tuple< double, double, double > > &	categoryBinCentres )

inline

Given a particle mass hypothesis' pdgId, store the list of selection cuts (one for each category) into the payload.

Parameters

pdg	the particle mass hypothesis' pdgId.
cutfiles	a list of text files w/ cut strings, for each (clusterTheta, p, charge) category. The format of the cut must comply with the `GeneralCut` syntax.
categoryBinCentres	a list of <double, double, double> representing the (clusterTheta, p, charge) bin centres. Used to check consistency of the xml vector indexing w/ the linearised TH3 category map.

Definition at line 209 of file ChargedPidMVAWeights.h.

    {
 
      if (!isValidPdg(pdg)) {
        B2FATAL("PDG: " << pdg << " is not that of a valid charged particle! Aborting...");
      }
 
      unsigned int idx(0);
      for (const auto& cutfile : cutfiles) {
 
        auto bin_centres_tuple = categoryBinCentres.at(idx);
 
        auto theta_bin_centre = std::get<0>(bin_centres_tuple);
        auto p_bin_centre = std::get<1>(bin_centres_tuple);
        auto charge_bin_centre = std::get<2>(bin_centres_tuple);
 
        auto h_idx = getMVAWeightIdx(theta_bin_centre, p_bin_centre, charge_bin_centre);
        if (idx != h_idx) {
          B2FATAL("Cut file:\n" << cutfile << "\nindex in input vector:\n" << idx << "\ndoes not correspond to:\n" << h_idx <<
                  "\n, i.e. the linearised index of the 3D bin centered in (clusterTheta, p, charge) = (" << theta_bin_centre << ", " << p_bin_centre
                  << ", " <<
                  charge_bin_centre <<
                  ")\nPlease check how the input cut file list is being filled.");
        }
 
        std::ifstream ifs(cutfile);
        std::string cut((std::istreambuf_iterator<char>(ifs)), (std::istreambuf_iterator<char>()));
 
        // Strip trailing newline.
        cut.erase(std::remove(cut.begin(), cut.end(), '\n'), cut.end());
 
        m_cuts[pdg].push_back(cut);
 
        ++idx;
      }
 
    }

◆ storeCutsMultiClass()

void storeCutsMultiClass	(	const std::vector< std::string > &	cutfiles,
		const std::vector< std::tuple< double, double, double > > &	categoryBinCentres )

inline

For the multi-class mode, store the list of selection cuts (one for each category) into the payload.

Uses the special value of pdg=0 reserved for multi-class mode.

Parameters

cutfiles	a list of text files w/ cut strings, for each (clusterTheta, p, charge) category. The format of the cut must comply with the `GeneralCut` syntax.
categoryBinCentres	a list of <double, double, double> representing the (clusterTheta, p, charge) bin centres. Used to check consistency of the xml vector indexing w/ the linearised TH3 category map.

Definition at line 258 of file ChargedPidMVAWeights.h.

    {
      storeCuts(0, cutfiles, categoryBinCentres);
    }

◆ storeMVAWeights()

void storeMVAWeights	(	const int	pdg,
		const std::vector< std::string > &	filepaths,
		const std::vector< std::tuple< double, double, double > > &	categoryBinCentres )

inline

Given a particle mass hypothesis' pdgId, store the list of MVA weight files (one for each category) into the payload.

Parameters

pdg	the particle mass hypothesis' pdgId.
filepaths	a list of xml (root) file paths for several (clusterTheta, p, charge) categories.
categoryBinCentres	a list of <double, double, double> representing the (clusterTheta, p, charge) bin centres. Used to check consistency of the xml vector indexing w/ the linearised TH3 category map.

Definition at line 134 of file ChargedPidMVAWeights.h.

    {
 
      if (!isValidPdg(pdg)) {
        B2FATAL("PDG: " << pdg << " is not that of a valid charged particle! Aborting...");
      }
 
      unsigned int idx(0);
      for (const auto& path : filepaths) {
 
        // Index consistency check.
        auto bin_centres_tuple = categoryBinCentres.at(idx);
 
        auto theta_bin_centre = std::get<0>(bin_centres_tuple);
        auto p_bin_centre = std::get<1>(bin_centres_tuple);
        auto charge_bin_centre = std::get<2>(bin_centres_tuple);
 
        auto h_idx = getMVAWeightIdx(theta_bin_centre, p_bin_centre, charge_bin_centre);
        if (idx != h_idx) {
          B2FATAL("xml file:\n" << path << "\nindex in input vector:\n" << idx << "\ndoes not correspond to:\n" << h_idx <<
                  "\n, i.e. the linearised index of the 3D bin centered in (clusterTheta, p, charge) = (" << theta_bin_centre << ", " << p_bin_centre
                  << ", " <<
                  charge_bin_centre <<
                  ")\nPlease check how the input xml file list is being filled.");
        }
 
        MVA::Weightfile weightfile;
        if (path.ends_with(".root")) {
          weightfile = MVA::Weightfile::loadFromROOTFile(path);
        } else  if (path.ends_with(".xml")) {
          weightfile = MVA::Weightfile::loadFromXMLFile(path);
        } else {
          B2WARNING("Unknown file extension for file: " << path << ", fallback to xml...");
          weightfile = MVA::Weightfile::loadFromXMLFile(path);
        }
 
        // Serialize the MVA::Weightfile object into a string for storage in the database,
        // otherwise there are issues w/ dictionary generation for the payload class...
        std::stringstream ss;
        MVA::Weightfile::saveToStream(weightfile, ss);
        m_weightfiles[pdg].push_back(ss.str());
 
        ++idx;
      }
 
    }

◆ storeMVAWeightsMultiClass()

void storeMVAWeightsMultiClass	(	const std::vector< std::string > &	filepaths,
		const std::vector< std::tuple< double, double, double > > &	categoryBinCentres )

inline

For the multi-class mode, store the list of MVA weight files (one for each category) into the payload.

Uses the special value of pdg=0 reserved for multi-class mode.

Parameters

filepaths	a list of xml (root) file paths for several (clusterTheta, p, charge) categories.
categoryBinCentres	a list of <double, double, double> representing the (clusterTheta, p, charge) bin centres. Used to check consistency of the xml vector indexing w/ the linearised TH3 category map.

Definition at line 192 of file ChargedPidMVAWeights.h.

    {
      storeMVAWeights(0, filepaths, categoryBinCentres);
    }

Member Data Documentation

◆ m_aliases

VariablesByAlias m_aliases

private

A map that associates variable aliases used in the MVA training to variable names known to the VariableManager.

Definition at line 572 of file ChargedPidMVAWeights.h.

◆ m_ang_unit

TParameter<double> m_ang_unit

private

The angular unit used for defining the bins grid.

Definition at line 518 of file ChargedPidMVAWeights.h.

◆ m_categories

std::unique_ptr<TH3F> m_categories

private

A 3D histogram whose bins represent the categories for which XML weight files are defined.

It is used to lookup the correct file in the payload, given a reconstructed set of (clusterTheta(theta), p, charge).

Definition at line 528 of file ChargedPidMVAWeights.h.

◆ m_cuts

WeightfilesByParticle m_cuts

private

Initial value:

= {
      { 0, std::vector<std::string>() },
      { Const::electron.getPDGCode(), std::vector<std::string>() },
      { Const::muon.getPDGCode(), std::vector<std::string>() },
      { Const::pion.getPDGCode(), std::vector<std::string>() },
      { Const::kaon.getPDGCode(), std::vector<std::string>() },
      { Const::proton.getPDGCode(), std::vector<std::string>() },
      { Const::deuteron.getPDGCode(), std::vector<std::string>() }
    }

For each charged particle mass hypothesis' pdgId, this map contains a list of selection cuts to be stored in the payload.

To each Weightfile (i.e., category) corresponds a cut. The indexing in each vector must reflect the one of the corresponding 'linearised' TH3F histogram contained in the m_grids map.

The dummy pdgId=0 key is reserved for multi-class, where a unique signal hypothesis is not defined.

Definition at line 558 of file ChargedPidMVAWeights.h.

                                   {
      { 0, std::vector<std::string>() },
      { Const::electron.getPDGCode(), std::vector<std::string>() },
      { Const::muon.getPDGCode(), std::vector<std::string>() },
      { Const::pion.getPDGCode(), std::vector<std::string>() },
      { Const::kaon.getPDGCode(), std::vector<std::string>() },
      { Const::proton.getPDGCode(), std::vector<std::string>() },
      { Const::deuteron.getPDGCode(), std::vector<std::string>() }
    };

◆ m_energy_unit

TParameter<double> m_energy_unit

private

The energy unit used for defining the bins grid.

Definition at line 517 of file ChargedPidMVAWeights.h.

◆ m_implicitNaNmasking

bool m_implicitNaNmasking

private

Flag to indicate whether the MVA variables have been NaN-masked directly in the weightfiles.

Definition at line 521 of file ChargedPidMVAWeights.h.

◆ m_thetaVarName

std::string m_thetaVarName

private

The name of the polar angle variable used in the MVA categorisation.

Must be a string that can be parsed by the VariableManager.

Definition at line 520 of file ChargedPidMVAWeights.h.

◆ m_weightfiles

WeightfilesByParticle m_weightfiles

private

Initial value:

= {
      { 0, std::vector<std::string>() },
      { Const::electron.getPDGCode(), std::vector<std::string>() },
      { Const::muon.getPDGCode(), std::vector<std::string>() },
      { Const::pion.getPDGCode(), std::vector<std::string>() },
      { Const::kaon.getPDGCode(), std::vector<std::string>() },
      { Const::proton.getPDGCode(), std::vector<std::string>() },
      { Const::deuteron.getPDGCode(), std::vector<std::string>() }
    }

For each charged particle mass hypothesis' pdgId, this map contains a list of (serialized) Weightfile objects to be stored in the payload.

Each weightfile in the list corresponds to a 3D category. The indexing in each vector must reflect the one of the corresponding 'linearised' TH3F histogram contained in the m_grids map.

The dummy pdgId=0 key is reserved for multi-class, where a unique signal hypothesis is not defined.

Definition at line 539 of file ChargedPidMVAWeights.h.

                                          {
      { 0, std::vector<std::string>() },
      { Const::electron.getPDGCode(), std::vector<std::string>() },
      { Const::muon.getPDGCode(), std::vector<std::string>() },
      { Const::pion.getPDGCode(), std::vector<std::string>() },
      { Const::kaon.getPDGCode(), std::vector<std::string>() },
      { Const::proton.getPDGCode(), std::vector<std::string>() },
      { Const::deuteron.getPDGCode(), std::vector<std::string>() }
    };

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

analysis/dbobjects/include/ChargedPidMVAWeights.h

Public Types

Public Member Functions

Private Types

Private Member Functions

Private Attributes

Detailed Description

Member Typedef Documentation

◆ VariablesByAlias

◆ WeightfilesByParticle

Member Enumeration Documentation

◆ ChargedPidMVATrainingMode

Constructor & Destructor Documentation

◆ ChargedPidMVAWeights() [1/2]

◆ ChargedPidMVAWeights() [2/2]

◆ ~ChargedPidMVAWeights()

Member Function Documentation

◆ ClassDef()

◆ dumpPayload()

◆ dumpPayloadMulticlass()

◆ findBin()

◆ getAliases()

◆ getCuts()

◆ getCutsMulticlass()

◆ getMVAWeightIdx() [1/2]

◆ getMVAWeightIdx() [2/2]

◆ getMVAWeights()

◆ getMVAWeightsMulticlass()

◆ getThetaVarName()

◆ getWeightCategories()

◆ hasImplicitNaNmasking()

◆ isValidPdg()

◆ setAngularUnit()

◆ setEnergyUnit()

◆ setWeightCategories()

◆ storeAliases()

◆ storeCuts()

◆ storeCutsMultiClass()

◆ storeMVAWeights()

◆ storeMVAWeightsMultiClass()

Member Data Documentation

◆ m_aliases

◆ m_ang_unit

◆ m_categories

◆ m_cuts

◆ m_energy_unit

◆ m_implicitNaNmasking

◆ m_thetaVarName

◆ m_weightfiles