central_moment: n and mean computed outside select case

Author: jvdp1

src/stdlib_experimental_stats_moment.fypp

@@ -82,12 +82,13 @@ contains
           return
         end if
 
+        n = size(x, dim)
+        mean = sum(x, dim) / n
+
         res = 0
         select case(dim)
           #:for fi in range(1, rank+1)
           case(${fi}$)
-            n = size(x, dim)
-            mean = sum(x, dim) / n
             do i = 1, size(x, dim)
               res = res + (x${select_subarray(rank, [(fi, 'i')])}$ - mean)**order
             end do
@@ -121,12 +122,13 @@ contains
           return
         end if
 
+        n = size(x, dim)
+        mean = sum(real(x, dp), dim) / n
+
         res = 0
         select case(dim)
           #:for fi in range(1, rank+1)
           case(${fi}$)
-            n = size(x, dim)
-            mean = sum(real(x, dp), dim) / n
             do i = 1, size(x, dim)
               res = res + (real(x${select_subarray(rank, [(fi, 'i')])}$, dp) - mean)**order
             end do
@@ -198,12 +200,13 @@ contains
         real(${k1}$) :: n${reduced_shape('x', rank, 'dim')}$
         ${t1}$ :: mean${reduced_shape('x', rank, 'dim')}$
 
+        n = count(mask, dim)
+        mean = sum(x, dim, mask) / n
+
         res = 0
         select case(dim)
           #:for fi in range(1, rank+1)
           case(${fi}$)
-            n = count(mask, dim)
-            mean = sum(x, dim, mask) / n
             do i = 1, size(x, dim)
               res = res + merge( (x${select_subarray(rank, [(fi, 'i')])}$ - mean)**order,&
                 #:if t1[0] == 'r'
@@ -238,12 +241,13 @@ contains
         real(dp) :: n${reduced_shape('x', rank, 'dim')}$
         real(dp) :: mean${reduced_shape('x', rank, 'dim')}$
 
+        n = count(mask, dim)
+        mean = sum(real(x, dp), dim, mask) / n
+
         res = 0
         select case(dim)
           #:for fi in range(1, rank+1)
           case(${fi}$)
-            n = count(mask, dim)
-            mean = sum(real(x, dp), dim, mask) / n
             do i = 1, size(x, dim)
               res = res + merge((real(x${select_subarray(rank, [(fi, 'i')])}$, dp) - mean)**order,&
                                   0._dp, mask${select_subarray(rank, [(fi, 'i')])}$)