InternalParticle.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>
#include <analysis/VertexFitting/TreeFitter/RecoTrack.h>
 
namespace TreeFitter {
 
  class InternalParticle : public ParticleBase {
 
  public:
 
    InternalParticle(Belle2::Particle* particle,
                     const ParticleBase* mother,
                     const ConstraintConfiguration& config,
                     bool forceFitAll
                    ) ;
 
    virtual ~InternalParticle() {};
 
    virtual ErrCode initCovariance(FitParams&) const override;
 
    ErrCode projectKineConstraint(const FitParams&, Projection&) const;
 
    ErrCode projectBeamConstraint(const FitParams&, Projection&) const;
 
    virtual void forceP4Sum(FitParams&) const override;
 
    virtual ErrCode initParticleWithMother(FitParams& fitparams) override;
 
    virtual ErrCode initMotherlessParticle(FitParams& fitparams) override;
 
    ErrCode projectConstraint(const Constraint::Type type, const FitParams& fitparams, Projection& p) const override;
 
    virtual int dim() const override;
 
    virtual int type() const override { return kInternalParticle ; }
 
    virtual int posIndex() const override;
 
    virtual int tauIndex() const override;
 
    virtual int momIndex() const override;
 
    virtual bool hasEnergy() const override { return true ; }
 
    virtual bool hasPosition() const override;
 
    virtual std::string parname(int index) const override ;
 
    virtual void addToConstraintList(constraintlist& list, int depth) const override ;
 
    void setMassConstraint(bool b) { m_massconstraint = b ; }
 
  protected:
 
    ErrCode initMomentum(FitParams& fitparams) const ;
 
  private:
 
    bool static compTrkTransverseMomentum(const RecoTrack* lhs, const RecoTrack* rhs);
 
    bool m_massconstraint ;
 
    bool m_beamconstraint ;
 
    bool m_shares_vertex_with_mother;
 
    bool m_geo_constraint;
 
    bool m_lifetimeconstraint ;
 
    bool m_isconversion ;
 
    bool m_automatic_vertex_constraining;
  } ;
 
}