CommonTools/RecoAlgos/src/TrackToCandidate.h - master - githubmirror/cmssw

CommonTools/RecoAlgos/src/TrackToCandidate.h

38 строк1 KB

Shahzad Malik Muzaffar

[CommonTools] Move public headers in to interface directory

16 авг 2021, 18:22

16 авг 2021, 18:225d62c0c

#ifndef RecoAlgos_TrackToCandidate_h
#define RecoAlgos_TrackToCandidate_h
#include "CommonTools/RecoAlgos/interface/MassiveCandidateConverter.h"
#include "CommonTools/RecoAlgos/interface/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, edm::ConsumesCollector iC) : MassiveCandidateConverter(cfg, iC) {}
    void convert(reco::TrackRef trkRef, reco::RecoChargedCandidate& c) const {
      const reco::Track& trk = *trkRef;
      c.setCharge(trk.charge());
      c.setVertex(trk.vertex());
      const 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;
    };
  }  // namespace helper

}  // namespace converter

#endif