ChargedCandidateConverter.cc

Go to the documentation of this file.

#include "RecoBTag/FeatureTools/interface/ChargedCandidateConverter.h"


namespace btagbtvdeep {

  void packedCandidateToFeatures(const pat::PackedCandidate * c_pf,
                 const pat::Jet & jet,
                 const TrackInfoBuilder & track_info,
                 const float drminpfcandsv, const float jetR,
                 ChargedCandidateFeatures & c_pf_features,
                 const bool flip) {

    commonCandidateToFeatures(c_pf, jet, track_info, drminpfcandsv, jetR, c_pf_features, flip);

    c_pf_features.vtx_ass = c_pf->pvAssociationQuality();

    c_pf_features.puppiw = c_pf->puppiWeight();

    // if PackedCandidate does not have TrackDetails this gives an Exception
    // because unpackCovariance might be called for pseudoTrack/bestTrack
    if (c_pf->hasTrackDetails()) {
      const auto & pseudo_track =  c_pf->pseudoTrack();
      c_pf_features.chi2 = catch_infs_and_bound(pseudo_track.normalizedChi2(),300,-1,300);
      // this returns the quality enum not a mask.
      c_pf_features.quality = pseudo_track.qualityMask();
    } else {
      // default negative chi2 and loose track if notTrackDetails
      c_pf_features.chi2 = catch_infs_and_bound(-1,300,-1,300);
      c_pf_features.quality =(1 << reco::TrackBase::loose);
    }

    c_pf_features.dxy = catch_infs(c_pf->dxy());
    c_pf_features.dz = catch_infs(c_pf->dz());
    c_pf_features.dxysig = c_pf->bestTrack() ? catch_infs(c_pf->dxy() / c_pf->dxyError()) : 0;
    c_pf_features.dzsig = c_pf->bestTrack() ? catch_infs(c_pf->dz() / c_pf->dzError()) : 0;
  }

  void recoCandidateToFeatures(const reco::PFCandidate * c_pf,
                   const reco::Jet & jet,
                   const TrackInfoBuilder & track_info,
                   const float drminpfcandsv, const float jetR, const float puppiw,
                   const int pv_ass_quality,
                   const reco::VertexRef & pv,
                   ChargedCandidateFeatures & c_pf_features,
                   const bool flip) {

    commonCandidateToFeatures(c_pf, jet, track_info, drminpfcandsv, jetR, c_pf_features, flip);

    c_pf_features.vtx_ass = vtx_ass_from_pfcand(*c_pf, pv_ass_quality, pv);
    c_pf_features.puppiw = puppiw;

    const auto & pseudo_track =  (c_pf->bestTrack()) ? *c_pf->bestTrack() : reco::Track();
    c_pf_features.chi2 = catch_infs_and_bound(std::floor(pseudo_track.normalizedChi2()),300,-1,300);
    c_pf_features.quality = quality_from_pfcand(*c_pf);

    // To be implemented if FatJet tag becomes RECO compatible
    // const auto *trk =
    // float dz =
    // float dxy =

    // c_pf_features.dxy =
    // c_pf_features.dz =
    // c_pf_features.dxysig =
    // c_pf_features.dzsig =
  }

}