Belle II Software light-2406-ragdoll
TreeFitterModule.h
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#pragma once
9
10#include <framework/core/Module.h>
11#include <framework/database/DBObjPtr.h>
12#include <framework/datastore/StoreArray.h>
13#include <framework/datastore/StoreObjPtr.h>
14
15#include <analysis/dataobjects/Particle.h>
16#include <analysis/dataobjects/ParticleList.h>
17
18#include <analysis/VertexFitting/TreeFitter/ConstraintConfiguration.h>
19
20#include <analysis/DecayDescriptor/DecayDescriptor.h>
21
22namespace Belle2 {
27 class Particle;
28
35 class TreeFitterModule : public Module {
36
37 public:
40
42 virtual void initialize() override;
43
45 virtual void beginRun() override;
46
48 virtual void event() override;
49
51 virtual void terminate() override;
52
53 private:
54
56 void plotFancyASCII();
57
60
62 std::string m_particleList;
63
72
79
81 std::vector<int> m_massConstraintList;
82
84 std::vector<int> m_geoConstraintListPDG;
85
89
91 std::vector<std::string> m_massConstraintListParticlename;
96
99
105
107 bool fitTree(Particle* head);
108
111
113 unsigned int m_nCandidatesAfter;
114
119
123
126
128 std::vector<double> m_customOriginVertex;
129
131 std::vector<double> m_customOriginCovariance;
132
136 std::vector<std::string> m_removeConstraintList;
137
142
145
148
150 Eigen::Matrix<double, 4, 1> m_beamMomE;
151
153 Eigen::Matrix<double, 4, 4> m_beamCovariance;
154
157
160
163
166
169
170 };
172}
The DecayDescriptor stores information about a decay tree or parts of a decay tree.
Base class for Modules.
Definition: Module.h:72
Class to store reconstructed particles.
Definition: Particle.h:75
Accessor to arrays stored in the data store.
Definition: StoreArray.h:113
Type-safe access to single objects in the data store.
Definition: StoreObjPtr.h:96
Module to fit an entire decay tree.
DecayDescriptor m_pDDescriptorForIgnoring
Decay descriptor of the ignored particles.
void plotFancyASCII()
plot ascii art and statistics
std::string m_ignoreFromVertexFit
decay string to select one particle that will be ignored to determine the vertex position
std::vector< double > m_customOriginVertex
vertex coordinates of the custom origin
virtual void initialize() override
initialize
virtual void event() override
performed for each event
Eigen::Matrix< double, 4, 1 > m_beamMomE
beam four-momentum
bool m_updateDaughters
flag if you want to update all particle momenta in the decay tree.
DecayDescriptor m_pDDescriptorInvisibles
Decay descriptor of the invisible particles.
std::vector< std::string > m_massConstraintListParticlename
vector carrying the names of the particles to be mass constraint
unsigned int m_nCandidatesAfter
after the fit
std::vector< int > m_massConstraintList
vector carrying the PDG codes of the particles to be mass constraint
std::vector< int > m_fixedToMotherVertexListPDG
list of pdg codes of particles where we use the same vertex for production and decay which is the ver...
StoreArray< Particle > m_particles
StoreArray of Particles.
virtual void terminate() override
stuff at the end
bool m_automatic_vertex_constraining
should the vertex be joined with the mother and should it be geometrically constrained?...
virtual void beginRun() override
performed at the start of run
bool m_ipConstraint
Use x-y-z beamspot constraint.
int m_originDimension
dimension to use for beam/origin constraint
int m_inflationFactorCovZ
inflate beamspot covariance of z by this number
std::vector< std::string > m_removeConstraintList
list of constraints not to be applied in tree fit WARNING only use if you know what you are doing
int m_massConstraintType
type of the mass constraint false: use normal one.
std::vector< int > m_geoConstraintListPDG
list of pdg codes of particles to use a geo constraint for
std::vector< double > m_customOriginCovariance
covariance of the custom origin
std::string m_particleList
name of the particle list fed to the fitter
double m_confidenceLevel
minimum confidence level to accept fit calculated as f(chiSquared, NDF) -2: accept all 0: only accept...
bool m_customOrigin
use a custom vertex as the production vertex of the highest hierarchy particle
std::string m_treatAsInvisible
decay string to select one particle that will be treated as invisible
Eigen::Matrix< double, 4, 4 > m_beamCovariance
beam covariance matrix
int m_beamConstraintPDG
PDG code of particle to be constrained to the beam 4-momentum.
bool m_useReferencing
linearise around a previous state of the Kalman Filter
bool fitTree(Particle *head)
this fits all particle candidates contained in the m_particleList
double m_precision
convergence precision for the newton method When the delta chiSquared between 2 iterations divided by...
StoreObjPtr< ParticleList > m_plist
input particle list
unsigned int m_nCandidatesBeforeFit
before the fit
Abstract base class for different kinds of events.
Definition: ClusterUtils.h:24