10 #include <tracking/ckf/cdc/filters/states/CDCStateFilterFactory.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/RecordingFilter.icc.h>
18 #include <tracking/trackFindingCDC/varsets/VariadicUnionVarSet.h>
20 #include <tracking/ckf/cdc/filters/states/CDCStateBasicVarSet.h>
21 #include <tracking/ckf/cdc/filters/states/CDCfromEclStateTruthVarSet.h>
23 #include <tracking/ckf/cdc/filters/states/MCTruthCDCStateFilter.h>
24 #include <tracking/ckf/cdc/filters/states/RoughCDCStateFilter.h>
25 #include <tracking/ckf/cdc/filters/states/RoughCDCfromEclStateFilter.h>
26 #include <tracking/ckf/cdc/filters/states/ExtrapolateAndUpdateCDCStateFilter.h>
27 #include <tracking/ckf/cdc/filters/states/DistanceCDCStateFilter.h>
28 #include <tracking/ckf/cdc/filters/states/MCTruthEclSeedFilter.h>
31 using namespace TrackFindingCDC;
35 using RecordingCDCStateFilter = RecordingFilter<VariadicUnionVarSet<CDCStateBasicVarSet>>;
38 using RecordingCDCfromEclStateFilter = RecordingFilter<VariadicUnionVarSet<CDCStateBasicVarSet, CDCfromEclStateTruthVarSet>>;
40 using AndCDCStateFilter = AndFilter<BaseCDCStateFilter>;
45 :
Super(defaultFilterName)
58 return "Reject cdc states";
64 {
"none",
"no track combination is valid"},
65 {
"all",
"set all track combinations as good"},
66 {
"mc_truth",
"filtering based on the mc truth information"},
67 {
"mc_truth_eclSeed",
"filtering based on the mc truth information"},
68 {
"rough",
"very rough filtering"},
69 {
"rough_eclSeed",
"very rough filtering, seed created from ECL shower"},
70 {
"extrapolate_and_update",
"Extrapolation and update"},
71 {
"distance",
"Give a weight based on the distance"},
72 {
"recording",
"record variables to a TTree"},
73 {
"recording_eclSeed",
"record variables to a TTree"},
74 {
"rough_and_recording",
"very rough filtering, seed created from SVD track"},
75 {
"rough_and_recording_eclSeed",
"very rough filtering, seed created from ECL shower"},
76 {
"distance_and_recording_eclSeed",
"Give a weight based on the distance"},
81 std::unique_ptr<BaseCDCStateFilter>
84 if (filterName ==
"none") {
85 return std::make_unique<TrackFindingCDC::NoneFilter<BaseCDCStateFilter>>();
86 }
else if (filterName ==
"all") {
87 return std::make_unique<TrackFindingCDC::AllFilter<BaseCDCStateFilter>>();
88 }
else if (filterName ==
"rough") {
89 return std::make_unique<RoughCDCStateFilter>();
90 }
else if (filterName ==
"rough_eclSeed") {
91 return std::make_unique<RoughCDCfromEclStateFilter>();
92 }
else if (filterName ==
"mc_truth") {
93 return std::make_unique<MCTruthCDCStateFilter>();
94 }
else if (filterName ==
"mc_truth_eclSeed") {
95 return std::make_unique<MCTruthEclSeedFilter>();
96 }
else if (filterName ==
"extrapolate_and_update") {
97 return std::make_unique<ExtrapolateAndUpdateCDCStateFilter>();
98 }
else if (filterName ==
"distance") {
99 return std::make_unique<DistanceCDCStateFilter>();
100 }
else if (filterName ==
"recording") {
101 return std::make_unique<RecordingCDCStateFilter>(
"CDCStateFilter.root");
102 }
else if (filterName ==
"rough_and_recording") {
103 return std::make_unique<AndCDCStateFilter>(
104 std::make_unique<RecordingCDCStateFilter>(
"CDCStateFilter.root"),
105 std::make_unique<RoughCDCStateFilter>()
107 }
else if (filterName ==
"recording_eclSeed") {
108 return std::make_unique<RecordingCDCfromEclStateFilter>(
"CDCfromECLStateFilter.root");
109 }
else if (filterName ==
"rough_and_recording_eclSeed") {
110 return std::make_unique<AndCDCStateFilter>(
111 std::make_unique<RecordingCDCfromEclStateFilter>(
"CDCfromECLStateFilter.root"),
112 std::make_unique<RoughCDCfromEclStateFilter>()
114 }
else if (filterName ==
"distance_and_recording_eclSeed") {
115 return std::make_unique<AndCDCStateFilter>(
116 std::make_unique<RecordingCDCfromEclStateFilter>(
"CDCfromECLStateFilter.root"),
117 std::make_unique<DistanceCDCStateFilter>()