Belle II Software  release-05-01-25
UnionRecordingFacetFilter.cc
1 /**************************************************************************
2  * BASF2 (Belle Analysis Framework 2) *
3  * Copyright(C) 2015 - Belle II Collaboration *
4  * *
5  * Author: The Belle II Collaboration *
6  * Contributors: Oliver Frost *
7  * *
8  * This software is provided "as is" without any warranty. *
9  **************************************************************************/
10 #include <tracking/trackFindingCDC/filters/facet/UnionRecordingFacetFilter.h>
11 
12 #include <tracking/trackFindingCDC/filters/facet/FitFacetVarSet.h>
13 #include <tracking/trackFindingCDC/filters/facet/BendFacetVarSet.h>
14 #include <tracking/trackFindingCDC/filters/facet/FitlessFacetVarSet.h>
15 #include <tracking/trackFindingCDC/filters/facet/BasicFacetVarSet.h>
16 
17 #include <tracking/trackFindingCDC/filters/facet/MVAFacetFilter.h>
18 
19 #include <tracking/trackFindingCDC/filters/base/UnionRecordingFilter.icc.h>
20 
21 using namespace Belle2;
22 using namespace TrackFindingCDC;
23 
25 
26 std::vector<std::string> UnionRecordingFacetFilter::getValidVarSetNames() const
27 {
28  std::vector<std::string> varSetNames = Super::getValidVarSetNames();
29  varSetNames.insert(varSetNames.end(), {"basic", "fitless", "bend", "fit", "mva"});
30  return varSetNames;
31 }
32 
33 std::unique_ptr<BaseVarSet<const CDCFacet> >
34 UnionRecordingFacetFilter::createVarSet(const std::string& name) const
35 {
36  if (name == "basic") {
37  return std::make_unique<BasicFacetVarSet>();
38  } else if (name == "fitless") {
39  return std::make_unique<FitlessFacetVarSet>();
40  } else if (name == "bend") {
41  return std::make_unique<BendFacetVarSet>();
42  } else if (name == "fit") {
43  return std::make_unique<FitFacetVarSet>();
44  } else if (name == "mva") {
45  MVAFacetFilter mvaFacetFilter;
46  return std::move(mvaFacetFilter).releaseVarSet();
47  } else {
48  return Super::createVarSet(name);
49  }
50 }
Belle2::TrackFindingCDC::UnionRecordingFacetFilter::getValidVarSetNames
std::vector< std::string > getValidVarSetNames() const final
Get the valid names of variable sets for facets.
Definition: UnionRecordingFacetFilter.cc:26
Belle2::TrackFindingCDC::UnionRecording< FacetFilterFactory ::CreatedFilter >::getValidVarSetNames
virtual std::vector< std::string > getValidVarSetNames() const
Getter for the list of valid names of concret variable sets.
Definition: UnionRecordingFilter.icc.h:101
Belle2::TrackFindingCDC::UnionRecordingFilter< FacetFilterFactory >
Belle2::TrackFindingCDC::MVAFacetFilter
Background facet detection based on MVA.
Definition: MVAFacetFilter.h:44
Belle2
Abstract base class for different kinds of events.
Definition: MillepedeAlgorithm.h:19
Belle2::TrackFindingCDC::OnVarSet::releaseVarSet
std::unique_ptr< AVarSet > releaseVarSet() &&
Steal the set of variables form this filter - filter becomes disfunctional afterwards.
Definition: FilterOnVarSet.icc.h:85
Belle2::TrackFindingCDC::UnionRecording< FacetFilterFactory ::CreatedFilter >::createVarSet
virtual std::unique_ptr< AVarSet > createVarSet(const std::string &name) const
Create a variable set for the given name.
Definition: UnionRecordingFilter.icc.h:131
Belle2::TrackFindingCDC::UnionRecordingFacetFilter::createVarSet
std::unique_ptr< BaseVarSet< const CDCFacet > > createVarSet(const std::string &name) const final
Create a concrete variables set for facets from a name.
Definition: UnionRecordingFacetFilter.cc:34