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);

  ~GsfVertexSmoother() override {}

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

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

  VertexSmoother<5>* clone() const override { 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