reconstruction data objects

Collaboration diagram for reconstruction data objects:

Classes
class	KLMMuonIDDNNInputVariable
	KLM MuonID DNN input variables datastore object to store the input variables for retraining KLMMuonIDDNNExpertModule. More...
class	VXDDedxLikelihood
	Container for likelihoods obtained by the VXD dE/dx PID (VXDDedxPIDModule). More...
class	VXDDedxTrack
	Debug output for VXDDedxPID module. More...

Functions
void	addHit (int sid, int layer, int adcCount, double dx, double dEdx)
	Add a single hit to the object.
void	addDedx (int layer, double distance, double dedxValue)
	add dE/dx information for a VXD layer
double	getDedx (Const::EDetector detector) const
	Get dE/dx truncated mean for given detector.
double	getDedxError (Const::EDetector detector) const
	Get the error on the dE/dx truncated mean for given detector.
double	getDedxMean (Const::EDetector detector) const
	Get the dE/dx mean for given detector.
void	clearLogLikelihoods ()
	Clear log likelihoods (set to zero) and reset the counter of added log likelihood values.
void	addLogLikelihoods (const std::vector< const TH2F * > &PDFs, Dedx::Detector detector, bool truncated)
	Calculate and add log likelihoods to array m_vxdLogl.
void	addLogLikelihoods (const std::vector< const TH2F * > &PDFs, double dedxValue, double minPDFValue)
	Calculate and add log likelihoods to array m_vxdLogl.

Detailed Description

Function Documentation

addDedx()

void addDedx	(	int	layer,
		double	distance,
		double	dedxValue
	)

add dE/dx information for a VXD layer

Definition at line 27 of file VXDDedxTrack.cc.

  {
    dedxLayer.push_back(layer);
    dist.push_back(distance);
    dedx.push_back(dedxValue);
    m_length += distance;
  }

addHit()

void addHit	(	int	sid,
		int	layer,
		int	adcCount,
		double	dx,
		double	dEdx
	)

Add a single hit to the object.

Definition at line 18 of file VXDDedxTrack.cc.

  {
    m_sensorID.push_back(sid);
    m_layer.push_back(layer);
    m_adcCount.push_back(adcCount);
    m_dx.push_back(dx);
    m_dEdx.push_back(dEdx);
  }

addLogLikelihoods() [1/2]

void addLogLikelihoods	(	const std::vector< const TH2F * > &	PDFs,
		Dedx::Detector	detector,
		bool	truncated
	)

Calculate and add log likelihoods to array m_vxdLogl.

Parameters

PDFs	PDF's
detector	detector (PXD or SVD)
truncated	if true, use dE/dx truncated mean, otherwise use dE/dx measured in layers

Definition at line 62 of file VXDDedxTrack.cc.

  {
    if (detector != Dedx::c_PXD and detector != Dedx::c_SVD) return;
 
    if (truncated) {
      addLogLikelihoods(PDFs, m_dedxAvgTruncated[detector], 1e-3);
    } else {
      for (size_t i = 0; i < dedx.size(); i++) {
        if (detector == Dedx::c_PXD and std::abs(dedxLayer[i]) > 2) continue;
        if (detector == Dedx::c_SVD and std::abs(dedxLayer[i]) < 3) continue;
        addLogLikelihoods(PDFs, dedx[i], 1e-5);
      }
    }
 
  }

addLogLikelihoods() [2/2]

void addLogLikelihoods ( const std::vector< const TH2F * > &  PDFs, double  dedxValue, double  minPDFValue  )

private

Calculate and add log likelihoods to array m_vxdLogl.

Parameters

PDFs	PDF's
dedxValue	value of dE/dx
minPDFValue	minimal PDF value (used when bins are found empty)

Definition at line 78 of file VXDDedxTrack.cc.

  {
    if (dedxValue <= 0) return;
 
    int binX = PDFs[0]->GetXaxis()->FindFixBin(m_p);
    int binY = PDFs[0]->GetYaxis()->FindFixBin(dedxValue);
    bool inRange = binX > 0 and binX <= PDFs[0]->GetNbinsX() and binY > 0 and binY <= PDFs[0]->GetNbinsY();
    for (unsigned int iPart = 0; iPart < Const::ChargedStable::c_SetSize; iPart++) {
      double pdfValue = 0;
      const auto& pdf = PDFs[iPart];
      if (inRange) {
        pdfValue = const_cast<TH2F*>(pdf)->Interpolate(m_p, dedxValue);
      } else {
        pdfValue = pdf->GetBinContent(binX, binY);
      }
      if (pdfValue != pdfValue) {
        B2ERROR("pdfValue is NAN for a track with p=" << m_p << " and dedx=" << dedxValue);
        clearLogLikelihoods();
        return;
      }
      if (pdfValue <= 0) pdfValue = minPDFValue;
      m_vxdLogl[iPart] += std::log(pdfValue);
    }
 
    m_numAdded++;
  }

clearLogLikelihoods()

void clearLogLikelihoods

(

)

Clear log likelihoods (set to zero) and reset the counter of added log likelihood values.

Definition at line 56 of file VXDDedxTrack.cc.

  {
    for (auto& logl : m_vxdLogl) logl = 0;
    m_numAdded = 0;
  }

getDedx()

double getDedx

(

Const::EDetector

detector

)

const

Get dE/dx truncated mean for given detector.

Definition at line 35 of file VXDDedxTrack.cc.

  {
    if (detector == Const::PXD) return m_dedxAvgTruncated[Dedx::c_PXD];
    if (detector == Const::SVD) return m_dedxAvgTruncated[Dedx::c_SVD];
    return 0.0;
  }

getDedxError()

double getDedxError

(

Const::EDetector

detector

)

const

Get the error on the dE/dx truncated mean for given detector.

Definition at line 42 of file VXDDedxTrack.cc.

  {
    if (detector == Const::PXD) return m_dedxAvgTruncatedErr[Dedx::c_PXD];
    if (detector == Const::SVD) return m_dedxAvgTruncatedErr[Dedx::c_SVD];
    return 0.0;
  }

getDedxMean()

double getDedxMean

(

Const::EDetector

detector

)

const

Get the dE/dx mean for given detector.

Definition at line 49 of file VXDDedxTrack.cc.

  {
    if (detector == Const::PXD) return m_dedxAvg[Dedx::c_PXD];
    if (detector == Const::SVD) return m_dedxAvg[Dedx::c_SVD];
    return 0.0;
  }