VertexHistory.h

Go to the documentation of this file.

#ifndef VertexHistory_h
#define VertexHistory_h

#include "DataFormats/VertexReco/interface/Vertex.h"
#include "DataFormats/VertexReco/interface/VertexFwd.h"

#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"

#include "SimDataFormats/Associations/interface/VertexAssociation.h"
#include "SimTracker/TrackHistory/interface/HistoryBase.h"
#include "SimTracker/TrackHistory/interface/Utils.h"

class VertexHistory : public HistoryBase {
public:
  /* Creates a VertexHistory with association given by pset.

     /param[in] config with the configuration values
  */
  VertexHistory(const edm::ParameterSet &, edm::ConsumesCollector);

  void newEvent(const edm::Event &, const edm::EventSetup &);

  /* Return false when the history cannot be determined upto a given depth.
     If not depth is pass to the function no restriction are apply to it.

     /param[in] trackingVertexRef of a simulated track
     /param[in] depth of the vertex history
     /param[out] boolean that is true when history can be determined
  */
  bool evaluate(TrackingVertexRef tvr) {
    if (enableSimToReco_) {
      std::pair<reco::VertexBaseRef, double> result = match(tvr, simToReco_, bestMatchByMaxValue_);
      recovertex_ = result.first;
      quality_ = result.second;
    }
    return HistoryBase::evaluate(tvr);
  }

  /* Return false when the track association is not possible (fake track).

     /param[in] VertexRef to a reco::track
     /param[out] boolean that is false when a fake track is detected
  */
  bool evaluate(reco::VertexBaseRef);

  const reco::VertexBaseRef &recoVertex() const { return recovertex_; }

  double quality() const { return quality_; }

private:
  bool bestMatchByMaxValue_;

  bool enableRecoToSim_, enableSimToReco_;

  double quality_;

  edm::InputTag vertexProducer_;

  edm::InputTag trackingTruth_;

  edm::InputTag vertexAssociator_;

  reco::VertexBaseRef recovertex_;

  reco::VertexRecoToSimCollection recoToSim_;

  reco::VertexSimToRecoCollection simToReco_;
};

#endif