FeatureTools/interface/JetConverter.h

0001 #ifndef RecoBTag_FeatureTools_JetConverter_h
0002 #define RecoBTag_FeatureTools_JetConverter_h
0003
0004 #include "DataFormats/BTauReco/interface/JetFeatures.h"
0005
0006 #include "DataFormats/JetReco/interface/Jet.h"
0007
0008 namespace btagbtvdeep {
0009
0010   class JetConverter {
0011   public:
0012     static void jetToFeatures(const reco::Jet& jet, JetFeatures& jet_features) {
0013       jet_features.pt = jet.pt();  // uncorrected
0014       jet_features.eta = jet.eta();
0015       jet_features.phi = jet.phi();
0016       jet_features.mass = jet.mass();
0017       jet_features.energy = jet.energy();
0018     }
0019   };
0020
0021 }  // namespace btagbtvdeep
0022
0023 #endif  //RecoBTag_FeatureTools_JetConverter_h