BC and smoothing stuff

wilfonba · Feb 20, 2025 · 95b76a4 · 95b76a4
1 parent 0a26135
commit 95b76a4
Show file tree

Hide file tree

Showing 6 changed files with 314 additions and 52 deletions.
diff --git a/src/common/m_derived_types.fpp b/src/common/m_derived_types.fpp
@@ -75,6 +75,9 @@ module m_derived_types
         integer :: beg
         integer :: end
 
+        logical :: b_extrap_ic
+        logical :: e_extrap_ic
+
         real(wp) :: vb1
         real(wp) :: vb2
         real(wp) :: vb3

diff --git a/src/pre_process/m_global_parameters.fpp b/src/pre_process/m_global_parameters.fpp
@@ -278,7 +278,7 @@ contains
 
         ! Computational domain parameters
         m = dflt_int; n = 0; p = 0
-        
+
         cyl_coord = .false.
 
         x_domain%beg = dflt_real

diff --git a/src/pre_process/m_mpi_proxy.fpp b/src/pre_process/m_mpi_proxy.fpp
@@ -344,6 +344,22 @@ contains
                     end if
                 end do
 
+                ! Boundary condition at the beginning
+                if (proc_coords(3) > 0 .or. (bc_z%beg == -1 .and. num_procs_z > 1)) then
+                    proc_coords(3) = proc_coords(3) - 1
+                    call MPI_CART_RANK(MPI_COMM_CART, proc_coords, &
+                                       bc_z%beg, ierr)
+                    proc_coords(3) = proc_coords(3) + 1
+                end if
+
+                ! Boundary condition at the end
+                if (proc_coords(3) > 0 .or. (bc_z%end == -1 .and. num_procs_z > 1)) then
+                    proc_coords(3) = proc_coords(3) + 1
+                    call MPI_CART_RANK(MPI_COMM_CART, proc_coords, &
+                                       bc_z%end, ierr)
+                    proc_coords(3) = proc_coords(3) - 1
+                end if
+
                 ! Beginning and end sub-domain boundary locations
                 if (parallel_io .neqv. .true.) then
                     if (old_grid .neqv. .true.) then
@@ -457,6 +473,22 @@ contains
                 end if
             end do
 
+            ! Boundary condition at the beginning
+            if (proc_coords(2) > 0 .or. (bc_y%beg == -1 .and. num_procs_y > 1)) then
+                proc_coords(2) = proc_coords(2) - 1
+                call MPI_CART_RANK(MPI_COMM_CART, proc_coords, &
+                                   bc_y%beg, ierr)
+                proc_coords(2) = proc_coords(2) + 1
+            end if
+
+            ! Boundary condition at the end
+            if (proc_coords(2) < num_procs_y - 1 .or. (bc_y%end == -1 .and. num_procs_y > 1)) then
+                proc_coords(2) = proc_coords(2) + 1
+                call MPI_CART_RANK(MPI_COMM_CART, proc_coords, &
+                                   bc_y%end, ierr)
+                proc_coords(2) = proc_coords(2) - 1
+            end if
+
             ! Beginning and end sub-domain boundary locations
             if (parallel_io .neqv. .true.) then
                 if (old_grid .neqv. .true.) then
@@ -522,6 +554,20 @@ contains
             end if
         end do
 
+        ! Boundary condition at the beginning
+        if (proc_coords(1) > 0 .or. (bc_x%beg == -1 .and. num_procs_x > 1)) then
+            proc_coords(1) = proc_coords(1) - 1
+            call MPI_CART_RANK(MPI_COMM_CART, proc_coords, bc_x%beg, ierr)
+            proc_coords(1) = proc_coords(1) + 1
+        end if
+
+        ! Boundary condition at the end
+        if (proc_coords(1) < num_procs_x - 1 .or. (bc_x%end == -1 .and. num_procs_x > 1)) then
+            proc_coords(1) = proc_coords(1) + 1
+            call MPI_CART_RANK(MPI_COMM_CART, proc_coords, bc_x%end, ierr)
+            proc_coords(1) = proc_coords(1) - 1
+        end if
+
         ! Beginning and end sub-domain boundary locations
         if (parallel_io .neqv. .true.) then
             if (old_grid .neqv. .true.) then