var/da/da_sound/da_jo_and_grady_sound.inc

   1 subroutine da_jo_and_grady_sound(iv, re, jo, jo_grad_y)
   2
   3    !-----------------------------------------------------------------------
   4    ! Purpose: TBD
   5    !-----------------------------------------------------------------------
   6
   7    implicit none
   8
   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.
  13
  14    integer                      :: n, k
  15    ! the following "global" objects are used only when testing
  16    type (iv_type) :: iv_glob         ! Global Innovation vector (O-B).
  17    type (y_type)  :: re_glob         ! Global Residual vector (O-A).
  18    type (y_type)  :: jo_grad_y_glob  ! Global Grad_y(Jo)
  19
  20    if (trace_use_dull) call da_trace_entry("da_jo_and_grady_sound")
  21
  22    jo % sound_u = 0.0
  23    jo % sound_v = 0.0
  24    jo % sound_t = 0.0
  25    jo % sound_q = 0.0
  26
  27    if (test_dm_exact) then
  28       if (iv%info(sound)%ntotal == 0) then
  29          if (trace_use_dull) call da_trace_exit("da_jo_and_grady_sound")
  30          return
  31       end if
  32    else
  33       if (iv%info(sound)%nlocal < 1) then
  34          if (trace_use_dull) call da_trace_exit("da_jo_and_grady_sound")
  35          return
  36       end if
  37    end if
  38
  39    do n=1, iv%info(sound)%nlocal
  40        do k=1, iv%info(sound)%levels(n)
  41           jo_grad_y%sound(n)%u(k) = -re%sound(n)%u(k) / (iv%sound(n)%u(k)%error * iv%sound(n)%u(k)%error)
  42           jo_grad_y%sound(n)%v(k) = -re%sound(n)%v(k) / (iv%sound(n)%v(k)%error * iv%sound(n)%v(k)%error)
  43           jo_grad_y%sound(n)%t(k) = -re%sound(n)%t(k) / (iv%sound(n)%t(k)%error * iv%sound(n)%t(k)%error)
  44           jo_grad_y%sound(n)%q(k) = -re%sound(n)%q(k) / (iv%sound(n)%q(k)%error * iv%sound(n)%q(k)%error)
  45       end do
  46    end do
  47
  48    ! Bitwise-exact reduction preserves operation order of serial code for
  49    ! testing, at the cost of much-increased run-time.  Turn it off when not
  50    ! testing.  This will always be .false. for a serial or 1-MPI-process run.
  51    if (test_dm_exact) then
  52       ! collect all obs in serial order and allocate global objects
  53       call da_to_global_sound(iv, re, jo_grad_y, iv_glob, re_glob, jo_grad_y_glob)
  54       ! perform remaining computations
  55       call da_jo_sound_uvtq(iv_glob, re_glob, jo_grad_y_glob, jo)
  56       ! free global objects
  57       call da_deallocate_global_sound(iv_glob, re_glob, jo_grad_y_glob)
  58    else
  59       ! perform remaining computations
  60       call da_jo_sound_uvtq(iv, re, jo_grad_y, jo)
  61    end if
  62
  63    jo % sound_u = 0.5 * jo % sound_u
  64    jo % sound_v = 0.5 * jo % sound_v
  65    jo % sound_t = 0.5 * jo % sound_t
  66    jo % sound_q = 0.5 * jo % sound_q
  67
  68    if (trace_use_dull) call da_trace_exit("da_jo_and_grady_sound")
  69
  70 end subroutine da_jo_and_grady_sound
  71
  72