GhostTrack.h

Go to the documentation of this file.

#ifndef RecoBTag_GhostTrack_h
#define RecoBTag_GhostTrack_h
 
#include <vector>
 
#include "DataFormats/GeometryVector/interface/GlobalPoint.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"
 
#include "TrackingTools/TransientTrack/interface/TransientTrack.h"
 
#include "RecoVertex/GhostTrackFitter/interface/GhostTrackState.h"
#include "RecoVertex/GhostTrackFitter/interface/GhostTrackPrediction.h"
 
namespace reco {
 
  class GhostTrack {
  public:
    GhostTrack(const GhostTrackPrediction &prior,
               const GhostTrackPrediction &prediction,
               const std::vector<GhostTrackState> &states,
               double ndof,
               double chi2)
        : prediction_(prediction), prior_(prior), states_(states), ndof_(ndof), chi2_(chi2) {}
 
    GhostTrack(const GhostTrackPrediction &prior,
               const GhostTrackPrediction &prediction,
               const std::vector<TransientTrack> &tracks,
               double ndof,
               double chi2,
               const std::vector<float> &weights = std::vector<float>(),
               const GlobalPoint &origin = GlobalPoint(),
               bool withOrigin = false);
 
    GhostTrack(const Track &ghostTrack,
               const std::vector<TransientTrack> &tracks,
               const std::vector<float> &weights = std::vector<float>(),
               const GhostTrackPrediction &prior = GhostTrackPrediction(),
               const GlobalPoint &origin = GlobalPoint(),
               bool withOrigin = false);
 
    const GhostTrackPrediction &prediction() const { return prediction_; }
    const GhostTrackPrediction &prior() const { return prior_; }
 
    const std::vector<GhostTrackState> &states() const { return states_; }
    double ndof() const { return ndof_; }
    double chi2() const { return chi2_; }
 
    operator Track() const { return prediction_.track(ndof_, chi2_); }
 
  private:
    void initStates(const std::vector<TransientTrack> &tracks, const std::vector<float> &weights, double offset);
 
    GhostTrackPrediction prediction_;
    GhostTrackPrediction prior_;
    std::vector<GhostTrackState> states_;
    double ndof_;
    double chi2_;
  };
 
}  // namespace reco
 
#endif  // RecoBTag_GhostTrack_h