Belle II Software development
FourHitFilter.cc
1/**************************************************************************
2 * basf2 (Belle II Analysis Software Framework) *
3 * Author: The Belle II Collaboration *
4 * *
5 * See git log for contributors and copyright holders. *
6 * This file is licensed under LGPL-3.0, see LICENSE.md. *
7 **************************************************************************/
8#include <tracking/vxdHoughTracking/filters/pathFilters/FourHitFilter.h>
9#include <tracking/trackFindingCDC/filters/base/Filter.icc.h>
10#include <tracking/trackFindingCDC/utilities/StringManipulation.h>
11#include <framework/core/ModuleParamList.templateDetails.h>
12#include <framework/geometry/BFieldManager.h>
13
14using namespace Belle2;
15using namespace TrackFindingCDC;
16using namespace vxdHoughTracking;
17
18void FourHitFilter::exposeParameters(ModuleParamList* moduleParamList, const std::string& prefix)
19{
20 moduleParamList->addParameter(TrackFindingCDC::prefixed(prefix, "circleRadiusDifferenceCut"), m_CircleRadiusDifferenceCut,
21 "Cut on the difference of the radii of the two circles that can be defined by two hit triplets.",
23
24 moduleParamList->addParameter(TrackFindingCDC::prefixed(prefix, "circleCenterPositionDifferenceCut"),
26 "Cut on the difference between the center positions of the two circles that can be defined by two hit triplets.",
28}
29
31{
32 const double bFieldZ = BFieldManager::getField(0, 0, 0).Z() / Unit::T;
33 m_fourHitVariables.setBFieldZ(bFieldZ);
34}
35
36TrackFindingCDC::Weight
38{
39 const std::vector<TrackFindingCDC::WithWeight<const VXDHoughState*>>& previousHits = pair.first;
40
41 // Do nothing if path is too short or too long
42 if (previousHits.size() != 3) {
43 return NAN;
44 }
45
46 const B2Vector3D& firstHitPos = previousHits.at(0)->getHit()->getPosition();
47 const B2Vector3D& secondHitPos = previousHits.at(1)->getHit()->getPosition();
48 const B2Vector3D& thirdHitPos = previousHits.at(2)->getHit()->getPosition();
49 const B2Vector3D& currentHitPos = pair.second->getHit()->getPosition();
50
51 m_fourHitVariables.setHits(firstHitPos, secondHitPos, thirdHitPos, currentHitPos);
52
53 if (m_fourHitVariables.getCircleRadiusDifference() > m_CircleRadiusDifferenceCut) {
54 return NAN;
55 }
56 if (m_fourHitVariables.getCircleCenterPositionDifference() > m_CircleCenterPositionDifferenceCut) {
57 return NAN;
58 }
59
60 return fabs(1.0 / m_fourHitVariables.getCircleRadiusDifference());
61}
The Module parameter list class.
std::pair< const std::vector< TrackFindingCDC::WithWeight< const VXDHoughState * > >, VXDHoughState * > Object
Definition Filter.dcl.h:35
static const double T
[tesla]
Definition Unit.h:120
double m_CircleRadiusDifferenceCut
Cut on difference of the two circle radii estimated from two triplets.
double m_CircleCenterPositionDifferenceCut
Cut on difference of the two circle center estimated from two triplets.
TrackFindingCDC::Weight operator()(const BasePathFilter::Object &pair) override
Return the weight based on the ThreeHitVariables.
void beginRun() override
set BField value for estimator
FourHitVariables m_fourHitVariables
Construct empty ThreeHitVariables instance.
void exposeParameters(ModuleParamList *moduleParamList, const std::string &prefix) override
Expose the parameters.
STL class.
void addParameter(const std::string &name, T &paramVariable, const std::string &description, const T &defaultValue)
Adds a new parameter to the module list.
B2Vector3< double > B2Vector3D
typedef for common usage with double
Definition B2Vector3.h:516
static void getField(const double *pos, double *field)
return the magnetic field at a given position.
Abstract base class for different kinds of events.