File indexing completed on 2024-04-06 12:24:27
0001 #include "RecoBTag/FeatureTools/interface/deep_helpers.h"
0002 #include "DataFormats/BTauReco/interface/ShallowTagInfoFeatures.h"
0003
0004 #include "DataFormats/BTauReco/interface/ShallowTagInfo.h"
0005 #include "DataFormats/BTauReco/interface/TaggingVariable.h"
0006
0007 #include "RecoBTag/FeatureTools/interface/ShallowTagInfoConverter.h"
0008
0009 namespace btagbtvdeep {
0010
0011 static constexpr std::size_t max_jetNSelectedTracks = 100;
0012
0013 void bTagToFeatures(const reco::TaggingVariableList& tag_info_vars, ShallowTagInfoFeatures& tag_info_features) {
0014 tag_info_features.trackSumJetEtRatio = tag_info_vars.get(reco::btau::trackSumJetEtRatio, -999);
0015 tag_info_features.trackSumJetDeltaR = tag_info_vars.get(reco::btau::trackSumJetDeltaR, -999);
0016 tag_info_features.vertexCategory = tag_info_vars.get(reco::btau::vertexCategory, -999);
0017 tag_info_features.trackSip2dValAboveCharm = tag_info_vars.get(reco::btau::trackSip2dValAboveCharm, -999);
0018 tag_info_features.trackSip2dSigAboveCharm = tag_info_vars.get(reco::btau::trackSip2dSigAboveCharm, -999);
0019 tag_info_features.trackSip3dValAboveCharm = tag_info_vars.get(reco::btau::trackSip3dValAboveCharm, -999);
0020 tag_info_features.trackSip3dSigAboveCharm = tag_info_vars.get(reco::btau::trackSip3dSigAboveCharm, -999);
0021 tag_info_features.jetNTracksEtaRel = tag_info_vars.get(reco::btau::jetNTracksEtaRel, -1);
0022 tag_info_features.jetNSelectedTracks =
0023 std::min(tag_info_vars.getList(reco::btau::trackMomentum, false).size(), max_jetNSelectedTracks);
0024 }
0025 }
