CMS 3D CMS Logo

SoftConversionProducer Class Reference

Id
SoftConversionProducer.h,v 1.6 2008/07/18 01:41:21 dwjang Exp
Date
2008/07/18 01:41:21
Revision
1.6
More...

#include <RecoEgamma/EgammaPhotonProducers/interface/SoftConversionProducer.h>

Inheritance diagram for SoftConversionProducer:

edm::EDProducer edm::ProducerBase edm::ProductRegistryHelper

List of all members.

Public Types

typedef std::vector< std::pair
< reco::TrackRef,
reco::CaloClusterPtr > > 		TrackClusterMap

Public Member Functions

virtual void beginJob (edm::EventSetup const &es)
virtual void endJob ()
bool NotAlreadyIn (const reco::Conversion &thisConv, const std::auto_ptr< reco::ConversionCollection > &outputColl) const
virtual void produce (edm::Event &evt, const edm::EventSetup &es)
 SoftConversionProducer (const edm::ParameterSet &ps)
bool trackQualityCut (const reco::TrackRef &trk)
 ~SoftConversionProducer ()

Private Attributes

edm::InputTag clusterBarrelCollection_
edm::InputTag clusterEndcapCollection_
double clustersMaxDeltaEta_
double clustersMaxDeltaPhi_
std::string clusterType_
edm::ParameterSet conf_
std::string conversionIOTrackProducer_
std::string conversionOITrackProducer_
std::string inOutTrackClusterAssociationCollection_
std::string outInTrackClusterAssociationCollection_
std::string softConversionCollection_
ConversionTrackEcalImpactPoint * theEcalImpactPositionFinder_
edm::ESHandle
< GeometricSearchTracker > 		theGeomSearchTracker_
edm::ESHandle< MagneticField > theMF_
ConversionTrackPairFinder * theTrackPairFinder_
ConversionVertexFinder * theVertexFinder_
double trackMaxChi2_
double trackMinHits_

Detailed Description

Id
SoftConversionProducer.h,v 1.6 2008/07/18 01:41:21 dwjang Exp
Date
2008/07/18 01:41:21
Revision
1.6

Author:
Dongwook Jang, Carnegie Mellon University Modified version of ConvertedPhotonProducer

Definition at line 27 of file SoftConversionProducer.h.

Member Typedef Documentation

typedef std::vector<std::pair<reco::TrackRef, reco::CaloClusterPtr> > SoftConversionProducer::TrackClusterMap

Definition at line 31 of file SoftConversionProducer.h.

Constructor & Destructor Documentation

SoftConversionProducer::SoftConversionProducer ( const edm::ParameterSet &  ps  ) 

Definition at line 36 of file SoftConversionProducer.cc.

References clusterBarrelCollection_, clusterEndcapCollection_, clustersMaxDeltaEta_, clustersMaxDeltaPhi_, clusterType_, conf_, conversionIOTrackProducer_, conversionOITrackProducer_, edm::ParameterSet::getParameter(), inOutTrackClusterAssociationCollection_, LogDebug, outInTrackClusterAssociationCollection_, softConversionCollection_, theEcalImpactPositionFinder_, theTrackPairFinder_, theVertexFinder_, trackMaxChi2_, and trackMinHits_.

00036                                                                             : conf_(config) {
00037 
00038   LogDebug("SoftConversionProducer") << " SoftConversionProducer CTOR " << "\n";
00039   
00040   conversionOITrackProducer_              = conf_.getParameter<std::string>("conversionOITrackProducer");
00041   conversionIOTrackProducer_              = conf_.getParameter<std::string>("conversionIOTrackProducer");
00042   outInTrackClusterAssociationCollection_ = conf_.getParameter<std::string>("outInTrackClusterAssociationCollection");
00043   inOutTrackClusterAssociationCollection_ = conf_.getParameter<std::string>("inOutTrackClusterAssociationCollection");
00044   clusterType_                            = conf_.getParameter<std::string>("clusterType");
00045   clusterBarrelCollection_                = conf_.getParameter<edm::InputTag>("clusterBarrelCollection");
00046   clusterEndcapCollection_                = conf_.getParameter<edm::InputTag>("clusterEndcapCollection");
00047   softConversionCollection_               = conf_.getParameter<std::string>("softConversionCollection");
00048   trackMaxChi2_                           = conf_.getParameter<double>("trackMaxChi2");
00049   trackMinHits_                           = conf_.getParameter<double>("trackMinHits");
00050   clustersMaxDeltaEta_                    = conf_.getParameter<double>("clustersMaxDeltaEta");
00051   clustersMaxDeltaPhi_                    = conf_.getParameter<double>("clustersMaxDeltaPhi");
00052   
00053   theTrackPairFinder_ = 0;
00054   theVertexFinder_ = 0;
00055   theEcalImpactPositionFinder_ = 0;
00056   
00057   // Register the product
00058   produces< reco::ConversionCollection >(softConversionCollection_);
00059 
00060 }

SoftConversionProducer::~SoftConversionProducer (  ) 

Definition at line 63 of file SoftConversionProducer.cc.

References theEcalImpactPositionFinder_, theTrackPairFinder_, and theVertexFinder_.

00063                                                 {
00064 
00065   if(theTrackPairFinder_) delete theTrackPairFinder_;
00066   if(theVertexFinder_) delete theVertexFinder_;
00067   if(theEcalImpactPositionFinder_) delete theEcalImpactPositionFinder_; 
00068 
00069 }

Member Function Documentation

void SoftConversionProducer::beginJob ( edm::EventSetup const &  es  )  [virtual]

Reimplemented from edm::EDProducer.

Definition at line 72 of file SoftConversionProducer.cc.

References edm::EventSetup::get(), theEcalImpactPositionFinder_, theMF_, theTrackPairFinder_, and theVertexFinder_.

00072                                                                            {
00073   
00074   //get magnetic field
00075   theEventSetup.get<IdealMagneticFieldRecord>().get(theMF_);  
00076   
00077   // instantiate the Track Pair Finder algorithm
00078   theTrackPairFinder_ = new ConversionTrackPairFinder ();
00079 
00080   // instantiate the Vertex Finder algorithm
00081   theVertexFinder_ = new ConversionVertexFinder ();
00082   
00083   // instantiate the algorithm for finding the position of the track extrapolation at the Ecal front face
00084   theEcalImpactPositionFinder_ = new   ConversionTrackEcalImpactPoint ( &(*theMF_) );
00085 
00086 }

void SoftConversionProducer::endJob ( void   )  [virtual]

Reimplemented from edm::EDProducer.

Definition at line 89 of file SoftConversionProducer.cc.

00089 {}

bool SoftConversionProducer::NotAlreadyIn ( const reco::Conversion &  thisConv,
const std::auto_ptr< reco::ConversionCollection > &  outputColl 
) const

Definition at line 244 of file SoftConversionProducer.cc.

References conv, it, and reco::Conversion::tracks().

Referenced by produce().

00245                                                                                                          {
00246 
00247   if(outputColl->size() == 0) return true;
00248 
00249   reco::ConversionCollection::const_iterator it = outputColl->begin();
00250   reco::ConversionCollection::const_iterator it_end = outputColl->end();
00251   for( ; it != it_end; it++){
00252     const reco::Conversion& conv = *it;
00253     if((thisConv.tracks()[0] == conv.tracks()[0] && thisConv.tracks()[1] == conv.tracks()[1]) ||
00254        (thisConv.tracks()[0] == conv.tracks()[1] && thisConv.tracks()[1] == conv.tracks()[0]))
00255       return false;
00256   }// for
00257   
00258   return true;
00259 }

void SoftConversionProducer::produce ( edm::Event &  evt,
const edm::EventSetup &  es 
) [virtual]

Implements edm::EDProducer.

Definition at line 92 of file SoftConversionProducer.cc.

References funct::abs(), clusterBarrelCollection_, clusterEndcapCollection_, clustersMaxDeltaEta_, clustersMaxDeltaPhi_, clusterType_, conversionIOTrackProducer_, conversionOITrackProducer_, DeltaPhi(), dPhi(), e, ConversionTrackEcalImpactPoint::find(), edm::EventSetup::get(), edm::Event::getByLabel(), TrajectoryStateOnSurface::globalParameters(), reco::TransientTrack::innermostMeasurementState(), inOutTrackClusterAssociationCollection_, int, reco::Vertex::isValid(), muonGeometry::mag(), GlobalTrajectoryParameters::momentum(), NotAlreadyIn(), outInTrackClusterAssociationCollection_, GlobalTrajectoryParameters::position(), edm::PtrVector< T >::push_back(), edm::Event::put(), ConversionVertexFinder::run(), softConversionCollection_, theEcalImpactPositionFinder_, theVertexFinder_, and trackQualityCut().

00092                                                                                            {
00093   
00094   edm::Handle<reco::TrackCollection> outInTrkHandle;
00095   theEvent.getByLabel(conversionOITrackProducer_,  outInTrkHandle);
00096    
00097   edm::Handle<reco::TrackCaloClusterPtrAssociation> outInTrkClusterAssocHandle;
00098   theEvent.getByLabel( conversionOITrackProducer_, outInTrackClusterAssociationCollection_, outInTrkClusterAssocHandle);
00099 
00100   edm::Handle<reco::TrackCollection> inOutTrkHandle;
00101   theEvent.getByLabel(conversionIOTrackProducer_, inOutTrkHandle);
00102   
00103   edm::Handle<reco::TrackCaloClusterPtrAssociation> inOutTrkClusterAssocHandle;
00104   theEvent.getByLabel( conversionIOTrackProducer_, inOutTrackClusterAssociationCollection_, inOutTrkClusterAssocHandle);
00105 
00106   edm::Handle<edm::View<reco::CaloCluster> > clusterBarrelHandle;
00107   theEvent.getByLabel(clusterBarrelCollection_, clusterBarrelHandle);
00108 
00109   edm::Handle<edm::View<reco::CaloCluster> > clusterEndcapHandle;
00110   if(clusterType_ == "BasicCluster"){
00111     //theEvent.getByLabel(clusterEndcapProducer_, clusterEndcapCollection_, clusterEndcapHandle);
00112     theEvent.getByLabel(clusterEndcapCollection_, clusterEndcapHandle);
00113   }
00114 
00115   // create temporary map to loop over tracks conveniently
00116   TrackClusterMap trackClusterMap;
00117 
00118   int nTracksOI = (int) outInTrkHandle->size();
00119   //  std::cout << "  nTracksOI " <<  nTracksOI << std::endl;
00120   for(int itrk=0; itrk<nTracksOI; itrk++){
00121     reco::TrackRef tRef(outInTrkHandle,itrk);
00122     if(!trackQualityCut(tRef)) continue;
00123     reco::CaloClusterPtr cRef = (*outInTrkClusterAssocHandle)[tRef];
00124     trackClusterMap.push_back(make_pair(tRef,cRef));
00125   }
00126 
00127   int nTracksIO = (int) inOutTrkHandle->size();
00128   // std::cout << "  nTracksIO " <<  nTracksIO << std::endl;
00129   for(int itrk=0; itrk<nTracksIO; itrk++){
00130     reco::TrackRef tRef(inOutTrkHandle,itrk);
00131     if(!trackQualityCut(tRef)) continue;
00132     reco::CaloClusterPtr cRef = (*inOutTrkClusterAssocHandle)[tRef];
00133     trackClusterMap.push_back(make_pair(tRef,cRef));
00134   }
00135 
00136   // Transform Track into TransientTrack (needed by the Vertex fitter)
00137   edm::ESHandle<TransientTrackBuilder> theTransientTrackBuilder;
00138   theEventSetup.get<TransientTrackRecord>().get("TransientTrackBuilder",theTransientTrackBuilder);
00139 
00140   // the output collection to be produced from this producer
00141   std::auto_ptr< reco::ConversionCollection > outputColl(new reco::ConversionCollection);
00142 
00143   // prepare iterator
00144   TrackClusterMap::iterator iter1    = trackClusterMap.begin();
00145   TrackClusterMap::iterator iter2    = trackClusterMap.begin();
00146   TrackClusterMap::iterator iter_end = trackClusterMap.end();
00147 
00148 
00149 
00150   // double-loop to make pairs
00151   for( ; iter1 != iter_end; iter1++) {
00152     //    std::cout << " back to start of loop 1 " << std::endl;
00153     const reco::TrackRef trk1 = iter1->first;
00154 
00155     for(iter2 = iter1+1; iter2 != iter_end; iter2++) {
00156       //      std::cout << " back to start of loop 2 " << std::endl;
00157       const reco::TrackRef trk2 = iter2->first;
00158       if(trk1 == trk2) continue;
00159 
00160 
00161       const reco::CaloClusterPtr cls1 = iter1->second;
00162       reco::TransientTrack tsk1 = theTransientTrackBuilder->build(*trk1);
00163 
00164       const reco::CaloClusterPtr cls2 = iter2->second;
00165       reco::TransientTrack tsk2 = theTransientTrackBuilder->build(*trk2);
00166 
00167       //      std::cout << " after transient " << std::endl;
00168       // std::cout << " eta1 " << cls1->position().Eta() << " eta2 " << cls2->position().Eta() << std::endl;
00169 
00170       if ( ( tsk1.innermostMeasurementState().globalParameters().position() - tsk2.innermostMeasurementState().globalParameters().position() ).mag() < 1e-7 &&
00171            ( tsk1.innermostMeasurementState().globalParameters().momentum() - tsk2.innermostMeasurementState().globalParameters().momentum() ).mag() < 1e-7 ) continue;
00172 
00173 
00174       double dEta = std::abs(cls1->position().Eta() - cls2->position().Eta());
00175       if(dEta > clustersMaxDeltaEta_) continue;
00176       double dPhi = std::abs(ROOT::Math::VectorUtil::DeltaPhi(cls1->position(),cls2->position()));
00177       if(dPhi > clustersMaxDeltaPhi_) continue;
00178 
00179       std::vector<reco::TransientTrack> toBeFitted;
00180       toBeFitted.push_back(tsk1);
00181       toBeFitted.push_back(tsk2);
00182 
00183       //      std::cout << " Before vertex " << std::endl;
00184       reco::Vertex theConversionVertex = (reco::Vertex) theVertexFinder_->run(toBeFitted);
00185       //std::cout << " After vertex " << std::endl;
00186 
00187       if(theConversionVertex.isValid()){
00188         //      std::cout << " valid vertex " << std::endl;
00189         reco::CaloClusterPtrVector scRefs;
00190         scRefs.push_back(cls1);
00191         scRefs.push_back(cls2);
00192         //std::cout << " after push back scref " << std::endl; 
00193         std::vector<reco::CaloClusterPtr> clusterRefs;
00194         clusterRefs.push_back(cls1);
00195         clusterRefs.push_back(cls2);
00196         //std::cout << " after push back slusterref " << std::endl; 
00197         std::vector<reco::TrackRef> trkRefs;
00198         trkRefs.push_back(trk1);
00199         trkRefs.push_back(trk2);
00200 
00201 
00202         //      std::cout << " Before impact finder " << std::endl;
00203         std::vector<math::XYZPoint> trkPositionAtEcal = theEcalImpactPositionFinder_->find(toBeFitted,clusterBarrelHandle);
00204         //      std::cout << " After impact finder " << std::endl;
00205         if((clusterType_ == "BasicCluster") && (std::abs(cls2->position().Eta()) > 1.5)){
00206           trkPositionAtEcal.clear();
00207           trkPositionAtEcal = theEcalImpactPositionFinder_->find(toBeFitted,clusterEndcapHandle);
00208         }
00209 
00210         reco::Conversion newCandidate(scRefs, trkRefs, trkPositionAtEcal, theConversionVertex, clusterRefs);
00211 
00212         // Check this candidate is already in the collection.
00213         // This is checking that two tracks in a conversion candidate are identicial.
00214 
00215         if(NotAlreadyIn(newCandidate,outputColl)) outputColl->push_back(newCandidate);
00216 
00217         //      printf("=====> run(%d), event(%d) <=====\n",theEvent.id().run(),theEvent.id().event());
00218         //      printf("Found a softConverion with vtxR(%f), vtxEta(%f), pt(%f), pt1(%f), pt2(%f)\n",
00219         //      newCandidate.conversionVertex().position().rho(),newCandidate.conversionVertex().position().eta(),
00220         //      newCandidate.pairMomentum().perp(),trk1->momentum().rho(),trk2->momentum().rho());
00221 
00222         clusterRefs.clear();
00223         trkRefs.clear();
00224         trkPositionAtEcal.clear();
00225       }// if(theConversionVertex.isValid()
00226 
00227       toBeFitted.clear();
00228 
00229     }// end of iter2
00230   }// end of iter1
00231 
00232   //  std::cout << " Putting stuff in the event " << std::endl;
00233   // put the collection into the event
00234   theEvent.put( outputColl, softConversionCollection_);
00235   
00236 }

bool SoftConversionProducer::trackQualityCut ( const reco::TrackRef &  trk  ) 

Definition at line 239 of file SoftConversionProducer.cc.

References trackMaxChi2_, and trackMinHits_.

Referenced by produce().

00239                                                                    {
00240   return (trk->numberOfValidHits() >= trackMinHits_ && trk->normalizedChi2() < trackMaxChi2_);
00241 }

Member Data Documentation

edm::InputTag SoftConversionProducer::clusterBarrelCollection_ [private]

Definition at line 53 of file SoftConversionProducer.h.

Referenced by produce(), and SoftConversionProducer().

edm::InputTag SoftConversionProducer::clusterEndcapCollection_ [private]

Definition at line 54 of file SoftConversionProducer.h.

Referenced by produce(), and SoftConversionProducer().

double SoftConversionProducer::clustersMaxDeltaEta_ [private]

Definition at line 59 of file SoftConversionProducer.h.

Referenced by produce(), and SoftConversionProducer().

double SoftConversionProducer::clustersMaxDeltaPhi_ [private]

Definition at line 60 of file SoftConversionProducer.h.

Referenced by produce(), and SoftConversionProducer().

std::string SoftConversionProducer::clusterType_ [private]

Definition at line 52 of file SoftConversionProducer.h.

Referenced by produce(), and SoftConversionProducer().

edm::ParameterSet SoftConversionProducer::conf_ [private]

Definition at line 62 of file SoftConversionProducer.h.

Referenced by SoftConversionProducer().

std::string SoftConversionProducer::conversionIOTrackProducer_ [private]

Definition at line 47 of file SoftConversionProducer.h.

Referenced by produce(), and SoftConversionProducer().

std::string SoftConversionProducer::conversionOITrackProducer_ [private]

Definition at line 46 of file SoftConversionProducer.h.

Referenced by produce(), and SoftConversionProducer().

std::string SoftConversionProducer::inOutTrackClusterAssociationCollection_ [private]

Definition at line 50 of file SoftConversionProducer.h.

Referenced by produce(), and SoftConversionProducer().

std::string SoftConversionProducer::outInTrackClusterAssociationCollection_ [private]

Definition at line 49 of file SoftConversionProducer.h.

Referenced by produce(), and SoftConversionProducer().

std::string SoftConversionProducer::softConversionCollection_ [private]

Definition at line 56 of file SoftConversionProducer.h.

Referenced by produce(), and SoftConversionProducer().

ConversionTrackEcalImpactPoint* SoftConversionProducer::theEcalImpactPositionFinder_ [private]

Definition at line 68 of file SoftConversionProducer.h.

Referenced by beginJob(), produce(), SoftConversionProducer(), and ~SoftConversionProducer().

edm::ESHandle<GeometricSearchTracker> SoftConversionProducer::theGeomSearchTracker_ [private]

Definition at line 64 of file SoftConversionProducer.h.

edm::ESHandle<MagneticField> SoftConversionProducer::theMF_ [private]

Definition at line 63 of file SoftConversionProducer.h.

Referenced by beginJob().

ConversionTrackPairFinder* SoftConversionProducer::theTrackPairFinder_ [private]

Definition at line 66 of file SoftConversionProducer.h.

Referenced by beginJob(), SoftConversionProducer(), and ~SoftConversionProducer().

ConversionVertexFinder* SoftConversionProducer::theVertexFinder_ [private]

Definition at line 67 of file SoftConversionProducer.h.

Referenced by beginJob(), produce(), SoftConversionProducer(), and ~SoftConversionProducer().

double SoftConversionProducer::trackMaxChi2_ [private]

Definition at line 57 of file SoftConversionProducer.h.

Referenced by SoftConversionProducer(), and trackQualityCut().

double SoftConversionProducer::trackMinHits_ [private]

Definition at line 58 of file SoftConversionProducer.h.

Referenced by SoftConversionProducer(), and trackQualityCut().

The documentation for this class was generated from the following files:
Generated on Tue Jun 9 18:32:39 2009 for CMSSW by  doxygen 1.5.4