File indexing completed on 2023-03-29 01:43:03
0001 #include "RecoBTag/FeatureTools/interface/NeutralCandidateConverter.h"
0002
0003 namespace btagbtvdeep {
0004
0005 void packedCandidateToFeatures(const pat::PackedCandidate* n_pf,
0006 const pat::Jet& jet,
0007 const bool isWeightedJet,
0008 const float drminpfcandsv,
0009 const float jetR,
0010 const float puppiw,
0011 NeutralCandidateFeatures& n_pf_features) {
0012 commonCandidateToFeatures(n_pf, jet, isWeightedJet, drminpfcandsv, jetR, puppiw, n_pf_features);
0013
0014 n_pf_features.hadFrac = n_pf->hcalFraction();
0015 n_pf_features.puppiw = puppiw;
0016 }
0017
0018 void recoCandidateToFeatures(const reco::PFCandidate* n_pf,
0019 const reco::Jet& jet,
0020 const bool isWeightedJet,
0021 const float drminpfcandsv,
0022 const float jetR,
0023 const float puppiw,
0024 NeutralCandidateFeatures& n_pf_features) {
0025 commonCandidateToFeatures(n_pf, jet, isWeightedJet, drminpfcandsv, jetR, puppiw, n_pf_features);
0026 n_pf_features.puppiw = puppiw;
0027
0028
0029
0030
0031 if (abs(n_pf->pdgId()) == 1 || abs(n_pf->pdgId()) == 130) {
0032 n_pf_features.hadFrac = n_pf->hcalEnergy() / (n_pf->ecalEnergy() + n_pf->hcalEnergy());
0033 } else {
0034 n_pf_features.hadFrac = 0;
0035 }
0036 }
0037
0038 }