Fix probe output

MFlowCode · Jun 7, 2024 · f756cd4 · f756cd4
1 parent a0a3ef8
commit f756cd4
Showing 1 changed file with 28 additions and 10 deletions.
diff --git a/src/simulation/m_data_output.fpp b/src/simulation/m_data_output.fpp
@@ -1230,14 +1230,20 @@ contains
                             vel(s) = q_cons_vf(cont_idx%end + s)%sf(j - 2, k - 2, l)/rho
                         end do
 
-                        call s_compute_pressure( &
-                            q_cons_vf(1)%sf(j - 2, k - 2, l), &
-                            q_cons_vf(alf_idx)%sf(j - 2, k - 2, l), &
-                            0.5d0*(q_cons_vf(2)%sf(j - 2, k - 2, l)**2.d0)/ &
-                            q_cons_vf(1)%sf(j - 2, k - 2, l), &
-                            pi_inf, gamma, rho, qv, pres, &
-                            q_cons_vf(stress_idx%beg)%sf(j - 2, k - 2, l), &
-                            q_cons_vf(mom_idx%beg)%sf(j - 2, k - 2, l), G)
+                        if (hypoelasticity) then
+                            call s_compute_pressure( &
+                                q_cons_vf(1)%sf(j - 2, k - 2, l), &
+                                q_cons_vf(alf_idx)%sf(j - 2, k - 2, l), &
+                                0.5d0*(q_cons_vf(2)%sf(j - 2, k - 2, l)**2.d0)/ &
+                                q_cons_vf(1)%sf(j - 2, k - 2, l), &
+                                pi_inf, gamma, rho, qv, pres, &
+                                q_cons_vf(stress_idx%beg)%sf(j - 2, k - 2, l), &
+                                q_cons_vf(mom_idx%beg)%sf(j - 2, k - 2, l), G)
+                        else
+                            call s_compute_pressure(q_cons_vf(E_idx)%sf(j - 2, k - 2, l), &
+                                                    q_cons_vf(alf_idx)%sf(j - 2, k - 2, l), &
+                                                    0.5d0*rho*dot_product(vel, vel), pi_inf, gamma, rho, qv, pres)
+                        end if
 
                         if (model_eqns == 4) then
                             lit_gamma = 1d0/fluid_pp(1)%gamma + 1d0
@@ -1307,8 +1313,20 @@ contains
                                 vel(s) = q_cons_vf(cont_idx%end + s)%sf(j - 2, k - 2, l - 2)/rho
                             end do
 
-                            call s_compute_pressure(q_cons_vf(E_idx)%sf(j - 2, k - 2, l - 2), &
-                                                    0d0, 0.5d0*rho*dot_product(vel, vel), pi_inf, gamma, rho, qv, pres)
+                            if (hypoelasticity) then
+                                call s_compute_pressure( &
+                                    q_cons_vf(1)%sf(j - 2, k - 2, l - 2), &
+                                    q_cons_vf(alf_idx)%sf(j - 2, k - 2, l - 2), &
+                                    0.5d0*(q_cons_vf(2)%sf(j - 2, k - 2, l - 2)**2.d0)/ &
+                                    q_cons_vf(1)%sf(j - 2, k - 2, l - 2), &
+                                    pi_inf, gamma, rho, qv, pres, &
+                                    q_cons_vf(stress_idx%beg)%sf(j - 2, k - 2, l - 2), &
+                                    q_cons_vf(mom_idx%beg)%sf(j - 2, k - 2, l - 2), G)
+                            else
+                                call s_compute_pressure(q_cons_vf(E_idx)%sf(j - 2, k - 2, l - 2), &
+                                                        q_cons_vf(alf_idx)%sf(j - 2, k - 2, l - 2), &
+                                                        0.5d0*rho*dot_product(vel, vel), pi_inf, gamma, rho, qv, pres)
+                            end if
 
                             ! Compute mixture sound speed
                             call s_compute_speed_of_sound(pres, rho, gamma, pi_inf, &