Belle II Software development
PXDStateTruthVarSet.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#include <tracking/ckf/pxd/filters/states/PXDStateTruthVarSet.h>
9
10#include <framework/core/ModuleParamList.h>
11#include <framework/dataobjects/EventMetaData.h>
12#include <tracking/trackFindingCDC/utilities/StringManipulation.h>
13
14#include <tracking/mcMatcher/TrackMatchLookUp.h>
15#include <tracking/dataobjects/RecoTrack.h>
16
17#include <TRandom.h>
18
19using namespace Belle2;
20using namespace TrackFindingCDC;
21
23{
24 const std::vector<TrackFindingCDC::WithWeight<const CKFToPXDState*>>& previousStates = pair->first;
25 const CKFToPXDState* state = pair->second;
26
27 std::vector<TrackFindingCDC::WithWeight<const CKFToPXDState*>> allStates = previousStates;
28 allStates.emplace_back(state, 0);
29
30 const RecoTrack* seedTrack = previousStates.front()->getSeed();
31 B2ASSERT("Path without seed?", seedTrack);
32
33 var<named("truth_event_id")>() = m_eventMetaData->getEvent();
34 var<named("truth_seed_number")>() = seedTrack->getArrayIndex();
35
36 // Default to 0
37 var<named("truth_position_x")>() = 0;
38 var<named("truth_position_y")>() = 0;
39 var<named("truth_position_z")>() = 0;
40 var<named("truth_momentum_x")>() = 0;
41 var<named("truth_momentum_y")>() = 0;
42 var<named("truth_momentum_z")>() = 0;
43 var<named("truth")>() = 0;
44
45 if (not m_mcUtil.allStatesCorrect(allStates)) {
46 if (m_UseFractionOfBackground and gRandom->Uniform() > m_BackgroundFraction) {
47 return false;
48 }
49 // Keep all variables set to false and return.
50 return true;
51 }
52
53 const std::string& seedTrackStoreArrayName = seedTrack->getArrayName();
54
55 TrackMatchLookUp mcCDCMatchLookUp("MCRecoTracks", seedTrackStoreArrayName);
56 const RecoTrack* cdcMCTrack = mcCDCMatchLookUp.getRelatedMCRecoTrack(*seedTrack);
57
58 var<named("truth")>() = 1;
59
60 var<named("truth_position_x")>() = cdcMCTrack->getPositionSeed().X();
61 var<named("truth_position_y")>() = cdcMCTrack->getPositionSeed().Y();
62 var<named("truth_position_z")>() = cdcMCTrack->getPositionSeed().Z();
63 var<named("truth_momentum_x")>() = cdcMCTrack->getMomentumSeed().X();
64 var<named("truth_momentum_y")>() = cdcMCTrack->getMomentumSeed().Y();
65 var<named("truth_momentum_z")>() = cdcMCTrack->getMomentumSeed().Z();
66
67 return true;
68}
69
70void PXDStateTruthVarSet::exposeParameters(ModuleParamList* moduleParamList, const std::string& prefix)
71{
72 moduleParamList->addParameter(TrackFindingCDC::prefixed(prefix, "useFractionOfBackground"), m_UseFractionOfBackground,
73 "Use only a fraction of background hits and combinations in recording to improve the signal fraction in the recording.",
75
76 moduleParamList->addParameter(TrackFindingCDC::prefixed(prefix, "backgroundFraction"), m_BackgroundFraction,
77 "Fraction of background hits and combinations to be used in recording to improve the signal fraction in the recording if " + \
78 TrackFindingCDC::prefixed(prefix, "useFractionOfBackground") + " is true.",
80}
Specialized CKF State for extrapolating into the PXD.
The Module parameter list class.
double m_BackgroundFraction
Fraction of background to use in case m_UseFractionOfBackground is true The default value of 0....
MCUtil m_mcUtil
MC information used in the PXD CKF.
virtual bool extract(const BasePXDStateFilter::Object *result) override
Generate and assign the variables from the object.
StoreObjPtr< EventMetaData > m_eventMetaData
Pointer to the object containing event number etc.
bool m_UseFractionOfBackground
Only use a fraction of background events for recording for MVA training.
void exposeParameters(ModuleParamList *moduleParamList, const std::string &prefix) override
Expose the parameters to the outside world.
This is the Reconstruction Event-Data Model Track.
Definition RecoTrack.h:79
ROOT::Math::XYZVector getPositionSeed() const
Return the position seed stored in the reco track. ATTENTION: This is not the fitted position.
Definition RecoTrack.h:480
ROOT::Math::XYZVector getMomentumSeed() const
Return the momentum seed stored in the reco track. ATTENTION: This is not the fitted momentum.
Definition RecoTrack.h:487
std::string getArrayName() const
Get name of array this object is stored in, or "" if not found.
int getArrayIndex() const
Returns this object's array index (in StoreArray), or -1 if not found.
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
Class to provide convenient methods to look up matching information between pattern recognition and M...
const RecoTrack * getRelatedMCRecoTrack(const RecoTrack &prRecoTrack) const
Looks for a related Monte Carlo track for the given pattern recognition track and return it if found.
STL class.
void addParameter(const std::string &name, T &paramVariable, const std::string &description, const T &defaultValue)
Adds a new parameter to the module list.
Abstract base class for different kinds of events.