da_setup_cv.inc

1 subroutine da_setup_cv(be)
2 
3    !---------------------------------------------------------------------------
4    ! Purpose: Calculate the size of the 1-dimensional control variable array.        
5    !---------------------------------------------------------------------------
6 
7    implicit none
8 
9    type (be_type), intent(inout) :: be       ! background error.
10 
11    integer                       :: iy, jx   ! Local horizontal domain dimensions.
12    integer                       :: ij       ! Product of horizontal dims.
13 
14    !--------------------------------------------------------------
15    ! 1] Define standard control variable size:
16    !--------------------------------------------------------------
17 
18    if (global) then
19       be % cv % size1c = (be % v1 % max_wave+1) * (be % v1 % max_wave+2)/2
20       be % cv % size2c = (be % v2 % max_wave+1) * (be % v2 % max_wave+2)/2
21       be % cv % size3c = (be % v3 % max_wave+1) * (be % v3 % max_wave+2)/2
22       be % cv % size4c = (be % v4 % max_wave+1) * (be % v4 % max_wave+2)/2
23       be % cv % size5c = (be % v5 % max_wave+1) * (be % v5 % max_wave+2)/2
24 
25       be % cv % size1  = 2 * be % cv % size1c * be % v1 % mz
26       be % cv % size2  = 2 * be % cv % size2c * be % v2 % mz
27       be % cv % size3  = 2 * be % cv % size3c * be % v3 % mz
28       be % cv % size4  = 2 * be % cv % size4c * be % v4 % mz
29       be % cv % size5  = 2 * be % cv % size5c * be % v5 % mz
30    else
31       iy = ite - its + 1
32       jx = jte - jts + 1
33       ij = iy * jx
34       be % cv % size1 = ij * be % v1 % mz
35       be % cv % size2 = ij * be % v2 % mz
36       be % cv % size3 = ij * be % v3 % mz
37       be % cv % size4 = ij * be % v4 % mz
38       be % cv % size5 = ij * be % v5 % mz
39    end if
40 
41    be % cv % size_jb = be % cv % size1 + be % cv % size2 + be % cv % size3 + &
42       be % cv % size4 + be % cv % size5
43 
44    !--------------------------------------------------------------
45    ! [2] Define flow-dependent control variable size:
46    !--------------------------------------------------------------
47 
48    be % cv % size_je = 0
49 
50    if (ensdim_alpha > 0) then
51       if (global) then
52          be % cv % size_alphac = (be % alpha % max_wave + 1) * &
53                                  (be % alpha % max_wave + 2)  / 2
54          be % cv % size_je  = 2 * be % cv % size_alphac * be % alpha % mz
55       else
56          be % cv % size_alphac = ij * be % alpha % mz
57          be % cv % size_je = be % cv % size_alphac
58       end if
59    end if
60 
61    be % cv % size = be % cv % size_jb +  be % cv % size_je
62 
63    !--------------------------------------------------------------
64    ! [3] Define domain-wide cv sizes for bit-repro option:
65    !--------------------------------------------------------------
66 
67    if (.not. global) then
68       iy = ide - ids + 1
69       jx = jde - jds + 1
70       ij = iy * jx
71       cv_size_domain_jb = ij * (be % v1 % mz + be % v2 % mz + be % v3 % mz + &
72                                  be % v4 % mz + be % v5 % mz)
73       cv_size_domain_je = ij * be % alpha % mz 
74    end if
75 
76    cv_size_domain = cv_size_domain_jb + cv_size_domain_je
77 
78 end subroutine da_setup_cv
79 
80