#include <FastTSGFromL2Muon.h>

Inheritance diagram for FastTSGFromL2Muon:

Public Member Functions
virtual void	beginRun (edm::Run const &run, edm::EventSetup const &es) override

	FastTSGFromL2Muon (const edm::ParameterSet &cfg)

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

virtual	~FastTSGFromL2Muon ()

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
bool	clean (reco::TrackRef muRef, RectangularEtaPhiTrackingRegion region, const BasicTrajectorySeed aSeed, const SimTrack &theSimTrack)

Private Attributes
edm::ParameterSet	theConfig

edm::InputTag	theL2CollectionLabel

double	thePtCut

MuonTrackingRegionBuilder *	theRegionBuilder

std::vector< edm::InputTag >	theSeedCollectionLabels

MuonServiceProxy *	theService

edm::InputTag	theSimTrackCollectionLabel

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

Definition at line 23 of file FastTSGFromL2Muon.h.

Constructor & Destructor Documentation

FastTSGFromL2Muon::FastTSGFromL2Muon ( const edm::ParameterSet & cfg )

Definition at line 21 of file FastTSGFromL2Muon.cc.

References edm::ParameterSet::getParameter(), MuonServiceProxy_cff::MuonServiceProxy, theL2CollectionLabel, thePtCut, theSeedCollectionLabels, theService, and theSimTrackCollectionLabel.

                                                                : theConfig(cfg) 
 {
   produces<L3MuonTrajectorySeedCollection>();
 
   edm::ParameterSet serviceParameters = 
     cfg.getParameter<edm::ParameterSet>("ServiceParameters");
   theService = new MuonServiceProxy(serviceParameters);
 
   thePtCut = cfg.getParameter<double>("PtCut");
 
   theL2CollectionLabel = cfg.getParameter<edm::InputTag>("MuonCollectionLabel");
   theSeedCollectionLabels = cfg.getParameter<std::vector<edm::InputTag> >("SeedCollectionLabels");
   theSimTrackCollectionLabel  = cfg.getParameter<edm::InputTag>("SimTrackCollectionLabel");
   // useTFileService_ = cfg.getUntrackedParameter<bool>("UseTFileService",false);
 
 }

FastTSGFromL2Muon::~FastTSGFromL2Muon ( )

virtual

Definition at line 38 of file FastTSGFromL2Muon.cc.

39 {

40 }

Member Function Documentation

void FastTSGFromL2Muon::beginRun	(	edm::Run const &	run,
		edm::EventSetup const &	es
	)

overridevirtual

Reimplemented from edm::EDProducer.

Definition at line 43 of file FastTSGFromL2Muon.cc.

References edm::EDConsumerBase::consumesCollector(), edm::ParameterSet::getParameter(), theConfig, theRegionBuilder, theService, and MuonServiceProxy::update().

 {
   //update muon proxy service
   theService->update(es);
   
   //region builder
   edm::ParameterSet regionBuilderPSet = 
     theConfig.getParameter<edm::ParameterSet>("MuonTrackingRegionBuilder");
   edm::ConsumesCollector iC  = consumesCollector();
   theRegionBuilder = new MuonTrackingRegionBuilder(regionBuilderPSet,theService,iC);
   
   /*
   if(useTFileService_) {
     edm::Service<TFileService> fs;
     h_nSeedPerTrack = fs->make<TH1F>("nSeedPerTrack","nSeedPerTrack",76,-0.5,75.5);
     h_nGoodSeedPerTrack = fs->make<TH1F>("nGoodSeedPerTrack","nGoodSeedPerTrack",75,-0.5,75.5);
     h_nGoodSeedPerEvent = fs->make<TH1F>("nGoodSeedPerEvent","nGoodSeedPerEvent",75,-0.5,75.5);
   } else {
     h_nSeedPerTrack = 0;
     h_nGoodSeedPerEvent = 0;
     h_nGoodSeedPerTrack = 0;
   }
   */
 
 }

bool FastTSGFromL2Muon::clean	(	reco::TrackRef	muRef,
		RectangularEtaPhiTrackingRegion *	region,
		const BasicTrajectorySeed *	aSeed,
		const SimTrack &	theSimTrack
	)

private

Definition at line 181 of file FastTSGFromL2Muon.cc.

References SiPixelRawToDigiRegional_cfi::deltaPhi, MuonErrorMatrixValues_cff::etaRange, PixelRecoRange< T >::max(), PixelRecoRange< T >::mean(), CoreSimTrack::momentum(), PtMinSelector_cfg::ptMin, TkTrackingRegionsMargin< T >::right(), and mathSSE::sqrt().

Referenced by produce().

                                           { 
     
   // Eta cleaner   
   const PixelRecoRange<float>& etaRange = region->etaRange() ;  
   double etaSeed = theSimTrack.momentum().Eta();
   double etaLimit  = (fabs(fabs(etaRange.max())-fabs(etaRange.mean())) <0.05) ? 
     0.05 : fabs(fabs(etaRange.max()) - fabs(etaRange.mean())) ;
   bool inEtaRange = 
     etaSeed >= (etaRange.mean() - etaLimit) && 
     etaSeed <= (etaRange.mean() + etaLimit) ;
   if  ( !inEtaRange ) return false;
   
   // Phi cleaner
   const TkTrackingRegionsMargin<float>& phiMargin = region->phiMargin();
   double phiSeed = theSimTrack.momentum().Phi(); 
   double phiLimit  = (phiMargin.right() < 0.05 ) ? 0.05 : phiMargin.right(); 
   bool inPhiRange = 
     (fabs(deltaPhi(phiSeed,double(region->direction().phi()))) < phiLimit );
   if  ( !inPhiRange ) return false;
   
   // pt cleaner
   double ptSeed  = std::sqrt(theSimTrack.momentum().Perp2());
   double ptMin   = (region->ptMin()>3.5) ? 3.5: region->ptMin();  
   bool inPtRange = ptSeed >= ptMin &&  ptSeed<= 2*(muRef->pt());
   return inPtRange;
   
 }

void FastTSGFromL2Muon::produce	(	edm::Event &	ev,
		const edm::EventSetup &	es
	)

overridevirtual

Implements edm::EDProducer.

Definition at line 71 of file FastTSGFromL2Muon.cc.

References clean(), edm::Event::getByLabel(), iseed, edm::Event::put(), TrajectorySeed::recHits(), MuonTrackingRegionBuilder::region(), query::result, MuonTrackingRegionBuilder::setEvent(), SiTrackerGSMatchedRecHit2D::simtrackId(), theL2CollectionLabel, thePtCut, theRegionBuilder, theSeedCollectionLabels, theService, theSimTrackCollectionLabel, and MuonServiceProxy::update().

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

 {
 
   // Initialize the output product
   std::auto_ptr<L3MuonTrajectorySeedCollection> result(new L3MuonTrajectorySeedCollection());
 
   //intialize service
   theService->update(es);
   
   // Region builder
   theRegionBuilder->setEvent(ev);
 
   // Retrieve the Monte Carlo truth (SimTracks)
   edm::Handle<edm::SimTrackContainer> theSimTracks;
   ev.getByLabel(theSimTrackCollectionLabel,theSimTracks);
 
   // Retrieve L2 muon collection
   edm::Handle<reco::TrackCollection> l2muonH;
   ev.getByLabel(theL2CollectionLabel ,l2muonH); 
 
   // Retrieve Seed collection
   unsigned seedCollections = theSeedCollectionLabels.size();
   std::vector<edm::Handle<edm::View<TrajectorySeed> > > theSeeds;
   theSeeds.resize(seedCollections);
   unsigned seed_size = 0;
   for ( unsigned iseed=0; iseed<seedCollections; ++iseed ) { 
     ev.getByLabel(theSeedCollectionLabels[iseed], theSeeds[iseed]);
     seed_size += theSeeds[iseed]->size();
   }
 
   // Loop on L2 muons
   unsigned int imu=0;
   unsigned int imuMax=l2muonH->size();
   // std::cout << "Found " << imuMax << " L2 muons" << std::endl;
   for (;imu!=imuMax;++imu){
 
     // Make a ref to l2 muon
     reco::TrackRef muRef(l2muonH, imu);
     
     // Cut on muons with low momenta
     if ( muRef->pt() < thePtCut 
      || muRef->innerMomentum().Rho() < thePtCut 
      || muRef->innerMomentum().R() < 2.5 ) continue;
     
     // Define the region of interest
     RectangularEtaPhiTrackingRegion * region = theRegionBuilder->region(muRef);
 
     // Copy the collection of seeds (ahem, this is time consuming!)
     std::vector<TrajectorySeed> tkSeeds;
     std::set<unsigned> tkIds;
     tkSeeds.reserve(seed_size);
     for ( unsigned iseed=0; iseed<seedCollections; ++iseed ) { 
       edm::Handle<edm::View<TrajectorySeed> > aSeedCollection = theSeeds[iseed];
       unsigned nSeeds = aSeedCollection->size();
       for (unsigned seednr = 0; seednr < nSeeds; ++seednr) {
             
     // The seed
     const BasicTrajectorySeed* aSeed = &((*aSeedCollection)[seednr]);
     
     // Find the first hit of the Seed
     TrajectorySeed::range theSeedingRecHitRange = aSeed->recHits();
     const SiTrackerGSMatchedRecHit2D * theFirstSeedingRecHit = 
       (const SiTrackerGSMatchedRecHit2D*) (&(*(theSeedingRecHitRange.first)));
 
     // The SimTrack id associated to that recHit
     int simTrackId = theFirstSeedingRecHit->simtrackId();
 
     // Track already associated to a seed
     std::set<unsigned>::iterator tkId = tkIds.find(simTrackId);
     if( tkId != tkIds.end() ) continue; 
 
     const SimTrack& theSimTrack = (*theSimTracks)[simTrackId]; 
 
     if ( clean(muRef,region,aSeed,theSimTrack) ) tkSeeds.push_back(*aSeed);
     tkIds.insert(simTrackId);
 
       } // End loop on seeds
  
     } // End loop on seed collections
   
     // Free memory
     delete region;
 
     // A plot
     // if(h_nSeedPerTrack) h_nSeedPerTrack->Fill(tkSeeds.size());
 
     // Another plot
     // if(h_nGoodSeedPerTrack) h_nGoodSeedPerTrack->Fill(tkSeeds.size());
     
 
     // Now create the Muon Trajectory Seed
     unsigned int is=0;
     unsigned int isMax=tkSeeds.size();
     for (;is!=isMax;++is){
       result->push_back( L3MuonTrajectorySeed(tkSeeds[is], muRef));
     } // End of tk seed loop
     
   } // End of l2 muon loop
 
   // std::cout << "Found " << result->size() << " seeds for muons" << std::endl;
 
   // And yet another plot
   // if(h_nGoodSeedPerEvent) h_nGoodSeedPerEvent->Fill(result->size());
   
   //put in the event
   ev.put(result);
 
 }