Trk::ClusterSplitProbabilityContainer Class Reference

Container to associate Cluster with cluster splitting probabilities. More...

#include <ClusterSplitProbabilityContainer.h>

Collaboration diagram for Trk::ClusterSplitProbabilityContainer:

Classes
struct	ProbabilityInfo

Public Types
using	ClusterIdentifier = unsigned int

Public Member Functions
const ProbabilityInfo &	splitProbability (const PrepRawData *cluster) const
ProbabilityInfo *	getSplitProbability (const PrepRawData *cluster)
ProbabilityInfo &	setSplitInformation (const PrepRawData *cluster, float prob1, float prob2)
void	setTooBigToBeSplit (const PrepRawData *cluster)
const std::unordered_map< ClusterIdentifier, ProbabilityInfo > &	splitProbMap () const

Static Public Member Functions
static const ProbabilityInfo &	getNoSplitProbability ()

Protected Attributes
std::unordered_map< ClusterIdentifier, ProbabilityInfo >	m_splitProbMap

Static Protected Attributes
static const ProbabilityInfo	s_unset

Friends
class	ClusterSplitProbabilityContainerCnv_p1

Detailed Description

Container to associate Cluster with cluster splitting probabilities.

Definition at line 16 of file ClusterSplitProbabilityContainer.h.

Member Typedef Documentation

ClusterIdentifier

using Trk::ClusterSplitProbabilityContainer::ClusterIdentifier = unsigned int

Definition at line 20 of file ClusterSplitProbabilityContainer.h.

Member Function Documentation

getNoSplitProbability()

const ProbabilityInfo & Trk::ClusterSplitProbabilityContainer::getNoSplitProbability ( )

inlinestatic

Definition at line 33 of file ClusterSplitProbabilityContainer.h.

{return  s_unset; }

getSplitProbability()

ProbabilityInfo * Trk::ClusterSplitProbabilityContainer::getSplitProbability ( const PrepRawData * cluster )

inline

Definition at line 40 of file ClusterSplitProbabilityContainer.h.

                                                                      {
         assert(cluster);
         std::unordered_map<ClusterIdentifier, ProbabilityInfo>::iterator iter = m_splitProbMap.find(cluster->getHashAndIndex().hashAndIndex());
         return (iter != m_splitProbMap.end()) ? &iter->second : nullptr;
      }

setSplitInformation()

ProbabilityInfo & Trk::ClusterSplitProbabilityContainer::setSplitInformation ( const PrepRawData * cluster, float prob1, float prob2 )

inline

Definition at line 45 of file ClusterSplitProbabilityContainer.h.

                                                                                                 {
         assert(cluster);
         // @TODO remove the precision clipping, which is currently implemented here to produce identical results wrt. the current version.
         constexpr  int splitmask =  0x7fff;
         prob1 = int(prob1 * splitmask)/(1.*splitmask);
         prob2 = int(prob2 * splitmask)/(1.*splitmask);
         std::pair<std::unordered_map<ClusterIdentifier, ProbabilityInfo>::iterator,bool>
            ret = m_splitProbMap.insert( std::make_pair( cluster->getHashAndIndex().hashAndIndex(), ProbabilityInfo(prob1, prob2) ) );
         if (!ret.second) {
            std::unordered_map<ClusterIdentifier, ProbabilityInfo>::iterator iter = m_splitProbMap.find(cluster->getHashAndIndex().hashAndIndex());
            if (iter == m_splitProbMap.end()) {
               throw std::runtime_error("Failed to set cluster splitting probability.");
            }
            iter->second.updateSplitProbabilities(prob1,prob2);
            return iter->second;
         }
         else {
            return ret.first->second;
         }
      }

setTooBigToBeSplit()

void Trk::ClusterSplitProbabilityContainer::setTooBigToBeSplit ( const PrepRawData * cluster )

inline

Definition at line 65 of file ClusterSplitProbabilityContainer.h.

                                                         {
         assert(cluster);
         m_splitProbMap.insert( std::make_pair( cluster->getHashAndIndex().hashAndIndex(), ProbabilityInfo(-1.,-1.) ) );
      }

splitProbability()

const ProbabilityInfo & Trk::ClusterSplitProbabilityContainer::splitProbability ( const PrepRawData * cluster ) const

inline

Definition at line 35 of file ClusterSplitProbabilityContainer.h.

                                                                               {
         assert(cluster);
         std::unordered_map<ClusterIdentifier, ProbabilityInfo>::const_iterator iter = m_splitProbMap.find(cluster->getHashAndIndex().hashAndIndex());
         return (iter != m_splitProbMap.end()) ? iter->second : s_unset;
      }

splitProbMap()

const std::unordered_map< ClusterIdentifier, ProbabilityInfo > & Trk::ClusterSplitProbabilityContainer::splitProbMap ( ) const

inline

Definition at line 71 of file ClusterSplitProbabilityContainer.h.

     {
       return m_splitProbMap;
     }

Friends And Related Symbol Documentation

ClusterSplitProbabilityContainerCnv_p1

friend class ClusterSplitProbabilityContainerCnv_p1

friend

Definition at line 18 of file ClusterSplitProbabilityContainer.h.

Member Data Documentation

m_splitProbMap

std::unordered_map<ClusterIdentifier, ProbabilityInfo> Trk::ClusterSplitProbabilityContainer::m_splitProbMap

protected

Definition at line 77 of file ClusterSplitProbabilityContainer.h.

s_unset

const Trk::ClusterSplitProbabilityContainer::ProbabilityInfo Trk::ClusterSplitProbabilityContainer::s_unset

staticprotected

Definition at line 78 of file ClusterSplitProbabilityContainer.h.

---

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

• ClusterSplitProbabilityContainer.h
• ClusterSplitProbabilityContainer.cxx