---

TrackRefitter Class Reference

Refit Tracks: Produce Tracks from TrackCollection. More...

#include <RecoTracker/TrackProducer/plugins/TrackRefitter.h>

Inheritance diagram for TrackRefitter:

List of all members.

Public Member Functions
virtual void	produce (edm::Event &, const edm::EventSetup &)
	Implementation of produce method.
	TrackRefitter (const edm::ParameterSet &iConfig)
	Constructor.
Private Types
enum	Constraint { none, momentum, vertex }
Private Attributes
Constraint	constraint_
TrackProducerAlgorithm < reco::Track >	theAlgo

---

Detailed Description

Refit Tracks: Produce Tracks from TrackCollection.

It performs a new final fit on a TrackCollection.

Date

2007/12/07 02:20:40

Revision

1.2

Author:

cerati

Definition at line 15 of file TrackRefitter.h.

---

Member Enumeration Documentation

enum TrackRefitter::Constraint [private]

Enumerator:

none
momentum
vertex

Definition at line 26 of file TrackRefitter.h.

{ none, momentum, vertex };

---

Constructor & Destructor Documentation

TrackRefitter::TrackRefitter

(

const edm::ParameterSet &

iConfig

)

[explicit]

Constructor.

Definition at line 13 of file TrackRefitter.cc.

References TrackProducerBase< reco::Track >::alias_, constraint_, Exception, edm::ParameterSet::getParameter(), momentum, none, TrackProducerBase< reco::Track >::setAlias(), TrackProducerBase< reco::Track >::setConf(), TrackProducerBase< reco::Track >::setSrc(), and vertex.

                                                          :
  KfTrackProducerBase(iConfig.getParameter<bool>("TrajectoryInEvent"),
                      iConfig.getParameter<bool>("useHitsSplitting")),
  theAlgo(iConfig)
{
  setConf(iConfig);
  setSrc( iConfig.getParameter<edm::InputTag>( "src" ), iConfig.getParameter<edm::InputTag>( "beamSpot" ));
  setAlias( iConfig.getParameter<std::string>( "@module_label" ) );
  std::string  constraint_str = iConfig.getParameter<std::string>( "constraint" );

  if (constraint_str == "") constraint_ = none;
  else if (constraint_str == "momentum") constraint_ = momentum;
  else if (constraint_str == "vertex") constraint_ = vertex;
  else {
    edm::LogError("TrackRefitter")<<"constraint: "<<constraint_str<<" not understood. Set it to 'momentum', 'vertex' or leave it empty";    
    throw cms::Exception("TrackRefitter") << "unknown type of contraint! Set it to 'momentum', 'vertex' or leave it empty";    
  }

  //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 TrackRefitter::produce	(	edm::Event &	theEvent,
		const edm::EventSetup &	setup
	)			[virtual]

Implementation of produce method.

Implements edm::EDProducer.

Definition at line 40 of file TrackRefitter.cc.

References TrackProducerBase< reco::Track >::bsSrc_, constraint_, e, edm::Handle< T >::failedToGet(), edm::Event::getByLabel(), edm::Event::getByType(), TrackProducerBase< reco::Track >::getFromES(), TrackProducerBase< reco::Track >::getFromEvt(), edm::Event::id(), LogDebug, momentum, none, edm::ESHandle< T >::product(), KfTrackProducerBase::putInEvt(), TrackProducerAlgorithm< T >::runWithMomentum(), TrackProducerAlgorithm< T >::runWithTrack(), TrackProducerAlgorithm< T >::runWithVertex(), theAlgo, and vertex.

{
  edm::LogInfo("TrackRefitter") << "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;
  //  getFromES(setup,theG,theMF,theFitter,thePropagator);
  edm::ESHandle<TransientTrackingRecHitBuilder> theBuilder;
  getFromES(setup,theG,theMF,theFitter,thePropagator,theBuilder);

  //
  //declare and get TrackCollection to be retrieved from the event
  //
  AlgoProductCollection algoResults;
  reco::BeamSpot bs;
  switch(constraint_){
  case none :
    {
      edm::Handle<reco::TrackCollection> theTCollection;
      getFromEvt(theEvent,theTCollection,bs);
      if (theTCollection.failedToGet()){
        edm::LogError("TrackRefitter")<<"could not get the reco::TrackCollection."; break;}
      LogDebug("TrackRefitter") << "run the algorithm" << "\n";
      try {
        theAlgo.runWithTrack(theG.product(), theMF.product(), *theTCollection, 
                             theFitter.product(), thePropagator.product(), theBuilder.product(), bs, algoResults);
      }catch (cms::Exception &e){ edm::LogError("TrackProducer") << "cms::Exception caught during theAlgo.runWithTrack." << "\n" << e << "\n"; throw; }
      break;
    }
  case momentum :
    {
      edm::Handle<TrackMomConstraintAssociationCollection> theTCollectionWithConstraint;
      theEvent.getByType(theTCollectionWithConstraint);
      edm::Handle<reco::BeamSpot> recoBeamSpotHandle;
      theEvent.getByLabel(bsSrc_,recoBeamSpotHandle);
      bs = *recoBeamSpotHandle;      
      if (theTCollectionWithConstraint.failedToGet()){
        edm::LogError("TrackRefitter")<<"could not get TrackMomConstraintAssociationCollection product."; break;}
      LogDebug("TrackRefitter") << "run the algorithm" << "\n";
      try {
        theAlgo.runWithMomentum(theG.product(), theMF.product(), *theTCollectionWithConstraint, 
                                theFitter.product(), thePropagator.product(), theBuilder.product(), bs, algoResults);
      }catch (cms::Exception &e){ edm::LogError("TrackProducer") << "cms::Exception caught during theAlgo.runWithTrack." << "\n" << e << "\n"; throw; }
      break;}
  case  vertex :
    {
      edm::Handle<TrackVtxConstraintAssociationCollection> theTCollectionWithConstraint;
      theEvent.getByType(theTCollectionWithConstraint);
      edm::Handle<reco::BeamSpot> recoBeamSpotHandle;
      theEvent.getByLabel(bsSrc_,recoBeamSpotHandle);
      bs = *recoBeamSpotHandle;      
      if (theTCollectionWithConstraint.failedToGet()){
        edm::LogError("TrackRefitter")<<"could not get TrackVtxConstraintAssociationCollection product."; break;}
      LogDebug("TrackRefitter") << "run the algorithm" << "\n";
      try {
      theAlgo.runWithVertex(theG.product(), theMF.product(), *theTCollectionWithConstraint, 
                            theFitter.product(), thePropagator.product(), theBuilder.product(), bs, algoResults);      
      }catch (cms::Exception &e){ edm::LogError("TrackProducer") << "cms::Exception caught during theAlgo.runWithTrack." << "\n" << e << "\n"; throw; }
    }
    //default... there cannot be any other possibility due to the check in the ctor
  }

  
  //put everything in th event
  putInEvt(theEvent, outputRHColl, outputTColl, outputTEColl, outputTrajectoryColl, algoResults);
  LogDebug("TrackRefitter") << "end" << "\n";
}

---

Member Data Documentation

Constraint TrackRefitter::constraint_ [private]

Definition at line 27 of file TrackRefitter.h.

Referenced by produce(), and TrackRefitter().

TrackProducerAlgorithm<reco::Track> TrackRefitter::theAlgo [private]

Definition at line 25 of file TrackRefitter.h.

Referenced by produce().

---

The documentation for this class was generated from the following files:

• RecoTracker/TrackProducer/plugins/TrackRefitter.h
• RecoTracker/TrackProducer/plugins/TrackRefitter.cc

---