Origin.h

/**************************************************************************
 *
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2018 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributor: wouter Hulsbergen, Jo-Frederik Krohn  *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
#pragma once
 
#include <analysis/VertexFitting/TreeFitter/ParticleBase.h>
 
#include <mdst/dbobjects/BeamSpot.h>
#include <framework/database/DBObjPtr.h>
 
namespace TreeFitter {
 
  class Origin : public ParticleBase {
 
  public:
 
    Origin(Belle2::Particle* particle,
           const ConstraintConfiguration& config,
           const bool forceFitAll
          );
 
    explicit Origin(Belle2::Particle* daughter);
 
    virtual ~Origin() {};
 
    virtual  ErrCode initParticleWithMother(FitParams& fitparams) override;
 
    virtual  ErrCode initMotherlessParticle(FitParams& fitparams) override;
 
    ErrCode initOrigin();
 
    virtual int dim() const override { return m_constraintDimension; }
 
    virtual ErrCode initCovariance(FitParams& fitpar) const override;
 
    /* particle type */
    virtual int type() const override { return kOrigin; }
 
    ErrCode projectOriginConstraint(const FitParams& fitpar, Projection&) const;
 
    virtual ErrCode projectConstraint(Constraint::Type, const FitParams&, Projection&) const override;
 
    virtual void addToConstraintList(constraintlist& list, int depth) const override;
 
    virtual int posIndex() const override { return index(); }
 
    virtual int momIndex() const override { return -1; }
 
    virtual int tauIndex() const override { return -1; }
 
    virtual bool hasEnergy() const override { return false; }
 
    virtual std::string name() const { return "Origin"; }
 
  private:
 
    const int m_constraintDimension;
 
    const std::vector<double> m_customOriginVertex;
 
    const std::vector<double> m_customOriginCovariance;
 
    Eigen::Matrix<double, Eigen::Dynamic, 1, Eigen::ColMajor, 3, 1> m_posVec;
 
    Eigen::Matrix<double, Eigen::Dynamic, Eigen::Dynamic, Eigen::ColMajor, 3, 3> m_covariance;
 
    const bool m_isBeamSpot;
 
    Belle2::DBObjPtr<Belle2::BeamSpot> m_beamSpot;
 
    const int m_inflationFactorCovZ;
  };
}