Add ghost associated H, W, Z, and Top counts to fat jet

BTagTrainingPreprocessing/src/BTaggingWriterConfiguration.hh

@@ -26,6 +26,12 @@ namespace cfg {
     "XbbScoreTop",
     "XbbScoreHiggs",
   };
+  const List FatJetParentTruthLabels = {
+    "GhostHBosonsCount",
+    "GhostWBosonsCount",
+    "GhostZBosonsCount",
+    "GhostTQuarksFinalCount",
+  };
 
   // BTagging
   const List BTagDoubles = {

BTagTrainingPreprocessing/util/dump-hbb.cxx

@@ -203,7 +203,7 @@ int main (int argc, char *argv[])
       fat_cfg.jet_float_variables.push_back(name);
     }
   }
-  fat_cfg.parent_truth_labels = {"GhostHBosonsCount"};
+  fat_cfg.parent_truth_labels = cfg::FatJetParentTruthLabels;
   fat_cfg.name = "fat_jet";
   BTagJetWriter fatjet_writer(output, fat_cfg);