RhoKinFitter.h

Go to the documentation of this file.

#ifndef RhoKinFitter_H
#define RhoKinFitter_H
//                                                                      //
// RhoKinFitter                 //
//                                                                      //
// V. Jha 2010  FZ Juelich                                              //
//                                                                      //

#include "RhoFitterBase.h"
#include "RhoCandList.h"
#include "TVector3.h"
#include "TMatrixDSym.h"
#include "TMatrixD.h"
#include "TLorentzVector.h"
#include "TDecompLU.h"

class RhoCandidate;

class RhoKinFitter : public RhoFitterBase {
 public:
  RhoKinFitter(RhoCandidate *b);
  virtual ~RhoKinFitter();

  void Add4MomConstraint(TLorentzVector lv);
  void AddMomConstraint(TVector3 v);
  void AddTotEConstraint(double energy);
  void AddTotMomConstraint(double momentum);
  void AddMassConstraint(double mass);
  Bool_t Fit();
  Bool_t FitAll()
  {
    Fatal("RhoKinFitter::FitAll()", "Not applicable. Please use Fit().");
    return kFALSE;
  };

  double Chi2Diff() { return fChi2Diff; }
  double GetPull() { return fPull; }
  double Chi2Contribution(const RhoCandidate &) const
  {
    Error("RhoKinFitter::Chi2Contribution", "No chi2 contribution available.");
    return -9999.;
  };

 private:
  void SetMatrices();
  void ZeroMatrices();
  void ReadMatrix();
  void ReadKinMatrix();
  void ReadMassKinMatrix();
  void Read4MomKinMatrix();
  void ReadMomKinMatrix();
  void ReadTotEKinMatrix();
  void ReadTotMomKinMatrix();
  Bool_t Solve();
  void SetOutput();

  TMatrixD fAl0;   // original parameters
  TMatrixD fAl1;   // fitted parameters
  TMatrixD fV_al0; // Covariance matrix for original parameters
  TMatrixD fV_al1; // Covariance matrix for original parameters

  TMatrixD fmD; // Matrix of constraint derivitives
  TMatrixD fmE; // Matrix of vertex constraints
  TMatrixD fmd; // Vector of evaluated constraints
  TMatrixD fmPull;

  Int_t fNvar;     // Number of variables (=4 for 4 vector)
  Int_t fNpar;     // Number of parameters=Npart*fNvar
  Int_t fNcon;     // Number of constraints
  Int_t fNc;       // countNumber of constraints added
  Int_t fNiter;    // Number of times Solve has been called
  Int_t fNunKnown; // Number of unknowns
  Int_t fNumCon;   // Number of constraints ...

  double fMass;
  TLorentzVector flmm;
  TVector3 fmm;
  double fEc;
  double fMom;

  int f4MomConstraint;
  int fMomConstraint;
  int fTotEConstraint;
  int fMassConstraint;
  int fTotMomConstraint;

  double fChi2Diff;
  double fPull;

  ClassDef(RhoKinFitter, 1)
};

#endif