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< TrackingUtilities::CDCRecoHit3D, const TrackingUtilities::CDCRLWireHit * >, FirstOfPairInBox< HitInQuadraticBox >, 2, 2 > Z0TanLambdaLegendre< std::pair< TrackingUtilities::CDCRecoHit3D, const TrackingUtilities::CDCRLWireHit * >, FirstOfPairInBox< HitInZ0TanLambdaBox >, 2, 2 > Z0TanLambdaLegendre< std::pair< TrackingUtilities::CDCRecoHit3D, const TrackingUtilities::CDCRLWireHit * >, FirstOfPairInBox< HitInZ0TanLambdaBoxUsingZ >, 2, 2 > Z0TanLambdaLegendre< std::pair< std::pair< CDCSegment3D, TrackingUtilities::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 TrackingUtilities::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 78 of file BoxDivisionHoughTree.h.

◆ Arrays

using Arrays
privateinherited

Tuple type of the discrete value arrays.

Definition at line 81 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 129 of file BoxDivisionHoughTree.h.

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

◆ 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 158 of file BoxDivisionHoughTree.h.

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

◆ 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 105 of file BoxDivisionHoughTree.h.

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

◆ constructHoughPlane()

HoughBox constructHoughPlane ( )
inlineprivateinherited

Construct the box of the top node of the tree.

Definition at line 246 of file BoxDivisionHoughTree.h.

247 {
248 return constructHoughPlaneImpl(std::make_index_sequence<sizeof...(divisions)>());
249 }

◆ 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 240 of file BoxDivisionHoughTree.h.

241 {
242 return HoughBox(Type<Is>::getRange(std::get<Is>(m_arrays))...);
243 }

◆ fell()

void fell ( )
inlineinherited

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

Definition at line 182 of file BoxDivisionHoughTree.h.

183 {
184 HoughTree* const tree = this->m_houghTree.get();
185 if (tree != nullptr) {
186 tree->fell();
187 }
188 }

◆ findSingleBest()

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

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

Definition at line 59 of file SimpleBoxDivisionHoughTree.h.

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

◆ getArray()

const Array< I > & getArray ( ) const
inlineinherited

Getter for the array of discrete value for coordinate I.

Definition at line 232 of file BoxDivisionHoughTree.h.

233 {
234 return std::get<I>(m_arrays);
235 }

◆ getDivision()

size_t getDivision ( size_t i) const
inlineinherited

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

Definition at line 85 of file BoxDivisionHoughTree.h.

86 {
87 return m_divisions[i];
88 }

◆ 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 76 of file SimpleBoxDivisionHoughTree.h.

77 {
78 return m_maximumX;
79 }

◆ 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 82 of file SimpleBoxDivisionHoughTree.h.

83 {
84 return m_maximumY;
85 }

◆ getMaxLevel()

int getMaxLevel ( ) const
inlineinherited

Getter for the currently set maximal level.

Definition at line 207 of file BoxDivisionHoughTree.h.

208 {
209 return m_maxLevel;
210 }

◆ 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 88 of file SimpleBoxDivisionHoughTree.h.

89 {
90 return m_overlapX;
91 }

◆ 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 94 of file SimpleBoxDivisionHoughTree.h.

95 {
96 return m_overlapY;
97 }

◆ 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 219 of file BoxDivisionHoughTree.h.

220 {
221 return m_sectorLevelSkip;
222 }

◆ getTree()

HoughTree * getTree ( ) const
inlineinherited

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

Definition at line 201 of file BoxDivisionHoughTree.h.

202 {
203 return m_houghTree.get();
204 }

◆ 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 49 of file SimpleBoxDivisionHoughTree.h.

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

◆ raze()

void raze ( )
inlineinherited

Release all memory that the tree acquired during the runs.

Definition at line 191 of file BoxDivisionHoughTree.h.

192 {
193 HoughTree* const tree = this->m_houghTree.get();
194 if (tree != nullptr) {
195 tree->raze();
196 }
197 }

◆ seed()

void seed ( const AItemPtrs & items)
inlineinherited

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

Definition at line 175 of file BoxDivisionHoughTree.h.

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

◆ setMaxLevel()

void setMaxLevel ( int maxLevel)
inlineinherited

Setter maximal level of the hough tree.

Definition at line 213 of file BoxDivisionHoughTree.h.

214 {
215 m_maxLevel = maxLevel;
216 }

◆ 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 225 of file BoxDivisionHoughTree.h.

226 {
227 m_sectorLevelSkip = sectorLevelSkip;
228 }

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 268 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 253 of file BoxDivisionHoughTree.h.

253{{divisions ...}};

◆ m_houghTree

std::unique_ptr<HoughTree> m_houghTree
privateinherited

Dynamic hough tree structure traversed in the leaf search.

Definition at line 256 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 101 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 104 of file SimpleBoxDivisionHoughTree.h.

◆ m_maxLevel

int m_maxLevel
privateinherited

Number of the maximum tree level.

Definition at line 259 of file BoxDivisionHoughTree.h.

◆ m_overlaps

HoughBox::Delta m_overlaps
privateinherited

An tuple of division overlaps in each coordinate.

Definition at line 265 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 107 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 110 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 262 of file BoxDivisionHoughTree.h.


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