#include <PFConversionProducer.h>

Inheritance diagram for PFConversionProducer:

Public Member Functions
	PFConversionProducer (const edm::ParameterSet &)
	Constructor.
	~PFConversionProducer ()
	Destructor.
Private Member Functions
virtual void	beginRun (edm::Run &, const edm::EventSetup &)
virtual void	endRun ()
virtual void	produce (edm::Event &, const edm::EventSetup &)
	Produce the PFRecTrack collection.
Private Attributes
edm::InputTag	pfConversionContainer_
edm::InputTag	pfTrackContainer_
PFTrackTransformer *	pfTransformer_
	PFTrackTransformer.
edm::InputTag	vtx_h

Detailed Description

Definition at line 11 of file PFConversionProducer.h.

Constructor & Destructor Documentation

PFConversionProducer::PFConversionProducer ( const edm::ParameterSet & iConfig ) [explicit]

Constructor.

Definition at line 25 of file PFConversionProducer.cc.

References edm::ParameterSet::getParameter(), pfConversionContainer_, and vtx_h.

                                                                     :
  pfTransformer_(0)
{
  produces<reco::PFRecTrackCollection>();
  produces<reco::PFConversionCollection>();

  pfConversionContainer_ = 
    iConfig.getParameter< InputTag >("conversionCollection");
  vtx_h=iConfig.getParameter<edm::InputTag>("PrimaryVertexLabel");
}

PFConversionProducer::~PFConversionProducer ( )

Destructor.

Definition at line 36 of file PFConversionProducer.cc.

References pfTransformer_.

{
  delete pfTransformer_;
}

Member Function Documentation

void PFConversionProducer::beginRun	(	edm::Run &	run,
		const edm::EventSetup &	iSetup
	)		`[private, virtual]`

Reimplemented from edm::EDProducer.

Definition at line 188 of file PFConversionProducer.cc.

References edm::EventSetup::get(), PFTrackTransformer::OnlyProp(), and pfTransformer_.

{
  ESHandle<MagneticField> magneticField;
  iSetup.get<IdealMagneticFieldRecord>().get(magneticField);
  pfTransformer_= new PFTrackTransformer(math::XYZVector(magneticField->inTesla(GlobalPoint(0,0,0))));
  pfTransformer_->OnlyProp();
}

void PFConversionProducer::endRun ( void ) [private, virtual]

Definition at line 199 of file PFConversionProducer.cc.

References pfTransformer_.

                             {
  delete pfTransformer_;
}

void PFConversionProducer::produce	(	edm::Event &	iEvent,
		const edm::EventSetup &	iSetup
	)		`[private, virtual]`

Produce the PFRecTrack collection.

Implements edm::EDProducer.

Definition at line 42 of file PFConversionProducer.cc.

References PFTrackTransformer::addPoints(), reco::Conversion::arbitratedMergedEcalGeneral, TransientTrackBuilder::build(), ChiSquaredProbability(), alignCSCRings::e, reco::PFTrajectoryPoint::ECALEntrance, cropTnPTrees::frac, edm::EventSetup::get(), edm::Event::getByLabel(), edm::Event::getRefBeforePut(), reco::Conversion::highPurity, TrackingRecHit::isValid(), reco::PFTrajectoryPoint::isValid(), edm::HandleBase::isValid(), edm::Ref< C, T, F >::key(), reco::PFRecTrack::KF, AlCaHLTBitMon_ParallelJobs::p, pfConversionContainer_, pfTrackContainer_, pfTransformer_, edm::ESHandle< T >::product(), edm::Event::put(), edm::second(), TrackingRecHit::sharesInput(), IPTools::signedTransverseImpactParameter(), TrackingRecHit::some, TrackValidation_HighPurity_cff::valid, and vtx_h.

{
  
  //create the empty collections 
  auto_ptr< reco::PFConversionCollection > 
    pfConversionColl (new reco::PFConversionCollection);
  auto_ptr< reco::PFRecTrackCollection > 
    pfRecTrackColl (new reco::PFRecTrackCollection);
  
  edm::ESHandle<TransientTrackBuilder> builder;
  iSetup.get<TransientTrackRecord>().get("TransientTrackBuilder", builder);
  TransientTrackBuilder thebuilder = *(builder.product());
  reco::PFRecTrackRefProd pfTrackRefProd = iEvent.getRefBeforePut<reco::PFRecTrackCollection>();
  Handle<reco::ConversionCollection> convCollH;
  iEvent.getByLabel(pfConversionContainer_, convCollH);
  
  const reco::ConversionCollection& convColl = *(convCollH.product());
  
  Handle<reco::TrackCollection> trackColl;
  iEvent.getByLabel(pfTrackContainer_, trackColl);
  Handle<reco::VertexCollection> vertex;
  iEvent.getByLabel(vtx_h, vertex);
  //Find PV for IP calculation, if there is no PV in collection than use dummy 
  reco::Vertex dummy;
  const reco::Vertex* pv=&dummy;    
  if (vertex.isValid()) 
    {
      pv = &*vertex->begin();
    } 
  else 
    { // create a dummy PV
      reco::Vertex::Error e;
      e(0, 0) = 0.0015 * 0.0015;
      e(1, 1) = 0.0015 * 0.0015;
      e(2, 2) = 15. * 15.;
      reco::Vertex::Point p(0, 0, 0);
      dummy = reco::Vertex(p, e, 0, 0, 0);   
    } 
  
  int idx = 0; //index of track in PFRecTrack collection 
  multimap<unsigned int, unsigned int> trackmap; //Map of Collections and tracks  
  std::vector<unsigned int> conv_coll(0);
   
  // CLEAN CONVERSION COLLECTION FOR DUPLICATES     
  for( unsigned int icoll1=0; icoll1 < convColl.size(); icoll1++) 
    { 
      if (( !convColl[icoll1].quality(reco::Conversion::arbitratedMergedEcalGeneral)) || (!convColl[icoll1].quality(reco::Conversion::highPurity))) continue;
      
      bool greater_prob=false;
      std::vector<edm::RefToBase<reco::Track> > tracksRefColl1 = convColl[icoll1].tracks();      
      for(unsigned it1 = 0; it1 < tracksRefColl1.size(); it1++)
        {
          reco::TrackRef trackRef1 = (tracksRefColl1[it1]).castTo<reco::TrackRef>();
         
          for( unsigned int icoll2=0; icoll2 < convColl.size(); icoll2++) 
            {
              if(icoll1==icoll2)continue;
              if (( !convColl[icoll2].quality(reco::Conversion::arbitratedMergedEcalGeneral)) || (!convColl[icoll2].quality(reco::Conversion::highPurity))) continue;
              std::vector<edm::RefToBase<reco::Track> > tracksRefColl2 = convColl[icoll2].tracks();     
              for(unsigned it2 = 0; it2 < tracksRefColl2.size(); it2++)
                {
                  reco::TrackRef trackRef2 = (tracksRefColl2[it2]).castTo<reco::TrackRef>();
                  double like1=-999;
                  double like2=-999;
                  //number of shared hits
                  int shared=0;
                  for(trackingRecHit_iterator iHit1=trackRef1->recHitsBegin(); iHit1!=trackRef1->recHitsEnd(); iHit1++) 
                    {
                      const TrackingRecHit *h_1=iHit1->get();
                      if(h_1->isValid()){                 
                        for(trackingRecHit_iterator iHit2=trackRef2->recHitsBegin(); iHit2!=trackRef2->recHitsEnd(); iHit2++)
                          {
                            const TrackingRecHit *h_2=iHit2->get();
                            if(h_2->isValid() && h_1->sharesInput(h_2, TrackingRecHit::some))shared++;//count number of shared hits
                          }
                      }
                    }             
                  float frac=0;
                  //number of valid hits in tracks that are duplicates
                  float size1=trackRef1->found();
                  float size2=trackRef2->found();
                  //divide number of shared hits by the total number of hits for the track with less hits
                  if(size1>size2)frac=(double)shared/size2;
                  else frac=(double)shared/size1;
                  if(frac>0.9)
                    {
                      like1=ChiSquaredProbability(convColl[icoll1].conversionVertex().chi2(), convColl[icoll1].conversionVertex().ndof());
                      like2=ChiSquaredProbability(convColl[icoll2].conversionVertex().chi2(), convColl[icoll2].conversionVertex().ndof());
                    }
                  if(like2>like1)
                    {greater_prob=true;  break;}
                }//end loop over tracks in collection 2

              if(greater_prob)break; //if a duplicate track is found in a collection with greater Chi^2 probability for Vertex fit then break out of comparison loop
            }//end loop over collection 2 checking
          if(greater_prob)break;//if a duplicate track is found in a collection with greater Chi^2 probability for Vertex fit then one does not need to check the other track the collection will not be stored
        } //end loop over tracks in collection 1
      if(!greater_prob)conv_coll.push_back(icoll1);
    }//end loop over collection 1
  
  //Finally fill empty collections
  for(unsigned iColl=0; iColl<conv_coll.size(); iColl++)
    {
      unsigned int collindex=conv_coll[iColl];
      //std::cout<<"Filling this collection"<<collindex<<endl;
      std::vector<reco::PFRecTrackRef> pfRecTkcoll;     
      
      std::vector<edm::RefToBase<reco::Track> > tracksRefColl = convColl[collindex].tracks();     
      // convert the secondary tracks
      for(unsigned it = 0; it < tracksRefColl.size(); it++)
        {
          reco::TrackRef trackRef = (tracksRefColl[it]).castTo<reco::TrackRef>();      
          reco::PFRecTrack pfRecTrack( trackRef->charge(), 
                                       reco::PFRecTrack::KF, 
                                       trackRef.key(), 
                                       trackRef );             
          //std::cout<<"Track Pt "<<trackRef->pt()<<std::endl;
          Trajectory FakeTraj;
          bool valid = pfTransformer_->addPoints( pfRecTrack, *trackRef, FakeTraj);
          if(valid) 
            {
              double stip=-999;
              const reco::PFTrajectoryPoint& atECAL=pfRecTrack.extrapolatedPoint(reco::PFTrajectoryPoint::ECALEntrance);
              //if extrapolation to ECAL is valid then calculate STIP
              if(atECAL.isValid())
                {
                  GlobalVector direction(pfRecTrack.extrapolatedPoint(reco::PFTrajectoryPoint::ECALEntrance).position().x(),
                                         pfRecTrack.extrapolatedPoint(reco::PFTrajectoryPoint::ECALEntrance).position().y(), 
                                         pfRecTrack.extrapolatedPoint(reco::PFTrajectoryPoint::ECALEntrance).position().z());
                  stip = IPTools::signedTransverseImpactParameter(thebuilder.build(*trackRef), direction, *pv).second.significance();
                }
              pfRecTrack.setSTIP(stip);     
              pfRecTkcoll.push_back(reco::PFRecTrackRef( pfTrackRefProd, idx++));         
              pfRecTrackColl->push_back(pfRecTrack);        
            }
        }//end loop over tracks
      //store reference to the Conversion collection
      reco::ConversionRef niRef(convCollH, collindex);
      pfConversionColl->push_back( reco::PFConversion( niRef, pfRecTkcoll ));
    }//end loop over collections
  iEvent.put(pfRecTrackColl);
  iEvent.put(pfConversionColl);    
}