aboutsummaryrefslogtreecommitdiff
path: root/src/include/upwind_first.h
blob: 6c99d2980c71db1b48e9c5b5001e3fd6c32d4d2e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
! Calculate first order upwinded differences.
! $Header$

idx = one / cctk_delta_space(1)
idy = one / cctk_delta_space(2)
idz = one / cctk_delta_space(3)

do k = 1, nz
  do j = 1, ny
    do i = 1, nx
      if ( eh_mask(i,j,k) .ge. 0 ) then
        if ( ( .not. btest(eh_mask(i,j,k),0) ) .and. &
             ( .not. btest(eh_mask(i,j,k),1) ) ) then
          al = idx * ( f(i,j,k) - f(i-1,j,k) )
          ar = idx * ( f(i+1,j,k) - f(i,j,k) )
        else if ( btest(eh_mask(i,j,k),0) ) then
          al = zero
          ar = idx * ( f(i+1,j,k) - f(i,j,k) )
        else
          al = idx * ( f(i,j,k) - f(i-1,j,k) )
          ar = zero
        end if
        if ( ( .not. btest(eh_mask(i,j,k),2) ) .and. &
             ( .not. btest(eh_mask(i,j,k),3) ) ) then
          bl = idy * ( f(i,j,k) - f(i,j-1,k) )
          br = idy * ( f(i,j+1,k) - f(i,j,k) )
        else if ( btest(eh_mask(i,j,k),2) ) then
          bl = zero
          br = idy * ( f(i,j+1,k) - f(i,j,k) )
        else
          bl = idy * ( f(i,j,k) - f(i,j-1,k) )
          br = zero
        end if
        if ( ( .not. btest(eh_mask(i,j,k),4) ) .and. &
             ( .not. btest(eh_mask(i,j,k),5) ) ) then
          cl = idz * ( f(i,j,k) - f(i,j,k-1) )
          cr = idz * ( f(i,j,k+1) - f(i,j,k) )
        else if ( btest(eh_mask(i,j,k),4) ) then
          cl = zero
          cr = idz * ( f(i,j,k+1) - f(i,j,k) )
        else
          cl = idz * ( f(i,j,k) - f(i,j,k-1) )
          cr = zero
        end if

        alminus = half*(al-abs(al))
        alplus = half*(al+abs(al))
        arminus = half*(ar-abs(ar))
        arplus = half*(ar+abs(ar))

        blminus = half*(bl-abs(bl))
        blplus = half*(bl+abs(bl))
        brminus = half*(br-abs(br))
        brplus = half*(br+abs(br))

        clminus = half*(cl-abs(cl))
        clplus = half*(cl+abs(cl))
        crminus = half*(cr-abs(cr))
        crplus = half*(cr+abs(cr))

        if ( f(i,j,k) .gt. 0 ) then
          if ( abs(alplus) .gt. abs(arminus) ) then
            dfx(i,j,k) = alplus
          else
            dfx(i,j,k) = arminus
          end if
          if ( abs(blplus) .gt. abs(brminus) ) then
            dfy(i,j,k) = blplus
          else
            dfy(i,j,k) = brminus
          end if
          if ( abs(clplus) .gt. abs(crminus) ) then
            dfz(i,j,k) = clplus
          else
            dfz(i,j,k) = crminus
          end if
        endif

        if ( f(i,j,k) .lt. 0 ) then
          if ( abs(alminus) .gt. abs(arplus) ) then
            dfx(i,j,k) = alminus
          else
            dfx(i,j,k) = arplus
          end if
          if ( abs(blminus) .gt. abs(brplus) ) then
            dfy(i,j,k) = blminus
          else
            dfy(i,j,k) = brplus
          end if
          if ( abs(clminus) .gt. abs(crplus) ) then
            dfz(i,j,k) = clminus
          else
            dfz(i,j,k) = crplus
          end if
        end if
      else
        dfx(i,j,k) = zero
        dfy(i,j,k) = zero
        dfz(i,j,k) = zero
      end if
    end do
  end do
end do