1
1
#:include "common.fypp"
2
2
#:set RANKS = range(1, MAXRANK + 1)
3
- #:set RC_KINDS_TYPES = REAL_KINDS_TYPES + CMPLX_KINDS_TYPES
4
3
submodule (stdlib_experimental_stats) stdlib_experimental_stats_var
5
4
6
5
use, intrinsic:: ieee_arithmetic, only: ieee_value, ieee_quiet_nan
@@ -10,7 +9,7 @@ submodule (stdlib_experimental_stats) stdlib_experimental_stats_var
10
9
11
10
contains
12
11
13
- #:for k1, t1 in RC_KINDS_TYPES
12
+ #:for k1, t1 in REAL_KINDS_TYPES
14
13
#:for rank in RANKS
15
14
#:set RName = rname("var_all",rank, t1, k1)
16
15
module function ${RName}$(x, mask) result(res)
@@ -60,7 +59,7 @@ contains
60
59
#:endfor
61
60
62
61
63
- #:for k1, t1 in RC_KINDS_TYPES
62
+ #:for k1, t1 in REAL_KINDS_TYPES
64
63
#:for rank in RANKS
65
64
#:set RName = rname("var",rank, t1, k1)
66
65
module function ${RName}$(x, dim, mask) result(res)
@@ -136,7 +135,7 @@ contains
136
135
#:endfor
137
136
138
137
139
- #:for k1, t1 in RC_KINDS_TYPES
138
+ #:for k1, t1 in REAL_KINDS_TYPES
140
139
#:for rank in RANKS
141
140
#:set RName = rname("var_mask_all",rank, t1, k1)
142
141
module function ${RName}$(x, mask) result(res)
@@ -176,7 +175,7 @@ contains
176
175
#:endfor
177
176
178
177
179
- #:for k1, t1 in RC_KINDS_TYPES
178
+ #:for k1, t1 in REAL_KINDS_TYPES
180
179
#:for rank in RANKS
181
180
#:set RName = rname("var_mask",rank, t1, k1)
182
181
module function ${RName}$(x, dim, mask) result(res)
@@ -197,11 +196,7 @@ contains
197
196
mean = sum(x, dim, mask) / n
198
197
do i = 1, size(x, dim)
199
198
res = res + merge( (x${rankindice(':', 'i', rank, fi )}$ - mean)**2,&
200
- #:if t1[0] == 'r'
201
199
0._${k1}$,&
202
- #:else
203
- cmplx(0._${k1}$, 0._${k1}$, ${k1}$),&
204
- #:endif
205
200
mask${rankindice(':', 'i', rank, fi)}$)
206
201
end do
207
202
#:endfor
0 commit comments