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