TFitConstraintEp.h

Go to the documentation of this file.

#ifndef TFitConstraintEp_hh
#define TFitConstraintEp_hh

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

class TFitConstraintEp: public TAbsFitConstraint {

public :

  enum component {
    pX,
    pY,
    pZ,
    E
  };

  TFitConstraintEp( );

  TFitConstraintEp(  const TString &name, const TString &title,
                     TFitConstraintEp::component thecomponent,
                     Double_t constraint = 0.);

  TFitConstraintEp(  std::vector<TAbsFitParticle*>* particles, 
             TFitConstraintEp::component thecomponent, 
             Double_t constraint = 0.);

  TFitConstraintEp(  const TString &name, const TString &title,
             std::vector<TAbsFitParticle*>* particles, 
             TFitConstraintEp::component thecomponent, 
             Double_t constraint = 0.);
  virtual ~TFitConstraintEp();

  void addParticle( TAbsFitParticle* particle );
  void addParticles( TAbsFitParticle* p1, TAbsFitParticle* p2 = 0, TAbsFitParticle* p3 = 0, TAbsFitParticle* p4 = 0,
             TAbsFitParticle* p5 = 0, TAbsFitParticle* p6 = 0, TAbsFitParticle* p7 = 0, TAbsFitParticle* p8 = 0,
             TAbsFitParticle* p9 = 0, TAbsFitParticle* p10 = 0);
  void setConstraint(Double_t constraint){_constraint = constraint;};

  // returns derivative df/dP with P=(p,E) and f the constraint f=0.
  // The matrix contains one row (df/dp, df/dE).
  virtual TMatrixD* getDerivative( TAbsFitParticle* particle );
  virtual Double_t getInitValue();
  virtual Double_t getCurrentValue();

  virtual TString getInfoString();
  virtual void print(); 
 
protected :


private:
  std::vector<TAbsFitParticle*> _particles;    // Vector containing constrained particles
  Double_t _constraint;                   // Value of constraint
  TFitConstraintEp::component _component; // 4vector component to be constrained

};

#endif