NeutralCandidateConverter.h

Go to the documentation of this file.

#ifndef RecoBTag_FeatureTools_NeutralCandidateConverter_h
#define RecoBTag_FeatureTools_NeutralCandidateConverter_h

#include "RecoBTag/FeatureTools/interface/deep_helpers.h"
#include "DataFormats/BTauReco/interface/NeutralCandidateFeatures.h"

#include "DataFormats/PatCandidates/interface/PackedCandidate.h"
#include "DataFormats/ParticleFlowCandidate/interface/PFCandidate.h"
#include "DataFormats/JetReco/interface/Jet.h"
#include "DataFormats/PatCandidates/interface/Jet.h"

namespace btagbtvdeep {

  void packedCandidateToFeatures(const pat::PackedCandidate* n_pf,
                                 const pat::Jet& jet,
                                 const bool isWeightedJet,
                                 const float drminpfcandsv,
                                 const float jetR,
                                 const float puppiw,
                                 NeutralCandidateFeatures& n_pf_features);

  void recoCandidateToFeatures(const reco::PFCandidate* n_pf,
                               const reco::Jet& jet,
                               const bool isWeightedJet,
                               const float drminpfcandsv,
                               const float jetR,
                               const float puppiw,
                               NeutralCandidateFeatures& n_pf_features);

  template <typename CandidateType>
  static void commonCandidateToFeatures(const CandidateType* n_pf,
                                        const reco::Jet& jet,
                                        const bool& isWeightedJet,
                                        const float& drminpfcandsv,
                                        const float& jetR,
                                        const float& puppiw,
                                        NeutralCandidateFeatures& n_pf_features) {
    std::pair<float, float> drSubjetFeatures = getDRSubjetFeatures(jet, n_pf);
    n_pf_features.drsubjet1 = drSubjetFeatures.first;
    n_pf_features.drsubjet2 = drSubjetFeatures.second;

    float constituentWeight = 1.;
    if (isWeightedJet)
      constituentWeight = puppiw;

    // Jet relative vars
    n_pf_features.ptrel = catch_infs_and_bound((n_pf->pt() * constituentWeight) / jet.pt(), 0, -1, 0, -1);
    n_pf_features.ptrel_noclip = (n_pf->pt() * constituentWeight) / jet.pt();
    n_pf_features.erel = (n_pf->energy() * constituentWeight) / jet.energy();

    n_pf_features.deltaR = catch_infs_and_bound(reco::deltaR(*n_pf, jet), 0, -0.6, 0, -0.6);
    n_pf_features.deltaR_noclip = reco::deltaR(*n_pf, jet);

    n_pf_features.isGamma = 0;
    if (std::abs(n_pf->pdgId()) == 22)
      n_pf_features.isGamma = 1;

    n_pf_features.drminsv = catch_infs_and_bound(drminpfcandsv, 0, -1. * jetR, 0, -1. * jetR);

    n_pf_features.etarel = catch_infs_and_bound(std::fabs(n_pf->eta() - jet.eta()), 0, -2, 0, -0.5);
    n_pf_features.phirel = catch_infs_and_bound(std::fabs(reco::deltaPhi(n_pf->phi(), jet.phi())), 0, -2, 0, -0.5);
    n_pf_features.pt = n_pf->pt();
    n_pf_features.eta = n_pf->eta();
    n_pf_features.phi = n_pf->phi();
    n_pf_features.e = n_pf->energy();
    n_pf_features.px = n_pf->px();
    n_pf_features.py = n_pf->py();
    n_pf_features.pz = n_pf->pz();
  }

}  // namespace btagbtvdeep

#endif  //RecoBTag_FeatureTools_NeutralCandidateConverter_h