Run processor

Author: qgp

machine_learning_hep/data/data_run3/database_ml_parameters_D0Jet_pp.yml

@@ -57,6 +57,14 @@ D0Jet_pp:
         trees:
           O2hfd0collbase: [fNumContrib, fPosZ]
         filter: "abs(fPosZ) < 10."
+      collcnt_mc:
+        level: mc
+        trees:
+          O2collcount: [fReadCounts, fReadCountsWithTVX, fReadCountsWithTVXAndZVertexAndSel8, fReadCountsWithTVXAndZVertexAndSelMC]
+      bccnt_mc:
+        level: mc
+        trees:
+          O2bccount: [fReadCountsWithTVX, fReadCountsWithTVXAndNoTFB, fReadCountsWithTVXAndNoTFBAndNoITSROFB]
       collcnt:
         level: data
         trees:
@@ -220,12 +228,18 @@ D0Jet_pp:
         level: fd
         source: collfd
         file: AnalysisResultsEvt.parquet
-      # collcnt:
-      #   level: all
-      #   file: AnalysisResultsCollCnt.parquet
-      # bccnt:
-      #   level: all
-      #   file: AnalysisResultsBcCnt.parquet
+      collcnt_mc:
+        level: mc
+        file: AnalysisResultsCollCnt.parquet
+      bccnt_mc:
+        level: mc
+        file: AnalysisResultsBcCnt.parquet
+      collcnt:
+        level: data
+        file: AnalysisResultsCollCnt.parquet
+      bccnt:
+        level: data
+        file: AnalysisResultsBcCnt.parquet
 
   variables:
     var_all: [fDecayLength, fDecayLengthXY, fDecayLengthNormalised, fDecayLengthXYNormalised, fCpa, fCpaXY, fImpactParameter0, fImpactParameter1, fErrorImpactParameter0, fErrorImpactParameter1, fNSigTpcPiExpPi, fNSigTpcKaExpPi, fNSigTpcPiExpKa, fNSigTpcKaExpKa]

machine_learning_hep/processer.py

@@ -196,8 +196,8 @@ def __init__(
         self.l_gen = createlist(self.d_pkl, self.l_path, self.n_gen)
         self.l_evt = createlist(self.d_pkl, self.l_path, self.n_evt)
         self.l_evtorig = createlist(self.d_pkl, self.l_path, self.n_evtorig)
-        self.l_collcnt = createlist(self.d_pkl, self.l_path, self.n_collcnt)
-        self.l_bccnt = createlist(self.d_pkl, self.l_path, self.n_bccnt)
+        self.l_collcnt = createlist(self.d_pkl, self.l_path, self.n_collcnt) if self.datatype != "fd" else None
+        self.l_bccnt = createlist(self.d_pkl, self.l_path, self.n_bccnt) if self.datatype != "fd" else None
         self.l_histomass = createlist(self.d_results, self.l_path, self.n_filemass)
         self.l_histoeff = createlist(self.d_results, self.l_path, self.n_fileeff)
         # self.l_historesp = createlist(self.d_results, self.l_path, self.n_fileresp)