TrackToCandidate.h

Go to the documentation of this file.

#ifndef RecoAlgos_TrackToCandidate_h
#define RecoAlgos_TrackToCandidate_h
#include "CommonTools/RecoAlgos/src/MassiveCandidateConverter.h"
#include "CommonTools/RecoAlgos/src/CandidateProducer.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/RecoCandidate/interface/RecoChargedCandidate.h"
#include "DataFormats/RecoCandidate/interface/RecoChargedCandidateFwd.h"
#include "DataFormats/Candidate/interface/CandidateFwd.h"

namespace converter {

  struct TrackToCandidate : public MassiveCandidateConverter {
    typedef reco::Track value_type;
    typedef reco::TrackCollection Components;
    typedef reco::RecoChargedCandidate Candidate;
    TrackToCandidate(const edm::ParameterSet & cfg) :
      MassiveCandidateConverter(cfg) {
    }
    void convert(reco::TrackRef trkRef, reco::RecoChargedCandidate & c) const {
      const reco::Track & trk = * trkRef;
      c.setCharge(trk.charge());
      c.setVertex(trk.vertex());
      reco::Track::Vector p = trk.momentum();
      double t = sqrt(massSqr_ + p.mag2());
      c.setP4(reco::Candidate::LorentzVector(p.x(), p.y(), p.z(), t));
      c.setTrack(trkRef);
      c.setPdgId(particle_.pdgId());
    }
  };

  namespace helper {
    template<>
    struct CandConverter<reco::Track> {
      typedef TrackToCandidate type;
    };
  }

}

#endif