output bulk viscosity in plotfiles

AMReX-FHD · Sep 21, 2023 · 1fd3499 · 1fd3499
1 parent b2f3450
commit 1fd3499
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 4 deletions.
diff --git a/src_compressible_stag/compressible_functions_stag.H b/src_compressible_stag/compressible_functions_stag.H
@@ -32,7 +32,8 @@ void WritePlotFileStag(int step,
                        const MultiFab& surfcovMeans,
                        const MultiFab& surfcovVars,
                        const MultiFab& eta, 
-                       const MultiFab& kappa);
+                       const MultiFab& kappa,
+                       const MultiFab& zeta);
 
 void WriteSpatialCross3D(const Vector<Real>& spatialCross, int step, const Geometry& geom, const int ncross);
 

diff --git a/src_compressible_stag/main_driver.cpp b/src_compressible_stag/main_driver.cpp
@@ -717,7 +717,7 @@ void main_driver(const char* argv)
 
         if (plot_int > 0) {
             WritePlotFileStag(0, 0.0, geom, cu, cuMeans, cuVars, cumom, cumomMeans, cumomVars, 
-                          prim, primMeans, primVars, vel, velMeans, velVars, coVars, surfcov, surfcovMeans, surfcovVars, eta, kappa);
+                          prim, primMeans, primVars, vel, velMeans, velVars, coVars, surfcov, surfcovMeans, surfcovVars, eta, kappa, zeta);
 #if defined(TURB)
             if (turbForcing > 0) {
                 EvaluateWritePlotFileVelGrad(0, 0.0, geom, vel, vel_decomp);
@@ -1145,7 +1145,7 @@ void main_driver(const char* argv)
             //yzAverage(cuMeans, cuVars, primMeans, primVars, spatialCross,
             //          cuMeansAv, cuVarsAv, primMeansAv, primVarsAv, spatialCrossAv);
             WritePlotFileStag(step, time, geom, cu, cuMeans, cuVars, cumom, cumomMeans, cumomVars,
-                              prim, primMeans, primVars, vel, velMeans, velVars, coVars, surfcov, surfcovMeans, surfcovVars, eta, kappa);
+                              prim, primMeans, primVars, vel, velMeans, velVars, coVars, surfcov, surfcovMeans, surfcovVars, eta, kappa, zeta);
 
 #if defined(TURB)
             if (turbForcing > 0) {

diff --git a/src_compressible_stag/writePlotFileStag.cpp b/src_compressible_stag/writePlotFileStag.cpp
@@ -24,7 +24,8 @@ void WritePlotFileStag(int step,
                        const amrex::MultiFab& surfcovMeans,
                        const amrex::MultiFab& surfcovVars,
                        const amrex::MultiFab& eta,
-                       const amrex::MultiFab& kappa)
+                       const amrex::MultiFab& kappa,
+                       const amrex::MultiFab& zeta)
 {
     BL_PROFILE_VAR("writePlotFileStag()",writePlotFileStag);
 
@@ -38,11 +39,13 @@ void WritePlotFileStag(int step,
     // prim: [vx, vy, vz, T, p, Yk, Xk] -- 5 + 2*nspecies
     // shifted [vx, vy, vz] -- 3
     // eta, kappa -- 2
+    // zeta -- 1 (only when visc_type = 3)
     nplot += nvars;
     nplot += 3;
     nplot += 5 + 2*nspecies;
     nplot += 3;
     nplot += 2;
+    if (amrex::Math::abs(visc_type) == 3) nplot += 1;
 
     if (nspec_surfcov>0) nplot += nspec_surfcov;
 
@@ -132,6 +135,14 @@ void WritePlotFileStag(int step,
     amrex::MultiFab::Copy(plotfile,kappa,0,cnt,numvars,0);
     cnt+=numvars;
 
+    // instantaneous
+    // zeta -- 1
+    if (amrex::Math::abs(visc_type) == 3) {
+        numvars = 1;
+        amrex::MultiFab::Copy(plotfile,zeta,0,cnt,numvars,0);
+        cnt+=numvars;
+    }
+
     // instantaneous
     // surfcov -- nspec_surfcov
     if (nspec_surfcov>0) {
@@ -277,6 +288,7 @@ void WritePlotFileStag(int step,
 
     varNames[cnt++] = "eta";
     varNames[cnt++] = "kappa";
+    if (amrex::Math::abs(visc_type) == 3) varNames[cnt++] = "zeta";
 
     if (nspec_surfcov>0) {
         x = "surfcov_";