DeepFlavourFeatures

Macros

Line Code
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32
#ifndef DataFormats_BTauReco_DeepFlavourFeatures_h
#define DataFormats_BTauReco_DeepFlavourFeatures_h

#include "DataFormats/BTauReco/interface/JetFeatures.h"
#include "DataFormats/BTauReco/interface/SecondaryVertexFeatures.h"
#include "DataFormats/BTauReco/interface/ShallowTagInfoFeatures.h"
#include "DataFormats/BTauReco/interface/NeutralCandidateFeatures.h"
#include "DataFormats/BTauReco/interface/ChargedCandidateFeatures.h"
#include "DataFormats/BTauReco/interface/SeedingTrackFeatures.h"

#include <vector>

namespace btagbtvdeep {

  class DeepFlavourFeatures {
  public:
    JetFeatures jet_features;
    ShallowTagInfoFeatures tag_info_features;

    std::vector<SecondaryVertexFeatures> sv_features;

    std::vector<NeutralCandidateFeatures> n_pf_features;
    std::vector<ChargedCandidateFeatures> c_pf_features;

    std::vector<SeedingTrackFeatures> seed_features;

    std::size_t npv;  // used by deep flavour
  };

}  // namespace btagbtvdeep

#endif  //DataFormats_BTauReco_DeepFlavourFeatures_h