KinematicParticle.h

Go to the documentation of this file.

#ifndef KinematicParticle_H
#define KinematicParticle_H
 
#include "RecoVertex/KinematicFitPrimitives/interface/ParticleMass.h"
#include "RecoVertex/KinematicFitPrimitives/interface/KinematicState.h"
#include "RecoVertex/VertexPrimitives/interface/LinearizedTrackState.h"
 
#include "DataFormats/GeometrySurface/interface/ReferenceCounted.h"
 
class KinematicConstraint;
class KinematicTree;
 
class KinematicParticle : public ReferenceCounted {
public:
  typedef ReferenceCountingPointer<LinearizedTrackState<6> > RefCountedLinearizedTrackState;
 
  friend class KinematicParticleVertexFitter;
  friend class KinematicTree;
 
  KinematicParticle() {}
 
  ~KinematicParticle() override;
 
  virtual bool operator==(const KinematicParticle& other) const = 0;
 
  virtual bool operator==(const ReferenceCountingPointer<KinematicParticle>& other) const = 0;
 
  virtual bool operator!=(const KinematicParticle& other) const = 0;
 
  virtual bool operator<(const KinematicParticle& other) const;
 
  virtual KinematicState initialState() const;
 
  virtual KinematicState currentState() const;
 
  virtual KinematicState stateAtPoint(const GlobalPoint& point) const = 0;
 
  virtual ReferenceCountingPointer<KinematicParticle> refittedParticle(const KinematicState& state,
                                                                       float chi2,
                                                                       float ndf,
                                                                       KinematicConstraint* cons = nullptr) const = 0;
 
  virtual RefCountedLinearizedTrackState particleLinearizedTrackState(const GlobalPoint& point) const = 0;
 
  virtual KinematicConstraint* lastConstraint() const;
 
  virtual ReferenceCountingPointer<KinematicParticle> previousParticle() const;
 
  virtual KinematicTree* correspondingTree() const;
 
  virtual float chiSquared() const;
 
  virtual float degreesOfFreedom() const;
 
  const MagneticField* magneticField() const { return theField; }
 
  reco::TransientTrack refittedTransientTrack() const;
 
protected:
  virtual void setTreePointer(KinematicTree* tr) const;
 
  const MagneticField* theField;
 
  //pointer to the tree current
  //particle  belongs to
  mutable KinematicTree* tree;
 
  //last constraint applied
  mutable KinematicConstraint* lConstraint;
 
  //previous particle
  mutable ReferenceCountingPointer<KinematicParticle> pState;
 
  //initial kinematic state of
  //current particle
  KinematicState initState;
 
  //particle state at point
  mutable KinematicState cState;
 
  //chi2 and number of degrees of freedom
  float chi2;
 
  float ndf;
};
#endif