#include <UncleanSCRecoveryProducer.h>

Inheritance diagram for UncleanSCRecoveryProducer:

Public Member Functions
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

bool	hasAbilityToProduceInBeginLumis () const final

bool	hasAbilityToProduceInBeginRuns () const final

bool	hasAbilityToProduceInEndLumis () const final

bool	hasAbilityToProduceInEndRuns () const final

bool	wantsGlobalLuminosityBlocks () const final

bool	wantsGlobalRuns () const final

bool	wantsStreamLuminosityBlocks () const final

bool	wantsStreamRuns () const final

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

ModuleDescription const &	moduleDescription () const

	~EDProducerBase () override

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

std::vector< edm::ProductResolverIndex > const &	putTokenIndexToProductResolverIndex () const

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, ModuleToResolverIndicies const &iIndicies, std::string const &moduleLabel)

	~ProducerBase () noexcept(false) override

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

void	convertCurrentProcessAlias (std::string const &processName)
	Convert "@currentProcess" in InputTag process names to the actual current process name. More...

	EDConsumerBase ()

	EDConsumerBase (EDConsumerBase const &)=delete

	EDConsumerBase (EDConsumerBase &&)=default

ESProxyIndex const *	esGetTokenIndices (edm::Transition iTrans) const

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 &	itemsToGetFrom (BranchType iType) 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

ProductResolverIndexAndSkipBit	uncheckedIndexFrom (EDGetToken) const

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

void	updateLookup (eventsetup::ESRecordsToProxyIndices const &)

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
using	ModuleToResolverIndicies = std::unordered_multimap< std::string, std::tuple< edm::TypeID const , const char , edm::ProductResolverIndex >>

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::ProducerBase
ProducesCollector	producesCollector ()

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 ESProduct , typename ESRecord , Transition Tr = Transition::Event>
auto	esConsumes ()

template<typename ESProduct , typename ESRecord , Transition Tr = Transition::Event>
auto	esConsumes (ESInputTag const &tag)

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 15 of file UncleanSCRecoveryProducer.h.

Constructor & Destructor Documentation

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

Definition at line 36 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

override

Definition at line 50 of file UncleanSCRecoveryProducer.cc.

References bcCollection_, cleanScCollection_, reco::SuperCluster::clustersBegin(), reco::SuperCluster::clustersEnd(), reco::CaloCluster::common, reco::CaloCluster::energy(), reco::CaloCluster::eta(), edm::Event::getByToken(), mps_fire::i, reco::CaloCluster::isInClean(), reco::CaloCluster::isInUnclean(), edm::OrphanHandleBase::isValid(), LogTrace, eostools::move(), reco::CaloCluster::position(), edm::Handle< T >::product(), edm::PtrVector< T >::push_back(), edm::Event::put(), scCollection_, reco::SuperCluster::seed(), SurveyInfoScenario_cff::seed, reco::CaloCluster::setFlags(), HLT_2018_cff::superClusters, uncleanBcCollection_, reco::CaloCluster::uncleanOnly, and uncleanScCollection_.

                                                                                                  {
   // __________________________________________________________________________
   //
   // 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();
   }
 }