FitManager.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * right(C) 2018 - Belle II Collaboration                                 *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributor: Wouter Hulsbergen, Francesco Tenchini, Jo-Frederik Krohn  *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
#pragma once
 
#include <analysis/dataobjects/Particle.h>
#include <analysis/VertexFitting/TreeFitter/ErrCode.h>
#include <analysis/VertexFitting/TreeFitter/ConstraintConfiguration.h>
 
namespace TreeFitter {
  class DecayChain;
  class FitParams;
  class ParticleBase;
  class ConstraintConfiguration;
 
  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_niter(-1), m_prec(0.01), m_updateDaugthers(false), m_ndf(0),
      m_fitparams(0), m_useReferencing(false)
    {}
 
    FitManager(Belle2::Particle* particle,
               const ConstraintConfiguration& config,
               double prec = 0.01,
               bool updateDaughters = false,
               const bool useReferencing = false
              );
 
    FitManager(const FitManager& other) = delete;
 
    FitManager& operator=(const FitManager& other) = delete;
 
    ~FitManager();
 
    bool fit();
 
    void setExtraInfo(Belle2::Particle* part, const std::string& name, const double value) const;
 
    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;
 
    int posIndex(Belle2::Particle* particle) const;
 
    int momIndex(Belle2::Particle* particle) const;
 
    int tauIndex(Belle2::Particle* particle) const;
 
    double chiSquare() const { return m_chiSquare ; }
 
    int nDof() const;
 
    int status() const { return m_status; }
 
    int nIter() const { return m_niter; }
 
    const ErrCode& errCode() { return m_errCode; }
 
    const DecayChain* decaychain() const { return m_decaychain; }
 
    Belle2::Particle* particle() { return m_particle; }
 
  private:
    Belle2::Particle* m_particle;
 
    DecayChain* m_decaychain;
 
    int m_status;
 
    double m_chiSquare;
 
    int m_niter;
 
    double m_prec;
 
    ErrCode m_errCode;
 
    const bool m_updateDaugthers;
 
    int m_ndf;
 
    FitParams* m_fitparams;
 
    bool m_useReferencing;
 
    const ConstraintConfiguration m_config;
  };
}