[Output] Write initial VTK config; shift others by one

src/one-body-sample.cc

@@ -82,7 +82,6 @@
 #include "state.hh"
 #include "timestepping.hh"
 #include "vtk.hh"
-
 size_t const dims = DIM;
 
 void initPython() {
@@ -296,6 +295,13 @@ int main(int argc, char *argv[]) {
                 typename MyAssembler::CellBasis> const
     vtkWriter(myAssembler.cellBasis, myAssembler.vertexBasis, "obs");
 
+    if (parset.get<bool>("io.writeVTK")) {
+      ScalarVector stress;
+      myAssembler.assembleVonMisesStress(
+          body.getYoungModulus(), body.getPoissonRatio(), u_initial, stress);
+      vtkWriter.write(0, u_initial, v_initial, alpha_initial, stress);
+    }
+
     // Set up TNNMG solver
     using NonlinearFactory =
         SolverFactory<dims, MyBlockProblem<ConvexProblem<
@@ -433,7 +439,7 @@ int main(int argc, char *argv[]) {
         ScalarVector stress;
         myAssembler.assembleVonMisesStress(body.getYoungModulus(),
                                            body.getPoissonRatio(), u, stress);
-        vtkWriter.write(timeStep - 1, u, v, alpha, stress);
+        vtkWriter.write(timeStep, u, v, alpha, stress);
       }
     }
     iterationWriter.close();