Belle II Software development
CDCCKFStateCreator Class Referenceabstract

Create CKF states, based on the current path. Perform some basic selection at this stage (based on phi, max. jump of layers) More...

#include <CDCCKFStateCreator.h>

Inheritance diagram for CDCCKFStateCreator:
Findlet< CDCCKFState, const CDCCKFState, const TrackingUtilities::CDCWireHit *const > CompositeProcessingSignalListener ProcessingSignalListener

Classes

struct  CDCCKFWireHitCache
 Store basic wire info for faster access. More...
 

Public Types

using IOTypes
 Types that should be served to apply on invocation.
 
using IOTypes
 Types that should be served to apply on invocation.
 
using IOVectors
 Vector types that should be served to apply on invocation.
 
using IOVectors
 Vector types that should be served to apply on invocation.
 

Public Member Functions

void setMaximalLayerJump (int maximalLayerJump)
 Set maximal layer jump for state creation.
 
void setMaximalLayerJumpBackwardSeed (int maximalLayerJumpBackwardSeed)
 Set maximal layer jump for backward seed tracks.
 
void setMaximalDeltaPhi (double maximalDeltaPhi)
 Set maximal delta phi for state creation.
 
void setHitFindingDirection (const std::string &hitFindingDirection)
 Set hit finding direction.
 
void exposeParameters (ModuleParamList *moduleParamList, const std::string &prefix) override
 Expose the parameters of the sub findlets.
 
void beginEvent () override
 Clear the wireHit cache.
 
void apply (std::vector< CDCCKFState > &nextStates, const CDCCKFPath &path, const std::vector< const TrackingUtilities::CDCWireHit * > &wireHits) override
 Main method of the findlet. Select + create states (output parameter nextStates) suitable for the input path, based on input wireHits.
 
virtual std::string getDescription ()
 Brief description of the purpose of the concrete findlet.
 
virtual std::string getDescription ()
 Brief description of the purpose of the concrete findlet.
 
virtual void apply (ToVector< AIOTypes > &... ioVectors)=0
 Main function executing the algorithm.
 
void initialize () override
 Receive and dispatch signal before the start of the event processing.
 
void beginRun () override
 Receive and dispatch signal for the beginning of a new run.
 
void endRun () override
 Receive and dispatch signal for the end of the run.
 
void terminate () override
 Receive and dispatch Signal for termination of the event processing.
 

Protected Types

using ToVector
 Short hand for ToRangeImpl.
 
using ToVector
 Short hand for ToRangeImpl.
 

Protected Member Functions

void addProcessingSignalListener (ProcessingSignalListener *psl)
 Register a processing signal listener to be notified.
 
int getNProcessingSignalListener ()
 Get the number of currently registered listeners.
 

Private Types

using Super = TrackingUtilities::Findlet<CDCCKFState, const CDCCKFState, const TrackingUtilities::CDCWireHit* const>
 Parent class.
 

Private Attributes

int m_maximalLayerJump = 2
 Maximum allowed step over layers.
 
int m_maximalLayerJump_backwardSeed = 3
 Maximum allowed step over layers (if outside->in CKF) for first step after seed (e.g. ECLShower)
 
double m_maximalDeltaPhi = TMath::Pi() / 8
 Maximal distance in phi between the path last hit/seed and the candidate hit.
 
std::string m_param_writeOutDirectionAsString = "forward"
 Parameter for the direction in which the tracks are built.
 
TrackingUtilities::EForwardBackward m_param_writeOutDirection = TrackingUtilities::EForwardBackward::c_Unknown
 Direction parameter converted from the string parameters.
 
bool doForward = true
 Direction parameter converted to boolean for convenience.
 
std::vector< CDCCKFWireHitCachem_wireHitCache = {}
 Cache to store frequently used information.
 
std::vector< ProcessingSignalListener * > m_subordinaryProcessingSignalListeners
 References to subordinary signal processing listener contained in this findlet.
 
bool m_initialized
 Flag to keep track whether initialization happened before.
 
bool m_terminated
 Flag to keep track whether termination happened before.
 
std::string m_initializedAs
 Name of the type during initialisation.
 

Detailed Description

Create CKF states, based on the current path. Perform some basic selection at this stage (based on phi, max. jump of layers)

Definition at line 35 of file CDCCKFStateCreator.h.

Member Typedef Documentation

◆ IOTypes [1/2]

using IOTypes
inherited

Types that should be served to apply on invocation.

Definition at line 30 of file Findlet.h.

◆ IOTypes [2/2]

using IOTypes
inherited

Types that should be served to apply on invocation.

Definition at line 30 of file Findlet.h.

◆ IOVectors [1/2]

using IOVectors
inherited

Vector types that should be served to apply on invocation.

Definition at line 53 of file Findlet.h.

◆ IOVectors [2/2]

using IOVectors
inherited

Vector types that should be served to apply on invocation.

Definition at line 53 of file Findlet.h.

◆ Super

Parent class.

Definition at line 40 of file CDCCKFStateCreator.h.

◆ ToVector [1/2]

using ToVector
protectedinherited

Short hand for ToRangeImpl.

Definition at line 49 of file Findlet.h.

◆ ToVector [2/2]

using ToVector
protectedinherited

Short hand for ToRangeImpl.

Definition at line 49 of file Findlet.h.

Member Function Documentation

◆ addProcessingSignalListener()

void addProcessingSignalListener ( ProcessingSignalListener * psl)
protectedinherited

Register a processing signal listener to be notified.

Definition at line 53 of file CompositeProcessingSignalListener.cc.

56{
58}
Interface for a minimal algorithm part that wants to expose some parameters to a module.
Definition Findlet.h:26

◆ apply()

void apply ( std::vector< CDCCKFState > & nextStates,
const CDCCKFPath & path,
const std::vector< const TrackingUtilities::CDCWireHit * > & wireHits )
inlineoverride

Main method of the findlet. Select + create states (output parameter nextStates) suitable for the input path, based on input wireHits.

Definition at line 115 of file CDCCKFStateCreator.h.

117 {
118 // TODO: as we do not need any information on the current state (track state) of the path, we could in principle
119 // TODO: precalculate everything in here
120
121 // Create cache over wirehits, if empty:
122 if (m_wireHitCache.empty()) {
123 const size_t nHits = wireHits.size();
124 m_wireHitCache.reserve(nHits);
125 for (auto hitPtr : wireHits) {
126 // to speed things up, don't consider background/taken hits at all (and not just in the loop below).
127 // I can't just remove them from the list, otherwise the relation to the wireHits is broken
128 // so set the layer index to a high number.
129 if (hitPtr->getAutomatonCell().hasBackgroundFlag() || hitPtr->getAutomatonCell().hasTakenFlag()) {
130 m_wireHitCache.push_back(CDCCKFWireHitCache{99999, 0.});
131 } else {
132 m_wireHitCache.push_back(CDCCKFWireHitCache{hitPtr->getWire().getICLayer(), hitPtr->getRefPos2D().Phi()});
133 }
134 }
135 }
136
137 // Cache last-on-the-path state info too:
138 const auto& lastState = path.back();
139 double lastPhi = 0;
140 double lastICLayer = -1;
141 if (lastState.isSeed()) {
142 if (doForward) {
143 lastICLayer = 0;
144 } else {
145 const auto& wireTopology = CDC::CDCWireTopology::getInstance();
146 const auto& wires = wireTopology.getWires();
147 const float maxForwardZ = wires.back().getForwardZ(); // 157.615
148 const float maxBackwardZ = wires.back().getBackwardZ(); // -72.0916
149
150 const ROOT::Math::XYZVector seedPos(lastState.getSeed()->getPositionSeed());
151 const float seedPosZ = seedPos.z();
152
153 if (seedPosZ < maxForwardZ && seedPosZ > maxBackwardZ) {
154 lastICLayer = 56;
155 } else {
156 // do straight extrapolation of seed momentum to CDC outer walls
157 ROOT::Math::XYZVector seedMomZOne(lastState.getSeed()->getMomentumSeed());
158 seedMomZOne = seedMomZOne / seedMomZOne.z();
159 // const float maxZ = seedPosZ > 0 ? maxForwardZ : maxBackwardZ;
160 // const ROOT::Math::XYZVector extrapolatedPos = seedPos - seedMom / seedMom.R() * (seedPosZ - maxZ);
161
162 // find closest iCLayer
163 float minDist = 99999;
164 for (const auto& wire : wires) {
165 const float maxZ = seedPosZ > 0 ? wire.getForwardZ() : wire.getBackwardZ();
166 const ROOT::Math::XYZVector extrapolatedPos = seedPos - seedMomZOne * (seedPosZ - maxZ);
167
168 const auto distance = wire.getDistance(extrapolatedPos);
169 if (distance < minDist) {
170 minDist = distance;
171 lastICLayer = wire.getICLayer();
172 }
173 }
174 B2DEBUG(29, lastICLayer << " (d=" << minDist << ")");
175 }
176 }
177 } else {
178 lastPhi = lastState.getWireHit()->getRefPos2D().Phi();
179 lastICLayer = lastState.getWireHit()->getWire().getICLayer();
180 }
181
182 // Get sorted vector of wireHits on the path for faster search
183 std::vector<const TrackingUtilities::CDCWireHit*> wireHitsOnPath;
184 for (auto const& state : path) {
185 if (! state.isSeed()) {
186 wireHitsOnPath.push_back(state.getWireHit());
187 }
188 }
189 std::sort(wireHitsOnPath.begin(), wireHitsOnPath.end());
190
191 size_t nHits = wireHits.size();
192 for (size_t i = 0; i < nHits; i++) {
193 // adjust direction of loop (minimal speed gain)
194 int idx = doForward ? i : nHits - i - 1;
195
196 const auto iCLayer = m_wireHitCache[idx].icLayer; // wireHit->getWire().getICLayer();
197 if (m_param_writeOutDirection == TrackingUtilities::EForwardBackward::c_Backward && lastState.isSeed()) {
198 if (std::abs(lastICLayer - iCLayer) > m_maximalLayerJump_backwardSeed) {
199 continue;
200 }
201 } else if (std::abs(lastICLayer - iCLayer) > m_maximalLayerJump) {
202 continue;
203 }
204
205 if (! lastState.isSeed()) {
206 double deltaPhi = TrackingUtilities::AngleUtil::normalised(lastPhi - m_wireHitCache[idx].phi);
207 if (fabs(deltaPhi) > m_maximalDeltaPhi) {
208 continue;
209 }
210 }
211
212 const TrackingUtilities::CDCWireHit* wireHit = wireHits[idx];
213
214 if (std::binary_search(wireHitsOnPath.begin(), wireHitsOnPath.end(), wireHit)) {
215 continue;
216 }
217
218 nextStates.emplace_back(wireHit);
219 }
220 }

◆ beginEvent()

void beginEvent ( )
inlineoverridevirtual

Clear the wireHit cache.

Reimplemented from ProcessingSignalListener.

Definition at line 97 of file CDCCKFStateCreator.h.

98 {
99 Super::beginEvent();
100 m_wireHitCache.clear();
101
102 // Determine direction of track building
103 m_param_writeOutDirection = fromString(m_param_writeOutDirectionAsString);
104
105 if (m_param_writeOutDirection == TrackingUtilities::EForwardBackward::c_Forward) {
106 doForward = true;
107 } else if (m_param_writeOutDirection == TrackingUtilities::EForwardBackward::c_Backward) {
108 doForward = false;
109 } else {
110 B2FATAL("CDCCKFStateCreator: No valid direction specified. Please use forward/backward.");
111 }
112 }
TrackingUtilities::EForwardBackward fromString(const std::string &directionString)
Helper function to turn a direction string into a valid forward backward information.

◆ beginRun()

void beginRun ( )
overrideinherited

Receive and dispatch signal for the beginning of a new run.

Definition at line 33 of file CompositeProcessingSignalListener.cc.

24{
27 psl->beginRun();
28 }
29}
void beginRun() override
Receive and dispatch signal for the beginning of a new run.
virtual void beginRun()
Receive signal for the beginning of a new run.

◆ endRun()

void endRun ( )
overrideinherited

Receive and dispatch signal for the end of the run.

Definition at line 39 of file CompositeProcessingSignalListener.cc.

40{
42 psl->endRun();
43 }
45}
void endRun() override
Receive and dispatch signal for the end of the run.
virtual void endRun()
Receive signal for the end of the run.

◆ exposeParameters()

void exposeParameters ( ModuleParamList * moduleParamList,
const std::string & prefix )
inlineoverridevirtual

Expose the parameters of the sub findlets.

Reimplemented from CompositeProcessingSignalListener.

Definition at line 84 of file CDCCKFStateCreator.h.

85 {
86 moduleParamList->addParameter(TrackingUtilities::prefixed(prefix, "maximalLayerJump"),
87 m_maximalLayerJump, "Maximal jump over N layers", m_maximalLayerJump);
88 moduleParamList->addParameter(TrackingUtilities::prefixed(prefix, "maximalLayerJumpBackwardSeed"),
89 m_maximalLayerJump_backwardSeed, "Maximal jump over N layers", m_maximalLayerJump_backwardSeed);
90 moduleParamList->addParameter(TrackingUtilities::prefixed(prefix, "maximalDeltaPhi"),
91 m_maximalDeltaPhi, "Maximal distance in phi between wires for Z=0 plane", m_maximalDeltaPhi);
92 moduleParamList->addParameter(TrackingUtilities::prefixed(prefix, "hitFindingDirection"),
93 m_param_writeOutDirectionAsString, "Start from innermost/outermost CDC layers", m_param_writeOutDirectionAsString);
94 }

◆ getDescription() [1/2]

virtual std::string getDescription ( )
inlinevirtualinherited

Brief description of the purpose of the concrete findlet.

Definition at line 60 of file Findlet.h.

61 {
62 return "(no description)";
63 }

◆ getDescription() [2/2]

virtual std::string getDescription ( )
inlinevirtualinherited

Brief description of the purpose of the concrete findlet.

Definition at line 60 of file Findlet.h.

61 {
62 return "(no description)";
63 }

◆ getNProcessingSignalListener()

int getNProcessingSignalListener ( )
protectedinherited

Get the number of currently registered listeners.

Definition at line 56 of file CompositeProcessingSignalListener.cc.

61{
63}

◆ initialize()

void initialize ( )
overrideinherited

Receive and dispatch signal before the start of the event processing.

Definition at line 30 of file CompositeProcessingSignalListener.cc.

16{
19 psl->initialize();
20 }
21}
void initialize() override
Receive and dispatch signal before the start of the event processing.
virtual void initialize()
Receive signal before the start of the event processing.

◆ setHitFindingDirection()

void setHitFindingDirection ( const std::string & hitFindingDirection)
inline

Set hit finding direction.

Parameters
hitFindingDirectionStart from innermost/outermost CDC layers

Definition at line 78 of file CDCCKFStateCreator.h.

79 {
80 m_param_writeOutDirectionAsString = hitFindingDirection;
81 }

◆ setMaximalDeltaPhi()

void setMaximalDeltaPhi ( double maximalDeltaPhi)
inline

Set maximal delta phi for state creation.

Parameters
maximalDeltaPhiMaximum distance in phi between wires for Z=0 plane

Definition at line 72 of file CDCCKFStateCreator.h.

72{ m_maximalDeltaPhi = maximalDeltaPhi; }

◆ setMaximalLayerJump()

void setMaximalLayerJump ( int maximalLayerJump)
inline

Set maximal layer jump for state creation.

Parameters
maximalLayerJumpMaximum number of layers to jump

Definition at line 57 of file CDCCKFStateCreator.h.

57{ m_maximalLayerJump = maximalLayerJump; }

◆ setMaximalLayerJumpBackwardSeed()

void setMaximalLayerJumpBackwardSeed ( int maximalLayerJumpBackwardSeed)
inline

Set maximal layer jump for backward seed tracks.

Parameters
maximalLayerJumpBackwardSeedMaximum number of layers to jump for backward seeds

Definition at line 63 of file CDCCKFStateCreator.h.

64 {
65 m_maximalLayerJump_backwardSeed = maximalLayerJumpBackwardSeed;
66 }

◆ terminate()

void terminate ( )
overrideinherited

Receive and dispatch Signal for termination of the event processing.

Definition at line 42 of file CompositeProcessingSignalListener.cc.

48{
50 psl->terminate();
51 }
53}
void terminate() override
Receive and dispatch Signal for termination of the event processing.
virtual void terminate()
Receive Signal for termination of the event processing.

Member Data Documentation

◆ doForward

bool doForward = true
private

Direction parameter converted to boolean for convenience.

Definition at line 234 of file CDCCKFStateCreator.h.

◆ m_initialized

bool m_initialized
privateinherited

Flag to keep track whether initialization happened before.

Definition at line 52 of file ProcessingSignalListener.h.

◆ m_initializedAs

std::string m_initializedAs
privateinherited

Name of the type during initialisation.

Definition at line 58 of file ProcessingSignalListener.h.

◆ m_maximalDeltaPhi

double m_maximalDeltaPhi = TMath::Pi() / 8
private

Maximal distance in phi between the path last hit/seed and the candidate hit.

Definition at line 228 of file CDCCKFStateCreator.h.

◆ m_maximalLayerJump

int m_maximalLayerJump = 2
private

Maximum allowed step over layers.

Definition at line 224 of file CDCCKFStateCreator.h.

◆ m_maximalLayerJump_backwardSeed

int m_maximalLayerJump_backwardSeed = 3
private

Maximum allowed step over layers (if outside->in CKF) for first step after seed (e.g. ECLShower)

Definition at line 226 of file CDCCKFStateCreator.h.

◆ m_param_writeOutDirection

TrackingUtilities::EForwardBackward m_param_writeOutDirection = TrackingUtilities::EForwardBackward::c_Unknown
private

Direction parameter converted from the string parameters.

Definition at line 232 of file CDCCKFStateCreator.h.

◆ m_param_writeOutDirectionAsString

std::string m_param_writeOutDirectionAsString = "forward"
private

Parameter for the direction in which the tracks are built.

Definition at line 230 of file CDCCKFStateCreator.h.

◆ m_subordinaryProcessingSignalListeners

std::vector<ProcessingSignalListener*> m_subordinaryProcessingSignalListeners
privateinherited

References to subordinary signal processing listener contained in this findlet.

Definition at line 60 of file CompositeProcessingSignalListener.h.

◆ m_terminated

bool m_terminated
privateinherited

Flag to keep track whether termination happened before.

Definition at line 55 of file ProcessingSignalListener.h.

◆ m_wireHitCache

std::vector<CDCCKFWireHitCache> m_wireHitCache = {}
private

Cache to store frequently used information.

Definition at line 237 of file CDCCKFStateCreator.h.

237{};

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