#include <RecoVertex/NuclearInteractionProducer/interface/NuclearInteractionEDProducer.h>

Inheritance diagram for NuclearInteractionEDProducer:

Public Types
typedef edm::RefVector < TrajectorySeedCollection >	TrajectorySeedRefVector

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
	NuclearInteractionEDProducer (const edm::ParameterSet &)

	~NuclearInteractionEDProducer ()

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
virtual void	beginJob ()

virtual void	endJob ()

void	findAdditionalSecondaryTracks (reco::NuclearInteraction &nucl, const edm::Handle< reco::TrackCollection > &additionalSecTracks)

bool	isInside (const reco::TrackRef &track, const TrajectorySeedRefVector &seeds)

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

Private Attributes
std::string	additionalSecondaryProducer_

edm::ParameterSet	conf_

std::auto_ptr< NuclearLikelihood >	likelihoodCalculator

edm::ESWatcher < IdealMagneticFieldRecord >	magFieldWatcher_

std::string	primaryProducer_

std::string	secondaryProducer_

std::string	seedsProducer_

edm::ESWatcher < TransientTrackRecord >	transientTrackWatcher_

std::auto_ptr < NuclearVertexBuilder >	vertexBuilder

Additional Inherited Members
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::EDProducer
CurrentProcessingContext const *	currentContext () const

Protected Member Functions inherited from edm::ProducerBase
template<class TProducer , class TMethod >
void	callWhenNewProductsRegistered (TProducer *iProd, TMethod iMethod)

Detailed Description

Description: Associate nuclear seeds to primary tracks and associate secondary tracks to primary tracks

Implementation: <Notes on="" implementation>="">

Definition at line 49 of file NuclearInteractionEDProducer.h.

Member Typedef Documentation

typedef edm::RefVector<TrajectorySeedCollection> NuclearInteractionEDProducer::TrajectorySeedRefVector

Definition at line 52 of file NuclearInteractionEDProducer.h.

Constructor & Destructor Documentation

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

explicit

Definition at line 15 of file NuclearInteractionEDProducer.cc.

                                                                                          : 
 conf_(iConfig), 
 primaryProducer_(iConfig.getParameter<std::string>("primaryProducer")),
 seedsProducer_(iConfig.getParameter<std::string>("seedsProducer")),
 secondaryProducer_(iConfig.getParameter<std::string>("secondaryProducer")),
 additionalSecondaryProducer_(iConfig.getParameter<std::string>("additionalSecondaryProducer"))
 {
   produces<reco::NuclearInteractionCollection>();
 }

NuclearInteractionEDProducer::~NuclearInteractionEDProducer ( )

Definition at line 25 of file NuclearInteractionEDProducer.cc.

26 {

27 }

Member Function Documentation

void NuclearInteractionEDProducer::beginJob ( void )

privatevirtual

Reimplemented from edm::EDProducer.

Definition at line 129 of file NuclearInteractionEDProducer.cc.

130 {

131 }

void NuclearInteractionEDProducer::endJob ( void )

privatevirtual

Reimplemented from edm::EDProducer.

Definition at line 133 of file NuclearInteractionEDProducer.cc.

133 {}

void NuclearInteractionEDProducer::findAdditionalSecondaryTracks	(	reco::NuclearInteraction &	nucl,
		const edm::Handle< reco::TrackCollection > &	additionalSecTracks
	)

private

Definition at line 142 of file NuclearInteractionEDProducer.cc.

References i, likelihoodCalculator, LogDebug, reco::NuclearInteraction::seeds(), and vertexBuilder.

Referenced by produce().

                                                                                                  {
       
       LogDebug("NuclearInteractionMaker") << "Check if one of the " << additionalSecTracks->size() 
                                           << " additional secondary track is compatible";
       reco::TrackRefVector allSecondary;
       for(unsigned int i=0; i< additionalSecTracks->size(); i++) {
                  reco::TrackRef sec(additionalSecTracks, i);
                  if( vertexBuilder->isCompatible( sec ) ) {
                       // check if sec is already a secondary track (with id = tkId) 
                       // else add this new track
                       vertexBuilder->addSecondaryTrack( sec );
                  }
       }
       likelihoodCalculator->calculate( vertexBuilder->getVertex() );
 
       nucl = reco::NuclearInteraction(nucl.seeds(), vertexBuilder->getVertex(), likelihoodCalculator->result() );
 }

bool NuclearInteractionEDProducer::isInside	(	const reco::TrackRef &	track,
		const TrajectorySeedRefVector &	seeds
	)

private

Definition at line 136 of file NuclearInteractionEDProducer.cc.

References i, combine::key, edm::Ref< C, T, F >::key(), and edm::RefVector< C, T, F >::size().

Referenced by produce().

                                                                                                             {
     unsigned int seedKey = track->seedRef().key();
     for (unsigned int i=0; i< seeds.size(); i++) { if( seeds[i].key() == seedKey ) return true; }
     return false;
 }

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

privatevirtual

Get magnetic field

Get the primary tracks

Get the primary trajectories (produced by the Refitter)

Get the AssociationMap between primary tracks and trajectories

Get the AssociationMap between seeds and primary trajectories

Get the secondary tracks

Definition of the output

Loop on all primary trajectories

Get the primary track from the trajectory
Get the seeds from the trajectory
Get the secondary tracks
Get the vertex and the likelihood
Build the nuclear interaction
Search for additional secondary tracks in an other track collection and recompute the nuclear interaction

Implements edm::EDProducer.

Definition at line 35 of file NuclearInteractionEDProducer.cc.

References additionalSecondaryProducer_, edm::ESWatcher< T >::check(), conf_, edm::AssociationMap< Tag >::end(), edm::AssociationMap< Tag >::find(), findAdditionalSecondaryTracks(), edm::EventSetup::get(), edm::Event::getByLabel(), i, isInside(), edm::HandleBase::isValid(), gen::k, likelihoodCalculator, LogDebug, magFieldWatcher_, primaryProducer_, reco::print(), edm::Handle< T >::product(), edm::ESHandle< class >::product(), edm::Event::put(), secondaryProducer_, seedsProducer_, transientTrackWatcher_, edm::helpers::KeyVal< K, V >::val, and vertexBuilder.

 {
   if ( magFieldWatcher_.check(iSetup) || transientTrackWatcher_.check(iSetup) ) {
     edm::ESHandle<MagneticField> magField;
     iSetup.get<IdealMagneticFieldRecord>().get(magField);
 
     edm::ESHandle<TransientTrackBuilder> builder;
     iSetup.get<TransientTrackRecord>().get("TransientTrackBuilder",builder);
 
     vertexBuilder = std::auto_ptr< NuclearVertexBuilder >(new NuclearVertexBuilder( magField.product(), builder.product(), conf_) );
     likelihoodCalculator = std::auto_ptr< NuclearLikelihood >(new NuclearLikelihood);
   }
 
    edm::Handle<reco::TrackCollection>  primaryTrackCollection;
    iEvent.getByLabel( primaryProducer_, primaryTrackCollection );
 
    edm::Handle< TrajectoryCollection > primaryTrajectoryCollection;
    iEvent.getByLabel( primaryProducer_, primaryTrajectoryCollection );
 
    edm::Handle< TrajTrackAssociationCollection > refMapH;
    iEvent.getByLabel( primaryProducer_, refMapH );
    const TrajTrackAssociationCollection& refMap = *(refMapH.product());
 
    edm::Handle<TrajectoryToSeedsMap>  nuclMapH;
    iEvent.getByLabel(seedsProducer_, nuclMapH);
    const TrajectoryToSeedsMap& nuclMap = *(nuclMapH.product());
 
    edm::Handle<reco::TrackCollection>  secondaryTrackCollection;
    iEvent.getByLabel( secondaryProducer_, secondaryTrackCollection );
 
    // Get eventual additional secondary tracks
    edm::Handle<reco::TrackCollection>  additionalSecTracks;
    iEvent.getByLabel( additionalSecondaryProducer_, additionalSecTracks); 
 
    std::auto_ptr<reco::NuclearInteractionCollection> theNuclearInteractions(new reco::NuclearInteractionCollection);
 
    typedef edm::Ref<TrajectoryCollection> TrajectoryRef;
 
    for(unsigned int i = 0; i < primaryTrajectoryCollection->size() ; i++) {
 
          TrajectoryRef  trajRef( primaryTrajectoryCollection, i );
 
          TrajTrackAssociationCollection::const_iterator itPrimTrack = refMap.find( trajRef );
          if( itPrimTrack == refMap.end() || (itPrimTrack->val).isNull() ) continue;
          const reco::TrackRef& primary_track = itPrimTrack->val;
 
          TrajectoryToSeedsMap::const_iterator itSeeds = nuclMap.find( trajRef );
          if( itSeeds == nuclMap.end() || (itSeeds->val).isNull()) continue; 
          const TrajectorySeedRefVector& seeds = itSeeds->val;
 
          std::vector<reco::TrackRef> secondary_tracks;
          for( unsigned int k=0; k < secondaryTrackCollection->size(); k++) {
                      reco::TrackRef currentTrk(secondaryTrackCollection, k);
                      if( isInside( currentTrk, seeds ) ) secondary_tracks.push_back(currentTrk);
          }
               
          vertexBuilder->build(primary_track, secondary_tracks);
          likelihoodCalculator->calculate( vertexBuilder->getVertex() );
 
          reco::NuclearInteraction nuclInter(seeds, vertexBuilder->getVertex(), likelihoodCalculator->result() );
 
          if( additionalSecTracks.isValid() )
                  findAdditionalSecondaryTracks(nuclInter, additionalSecTracks); 
 
          theNuclearInteractions->push_back( nuclInter );
 
          std::ostringstream  str;
          print(str, nuclInter, vertexBuilder);
          edm::LogInfo("NuclearInteractionMaker") << str.str();
 
    }
 
 
    LogDebug("NuclearInteractionMaker") << "End of NuclearInteractionMaker - Number of nuclear interactions found :" << theNuclearInteractions->size();
    iEvent.put(theNuclearInteractions);
 }