FeatureTools/src/BoostedDoubleSVTagInfoConverter.cc

0001 #include "RecoBTag/FeatureTools/interface/BoostedDoubleSVTagInfoConverter.h"
0002 #include "RecoBTag/FeatureTools/interface/deep_helpers.h"
0003 #include "DataFormats/BTauReco/interface/BoostedDoubleSVTagInfoFeatures.h"
0004
0005 #include "DataFormats/BTauReco/interface/BoostedDoubleSVTagInfo.h"
0006 #include "DataFormats/BTauReco/interface/TaggingVariable.h"
0007
0008 namespace btagbtvdeep {
0009
0010   void doubleBTagToFeatures(const reco::TaggingVariableList& tag_info_vars,
0011                             BoostedDoubleSVTagInfoFeatures& tag_info_features) {
0012     tag_info_features.jetNTracks = tag_info_vars.get(reco::btau::jetNTracks, -999);
0013     tag_info_features.jetNSecondaryVertices = tag_info_vars.get(reco::btau::jetNSecondaryVertices, -999);
0014     tag_info_features.trackSip3dSig_0 = tag_info_vars.get(reco::btau::trackSip3dSig_0, -999);
0015     tag_info_features.trackSip3dSig_1 = tag_info_vars.get(reco::btau::trackSip3dSig_1, -999);
0016     tag_info_features.trackSip3dSig_2 = tag_info_vars.get(reco::btau::trackSip3dSig_2, -999);
0017     tag_info_features.trackSip3dSig_3 = tag_info_vars.get(reco::btau::trackSip3dSig_3, -999);
0018     tag_info_features.tau1_trackSip3dSig_0 = tag_info_vars.get(reco::btau::tau1_trackSip3dSig_0, -999);
0019     tag_info_features.tau1_trackSip3dSig_1 = tag_info_vars.get(reco::btau::tau1_trackSip3dSig_1, -999);
0020     tag_info_features.tau2_trackSip3dSig_0 = tag_info_vars.get(reco::btau::tau2_trackSip3dSig_0, -999);
0021     tag_info_features.tau2_trackSip3dSig_1 = tag_info_vars.get(reco::btau::tau2_trackSip3dSig_1, -999);
0022     tag_info_features.trackSip2dSigAboveBottom_0 = tag_info_vars.get(reco::btau::trackSip2dSigAboveBottom_0, -999);
0023     tag_info_features.trackSip2dSigAboveBottom_1 = tag_info_vars.get(reco::btau::trackSip2dSigAboveBottom_1, -999);
0024     tag_info_features.trackSip2dSigAboveCharm = tag_info_vars.get(reco::btau::trackSip2dSigAboveCharm, -999);
0025     tag_info_features.tau1_trackEtaRel_0 = tag_info_vars.get(reco::btau::tau1_trackEtaRel_0, -999);
0026     tag_info_features.tau1_trackEtaRel_1 = tag_info_vars.get(reco::btau::tau1_trackEtaRel_1, -999);
0027     tag_info_features.tau1_trackEtaRel_2 = tag_info_vars.get(reco::btau::tau1_trackEtaRel_2, -999);
0028     tag_info_features.tau2_trackEtaRel_0 = tag_info_vars.get(reco::btau::tau2_trackEtaRel_0, -999);
0029     tag_info_features.tau2_trackEtaRel_1 = tag_info_vars.get(reco::btau::tau2_trackEtaRel_1, -999);
0030     tag_info_features.tau2_trackEtaRel_2 = tag_info_vars.get(reco::btau::tau2_trackEtaRel_2, -999);
0031     tag_info_features.tau1_vertexMass = tag_info_vars.get(reco::btau::tau1_vertexMass, -999);
0032     tag_info_features.tau1_vertexEnergyRatio = tag_info_vars.get(reco::btau::tau1_vertexEnergyRatio, -999);
0033     tag_info_features.tau1_flightDistance2dSig = tag_info_vars.get(reco::btau::tau1_flightDistance2dSig, -999);
0034     tag_info_features.tau1_vertexDeltaR = tag_info_vars.get(reco::btau::tau1_vertexDeltaR, -999);
0035     tag_info_features.tau2_vertexMass = tag_info_vars.get(reco::btau::tau2_vertexMass, -999);
0036     tag_info_features.tau2_vertexEnergyRatio = tag_info_vars.get(reco::btau::tau2_vertexEnergyRatio, -999);
0037     tag_info_features.tau2_flightDistance2dSig = tag_info_vars.get(reco::btau::tau2_flightDistance2dSig, -999);
0038     tag_info_features.tau2_vertexDeltaR = tag_info_vars.get(reco::btau::tau2_vertexDeltaR, -999);  // not used
0039     tag_info_features.z_ratio = tag_info_vars.get(reco::btau::z_ratio, -999);
0040   }
0041
0042 }  // namespace btagbtvdeep