修改BOLL算法

wangyiqian · Sep 1, 2022 · 4f2979b · 4f2979b
1 parent 5463f3e
commit 4f2979b
Showing 1 changed file with 15 additions and 10 deletions.
diff --git a/lib/src/main/java/com/github/wangyiqian/stockchart/index/BollCalculator.kt b/lib/src/main/java/com/github/wangyiqian/stockchart/index/BollCalculator.kt
@@ -57,19 +57,24 @@ object BollCalculator: ICalculator {
 
             sum += kEntity.getClosePrice()
 
-            var ma = sum / (pEnd - pFrom + 1)
-            result[mbIdx][kEntityIdx] = ma
-            if(pEnd - pFrom + 1 == n){
+            if (pEnd - pFrom + 1 == n) {
+                val ma = sum / n
+                result[mbIdx][kEntityIdx] = ma
                 sum -= input[pFrom].getClosePrice()
                 pFrom += 1
+
+                var squareSum = 0f
+                for (i in pEnd downTo pEnd - n + 1) {
+                    squareSum += (input[i].getClosePrice() - ma).pow(2)
+                }
+                val std = sqrt(squareSum / n)
+                result[upIdx][kEntityIdx] = ma + k * std
+                result[dnIdx][kEntityIdx] = ma - k * std
+            } else {
+                result[mbIdx][kEntityIdx] = null
+                result[upIdx][kEntityIdx] = null
+                result[dnIdx][kEntityIdx] = null
             }
-            var squareSum = 0f
-            for (i in pEnd downTo pFrom) {
-                squareSum += (input[i].getClosePrice() - ma).pow(2)
-            }
-            val std = sqrt(squareSum / n)
-            result[upIdx][kEntityIdx] = ma + k * std
-            result[dnIdx][kEntityIdx] = ma - k * std
         }
         return result
     }