Functions
int	argMax (std::vector< float > const &vec)

std::array< float, NUM_ELEMENT_FEATURES >	getElementProperties (const reco::PFBlockElement &orig)

const std::vector< const reco::PFBlockElement * >	getPFElements (const reco::PFBlockCollection &blocks)

reco::PFCandidate	makeCandidate (int pred_pid, int pred_charge, float pred_pt, float pred_eta, float pred_sin_phi, float pred_cos_phi, float pred_e)

float	normalize (float in)

void	setCandidateRefs (reco::PFCandidate &cand, const std::vector< const reco::PFBlockElement *> elems, size_t ielem_originator)

Variables
static constexpr int	BATCH_SIZE = 1

static const std::map< int, int >	elem_type_encoding

static constexpr unsigned int	IDX_CHARGE = 8

static constexpr unsigned int	IDX_CLASS = 7

static constexpr unsigned int	IDX_COS_PHI = 12

static constexpr unsigned int	IDX_ENERGY = 13

static constexpr unsigned int	IDX_ETA = 10

static constexpr unsigned int	IDX_PT = 9

static constexpr unsigned int	IDX_SIN_PHI = 11

static constexpr int	LSH_BIN_SIZE = 64

static constexpr unsigned int	NUM_ELEMENT_FEATURES = 25

static constexpr int	NUM_MAX_ELEMENTS_BATCH = 200 * LSH_BIN_SIZE

static constexpr unsigned int	NUM_OUTPUT_FEATURES = 14

static const std::vector< int >	pdgid_encoding = {0, 211, 130, 1, 2, 22, 11, 13}

static constexpr float	PI_MASS = 0.13957

Function Documentation

◆ argMax()

int reco::mlpf::argMax ( std::vector< float > const & vec )

Definition at line 216 of file MLPFModel.cc.

References HLT_2024v13_cff::distance.

Referenced by MLPFProducer::produce().

                                           {
     return static_cast<int>(std::distance(vec.begin(), max_element(vec.begin(), vec.end())));
   }

◆ getElementProperties()

std::array< float, NUM_ELEMENT_FEATURES > reco::mlpf::getElementProperties ( const reco::PFBlockElement & orig )

Definition at line 17 of file MLPFModel.cc.

Referenced by MLPFProducer::produce().

                                                                                              {
     const auto type = orig.type();
     float pt = 0.0;
     float deltap = 0.0;
     float sigmadeltap = 0.0;
     float px = 0.0;
     float py = 0.0;
     float pz = 0.0;
     float eta = 0.0;
     float phi = 0.0;
     float energy = 0.0;
     float corr_energy = 0.0;
     float trajpoint = 0.0;
     float eta_ecal = 0.0;
     float phi_ecal = 0.0;
     float eta_hcal = 0.0;
     float phi_hcal = 0.0;
     float charge = 0;
     float layer = 0;
     float depth = 0;
     float muon_dt_hits = 0.0;
     float muon_csc_hits = 0.0;
     float muon_type = 0.0;
     float cluster_flags = 0.0;
     float gsf_electronseed_trkorecal = 0.0;
     float num_hits = 0.0;
 
     if (type == reco::PFBlockElement::TRACK) {
       const auto& matched_pftrack = orig.trackRefPF();
       if (matched_pftrack.isNonnull()) {
         const auto& atECAL = matched_pftrack->extrapolatedPoint(reco::PFTrajectoryPoint::ECALShowerMax);
         const auto& atHCAL = matched_pftrack->extrapolatedPoint(reco::PFTrajectoryPoint::HCALEntrance);
         if (atHCAL.isValid()) {
           eta_hcal = atHCAL.positionREP().eta();
           phi_hcal = atHCAL.positionREP().phi();
         }
         if (atECAL.isValid()) {
           eta_ecal = atECAL.positionREP().eta();
           phi_ecal = atECAL.positionREP().phi();
         }
       }
       const auto& ref = ((const reco::PFBlockElementTrack*)&orig)->trackRef();
       pt = ref->pt();
       px = ref->px();
       py = ref->py();
       pz = ref->pz();
       eta = ref->eta();
       phi = ref->phi();
       energy = ref->p();
       charge = ref->charge();
       num_hits = ref->recHitsSize();
 
       reco::MuonRef muonRef = orig.muonRef();
       if (muonRef.isNonnull()) {
         reco::TrackRef standAloneMu = muonRef->standAloneMuon();
         if (standAloneMu.isNonnull()) {
           muon_dt_hits = standAloneMu->hitPattern().numberOfValidMuonDTHits();
           muon_csc_hits = standAloneMu->hitPattern().numberOfValidMuonCSCHits();
         }
         muon_type = muonRef->type();
       }
 
     } else if (type == reco::PFBlockElement::BREM) {
       const auto* orig2 = (const reco::PFBlockElementBrem*)&orig;
       const auto& ref = orig2->GsftrackRef();
       trajpoint = orig2->indTrajPoint();
       if (ref.isNonnull()) {
         deltap = orig2->DeltaP();
         sigmadeltap = orig2->SigmaDeltaP();
         pt = ref->pt();
         px = ref->px();
         py = ref->py();
         pz = ref->pz();
         eta = ref->eta();
         phi = ref->phi();
         energy = ref->p();
         charge = ref->charge();
       }
 
       const auto& gsfextraref = ref->extra();
       if (gsfextraref.isAvailable() && gsfextraref->seedRef().isAvailable()) {
         reco::ElectronSeedRef seedref = gsfextraref->seedRef().castTo<reco::ElectronSeedRef>();
         if (seedref.isAvailable()) {
           if (seedref->isEcalDriven()) {
             gsf_electronseed_trkorecal = 1.0;
           } else if (seedref->isTrackerDriven()) {
             gsf_electronseed_trkorecal = 2.0;
           }
         }
       }
 
     } else if (type == reco::PFBlockElement::GSF) {
       //requires to keep GsfPFRecTracks
       const auto* orig2 = (const reco::PFBlockElementGsfTrack*)&orig;
       const auto& vec = orig2->Pin();
       pt = vec.pt();
       px = vec.px();
       py = vec.py();
       pz = vec.pz();
       eta = vec.eta();
       phi = vec.phi();
       energy = vec.energy();
 
       const auto& vec2 = orig2->Pout();
       eta_ecal = vec2.eta();
       phi_ecal = vec2.phi();
 
       if (!orig2->GsftrackRefPF().isNull()) {
         charge = orig2->GsftrackRefPF()->charge();
         num_hits = orig2->GsftrackRefPF()->PFRecBrem().size();
       }
 
       const auto& ref = orig2->GsftrackRef();
 
       const auto& gsfextraref = ref->extra();
       if (gsfextraref.isAvailable() && gsfextraref->seedRef().isAvailable()) {
         reco::ElectronSeedRef seedref = gsfextraref->seedRef().castTo<reco::ElectronSeedRef>();
         if (seedref.isAvailable()) {
           if (seedref->isEcalDriven()) {
             gsf_electronseed_trkorecal = 1.0;
           } else if (seedref->isTrackerDriven()) {
             gsf_electronseed_trkorecal = 2.0;
           }
         }
       };
 
     } else if (type == reco::PFBlockElement::ECAL || type == reco::PFBlockElement::PS1 ||
                type == reco::PFBlockElement::PS2 || type == reco::PFBlockElement::HCAL ||
                type == reco::PFBlockElement::HO || type == reco::PFBlockElement::HFHAD ||
                type == reco::PFBlockElement::HFEM) {
       const auto& ref = ((const reco::PFBlockElementCluster*)&orig)->clusterRef();
       if (ref.isNonnull()) {
         cluster_flags = ref->flags();
         eta = ref->eta();
         phi = ref->phi();
         pt = ref->pt();
         px = ref->position().x();
         py = ref->position().y();
         pz = ref->position().z();
         energy = ref->energy();
         corr_energy = ref->correctedEnergy();
         layer = ref->layer();
         depth = ref->depth();
         num_hits = ref->recHitFractions().size();
       }
     } else if (type == reco::PFBlockElement::SC) {
       const auto& clref = ((const reco::PFBlockElementSuperCluster*)&orig)->superClusterRef();
       if (clref.isNonnull()) {
         cluster_flags = clref->flags();
         eta = clref->eta();
         phi = clref->phi();
         px = clref->position().x();
         py = clref->position().y();
         pz = clref->position().z();
         energy = clref->energy();
         num_hits = clref->clustersSize();
       }
     }
 
     float typ_idx = static_cast<float>(elem_type_encoding.at(orig.type()));
 
     //Must be the same order as in tf_model.py
     return {{typ_idx,
              pt,
              eta,
              phi,
              energy,
              layer,
              depth,
              charge,
              trajpoint,
              eta_ecal,
              phi_ecal,
              eta_hcal,
              phi_hcal,
              muon_dt_hits,
              muon_csc_hits,
              muon_type,
              px,
              py,
              pz,
              deltap,
              sigmadeltap,
              gsf_electronseed_trkorecal,
              num_hits,
              cluster_flags,
              corr_energy}};
   }

◆ getPFElements()

const std::vector< const reco::PFBlockElement * > reco::mlpf::getPFElements ( const reco::PFBlockCollection & blocks )

Definition at line 263 of file MLPFModel.cc.

References groupFilesInBlocks::block, gather_cfg::blocks, and NUM_MAX_ELEMENTS_BATCH.

Referenced by MLPFProducer::produce().

                                                                                                 {
     std::vector<reco::PFCandidate> pOutputCandidateCollection;
 
     std::vector<const reco::PFBlockElement*> all_elements;
     for (const auto& block : blocks) {
       const auto& elems = block.elements();
       for (const auto& elem : elems) {
         if (all_elements.size() < NUM_MAX_ELEMENTS_BATCH) {
           all_elements.push_back(&elem);
         } else {
           //model needs to be created with a bigger LSH codebook size
           edm::LogError("MLPFProducer") << "too many input PFElements for predefined model size: " << elems.size();
           break;
         }
       }
     }
     return all_elements;
   }

◆ makeCandidate()

reco::PFCandidate reco::mlpf::makeCandidate	(	int	pred_pid,
		int	pred_charge,
		float	pred_pt,
		float	pred_eta,
		float	pred_sin_phi,
		float	pred_cos_phi,
		float	pred_e
	)

Definition at line 220 of file MLPFModel.cc.

References reco::charge(), reco::PFCandidate::e, reco::PFCandidate::egamma_HF, reco::PFCandidate::gamma, reco::PFCandidate::h, reco::PFCandidate::h0, reco::PFCandidate::h_HF, reco::PFCandidate::mu, PbPb_ZMuSkimMuonDPG_cff::particleType, PI_MASS, and reco::PFCandidate::X.

Referenced by MLPFProducer::produce().

                                                 {
     float pred_phi = std::atan2(pred_sin_phi, pred_cos_phi);
 
     //set the charge to +1 or -1 for PFCandidates that are charged, according to the sign of the predicted charge
     reco::PFCandidate::Charge charge = 0;
     if (pred_pid == 11 || pred_pid == 13 || pred_pid == 211) {
       charge = pred_charge > 0 ? +1 : -1;
     }
 
     math::PtEtaPhiELorentzVectorD p4(pred_pt, pred_eta, pred_phi, pred_e);
 
     reco::PFCandidate::ParticleType particleType(reco::PFCandidate::X);
     if (pred_pid == 211)
       particleType = reco::PFCandidate::h;
     else if (pred_pid == 130)
       particleType = reco::PFCandidate::h0;
     else if (pred_pid == 22)
       particleType = reco::PFCandidate::gamma;
     else if (pred_pid == 11)
       particleType = reco::PFCandidate::e;
     else if (pred_pid == 13)
       particleType = reco::PFCandidate::mu;
     else if (pred_pid == 1)
       particleType = reco::PFCandidate::h_HF;
     else if (pred_pid == 2)
       particleType = reco::PFCandidate::egamma_HF;
 
     reco::PFCandidate cand(charge, math::XYZTLorentzVector(p4.X(), p4.Y(), p4.Z(), p4.E()), particleType);
     cand.setMass(0.0);
     if (pred_pid == 211)
       cand.setMass(PI_MASS);
     //cand.setPdgId(pred_pid);
     //cand.setCharge(charge);
 
     return cand;
   }

◆ normalize()

float reco::mlpf::normalize ( float in )

Definition at line 207 of file MLPFModel.cc.

References funct::abs(), recoMuon::in, and edm::isNotFinite().

Referenced by EcalShapeBase::buildMe(), drawSingleHistogram(), Rivet::CMS_2013_I1224539_DIJET::finalize(), Geom::Phi< float >::normalize(), MLPFProducer::produce(), and EcalShapeBase::setEventSetup().

                             {
     if (std::abs(in) > 1e4f) {
       return 0.0;
     } else if (edm::isNotFinite(in)) {
       return 0.0;
     }
     return in;
   }

◆ setCandidateRefs()

void reco::mlpf::setCandidateRefs	(	reco::PFCandidate &	cand,
		const std::vector< const reco::PFBlockElement *>	elems,
		size_t	ielem_originator
	)

Definition at line 284 of file MLPFModel.cc.

References funct::abs(), reco::PFBlockElement::GSF, reco::PFBlockElementGsfTrack::GsftrackRef(), edm::Ref< C, T, F >::isNonnull(), reco::PFBlockElementTrack::muonRef(), reco::PFBlockElement::TRACK, reco::PFBlockElement::trackRef(), and reco::PFBlockElement::type().

Referenced by MLPFProducer::produce().

                                                  {
     const reco::PFBlockElement* elem = elems[ielem_originator];
 
     //set the track ref in case the originating element was a track
     if (std::abs(cand.pdgId()) == 211 && elem->type() == reco::PFBlockElement::TRACK && elem->trackRef().isNonnull()) {
       const auto* eltTrack = dynamic_cast<const reco::PFBlockElementTrack*>(elem);
       cand.setTrackRef(eltTrack->trackRef());
       cand.setVertex(eltTrack->trackRef()->vertex());
       cand.setPositionAtECALEntrance(eltTrack->positionAtECALEntrance());
     }
 
     //set the muon ref
     if (std::abs(cand.pdgId()) == 13) {
       const auto* eltTrack = dynamic_cast<const reco::PFBlockElementTrack*>(elem);
       const auto& muonRef = eltTrack->muonRef();
       cand.setTrackRef(muonRef->track());
       cand.setMuonTrackType(muonRef->muonBestTrackType());
       cand.setVertex(muonRef->track()->vertex());
       cand.setMuonRef(muonRef);
     }
 
     if (std::abs(cand.pdgId()) == 11 && elem->type() == reco::PFBlockElement::GSF) {
       const auto* eltTrack = dynamic_cast<const reco::PFBlockElementGsfTrack*>(elem);
       const auto& ref = eltTrack->GsftrackRef();
       cand.setGsfTrackRef(ref);
     }
   }

Variable Documentation

◆ BATCH_SIZE

constexpr int reco::mlpf::BATCH_SIZE = 1

static

Definition at line 18 of file MLPFModel.h.

◆ elem_type_encoding

const std::map<int, int> reco::mlpf::elem_type_encoding

static

Initial value:

= {
      {0, 0},
      {1, 1},
      {2, 2},
      {3, 3},
      {4, 4},
      {5, 5},
      {6, 6},
      {7, 7},
      {8, 8},
      {9, 9},
      {10, 10},
      {11, 11},
  }

Definition at line 47 of file MLPFModel.h.

Referenced by getElementProperties().