GPU: Pass firstOrbitOfTF from DataHeader to TPC reco workflow

Author: davidrohr

Detectors/TPC/reconstruction/src/GPUCATracking.cxx

@@ -64,6 +64,7 @@ int GPUCATracking::runTracking(GPUO2InterfaceIOPtrs* data, GPUInterfaceOutputs*
   GPUTPCDigitsMCInput gpuDigitsMapMC;
   GPUTrackingInOutPointers ptrs;
 
+  ptrs.settingsTF = data->settingsTF;
   ptrs.tpcCompressedClusters = data->compressedClusters;
   ptrs.tpcZS = data->tpcZS;
   if (data->o2Digits) {

Detectors/TPC/workflow/src/CATrackerSpec.cxx

@@ -328,6 +328,7 @@ DataProcessorSpec getCATrackerSpec(CompletionPolicyData* policyData, ca::Config
       const unsigned int* tpcZSmetaSizes2[GPUTrackingInOutZS::NSLICES][GPUTrackingInOutZS::NENDPOINTS];
       std::array<unsigned int, NEndpoints * NSectors> tpcZSonTheFlySizes;
       gsl::span<const ZeroSuppressedContainer8kb> inputZS;
+      o2::gpu::GPUSettingsTF tfSettings;
 
       bool getWorkflowTPCInput_clusters = false, getWorkflowTPCInput_mc = false, getWorkflowTPCInput_digits = false;
 
@@ -577,6 +578,11 @@ DataProcessorSpec getCATrackerSpec(CompletionPolicyData* policyData, ca::Config
         outputRegions.clusterLabels.allocator = [&clustersMCBuffer](size_t size) -> void* { return &clustersMCBuffer; };
       }
 
+      const auto* dh = o2::header::get<o2::header::DataHeader*>(pc.inputs().getByPos(0).header);
+      tfSettings.tfStartOrbit = dh->firstTForbit;
+      tfSettings.hasTfStartOrbit = 1;
+      ptrs.settingsTF = &tfSettings;
+
       int retVal = tracker->runTracking(&ptrs, &outputRegions);
       if (processAttributes->suppressOutput) {
         return;

GPU/GPUTracking/DataTypes/GPUDataTypes.h

@@ -104,6 +104,7 @@ struct GPUTPCMCInfo;
 struct GPUTPCClusterData;
 struct GPUTRDTrackletLabels;
 struct GPUTPCDigitsMCInput;
+struct GPUSettingsTF;
 
 class GPUDataTypes
 {
@@ -248,6 +249,7 @@ struct GPUTrackingInOutPointers {
   unsigned int nTRDTrackletsMC = 0;
   const GPUTRDTrackGPU* trdTracks = nullptr;
   unsigned int nTRDTracks = 0;
+  const GPUSettingsTF* settingsTF = nullptr;
 };
 #else
 struct GPUTrackingInOutPointers {

GPU/GPUTracking/DataTypes/GPUSettings.h

@@ -60,6 +60,16 @@ struct GPUSettingsGRP {
   int needsClusterer = 0;       // Set to true if the data requires the clusterizer
 };
 
+// Parameters of the current time frame
+struct GPUSettingsTF {
+  int hasTfStartOrbit = 0;
+  int tfStartOrbit = 0;
+  int hasRunStartOrbit = 0;
+  int runStartOrbit = 0;
+  int hasSimStartOrbit = 0;
+  int simStartOrbit = 0;
+};
+
 // Settings defining the setup of the GPUReconstruction processing (basically selecting the device / class instance)
 struct GPUSettingsDeviceBackend {
   unsigned int deviceType = GPUDataTypes::DeviceType::CPU; // Device type, shall use GPUDataTypes::DEVICE_TYPE constants, e.g. CPU / CUDA

GPU/GPUTracking/Definitions/GPUSettingsList.h

@@ -218,7 +218,7 @@ EndConfig()
 
 #ifdef BeginConfig
 // Settings concerning the standlone timeframe from run 2 events assembly tool
-BeginSubConfig(GPUSettingsTF, TF, configStandalone, "TF", 't', "Timeframe settings")
+BeginSubConfig(GPUSettingsTFSim, TF, configStandalone, "TF", 't', "Timeframe settings")
 AddOption(nMerge, int, 0, "", 0, "Merge n events in a timeframe", min(0))
 AddOption(averageDistance, float, 50., "", 0, "Average distance in cm of events merged into timeframe", min(0.f))
 AddOption(randomizeDistance, bool, true, "", 0, "Randomize distance around average distance of merged events")
@@ -309,7 +309,7 @@ AddHelp("help", 'h')
 AddHelpAll("helpall", 'H')
 AddSubConfig(GPUSettingsRec, rec)
 AddSubConfig(GPUSettingsProcessing, proc)
-AddSubConfig(GPUSettingsTF, TF)
+AddSubConfig(GPUSettingsTFSim, TF)
 AddSubConfig(GPUSettingsQA, QA)
 AddSubConfig(GPUSettingsDisplay, GL)
 AddSubConfig(GPUSettingsDisplayLight, GLlight)

GPU/GPUTracking/Global/GPUChainTrackingClusterizer.cxx

@@ -91,7 +91,7 @@ std::pair<unsigned int, unsigned int> GPUChainTracking::TPCClusterizerDecodeZSCo
   unsigned int nDigits = 0;
   unsigned int nPages = 0;
   bool doGPU = mRec->GetRecoStepsGPU() & GPUDataTypes::RecoStep::TPCClusterFinding;
-  int firstHBF = o2::raw::RDHUtils::getHeartBeatOrbit(*(const RAWDataHeaderGPU*)mIOPtrs.tpcZS->slice[iSlice].zsPtr[0][0]);
+  int firstHBF = (mIOPtrs.settingsTF && mIOPtrs.settingsTF->hasTfStartOrbit) ? mIOPtrs.settingsTF->tfStartOrbit : o2::raw::RDHUtils::getHeartBeatOrbit(*(const RAWDataHeaderGPU*)mIOPtrs.tpcZS->slice[iSlice].zsPtr[0][0]);
 
   for (unsigned short j = 0; j < GPUTrackingInOutZS::NENDPOINTS; j++) {
 #ifndef GPUCA_NO_VC
@@ -506,7 +506,7 @@ int GPUChainTracking::RunTPCClusterizer(bool synchronizeOutput)
         }
 
         if (mIOPtrs.tpcZS) {
-          int firstHBF = o2::raw::RDHUtils::getHeartBeatOrbit(*(const RAWDataHeaderGPU*)mIOPtrs.tpcZS->slice[iSlice].zsPtr[0][0]);
+          int firstHBF = (mIOPtrs.settingsTF && mIOPtrs.settingsTF->hasTfStartOrbit) ? mIOPtrs.settingsTF->tfStartOrbit : o2::raw::RDHUtils::getHeartBeatOrbit(*(const RAWDataHeaderGPU*)mIOPtrs.tpcZS->slice[iSlice].zsPtr[0][0]);
           runKernel<GPUTPCCFDecodeZS, GPUTPCCFDecodeZS::decodeZS>(GetGridBlk(doGPU ? clusterer.mPmemory->counters.nPagesSubslice : GPUTrackingInOutZS::NENDPOINTS, lane), {iSlice}, {}, firstHBF);
           TransferMemoryResourceLinkToHost(RecoStep::TPCClusterFinding, clusterer.mMemoryId, lane);
         }

GPU/GPUTracking/Interface/GPUO2InterfaceConfiguration.h

@@ -125,6 +125,9 @@ struct GPUO2InterfaceIOPtrs {
 #endif
   // Output for entropy-reduced clusters of TPC compression
   const o2::tpc::CompressedClustersFlat* compressedClusters = nullptr;
+
+  // Ptr to parameters for current TF
+  const GPUSettingsTF* settingsTF = nullptr;
 };
 } // namespace gpu
 } // namespace o2