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

Inheritance diagram for TrackerToMuonPropagator:

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

	~TrackerToMuonPropagator ()

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

ModuleDescription const &	moduleDescription () const

virtual	~EDProducer ()

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

void	registerProducts (ProducerBase , ProductRegistry , ModuleDescription const &)

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

virtual	~ProducerBase ()

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

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

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

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

std::vector < ProductHolderIndexAndSkipBit > const &	itemsToGetFromEvent () const

void	labelsForToken (EDGetToken iToken, Labels &oLabels) const

void	modulesDependentUpon (const std::string &iProcessName, std::vector< const char * > &oModuleLabels) const

bool	registeredToConsume (ProductHolderIndex, bool, BranchType) const

bool	registeredToConsumeMany (TypeID const &, BranchType) const

void	updateLookup (BranchType iBranchType, ProductHolderIndexHelper const &)

virtual	~EDConsumerBase ()

Private Member Functions
virtual void	beginJob () override

virtual void	endJob () override

virtual 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
typedef ProductRegistryHelper::TypeLabelList	TypeLabelList

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

static void	fillDescriptions (ConfigurationDescriptions &descriptions)

static void	prevalidate (ConfigurationDescriptions &descriptions)

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

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 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 63 of file TrackerToMuonPropagator.cc.

Constructor & Destructor Documentation

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

explicit

Definition at line 93 of file TrackerToMuonPropagator.cc.

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

 {
    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 = NULL;
   
    produces<std::vector<Trajectory> >();
    produces<TrajTrackAssociationCollection>();
 }

TrackerToMuonPropagator::~TrackerToMuonPropagator ( )

Definition at line 109 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 273 of file TrackerToMuonPropagator.cc.

274 {

275 }

void TrackerToMuonPropagator::endJob ( void )

overrideprivatevirtual

Reimplemented from edm::EDProducer.

Definition at line 279 of file TrackerToMuonPropagator.cc.

279 {

280 }

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

overrideprivatevirtual

Implements edm::EDProducer.

Definition at line 124 of file TrackerToMuonPropagator.cc.

References alongMomentum, MuonTransientTrackingRecHitBuilder::build(), DeDxDiscriminatorTools::charge(), MuonSubdetId::CSC, MuonSubdetId::DT, alignCSCRings::e, edm::hlt::Exception, Trajectory::firstMeasurement(), TrajectoryMeasurement::forwardPredictedState(), edm::EventSetup::get(), edm::Event::getByLabel(), susybsm::HSCParticleType::globalMuon, globalMuons_cfi::globalMuons, i, TrajectoryStateOnSurface::isValid(), m_globalMuons, m_globalMuonTracks, m_propagator, m_refitTracker, m_trackTransformer, DetId::Muon, trajectoryStateTransform::persistentState(), Trajectory::push(), edm::OwnVector< T, P >::push_back(), edm::Event::put(), DetId::rawId(), TrajectoryMeasurement::recHit(), TrackTransformer::setServices(), edm::OwnVector< T, P >::size(), 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
    std::auto_ptr<std::vector<Trajectory> > trajectoryCollection(new 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 (trackingRecHit_iterator hit = globalMuon->combinedMuon()->recHitsBegin();  hit != globalMuon->combinedMuon()->recHitsEnd();  ++hit) {
      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.size() > 0) {
      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(trajectoryCollection);
 
    // create the trajectory <-> track association map
    std::auto_ptr<TrajTrackAssociationCollection> trajTrackMap(new 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(trajTrackMap);
 }