SeedToTrackProducer Class Reference

#include <hugues/SeedToTrackProducer/plugins/SeedToTrackProducer.cc>

Inheritance diagram for SeedToTrackProducer:

Public Member Functions
	SeedToTrackProducer (const edm::ParameterSet &)
	~SeedToTrackProducer () override
Public Member Functions inherited from edm::one::EDProducer<>
	EDProducer ()=default
SerialTaskQueue *	globalLuminosityBlocksQueue () final
SerialTaskQueue *	globalRunsQueue () final
bool	hasAbilityToProduceInLumis () const final
bool	hasAbilityToProduceInRuns () const final
bool	wantsGlobalLuminosityBlocks () const final
bool	wantsGlobalRuns () const final
Public Member Functions inherited from edm::one::EDProducerBase
	EDProducerBase ()
ModuleDescription const &	moduleDescription () const
bool	wantsStreamLuminosityBlocks () const
bool	wantsStreamRuns () const
	~EDProducerBase () 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
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)
virtual	~EDConsumerBase () noexcept(false)

Private Member Functions
void	beginJob () override
void	endJob () override
void	produce (edm::Event &, const edm::EventSetup &) override
virtual TrajectoryStateOnSurface	seedTransientState (const TrajectorySeed &)

Private Attributes
edm::EDGetTokenT< edm::View< TrajectorySeed > >	L2seedsTagS_
edm::EDGetTokenT< TrajectorySeedCollection >	L2seedsTagT_
edm::ESHandle< MagneticField >	theMGField
edm::ESHandle< GlobalTrackingGeometry >	theTrackingGeometry

Additional Inherited Members
Public Types inherited from edm::one::EDProducerBase
typedef EDProducerBase	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::one::EDProducerBase
static const std::string &	baseType ()
static void	fillDescriptions (ConfigurationDescriptions &descriptions)
static void	prevalidate (ConfigurationDescriptions &descriptions)
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]

Description:

Definition at line 49 of file SeedToTrackProducer.h.

Constructor & Destructor Documentation

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

explicit

Definition at line 28 of file SeedToTrackProducer.cc.

References edm::ParameterSet::getParameter(), L2seedsTagS_, and L2seedsTagT_.

{
    
  L2seedsTagT_ = consumes<TrajectorySeedCollection>(iConfig.getParameter<edm::InputTag>("L2seedsCollection"));
  L2seedsTagS_ = consumes<edm::View<TrajectorySeed> >(iConfig.getParameter<edm::InputTag>("L2seedsCollection"));

    produces<reco::TrackCollection>();
    produces<reco::TrackExtraCollection>();
    produces<TrackingRecHitCollection>();
}

SeedToTrackProducer::~SeedToTrackProducer ( )

override

Definition at line 39 of file SeedToTrackProducer.cc.

{
}

Member Function Documentation

void SeedToTrackProducer::beginJob ( void  )

overrideprivatevirtual

Reimplemented from edm::one::EDProducerBase.

Definition at line 152 of file SeedToTrackProducer.cc.

{
}

void SeedToTrackProducer::endJob ( void  )

overrideprivatevirtual

Reimplemented from edm::one::EDProducerBase.

Definition at line 158 of file SeedToTrackProducer.cc.

References DEFINE_FWK_MODULE.

                            {
}

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

overrideprivate

Definition at line 49 of file SeedToTrackProducer.cc.

References reco::TrackBase::appendHitPattern(), clone(), funct::cos(), TrajectoryStateOnSurface::curvilinearError(), PV3DBase< T, PVType, FrameType >::eta(), reco::TrackBase::eta(), edm::EventSetup::get(), edm::Event::getByToken(), edm::Event::getRefBeforePut(), TrajectoryStateOnSurface::globalMomentum(), susybsm::HSCParticleType::globalMuon, mps_fire::i, training_settings::idx, edm::HandleBase::isValid(), SeedToTrackProducer_cfi::L2seedsCollection, L2seedsTagS_, L2seedsTagT_, CurvilinearTrajectoryError::matrix(), eostools::move(), PV3DBase< T, PVType, FrameType >::perp(), PV3DBase< T, PVType, FrameType >::phi(), reco::TrackBase::phi(), reco::TrackBase::phiError(), edm::Handle< T >::product(), reco::TrackBase::pt(), reco::TrackBase::ptError(), edm::Event::put(), SurveyInfoScenario_cff::seed, seedTransientState(), trackingPlots::selectedTracks, reco::Track::setExtra(), funct::sin(), theMGField, and theTrackingGeometry.

{
    using namespace edm;
    using namespace std;

    std::unique_ptr<reco::TrackCollection> selectedTracks(new reco::TrackCollection);
    std::unique_ptr<reco::TrackExtraCollection> selectedTrackExtras( new reco::TrackExtraCollection() );
    std::unique_ptr<TrackingRecHitCollection> selectedTrackHits( new TrackingRecHitCollection() );
    
    reco::TrackRefProd rTracks = iEvent.getRefBeforePut<reco::TrackCollection>();
    reco::TrackExtraRefProd rTrackExtras = iEvent.getRefBeforePut<reco::TrackExtraCollection>();
    TrackingRecHitRefProd rHits = iEvent.getRefBeforePut<TrackingRecHitCollection>();
    
    edm::Ref<reco::TrackExtraCollection>::key_type hidx = 0;
    edm::Ref<reco::TrackExtraCollection>::key_type idx = 0;
    
    // magnetic fied and detector geometry
    iSetup.get<IdealMagneticFieldRecord>().get(theMGField);
    iSetup.get<GlobalTrackingGeometryRecord>().get(theTrackingGeometry);

    edm::ESHandle<TrackerTopology> httopo;
    iSetup.get<TrackerTopologyRcd>().get(httopo);
    const TrackerTopology& ttopo = *httopo;

    // now read the L2 seeds collection :
    edm::Handle<TrajectorySeedCollection> L2seedsCollection;
    iEvent.getByToken(L2seedsTagT_,L2seedsCollection);
    const std::vector<TrajectorySeed>* L2seeds = nullptr;
    if (L2seedsCollection.isValid()) L2seeds = L2seedsCollection.product();
    else  edm::LogError("SeedToTrackProducer") << "L2 seeds collection not found !! " << endl;
    
    edm::Handle<edm::View<TrajectorySeed> > seedHandle;
    iEvent.getByToken(L2seedsTagS_, seedHandle);
    
    
    int countRH = 0;

    // now  loop on the seeds :
    for (unsigned int i = 0; i < L2seeds->size() ; i++){
        
        
        //get the kinematic extrapolation from the seed
        TrajectoryStateOnSurface theTrajectory = seedTransientState(L2seeds->at(i));
        float seedEta =  theTrajectory.globalMomentum().eta();
        float seedPhi =  theTrajectory.globalMomentum().phi();
        float seedPt =  theTrajectory.globalMomentum().perp();
        CovarianceMatrix matrixSeedErr = theTrajectory.curvilinearError().matrix();
        edm::LogVerbatim("SeedToTrackProducer") <<  "seedPt=" << seedPt << " seedEta=" << seedEta << " seedPhi=" << seedPhi << endl;
        /*AlgebraicSymMatrix66 errors = theTrajectory.cartesianError().matrix();
        double partialPterror = errors(3,3)*pow(theTrajectory.globalMomentum().x(),2) + errors(4,4)*pow(theTrajectory.globalMomentum().y(),2);
    edm::LogVerbatim("SeedToTrackProducer") <<  "seedPtError=" << sqrt(partialPterror)/theTrajectory.globalMomentum().perp() << "seedPhiError=" << theTrajectory.curvilinearError().matrix()(2,2) << endl;*/ 
        //fill the track in a way that its pt, phi and eta will be the same as the seed
        math::XYZPoint initPoint(0,0,0);
        math::XYZVector initMom(seedPt*cos(seedPhi),seedPt*sin(seedPhi),seedPt*sinh(seedEta));
        reco::Track theTrack(1, 1, //dummy Chi2 and ndof
                             initPoint, initMom,
                             1, matrixSeedErr,
                             reco::TrackBase::TrackAlgorithm::globalMuon, reco::TrackBase::TrackQuality::tight);

        //fill the extra track with dummy information
        math::XYZPoint dummyFinalPoint(1,1,1);
        math::XYZVector dummyFinalMom(0,0,10);
        edm::RefToBase<TrajectorySeed> seed(seedHandle, i);
        CovarianceMatrix matrixExtra = ROOT::Math::SMatrixIdentity();
        reco::TrackExtra theTrackExtra(dummyFinalPoint, dummyFinalMom, true, initPoint, initMom, true,
                                  matrixSeedErr, 1,
                                  matrixExtra, 2,
                                  (L2seeds->at(i)).direction(), seed);
        theTrack.setExtra( reco::TrackExtraRef( rTrackExtras, idx++ ) );
        edm::LogVerbatim("SeedToTrackProducer") << "trackPt=" << theTrack.pt() << " trackEta=" << theTrack.eta() << " trackPhi=" << theTrack.phi() << endl;
        edm::LogVerbatim("SeedToTrackProducer") << "trackPtError=" << theTrack.ptError() << "trackPhiError=" << theTrack.phiError() << endl;
 
        //fill the seed segments in the track
        unsigned int nHitsAdded = 0;
        for(TrajectorySeed::recHitContainer::const_iterator itRecHits=(L2seeds->at(i)).recHits().first; itRecHits!=(L2seeds->at(i)).recHits().second; ++itRecHits, ++countRH) {
            TrackingRecHit* hit = (itRecHits)->clone();
            theTrack.appendHitPattern(*hit, ttopo);
            selectedTrackHits->push_back(hit);
            nHitsAdded++;
        }
        theTrackExtra.setHits( rHits, hidx, nHitsAdded );
        hidx += nHitsAdded;
        selectedTracks->push_back(theTrack);
        selectedTrackExtras->push_back(theTrackExtra);

    }
    iEvent.put(std::move(selectedTracks));
    iEvent.put(std::move(selectedTrackExtras));
    iEvent.put(std::move(selectedTrackHits));
    
}

TrajectoryStateOnSurface SeedToTrackProducer::seedTransientState ( const TrajectorySeed &  tmpSeed )

privatevirtual

Definition at line 141 of file SeedToTrackProducer.cc.

References PTrajectoryStateOnDet::detId(), GlobalTrackingGeometry::idToDet(), TrajectorySeed::startingState(), theMGField, theTrackingGeometry, and trajectoryStateTransform::transientState().

Referenced by produce().

                                                                                             {
    
    PTrajectoryStateOnDet tmpTSOD = tmpSeed.startingState();
    DetId tmpDetId(tmpTSOD.detId());
    const GeomDet* tmpGeomDet = theTrackingGeometry->idToDet(tmpDetId);
    TrajectoryStateOnSurface tmpTSOS = trajectoryStateTransform::transientState(tmpTSOD, &(tmpGeomDet->surface()), &(*theMGField));
    return tmpTSOS;
}

Member Data Documentation

edm::EDGetTokenT<edm::View<TrajectorySeed> > SeedToTrackProducer::L2seedsTagS_

private

Definition at line 62 of file SeedToTrackProducer.h.

Referenced by produce(), and SeedToTrackProducer().

edm::EDGetTokenT<TrajectorySeedCollection> SeedToTrackProducer::L2seedsTagT_

private

Definition at line 61 of file SeedToTrackProducer.h.

Referenced by produce(), and SeedToTrackProducer().

edm::ESHandle<MagneticField> SeedToTrackProducer::theMGField

private

Definition at line 64 of file SeedToTrackProducer.h.

Referenced by produce(), and seedTransientState().

edm::ESHandle<GlobalTrackingGeometry> SeedToTrackProducer::theTrackingGeometry

private

Definition at line 65 of file SeedToTrackProducer.h.

Referenced by produce(), and seedTransientState().