AnnealingGhostTrackFitter.h

Go to the documentation of this file.

#ifndef RecoBTag_AnnealingGhostTrackFitter_h
#define RecoBTag_AnnealingGhostTrackFitter_h
 
#include <vector>
#include <memory>
 
#include "RecoVertex/VertexTools/interface/AnnealingSchedule.h"
 
#include "RecoVertex/GhostTrackFitter/interface/GhostTrackFitter.h"
#include "RecoVertex/GhostTrackFitter/interface/SequentialGhostTrackFitter.h"
 
namespace reco {
 
  class GhostTrackPrediction;
  class GhostTrackState;
 
  class AnnealingGhostTrackFitter : public SequentialGhostTrackFitter {
  public:
    AnnealingGhostTrackFitter();
    AnnealingGhostTrackFitter(const AnnealingGhostTrackFitter &orig)
        : annealing(orig.annealing->clone()), firstStep(orig.firstStep) {}
    ~AnnealingGhostTrackFitter() override {}
 
  private:
    FitterImpl *clone() const override { return new AnnealingGhostTrackFitter(*this); }
 
    bool stable(const GhostTrackPrediction &before, const GhostTrackPrediction &after) const override {
      return SequentialGhostTrackFitter::stable(before, after) && annealing->isAnnealed();
    }
 
    void reset() override {
      annealing->resetAnnealing();
      firstStep = true;
    }
    void postFit(const GhostTrackFitter::PredictionUpdater &updater,
                 const GhostTrackPrediction &pred,
                 std::vector<GhostTrackState> &states) override;
 
    std::unique_ptr<AnnealingSchedule> annealing;
    bool firstStep;
  };
 
}  // namespace reco
 
#endif  // RecoBTag_AnnealingGhostTrackFitter_h