FitParams.h

Go to the documentation of this file.

// ******************************************************
// DecayTreeFitter Package
// We thank the original author Wouter Hulsbergen
// (BaBar, LHCb) for providing the sources.
// http://arxiv.org/abs/physics/0503191v1 (2005)
// Adaptation & Development for PANDA: Ralf Kliemt (2015)
// ******************************************************
#ifndef FITPARAMS_H
#define FITPARAMS_H 1

#include <iostream>
#include <vector>
#include <cmath>
#include <map>
#include "TVectorD.h"
#include "TMatrixDSym.h"
#include "ChiSquare.h"
#include "Rtypes.h"

namespace DecayTreeFitter {
class ParticleBase;

class FitParams {
 public:
  // Class that contains the parameters and covariance for the
  // vertex fit.
  FitParams(int dim);

  FitParams(const FitParams &par);
  virtual ~FitParams();

  TMatrixDSym &cov() { return m_cov; }
  TVectorD &par() { return m_par; }
  double &par(int row) { return m_par(row); }

  TMatrixDSym cov(const std::vector<int> &indexVec) const;
  TVectorD par(const std::vector<int> &indexVec) const;

  const TMatrixDSym &cov() const { return m_cov; }
  const TVectorD &par() const { return m_par; }
  const double &par(int row) const { return m_par(row); }

  // TMatrixDSym& scale() { return m_scale ; }

  int &nConstraintsVec(int row) { return m_nConstraintsVec[row /*-1*/]; }

  // int dim() const { return m_par.num_row() ; }
  int dim() const { return m_dim; }
  double chiSquare() const { return m_chiSquare; }

  int nConstraints() const { return m_nConstraints; }
  int nDof() const { return nConstraints() - dim(); }
  double err(int row) const { return sqrt(m_cov(row, row)); }

  void resize(int newdim);
  void reset(int newdim);
  void resetPar();
  void resetCov(double scale = 100);
  void print() const;
  bool testCov() const;
  void addChiSquare(double chi2, int nconstraints, const ParticleBase *p);
  ChiSquare chiSquare(const ParticleBase &p) const;

 protected:
  FitParams(){};

 private:
  int m_dim;
  TVectorD m_par;
  TMatrixDSym m_cov;
  // TMatrixDSym m_scale ;
  double m_chiSquare;
  int m_nConstraints;
  std::vector<int> m_nConstraintsVec; // vector with number of constraints per parameter
  std::map<const ParticleBase *, ChiSquare> m_chiSquareMap;
  ClassDef(FitParams, 1)
};
} // namespace DecayTreeFitter

#endif