AODProducerWorkflowSpec.cxx

// Copyright 2019-2020 CERN and copyright holders of ALICE O2.
// See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
// All rights not expressly granted are reserved.
//
// This software is distributed under the terms of the GNU General Public
// License v3 (GPL Version 3), copied verbatim in the file "COPYING".
//
// 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.

/// @file   AODProducerWorkflowSpec.cxx

#include "AODProducerWorkflow/AODProducerWorkflowSpec.h"
#include "AODProducerWorkflow/AODMcProducerHelpers.h"
#include "AODProducerWorkflow/AODProducerHelpers.h"
#include "DataFormatsEMCAL/TriggerRecord.h"
#include "DataFormatsEMCAL/EventHandler.h"
#include "DataFormatsFT0/RecPoints.h"
#include "DataFormatsFDD/RecPoint.h"
#include "DataFormatsFV0/RecPoints.h"
#include "DataFormatsGlobalTracking/RecoContainer.h"
#include "DataFormatsCTP/Digits.h"
#include "DataFormatsCTP/Configuration.h"
#include "DataFormatsCPV/Cluster.h"
#include "DataFormatsCPV/TriggerRecord.h"
#include "DataFormatsITS/TrackITS.h"
#include "DataFormatsMCH/ROFRecord.h"
#include "DataFormatsMCH/TrackMCH.h"
#include "DataFormatsMCH/Cluster.h"
#include "DataFormatsMID/Track.h"
#include "DataFormatsMFT/TrackMFT.h"
#include "DataFormatsPHOS/TriggerRecord.h"
#include "DataFormatsPHOS/EventHandler.h"
#include "DataFormatsTPC/TrackTPC.h"
#include "DataFormatsTRD/TriggerRecord.h"
#include "DataFormatsZDC/BCRecData.h"
#include "DataFormatsZDC/ZDCEnergy.h"
#include "DataFormatsZDC/ZDCTDCData.h"
#include "DataFormatsParameters/GRPECSObject.h"
#include "MathUtils/Utils.h"
#include "CCDB/BasicCCDBManager.h"
#include "CommonConstants/Triggers.h"
#include "CommonConstants/PhysicsConstants.h"
#include "CommonDataFormat/InteractionRecord.h"
#include "DataFormatsTRD/TrackTRD.h"
#include "DataFormatsTRD/TrackTriggerRecord.h"
#include "DataFormatsGlobalTracking/RecoContainer.h"
#include "Framework/AnalysisDataModel.h"
#include "Framework/ConfigParamRegistry.h"
#include "Framework/DataTypes.h"
#include "Framework/TableBuilder.h"
#include "Framework/CCDBParamSpec.h"
#include "CommonUtils/TreeStreamRedirector.h"
#include "FT0Base/Geometry.h"
#include "GlobalTracking/MatchTOF.h"
#include "ReconstructionDataFormats/Cascade.h"
#include "GlobalTracking/MatchGlobalFwd.h"
#include "MCHTracking/TrackExtrap.h"
#include "MCHTracking/TrackParam.h"
#include "ITSMFTBase/DPLAlpideParam.h"
#include "DetectorsVertexing/PVertexerParams.h"
#include "ReconstructionDataFormats/GlobalFwdTrack.h"
#include "ReconstructionDataFormats/GlobalTrackID.h"
#include "ReconstructionDataFormats/PrimaryVertex.h"
#include "ReconstructionDataFormats/StrangeTrack.h"
#include "ReconstructionDataFormats/Track.h"
#include "ReconstructionDataFormats/TrackTPCITS.h"
#include "ReconstructionDataFormats/TrackMCHMID.h"
#include "ReconstructionDataFormats/MatchInfoHMP.h"
#include "ReconstructionDataFormats/V0.h"
#include "ReconstructionDataFormats/VtxTrackIndex.h"
#include "ReconstructionDataFormats/VtxTrackRef.h"
#include "SimulationDataFormat/DigitizationContext.h"
#include "SimulationDataFormat/MCEventLabel.h"
#include "SimulationDataFormat/MCTrack.h"
#include "SimulationDataFormat/MCTruthContainer.h"
#include "SimulationDataFormat/MCUtils.h"
#include "SimulationDataFormat/MCGenProperties.h"
#include "ZDCBase/Constants.h"
#include "TPCBase/ParameterElectronics.h"
#include "GPUTPCGMMergedTrackHit.h"
#include "TOFBase/Utils.h"
#include "O2Version.h"
#include "TMath.h"
#include "MathUtils/Utils.h"
#include "Math/SMatrix.h"
#include "TString.h"
#include <limits>
#include <map>
#include <numeric>
#include <type_traits>
#include <unordered_map>
#include <set>
#include <string>
#include <vector>
#include <thread>
#include "TLorentzVector.h"
#include "TVector3.h"
#include "MathUtils/Tsallis.h"
#include <random>
#ifdef WITH_OPENMP
#include <omp.h>
#endif
#include <filesystem>
#include <nlohmann/json.hpp>

using namespace o2::framework;
using namespace o2::math_utils::detail;
using PVertex = o2::dataformats::PrimaryVertex;
using GIndex = o2::dataformats::VtxTrackIndex;
using DataRequest = o2::globaltracking::DataRequest;
using GID = o2::dataformats::GlobalTrackID;
using DetID = o2::detectors::DetID;
using SMatrix55Sym = ROOT::Math::SMatrix<double, 5, 5, ROOT::Math::MatRepSym<double, 5>>;

namespace o2::aodproducer
{

void AODProducerWorkflowDPL::createCTPReadout(const o2::globaltracking::RecoContainer& recoData, std::vector<o2::ctp::CTPDigit>& ctpDigits, ProcessingContext& pc)
{
  // Extraxt CTP Config from CCDB
  const auto ctpcfg = pc.inputs().get<o2::ctp::CTPConfiguration*>("ctpconfig");
  ctpcfg->printStream(std::cout);
  // o2::ctp::CTPConfiguration ctpcfg = o2::ctp::CTPRunManager::getConfigFromCCDB(-1, std::to_string(runNumber)); // how to get run
  //  Extract inputs from recoData
  uint64_t classMaskEMCAL = 0, classMaskTRD = 0, classMaskPHOSCPV = 0;
  for (const auto& trgclass : ctpcfg->getCTPClasses()) {
    if (trgclass.cluster->getClusterDetNames().find("EMC") != std::string::npos) {
      classMaskEMCAL = trgclass.classMask;
    }
    if (trgclass.cluster->getClusterDetNames().find("PHS") != std::string::npos) {
      classMaskPHOSCPV = trgclass.classMask;
    }
    if (trgclass.cluster->getClusterDetNames().find("TRD") != std::string::npos) {
      classMaskTRD = trgclass.classMask;
    }
  }
  LOG(info) << "createCTPReadout: Class Mask EMCAL -> " << classMaskEMCAL;
  LOG(info) << "createCTPReadout: Class Mask PHOS/CPV -> " << classMaskPHOSCPV;
  LOG(info) << "createCTPReadout: Class Mask TRD -> " << classMaskTRD;

  // const auto& fddRecPoints = recoData.getFDDRecPoints();
  // const auto& fv0RecPoints = recoData.getFV0RecPoints();
  const auto& triggerrecordEMCAL = recoData.getEMCALTriggers();
  const auto& triggerrecordPHOSCPV = recoData.getPHOSTriggers();
  const auto& triggerrecordTRD = recoData.getTRDTriggerRecords();
  // For EMCAL filter remove calibration triggers
  std::vector<o2::emcal::TriggerRecord> triggerRecordEMCALPhys;
  for (const auto& trg : triggerrecordEMCAL) {
    if (trg.getTriggerBits() & o2::trigger::Cal) {
      continue;
    }
    triggerRecordEMCALPhys.push_back(trg);
  }
  // const auto& triggerrecordTRD =recoData.getITSTPCTRDTriggers()
  //

  // Find TVX triggers, only TRD/EMCAL/PHOS/CPV triggers in coincidence will be accepted
  std::set<uint64_t> bcsMapT0triggers;
  const auto& ft0RecPoints = recoData.getFT0RecPoints();
  for (auto& ft0RecPoint : ft0RecPoints) {
    auto t0triggers = ft0RecPoint.getTrigger();
    if (t0triggers.getVertex()) {
      uint64_t globalBC = ft0RecPoint.getInteractionRecord().toLong();
      bcsMapT0triggers.insert(globalBC);
    }
  }

  auto genericCTPDigitizer = [&bcsMapT0triggers, &ctpDigits](auto triggerrecords, uint64_t classmask) -> int {
    // Strategy:
    // find detector trigger based on trigger record from readout and add CTPDigit if trigger there
    int cntwarnings = 0;
    uint32_t orbitPrev = 0;
    uint16_t bcPrev = 0;
    for (auto& trigger : triggerrecords) {
      auto orbitPrevT = orbitPrev;
      auto bcPrevT = bcPrev;
      bcPrev = trigger.getBCData().bc;
      orbitPrev = trigger.getBCData().orbit;
      // dedicated for TRD: remove bogus triggers
      if (orbitPrev < orbitPrevT || bcPrev >= o2::constants::lhc::LHCMaxBunches || (orbitPrev == orbitPrevT && bcPrev < bcPrevT)) {
        cntwarnings++;
        // LOGP(warning, "Bogus TRD trigger at bc:{}/orbit:{} (previous was {}/{}), with {} tracklets and {} digits",bcPrev, orbitPrev, bcPrevT, orbitPrevT, trig.getNumberOfTracklets(), trig.getNumberOfDigits());
      } else {
        uint64_t globalBC = trigger.getBCData().toLong();
        auto t0entry = bcsMapT0triggers.find(globalBC);
        if (t0entry != bcsMapT0triggers.end()) {
          auto ctpdig = std::find_if(ctpDigits.begin(), ctpDigits.end(), [globalBC](const o2::ctp::CTPDigit& dig) { return static_cast<uint64_t>(dig.intRecord.toLong()) == globalBC; });
          if (ctpdig != ctpDigits.end()) {
            // CTP digit existing from other trigger, merge detector class mask
            ctpdig->CTPClassMask |= std::bitset<64>(classmask);
            LOG(debug) << "createCTPReadout: Merging " << classmask << " CTP digits with existing digit, CTP mask " << ctpdig->CTPClassMask;
          } else {
            // New CTP digit needed
            LOG(debug) << "createCTPReadout: New CTP digit needed for class " << classmask << std::endl;
            auto& ctpdigNew = ctpDigits.emplace_back();
            ctpdigNew.intRecord.setFromLong(globalBC);
            ctpdigNew.CTPClassMask = classmask;
          }
        } else {
          LOG(warning) << "createCTPReadout: Found " << classmask << " and no MTVX:" << globalBC;
        }
      }
    }
    return cntwarnings;
  };

  auto warningsTRD = genericCTPDigitizer(triggerrecordTRD, classMaskTRD);
  auto warningsEMCAL = genericCTPDigitizer(triggerRecordEMCALPhys, classMaskEMCAL);
  auto warningsPHOSCPV = genericCTPDigitizer(triggerrecordPHOSCPV, classMaskPHOSCPV);

  LOG(info) << "createCTPReadout:# of TRD bogus triggers:" << warningsTRD;
  LOG(info) << "createCTPReadout:# of EMCAL bogus triggers:" << warningsEMCAL;
  LOG(info) << "createCTPReadout:# of PHOS/CPV bogus triggers:" << warningsPHOSCPV;
}

void AODProducerWorkflowDPL::collectBCs(const o2::globaltracking::RecoContainer& data,
                                        const std::vector<o2::InteractionTimeRecord>& mcRecords,
                                        std::map<uint64_t, int>& bcsMap)
{
  const auto& primVertices = data.getPrimaryVertices();
  const auto& fddRecPoints = data.getFDDRecPoints();
  const auto& ft0RecPoints = data.getFT0RecPoints();
  const auto& fv0RecPoints = data.getFV0RecPoints();
  const auto& caloEMCCellsTRGR = data.getEMCALTriggers();
  const auto& caloPHOSCellsTRGR = data.getPHOSTriggers();
  const auto& cpvTRGR = data.getCPVTriggers();
  const auto& ctpDigits = data.getCTPDigits();
  const auto& zdcBCRecData = data.getZDCBCRecData();

  bcsMap[mStartIR.toLong()] = 1; // store the start of TF

  // collecting non-empty BCs and enumerating them
  for (auto& rec : mcRecords) {
    uint64_t globalBC = rec.toLong();
    bcsMap[globalBC] = 1;
  }

  for (auto& fddRecPoint : fddRecPoints) {
    uint64_t globalBC = fddRecPoint.getInteractionRecord().toLong();
    bcsMap[globalBC] = 1;
  }

  for (auto& ft0RecPoint : ft0RecPoints) {
    uint64_t globalBC = ft0RecPoint.getInteractionRecord().toLong();
    bcsMap[globalBC] = 1;
  }

  for (auto& fv0RecPoint : fv0RecPoints) {
    uint64_t globalBC = fv0RecPoint.getInteractionRecord().toLong();
    bcsMap[globalBC] = 1;
  }

  for (auto& zdcRecData : zdcBCRecData) {
    uint64_t globalBC = zdcRecData.ir.toLong();
    bcsMap[globalBC] = 1;
  }

  for (auto& vertex : primVertices) {
    auto& timeStamp = vertex.getTimeStamp();
    double tsTimeStamp = timeStamp.getTimeStamp() * 1E3; // mus to ns
    uint64_t globalBC = relativeTime_to_GlobalBC(tsTimeStamp);
    bcsMap[globalBC] = 1;
  }

  for (auto& emcaltrg : caloEMCCellsTRGR) {
    uint64_t globalBC = emcaltrg.getBCData().toLong();
    bcsMap[globalBC] = 1;
  }

  for (auto& phostrg : caloPHOSCellsTRGR) {
    uint64_t globalBC = phostrg.getBCData().toLong();
    bcsMap[globalBC] = 1;
  }

  for (auto& cpvtrg : cpvTRGR) {
    uint64_t globalBC = cpvtrg.getBCData().toLong();
    bcsMap[globalBC] = 1;
  }

  for (auto& ctpDigit : ctpDigits) {
    uint64_t globalBC = ctpDigit.intRecord.toLong();
    bcsMap[globalBC] = 1;
  }

  int bcID = 0;
  for (auto& item : bcsMap) {
    item.second = bcID;
    bcID++;
  }
}

template <typename TracksCursorType, typename TracksCovCursorType>
void AODProducerWorkflowDPL::addToTracksTable(TracksCursorType& tracksCursor, TracksCovCursorType& tracksCovCursor,
                                              const o2::track::TrackParCov& track, int collisionID, aod::track::TrackTypeEnum type)
{
  tracksCursor(collisionID,
               type,
               truncateFloatFraction(track.getX(), mTrackX),
               truncateFloatFraction(track.getAlpha(), mTrackAlpha),
               track.getY(),
               track.getZ(),
               truncateFloatFraction(track.getSnp(), mTrackSnp),
               truncateFloatFraction(track.getTgl(), mTrackTgl),
               truncateFloatFraction(track.getQ2Pt(), mTrack1Pt));
  // trackscov
  float sY = TMath::Sqrt(track.getSigmaY2()), sZ = TMath::Sqrt(track.getSigmaZ2()), sSnp = TMath::Sqrt(track.getSigmaSnp2()),
        sTgl = TMath::Sqrt(track.getSigmaTgl2()), sQ2Pt = TMath::Sqrt(track.getSigma1Pt2());
  tracksCovCursor(truncateFloatFraction(sY, mTrackCovDiag),
                  truncateFloatFraction(sZ, mTrackCovDiag),
                  truncateFloatFraction(sSnp, mTrackCovDiag),
                  truncateFloatFraction(sTgl, mTrackCovDiag),
                  truncateFloatFraction(sQ2Pt, mTrackCovDiag),
                  (Char_t)(128. * track.getSigmaZY() / (sZ * sY)),
                  (Char_t)(128. * track.getSigmaSnpY() / (sSnp * sY)),
                  (Char_t)(128. * track.getSigmaSnpZ() / (sSnp * sZ)),
                  (Char_t)(128. * track.getSigmaTglY() / (sTgl * sY)),
                  (Char_t)(128. * track.getSigmaTglZ() / (sTgl * sZ)),
                  (Char_t)(128. * track.getSigmaTglSnp() / (sTgl * sSnp)),
                  (Char_t)(128. * track.getSigma1PtY() / (sQ2Pt * sY)),
                  (Char_t)(128. * track.getSigma1PtZ() / (sQ2Pt * sZ)),
                  (Char_t)(128. * track.getSigma1PtSnp() / (sQ2Pt * sSnp)),
                  (Char_t)(128. * track.getSigma1PtTgl() / (sQ2Pt * sTgl)));
}

template <typename TracksExtraCursorType>
void AODProducerWorkflowDPL::addToTracksExtraTable(TracksExtraCursorType& tracksExtraCursor, TrackExtraInfo& extraInfoHolder)
{
  // In case of TPC-only tracks, do not truncate the time error since we encapsulate there a special encoding of
  // the deltaFwd/Bwd times
  auto trackTimeRes = extraInfoHolder.trackTimeRes;
  if (!extraInfoHolder.isTPConly) {
    trackTimeRes = truncateFloatFraction(trackTimeRes, mTrackTimeError);
  }

  // extra
  tracksExtraCursor(truncateFloatFraction(extraInfoHolder.tpcInnerParam, mTrack1Pt),
                    extraInfoHolder.flags,
                    extraInfoHolder.itsClusterSizes,
                    extraInfoHolder.tpcNClsFindable,
                    extraInfoHolder.tpcNClsFindableMinusFound,
                    extraInfoHolder.tpcNClsFindableMinusPID,
                    extraInfoHolder.tpcNClsFindableMinusCrossedRows,
                    extraInfoHolder.tpcNClsShared,
                    extraInfoHolder.trdPattern,
                    truncateFloatFraction(extraInfoHolder.itsChi2NCl, mTrackChi2),
                    truncateFloatFraction(extraInfoHolder.tpcChi2NCl, mTrackChi2),
                    truncateFloatFraction(extraInfoHolder.trdChi2, mTrackChi2),
                    truncateFloatFraction(extraInfoHolder.tofChi2, mTrackChi2),
                    truncateFloatFraction(extraInfoHolder.tpcSignal, mTrackSignal),
                    truncateFloatFraction(extraInfoHolder.trdSignal, mTrackSignal),
                    truncateFloatFraction(extraInfoHolder.length, mTrackSignal),
                    truncateFloatFraction(extraInfoHolder.tofExpMom, mTrack1Pt),
                    truncateFloatFraction(extraInfoHolder.trackEtaEMCAL, mTrackPosEMCAL),
                    truncateFloatFraction(extraInfoHolder.trackPhiEMCAL, mTrackPosEMCAL),
                    truncateFloatFraction(extraInfoHolder.trackTime, mTrackTime),
                    trackTimeRes);
}

template <typename TracksQACursorType>
void AODProducerWorkflowDPL::addToTracksQATable(TracksQACursorType& tracksQACursor, TrackQA& trackQAInfoHolder)
{
  tracksQACursor(
    trackQAInfoHolder.trackID,
    truncateFloatFraction(trackQAInfoHolder.tpcTime0, mTPCTime0),
    truncateFloatFraction(trackQAInfoHolder.tpcdEdxNorm, mTrackSignal),
    trackQAInfoHolder.tpcdcaR,
    trackQAInfoHolder.tpcdcaZ,
    trackQAInfoHolder.tpcClusterByteMask,
    trackQAInfoHolder.tpcdEdxMax0R,
    trackQAInfoHolder.tpcdEdxMax1R,
    trackQAInfoHolder.tpcdEdxMax2R,
    trackQAInfoHolder.tpcdEdxMax3R,
    trackQAInfoHolder.tpcdEdxTot0R,
    trackQAInfoHolder.tpcdEdxTot1R,
    trackQAInfoHolder.tpcdEdxTot2R,
    trackQAInfoHolder.tpcdEdxTot3R,
    trackQAInfoHolder.dRefContY,
    trackQAInfoHolder.dRefContZ,
    trackQAInfoHolder.dRefContSnp,
    trackQAInfoHolder.dRefContTgl,
    trackQAInfoHolder.dRefContQ2Pt,
    trackQAInfoHolder.dRefGloY,
    trackQAInfoHolder.dRefGloZ,
    trackQAInfoHolder.dRefGloSnp,
    trackQAInfoHolder.dRefGloTgl,
    trackQAInfoHolder.dRefGloQ2Pt,
    trackQAInfoHolder.dTofdX,
    trackQAInfoHolder.dTofdZ);
}

template <typename mftTracksCursorType, typename AmbigMFTTracksCursorType>
void AODProducerWorkflowDPL::addToMFTTracksTable(mftTracksCursorType& mftTracksCursor, AmbigMFTTracksCursorType& ambigMFTTracksCursor,
                                                 GIndex trackID, const o2::globaltracking::RecoContainer& data, int collisionID,
                                                 std::uint64_t collisionBC, const std::map<uint64_t, int>& bcsMap)
{
  // mft tracks
  int bcSlice[2] = {-1, -1};
  const auto& track = data.getMFTTrack(trackID);
  const auto& rof = data.getMFTTracksROFRecords()[mMFTROFs[trackID.getIndex()]];
  float trackTime = rof.getBCData().differenceInBC(mStartIR) * o2::constants::lhc::LHCBunchSpacingNS + mMFTROFrameHalfLengthNS + mMFTROFBiasNS;
  float trackTimeRes = mMFTROFrameHalfLengthNS;
  bool needBCSlice = collisionID < 0;
  std::uint64_t bcOfTimeRef;
  if (needBCSlice) {
    double error = mTimeMarginTrackTime + trackTimeRes;
    bcOfTimeRef = fillBCSlice(bcSlice, trackTime - error, trackTime + error, bcsMap);
  } else {
    bcOfTimeRef = collisionBC - mStartIR.toLong(); // by default (unambiguous) track time is wrt collision BC
  }
  trackTime -= bcOfTimeRef * o2::constants::lhc::LHCBunchSpacingNS;

  // the Cellular Automaton track-finding algorithm flag is stored in first of the 4 bits not used for the cluster size
  uint64_t mftClusterSizesAndTrackFlags = track.getClusterSizes();
  mftClusterSizesAndTrackFlags |= (track.isCA()) ? (1ULL << (60)) : 0;

  mftTracksCursor(collisionID,
                  track.getX(),
                  track.getY(),
                  truncateFloatFraction(track.getZ(), mTrackX), // for the forward tracks Z has the same role as X in barrel
                  truncateFloatFraction(track.getPhi(), mTrackAlpha),
                  truncateFloatFraction(track.getTanl(), mTrackTgl),
                  truncateFloatFraction(track.getInvQPt(), mTrack1Pt),
                  mftClusterSizesAndTrackFlags,
                  truncateFloatFraction(track.getTrackChi2(), mTrackChi2),
                  truncateFloatFraction(trackTime, mTrackTime),
                  truncateFloatFraction(trackTimeRes, mTrackTimeError));
  if (needBCSlice) {
    ambigMFTTracksCursor(mTableTrMFTID, bcSlice);
  }
}

template <typename TracksCursorType, typename TracksCovCursorType, typename TracksExtraCursorType, typename TracksQACursorType, typename AmbigTracksCursorType,
          typename MFTTracksCursorType, typename MFTTracksCovCursorType, typename AmbigMFTTracksCursorType,
          typename FwdTracksCursorType, typename FwdTracksCovCursorType, typename AmbigFwdTracksCursorType, typename FwdTrkClsCursorType>
void AODProducerWorkflowDPL::fillTrackTablesPerCollision(int collisionID,
                                                         std::uint64_t collisionBC,
                                                         const o2::dataformats::VtxTrackRef& trackRef,
                                                         const gsl::span<const GIndex>& GIndices,
                                                         const o2::globaltracking::RecoContainer& data,
                                                         TracksCursorType& tracksCursor,
                                                         TracksCovCursorType& tracksCovCursor,
                                                         TracksExtraCursorType& tracksExtraCursor,
                                                         TracksQACursorType& tracksQACursor,
                                                         AmbigTracksCursorType& ambigTracksCursor,
                                                         MFTTracksCursorType& mftTracksCursor,
                                                         MFTTracksCovCursorType& mftTracksCovCursor,
                                                         AmbigMFTTracksCursorType& ambigMFTTracksCursor,
                                                         FwdTracksCursorType& fwdTracksCursor,
                                                         FwdTracksCovCursorType& fwdTracksCovCursor,
                                                         AmbigFwdTracksCursorType& ambigFwdTracksCursor,
                                                         FwdTrkClsCursorType& fwdTrkClsCursor,
                                                         const std::map<uint64_t, int>& bcsMap)
{
  for (int src = GIndex::NSources; src--;) {
    if (!GIndex::isTrackSource(src)) {
      continue;
    }
    int start = trackRef.getFirstEntryOfSource(src);
    int end = start + trackRef.getEntriesOfSource(src);
    int nToReserve = end - start; // + last index for a given table
    if (src == GIndex::Source::MFT) {
      mftTracksCursor.reserve(nToReserve + mftTracksCursor.lastIndex());
    } else if (src == GIndex::Source::MCH || src == GIndex::Source::MFTMCH || src == GIndex::Source::MCHMID) {
      fwdTracksCursor.reserve(nToReserve + fwdTracksCursor.lastIndex());
      fwdTracksCovCursor.reserve(nToReserve + fwdTracksCovCursor.lastIndex());
      if (src == GIndex::Source::MFTMCH) {
        mftTracksCovCursor.reserve(nToReserve + mftTracksCovCursor.lastIndex());
      }
    } else {
      tracksCursor.reserve(nToReserve + tracksCursor.lastIndex());
      tracksCovCursor.reserve(nToReserve + tracksCovCursor.lastIndex());
      tracksExtraCursor.reserve(nToReserve + tracksExtraCursor.lastIndex());
    }
    for (int ti = start; ti < end; ti++) {
      const auto& trackIndex = GIndices[ti];
      if (GIndex::includesSource(src, mInputSources)) {
        if (src == GIndex::Source::MFT) {                                                                // MFT tracks are treated separately since they are stored in a different table
          if (trackIndex.isAmbiguous() && mGIDToTableMFTID.find(trackIndex) != mGIDToTableMFTID.end()) { // was it already stored ?
            continue;
          }
          addToMFTTracksTable(mftTracksCursor, ambigMFTTracksCursor, trackIndex, data, collisionID, collisionBC, bcsMap);
          mGIDToTableMFTID.emplace(trackIndex, mTableTrMFTID);
          mTableTrMFTID++;
        } else if (src == GIndex::Source::MCH || src == GIndex::Source::MFTMCH || src == GIndex::Source::MCHMID) { // FwdTracks tracks are treated separately since they are stored in a different table
          if (trackIndex.isAmbiguous() && mGIDToTableFwdID.find(trackIndex) != mGIDToTableFwdID.end()) {           // was it already stored ?
            continue;
          }
          addToFwdTracksTable(fwdTracksCursor, fwdTracksCovCursor, ambigFwdTracksCursor, mftTracksCovCursor, trackIndex, data, collisionID, collisionBC, bcsMap);
          mGIDToTableFwdID.emplace(trackIndex, mTableTrFwdID);
          addClustersToFwdTrkClsTable(data, fwdTrkClsCursor, trackIndex, mTableTrFwdID);
          mTableTrFwdID++;
        } else {
          // barrel track: normal tracks table
          if (trackIndex.isAmbiguous() && mGIDToTableID.find(trackIndex) != mGIDToTableID.end()) { // was it already stored ?
            continue;
          }

          float weight = 0;
          static std::uniform_real_distribution<> distr(0., 1.);
          bool writeQAData = o2::math_utils::Tsallis::downsampleTsallisCharged(data.getTrackParam(trackIndex).getPt(), mTrackQCFraction, mSqrtS, weight, distr(mGenerator));
          auto extraInfoHolder = processBarrelTrack(collisionID, collisionBC, trackIndex, data, bcsMap);

          if (writeQAData) {
            auto trackQAInfoHolder = processBarrelTrackQA(collisionID, collisionBC, trackIndex, data, bcsMap);
            if (std::bitset<8>(trackQAInfoHolder.tpcClusterByteMask).count() >= mTrackQCNTrCut) {
              trackQAInfoHolder.trackID = mTableTrID;
              // LOGP(info, "orig time0 in bc: {} diffBCRef: {}, ttime: {} -> {}", trackQAInfoHolder.tpcTime0*8, extraInfoHolder.diffBCRef, extraInfoHolder.trackTime, (trackQAInfoHolder.tpcTime0 * 8 - extraInfoHolder.diffBCRef) * o2::constants::lhc::LHCBunchSpacingNS - extraInfoHolder.trackTime);
              trackQAInfoHolder.tpcTime0 = (trackQAInfoHolder.tpcTime0 * 8 - extraInfoHolder.diffBCRef) * o2::constants::lhc::LHCBunchSpacingNS - extraInfoHolder.trackTime;
              // difference between TPC track time0 and stored track nominal time in ns instead of TF start
              addToTracksQATable(tracksQACursor, trackQAInfoHolder);
            } else {
              writeQAData = false;
            }
          }

          // include specific selection of tpc standalone tracks if thinning is active
          if (mThinTracks && extraInfoHolder.isTPConly && !writeQAData) { // if trackQA is written then no check has to be done
            auto trk = data.getTPCTrack(trackIndex);
            if (trk.getNClusters() >= mTrackQCNCls && trk.getPt() >= mTrackQCPt) {
              o2::dataformats::DCA dcaInfo{999.f, 999.f, 999.f, 999.f, 999.f};
              o2::dataformats::VertexBase v = mVtx.getMeanVertex(collisionID < 0 ? 0.f : data.getPrimaryVertex(collisionID).getZ());
              if (o2::base::Propagator::Instance()->propagateToDCABxByBz(v, trk, 2., mMatCorr, &dcaInfo) && std::abs(dcaInfo.getY()) < mTrackQCDCAxy) {
                writeQAData = true; // just setting this to not thin the track
              }
            }
          }

          // Skip thinning if not enabled or track is not tpc standalone or assoc. to a V0 or qa'ed
          if (mThinTracks && src == GIndex::Source::TPC && mGIDUsedBySVtx.find(trackIndex) == mGIDUsedBySVtx.end() && mGIDUsedByStr.find(trackIndex) == mGIDUsedByStr.end() && !writeQAData) {
            mGIDToTableID.emplace(trackIndex, -1); // pretend skipped tracks are stored; this is safe since they are are not written to disk and -1 indicates to all users to not use this track
            continue;
          }

          if (!extraInfoHolder.isTPConly && extraInfoHolder.trackTimeRes < 0.f) { // failed or rejected?
            LOG(warning) << "Barrel track " << trackIndex << " has no time set, rejection is not expected : time=" << extraInfoHolder.trackTime
                         << " timeErr=" << extraInfoHolder.trackTimeRes << " BCSlice: " << extraInfoHolder.bcSlice[0] << ":" << extraInfoHolder.bcSlice[1];
            continue;
          }
          const auto& trOrig = data.getTrackParam(trackIndex);
          bool isProp = false;
          if (mPropTracks && trOrig.getX() < mMinPropR &&
              mGIDUsedBySVtx.find(trackIndex) == mGIDUsedBySVtx.end() &&
              mGIDUsedByStr.find(trackIndex) == mGIDUsedByStr.end()) { // Do not propagate track assoc. to V0s and str. tracking
            auto trackPar(trOrig);
            isProp = propagateTrackToPV(trackPar, data, collisionID);
            if (isProp) {
              addToTracksTable(tracksCursor, tracksCovCursor, trackPar, collisionID, aod::track::Track);
            }
          }
          if (!isProp) {
            addToTracksTable(tracksCursor, tracksCovCursor, trOrig, collisionID, aod::track::TrackIU);
          }
          addToTracksExtraTable(tracksExtraCursor, extraInfoHolder);

          //  collecting table indices of barrel tracks for V0s table
          if (extraInfoHolder.bcSlice[0] >= 0 && collisionID < 0) {
            ambigTracksCursor(mTableTrID, extraInfoHolder.bcSlice);
          }
          mGIDToTableID.emplace(trackIndex, mTableTrID);
          mTableTrID++;
        }
      }
    }
  }
  if (collisionID < 0) {
    return;
  }
  /// Add strangeness tracks to the table
  auto sTracks = data.getStrangeTracks();
  tracksCursor.reserve(mVertexStrLUT[collisionID + 1] + tracksCursor.lastIndex());
  tracksCovCursor.reserve(mVertexStrLUT[collisionID + 1] + tracksCovCursor.lastIndex());
  tracksExtraCursor.reserve(mVertexStrLUT[collisionID + 1] + tracksExtraCursor.lastIndex());
  for (int iS{mVertexStrLUT[collisionID]}; iS < mVertexStrLUT[collisionID + 1]; ++iS) {
    auto& collStrTrk = mCollisionStrTrk[iS];
    auto& sTrk = sTracks[collStrTrk.second];
    TrackExtraInfo extraInfo;
    extraInfo.itsChi2NCl = sTrk.mTopoChi2; // TODO: this is the total chi2 of adding the ITS clusters, the topology chi2 meaning might change in the future
    extraInfo.itsClusterSizes = sTrk.getClusterSizes();
    addToTracksTable(tracksCursor, tracksCovCursor, sTrk.mMother, collisionID, aod::track::StrangeTrack);
    addToTracksExtraTable(tracksExtraCursor, extraInfo);
    mStrTrkIndices[collStrTrk.second] = mTableTrID;
    mTableTrID++;
  }
}

void AODProducerWorkflowDPL::fillIndexTablesPerCollision(const o2::dataformats::VtxTrackRef& trackRef, const gsl::span<const GIndex>& GIndices, const o2::globaltracking::RecoContainer& data)
{
  const auto& mchmidMatches = data.getMCHMIDMatches();

  for (int src : {GIndex::Source::MCHMID, GIndex::Source::MFTMCH, GIndex::Source::MCH, GIndex::Source::MFT}) {
    int start = trackRef.getFirstEntryOfSource(src);
    int end = start + trackRef.getEntriesOfSource(src);
    for (int ti = start; ti < end; ti++) {
      auto& trackIndex = GIndices[ti];
      if (GIndex::includesSource(src, mInputSources)) {
        if (src == GIndex::Source::MFT) {
          if (trackIndex.isAmbiguous() && mGIDToTableMFTID.find(trackIndex) != mGIDToTableMFTID.end()) {
            continue;
          }
          mGIDToTableMFTID.emplace(trackIndex, mIndexMFTID);
          mIndexTableMFT[trackIndex.getIndex()] = mIndexMFTID;
          mIndexMFTID++;
        } else if (src == GIndex::Source::MCH || src == GIndex::Source::MFTMCH || src == GIndex::Source::MCHMID) {
          if (trackIndex.isAmbiguous() && mGIDToTableFwdID.find(trackIndex) != mGIDToTableFwdID.end()) {
            continue;
          }
          mGIDToTableFwdID.emplace(trackIndex, mIndexFwdID);
          if (src == GIndex::Source::MCH) {
            mIndexTableFwd[trackIndex.getIndex()] = mIndexFwdID;
          } else if (src == GIndex::Source::MCHMID) {
            const auto& mchmidMatch = mchmidMatches[trackIndex.getIndex()];
            const auto mchTrackID = mchmidMatch.getMCHRef().getIndex();
            mIndexTableFwd[mchTrackID] = mIndexFwdID;
          }
          mIndexFwdID++;
        }
      }
    }
  }
}

template <typename FwdTracksCursorType, typename FwdTracksCovCursorType, typename AmbigFwdTracksCursorType, typename mftTracksCovCursorType>
void AODProducerWorkflowDPL::addToFwdTracksTable(FwdTracksCursorType& fwdTracksCursor, FwdTracksCovCursorType& fwdTracksCovCursor,
                                                 AmbigFwdTracksCursorType& ambigFwdTracksCursor, mftTracksCovCursorType& mftTracksCovCursor, GIndex trackID,
                                                 const o2::globaltracking::RecoContainer& data, int collisionID, std::uint64_t collisionBC,
                                                 const std::map<uint64_t, int>& bcsMap)
{
  const auto& mchTracks = data.getMCHTracks();
  const auto& midTracks = data.getMIDTracks();
  const auto& mchmidMatches = data.getMCHMIDMatches();
  const auto& mchClusters = data.getMCHTrackClusters();

  FwdTrackInfo fwdInfo;
  FwdTrackCovInfo fwdCovInfo;
  int bcSlice[2] = {-1, -1};

  // helper lambda for mch bitmap -- common for global and standalone tracks
  auto getMCHBitMap = [&](int mchTrackID) {
    if (mchTrackID != -1) { // check matching just in case
      const auto& mchTrack = mchTracks[mchTrackID];
      int first = mchTrack.getFirstClusterIdx();
      int last = mchTrack.getLastClusterIdx();
      for (int i = first; i <= last; i++) { // check chamberIds of all clusters
        const auto& cluster = mchClusters[i];
        int chamberId = cluster.getChamberId();
        fwdInfo.mchBitMap |= 1 << chamberId;
      }
    }
  };

  auto getMIDBitMapBoards = [&](int midTrackID) {
    if (midTrackID != -1) { // check matching just in case
      const auto& midTrack = midTracks[midTrackID];
      fwdInfo.midBitMap = midTrack.getHitMap();
      fwdInfo.midBoards = midTrack.getEfficiencyWord();
    }
  };

  auto extrapMCHTrack = [&](int mchTrackID) {
    const auto& track = mchTracks[mchTrackID];

    // mch standalone tracks extrapolated to vertex
    // compute 3 sets of tracks parameters :
    // - at vertex
    // - at DCA
    // - at the end of the absorber
    // extrapolate to vertex
    float vx = 0, vy = 0, vz = 0;
    if (collisionID >= 0) {
      const auto& v = data.getPrimaryVertex(collisionID);
      vx = v.getX();
      vy = v.getY();
      vz = v.getZ();
    }

    o2::mch::TrackParam trackParamAtVertex(track.getZ(), track.getParameters(), track.getCovariances());
    if (mPropMuons) {
      double errVtx{0.0}; // FIXME: get errors associated with vertex if available
      double errVty{0.0};
      if (!o2::mch::TrackExtrap::extrapToVertex(trackParamAtVertex, vx, vy, vz, errVtx, errVty)) {
        return false;
      }
    }

    // extrapolate to DCA
    o2::mch::TrackParam trackParamAtDCA(track.getZ(), track.getParameters());
    if (!o2::mch::TrackExtrap::extrapToVertexWithoutBranson(trackParamAtDCA, vz)) {
      return false;
    }

    // extrapolate to the end of the absorber
    o2::mch::TrackParam trackParamAtRAbs(track.getZ(), track.getParameters());
    if (!o2::mch::TrackExtrap::extrapToZ(trackParamAtRAbs, -505.)) { // FIXME: replace hardcoded 505
      return false;
    }

    double dcaX = trackParamAtDCA.getNonBendingCoor() - vx;
    double dcaY = trackParamAtDCA.getBendingCoor() - vy;
    double dca = std::sqrt(dcaX * dcaX + dcaY * dcaY);

    double xAbs = trackParamAtRAbs.getNonBendingCoor();
    double yAbs = trackParamAtRAbs.getBendingCoor();

    double dpdca = track.getP() * dca;
    double dchi2 = track.getChi2OverNDF();

    auto fwdmuon = mMatching.MCHtoFwd(trackParamAtVertex);

    fwdInfo.x = fwdmuon.getX();
    fwdInfo.y = fwdmuon.getY();
    fwdInfo.z = fwdmuon.getZ();
    fwdInfo.phi = fwdmuon.getPhi();
    fwdInfo.tanl = fwdmuon.getTgl();
    fwdInfo.invqpt = fwdmuon.getInvQPt();
    fwdInfo.rabs = std::sqrt(xAbs * xAbs + yAbs * yAbs);
    fwdInfo.chi2 = dchi2;
    fwdInfo.pdca = dpdca;
    fwdInfo.nClusters = track.getNClusters();

    fwdCovInfo.sigX = TMath::Sqrt(fwdmuon.getCovariances()(0, 0));
    fwdCovInfo.sigY = TMath::Sqrt(fwdmuon.getCovariances()(1, 1));
    fwdCovInfo.sigPhi = TMath::Sqrt(fwdmuon.getCovariances()(2, 2));
    fwdCovInfo.sigTgl = TMath::Sqrt(fwdmuon.getCovariances()(3, 3));
    fwdCovInfo.sig1Pt = TMath::Sqrt(fwdmuon.getCovariances()(4, 4));
    fwdCovInfo.rhoXY = (Char_t)(128. * fwdmuon.getCovariances()(0, 1) / (fwdCovInfo.sigX * fwdCovInfo.sigY));
    fwdCovInfo.rhoPhiX = (Char_t)(128. * fwdmuon.getCovariances()(0, 2) / (fwdCovInfo.sigPhi * fwdCovInfo.sigX));
    fwdCovInfo.rhoPhiY = (Char_t)(128. * fwdmuon.getCovariances()(1, 2) / (fwdCovInfo.sigPhi * fwdCovInfo.sigY));
    fwdCovInfo.rhoTglX = (Char_t)(128. * fwdmuon.getCovariances()(0, 3) / (fwdCovInfo.sigTgl * fwdCovInfo.sigX));
    fwdCovInfo.rhoTglY = (Char_t)(128. * fwdmuon.getCovariances()(1, 3) / (fwdCovInfo.sigTgl * fwdCovInfo.sigY));
    fwdCovInfo.rhoTglPhi = (Char_t)(128. * fwdmuon.getCovariances()(2, 3) / (fwdCovInfo.sigTgl * fwdCovInfo.sigPhi));
    fwdCovInfo.rho1PtX = (Char_t)(128. * fwdmuon.getCovariances()(0, 4) / (fwdCovInfo.sig1Pt * fwdCovInfo.sigX));
    fwdCovInfo.rho1PtY = (Char_t)(128. * fwdmuon.getCovariances()(1, 4) / (fwdCovInfo.sig1Pt * fwdCovInfo.sigY));
    fwdCovInfo.rho1PtPhi = (Char_t)(128. * fwdmuon.getCovariances()(2, 4) / (fwdCovInfo.sig1Pt * fwdCovInfo.sigPhi));
    fwdCovInfo.rho1PtTgl = (Char_t)(128. * fwdmuon.getCovariances()(3, 4) / (fwdCovInfo.sig1Pt * fwdCovInfo.sigTgl));

    return true;
  };

  if (trackID.getSource() == GIndex::MCH) { // This is an MCH track
    int mchTrackID = trackID.getIndex();
    getMCHBitMap(mchTrackID);
    if (!extrapMCHTrack(mchTrackID)) {
      LOGF(warn, "Unable to extrapolate MCH track with ID %d! Dummy parameters will be used", mchTrackID);
    }
    fwdInfo.trackTypeId = o2::aod::fwdtrack::MCHStandaloneTrack;
    const auto& rof = data.getMCHTracksROFRecords()[mMCHROFs[mchTrackID]];
    auto time = rof.getTimeMUS(mStartIR).first;
    fwdInfo.trackTime = time.getTimeStamp() * 1.e3;
    fwdInfo.trackTimeRes = time.getTimeStampError() * 1.e3;
  } else if (trackID.getSource() == GIndex::MCHMID) { // This is an MCH-MID track
    fwdInfo.trackTypeId = o2::aod::fwdtrack::MuonStandaloneTrack;
    auto mchmidMatch = mchmidMatches[trackID.getIndex()];
    auto mchTrackID = mchmidMatch.getMCHRef().getIndex();
    if (!extrapMCHTrack(mchTrackID)) {
      LOGF(warn, "Unable to extrapolate MCH track with ID %d! Dummy parameters will be used", mchTrackID);
    }
    auto midTrackID = mchmidMatch.getMIDRef().getIndex();
    fwdInfo.chi2matchmchmid = mchmidMatch.getMatchChi2OverNDF();
    getMCHBitMap(mchTrackID);
    getMIDBitMapBoards(midTrackID);
    auto time = mchmidMatch.getTimeMUS(mStartIR).first;
    fwdInfo.trackTime = time.getTimeStamp() * 1.e3;
    fwdInfo.trackTimeRes = time.getTimeStampError() * 1.e3;
  } else { // This is a GlobalMuonTrack or a GlobalForwardTrack
    const auto& track = data.getGlobalFwdTrack(trackID);
    const auto& mftTracks = data.getMFTTracks();
    const auto& mfttrack = mftTracks[track.getMFTTrackID()];
    if (!extrapMCHTrack(track.getMCHTrackID())) {
      LOGF(warn, "Unable to extrapolate MCH track with ID %d! Dummy parameters will be used", track.getMCHTrackID());
    }
    fwdInfo.x = track.getX();
    fwdInfo.y = track.getY();
    fwdInfo.z = track.getZ();
    fwdInfo.phi = track.getPhi();
    fwdInfo.tanl = track.getTanl();
    fwdInfo.invqpt = track.getInvQPt();
    fwdInfo.chi2 = track.getTrackChi2();
    // fwdInfo.nClusters = track.getNumberOfPoints();
    fwdInfo.chi2matchmchmid = track.getMIDMatchingChi2();
    fwdInfo.chi2matchmchmft = track.getMFTMCHMatchingChi2();
    fwdInfo.matchscoremchmft = track.getMFTMCHMatchingScore();
    fwdInfo.matchmfttrackid = mIndexTableMFT[track.getMFTTrackID()];
    fwdInfo.matchmchtrackid = mIndexTableFwd[track.getMCHTrackID()];
    fwdInfo.trackTime = track.getTimeMUS().getTimeStamp() * 1.e3;
    fwdInfo.trackTimeRes = track.getTimeMUS().getTimeStampError() * 1.e3;

    getMCHBitMap(track.getMCHTrackID());
    getMIDBitMapBoards(track.getMIDTrackID());

    fwdCovInfo.sigX = TMath::Sqrt(track.getCovariances()(0, 0));
    fwdCovInfo.sigY = TMath::Sqrt(track.getCovariances()(1, 1));
    fwdCovInfo.sigPhi = TMath::Sqrt(track.getCovariances()(2, 2));
    fwdCovInfo.sigTgl = TMath::Sqrt(track.getCovariances()(3, 3));
    fwdCovInfo.sig1Pt = TMath::Sqrt(track.getCovariances()(4, 4));
    fwdCovInfo.rhoXY = (Char_t)(128. * track.getCovariances()(0, 1) / (fwdCovInfo.sigX * fwdCovInfo.sigY));
    fwdCovInfo.rhoPhiX = (Char_t)(128. * track.getCovariances()(0, 2) / (fwdCovInfo.sigPhi * fwdCovInfo.sigX));
    fwdCovInfo.rhoPhiY = (Char_t)(128. * track.getCovariances()(1, 2) / (fwdCovInfo.sigPhi * fwdCovInfo.sigY));
    fwdCovInfo.rhoTglX = (Char_t)(128. * track.getCovariances()(0, 3) / (fwdCovInfo.sigTgl * fwdCovInfo.sigX));
    fwdCovInfo.rhoTglY = (Char_t)(128. * track.getCovariances()(1, 3) / (fwdCovInfo.sigTgl * fwdCovInfo.sigY));
    fwdCovInfo.rhoTglPhi = (Char_t)(128. * track.getCovariances()(2, 3) / (fwdCovInfo.sigTgl * fwdCovInfo.sigPhi));
    fwdCovInfo.rho1PtX = (Char_t)(128. * track.getCovariances()(0, 4) / (fwdCovInfo.sig1Pt * fwdCovInfo.sigX));
    fwdCovInfo.rho1PtY = (Char_t)(128. * track.getCovariances()(1, 4) / (fwdCovInfo.sig1Pt * fwdCovInfo.sigY));
    fwdCovInfo.rho1PtPhi = (Char_t)(128. * track.getCovariances()(2, 4) / (fwdCovInfo.sig1Pt * fwdCovInfo.sigPhi));
    fwdCovInfo.rho1PtTgl = (Char_t)(128. * track.getCovariances()(3, 4) / (fwdCovInfo.sig1Pt * fwdCovInfo.sigTgl));

    fwdInfo.trackTypeId = (fwdInfo.chi2matchmchmid >= 0) ? o2::aod::fwdtrack::GlobalMuonTrack : o2::aod::fwdtrack::GlobalForwardTrack;

    float sX = TMath::Sqrt(mfttrack.getSigma2X()), sY = TMath::Sqrt(mfttrack.getSigma2Y()), sPhi = TMath::Sqrt(mfttrack.getSigma2Phi()),
          sTgl = TMath::Sqrt(mfttrack.getSigma2Tanl()), sQ2Pt = TMath::Sqrt(mfttrack.getSigma2InvQPt());

    mftTracksCovCursor(fwdInfo.matchmfttrackid,
                       truncateFloatFraction(sX, mTrackCovDiag),
                       truncateFloatFraction(sY, mTrackCovDiag),
                       truncateFloatFraction(sPhi, mTrackCovDiag),
                       truncateFloatFraction(sTgl, mTrackCovDiag),
                       truncateFloatFraction(sQ2Pt, mTrackCovDiag),
                       (Char_t)(128. * mfttrack.getCovariances()(0, 1) / (sX * sY)),
                       (Char_t)(128. * mfttrack.getCovariances()(0, 2) / (sPhi * sX)),
                       (Char_t)(128. * mfttrack.getCovariances()(1, 2) / (sPhi * sY)),
                       (Char_t)(128. * mfttrack.getCovariances()(0, 3) / (sTgl * sX)),
                       (Char_t)(128. * mfttrack.getCovariances()(1, 3) / (sTgl * sY)),
                       (Char_t)(128. * mfttrack.getCovariances()(2, 3) / (sTgl * sPhi)),
                       (Char_t)(128. * mfttrack.getCovariances()(0, 4) / (sQ2Pt * sX)),
                       (Char_t)(128. * mfttrack.getCovariances()(1, 4) / (sQ2Pt * sY)),
                       (Char_t)(128. * mfttrack.getCovariances()(2, 4) / (sQ2Pt * sPhi)),
                       (Char_t)(128. * mfttrack.getCovariances()(3, 4) / (sQ2Pt * sTgl)));
  }

  std::uint64_t bcOfTimeRef;
  bool needBCSlice = collisionID < 0;
  if (needBCSlice) { // need to store BC slice
    float err = mTimeMarginTrackTime + fwdInfo.trackTimeRes;
    bcOfTimeRef = fillBCSlice(bcSlice, fwdInfo.trackTime - err, fwdInfo.trackTime + err, bcsMap);
  } else {
    bcOfTimeRef = collisionBC - mStartIR.toLong(); // by default track time is wrt collision BC (unless no collision assigned)
  }
  fwdInfo.trackTime -= bcOfTimeRef * o2::constants::lhc::LHCBunchSpacingNS;

  fwdTracksCursor(collisionID,
                  fwdInfo.trackTypeId,
                  fwdInfo.x,
                  fwdInfo.y,
                  truncateFloatFraction(fwdInfo.z, mTrackX), // for the forward tracks Z has the same role as X in the barrel
                  truncateFloatFraction(fwdInfo.phi, mTrackAlpha),
                  truncateFloatFraction(fwdInfo.tanl, mTrackTgl),
                  truncateFloatFraction(fwdInfo.invqpt, mTrack1Pt),
                  fwdInfo.nClusters,
                  truncateFloatFraction(fwdInfo.pdca, mTrackX),
                  truncateFloatFraction(fwdInfo.rabs, mTrackX),
                  truncateFloatFraction(fwdInfo.chi2, mTrackChi2),
                  truncateFloatFraction(fwdInfo.chi2matchmchmid, mTrackChi2),
                  truncateFloatFraction(fwdInfo.chi2matchmchmft, mTrackChi2),
                  truncateFloatFraction(fwdInfo.matchscoremchmft, mTrackChi2),
                  fwdInfo.matchmfttrackid,
                  fwdInfo.matchmchtrackid,
                  fwdInfo.mchBitMap,
                  fwdInfo.midBitMap,
                  fwdInfo.midBoards,
                  truncateFloatFraction(fwdInfo.trackTime, mTrackTime),
                  truncateFloatFraction(fwdInfo.trackTimeRes, mTrackTimeError));

  fwdTracksCovCursor(truncateFloatFraction(fwdCovInfo.sigX, mTrackCovDiag),
                     truncateFloatFraction(fwdCovInfo.sigY, mTrackCovDiag),
                     truncateFloatFraction(fwdCovInfo.sigPhi, mTrackCovDiag),
                     truncateFloatFraction(fwdCovInfo.sigTgl, mTrackCovDiag),
                     truncateFloatFraction(fwdCovInfo.sig1Pt, mTrackCovDiag),
                     fwdCovInfo.rhoXY,
                     fwdCovInfo.rhoPhiX,
                     fwdCovInfo.rhoPhiY,
                     fwdCovInfo.rhoTglX,
                     fwdCovInfo.rhoTglY,
                     fwdCovInfo.rhoTglPhi,
                     fwdCovInfo.rho1PtX,
                     fwdCovInfo.rho1PtY,
                     fwdCovInfo.rho1PtPhi,
                     fwdCovInfo.rho1PtTgl);

  if (needBCSlice) {
    ambigFwdTracksCursor(mTableTrFwdID, bcSlice);
  }
}

//------------------------------------------------------------------
void AODProducerWorkflowDPL::updateMCHeader(MCCollisionCursor& collisionCursor,
                                            XSectionCursor& xSectionCursor,
                                            PdfInfoCursor& pdfInfoCursor,
                                            HeavyIonCursor& heavyIonCursor,
                                            const MCEventHeader& header,
                                            int collisionID,
                                            int bcID,
                                            float time,
                                            short generatorID,
                                            int sourceID)
{
  using aodmchelpers::updateHepMCHeavyIon;
  using aodmchelpers::updateHepMCPdfInfo;
  using aodmchelpers::updateHepMCXSection;
  using aodmchelpers::updateMCCollisions;

  auto genID = updateMCCollisions(collisionCursor,
                                  bcID,
                                  time,
                                  header,
                                  generatorID,
                                  sourceID,
                                  mCollisionPosition);
  mXSectionUpdate = (updateHepMCXSection(xSectionCursor, //
                                         collisionID,    //
                                         genID,          //
                                         header,         //
                                         mXSectionUpdate)
                       ? HepMCUpdate::always
                       : HepMCUpdate::never);
  mPdfInfoUpdate = (updateHepMCPdfInfo(pdfInfoCursor, //
                                       collisionID,   //
                                       genID,         //
                                       header,        //
                                       mPdfInfoUpdate)
                      ? HepMCUpdate::always
                      : HepMCUpdate::never);
  mHeavyIonUpdate = (updateHepMCHeavyIon(heavyIonCursor, //
                                         collisionID,    //
                                         genID,          //
                                         header,
                                         mHeavyIonUpdate)
                       ? HepMCUpdate::always
                       : HepMCUpdate::never);
}

void dimensionMCKeepStore(std::vector<std::vector<std::unordered_map<int, int>>>& store, int Nsources, int NEvents)
{
  store.resize(Nsources);
  for (int s = 0; s < Nsources; ++s) {
    store[s].resize(NEvents);
  }
}

void clearMCKeepStore(std::vector<std::vector<std::unordered_map<int, int>>>& store)
{
  for (auto s = 0U; s < store.size(); ++s) {
    for (auto e = 0U; e < store[s].size(); ++e) {
      store[s][e].clear();
    }
  }
}

// helper function to add a particle/track to the MC keep store
void keepMCParticle(std::vector<std::vector<std::unordered_map<int, int>>>& store, int source, int event, int track, int value = 1)
{
  if (track < 0) {
    LOG(warn) << "trackID is smaller than 0. Neglecting";
    return;
  }
  store[source][event][track] = value;
}

void AODProducerWorkflowDPL::fillMCParticlesTable(o2::steer::MCKinematicsReader& mcReader,
                                                  MCParticlesCursor& mcParticlesCursor,
                                                  const gsl::span<const o2::dataformats::VtxTrackRef>& primVer2TRefs,
                                                  const gsl::span<const GIndex>& GIndices,
                                                  const o2::globaltracking::RecoContainer& data,
                                                  const std::vector<std::vector<int>>& mcColToEvSrc)
{
  int NSources = 0;
  int NEvents = 0;
  for (auto& p : mcColToEvSrc) {
    NSources = std::max(p[1], NSources);
    NEvents = std::max(p[2], NEvents);
  }
  NSources++; // 0 - indexed
  NEvents++;
  LOG(info) << " number of events " << NEvents;
  LOG(info) << " number of sources " << NSources;
  dimensionMCKeepStore(mToStore, NSources, NEvents);

  std::vector<int> particleIDsToKeep;

  auto markMCTrackForSrc = [&](std::array<GID, GID::NSources>& contributorsGID, uint8_t src) {
    auto mcLabel = data.getTrackMCLabel(contributorsGID[src]);
    if (!mcLabel.isValid()) {
      return;
    }
    keepMCParticle(mToStore, mcLabel.getSourceID(), mcLabel.getEventID(), mcLabel.getTrackID());
  };

  // mark reconstructed MC particles to store them into the table
  for (auto& trackRef : primVer2TRefs) {
    for (int src = GIndex::NSources; src--;) {
      int start = trackRef.getFirstEntryOfSource(src);
      int end = start + trackRef.getEntriesOfSource(src);
      for (int ti = start; ti < end; ti++) {
        auto& trackIndex = GIndices[ti];
        if (GIndex::includesSource(src, mInputSources)) {
          auto mcTruth = data.getTrackMCLabel(trackIndex);
          if (!mcTruth.isValid()) {
            continue;
          }
          keepMCParticle(mToStore, mcTruth.getSourceID(), mcTruth.getEventID(), mcTruth.getTrackID());
          // treating contributors of global tracks
          auto contributorsGID = data.getSingleDetectorRefs(trackIndex);
          if (contributorsGID[GIndex::Source::TPC].isIndexSet()) {
            markMCTrackForSrc(contributorsGID, GIndex::Source::TPC);
          }
          if (contributorsGID[GIndex::Source::ITS].isIndexSet()) {
            markMCTrackForSrc(contributorsGID, GIndex::Source::ITS);
          }
          if (contributorsGID[GIndex::Source::TOF].isIndexSet()) {
            const auto& labelsTOF = data.getTOFClustersMCLabels()->getLabels(contributorsGID[GIndex::Source::TOF]);
            for (auto& mcLabel : labelsTOF) {
              if (!mcLabel.isValid()) {
                continue;
              }
              keepMCParticle(mToStore, mcLabel.getSourceID(), mcLabel.getEventID(), mcLabel.getTrackID());
            }
          }
        }
      }
    }
  }
  // mark calorimeter signals as reconstructed particles
  if (mInputSources[GIndex::EMC]) {
    auto& mcCaloEMCCellLabels = data.getEMCALCellsMCLabels()->getTruthArray();
    for (auto& mcTruth : mcCaloEMCCellLabels) {
      if (!mcTruth.isValid()) {
        continue;
      }
      keepMCParticle(mToStore, mcTruth.getSourceID(), mcTruth.getEventID(), mcTruth.getTrackID());
    }
  }
  if (mInputSources[GIndex::PHS]) {
    auto& mcCaloPHOSCellLabels = data.getPHOSCellsMCLabels()->getTruthArray();
    for (auto& mcTruth : mcCaloPHOSCellLabels) {
      if (!mcTruth.isValid()) {
        continue;
      }
      keepMCParticle(mToStore, mcTruth.getSourceID(), mcTruth.getEventID(), mcTruth.getTrackID());
    }
  }
  using namespace aodmchelpers;
  using MCTrackNavigator = o2::mcutils::MCTrackNavigator;

  size_t offset = 0;
  for (auto& colInfo : mcColToEvSrc) { // loop over "<eventID, sourceID> <-> combined MC col. ID" key pairs
    int event = colInfo[2];
    int source = colInfo[1];
    int mcColId = colInfo[0];
    std::vector<MCTrack> const& mcParticles = mcReader.getTracks(source, event);
    LOG(debug) << "Event=" << event << " source=" << source << " collision=" << mcColId;
    auto& preselect = mToStore[source][event];

    offset = updateParticles(mcParticlesCursor,
                             mcColId,
                             mcParticles,
                             preselect,
                             offset,
                             mRecoOnly,
                             source == 0, // background
                             mMcParticleW,
                             mMcParticleMom,
                             mMcParticlePos);

    mcReader.releaseTracksForSourceAndEvent(source, event);
  }
}

template <typename MCTrackLabelCursorType, typename MCMFTTrackLabelCursorType, typename MCFwdTrackLabelCursorType>
void AODProducerWorkflowDPL::fillMCTrackLabelsTable(MCTrackLabelCursorType& mcTrackLabelCursor,
                                                    MCMFTTrackLabelCursorType& mcMFTTrackLabelCursor,
                                                    MCFwdTrackLabelCursorType& mcFwdTrackLabelCursor,
                                                    const o2::dataformats::VtxTrackRef& trackRef,
                                                    const gsl::span<const GIndex>& primVerGIs,
                                                    const o2::globaltracking::RecoContainer& data,
                                                    int vertexId)
{
  // labelMask (temporary) usage:
  //   bit 13 -- ITS/TPC with ITS label (track of AB tracklet) different from TPC
  //   bit 14 -- isNoise() == true
  //   bit 15 -- isFake() == true (defined by the fakeness of the top level global track, i.e. if TOF is present, fake means that the track of the TPC label does not contribute to TOF cluster)
  // labelID = -1 -- label is not set

  for (int src = GIndex::NSources; src--;) {
    int start = trackRef.getFirstEntryOfSource(src);
    int end = start + trackRef.getEntriesOfSource(src);
    mcMFTTrackLabelCursor.reserve(end - start + mcMFTTrackLabelCursor.lastIndex());
    mcFwdTrackLabelCursor.reserve(end - start + mcFwdTrackLabelCursor.lastIndex());
    mcTrackLabelCursor.reserve(end - start + mcTrackLabelCursor.lastIndex());
    for (int ti = start; ti < end; ti++) {
      const auto trackIndex = primVerGIs[ti];

      // check if the label was already stored (or the track was rejected for some reason in the fillTrackTablesPerCollision)
      auto needToStore = [trackIndex](std::unordered_map<GIndex, int>& mp) {
        auto entry = mp.find(trackIndex);
        if (entry == mp.end() || entry->second == -1) {
          return false;
        }
        entry->second = -1;
        return true;
      };

      if (GIndex::includesSource(src, mInputSources)) {
        auto mcTruth = data.getTrackMCLabel(trackIndex);
        MCLabels labelHolder{};
        if ((src == GIndex::Source::MFT) || (src == GIndex::Source::MFTMCH) || (src == GIndex::Source::MCH) || (src == GIndex::Source::MCHMID)) { // treating mft and fwd labels separately
          if (!needToStore(src == GIndex::Source::MFT ? mGIDToTableMFTID : mGIDToTableFwdID)) {
            continue;
          }
          if (mcTruth.isValid()) { // if not set, -1 will be stored
            labelHolder.labelID = (mToStore[mcTruth.getSourceID()][mcTruth.getEventID()])[mcTruth.getTrackID()];
          }
          if (mcTruth.isFake()) {
            labelHolder.fwdLabelMask |= (0x1 << 7);
          }
          if (mcTruth.isNoise()) {
            labelHolder.fwdLabelMask |= (0x1 << 6);
          }
          if (src == GIndex::Source::MFT) {
            mcMFTTrackLabelCursor(labelHolder.labelID,
                                  labelHolder.fwdLabelMask);
          } else {
            mcFwdTrackLabelCursor(labelHolder.labelID,
                                  labelHolder.fwdLabelMask);
          }
        } else {
          if (!needToStore(mGIDToTableID)) {
            continue;
          }
          if (mcTruth.isValid()) { // if not set, -1 will be stored
            labelHolder.labelID = (mToStore[mcTruth.getSourceID()][mcTruth.getEventID()])[mcTruth.getTrackID()]; // defined by TPC if it contributes, otherwise: by ITS
            if (mcTruth.isFake()) {
              labelHolder.labelMask |= (0x1 << 15);
            }
            if (trackIndex.includesDet(DetID::TPC) && trackIndex.getSource() != GIndex::Source::TPC) { // this is global track
              auto contributorsGID = data.getSingleDetectorRefs(trackIndex);
              if (contributorsGID[GIndex::Source::ITSTPC].isIndexSet()) { // there is a match to ITS tracks or ITSAB tracklet!
                if (data.getTrackMCLabel(contributorsGID[GIndex::Source::ITSTPC]).isFake()) {
                  labelHolder.labelMask |= (0x1 << 13);
                }
              }
            }
          } else if (mcTruth.isNoise()) {
            labelHolder.labelMask |= (0x1 << 14);
          }
          mcTrackLabelCursor(labelHolder.labelID, labelHolder.labelMask);
        }
      }
    }
  }

  // filling the tables with the strangeness tracking labels
  auto sTrackLabels = data.getStrangeTracksMCLabels();
  // check if vertexId and vertexId + 1 maps into mVertexStrLUT
  if (!(vertexId < 0 || vertexId >= mVertexStrLUT.size() - 1)) {
    mcTrackLabelCursor.reserve(mVertexStrLUT[vertexId + 1] + mcTrackLabelCursor.lastIndex());
    for (int iS{mVertexStrLUT[vertexId]}; iS < mVertexStrLUT[vertexId + 1]; ++iS) {
      auto& collStrTrk = mCollisionStrTrk[iS];
      auto& label = sTrackLabels[collStrTrk.second];
      MCLabels labelHolder;
      labelHolder.labelID = label.isValid() ? (mToStore[label.getSourceID()][label.getEventID()])[label.getTrackID()] : -1;
      labelHolder.labelMask = (label.isFake() << 15) | (label.isNoise() << 14);
      mcTrackLabelCursor(labelHolder.labelID, labelHolder.labelMask);
    }
  }
}

template <typename V0CursorType, typename CascadeCursorType, typename Decay3BodyCursorType>
void AODProducerWorkflowDPL::fillSecondaryVertices(const o2::globaltracking::RecoContainer& recoData, V0CursorType& v0Cursor, CascadeCursorType& cascadeCursor, Decay3BodyCursorType& decay3BodyCursor)
{

  auto v0s = recoData.getV0sIdx();
  auto cascades = recoData.getCascadesIdx();
  auto decays3Body = recoData.getDecays3BodyIdx();

  v0Cursor.reserve(v0s.size());
  // filling v0s table
  for (size_t iv0 = 0; iv0 < v0s.size(); iv0++) {
    const auto& v0 = v0s[iv0];
    auto trPosID = v0.getProngID(0);
    auto trNegID = v0.getProngID(1);
    uint8_t v0flags = v0.getBits();
    int posTableIdx = -1, negTableIdx = -1, collID = -1;
    auto item = mGIDToTableID.find(trPosID);
    if (item != mGIDToTableID.end()) {
      posTableIdx = item->second;
    } else {
      LOG(warn) << "Could not find a positive track index for prong ID " << trPosID;
    }
    item = mGIDToTableID.find(trNegID);
    if (item != mGIDToTableID.end()) {
      negTableIdx = item->second;
    } else {
      LOG(warn) << "Could not find a negative track index for prong ID " << trNegID;
    }
    auto itemV = mVtxToTableCollID.find(v0.getVertexID());
    if (itemV == mVtxToTableCollID.end()) {
      LOG(warn) << "Could not find V0 collisionID for the vertex ID " << v0.getVertexID();
    } else {
      collID = itemV->second;
    }
    if (posTableIdx != -1 and negTableIdx != -1 and collID != -1) {
      v0Cursor(collID, posTableIdx, negTableIdx, v0flags);
      mV0ToTableID[int(iv0)] = mTableV0ID++;
    }
  }

  // filling cascades table
  cascadeCursor.reserve(cascades.size());
  for (auto& cascade : cascades) {
    auto itemV0 = mV0ToTableID.find(cascade.getV0ID());
    if (itemV0 == mV0ToTableID.end()) {
      continue;
    }
    int v0tableID = itemV0->second, bachTableIdx = -1, collID = -1;
    auto bachelorID = cascade.getBachelorID();
    auto item = mGIDToTableID.find(bachelorID);
    if (item != mGIDToTableID.end()) {
      bachTableIdx = item->second;
    } else {
      LOG(warn) << "Could not find a bachelor track index";
      continue;
    }
    auto itemV = mVtxToTableCollID.find(cascade.getVertexID());
    if (itemV != mVtxToTableCollID.end()) {
      collID = itemV->second;
    } else {
      LOG(warn) << "Could not find cascade collisionID for the vertex ID " << cascade.getVertexID();
      continue;
    }
    cascadeCursor(collID, v0tableID, bachTableIdx);
  }

  // filling 3 body decays table
  decay3BodyCursor.reserve(decays3Body.size());
  for (size_t i3Body = 0; i3Body < decays3Body.size(); i3Body++) {
    const auto& decay3Body = decays3Body[i3Body];
    GIndex trIDs[3]{
      decay3Body.getProngID(0),
      decay3Body.getProngID(1),
      decay3Body.getProngID(2)};
    int tableIdx[3]{-1, -1, -1}, collID = -1;
    bool missing{false};
    for (int i{0}; i < 3; ++i) {
      auto item = mGIDToTableID.find(trIDs[i]);
      if (item != mGIDToTableID.end()) {
        tableIdx[i] = item->second;
      } else {
        LOG(warn) << fmt::format("Could not find a track index for prong ID {}", (int)trIDs[i]);
        missing = true;
      }
    }
    auto itemV = mVtxToTableCollID.find(decay3Body.getVertexID());
    if (itemV == mVtxToTableCollID.end()) {
      LOG(warn) << "Could not find 3 body collisionID for the vertex ID " << decay3Body.getVertexID();
      missing = true;
    } else {
      collID = itemV->second;
    }
    if (missing) {
      continue;
    }
    decay3BodyCursor(collID, tableIdx[0], tableIdx[1], tableIdx[2]);
  }
}

template <typename FwdTrkClsCursorType>
void AODProducerWorkflowDPL::addClustersToFwdTrkClsTable(const o2::globaltracking::RecoContainer& recoData, FwdTrkClsCursorType& fwdTrkClsCursor, GIndex trackID, int fwdTrackId)
{
  const auto& mchTracks = recoData.getMCHTracks();
  const auto& mchmidMatches = recoData.getMCHMIDMatches();
  const auto& mchClusters = recoData.getMCHTrackClusters();

  int mchTrackID = -1;
  if (trackID.getSource() == GIndex::MCH) { // This is an MCH track
    mchTrackID = trackID.getIndex();
  } else if (trackID.getSource() == GIndex::MCHMID) { // This is an MCH-MID track
    auto mchmidMatch = mchmidMatches[trackID.getIndex()];
    mchTrackID = mchmidMatch.getMCHRef().getIndex();
  } // Others are Global Forward Tracks, their clusters will be or were added with the corresponding MCH track

  if (mchTrackID > -1 && mchTrackID < mchTracks.size()) {
    const auto& mchTrack = mchTracks[mchTrackID];
    fwdTrkClsCursor.reserve(mchTrack.getNClusters() + fwdTrkClsCursor.lastIndex());
    int first = mchTrack.getFirstClusterIdx();
    int last = mchTrack.getLastClusterIdx();
    for (int i = first; i <= last; i++) {
      const auto& cluster = mchClusters[i];
      fwdTrkClsCursor(fwdTrackId,
                      truncateFloatFraction(cluster.x, mMuonCl),
                      truncateFloatFraction(cluster.y, mMuonCl),
                      truncateFloatFraction(cluster.z, mMuonCl),
                      (((cluster.ey < 5.) & 0x1) << 12) | (((cluster.ex < 5.) & 0x1) << 11) | cluster.getDEId());
    }
  }
}

template <typename HMPCursorType>
void AODProducerWorkflowDPL::fillHMPID(const o2::globaltracking::RecoContainer& recoData, HMPCursorType& hmpCursor)
{
  auto hmpMatches = recoData.getHMPMatches();

  hmpCursor.reserve(hmpMatches.size());

  // filling HMPs table
  for (size_t iHmp = 0; iHmp < hmpMatches.size(); iHmp++) {

    const auto& match = hmpMatches[iHmp];

    float xTrk, yTrk, theta, phi;
    float xMip, yMip;
    int charge, nph;

    match.getHMPIDtrk(xTrk, yTrk, theta, phi);
    match.getHMPIDmip(xMip, yMip, charge, nph);

    auto photChargeVec = match.getPhotCharge();

    float photChargeVec2[10]; // = {0.,0.,0.,0.,0.,0.,0.,0.,0.,0.};

    for (Int_t i = 0; i < 10; i++) {
      photChargeVec2[i] = photChargeVec[i];
    }
    auto tref = mGIDToTableID.find(match.getTrackRef());
    if (tref != mGIDToTableID.end()) {
      hmpCursor(tref->second, match.getHMPsignal(), xTrk, yTrk, xMip, yMip, nph, charge, match.getIdxHMPClus(), match.getHmpMom(), photChargeVec2);
    } else {
      LOG(error) << "Could not find AOD track table entry for HMP-matched track " << match.getTrackRef().asString();
    }
  }
}

void AODProducerWorkflowDPL::prepareStrangenessTracking(const o2::globaltracking::RecoContainer& recoData)
{
  auto v0s = recoData.getV0sIdx();
  auto cascades = recoData.getCascadesIdx();
  auto decays3Body = recoData.getDecays3BodyIdx();

  int sTrkID = 0;
  mCollisionStrTrk.clear();
  mCollisionStrTrk.reserve(recoData.getStrangeTracks().size());
  mVertexStrLUT.clear();
  mVertexStrLUT.resize(recoData.getPrimaryVertices().size() + 1, 0);
  for (auto& sTrk : recoData.getStrangeTracks()) {
    auto ITSIndex = GIndex{sTrk.mITSRef, GIndex::ITS};
    int vtxId{0};
    if (sTrk.mPartType == dataformats::kStrkV0) {
      vtxId = v0s[sTrk.mDecayRef].getVertexID();
    } else if (sTrk.mPartType == dataformats::kStrkCascade) {
      vtxId = cascades[sTrk.mDecayRef].getVertexID();
    } else {
      vtxId = decays3Body[sTrk.mDecayRef].getVertexID();
    }
    mCollisionStrTrk.emplace_back(vtxId, sTrkID++);
    mVertexStrLUT[vtxId]++;
  }
  std::exclusive_scan(mVertexStrLUT.begin(), mVertexStrLUT.end(), mVertexStrLUT.begin(), 0);

  // sort by collision ID
  std::sort(mCollisionStrTrk.begin(), mCollisionStrTrk.end(), [](const auto& a, const auto& b) { return a.first < b.first; });
  mStrTrkIndices.clear();
  mStrTrkIndices.resize(mCollisionStrTrk.size(), -1);
}

template <typename V0C, typename CC, typename D3BC>
void AODProducerWorkflowDPL::fillStrangenessTrackingTables(const o2::globaltracking::RecoContainer& recoData, V0C& v0Curs, CC& cascCurs, D3BC& d3BodyCurs)
{
  int itsTableIdx = -1;
  int sTrkID = 0;
  int nV0 = 0;
  int nCasc = 0;
  int nD3Body = 0;

  for (const auto& sTrk : recoData.getStrangeTracks()) {
    if (sTrk.mPartType == dataformats::kStrkV0) {
      nV0++;
    } else if (sTrk.mPartType == dataformats::kStrkCascade) {
      nCasc++;
    } else {
      nD3Body++;
    }
  }

  v0Curs.reserve(nV0);
  cascCurs.reserve(nCasc);
  d3BodyCurs.reserve(nD3Body);

  for (const auto& sTrk : recoData.getStrangeTracks()) {
    auto ITSIndex = GIndex{sTrk.mITSRef, GIndex::ITS};
    auto item = mGIDToTableID.find(ITSIndex);
    if (item != mGIDToTableID.end()) {
      itsTableIdx = item->second;
    } else {
      LOG(warn) << "Could not find a ITS strange track index " << ITSIndex;
      continue;
    }
    if (sTrk.mPartType == dataformats::kStrkV0) {
      v0Curs(mStrTrkIndices[sTrkID++],
             itsTableIdx,
             sTrk.mDecayRef,
             sTrk.mDecayVtx[0],
             sTrk.mDecayVtx[1],
             sTrk.mDecayVtx[2],
             sTrk.mMasses[0],
             sTrk.mMasses[1],
             sTrk.mMatchChi2,
             sTrk.mTopoChi2,
             sTrk.getAverageClusterSize());
    } else if (sTrk.mPartType == dataformats::kStrkCascade) {
      cascCurs(mStrTrkIndices[sTrkID++],
               itsTableIdx,
               sTrk.mDecayRef,
               sTrk.mDecayVtx[0],
               sTrk.mDecayVtx[1],
               sTrk.mDecayVtx[2],
               sTrk.mMasses[0],
               sTrk.mMasses[1],
               sTrk.mMatchChi2,
               sTrk.mTopoChi2,
               sTrk.getAverageClusterSize());
    } else {
      d3BodyCurs(mStrTrkIndices[sTrkID++],
                 itsTableIdx,
                 sTrk.mDecayRef,
                 sTrk.mDecayVtx[0],
                 sTrk.mDecayVtx[1],
                 sTrk.mDecayVtx[2],
                 sTrk.mMasses[0],
                 sTrk.mMasses[1],
                 sTrk.mMatchChi2,
                 sTrk.mTopoChi2,
                 sTrk.getAverageClusterSize());
    }
  }
}

void AODProducerWorkflowDPL::countTPCClusters(const o2::globaltracking::RecoContainer& data)
{
  const auto& tpcTracks = data.getTPCTracks();
  const auto& tpcClusRefs = data.getTPCTracksClusterRefs();
  const auto& tpcClusShMap = data.clusterShMapTPC;
  const auto& tpcClusAcc = data.getTPCClusters();
  constexpr int maxRows = 152;
  constexpr int neighbour = 2;
  int ntr = tpcTracks.size();
  mTPCCounters.clear();
  mTPCCounters.resize(ntr);
#ifdef WITH_OPENMP
  int ngroup = std::min(50, std::max(1, ntr / mNThreads));
#pragma omp parallel for schedule(dynamic, ngroup) num_threads(mNThreads)
#endif
  for (int itr = 0; itr < ntr; itr++) {
    std::array<bool, maxRows> clMap{}, shMap{};
    uint8_t sectorIndex, rowIndex;
    uint32_t clusterIndex;
    auto& counters = mTPCCounters[itr];
    const auto& track = tpcTracks[itr];
    for (int i = 0; i < track.getNClusterReferences(); i++) {
      o2::tpc::TrackTPC::getClusterReference(tpcClusRefs, i, sectorIndex, rowIndex, clusterIndex, track.getClusterRef());
      unsigned int absoluteIndex = tpcClusAcc.clusterOffset[sectorIndex][rowIndex] + clusterIndex;
      clMap[rowIndex] = true;
      if (tpcClusShMap[absoluteIndex] & o2::gpu::GPUTPCGMMergedTrackHit::flagShared) {
        if (!shMap[rowIndex]) {
          counters.shared++;
        }
        shMap[rowIndex] = true;
      }
    }
    int last = -1;
    for (int i = 0; i < maxRows; i++) {
      if (clMap[i]) {
        counters.crossed++;
        counters.found++;
        last = i;
      } else if ((i - last) <= neighbour) {
        counters.crossed++;
      } else {
        int lim = std::min(i + 1 + neighbour, maxRows);
        for (int j = i + 1; j < lim; j++) {
          if (clMap[j]) {
            counters.crossed++;
            break;
          }
        }
      }
    }
  }
}

uint8_t AODProducerWorkflowDPL::getTRDPattern(const o2::trd::TrackTRD& track)
{
  uint8_t pattern = 0;
  for (int il = o2::trd::TrackTRD::EGPUTRDTrack::kNLayers - 1; il >= 0; il--) {
    if (track.getTrackletIndex(il) != -1) {
      pattern |= 0x1 << il;
    }
  }
  if (track.getHasNeighbor()) {
    pattern |= 0x1 << 6;
  }
  if (track.getHasPadrowCrossing()) {
    pattern |= 0x1 << 7;
  }
  return pattern;
}

template <typename TCaloHandler, typename TCaloCursor, typename TCaloTRGCursor, typename TMCCaloLabelCursor>
void AODProducerWorkflowDPL::addToCaloTable(TCaloHandler& caloHandler, TCaloCursor& caloCellCursor, TCaloTRGCursor& caloTRGCursor,
                                            TMCCaloLabelCursor& mcCaloCellLabelCursor, int eventID, int bcID, int8_t caloType)
{
  auto inputEvent = caloHandler.buildEvent(eventID);
  auto cellsInEvent = inputEvent.mCells;        // get cells belonging to current event
  auto cellMClabels = inputEvent.mMCCellLabels; // get MC labels belonging to current event (only implemented for EMCal currently!)
  caloCellCursor.reserve(cellsInEvent.size() + caloCellCursor.lastIndex());
  caloTRGCursor.reserve(cellsInEvent.size() + caloTRGCursor.lastIndex());
  if (mUseMC) {
    mcCaloCellLabelCursor.reserve(cellsInEvent.size() + mcCaloCellLabelCursor.lastIndex());
  }
  for (auto iCell = 0U; iCell < cellsInEvent.size(); iCell++) {
    caloCellCursor(bcID,
                   CellHelper::getCellNumber(cellsInEvent[iCell]),
                   truncateFloatFraction(CellHelper::getAmplitude(cellsInEvent[iCell]), mCaloAmp),
                   truncateFloatFraction(CellHelper::getTimeStamp(cellsInEvent[iCell]), mCaloTime),
                   cellsInEvent[iCell].getType(),
                   caloType); // 1 = emcal, -1 = undefined, 0 = phos

    // todo: fix dummy values in CellHelper when it is clear what is filled for trigger information
    if (CellHelper::isTRU(cellsInEvent[iCell])) { // Write only trigger cells into this table
      caloTRGCursor(bcID,
                    CellHelper::getFastOrAbsID(cellsInEvent[iCell]),
                    CellHelper::getLnAmplitude(cellsInEvent[iCell]),
                    CellHelper::getTriggerBits(cellsInEvent[iCell]),
                    caloType);
    }
    if (mUseMC) {
      // Common for PHOS and EMCAL
      //  loop over all MC Labels for the current cell
      std::vector<int32_t> particleIds;
      std::vector<float> amplitudeFraction;
      if (!mEMCselectLeading) {
        particleIds.reserve(cellMClabels.size());
        amplitudeFraction.reserve(cellMClabels.size());
      }
      float tmpMaxAmplitude = 0;
      int32_t tmpindex = 0;
      for (auto& mclabel : cellMClabels[iCell]) {
        // do not fill noise lables!
        if (mclabel.isValid()) {
          if (mEMCselectLeading) {
            if (mclabel.getAmplitudeFraction() > tmpMaxAmplitude) {
              // Check if this MCparticle added to be kept?
              if (mToStore.at(mclabel.getSourceID()).at(mclabel.getEventID()).find(mclabel.getTrackID()) !=
                  mToStore.at(mclabel.getSourceID()).at(mclabel.getEventID()).end()) {
                tmpMaxAmplitude = mclabel.getAmplitudeFraction();
                tmpindex = (mToStore.at(mclabel.getSourceID()).at(mclabel.getEventID())).at(mclabel.getTrackID());
              }
            }
          } else {
            auto trackStore = mToStore.at(mclabel.getSourceID()).at(mclabel.getEventID());
            auto iter = trackStore.find(mclabel.getTrackID());
            if (iter != trackStore.end()) {
              amplitudeFraction.emplace_back(mclabel.getAmplitudeFraction());
              particleIds.emplace_back(iter->second);
            } else {
              particleIds.emplace_back(-1); // should the mc particle not be in mToStore make sure something (e.g. -1) is saved in particleIds so the length of particleIds is the same es amplitudeFraction!
              amplitudeFraction.emplace_back(0.f);
              LOG(warn) << "CaloTable: Could not find track for mclabel (" << mclabel.getSourceID() << "," << mclabel.getEventID() << "," << mclabel.getTrackID() << ") in the AOD MC store";
              if (mMCKineReader) {
                auto mctrack = mMCKineReader->getTrack(mclabel);
                TVector3 vec;
                mctrack->GetStartVertex(vec);
                LOG(warn) << " ... this track is of PDG " << mctrack->GetPdgCode() << " produced by " << mctrack->getProdProcessAsString() << " at (" << vec.X() << "," << vec.Y() << "," << vec.Z() << ")";
              }
            }
          }
        }
      } // end of loop over all MC Labels for the current cell
      if (mEMCselectLeading) {
        amplitudeFraction.emplace_back(tmpMaxAmplitude);
        particleIds.emplace_back(tmpindex);
      }
      if (particleIds.size() == 0) {
        particleIds.emplace_back(-1);
        amplitudeFraction.emplace_back(0.f);
      }
      mcCaloCellLabelCursor(particleIds,
                            amplitudeFraction);
    }
  } // end of loop over cells in current event
}

// fill calo related tables (cells and calotrigger table)
template <typename TCaloCursor, typename TCaloTRGCursor, typename TMCCaloLabelCursor>
void AODProducerWorkflowDPL::fillCaloTable(TCaloCursor& caloCellCursor, TCaloTRGCursor& caloTRGCursor,
                                           TMCCaloLabelCursor& mcCaloCellLabelCursor, const std::map<uint64_t, int>& bcsMap,
                                           const o2::globaltracking::RecoContainer& data)
{
  // get calo information
  auto caloEMCCells = data.getEMCALCells();
  auto caloEMCCellsTRGR = data.getEMCALTriggers();
  auto mcCaloEMCCellLabels = data.getEMCALCellsMCLabels();

  auto caloPHOSCells = data.getPHOSCells();
  auto caloPHOSCellsTRGR = data.getPHOSTriggers();
  auto mcCaloPHOSCellLabels = data.getPHOSCellsMCLabels();

  if (!mInputSources[GIndex::PHS]) {
    caloPHOSCells = {};
    caloPHOSCellsTRGR = {};
    mcCaloPHOSCellLabels = {};
  }

  if (!mInputSources[GIndex::EMC]) {
    caloEMCCells = {};
    caloEMCCellsTRGR = {};
    mcCaloEMCCellLabels = {};
  }

  o2::emcal::EventHandler<o2::emcal::Cell> emcEventHandler;
  o2::phos::EventHandler<o2::phos::Cell> phsEventHandler;

  // get cell belonging to an eveffillnt instead of timeframe
  emcEventHandler.reset();
  emcEventHandler.setCellData(caloEMCCells, caloEMCCellsTRGR);
  emcEventHandler.setCellMCTruthContainer(mcCaloEMCCellLabels);

  phsEventHandler.reset();
  phsEventHandler.setCellData(caloPHOSCells, caloPHOSCellsTRGR);
  phsEventHandler.setCellMCTruthContainer(mcCaloPHOSCellLabels);

  int emcNEvents = emcEventHandler.getNumberOfEvents();
  int phsNEvents = phsEventHandler.getNumberOfEvents();

  std::vector<std::tuple<uint64_t, int8_t, int>> caloEvents; // <bc, caloType, eventID>

  caloEvents.reserve(emcNEvents + phsNEvents);

  for (int iev = 0; iev < emcNEvents; ++iev) {
    uint64_t bc = emcEventHandler.getInteractionRecordForEvent(iev).toLong();
    caloEvents.emplace_back(std::make_tuple(bc, 1, iev));
  }

  for (int iev = 0; iev < phsNEvents; ++iev) {
    uint64_t bc = phsEventHandler.getInteractionRecordForEvent(iev).toLong();
    caloEvents.emplace_back(std::make_tuple(bc, 0, iev));
  }

  std::sort(caloEvents.begin(), caloEvents.end(),
            [](const auto& left, const auto& right) { return std::get<0>(left) < std::get<0>(right); });

  // loop over events
  for (int i = 0; i < emcNEvents + phsNEvents; ++i) {
    uint64_t globalBC = std::get<0>(caloEvents[i]);
    int8_t caloType = std::get<1>(caloEvents[i]);
    int eventID = std::get<2>(caloEvents[i]);
    auto item = bcsMap.find(globalBC);
    int bcID = -1;
    if (item != bcsMap.end()) {
      bcID = item->second;
    } else {
      LOG(warn) << "Error: could not find a corresponding BC ID for a calo point; globalBC = " << globalBC << ", caloType = " << (int)caloType;
    }
    if (caloType == 0) { // phos
      addToCaloTable(phsEventHandler, caloCellCursor, caloTRGCursor, mcCaloCellLabelCursor, eventID, bcID, caloType);
    }
    if (caloType == 1) { // emc
      addToCaloTable(emcEventHandler, caloCellCursor, caloTRGCursor, mcCaloCellLabelCursor, eventID, bcID, caloType);
    }
  }

  caloEvents.clear();
}

void AODProducerWorkflowDPL::init(InitContext& ic)
{
  mTimer.Stop();
  o2::base::GRPGeomHelper::instance().setRequest(mGGCCDBRequest);
  mLPMProdTag = ic.options().get<string>("lpmp-prod-tag");
  mAnchorPass = ic.options().get<string>("anchor-pass");
  mAnchorProd = ic.options().get<string>("anchor-prod");
  mUser = ic.options().get<string>("created-by");
  mRecoPass = ic.options().get<string>("reco-pass");
  mTFNumber = ic.options().get<int64_t>("aod-timeframe-id");
  mRecoOnly = ic.options().get<int>("reco-mctracks-only");
  mTruncate = ic.options().get<int>("enable-truncation");
  mRunNumber = ic.options().get<int>("run-number");
  mCTPReadout = ic.options().get<int>("ctpreadout-create");
  mNThreads = std::max(1, ic.options().get<int>("nthreads"));
  mEMCselectLeading = ic.options().get<bool>("emc-select-leading");
  mThinTracks = ic.options().get<bool>("thin-tracks");
  mPropTracks = ic.options().get<bool>("propagate-tracks");
  mPropMuons = ic.options().get<bool>("propagate-muons");
  if (auto s = ic.options().get<std::string>("with-streamers"); !s.empty()) {
    mStreamerFlags.set(s);
    if (mStreamerFlags) {
      LOGP(info, "Writing streamer data with mask:");
      LOG(info) << mStreamerFlags;
    } else {
      LOGP(warn, "Specified non-default empty streamer mask!");
    }
  }
  mTrackQCFraction = ic.options().get<float>("trackqc-fraction");
  mTrackQCNTrCut = ic.options().get<int64_t>("trackqc-NTrCut");
  mTrackQCDCAxy = ic.options().get<float>("trackqc-tpc-dca");
  mTrackQCPt = ic.options().get<float>("trackqc-tpc-pt");
  mTrackQCNCls = ic.options().get<int>("trackqc-tpc-cls");
  if (auto seed = ic.options().get<int>("seed"); seed == 0) {
    LOGP(info, "Using random device for seeding");
    std::random_device rd;
    std::array<int, std::mt19937::state_size> seed_data{};
    std::generate(std::begin(seed_data), std::end(seed_data), std::ref(rd));
    std::seed_seq seq(std::begin(seed_data), std::end(seed_data));
    mGenerator = std::mt19937(seq);
  } else {
    LOGP(info, "Using seed {} for sampling", seed);
    mGenerator.seed(seed);
  }
#ifdef WITH_OPENMP
  LOGP(info, "Multi-threaded parts will run with {} OpenMP threads", mNThreads);
#else
  mNThreads = 1;
  LOG(info) << "OpenMP is disabled";
#endif
  if (mTFNumber == -1L) {
    LOG(info) << "TFNumber will be obtained from CCDB";
  }
  if (mRunNumber == -1L) {
    LOG(info) << "The Run number will be obtained from DPL headers";
  }

  // set no truncation if selected by user
  if (mTruncate != 1) {
    LOG(info) << "Truncation is not used!";
    mCollisionPosition = 0xFFFFFFFF;
    mCollisionPositionCov = 0xFFFFFFFF;
    mTrackX = 0xFFFFFFFF;
    mTrackAlpha = 0xFFFFFFFF;
    mTrackSnp = 0xFFFFFFFF;
    mTrackTgl = 0xFFFFFFFF;
    mTrack1Pt = 0xFFFFFFFF;
    mTrackChi2 = 0xFFFFFFFF;
    mTrackCovDiag = 0xFFFFFFFF;
    mTrackCovOffDiag = 0xFFFFFFFF;
    mTrackSignal = 0xFFFFFFFF;
    mTrackTime = 0xFFFFFFFF;
    mTPCTime0 = 0xFFFFFFFF;
    mTrackTimeError = 0xFFFFFFFF;
    mTrackPosEMCAL = 0xFFFFFFFF;
    mTracklets = 0xFFFFFFFF;
    mMcParticleW = 0xFFFFFFFF;
    mMcParticlePos = 0xFFFFFFFF;
    mMcParticleMom = 0xFFFFFFFF;
    mCaloAmp = 0xFFFFFFFF;
    mCaloTime = 0xFFFFFFFF;
    mCPVPos = 0xFFFFFFFF;
    mCPVAmpl = 0xFFFFFFFF;
    mMuonTr1P = 0xFFFFFFFF;
    mMuonTrThetaX = 0xFFFFFFFF;
    mMuonTrThetaY = 0xFFFFFFFF;
    mMuonTrZmu = 0xFFFFFFFF;
    mMuonTrBend = 0xFFFFFFFF;
    mMuonTrNonBend = 0xFFFFFFFF;
    mMuonTrCov = 0xFFFFFFFF;
    mMuonCl = 0xFFFFFFFF;
    mMuonClErr = 0xFFFFFFFF;
    mV0Time = 0xFFFFFFFF;
    mV0ChannelTime = 0xFFFFFFFF;
    mFDDTime = 0xFFFFFFFF;
    mFDDChannelTime = 0xFFFFFFFF;
    mT0Time = 0xFFFFFFFF;
    mT0ChannelTime = 0xFFFFFFFF;
    mV0Amplitude = 0xFFFFFFFF;
    mFDDAmplitude = 0xFFFFFFFF;
    mT0Amplitude = 0xFFFFFFFF;
  }
  // Initialize ZDC helper maps
  for (int ic = 0; ic < o2::zdc::NChannels; ic++) {
    mZDCEnergyMap[ic] = -std::numeric_limits<float>::infinity();
  }
  for (int ic = 0; ic < o2::zdc::NTDCChannels; ic++) {
    mZDCTDCMap[ic] = -std::numeric_limits<float>::infinity();
  }

  std::string hepmcUpdate = ic.options().get<std::string>("hepmc-update");
  HepMCUpdate when = (hepmcUpdate == "never" ? HepMCUpdate::never : hepmcUpdate == "always" ? HepMCUpdate::always
                                                                  : hepmcUpdate == "all"    ? HepMCUpdate::allKeys
                                                                                            : HepMCUpdate::anyKey);
  mXSectionUpdate = when;
  mPdfInfoUpdate = when;
  mHeavyIonUpdate = when;

  mTimer.Reset();

  if (mStreamerFlags) {
    mStreamer = std::make_unique<o2::utils::TreeStreamRedirector>("AO2DStreamer.root", "RECREATE");
  }
}

namespace
{
void add_additional_meta_info(std::vector<TString>& keys, std::vector<TString>& values)
{
  // see if we should put additional meta info (e.g. from MC)
  auto aod_external_meta_info_file = getenv("AOD_ADDITIONAL_METADATA_FILE");
  if (aod_external_meta_info_file != nullptr) {
    LOG(info) << "Trying to inject additional AOD meta-data from " << aod_external_meta_info_file;
    if (std::filesystem::exists(aod_external_meta_info_file)) {
      std::ifstream input_file(aod_external_meta_info_file);
      if (input_file) {
        nlohmann::json json_data;
        try {
          input_file >> json_data;
        } catch (nlohmann::json::parse_error& e) {
          std::cerr << "JSON Parse Error: " << e.what() << "\n";
          std::cerr << "Exception ID: " << e.id << "\n";
          std::cerr << "Byte position: " << e.byte << "\n";
          return;
        }
        // If parsing succeeds, iterate over key-value pairs
        for (const auto& [key, value] : json_data.items()) {
          LOG(info) << "Adding AOD MetaData" << key << " : " << value;
          keys.push_back(key.c_str());
          values.push_back(value.get<std::string>());
        }
      }
    }
  }
}
} // namespace

void AODProducerWorkflowDPL::run(ProcessingContext& pc)
{
  mTimer.Start(false);
  o2::globaltracking::RecoContainer recoData;
  recoData.collectData(pc, *mDataRequest);
  updateTimeDependentParams(pc); // Make sure that this is called after the RecoContainer collect data, since some condition objects are fetched there

  mStartIR = recoData.startIR;

  auto primVertices = recoData.getPrimaryVertices();
  auto primVer2TRefs = recoData.getPrimaryVertexMatchedTrackRefs();
  auto primVerGIs = recoData.getPrimaryVertexMatchedTracks();
  auto primVerLabels = recoData.getPrimaryVertexMCLabels();

  auto fddChData = recoData.getFDDChannelsData();
  auto fddRecPoints = recoData.getFDDRecPoints();
  auto ft0ChData = recoData.getFT0ChannelsData();
  auto ft0RecPoints = recoData.getFT0RecPoints();
  auto fv0ChData = recoData.getFV0ChannelsData();
  auto fv0RecPoints = recoData.getFV0RecPoints();

  auto zdcEnergies = recoData.getZDCEnergy();
  auto zdcBCRecData = recoData.getZDCBCRecData();
  auto zdcTDCData = recoData.getZDCTDCData();

  auto cpvClusters = recoData.getCPVClusters();
  auto cpvTrigRecs = recoData.getCPVTriggers();

  auto ctpDigits = recoData.getCTPDigits();
  const auto& tinfo = pc.services().get<o2::framework::TimingInfo>();
  std::vector<o2::ctp::CTPDigit> ctpDigitsCreated;
  if (mCTPReadout == 1) {
    LOG(info) << "CTP : creating ctpreadout in AOD producer";
    createCTPReadout(recoData, ctpDigitsCreated, pc);
    LOG(info) << "CTP : ctpreadout created from AOD";
    ctpDigits = gsl::span<o2::ctp::CTPDigit>(ctpDigitsCreated);
  }
  LOG(debug) << "FOUND " << primVertices.size() << " primary vertices";
  LOG(debug) << "FOUND " << ft0RecPoints.size() << " FT0 rec. points";
  LOG(debug) << "FOUND " << fv0RecPoints.size() << " FV0 rec. points";
  LOG(debug) << "FOUND " << fddRecPoints.size() << " FDD rec. points";
  LOG(debug) << "FOUND " << cpvClusters.size() << " CPV clusters";
  LOG(debug) << "FOUND " << cpvTrigRecs.size() << " CPV trigger records";

  LOG(info) << "FOUND " << primVertices.size() << " primary vertices";

  using namespace o2::aodhelpers;

  auto bcCursor = createTableCursor<o2::aod::BCs>(pc);
  auto bcFlagsCursor = createTableCursor<o2::aod::BCFlags>(pc);
  auto cascadesCursor = createTableCursor<o2::aod::Cascades>(pc);
  auto collisionsCursor = createTableCursor<o2::aod::Collisions>(pc);
  auto decay3BodyCursor = createTableCursor<o2::aod::Decay3Bodys>(pc);
  auto trackedCascadeCursor = createTableCursor<o2::aod::TrackedCascades>(pc);
  auto trackedV0Cursor = createTableCursor<o2::aod::TrackedV0s>(pc);
  auto tracked3BodyCurs = createTableCursor<o2::aod::Tracked3Bodys>(pc);
  auto fddCursor = createTableCursor<o2::aod::FDDs>(pc);
  auto fddExtraCursor = createTableCursor<o2::aod::FDDsExtra>(pc);
  auto ft0Cursor = createTableCursor<o2::aod::FT0s>(pc);
  auto ft0ExtraCursor = createTableCursor<o2::aod::FT0sExtra>(pc);
  auto fv0aCursor = createTableCursor<o2::aod::FV0As>(pc);
  auto fv0aExtraCursor = createTableCursor<o2::aod::FV0AsExtra>(pc);
  auto fwdTracksCursor = createTableCursor<o2::aod::StoredFwdTracks>(pc);
  auto fwdTracksCovCursor = createTableCursor<o2::aod::StoredFwdTracksCov>(pc);
  auto fwdTrkClsCursor = createTableCursor<o2::aod::FwdTrkCls>(pc);
  auto mftTracksCursor = createTableCursor<o2::aod::StoredMFTTracks>(pc);
  auto mftTracksCovCursor = createTableCursor<o2::aod::StoredMFTTracksCov>(pc);
  auto tracksCursor = createTableCursor<o2::aod::StoredTracksIU>(pc);
  auto tracksCovCursor = createTableCursor<o2::aod::StoredTracksCovIU>(pc);
  auto tracksExtraCursor = createTableCursor<o2::aod::StoredTracksExtra>(pc);
  auto tracksQACursor = createTableCursor<o2::aod::TracksQAVersion>(pc);
  auto ambigTracksCursor = createTableCursor<o2::aod::AmbiguousTracks>(pc);
  auto ambigMFTTracksCursor = createTableCursor<o2::aod::AmbiguousMFTTracks>(pc);
  auto ambigFwdTracksCursor = createTableCursor<o2::aod::AmbiguousFwdTracks>(pc);
  auto v0sCursor = createTableCursor<o2::aod::V0s>(pc);
  auto zdcCursor = createTableCursor<o2::aod::Zdcs>(pc);
  auto hmpCursor = createTableCursor<o2::aod::HMPIDs>(pc);
  auto caloCellsCursor = createTableCursor<o2::aod::Calos>(pc);
  auto caloCellsTRGTableCursor = createTableCursor<o2::aod::CaloTriggers>(pc);
  auto cpvClustersCursor = createTableCursor<o2::aod::CPVClusters>(pc);
  auto originCursor = createTableCursor<o2::aod::Origins>(pc);

  // Declare MC cursors type without adding the output for a table
  o2::framework::Produces<o2::aod::McCollisionLabels> mcColLabelsCursor;
  o2::framework::Produces<o2::aod::McCollisions> mcCollisionsCursor;
  o2::framework::Produces<o2::aod::HepMCXSections> hepmcXSectionsCursor;
  o2::framework::Produces<o2::aod::HepMCPdfInfos> hepmcPdfInfosCursor;
  o2::framework::Produces<o2::aod::HepMCHeavyIons> hepmcHeavyIonsCursor;
  o2::framework::Produces<o2::aod::McMFTTrackLabels> mcMFTTrackLabelCursor;
  o2::framework::Produces<o2::aod::McFwdTrackLabels> mcFwdTrackLabelCursor;
  o2::framework::Produces<o2::aod::StoredMcParticles_001> mcParticlesCursor;
  o2::framework::Produces<o2::aod::McTrackLabels> mcTrackLabelCursor;
  o2::framework::Produces<o2::aod::McCaloLabels_001> mcCaloLabelsCursor;
  if (mUseMC) { // This creates the actual writercursor
    mcColLabelsCursor = createTableCursor<o2::aod::McCollisionLabels>(pc);
    mcCollisionsCursor = createTableCursor<o2::aod::McCollisions>(pc);
    hepmcXSectionsCursor = createTableCursor<o2::aod::HepMCXSections>(pc);
    hepmcPdfInfosCursor = createTableCursor<o2::aod::HepMCPdfInfos>(pc);
    hepmcHeavyIonsCursor = createTableCursor<o2::aod::HepMCHeavyIons>(pc);
    mcMFTTrackLabelCursor = createTableCursor<o2::aod::McMFTTrackLabels>(pc);
    mcFwdTrackLabelCursor = createTableCursor<o2::aod::McFwdTrackLabels>(pc);
    mcParticlesCursor = createTableCursor<o2::aod::StoredMcParticles_001>(pc);
    mcTrackLabelCursor = createTableCursor<o2::aod::McTrackLabels>(pc);
    mcCaloLabelsCursor = createTableCursor<o2::aod::McCaloLabels_001>(pc);
  }

  std::unique_ptr<o2::steer::MCKinematicsReader> mcReader;
  if (mUseMC) {
    mcReader = std::make_unique<o2::steer::MCKinematicsReader>("collisioncontext.root");
  }
  mMCKineReader = mcReader.get(); // for use in different functions
  std::map<uint64_t, int> bcsMap;
  collectBCs(recoData, mUseMC ? mcReader->getDigitizationContext()->getEventRecords() : std::vector<o2::InteractionTimeRecord>{}, bcsMap);
  if (!primVer2TRefs.empty()) { // if the vertexing was done, the last slot refers to orphan tracks
    addRefGlobalBCsForTOF(primVer2TRefs.back(), primVerGIs, recoData, bcsMap);
  }
  // initialize the bunch crossing container for further use below
  mBCLookup.init(bcsMap);

  uint64_t tfNumber;
  const int runNumber = (mRunNumber == -1) ? int(tinfo.runNumber) : mRunNumber;
  if (mTFNumber == -1L) {
    // TODO has to use absolute time of TF
    tfNumber = uint64_t(tinfo.firstTForbit) + (uint64_t(tinfo.runNumber) << 32); // getTFNumber(mStartIR, runNumber);
  } else {
    tfNumber = mTFNumber;
  }

  std::vector<float> aAmplitudes, aTimes;
  std::vector<uint8_t> aChannels;
  fv0aCursor.reserve(fv0RecPoints.size());
  for (auto& fv0RecPoint : fv0RecPoints) {
    aAmplitudes.clear();
    aChannels.clear();
    aTimes.clear();
    const auto channelData = fv0RecPoint.getBunchChannelData(fv0ChData);
    for (auto& channel : channelData) {
      if (channel.charge > 0) {
        aAmplitudes.push_back(truncateFloatFraction(channel.charge, mV0Amplitude));
        aTimes.push_back(truncateFloatFraction(channel.time * 1.E-3, mV0ChannelTime));
        aChannels.push_back(channel.channel);
      }
    }
    uint64_t bc = fv0RecPoint.getInteractionRecord().toLong();
    auto item = bcsMap.find(bc);
    int bcID = -1;
    if (item != bcsMap.end()) {
      bcID = item->second;
    } else {
      LOG(fatal) << "Error: could not find a corresponding BC ID for a FV0 rec. point; BC = " << bc;
    }
    fv0aCursor(bcID,
               aAmplitudes,
               aChannels,
               truncateFloatFraction(fv0RecPoint.getCollisionGlobalMeanTime() * 1E-3, mV0Time), // ps to ns
               fv0RecPoint.getTrigger().getTriggersignals());

    if (mEnableFITextra) {
      fv0aExtraCursor(bcID,
                      aTimes);
    }
  }

  std::vector<float> zdcEnergy, zdcAmplitudes, zdcTime;
  std::vector<uint8_t> zdcChannelsE, zdcChannelsT;
  zdcCursor.reserve(zdcBCRecData.size());
  for (auto zdcRecData : zdcBCRecData) {
    uint64_t bc = zdcRecData.ir.toLong();
    auto item = bcsMap.find(bc);
    int bcID = -1;
    if (item != bcsMap.end()) {
      bcID = item->second;
    } else {
      LOG(fatal) << "Error: could not find a corresponding BC ID for a ZDC rec. point; BC = " << bc;
    }
    int fe, ne, ft, nt, fi, ni;
    zdcRecData.getRef(fe, ne, ft, nt, fi, ni);
    zdcEnergy.clear();
    zdcChannelsE.clear();
    zdcAmplitudes.clear();
    zdcTime.clear();
    zdcChannelsT.clear();
    for (int ie = 0; ie < ne; ie++) {
      auto& zdcEnergyData = zdcEnergies[fe + ie];
      zdcEnergy.emplace_back(zdcEnergyData.energy());
      zdcChannelsE.emplace_back(zdcEnergyData.ch());
    }
    for (int it = 0; it < nt; it++) {
      auto& tdc = zdcTDCData[ft + it];
      zdcAmplitudes.emplace_back(tdc.amplitude());
      zdcTime.emplace_back(tdc.value());
      zdcChannelsT.emplace_back(o2::zdc::TDCSignal[tdc.ch()]);
    }
    zdcCursor(bcID,
              zdcEnergy,
              zdcChannelsE,
              zdcAmplitudes,
              zdcTime,
              zdcChannelsT);
  }

  // keep track event/source id for each mc-collision
  // using map and not unordered_map to ensure
  // correct ordering when iterating over container elements
  std::vector<std::vector<int>> mcColToEvSrc;

  if (mUseMC) {
    using namespace o2::aodmchelpers;

    // filling mcCollision table
    int nMCCollisions = mcReader->getDigitizationContext()->getNCollisions();
    const auto& mcRecords = mcReader->getDigitizationContext()->getEventRecords();
    const auto& mcParts = mcReader->getDigitizationContext()->getEventParts();

    // count all parts
    int totalNParts = 0;
    for (int iCol = 0; iCol < nMCCollisions; iCol++) {
      totalNParts += mcParts[iCol].size();
    }
    mcCollisionsCursor.reserve(totalNParts);

    for (int iCol = 0; iCol < nMCCollisions; iCol++) {
      const auto time = mcRecords[iCol].getTimeOffsetWrtBC();
      auto globalBC = mcRecords[iCol].toLong();
      auto item = bcsMap.find(globalBC);
      int bcID = -1;
      if (item != bcsMap.end()) {
        bcID = item->second;
      } else {
        LOG(fatal) << "Error: could not find a corresponding BC ID "
                   << "for MC collision; BC = " << globalBC
                   << ", mc collision = " << iCol;
      }
      auto& colParts = mcParts[iCol];
      auto nParts = colParts.size();
      for (auto colPart : colParts) {
        auto eventID = colPart.entryID;
        auto sourceID = colPart.sourceID;
        // enable embedding: if several colParts exist, then they are
        // saved as one collision
        if (nParts == 1 || sourceID == 0) {
          // FIXME:
          // use generators' names for generatorIDs (?)
          auto& header = mcReader->getMCEventHeader(sourceID, eventID);
          updateMCHeader(mcCollisionsCursor.cursor,
                         hepmcXSectionsCursor.cursor,
                         hepmcPdfInfosCursor.cursor,
                         hepmcHeavyIonsCursor.cursor,
                         header,
                         iCol,
                         bcID,
                         time,
                         0,
                         sourceID);
        }
        mcColToEvSrc.emplace_back(std::vector<int>{iCol, sourceID, eventID}); // point background and injected signal events to one collision
      }
    }
  }

  std::sort(mcColToEvSrc.begin(), mcColToEvSrc.end(),
            [](const std::vector<int>& left, const std::vector<int>& right) { return (left[0] < right[0]); });

  // vector of FDD amplitudes
  int16_t aFDDAmplitudesA[8] = {0u}, aFDDAmplitudesC[8] = {0u};
  float aFDDTimesA[8] = {0.f}, aFDDTimesC[8] = {0.f};
  // filling FDD table
  fddCursor.reserve(fddRecPoints.size());
  for (const auto& fddRecPoint : fddRecPoints) {
    for (int i = 0; i < 8; i++) {
      aFDDAmplitudesA[i] = 0;
      aFDDAmplitudesC[i] = 0;
      aFDDTimesA[i] = 0.f;
      aFDDTimesC[i] = 0.f;
    }
    uint64_t globalBC = fddRecPoint.getInteractionRecord().toLong();
    uint64_t bc = globalBC;
    auto item = bcsMap.find(bc);
    int bcID = -1;
    if (item != bcsMap.end()) {
      bcID = item->second;
    } else {
      LOG(fatal) << "Error: could not find a corresponding BC ID for a FDD rec. point; BC = " << bc;
    }
    const auto channelData = fddRecPoint.getBunchChannelData(fddChData);
    for (const auto& channel : channelData) {
      if (channel.mPMNumber < 8) {
        aFDDAmplitudesC[channel.mPMNumber] = channel.mChargeADC;                                      // amplitude
        aFDDTimesC[channel.mPMNumber] = truncateFloatFraction(channel.mTime * 1E-3, mFDDChannelTime); // time
      } else {
        aFDDAmplitudesA[channel.mPMNumber - 8] = channel.mChargeADC;                                      // amplitude
        aFDDTimesA[channel.mPMNumber - 8] = truncateFloatFraction(channel.mTime * 1E-3, mFDDChannelTime); // time
      }
    }

    fddCursor(bcID,
              aFDDAmplitudesA,
              aFDDAmplitudesC,
              truncateFloatFraction(fddRecPoint.getCollisionTimeA() * 1E-3, mFDDTime), // ps to ns
              truncateFloatFraction(fddRecPoint.getCollisionTimeC() * 1E-3, mFDDTime), // ps to ns
              fddRecPoint.getTrigger().getTriggersignals());
    if (mEnableFITextra) {
      fddExtraCursor(bcID,
                     aFDDTimesA,
                     aFDDTimesC);
    }
  }

  // filling FT0 table
  std::vector<float> aAmplitudesA, aAmplitudesC, aTimesA, aTimesC;
  std::vector<uint8_t> aChannelsA, aChannelsC;
  ft0Cursor.reserve(ft0RecPoints.size());
  for (auto& ft0RecPoint : ft0RecPoints) {
    aAmplitudesA.clear();
    aAmplitudesC.clear();
    aTimesA.clear();
    aTimesC.clear();
    aChannelsA.clear();
    aChannelsC.clear();
    const auto channelData = ft0RecPoint.getBunchChannelData(ft0ChData);
    for (auto& channel : channelData) {
      // TODO: switch to calibrated amplitude
      if (channel.QTCAmpl > 0) {
        constexpr int nFT0ChannelsAside = o2::ft0::Geometry::NCellsA * 4;
        if (channel.ChId < nFT0ChannelsAside) {
          aChannelsA.push_back(channel.ChId);
          aAmplitudesA.push_back(truncateFloatFraction(channel.QTCAmpl, mT0Amplitude));
          aTimesA.push_back(truncateFloatFraction(channel.CFDTime * 1E-3, mT0ChannelTime));
        } else {
          aChannelsC.push_back(channel.ChId - nFT0ChannelsAside);
          aAmplitudesC.push_back(truncateFloatFraction(channel.QTCAmpl, mT0Amplitude));
          aTimesC.push_back(truncateFloatFraction(channel.CFDTime * 1E-3, mT0ChannelTime));
        }
      }
    }
    uint64_t globalBC = ft0RecPoint.getInteractionRecord().toLong();
    uint64_t bc = globalBC;
    auto item = bcsMap.find(bc);
    int bcID = -1;
    if (item != bcsMap.end()) {
      bcID = item->second;
    } else {
      LOG(fatal) << "Error: could not find a corresponding BC ID for a FT0 rec. point; BC = " << bc;
    }
    ft0Cursor(bcID,
              aAmplitudesA,
              aChannelsA,
              aAmplitudesC,
              aChannelsC,
              truncateFloatFraction(ft0RecPoint.getCollisionTimeA() * 1E-3, mT0Time), // ps to ns
              truncateFloatFraction(ft0RecPoint.getCollisionTimeC() * 1E-3, mT0Time), // ps to ns
              ft0RecPoint.getTrigger().getTriggersignals());
    if (mEnableFITextra) {
      ft0ExtraCursor(bcID,
                     aTimesA,
                     aTimesC);
    }
  }

  if (mUseMC) {
    // filling MC collision labels
    mcColLabelsCursor.reserve(primVerLabels.size());
    for (auto& label : primVerLabels) {
      auto it = std::find_if(mcColToEvSrc.begin(), mcColToEvSrc.end(),
                             [&label](const auto& mcColInfo) { return mcColInfo[1] == label.getSourceID() && mcColInfo[2] == label.getEventID(); });
      int32_t mcCollisionID = -1;
      if (it != mcColToEvSrc.end()) {
        mcCollisionID = it->at(0);
      }
      uint16_t mcMask = 0; // todo: set mask using normalized weights?
      mcColLabelsCursor(mcCollisionID, mcMask);
    }
  }

  cacheTriggers(recoData);
  countTPCClusters(recoData);

  int collisionID = 0;
  mIndexTableMFT.resize(recoData.getMFTTracks().size());
  mIndexTableFwd.resize(recoData.getMCHTracks().size());

  auto& trackReffwd = primVer2TRefs.back();
  fillIndexTablesPerCollision(trackReffwd, primVerGIs, recoData);
  collisionID = 0;
  for (auto& vertex : primVertices) {
    auto& trackReffwd = primVer2TRefs[collisionID];
    fillIndexTablesPerCollision(trackReffwd, primVerGIs, recoData); // this function must follow the same track order as 'fillTrackTablesPerCollision' to fill the map of track indices
    collisionID++;
  }

  /// Strangeness tracking requires its index LUTs to be filled before the tracks are filled
  prepareStrangenessTracking(recoData);

  mGIDToTableFwdID.clear(); // reset the tables to be used by 'fillTrackTablesPerCollision'
  mGIDToTableMFTID.clear();

  if (mPropTracks || mThinTracks) {
    auto v0s = recoData.getV0sIdx();
    auto cascades = recoData.getCascadesIdx();
    auto decays3Body = recoData.getDecays3BodyIdx();
    mGIDUsedBySVtx.reserve(v0s.size() * 2 + cascades.size() + decays3Body.size() * 3);
    for (const auto& v0 : v0s) {
      mGIDUsedBySVtx.insert(v0.getProngID(0));
      mGIDUsedBySVtx.insert(v0.getProngID(1));
    }
    for (const auto& cascade : cascades) {
      mGIDUsedBySVtx.insert(cascade.getBachelorID());
    }
    for (const auto& id3Body : decays3Body) {
      mGIDUsedBySVtx.insert(id3Body.getProngID(0));
      mGIDUsedBySVtx.insert(id3Body.getProngID(1));
      mGIDUsedBySVtx.insert(id3Body.getProngID(2));
    }

    mGIDUsedByStr.reserve(recoData.getStrangeTracks().size());
    for (const auto& sTrk : recoData.getStrangeTracks()) {
      mGIDUsedByStr.emplace(sTrk.mITSRef, GIndex::ITS);
    }
  }

  // filling unassigned tracks first
  // so that all unassigned tracks are stored in the beginning of the table together
  auto& trackRef = primVer2TRefs.back(); // references to unassigned tracks are at the end
  // fixme: interaction time is undefined for unassigned tracks (?)
  fillTrackTablesPerCollision(-1, std::uint64_t(-1), trackRef, primVerGIs, recoData, tracksCursor, tracksCovCursor, tracksExtraCursor, tracksQACursor,
                              ambigTracksCursor, mftTracksCursor, mftTracksCovCursor, ambigMFTTracksCursor,
                              fwdTracksCursor, fwdTracksCovCursor, ambigFwdTracksCursor, fwdTrkClsCursor, bcsMap);

  // filling collisions and tracks into tables
  collisionID = 0;
  collisionsCursor.reserve(primVertices.size());
  for (auto& vertex : primVertices) {
    auto& cov = vertex.getCov();
    auto& timeStamp = vertex.getTimeStamp();                       // this is a relative time
    const double interactionTime = timeStamp.getTimeStamp() * 1E3; // mus to ns
    uint64_t globalBC = relativeTime_to_GlobalBC(interactionTime);
    uint64_t localBC = relativeTime_to_LocalBC(interactionTime);
    LOG(debug) << "global BC " << globalBC << " local BC " << localBC << " relative interaction time " << interactionTime;
    // collision timestamp in ns wrt the beginning of collision BC
    const float relInteractionTime = static_cast<float>(localBC * o2::constants::lhc::LHCBunchSpacingNS - interactionTime);
    auto item = bcsMap.find(globalBC);
    int bcID = -1;
    if (item != bcsMap.end()) {
      bcID = item->second;
    } else {
      LOG(fatal) << "Error: could not find a corresponding BC ID for a collision; BC = " << globalBC << ", collisionID = " << collisionID;
    }
    collisionsCursor(bcID,
                     truncateFloatFraction(vertex.getX(), mCollisionPosition),
                     truncateFloatFraction(vertex.getY(), mCollisionPosition),
                     truncateFloatFraction(vertex.getZ(), mCollisionPosition),
                     truncateFloatFraction(cov[0], mCollisionPositionCov),
                     truncateFloatFraction(cov[1], mCollisionPositionCov),
                     truncateFloatFraction(cov[2], mCollisionPositionCov),
                     truncateFloatFraction(cov[3], mCollisionPositionCov),
                     truncateFloatFraction(cov[4], mCollisionPositionCov),
                     truncateFloatFraction(cov[5], mCollisionPositionCov),
                     vertex.getFlags(),
                     truncateFloatFraction(vertex.getChi2(), mCollisionPositionCov),
                     vertex.getNContributors(),
                     truncateFloatFraction(relInteractionTime, mCollisionPosition),
                     truncateFloatFraction(timeStamp.getTimeStampError() * 1E3, mCollisionPositionCov));
    mVtxToTableCollID[collisionID] = mTableCollID++;

    auto& trackRef = primVer2TRefs[collisionID];
    // passing interaction time in [ps]
    fillTrackTablesPerCollision(collisionID, globalBC, trackRef, primVerGIs, recoData, tracksCursor, tracksCovCursor, tracksExtraCursor, tracksQACursor, ambigTracksCursor,
                                mftTracksCursor, mftTracksCovCursor, ambigMFTTracksCursor,
                                fwdTracksCursor, fwdTracksCovCursor, ambigFwdTracksCursor, fwdTrkClsCursor, bcsMap);
    collisionID++;
  }

  fillSecondaryVertices(recoData, v0sCursor, cascadesCursor, decay3BodyCursor);
  fillHMPID(recoData, hmpCursor);
  fillStrangenessTrackingTables(recoData, trackedV0Cursor, trackedCascadeCursor, tracked3BodyCurs);

  // helper map for fast search of a corresponding class mask for a bc
  auto emcalIncomplete = filterEMCALIncomplete(recoData.getEMCALTriggers());
  std::unordered_map<uint64_t, std::pair<uint64_t, uint64_t>> bcToClassMask;
  if (mInputSources[GID::CTP]) {
    LOG(debug) << "CTP input available";
    for (auto& ctpDigit : ctpDigits) {
      uint64_t bc = ctpDigit.intRecord.toLong();
      uint64_t classMask = ctpDigit.CTPClassMask.to_ulong();
      uint64_t inputMask = ctpDigit.CTPInputMask.to_ulong();
      if (emcalIncomplete.find(bc) != emcalIncomplete.end()) {
        // reject EMCAL triggers as BC was rejected as incomplete at readout level
        auto classMaskOrig = classMask;
        classMask = classMask & ~mEMCALTrgClassMask;
        LOG(debug) << "Found EMCAL incomplete event, mask before " << std::bitset<64>(classMaskOrig) << ", after " << std::bitset<64>(classMask);
      }
      bcToClassMask[bc] = {classMask, inputMask};
      // LOG(debug) << Form("classmask:0x%llx", classMask);
    }
  }

  // filling BC table
  bcCursor.reserve(bcsMap.size());
  for (auto& item : bcsMap) {
    uint64_t bc = item.first;
    std::pair<uint64_t, uint64_t> masks{0, 0};
    if (mInputSources[GID::CTP]) {
      auto bcClassPair = bcToClassMask.find(bc);
      if (bcClassPair != bcToClassMask.end()) {
        masks = bcClassPair->second;
      }
    }
    bcCursor(runNumber,
             bc,
             masks.first,
             masks.second);
  }

  bcToClassMask.clear();

  // filling BC flags table:
  auto bcFlags = fillBCFlags(recoData, bcsMap);
  bcFlagsCursor.reserve(bcFlags.size());
  for (auto f : bcFlags) {
    bcFlagsCursor(f);
  }

  // fill cpvcluster table
  if (mInputSources[GIndex::CPV]) {
    float posX, posZ;
    cpvClustersCursor.reserve(cpvTrigRecs.size());
    for (auto& cpvEvent : cpvTrigRecs) {
      uint64_t bc = cpvEvent.getBCData().toLong();
      auto item = bcsMap.find(bc);
      int bcID = -1;
      if (item != bcsMap.end()) {
        bcID = item->second;
      } else {
        LOG(fatal) << "Error: could not find a corresponding BC ID for a CPV Trigger Record; BC = " << bc;
      }
      for (int iClu = cpvEvent.getFirstEntry(); iClu < cpvEvent.getFirstEntry() + cpvEvent.getNumberOfObjects(); iClu++) {
        auto& clu = cpvClusters[iClu];
        clu.getLocalPosition(posX, posZ);
        cpvClustersCursor(bcID,
                          truncateFloatFraction(posX, mCPVPos),
                          truncateFloatFraction(posZ, mCPVPos),
                          truncateFloatFraction(clu.getEnergy(), mCPVAmpl),
                          clu.getPackedClusterStatus());
      }
    }
  }

  if (mUseMC) {
    TStopwatch timer;
    timer.Start();
    // filling mc particles table
    fillMCParticlesTable(*mcReader,
                         mcParticlesCursor.cursor,
                         primVer2TRefs,
                         primVerGIs,
                         recoData,
                         mcColToEvSrc);
    timer.Stop();
    LOG(info) << "FILL MC took " << timer.RealTime() << " s";
    mcColToEvSrc.clear();

    // ------------------------------------------------------
    // filling track labels

    // need to go through labels in the same order as for tracks
    fillMCTrackLabelsTable(mcTrackLabelCursor, mcMFTTrackLabelCursor, mcFwdTrackLabelCursor, primVer2TRefs.back(), primVerGIs, recoData);
    for (auto iref = 0U; iref < primVer2TRefs.size() - 1; iref++) {
      auto& trackRef = primVer2TRefs[iref];
      fillMCTrackLabelsTable(mcTrackLabelCursor, mcMFTTrackLabelCursor, mcFwdTrackLabelCursor, trackRef, primVerGIs, recoData, iref);
    }
  }

  // Fill calo tables and if MC also the MCCaloTable, therefore, has to be after fillMCParticlesTable call!
  if (mInputSources[GIndex::PHS] || mInputSources[GIndex::EMC]) {
    fillCaloTable(caloCellsCursor, caloCellsTRGTableCursor, mcCaloLabelsCursor, bcsMap, recoData);
  }

  bcsMap.clear();
  clearMCKeepStore(mToStore);
  mGIDToTableID.clear();
  mTableTrID = 0;
  mGIDToTableFwdID.clear();
  mTableTrFwdID = 0;
  mGIDToTableMFTID.clear();
  mTableTrMFTID = 0;
  mVtxToTableCollID.clear();
  mTableCollID = 0;
  mV0ToTableID.clear();
  mTableV0ID = 0;

  mIndexTableFwd.clear();
  mIndexFwdID = 0;
  mIndexTableMFT.clear();
  mIndexMFTID = 0;

  mBCLookup.clear();

  mGIDUsedBySVtx.clear();
  mGIDUsedByStr.clear();

  originCursor(tfNumber);

  // sending metadata to writer
  TString dataType = mUseMC ? "MC" : "RAW";
  TString O2Version = o2::fullVersion();
  TString ROOTVersion = ROOT_RELEASE;
  mMetaDataKeys = {"DataType", "Run", "O2Version", "ROOTVersion", "RecoPassName", "AnchorProduction", "AnchorPassName", "LPMProductionTag", "CreatedBy"};
  mMetaDataVals = {dataType, "3", O2Version, ROOTVersion, mRecoPass, mAnchorProd, mAnchorPass, mLPMProdTag, mUser};
  add_additional_meta_info(mMetaDataKeys, mMetaDataVals);

  pc.outputs().snapshot(Output{"AMD", "AODMetadataKeys", 0}, mMetaDataKeys);
  pc.outputs().snapshot(Output{"AMD", "AODMetadataVals", 0}, mMetaDataVals);

  pc.outputs().snapshot(Output{"TFN", "TFNumber", 0}, tfNumber);
  pc.outputs().snapshot(Output{"TFF", "TFFilename", 0}, "");

  mTimer.Stop();
}

void AODProducerWorkflowDPL::cacheTriggers(const o2::globaltracking::RecoContainer& recoData)
{
  // ITS tracks->ROF
  {
    mITSROFs.clear();
    const auto& rofs = recoData.getITSTracksROFRecords();
    uint16_t count = 0;
    for (const auto& rof : rofs) {
      int first = rof.getFirstEntry(), last = first + rof.getNEntries();
      for (int i = first; i < last; i++) {
        mITSROFs.push_back(count);
      }
      count++;
    }
  }
  // MFT tracks->ROF
  {
    mMFTROFs.clear();
    const auto& rofs = recoData.getMFTTracksROFRecords();
    uint16_t count = 0;
    for (const auto& rof : rofs) {
      int first = rof.getFirstEntry(), last = first + rof.getNEntries();
      for (int i = first; i < last; i++) {
        mMFTROFs.push_back(count);
      }
      count++;
    }
  }
  // ITSTPCTRD tracks -> TRD trigger
  {
    mITSTPCTRDTriggers.clear();
    const auto& itstpctrigs = recoData.getITSTPCTRDTriggers();
    int count = 0;
    for (const auto& trig : itstpctrigs) {
      int first = trig.getFirstTrack(), last = first + trig.getNumberOfTracks();
      for (int i = first; i < last; i++) {
        mITSTPCTRDTriggers.push_back(count);
      }
      count++;
    }
  }
  // TPCTRD tracks -> TRD trigger
  {
    mTPCTRDTriggers.clear();
    const auto& tpctrigs = recoData.getTPCTRDTriggers();
    int count = 0;
    for (const auto& trig : tpctrigs) {
      int first = trig.getFirstTrack(), last = first + trig.getNumberOfTracks();
      for (int i = first; i < last; i++) {
        mTPCTRDTriggers.push_back(count);
      }
      count++;
    }
  }
  // MCH tracks->ROF
  {
    mMCHROFs.clear();
    const auto& rofs = recoData.getMCHTracksROFRecords();
    uint16_t count = 0;
    for (const auto& rof : rofs) {
      int first = rof.getFirstIdx(), last = first + rof.getNEntries();
      for (int i = first; i < last; i++) {
        mMCHROFs.push_back(count);
      }
      count++;
    }
  }
}

AODProducerWorkflowDPL::TrackExtraInfo AODProducerWorkflowDPL::processBarrelTrack(int collisionID, std::uint64_t collisionBC, GIndex trackIndex,
                                                                                  const o2::globaltracking::RecoContainer& data, const std::map<uint64_t, int>& bcsMap)
{
  TrackExtraInfo extraInfoHolder;
  if (collisionID < 0) {
    extraInfoHolder.flags |= o2::aod::track::OrphanTrack;
  }
  bool needBCSlice = collisionID < 0;                     // track is associated to multiple vertices
  uint64_t bcOfTimeRef = collisionBC - mStartIR.toLong(); // by default track time is wrt collision BC (unless no collision assigned)

  auto setTrackTime = [&](double t, double terr, bool gaussian) {
    // set track time and error, for ambiguous tracks define the bcSlice as it was used in vertex-track association
    // provided track time (wrt TF start) and its error should be in ns, gaussian flag tells if the error is assumed to be gaussin or half-interval
    if (!gaussian) {
      extraInfoHolder.flags |= o2::aod::track::TrackTimeResIsRange;
    }
    extraInfoHolder.trackTimeRes = terr;
    if (needBCSlice) { // need to define BC slice
      double error = this->mTimeMarginTrackTime + (gaussian ? extraInfoHolder.trackTimeRes * this->mNSigmaTimeTrack : extraInfoHolder.trackTimeRes);
      bcOfTimeRef = fillBCSlice(extraInfoHolder.bcSlice, t - error, t + error, bcsMap);
    }
    extraInfoHolder.trackTime = float(t - bcOfTimeRef * o2::constants::lhc::LHCBunchSpacingNS);
    extraInfoHolder.diffBCRef = int(bcOfTimeRef);
    LOGP(debug, "time : {}/{} -> {}/{} -> trunc: {}/{} CollID: {} Amb: {}", t, terr, t - bcOfTimeRef * o2::constants::lhc::LHCBunchSpacingNS, terr,
         truncateFloatFraction(extraInfoHolder.trackTime, mTrackTime), truncateFloatFraction(extraInfoHolder.trackTimeRes, mTrackTimeError),
         collisionID, trackIndex.isAmbiguous());
  };
  auto contributorsGID = data.getSingleDetectorRefs(trackIndex);
  const auto& trackPar = data.getTrackParam(trackIndex);
  extraInfoHolder.flags |= trackPar.getPID() << 28;
  auto src = trackIndex.getSource();
  if (contributorsGID[GIndex::Source::TOF].isIndexSet()) { // ITS-TPC-TRD-TOF, ITS-TPC-TOF, TPC-TRD-TOF, TPC-TOF
    const auto& tofMatch = data.getTOFMatch(trackIndex);
    extraInfoHolder.tofChi2 = tofMatch.getChi2();
    const auto& tofInt = tofMatch.getLTIntegralOut();
    float intLen = tofInt.getL();
    extraInfoHolder.length = intLen;
    const float mass = o2::constants::physics::MassPionCharged; // default pid = pion
    if (tofInt.getTOF(o2::track::PID::Pion) > 0.f) {
      float expBeta = (intLen / (tofInt.getTOF(o2::track::PID::Pion) * cSpeed));
      if (expBeta > o2::constants::math::Almost1) {
        expBeta = o2::constants::math::Almost1;
      }
      extraInfoHolder.tofExpMom = mass * expBeta / std::sqrt(1.f - expBeta * expBeta);
    }
    // correct the time of the track
    const double massZ = o2::track::PID::getMass2Z(trackPar.getPID());
    const double energy = sqrt((massZ * massZ) + (extraInfoHolder.tofExpMom * extraInfoHolder.tofExpMom));
    const double exp = extraInfoHolder.length * energy / (cSpeed * extraInfoHolder.tofExpMom);
    auto tofSignal = (tofMatch.getSignal() - exp) * 1e-3; // time in ns wrt TF start
    setTrackTime(tofSignal, 0.2, true);                   // FIXME: calculate actual resolution (if possible?)
  }
  if (contributorsGID[GIndex::Source::TRD].isIndexSet()) {                                        // ITS-TPC-TRD-TOF, TPC-TRD-TOF, TPC-TRD, ITS-TPC-TRD
    const auto& trdOrig = data.getTrack<o2::trd::TrackTRD>(contributorsGID[GIndex::Source::TRD]); // refitted TRD trac
    extraInfoHolder.trdChi2 = trdOrig.getChi2();
    extraInfoHolder.trdSignal = trdOrig.getSignal();
    extraInfoHolder.trdPattern = getTRDPattern(trdOrig);
    if (extraInfoHolder.trackTimeRes < 0.) { // time is not set yet, this is possible only for TPC-TRD and ITS-TPC-TRD tracks, since those with TOF are set upstream
      // TRD is triggered: time uncertainty is within a BC
      const auto& trdTrig = (src == GIndex::Source::TPCTRD) ? data.getTPCTRDTriggers()[mTPCTRDTriggers[trackIndex.getIndex()]] : data.getITSTPCTRDTriggers()[mITSTPCTRDTriggers[trackIndex.getIndex()]];
      double ttrig = trdTrig.getBCData().differenceInBC(mStartIR) * o2::constants::lhc::LHCBunchSpacingNS; // 1st get time wrt TF start
      setTrackTime(ttrig, 1., true);                                                                       // FIXME: calculate actual resolution (if possible?)
    }
  }
  if (contributorsGID[GIndex::Source::ITS].isIndexSet()) {
    const auto& itsTrack = data.getITSTrack(contributorsGID[GIndex::ITS]);
    int nClusters = itsTrack.getNClusters();
    float chi2 = itsTrack.getChi2();
    extraInfoHolder.itsChi2NCl = nClusters != 0 ? chi2 / (float)nClusters : 0;
    extraInfoHolder.itsClusterSizes = itsTrack.getClusterSizes();
    if (src == GIndex::ITS) { // standalone ITS track should set its time from the ROF
      const auto& rof = data.getITSTracksROFRecords()[mITSROFs[trackIndex.getIndex()]];
      double t = rof.getBCData().differenceInBC(mStartIR) * o2::constants::lhc::LHCBunchSpacingNS + mITSROFrameHalfLengthNS + mITSROFBiasNS;
      setTrackTime(t, mITSROFrameHalfLengthNS, false);
    }
  } else if (contributorsGID[GIndex::Source::ITSAB].isIndexSet()) { // this is an ITS-TPC afterburner contributor
    extraInfoHolder.itsClusterSizes = data.getITSABRefs()[contributorsGID[GIndex::Source::ITSAB].getIndex()].getClusterSizes();
  }
  if (contributorsGID[GIndex::Source::TPC].isIndexSet()) {
    const auto& tpcOrig = data.getTPCTrack(contributorsGID[GIndex::TPC]);
    const auto& tpcClData = mTPCCounters[contributorsGID[GIndex::TPC]];
    const auto& dEdx = tpcOrig.getdEdx().dEdxTotTPC > 0 ? tpcOrig.getdEdx() : tpcOrig.getdEdxAlt();
    if (tpcOrig.getdEdx().dEdxTotTPC == 0) {
      extraInfoHolder.flags |= o2::aod::track::TPCdEdxAlt;
    }
    extraInfoHolder.tpcInnerParam = tpcOrig.getP() / tpcOrig.getAbsCharge();
    extraInfoHolder.tpcChi2NCl = tpcOrig.getNClusters() ? tpcOrig.getChi2() / tpcOrig.getNClusters() : 0;
    extraInfoHolder.tpcSignal = dEdx.dEdxTotTPC;
    extraInfoHolder.tpcNClsFindable = tpcOrig.getNClusters();
    extraInfoHolder.tpcNClsFindableMinusFound = tpcOrig.getNClusters() - tpcClData.found;
    extraInfoHolder.tpcNClsFindableMinusCrossedRows = tpcOrig.getNClusters() - tpcClData.crossed;
    extraInfoHolder.tpcNClsShared = tpcClData.shared;
    uint32_t clsUsedForPID = dEdx.NHitsIROC + dEdx.NHitsOROC1 + dEdx.NHitsOROC2 + dEdx.NHitsOROC3;
    extraInfoHolder.tpcNClsFindableMinusPID = tpcOrig.getNClusters() - clsUsedForPID;
    if (src == GIndex::TPC) { // standalone TPC track should set its time from their timebins range
      if (needBCSlice) {
        double t = (tpcOrig.getTime0() + 0.5 * (tpcOrig.getDeltaTFwd() - tpcOrig.getDeltaTBwd())) * mTPCBinNS; // central value
        double terr = 0.5 * (tpcOrig.getDeltaTFwd() + tpcOrig.getDeltaTBwd()) * mTPCBinNS;
        double err = mTimeMarginTrackTime + terr;
        bcOfTimeRef = fillBCSlice(extraInfoHolder.bcSlice, t - err, t + err, bcsMap);
      }
      aod::track::extensions::TPCTimeErrEncoding p;
      p.setDeltaTFwd(tpcOrig.getDeltaTFwd());
      p.setDeltaTBwd(tpcOrig.getDeltaTBwd());
      extraInfoHolder.trackTimeRes = p.getTimeErr();
      extraInfoHolder.trackTime = float(tpcOrig.getTime0() * mTPCBinNS - bcOfTimeRef * o2::constants::lhc::LHCBunchSpacingNS);
      extraInfoHolder.diffBCRef = int(bcOfTimeRef);
      extraInfoHolder.isTPConly = true; // no truncation
      extraInfoHolder.flags |= o2::aod::track::TrackTimeAsym;
    } else if (src == GIndex::ITSTPC) { // its-tpc matched tracks have gaussian time error and the time was not set above
      const auto& trITSTPC = data.getTPCITSTrack(trackIndex);
      auto ts = trITSTPC.getTimeMUS();
      setTrackTime(ts.getTimeStamp() * 1.e3, ts.getTimeStampError() * 1.e3, true);
    }
  }

  extrapolateToCalorimeters(extraInfoHolder, data.getTrackParamOut(trackIndex));
  // set bit encoding for PVContributor property as part of the flag field
  if (trackIndex.isPVContributor()) {
    extraInfoHolder.flags |= o2::aod::track::PVContributor;
  }
  return extraInfoHolder;
}

AODProducerWorkflowDPL::TrackQA AODProducerWorkflowDPL::processBarrelTrackQA(int collisionID, std::uint64_t collisionBC, GIndex trackIndex,
                                                                             const o2::globaltracking::RecoContainer& data, const std::map<uint64_t, int>& bcsMap)
{
  TrackQA trackQAHolder;
  auto contributorsGID = data.getTPCContributorGID(trackIndex);
  const auto& trackPar = data.getTrackParam(trackIndex);
  if (contributorsGID.isIndexSet()) {
    auto prop = o2::base::Propagator::Instance();
    const auto& tpcOrig = data.getTPCTrack(contributorsGID);
    /// getDCA - should be done  with the copy of TPC only track
    o2::track::TrackParametrization<float> tpcTMP = tpcOrig;                                             /// get backup of the track
    const o2::base::Propagator::MatCorrType mMatType = o2::base::Propagator::MatCorrType::USEMatCorrLUT; /// should be parameterized
    const o2::dataformats::VertexBase v = mVtx.getMeanVertex(collisionID < 0 ? 0.f : data.getPrimaryVertex(collisionID).getZ());
    std::array<float, 2> dcaInfo{-999., -999.};
    if (prop->propagateToDCABxByBz({v.getX(), v.getY(), v.getZ()}, tpcTMP, 2.f, mMatType, &dcaInfo)) {
      trackQAHolder.tpcdcaR = 100. * dcaInfo[0] / sqrt(1. + trackPar.getQ2Pt() * trackPar.getQ2Pt());
      trackQAHolder.tpcdcaZ = 100. * dcaInfo[1] / sqrt(1. + trackPar.getQ2Pt() * trackPar.getQ2Pt());
    }
    // This allows to safely clamp any float to one byte, using the
    // minmal/maximum values as under-/overflow borders and rounding to the nearest integer
    auto safeInt8Clamp = [](auto value) -> int8_t {
      using ValType = decltype(value);
      return static_cast<int8_t>(TMath::Nint(std::clamp(value, static_cast<ValType>(std::numeric_limits<int8_t>::min()), static_cast<ValType>(std::numeric_limits<int8_t>::max()))));
    };
    auto safeUInt8Clamp = [](auto value) -> uint8_t {
      using ValType = decltype(value);
      return static_cast<uint8_t>(TMath::Nint(std::clamp(value, static_cast<ValType>(std::numeric_limits<uint8_t>::min()), static_cast<ValType>(std::numeric_limits<uint8_t>::max()))));
    };

    /// get tracklet byteMask
    uint8_t clusterCounters[8] = {0};
    {
      uint8_t sectorIndex, rowIndex;
      uint32_t clusterIndex;
      const auto& tpcClusRefs = data.getTPCTracksClusterRefs();
      for (int i = 0; i < tpcOrig.getNClusterReferences(); i++) {
        o2::tpc::TrackTPC::getClusterReference(tpcClusRefs, i, sectorIndex, rowIndex, clusterIndex, tpcOrig.getClusterRef());
        char indexTracklet = (rowIndex % 152) / 19;
        clusterCounters[indexTracklet]++;
      }
    }
    uint8_t byteMask = 0;
    for (int i = 0; i < 8; i++) {
      if (clusterCounters[i] > 5) {
        byteMask |= (1 << i);
      }
    }
    trackQAHolder.tpcTime0 = tpcOrig.getTime0();
    trackQAHolder.tpcClusterByteMask = byteMask;
    const auto& dEdxInfoAlt = tpcOrig.getdEdxAlt(); // tpcOrig.getdEdx()
    const float dEdxNorm = (dEdxInfoAlt.dEdxTotTPC > 0) ? 100. / dEdxInfoAlt.dEdxTotTPC : 0;
    trackQAHolder.tpcdEdxNorm = dEdxInfoAlt.dEdxTotTPC;
    trackQAHolder.tpcdEdxMax0R = safeUInt8Clamp(dEdxInfoAlt.dEdxMaxIROC * dEdxNorm);
    trackQAHolder.tpcdEdxMax1R = safeUInt8Clamp(dEdxInfoAlt.dEdxMaxOROC1 * dEdxNorm);
    trackQAHolder.tpcdEdxMax2R = safeUInt8Clamp(dEdxInfoAlt.dEdxMaxOROC2 * dEdxNorm);
    trackQAHolder.tpcdEdxMax3R = safeUInt8Clamp(dEdxInfoAlt.dEdxMaxOROC3 * dEdxNorm);
    //
    trackQAHolder.tpcdEdxTot0R = safeUInt8Clamp(dEdxInfoAlt.dEdxTotIROC * dEdxNorm);
    trackQAHolder.tpcdEdxTot1R = safeUInt8Clamp(dEdxInfoAlt.dEdxTotOROC1 * dEdxNorm);
    trackQAHolder.tpcdEdxTot2R = safeUInt8Clamp(dEdxInfoAlt.dEdxTotOROC2 * dEdxNorm);
    trackQAHolder.tpcdEdxTot3R = safeUInt8Clamp(dEdxInfoAlt.dEdxTotOROC3 * dEdxNorm);
    ///
    float scaleTOF{0};
    auto contributorsGIDA = data.getSingleDetectorRefs(trackIndex);
    if (contributorsGIDA[GIndex::Source::TOF].isIndexSet()) { // ITS-TPC-TRD-TOF, ITS-TPC-TOF, TPC-TRD-TOF, TPC-TOF
      const auto& tofMatch = data.getTOFMatch(trackIndex);
      const float qpt = trackPar.getQ2Pt();
      scaleTOF = std::sqrt(o2::aod::track::trackQAScaledTOF[0] * o2::aod::track::trackQAScaledTOF[0] + qpt * qpt * o2::aod::track::trackQAScaledTOF[1] * o2::aod::track::trackQAScaledTOF[1]) / (2. * o2::aod::track::trackQAScaleBins);
      trackQAHolder.dTofdX = safeInt8Clamp(tofMatch.getDXatTOF() / scaleTOF);
      trackQAHolder.dTofdZ = safeInt8Clamp(tofMatch.getDZatTOF() / scaleTOF);
    }

    // Add matching information at a reference point (defined by
    // o2::aod::track::trackQARefRadius) in the same frame as the global track
    // without material corrections and error propagation
    if (auto itsContGID = data.getITSContributorGID(trackIndex); itsContGID.isIndexSet() && itsContGID.getSource() != GIndex::ITSAB) {
      const auto& itsOrig = data.getITSTrack(itsContGID);
      o2::track::TrackPar gloCopy = trackPar;
      o2::track::TrackPar itsCopy = itsOrig;
      o2::track::TrackPar tpcCopy = tpcOrig;
      if (prop->propagateToX(gloCopy, o2::aod::track::trackQARefRadius, prop->getNominalBz(), o2::base::Propagator::MAX_SIN_PHI, o2::base::Propagator::MAX_STEP, mMatCorr) &&
          prop->propagateToAlphaX(tpcCopy, gloCopy.getAlpha(), o2::aod::track::trackQARefRadius, false, o2::base::Propagator::MAX_SIN_PHI, o2::base::Propagator::MAX_STEP, 1, mMatCorr) &&
          prop->propagateToAlphaX(itsCopy, gloCopy.getAlpha(), o2::aod::track::trackQARefRadius, false, o2::base::Propagator::MAX_SIN_PHI, o2::base::Propagator::MAX_STEP, 1, mMatCorr)) {
        // All tracks are now at the same radius and in the same frame and we can calculate the deltas wrt. to the global track
        // The scale is defined by the global track scaling depending on beta0
        const float beta0 = std::sqrt(std::min(50.f / tpcOrig.getdEdx().dEdxMaxTPC, 1.f));
        const float qpt = gloCopy.getQ2Pt();
        const float x = qpt / beta0;
        // scaling is defined as sigmaBins/sqrt(p0^2 + (p1 * q/pt / beta)^2)
        auto scaleCont = [&x](int i) -> float {
          return o2::aod::track::trackQAScaleBins / std::sqrt(o2::aod::track::trackQAScaleContP0[i] * o2::aod::track::trackQAScaleContP0[i] + (o2::aod::track::trackQAScaleContP1[i] * x) * (o2::aod::track::trackQAScaleContP1[i] * x));
        };
        auto scaleGlo = [&x](int i) -> float {
          return o2::aod::track::trackQAScaleBins / std::sqrt(o2::aod::track::trackQAScaleGloP0[i] * o2::aod::track::trackQAScaleGloP0[i] + (o2::aod::track::trackQAScaleGloP1[i] * x) * (o2::aod::track::trackQAScaleGloP1[i] * x));
        };

        // Calculate deltas for contributors
        trackQAHolder.dRefContY = safeInt8Clamp((itsCopy.getY() - tpcCopy.getY()) * scaleCont(0));
        trackQAHolder.dRefContZ = safeInt8Clamp((itsCopy.getZ() - tpcCopy.getZ()) * scaleCont(1));
        trackQAHolder.dRefContSnp = safeInt8Clamp((itsCopy.getSnp() - tpcCopy.getSnp()) * scaleCont(2));
        trackQAHolder.dRefContTgl = safeInt8Clamp((itsCopy.getTgl() - tpcCopy.getTgl()) * scaleCont(3));
        trackQAHolder.dRefContQ2Pt = safeInt8Clamp((itsCopy.getQ2Pt() - tpcCopy.getQ2Pt()) * scaleCont(4));
        // Calculate deltas for global track against averaged contributors
        trackQAHolder.dRefGloY = safeInt8Clamp(((itsCopy.getY() + tpcCopy.getY()) * 0.5f - gloCopy.getY()) * scaleGlo(0));
        trackQAHolder.dRefGloZ = safeInt8Clamp(((itsCopy.getZ() + tpcCopy.getZ()) * 0.5f - gloCopy.getZ()) * scaleGlo(1));
        trackQAHolder.dRefGloSnp = safeInt8Clamp(((itsCopy.getSnp() + tpcCopy.getSnp()) * 0.5f - gloCopy.getSnp()) * scaleGlo(2));
        trackQAHolder.dRefGloTgl = safeInt8Clamp(((itsCopy.getTgl() + tpcCopy.getTgl()) * 0.5f - gloCopy.getTgl()) * scaleGlo(3));
        trackQAHolder.dRefGloQ2Pt = safeInt8Clamp(((itsCopy.getQ2Pt() + tpcCopy.getQ2Pt()) * 0.5f - gloCopy.getQ2Pt()) * scaleGlo(4));
        //

        if (mStreamerFlags[AODProducerStreamerFlags::TrackQA]) {
          (*mStreamer) << "trackQA"
                       << "trackITSOrig=" << itsOrig
                       << "trackTPCOrig=" << tpcOrig
                       << "trackITSTPCOrig=" << trackPar
                       << "trackITSProp=" << itsCopy
                       << "trackTPCProp=" << tpcCopy
                       << "trackITSTPCProp=" << gloCopy
                       << "refRadius=" << o2::aod::track::trackQARefRadius
                       << "scaleBins=" << o2::aod::track::trackQAScaleBins
                       << "scaleCont0=" << scaleCont(0)
                       << "scaleCont1=" << scaleCont(1)
                       << "scaleCont2=" << scaleCont(2)
                       << "scaleCont3=" << scaleCont(3)
                       << "scaleCont4=" << scaleCont(4)
                       << "scaleGlo0=" << scaleGlo(0)
                       << "scaleGlo1=" << scaleGlo(1)
                       << "scaleGlo2=" << scaleGlo(2)
                       << "scaleGlo3=" << scaleGlo(3)
                       << "scaleGlo4=" << scaleGlo(4)
                       << "trackQAHolder.tpcTime0=" << trackQAHolder.tpcTime0
                       << "trackQAHolder.tpcdEdxNorm=" << trackQAHolder.tpcdEdxNorm
                       << "trackQAHolder.tpcdcaR=" << trackQAHolder.tpcdcaR
                       << "trackQAHolder.tpcdcaZ=" << trackQAHolder.tpcdcaZ
                       << "trackQAHolder.tpcdcaClusterByteMask=" << trackQAHolder.tpcClusterByteMask
                       << "trackQAHolder.tpcdEdxMax0R=" << trackQAHolder.tpcdEdxMax0R
                       << "trackQAHolder.tpcdEdxMax1R=" << trackQAHolder.tpcdEdxMax1R
                       << "trackQAHolder.tpcdEdxMax2R=" << trackQAHolder.tpcdEdxMax2R
                       << "trackQAHolder.tpcdEdxMax3R=" << trackQAHolder.tpcdEdxMax3R
                       << "trackQAHolder.tpcdEdxTot0R=" << trackQAHolder.tpcdEdxTot0R
                       << "trackQAHolder.tpcdEdxTot1R=" << trackQAHolder.tpcdEdxTot1R
                       << "trackQAHolder.tpcdEdxTot2R=" << trackQAHolder.tpcdEdxTot2R
                       << "trackQAHolder.tpcdEdxTot3R=" << trackQAHolder.tpcdEdxTot3R
                       << "trackQAHolder.dRefContY=" << trackQAHolder.dRefContY
                       << "trackQAHolder.dRefContZ=" << trackQAHolder.dRefContZ
                       << "trackQAHolder.dRefContSnp=" << trackQAHolder.dRefContSnp
                       << "trackQAHolder.dRefContTgl=" << trackQAHolder.dRefContTgl
                       << "trackQAHolder.dRefContQ2Pt=" << trackQAHolder.dRefContQ2Pt
                       << "trackQAHolder.dRefGloY=" << trackQAHolder.dRefGloY
                       << "trackQAHolder.dRefGloZ=" << trackQAHolder.dRefGloZ
                       << "trackQAHolder.dRefGloSnp=" << trackQAHolder.dRefGloSnp
                       << "trackQAHolder.dRefGloTgl=" << trackQAHolder.dRefGloTgl
                       << "trackQAHolder.dRefGloQ2Pt=" << trackQAHolder.dRefGloQ2Pt
                       << "trackQAHolder.dTofdX=" << trackQAHolder.dTofdX
                       << "trackQAHolder.dTofdZ=" << trackQAHolder.dTofdZ
                       << "scaleTOF=" << scaleTOF
                       << "\n";
        }
      }
    }
  }

  return trackQAHolder;
}

bool AODProducerWorkflowDPL::propagateTrackToPV(o2::track::TrackParametrizationWithError<float>& trackPar,
                                                const o2::globaltracking::RecoContainer& data,
                                                int colID)
{
  o2::dataformats::DCA dcaInfo;
  dcaInfo.set(999.f, 999.f, 999.f, 999.f, 999.f);
  o2::dataformats::VertexBase v = mVtx.getMeanVertex(colID < 0 ? 0.f : data.getPrimaryVertex(colID).getZ());
  return o2::base::Propagator::Instance()->propagateToDCABxByBz(v, trackPar, 2.f, mMatCorr, &dcaInfo);
}

void AODProducerWorkflowDPL::extrapolateToCalorimeters(TrackExtraInfo& extraInfoHolder, const o2::track::TrackPar& track)
{
  constexpr float XEMCAL = 440.f, XPHOS = 460.f, XEMCAL2 = XEMCAL * XEMCAL;
  constexpr float ETAEMCAL = 0.75;                                  // eta of EMCAL/DCAL with margin
  constexpr float ZEMCALFastCheck = 460.;                           // Max Z (with margin to check with straightline extrapolarion)
  constexpr float ETADCALINNER = 0.22;                              // eta of the DCAL PHOS Hole (at XEMCAL)
  constexpr float ETAPHOS = 0.13653194;                             // nominal eta of the PHOS acceptance (at XPHOS): -log(tan((TMath::Pi()/2 - atan2(63, 460))/2))
  constexpr float ETAPHOSMARGIN = 0.17946979;                       // etat of the PHOS acceptance with 20 cm margin (at XPHOS): -log(tan((TMath::Pi()/2 + atan2(63+20., 460))/2)), not used, for the ref only
  constexpr float ETADCALPHOSSWITCH = (ETADCALINNER + ETAPHOS) / 2; // switch to DCAL to PHOS check if eta < this value
  constexpr short SNONE = 0, SEMCAL = 0x1, SPHOS = 0x2;
  constexpr short SECTORTYPE[18] = {
    SNONE, SNONE, SNONE, SNONE,                     // 0:3
    SEMCAL, SEMCAL, SEMCAL, SEMCAL, SEMCAL, SEMCAL, // 3:9 EMCAL only
    SNONE, SNONE,                                   // 10:11
    SPHOS,                                          // 12 PHOS only
    SPHOS | SEMCAL, SPHOS | SEMCAL, SPHOS | SEMCAL, // 13:15 PHOS & DCAL
    SEMCAL,                                         // 16 DCAL only
    SNONE                                           // 17
  };

  o2::track::TrackPar outTr{track};
  auto prop = o2::base::Propagator::Instance();
  // 1st propagate to EMCAL nominal radius
  float xtrg = 0;
  // quick check with straight line propagtion
  if (!outTr.getXatLabR(XEMCAL, xtrg, prop->getNominalBz(), o2::track::DirType::DirOutward) ||
      (std::abs(outTr.getZAt(xtrg, 0)) > ZEMCALFastCheck) ||
      !prop->PropagateToXBxByBz(outTr, xtrg, 0.95, 10, o2::base::Propagator::MatCorrType::USEMatCorrLUT)) {
    LOGP(debug, "preliminary step: does not reach R={} {}", XEMCAL, outTr.asString());
    return;
  }
  // we do not necessarilly reach wanted radius in a single propagation
  if ((outTr.getX() * outTr.getX() + outTr.getY() * outTr.getY() < XEMCAL2) &&
      (!outTr.rotateParam(outTr.getPhi()) ||
       !outTr.getXatLabR(XEMCAL, xtrg, prop->getNominalBz(), o2::track::DirType::DirOutward) ||
       !prop->PropagateToXBxByBz(outTr, xtrg, 0.95, 10, o2::base::Propagator::MatCorrType::USEMatCorrLUT))) {
    LOGP(debug, "does not reach R={} {}", XEMCAL, outTr.asString());
    return;
  }
  // rotate to proper sector
  int sector = o2::math_utils::angle2Sector(outTr.getPhiPos());

  auto propExactSector = [&outTr, &sector, prop](float xprop) -> bool { // propagate exactly to xprop in the proper sector frame
    int ntri = 0;
    while (ntri < 2) {
      auto outTrTmp = outTr;
      float alpha = o2::math_utils::sector2Angle(sector);
      if ((std::abs(outTr.getZ()) > ZEMCALFastCheck) || !outTrTmp.rotateParam(alpha) ||
          !prop->PropagateToXBxByBz(outTrTmp, xprop, 0.95, 10, o2::base::Propagator::MatCorrType::USEMatCorrLUT)) {
        LOGP(debug, "failed on rotation to {} (sector {}) or propagation to X={} {}", alpha, sector, xprop, outTrTmp.asString());
        return false;
      }
      // make sure we are still in the target sector
      int sectorTmp = o2::math_utils::angle2Sector(outTrTmp.getPhiPos());
      if (sectorTmp == sector) {
        outTr = outTrTmp;
        break;
      }
      sector = sectorTmp;
      ntri++;
    }
    if (ntri == 2) {
      LOGP(debug, "failed to rotate to sector, {}", outTr.asString());
      return false;
    }
    return true;
  };

  // we are at the EMCAL X, check if we are in the good sector
  if (!propExactSector(XEMCAL) || SECTORTYPE[sector] == SNONE) { // propagation failed or neither EMCAL not DCAL/PHOS
    return;
  }

  // check if we are in a good eta range
  float r = std::sqrt(outTr.getX() * outTr.getX() + outTr.getY() * outTr.getY()), tg = std::atan2(r, outTr.getZ());
  float eta = -std::log(std::tan(0.5f * tg)), etaAbs = std::abs(eta);
  if (etaAbs > ETAEMCAL) {
    LOGP(debug, "eta = {} is off at EMCAL radius", eta, outTr.asString());
    return;
  }
  // are we in the PHOS hole (with margin)?
  if ((SECTORTYPE[sector] & SPHOS) && etaAbs < ETADCALPHOSSWITCH) { // propagate to PHOS radius
    if (!propExactSector(XPHOS)) {
      return;
    }
    r = std::sqrt(outTr.getX() * outTr.getX() + outTr.getY() * outTr.getY());
    tg = std::atan2(r, outTr.getZ());
    eta = -std::log(std::tan(0.5f * tg));
  } else if (!(SECTORTYPE[sector] & SEMCAL)) { // are in the sector with PHOS only
    return;
  }
  extraInfoHolder.trackPhiEMCAL = outTr.getPhiPos();
  extraInfoHolder.trackEtaEMCAL = eta;
  LOGP(debug, "eta = {} phi = {} sector {} for {}", extraInfoHolder.trackEtaEMCAL, extraInfoHolder.trackPhiEMCAL, sector, outTr.asString());
  //
}

std::set<uint64_t> AODProducerWorkflowDPL::filterEMCALIncomplete(const gsl::span<const o2::emcal::TriggerRecord> triggers)
{
  std::set<uint64_t> emcalIncompletes;
  for (const auto& trg : triggers) {
    if (trg.getTriggerBits() & o2::emcal::triggerbits::Inc) {
      // trigger record masked at incomplete at readout level
      emcalIncompletes.insert(trg.getBCData().toLong());
    }
  }
  return emcalIncompletes;
}

void AODProducerWorkflowDPL::updateTimeDependentParams(ProcessingContext& pc)
{
  o2::base::GRPGeomHelper::instance().checkUpdates(pc);
  static bool initOnceDone = false;
  if (!initOnceDone) { // this params need to be queried only once
    initOnceDone = true;
    // Note: DPLAlpideParam for ITS and MFT will be loaded by the RecoContainer
    mSqrtS = o2::base::GRPGeomHelper::instance().getGRPLHCIF()->getSqrtS();
    // apply settings
    auto grpECS = o2::base::GRPGeomHelper::instance().getGRPECS();
    o2::BunchFilling bcf = o2::base::GRPGeomHelper::instance().getGRPLHCIF()->getBunchFilling();
    std::bitset<3564> bs = bcf.getBCPattern();
    for (auto i = 0U; i < bs.size(); i++) {
      if (bs.test(i)) {
        o2::tof::Utils::addInteractionBC(i);
      }
    }

    const auto& alpParamsITS = o2::itsmft::DPLAlpideParam<o2::detectors::DetID::ITS>::Instance();
    mITSROFrameHalfLengthNS = 0.5 * (grpECS->isDetContinuousReadOut(o2::detectors::DetID::ITS) ? alpParamsITS.roFrameLengthInBC * o2::constants::lhc::LHCBunchSpacingNS : alpParamsITS.roFrameLengthTrig);
    mITSROFBiasNS = o2::itsmft::DPLAlpideParam<o2::detectors::DetID::ITS>::Instance().roFrameBiasInBC * o2::constants::lhc::LHCBunchSpacingNS;
    const auto& alpParamsMFT = o2::itsmft::DPLAlpideParam<o2::detectors::DetID::MFT>::Instance();
    mMFTROFrameHalfLengthNS = 0.5 * (grpECS->isDetContinuousReadOut(o2::detectors::DetID::MFT) ? alpParamsMFT.roFrameLengthInBC * o2::constants::lhc::LHCBunchSpacingNS : alpParamsMFT.roFrameLengthTrig);
    mMFTROFBiasNS = o2::itsmft::DPLAlpideParam<o2::detectors::DetID::MFT>::Instance().roFrameBiasInBC * o2::constants::lhc::LHCBunchSpacingNS;

    // RS FIXME: this is not yet fetched from the CCDB
    auto& elParam = o2::tpc::ParameterElectronics::Instance();
    mTPCBinNS = elParam.ZbinWidth * 1.e3;

    const auto& pvParams = o2::vertexing::PVertexerParams::Instance();
    mNSigmaTimeTrack = pvParams.nSigmaTimeTrack;
    mTimeMarginTrackTime = pvParams.timeMarginTrackTime * 1.e3;
    mFieldON = std::abs(o2::base::Propagator::Instance()->getNominalBz()) > 0.01;

    pc.inputs().get<o2::ctp::CTPConfiguration*>("ctpconfig");
  }
  if (mPropTracks) {
    pc.inputs().get<o2::dataformats::MeanVertexObject*>("meanvtx");
  }
}

//_______________________________________
void AODProducerWorkflowDPL::finaliseCCDB(ConcreteDataMatcher& matcher, void* obj)
{
  // Note: strictly speaking, for Configurable params we don't need finaliseCCDB check, the singletons are updated at the CCDB fetcher level
  if (o2::base::GRPGeomHelper::instance().finaliseCCDB(matcher, obj)) {
    if (matcher == ConcreteDataMatcher("GLO", "GRPMAGFIELD", 0)) {
      o2::mch::TrackExtrap::setField();
    }
    return;
  }
  if (matcher == ConcreteDataMatcher("ITS", "ALPIDEPARAM", 0)) {
    LOG(info) << "ITS Alpide param updated";
    const auto& par = o2::itsmft::DPLAlpideParam<o2::detectors::DetID::ITS>::Instance();
    par.printKeyValues();
    return;
  }
  if (matcher == ConcreteDataMatcher("MFT", "ALPIDEPARAM", 0)) {
    LOG(info) << "MFT Alpide param updated";
    const auto& par = o2::itsmft::DPLAlpideParam<o2::detectors::DetID::MFT>::Instance();
    par.printKeyValues();
    return;
  }
  if (matcher == ConcreteDataMatcher("GLO", "MEANVERTEX", 0)) {
    LOG(info) << "Imposing new MeanVertex: " << ((const o2::dataformats::MeanVertexObject*)obj)->asString();
    mVtx = *(const o2::dataformats::MeanVertexObject*)obj;
    return;
  }
  if (matcher == ConcreteDataMatcher("CTP", "CTPCONFIG", 0)) {
    // construct mask with EMCAL trigger classes for rejection of incomplete triggers
    auto ctpconfig = *(const o2::ctp::CTPConfiguration*)obj;
    mEMCALTrgClassMask = 0;
    for (const auto& trgclass : ctpconfig.getCTPClasses()) {
      if (trgclass.cluster->maskCluster[o2::detectors::DetID::EMC]) {
        mEMCALTrgClassMask |= trgclass.classMask;
      }
    }
    LOG(info) << "Loaded EMCAL trigger class mask: " << std::bitset<64>(mEMCALTrgClassMask);
  }
}

void AODProducerWorkflowDPL::addRefGlobalBCsForTOF(const o2::dataformats::VtxTrackRef& trackRef, const gsl::span<const GIndex>& GIndices,
                                                   const o2::globaltracking::RecoContainer& data, std::map<uint64_t, int>& bcsMap)
{
  // Orphan tracks need to refer to some globalBC and for tracks with TOF this BC should be whithin an orbit
  // from the track abs time (to guarantee time precision). Therefore, we may need to insert some dummy globalBCs
  // to guarantee proper reference.
  // complete globalBCs by dummy entries necessary to provide BC references for TOF tracks with requested precision
  // to provide a reference for the time of the orphan tracks we should make sure that there are no gaps longer
  // than needed to store the time with sufficient precision

  // estimate max distance in BCs between TOF time and eventual reference BC
  int nbitsFrac = 24 - (32 - o2::math_utils::popcount(mTrackTime)); // number of bits used to encode the fractional part of float truncated by the mask
  int nbitsLoss = std::max(0, int(std::log2(TOFTimePrecPS)));       // allowed bit loss guaranteeing needed precision in PS
  assert(nbitsFrac > 1);
  std::uint64_t maxRangePS = std::uint64_t(0x1) << (nbitsFrac + nbitsLoss);
  int maxGapBC = maxRangePS / (o2::constants::lhc::LHCBunchSpacingNS * 1e3); // max gap in BCs allowing to store time with required precision
  LOG(info) << "Max gap of " << maxGapBC << " BCs to closest globalBC reference is needed for TOF tracks to provide precision of "
            << TOFTimePrecPS << " ps";

  // check if there are tracks orphan tracks at all
  if (!trackRef.getEntries()) {
    return;
  }
  // the bscMap has at least TF start BC
  std::uint64_t maxBC = mStartIR.toLong();
  const auto& tofClus = data.getTOFClusters();
  for (int src = GIndex::NSources; src--;) {
    if (!GIndex::getSourceDetectorsMask(src)[o2::detectors::DetID::TOF]) { // check only tracks with TOF contribution
      continue;
    }
    int start = trackRef.getFirstEntryOfSource(src);
    int end = start + trackRef.getEntriesOfSource(src);
    for (int ti = start; ti < end; ti++) {
      auto& trackIndex = GIndices[ti];
      const auto& tofMatch = data.getTOFMatch(trackIndex);
      const auto& tofInt = tofMatch.getLTIntegralOut();
      float intLen = tofInt.getL();
      float tofExpMom = 0.;
      if (tofInt.getTOF(o2::track::PID::Pion) > 0.f) {
        float expBeta = (intLen / (tofInt.getTOF(o2::track::PID::Pion) * cSpeed));
        if (expBeta > o2::constants::math::Almost1) {
          expBeta = o2::constants::math::Almost1;
        }
        tofExpMom = o2::constants::physics::MassPionCharged * expBeta / std::sqrt(1.f - expBeta * expBeta);
      } else {
        continue;
      }
      double massZ = o2::track::PID::getMass2Z(data.getTrackParam(trackIndex).getPID());
      double energy = sqrt((massZ * massZ) + (tofExpMom * tofExpMom));
      double exp = intLen * energy / (cSpeed * tofExpMom);
      auto tofSignal = (tofMatch.getSignal() - exp) * 1e-3; // time in ns wrt TF start
      auto bc = relativeTime_to_GlobalBC(tofSignal);

      auto it = bcsMap.lower_bound(bc);
      if (it == bcsMap.end() || it->first > bc + maxGapBC) {
        bcsMap.emplace_hint(it, bc, 1);
        LOG(debug) << "adding dummy BC " << bc;
      }
      if (bc > maxBC) {
        maxBC = bc;
      }
    }
  }
  // make sure there is a globalBC exceeding the max encountered bc
  if ((--bcsMap.end())->first <= maxBC) {
    bcsMap.emplace_hint(bcsMap.end(), maxBC + 1, 1);
  }
  // renumber BCs
  int bcID = 0;
  for (auto& item : bcsMap) {
    item.second = bcID;
    bcID++;
  }
}

std::uint64_t AODProducerWorkflowDPL::fillBCSlice(int (&slice)[2], double tmin, double tmax, const std::map<uint64_t, int>& bcsMap) const
{
  // for ambiguous tracks (no or multiple vertices) we store the BC slice corresponding to track time window used for track-vertex matching,
  // see VertexTrackMatcher::extractTracks creator method, i.e. central time estimated +- uncertainty defined as:
  // 1) for tracks having a gaussian time error: PVertexerParams.nSigmaTimeTrack * trackSigma + PVertexerParams.timeMarginTrackTime
  // 2) for tracks having time uncertainty in a fixed time interval (TPC,ITS,MFT..): half of the interval + PVertexerParams.timeMarginTrackTime
  // The track time in the TrackExtraInfo is stored in ns wrt the collision BC for unambigous tracks and wrt bcSlice[0] for ambiguous ones,
  // with convention for errors: trackSigma in case (1) and half of the time interval for case (2) above.

  // find indices of widest slice of global BCs in the map compatible with provided BC range. bcsMap is guaranteed to be non-empty.
  // We also assume that tmax >= tmin.

  uint64_t bcMin = relativeTime_to_GlobalBC(tmin), bcMax = relativeTime_to_GlobalBC(tmax);

  /*
    // brute force way of searching bcs via direct binary search in the map
    auto lower = bcsMap.lower_bound(bcMin), upper = bcsMap.upper_bound(bcMax);

    if (lower == bcsMap.end()) {
      --lower;
    }
    if (upper != lower) {
      --upper;
    }
    slice[0] = std::distance(bcsMap.begin(), lower);
    slice[1] = std::distance(bcsMap.begin(), upper);
  */

  // faster way to search in bunch crossing via the accelerated bunch crossing lookup structure
  auto p = mBCLookup.lower_bound(bcMin);
  // assuming that bcMax will be >= bcMin and close to bcMin; we can find
  // the upper bound quickly by lineary iterating from p.first to the point where
  // the time becomes larger than bcMax.
  // (if this is not the case we could determine it with a similar call to mBCLookup)
  auto& bcvector = mBCLookup.getBCTimeVector();
  auto upperindex = p.first;
  while (upperindex < bcvector.size() && bcvector[upperindex] <= bcMax) {
    upperindex++;
  }
  if (upperindex != p.first) {
    upperindex--;
  }
  slice[0] = p.first;
  slice[1] = upperindex;

  auto bcOfTimeRef = p.second - this->mStartIR.toLong();
  LOG(debug) << "BC slice t:" << tmin << " " << slice[0]
             << " t: " << tmax << " " << slice[1]
             << " bcref: " << bcOfTimeRef;
  return bcOfTimeRef;
}

std::vector<uint8_t> AODProducerWorkflowDPL::fillBCFlags(const o2::globaltracking::RecoContainer& data, std::map<uint64_t, int>& bcsMap) const
{
  std::vector<uint8_t> flags(bcsMap.size());

  // flag BCs belonging to UPC mode ITS ROFs
  auto bcIt = bcsMap.begin();
  auto itsrofs = data.getITSTracksROFRecords();
  auto lROF = o2::itsmft::DPLAlpideParam<o2::detectors::DetID::ITS>::Instance().roFrameLengthInBC;
  auto bROF = o2::itsmft::DPLAlpideParam<o2::detectors::DetID::ITS>::Instance().roFrameBiasInBC;
  for (auto& rof : itsrofs) {
    if (!rof.getFlag(o2::itsmft::ROFRecord::VtxUPCMode)) {
      continue;
    }
    uint64_t globalBC0 = rof.getBCData().toLong() + bROF, globalBC1 = globalBC0 + lROF - 1;
    // BCs are sorted, iterate until the start of ROF
    while (bcIt != bcsMap.end()) {
      if (bcIt->first < globalBC0) {
        ++bcIt;
        continue;
      }
      if (bcIt->first > globalBC1) {
        break;
      }
      flags[bcIt->second] |= o2::aod::bc::ITSUPCMode;
      ++bcIt;
    }
  }
  return flags;
}

void AODProducerWorkflowDPL::endOfStream(EndOfStreamContext& /*ec*/)
{
  LOGF(info, "aod producer dpl total timing: Cpu: %.3e Real: %.3e s in %d slots",
       mTimer.CpuTime(), mTimer.RealTime(), mTimer.Counter() - 1);

  mStreamer.reset();
}

DataProcessorSpec getAODProducerWorkflowSpec(GID::mask_t src, bool enableSV, bool enableStrangenessTracking, bool useMC, bool CTPConfigPerRun, bool enableFITextra)
{
  auto dataRequest = std::make_shared<DataRequest>();
  dataRequest->inputs.emplace_back("ctpconfig", "CTP", "CTPCONFIG", 0, Lifetime::Condition, ccdbParamSpec("CTP/Config/Config", CTPConfigPerRun));

  dataRequest->requestTracks(src, useMC);
  dataRequest->requestPrimaryVertices(useMC);
  if (src[GID::CTP]) {
    dataRequest->requestCTPDigits(useMC);
  }
  if (enableSV) {
    dataRequest->requestSecondaryVertices(useMC);
  }
  if (enableStrangenessTracking) {
    dataRequest->requestStrangeTracks(useMC);
    LOGF(info, "requestStrangeTracks Finish");
  }
  if (src[GID::ITS]) {
    dataRequest->requestClusters(GIndex::getSourcesMask("ITS"), false);
  }
  if (src[GID::TPC]) {
    dataRequest->requestClusters(GIndex::getSourcesMask("TPC"), false); // no need to ask for TOF clusters as they are requested with TOF tracks
  }
  if (src[GID::TOF]) {
    dataRequest->requestTOFClusters(useMC);
  }
  if (src[GID::PHS]) {
    dataRequest->requestPHOSCells(useMC);
  }
  if (src[GID::TRD]) {
    dataRequest->requestTRDTracklets(false);
  }
  if (src[GID::EMC]) {
    dataRequest->requestEMCALCells(useMC);
  }
  if (src[GID::CPV]) {
    dataRequest->requestCPVClusters(useMC);
  }

  auto ggRequest = std::make_shared<o2::base::GRPGeomRequest>(true,                              // orbitResetTime
                                                              true,                              // GRPECS=true
                                                              true,                              // GRPLHCIF
                                                              true,                              // GRPMagField
                                                              true,                              // askMatLUT
                                                              o2::base::GRPGeomRequest::Aligned, // geometry
                                                              dataRequest->inputs,
                                                              true); // query only once all objects except mag.field

  dataRequest->inputs.emplace_back("meanvtx", "GLO", "MEANVERTEX", 0, Lifetime::Condition, ccdbParamSpec("GLO/Calib/MeanVertex", {}, 1));

  using namespace o2::aod;
  using namespace o2::aodproducer;

  std::vector<OutputSpec> outputs{
    OutputForTable<BCs>::spec(),
    OutputForTable<BCFlags>::spec(),
    OutputForTable<Cascades>::spec(),
    OutputForTable<Collisions>::spec(),
    OutputForTable<Decay3Bodys>::spec(),
    OutputForTable<FDDs>::spec(),
    OutputForTable<FDDsExtra>::spec(),
    OutputForTable<FT0s>::spec(),
    OutputForTable<FT0sExtra>::spec(),
    OutputForTable<FV0As>::spec(),
    OutputForTable<FV0AsExtra>::spec(),
    OutputForTable<StoredFwdTracks>::spec(),
    OutputForTable<StoredFwdTracksCov>::spec(),
    OutputForTable<StoredMFTTracks>::spec(),
    OutputForTable<StoredMFTTracksCov>::spec(),
    OutputForTable<StoredTracksIU>::spec(),
    OutputForTable<StoredTracksCovIU>::spec(),
    OutputForTable<StoredTracksExtra>::spec(),
    OutputForTable<TracksQA>::spec(),
    OutputForTable<TrackedCascades>::spec(),
    OutputForTable<TrackedV0s>::spec(),
    OutputForTable<Tracked3Bodys>::spec(),
    OutputForTable<AmbiguousTracks>::spec(),
    OutputForTable<AmbiguousMFTTracks>::spec(),
    OutputForTable<AmbiguousFwdTracks>::spec(),
    OutputForTable<FwdTrkCls>::spec(),
    OutputForTable<V0s>::spec(),
    OutputForTable<HMPIDs>::spec(),
    OutputForTable<Zdcs>::spec(),
    OutputForTable<Calos>::spec(),
    OutputForTable<CaloTriggers>::spec(),
    OutputForTable<CPVClusters>::spec(),
    OutputForTable<Origins>::spec(),
    OutputSpec{"TFN", "TFNumber"},
    OutputSpec{"TFF", "TFFilename"},
    OutputSpec{"AMD", "AODMetadataKeys"},
    OutputSpec{"AMD", "AODMetadataVals"}};

  if (useMC) {
    outputs.insert(outputs.end(),
                   {OutputForTable<McCollisions>::spec(),
                    OutputForTable<HepMCXSections>::spec(),
                    OutputForTable<HepMCPdfInfos>::spec(),
                    OutputForTable<HepMCHeavyIons>::spec(),
                    OutputForTable<McMFTTrackLabels>::spec(),
                    OutputForTable<McFwdTrackLabels>::spec(),
                    OutputForTable<StoredMcParticles_001>::spec(),
                    OutputForTable<McTrackLabels>::spec(),
                    OutputForTable<McCaloLabels_001>::spec(),
                    // todo: use addTableToOuput helper?
                    //  currently the description is MCCOLLISLABEL, so
                    //  the name in AO2D would be O2mccollislabel
                    // addTableToOutput<McCollisionLabels>(outputs);
                    {OutputLabel{"McCollisionLabels"}, "AOD", "MCCOLLISIONLABEL", 0, Lifetime::Timeframe}});
  }

  return DataProcessorSpec{
    "aod-producer-workflow",
    dataRequest->inputs,
    outputs,
    AlgorithmSpec{adaptFromTask<AODProducerWorkflowDPL>(src, dataRequest, ggRequest, enableSV, useMC, enableFITextra)},
    Options{
      ConfigParamSpec{"run-number", VariantType::Int64, -1L, {"The run-number. If left default we try to get it from DPL header."}},
      ConfigParamSpec{"aod-timeframe-id", VariantType::Int64, -1L, {"Set timeframe number"}},
      ConfigParamSpec{"fill-calo-cells", VariantType::Int, 1, {"Fill calo cells into cell table"}},
      ConfigParamSpec{"enable-truncation", VariantType::Int, 1, {"Truncation parameter: 1 -- on, != 1 -- off"}},
      ConfigParamSpec{"lpmp-prod-tag", VariantType::String, "", {"LPMProductionTag"}},
      ConfigParamSpec{"anchor-pass", VariantType::String, "", {"AnchorPassName"}},
      ConfigParamSpec{"anchor-prod", VariantType::String, "", {"AnchorProduction"}},
      ConfigParamSpec{"reco-pass", VariantType::String, "", {"RecoPassName"}},
      ConfigParamSpec{"created-by", VariantType::String, "", {"Who created this AO2D"}},
      ConfigParamSpec{"nthreads", VariantType::Int, std::max(1, int(std::thread::hardware_concurrency() / 2)), {"Number of threads"}},
      ConfigParamSpec{"reco-mctracks-only", VariantType::Int, 0, {"Store only reconstructed MC tracks and their mothers/daughters. 0 -- off, != 0 -- on"}},
      ConfigParamSpec{"ctpreadout-create", VariantType::Int, 0, {"Create CTP digits from detector readout and CTP inputs. !=1 -- off, 1 -- on"}},
      ConfigParamSpec{"emc-select-leading", VariantType::Bool, false, {"Flag to select if only the leading contributing particle for an EMCal cell should be stored"}},
      ConfigParamSpec{"propagate-tracks", VariantType::Bool, false, {"Propagate tracks (not used for secondary vertices) to IP"}},
      ConfigParamSpec{"hepmc-update", VariantType::String, "always", {"When to update HepMC Aux tables: always - force update, never - never update, all - if all keys are present, any - when any key is present (not valid yet)"}},
      ConfigParamSpec{"propagate-muons", VariantType::Bool, false, {"Propagate muons to IP"}},
      ConfigParamSpec{"thin-tracks", VariantType::Bool, false, {"Produce thinned track tables"}},
      ConfigParamSpec{"trackqc-fraction", VariantType::Float, float(0.1), {"Fraction of tracks to QC"}},
      ConfigParamSpec{"trackqc-NTrCut", VariantType::Int64, 4L, {"Minimal length of the track - in amount of tracklets"}},
      ConfigParamSpec{"trackqc-tpc-dca", VariantType::Float, 3.f, {"Keep TPC standalone track with this DCAxy to the PV"}},
      ConfigParamSpec{"trackqc-tpc-cls", VariantType::Int, 80, {"Keep TPC standalone track with this #clusters"}},
      ConfigParamSpec{"trackqc-tpc-pt", VariantType::Float, 0.2f, {"Keep TPC standalone track with this pt"}},
      ConfigParamSpec{"with-streamers", VariantType::String, "", {"Bit-mask to steer writing of intermediate streamer files"}},
      ConfigParamSpec{"seed", VariantType::Int, 0, {"Set seed for random generator used for sampling (0 (default) means using a random_device)"}},
    }};
}

} // namespace o2::aodproducer