Composite.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 * External Contributor: Wouter Hulsbergen                                *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
#pragma once
 
#include <analysis/VertexFitting/TreeFitter/ParticleBase.h>
 
namespace TreeFitter {
 
  class Composite : public ParticleBase {
 
  public:
    Composite(Belle2::Particle* bc, const ParticleBase* mother, const ConstraintConfiguration& config,
              bool massconstraint = false);
 
    virtual ~Composite() {};
 
    virtual ErrCode initParticleWithMother(FitParams& fitparams) override;
 
    virtual ErrCode initMotherlessParticle(FitParams& fitparams) override;
 
    void updateParams();
 
    ErrCode projectComposite(const FitParams& fitparams, Projection& p) const;
 
    int dimMeas() const        { return m_hasEnergy ? 7 : 6 ; }
 
    virtual int dim() const override { return m_hasEnergy ? 8 : 7 ; }// (x,y,z,t,px,py,pz,(E))
 
    int dimM() const        { return m_hasEnergy ? 7 : 6 ; }
 
    virtual ErrCode projectConstraint(Constraint::Type, const FitParams&, Projection&) const override ;
 
    virtual int type() const override { return kComposite ; }
 
    virtual int posIndex() const override { return index()   ; }
 
    virtual int tauIndex() const override { return index() + 3 ; }
 
    virtual int momIndex() const override { return index() + 4 ; }
 
    virtual bool hasEnergy() const override { return m_hasEnergy ; }
 
    virtual bool hasPosition() const override { return true ; }
 
    virtual void addToConstraintList(constraintlist& alist, int depth) const override
    {
      alist.push_back(Constraint(this, Constraint::composite, depth, dimM())) ;
      alist.push_back(Constraint(this, Constraint::geometric, depth, 3)) ;
      if (m_massconstraint) {
        alist.push_back(Constraint(this, Constraint::mass, depth, 1, 3));
      }
    }
 
  protected:
 
    Eigen::Matrix<double, 7, 1> m_params;
 
    Eigen::Matrix < double, -1, -1, 0, 7, 7 >  m_covariance;
 
    bool m_hasEnergy;
 
    bool m_massconstraint;
  };
}