#include <EgammaHLTNxNClusterProducer.h>

Inheritance diagram for EgammaHLTNxNClusterProducer:

Public Member Functions
	EgammaHLTNxNClusterProducer (const edm::ParameterSet &ps)

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

	~EgammaHLTNxNClusterProducer ()

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 Member Functions
bool	checkStatusOfEcalRecHit (const EcalChannelStatus &channelStatus, const EcalRecHit &rh)

void	makeNxNClusters (edm::Event &evt, const edm::EventSetup &es, const EcalRecHitCollection *hits, const reco::CaloID::Detectors detector)

Private Attributes
std::string	barrelClusterCollection_

edm::InputTag	barrelHitProducer_

std::string	barrelHits_

int	clusEtaSize_

int	clusPhiSize_

double	clusSeedThr_

double	clusSeedThrEndCap_

int	debug_

bool	doBarrel_

bool	doEndcaps_

std::string	endcapClusterCollection_

edm::InputTag	endcapHitProducer_

std::string	endcapHits_

int	flagLevelRecHitsToUse_

int	maxNumberofClusters_

int	maxNumberofSeeds_

PositionCalc	posCalculator_

std::map< std::string, double >	providedParameters

int	statusLevelRecHitsToUse_

bool	useDBStatus_

bool	useRecoFlag_

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 57 of file EgammaHLTNxNClusterProducer.h.

Constructor & Destructor Documentation

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

Definition at line 45 of file EgammaHLTNxNClusterProducer.cc.

References edm::ParameterSet::getParameter().

 {
   
   doBarrel_   = ps.getParameter<bool>("doBarrel");
   doEndcaps_   = ps.getParameter<bool>("doEndcaps");
     
   barrelHitProducer_ = ps.getParameter< edm::InputTag > ("barrelHitProducer");
   endcapHitProducer_ = ps.getParameter< edm::InputTag > ("endcapHitProducer");
   
   clusEtaSize_ = ps.getParameter<int> ("clusEtaSize");
   clusPhiSize_ = ps.getParameter<int> ("clusPhiSize");
   
   // The names of the produced cluster collections
   barrelClusterCollection_  = ps.getParameter<std::string>("barrelClusterCollection");
   endcapClusterCollection_  = ps.getParameter<std::string>("endcapClusterCollection");
   
   clusSeedThr_ = ps.getParameter<double> ("clusSeedThr");
   clusSeedThrEndCap_ = ps.getParameter<double> ("clusSeedThrEndCap");
   
   useRecoFlag_ = ps.getParameter<bool>("useRecoFlag");
   flagLevelRecHitsToUse_ = ps.getParameter<int>("flagLevelRecHitsToUse"); 
   
   useDBStatus_ = ps.getParameter<bool>("useDBStatus");
   statusLevelRecHitsToUse_ = ps.getParameter<int>("statusLevelRecHitsToUse"); 
   
     // Parameters for the position calculation:
   posCalculator_ = PositionCalc( ps.getParameter<edm::ParameterSet>("posCalcParameters") );
 
   //max number of seeds / clusters, once reached, then return 0 
   maxNumberofSeeds_    = ps.getParameter<int> ("maxNumberofSeeds");
   maxNumberofClusters_ = ps.getParameter<int> ("maxNumberofClusters");
   
   debug_ = ps.getParameter<int> ("debugLevel");
   
   produces< reco::BasicClusterCollection >(barrelClusterCollection_);
   produces< reco::BasicClusterCollection >(endcapClusterCollection_);
   
 }

EgammaHLTNxNClusterProducer::~EgammaHLTNxNClusterProducer ( )

Definition at line 92 of file EgammaHLTNxNClusterProducer.cc.

 {
   //delete island_p;
 }

Member Function Documentation

bool EgammaHLTNxNClusterProducer::checkStatusOfEcalRecHit	(	const EcalChannelStatus &	channelStatus,
		const EcalRecHit &	rh
	)

private

from recoFlag()

good

good || PoorCalib

good || PoorCalib || LeadingEdgeRecovered || kNeighboursRecovered,

Definition at line 135 of file EgammaHLTNxNClusterProducer.cc.

References EcalRecHit::id(), DetId::rawId(), EcalRecHit::recoFlag(), and ntuplemaker::status.

                                                                                                                     {
   
   if(useRecoFlag_ ){ 
     int flag = rh.recoFlag();
     if( flagLevelRecHitsToUse_ ==0){ 
       if( flag != 0) return false; 
     }
     else if( flagLevelRecHitsToUse_ ==1){ 
       if( flag !=0 && flag != 4 ) return false; 
     }
     else if( flagLevelRecHitsToUse_ ==2){ 
       if( flag !=0 && flag != 4 && flag != 6 && flag != 7) return false; 
     }
   }
   if ( useDBStatus_){ 
     int status =  int(channelStatus[rh.id().rawId()].getStatusCode()); 
     if ( status > statusLevelRecHitsToUse_ ) return false; 
   }
   
   return true; 
 }

void EgammaHLTNxNClusterProducer::makeNxNClusters	(	edm::Event &	evt,
		const edm::EventSetup &	es,
		const EcalRecHitCollection *	hits,
		const reco::CaloID::Detectors	detector
	)

private

get status from DB

if too much clusters made, then return 0 also

Definition at line 160 of file EgammaHLTNxNClusterProducer.cc.

References edm::SortedCollection< T, SORT >::begin(), reco::CaloID::DET_ECAL_BARREL, cond::rpcobgas::detid, DetId::Ecal, EcalBarrel, EcalEndcap, EcalPreshower, edm::SortedCollection< T, SORT >::end(), relval_parameters_module::energy, edm::EventID::event(), spr::find(), edm::SortedCollection< T, SORT >::find(), edm::EventSetup::get(), CaloSubdetectorTopology::getWindow(), edm::EventBase::id(), reco::CaloCluster::island, LogDebug, edm::Event::put(), edm::EventID::run(), and python.multivaluedict::sort().

 {
   
   
   edm::ESHandle<EcalChannelStatus> csHandle;
   if ( useDBStatus_ ) es.get<EcalChannelStatusRcd>().get(csHandle);
   const EcalChannelStatus &channelStatus = *csHandle; 
     
   
   std::vector<EcalRecHit> seeds;
   
   double clusterSeedThreshold ; 
   if (detector == reco::CaloID::DET_ECAL_BARREL){
     clusterSeedThreshold = clusSeedThr_;
   }else{
     clusterSeedThreshold = clusSeedThrEndCap_; 
   }
   
 
   for(EcalRecHitCollection::const_iterator itt = hits->begin(); itt != hits->end(); itt++){
     double energy = itt->energy();
     if( ! checkStatusOfEcalRecHit(channelStatus, *itt) ) continue; 
     if (energy > clusterSeedThreshold ) seeds.push_back(*itt);
     
     if( int(seeds.size()) > maxNumberofSeeds_){ //too many seeds, like beam splash events
       seeds.clear();  
       break; 
     }
   }
   
   // get the geometry and topology from the event setup:
   edm::ESHandle<CaloGeometry> geoHandle;
   es.get<CaloGeometryRecord>().get(geoHandle);
   
   const CaloSubdetectorGeometry *geometry_p;
   CaloSubdetectorTopology *topology_p;
   if (detector == reco::CaloID::DET_ECAL_BARREL) {
     geometry_p = geoHandle->getSubdetectorGeometry(DetId::Ecal, EcalBarrel);
     topology_p = new EcalBarrelTopology(geoHandle);
   }else {
     geometry_p = geoHandle->getSubdetectorGeometry(DetId::Ecal, EcalEndcap);
     topology_p = new EcalEndcapTopology(geoHandle); 
   }
   
   const CaloSubdetectorGeometry *geometryES_p;
   geometryES_p = geoHandle->getSubdetectorGeometry(DetId::Ecal, EcalPreshower);
     
 
   
   std::vector<reco::BasicCluster> clusters;
   std::vector<DetId> usedXtals;
   
   // sort seed according to Energy
   sort(seeds.begin(), seeds.end(), ecalRecHitSort());
   
   
   
   for (std::vector<EcalRecHit>::iterator itseed=seeds.begin(); itseed!=seeds.end(); itseed++) {
     DetId seed_id = itseed->id();
     std::vector<DetId>::const_iterator usedIds;
     
     std::vector<DetId>::iterator  itdet = find(usedXtals.begin(),usedXtals.end(),seed_id);
     if(itdet != usedXtals.end()) continue; 
     
     std::vector<DetId> clus_v = topology_p->getWindow(seed_id,clusEtaSize_,clusPhiSize_);   
     std::vector<std::pair<DetId, float> > clus_used;
     
     float clus_energy = 0; 
     
     for (std::vector<DetId>::iterator det=clus_v.begin(); det!=clus_v.end(); det++) {
     DetId detid = *det;
     
     //not yet used 
     std::vector<DetId>::iterator itdet = find(usedXtals.begin(),usedXtals.end(),detid);
     if(itdet != usedXtals.end()) continue; 
     //inside the collection
     EcalRecHitCollection::const_iterator hit  = hits->find(detid); 
     if( hit == hits->end()) continue; 
     
     if( ! checkStatusOfEcalRecHit(channelStatus, *hit) ) continue; 
     
     usedXtals.push_back(detid);
     clus_used.push_back(std::pair<DetId, float>(detid, 1.) );
     clus_energy += hit->energy();
     
     } 
     
     if( clus_energy <= 0 ) continue; 
     
     math::XYZPoint clus_pos = posCalculator_.Calculate_Location(clus_used,hits,geometry_p,geometryES_p);
     
     if (debug_>=2 ) LogDebug("")<<"nxn_cluster in run "<< evt.id().run()<<" event "<<evt.id().event()<<" energy: "<<clus_energy <<" eta: "<< clus_pos.Eta()<<" phi: "<< clus_pos.Phi()<<" nRecHits: "<< clus_used.size() <<std::endl;
     
     clusters.push_back(reco::BasicCluster(clus_energy, clus_pos, reco::CaloID(detector), clus_used, reco::CaloCluster::island, seed_id));
     if( int(clusters.size()) > maxNumberofClusters_){ 
       clusters.clear(); 
       break; 
     }
     
   }
   
   
   //Create empty output collections
   std::auto_ptr< reco::BasicClusterCollection > clusters_p(new reco::BasicClusterCollection);
   clusters_p->assign(clusters.begin(), clusters.end());
   if (detector == reco::CaloID::DET_ECAL_BARREL){
     if(debug_>=1) LogDebug("")<<"nxnclusterProducer: "<<clusters_p->size() <<" made in barrel"<<std::endl;
     evt.put(clusters_p, barrelClusterCollection_);
   }
   else {
     if(debug_>=1) LogDebug("")<<"nxnclusterProducer: "<<clusters_p->size() <<" made in endcap"<<std::endl;
     evt.put(clusters_p, endcapClusterCollection_);
   }
   
 }

void EgammaHLTNxNClusterProducer::produce	(	edm::Event &	evt,
		const edm::EventSetup &	es
	)

virtual

Implements edm::EDProducer.

Definition at line 98 of file EgammaHLTNxNClusterProducer.cc.

References reco::CaloID::DET_ECAL_BARREL, reco::CaloID::DET_ECAL_ENDCAP, edm::EventID::event(), edm::Event::getByLabel(), edm::EventBase::id(), edm::HandleBase::isValid(), LogDebug, edm::Handle< T >::product(), edm::EventID::run(), and edm::SortedCollection< T, SORT >::size().

 {
   
   
   if(doBarrel_){
     Handle<EcalRecHitCollection> barrelRecHitsHandle;
     evt.getByLabel(barrelHitProducer_,barrelRecHitsHandle);
     if (!barrelRecHitsHandle.isValid()) {
       LogDebug("") << "EgammaHLTNxNClusterProducer Error! can't get product eb hit!" << std::endl;
     }
     
     const EcalRecHitCollection *hits_eb = barrelRecHitsHandle.product();
     if( debug_>=2 ) LogDebug("")<<"EgammaHLTNxNClusterProducer nEBrechits: "<< evt.id().run()<<" event "<<evt.id().event() <<" "<< hits_eb->size()<<std::endl;
     
     makeNxNClusters(evt,es,hits_eb, reco::CaloID::DET_ECAL_BARREL);
     
   }
   
   
   if(doEndcaps_){
     Handle<EcalRecHitCollection> endcapRecHitsHandle;
     evt.getByLabel(endcapHitProducer_,endcapRecHitsHandle);
     if (!endcapRecHitsHandle.isValid()) {
       LogDebug("") << "EgammaHLTNxNClusterProducer Error! can't get product ee hit!" << std::endl;
     }
     
     const EcalRecHitCollection *hits_ee = endcapRecHitsHandle.product();
     if( debug_>=2 ) LogDebug("")<<"EgammaHLTNxNClusterProducer nEErechits: "<< evt.id().run()<<" event "<<evt.id().event() <<" "<< hits_ee->size()<<std::endl;
     makeNxNClusters(evt,es,hits_ee, reco::CaloID::DET_ECAL_ENDCAP);
   }
   
   
   
 }