BTauReco/interface/DeepDoubleXFeatures.h

0001 #ifndef DataFormats_BTauReco_DeepDoubleXFeatures_h
0002 #define DataFormats_BTauReco_DeepDoubleXFeatures_h
0003
0004 #include <vector>
0005
0006 #include "DataFormats/BTauReco/interface/JetFeatures.h"
0007 #include "DataFormats/BTauReco/interface/SecondaryVertexFeatures.h"
0008 #include "DataFormats/BTauReco/interface/BoostedDoubleSVTagInfoFeatures.h"
0009 #include "DataFormats/BTauReco/interface/ChargedCandidateFeatures.h"
0010 #include "DataFormats/BTauReco/interface/NeutralCandidateFeatures.h"
0011
0012 namespace btagbtvdeep {
0013
0014   class DeepDoubleXFeatures {
0015   public:
0016     bool empty() const { return is_empty_; }
0017
0018     void filled() { is_empty_ = false; }
0019
0020     JetFeatures jet_features;
0021     BoostedDoubleSVTagInfoFeatures tag_info_features;
0022
0023     std::vector<ChargedCandidateFeatures> c_pf_features;
0024
0025     std::vector<NeutralCandidateFeatures> n_pf_features;
0026
0027     std::vector<SecondaryVertexFeatures> sv_features;
0028
0029     std::size_t npv;  // used by deep flavour
0030
0031   private:
0032     bool is_empty_ = true;
0033   };
0034
0035 }  // namespace btagbtvdeep
0036
0037 #endif  //DataFormats_BTauReco_DeepDoubleXFeatures_h