 |
Belle II Software
release-05-01-25
|
11 #include <tracking/modules/trackQualityEstimator/TrackQualityEstimatorMVAModule.h>
21 setDescription(
"The quality estimator module for a fully reconstructed track");
26 "Name of the recoTrack StoreArray.",
29 addParam(
"SVDCDCRecoTracksStoreArrayName",
31 "Name of the SVD-CDC StoreArray.",
34 addParam(
"CDCRecoTracksStoreArrayName",
36 "Name of the CDC StoreArray.",
39 addParam(
"SVDRecoTracksStoreArrayName",
41 "Name of the SVD StoreArray.",
44 addParam(
"PXDRecoTracksStoreArrayName",
46 "Name of the PXD StoreArray.",
51 "Name of the fitted mdst Tracks StoreArray.",
56 "Identifier of weightfile in Database or local root/xml file.",
61 "Whether to use eventwise features.",
93 if (svdcdcRecoTrack) {
105 const float qualityIndicator =
m_mvaExpert->predict();
107 recoTrack.setQualityIndicator(qualityIndicator);
void initialize() override
Initializes the Module.
std::unique_ptr< RecoTrackExtractor > m_recoTrackExtractor
pointer to object that extracts info from the root RecoTrack
std::unique_ptr< EventInfoExtractor > m_eventInfoExtractor
pointer to object that extracts info from the whole event
void beginRun() override
Launches mvaExpert and sets the magnetic field strength.
void setDescription(const std::string &description)
Sets the description of the module.
std::unique_ptr< HitInfoExtractor > m_hitInfoExtractor
pointer to object that extracts info from the hits within the RecoTrack
std::string m_weightFileIdentifier
identifier of weightfile in Database or local root/xml file
std::string m_tracksStoreArrayName
Name of the StoreArray with mdst Tracks from track fit.
#define REG_MODULE(moduleName)
Register the given module (without 'Module' suffix) with the framework.
@ c_ParallelProcessingCertified
This module can be run in parallel processing mode safely (All I/O must be done through the data stor...
TO * getRelatedTo(const std::string &name="", const std::string &namedRelation="") const
Get the object to which this object has a relation.
std::vector< Named< float * > > m_variableSet
set of named variables to be used in MVA
std::string m_svdRecoTracksStoreArrayName
Name of the SVD StoreArray.
void setPropertyFlags(unsigned int propertyFlags)
Sets the flags for the module properties.
This is the Reconstruction Event-Data Model Track.
Abstract base class for different kinds of events.
bool m_collectEventFeatures
Parameter to enable event-wise features.
std::string m_cdcRecoTracksStoreArrayName
Name of the CDC StoreArray.
std::unique_ptr< SubRecoTrackExtractor > m_subRecoTrackExtractor
pointer to object that extracts info from the related sub RecoTracks
void addParam(const std::string &name, T ¶mVariable, const std::string &description, const T &defaultValue)
Adds a new parameter to the module.
std::string m_svdcdcRecoTracksStoreArrayName
Name of the SVD-CDC StoreArray.
StoreArray< RecoTrack > m_recoTracks
Store Array of the recoTracks.
std::string m_pxdRecoTracksStoreArrayName
Name of the PXD StoreArray.
std::string m_recoTracksStoreArrayName
Name of the recoTrack StoreArray.
void event() override
Applies the selected quality estimation method for a given set of TCs.
TrackQualityEstimatorMVAModule()
Constructor of the module.
std::unique_ptr< MVAExpert > m_mvaExpert
pointer to the object to interact with the MVA package