8#include <tracking/trackFindingCDC/filters/segmentTriple/MCSegmentTripleFilter.h>
10#include <tracking/trackFindingCDC/mclookup/CDCMCSegment2DLookUp.h>
12#include <tracking/trackFindingCDC/eventdata/tracks/CDCSegmentTriple.h>
13#include <tracking/trackFindingCDC/eventdata/tracks/CDCAxialSegmentPair.h>
15#include <tracking/trackFindingCDC/eventdata/segments/CDCSegment2D.h>
16#include <tracking/trackFindingCDC/eventdata/trajectories/CDCTrajectory3D.h>
17#include <tracking/trackFindingCDC/eventdata/trajectories/CDCTrajectorySZ.h>
19#include <tracking/trackFindingCDC/filters/base/MCSymmetricFilter.icc.h>
21#include <framework/logging/Logger.h>
24using namespace TrackFindingCDC;
36 const std::string& prefix)
49 const CDCAxialSegment2D* ptrStartSegment = segmentTriple.
getStartSegment();
50 const CDCStereoSegment2D* ptrMiddleSegment = segmentTriple.
getMiddleSegment();
51 const CDCAxialSegment2D* ptrEndSegment = segmentTriple.
getEndSegment();
53 const CDCAxialSegment2D& startSegment = *ptrStartSegment;
54 const CDCAxialSegment2D& middleSegment = *ptrMiddleSegment;
55 const CDCAxialSegment2D& endSegment = *ptrEndSegment;
61 if (std::isnan(pairWeight))
return NAN;
66 EForwardBackward startToMiddleFBInfo = mcSegmentLookUp.
areAlignedInMCTrack(ptrStartSegment, ptrMiddleSegment);
67 if (startToMiddleFBInfo == EForwardBackward::c_Invalid)
return NAN;
69 EForwardBackward middleToEndFBInfo = mcSegmentLookUp.
areAlignedInMCTrack(ptrMiddleSegment, ptrEndSegment);
70 if (middleToEndFBInfo == EForwardBackward::c_Invalid)
return NAN;
73 if (startToMiddleFBInfo != middleToEndFBInfo)
return NAN;
76 if ((startToMiddleFBInfo == EForwardBackward::c_Forward and middleToEndFBInfo == EForwardBackward::c_Forward) or
77 (
getAllowReverse() and startToMiddleFBInfo == EForwardBackward::c_Backward and middleToEndFBInfo == EForwardBackward::c_Backward)) {
82 Weight cellWeight = startSegment.size() + middleSegment.size() + endSegment.size();
83 return startToMiddleFBInfo > 0 ? cellWeight : -cellWeight;
99 const CDCAxialSegment2D* ptrStartSegment = segmentTriple.
getStartSegment();
100 if (not ptrStartSegment) {
101 B2WARNING(
"Start segment of segmentTriple is nullptr. Could not set fits.");
The Module parameter list class.
Class representing a pair of reconstructed axial segments in adjacent superlayer.
CDCTrajectory3D getTrajectory3D(const ACDCHitCollection *ptrHits) const
Returns the trajectory of the collection of hits.
EForwardBackward areAlignedInMCTrack(const ACDCHitCollection *ptrFromHits, const ACDCHitCollection *ptrToHits) const
Returns if the second collection of hits follows the first collection of hits in their common Monte C...
Specialisation of the lookup for the truth values of two dimensional segments.
static const CDCMCSegment2DLookUp & getInstance()
Getter for the singletone instance.
Class representing a triple of reconstructed segments in adjacent superlayer.
const CDCStereoSegment2D * getMiddleSegment() const
Getter for the middle stereo segment.
const CDCAxialSegment2D * getEndSegment() const
Getter for the end axial segment.
void setTrajectory3D(const CDCTrajectory3D &trajectory3D) const
Setter for the three dimensional helix trajectory.
CDCTrajectorySZ getTrajectorySZ() const
Getter for the linear trajectory in the sz direction.
const CDCAxialSegment2D * getStartSegment() const
Getter for the start axial segment.
Particle full three dimensional trajectory.
bool isFitted() const
Indicates if the line has been fitted.
void addProcessingSignalListener(ProcessingSignalListener *psl)
MCSymmetric< BaseSegmentTripleFilter > Super
Type of the super class.
MCAxialSegmentPairFilter m_mcAxialSegmentPairFilter
Instance of the cell filter to reject neighborhoods of false cells.
void initialize() final
Initialize the before event processing.
Weight operator()(const CDCSegmentTriple &segmentTriple) final
Check if the segment triple is aligned in the Monte Carlo track. Signals NAN if not.
void setTrajectoryOf(const CDCSegmentTriple &segmentTriple) const
Sets the trajectories of the segment triple from Monte Carlo information. Is executed for good segmen...
void exposeParameters(ModuleParamList *moduleParamList, const std::string &prefix) final
Expose the parameters to a module.
void setAllowReverse(bool allowReverse) override
Setter for the allow reverse parameter.
MCSegmentTripleFilter(bool allowReverse=true)
Constructor initializing the symmetry flag.
Mixin for filters that use Monte Carlo information.
void initialize() override
bool getAllowReverse() const
Abstract base class for different kinds of events.