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