ME0MuonConverter.cc

Go to the documentation of this file.

#include <FWCore/PluginManager/interface/ModuleDef.h>
#include <FWCore/Framework/interface/MakerMacros.h>

#include <DataFormats/Common/interface/Handle.h>
#include <FWCore/Framework/interface/ESHandle.h>
#include <FWCore/MessageLogger/interface/MessageLogger.h> 

#include <Geometry/Records/interface/MuonGeometryRecord.h>



#include "DataFormats/GeometryVector/interface/GlobalVector.h"
#include "DataFormats/GeometryVector/interface/GlobalPoint.h"
#include "TrackingTools/GeomPropagators/interface/Propagator.h"
#include "TrackingTools/TrajectoryState/interface/TrajectoryStateOnSurface.h"
#include "DataFormats/TrajectorySeed/interface/PropagationDirection.h"

#include "TLorentzVector.h"

#include "DataFormats/TrackReco/interface/TrackFwd.h"
#include "DataFormats/TrackReco/interface/Track.h"

#include "TrackingTools/Records/interface/TrackingComponentsRecord.h"


#include "DataFormats/RecoCandidate/interface/RecoChargedCandidate.h"
#include "DataFormats/RecoCandidate/interface/RecoChargedCandidateFwd.h"

#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/stream/EDProducer.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Utilities/interface/InputTag.h"


#include "DataFormats/RecoCandidate/interface/RecoChargedCandidate.h"
#include "DataFormats/RecoCandidate/interface/RecoChargedCandidateFwd.h"


#include "FWCore/Framework/interface/EDProducer.h"
#include "FWCore/Utilities/interface/InputTag.h"

#include <DataFormats/GEMRecHit/interface/ME0SegmentCollection.h>



#include <DataFormats/MuonReco/interface/ME0Muon.h>
#include <DataFormats/MuonReco/interface/ME0MuonCollection.h>



class ME0MuonConverter : public edm::stream::EDProducer<> {
public:
  explicit ME0MuonConverter(const edm::ParameterSet&);
  ~ME0MuonConverter() override;
  void produce(edm::Event&, const edm::EventSetup&) override;

    

private:

  //edm::EDGetTokenT<std::vector<reco::ME0Muon>> OurMuonsToken_;
  edm::EDGetTokenT<ME0MuonCollection> OurMuonsToken_;
};


ME0MuonConverter::ME0MuonConverter(const edm::ParameterSet& pas) {
    
  produces<std::vector<reco::RecoChargedCandidate> >();  
  edm::InputTag OurMuonsTag ("me0SegmentMatching");
  OurMuonsToken_ = consumes<ME0MuonCollection>(OurMuonsTag);

}

ME0MuonConverter::~ME0MuonConverter() {}

void ME0MuonConverter::produce(edm::Event& ev, const edm::EventSetup& setup) {

  using namespace edm;

  using namespace reco;

  // Handle <std::vector<ME0Muon> > OurMuons;
  // ev.getByToken <std::vector<ME0Muon> > (OurMuonsToken_, OurMuons);


  Handle <ME0MuonCollection> OurMuons;
  ev.getByToken(OurMuonsToken_, OurMuons);

  auto oc = std::make_unique<RecoChargedCandidateCollection>();

  for (std::vector<ME0Muon>::const_iterator thisMuon = OurMuons->begin();
       thisMuon != OurMuons->end(); ++thisMuon){
    TrackRef tkRef = thisMuon->innerTrack();
    
    Particle::Charge q = tkRef->charge();
    Particle::LorentzVector p4(tkRef->px(), tkRef->py(), tkRef->pz(), tkRef->p());
    Particle::Point vtx(tkRef->vx(),tkRef->vy(), tkRef->vz());

    int pid = 0;
    if(abs(q)==1) pid = q < 0 ? 13 : -13;
    reco::RecoChargedCandidate cand(q, p4, vtx, pid);
    cand.setTrack(thisMuon->innerTrack());
    oc->push_back(cand);
  }
    
  ev.put(std::move(oc));
}


 DEFINE_FWK_MODULE(ME0MuonConverter);