RhoKinVtxFitter.h

Go to the documentation of this file.

#ifndef RhoKinVtxFitter_H
#define RhoKinVtxFitter_H
//                                                                      //
// RhoKinVtxFitter                  //
//                                                                      //
// Author: V. Jha 2010                //
//                                                                      //

#include "RhoFitterBase.h"
#include "RhoCandList.h"

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

class RhoCandidate;

class RhoKinVtxFitter : public RhoFitterBase {
 public:
  RhoKinVtxFitter(RhoCandidate *b);
  virtual ~RhoKinVtxFitter();
  void AddMassConstraint(double mass);
  // void AddPointingConstraint(TVector3 pVtx);
  double GetPull() { return fPull; }

  void SetNMaxIterations(int nit = 20)
  {
    fNMaxIterations = nit;
    fIterateExact = false;
  };
  void SetNIterationsExact(int nit = 2)
  {
    fNMaxIterations = nit;
    fIterateExact = true;
  };
  void SetMinDChisq(double m = 0.001) { fMinDChisq = fabs(m); };

 private:
  Bool_t FitNode(RhoCandidate *b);

  void SetMatrices();
  void ResetMatrices();
  void ReadMatrix();

  void ReadKinMatrix();
  void ReadMassKinMatrix();
  // void ReadPointingKinMatrix() {ReadPointingKinMatrix(fHeadOfTree);};
  // void ReadPointingKinMatrix(RhoCandidate* head);

  Bool_t Compute(RhoCandidate *c);
  void SetOutput(RhoCandidate *head);

  void TransportToVertex(TMatrixD &, TMatrixD &, TMatrixD &, TMatrixD &, TMatrixD &);
  void GetCovariance(TMatrixD &a_cov0, TMatrixD &cov_al_x, TMatrixD &V_vtx, TMatrixD &covS);

  TMatrixD al0;   // original parameters
  TMatrixD al1;   // fitted parameters
  TMatrixD V_al0; // Covariance matrix for original parameters
  TMatrixD V_al1; // Covariance matrix for original parameters
  TMatrixD covC;  // Covariance matrix for composite

  TMatrixD mD; // Matrix of constraint derivitives
  TMatrixD mE; // Matrix of vertex constraints
  TMatrixD md; // Vector of evaluated constraints
  TMatrixD mPull;

  Int_t fNvar;  // Number of variables (=4 for 4 vector)
  Int_t fNpar;  // Number of parameters=Npart*fNvar
  Int_t fNpart; // Number of particles
  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 fNumKnown; // Number of unknowns
  Int_t NumCon;    // Number of constraints ...

  int niter;
  double fChi2Diff;
  double fPull;

  // VtxVertex                   _vertex;
  TMatrixD vtx_ex;
  TMatrixD vtx_st;

  double fMass;
  // TVector3 fpVtx;

  int fMassConstraint;
  // int fPointConstraint;

  // control abortion of fit
  double fMinDChisq;
  int fNMaxIterations;
  bool fIterateExact;

  int fnDof;
  double fchiSquare;

  ClassDef(RhoKinVtxFitter, 1) // A Kinematic Vertex Fitter
};

#endif