Merge remote-tracking branch 'origin/release-v4.6.1'
[WRF.git] / var / da / da_geoamv / da_jo_and_grady_geoamv.inc
blob88dcda3d2734a360b4081d6f6a7646710d80c74f
1 subroutine da_jo_and_grady_geoamv( iv, re, jo, jo_grad_y)
3    !-------------------------------------------------------------------------
4    ! Purpose:  Calculates Cost function and Gradient for Geo. CVMs 
5    !-------------------------------------------------------------------------
7    implicit none
9    type(iv_type), intent(in)    :: iv          ! Innovation vector.
10    type(y_type),  intent(in)    :: re          ! Residual vector.
11    type(y_type),  intent(inout) :: jo_grad_y   ! Grad_y(Jo)
12    type(jo_type), intent(inout) :: jo          ! Obs cost function.
14    integer :: n, k
16    if (trace_use_dull) call da_trace_entry("da_jo_and_grady_geoamv")
18    jo % geoamv_u = 0.0
19    jo % geoamv_v = 0.0
21    if (iv%info(geoamv)%nlocal > 0) then
22       do n=1, iv%info(geoamv)%nlocal
23          do k=1, iv%info(geoamv)%levels(n)
24             jo_grad_y%geoamv(n)%u(k) = -re%geoamv(n)%u(k) / ( iv%geoamv(n)%u(k)%error * iv%geoamv(n)%u(k)%error)
25             jo_grad_y%geoamv(n)%v(k) = -re%geoamv(n)%v(k) / ( iv%geoamv(n)%v(k)%error * iv%geoamv(n)%v(k)%error)
26          end do
28          do k=1, iv%info(geoamv)%levels(n)
29             if (iv%info(geoamv)%proc_domain(1,n)) then
30                jo % geoamv_u = jo % geoamv_u - re%geoamv(n)%u(k) * jo_grad_y%geoamv(n)%u(k)
31                jo % geoamv_v = jo % geoamv_v - re%geoamv(n)%v(k) * jo_grad_y%geoamv(n)%v(k)
32             end if
33          end do
34       end do
36       jo % geoamv_u = 0.5 * jo % geoamv_u
37       jo % geoamv_v = 0.5 * jo % geoamv_v
38    end if
40    if (trace_use_dull) call da_trace_exit("da_jo_and_grady_geoamv")
41      
42 end subroutine da_jo_and_grady_geoamv