Belle II Software prerelease-10-00-00a
Composite.h
1/**************************************************************************
2 * basf2 (Belle II Analysis Software Framework) *
3 * Author: The Belle II Collaboration *
4 * External Contributor: Wouter Hulsbergen *
5 * *
6 * See git log for contributors and copyright holders. *
7 * This file is licensed under LGPL-3.0, see LICENSE.md. *
8 **************************************************************************/
9#pragma once
10
11#include <analysis/VertexFitting/TreeFitter/ParticleBase.h>
12
13namespace TreeFitter {
14
17 class Composite : public ParticleBase {
18
19 public:
22 bool massconstraint = false);
23
25 virtual ~Composite() {};
26
28 virtual ErrCode initParticleWithMother(FitParams& fitparams) override;
29
31 virtual ErrCode initMotherlessParticle(FitParams& fitparams) override;
32
34 void updateParams();
35
37 ErrCode projectComposite(const FitParams& fitparams, Projection& p) const;
38
40 virtual int dim() const override { return m_hasEnergy ? 8 : 7 ; }// (x,y,z,t,px,py,pz,(E))
41
43 int dimM() const { return m_hasEnergy ? 7 : 6 ; }
44
46 virtual ErrCode projectConstraint(Constraint::Type, const FitParams&, Projection&) const override ;
47
49 virtual int type() const override { return kComposite ; }
50
52 virtual int posIndex() const override { return index() ; }
53
55 virtual int tauIndex() const override { return index() + 3 ; }
56
58 virtual int momIndex() const override { return index() + 4 ; }
59
61 virtual bool hasEnergy() const override { return m_hasEnergy ; }
62
64 virtual bool hasPosition() const override { return true ; }
65
67 virtual void addToConstraintList(constraintlist& alist, int depth) const override
68 {
69 alist.push_back(Constraint(this, Constraint::composite, depth, dimM())) ;
70 alist.push_back(Constraint(this, Constraint::geometric, depth, 3)) ;
71 if (m_massconstraint) {
72 alist.push_back(Constraint(this, Constraint::mass, depth, 1, 3));
73 }
74 }
75
76 protected:
77
79 Eigen::Matrix<double, 7, 1> m_params;
80
82 Eigen::Matrix < double, -1, -1, 0, 7, 7 > m_covariance;
83
86
89 };
90}
Class to store reconstructed particles.
Definition Particle.h:76
virtual bool hasPosition() const override
return true FIXME
Definition Composite.h:64
virtual ~Composite()
destructor
Definition Composite.h:25
int dimM() const
get dimension of measurement
Definition Composite.h:43
virtual int tauIndex() const override
get tau (lifetime) index in statevector
Definition Composite.h:55
Composite(Belle2::Particle *bc, const ParticleBase *mother, const ConstraintConfiguration &config, bool massconstraint=false)
constructor
Definition Composite.cc:19
virtual int dim() const override
get dimension of constraint
Definition Composite.h:40
void updateParams()
update changed params
Definition Composite.cc:43
virtual ErrCode projectConstraint(Constraint::Type, const FitParams &, Projection &) const override
project this constraint
Definition Composite.cc:97
bool m_massconstraint
flag
Definition Composite.h:88
Eigen::Matrix< double, 7, 1 > m_params
column vector to store the measurement
Definition Composite.h:79
virtual int momIndex() const override
get momentum index in statevector
Definition Composite.h:58
Eigen::Matrix< double, -1, -1, 0, 7, 7 > m_covariance
only lower triangle filled!
Definition Composite.h:82
virtual int type() const override
get type
Definition Composite.h:49
virtual bool hasEnergy() const override
return of this constraint/particle has an energy component
Definition Composite.h:61
virtual void addToConstraintList(constraintlist &alist, int depth) const override
add this to list
Definition Composite.h:67
virtual ErrCode initParticleWithMother(FitParams &fitparams) override
init particle in case it has a mother
Definition Composite.cc:26
virtual ErrCode initMotherlessParticle(FitParams &fitparams) override
init particle in case it has no mother
Definition Composite.cc:31
ErrCode projectComposite(const FitParams &fitparams, Projection &p) const
project this particle constraint
Definition Composite.cc:75
virtual int posIndex() const override
get position index in statevectof x,y,z,tau,px,py,pz
Definition Composite.h:52
class to manage the order of constraints and their filtering
Definition Constraint.h:20
Type
type of constraints the order of these constraints is important: it is the order in which they are ap...
Definition Constraint.h:27
abstract errorocode be aware that the default is success
Definition ErrCode.h:14
Class to store and manage fitparams (statevector)
Definition FitParams.h:20
ParticleBase(Belle2::Particle *particle, const ParticleBase *mother, const ConstraintConfiguration *config=nullptr)
default constructor
int index() const
get index
const ParticleBase * mother() const
getMother() / hasMother()
std::vector< Constraint > constraintlist
alias
class to store the projected residuals and the corresponding jacobian as well as the covariance matri...
Definition Projection.h:18