The class for PXD cluster position lookup table payload. More...

#include <PXDClusterPositionEstimatorPar.h>

Inheritance diagram for PXDClusterPositionEstimatorPar:

Collaboration diagram for PXDClusterPositionEstimatorPar:

[legend]

Public Member Functions
	PXDClusterPositionEstimatorPar ()
	Default constructor.

	~PXDClusterPositionEstimatorPar ()
	Destructor.

void	addGrid (int clusterkind, const TH2F &grid)
	Add grid for clusterkind.

const std::map< int, TH2F > &	getGridMap () const
	Return grid map.

void	setShapeClassifier (const PXDClusterShapeClassifierPar &classifier, int uBin, int vBin, int clusterkind)
	Set shape classifier.

const PXDClusterShapeClassifierPar &	getShapeClassifier (int uBin, int vBin, int clusterkind) const
	Returns shape classifier.

const PXDClusterShapeClassifierPar &	getShapeClassifier (double thetaU, double thetaV, int clusterkind) const
	Returns shape classifier for incidence angles and clusterkind.

bool	hasClassifier (double thetaU, double thetaV, int clusterkind) const
	Returns True if there is a classifier available.

const PXDClusterOffsetPar *	getOffset (int shape_index, float eta, double thetaU, double thetaV, int clusterkind) const
	Returns correction (offset) for cluster shape relative to center of pixel (startU/startV) if available, otherwise returns nullptr.

float	getShapeLikelyhood (int shape_index, double thetaU, double thetaV, int clusterkind) const
	Returns shape likelihood.

Private Member Functions
	ClassDef (PXDClusterPositionEstimatorPar, 2)
	ClassDef, must be the last term before the closing {}.

Private Attributes
std::map< int, TH2F >	m_gridmap
	Map of angular grids for different clusterkinds.

std::map< int, std::vector< PXDClusterShapeClassifierPar > >	m_shapeClassifiers
	Map of cluster shape classifiers for different clusterkinds.

Detailed Description

The class for PXD cluster position lookup table payload.

Definition at line 27 of file PXDClusterPositionEstimatorPar.h.

Constructor & Destructor Documentation

◆ PXDClusterPositionEstimatorPar()

PXDClusterPositionEstimatorPar ( )

inline

Default constructor.

Definition at line 30 of file PXDClusterPositionEstimatorPar.h.

30{}

◆ ~PXDClusterPositionEstimatorPar()

~PXDClusterPositionEstimatorPar ( )

inline

Destructor.

Definition at line 33 of file PXDClusterPositionEstimatorPar.h.

33{}

Member Function Documentation

◆ addGrid()

void addGrid	(	int	clusterkind,
		const TH2F &	grid )

inline

Add grid for clusterkind.

Definition at line 36 of file PXDClusterPositionEstimatorPar.h.

    {
      m_gridmap[clusterkind] = grid;
      m_shapeClassifiers[clusterkind] = std::vector<PXDClusterShapeClassifierPar>();
 
      for (auto uBin = 1; uBin <= m_gridmap[clusterkind].GetXaxis()->GetNbins(); uBin++) {
        for (auto vBin = 1; vBin <= m_gridmap[clusterkind].GetYaxis()->GetNbins(); vBin++) {
          auto size = m_shapeClassifiers[clusterkind].size();
          m_gridmap[clusterkind].SetBinContent(uBin, vBin, size);
          m_shapeClassifiers[clusterkind].push_back(PXDClusterShapeClassifierPar());
        }
      }
    }

◆ getGridMap()

const std::map< int, TH2F > & getGridMap ( ) const

inline

Return grid map.

Definition at line 51 of file PXDClusterPositionEstimatorPar.h.

    {
      return m_gridmap;
    }

◆ getOffset()

const PXDClusterOffsetPar * getOffset	(	int	shape_index,
		float	eta,
		double	thetaU,
		double	thetaV,
		int	clusterkind ) const

inline

Returns correction (offset) for cluster shape relative to center of pixel (startU/startV) if available, otherwise returns nullptr.

Definition at line 102 of file PXDClusterPositionEstimatorPar.h.

    {
      //Check if there is a classifier
      if (not hasClassifier(thetaU, thetaV, clusterkind)) {
        return nullptr;
      }
      const PXDClusterShapeClassifierPar& classifier = getShapeClassifier(thetaU, thetaV, clusterkind);
      return classifier.getOffset(shape_index, eta);
    }

◆ getShapeClassifier() [1/2]

const PXDClusterShapeClassifierPar & getShapeClassifier	(	double	thetaU,
		double	thetaV,
		int	clusterkind ) const

inline

Returns shape classifier for incidence angles and clusterkind.

Definition at line 72 of file PXDClusterPositionEstimatorPar.h.

    {
      auto grid = m_gridmap.at(clusterkind);
      int uBin = grid.GetXaxis()->FindBin(thetaU);
      int vBin = grid.GetYaxis()->FindBin(thetaV);
      int key = grid.GetBinContent(uBin, vBin);
      return m_shapeClassifiers.at(clusterkind)[key];
    }

◆ getShapeClassifier() [2/2]

const PXDClusterShapeClassifierPar & getShapeClassifier	(	int	uBin,
		int	vBin,
		int	clusterkind ) const

inline

Returns shape classifier.

Definition at line 64 of file PXDClusterPositionEstimatorPar.h.

    {
      int key = m_gridmap.at(clusterkind).GetBinContent(uBin, vBin);
      const PXDClusterShapeClassifierPar& classifier = m_shapeClassifiers.at(clusterkind)[key];
      return classifier;
    }

◆ getShapeLikelyhood()

float getShapeLikelyhood	(	int	shape_index,
		double	thetaU,
		double	thetaV,
		int	clusterkind ) const

inline

Returns shape likelihood.

Returns zero if shape not known, otherwise positive float

Definition at line 113 of file PXDClusterPositionEstimatorPar.h.

    {
      // Return zero if there no classifier
      if (not hasClassifier(thetaU, thetaV, clusterkind)) {
        return 0;
      }
      auto likelyhoodMap = getShapeClassifier(thetaU, thetaV, clusterkind).getShapeLikelyhoodMap();
 
      // Return zero if no likelihood was estimated
      auto it = likelyhoodMap.find(shape_index);
      if (it == likelyhoodMap.end())
        return 0;
 
      return it->second;
    }

◆ hasClassifier()

bool hasClassifier	(	double	thetaU,
		double	thetaV,
		int	clusterkind ) const

inline

Returns True if there is a classifier available.

Definition at line 82 of file PXDClusterPositionEstimatorPar.h.

    {
      //Check clusterkind is valid
      if (m_gridmap.find(clusterkind) == m_gridmap.end()) {
        return false;
      }
 
      // Check thetaU, thetaV are inside grid
      auto grid = m_gridmap.at(clusterkind);
      int uBin = grid.GetXaxis()->FindBin(thetaU);
      int vBin = grid.GetYaxis()->FindBin(thetaV);
      int uBins = grid.GetXaxis()->GetNbins();
      int vBins = grid.GetYaxis()->GetNbins();
      if ((uBin < 1) || (uBin > uBins) || (vBin < 1) || (vBin > vBins))
        return false;
 
      return true;
    }

◆ setShapeClassifier()

void setShapeClassifier	(	const PXDClusterShapeClassifierPar &	classifier,
		int	uBin,
		int	vBin,
		int	clusterkind )

inline

Set shape classifier.

Definition at line 57 of file PXDClusterPositionEstimatorPar.h.

    {
      int key = m_gridmap[clusterkind].GetBinContent(uBin, vBin);
      m_shapeClassifiers[clusterkind][key] = classifier;
    }

Member Data Documentation

◆ m_gridmap

std::map<int, TH2F> m_gridmap

private

Map of angular grids for different clusterkinds.

Definition at line 132 of file PXDClusterPositionEstimatorPar.h.

◆ m_shapeClassifiers

std::map<int, std::vector<PXDClusterShapeClassifierPar> > m_shapeClassifiers

private

Map of cluster shape classifiers for different clusterkinds.

Definition at line 134 of file PXDClusterPositionEstimatorPar.h.

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

pxd/dbobjects/include/PXDClusterPositionEstimatorPar.h

Public Member Functions

Private Member Functions

Private Attributes

Detailed Description

Constructor & Destructor Documentation

◆ PXDClusterPositionEstimatorPar()

◆ ~PXDClusterPositionEstimatorPar()

Member Function Documentation

◆ addGrid()

◆ getGridMap()

◆ getOffset()

◆ getShapeClassifier() [1/2]

◆ getShapeClassifier() [2/2]

◆ getShapeLikelyhood()

◆ hasClassifier()

◆ setShapeClassifier()

Member Data Documentation

◆ m_gridmap

◆ m_shapeClassifiers