ParametersDefinerForTP.h

Go to the documentation of this file.

#ifndef TrackAssociation_ParametersDefinerForTP_h
#define TrackAssociation_ParametersDefinerForTP_h

#include <memory>

#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(){};
  ParametersDefinerForTP(const edm::ParameterSet& iConfig);
  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) { }

  virtual std::unique_ptr<ParametersDefinerForTP> clone() const { return std::unique_ptr<ParametersDefinerForTP>(new ParametersDefinerForTP(*this)); }

  edm::InputTag beamSpotInputTag_;

};


#endif