#include <ElectronRecalibSuperClusterAssociator.h>

Inheritance diagram for ElectronRecalibSuperClusterAssociator:

Public Member Functions
	ElectronRecalibSuperClusterAssociator (const edm::ParameterSet &conf)

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

virtual	~ElectronRecalibSuperClusterAssociator ()

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 Attributes
std::string	electronCollection_

std::string	electronProducer_

std::string	scCollection_

std::string	scIslandCollection_

std::string	scIslandProducer_

std::string	scProducer_

Additional Inherited Members
Public Types inherited from edm::EDProducer
typedef EDProducer	ModuleType

typedef WorkerT< EDProducer >	WorkerType

Public Types inherited from edm::ProducerBase
typedef ProductRegistryHelper::TypeLabelList	TypeLabelList

Static Public Member Functions inherited from edm::EDProducer
static const std::string &	baseType ()

static void	fillDescriptions (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

Definition at line 24 of file ElectronRecalibSuperClusterAssociator.h.

Constructor & Destructor Documentation

ElectronRecalibSuperClusterAssociator::ElectronRecalibSuperClusterAssociator ( const edm::ParameterSet & conf )

explicit

Definition at line 24 of file ElectronRecalibSuperClusterAssociator.cc.

References gather_cfg::cout, and edm::ParameterSet::getParameter().

 {
 #ifdef DEBUG
   std::cout<< "ElectronRecalibSuperClusterAssociator::ElectronRecalibSuperClusterAssociator" << std::endl;
 #endif
 
   //register your products
   produces<GsfElectronCollection>();
   produces<GsfElectronCoreCollection>() ;
   produces<SuperClusterCollection>();
   
   scProducer_ = iConfig.getParameter<std::string>("scProducer");
   scCollection_ = iConfig.getParameter<std::string>("scCollection");
 
   scIslandProducer_ = iConfig.getParameter<std::string>("scIslandProducer");
   scIslandCollection_ = iConfig.getParameter<std::string>("scIslandCollection");
 
   electronProducer_ = iConfig.getParameter<std::string > ("electronProducer");
   electronCollection_ = iConfig.getParameter<std::string > ("electronCollection");
 #ifdef DEBUG
   std::cout<< "ElectronRecalibSuperClusterAssociator::ElectronRecalibSuperClusterAssociator::end" << std::endl;
 #endif
 }

ElectronRecalibSuperClusterAssociator::~ElectronRecalibSuperClusterAssociator ( )

virtual

Definition at line 48 of file ElectronRecalibSuperClusterAssociator.cc.

49 {

50 }

Member Function Documentation

void ElectronRecalibSuperClusterAssociator::produce	(	edm::Event &	e,
		const edm::EventSetup &	c
	)

virtual

fixme : should have a vector of ptr of ref, to avoid copying

Implements edm::EDProducer.

Definition at line 53 of file ElectronRecalibSuperClusterAssociator.cc.

References ExpressReco_HICollisions_FallBack::cerr, reco::SuperCluster::clustersBegin(), reco::SuperCluster::clustersEnd(), gather_cfg::cout, ExpressReco_HICollisions_FallBack::electronCollection, reco::CaloCluster::energy(), edm::Event::getByLabel(), edm::Event::getRefBeforePut(), edm::HandleBase::isValid(), reco::CaloCluster::position(), funct::pow(), edm::Handle< T >::product(), edm::PtrVector< T >::push_back(), edm::Event::put(), ExpressReco_HICollisions_FallBack::scCollection, reco::SuperCluster::seed(), reco::GsfElectronCore::setGsfTrack(), reco::LeafCandidate::setP4(), reco::GsfElectronCore::setSuperCluster(), and mathSSE::sqrt().

Referenced by python.JSONExport.JsonExport::export(), and python.HTMLExport.HTMLExport::export().

 {
 #ifdef DEBUG
   std::cout<< "ElectronRecalibSuperClusterAssociator::produce" << std::endl;
 #endif
   // Create the output collections   
   std::auto_ptr<GsfElectronCollection> pOutEle(new GsfElectronCollection);
   std::auto_ptr<GsfElectronCoreCollection> pOutEleCore(new GsfElectronCoreCollection);
   std::auto_ptr<SuperClusterCollection> pOutNewEndcapSC(new SuperClusterCollection);
 
   reco::SuperClusterRefProd rSC = e.getRefBeforePut<SuperClusterCollection>();
   edm::Ref<SuperClusterCollection>::key_type idxSC = 0;
 
   //Get Hybrid SuperClusters
   Handle<reco::SuperClusterCollection> pSuperClusters;
   e.getByLabel(scProducer_, scCollection_, pSuperClusters);
   if (!pSuperClusters.isValid()) {
     std::cerr << "Error! can't get the product SuperClusterCollection "<< std::endl;
   }
   const reco::SuperClusterCollection* scCollection = pSuperClusters.product();
   
 #ifdef DEBUG
   std::cout<<"scCollection->size()"<<scCollection->size()<<std::endl;
 #endif
   
   //Get Island SuperClusters
   Handle<reco::SuperClusterCollection> pIslandSuperClusters;
   e.getByLabel(scIslandProducer_, scIslandCollection_, pIslandSuperClusters);
   if (!pIslandSuperClusters.isValid()) {
     std::cerr << "Error! can't get the product IslandSuperClusterCollection "<< std::endl;
   }
   const reco::SuperClusterCollection* scIslandCollection = pIslandSuperClusters.product();
   
 #ifdef DEBUG
   std::cout<<"scEECollection->size()"<<scIslandCollection->size()<<std::endl;
 #endif
 
   // Get Electrons
   Handle<reco::GsfElectronCollection> pElectrons;
   e.getByLabel(electronProducer_, electronCollection_, pElectrons);
   if (!pElectrons.isValid()) {
     std::cerr << "Error! can't get the product ElectronCollection "<< std::endl;
   }
   const reco::GsfElectronCollection* electronCollection = pElectrons.product();
   
   for(reco::GsfElectronCollection::const_iterator eleIt = electronCollection->begin(); eleIt != electronCollection->end(); eleIt++)
     {
       float DeltaRMineleSCbarrel(0.15); 
       float DeltaRMineleSCendcap(0.15); 
       const reco::SuperCluster* nearestSCbarrel=0;
       const reco::SuperCluster* nearestSCendcap=0;
       int iscRef=-1;
       int iSC=0;
       
       // first loop is on EB superClusters
       for(reco::SuperClusterCollection::const_iterator scIt = scCollection->begin();
       scIt != scCollection->end(); scIt++){
 #ifdef DEBUG    
     std::cout << scIt->energy() << " " << scIt->eta() << " " << scIt->phi() << " " << eleIt->eta() << " " << eleIt->phi() << std::endl;
 #endif
     
     double DeltaReleSC = sqrt ( pow(  eleIt->eta() - scIt->eta(),2) + pow(eleIt->phi() - scIt->phi(),2));
     
     if(DeltaReleSC<DeltaRMineleSCbarrel)
       {
         DeltaRMineleSCbarrel = DeltaReleSC;
         nearestSCbarrel = &*scIt;
         iscRef = iSC;
       }
     iSC++;
       }
       iSC = 0;
       
       // second loop is on EE superClusters
       int iscRefendcap=-1;
       
       for(reco::SuperClusterCollection::const_iterator scItEndcap = scIslandCollection->begin();
       scItEndcap != scIslandCollection->end(); scItEndcap++){
 #ifdef DEBUG    
     std::cout << "EE " << scItEndcap->energy() << " " << scItEndcap->eta() << " " << scItEndcap->phi() << " " << eleIt->eta() << " " << eleIt->phi() << std::endl;
 #endif
     
     double DeltaReleSC = sqrt ( pow(  eleIt->eta() - scItEndcap->eta(),2) + pow(eleIt->phi() - scItEndcap->phi(),2));
     
     if(DeltaReleSC<DeltaRMineleSCendcap)
       {
         DeltaRMineleSCendcap = DeltaReleSC;
         nearestSCendcap = &*scItEndcap;
         iscRefendcap = iSC;
       }
     iSC++;
       }
 
       GsfElectronCoreRefProd rEleCore=e.getRefBeforePut<GsfElectronCoreCollection>();
       edm::Ref<GsfElectronCoreCollection>::key_type idxEleCore = 0;
 
       if(nearestSCbarrel && !nearestSCendcap){
     reco::GsfElectronCore newEleCore(*(eleIt->core()));
     newEleCore.setGsfTrack(eleIt->gsfTrack());
     reco::SuperClusterRef scRef(reco::SuperClusterRef(pSuperClusters, iscRef));
     newEleCore.setSuperCluster(scRef);
     reco::GsfElectronCoreRef newEleCoreRef(reco::GsfElectronCoreRef(rEleCore, idxEleCore ++));
     pOutEleCore->push_back(newEleCore);
         reco::GsfElectron newEle(*eleIt,newEleCoreRef,CaloClusterPtr(),
 //                TrackRef(),GsfTrackRefVector());
                   TrackRef(),TrackBaseRef(), GsfTrackRefVector());
     newEle.setP4(eleIt->p4()*(nearestSCbarrel->energy()/eleIt->ecalEnergy()));
 
     pOutEle->push_back(newEle);
 #ifdef DEBUG
     std::cout << "Association is with EB superCluster "<< std::endl;
 #endif  
       }  
 
       if(!nearestSCbarrel && nearestSCendcap)
     {
 #ifdef DEBUG
     std::cout << "Starting Association is with EE superCluster "<< std::endl;
 #endif  
 
     float preshowerEnergy=eleIt->superCluster()->preshowerEnergy(); 
 #ifdef DEBUG
       std::cout << "preshowerEnergy"<< preshowerEnergy << std::endl;
 #endif
       CaloClusterPtrVector newBCRef;
       for (CaloCluster_iterator bcRefIt=nearestSCendcap->clustersBegin();bcRefIt!=nearestSCendcap->clustersEnd();++bcRefIt){
         CaloClusterPtr cPtr(*bcRefIt);
         newBCRef.push_back(cPtr);
       }
      
 
       reco::SuperCluster newSC(nearestSCendcap->energy() + preshowerEnergy, nearestSCendcap->position() , nearestSCendcap->seed(),newBCRef , preshowerEnergy );
       pOutNewEndcapSC->push_back(newSC);
       reco::SuperClusterRef scRef(reco::SuperClusterRef(rSC, idxSC ++));
 
       reco::GsfElectronCore newEleCore(*(eleIt->core()));
       newEleCore.setGsfTrack(eleIt->gsfTrack());
       newEleCore.setSuperCluster(scRef);
       reco::GsfElectronCoreRef newEleCoreRef(reco::GsfElectronCoreRef(rEleCore, idxEleCore ++));
       pOutEleCore->push_back(newEleCore);
       reco::GsfElectron newEle(*eleIt,newEleCoreRef,CaloClusterPtr(),
 //                TrackRef(),GsfTrackRefVector());
                   TrackRef(),TrackBaseRef(), GsfTrackRefVector());
            
           newEle.setP4(eleIt->p4()*(newSC.energy()/eleIt->ecalEnergy())) ;
       pOutEle->push_back(newEle);
 
 #ifdef DEBUG
     std::cout << "Association is with EE superCluster "<< std::endl;
 #endif  
       }  
     
       if(nearestSCbarrel && nearestSCendcap){
     reco::GsfElectronCore newEleCore(*(eleIt->core()));
     newEleCore.setGsfTrack(eleIt->gsfTrack());
 
     
     if(DeltaRMineleSCendcap>=DeltaRMineleSCbarrel)
       {
         reco::SuperClusterRef scRef(reco::SuperClusterRef(pSuperClusters, iscRef));
         newEleCore.setSuperCluster(scRef);
         reco::GsfElectronCoreRef newEleCoreRef(reco::GsfElectronCoreRef(rEleCore, idxEleCore ++));
         pOutEleCore->push_back(newEleCore);
         reco::GsfElectron newEle(*eleIt,newEleCoreRef,CaloClusterPtr(),
 //                TrackRef(),GsfTrackRefVector());
                   TrackRef(),TrackBaseRef(), GsfTrackRefVector());
         newEle.setP4(eleIt->p4()*(nearestSCbarrel->energy()/eleIt->ecalEnergy()));
         pOutEle->push_back(newEle);
 
 
 #ifdef DEBUG
         std::cout << "Association is with EB superCluster, after quarrel "<< std::endl;
 #endif  
       }
     else if(DeltaRMineleSCendcap<DeltaRMineleSCbarrel)
       {
         float preshowerEnergy=eleIt->superCluster()->preshowerEnergy(); 
         CaloClusterPtrVector newBCRef;
         for (CaloCluster_iterator bcRefIt=nearestSCendcap->clustersBegin();bcRefIt!=nearestSCendcap->clustersEnd();++bcRefIt){
           CaloClusterPtr cPtr(*bcRefIt);
           newBCRef.push_back(*bcRefIt);}
         reco::SuperCluster newSC(nearestSCendcap->energy() + preshowerEnergy,  nearestSCendcap->position() , nearestSCendcap->seed(), newBCRef , preshowerEnergy );
         pOutNewEndcapSC->push_back(newSC);
         reco::SuperClusterRef scRef(reco::SuperClusterRef(rSC, idxSC ++));
         newEleCore.setSuperCluster(scRef);
         reco::GsfElectronCoreRef newEleCoreRef(reco::GsfElectronCoreRef(rEleCore, idxEleCore ++));
         pOutEleCore->push_back(newEleCore);
         reco::GsfElectron newEle(*eleIt,newEleCoreRef,CaloClusterPtr(),
 //                TrackRef(),GsfTrackRefVector());
                   TrackRef(),TrackBaseRef(), GsfTrackRefVector());
         newEle.setP4(eleIt->p4()*(newSC.energy()/eleIt->ecalEnergy())) ;
         pOutEle->push_back(newEle);
 #ifdef DEBUG
         std::cout << "Association is with EE superCluster, after quarrel "<< std::endl;
 #endif  
       } 
 
       }
       
 
     }
   
   
   
 #ifdef DEBUG
   std::cout << "Filled new electrons  " << pOutEle->size() << std::endl;
   std::cout << "Filled new electronsCore  " << pOutEleCore->size() << std::endl;
   std::cout << "Filled new endcapSC  " << pOutNewEndcapSC->size() << std::endl;
 #endif  
   
   // put result into the Event
 
   e.put(pOutEle);
   e.put(pOutEleCore);
   e.put(pOutNewEndcapSC);
   
 }