Belle II Software  release-05-01-25
UnionRecordingSegmentRelationFilter.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/segmentRelation/UnionRecordingSegmentRelationFilter.h>
11 
12 #include <tracking/trackFindingCDC/filters/segmentRelation/BasicSegmentRelationVarSet.h>
13 #include <tracking/trackFindingCDC/filters/segmentRelation/HitGapSegmentRelationVarSet.h>
14 #include <tracking/trackFindingCDC/filters/segmentRelation/FitlessSegmentRelationVarSet.h>
15 #include <tracking/trackFindingCDC/filters/segmentRelation/FitSegmentRelationVarSet.h>
16 
17 #include <tracking/trackFindingCDC/filters/segmentRelation/MVAFeasibleSegmentRelationFilter.h>
18 #include <tracking/trackFindingCDC/filters/segmentRelation/MVARealisticSegmentRelationFilter.h>
19 
20 #include <tracking/trackFindingCDC/filters/base/UnionRecordingFilter.icc.h>
21 
22 
23 using namespace Belle2;
24 using namespace TrackFindingCDC;
25 
27 
28 std::vector<std::string>
30 {
31  std::vector<std::string> varSetNames = Super::getValidVarSetNames();
32  varSetNames.insert(varSetNames.end(), {"basic", "hit_gap", "feasible", "fitless", "fit", "realistic"});
33  return varSetNames;
34 }
35 
36 std::unique_ptr<BaseVarSet<Relation<const CDCSegment2D> > >
38 {
39  if (name == "basic") {
40  return std::make_unique<BasicSegmentRelationVarSet>();
41  } else if (name == "hit_gap") {
42  return std::make_unique<HitGapSegmentRelationVarSet>();
43  } else if (name == "feasible") {
45  return std::move(filter).releaseVarSet();
46  } else if (name == "fitless") {
47  return std::make_unique<FitlessSegmentRelationVarSet>();
48  } else if (name == "fit") {
49  return std::make_unique<FitSegmentRelationVarSet>();
50  } else if (name == "realistic") {
52  return std::move(filter).releaseVarSet();
53  } else {
54  return Super::createVarSet(name);
55  }
56 }
Belle2::TrackFindingCDC::UnionRecording< SegmentRelationFilterFactory ::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< SegmentRelationFilterFactory >
Belle2::TrackFindingCDC::UnionRecordingSegmentRelationFilter::createVarSet
std::unique_ptr< BaseVarSet< Relation< const CDCSegment2D > > > createVarSet(const std::string &name) const override
Create a concrete variables set for segment relations from a name.
Definition: UnionRecordingSegmentRelationFilter.cc:37
Belle2::filter
std::map< ExpRun, std::pair< double, double > > filter(const std::map< ExpRun, std::pair< double, double >> &runs, double cut, std::map< ExpRun, std::pair< double, double >> &runsRemoved)
filter events to remove runs shorter than cut, it stores removed runs in runsRemoved
Definition: Splitter.cc:43
Belle2
Abstract base class for different kinds of events.
Definition: MillepedeAlgorithm.h:19
Belle2::TrackFindingCDC::MVAFeasibleSegmentRelationFilter
Filter for the constuction of segment pairs based on simple criteria without the common fit.
Definition: MVAFeasibleSegmentRelationFilter.h:37
Belle2::TrackFindingCDC::UnionRecording< SegmentRelationFilterFactory ::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::MVARealisticSegmentRelationFilter
Final filter for the constuction of segment pairs.
Definition: MVARealisticSegmentRelationFilter.h:40
Belle2::TrackFindingCDC::UnionRecordingSegmentRelationFilter::getValidVarSetNames
std::vector< std::string > getValidVarSetNames() const override
Get the valid names of variable sets for segment relations.
Definition: UnionRecordingSegmentRelationFilter.cc:29