Belle II Software development
PXDStateBasicVarSet.cc
1/**************************************************************************
2 * basf2 (Belle II Analysis Software Framework) *
3 * Author: The Belle II Collaboration *
4 * *
5 * See git log for contributors and copyright holders. *
6 * This file is licensed under LGPL-3.0, see LICENSE.md. *
7 **************************************************************************/
8
9#include <tracking/ckf/pxd/filters/states/PXDStateBasicVarSet.h>
10
11#include <tracking/trackFindingCDC/eventdata/trajectories/CDCTrajectory3D.h>
12#include <tracking/trackFindingCDC/eventdata/trajectories/CDCTrajectory2D.h>
13#include <tracking/trackFindingCDC/eventdata/trajectories/CDCTrajectorySZ.h>
14#include <tracking/trackFindingCDC/geometry/Vector3D.h>
15
16#include <tracking/spacePointCreation/SpacePoint.h>
17#include <tracking/dataobjects/RecoTrack.h>
18
19#include <pxd/dataobjects/PXDCluster.h>
20
21using namespace Belle2;
22using namespace TrackFindingCDC;
23
25{
26 const std::vector<TrackFindingCDC::WithWeight<const CKFToPXDState*>>& previousStates = pair->first;
27 CKFToPXDState* state = pair->second;
28
29 const RecoTrack* seedTrack = previousStates.front()->getSeed();
30 B2ASSERT("Path without seed?", seedTrack);
31
32 const SpacePoint* spacePoint = state->getHit();
33 B2ASSERT("Path without hit?", spacePoint);
34
35 genfit::MeasuredStateOnPlane firstMeasurement;
36 if (state->mSoPSet()) {
37 firstMeasurement = state->getMeasuredStateOnPlane();
38 } else {
39 firstMeasurement = previousStates.back()->getMeasuredStateOnPlane();
40 }
41
42 Vector3D position = Vector3D(firstMeasurement.getPos());
43 Vector3D momentum = Vector3D(firstMeasurement.getMom());
44
45 const CDCTrajectory3D trajectory(position, 0, momentum, seedTrack->getChargeSeed());
46
47 const Vector3D& hitPosition = static_cast<Vector3D>(spacePoint->getPosition());
48
49 const double arcLength = trajectory.calcArcLength2D(hitPosition);
50 const Vector2D& trackPositionAtHit2D = trajectory.getTrajectory2D().getPos2DAtArcLength2D(arcLength);
51 double trackPositionAtHitZ = trajectory.getTrajectorySZ().mapSToZ(arcLength);
52
53 Vector3D trackPositionAtHit(trackPositionAtHit2D, trackPositionAtHitZ);
54 Vector3D distance = trackPositionAtHit - hitPosition;
55
56 var<named("distance")>() = static_cast<Float_t>(distance.norm());
57 var<named("xy_distance")>() = static_cast<Float_t>(distance.xy().norm());
58 var<named("z_distance")>() = static_cast<Float_t>(distance.z());
59
60 Vector3D mSoP_distance = position - hitPosition;
61
62 var<named("mSoP_distance")>() = static_cast<Float_t>(mSoP_distance.norm());
63 var<named("mSoP_xy_distance")>() = static_cast<Float_t>(mSoP_distance.xy().norm());
64 var<named("mSoP_z_distance")>() = static_cast<Float_t>(mSoP_distance.z());
65
66 var<named("same_hemisphere")>() = fabs(position.phi() - hitPosition.phi()) < TMath::PiOver2();
67
68 var<named("arcLengthOfHitPosition")>() = static_cast<Float_t>(trajectory.calcArcLength2D(hitPosition));
69 var<named("arcLengthOfCenterPosition")>() = static_cast<Float_t>(trajectory.calcArcLength2D(Vector3D(0, 0, 0)));
70
71 var<named("layer")>() = spacePoint->getVxdID().getLayerNumber();
72 var<named("number")>() = previousStates.size();
73
74 var<named("pt")>() = static_cast<Float_t>(momentum.xy().norm());
75 var<named("tan_lambda")>() = static_cast<Float_t>(trajectory.getTanLambda());
76 var<named("phi")>() = static_cast<Float_t>(momentum.phi());
77
78 const VxdID& sensorInfo = spacePoint->getVxdID();
79
80 var<named("ladder")>() = sensorInfo.getLadderNumber();
81 var<named("sensor")>() = sensorInfo.getSensorNumber();
82 var<named("segment")>() = sensorInfo.getSegmentNumber();
83 var<named("id")>() = sensorInfo.getID();
84
85 const auto& clusters = spacePoint->getRelationsTo<PXDCluster>();
86 B2ASSERT("Must be related to exactly 1 cluster", clusters.size() == 1);
87 var<named("cluster_charge")>() = clusters[0]->getCharge();
88 var<named("cluster_seed_charge")>() = clusters[0]->getSeedCharge();
89 var<named("cluster_size")>() = clusters[0]->getSize();
90 var<named("cluster_size_u")>() = clusters[0]->getUSize();
91 var<named("cluster_size_v")>() = clusters[0]->getVSize();
92
93 var<named("last_layer")>() = 0;
94 var<named("last_ladder")>() = 0;
95 var<named("last_sensor")>() = 0;
96 var<named("last_segment")>() = 0;
97 var<named("last_id")>() = 0;
98 var<named("last_cluster_charge")>() = 0;
99 var<named("last_cluster_seed_charge")>() = 0;
100 var<named("last_cluster_size")>() = 0;
101 var<named("last_cluster_size_u")>() = 0;
102 var<named("last_cluster_size_v")>() = 0;
103
104 const CKFToPXDState* parent = previousStates.back();
105 const SpacePoint* parentSpacePoint = parent->getHit();
106 if (parentSpacePoint) {
107 const VxdID& parentSensorInfo = parentSpacePoint->getVxdID();
108
109 var<named("last_layer")>() = parentSensorInfo.getLayerNumber();
110 var<named("last_ladder")>() = parentSensorInfo.getLadderNumber();
111 var<named("last_sensor")>() = parentSensorInfo.getSensorNumber();
112 var<named("last_segment")>() = parentSensorInfo.getSegmentNumber();
113 var<named("last_id")>() = parentSensorInfo.getID();
114
115 const auto& parentclusters = parentSpacePoint->getRelationsTo<PXDCluster>();
116 B2ASSERT("Must be related to exactly 1 cluster", parentclusters.size() == 1);
117 var<named("last_cluster_charge")>() = parentclusters[0]->getCharge();
118 var<named("last_cluster_seed_charge")>() = parentclusters[0]->getSeedCharge();
119 var<named("last_cluster_size")>() = parentclusters[0]->getSize();
120 var<named("last_cluster_size_u")>() = parentclusters[0]->getUSize();
121 var<named("last_cluster_size_v")>() = parentclusters[0]->getVSize();
122 }
123
124 const double residual = m_kalmanStepper.calculateResidual(firstMeasurement, *state);
125 var<named("residual")>() = residual;
126
127 if (state->isFitted()) {
128 var<named("chi2")>() = static_cast<Float_t>(state->getChi2());
129 } else {
130 var<named("chi2")>() = -999;
131 }
132
133 const TMatrixDSym& cov5 = firstMeasurement.getCov();
134 const Float_t sigmaUV = std::sqrt(std::max(cov5(4, 4), cov5(3, 3)));
135 var<named("sigma_uv")>() = sigmaUV;
136 var<named("residual_over_sigma")>() = residual / sigmaUV;
137
138 return true;
139}
Specialized CKF State for extrapolating into the PXD.
The PXD Cluster class This class stores all information about reconstructed PXD clusters The position...
Definition PXDCluster.h:30
virtual bool extract(const BasePXDStateFilter::Object *pair) override
Generate and assign the variables from the VXD-CDC-pair.
PXDKalmanStepper m_kalmanStepper
Kalmap update filter used in this var set.
This is the Reconstruction Event-Data Model Track.
Definition RecoTrack.h:79
short int getChargeSeed() const
Return the charge seed stored in the reco track. ATTENTION: This is not the fitted charge.
Definition RecoTrack.h:508
RelationVector< TO > getRelationsTo(const std::string &name="", const std::string &namedRelation="") const
Get the relations that point from this object to another store array.
SpacePoint typically is build from 1 PXDCluster or 1-2 SVDClusters.
Definition SpacePoint.h:42
VxdID getVxdID() const
Return the VxdID of the sensor on which the the cluster of the SpacePoint lives.
Definition SpacePoint.h:148
const B2Vector3D & getPosition() const
return the position vector in global coordinates
Definition SpacePoint.h:138
Vector2D getPos2DAtArcLength2D(double arcLength2D)
Getter for the position at a given two dimensional arc length.
Particle full three dimensional trajectory.
double calcArcLength2D(const Vector3D &point) const
Calculate the travel distance from the start position of the trajectory.
CDCTrajectory2D getTrajectory2D() const
Getter for the two dimensional trajectory.
CDCTrajectorySZ getTrajectorySZ() const
Getter for the sz trajectory.
double getTanLambda() const
Getter for the slope of z over the transverse travel distance s.
double mapSToZ(const double s=0) const
Translates the travel distance to the z coordinate.
std::pair< const std::vector< TrackFindingCDC::WithWeight< const CKFToPXDState * > >, CKFToPXDState * > Object
Definition Filter.dcl.h:35
static constexpr int named(const char *name)
Definition VarSet.h:78
A two dimensional vector which is equipped with functions for correct handling of orientation relate...
Definition Vector2D.h:32
Class to uniquely identify a any structure of the PXD and SVD.
Definition VxdID.h:33
baseType getID() const
Get the unique id.
Definition VxdID.h:94
baseType getSensorNumber() const
Get the sensor id.
Definition VxdID.h:100
baseType getSegmentNumber() const
Get the sensor segment.
Definition VxdID.h:102
baseType getLadderNumber() const
Get the ladder id.
Definition VxdID.h:98
baseType getLayerNumber() const
Get the layer id.
Definition VxdID.h:96
STL class.
HepGeom::Vector3D< double > Vector3D
3D Vector
Definition Cell.h:34
Abstract base class for different kinds of events.