Add process function in straevselextrasconverter.cxx

Author: romainschotter

PWGLF/TableProducer/Strangeness/Converters/straevselextrasconverter.cxx

@@ -25,8 +25,17 @@ using namespace o2::aod::evsel;
 struct straevselextrasconverter {
   Produces<aod::StraEvSelExtras_001> straEvSelExtras_001;
 
-  void process(soa::Join<aod::StraEvSels_005, aod::StraEvSelExtras_000> const& straEvSels_005)
+  void init(InitContext&)
   {
+    LOGF(info, "Initializing now: cross-checking correctness...");
+    if (doprocessAll + doprocessStraEvSelsOnly  > 1) {
+      LOGF(fatal, "You have enabled more than one process function. Please check your configuration! Aborting now.");
+    }
+  }
+
+  void processAll(soa::Join<aod::StraEvSels_005, aod::StraEvSelExtras_000> const& straEvSels_005)
+  {
+    straEvSelExtras_001.reserve(straEvSels_005.size());
     for (auto& values : straEvSels_005) {
       straEvSelExtras_001(values.multZNA(),
                           values.multZNC(),
@@ -57,6 +66,42 @@ struct straevselextrasconverter {
                           values.energyCommonZNC());
     }
   }
+
+  void processStraEvSelsOnly(aod::StraEvSels_005 const& straEvSels_005)
+  {
+    straEvSelExtras_001.reserve(straEvSels_005.size());
+    for (auto& values : straEvSels_005) {
+      straEvSelExtras_001(values.multZNA(),
+                          values.multZNC(),
+                          values.multZEM1(),
+                          values.multZEM2(),
+                          values.multZPA(),
+                          values.multZPC(),
+                          values.multNTracksITSTPC(),
+                          values.multAllTracksTPCOnly(),
+                          values.multAllTracksITSTPC(),
+                          values.trackOccupancyInTimeRange(),
+                          values.ft0cOccupancyInTimeRange(),
+                          -999.,  // dummy timeFDDA,
+                          -999.,  // dummy timeFDDC,
+                          -999.,  // dummy timeFV0A,
+                          -999.,  // dummy timeFT0A,
+                          -999.,  // dummy timeFT0C,
+                          0,      // dummy triggerMaskFT0,
+                          values.gapSide(),
+                          values.totalFT0AmplitudeA(),
+                          values.totalFT0AmplitudeC(),
+                          values.totalFV0AmplitudeA(),
+                          values.totalFDDAmplitudeA(),
+                          values.totalFDDAmplitudeC(),
+                          -999.,  // dummy timeZNA,
+                          -999.,  // dummy timeZNC,
+                          values.energyCommonZNA(),
+                          values.energyCommonZNC());
+    }
+  }
+  PROCESS_SWITCH(straevselextrasconverter, processAll, "Store StraEvSels005 and StraEvSelExtras000 into StraEvSelExtras001", true);
+  PROCESS_SWITCH(straevselextrasconverter, processStraEvSelsOnly, "Store only StraEvSels005 into StraEvSelExtras001. Other columns are set to dummy values", false);
 };
 
 WorkflowSpec defineDataProcessing(ConfigContext const& cfgc)