10 #include <tracking/ckf/pxd/filters/states/PXDStateFilterFactory.h>
12 #include <tracking/trackFindingCDC/filters/base/FilterFactory.icc.h>
13 #include <tracking/trackFindingCDC/filters/base/AllFilter.icc.h>
14 #include <tracking/trackFindingCDC/filters/base/NoneFilter.icc.h>
15 #include <tracking/trackFindingCDC/filters/base/AndFilter.icc.h>
16 #include <tracking/trackFindingCDC/filters/base/NegativeFilter.icc.h>
17 #include <tracking/trackFindingCDC/filters/base/ChoosableFromVarSetFilter.icc.h>
18 #include <tracking/trackFindingCDC/filters/base/RecordingFilter.icc.h>
19 #include <tracking/trackFindingCDC/filters/base/MVAFilter.icc.h>
20 #include <tracking/trackFindingCDC/filters/base/SloppyFilter.icc.h>
21 #include <tracking/trackFindingCDC/filters/base/TruthVarFilter.icc.h>
23 #include <tracking/trackFindingCDC/varsets/VariadicUnionVarSet.h>
25 #include <tracking/ckf/general/filters/AdvanceFilter.h>
26 #include <tracking/ckf/general/filters/KalmanFilter.h>
27 #include <tracking/ckf/pxd/utilities/PXDAdvancer.h>
28 #include <tracking/ckf/pxd/utilities/PXDKalmanStepper.h>
30 #include <tracking/ckf/pxd/filters/states/PXDStateBasicVarSet.h>
31 #include <tracking/ckf/pxd/filters/states/PXDStateTruthVarSet.h>
32 #include <tracking/ckf/pxd/filters/states/SimplePXDStateFilter.h>
33 #include <tracking/ckf/pxd/filters/states/AllPXDStateFilter.h>
34 #include <tracking/ckf/pxd/filters/states/NonIPCrossingPXDStateFilter.h>
37 using namespace TrackFindingCDC;
41 using MCPXDStateFilter = TruthVarFilter<PXDStateTruthVarSet>;
44 using MCOrderingPXDStateFilter = ChoosableFromVarSetFilter<PXDStateTruthVarSet>;
47 using SloppyMCPXDStateFilter = Sloppy<TruthVarFilter<PXDStateTruthVarSet>>;
50 using RecordingPXDStateFilter = RecordingFilter<VariadicUnionVarSet<PXDStateTruthVarSet, PXDStateBasicVarSet>>;
53 using MVAPXDStateFilter = NegativeFilter<MVAFilter<PXDStateBasicVarSet>>;
56 using AndPXDStateFilter = AndFilter<BasePXDStateFilter>;
59 class SloppyRecordingPXDStateFilter :
public RecordingPXDStateFilter {
61 explicit SloppyRecordingPXDStateFilter(
const std::string& defaultRootFileName) : RecordingPXDStateFilter(defaultRootFileName)
63 setSkimFilter(std::make_unique<SloppyMCPXDStateFilter>());
70 :
Super(defaultFilterName)
83 return "Reject pxd states";
89 {
"none",
"no track combination is valid"},
90 {
"all",
"set all track combinations as good"},
91 {
"advance",
"extrapolate the states"},
92 {
"fit",
"update the mSoP using a Kalman Filter"},
93 {
"truth",
"monte carlo truth"},
94 {
"ordering_truth",
"monte carlo truth ordering"},
95 {
"sloppy_truth",
"sloppy monte carlo truth"},
96 {
"simple",
"simple filter to be used in pxd"},
97 {
"recording",
"record variables to a TTree"},
98 {
"mva_with_direction_check",
"MVA filter with direction check"},
99 {
"mva",
"MVA filter"},
100 {
"sloppy_recording",
"record variables to a TTree"},
104 std::unique_ptr<BasePXDStateFilter>
107 if (filterName ==
"none") {
108 return std::make_unique<TrackFindingCDC::NoneFilter<BasePXDStateFilter>>();
109 }
else if (filterName ==
"all") {
110 return std::make_unique<AllPXDStateFilter>();
111 }
else if (filterName ==
"advance") {
112 return std::make_unique<AdvanceFilter<CKFToPXDState, PXDAdvancer>>();
113 }
else if (filterName ==
"fit") {
114 return std::make_unique<KalmanFilter<CKFToPXDState, PXDKalmanStepper>>();
115 }
else if (filterName ==
"simple") {
116 return std::make_unique<SimplePXDStateFilter>();
117 }
else if (filterName ==
"truth") {
118 return std::make_unique<MCPXDStateFilter>();
119 }
else if (filterName ==
"ordering_truth") {
120 return std::make_unique<MCOrderingPXDStateFilter>(
"truth_inverted");
121 }
else if (filterName ==
"sloppy_truth") {
122 return std::make_unique<SloppyMCPXDStateFilter>();
123 }
else if (filterName ==
"recording") {
124 return std::make_unique<RecordingPXDStateFilter>(
"PXDStateFilter.root");
125 }
else if (filterName ==
"mva_with_direction_check") {
126 return std::make_unique<AndPXDStateFilter>(
127 std::make_unique<NonIPCrossingPXDStateFilter>(),
128 std::make_unique<MVAPXDStateFilter>(
"ckf_CDCPXDStateFilter_1")
130 }
else if (filterName ==
"mva") {
131 return std::make_unique<AndPXDStateFilter>(
132 std::make_unique<MVAPXDStateFilter>(
"ckf_CDCPXDStateFilter_1"),
133 std::make_unique<NonIPCrossingPXDStateFilter>());
134 }
else if (filterName ==
"sloppy_recording") {
135 return std::make_unique<SloppyRecordingPXDStateFilter>(
"PXDStateFilter.root");