GsfVertexSmoother.h

Go to the documentation of this file.

#ifndef _GsfVertexSmoother_H_
#define _GsfVertexSmoother_H_


#include "RecoVertex/VertexPrimitives/interface/VertexSmoother.h"
#include "RecoVertex/GaussianSumVertexFit/interface/GsfVertexUpdator.h"
#include "RecoVertex/GaussianSumVertexFit/interface/GsfVertexMerger.h"
#include "RecoVertex/KalmanVertexFit/interface/KalmanVertexUpdator.h"
#include "RecoVertex/KalmanVertexFit/interface/KalmanVertexTrackUpdator.h"
#include "RecoVertex/KalmanVertexFit/interface/KalmanSmoothedVertexChi2Estimator.h"
#include "RecoVertex/GaussianSumVertexFit/interface/GsfVertexWeightCalculator.h"
#include "RecoVertex/VertexTools/interface/VertexTrackFactory.h"
#include "DataFormats/GeometryCommonDetAlgo/interface/DeepCopyPointerByClone.h"
#include "RecoVertex/KalmanVertexFit/interface/KVFHelper.h"


class GsfVertexSmoother : public VertexSmoother<5> {

public:

  GsfVertexSmoother(bool limit, const GsfVertexMerger * merger);

  virtual ~GsfVertexSmoother() {}

  virtual CachingVertex<5> smooth(const CachingVertex<5> & vertex) const;

  const VertexUpdator<5> * vertexUpdator() const
  {return &theUpdator;}

  virtual VertexSmoother<5> * clone() const 
  {
    return new GsfVertexSmoother(* this);
  }
  
private:

  typedef CachingVertex<5>::RefCountedVertexTrack RefCountedVertexTrack;
  typedef VertexTrack<5>::RefCountedLinearizedTrackState RefCountedLinearizedTrackState;
  typedef VertexTrack<5>::RefCountedRefittedTrackState RefCountedRefittedTrackState;
  typedef VertexTrack<5>::AlgebraicMatrix3M AlgebraicMatrix3M;
  typedef VertexTrack<5>::AlgebraicSymMatrixOO AlgebraicSymMatrixOO;

  typedef std::vector<VertexState> VSC;
  typedef std::vector<RefCountedLinearizedTrackState> LTC;
  typedef std::pair<double, double> VtxTrkChi2Pair;
  typedef std::pair<RefCountedRefittedTrackState, double> TrackWeightPair;
  typedef std::pair<RefCountedRefittedTrackState, VtxTrkChi2Pair> TrackChi2Pair;
  typedef std::pair<TrackWeightPair, VtxTrkChi2Pair> RefittedTrackComponent;

  VertexState meanVertex(const VertexState & vertexA,
             const VertexState & vertexB) const;

  TrackChi2Pair vertexAndTrackUpdate(const VertexState & oldVertex,
    const RefCountedVertexTrack track, const GlobalPoint & referencePosition) const;

  RefittedTrackComponent createNewComponent(const VertexState & oldVertex,
     const RefCountedLinearizedTrackState linTrack, float weight) const;

  TrackChi2Pair assembleTrackComponents(
    const std::vector<RefittedTrackComponent> & trackComponents,
    const GlobalPoint & referencePosition) const;

  double priorVertexChi2(const VertexState priorVertex, 
    const VertexState fittedVertex) const;

  bool limitComponents;
  DeepCopyPointerByClone<GsfVertexMerger> theMerger;
  GsfVertexUpdator theUpdator;
  KalmanVertexUpdator<5> kalmanVertexUpdator;
  KalmanSmoothedVertexChi2Estimator<5> smoothedChi2Estimator;
  KalmanVertexTrackUpdator<5> theVertexTrackUpdator;       
  GsfVertexWeightCalculator theWeightCalculator;
  VertexTrackFactory<5> theVTFactory;
  KVFHelper<5> helper;
};

#endif