#include <Alignment/TrackerToMuonPropagator/src/TrackerToMuonPropagator.cc>

Inheritance diagram for TrackerToMuonPropagator:

Public Member Functions
	TrackerToMuonPropagator (const edm::ParameterSet &)

	~TrackerToMuonPropagator () override

Public Member Functions inherited from edm::EDProducer
	EDProducer ()

SerialTaskQueue *	globalLuminosityBlocksQueue ()

SerialTaskQueue *	globalRunsQueue ()

ModuleDescription const &	moduleDescription () const

	~EDProducer () override

Public Member Functions inherited from edm::ProducerBase
void	callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)

std::vector< edm::ProductResolverIndex > const &	indiciesForPutProducts (BranchType iBranchType) const

	ProducerBase ()

std::vector< edm::ProductResolverIndex > const &	putTokenIndexToProductResolverIndex () const

void	registerProducts (ProducerBase , ProductRegistry , ModuleDescription const &)

std::function< void(BranchDescription const &)>	registrationCallback () const
	used by the fwk to register list of products More...

void	resolvePutIndicies (BranchType iBranchType, ModuleToResolverIndicies const &iIndicies, std::string const &moduleLabel)

	~ProducerBase () noexcept(false) override

Public Member Functions inherited from edm::EDConsumerBase
std::vector< ConsumesInfo >	consumesInfo () const

void	convertCurrentProcessAlias (std::string const &processName)
	Convert "@currentProcess" in InputTag process names to the actual current process name. More...

	EDConsumerBase ()

	EDConsumerBase (EDConsumerBase const &)=delete

	EDConsumerBase (EDConsumerBase &&)=default

ESProxyIndex const *	esGetTokenIndices (edm::Transition iTrans) const

ProductResolverIndexAndSkipBit	indexFrom (EDGetToken, BranchType, TypeID const &) const

void	itemsMayGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const

void	itemsToGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const

std::vector< ProductResolverIndexAndSkipBit > const &	itemsToGetFrom (BranchType iType) const

void	labelsForToken (EDGetToken iToken, Labels &oLabels) const

void	modulesWhoseProductsAreConsumed (std::vector< ModuleDescription const * > &modules, ProductRegistry const &preg, std::map< std::string, ModuleDescription const * > const &labelsToDesc, std::string const &processName) const

EDConsumerBase const &	operator= (EDConsumerBase const &)=delete

EDConsumerBase &	operator= (EDConsumerBase &&)=default

bool	registeredToConsume (ProductResolverIndex, bool, BranchType) const

bool	registeredToConsumeMany (TypeID const &, BranchType) const

ProductResolverIndexAndSkipBit	uncheckedIndexFrom (EDGetToken) const

void	updateLookup (BranchType iBranchType, ProductResolverIndexHelper const &, bool iPrefetchMayGet)

void	updateLookup (eventsetup::ESRecordsToProxyIndices const &)

virtual	~EDConsumerBase () noexcept(false)

Private Member Functions
void	beginJob () override

void	endJob () override

void	produce (edm::Event &, const edm::EventSetup &) override

Private Attributes
edm::InputTag	m_globalMuons

edm::InputTag	m_globalMuonTracks

std::string	m_propagator

bool	m_refitTracker

TrackTransformer *	m_trackTransformer

Additional Inherited Members
Public Types inherited from edm::EDProducer
typedef EDProducer	ModuleType

Public Types inherited from edm::ProducerBase
using	ModuleToResolverIndicies = std::unordered_multimap< std::string, std::tuple< edm::TypeID const , const char , edm::ProductResolverIndex >>

typedef ProductRegistryHelper::TypeLabelList	TypeLabelList

Public Types inherited from edm::EDConsumerBase
typedef ProductLabels	Labels

Static Public Member Functions inherited from edm::EDProducer
static const std::string &	baseType ()

static void	fillDescriptions (ConfigurationDescriptions &descriptions)

static void	prevalidate (ConfigurationDescriptions &descriptions)

static bool	wantsGlobalLuminosityBlocks ()

static bool	wantsGlobalRuns ()

static bool	wantsStreamLuminosityBlocks ()

static bool	wantsStreamRuns ()

Protected Member Functions inherited from edm::ProducerBase
ProducesCollector	producesCollector ()

Protected Member Functions inherited from edm::EDConsumerBase
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	consumes (edm::InputTag const &tag)

EDGetToken	consumes (const TypeToGet &id, edm::InputTag const &tag)

template<BranchType B>
EDGetToken	consumes (TypeToGet const &id, edm::InputTag const &tag)

ConsumesCollector	consumesCollector ()
	Use a ConsumesCollector to gather consumes information from helper functions. More...

template<typename ProductType , BranchType B = InEvent>
void	consumesMany ()

void	consumesMany (const TypeToGet &id)

template<BranchType B>
void	consumesMany (const TypeToGet &id)

template<typename ESProduct , typename ESRecord , Transition Tr = Transition::Event>
auto	esConsumes ()

template<typename ESProduct , typename ESRecord , Transition Tr = Transition::Event>
auto	esConsumes (ESInputTag const &tag)

template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	mayConsume (edm::InputTag const &tag)

EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

template<BranchType B>
EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

Detailed Description

Description: <one line="" class="" summary>="">

Implementation: <Notes on="" implementation>="">

Definition at line 62 of file TrackerToMuonPropagator.cc.

Constructor & Destructor Documentation

TrackerToMuonPropagator::TrackerToMuonPropagator ( const edm::ParameterSet & iConfig )

explicit

Definition at line 92 of file TrackerToMuonPropagator.cc.

References edm::ParameterSet::getParameter(), m_globalMuons, m_globalMuonTracks, m_propagator, m_refitTracker, m_trackTransformer, AlCaHLTBitMon_QueryRunRegistry::string, and HLT_2018_cff::TrackTransformer.

                                                                                {
   m_globalMuons = iConfig.getParameter<edm::InputTag>("globalMuons");
   m_globalMuonTracks = iConfig.getParameter<edm::InputTag>("globalMuonTracks");
   m_propagator = iConfig.getParameter<std::string>("propagator");
   m_refitTracker = iConfig.getParameter<bool>("refitTrackerTrack");
   if (m_refitTracker) {
     m_trackTransformer = new TrackTransformer(iConfig.getParameter<edm::ParameterSet>("trackerTrackTransformer"));
   } else
     m_trackTransformer = nullptr;
 
   produces<std::vector<Trajectory>>();
   produces<TrajTrackAssociationCollection>();
 }

TrackerToMuonPropagator::~TrackerToMuonPropagator ( )

override

Definition at line 106 of file TrackerToMuonPropagator.cc.

                                                   {
   // do anything here that needs to be done at desctruction time
   // (e.g. close files, deallocate resources etc.)
 }

Member Function Documentation

void TrackerToMuonPropagator::beginJob ( void )

overrideprivatevirtual

Reimplemented from edm::EDProducer.

Definition at line 270 of file TrackerToMuonPropagator.cc.

270 {}

void TrackerToMuonPropagator::endJob ( void )

overrideprivatevirtual

Reimplemented from edm::EDProducer.

Definition at line 273 of file TrackerToMuonPropagator.cc.

References DEFINE_FWK_MODULE.

Referenced by o2olib.O2ORunMgr::executeJob().

273 {}

void TrackerToMuonPropagator::produce	(	edm::Event &	iEvent,
		const edm::EventSetup &	iSetup
	)

overrideprivate

Definition at line 116 of file TrackerToMuonPropagator.cc.

References alongMomentum, MuonTransientTrackingRecHitBuilder::build(), ALCARECOTkAlJpsiMuMu_cff::charge, MuonSubdetId::CSC, MuonSubdetId::DT, MillePedeFileConverter_cfg::e, edm::OwnVector< T, P >::empty(), Exception, Trajectory::firstMeasurement(), TrajectoryMeasurement::forwardPredictedState(), edm::EventSetup::get(), edm::Event::getByLabel(), susybsm::HSCParticleType::globalMuon, globalMuons_cfi::globalMuons, mps_fire::i, DTGeometry::idToDet(), TrackerGeometry::idToDet(), CSCGeometry::idToDet(), TrajectoryStateOnSurface::isValid(), m_globalMuons, m_globalMuonTracks, m_propagator, m_refitTracker, m_trackTransformer, HLT_2018_cff::magneticField, eostools::move(), DetId::Muon, trajectoryStateTransform::persistentState(), Propagator::propagate(), TrackCandidateProducer_cfi::propagator, Trajectory::push(), edm::OwnVector< T, P >::push_back(), edm::Event::put(), DetId::rawId(), TrajectoryMeasurement::recHit(), TrackTransformer::setServices(), edm::OwnVector< T, P >::size(), GeomDet::surface(), and TrackTransformer::transform().

Referenced by JSONExport.JsonExport::export(), HTMLExport.HTMLExport::export(), and HTMLExport.HTMLExportStatic::export().

                                                                                    {
   if (m_trackTransformer)
     m_trackTransformer->setServices(iSetup);
 
   edm::Handle<reco::MuonCollection> globalMuons;
   iEvent.getByLabel(m_globalMuons, globalMuons);
 
   edm::Handle<reco::TrackCollection> globalMuonTracks;
   iEvent.getByLabel(m_globalMuonTracks, globalMuonTracks);
 
   edm::ESHandle<Propagator> propagator;
   iSetup.get<TrackingComponentsRecord>().get(m_propagator, propagator);
 
   edm::ESHandle<TrackerGeometry> trackerGeometry;
   iSetup.get<TrackerDigiGeometryRecord>().get(trackerGeometry);
 
   edm::ESHandle<DTGeometry> dtGeometry;
   iSetup.get<MuonGeometryRecord>().get(dtGeometry);
 
   edm::ESHandle<CSCGeometry> cscGeometry;
   iSetup.get<MuonGeometryRecord>().get(cscGeometry);
 
   edm::ESHandle<MagneticField> magneticField;
   iSetup.get<IdealMagneticFieldRecord>().get(magneticField);
 
   edm::ESHandle<GlobalTrackingGeometry> globalGeometry;
   iSetup.get<GlobalTrackingGeometryRecord>().get(globalGeometry);
 
   // Create these factories once per event
 
   MuonTransientTrackingRecHitBuilder muonTransBuilder;
 
   // Create a collection of Trajectories, to put in the Event
   auto trajectoryCollection = std::make_unique<std::vector<Trajectory>>();
 
   // Remember which trajectory is associated with which track
   std::map<edm::Ref<std::vector<Trajectory>>::key_type, edm::Ref<reco::TrackCollection>::key_type> reference_map;
   edm::Ref<std::vector<Trajectory>>::key_type trajCounter = 0;
 
   for (reco::MuonCollection::const_iterator globalMuon = globalMuons->begin(); globalMuon != globalMuons->end();
        ++globalMuon) {
     // get the counter for this global muon (that's why we needed to extract the collection explicitly
     edm::Ref<reco::TrackCollection>::key_type trackCounter = 0;
     reco::TrackCollection::const_iterator globalMuonTrack = globalMuonTracks->begin();
     for (; globalMuonTrack != globalMuonTracks->end(); ++globalMuonTrack) {
       trackCounter++;
       if (fabs(globalMuon->combinedMuon()->phi() - globalMuonTrack->phi()) < 1e-10 &&
           fabs(globalMuon->combinedMuon()->eta() - globalMuonTrack->eta()) < 1e-10)
         break;
     }
     if (globalMuonTrack == globalMuonTracks->end()) {
       throw cms::Exception("BadConfig") << "The tracks label doesn't correspond to the same objects as the muons label"
                                         << std::endl;
     }
 
     TrajectoryStateOnSurface tracker_tsos;
     DetId outerDetId;
     if (m_refitTracker) {
       std::vector<Trajectory> trackerTrajectories = m_trackTransformer->transform(*globalMuon->track());
       if (trackerTrajectories.size() == 1) {
         const Trajectory trackerTrajectory = *(trackerTrajectories.begin());
 
         // surprisingly, firstMeasurement() corresponds to the outermost state of the tracker
         tracker_tsos = trackerTrajectory.firstMeasurement().forwardPredictedState();
         outerDetId = trackerTrajectory.firstMeasurement().recHit()->geographicalId();
       } else
         continue;
     } else {
       // get information about the outermost tracker hit
       GlobalPoint outerPosition(globalMuon->track()->outerPosition().x(),
                                 globalMuon->track()->outerPosition().y(),
                                 globalMuon->track()->outerPosition().z());
       GlobalVector outerMomentum(globalMuon->track()->outerMomentum().x(),
                                  globalMuon->track()->outerMomentum().y(),
                                  globalMuon->track()->outerMomentum().z());
       int charge = globalMuon->track()->charge();
       const reco::Track::CovarianceMatrix outerStateCovariance = globalMuon->track()->outerStateCovariance();
       outerDetId = DetId(globalMuon->track()->outerDetId());
 
       // construct the information necessary to make a TrajectoryStateOnSurface
       GlobalTrajectoryParameters globalTrajParams(outerPosition, outerMomentum, charge, &(*magneticField));
       CurvilinearTrajectoryError curviError(outerStateCovariance);
       FreeTrajectoryState tracker_state(globalTrajParams, curviError);
 
       // starting point for propagation into the muon system
       tracker_tsos =
           TrajectoryStateOnSurface(globalTrajParams, curviError, trackerGeometry->idToDet(outerDetId)->surface());
     }
 
     TrajectoryStateOnSurface last_tsos = tracker_tsos;
 
     // loop over the muon hits, keeping track of the successful extrapolations
     edm::OwnVector<TrackingRecHit> muonHits;
     std::vector<TrajectoryStateOnSurface> TSOSes;
     for (auto const& hit : globalMuon->combinedMuon()->recHits()) {
       DetId id = hit->geographicalId();
 
       TrajectoryStateOnSurface extrapolation;
       bool extrapolated = false;
       if (id.det() == DetId::Muon && id.subdetId() == MuonSubdetId::DT) {
         extrapolation = propagator->propagate(last_tsos, dtGeometry->idToDet(id)->surface());
         extrapolated = true;
       } else if (id.det() == DetId::Muon && id.subdetId() == MuonSubdetId::CSC) {
         extrapolation = propagator->propagate(last_tsos, cscGeometry->idToDet(id)->surface());
         extrapolated = true;
       }
 
       if (extrapolated && extrapolation.isValid()) {
         muonHits.push_back(hit->clone());
         TSOSes.push_back(extrapolation);
       }
     }  // end loop over standAloneMuon hits
 
     // if it has any successful extrapolations, make them into a Trajectory
     if (!muonHits.empty()) {
       PTrajectoryStateOnDet const& PTraj = trajectoryStateTransform::persistentState(tracker_tsos, outerDetId.rawId());
       TrajectorySeed trajectorySeed(PTraj, muonHits, alongMomentum);
       Trajectory trajectory(trajectorySeed, alongMomentum);
 
       for (unsigned int i = 0; i < muonHits.size(); i++) {
         TrajectoryMeasurement::ConstRecHitPointer hitPtr(muonTransBuilder.build(&(muonHits[i]), globalGeometry));
         TrajectoryStateOnSurface TSOS = TSOSes[i];
         trajectory.push(TrajectoryMeasurement(TSOS, TSOS, TSOS, hitPtr));
       }  // end filling Trajectory
 
       trajectoryCollection->push_back(trajectory);
 
       // Remember which Trajectory is associated with which Track
       trajCounter++;
       reference_map[trajCounter] = trackCounter;
 
     }  // end if we have some good extrapolations
 
   }  // end loop over globalMuons
 
   unsigned int numTrajectories = trajectoryCollection->size();
 
   // insert the trajectories into the Event
   edm::OrphanHandle<std::vector<Trajectory>> ohTrajs = iEvent.put(std::move(trajectoryCollection));
 
   // create the trajectory <-> track association map
   auto trajTrackMap = std::make_unique<TrajTrackAssociationCollection>();
 
   for (trajCounter = 0; trajCounter < numTrajectories; trajCounter++) {
     edm::Ref<reco::TrackCollection>::key_type trackCounter = reference_map[trajCounter];
 
     trajTrackMap->insert(edm::Ref<std::vector<Trajectory>>(ohTrajs, trajCounter),
                          edm::Ref<reco::TrackCollection>(globalMuonTracks, trackCounter));
   }
   // and put it in the Event, also
   iEvent.put(std::move(trajTrackMap));
 }