ConversionTrackRefFix.cc

Go to the documentation of this file.

#include "FastSimulation/Tracking/plugins/ConversionTrackRefFix.h"

#include "DataFormats/TrackReco/interface/Track.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "DataFormats/EgammaTrackReco/interface/ConversionTrack.h"

using namespace std;
using namespace reco;
using namespace edm;

ConversionTrackRefFix::ConversionTrackRefFix(const edm::ParameterSet& iConfig)
{
  InputTag conversionTracksTag = iConfig.getParameter<InputTag>("src");
  InputTag newTracksTag = iConfig.getParameter<InputTag>("newTrackCollection");
  
  produces<ConversionTrackCollection>();

  conversionTracksToken = consumes<ConversionTrackCollection>(conversionTracksTag);
  newTracksToken = consumes<TrackCollection>(newTracksTag);
}

ConversionTrackRefFix::~ConversionTrackRefFix(){}


void
ConversionTrackRefFix::produce(edm::Event& iEvent, const edm::EventSetup& iSetup)
{
  Handle<ConversionTrackCollection> conversionTracks;
  iEvent.getByToken(conversionTracksToken,conversionTracks);

  Handle<TrackCollection> newTracks;
  iEvent.getByToken(newTracksToken,newTracks);

  unique_ptr<ConversionTrackCollection> output(new ConversionTrackCollection);
  
  for(const ConversionTrack &  conversion : *(conversionTracks.product())){
    size_t trackIndex = conversion.trackRef().key();
    output->push_back(ConversionTrack(TrackBaseRef(TrackRef(newTracks,trackIndex))));
    output->back().setTrajRef(conversion.trajRef());
    output->back().setIsTrackerOnly(conversion.isTrackerOnly());
    output->back().setIsArbitratedEcalSeeded(conversion.isArbitratedEcalSeeded());
    output->back().setIsArbitratedMerged(conversion.isArbitratedMerged());
    output->back().setIsArbitratedMergedEcalGeneral(conversion.isArbitratedMergedEcalGeneral());
  }

  iEvent.put(std::move(output));

}

#include "FWCore/Framework/interface/MakerMacros.h"
DEFINE_FWK_MODULE(ConversionTrackRefFix);