#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 ()

ModuleDescription const &	moduleDescription () const

virtual	~EDProducer ()

Public Member Functions inherited from edm::ProducerBase
void	callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)

	ProducerBase ()

void	registerProducts (ProducerBase , ProductRegistry , ModuleDescription const &)

std::function< void(BranchDescription const &)>	registrationCallback () const
	used by the fwk to register list of products More...

virtual	~ProducerBase ()

Public Member Functions inherited from edm::EDConsumerBase
std::vector< ConsumesInfo >	consumesInfo () const

	EDConsumerBase ()

ProductHolderIndexAndSkipBit	indexFrom (EDGetToken, BranchType, TypeID const &) const

void	itemsMayGet (BranchType, std::vector< ProductHolderIndexAndSkipBit > &) const

void	itemsToGet (BranchType, std::vector< ProductHolderIndexAndSkipBit > &) const

std::vector < ProductHolderIndexAndSkipBit > const &	itemsToGetFromEvent () const

void	labelsForToken (EDGetToken iToken, Labels &oLabels) const

void	modulesDependentUpon (const std::string &iProcessName, std::vector< const char * > &oModuleLabels) const

void	modulesWhoseProductsAreConsumed (std::vector< ModuleDescription const * > &modules, ProductRegistry const &preg, std::map< std::string, ModuleDescription const * > const &labelsToDesc, std::string const &processName) const

bool	registeredToConsume (ProductHolderIndex, bool, BranchType) const

bool	registeredToConsumeMany (TypeID const &, BranchType) const

void	updateLookup (BranchType iBranchType, ProductHolderIndexHelper const &)

virtual	~EDConsumerBase ()

Private Attributes
edm::EDGetTokenT < reco::SuperClusterCollection >	ebScToken_

edm::EDGetTokenT < reco::SuperClusterCollection >	eeScToken_

edm::InputTag	electronSrc_

edm::EDGetTokenT < reco::GsfElectronCollection >	electronToken_

std::string	outputLabel_

edm::InputTag	superClusterCollectionEB_

edm::InputTag	superClusterCollectionEE_

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

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

Public Types inherited from edm::EDConsumerBase
typedef ProductLabels	Labels

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::EDConsumerBase
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	consumes (edm::InputTag const &tag)

EDGetToken	consumes (const TypeToGet &id, edm::InputTag const &tag)

template<BranchType B>
EDGetToken	consumes (TypeToGet const &id, edm::InputTag const &tag)

ConsumesCollector	consumesCollector ()
	Use a ConsumesCollector to gather consumes information from helper functions. More...

template<typename ProductType , BranchType B = InEvent>
void	consumesMany ()

void	consumesMany (const TypeToGet &id)

template<BranchType B>
void	consumesMany (const TypeToGet &id)

template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	mayConsume (edm::InputTag const &tag)

EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

template<BranchType B>
EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

Detailed Description

Definition at line 26 of file ElectronRecalibSuperClusterAssociator.h.

Constructor & Destructor Documentation

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

explicit

Definition at line 25 of file ElectronRecalibSuperClusterAssociator.cc.

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

 {
 #ifdef DEBUG
   std::cout<< "ElectronRecalibSuperClusterAssociator::ElectronRecalibSuperClusterAssociator" << std::endl;
 #endif
 
   //register your products
   produces<GsfElectronCollection>();
   produces<GsfElectronCoreCollection>() ;
   //  produces<SuperClusterCollection>();
   
   superClusterCollectionEB_ = iConfig.getParameter<edm::InputTag > ("superClusterCollectionEB");
   superClusterCollectionEE_ = iConfig.getParameter<edm::InputTag > ("superClusterCollectionEE");
 
   outputLabel_ = iConfig.getParameter<std::string>("outputLabel");
   electronSrc_ = iConfig.getParameter<edm::InputTag > ("electronSrc");
 
   electronToken_ = consumes<reco::GsfElectronCollection>(electronSrc_);
   ebScToken_     = consumes<reco::SuperClusterCollection>(superClusterCollectionEB_);
   eeScToken_     = consumes<reco::SuperClusterCollection>(superClusterCollectionEE_);
   
 #ifdef DEBUG
   std::cout<< "ElectronRecalibSuperClusterAssociator::ElectronRecalibSuperClusterAssociator::end" << std::endl;
 #endif
 }

ElectronRecalibSuperClusterAssociator::~ElectronRecalibSuperClusterAssociator ( )

virtual

Definition at line 51 of file ElectronRecalibSuperClusterAssociator.cc.

52 {

53 }

Member Function Documentation

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

virtual

Implements edm::EDProducer.

Definition at line 56 of file ElectronRecalibSuperClusterAssociator.cc.

References gather_cfg::cout, reco::deltaR2(), edm::Event::getByToken(), edm::Event::getRefBeforePut(), reco::GsfElectron::P4_FROM_SUPER_CLUSTER, edm::Event::put(), reco::GsfElectron::setCorrectedEcalEnergy(), reco::GsfElectron::setCorrectedEcalEnergyError(), reco::GsfElectron::setP4(), reco::GsfElectronCore::setParentSuperCluster(), reco::GsfElectronCore::setSuperCluster(), and mathSSE::sqrt().

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

 {
 #ifdef DEBUG
   std::cout<< "GEDElectronRecalibSuperClusterAssociator::produce" << std::endl;
 #endif
 
   // Create the output collections   
   std::auto_ptr<GsfElectronCollection> pOutEle(new GsfElectronCollection);
   std::auto_ptr<GsfElectronCoreCollection> pOutEleCore(new GsfElectronCoreCollection);
 
   // Get SuperClusters in EB
   Handle<reco::SuperClusterCollection> superClusterEBHandle;
   e.getByToken(ebScToken_, superClusterEBHandle); 
   //const reco::SuperClusterCollection* scCollection = superClusterEBHandle.product();
   
 #ifdef DEBUG
   std::cout<<"EB scCollection->size()" << superClusterEBHandle->size()<<std::endl;
 #endif
 
   // Get SuperClusters in EE
   Handle<reco::SuperClusterCollection> superClusterEEHandle;
   e.getByToken(eeScToken_, superClusterEEHandle);
   //  const reco::SuperClusterCollection* eeScCollection = superClusterEEHandle.product();
 
 #ifdef DEBUG
   std::cout<<"EE scCollection->size()" << superClusterEEHandle->size() << std::endl;
 #endif
 
   // Get Electrons
   edm::Handle<reco::GsfElectronCollection> eleHandle;
   e.getByToken(electronToken_, eleHandle);
   //  const reco::GsfElectronCollection* electronCollection = eleHandle.product();
 
   //  GsfElectronCoreRefProd rEleCore = const_cast<edm::Event&>(iEvent).getRefBeforePut<GsfElectronCoreCollection>();
   GsfElectronCoreRefProd rEleCore = e.getRefBeforePut<GsfElectronCoreCollection>();
   edm::Ref<GsfElectronCoreCollection>::key_type idxEleCore = 0;
   
   for(reco::GsfElectronCollection::const_iterator eleIt = eleHandle->begin(); eleIt != eleHandle->end(); eleIt++)
     {
       float DeltaRMineleSCbarrel(0.15); //initial minDeltaR
       float DeltaRMineleSCendcap(0.15); 
       const reco::SuperCluster* nearestSCbarrel=0;
       const reco::SuperCluster* nearestSCendcap=0;
       int iscRef=-1, iscRefendcap=-1;
       int iSC=0;
       
       if(eleIt->trackerDrivenSeed()){
     edm::LogError("trackerDriven") << "skipping trackerDriven electrons";
     continue;
       }
       // first loop is on EB superClusters
       iSC=0;
       for(reco::SuperClusterCollection::const_iterator scIt = superClusterEBHandle->begin();
       scIt != superClusterEBHandle->end(); scIt++, iSC++){
 
     double DeltaReleSC = sqrt(reco::deltaR2(eleIt->eta(), eleIt->phi(),
                         scIt->eta(), scIt->phi()));
     
     if(DeltaReleSC<DeltaRMineleSCbarrel) //save the nearest SC
       {
         DeltaRMineleSCbarrel = DeltaReleSC;
         nearestSCbarrel = &*scIt;
         iscRef = iSC;
       }
 #ifdef DEBUG    
     std::cout << "EB: " << scIt - superClusterEBHandle->begin() << " " << iSC << " " << iscRef 
           << "\t" << std::setprecision(4) << scIt->energy() 
           << " " << scIt->eta() << " " << scIt->phi() 
           << "\t--\t" << eleIt->energy() << " " << eleIt->eta() << " " << eleIt->phi() 
           << "\t" << DeltaRMineleSCbarrel
           << std::endl;
 #endif
       }
       
       // second loop is on EE superClusters
       iSC=0;
       for(reco::SuperClusterCollection::const_iterator scIt = superClusterEEHandle->begin();
       scIt != superClusterEEHandle->end(); scIt++, iSC++){
 #ifdef DEBUG    
     std::cout << "EE: " << scIt - superClusterEEHandle->begin() << " " << iSC << " " << iscRef 
           << "\t" << std::setprecision(4) << scIt->energy() 
           << " " << scIt->eta() << " " << scIt->phi() 
           << "\t--\t " << eleIt->energy() << " " << eleIt->eta() << " " << eleIt->phi() 
           << "\t" << DeltaRMineleSCendcap
           << std::endl;
 #endif
     
     double DeltaReleSC = sqrt(reco::deltaR2(eleIt->eta(), eleIt->phi(),
                         scIt->eta(), scIt->phi()));
     
     if(DeltaReleSC<DeltaRMineleSCendcap)
       {
         DeltaRMineleSCendcap = DeltaReleSC;
         nearestSCendcap = &*scIt;
         iscRefendcap = iSC;
       }
       }
       //      if(eleIt->isEB()) assert(DeltaRMineleSCbarrel < DeltaRMineleSCendcap);
       //else assert(DeltaRMineleSCbarrel > DeltaRMineleSCendcap);
       if(eleIt->isEB() && DeltaRMineleSCbarrel > DeltaRMineleSCendcap){
     edm::LogError("ElectronRecalibAssociator") << "EB electron, but nearest SC is in EE";;
     continue;
       }
 
       if(eleIt->isEB() && nearestSCbarrel){
     pOutEleCore->push_back(*eleIt->core()); // clone the old core and add to the collection of new cores
     reco::GsfElectronCoreRef newEleCoreRef(rEleCore, idxEleCore++); // reference to the new electron core in the new collection
     reco::GsfElectronCore & newEleCore = pOutEleCore->back(); // pick the clone
     //newEleCore.setGsfTrack(eleIt->gsfTrack());           // set the gsf track (not needed since it is not changed)
     reco::SuperClusterRef scRef(reco::SuperClusterRef(superClusterEBHandle, iscRef));  // Reference to the new SC
 #ifndef CMSSW_5_3_X
         newEleCore.setParentSuperCluster(scRef);             // mustache 
 #endif
     newEleCore.setSuperCluster(scRef); // let's check this! if it is possible to recreate the pfSC
     
     pOutEle->push_back(reco::GsfElectron(*eleIt,newEleCoreRef));
     reco::GsfElectron& newEle = pOutEle->back();
     
     //-- first possibility: set the new p4SC using refined SC
     newEle.setP4(reco::GsfElectron::P4_FROM_SUPER_CLUSTER, 
              eleIt->p4(reco::GsfElectron::P4_FROM_SUPER_CLUSTER),
              eleIt->p4Error(reco::GsfElectron::P4_FROM_SUPER_CLUSTER), false);  //*newEle.superCluster()->energy()/eleIt->superCluster()->energy());
     
     //-- second possibility: set the new p4SC using mustache SC
     //newEle.setP4(reco::GsfElectron::P4_FROM_SUPER_CLUSTER, eleIt->p4(reco::GsfElectron::P4_FROM_SUPER_CLUSTER)*newEle.parentSuperCluster()->energy()/eleIt->parentSuperCluster()->energy(), eleIt->p4Error(reco::GsfElectron::P4_FROM_SUPER_CLUSTER), false); 
 
     //-- update the correctedEcalEnergy
     newEle.setCorrectedEcalEnergy(eleIt->ecalEnergy()*(scRef->energy()/eleIt->p4(reco::GsfElectron::P4_FROM_SUPER_CLUSTER).energy()));
     newEle.setCorrectedEcalEnergyError(eleIt->ecalEnergyError()*(scRef->energy()/eleIt->ecalEnergy()));
     
       }  else if(!(eleIt->isEB()) && nearestSCendcap)
     {
     pOutEleCore->push_back(*eleIt->core()); // clone the old core and add to the collection of new cores
     reco::GsfElectronCoreRef newEleCoreRef(rEleCore, idxEleCore++); // reference to the new electron core in the new collection
     reco::GsfElectronCore & newEleCore = pOutEleCore->back(); // pick the clone
     //newEleCore.setGsfTrack(eleIt->gsfTrack());           // set the gsf track (not needed since it is not changed)
     reco::SuperClusterRef scRef(reco::SuperClusterRef(superClusterEEHandle, iscRefendcap));  // Reference to the new SC
 #ifndef CMSSW_5_3_X
         newEleCore.setParentSuperCluster(scRef);             // mustache 
 #endif
     newEleCore.setSuperCluster(scRef); // let's check this! if it is possible to recreate the pfSC
     
     pOutEle->push_back(reco::GsfElectron(*eleIt,newEleCoreRef));
     reco::GsfElectron& newEle = pOutEle->back();
     
     //-- first possibility: set the new p4SC using refined SC
     newEle.setP4(reco::GsfElectron::P4_FROM_SUPER_CLUSTER, 
              eleIt->p4(reco::GsfElectron::P4_FROM_SUPER_CLUSTER),
              eleIt->p4Error(reco::GsfElectron::P4_FROM_SUPER_CLUSTER), false);  //*newEle.superCluster()->energy()/eleIt->superCluster()->energy());
     
     //-- second possibility: set the new p4SC using mustache SC
     //newEle.setP4(reco::GsfElectron::P4_FROM_SUPER_CLUSTER, eleIt->p4(reco::GsfElectron::P4_FROM_SUPER_CLUSTER)*newEle.parentSuperCluster()->energy()/eleIt->parentSuperCluster()->energy(), eleIt->p4Error(reco::GsfElectron::P4_FROM_SUPER_CLUSTER), false); 
 
     //-- update the correctedEcalEnergy
     newEle.setCorrectedEcalEnergy(eleIt->ecalEnergy()*(scRef->energy()/eleIt->p4(reco::GsfElectron::P4_FROM_SUPER_CLUSTER).energy()));
     newEle.setCorrectedEcalEnergyError(eleIt->ecalEnergyError()*(scRef->energy()/eleIt->ecalEnergy()));
     }else{
     edm::LogError("Failed SC association") << "No SC to be associated to the electron";
       }
     }
   
   
   
 #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);
   
 }