ParametersDefinerForTP.h

Go to the documentation of this file.

#ifndef TrackAssociation_ParametersDefinerForTP_h
#define TrackAssociation_ParametersDefinerForTP_h

#include <SimDataFormats/TrackingAnalysis/interface/TrackingParticle.h>
#include "SimGeneral/TrackingAnalysis/interface/SimHitTPAssociationProducer.h"
#include "DataFormats/Candidate/interface/Candidate.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/ESHandle.h"      
#include "FWCore/Framework/interface/EventSetup.h"

class ParametersDefinerForTP {

 public:
  ParametersDefinerForTP(){};
  virtual ~ParametersDefinerForTP() {};

    typedef int Charge; 
    typedef math::XYZPointD Point; 
    typedef math::XYZTLorentzVectorD LorentzVector; 

  virtual TrackingParticle::Vector momentum(const edm::Event& iEvent, const edm::EventSetup& iSetup, 
    const Charge ch, const Point & vtx, const LorentzVector& lv) const;

  virtual TrackingParticle::Vector momentum(const edm::Event& iEvent, const edm::EventSetup& iSetup, const TrackingParticleRef tpr) const{
    return momentum(iEvent, iSetup, tpr->charge(),tpr->vertex(),tpr->p4());
  }

  virtual TrackingParticle::Vector momentum(const edm::Event& iEvent, const edm::EventSetup& iSetup, const reco::Candidate& tp) const {
    return momentum(iEvent, iSetup, tp.charge(),tp.vertex(),tp.p4());
  }

  virtual TrackingParticle::Point vertex(const edm::Event& iEvent, const edm::EventSetup& iSetup,
    const Charge ch, const Point & vtx, const LorentzVector& lv) const;
 
  virtual TrackingParticle::Point vertex(const edm::Event& iEvent, const edm::EventSetup& iSetup, const TrackingParticleRef& tpr) const{
    return vertex(iEvent, iSetup, tpr->charge(),tpr->vertex(),tpr->p4());
  }

  virtual TrackingParticle::Point vertex(const edm::Event& iEvent, const edm::EventSetup& iSetup, const reco::Candidate& tp) const {
    return vertex(iEvent, iSetup, tp.charge(),tp.vertex(),tp.p4());
  }

  virtual void initEvent(edm::Handle<SimHitTPAssociationProducer::SimHitTPAssociationList> simHitsTPAssocToSet) const { }

};


#endif