added code for RZ case (mainly var name handlings)

guj · Nov 2, 2023 · c19ebdd · c19ebdd
1 parent 56ac633
commit c19ebdd
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/Source/Diagnostics/FlushFormats/warpxWriter.H b/Source/Diagnostics/FlushFormats/warpxWriter.H
@@ -295,14 +295,15 @@ AMReX_warpxWriter::SavePosId_RZ(PIt& pti,
                            [](amrex::ParticleReal const *p){ delete[] p; }
                            );
       for (auto i=0; i<numParticleOnTile; i++) {
-    auto const r = aos[i].pos(0);  // {0: "r", 1: "z"}
-    x.get()[i] = r * std::cos(theta[i]);
-    y.get()[i] = r * std::sin(theta[i]);
+        auto const r = aos[i].pos(0);  // {0: "r", 1: "z"}
+        x.get()[i] = r * std::cos(theta[i]);
+        y.get()[i] = r * std::sin(theta[i]);
       }
       currSpecies["position"]["x"].storeChunk(x, {offset}, {numParticleOnTile64});
       currSpecies["position"]["y"].storeChunk(y, {offset}, {numParticleOnTile64});
     }
   }
+  SaveSpecieId(pti, currSpecies, offset);
 #else
   SavePosId(pti, currSpecies, offset);
 #endif