JetConverter.h

Go to the documentation of this file.

#ifndef RecoBTag_FeatureTools_JetConverter_h
#define RecoBTag_FeatureTools_JetConverter_h

#include "DataFormats/BTauReco/interface/JetFeatures.h"

#include "DataFormats/JetReco/interface/Jet.h"

namespace btagbtvdeep {

  class JetConverter {
  public:

    static void jetToFeatures(const reco::Jet & jet,
                  JetFeatures & jet_features) {
      jet_features.pt = jet.pt(); // uncorrected
      jet_features.eta = jet.eta();
      jet_features.phi = jet.phi();   
      jet_features.mass = jet.mass();
      jet_features.energy = jet.energy();
    }
  };

}

#endif //RecoBTag_FeatureTools_JetConverter_h