 |
Belle II Software
release-05-01-25
|
16 #include <pxd/dbobjects/PXDClusterOffsetPar.h>
28 class PXDClusterShapeClassifierPar:
public TObject {
40 const PXDClusterOffsetPar*
getOffset(
int shape_index,
float eta)
const
46 return &
m_offsets.at(shape_index)[eta_index];
54 m_offsets[shape_index] = std::vector<PXDClusterOffsetPar>();
64 void addEtaOffset(
int shape_index, PXDClusterOffsetPar& offset) {
m_offsets[shape_index].push_back(offset);}
67 unsigned int getEtaIndex(
int shape_index,
float eta)
const
70 for (
int i = etaPercentiles.size() - 1; i >= 0; --i) {
71 if (eta >= etaPercentiles[i])
88 std::map<int, std::vector<PXDClusterOffsetPar> >
m_offsets;
The class for PXD cluster shape classifier payload.
std::map< int, std::vector< PXDClusterOffsetPar > > m_offsets
Map of position offsets (corrections)
std::map< int, float > m_shape_likelyhoods
Map of shape likelyhoods.
void addShapeLikelyhood(int shape_index, float likelyhood)
Add shape likelyhood.
const std::map< int, std::vector< PXDClusterOffsetPar > > & getOffsetMap() const
Return offset map for position correction.
const std::map< int, std::vector< float > > & getPercentilesMap() const
Return percentiles map for position correction.
std::map< int, std::vector< float > > m_likelyhoods
Map of likelyhoods
void addEtaPercentile(int shape_index, float percentile)
Add eta percentile to shape for position correction.
PXDClusterShapeClassifierPar()
Default constructor.
ClassDef(PXDClusterShapeClassifierPar, 2)
ClassDef, must be the last term before the closing {}.
Abstract base class for different kinds of events.
unsigned int getEtaIndex(int shape_index, float eta) const
Get eta index for position correction.
const std::map< int, float > & getShapeLikelyhoodMap() const
Return shape likelyhood map
void addShape(int shape_index)
Add shape for position correction.
void addEtaOffset(int shape_index, PXDClusterOffsetPar &offset)
Add offset to shape for position correction.
const std::map< int, std::vector< float > > & getLikelyhoodMap() const
Return likelyhood map for position correction.
const PXDClusterOffsetPar * getOffset(int shape_index, float eta) const
Returns position offset if available, otherwise returns nullptr.
void addEtaLikelyhood(int shape_index, float likelyhood)
Add eta likelyhood to shape for position correction.
std::map< int, std::vector< float > > m_percentiles
Map of percentiles