TKinFitter.h

Go to the documentation of this file.

#ifndef TKinFitter_h
#define TKinFitter_h

#include <vector>
#include "TMatrixD.h"
#include "TNamed.h"
#include "PhysicsTools/KinFitter/interface/TAbsFitParticle.h"

class TAbsFitConstraint;
class TH1D;
class TLorentzVector;

class TKinFitter : public TNamed {

public :

  TKinFitter();
  TKinFitter(const TString &name, const TString &title);  
  ~TKinFitter() override;
  void reset();         
  void resetStatus();   

  Int_t fit();

  void addMeasParticle( TAbsFitParticle* particle );
  void addMeasParticles( TAbsFitParticle* p1, TAbsFitParticle* p2 = nullptr, TAbsFitParticle* p3 = nullptr, 
             TAbsFitParticle* p4 = nullptr, TAbsFitParticle* p5 = nullptr, TAbsFitParticle* p6 = nullptr,
             TAbsFitParticle* p7 = nullptr, TAbsFitParticle* p8 = nullptr, TAbsFitParticle* p9 = nullptr);
  void addUnmeasParticle( TAbsFitParticle* particle );
  void addUnmeasParticles( TAbsFitParticle* p1, TAbsFitParticle* p2 = nullptr, TAbsFitParticle* p3 = nullptr, 
               TAbsFitParticle* p4 = nullptr, TAbsFitParticle* p5 = nullptr, TAbsFitParticle* p6 = nullptr,
               TAbsFitParticle* p7 = nullptr, TAbsFitParticle* p8 = nullptr, TAbsFitParticle* p9 = nullptr);
  void addConstraint( TAbsFitConstraint* constraint );

  Int_t getNDF() { return  (_constraints.size() - _nParA); }
  Int_t getNParA() { return _nParA; }
  Int_t getNParB() { return _nParB; }
  void setMaxNbIter( Int_t maxNbIter ) { _maxNbIter = maxNbIter; }
  Int_t getMaxNumberIter() { return _maxNbIter; }
  Int_t getNbIter() { return _nbIter; }
  Int_t getStatus() { return _status; }
  void setMaxDeltaS( Double_t maxDeltaS ) { _maxDeltaS = TMath::Abs( maxDeltaS ); }
  Double_t getMaxDeltaS() { return _maxDeltaS; }
  void setMaxF( Double_t maxF ) { _maxF = TMath::Abs( maxF ); }
  Double_t getMaxF() { return _maxF; }
  const TMatrixD* getCovMatrix() { return &_V; }
  void setCovMatrix( TMatrixD &V );
  const TMatrixD* getCovMatrixFit() { return &_yaVFit; }
  Double_t getS();
  Double_t getF();
  void setVerbosity( Int_t verbosity = 1 );
  Int_t getVerbosity( ) { return _verbosity; }

  Int_t nbMeasParticles() { return _measParticles.size(); }
  const TAbsFitParticle* getMeasParticle( Int_t index ) { return _measParticles[index]; }
  const TLorentzVector* get4Vec( Int_t index ) { return (_measParticles[index])->getCurr4Vec(); }

  Int_t nbUnmeasParticles() { return _unmeasParticles.size(); }
  const TAbsFitParticle* getUnmeasParticle( Int_t index ) { return _unmeasParticles[index]; }
  Int_t nbConstraints() { return _constraints.size(); }

  void print();

protected:

  Bool_t calcA();
  Bool_t calcB();
  Bool_t calcVA();
  Bool_t calcVB();
  Bool_t calcC();

  Bool_t calcC11();
  Bool_t calcC21();
  Bool_t calcC22();
  Bool_t calcC31();
  Bool_t calcC32();
  Bool_t calcC33();

  Bool_t calcDeltaA();
  Bool_t calcDeltaY();
  Bool_t calcLambda();

  Bool_t calcV();
  Bool_t calcVFit();

  Bool_t applyDeltaA();
  Bool_t applyDeltaY();
  void applyVFit();

  Bool_t converged(Double_t F, Double_t prevS, Double_t currS);

  TString getStatusString();
  void countMeasParams();
  void countUnmeasParams();
  void resetParams();

  void printMatrix(const TMatrixD &matrix, const TString& name = "");

private :

  Int_t _maxNbIter;       // Maximum number of iterations
  Double_t _maxDeltaS;    // Convergence criterium for deltaS
  Double_t _maxF;       // Convergence criterium for F
  Int_t _verbosity;       // Verbosty of the fitter 0: quiet, 1: print result, 2: print iterations, 3: print also matrices

  TMatrixD _A;      // Jacobi Matrix of unmeasured parameters
  TMatrixD _AT;     // Transposed Jacobi Matrix of unmeasured parameters
  TMatrixD _B;      // Jacobi Matrix of measured parameters
  TMatrixD _BT;     // Transposed Jacobi Matrix of measured parameters
  TMatrixD _V;      // Covariance matrix
  TMatrixD _Vinv;   // Inverse covariance matrix
  TMatrixD _VB;     // VB    = ( B*V*BT )^(-1)
  TMatrixD _VBinv;  // VBinv = ( B*V*BT )
  TMatrixD _VA;     // VA    = ( AT*VB*A )
  TMatrixD _VAinv;  // VAinv = ( AT*VB*A )^(-1)
  TMatrixD _c;      // Vector c = A*delta(a*) + B*delta(y*) - f*

  TMatrixD _C11;     // Matrix C11
  TMatrixD _C11T;    // Matrix C11T
  TMatrixD _C21;     // Matrix C21
  TMatrixD _C21T;    // Matrix C21T
  TMatrixD _C22;     // Matrix C22
  TMatrixD _C22T;    // Matrix C22T
  TMatrixD _C31;     // Matrix C31
  TMatrixD _C31T;    // Matrix C31T
  TMatrixD _C32;     // Matrix C32
  TMatrixD _C32T;    // Matrix C32T
  TMatrixD _C33;     // Matrix C33
  TMatrixD _C33T;    // Matrix C33T

  TMatrixD _deltaA;  // The correction vector deltaA for unmeasured particles of the current iteration
  TMatrixD _deltaY;  // The correction vector deltaY for measured particles of the current iteration
  TMatrixD _deltaAstar; // The correction vector deltaA for unmeasured particles of the previous iteration
  TMatrixD _deltaYstar; // The correction vector deltaY for measured particles of the previous iteration
  TMatrixD _lambda;  // The column vector of Lagrange multiplicators (likelihood L = S + 2 sum_i lambda_i * f_i)
  TMatrixD _lambdaT; // The row vector of Lagrange multiplicators (likelihood L = S + 2 sum_i lambda_i * f_i)

  TMatrixD _lambdaVFit;   // Covariance matrix of lambda after the fit
  TMatrixD _yaVFit;       // Combined covariance matrix of y and a after the fit

  Int_t _nParA;     // Number of unmeasured parameters
  Int_t _nParB;     // Number of measured parameters

  std::vector<TAbsFitConstraint*> _constraints;    // vector with constraints
  std::vector<TAbsFitParticle*> _measParticles;    // vector with measured particles
  std::vector<TAbsFitParticle*> _unmeasParticles;  // vector with unmeasured particles

  Int_t _status;        // Status of the last fit;_
  Int_t _nbIter;        // number of iteration performed in the fit

  ClassDefOverride(TKinFitter, 0)
};

#endif