11#:include "common.fypp"
22#:set RANKS = range(1, MAXRANK + 1)
3- #:set RC_KINDS_TYPES = REAL_KINDS_TYPES + CMPLX_KINDS_TYPES
43submodule (stdlib_experimental_stats) stdlib_experimental_stats_var
54
65 use, intrinsic:: ieee_arithmetic, only: ieee_value, ieee_quiet_nan
@@ -10,7 +9,7 @@ submodule (stdlib_experimental_stats) stdlib_experimental_stats_var
109
1110contains
1211
13- #:for k1, t1 in RC_KINDS_TYPES
12+ #:for k1, t1 in REAL_KINDS_TYPES
1413 #:for rank in RANKS
1514 #:set RName = rname("var_all",rank, t1, k1)
1615 module function ${RName}$(x, mask) result(res)
@@ -60,7 +59,7 @@ contains
6059 #:endfor
6160
6261
63- #:for k1, t1 in RC_KINDS_TYPES
62+ #:for k1, t1 in REAL_KINDS_TYPES
6463 #:for rank in RANKS
6564 #:set RName = rname("var",rank, t1, k1)
6665 module function ${RName}$(x, dim, mask) result(res)
@@ -136,7 +135,7 @@ contains
136135 #:endfor
137136
138137
139- #:for k1, t1 in RC_KINDS_TYPES
138+ #:for k1, t1 in REAL_KINDS_TYPES
140139 #:for rank in RANKS
141140 #:set RName = rname("var_mask_all",rank, t1, k1)
142141 module function ${RName}$(x, mask) result(res)
@@ -176,7 +175,7 @@ contains
176175 #:endfor
177176
178177
179- #:for k1, t1 in RC_KINDS_TYPES
178+ #:for k1, t1 in REAL_KINDS_TYPES
180179 #:for rank in RANKS
181180 #:set RName = rname("var_mask",rank, t1, k1)
182181 module function ${RName}$(x, dim, mask) result(res)
@@ -197,11 +196,7 @@ contains
197196 mean = sum(x, dim, mask) / n
198197 do i = 1, size(x, dim)
199198 res = res + merge( (x${rankindice(':', 'i', rank, fi )}$ - mean)**2,&
200- #:if t1[0] == 'r'
201199 0._${k1}$,&
202- #:else
203- cmplx(0._${k1}$, 0._${k1}$, ${k1}$),&
204- #:endif
205200 mask${rankindice(':', 'i', rank, fi)}$)
206201 end do
207202 #:endfor
0 commit comments