Test the MVA-based charged PID. More...

Inheritance diagram for ChargedParticleIdentificatorTest:

Collaboration diagram for ChargedParticleIdentificatorTest:

[legend]

Public Attributes
ChargedPidMVAWeights	m_dbrep
	Database representation of MVA weightfiles.

Const::ChargedStable	m_testHypo = Const::electron
	The signal charged particle hypothesis to test.

std::vector< double >	m_thetabins = {0.2164208, 0.5480334, 0.561996, 2.2462387, 2.2811453, 2.7070057}
	The clusterTheta bin edges in [rad].

std::vector< double >	m_pbins = {0.2, 0.6, 0.75, 1.0, 3.0, 7.0}
	The p bin edges in [GeV/c].

std::vector< double >	m_chbins = { -1.5, -0.5, 0.5, 1.5}
	The charge bin edges.

std::string	m_basename = "dummy_weightfile"
	Base common name for all dummy weight files.

std::vector< std::string >	m_dummyfiles
	List of dummy xml file names.

Protected Member Functions
void	SetUp () override
	Prepare resources for the tests.

void	TearDown () override
	Release all resources.

Detailed Description

Test the MVA-based charged PID.

Definition at line 27 of file chargedParticleIdentificator.cc.

Member Function Documentation

◆ SetUp()

void SetUp ( )

inlineoverrideprotected

Prepare resources for the tests.

Definition at line 73 of file chargedParticleIdentificator.cc.

    {
 
      m_dbrep.setWeightCategories(m_thetabins.data(), m_thetabins.size() - 1,
                                  m_pbins.data(), m_pbins.size() - 1,
                                  m_chbins.data(), m_chbins.size() - 1);
 
      std::vector<std::tuple<double, double, double>> gridBinCentres;
 
      for (unsigned int kch(0); kch < m_chbins.size() - 1; kch++) {
        auto ch_bin_centre = (m_chbins.at(kch) + m_chbins.at(kch + 1)) / 2.0;
        for (unsigned int ip(0); ip < m_pbins.size() - 1; ip++) {
          auto p_bin_centre = (m_pbins.at(ip) + m_pbins.at(ip + 1)) / 2.0;
          for (unsigned int jth(0); jth < m_thetabins.size() - 1; jth++) {
            auto th_bin_centre = (m_thetabins.at(jth) + m_thetabins.at(jth + 1)) / 2.0;
            auto fname = m_basename
                         + "__clusterTheta__" + std::to_string(m_thetabins.at(jth)) + "_" + std::to_string(m_thetabins.at(jth + 1))
                         + "__p__" + std::to_string(m_pbins.at(ip)) + "_" + std::to_string(m_pbins.at(ip + 1))
                         + "__charge__" + std::to_string(ch_bin_centre);
 
            std::replace(fname.begin(), fname.end(), '.', '_');
            fname += ".xml";
            std::ofstream dummyfile(fname);
            dummyfile.close();
            m_dummyfiles.push_back(fname);
 
            auto centre = std::make_tuple(th_bin_centre, p_bin_centre, ch_bin_centre);
            gridBinCentres.push_back(centre);
          }
        }
      }
      m_dbrep.storeMVAWeights(m_testHypo.getPDGCode(), m_dummyfiles, gridBinCentres);
 
    }

◆ TearDown()

void TearDown ( )

inlineoverrideprotected

Release all resources.

Definition at line 111 of file chargedParticleIdentificator.cc.

    {
 
      // Delete all dummy files.
      for (const auto& fname : m_dummyfiles) {
        if (remove(fname.c_str())) {
          B2ERROR("Couldn't remove file: " << fname);
        }
      }
 
    }