FitManager.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/ErrCode.h>
#include <analysis/VertexFitting/TreeFitter/ConstraintConfiguration.h>
 
namespace Belle2 {
  class Particle;
}
 
namespace TreeFitter {
  class DecayChain;
  class FitParams;
  class ParticleBase;

  class FitManager {
  public:

    enum VertexStatus { Success = 0, NonConverged, BadInput, Failed, UnFitted };

    FitManager() : m_particle(0), m_decaychain(0), m_status(VertexStatus::UnFitted),
      m_chiSquare(-1), m_prec(0.01), m_updateDaugthers(false), m_ndf(0),
      m_fitparams(0)
    {}

    FitManager(Belle2::Particle* particle,
               const ConstraintConfiguration& config,
               double prec = 0.01,
               bool updateDaughters = false
              );

    FitManager(const FitManager& other) = delete;

    FitManager& operator=(const FitManager& other) = delete;

    ~FitManager();

    bool fit();

    bool updateCand(Belle2::Particle& particle, const bool isTreeHead) const;

    void updateCand(const ParticleBase& pb, Belle2::Particle& cand, const bool isTreeHead) const;

    void updateTree(Belle2::Particle& particle, const bool isTreeHead) const;

    void getCovFromPB(const ParticleBase* pb, TMatrixFSym& returncov) const;

    std::tuple<double, double> getLifeTime(Belle2::Particle& cand) const;

    std::tuple<double, double> getDecayLength(const ParticleBase* pb) const;

    std::tuple<double, double> getDecayLength(const ParticleBase* pb, const FitParams& fitparams) const;

    std::tuple<double, double> getDecayLength(Belle2::Particle& cand) const;

    Belle2::Particle* particle() { return m_particle; }
 
  private:
    Belle2::Particle* m_particle;

    DecayChain* m_decaychain;

    int m_status;

    double m_chiSquare;

    double m_prec;

    ErrCode m_errCode;

    const bool m_updateDaugthers;

    int m_ndf;

    FitParams* m_fitparams;

    const ConstraintConfiguration m_config;
  };
}