m_pressure_relaxation.fpp.f90

Go to the documentation of this file.

# 1 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
!>
!! @file
!! @brief Contains module m_pressure_relaxation
 
# 1 "/home/runner/work/MFC/MFC/src/common/include/case.fpp" 1
! This file exists so that Fypp can be run without generating case.fpp files for
! each target. This is useful when generating documentation, for example. This
! should also let MFC be built with CMake directly, without invoking mfc.sh.
 
! For pre-process.
# 8 "/home/runner/work/MFC/MFC/src/common/include/case.fpp"
 
! For moving immersed boundaries in simulation
# 12 "/home/runner/work/MFC/MFC/src/common/include/case.fpp"
# 6 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp" 2
# 1 "/home/runner/work/MFC/MFC/src/common/include/macros.fpp" 1
# 1 "/home/runner/work/MFC/MFC/src/common/include/parallel_macros.fpp" 1
# 1 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp" 1
# 2 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
# 3 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
# 4 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
# 5 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
# 6 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
 
# 8 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
# 9 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
# 10 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
 
# 17 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
 
# 46 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
 
# 58 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
 
# 68 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
 
# 98 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
 
# 110 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
 
# 120 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
 
# 167 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
! New line at end of file is required for FYPP
# 2 "/home/runner/work/MFC/MFC/src/common/include/parallel_macros.fpp" 2
# 1 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp" 1
# 1 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp" 1
# 2 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
# 3 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
# 4 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
# 5 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
# 6 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
 
# 8 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
# 9 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
# 10 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
 
# 17 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
 
# 46 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
 
# 58 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
 
# 68 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
 
# 98 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
 
# 110 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
 
# 120 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
 
# 167 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
! New line at end of file is required for FYPP
# 2 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp" 2
 
# 4 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
# 5 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
# 6 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
# 7 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
# 8 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
 
# 20 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
 
# 43 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
 
# 48 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
 
# 53 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
 
# 58 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
 
# 63 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
 
# 68 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
 
# 76 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
 
# 81 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
 
# 86 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
 
# 91 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
 
# 96 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
 
# 101 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
 
# 106 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
 
# 111 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
 
# 116 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
 
# 121 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
 
# 151 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
 
# 192 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
 
# 206 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
 
# 231 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
 
# 242 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
 
# 244 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
# 255 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
 
# 284 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
 
# 294 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
 
# 304 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
 
# 313 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
 
# 330 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
 
# 340 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
 
# 347 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
 
# 353 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
 
# 359 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
 
# 365 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
 
# 371 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
 
# 377 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
! New line at end of file is required for FYPP
# 3 "/home/runner/work/MFC/MFC/src/common/include/parallel_macros.fpp" 2
# 1 "/home/runner/work/MFC/MFC/src/common/include/acc_macros.fpp" 1
# 1 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp" 1
# 2 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
# 3 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
# 4 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
# 5 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
# 6 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
 
# 8 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
# 9 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
# 10 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
 
# 17 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
 
# 46 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
 
# 58 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
 
# 68 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
 
# 98 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
 
# 110 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
 
# 120 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
 
# 167 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
! New line at end of file is required for FYPP
# 2 "/home/runner/work/MFC/MFC/src/common/include/acc_macros.fpp" 2
 
# 7 "/home/runner/work/MFC/MFC/src/common/include/acc_macros.fpp"
 
# 17 "/home/runner/work/MFC/MFC/src/common/include/acc_macros.fpp"
 
# 22 "/home/runner/work/MFC/MFC/src/common/include/acc_macros.fpp"
 
# 27 "/home/runner/work/MFC/MFC/src/common/include/acc_macros.fpp"
 
# 32 "/home/runner/work/MFC/MFC/src/common/include/acc_macros.fpp"
 
# 37 "/home/runner/work/MFC/MFC/src/common/include/acc_macros.fpp"
 
# 42 "/home/runner/work/MFC/MFC/src/common/include/acc_macros.fpp"
 
# 47 "/home/runner/work/MFC/MFC/src/common/include/acc_macros.fpp"
 
# 52 "/home/runner/work/MFC/MFC/src/common/include/acc_macros.fpp"
 
# 57 "/home/runner/work/MFC/MFC/src/common/include/acc_macros.fpp"
 
# 62 "/home/runner/work/MFC/MFC/src/common/include/acc_macros.fpp"
 
# 73 "/home/runner/work/MFC/MFC/src/common/include/acc_macros.fpp"
 
# 78 "/home/runner/work/MFC/MFC/src/common/include/acc_macros.fpp"
 
# 83 "/home/runner/work/MFC/MFC/src/common/include/acc_macros.fpp"
 
# 88 "/home/runner/work/MFC/MFC/src/common/include/acc_macros.fpp"
 
# 103 "/home/runner/work/MFC/MFC/src/common/include/acc_macros.fpp"
 
# 131 "/home/runner/work/MFC/MFC/src/common/include/acc_macros.fpp"
 
# 160 "/home/runner/work/MFC/MFC/src/common/include/acc_macros.fpp"
 
# 175 "/home/runner/work/MFC/MFC/src/common/include/acc_macros.fpp"
 
# 193 "/home/runner/work/MFC/MFC/src/common/include/acc_macros.fpp"
 
# 215 "/home/runner/work/MFC/MFC/src/common/include/acc_macros.fpp"
 
# 244 "/home/runner/work/MFC/MFC/src/common/include/acc_macros.fpp"
 
# 259 "/home/runner/work/MFC/MFC/src/common/include/acc_macros.fpp"
 
# 269 "/home/runner/work/MFC/MFC/src/common/include/acc_macros.fpp"
 
# 278 "/home/runner/work/MFC/MFC/src/common/include/acc_macros.fpp"
 
# 294 "/home/runner/work/MFC/MFC/src/common/include/acc_macros.fpp"
 
# 304 "/home/runner/work/MFC/MFC/src/common/include/acc_macros.fpp"
 
# 311 "/home/runner/work/MFC/MFC/src/common/include/acc_macros.fpp"
! New line at end of file is required for FYPP
# 4 "/home/runner/work/MFC/MFC/src/common/include/parallel_macros.fpp" 2
 
! GPU parallel region (scalar reductions, maxval/minval)
# 23 "/home/runner/work/MFC/MFC/src/common/include/parallel_macros.fpp"
 
! GPU parallel loop over threads (most common GPU macro)
# 43 "/home/runner/work/MFC/MFC/src/common/include/parallel_macros.fpp"
 
! Required closing for GPU_PARALLEL_LOOP
# 55 "/home/runner/work/MFC/MFC/src/common/include/parallel_macros.fpp"
 
! Mark routine for device compilation
# 112 "/home/runner/work/MFC/MFC/src/common/include/parallel_macros.fpp"
 
! Declare device-resident data
# 130 "/home/runner/work/MFC/MFC/src/common/include/parallel_macros.fpp"
 
! Inner loop within a GPU parallel region
# 145 "/home/runner/work/MFC/MFC/src/common/include/parallel_macros.fpp"
 
! Scoped GPU data region
# 164 "/home/runner/work/MFC/MFC/src/common/include/parallel_macros.fpp"
 
! Host code with device pointers (for MPI with GPU buffers)
# 193 "/home/runner/work/MFC/MFC/src/common/include/parallel_macros.fpp"
 
! Allocate device memory (unscoped)
# 207 "/home/runner/work/MFC/MFC/src/common/include/parallel_macros.fpp"
 
! Free device memory
# 219 "/home/runner/work/MFC/MFC/src/common/include/parallel_macros.fpp"
 
! Atomic operation on device
# 231 "/home/runner/work/MFC/MFC/src/common/include/parallel_macros.fpp"
 
! End atomic capture block
# 242 "/home/runner/work/MFC/MFC/src/common/include/parallel_macros.fpp"
 
! Copy data between host and device
# 254 "/home/runner/work/MFC/MFC/src/common/include/parallel_macros.fpp"
 
! Synchronization barrier
# 266 "/home/runner/work/MFC/MFC/src/common/include/parallel_macros.fpp"
 
! Import GPU library module (openacc or omp_lib)
# 275 "/home/runner/work/MFC/MFC/src/common/include/parallel_macros.fpp"
 
! Emit code only for AMD compiler
# 282 "/home/runner/work/MFC/MFC/src/common/include/parallel_macros.fpp"
 
! Emit code for non-Cray compilers
# 289 "/home/runner/work/MFC/MFC/src/common/include/parallel_macros.fpp"
 
! Emit code only for Cray compiler
# 296 "/home/runner/work/MFC/MFC/src/common/include/parallel_macros.fpp"
 
! Emit code for non-NVIDIA compilers
# 303 "/home/runner/work/MFC/MFC/src/common/include/parallel_macros.fpp"
 
# 305 "/home/runner/work/MFC/MFC/src/common/include/parallel_macros.fpp"
# 306 "/home/runner/work/MFC/MFC/src/common/include/parallel_macros.fpp"
! New line at end of file is required for FYPP
# 2 "/home/runner/work/MFC/MFC/src/common/include/macros.fpp" 2
 
# 14 "/home/runner/work/MFC/MFC/src/common/include/macros.fpp"
 
! Caution: This macro requires the use of a binding script to set CUDA_VISIBLE_DEVICES, such that we have one GPU device per MPI
! rank. That's because for both cudaMemAdvise (preferred location) and cudaMemPrefetchAsync we use location = device_id = 0. For an
! example see misc/nvidia_uvm/bind.sh.
# 55 "/home/runner/work/MFC/MFC/src/common/include/macros.fpp"
 
! Allocate and create GPU device memory
# 75 "/home/runner/work/MFC/MFC/src/common/include/macros.fpp"
 
! Free GPU device memory and deallocate
# 83 "/home/runner/work/MFC/MFC/src/common/include/macros.fpp"
 
! Cray-specific GPU pointer setup for vector fields
# 107 "/home/runner/work/MFC/MFC/src/common/include/macros.fpp"
 
! Cray-specific GPU pointer setup for scalar fields
# 123 "/home/runner/work/MFC/MFC/src/common/include/macros.fpp"
 
! Cray-specific GPU pointer setup for acoustic source spatials
# 148 "/home/runner/work/MFC/MFC/src/common/include/macros.fpp"
 
# 154 "/home/runner/work/MFC/MFC/src/common/include/macros.fpp"
 
# 161 "/home/runner/work/MFC/MFC/src/common/include/macros.fpp"
! New line at end of file is required for FYPP
# 7 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp" 2
 
!> @brief Pressure relaxation for the six-equation multi-component model via Newton--Raphson equilibration and volume-fraction
!! correction
module m_pressure_relaxation
 
    use m_derived_types
    use m_global_parameters
    use m_constants, only: model_eqns_5eq
 
    implicit none
 
    private; public :: s_pressure_relaxation_procedure, s_initialize_pressure_relaxation_module, &
        & s_finalize_pressure_relaxation_module
 
    real(wp), allocatable, dimension(:,:) :: res_pr
 
# 22 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#if defined(MFC_OpenACC)
# 22 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
!$acc declare create(Res_pr) 
# 22 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#elif defined(MFC_OpenMP)
# 22 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
!$omp declare target (Res_pr) 
# 22 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#endif
 
contains
 
    !> Initialize the pressure relaxation module
    impure subroutine s_initialize_pressure_relaxation_module
 
        integer :: i, j
 
        if (viscous) then
#ifdef MFC_DEBUG
# 32 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
    block
# 32 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
        use iso_fortran_env, only: output_unit
# 32 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
 
# 32 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
        print *, 'm_pressure_relaxation.fpp:32: ', '@:ALLOCATE(Res_pr(1:2, 1:Re_size_max))'
# 32 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
 
# 32 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
        call flush (output_unit)
# 32 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
    end block
# 32 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#endif
# 32 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
    allocate (res_pr(1:2, 1:re_size_max))
# 32 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
 
# 32 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
 
# 32 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#if defined(MFC_OpenACC)
# 32 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
!$acc enter data create(Res_pr) 
# 32 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#elif defined(MFC_OpenMP)
# 32 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
!$omp target enter data map(always,alloc:Res_pr) 
# 32 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#endif
            do i = 1, 2
                do j = 1, re_size(i)
                    res_pr(i, j) = fluid_pp(re_idx(i, j))%Re(i)
                end do
            end do
 
# 38 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#if defined(MFC_OpenACC)
# 38 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
!$acc update device(Res_pr, Re_idx, Re_size) 
# 38 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#elif defined(MFC_OpenMP)
# 38 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
!$omp target update to(Res_pr, Re_idx, Re_size) 
# 38 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#endif
        end if
 
    end subroutine s_initialize_pressure_relaxation_module
 
    !> Finalize the pressure relaxation module
    impure subroutine s_finalize_pressure_relaxation_module
 
        if (viscous) then
#ifdef MFC_DEBUG
# 47 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
    block
# 47 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
        use iso_fortran_env, only: output_unit
# 47 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
 
# 47 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
        print *, 'm_pressure_relaxation.fpp:47: ', '@:DEALLOCATE(Res_pr)'
# 47 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
 
# 47 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
        call flush (output_unit)
# 47 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
    end block
# 47 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#endif
# 47 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
 
# 47 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#if defined(MFC_OpenACC)
# 47 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
!$acc exit data delete(Res_pr) 
# 47 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#elif defined(MFC_OpenMP)
# 47 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
!$omp target exit data map(release:Res_pr) 
# 47 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#endif
# 47 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
    deallocate (res_pr)
        end if
 
    end subroutine s_finalize_pressure_relaxation_module
 
    !> The main pressure relaxation procedure
    subroutine s_pressure_relaxation_procedure(q_cons_vf)
 
        type(scalar_field), dimension(sys_size), intent(inout) :: q_cons_vf
        integer                                                :: j, k, l
 
 
# 58 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
 
# 58 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#if defined(MFC_OpenACC)
# 58 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
!$acc parallel loop collapse(3) gang vector default(present) private(j, k, l)
# 58 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#elif defined(MFC_OpenMP)
# 58 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
 
# 58 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
 
# 58 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
 
# 58 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
!$omp target teams loop defaultmap(firstprivate:scalar) bind(teams,parallel) collapse(3) defaultmap(tofrom:aggregate) defaultmap(tofrom:allocatable) defaultmap(tofrom:pointer) private(j, k, l)
# 58 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#endif
        do l = 0, p
            do k = 0, n
                do j = 0, m
                    call s_relax_cell_pressure(q_cons_vf, j, k, l)
                end do
            end do
        end do
 
# 66 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#if defined(MFC_OpenACC)
# 66 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
!$acc end parallel loop
# 66 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#elif defined(MFC_OpenMP)
# 66 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
 
# 66 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
!$omp end target teams loop
# 66 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#endif
 
    end subroutine s_pressure_relaxation_procedure
 
    !> Process pressure relaxation for a single cell
    subroutine s_relax_cell_pressure(q_cons_vf, j, k, l)
 
 
# 73 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#if MFC_OpenACC
# 73 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
!$acc routine seq 
# 73 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#elif MFC_OpenMP
# 73 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
 
# 73 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
 
# 73 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
!$omp declare target device_type(any) 
# 73 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#endif
 
        type(scalar_field), dimension(sys_size), intent(inout) :: q_cons_vf
        integer, intent(in)                                    :: j, k, l
 
        ! Volume fraction correction
        if (mpp_lim) call s_correct_volume_fractions(q_cons_vf, j, k, l)
 
        ! Pressure equilibration
        if (s_needs_pressure_relaxation(q_cons_vf, j, k, l)) then
            call s_equilibrate_pressure(q_cons_vf, j, k, l)
        end if
 
        ! Internal energy correction
        call s_correct_internal_energies(q_cons_vf, j, k, l)
 
    end subroutine s_relax_cell_pressure
 
    !> Check if pressure relaxation is needed for this cell
    logical function s_needs_pressure_relaxation(q_cons_vf, j, k, l)
 
 
# 94 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#if MFC_OpenACC
# 94 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
!$acc routine seq 
# 94 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#elif MFC_OpenMP
# 94 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
 
# 94 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
 
# 94 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
!$omp declare target device_type(any) 
# 94 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#endif
 
        type(scalar_field), dimension(sys_size), intent(in) :: q_cons_vf
        integer, intent(in)                                 :: j, k, l
        integer                                             :: i
 
        s_needs_pressure_relaxation = .true.
 
# 101 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#if defined(MFC_OpenACC)
# 101 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
!$acc loop seq 
# 101 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#elif defined(MFC_OpenMP)
# 101 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
 
# 101 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#endif
        do i = 1, num_fluids
            if (q_cons_vf(i + eqn_idx%adv%beg - 1)%sf(j, k, l) > (1._wp - sgm_eps)) then
                s_needs_pressure_relaxation = .false.
            end if
        end do
 
    end function s_needs_pressure_relaxation
 
    !> Correct volume fractions to physical bounds
    subroutine s_correct_volume_fractions(q_cons_vf, j, k, l)
 
 
# 113 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#if MFC_OpenACC
# 113 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
!$acc routine seq 
# 113 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#elif MFC_OpenMP
# 113 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
 
# 113 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
 
# 113 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
!$omp declare target device_type(any) 
# 113 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#endif
 
        type(scalar_field), dimension(sys_size), intent(inout) :: q_cons_vf
        integer, intent(in)                                    :: j, k, l
        real(wp)                                               :: sum_alpha
        integer                                                :: i
 
        sum_alpha = 0._wp
 
# 121 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#if defined(MFC_OpenACC)
# 121 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
!$acc loop seq 
# 121 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#elif defined(MFC_OpenMP)
# 121 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
 
# 121 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#endif
        do i = 1, num_fluids
            if ((q_cons_vf(i + eqn_idx%cont%beg - 1)%sf(j, k, l) < 0._wp) .or. (q_cons_vf(i + eqn_idx%adv%beg - 1)%sf(j, k, &
                & l) < 0._wp)) then
                q_cons_vf(i + eqn_idx%cont%beg - 1)%sf(j, k, l) = 0._wp
                q_cons_vf(i + eqn_idx%adv%beg - 1)%sf(j, k, l) = 0._wp
                q_cons_vf(i + eqn_idx%int_en%beg - 1)%sf(j, k, l) = 0._wp
            end if
            if (q_cons_vf(i + eqn_idx%adv%beg - 1)%sf(j, k, l) > 1._wp) q_cons_vf(i + eqn_idx%adv%beg - 1)%sf(j, k, l) = 1._wp
            sum_alpha = sum_alpha + q_cons_vf(i + eqn_idx%adv%beg - 1)%sf(j, k, l)
        end do
 
 
# 133 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#if defined(MFC_OpenACC)
# 133 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
!$acc loop seq 
# 133 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#elif defined(MFC_OpenMP)
# 133 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
 
# 133 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#endif
        do i = 1, num_fluids
            q_cons_vf(i + eqn_idx%adv%beg - 1)%sf(j, k, l) = q_cons_vf(i + eqn_idx%adv%beg - 1)%sf(j, k, l)/sum_alpha
        end do
 
    end subroutine s_correct_volume_fractions
 
    !> Main pressure equilibration using Newton-Raphson
    subroutine s_equilibrate_pressure(q_cons_vf, j, k, l)
 
 
# 143 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#if MFC_OpenACC
# 143 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
!$acc routine seq 
# 143 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#elif MFC_OpenMP
# 143 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
 
# 143 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
 
# 143 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
!$omp declare target device_type(any) 
# 143 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#endif
 
        type(scalar_field), dimension(sys_size), intent(inout) :: q_cons_vf
        integer, intent(in)                                    :: j, k, l
        real(wp)                                               :: pres_relax, f_pres, df_pres
# 151 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
            real(wp), dimension(num_fluids) :: pres_K_init, rho_K_s
# 153 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
        integer, parameter :: MAX_ITER = 50
        ! Pressure relaxation convergence tolerance
        real(wp), parameter :: TOLERANCE = 1.e-10_wp
        integer             :: iter, i
 
        ! Initialize pressures
        pres_relax = 0._wp
 
# 160 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#if defined(MFC_OpenACC)
# 160 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
!$acc loop seq 
# 160 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#elif defined(MFC_OpenMP)
# 160 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
 
# 160 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#endif
        do i = 1, num_fluids
            if (q_cons_vf(i + eqn_idx%adv%beg - 1)%sf(j, k, l) > sgm_eps) then
                ! Phasic internal energy carries the formation energy: alpha_rho_k*qv_k must be
                ! removed before inverting the stiffened-gas EOS, or a nonzero qv inflates the
                ! phasic pressure by rho_k*qv_k/gamma_k (this is what breaks the reactive burn).
                pres_k_init(i) = ((q_cons_vf(i + eqn_idx%int_en%beg - 1)%sf(j, k, l) - q_cons_vf(i + eqn_idx%cont%beg - 1)%sf(j, &
                            & k, l)*qvs(i))/q_cons_vf(i + eqn_idx%adv%beg - 1)%sf(j, k, l) - pi_infs(i))/gammas(i)
                if (pres_k_init(i) <= -(1._wp - 1.e-8_wp)*ps_inf(i) + 1.e-8_wp) pres_k_init(i) = -(1._wp - 1.e-8_wp)*ps_inf(i) &
                    & + 1.e-8_wp
            else
                pres_k_init(i) = 0._wp
            end if
            pres_relax = pres_relax + q_cons_vf(i + eqn_idx%adv%beg - 1)%sf(j, k, l)*pres_k_init(i)
        end do
 
        ! Newton-Raphson iteration
        f_pres = 1.e-9_wp
        df_pres = 1.e9_wp
 
# 179 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#if defined(MFC_OpenACC)
# 179 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
!$acc loop seq 
# 179 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#elif defined(MFC_OpenMP)
# 179 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
 
# 179 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#endif
        do iter = 0, max_iter - 1
            if (abs(f_pres) > tolerance) then
                pres_relax = pres_relax - f_pres/df_pres
 
                ! Enforce pressure bounds
                do i = 1, num_fluids
                    if (pres_relax <= -(1._wp - 1.e-8_wp)*ps_inf(i) + 1.e-8_wp) pres_relax = -(1._wp - 1.e-8_wp)*ps_inf(i) &
                        & + 1.e-8_wp
                end do
 
                ! Newton-Raphson step
                f_pres = -1._wp
                df_pres = 0._wp
 
# 193 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#if defined(MFC_OpenACC)
# 193 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
!$acc loop seq 
# 193 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#elif defined(MFC_OpenMP)
# 193 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
 
# 193 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#endif
                do i = 1, num_fluids
                    if (q_cons_vf(i + eqn_idx%adv%beg - 1)%sf(j, k, l) > sgm_eps) then
                        ! Isentropic relation: rho = rho0 * (p/p0)^(1/gamma), Saurel et al. JFM (2009)
                        rho_k_s(i) = q_cons_vf(i + eqn_idx%cont%beg - 1)%sf(j, k, l)/max(q_cons_vf(i + eqn_idx%adv%beg - 1)%sf(j, &
                                & k, l), sgm_eps)*((pres_relax + ps_inf(i))/(pres_k_init(i) + ps_inf(i)))**(1._wp/gs_min(i))
                        f_pres = f_pres + q_cons_vf(i + eqn_idx%cont%beg - 1)%sf(j, k, l)/rho_k_s(i)
                        df_pres = df_pres - q_cons_vf(i + eqn_idx%cont%beg - 1)%sf(j, k, &
                                                      & l)/(gs_min(i)*rho_k_s(i)*(pres_relax + ps_inf(i)))
                    end if
                end do
            end if
        end do
 
        ! Update volume fractions
 
# 208 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#if defined(MFC_OpenACC)
# 208 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
!$acc loop seq 
# 208 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#elif defined(MFC_OpenMP)
# 208 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
 
# 208 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#endif
        do i = 1, num_fluids
            if (q_cons_vf(i + eqn_idx%adv%beg - 1)%sf(j, k, l) > sgm_eps) q_cons_vf(i + eqn_idx%adv%beg - 1)%sf(j, k, &
                & l) = q_cons_vf(i + eqn_idx%cont%beg - 1)%sf(j, k, l)/rho_k_s(i)
        end do
 
    end subroutine s_equilibrate_pressure
 
    !> Correct internal energies using equilibrated pressure
    subroutine s_correct_internal_energies(q_cons_vf, j, k, l)
 
 
# 219 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#if MFC_OpenACC
# 219 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
!$acc routine seq 
# 219 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#elif MFC_OpenMP
# 219 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
 
# 219 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
 
# 219 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
!$omp declare target device_type(any) 
# 219 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#endif
 
        type(scalar_field), dimension(sys_size), intent(inout) :: q_cons_vf
        integer, intent(in)                                    :: j, k, l
# 226 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
            real(wp), dimension(num_fluids) :: alpha_rho, alpha
# 228 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
        real(wp)               :: rho, dyn_pres, gamma, pi_inf, pres_relax, sum_alpha, qv_mix
        real(wp), dimension(2) :: Re
        integer                :: i, q
 
 
# 232 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#if defined(MFC_OpenACC)
# 232 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
!$acc loop seq 
# 232 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#elif defined(MFC_OpenMP)
# 232 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
 
# 232 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#endif
        do i = 1, num_fluids
            alpha_rho(i) = q_cons_vf(i)%sf(j, k, l)
            alpha(i) = q_cons_vf(eqn_idx%E + i)%sf(j, k, l)
        end do
 
        ! Compute mixture properties (combined bubble and standard logic)
        rho = 0._wp
        gamma = 0._wp
        pi_inf = 0._wp
 
        if (bubbles_euler) then
            if (mpp_lim .and. (model_eqns == model_eqns_5eq) .and. (num_fluids > 2)) then
 
# 245 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#if defined(MFC_OpenACC)
# 245 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
!$acc loop seq 
# 245 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#elif defined(MFC_OpenMP)
# 245 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
 
# 245 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#endif
                do i = 1, num_fluids
                    rho = rho + alpha_rho(i)
                    gamma = gamma + alpha(i)*gammas(i)
                    pi_inf = pi_inf + alpha(i)*pi_infs(i)
                end do
            else if ((model_eqns == model_eqns_5eq) .and. (num_fluids > 2)) then
 
# 252 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#if defined(MFC_OpenACC)
# 252 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
!$acc loop seq 
# 252 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#elif defined(MFC_OpenMP)
# 252 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
 
# 252 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#endif
                do i = 1, num_fluids - 1
                    rho = rho + alpha_rho(i)
                    gamma = gamma + alpha(i)*gammas(i)
                    pi_inf = pi_inf + alpha(i)*pi_infs(i)
                end do
            else
                rho = alpha_rho(1)
                gamma = gammas(1)
                pi_inf = pi_infs(1)
            end if
        else
            sum_alpha = 0._wp
            if (mpp_lim) then
 
# 266 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#if defined(MFC_OpenACC)
# 266 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
!$acc loop seq 
# 266 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#elif defined(MFC_OpenMP)
# 266 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
 
# 266 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#endif
                do i = 1, num_fluids
                    alpha_rho(i) = max(0._wp, alpha_rho(i))
                    alpha(i) = min(max(0._wp, alpha(i)), 1._wp)
                    sum_alpha = sum_alpha + alpha(i)
                end do
                alpha = alpha/max(sum_alpha, sgm_eps)
            end if
 
 
# 275 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#if defined(MFC_OpenACC)
# 275 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
!$acc loop seq 
# 275 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#elif defined(MFC_OpenMP)
# 275 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
 
# 275 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#endif
            do i = 1, num_fluids
                rho = rho + alpha_rho(i)
                gamma = gamma + alpha(i)*gammas(i)
                pi_inf = pi_inf + alpha(i)*pi_infs(i)
            end do
 
            if (viscous) then
 
# 283 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#if defined(MFC_OpenACC)
# 283 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
!$acc loop seq 
# 283 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#elif defined(MFC_OpenMP)
# 283 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
 
# 283 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#endif
                do i = 1, 2
                    re(i) = dflt_real
                    if (re_size(i) > 0) re(i) = 0._wp
 
# 287 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#if defined(MFC_OpenACC)
# 287 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
!$acc loop seq 
# 287 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#elif defined(MFC_OpenMP)
# 287 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
 
# 287 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#endif
                    do q = 1, re_size(i)
                        re(i) = alpha(re_idx(i, q))/res_pr(i, q) + re(i)
                    end do
                    re(i) = 1._wp/max(re(i), sgm_eps)
                end do
            end if
        end if
 
        ! Compute dynamic pressure and update internal energies
        dyn_pres = 0._wp
 
# 298 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#if defined(MFC_OpenACC)
# 298 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
!$acc loop seq 
# 298 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#elif defined(MFC_OpenMP)
# 298 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
 
# 298 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#endif
        do i = eqn_idx%mom%beg, eqn_idx%mom%end
            dyn_pres = dyn_pres + 5.e-1_wp*q_cons_vf(i)%sf(j, k, l)*q_cons_vf(i)%sf(j, k, l)/max(rho, sgm_eps)
        end do
 
        ! Mixture formation energy: the relaxed mixture pressure is recovered from the
        ! conserved total energy with the qv reference removed (consistent with s_compute_pressure).
        qv_mix = 0._wp
 
# 306 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#if defined(MFC_OpenACC)
# 306 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
!$acc loop seq 
# 306 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#elif defined(MFC_OpenMP)
# 306 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
 
# 306 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#endif
        do i = 1, num_fluids
            qv_mix = qv_mix + alpha_rho(i)*qvs(i)
        end do
 
        pres_relax = (q_cons_vf(eqn_idx%E)%sf(j, k, l) - dyn_pres - qv_mix - pi_inf)/gamma
 
 
# 313 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#if defined(MFC_OpenACC)
# 313 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
!$acc loop seq 
# 313 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#elif defined(MFC_OpenMP)
# 313 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
 
# 313 "/home/runner/work/MFC/MFC/src/simulation/m_pressure_relaxation.fpp"
#endif
        do i = 1, num_fluids
            q_cons_vf(i + eqn_idx%int_en%beg - 1)%sf(j, k, l) = q_cons_vf(i + eqn_idx%adv%beg - 1)%sf(j, k, &
                      & l)*(gammas(i)*pres_relax + pi_infs(i)) + q_cons_vf(i + eqn_idx%cont%beg - 1)%sf(j, k, l)*qvs(i)
        end do
 
    end subroutine s_correct_internal_energies
 
end module m_pressure_relaxation