move neutral_mld & u_dynH to richnum_mode if statement

obl_inmom.f90

@@ -91,9 +91,13 @@ module obl_inmom
       call legacy_n2(den, hhq, zw, g, lu, n2)
       call legacy_rit(n2, s2, border_shift, lu, rit(:,:,2:size(rit, 3)))
       call legacy_rit_top(rlh, taux, tauy, border_shift, lu, rit(:,:,1))
-    end if
-    ! print *, "hello aft rit"
       call sync_xy_border_3d(rit)
+      ! neutral mld & u_dynH for pph_dyn
+      call legacy_neutral_mld(rlh, taux, tauy, border_shift, lu, neutral_mld, u_dynH)
+      call sync_xy_border_2d(neutral_mld)
+      call sync_xy_border_2d(u_dynH)
+      !print *, "neutral_mld:", neutral_mld
+    end if
 
     do j = 1, ny
       do i = 1, nx
@@ -115,11 +119,6 @@ module obl_inmom
             call pph_kh(kh(i,j,:), rit(i,j,:), pphParams, nz)
             call pph_km(km(i,j,:), rit(i,j,:), pphParams, nz)
           else if (kh_km_mode == 3) then
-            if (i == 1 .and. j == 1) then
-                call legacy_neutral_mld(rlh, taux, tauy, border_shift, lu, neutral_mld, u_dynH)
-                call sync_xy_border_2d(neutral_mld)
-                call sync_xy_border_2d(u_dynH)
-            end if
             call pph_dyn_kh(kh(i,j,:), rit(i,j,:), u_dynH(i,j), neutral_mld(i,j), pphdynParams, nz)
             call pph_dyn_km(km(i,j,:), rit(i,j,:), u_dynH(i,j), neutral_mld(i,j), pphdynParams, nz)
           end if