MultiVertexFitter.h

Go to the documentation of this file.

#ifndef _MultiVertexFitter_H_
#define _MultiVertexFitter_H_

#include <vector>
#include <set>
#include <utility>
#include <map>
#include "RecoVertex/VertexPrimitives/interface/CachingVertex.h"
#include "DataFormats/GeometryVector/interface/GlobalPoint.h"
#include "RecoVertex/MultiVertexFit/interface/DefaultMVFAnnealing.h"
#include "TrackingTools/TransientTrack/interface/TransientTrack.h"
#include "RecoVertex/LinearizationPointFinders/interface/DefaultLinearizationPointFinder.h"
#include "RecoVertex/MultiVertexFit/interface/LinTrackCache.h"

class MultiVertexFitter {
public:
  MultiVertexFitter(const AnnealingSchedule &sched = DefaultMVFAnnealing(),
                    const LinearizationPointFinder &seeder = DefaultLinearizationPointFinder(),
                    float revive_below = -1.);
  MultiVertexFitter(const MultiVertexFitter &);
  ~MultiVertexFitter();

  typedef std::pair<reco::TransientTrack, float> TrackAndWeight;
  typedef std::map<int, double> SeedToWeightMap;
  typedef std::map<reco::TransientTrack, SeedToWeightMap> TrackAndSeedToWeightMap;

  std::vector<CachingVertex<5> > vertices(
      const std::vector<std::vector<reco::TransientTrack> > &,
      const std::vector<reco::TransientTrack> &primaries = std::vector<reco::TransientTrack>());

  std::vector<CachingVertex<5> > vertices(
      const std::vector<std::vector<TrackAndWeight> > &,
      const std::vector<reco::TransientTrack> &primaries = std::vector<reco::TransientTrack>());

  std::vector<CachingVertex<5> > vertices(
      const std::vector<CachingVertex<5> > &,
      const std::vector<reco::TransientTrack> &primaries = std::vector<reco::TransientTrack>());

  std::vector<CachingVertex<5> > vertices(
      const std::vector<TransientVertex> &,
      const std::vector<reco::TransientTrack> &primaries = std::vector<reco::TransientTrack>());

private:
  std::vector<CachingVertex<5> > fit();
  void lostVertexClaimer();
  void updateWeights();
  bool updateSeeds();

  void clear();
  void createSeed(const std::vector<reco::TransientTrack> &tracks);
  void createSeed(const std::vector<TrackAndWeight> &tracks);
  void createPrimaries(const std::vector<reco::TransientTrack> &tracks);
  void printWeights() const;
  void printWeights(const reco::TransientTrack &) const;
  void printSeeds() const;

  int seedNr();
  void resetSeedNr();

private:
  typedef CachingVertex<5>::RefCountedVertexTrack RefCountedVertexTrack;
  typedef ReferenceCountingPointer<LinearizedTrackState<5> > RefCountedLinearizedTrackState;

  // vertex seeds. Come with a seed number
  // makes the weight table simpler, faster,
  // and more reliable (I hope)
  std::vector<std::pair<int, CachingVertex<5> > > theVertexStates;
  int theVertexStateNr;
  float theReviveBelow;
  std::vector<reco::TransientTrack> theTracks;
  std::set<reco::TransientTrack> thePrimaries;
  AnnealingSchedule *theAssComp;
  LinearizationPointFinder *theSeeder;
  TrackAndSeedToWeightMap theWeights;
  LinTrackCache theCache;
};

#endif