ConversionTrackProducer Class Reference

#include <ConversionTrackProducer.h>

Inheritance diagram for ConversionTrackProducer:

Public Member Functions
	ConversionTrackProducer (const edm::ParameterSet &conf)
void	produce (edm::Event &e, const edm::EventSetup &c) override
	~ConversionTrackProducer () override
Public Member Functions inherited from edm::stream::EDProducer<>
	EDProducer ()=default
bool	hasAbilityToProduceInBeginLumis () const final
bool	hasAbilityToProduceInBeginRuns () const final
bool	hasAbilityToProduceInEndLumis () const final
bool	hasAbilityToProduceInEndRuns () const final

Private Types
typedef edm::AssociationMap< edm::OneToOne< std::vector< Trajectory >, reco::GsfTrackCollection, unsigned short > >	TrajGsfTrackAssociationCollection

Private Attributes
edm::EDGetTokenT< reco::BeamSpot >	beamSpotInputTag
edm::ParameterSet	conf_
IdealHelixParameters	ConvTrackPreSelector
bool	filterOnConvTrackHyp
edm::EDGetTokenT< edm::View< reco::Track > >	genericTracks
edm::EDGetTokenT< TrajGsfTrackAssociationCollection >	gsfTrajectories
edm::EDGetTokenT< TrajTrackAssociationCollection >	kfTrajectories
double	minConvRadius
std::unique_ptr< reco::ConversionTrackCollection >	outputTrks
bool	setArbitratedEcalSeeded
bool	setArbitratedMerged
bool	setArbitratedMergedEcalGeneral
bool	setIsGsfTrackOpen
bool	setTrackerOnly
std::string	trackProducer
bool	useTrajectory

Additional Inherited Members
Public Types inherited from edm::stream::EDProducer<>
typedef CacheContexts< T... >	CacheTypes
typedef CacheTypes::GlobalCache	GlobalCache
typedef AbilityChecker< T... >	HasAbility
typedef CacheTypes::LuminosityBlockCache	LuminosityBlockCache
typedef LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCache >	LuminosityBlockContext
typedef CacheTypes::LuminosityBlockSummaryCache	LuminosityBlockSummaryCache
typedef CacheTypes::RunCache	RunCache
typedef RunContextT< RunCache, GlobalCache >	RunContext
typedef CacheTypes::RunSummaryCache	RunSummaryCache

Detailed Description

Definition at line 43 of file ConversionTrackProducer.h.

Member Typedef Documentation

typedef edm::AssociationMap<edm::OneToOne<std::vector<Trajectory>, reco::GsfTrackCollection, unsigned short> > ConversionTrackProducer::TrajGsfTrackAssociationCollection

private

Definition at line 45 of file ConversionTrackProducer.h.

Constructor & Destructor Documentation

ConversionTrackProducer::ConversionTrackProducer ( const edm::ParameterSet &  conf )

explicit

Definition at line 35 of file ConversionTrackProducer.cc.

References genericTracks, gsfTrajectories, kfTrajectories, and trackProducer.

    : conf_(conf),
      trackProducer(conf.getParameter<std::string>("TrackProducer")),
      useTrajectory(conf.getParameter<bool>("useTrajectory")),
      setTrackerOnly(conf.getParameter<bool>("setTrackerOnly")),
      setIsGsfTrackOpen(conf.getParameter<bool>("setIsGsfTrackOpen")),
      setArbitratedEcalSeeded(conf.getParameter<bool>("setArbitratedEcalSeeded")),
      setArbitratedMerged(conf.getParameter<bool>("setArbitratedMerged")),
      setArbitratedMergedEcalGeneral(conf.getParameter<bool>("setArbitratedMergedEcalGeneral")),
      beamSpotInputTag(consumes<reco::BeamSpot>(conf.getParameter<edm::InputTag>("beamSpotInputTag"))),
      filterOnConvTrackHyp(conf.getParameter<bool>("filterOnConvTrackHyp")),
      minConvRadius(conf.getParameter<double>("minConvRadius")) {
  edm::InputTag thetp(trackProducer);
  genericTracks = consumes<edm::View<reco::Track> >(thetp);
  kfTrajectories = consumes<TrajTrackAssociationCollection>(thetp);
  gsfTrajectories = consumes<TrajGsfTrackAssociationCollection>(thetp);
  produces<reco::ConversionTrackCollection>();
}

ConversionTrackProducer::~ConversionTrackProducer ( )

override

Definition at line 55 of file ConversionTrackProducer.cc.

{}

Member Function Documentation

void ConversionTrackProducer::produce ( edm::Event &  e, const edm::EventSetup &  c  )

override

Definition at line 58 of file ConversionTrackProducer.cc.

References pwdgSkimBPark_cfi::beamSpot, beamSpotInputTag, edm::AssociationMap< Tag >::begin(), edm::AssociationMap< edm::OneToOne< std::vector< Trajectory >, reco::TrackCollection, unsigned short > >::const_iterator, edm::AssociationMap< edm::OneToOne< std::vector< Trajectory >, reco::GsfTrackCollection, unsigned short > >::const_iterator, ConvTrackPreSelector, edm::AssociationMap< Tag >::end(), filterOnConvTrackHyp, genericTracks, edm::EventSetup::get(), edm::Event::getByToken(), gsfTrajectories, mps_fire::i, IdealHelixParameters::isTangentPointDistanceLessThan(), kfTrajectories, minConvRadius, eostools::move(), outputTrks, edm::ESHandle< T >::product(), edm::Event::put(), setArbitratedEcalSeeded, setArbitratedMerged, setArbitratedMergedEcalGeneral, reco::ConversionTrack::setIsArbitratedEcalSeeded(), reco::ConversionTrack::setIsArbitratedMerged(), reco::ConversionTrack::setIsArbitratedMergedEcalGeneral(), reco::ConversionTrack::setIsGsfTrackOpen(), setIsGsfTrackOpen, reco::ConversionTrack::setIsTrackerOnly(), IdealHelixParameters::setMagnField(), setTrackerOnly, reco::ConversionTrack::setTrajRef(), and useTrajectory.

                                                                          {
  //get input collection (through edm::View)
  edm::Handle<edm::View<reco::Track> > hTrks;
  e.getByToken(genericTracks, hTrks);

  //get association maps between trajectories and tracks and build temporary maps
  edm::Handle<TrajTrackAssociationCollection> hTTAss;
  edm::Handle<TrajGsfTrackAssociationCollection> hTTAssGsf;

  std::map<reco::TrackRef, edm::Ref<std::vector<Trajectory> > > tracktrajmap;
  std::map<reco::GsfTrackRef, edm::Ref<std::vector<Trajectory> > > gsftracktrajmap;

  if (useTrajectory) {
    if (!hTrks->empty()) {
      if (dynamic_cast<const reco::GsfTrack*>(&hTrks->at(0))) {
        //fill map for gsf tracks
        e.getByToken(gsfTrajectories, hTTAssGsf);
        for (TrajGsfTrackAssociationCollection::const_iterator iPair = hTTAssGsf->begin(); iPair != hTTAssGsf->end();
             ++iPair) {
          gsftracktrajmap[iPair->val] = iPair->key;
        }

      } else {
        //fill map for standard tracks
        e.getByToken(kfTrajectories, hTTAss);
        for (TrajTrackAssociationCollection::const_iterator iPair = hTTAss->begin(); iPair != hTTAss->end(); ++iPair) {
          tracktrajmap[iPair->val] = iPair->key;
        }
      }
    }
  }

  // Step B: create empty output collection
  outputTrks = std::make_unique<reco::ConversionTrackCollection>();

  //--------------------------------------------------
  //Added by D. Giordano
  // 2011/08/05
  // Reduction of the track sample based on geometric hypothesis for conversion tracks

  edm::Handle<reco::BeamSpot> beamSpotHandle;
  e.getByToken(beamSpotInputTag, beamSpotHandle);

  edm::ESHandle<MagneticField> magFieldHandle;
  es.get<IdealMagneticFieldRecord>().get(magFieldHandle);

  if (filterOnConvTrackHyp && !beamSpotHandle.isValid()) {
    edm::LogError("Invalid Collection") << "invalid collection for the BeamSpot";
    throw;
  }

  ConvTrackPreSelector.setMagnField(magFieldHandle.product());

  //----------------------------------------------------------

  // Simple conversion of tracks to conversion tracks, setting appropriate flags from configuration
  for (size_t i = 0; i < hTrks->size(); ++i) {
    //--------------------------------------------------
    //Added by D. Giordano
    // 2011/08/05
    // Reduction of the track sample based on geometric hypothesis for conversion tracks

    math::XYZVector beamSpot = math::XYZVector(beamSpotHandle->position());
    edm::RefToBase<reco::Track> trackBaseRef = hTrks->refAt(i);
    if (filterOnConvTrackHyp &&
        ConvTrackPreSelector.isTangentPointDistanceLessThan(minConvRadius, trackBaseRef.get(), beamSpot))
      continue;
    //--------------------------------------------------

    reco::ConversionTrack convTrack(trackBaseRef);
    convTrack.setIsTrackerOnly(setTrackerOnly);
    convTrack.setIsGsfTrackOpen(setIsGsfTrackOpen);
    convTrack.setIsArbitratedEcalSeeded(setArbitratedEcalSeeded);
    convTrack.setIsArbitratedMerged(setArbitratedMerged);
    convTrack.setIsArbitratedMergedEcalGeneral(setArbitratedMergedEcalGeneral);

    //fill trajectory association if configured, using correct map depending on track type
    if (useTrajectory) {
      if (!gsftracktrajmap.empty()) {
        convTrack.setTrajRef(gsftracktrajmap.find(trackBaseRef.castTo<reco::GsfTrackRef>())->second);
      } else {
        convTrack.setTrajRef(tracktrajmap.find(trackBaseRef.castTo<reco::TrackRef>())->second);
      }
    }

    outputTrks->push_back(convTrack);
  }

  e.put(std::move(outputTrks));
  return;

}  //end produce

Member Data Documentation

edm::EDGetTokenT<reco::BeamSpot> ConversionTrackProducer::beamSpotInputTag

private

Definition at line 73 of file ConversionTrackProducer.h.

Referenced by produce().

edm::ParameterSet ConversionTrackProducer::conf_

private

Definition at line 55 of file ConversionTrackProducer.h.

IdealHelixParameters ConversionTrackProducer::ConvTrackPreSelector

private

Definition at line 76 of file ConversionTrackProducer.h.

Referenced by produce().

bool ConversionTrackProducer::filterOnConvTrackHyp

private

Definition at line 74 of file ConversionTrackProducer.h.

Referenced by produce().

edm::EDGetTokenT<edm::View<reco::Track> > ConversionTrackProducer::genericTracks

private

Definition at line 58 of file ConversionTrackProducer.h.

Referenced by ConversionTrackProducer(), and produce().

edm::EDGetTokenT<TrajGsfTrackAssociationCollection> ConversionTrackProducer::gsfTrajectories

private

Definition at line 60 of file ConversionTrackProducer.h.

Referenced by ConversionTrackProducer(), and produce().

edm::EDGetTokenT<TrajTrackAssociationCollection> ConversionTrackProducer::kfTrajectories

private

Definition at line 59 of file ConversionTrackProducer.h.

Referenced by ConversionTrackProducer(), and produce().

double ConversionTrackProducer::minConvRadius

private

Definition at line 75 of file ConversionTrackProducer.h.

Referenced by produce().

std::unique_ptr<reco::ConversionTrackCollection> ConversionTrackProducer::outputTrks

private

Definition at line 79 of file ConversionTrackProducer.h.

Referenced by produce().

bool ConversionTrackProducer::setArbitratedEcalSeeded

private

Definition at line 64 of file ConversionTrackProducer.h.

Referenced by produce().

bool ConversionTrackProducer::setArbitratedMerged

private

Definition at line 65 of file ConversionTrackProducer.h.

Referenced by produce().

bool ConversionTrackProducer::setArbitratedMergedEcalGeneral

private

Definition at line 66 of file ConversionTrackProducer.h.

Referenced by produce().

bool ConversionTrackProducer::setIsGsfTrackOpen

private

Definition at line 63 of file ConversionTrackProducer.h.

Referenced by produce().

bool ConversionTrackProducer::setTrackerOnly

private

Definition at line 62 of file ConversionTrackProducer.h.

Referenced by produce().

std::string ConversionTrackProducer::trackProducer

private

Definition at line 57 of file ConversionTrackProducer.h.

Referenced by ConversionTrackProducer().

bool ConversionTrackProducer::useTrajectory

private

Definition at line 61 of file ConversionTrackProducer.h.

Referenced by produce().