[Common] Add hepMC info to extras table task

Common/TableProducer/multiplicityExtraTable.cxx

@@ -8,7 +8,7 @@
 // In applying this license CERN does not waive the privileges and immunities
 // granted to it by virtue of its status as an Intergovernmental Organization
 // or submit itself to any jurisdiction.
 #include <CCDB/BasicCCDBManager.h> // megalinter thinks this is a C header...
 #include <bitset>
 #include "Framework/ConfigParamSpec.h"
 #include "Framework/runDataProcessing.h"
@@ -36,6 +36,9 @@
   Produces<aod::Mults2BC> mult2bc;
   Produces<aod::BC2Mults> bc2mult;
 
+  // auxiliary for MC
+  Produces<aod::MultHepMCHIs> multHepMCHIs;
+
   // Allow for downscaling of BC table for less space use in derived data
   Configurable<float> bcDownscaleFactor{"bcDownscaleFactor", 2, "Downscale factor for BC table (0: save nothing, 1: save all)"};
   Configurable<float> minFT0CforBCTable{"minFT0CforBCTable", 25.0f, "Minimum FT0C amplitude to fill BC table to reduce data"};
@@ -45,7 +48,7 @@
 
   float tru(float value)
   {
     if (bcTableFloatPrecision < 1e-4)
       return value; // make sure nothing bad happens in case zero (best precision)
     return bcTableFloatPrecision * std::round(value / bcTableFloatPrecision) + 0.5f * bcTableFloatPrecision;
   };
@@ -102,7 +105,7 @@
       float multFT0C = 0.f;
       if (bc.has_ft0()) {
         auto ft0 = bc.ft0();
         for (auto amplitude : ft0.amplitudeC()) {
           multFT0C += amplitude;
         }
       } else {
@@ -181,10 +184,10 @@
         multFT0TriggerBits = static_cast<uint8_t>(triggers.to_ulong());
 
         // calculate T0 charge
         for (auto amplitude : ft0.amplitudeA()) {
           multFT0A += amplitude;
         }
         for (auto amplitude : ft0.amplitudeC()) {
           multFT0C += amplitude;
         }
         posZFT0 = ft0.posZ();
@@ -198,7 +201,7 @@
         std::bitset<8> fV0Triggers = fv0.triggerMask();
         multFV0TriggerBits = static_cast<uint8_t>(fV0Triggers.to_ulong());
 
         for (auto amplitude : fv0.amplitude()) {
           multFV0A += amplitude;
         }
         isFV0OrA = fV0Triggers[o2::fit::Triggers::bitA];
@@ -211,10 +214,10 @@
         std::bitset<8> fFDDTriggers = fdd.triggerMask();
         multFDDTriggerBits = static_cast<uint8_t>(fFDDTriggers.to_ulong());
 
         for (auto amplitude : fdd.chargeA()) {
           multFDDA += amplitude;
         }
         for (auto amplitude : fdd.chargeC()) {
           multFDDC += amplitude;
         }
       } else {
@@ -266,20 +269,33 @@
 
       if (ii - 1 >= 0)
         deltaPrevious = timeArray[ii] - timeArray[ii - 1];
       if (ii - 2 >= 0)
         deltaPrePrevious = timeArray[ii] - timeArray[ii - 2];
       if (ii + 1 < collisions.size())
         deltaNext = timeArray[ii + 1] - timeArray[ii];
       if (ii + 2 < collisions.size())
         deltaNeNext = timeArray[ii + 2] - timeArray[ii];
 
       multNeigh(deltaPrePrevious, deltaPrevious, deltaNext, deltaNeNext);
     }
   }
 
+  void processHepMCHeavyIons(aod::HepMCHeavyIons const& hepmchis)
+  {
+    for (auto const& hepmchi : hepmchis) {
+      multHepMCHIs(hepmchi.mcCollisionId(),
+                   hepmchi.ncollHard(),
+                   hepmchi.npartProj(),
+                   hepmchi.npartTarg(),
+                   hepmchi.ncoll(),
+                   hepmchi.impactParameter());
+    }
+  }
+
   // Process switches
   PROCESS_SWITCH(MultiplicityExtraTable, processBCs, "Produce BC tables", true);
   PROCESS_SWITCH(MultiplicityExtraTable, processCollisionNeighbors, "Produce neighbor timing tables", true);
+  PROCESS_SWITCH(MultiplicityExtraTable, processHepMCHeavyIons, "Produce MultHepMCHIs tables", false);
 };
 
 WorkflowSpec defineDataProcessing(ConfigContext const& cfgc)