GPU: Fix standalone dump: don't dump MC labels when there are no clusters / etc.

Author: davidrohr

GPU/GPUTracking/Base/GPUReconstruction.h

@@ -292,7 +292,7 @@ class GPUReconstruction
 
   // Private helper functions for reading / writing / allocating IO buffer from/to file
   template <class T, class S>
-  void DumpData(FILE* fp, const T* const* entries, const S* num, InOutPointerType type);
+  unsigned int DumpData(FILE* fp, const T* const* entries, const S* num, InOutPointerType type);
   template <class T, class S>
   size_t ReadData(FILE* fp, const T** entries, S* num, std::unique_ptr<T[]>* mem, InOutPointerType type, T** nonConstPtrs = nullptr);
   template <class T>
@@ -508,15 +508,15 @@ inline void GPUReconstruction::SetupGPUProcessor(T* proc, bool allocate)
 }
 
 template <class T, class S>
-inline void GPUReconstruction::DumpData(FILE* fp, const T* const* entries, const S* num, InOutPointerType type)
+inline unsigned int GPUReconstruction::DumpData(FILE* fp, const T* const* entries, const S* num, InOutPointerType type)
 {
   int count = getNIOTypeMultiplicity(type);
   unsigned int numTotal = 0;
   for (int i = 0; i < count; i++) {
     numTotal += num[i];
   }
   if (numTotal == 0) {
-    return;
+    return 0;
   }
   fwrite(&type, sizeof(type), 1, fp);
   for (int i = 0; i < count; i++) {
@@ -525,6 +525,10 @@ inline void GPUReconstruction::DumpData(FILE* fp, const T* const* entries, const
       fwrite(entries[i], sizeof(*entries[i]), num[i], fp);
     }
   }
+  if (mProcessingSettings.debugLevel >= 2) {
+    GPUInfo("Dumped %lld %s", (long long int)numTotal, IOTYPENAMES[type]);
+  }
+  return numTotal;
 }
 
 template <class T, class S>

GPU/GPUTracking/Global/GPUChain.h

@@ -127,9 +127,9 @@ class GPUChain
     mRec->AllocateIOMemoryHelper<T>(n, ptr, u);
   }
   template <class T, class S>
-  inline void DumpData(FILE* fp, const T* const* entries, const S* num, InOutPointerType type)
+  inline unsigned int DumpData(FILE* fp, const T* const* entries, const S* num, InOutPointerType type)
   {
-    mRec->DumpData<T>(fp, entries, num, type);
+    return mRec->DumpData<T>(fp, entries, num, type);
   }
   template <class T, class S>
   inline size_t ReadData(FILE* fp, const T** entries, S* num, std::unique_ptr<T[]>* mem, InOutPointerType type, T** nonConstPtrs = nullptr)

GPU/GPUTracking/Global/GPUChainTrackingIO.cxx

@@ -77,17 +77,17 @@ void GPUChainTracking::DumpData(const char* filename)
   DumpData(fp, mIOPtrs.rawClusters, mIOPtrs.nRawClusters, InOutPointerType::RAW_CLUSTERS);
 #ifdef HAVE_O2HEADERS
   if (mIOPtrs.clustersNative) {
-    DumpData(fp, &mIOPtrs.clustersNative->clustersLinear, &mIOPtrs.clustersNative->nClustersTotal, InOutPointerType::CLUSTERS_NATIVE);
-    fwrite(&mIOPtrs.clustersNative->nClusters[0][0], sizeof(mIOPtrs.clustersNative->nClusters[0][0]), NSLICES * GPUCA_ROW_COUNT, fp);
-    if (mIOPtrs.clustersNative->clustersMCTruth) {
-      const auto& buffer = mIOPtrs.clustersNative->clustersMCTruth->getBuffer();
-      std::pair<const char*, size_t> tmp = {buffer.data(), buffer.size()};
-      DumpData(fp, &tmp.first, &tmp.second, InOutPointerType::CLUSTER_NATIVE_MC);
+    if (DumpData(fp, &mIOPtrs.clustersNative->clustersLinear, &mIOPtrs.clustersNative->nClustersTotal, InOutPointerType::CLUSTERS_NATIVE)) {
+      fwrite(&mIOPtrs.clustersNative->nClusters[0][0], sizeof(mIOPtrs.clustersNative->nClusters[0][0]), NSLICES * GPUCA_ROW_COUNT, fp);
+      if (mIOPtrs.clustersNative->clustersMCTruth) {
+        const auto& buffer = mIOPtrs.clustersNative->clustersMCTruth->getBuffer();
+        std::pair<const char*, size_t> tmp = {buffer.data(), buffer.size()};
+        DumpData(fp, &tmp.first, &tmp.second, InOutPointerType::CLUSTER_NATIVE_MC);
+      }
     }
   }
   if (mIOPtrs.tpcPackedDigits) {
-    DumpData(fp, mIOPtrs.tpcPackedDigits->tpcDigits, mIOPtrs.tpcPackedDigits->nTPCDigits, InOutPointerType::TPC_DIGIT);
-    if (mIOPtrs.tpcPackedDigits->tpcDigitsMC) {
+    if (DumpData(fp, mIOPtrs.tpcPackedDigits->tpcDigits, mIOPtrs.tpcPackedDigits->nTPCDigits, InOutPointerType::TPC_DIGIT) && mIOPtrs.tpcPackedDigits->tpcDigitsMC) {
       const char* ptrs[NSLICES];
       size_t sizes[NSLICES];
       for (unsigned int i = 0; i < NSLICES; i++) {
@@ -121,8 +121,9 @@ void GPUChainTracking::DumpData(const char* filename)
       }
     }
     total *= TPCZSHDR::TPC_ZS_PAGE_SIZE;
-    DumpData(fp, &ptr, &total, InOutPointerType::TPC_ZS);
-    fwrite(&counts, sizeof(counts), 1, fp);
+    if (DumpData(fp, &ptr, &total, InOutPointerType::TPC_ZS)) {
+      fwrite(&counts, sizeof(counts), 1, fp);
+    }
   }
 #endif
   DumpData(fp, mIOPtrs.sliceTracks, mIOPtrs.nSliceTracks, InOutPointerType::SLICE_OUT_TRACK);