Belle II Software development
SimpleBoxDivisionHoughTree< AHitPtr, AInBoxAlgorithm, divisionX, divisionY > Class Template Reference

Convenience class for the typical usage-case: A box divisioned hough tree with maximum and minimum values in both directions. More...

#include <SimpleBoxDivisionHoughTree.h>

Inheritance diagram for SimpleBoxDivisionHoughTree< AHitPtr, AInBoxAlgorithm, divisionX, divisionY >:
BoxDivisionHoughTree< AHitPtr, AInBoxAlgorithm::HoughBox, divisionX, divisionY > DebugableSimpleBoxDivisionHoughTree< AHitPtr, AInBoxAlgorithm, divisionX, divisionY > QuadraticLegendre< std::pair< CDCRecoHit3D, const CDCRLWireHit * >, FirstOfPairInBox< HitInQuadraticBox >, 2, 2 > Z0TanLambdaLegendre< std::pair< CDCRecoHit3D, const CDCRLWireHit * >, FirstOfPairInBox< HitInZ0TanLambdaBox >, 2, 2 > Z0TanLambdaLegendre< std::pair< CDCRecoHit3D, const CDCRLWireHit * >, FirstOfPairInBox< HitInZ0TanLambdaBoxUsingZ >, 2, 2 > Z0TanLambdaLegendre< std::pair< std::pair< CDCSegment3D, CDCTrajectorySZ >, const CDCSegment2D * >, SegmentInZ0TanLambdaAlgorithm, 2, 2 >

Public Types

using BoxDivision
 Type of the box division strategy.
 
using HoughTree
 Type of the fast hough tree structure.
 
using Type
 Type of the coordinate I.
 
using HasType
 Predicate that the given type is indeed a coordinate of the hough space.
 
using TypeIndex
 Function to get the coordinate index from its type.
 
using Node
 Type of the nodes used in the tree for the search.
 

Public Member Functions

 SimpleBoxDivisionHoughTree (float maximumX, float maximumY, Width< 0 > overlapX=0, Width< 1 > overlapY=0)
 Constructor using the given maximal level.
 
void initialize ()
 Initialize the tree with the given values.
 
std::vector< std::pair< HoughBox, std::vector< AHitPtr > > > findSingleBest (const Weight &minWeight)
 Find only the leave with the highest weight = number of items.
 
float getMaximumX () const
 Return the maximum value in x direction.
 
float getMaximumY () const
 Return the maximum value in y direction.
 
Width< 0 > getOverlapX () const
 Return the overlap in x direction.
 
Width< 1 > getOverlapY () const
 Return the overlap in y direction.
 
size_t getDivision (size_t i) const
 Getter the number of divisions at each level for coordinate index I.
 
void constructArray (double lowerBound, double upperBound, Width< I > nBinOverlap=0, Width< I > nBinWidth=0)
 Construct the discrete value array at coordinate index I.
 
void assignArray (Array< I > array, Width< I > overlap=0)
 Provide an externally constructed array by coordinate index.
 
std::enable_if_t< HasType< T >::value, void > assignArray (Array< TypeIndex< T >::value > array, Width< TypeIndex< T >::value > overlap=0)
 Provide an externally constructed array by coordinate type.
 
void seed (const AItemPtrs &items)
 Prepare the leave finding by filling the top node with given hits.
 
void fell ()
 Terminates the processing by striping all hit information from the tree.
 
void raze ()
 Release all memory that the tree acquired during the runs.
 
HoughTreegetTree () const
 Getter for the tree used in the search in the hough plane.
 
int getMaxLevel () const
 Getter for the currently set maximal level.
 
void setMaxLevel (int maxLevel)
 Setter maximal level of the hough tree.
 
int getSectorLevelSkip () const
 Getter for number of levels to skip in first level to form a finer sectored hough space.
 
void setSectorLevelSkip (int sectorLevelSkip)
 Setter for number of levels to skip in first level to form a finer sectored hough space.
 
const Array< I > & getArray () const
 Getter for the array of discrete value for coordinate I.
 

Private Types

using Super = BoxDivisionHoughTree<AHitPtr, typename AInBoxAlgorithm::HoughBox, divisionX, divisionY>
 The Super class.
 
using HoughBox = typename AInBoxAlgorithm::HoughBox
 The HoughBox we use.
 
template<size_t I>
using Width = typename HoughBox::template Width<I>
 Type of the width in coordinate I.
 
using Array
 Type of the discrete value array to coordinate index I.
 
using Arrays
 Tuple type of the discrete value arrays.
 

Private Member Functions

HoughBox constructHoughPlaneImpl (const std::index_sequence< Is... > &is)
 Construct the box of the top node of the tree. Implementation unroling the indices.
 
HoughBox constructHoughPlane ()
 Construct the box of the top node of the tree.
 

Private Attributes

float m_maximumX = 0
 The maximum value in X direction.
 
float m_maximumY = 0
 The maximum value in y direction.
 
Width< 0 > m_overlapX = 0
 The overlap in X direction.
 
Width< 1 > m_overlapY = 0
 The overlap in Y direction.
 
const std::array< size_t, sizeof ...(divisions)> m_divisions
 Array of the number of divisions at each level.
 
std::unique_ptr< HoughTreem_houghTree
 Dynamic hough tree structure traversed in the leaf search.
 
int m_maxLevel
 Number of the maximum tree level.
 
int m_sectorLevelSkip
 Number of levels to skip in first level to form a finer sectored hough space.
 
HoughBox::Delta m_overlaps
 An tuple of division overlaps in each coordinate.
 
Arrays m_arrays
 A tuple of value arrays providing the memory for the discrete bin bounds.
 

Detailed Description

template<class AHitPtr, class AInBoxAlgorithm, size_t divisionX, size_t divisionY>
class Belle2::TrackFindingCDC::SimpleBoxDivisionHoughTree< AHitPtr, AInBoxAlgorithm, divisionX, divisionY >

Convenience class for the typical usage-case: A box divisioned hough tree with maximum and minimum values in both directions.

Definition at line 20 of file SimpleBoxDivisionHoughTree.h.

Member Typedef Documentation

◆ Array

using Array
privateinherited

Type of the discrete value array to coordinate index I.

Definition at line 77 of file BoxDivisionHoughTree.h.

◆ Arrays

using Arrays
privateinherited

Tuple type of the discrete value arrays.

Definition at line 80 of file BoxDivisionHoughTree.h.

◆ BoxDivision

using BoxDivision
inherited

Type of the box division strategy.

Definition at line 41 of file BoxDivisionHoughTree.h.

◆ HasType

using HasType
inherited

Predicate that the given type is indeed a coordinate of the hough space.

Definition at line 52 of file BoxDivisionHoughTree.h.

◆ HoughBox

template<class AHitPtr, class AInBoxAlgorithm, size_t divisionX, size_t divisionY>
using HoughBox = typename AInBoxAlgorithm::HoughBox
private

The HoughBox we use.

Definition at line 27 of file SimpleBoxDivisionHoughTree.h.

◆ HoughTree

using HoughTree
inherited

Type of the fast hough tree structure.

Definition at line 44 of file BoxDivisionHoughTree.h.

◆ Node

using Node
inherited

Type of the nodes used in the tree for the search.

Definition at line 63 of file BoxDivisionHoughTree.h.

◆ Super

template<class AHitPtr, class AInBoxAlgorithm, size_t divisionX, size_t divisionY>
using Super = BoxDivisionHoughTree<AHitPtr, typename AInBoxAlgorithm::HoughBox, divisionX, divisionY>
private

The Super class.

Definition at line 24 of file SimpleBoxDivisionHoughTree.h.

◆ Type

using Type
inherited

Type of the coordinate I.

Definition at line 48 of file BoxDivisionHoughTree.h.

◆ TypeIndex

using TypeIndex
inherited

Function to get the coordinate index from its type.

Definition at line 56 of file BoxDivisionHoughTree.h.

◆ Width

template<class AHitPtr, class AInBoxAlgorithm, size_t divisionX, size_t divisionY>
template<size_t I>
using Width = typename HoughBox::template Width<I>
private

Type of the width in coordinate I.

Definition at line 31 of file SimpleBoxDivisionHoughTree.h.

Constructor & Destructor Documentation

◆ SimpleBoxDivisionHoughTree()

template<class AHitPtr, class AInBoxAlgorithm, size_t divisionX, size_t divisionY>
SimpleBoxDivisionHoughTree ( float maximumX,
float maximumY,
Width< 0 > overlapX = 0,
Width< 1 > overlapY = 0 )
inline

Constructor using the given maximal level.

Definition at line 35 of file SimpleBoxDivisionHoughTree.h.

39 : Super(0)
40 , m_maximumX(maximumX)
41 , m_maximumY(maximumY)
42 , m_overlapX(overlapX)
43 , m_overlapY(overlapY)
44 {
45 }

Member Function Documentation

◆ assignArray() [1/2]

void assignArray ( Array< I > array,
Width< I > overlap = 0 )
inlineinherited

Provide an externally constructed array by coordinate index.

Definition at line 128 of file BoxDivisionHoughTree.h.

129 {
130 std::get<I>(m_arrays) = std::move(array);
131 std::get<I>(m_overlaps) = overlap;
132
133 // In case of a discrete axes, check whether the size of the array is sensible
134 // such that the bin width at the highest granularity level is a whole number given the overlap.
135 if (std::is_integral<Width<I> >::value) {
136 const int division = getDivision(I);
137 const int granularityLevel = m_maxLevel + m_sectorLevelSkip;
138 const long int nBins = std::pow(division, granularityLevel);
139 const long int nPositions = std::get<I>(m_arrays).size();
140 const long int nWidthTimeNBins = nPositions - 1 + (nBins - 1) * overlap;
141
142 B2ASSERT("Not enough positions in array to cover all bins.\n"
143 "Expected: positions = " << nBins - (nBins - 1) * overlap + 1 << " at least.\n"
144 "Actual: positions = " << nPositions << " (overlap = " << overlap << ", bins = " << nBins << ")\n",
145 nWidthTimeNBins >= nBins);
146
147 B2ASSERT("Number of positions in array introduces inhomogeneous width at the highest granularity level.\n"
148 "Expected: positions = 'width * bins - (bins - 1) * overlap + 1'\n"
149 "Actual: positions = " << nPositions << " (overlap = " << overlap << ", bins = " << nBins << ")\n",
150 nWidthTimeNBins % nBins == 0);
151 }
152 }

◆ assignArray() [2/2]

std::enable_if_t< HasType< T >::value, void > assignArray ( Array< TypeIndex< T >::value > array,
Width< TypeIndex< T >::value > overlap = 0 )
inlineinherited

Provide an externally constructed array by coordinate type.

Definition at line 157 of file BoxDivisionHoughTree.h.

158 {
159 assignArray<TypeIndex<T>::value>(std::move(array), overlap);
160 }

◆ constructArray()

void constructArray ( double lowerBound,
double upperBound,
Width< I > nBinOverlap = 0,
Width< I > nBinWidth = 0 )
inlineinherited

Construct the discrete value array at coordinate index I.

This function is only applicable for discrete axes. For continuous axes assignArray should be call with an array containing only the lower and upper bound of the axes range and an optional overlap.

Parameters
lowerBoundLower bound of the value range
upperBoundUpper bound of the value range
nBinOverlapOverlap of neighboring bins. Default is no overlap. Usually this is counted in number of discrete values
nBinWidthWidth of the bins at lowest level. Default is width of 1. Usually this is counted in numbers of discrete values

Definition at line 104 of file BoxDivisionHoughTree.h.

108 {
109 static_assert(std::is_integral<Width<I> >::value, "Method only applicable to discrete axes");
110 const size_t division = getDivision(I);
111 const int granularityLevel = m_maxLevel + m_sectorLevelSkip;
112 const size_t nBins = std::pow(division, granularityLevel);
113
114 if (nBinWidth == 0) {
115 nBinWidth = nBinOverlap + 1;
116 }
117
118 B2ASSERT("Width " << nBinWidth << "is not bigger than overlap " << nBinOverlap,
119 nBinOverlap < nBinWidth);
120
121 const auto nPositions = (nBinWidth - nBinOverlap) * nBins + nBinOverlap + 1;
122 std::get<I>(m_arrays) = linspace<float>(lowerBound, upperBound, nPositions);
123 std::get<I>(m_overlaps) = nBinOverlap;
124 }

◆ constructHoughPlane()

HoughBox constructHoughPlane ( )
inlineprivateinherited

Construct the box of the top node of the tree.

Definition at line 239 of file BoxDivisionHoughTree.h.

240 {
241 return constructHoughPlaneImpl(std::make_index_sequence<sizeof...(divisions)>());
242 }

◆ constructHoughPlaneImpl()

HoughBox constructHoughPlaneImpl ( const std::index_sequence< Is... > & is)
inlineprivateinherited

Construct the box of the top node of the tree. Implementation unroling the indices.

Definition at line 233 of file BoxDivisionHoughTree.h.

234 {
235 return HoughBox(Type<Is>::getRange(std::get<Is>(m_arrays))...);
236 }

◆ fell()

void fell ( )
inlineinherited

Terminates the processing by striping all hit information from the tree.

Definition at line 181 of file BoxDivisionHoughTree.h.

182 {
183 m_houghTree->fell();
184 }

◆ findSingleBest()

template<class AHitPtr, class AInBoxAlgorithm, size_t divisionX, size_t divisionY>
std::vector< std::pair< HoughBox, std::vector< AHitPtr > > > findSingleBest ( const Weight & minWeight)
inline

Find only the leave with the highest weight = number of items.

Definition at line 58 of file SimpleBoxDivisionHoughTree.h.

59 {
60 AInBoxAlgorithm inBoxAlgorithm;
61 auto skipLowWeightNode = [minWeight](const typename Super::Node * node) {
62 return not(node->getWeight() >= minWeight);
63 };
64 auto found = this->getTree()->findHeaviestLeafSingle(inBoxAlgorithm, this->getMaxLevel(), skipLowWeightNode);
65
66 std::vector<std::pair<HoughBox, std::vector<AHitPtr>>> result;
67 if (found) {
68 // Move the found content over. unique_ptr still destroys the left overs.
69 result.push_back(std::move(*found));
70 }
71 return result;
72 }

◆ getArray()

const Array< I > & getArray ( ) const
inlineinherited

Getter for the array of discrete value for coordinate I.

Definition at line 225 of file BoxDivisionHoughTree.h.

226 {
227 return std::get<I>(m_arrays);
228 }

◆ getDivision()

size_t getDivision ( size_t i) const
inlineinherited

Getter the number of divisions at each level for coordinate index I.

Definition at line 84 of file BoxDivisionHoughTree.h.

85 {
86 return m_divisions[i];
87 }

◆ getMaximumX()

template<class AHitPtr, class AInBoxAlgorithm, size_t divisionX, size_t divisionY>
float getMaximumX ( ) const
inline

Return the maximum value in x direction.

Definition at line 75 of file SimpleBoxDivisionHoughTree.h.

76 {
77 return m_maximumX;
78 }

◆ getMaximumY()

template<class AHitPtr, class AInBoxAlgorithm, size_t divisionX, size_t divisionY>
float getMaximumY ( ) const
inline

Return the maximum value in y direction.

Definition at line 81 of file SimpleBoxDivisionHoughTree.h.

82 {
83 return m_maximumY;
84 }

◆ getMaxLevel()

int getMaxLevel ( ) const
inlineinherited

Getter for the currently set maximal level.

Definition at line 200 of file BoxDivisionHoughTree.h.

201 {
202 return m_maxLevel;
203 }

◆ getOverlapX()

template<class AHitPtr, class AInBoxAlgorithm, size_t divisionX, size_t divisionY>
Width< 0 > getOverlapX ( ) const
inline

Return the overlap in x direction.

Definition at line 87 of file SimpleBoxDivisionHoughTree.h.

88 {
89 return m_overlapX;
90 }

◆ getOverlapY()

template<class AHitPtr, class AInBoxAlgorithm, size_t divisionX, size_t divisionY>
Width< 1 > getOverlapY ( ) const
inline

Return the overlap in y direction.

Definition at line 93 of file SimpleBoxDivisionHoughTree.h.

94 {
95 return m_overlapY;
96 }

◆ getSectorLevelSkip()

int getSectorLevelSkip ( ) const
inlineinherited

Getter for number of levels to skip in first level to form a finer sectored hough space.

Definition at line 212 of file BoxDivisionHoughTree.h.

213 {
214 return m_sectorLevelSkip;
215 }

◆ getTree()

HoughTree * getTree ( ) const
inlineinherited

Getter for the tree used in the search in the hough plane.

Definition at line 194 of file BoxDivisionHoughTree.h.

195 {
196 return m_houghTree.get();
197 }

◆ initialize()

template<class AHitPtr, class AInBoxAlgorithm, size_t divisionX, size_t divisionY>
void initialize ( )
inline

Initialize the tree with the given values.

Definition at line 48 of file SimpleBoxDivisionHoughTree.h.

49 {
50 Super::template constructArray<0>(-getMaximumX(), getMaximumX(), getOverlapX());
51 Super::template constructArray<1>(-getMaximumY(), getMaximumY(), getOverlapY());
52
53 Super::initialize();
54 }

◆ raze()

void raze ( )
inlineinherited

Release all memory that the tree acquired during the runs.

Definition at line 187 of file BoxDivisionHoughTree.h.

188 {
189 m_houghTree->raze();
190 }

◆ seed()

void seed ( const AItemPtrs & items)
inlineinherited

Prepare the leave finding by filling the top node with given hits.

Definition at line 174 of file BoxDivisionHoughTree.h.

175 {
176 if (not m_houghTree) initialize();
177 m_houghTree->seed(items);
178 }

◆ setMaxLevel()

void setMaxLevel ( int maxLevel)
inlineinherited

Setter maximal level of the hough tree.

Definition at line 206 of file BoxDivisionHoughTree.h.

207 {
208 m_maxLevel = maxLevel;
209 }

◆ setSectorLevelSkip()

void setSectorLevelSkip ( int sectorLevelSkip)
inlineinherited

Setter for number of levels to skip in first level to form a finer sectored hough space.

Definition at line 218 of file BoxDivisionHoughTree.h.

219 {
220 m_sectorLevelSkip = sectorLevelSkip;
221 }

Member Data Documentation

◆ m_arrays

Arrays m_arrays
privateinherited

A tuple of value arrays providing the memory for the discrete bin bounds.

Definition at line 261 of file BoxDivisionHoughTree.h.

◆ m_divisions

const std::array<size_t, sizeof ...(divisions)> m_divisions
privateinherited

Array of the number of divisions at each level.

Definition at line 246 of file BoxDivisionHoughTree.h.

246{{divisions ...}};

◆ m_houghTree

std::unique_ptr<HoughTree> m_houghTree
privateinherited

Dynamic hough tree structure traversed in the leaf search.

Definition at line 249 of file BoxDivisionHoughTree.h.

◆ m_maximumX

template<class AHitPtr, class AInBoxAlgorithm, size_t divisionX, size_t divisionY>
float m_maximumX = 0
private

The maximum value in X direction.

Definition at line 100 of file SimpleBoxDivisionHoughTree.h.

◆ m_maximumY

template<class AHitPtr, class AInBoxAlgorithm, size_t divisionX, size_t divisionY>
float m_maximumY = 0
private

The maximum value in y direction.

Definition at line 103 of file SimpleBoxDivisionHoughTree.h.

◆ m_maxLevel

int m_maxLevel
privateinherited

Number of the maximum tree level.

Definition at line 252 of file BoxDivisionHoughTree.h.

◆ m_overlaps

HoughBox::Delta m_overlaps
privateinherited

An tuple of division overlaps in each coordinate.

Definition at line 258 of file BoxDivisionHoughTree.h.

◆ m_overlapX

template<class AHitPtr, class AInBoxAlgorithm, size_t divisionX, size_t divisionY>
Width<0> m_overlapX = 0
private

The overlap in X direction.

Definition at line 106 of file SimpleBoxDivisionHoughTree.h.

◆ m_overlapY

template<class AHitPtr, class AInBoxAlgorithm, size_t divisionX, size_t divisionY>
Width<1> m_overlapY = 0
private

The overlap in Y direction.

Definition at line 109 of file SimpleBoxDivisionHoughTree.h.

◆ m_sectorLevelSkip

int m_sectorLevelSkip
privateinherited

Number of levels to skip in first level to form a finer sectored hough space.

Definition at line 255 of file BoxDivisionHoughTree.h.


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