#include <SCEnergyCorrectorDRN.h>

Public Member Functions
TritonOutput< float >	getOutput (const TritonOutputMap &iOutput)

void	makeInput (const edm::Event &iEvent, TritonInputMap &iInput, const reco::SuperClusterCollection &inputSCs) const

	SCEnergyCorrectorDRN ()

	SCEnergyCorrectorDRN (const edm::ParameterSet &iConfig, edm::ConsumesCollector cc)

void	setEvent (const edm::Event &e)

void	setEventSetup (const edm::EventSetup &es)

template<edm::Transition tr = edm::Transition::BeginLuminosityBlock>
void	setTokens (const edm::ParameterSet &iConfig, edm::ConsumesCollector cc)

Static Public Member Functions
static void	fillPSetDescription (edm::ParameterSetDescription &desc)

static edm::ParameterSetDescription	makePSetDescription ()

Private Attributes
const CaloGeometry *	caloGeom_

edm::ESGetToken< CaloGeometry, CaloGeometryRecord >	caloGeomToken_

const CaloTopology *	caloTopo_

edm::ESGetToken< CaloTopology, CaloTopologyRecord >	caloTopoToken_

edm::Handle< EcalRecHitCollection >	recHitsEB_

edm::Handle< EcalRecHitCollection >	recHitsEE_

edm::Handle< double >	rhoHandle_

edm::EDGetTokenT< double >	rhoToken_

edm::EDGetTokenT< EcalRecHitCollection >	tokenEBRecHits_

edm::EDGetTokenT< EcalRecHitCollection >	tokenEERecHits_

Detailed Description

Definition at line 47 of file SCEnergyCorrectorDRN.h.

Constructor & Destructor Documentation

◆ SCEnergyCorrectorDRN() [1/2]

SCEnergyCorrectorDRN::SCEnergyCorrectorDRN ( )

Definition at line 23 of file SCEnergyCorrectorDRN.cc.

23 : caloTopo_(nullptr), caloGeom_(nullptr) {}

SCEnergyCorrectorDRN::caloGeom_

const CaloGeometry * caloGeom_

Definition: SCEnergyCorrectorDRN.h:68

SCEnergyCorrectorDRN::caloTopo_

const CaloTopology * caloTopo_

Definition: SCEnergyCorrectorDRN.h:67

◆ SCEnergyCorrectorDRN() [2/2]

SCEnergyCorrectorDRN::SCEnergyCorrectorDRN	(	const edm::ParameterSet &	iConfig,
		edm::ConsumesCollector	cc
	)

Definition at line 25 of file SCEnergyCorrectorDRN.cc.

References gpuPixelDoublets::cc, and setTokens().

     : SCEnergyCorrectorDRN() {
   setTokens(iConfig, cc);
 }

Member Function Documentation

◆ fillPSetDescription()

void SCEnergyCorrectorDRN::fillPSetDescription ( edm::ParameterSetDescription & desc )

static

Definition at line 30 of file SCEnergyCorrectorDRN.cc.

References submitPVResolutionJobs::desc, and ProducerED_cfi::InputTag.

Referenced by makePSetDescription().

                                                                                {
   desc.add<edm::InputTag>("ecalRecHitsEE", edm::InputTag("ecalRecHit", "reducedEcalRecHitsEE"));
   desc.add<edm::InputTag>("ecalRecHitsEB", edm::InputTag("ecalRecHit", "reducedEcalRecHitsEB"));
   desc.add<edm::InputTag>("rhoFastJet", edm::InputTag("fixedGridRhoAll"));
 }

◆ getOutput()

TritonOutput< float > SCEnergyCorrectorDRN::getOutput ( const TritonOutputMap & iOutput )

Definition at line 115 of file SCEnergyCorrectorDRN.cc.

References skim900GeV_StreamA_MinBiasPD_cfg::output1.

Referenced by SCEnergyCorrectorDRNProducer::produce().

                                                                                   {
   //check the results
   const auto& output1 = iOutput.begin()->second;
   // convert from server format
   const auto& serverout = output1.fromServer<float>();
 
   return serverout;
 }

◆ makeInput()

void SCEnergyCorrectorDRN::makeInput	(	const edm::Event &	iEvent,
		TritonInputMap &	iInput,
		const reco::SuperClusterCollection &	inputSCs
	)		const

Definition at line 53 of file SCEnergyCorrectorDRN.cc.

References caloGeom_, E_RANGE, EcalBarrel, DivergingColor::frac, CaloGeometry::getGeometry(), hfClusterShapes_cfi::hits, input1, input2, SCEnergyCorrectorDRNProducer_cfi::inputSCs, dqmiodumpmetadata::n, nHits, position, recHitsEB_, recHitsEE_, RHO_MAX, rhoHandle_, x, X_MAX, X_RANGE, y, Y_MAX, Y_RANGE, z, Z_MAX, and Z_RANGE.

Referenced by SCEnergyCorrectorDRNProducer::acquire().

                                                                                        {
   std::vector<unsigned> nHits;
   nHits.reserve(inputSCs.size());
   unsigned totalHits = 0;
   unsigned n;
   for (const auto& inputSC : inputSCs) {
     n = inputSC.hitsAndFractions().size();
     totalHits += n;
     nHits.push_back(n);
   }
 
   //set shapes
   auto& input1 = iInput.at("x__0");
   input1.setShape(0, totalHits);
   auto data1 = input1.allocate<float>();
   auto& vdata1 = (*data1)[0];
 
   auto& input2 = iInput.at("batch__1");
   input2.setShape(0, totalHits);
   auto data2 = input2.allocate<int64_t>();
   auto& vdata2 = (*data2)[0];
 
   auto& input3 = iInput.at("graphx__2");
   input3.setShape(0, 2 * nHits.size());
   auto data3 = input3.allocate<float>();
   auto& vdata3 = (*data3)[0];
 
   //fill
   unsigned batchNum = 0;
   float En, frac, x, y, z;
   for (const auto& inputSC : inputSCs) {
     const auto& hits = inputSC.hitsAndFractions();
     const bool isEB = hits[0].first.subdetId() == EcalBarrel;
     const auto& recHitsProduct = isEB ? recHitsEB_.product() : recHitsEE_.product();
     for (const auto& hit : hits) {
       En = EcalClusterTools::recHitEnergy(hit.first, recHitsProduct);
       frac = hit.second;
       GlobalPoint position = caloGeom_->getGeometry(hit.first)->getPosition();
       x = (position.x() + X_MAX) / X_RANGE;
       y = (position.y() + Y_MAX) / Y_RANGE;
       z = (position.z() + Z_MAX) / Z_RANGE;
       vdata1.push_back(x);
       vdata1.push_back(y);
       vdata1.push_back(z);
       vdata1.push_back(En * frac / E_RANGE);
       //Triton does not currently support batching for pytorch GNNs
       //We pass batch indices explicitely
       vdata2.push_back(batchNum);
     }
     vdata3.push_back(*rhoHandle_ / RHO_MAX);
     vdata3.push_back(0.0);
     ++batchNum;
   }
 
   // convert to server format
   input1.toServer(data1);
   input2.toServer(data2);
   input3.toServer(data3);
 }

◆ makePSetDescription()

edm::ParameterSetDescription SCEnergyCorrectorDRN::makePSetDescription ( )

static

Definition at line 36 of file SCEnergyCorrectorDRN.cc.

References submitPVResolutionJobs::desc, and fillPSetDescription().

Referenced by SCEnergyCorrectorDRNProducer::fillDescriptions().

                                                                      {
   edm::ParameterSetDescription desc;
   fillPSetDescription(desc);
   return desc;
 }

◆ setEvent()

void SCEnergyCorrectorDRN::setEvent ( const edm::Event & e )

Definition at line 47 of file SCEnergyCorrectorDRN.cc.

References recHitsEB_, recHitsEE_, rhoHandle_, rhoToken_, tokenEBRecHits_, and tokenEERecHits_.

Referenced by SCEnergyCorrectorDRNProducer::acquire().

                                                          {
   event.getByToken(tokenEBRecHits_, recHitsEB_);
   event.getByToken(tokenEERecHits_, recHitsEE_);
   event.getByToken(rhoToken_, rhoHandle_);
 }

◆ setEventSetup()

void SCEnergyCorrectorDRN::setEventSetup ( const edm::EventSetup & es )

Definition at line 42 of file SCEnergyCorrectorDRN.cc.

References caloGeom_, caloGeomToken_, caloTopo_, caloTopoToken_, and edm::EventSetup::getData().

Referenced by SCEnergyCorrectorDRNProducer::beginLuminosityBlock().

                                                                 {
   caloTopo_ = &es.getData(caloTopoToken_);
   caloGeom_ = &es.getData(caloGeomToken_);
 }

◆ setTokens()

template<edm::Transition esTransition>

void SCEnergyCorrectorDRN::setTokens	(	const edm::ParameterSet &	iConfig,
		edm::ConsumesCollector	cc
	)

Definition at line 83 of file SCEnergyCorrectorDRN.h.

References caloGeomToken_, caloTopoToken_, gpuPixelDoublets::cc, edm::ParameterSet::getParameter(), rhoToken_, tokenEBRecHits_, and tokenEERecHits_.

Referenced by SCEnergyCorrectorDRN().

                                                                                             {
   tokenEBRecHits_ = cc.consumes<EcalRecHitCollection>(iConfig.getParameter<edm::InputTag>("ecalRecHitsEB"));
   tokenEERecHits_ = cc.consumes<EcalRecHitCollection>(iConfig.getParameter<edm::InputTag>("ecalRecHitsEE"));
   caloGeomToken_ = cc.esConsumes<CaloGeometry, CaloGeometryRecord, esTransition>();
   caloTopoToken_ = cc.esConsumes<CaloTopology, CaloTopologyRecord, esTransition>();
   rhoToken_ = cc.consumes<double>(iConfig.getParameter<edm::InputTag>("rhoFastJet"));
 }