Belle II Software  release-05-01-25
CDCStateFilterFactory.cc
1 /**************************************************************************
2  * BASF2 (Belle Analysis Framework 2) *
3  * Copyright(C) 2017 - Belle II Collaboration *
4  * *
5  * Author: The Belle II Collaboration *
6  * Contributors: Nils Braun *
7  * *
8  * This software is provided "as is" without any warranty. *
9  **************************************************************************/
10 #include <tracking/ckf/cdc/filters/states/CDCStateFilterFactory.h>
11 
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>
17 
18 #include <tracking/trackFindingCDC/varsets/VariadicUnionVarSet.h>
19 
20 #include <tracking/ckf/cdc/filters/states/CDCStateBasicVarSet.h>
21 #include <tracking/ckf/cdc/filters/states/CDCfromEclStateTruthVarSet.h>
22 
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>
29 
30 using namespace Belle2;
31 using namespace TrackFindingCDC;
32 
33 namespace {
35  using RecordingCDCStateFilter = RecordingFilter<VariadicUnionVarSet<CDCStateBasicVarSet>>;
36  //using RecordingCDCStateFilter = RecordingFilter<VariadicUnionVarSet<CDCStateBasicVarSet, CDCStateTruthVarSet>>;
38  using RecordingCDCfromEclStateFilter = RecordingFilter<VariadicUnionVarSet<CDCStateBasicVarSet, CDCfromEclStateTruthVarSet>>;
40  using AndCDCStateFilter = AndFilter<BaseCDCStateFilter>;
41 }
42 
43 
44 CDCStateFilterFactory::CDCStateFilterFactory(const std::string& defaultFilterName)
45  : Super(defaultFilterName)
46 {
47 }
48 
50 
52 {
53  return "CDCState";
54 }
55 
57 {
58  return "Reject cdc states";
59 }
60 
61 std::map<std::string, std::string> CDCStateFilterFactory::getValidFilterNamesAndDescriptions() const
62 {
63  return {
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"},
77 
78  };
79 }
80 
81 std::unique_ptr<BaseCDCStateFilter>
82 CDCStateFilterFactory::create(const std::string& filterName) const
83 {
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>()
106  );
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>()
113  );
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>()
118  );
119  } else {
120  return Super::create(filterName);
121  }
122 }
Belle2::TrackFindingCDC::FilterFactory< BaseCDCStateFilter >
Belle2::CDCStateFilterFactory::create
std::unique_ptr< BaseCDCStateFilter > create(const std::string &filterName) const override
Create a filter with the given name.
Definition: CDCStateFilterFactory.cc:82
Belle2::TrackFindingCDC::FilterFactory< BaseCDCStateFilter >::create
virtual std::unique_ptr< BaseCDCStateFilter > create(const std::string &filterName) const
Create a filter with the given name, does not set filter specific parameters.
Definition: FilterFactory.icc.h:54
Belle2::CDCStateFilterFactory::getValidFilterNamesAndDescriptions
std::map< std::string, std::string > getValidFilterNamesAndDescriptions() const override
Getter for valid filter names and a description for each.
Definition: CDCStateFilterFactory.cc:61
Belle2::CDCStateFilterFactory::~CDCStateFilterFactory
~CDCStateFilterFactory()
Default destructor.
Belle2
Abstract base class for different kinds of events.
Definition: MillepedeAlgorithm.h:19
Belle2::CDCStateFilterFactory::CDCStateFilterFactory
CDCStateFilterFactory(const std::string &defaultFilterName="all")
Constructor forwarding the default filter name.
Definition: CDCStateFilterFactory.cc:44
Belle2::CDCStateFilterFactory::getFilterPurpose
std::string getFilterPurpose() const override
Getter for a descriptive purpose of the constructed filters.
Definition: CDCStateFilterFactory.cc:56
Belle2::CDCStateFilterFactory::getIdentifier
std::string getIdentifier() const override
Getter for a short identifier for the factory.
Definition: CDCStateFilterFactory.cc:51