fix surfcov restart with random numbers

AMReX-FHD · Feb 11, 2025 · 1212ec0 · 1212ec0
1 parent 06bf248
commit 1212ec0
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/src_compressible_stag/Checkpoint.cpp b/src_compressible_stag/Checkpoint.cpp
@@ -1087,6 +1087,10 @@ void ReadCheckPoint2D(int& step,
     Read_Copy_MF_Checkpoint(cumom[1],"cumomy",checkpointname,ba_old,dmap_old,1,1,1);
     Read_Copy_MF_Checkpoint(cumom[2],"cumomz",checkpointname,ba_old,dmap_old,1,1,2);
 
+    if (n_ads_spec>0) {
+        Read_Copy_MF_Checkpoint(surfcov,"surfcov",checkpointname,ba_old,dmap_old,n_ads_spec,0);
+    }
+
     // Set all stats to zero if reset stats, else read
     if (reset_stats == 1) {
         cuMeans.setVal(0.0);
@@ -1100,6 +1104,13 @@ void ReadCheckPoint2D(int& step,
             cumomVars[d].setVal(0.);
         }
         coVars.setVal(0.0);
+        if (n_ads_spec>0) {
+            for (int m=0;m<n_ads_spec;m++) {
+                surfcovMeans.setVal(0.0);
+                surfcovVars.setVal(0.0);
+		surfcovcoVars.setVal(0.0);
+            }
+        }
         spatialCross.setVal(0.0);
     }
     else {
@@ -1126,6 +1137,12 @@ void ReadCheckPoint2D(int& step,
         Read_Copy_MF_Checkpoint(cumomVars[0],"cumomvarx",checkpointname,ba_old,dmap_old,1,0,0);
         Read_Copy_MF_Checkpoint(cumomVars[1],"cumomvary",checkpointname,ba_old,dmap_old,1,0,1);
         Read_Copy_MF_Checkpoint(cumomVars[2],"cumomvarz",checkpointname,ba_old,dmap_old,1,0,2);
+
+        if (n_ads_spec>0) {
+            Read_Copy_MF_Checkpoint(surfcovMeans,"surfcovMeans",checkpointname,ba_old,dmap_old,n_ads_spec,0);
+            Read_Copy_MF_Checkpoint(surfcovVars,"surfcovVars",checkpointname,ba_old,dmap_old,n_ads_spec,0);
+	    Read_Copy_MF_Checkpoint(surfcovcoVars,"surfcovcoVars",checkpointname,ba_old,dmap_old,n_ads_spec*6,0);
+        }
     }
 
     // FillBoundaries

diff --git a/src_compressible_stag/main_driver.cpp b/src_compressible_stag/main_driver.cpp
@@ -1030,6 +1030,8 @@ void main_driver(const char* argv)
 	    } else {
                 Abort("splitting_MFsurfchem can be 0 or 1");
             }
+
+            VisMF::Write(surfcov,"a_surfcov");
 	}
 
         // FHD
-Original file line number
+Diff line change
@@ Expand Up / @@ -1030,6 +1030,8 @@ void main_driver(const char* argv) @@
     	    } else {
                     Abort("splitting_MFsurfchem can be 0 or 1");
                 }
+                VisMF::Write(surfcov,"a_surfcov");
     	}
             // FHD
@@ Expand Down @@