BTauReco/interface/DeepFlavourFeatures.h

0001 #ifndef DataFormats_BTauReco_DeepFlavourFeatures_h
0002 #define DataFormats_BTauReco_DeepFlavourFeatures_h
0003
0004 #include "DataFormats/BTauReco/interface/JetFeatures.h"
0005 #include "DataFormats/BTauReco/interface/SecondaryVertexFeatures.h"
0006 #include "DataFormats/BTauReco/interface/ShallowTagInfoFeatures.h"
0007 #include "DataFormats/BTauReco/interface/NeutralCandidateFeatures.h"
0008 #include "DataFormats/BTauReco/interface/ChargedCandidateFeatures.h"
0009 #include "DataFormats/BTauReco/interface/SeedingTrackFeatures.h"
0010
0011 #include <vector>
0012
0013 namespace btagbtvdeep {
0014
0015   class DeepFlavourFeatures {
0016   public:
0017     JetFeatures jet_features;
0018     ShallowTagInfoFeatures tag_info_features;
0019
0020     std::vector<SecondaryVertexFeatures> sv_features;
0021
0022     std::vector<NeutralCandidateFeatures> n_pf_features;
0023     std::vector<ChargedCandidateFeatures> c_pf_features;
0024
0025     std::vector<SeedingTrackFeatures> seed_features;
0026
0027     std::size_t npv;  // used by deep flavour
0028   };
0029
0030 }  // namespace btagbtvdeep
0031
0032 #endif  //DataFormats_BTauReco_DeepFlavourFeatures_h