|
|
|
#include <TrackProducerWithSCAssociation.h>
Public Member Functions | |
| std::vector< reco::TransientTrack > | getTransient (edm::Event &, const edm::EventSetup &) |
| virtual void | produce (edm::Event &, const edm::EventSetup &) |
| TrackProducerWithSCAssociation (const edm::ParameterSet &iConfig) | |
Private Member Functions | |
| void | putInEvt (edm::Event &evt, const Propagator *thePropagator, const MeasurementTracker *theMeasTk, std::auto_ptr< TrackingRecHitCollection > &selHits, std::auto_ptr< reco::TrackCollection > &selTracks, std::auto_ptr< reco::TrackExtraCollection > &selTrackExtras, std::auto_ptr< std::vector< Trajectory > > &selTrajectories, AlgoProductCollection &algoResults) |
Private Attributes | |
| std::string | conversionTrackCandidateProducer_ |
| std::string | myname_ |
| bool | myTrajectoryInEvent_ |
| edm::OrphanHandle < reco::TrackCollection > | rTracks_ |
| TrackProducerAlgorithm < reco::Track > | theAlgo |
| std::string | trackCSuperClusterAssociationCollection_ |
| std::string | trackSuperClusterAssociationCollection_ |
| bool | validTrackCandidateSCAssociationInput_ |
Definition at line 20 of file TrackProducerWithSCAssociation.h.
| TrackProducerWithSCAssociation::TrackProducerWithSCAssociation | ( | const edm::ParameterSet & | iConfig | ) | [explicit] |
Definition at line 20 of file TrackProducerWithSCAssociation.cc.
References TrackProducerBase< reco::Track >::alias_, conversionTrackCandidateProducer_, edm::ParameterSet::exists(), edm::ParameterSet::getParameter(), myname_, myTrajectoryInEvent_, TrackProducerBase< reco::Track >::setAlias(), TrackProducerBase< reco::Track >::setClusterRemovalInfo(), TrackProducerBase< reco::Track >::setConf(), TrackProducerBase< reco::Track >::setSrc(), GlobalPosition_Frontier_DevDB_cff::tag, trackCSuperClusterAssociationCollection_, and trackSuperClusterAssociationCollection_.
: TrackProducerBase<reco::Track>(iConfig.getParameter<bool>("TrajectoryInEvent")), theAlgo(iConfig) { setConf(iConfig); setSrc( iConfig.getParameter<edm::InputTag>( "src" ), iConfig.getParameter<edm::InputTag>( "beamSpot" )); setAlias( iConfig.getParameter<std::string>( "@module_label" ) ); if ( iConfig.exists("clusterRemovalInfo") ) { edm::InputTag tag = iConfig.getParameter<edm::InputTag>("clusterRemovalInfo"); if (!(tag == edm::InputTag())) { setClusterRemovalInfo( tag ); } } myname_ = iConfig.getParameter<std::string>("ComponentName"); conversionTrackCandidateProducer_ = iConfig.getParameter<std::string>("producer"); trackCSuperClusterAssociationCollection_ = iConfig.getParameter<std::string>("trackCandidateSCAssociationCollection"); trackSuperClusterAssociationCollection_ = iConfig.getParameter<std::string>("recoTrackSCAssociationCollection"); myTrajectoryInEvent_ = iConfig.getParameter<bool>("TrajectoryInEvent"); //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>(); // produces< reco::TrackSuperClusterAssociationCollection > (trackSuperClusterAssociationCollection_ ); produces< reco::TrackCaloClusterPtrAssociation > (trackSuperClusterAssociationCollection_ ); }
| std::vector< reco::TransientTrack > TrackProducerWithSCAssociation::getTransient | ( | edm::Event & | theEvent, |
| const edm::EventSetup & | setup | ||
| ) |
Definition at line 200 of file TrackProducerWithSCAssociation.cc.
References alignCSCRings::e, TrackProducerBase< reco::Track >::getFromES(), TrackProducerBase< reco::Track >::getFromEvt(), edm::EventBase::id(), parseEventContent::prod, edm::ESHandle< T >::product(), TrackProducerAlgorithm< T >::runWithCandidate(), and theAlgo.
{
edm::LogInfo("TrackProducerWithSCAssociation") << "Analyzing event number: " << theEvent.id() << "\n";
//
// create empty output collections
//
std::vector<reco::TransientTrack> ttks;
//
//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<TransientTrackingRecHitBuilder> theBuilder;
edm::ESHandle<MeasurementTracker> theMeasTk;
getFromES(setup,theG,theMF,theFitter,thePropagator,theMeasTk,theBuilder);
//
//declare and get TrackColection to be retrieved from the event
//
AlgoProductCollection algoResults;
reco::BeamSpot bs;
try{
edm::Handle<TrackCandidateCollection> theTCCollection;
getFromEvt(theEvent,theTCCollection,bs);
//
//run the algorithm
//
//LogDebug("TrackProducerWithSCAssociation") << "TrackProducerWithSCAssociation run the algorithm" << "\n";
theAlgo.runWithCandidate(theG.product(), theMF.product(), *theTCCollection,
theFitter.product(), thePropagator.product(), theBuilder.product(), bs, algoResults);
} catch (cms::Exception &e){ edm::LogInfo("TrackProducerWithSCAssociation") << "cms::Exception caught!!!" << "\n" << e << "\n";}
for (AlgoProductCollection::iterator prod=algoResults.begin();prod!=algoResults.end(); prod++){
ttks.push_back( reco::TransientTrack(*(((*prod).second).first),thePropagator.product()->magneticField() ));
}
//LogDebug("TrackProducerWithSCAssociation") << "TrackProducerWithSCAssociation end" << "\n";
return ttks;
}
| void TrackProducerWithSCAssociation::produce | ( | edm::Event & | theEvent, |
| const edm::EventSetup & | setup | ||
| ) | [virtual] |
Implements edm::EDProducer.
Definition at line 54 of file TrackProducerWithSCAssociation.cc.
References TrackProducerAlgorithm< T >::buildTrack(), cont, conversionTrackCandidateProducer_, PTrajectoryStateOnDet::detId(), alignCSCRings::e, edm::HandleBase::failedToGet(), edm::helper::Filler< Map >::fill(), edm::Event::getByLabel(), TrackProducerBase< reco::Track >::getFromES(), TrackProducerBase< reco::Track >::getFromEvt(), i, edm::helper::Filler< Map >::insert(), edm::HandleBase::isValid(), convertSQLiteXML::ok, edm::ESHandle< T >::product(), edm::Handle< T >::product(), edm::Event::put(), putInEvt(), TrackCandidate::recHits(), rTracks_, TrackCandidate::seed(), TrackCandidate::seedRef(), edm::ValueMap< T >::size(), theAlgo, trackCSuperClusterAssociationCollection_, trackSuperClusterAssociationCollection_, TrackCandidate::trajectoryStateOnDet(), trajectoryStateTransform::transientState(), and validTrackCandidateSCAssociationInput_.
{
//edm::LogInfo("TrackProducerWithSCAssociation") << "Analyzing event number: " << theEvent.id() << "\n";
//LogDebug("TrackProducerWithSCAssociation") << "Analyzing event number: " << theEvent.id() << "\n";
// std::cout << " TrackProducerWithSCAssociation 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>);
// Reco Track - Super Cluster Association
std::auto_ptr<reco::TrackCaloClusterPtrAssociation> scTrkAssoc_p(new reco::TrackCaloClusterPtrAssociation);
//
//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<TransientTrackingRecHitBuilder> theBuilder;
edm::ESHandle<MeasurementTracker> theMeasTk;
getFromES(setup,theG,theMF,theFitter,thePropagator,theMeasTk,theBuilder);
//
//declare and get TrackColection to be retrieved from the event
edm::Handle<TrackCandidateCollection> theTCCollection;
validTrackCandidateSCAssociationInput_=true;
edm::Handle<reco::TrackCandidateCaloClusterPtrAssociation> trkCandidateSCAssocHandle;
theEvent.getByLabel(conversionTrackCandidateProducer_, trackCSuperClusterAssociationCollection_ , trkCandidateSCAssocHandle);
if ( !trkCandidateSCAssocHandle.isValid() ) {
// std::cout << "Error! Can't get the product "<<trackCSuperClusterAssociationCollection_.c_str() << " but keep running. Empty collection will be produced " << "\n";
edm::LogError("TrackProducerWithSCAssociation") << "Error! Can't get the product "<<trackCSuperClusterAssociationCollection_.c_str() << " but keep running. Empty collection will be produced " << "\n";
validTrackCandidateSCAssociationInput_=false;
}
reco::TrackCandidateCaloClusterPtrAssociation scTrkCandAssCollection = *(trkCandidateSCAssocHandle.product());
if ( scTrkCandAssCollection.size() ==0 ) validTrackCandidateSCAssociationInput_=false;
std::vector<int> tccLocations;
AlgoProductCollection algoResults;
reco::BeamSpot bs;
getFromEvt(theEvent,theTCCollection,bs);
if (theTCCollection.failedToGet()){
edm::LogError("TrackProducerWithSCAssociation") <<"TrackProducerWithSCAssociation could not get the TrackCandidateCollection.";}
else{
//
//run the algorithm
//
// LogDebug("TrackProducerWithSCAssociation") << "TrackProducerWithSCAssociation run the algorithm" << "\n";
// theAlgo.runWithCandidate(theG.product(), theMF.product(), *theTCCollection,
// theFitter.product(), thePropagator.product(), theBuilder.product(), algoResults);
// we have to copy this method from the algo in order to get the association track-seed
// this is ugly temporary code that should be replaced!!!!!
// start of copied code ======================================================
// std::cout << "TrackProducerWithSCAssociation Number of TrackCandidates: " << theTCCollection->size() << "\n";
try{
int cont = 0;
int tcc=0;
for (TrackCandidateCollection::const_iterator i=theTCCollection->begin(); i!=theTCCollection->end();i++)
{
const TrackCandidate * theTC = &(*i);
PTrajectoryStateOnDet state = theTC->trajectoryStateOnDet();
const TrackCandidate::range& recHitVec=theTC->recHits();
const TrajectorySeed& seed = theTC->seed();
//convert PTrajectoryStateOnDet to TrajectoryStateOnSurface
DetId detId(state.detId());
TrajectoryStateOnSurface theTSOS = trajectoryStateTransform::transientState( state,
&(theG.product()->idToDet(detId)->surface()),
theMF.product());
//LogDebug("TrackProducerWithSCAssociation") << "TrackProducerWithSCAssociation Initial TSOS\n" << theTSOS << "\n";
//convert the TrackingRecHit vector to a TransientTrackingRecHit vector
//meanwhile computes the number of degrees of freedom
TransientTrackingRecHit::RecHitContainer hits;
float ndof=0;
for (edm::OwnVector<TrackingRecHit>::const_iterator i=recHitVec.first;
i!=recHitVec.second; i++){
hits.push_back(theBuilder.product()->build(&(*i) ));
}
//build Track
// LogDebug("TrackProducerWithSCAssociation") << "TrackProducerWithSCAssociation going to buildTrack"<< "\n";
bool ok = theAlgo.buildTrack(theFitter.product(),thePropagator.product(),algoResults, hits, theTSOS, seed, ndof, bs, theTC->seedRef());
// LogDebug("TrackProducerWithSCAssociation") << "TrackProducerWithSCAssociation buildTrack result: " << ok << "\n";
if(ok) {
cont++;
tccLocations.push_back(tcc);
}
tcc++;
}
edm::LogInfo("TrackProducerWithSCAssociation") << "Number of Tracks found: " << cont << "\n";
//LogDebug("TrackProducerWithSCAssociation") << "TrackProducerWithSCAssociation Number of Tracks found: " << cont << "\n";
// end of copied code ======================================================
} catch (cms::Exception &e){ edm::LogInfo("TrackProducerWithSCAssociation") << "cms::Exception caught!!!" << "\n" << e << "\n";}
//
//put everything in the event
// we copy putInEvt to get OrphanHandle filled...
putInEvt(theEvent,thePropagator.product(),theMeasTk.product(),
outputRHColl, outputTColl, outputTEColl, outputTrajectoryColl, algoResults);
// now construct associationmap and put it in the event
if ( validTrackCandidateSCAssociationInput_ ) {
int itrack=0;
std::vector<edm::Ptr<reco::CaloCluster> > caloPtrVec;
for(AlgoProductCollection::iterator i=algoResults.begin(); i!=algoResults.end();i++){
edm::Ref<TrackCandidateCollection> trackCRef(theTCCollection,tccLocations[itrack]);
const edm::Ptr<reco::CaloCluster>& aClus = (*trkCandidateSCAssocHandle)[trackCRef];
caloPtrVec.push_back( aClus );
itrack++;
}
edm::ValueMap<reco::CaloClusterPtr>::Filler filler(*scTrkAssoc_p);
filler.insert(rTracks_, caloPtrVec.begin(), caloPtrVec.end());
filler.fill();
}
theEvent.put(scTrkAssoc_p,trackSuperClusterAssociationCollection_ );
}
}
| void TrackProducerWithSCAssociation::putInEvt | ( | edm::Event & | evt, |
| const Propagator * | thePropagator, | ||
| const MeasurementTracker * | theMeasTk, | ||
| std::auto_ptr< TrackingRecHitCollection > & | selHits, | ||
| std::auto_ptr< reco::TrackCollection > & | selTracks, | ||
| std::auto_ptr< reco::TrackExtraCollection > & | selTrackExtras, | ||
| std::auto_ptr< std::vector< Trajectory > > & | selTrajectories, | ||
| AlgoProductCollection & | algoResults | ||
| ) | [private] |
Definition at line 251 of file TrackProducerWithSCAssociation.cc.
References reco::TrackExtraBase::add(), alongMomentum, TrajectoryStateOnSurface::curvilinearError(), Trajectory::direction(), Trajectory::firstMeasurement(), edm::Event::getRefBeforePut(), TrajectoryStateOnSurface::globalParameters(), i, edm::ESHandleBase::isValid(), j, gen::k, Trajectory::lastMeasurement(), GlobalTrajectoryParameters::momentum(), myTrajectoryInEvent_, AlCaHLTBitMon_ParallelJobs::p, GlobalTrajectoryParameters::position(), edm::Event::put(), TrajectoryMeasurement::recHit(), Trajectory::recHits(), rTracks_, Trajectory::seedRef(), reco::Track::setExtra(), reco::TrackBase::setHitPattern(), TrackProducerBase< reco::Track >::setSecondHitPattern(), lumiQTWidget::t, TrackProducerBase< reco::Track >::theSchool, TrackProducerBase< reco::Track >::trajectoryInEvent_, TrajectoryMeasurement::updatedState(), v, PV3DBase< T, PVType, FrameType >::x(), PV3DBase< T, PVType, FrameType >::y(), and PV3DBase< T, PVType, FrameType >::z().
Referenced by produce().
{
TrackingRecHitRefProd rHits = evt.getRefBeforePut<TrackingRecHitCollection>();
reco::TrackExtraRefProd rTrackExtras = evt.getRefBeforePut<reco::TrackExtraCollection>();
edm::Ref<reco::TrackExtraCollection>::key_type idx = 0;
edm::Ref<reco::TrackExtraCollection>::key_type hidx = 0;
edm::Ref<reco::TrackCollection>::key_type iTkRef = 0;
edm::Ref< std::vector<Trajectory> >::key_type iTjRef = 0;
std::map<unsigned int, unsigned int> tjTkMap;
for(AlgoProductCollection::iterator i=algoResults.begin(); i!=algoResults.end();i++){
Trajectory * theTraj = (*i).first;
if(myTrajectoryInEvent_) {
selTrajectories->push_back(*theTraj);
iTjRef++;
}
const TrajectoryFitter::RecHitContainer& transHits = theTraj->recHits();
reco::Track * theTrack = (*i).second.first;
PropagationDirection seedDir = (*i).second.second;
//LogDebug("TrackProducer") << "In KfTrackProducerBase::putInEvt - seedDir=" << seedDir;
reco::Track t = * theTrack;
selTracks->push_back( t );
iTkRef++;
// Store indices in local map (starts at 0)
if(trajectoryInEvent_) tjTkMap[iTjRef-1] = iTkRef-1;
//sets the outermost and innermost TSOSs
TrajectoryStateOnSurface outertsos;
TrajectoryStateOnSurface innertsos;
unsigned int innerId, outerId;
// --- NOTA BENE: the convention is to sort hits and measurements "along the momentum".
// This is consistent with innermost and outermost labels only for tracks from LHC collision
if (theTraj->direction() == alongMomentum) {
outertsos = theTraj->lastMeasurement().updatedState();
innertsos = theTraj->firstMeasurement().updatedState();
outerId = theTraj->lastMeasurement().recHit()->geographicalId().rawId();
innerId = theTraj->firstMeasurement().recHit()->geographicalId().rawId();
} else {
outertsos = theTraj->firstMeasurement().updatedState();
innertsos = theTraj->lastMeasurement().updatedState();
outerId = theTraj->firstMeasurement().recHit()->geographicalId().rawId();
innerId = theTraj->lastMeasurement().recHit()->geographicalId().rawId();
}
// ---
//build the TrackExtra
GlobalPoint v = outertsos.globalParameters().position();
GlobalVector p = outertsos.globalParameters().momentum();
math::XYZVector outmom( p.x(), p.y(), p.z() );
math::XYZPoint outpos( v.x(), v.y(), v.z() );
v = innertsos.globalParameters().position();
p = innertsos.globalParameters().momentum();
math::XYZVector inmom( p.x(), p.y(), p.z() );
math::XYZPoint inpos( v.x(), v.y(), v.z() );
reco::TrackExtraRef teref= reco::TrackExtraRef ( rTrackExtras, idx ++ );
reco::Track & track = selTracks->back();
track.setExtra( teref );
//======= I want to set the second hitPattern here =============
if (theSchool.isValid())
{
NavigationSetter setter( *theSchool );
setSecondHitPattern(theTraj,track,thePropagator,theMeasTk);
}
//==============================================================
selTrackExtras->push_back( reco::TrackExtra (outpos, outmom, true, inpos, inmom, true,
outertsos.curvilinearError(), outerId,
innertsos.curvilinearError(), innerId,
seedDir,theTraj->seedRef()));
reco::TrackExtra & tx = selTrackExtras->back();
// --- NOTA BENE: the convention is to sort hits and measurements "along the momentum".
// This is consistent with innermost and outermost labels only for tracks from LHC collisions
size_t k = 0;
if (theTraj->direction() == alongMomentum) {
for( TrajectoryFitter::RecHitContainer::const_iterator j = transHits.begin();
j != transHits.end(); j ++ ) {
if ((**j).hit()!=0){
TrackingRecHit * hit = (**j).hit()->clone();
track.setHitPattern( * hit, k++ );
selHits->push_back( hit );
tx.add( TrackingRecHitRef( rHits, hidx ++ ) );
}
}
}else{
for( TrajectoryFitter::RecHitContainer::const_iterator j = transHits.end()-1;
j != transHits.begin()-1; --j ) {
if ((**j).hit()!=0){
TrackingRecHit * hit = (**j).hit()->clone();
track.setHitPattern( * hit, k++ );
selHits->push_back( hit );
tx.add( TrackingRecHitRef( rHits, hidx ++ ) );
}
}
}
// ----
delete theTrack;
delete theTraj;
}
//LogTrace("TrackingRegressionTest") << "========== TrackProducer Info ===================";
//LogDebug("TrackProducerWithSCAssociation") << "number of finalTracks: " << selTracks->size() << std::endl;
//for (reco::TrackCollection::const_iterator it = selTracks->begin(); it != selTracks->end(); it++) {
//LogDebug("TrackProducerWithSCAssociation") << "track's n valid and invalid hit, chi2, pt : "
// << it->found() << " , "
// << it->lost() <<" , "
// << it->normalizedChi2() << " , "
// << it->pt() << std::endl;
// }
//LogTrace("TrackingRegressionTest") << "=================================================";
rTracks_ = evt.put( selTracks );
evt.put( selTrackExtras );
evt.put( selHits );
if(myTrajectoryInEvent_) {
edm::OrphanHandle<std::vector<Trajectory> > rTrajs = evt.put(selTrajectories);
// Now Create traj<->tracks association map
std::auto_ptr<TrajTrackAssociationCollection> trajTrackMap( new TrajTrackAssociationCollection() );
for ( std::map<unsigned int, unsigned int>::iterator i = tjTkMap.begin();
i != tjTkMap.end(); i++ ) {
edm::Ref<std::vector<Trajectory> > trajRef( rTrajs, (*i).first );
edm::Ref<reco::TrackCollection> tkRef( rTracks_, (*i).second );
trajTrackMap->insert( edm::Ref<std::vector<Trajectory> >( rTrajs, (*i).first ),
edm::Ref<reco::TrackCollection>( rTracks_, (*i).second ) );
}
evt.put( trajTrackMap );
}
}
std::string TrackProducerWithSCAssociation::conversionTrackCandidateProducer_ [private] |
Definition at line 33 of file TrackProducerWithSCAssociation.h.
Referenced by produce(), and TrackProducerWithSCAssociation().
std::string TrackProducerWithSCAssociation::myname_ [private] |
Definition at line 31 of file TrackProducerWithSCAssociation.h.
Referenced by TrackProducerWithSCAssociation().
bool TrackProducerWithSCAssociation::myTrajectoryInEvent_ [private] |
Definition at line 37 of file TrackProducerWithSCAssociation.h.
Referenced by putInEvt(), and TrackProducerWithSCAssociation().
Reimplemented from TrackProducerBase< reco::Track >.
Definition at line 36 of file TrackProducerWithSCAssociation.h.
Referenced by produce(), and putInEvt().
Definition at line 32 of file TrackProducerWithSCAssociation.h.
Referenced by getTransient(), and produce().
std::string TrackProducerWithSCAssociation::trackCSuperClusterAssociationCollection_ [private] |
Definition at line 34 of file TrackProducerWithSCAssociation.h.
Referenced by produce(), and TrackProducerWithSCAssociation().
std::string TrackProducerWithSCAssociation::trackSuperClusterAssociationCollection_ [private] |
Definition at line 35 of file TrackProducerWithSCAssociation.h.
Referenced by produce(), and TrackProducerWithSCAssociation().
Definition at line 38 of file TrackProducerWithSCAssociation.h.
Referenced by produce().
1.7.3