8#include <tracking/vxdHoughTracking/findlets/RecoTrackStorer.h>
9#include <framework/core/ModuleParamList.h>
10#include <framework/geometry/BFieldManager.h>
11#include <tracking/dataobjects/RecoTrack.h>
12#include <svd/dataobjects/SVDCluster.h>
13#include <tracking/spacePointCreation/SpacePoint.h>
14#include <tracking/spacePointCreation/SpacePointTrackCand.h>
15#include <tracking/trackFindingVXD/trackQualityEstimators/QualityEstimatorCircleFit.h>
16#include <tracking/trackFindingVXD/trackQualityEstimators/QualityEstimatorMC.h>
17#include <tracking/trackFindingVXD/trackQualityEstimators/QualityEstimatorRiemannHelixFit.h>
18#include <tracking/trackFindingVXD/trackQualityEstimators/QualityEstimatorTripletFit.h>
19#include <tracking/trackFindingCDC/utilities/StringManipulation.h>
20#include <tracking/trackFindingCDC/utilities/Algorithms.h>
21#include <tracking/trackFindingCDC/utilities/WeightedRelation.h>
24using namespace TrackFindingCDC;
25using namespace vxdHoughTracking;
43 moduleParamList->
addParameter(TrackFindingCDC::prefixed(prefix,
"SVDSpacePointTrackCandsStoreArrayName"),
48 "Identifier which estimation method to use. Valid identifiers are: [mcInfo, circleFit, tripletFit, helixFit].",
51 moduleParamList->
addParameter(TrackFindingCDC::prefixed(prefix,
"ResultStorerMCRecoTracksStoreArrayName"),
53 "Only required for MCInfo method. Name of StoreArray containing MCRecoTracks.",
56 moduleParamList->
addParameter(TrackFindingCDC::prefixed(prefix,
"ResultStorerMCStrictQualityEstimator"),
58 "Only required for MCInfo method. If false combining several MCTracks is allowed.",
78 m_estimator = std::make_unique<QualityEstimatorTripletFit>();
80 m_estimator = std::make_unique<QualityEstimatorCircleFit>();
82 m_estimator = std::make_unique<QualityEstimatorRiemannHelixFit>();
110 const std::vector<const SpacePoint*>& spacePoints)
112 for (
auto& thisSPTC : finishedResults) {
118 auto sortedHits = thisSPTC.getSortedHits();
119 const auto& estimatorResult =
m_estimator->estimateQualityAndProperties(sortedHits);
121 const ROOT::Math::XYZVector trackPosition(sortedHits.front()->X(), sortedHits.front()->Y(), sortedHits.front()->Z());
122 const ROOT::Math::XYZVector& trackMomentum = *estimatorResult.p;
123 const short& trackChargeSeed = estimatorResult.curvatureSign ? -1 * (*(estimatorResult.curvatureSign)) : 0;
124 const double qi = estimatorResult.qualityIndicator;
131 TVectorD stateSeed(6);
132 TMatrixDSym covSeed(6);
133 covSeed(0, 0) = 0.01 ; covSeed(1, 1) = 0.01 ; covSeed(2, 2) = 0.04 ;
134 covSeed(3, 3) = 0.01 ; covSeed(4, 4) = 0.01 ; covSeed(5, 5) = 0.04 ;
136 stateSeed(0) = (sortedHits.front()->X());
137 stateSeed(1) = (sortedHits.front()->Y());
138 stateSeed(2) = (sortedHits.front()->Z());
139 if (estimatorResult.p) {
140 auto momentumSeed = *(estimatorResult.p);
141 stateSeed(3) = momentumSeed.X();
142 stateSeed(4) = momentumSeed.Y();
143 stateSeed(5) = momentumSeed.Z();
150 thisSPTC.set6DSeed(stateSeed);
151 thisSPTC.setCovSeed(covSeed);
152 thisSPTC.setChargeSeed(trackChargeSeed);
166 unsigned int sortingParameter = 0;
167 for (
const SpacePoint* spacePoint : sortedHits) {
171 for (
const SVDCluster& relatedCluster : relatedClusters) {
173 newRecoTrack->
addSVDHit(&relatedCluster, sortingParameter, Belle2::RecoHitInformation::c_SVDHough);
179 for (
const SpacePoint* spacePoint : spacePoints) {
181 spacePoint->setAssignmentState(
true);
186 for (
const SVDCluster& relatedCluster : relatedClusters) {
188 spacePoint->setAssignmentState(
true);
@ c_DontWriteOut
Object/array should be NOT saved by output modules.
@ c_ErrorIfAlreadyRegistered
If the object/array was already registered, produce an error (aborting initialisation).
The Module parameter list class.
Class implementing the algorithm used for the MC based quality estimation.
void forceUpdateClusterNames()
Setter to force the class to update its cluster names.
This is the Reconstruction Event-Data Model Track.
void setSeedCovariance(const TMatrixDSym &seedCovariance)
Set the covariance of the seed. ATTENTION: This is not the fitted covariance.
void setQualityIndicator(const float qualityIndicator)
Set the quality index attached to this RecoTrack. 0 means likely fake.
static void registerRequiredRelations(StoreArray< RecoTrack > &recoTracks, std::string const &pxdHitsStoreArrayName="", std::string const &svdHitsStoreArrayName="", std::string const &cdcHitsStoreArrayName="", std::string const &bklmHitsStoreArrayName="", std::string const &eklmHitsStoreArrayName="", std::string const &recoHitInformationStoreArrayName="")
Convenience method which registers all relations required to fully use a RecoTrack.
bool addSVDHit(const UsedSVDHit *svdHit, const unsigned int sortingParameter, OriginTrackFinder foundByTrackFinder=OriginTrackFinder::c_undefinedTrackFinder)
Adds a svd hit with the given information to the reco track.
Class for type safe access to objects that are referred to in relations.
void addRelationTo(const RelationsInterface< BASE > *object, float weight=1.0, const std::string &namedRelation="") const
Add a relation from this object to another object (with caching).
The SVD Cluster class This class stores all information about reconstructed SVD clusters.
Storage for (VXD) SpacePoint-based track candidates.
@ c_isActive
bit 11: SPTC is active (i.e.
SpacePoint typically is build from 1 PXDCluster or 1-2 SVDClusters.
bool isRequired(const std::string &name="")
Ensure this array/object has been registered previously.
Accessor to arrays stored in the data store.
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 beginEvent() override
Receive and dispatch signal for the start of a new event.
virtual void exposeParameters(ModuleParamList *moduleParamList, const std::string &prefix)
Expose the set of parameters of the filter to the module parameter list.
static const double T
[tesla]
void apply(std::vector< SpacePointTrackCand > &finishedResults, const std::vector< const SpacePoint * > &spacePoints) override
Store the finishey SpacePointTrackCands into RecoTracks and tag the SpacePoints.
~RecoTrackStorer()
Default destructor.
bool m_MCStrictQualityEstimator
Only required for MCInfo method.
void initialize() override
Create the store arrays.
std::string m_SVDSpacePointTrackCandsStoreArrayName
StoreArray name of the SpacePointTrackCandidate StoreArray.
std::string m_EstimationMethod
Identifier which estimation method to use.
std::string m_SVDClustersStoreArrayName
StoreArray name of the SVDCluster StoreArray.
std::set< const SVDCluster * > m_usedClusters
Store the used clusters in the results.
RecoTrackStorer()
Constructor.
StoreArray< SpacePointTrackCand > m_storeSpacePointTrackCands
Output SpacePointTrackCand Store Array.
TrackFindingCDC::Findlet< SpacePointTrackCand, const SpacePoint *const > Super
Parent class.
void beginRun() override
Begin run.
std::string m_MCRecoTracksStoreArrayName
sets the name of the expected StoreArray containing MCRecoTracks. Only required for MCInfo method
void beginEvent() override
Reset internal vectors.
std::unique_ptr< QualityEstimatorBase > m_estimator
pointer to the selected QualityEstimator
StoreArray< RecoTrack > m_storeRecoTracks
Output RecoTracks Store Array.
std::set< const SpacePoint * > m_usedSpacePoints
Store the used space points in the results.
void exposeParameters(ModuleParamList *moduleParamList, const std::string &prefix) override
Expose the parameters of the sub findlets.
std::string m_RecoTracksStoreArrayName
StoreArray name of the output Track StoreArray.
void addParameter(const std::string &name, T ¶mVariable, const std::string &description, const T &defaultValue)
Adds a new parameter to the module list.
static void getField(const double *pos, double *field)
return the magnetic field at a given position.
Abstract base class for different kinds of events.