#include <UncleanSCRecoveryProducer.h>

Inheritance diagram for UncleanSCRecoveryProducer:

Public Member Functions
virtual void	produce (edm::StreamID, edm::Event &, const edm::EventSetup &) const override

	UncleanSCRecoveryProducer (const edm::ParameterSet &ps)

Public Member Functions inherited from edm::global::EDProducer<>
	EDProducer ()=default

Public Member Functions inherited from edm::global::EDProducerBase
	EDProducerBase ()

ModuleDescription const &	moduleDescription () const

virtual	~EDProducerBase ()

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

std::vector< edm::ProductResolverIndex > const &	indiciesForPutProducts (BranchType iBranchType) const

	ProducerBase ()

void	registerProducts (ProducerBase , ProductRegistry , ModuleDescription const &)

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

void	resolvePutIndicies (BranchType iBranchType, std::unordered_multimap< std::string, edm::ProductResolverIndex > const &iIndicies, std::string const &moduleLabel)

virtual	~ProducerBase () noexcept(false)

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

	EDConsumerBase ()

	EDConsumerBase (EDConsumerBase const &)=delete

	EDConsumerBase (EDConsumerBase &&)=default

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

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

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

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

void	labelsForToken (EDGetToken iToken, Labels &oLabels) const

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

EDConsumerBase const &	operator= (EDConsumerBase const &)=delete

EDConsumerBase &	operator= (EDConsumerBase &&)=default

bool	registeredToConsume (ProductResolverIndex, bool, BranchType) const

bool	registeredToConsumeMany (TypeID const &, BranchType) const

void	updateLookup (BranchType iBranchType, ProductResolverIndexHelper const &, bool iPrefetchMayGet)

virtual	~EDConsumerBase () noexcept(false)

Private Attributes
const std::string	bcCollection_

const edm::EDGetTokenT< reco::BasicClusterCollection >	cleanBcCollection_

const edm::EDGetTokenT< reco::SuperClusterCollection >	cleanScCollection_

const std::string	scCollection_

const edm::EDGetTokenT< reco::BasicClusterCollection >	uncleanBcCollection_

const edm::EDGetTokenT< reco::SuperClusterCollection >	uncleanScCollection_

Additional Inherited Members
Public Types inherited from edm::global::EDProducerBase
typedef EDProducerBase	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::global::EDProducerBase
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 17 of file UncleanSCRecoveryProducer.h.

Constructor & Destructor Documentation

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

Definition at line 38 of file UncleanSCRecoveryProducer.cc.

References bcCollection_, and scCollection_.

                                                                              :
   cleanBcCollection_(consumes<reco::BasicClusterCollection>(ps.getParameter<edm::InputTag>("cleanBcCollection"))),
   cleanScCollection_(consumes<reco::SuperClusterCollection>(ps.getParameter<edm::InputTag>("cleanScCollection"))),
   uncleanBcCollection_(consumes<reco::BasicClusterCollection>(ps.getParameter<edm::InputTag>("uncleanBcCollection"))),
   uncleanScCollection_(consumes<reco::SuperClusterCollection>(ps.getParameter<edm::InputTag>("uncleanScCollection"))),
   bcCollection_(ps.getParameter<std::string>("bcCollection")),
   scCollection_(ps.getParameter<std::string>("scCollection"))
 {
         // the products:
         produces< reco::BasicClusterCollection >(bcCollection_);
         produces< reco::SuperClusterCollection >(scCollection_);
 }

Member Function Documentation

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

overridevirtual

Definition at line 52 of file UncleanSCRecoveryProducer.cc.

 {
         // __________________________________________________________________________
         //
         // cluster collections:
         edm::Handle<reco::BasicClusterCollection> pCleanBC;
         edm::Handle<reco::SuperClusterCollection> pCleanSC;
         //
         edm::Handle<reco::BasicClusterCollection> pUncleanBC;
         edm::Handle<reco::SuperClusterCollection> pUncleanSC;
         // clean collections ________________________________________________________
         evt.getByToken(cleanScCollection_, pCleanSC);      
         const  reco::SuperClusterCollection cleanSC = *(pCleanSC.product());
 
         // unclean collections ______________________________________________________
         evt.getByToken(uncleanBcCollection_, pUncleanBC);
         const  reco::BasicClusterCollection uncleanBC = *(pUncleanBC.product());
         //
         evt.getByToken(uncleanScCollection_, pUncleanSC);
         const  reco::SuperClusterCollection uncleanSC = *(pUncleanSC.product());
         int uncleanSize = pUncleanSC->size();
         int cleanSize =   pCleanSC->size();
 
         LogTrace("EcalCleaning")  << "Size of Clean Collection: " << cleanSize 
                 << ", uncleanSize: " << uncleanSize;
 
         // collections are all taken now ____________________________________________
         //
         // the collections to be produced ___________________________________________
         reco::BasicClusterCollection basicClusters;
         reco::SuperClusterCollection superClusters;
         //
         //
         // collect all the basic clusters of the SC that belong to the unclean
         // collection and put them into the basicClusters vector
         // keep the information of which SC they belong to
         //
         // loop over the unclean sc: all SC will join the new collection
         std::vector< std::pair<int, int> > basicClusterOwner; // counting all
 
         std::vector<DetId> scUncleanSeedDetId;  // counting the unclean
         for (int isc =0; isc< uncleanSize; ++isc) {
                 const reco::SuperCluster unsc = uncleanSC[isc];    
                 scUncleanSeedDetId.push_back(unsc.seed()->seed());
                 reco::CaloCluster_iterator bciter = unsc.clustersBegin();
                 for (; bciter != unsc.clustersEnd(); ++bciter) {
                         // the basic clusters
                         basicClusters.push_back(**bciter);
                         // index of the unclean SC
                         basicClusterOwner.push_back( std::make_pair(isc,0) ); 
                 }
         }
         // loop over the clean: only the ones which are in common with the unclean
         // are taken into account
 
         std::vector<DetId> scCleanSeedDetId;  // counting the clean
         std::vector<int> isToBeKept;
         for (int isc =0; isc< cleanSize; ++isc) {
                 reco::SuperClusterRef cscRef( pCleanSC, isc );    
                 scCleanSeedDetId.push_back(cscRef->seed()->seed());
                 for (reco::CaloCluster_iterator bciter = cscRef->clustersBegin(); bciter != cscRef->clustersEnd(); ++bciter) {
                         // the basic clusters
                         basicClusters.push_back(**bciter);
                         // index of the clean SC
                         basicClusterOwner.push_back( std::make_pair(isc,1) ); 
                 }
                 if (cscRef->isInUnclean()) isToBeKept.push_back(1);
                 else isToBeKept.push_back(0);
         }
         //
         // now export the basic clusters into the event and get them back
         auto basicClusters_p = std::make_unique<reco::BasicClusterCollection>();
         basicClusters_p->assign(basicClusters.begin(), basicClusters.end());
         edm::OrphanHandle<reco::BasicClusterCollection> bccHandle =  
                 evt.put(std::move(basicClusters_p), bcCollection_);
         if (!(bccHandle.isValid())) {
 
                 edm::LogWarning("MissingInput") << "could not handle the new BasicClusters!";
                 return;
         }
         reco::BasicClusterCollection basicClustersProd = *bccHandle;
 
         LogTrace("EcalCleaning") <<"Got the BasicClusters from the event again";
         int bcSize = bccHandle->size();
         //
         // now we can create the SC collection
         //
         // run over the unclean SC: all to be kept here
         for (int isc=0; isc< uncleanSize; ++isc) {
                 reco::CaloClusterPtrVector clusterPtrVector;
                 // the seed is the basic cluster with the highest energy
                 reco::CaloClusterPtr seed; 
                 for (int jbc=0; jbc< bcSize; ++jbc) {
                         std::pair<int,int> theBcOwner = basicClusterOwner[jbc];
                         if (theBcOwner.first == isc && theBcOwner.second == 0) {
                                 reco::CaloClusterPtr currentClu=reco::CaloClusterPtr(bccHandle,jbc);
                                 clusterPtrVector.push_back(currentClu);
                                 if (scUncleanSeedDetId[isc] == currentClu->seed()) {
                                         seed = currentClu;
                                 }
                         }
                 }
                 const reco::SuperCluster unsc = uncleanSC[isc]; 
                 reco::SuperCluster newSC(unsc.energy(), unsc.position(), seed, clusterPtrVector );
                 newSC.setFlags(reco::CaloCluster::uncleanOnly);
                 superClusters.push_back(newSC);
         }
         // run over the clean SC: only those who are in common between the
         // clean and unclean collection are kept
         for (int isc=0; isc< cleanSize; ++isc) {
                 reco::SuperClusterRef cscRef( pCleanSC, isc); 
                 if (not cscRef->isInUnclean()) continue;
                 reco::CaloClusterPtrVector clusterPtrVector;
                 // the seed is the basic cluster with the highest energy
                 reco::CaloClusterPtr seed; 
                 for (int jbc=0; jbc< bcSize; ++jbc) {
                         std::pair<int,int> theBcOwner = basicClusterOwner[jbc];
                         if (theBcOwner.first == isc && theBcOwner.second == 1) {
                                 reco::CaloClusterPtr currentClu=reco::CaloClusterPtr(bccHandle,jbc);
                                 clusterPtrVector.push_back(currentClu);
                                 if (scCleanSeedDetId[isc] == currentClu->seed()) {
                                         seed = currentClu;
                                 }
                         }
                 }
                 reco::SuperCluster newSC(cscRef->energy(), cscRef->position(), 
                                          seed, clusterPtrVector );
                 newSC.setFlags(reco::CaloCluster::common);
                 superClusters.push_back(newSC);
         }
 
         auto superClusters_p = std::make_unique<reco::SuperClusterCollection>();
         superClusters_p->assign(superClusters.begin(), superClusters.end());
 
         evt.put(std::move(superClusters_p), scCollection_);
 
         LogTrace("EcalCleaning")<<"Clusters (Basic/Super) added to the Event! :-)";
 
         // ----- debugging ----------
         // print the new collection SC quantities
         // print out the clean collection SC
         LogTrace("EcalCleaning") << "Clean Collection SC ";
         for (int i=0; i < cleanSize; ++i) {
                 const reco::SuperCluster csc = cleanSC[i];
                 LogTrace("EcalCleaning") << " >>> clean    #" << i << "; Energy: " << csc.energy()
                         << " eta: " << csc.eta() 
                         << " sc seed detid: " << csc.seed()->seed().rawId();
         }
         // the unclean SC
         LogTrace("EcalCleaning") << "Unclean Collection SC ";
         for (int i=0; i < uncleanSize; ++i) {
                 const reco::SuperCluster usc = uncleanSC[i];
                 LogTrace("EcalCleaning") << " >>> unclean  #" << i << "; Energy: " << usc.energy()
                         << " eta: " << usc.eta() 
                         << " sc seed detid: " << usc.seed()->seed().rawId();
         }
         // the new collection
         LogTrace("EcalCleaning")<<"The new SC clean collection with size "<< superClusters.size();
         for (unsigned int i=0; i <  superClusters.size(); ++i) {
                 const reco::SuperCluster nsc = superClusters[i];
                 LogTrace("EcalCleaning")<< " >>> newSC    #" << i << "; Energy: " << nsc.energy()
                         << " eta: " << nsc.eta()  << " isClean=" 
                         << nsc.isInClean() << " isUnclean=" << nsc.isInUnclean()
                         << " sc seed detid: " << nsc.seed()->seed().rawId();
         }
 }