Back to home page

Project CMSSW displayed by LXR

 
 

    


File indexing completed on 2021-02-14 13:33:39

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 }  // namespace btagbtvdeep