GPU Display: Add option to tie TPC clusters to nominal x of pad rows

davidrohr · davidrohr · commit cf7d568fcc93 · 2020-11-20T09:08:36.000+01:00
diff --git a/GPU/GPUTracking/Base/GPUSettingsList.h b/GPU/GPUTracking/Base/GPUSettingsList.h
@@ -142,6 +142,7 @@ EndConfig()
 // Settings concerning the event display
 BeginSubConfig(GPUSettingsDisplay, GL, configStandalone, "GL", 'g', "OpenGL display settings")
 AddOption(clustersOnly, bool, false, "", 0, "Visualize clusters only")
+AddOption(clustersOnNominalRow, bool, false, "", 0, "Show clusters at nominal x of pad row for early-transformed data")
 AddHelp("help", 'h')
 EndConfig()
 
diff --git a/GPU/GPUTracking/Standalone/display/GPUDisplay.cxx b/GPU/GPUTracking/Standalone/display/GPUDisplay.cxx
@@ -1268,6 +1268,7 @@ int GPUDisplay::DrawGLScene_internal(bool mixAnimation, float mAnimateTime)
         if (mMerger.Param().par.earlyTpcTransform) {
           const auto& cl = ioptrs().clusterData[iSlice][i];
           cid = cl.id;
+          row = cl.row;
         } else {
           cid = ioptrs().clustersNative->clusterOffset[iSlice][0] + i;
           while (row < GPUCA_ROW_COUNT && ioptrs().clustersNative->clusterOffset[iSlice][row + 1] <= (unsigned int)cid) {
@@ -1282,7 +1283,7 @@ int GPUDisplay::DrawGLScene_internal(bool mixAnimation, float mAnimateTime)
         float4* ptr = &mGlobalPos[cid];
         if (mMerger.Param().par.earlyTpcTransform) {
           const auto& cl = ioptrs().clusterData[iSlice][i];
-          mChain->GetParam().Slice2Global(iSlice, cl.x + mXadd, cl.y, cl.z, &ptr->x, &ptr->y, &ptr->z);
+          mChain->GetParam().Slice2Global(iSlice, (mConfig.clustersOnNominalRow ? mMerger.Param().tpcGeometry.Row2X(row) : cl.x) + mXadd, cl.y, cl.z, &ptr->x, &ptr->y, &ptr->z);
         } else {
           float x, y, z;
           const auto& cln = ioptrs().clustersNative->clusters[iSlice][0][i];
