MTFTrackProducer Class Reference

#include <MTFTrackProducer.h>

Inheritance diagram for MTFTrackProducer:

Public Types
typedef std::vector< Trajectory >	TrajectoryCollection
Public Types inherited from TrackProducerBase< reco::Track >
typedef std::pair< Trajectory *, std::pair< reco::Track *, PropagationDirection > >	AlgoProduct
typedef std::vector< AlgoProduct >	AlgoProductCollection
typedef std::vector< reco::Track >	TrackCollection
Public Types inherited from edm::EDProducer
typedef EDProducer	ModuleType
typedef WorkerT< EDProducer >	WorkerType
Public Types inherited from edm::ProducerBase
typedef ProductRegistryHelper::TypeLabelList	TypeLabelList

Public Member Functions
	MTFTrackProducer (const edm::ParameterSet &iConfig)
	Constructor. More...
virtual void	produce (edm::Event &, const edm::EventSetup &)
	Implementation of produce method. More...
Public Member Functions inherited from KfTrackProducerBase
	KfTrackProducerBase (bool trajectoryInEvent, bool split)
	Constructor. More...
virtual void	putInEvt (edm::Event &, const Propagator *prop, const MeasurementTracker *measTk, std::auto_ptr< TrackingRecHitCollection > &, std::auto_ptr< reco::TrackCollection > &, std::auto_ptr< reco::TrackExtraCollection > &, std::auto_ptr< std::vector< Trajectory > > &, AlgoProductCollection &)
	Put produced collections in the event. More...
Public Member Functions inherited from TrackProducerBase< reco::Track >
const edm::ParameterSet &	getConf () const
virtual void	getFromES (const edm::EventSetup &, edm::ESHandle< TrackerGeometry > &, edm::ESHandle< MagneticField > &, edm::ESHandle< TrajectoryFitter > &, edm::ESHandle< Propagator > &, edm::ESHandle< MeasurementTracker > &, edm::ESHandle< TransientTrackingRecHitBuilder > &)
	Get needed services from the Event Setup. More...
virtual void	getFromEvt (edm::Event &, edm::Handle< TrackCandidateCollection > &, reco::BeamSpot &)
	Get TrackCandidateCollection from the Event (needed by TrackProducer) More...
virtual void	getFromEvt (edm::Event &, edm::Handle< TrackCollection > &, reco::BeamSpot &)
	Get TrackCollection from the Event (needed by TrackRefitter) More...
void	setAlias (std::string alias)
	set the aliases of produced collections More...
void	setClusterRemovalInfo (const edm::InputTag &clusterRemovalInfo)
	Sets the information on cluster removal, and turns it on. More...
void	setConf (edm::ParameterSet conf)
	Set parameter set. More...
void	setSecondHitPattern (Trajectory *traj, reco::Track &track, const Propagator *prop, const MeasurementTracker *measTk)
void	setSrc (edm::InputTag src, edm::InputTag bsSrc)
	set label of source collection More...
	TrackProducerBase (bool trajectoryInEvent=false)
	Constructor. More...
virtual	~TrackProducerBase ()
	Destructor. More...
Public Member Functions inherited from edm::EDProducer
	EDProducer ()
virtual	~EDProducer ()
Public Member Functions inherited from edm::ProducerBase
	ProducerBase ()
void	registerProducts (ProducerBase *, ProductRegistry *, ModuleDescription const &)
boost::function< void(const BranchDescription &)>	registrationCallback () const
	used by the fwk to register list of products More...
virtual	~ProducerBase ()

Private Member Functions
void	getFromEvt (edm::Event &, edm::Handle< TrajectoryCollection > &, reco::BeamSpot &)

Private Attributes
MTFTrackProducerAlgorithm	theAlgo

Additional Inherited Members
Static Public Member Functions inherited from edm::EDProducer
static const std::string &	baseType ()
static void	fillDescriptions (ConfigurationDescriptions &descriptions)
Protected Member Functions inherited from edm::EDProducer
CurrentProcessingContext const *	currentContext () const
Protected Member Functions inherited from edm::ProducerBase
template<class TProducer , class TMethod >
void	callWhenNewProductsRegistered (TProducer *iProd, TMethod iMethod)
Protected Attributes inherited from TrackProducerBase< reco::Track >
std::string	alias_
edm::InputTag	bsSrc_
edm::InputTag	clusterRemovalInfo_
bool	rekeyClusterRefs_
edm::OrphanHandle < TrackCollection >	rTracks_
edm::ESHandle< NavigationSchool >	theSchool
bool	trajectoryInEvent_

Detailed Description

Definition at line 8 of file MTFTrackProducer.h.

Member Typedef Documentation

typedef std::vector<Trajectory> MTFTrackProducer::TrajectoryCollection

Definition at line 10 of file MTFTrackProducer.h.

Constructor & Destructor Documentation

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

explicit

Constructor.

Definition at line 18 of file MTFTrackProducer.cc.

References TrackProducerBase< reco::Track >::alias_, edm::ParameterSet::getParameter(), TrackProducerBase< reco::Track >::setAlias(), TrackProducerBase< reco::Track >::setConf(), and TrackProducerBase< reco::Track >::setSrc().

                                                                :
  KfTrackProducerBase(iConfig.getParameter<bool>("TrajectoryInEvent"),false),
  theAlgo(iConfig)
{
  setConf(iConfig);
  setSrc( iConfig.getParameter<edm::InputTag>( "src" ), iConfig.getParameter<edm::InputTag>( "beamSpot" ));
  setAlias( iConfig.getParameter<std::string>( "@module_label" ) );
  //register your products
  produces<reco::TrackCollection>().setBranchAlias( alias_ + "Tracks" );
  produces<reco::TrackExtraCollection>().setBranchAlias( alias_ + "TrackExtras" );
  produces<TrackingRecHitCollection>().setBranchAlias( alias_ + "RecHits" );
  produces<std::vector<Trajectory> >() ;
  produces<TrajTrackAssociationCollection>();
}

Member Function Documentation

void MTFTrackProducer::getFromEvt ( edm::Event &  theEvent, edm::Handle< TrajectoryCollection > &  theTrajectoryCollection, reco::BeamSpot &  bs  )

private

Definition at line 87 of file MTFTrackProducer.cc.

References TrackProducerBase< reco::Track >::bsSrc_, edm::Event::getByLabel(), TrackProducerBase< reco::Track >::getConf(), edm::ParameterSet::getParameter(), and TrackProducerBase< reco::Track >::src_.

Referenced by produce().

{

  edm::InputTag src_=getConf().getParameter<edm::InputTag>( "src" );
  theEvent.getByLabel(src_,theTrajectoryCollection );  
  
  //get the BeamSpot
  edm::Handle<reco::BeamSpot> recoBeamSpotHandle;
  theEvent.getByLabel(bsSrc_,recoBeamSpotHandle);
  bs = *recoBeamSpotHandle;
}

void MTFTrackProducer::produce ( edm::Event &  theEvent, const edm::EventSetup &  setup  )

virtual

Implementation of produce method.

Implements edm::EDProducer.

Definition at line 34 of file MTFTrackProducer.cc.

References edm::EventSetup::get(), TrackProducerBase< reco::Track >::getConf(), TrackProducerBase< reco::Track >::getFromES(), getFromEvt(), edm::ParameterSet::getParameter(), edm::EventBase::id(), LogDebug, edm::ESHandle< class >::product(), KfTrackProducerBase::putInEvt(), MTFTrackProducerAlgorithm::runWithCandidate(), and theAlgo.

{
  edm::LogInfo("MTFTrackProducer") << "Analyzing event number: " << theEvent.id() << "\n";
  //
  // create empty output collections
  //
  std::auto_ptr<TrackingRecHitCollection>    outputRHColl (new TrackingRecHitCollection);
  std::auto_ptr<reco::TrackCollection>       outputTColl(new reco::TrackCollection);
  std::auto_ptr<reco::TrackExtraCollection>  outputTEColl(new reco::TrackExtraCollection);
  std::auto_ptr<std::vector<Trajectory> >    outputTrajectoryColl(new std::vector<Trajectory>);

  //
  //declare and get stuff to be retrieved from ES
  //
  edm::ESHandle<TrackerGeometry> theG;
  edm::ESHandle<MagneticField> theMF;
  edm::ESHandle<TrajectoryFitter> theFitter;
  edm::ESHandle<Propagator> thePropagator;
  edm::ESHandle<MeasurementTracker>  theMeasTk;
  edm::ESHandle<TransientTrackingRecHitBuilder> theBuilder;
  getFromES(setup,theG,theMF,theFitter,thePropagator,theMeasTk,theBuilder);
  //get additional es_modules needed by the MTF 
  edm::ESHandle<MultiTrackFilterHitCollector> measurementCollectorHandle;
  edm::ESHandle<SiTrackerMultiRecHitUpdatorMTF> updatorHandle;  
  std::string measurementCollectorName = getConf().getParameter<std::string>("MeasurementCollector");
  setup.get<MultiRecHitRecord>().get(measurementCollectorName, measurementCollectorHandle);
  std::string  updatorName = getConf().getParameter<std::string>("UpdatorName");    
  setup.get<MultiRecHitRecord>().get(updatorName, updatorHandle);    

  AlgoProductCollection algoResults;

  edm::Handle<std::vector<Trajectory> > theTrajectoryCollection;

  reco::BeamSpot bs;


  getFromEvt(theEvent,theTrajectoryCollection,bs);

  measurementCollectorHandle->updateEvent(theEvent);    
  //
  //run the algorithm  
    //

  theAlgo.runWithCandidate(theG.product(), theMF.product(), *theTrajectoryCollection,
               theFitter.product(), theBuilder.product(), measurementCollectorHandle.product(), updatorHandle.product(),bs,algoResults);
  //
  //put everything in the event
  putInEvt(theEvent, thePropagator.product(), theMeasTk.product(), 
       outputRHColl, outputTColl, outputTEColl, outputTrajectoryColl, algoResults);
  LogDebug("MTFTrackProducer") << "end" << "\n";
}

Member Data Documentation

MTFTrackProducerAlgorithm MTFTrackProducer::theAlgo

private

Definition at line 18 of file MTFTrackProducer.h.

Referenced by produce().