#include <EgammaHLTGsfTrackVarProducer.h>

Inheritance diagram for EgammaHLTGsfTrackVarProducer:

Classes
class	TrackExtrapolator

Public Member Functions
	EgammaHLTGsfTrackVarProducer (const edm::ParameterSet &)

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

	~EgammaHLTGsfTrackVarProducer ()

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
edm::InputTag	beamSpotTag_

edm::InputTag	inputCollectionTag_

int	lowerTrackNrToRemoveCut_

edm::InputTag	recoEcalCandTag_

TrackExtrapolator	trackExtrapolator_

int	upperTrackNrToRemoveCut_

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)

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

Definition at line 32 of file EgammaHLTGsfTrackVarProducer.h.

Constructor & Destructor Documentation

EgammaHLTGsfTrackVarProducer::EgammaHLTGsfTrackVarProducer ( const edm::ParameterSet & config )

explicit

Definition at line 51 of file EgammaHLTGsfTrackVarProducer.cc.

References beamSpotTag_, edm::ParameterSet::getParameter(), inputCollectionTag_, lowerTrackNrToRemoveCut_, recoEcalCandTag_, and upperTrackNrToRemoveCut_.

 {
   recoEcalCandTag_ = config.getParameter<edm::InputTag>("recoEcalCandidateProducer");
   inputCollectionTag_             = config.getParameter<edm::InputTag>("inputCollection");
   beamSpotTag_                   = config.getParameter<edm::InputTag>("beamSpotProducer");
   upperTrackNrToRemoveCut_  = config.getParameter<int>("upperTrackNrToRemoveCut"); //zeros out dEtaIn,dPhiIn if nrTracks>= this
   lowerTrackNrToRemoveCut_  = config.getParameter<int>("lowerTrackNrToRemoveCut"); //zeros out dEtaIn,dPhiIn if nrTracks<= this
  
   
   
   //register your products
   produces < reco::RecoEcalCandidateIsolationMap >( "Deta" ).setBranchAlias( "deta" );
   produces < reco::RecoEcalCandidateIsolationMap >( "Dphi" ).setBranchAlias( "dphi" ); 
 }

EgammaHLTGsfTrackVarProducer::~EgammaHLTGsfTrackVarProducer ( )

Definition at line 66 of file EgammaHLTGsfTrackVarProducer.cc.

66 {}

Member Function Documentation

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

virtual

Implements edm::EDProducer.

Definition at line 75 of file EgammaHLTGsfTrackVarProducer.cc.

References SiPixelRawToDigiRegional_cfi::beamSpot, beamSpotTag_, edm::RefToBase< T >::castTo(), EgammaHLTGsfTrackVarProducer::TrackExtrapolator::extrapolateTrackPosToPoint(), edm::EventSetup::get(), edm::Event::getByLabel(), inputCollectionTag_, edm::AssociationMap< Tag >::insert(), edm::HandleBase::isValid(), lowerTrackNrToRemoveCut_, reco::BeamSpot::position(), edm::Event::put(), recoEcalCandTag_, EgammaHLTGsfTrackVarProducer::TrackExtrapolator::setup(), trackExtrapolator_, and upperTrackNrToRemoveCut_.

 {
 
   trackExtrapolator_.setup(iSetup);
 
   // Get the HLT filtered objects
   edm::Handle<reco::RecoEcalCandidateCollection> recoEcalCandHandle;
   iEvent.getByLabel(recoEcalCandTag_,recoEcalCandHandle);
 
   edm::Handle<reco::ElectronCollection> electronHandle;
   iEvent.getByLabel(inputCollectionTag_,electronHandle);
 
   edm::Handle<reco::GsfTrackCollection> gsfTracksHandle;
   if(!electronHandle.isValid()) iEvent.getByLabel (inputCollectionTag_,gsfTracksHandle);
 
   edm::Handle<reco::BeamSpot> recoBeamSpotHandle;
   iEvent.getByLabel(beamSpotTag_,recoBeamSpotHandle);
   // gets its position
   const reco::BeamSpot& beamSpot = *recoBeamSpotHandle; 
 
   edm::ESHandle<MagneticField> theMagField;
   iSetup.get<IdealMagneticFieldRecord>().get(theMagField);
 
   reco::RecoEcalCandidateIsolationMap dEtaMap;
   reco::RecoEcalCandidateIsolationMap dPhiMap;
   
   for(reco::RecoEcalCandidateCollection::const_iterator iRecoEcalCand = recoEcalCandHandle->begin(); iRecoEcalCand != recoEcalCandHandle->end(); iRecoEcalCand++){
     reco::RecoEcalCandidateRef recoEcalCandRef(recoEcalCandHandle,iRecoEcalCand-recoEcalCandHandle->begin());
    
     const reco::SuperClusterRef scRef = recoEcalCandRef->superCluster();
    
     //the idea is that we can take the tracks from properly associated electrons or just take all gsf tracks with that sc as a seed
     std::vector<const reco::GsfTrack*> gsfTracks;
     if(electronHandle.isValid()){
       for(reco::ElectronCollection::const_iterator eleIt = electronHandle->begin(); eleIt != electronHandle->end(); eleIt++){
     if(eleIt->superCluster()==scRef){
       gsfTracks.push_back(&*eleIt->gsfTrack());
     }
       }
     }else{ 
       for(reco::GsfTrackCollection::const_iterator trkIt =gsfTracksHandle->begin();trkIt!=gsfTracksHandle->end();++trkIt){
     edm::RefToBase<TrajectorySeed> seed = trkIt->extra()->seedRef() ;
     reco::ElectronSeedRef elseed = seed.castTo<reco::ElectronSeedRef>() ;
     edm::RefToBase<reco::CaloCluster> caloCluster = elseed->caloCluster() ;
     reco::SuperClusterRef scRefFromTrk = caloCluster.castTo<reco::SuperClusterRef>() ;
     if(scRefFromTrk==scRef){
       gsfTracks.push_back(&*trkIt);
     }
       }
       
     }
     float dEtaInValue=999999;
     float dPhiInValue=999999;
     
     if(static_cast<int>(gsfTracks.size())>=upperTrackNrToRemoveCut_){
       dEtaInValue=0;
       dPhiInValue=0;
     }else if(static_cast<int>(gsfTracks.size())<=lowerTrackNrToRemoveCut_){
       dEtaInValue=0;
       dPhiInValue=0;
     }else{
       for(size_t trkNr=0;trkNr<gsfTracks.size();trkNr++){
       
     GlobalPoint scPos(scRef->x(),scRef->y(),scRef->z());
     GlobalPoint trackExtrapToSC = trackExtrapolator_.extrapolateTrackPosToPoint(*gsfTracks[trkNr],scPos);
     EleRelPointPair scAtVtx(scRef->position(),trackExtrapToSC,beamSpot.position());
     
     if(fabs(scAtVtx.dEta())<dEtaInValue) dEtaInValue=fabs(scAtVtx.dEta()); //we are allowing them to come from different tracks
     if(fabs(scAtVtx.dPhi())<dPhiInValue) dPhiInValue=fabs(scAtVtx.dPhi());//we are allowing them to come from different tracks
       } 
     }
     
     dEtaMap.insert(recoEcalCandRef, dEtaInValue);
     dPhiMap.insert(recoEcalCandRef, dPhiInValue);
   }
 
   std::auto_ptr<reco::RecoEcalCandidateIsolationMap> dEtaMapForEvent(new reco::RecoEcalCandidateIsolationMap(dEtaMap));
   std::auto_ptr<reco::RecoEcalCandidateIsolationMap> dPhiMapForEvent(new reco::RecoEcalCandidateIsolationMap(dPhiMap));
   iEvent.put(dEtaMapForEvent, "Deta" );
   iEvent.put(dPhiMapForEvent, "Dphi" );
 
 }