InternalTrack.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2020 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributor: Francesco Tenchini, Jo-Frederik Krohn, Fabian Krinner     *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
#pragma once
 
#include <analysis/VertexFitting/TreeFitter/ParticleBase.h>
#include <analysis/VertexFitting/TreeFitter/RecoTrack.h>
 
namespace TreeFitter {
 
  class InternalTrack : public ParticleBase {
 
  public:
 
    InternalTrack(Belle2::Particle* particle,
                  const ParticleBase* mother,
                  const ConstraintConfiguration& config,
                  bool forceFitAll,
                  bool noEnergySum,
                  bool forceMassConstraint
                 );
 
    virtual ~InternalTrack() {};
 
    virtual ErrCode initCovariance(FitParams&) const override;
 
    ErrCode projectHelixConstraint(const FitParams&, Projection&) const;
 
    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 kInternalTrack ; }
 
    virtual int posIndex() const override;
 
    virtual int tauIndex() const override;
 
    virtual int momIndex() const override;
 
    virtual bool hasEnergy() const override;
 
    virtual bool hasPosition() const override;
 
    virtual void addToConstraintList(constraintlist& list, int depth) const override ;
 
    void setMassConstraint(bool b) { m_massconstraint = b ; }
 
    void forceP4Sum(FitParams& fitparams) const override;
 
  protected:
 
    ErrCode initMomentum(FitParams& fitparams) const ;
 
  private:
 
    bool static compTrkTransverseMomentum(const RecoTrack* lhs, const RecoTrack* rhs);
 
    bool m_massconstraint ;
 
    bool m_noEnergySum;
 
    bool m_isconversion;
 
    double m_bfield;
  } ;
 
}