GPU QA: Drop obsolete csvDump function

davidrohr · davidrohr · commit 827af3fd15c4 · 2023-01-20T11:01:42.000+01:00
diff --git a/GPU/GPUTracking/Definitions/GPUSettingsList.h b/GPU/GPUTracking/Definitions/GPUSettingsList.h
@@ -317,7 +317,6 @@ AddOption(filterCharge, int, 0, "", 0, "Filter for positive (+1) or negative (-1
 AddOption(filterPID, int, -1, "", 0, "Filter for Particle Type (0 Electron, 1 Muon, 2 Pion, 3 Kaon, 4 Proton)")
 AddOption(nativeFitResolutions, bool, false, "", 0, "Create resolution histograms in the native fit units (sin(phi), tan(lambda), Q/Pt)")
 AddOption(enableLocalOutput, bool, true, "", 0, "Enable normal output to local PDF files / console")
-AddOption(csvDump, bool, false, "", 0, "Dump all clusters and Pt information into csv file")
 AddOption(writeMCLabels, bool, false, "", 0, "Store mc labels to file for later matching")
 AddOption(writeRootFiles, bool, false, "", 0, "Create ROOT canvas files")
 AddOptionVec(matchMCLabels, std::string, "", 0, "Read labels from files and match them, only process tracks where labels differ")
diff --git a/GPU/GPUTracking/qa/GPUQA.cxx b/GPU/GPUTracking/qa/GPUQA.cxx
@@ -1708,118 +1708,6 @@ void GPUQA::RunQA(bool matchOnly, const std::vector<o2::tpc::TrackTPC>* tracksEx
     GPUInfo("QA Time: Cluster Counts:\t%6.0f us", timer.GetCurrentElapsedTime(true) * 1e6);
   }
 
-  // Create CSV DumpTrackHits
-  if (mConfig.csvDump) {
-    if (!mTracking->GetParam().par.earlyTpcTransform) {
-      GPUError("Unsupported settings for csv dump\n");
-      return;
-    }
-    int totalNCls = GetNMCLabels();
-    if (totalNCls == 0) {
-      for (unsigned int iSlice = 0; iSlice < GPUChainTracking::NSLICES; iSlice++) {
-        totalNCls += mTracking->mIOPtrs.nClusterData[iSlice];
-      }
-    }
-
-    std::vector<float> clusterInfo(totalNCls);
-    memset(clusterInfo.data(), 0, clusterInfo.size() * sizeof(clusterInfo[0]));
-    for (unsigned int i = 0; i < nReconstructedTracks; i++) {
-      const GPUTPCGMMergedTrack& track = mTracking->mIOPtrs.mergedTracks[i];
-      if (!track.OK()) {
-        continue;
-      }
-      for (unsigned int k = 0; k < track.NClusters(); k++) {
-        if (mTracking->mIOPtrs.mergedTrackHits[track.FirstClusterRef() + k].state & GPUTPCGMMergedTrackHit::flagReject) {
-          continue;
-        }
-        int hitId = mTracking->mIOPtrs.mergedTrackHits[track.FirstClusterRef() + k].num;
-        float pt = fabsf(1.f / track.GetParam().GetQPt());
-        if (pt > clusterInfo[hitId]) {
-          clusterInfo[hitId] = pt;
-        }
-      }
-    }
-    static int csvNum = 0;
-    char fname[256];
-    sprintf(fname, "dump.%d.csv", csvNum);
-    FILE* fp = fopen(fname, "w+");
-    fprintf(fp, "x;y;z;reconstructedPt;individualMomentum;individualTransverseMomentum;trackLabel1;trackLabel2;trackLabel3;removed\n\n");
-    int dumpClTot = 0, dumpClLeft = 0, dumpClRem = 0;
-    for (unsigned int iSlice = 0; iSlice < GPUChainTracking::NSLICES; iSlice++) {
-      for (unsigned int i = 0; i < mTracking->mIOPtrs.nClusterData[iSlice]; i++) {
-        const auto& cl = mTracking->mIOPtrs.clusterData[iSlice][i];
-        float x, y, z;
-        const int cid = cl.id;
-        mTracking->GetParam().Slice2Global(iSlice, cl.x, cl.y, cl.z, &x, &y, &z);
-
-        float totalWeight = 0.f;
-        if (mcPresent()) {
-          for (int j = 0; j < GetMCLabelNID(cid); j++) {
-            if (GetMCLabelID(cid, j) >= 0) {
-              totalWeight += GetMCLabelWeight(cid, j);
-            }
-          }
-        }
-
-        float maxPt = 0.;
-        float p = 0.;
-
-        if (totalWeight > 0) {
-          for (int j = 0; j < GetMCLabelNID(cid); j++) {
-            const mcLabelI_t label = GetMCLabel(cid, j);
-            if (!label.isFake() && GetMCLabelWeight(cid, j) > 0.3 * totalWeight) {
-              const mcInfo_t& info = GetMCTrack(label);
-              const additionalMCParameters& mc2 = GetMCTrackObj(mMCParam, label);
-              const float pt = fabsf(mc2.pt);
-              if (pt > maxPt) {
-                maxPt = pt;
-                p = std::sqrt(info.pX * info.pX + info.pY * info.pY + info.pZ * info.pZ);
-              }
-            }
-          }
-        }
-        int labels[3] = {};
-        if (mcPresent()) {
-          for (int j = 0; j < GetMCLabelNID(cid); j++) {
-            labels[j] = GetMCLabelID(cid, j);
-          }
-        }
-
-        dumpClTot++;
-        int attach = mTracking->mIOPtrs.mergedTrackHitAttachment[cid];
-        CHECK_CLUSTER_STATE();
-        if (protect || physics) {
-          continue;
-        }
-        if (attach && qpt < 50) {
-          continue;
-        }
-        dumpClLeft++;
-        if (attach) {
-          dumpClRem++;
-        }
-
-        fprintf(fp, "%f;%f;%f;%f;%f;%f;%d;%d;%d;%d\n", x, y, z, attach ? 1.f / qpt : 0.f, p, maxPt, labels[0], labels[1], labels[2], attach ? 1 : 0);
-      }
-    }
-    fclose(fp);
-    if (mcPresent()) {
-      sprintf(fname, "dump_event.%d.csv", csvNum++);
-      fp = fopen(fname, "w+");
-      fprintf(fp, "trackLabel;trackMomentum;trackMomentumTransverse;trackMomentumZ\n\n");
-      for (unsigned int iCol = 0; iCol < GetNMCCollissions(); iCol++) {
-        for (unsigned int i = 0; i < GetNMCTracks(iCol); i++) {
-          const mcInfo_t& info = GetMCTrack(i, iCol);
-          additionalMCParameters& mc2 = mMCParam[iCol][i];
-          if (mc2.nWeightCls > 0) {
-            fprintf(fp, "%u;%f;%f;%f\n", i, std::sqrt(info.pX * info.pX + info.pY * info.pY + info.pZ * info.pZ), mc2.pt, info.pZ);
-          }
-        }
-      }
-      fclose(fp);
-    }
-    GPUInfo("Wrote %s,%d clusters in total, %d left, %d to be removed", fname, dumpClTot, dumpClLeft, dumpClRem);
-  }
   mTrackingScratchBuffer.clear();
 }
 
