From 66b1057c9ab251e5cf5445ad676cf233c62bb152 Mon Sep 17 00:00:00 2001
From: zhi <zhichen0428@gmail.com>
Date: Thu, 12 Dec 2024 20:54:15 -0500
Subject: [PATCH] allow eos pressure in actual_trans_final too

---
 Source/hydro/trans.cpp | 31 ++++++++++++++++++++++++++-----
 1 file changed, 26 insertions(+), 5 deletions(-)

diff --git a/Source/hydro/trans.cpp b/Source/hydro/trans.cpp
index dfea939024..3bc05c081e 100644
--- a/Source/hydro/trans.cpp
+++ b/Source/hydro/trans.cpp
@@ -908,12 +908,33 @@ Castro::actual_trans_final(const Box& bx,  // NOLINT(readability-convert-member-
                 qo_arr(i,j,k,QREINT) = q_arr(i,j,k,QREINT);
             }
 
-            // Pretend QREINT has been fixed and transverse_use_eos != 1.
-            // If we are wrong, we will fix it later.
+            // Pretend QREINT has been fixed
+            // We can get pressure update via eos or using p-evolution equation
+
+            if (transverse_use_eos) {
+
+                // With the EOS route:
+                eos_rep_t eos_state;
+                eos_state.rho = rrnewn;
+                eos_state.e = qo_arr(i,j,k,QREINT) / rrnewn;
+                for (int n = 0; n < NumSpec; n++) {
+                    eos_state.xn[n] = qo_arr(i,j,k,QFS+n);
+                }
+#if NAUX_NET > 0
+                for (int n = 0; n < NumAux; n++) {
+                    eos_state.aux[n] = qo_arr(i,j,k,QFX+n);
+                }
+#endif
+                eos(eos_input_re, eos_state);
+                qo_arr(i,j,k,QPRES) = eos_state.p;
+
+            } else {
+
+                // add the transverse term to the p evolution eq here
+                Real pnewn = q_arr(i,j,k,QPRES) - pt1new - pt2new;
+                qo_arr(i,j,k,QPRES) = pnewn;
+            }
 
-            // add the transverse term to the p evolution eq here
-            Real pnewn = q_arr(i,j,k,QPRES) - pt1new - pt2new;
-            qo_arr(i,j,k,QPRES) = pnewn;
         }
         else {
             qo_arr(i,j,k,QPRES) = q_arr(i,j,k,QPRES);