HGCalElectronIDValueMapProducer Class Reference

#include <RecoEgamma/EgammaTools/plugins/HGCalElectronIDValueMapProducer.cc>

Inheritance diagram for HGCalElectronIDValueMapProducer:

Public Member Functions
	HGCalElectronIDValueMapProducer (const edm::ParameterSet &)
	~HGCalElectronIDValueMapProducer () override
Public Member Functions inherited from edm::stream::EDProducer<>
	EDProducer ()=default
Public Member Functions inherited from edm::stream::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
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
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
void	updateLookup (BranchType iBranchType, ProductResolverIndexHelper const &, bool iPrefetchMayGet)
virtual	~EDConsumerBase () noexcept(false)

Static Public Member Functions
static void	fillDescriptions (edm::ConfigurationDescriptions &descriptions)
Static Public Member Functions inherited from edm::stream::EDProducerBase
static const std::string &	baseType ()
static void	fillDescriptions (ConfigurationDescriptions &descriptions)
static void	prevalidate (ConfigurationDescriptions &descriptions)

Private Member Functions
void	beginStream (edm::StreamID) override
void	endStream () override
void	produce (edm::Event &, const edm::EventSetup &) override

Private Attributes
std::unique_ptr< HGCalEgammaIDHelper >	eIDHelper_
edm::EDGetTokenT< edm::View< reco::GsfElectron > >	electronsToken_
std::map< const std::string, std::vector< float > >	maps_
float	radius_

Static Private Attributes
static const std::vector< std::string >	valuesProduced_

Additional Inherited Members
Public Types inherited from edm::stream::EDProducer<>
typedef CacheContexts< T... >	CacheTypes
typedef CacheTypes::GlobalCache	GlobalCache
typedef AbilityChecker< T... >	HasAbility
typedef CacheTypes::LuminosityBlockCache	LuminosityBlockCache
typedef LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCache >	LuminosityBlockContext
typedef CacheTypes::LuminosityBlockSummaryCache	LuminosityBlockSummaryCache
typedef CacheTypes::RunCache	RunCache
typedef RunContextT< RunCache, GlobalCache >	RunContext
typedef CacheTypes::RunSummaryCache	RunSummaryCache
Public Types inherited from edm::stream::EDProducerBase
typedef EDProducerAdaptorBase	ModuleType
Public Types inherited from edm::ProducerBase
typedef ProductRegistryHelper::TypeLabelList	TypeLabelList
Public Types inherited from edm::EDConsumerBase
typedef ProductLabels	Labels
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

Description: [one line class summary]

Implementation: [Notes on implementation]

Definition at line 42 of file HGCalElectronIDValueMapProducer.cc.

Constructor & Destructor Documentation

HGCalElectronIDValueMapProducer::HGCalElectronIDValueMapProducer ( const edm::ParameterSet &  iConfig )

explicit

Definition at line 108 of file HGCalElectronIDValueMapProducer.cc.

References edm::EDConsumerBase::consumesCollector(), eIDHelper_, crabWrapper::key, maps_, and valuesProduced_.

                                                                                               :
  electronsToken_(consumes<edm::View<reco::GsfElectron>>(iConfig.getParameter<edm::InputTag>("electrons"))),
  radius_(iConfig.getParameter<double>("pcaRadius"))
{
  for(const auto& key : valuesProduced_) {
    maps_[key] = {};
    produces<edm::ValueMap<float>>(key);
  }

  eIDHelper_.reset(new HGCalEgammaIDHelper(iConfig, consumesCollector()));
}

HGCalElectronIDValueMapProducer::~HGCalElectronIDValueMapProducer ( )

override

Definition at line 121 of file HGCalElectronIDValueMapProducer.cc.

{
}

Member Function Documentation

void HGCalElectronIDValueMapProducer::beginStream ( edm::StreamID  )

overrideprivatevirtual

Reimplemented from edm::stream::EDProducerBase.

Definition at line 248 of file HGCalElectronIDValueMapProducer.cc.

{
}

void HGCalElectronIDValueMapProducer::endStream ( )

overrideprivatevirtual

Reimplemented from edm::stream::EDProducerBase.

Definition at line 254 of file HGCalElectronIDValueMapProducer.cc.

                                           {
}

void HGCalElectronIDValueMapProducer::fillDescriptions ( edm::ConfigurationDescriptions &  descriptions )

static

Definition at line 259 of file HGCalElectronIDValueMapProducer.cc.

References edm::ConfigurationDescriptions::add(), edm::ParameterSetDescription::add(), DEFINE_FWK_MODULE, and valuesProduced_.

                                                                                            {
  // Auto-generate hgcalElectronIDValueMap_cfi
  edm::ParameterSetDescription desc;
  desc.add<edm::InputTag>("electrons", edm::InputTag("ecalDrivenGsfElectronsFromMultiCl"));
  desc.add<double>("pcaRadius", 3.0);
  desc.add<std::vector<std::string>>("variables", valuesProduced_);
  desc.add<std::vector<double>>("dEdXWeights")->setComment("This must be copied from dEdX_weights in RecoLocalCalo.HGCalRecProducers.HGCalRecHit_cfi");
  desc.add<unsigned int>("isoNRings", 5);
  desc.add<double>("isoDeltaR", 0.15);
  desc.add<double>("isoDeltaRmin", 0.0);
  desc.add<edm::InputTag>("EERecHits", edm::InputTag("HGCalRecHit","HGCEERecHits"));
  desc.add<edm::InputTag>("FHRecHits", edm::InputTag("HGCalRecHit","HGCHEFRecHits"));
  desc.add<edm::InputTag>("BHRecHits", edm::InputTag("HGCalRecHit","HGCHEBRecHits"));
  descriptions.add("hgcalElectronIDValueMap", desc);
}

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

overrideprivate

Definition at line 128 of file HGCalElectronIDValueMapProducer.cc.

References eIDHelper_, metsig::electron, electronsToken_, Exception, objects.autophobj::filler, edm::Event::getByToken(), maps_, eostools::move(), MillePedeFileConverter_cfg::out, edm::Event::put(), radius_, and valuesProduced_.

{
  using namespace edm;

  Handle<edm::View<reco::GsfElectron>> electronsH;
  iEvent.getByToken(electronsToken_, electronsH);

  // Clear previous map
  for(auto&& kv : maps_) {
    kv.second.clear();
    kv.second.reserve(electronsH->size());
  }

  // Set up helper tool
  eIDHelper_->eventInit(iEvent,iSetup);

  for(const auto& electron : *electronsH) {
    if(electron.isEB()) {
      // Fill some dummy value
      for(auto&& kv : maps_) {
    kv.second.push_back(0.);
      }
    }
    else {
      eIDHelper_->computeHGCAL(electron, radius_);

      // check the PCA has worked out
      if (eIDHelper_->sigmaUU() == -1){
      for(auto&& kv : maps_) {
          kv.second.push_back(0.);
      }
      continue;
      }

      hgcal::LongDeps ld(eIDHelper_->energyPerLayer(radius_, true));
      float measuredDepth, expectedDepth, expectedSigma;
      float depthCompatibility = eIDHelper_->clusterDepthCompatibility(ld, measuredDepth, expectedDepth, expectedSigma);

      // Fill here all the ValueMaps from their appropriate functions

      // Energies / PT
      const auto* eleCluster = electron.electronCluster().get();
      const double sinTheta = eleCluster->position().rho() / eleCluster->position().r();
      maps_["ecOrigEt"].push_back(eleCluster->energy() * sinTheta );
      maps_["ecOrigEnergy"].push_back(eleCluster->energy());

      // energies calculated in an cylinder around the axis of the electron cluster
      float ec_tot_energy = ld.energyEE() + ld.energyFH() + ld.energyBH();
      maps_["ecEt"].push_back(ec_tot_energy * sinTheta );
      maps_["ecEnergy"].push_back(ec_tot_energy);
      maps_["ecEnergyEE"].push_back(ld.energyEE());
      maps_["ecEnergyFH"].push_back(ld.energyFH());
      maps_["ecEnergyBH"].push_back(ld.energyBH());

      // Cluster shapes
      // PCA related
      maps_["pcaEig1"].push_back(eIDHelper_->eigenValues()(0));
      maps_["pcaEig2"].push_back(eIDHelper_->eigenValues()(1));
      maps_["pcaEig3"].push_back(eIDHelper_->eigenValues()(2));
      maps_["pcaSig1"].push_back(eIDHelper_->sigmas()(0));
      maps_["pcaSig2"].push_back(eIDHelper_->sigmas()(1));
      maps_["pcaSig3"].push_back(eIDHelper_->sigmas()(2));
      maps_["pcaAxisX"].push_back(eIDHelper_->axis().x());
      maps_["pcaAxisY"].push_back(eIDHelper_->axis().y());
      maps_["pcaAxisZ"].push_back(eIDHelper_->axis().z());
      maps_["pcaPositionX"].push_back(eIDHelper_->barycenter().x());
      maps_["pcaPositionY"].push_back(eIDHelper_->barycenter().y());
      maps_["pcaPositionZ"].push_back(eIDHelper_->barycenter().z());

      // transverse shapes
      maps_["sigmaUU"].push_back(eIDHelper_->sigmaUU());
      maps_["sigmaVV"].push_back(eIDHelper_->sigmaVV());
      maps_["sigmaEE"].push_back(eIDHelper_->sigmaEE());
      maps_["sigmaPP"].push_back(eIDHelper_->sigmaPP());

      // long profile
      maps_["nLayers"].push_back(ld.nLayers());
      maps_["firstLayer"].push_back(ld.firstLayer());
      maps_["lastLayer"].push_back(ld.lastLayer());
      maps_["e4oEtot"].push_back(ld.e4oEtot());
      maps_["layerEfrac10"].push_back(ld.layerEfrac10());
      maps_["layerEfrac90"].push_back(ld.layerEfrac90());

      // depth
      maps_["measuredDepth"].push_back(measuredDepth);
      maps_["expectedDepth"].push_back(expectedDepth);
      maps_["expectedSigma"].push_back(expectedSigma);
      maps_["depthCompatibility"].push_back(depthCompatibility);

      // Isolation
      maps_["caloIsoRing0"].push_back(eIDHelper_->getIsolationRing(0));
      maps_["caloIsoRing1"].push_back(eIDHelper_->getIsolationRing(1));
      maps_["caloIsoRing2"].push_back(eIDHelper_->getIsolationRing(2));
      maps_["caloIsoRing3"].push_back(eIDHelper_->getIsolationRing(3));
      maps_["caloIsoRing4"].push_back(eIDHelper_->getIsolationRing(4));
    }
  }

  // Check we didn't make up a new variable and forget it in valuesProduced_
  if ( maps_.size() != valuesProduced_.size() ) {
    throw cms::Exception("HGCalElectronIDValueMapProducer") << "We have a miscoded value map producer, since map size changed";
  }

  for(auto&& kv : maps_) {
    // Check we didn't forget any values
    if ( kv.second.size() != electronsH->size() ) {
      throw cms::Exception("HGCalElectronIDValueMapProducer") << "We have a miscoded value map producer, since the variable " << kv.first << " wasn't filled.";
    }
    // Do the filling
    auto out = std::make_unique<edm::ValueMap<float>>();
    edm::ValueMap<float>::Filler filler(*out);
    filler.insert(electronsH, kv.second.begin(), kv.second.end());
    filler.fill();
    // and put it into the event
    iEvent.put(std::move(out), kv.first);
  }
}

Member Data Documentation

std::unique_ptr<HGCalEgammaIDHelper> HGCalElectronIDValueMapProducer::eIDHelper_

private

Definition at line 61 of file HGCalElectronIDValueMapProducer.cc.

Referenced by HGCalElectronIDValueMapProducer(), and produce().

edm::EDGetTokenT<edm::View<reco::GsfElectron> > HGCalElectronIDValueMapProducer::electronsToken_

private

Definition at line 55 of file HGCalElectronIDValueMapProducer.cc.

Referenced by produce().

std::map<const std::string, std::vector<float> > HGCalElectronIDValueMapProducer::maps_

private

Definition at line 59 of file HGCalElectronIDValueMapProducer.cc.

Referenced by HGCalElectronIDValueMapProducer(), and produce().

float HGCalElectronIDValueMapProducer::radius_

private

Definition at line 56 of file HGCalElectronIDValueMapProducer.cc.

Referenced by produce().

const std::vector< std::string > HGCalElectronIDValueMapProducer::valuesProduced_

staticprivate

Definition at line 58 of file HGCalElectronIDValueMapProducer.cc.

Referenced by fillDescriptions(), HGCalElectronIDValueMapProducer(), and produce().