Inheritance diagram for DRNCorrectionProducerT< T >:

Public Member Functions
void	acquire (edm::Event const &iEvent, edm::EventSetup const &iSetup, Input &input) override

	DRNCorrectionProducerT (const edm::ParameterSet &iConfig)

void	produce (edm::Event &iEvent, const edm::EventSetup &iSetup, Output const &iOutput) override

Public Member Functions inherited from TritonEDProducerT< G, Capabilities >
void	endStream () final

	TritonEDProducerT (edm::ParameterSet const &cfg)

virtual void	tritonEndStream ()

Public Member Functions inherited from SonicEDProducer< TritonClient, edm::GlobalCache< G >, Capabilities... >
void	produce (edm::Event &iEvent, edm::EventSetup const &iSetup) final

virtual void	produce (edm::Event &iEvent, edm::EventSetup const &iSetup, Output const &iOutput)=0

	SonicEDProducer (edm::ParameterSet const &cfg)

	~SonicEDProducer () override=default

Public Member Functions inherited from SonicAcquirer< TritonClient, edm::stream::EDProducer< edm::ExternalWork, Capabilities... > >
void	acquire (edm::Event const &iEvent, edm::EventSetup const &iSetup, edm::WaitingTaskWithArenaHolder holder) final

virtual void	acquire (edm::Event const &iEvent, edm::EventSetup const &iSetup, Input &iInput)=0

void	beginStream (edm::StreamID) override

	SonicAcquirer (edm::ParameterSet const &cfg)

	~SonicAcquirer () override=default

Public Member Functions inherited from edm::stream::EDProducer< edm::ExternalWork, Capabilities... >
	EDProducer ()=default

	EDProducer (const EDProducer &)=delete

bool	hasAbilityToProduceInBeginLumis () const final

bool	hasAbilityToProduceInBeginProcessBlocks () const final

bool	hasAbilityToProduceInBeginRuns () const final

bool	hasAbilityToProduceInEndLumis () const final

bool	hasAbilityToProduceInEndProcessBlocks () const final

bool	hasAbilityToProduceInEndRuns () const final

const EDProducer &	operator= (const EDProducer &)=delete

Static Public Member Functions
static void	fillDescriptions (edm::ConfigurationDescriptions &descriptions)

Static Public Member Functions inherited from TritonEDProducerT< G, Capabilities >
static void	globalEndJob (G *)

static std::unique_ptr< G >	initializeGlobalCache (edm::ParameterSet const &pset)

Private Member Functions
bool	isEB (const T &part)

bool	isEE (const T &part)

bool	skip (const T &part)

Private Attributes
edm::EDGetTokenT< EcalRecHitCollection >	EBRecHitsToken_

edm::EDGetTokenT< EcalRecHitCollection >	EERecHitsToken_

edm::EDGetTokenT< EcalRecHitCollection >	ESRecHitsToken_

edm::ESGetToken< CaloGeometry, CaloGeometryRecord >	geomToken_

size_t	nPart_

size_t	nValidPart_

edm::EDGetTokenT< edm::View< T > >	particleToken_

edm::ESGetToken< EcalPedestals, EcalPedestalsRcd >	pedToken_

edm::EDGetTokenT< double >	rhoToken_

Additional Inherited Members
Public Types inherited from SonicEDProducer< TritonClient, edm::GlobalCache< G >, Capabilities... >
typedef TritonClient ::Output	Output

Public Types inherited from SonicAcquirer< TritonClient, edm::stream::EDProducer< edm::ExternalWork, Capabilities... > >
typedef TritonClient ::Input	Input

Public Types inherited from edm::stream::EDProducer< edm::ExternalWork, Capabilities... >
using	CacheTypes = CacheContexts< T... >

using	GlobalCache = typename CacheTypes::GlobalCache

using	HasAbility = AbilityChecker< T... >

using	InputProcessBlockCache = typename CacheTypes::InputProcessBlockCache

using	LuminosityBlockCache = typename CacheTypes::LuminosityBlockCache

using	LuminosityBlockContext = LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCache >

using	LuminosityBlockSummaryCache = typename CacheTypes::LuminosityBlockSummaryCache

using	RunCache = typename CacheTypes::RunCache

using	RunContext = RunContextT< RunCache, GlobalCache >

using	RunSummaryCache = typename CacheTypes::RunSummaryCache

Protected Member Functions inherited from SonicAcquirer< TritonClient, edm::stream::EDProducer< edm::ExternalWork, Capabilities... > >
void	makeClient ()

Protected Attributes inherited from SonicAcquirer< TritonClient, edm::stream::EDProducer< edm::ExternalWork, Capabilities... > >
std::unique_ptr< TritonClient >	client_

edm::ParameterSet	clientPset_

std::string	debugName_

std::chrono::time_point< std::chrono::high_resolution_clock >	t_dispatch_

bool	verbose_

Detailed Description

template<typename T>
class DRNCorrectionProducerT< T >

Definition at line 96 of file DRNCorrectionProducerT.cc.

Constructor & Destructor Documentation

◆ DRNCorrectionProducerT()

template<typename T >

DRNCorrectionProducerT< T >::DRNCorrectionProducerT ( const edm::ParameterSet & iConfig )

explicit

Definition at line 123 of file DRNCorrectionProducerT.cc.

     : TritonEDProducer<>(iConfig),
       particleToken_(consumes(iConfig.getParameter<edm::InputTag>("particleSource"))),
       rhoToken_(consumes(iConfig.getParameter<edm::InputTag>("rhoName"))),
       EBRecHitsToken_(consumes<EcalRecHitCollection>(iConfig.getParameter<edm::InputTag>("reducedEcalRecHitsEB"))),
       EERecHitsToken_(consumes<EcalRecHitCollection>(iConfig.getParameter<edm::InputTag>("reducedEcalRecHitsEE"))),
       ESRecHitsToken_(consumes<EcalRecHitCollection>(iConfig.getParameter<edm::InputTag>("reducedEcalRecHitsES"))),
       pedToken_(esConsumes()),
       geomToken_(esConsumes()) {
   produces<edm::ValueMap<std::pair<float, float>>>();
 }

Member Function Documentation

◆ acquire()

template<typename T >

void DRNCorrectionProducerT< T >::acquire	(	edm::Event const &	iEvent,
		edm::EventSetup const &	iSetup,
		Input &	input
	)

override

Definition at line 156 of file DRNCorrectionProducerT.cc.

References DetId::Ecal, EcalBarrel, EcalEndcap, EcalPreshower, relativeConstraints::geom, edm::EventSetup::getData(), CaloSubdetectorGeometry::getGeometry(), iEvent, EcalRecHit::kESGood, EcalRecHit::kGood, EcalRecHit::kHasSwitchToGain1, EcalRecHit::kHasSwitchToGain6, EcalRecHit::kOutOfTime, EcalRecHit::kPoorCalib, sistrip::NOISE_MIN, DetId::rawId(), EcalDeadCellBoundaryEnergyFilter_cfi::recHitsEB, EcalDeadCellBoundaryEnergyFilter_cfi::recHitsEE, MuonErrorMatrixAdjuster_cfi::rescale, ZElectronSkim_cff::rho, optionsL1T::skip, and Z_RANGE.

                                                                                                           {
   /*
    * Get products from event and event setup
    */
   const auto& particles_ = iEvent.getHandle(particleToken_);
   float rho = iEvent.get(rhoToken_);
 
   const auto& ped = &iSetup.getData(pedToken_);
   const auto& geo = &iSetup.getData(geomToken_);
 
   const CaloSubdetectorGeometry* ecalEBGeom =
       static_cast<const CaloSubdetectorGeometry*>(geo->getSubdetectorGeometry(DetId::Ecal, EcalBarrel));
   const CaloSubdetectorGeometry* ecalEEGeom =
       static_cast<const CaloSubdetectorGeometry*>(geo->getSubdetectorGeometry(DetId::Ecal, EcalEndcap));
   const CaloSubdetectorGeometry* ecalESGeom =
       static_cast<const CaloSubdetectorGeometry*>(geo->getSubdetectorGeometry(DetId::Ecal, EcalPreshower));
 
   const auto& recHitsEB = iEvent.get(EBRecHitsToken_);
   const auto& recHitsEE = iEvent.get(EERecHitsToken_);
   const auto& recHitsES = iEvent.get(ESRecHitsToken_);
 
   nPart_ = particles_->size();
 
   if (nPart_ == 0) {
     client_->setBatchSize(0);
     return;
   } else {
     client_->setBatchSize(1);
   }
 
   /*
    * Determine how many particles, how many RecHits there are in each subdetector
    */
   unsigned nHitsECAL = 0, nHitsES = 0;
   nValidPart_ = 0;
   for (auto& part : *particles_) {
     const reco::SuperClusterRef& sc = part.superCluster();
 
     if (skip(part))
       continue;
 
     nHitsECAL += sc->hitsAndFractions().size();
 
     for (auto iES = sc->preshowerClustersBegin(); iES != sc->preshowerClustersEnd(); ++iES) {
       nHitsES += (*iES)->hitsAndFractions().size();
     }
 
     ++nValidPart_;
   }
 
   /*
    * Allocate DRN inputs ({SB} is one of ECAL, ES):
    * x{SB}: (x, y, z, energy, [noise]) continuous-valued inputs per RecHit
    * f{SB}: (flagVal) integer denoting RecHit flag values
    * gainECAL: (gain) integer in (0, 1, 2) denoting gain value
    * gx: (rho, H/E) additional high-level features.
    * batch{SB}: graph models require explicitely passing the particle index for each vertex
    */
   auto& inputxECAL = iInput.at("xECAL");
   inputxECAL.setShape(0, nHitsECAL);
   auto dataxECAL = inputxECAL.allocate<float>();
   auto& vdataxECAL = (*dataxECAL)[0];
 
   auto& inputfECAL = iInput.at("fECAL");
   inputfECAL.setShape(0, nHitsECAL);
   auto datafECAL = inputfECAL.allocate<int64_t>();
   auto& vdatafECAL = (*datafECAL)[0];
 
   auto& inputGainECAL = iInput.at("gainECAL");
   inputGainECAL.setShape(0, nHitsECAL);
   auto dataGainECAL = inputGainECAL.allocate<int64_t>();
   auto& vdataGainECAL = (*dataGainECAL)[0];
 
   auto& inputGx = iInput.at("gx");
   inputGx.setShape(0, nValidPart_);
   auto dataGx = inputGx.allocate<float>();
   auto& vdataGx = (*dataGx)[0];
 
   auto& inputBatchECAL = iInput.at("batchECAL");
   inputBatchECAL.setShape(0, nHitsECAL);
   auto dataBatchECAL = inputBatchECAL.allocate<int64_t>();
   auto& vdataBatchECAL = (*dataBatchECAL)[0];
 
   auto& inputxES = iInput.at("xES");
   inputxES.setShape(0, nHitsES);
   auto dataxES = inputxES.allocate<float>();
   auto& vdataxES = (*dataxES)[0];
 
   auto& inputfES = iInput.at("fES");
   inputfES.setShape(0, nHitsES);
   auto datafES = inputfES.allocate<int64_t>();
   auto& vdatafES = (*datafES)[0];
 
   auto& inputBatchES = iInput.at("batchES");
   inputBatchES.setShape(0, nHitsES);
   auto dataBatchES = inputBatchES.allocate<int64_t>();
   auto& vdataBatchES = (*dataBatchES)[0];
 
   /*
    * Fill input tensors by iterating over particles...
    */
   int64_t partNum = 0;
   std::shared_ptr<const CaloCellGeometry> geom;
   for (auto& part : *particles_) {
     const reco::SuperClusterRef& sc = part.superCluster();
 
     if (skip(part))
       continue;
 
     std::vector<std::pair<DetId, float>> hitsAndFractions = sc->hitsAndFractions();
     EcalRecHitCollection::const_iterator hit;
 
     //iterate over ECAL hits...
     for (const auto& detitr : hitsAndFractions) {
       DetId id = detitr.first.rawId();
       if (isEB(part)) {
         geom = ecalEBGeom->getGeometry(id);
         hit = recHitsEB.find(detitr.first);
       } else {
         geom = ecalEEGeom->getGeometry(id);
         hit = recHitsEE.find(detitr.first);
       }
 
       //fill xECAL
       auto pos = geom->getPosition();
       vdataxECAL.push_back(rescale(pos.x(), XY_MIN, XY_RANGE));
       vdataxECAL.push_back(rescale(pos.y(), XY_MIN, XY_RANGE));
       vdataxECAL.push_back(rescale(pos.z(), Z_MIN, Z_RANGE));
       vdataxECAL.push_back(rescale(hit->energy() * detitr.second, ECAL_MIN, ECAL_RANGE));
       vdataxECAL.push_back(rescale(ped->find(detitr.first)->rms(1), NOISE_MIN, NOISE_RANGE));
 
       //fill fECAL
       int64_t flagVal = 0;
       if (hit->checkFlag(EcalRecHit::kGood))
         flagVal += 1;
       if (hit->checkFlag(EcalRecHit::kOutOfTime))
         flagVal += 2;
       if (hit->checkFlag(EcalRecHit::kPoorCalib))
         flagVal += 4;
 
       vdatafECAL.push_back(flagVal);
 
       //fill gain
       int64_t gainVal = 0;
       if (hit->checkFlag(EcalRecHit::kHasSwitchToGain6))
         gainVal = 1;
       else if (hit->checkFlag(EcalRecHit::kHasSwitchToGain1))
         gainVal = 0;
       else
         gainVal = 2;
 
       vdataGainECAL.push_back(gainVal);
 
       //fill batch number
       vdataBatchECAL.push_back(partNum);
     }  //end iterate over ECAL hits
 
     //iterate over ES clusters...
     for (auto iES = sc->preshowerClustersBegin(); iES != sc->preshowerClustersEnd(); ++iES) {
       for (const auto& ESitr : (*iES)->hitsAndFractions()) {  //iterate over ES hits
         hit = recHitsES.find(ESitr.first);
         geom = ecalESGeom->getGeometry(ESitr.first);
         auto& pos = geom->getPosition();
 
         //fill xES
         vdataxES.push_back(rescale(pos.x(), XY_MIN, XY_RANGE));
         vdataxES.push_back(rescale(pos.y(), XY_MIN, XY_RANGE));
         vdataxES.push_back(rescale(pos.z(), Z_MIN, Z_RANGE));
         vdataxES.push_back(rescale(hit->energy(), ES_MIN, ES_RANGE));
 
         //fill fES
         int64_t flagVal = 0;
         if (hit->checkFlag(EcalRecHit::kESGood))
           flagVal += 1;
 
         vdatafES.push_back(flagVal);
 
         //fill batchES
         vdataBatchES.push_back(partNum);
       }  //end iterate over ES hits
     }    //end iterate over ES clusters
 
     //fill gx
     vdataGx.push_back(rescale(rho, RHO_MIN, RHO_RANGE));
     vdataGx.push_back(rescale(part.hadronicOverEm(), HOE_MIN, HOE_RANGE));
 
     //increment particle number
     ++partNum;
   }  // end iterate over particles
 
   /*
    * Convert input tensors to server data format
    */
   inputxECAL.toServer(dataxECAL);
   inputfECAL.toServer(datafECAL);
   inputGainECAL.toServer(dataGainECAL);
   inputBatchECAL.toServer(dataBatchECAL);
 
   inputGx.toServer(dataGx);
 
   inputxES.toServer(dataxES);
   inputfES.toServer(datafES);
   inputBatchES.toServer(dataBatchES);
 }

◆ fillDescriptions()

template<typename T >

void DRNCorrectionProducerT< T >::fillDescriptions ( edm::ConfigurationDescriptions & descriptions )

static

Definition at line 403 of file DRNCorrectionProducerT.cc.

References edm::ConfigurationDescriptions::addWithDefaultLabel(), submitPVResolutionJobs::desc, TritonClient::fillPSetDescription(), and HLT_2022v14_cff::InputTag.

                                                                                            {
   edm::ParameterSetDescription desc;
   TritonClient::fillPSetDescription(desc);
   desc.add<edm::InputTag>("particleSource");
   desc.add<edm::InputTag>("rhoName");
   desc.add<edm::InputTag>("reducedEcalRecHitsEB", edm::InputTag("reducedEcalRecHitsEB"));
   desc.add<edm::InputTag>("reducedEcalRecHitsEE", edm::InputTag("reducedEcalRecHitsEE"));
   desc.add<edm::InputTag>("reducedEcalRecHitsES", edm::InputTag("reducedEcalRecHitsES"));
   descriptions.addWithDefaultLabel(desc);
 }

◆ isEB()

template<typename T >

bool DRNCorrectionProducerT< T >::isEB ( const T & part )

private

Definition at line 136 of file DRNCorrectionProducerT.cc.

References EcalBarrel.

Referenced by Electron.Electron::absIsoWithFSR().

                                                   {
   return part.superCluster()->seed()->hitsAndFractions().at(0).first.subdetId() == EcalBarrel;
 }

◆ isEE()

template<typename T >

bool DRNCorrectionProducerT< T >::isEE ( const T & part )

private

Definition at line 141 of file DRNCorrectionProducerT.cc.

References EcalEndcap.

                                                   {
   return part.superCluster()->seed()->hitsAndFractions().at(0).first.subdetId() == EcalEndcap;
 }

◆ produce()

template<typename T >

void DRNCorrectionProducerT< T >::produce	(	edm::Event &	iEvent,
		const edm::EventSetup &	iSetup,
		Output const &	iOutput
	)

override

Definition at line 362 of file DRNCorrectionProducerT.cc.

References pfMETCorrectionType0_cfi::correction, f, trigObjTnPSource_cfi::filler, mps_fire::i, iEvent, eostools::move(), amptDefaultParameters_cff::mu, MillePedeFileConverter_cfg::out, L1TObjectsTimingClient_cff::resolution, and optionsL1T::skip.

                                                                                                             {
   const auto& particles_ = iEvent.getHandle(particleToken_);
 
   std::vector<std::pair<float, float>> corrections;
   corrections.reserve(nPart_);
 
   //if there are no particles, the fromServer() call will fail
   //but we can just put() an empty valueMap
   if (nPart_) {
     const auto& muOut = iOutput.at("mu").fromServer<float>();
     const auto& sigmaOut = iOutput.at("sigma").fromServer<float>();
 
     unsigned i = 0;
     float mu, sigma, Epred, sigmaPred, rawE;
     for (unsigned iPart = 0; iPart < nPart_; ++iPart) {
       const auto& part = particles_->at(iPart);
       if (!skip(part)) {
         mu = correction(muOut[0][0 + 6 * i]);
         sigma = resolution(sigmaOut[0][0 + 5 * i]);
         ++i;
 
         rawE = part.superCluster()->rawEnergy();
         Epred = mu * rawE;
         sigmaPred = sigma * rawE;
         corrections.emplace_back(Epred, sigmaPred);
       } else {
         corrections.emplace_back(-1.0f, -1.0f);
       }
     }
   }
 
   //fill
   auto out = std::make_unique<edm::ValueMap<std::pair<float, float>>>();
   edm::ValueMap<std::pair<float, float>>::Filler filler(*out);
   filler.insert(particles_, corrections.begin(), corrections.end());
   filler.fill();
 
   iEvent.put(std::move(out));
 }

◆ skip()

template<typename T >

bool DRNCorrectionProducerT< T >::skip ( const T & part )

private

Definition at line 146 of file DRNCorrectionProducerT.cc.

                                                   {
   /*
    * Separated out from acquire() and produce() to ensure that skipping check is identical in both
    * N.B. in MiniAOD there are sometimes particles with no RecHits
    * We can not apply our regression to these, so we skip them
    */
   return (!isEB(part) && !isEE(part)) || part.superCluster()->hitsAndFractions().empty();
 }

Member Data Documentation

◆ EBRecHitsToken_

template<typename T >

edm::EDGetTokenT<EcalRecHitCollection> DRNCorrectionProducerT< T >::EBRecHitsToken_

private

Definition at line 110 of file DRNCorrectionProducerT.cc.

◆ EERecHitsToken_

template<typename T >

edm::EDGetTokenT<EcalRecHitCollection> DRNCorrectionProducerT< T >::EERecHitsToken_

private

Definition at line 110 of file DRNCorrectionProducerT.cc.

◆ ESRecHitsToken_

template<typename T >

edm::EDGetTokenT<EcalRecHitCollection> DRNCorrectionProducerT< T >::ESRecHitsToken_

private

Definition at line 110 of file DRNCorrectionProducerT.cc.

◆ geomToken_

template<typename T >

edm::ESGetToken<CaloGeometry, CaloGeometryRecord> DRNCorrectionProducerT< T >::geomToken_

private

Definition at line 113 of file DRNCorrectionProducerT.cc.

◆ nPart_

template<typename T >

size_t DRNCorrectionProducerT< T >::nPart_

private

Definition at line 115 of file DRNCorrectionProducerT.cc.

◆ nValidPart_

template<typename T >

size_t DRNCorrectionProducerT< T >::nValidPart_

private

Definition at line 115 of file DRNCorrectionProducerT.cc.

◆ particleToken_

template<typename T >

edm::EDGetTokenT<edm::View<T> > DRNCorrectionProducerT< T >::particleToken_

private

Definition at line 106 of file DRNCorrectionProducerT.cc.

◆ pedToken_

template<typename T >

edm::ESGetToken<EcalPedestals, EcalPedestalsRcd> DRNCorrectionProducerT< T >::pedToken_

private

Definition at line 112 of file DRNCorrectionProducerT.cc.

◆ rhoToken_

template<typename T >

edm::EDGetTokenT<double> DRNCorrectionProducerT< T >::rhoToken_

private

Definition at line 108 of file DRNCorrectionProducerT.cc.

Public Member Functions

Static Public Member Functions

Private Member Functions

Private Attributes

Additional Inherited Members

Detailed Description

template<typename T> class DRNCorrectionProducerT< T >

Constructor & Destructor Documentation

◆ DRNCorrectionProducerT()

Member Function Documentation

◆ acquire()

◆ fillDescriptions()

◆ isEB()

◆ isEE()

◆ produce()

◆ skip()

Member Data Documentation

◆ EBRecHitsToken_

◆ EERecHitsToken_

◆ ESRecHitsToken_

◆ geomToken_

◆ nPart_

◆ nValidPart_

◆ particleToken_

◆ pedToken_

◆ rhoToken_

template<typename T>
class DRNCorrectionProducerT< T >