]> source.dussan.org Git - poi.git/commitdiff
add DSTDEV function support
authorPJ Fanning <fanningpj@apache.org>
Wed, 25 May 2022 20:12:17 +0000 (20:12 +0000)
committerPJ Fanning <fanningpj@apache.org>
Wed, 25 May 2022 20:12:17 +0000 (20:12 +0000)
git-svn-id: https://svn.apache.org/repos/asf/poi/trunk@1901258 13f79535-47bb-0310-9956-ffa450edef68

poi/src/main/java/org/apache/poi/ss/formula/eval/FunctionEval.java
poi/src/main/java/org/apache/poi/ss/formula/functions/DAverage.java
poi/src/main/java/org/apache/poi/ss/formula/functions/DCount.java
poi/src/main/java/org/apache/poi/ss/formula/functions/DStarRunner.java
poi/src/main/java/org/apache/poi/ss/formula/functions/DStdev.java [new file with mode: 0644]
poi/src/test/java/org/apache/poi/ss/formula/functions/TestDStdev.java [new file with mode: 0644]

index 2474eb8ec3c034e95f0bfdda709a14edfa369b8c..9c6a63dad5fd3a796cf6b8ee2684f63435c136f0 100644 (file)
@@ -110,7 +110,7 @@ public final class FunctionEval {
         retval[42] = new DStarRunner(DStarRunner.DStarAlgorithmEnum.DAVERAGE);
         retval[43] = new DStarRunner(DStarRunner.DStarAlgorithmEnum.DMIN);
         retval[44] = new DStarRunner(DStarRunner.DStarAlgorithmEnum.DMAX);
-        // 45: DSTDEV
+        retval[45] = new DStarRunner(DStarRunner.DStarAlgorithmEnum.DSTDEV);
         retval[46] = AggregateFunction.VAR;
         // 47: DVAR
         retval[48] = TextFunction.TEXT;
index a524ddbc8a1f809bf21fe2a08a2dca519017e863..2d4dea2cc253b18591d48d2cddd399165bf8d859 100644 (file)
@@ -29,7 +29,7 @@ import java.math.MathContext;
  * Gets the average value of a column in an area with given conditions.
  */
 public final class DAverage implements IDStarAlgorithm {
-    private int count;
+    private long count;
     private double total;
 
     @Override
@@ -43,10 +43,14 @@ public final class DAverage implements IDStarAlgorithm {
 
     @Override
     public ValueEval getResult() {
-        return count == 0 ? NumberEval.ZERO : new NumberEval(getAverage());
+        return count == 0 ? NumberEval.ZERO : new NumberEval(divide());
     }
 
-    private double getAverage() {
+    private double divide() {
+        return divide(total, count);
+    }
+
+    private static double divide(final double total, final long count) {
         return BigDecimal.valueOf(total)
                 .divide(BigDecimal.valueOf(count), MathContext.DECIMAL128)
                 .doubleValue();
index 0fda8fa32c37d04478bfc66d06c0a9bd197ac162..c2ea0ebeb8a907cade964d27092cc4a9ef5ffa25 100644 (file)
@@ -26,7 +26,7 @@ import org.apache.poi.ss.formula.eval.ValueEval;
  * Counts the value of a column in an area with given conditions.
  */
 public final class DCount implements IDStarAlgorithm {
-    private int count;
+    private long count;
 
     @Override
     public boolean processMatch(ValueEval eval) {
index df96bdb4e9e91c670372a2f7c884499be5d2ea5f..32be5ce26776890a8fe761ce03a586b13c856ca0 100644 (file)
@@ -62,6 +62,8 @@ public final class DStarRunner implements Function3Arg {
         DCOUNT(DCount::new),
         /** @see DAverage */
         DAVERAGE(DAverage::new),
+        /** @see DStdev */
+        DSTDEV(DStdev::new),
         ;
 
         private final Supplier<IDStarAlgorithm> implSupplier;
diff --git a/poi/src/main/java/org/apache/poi/ss/formula/functions/DStdev.java b/poi/src/main/java/org/apache/poi/ss/formula/functions/DStdev.java
new file mode 100644 (file)
index 0000000..97bd2b3
--- /dev/null
@@ -0,0 +1,54 @@
+/* ====================================================================
+   Licensed to the Apache Software Foundation (ASF) under one or more
+   contributor license agreements.  See the NOTICE file distributed with
+   this work for additional information regarding copyright ownership.
+   The ASF licenses this file to You under the Apache License, Version 2.0
+   (the "License"); you may not use this file except in compliance with
+   the License.  You may obtain a copy of the License at
+
+       http://www.apache.org/licenses/LICENSE-2.0
+
+   Unless required by applicable law or agreed to in writing, software
+   distributed under the License is distributed on an "AS IS" BASIS,
+   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+   See the License for the specific language governing permissions and
+   limitations under the License.
+==================================================================== */
+
+package org.apache.poi.ss.formula.functions;
+
+import org.apache.poi.ss.formula.eval.NumberEval;
+import org.apache.poi.ss.formula.eval.NumericValueEval;
+import org.apache.poi.ss.formula.eval.ValueEval;
+import org.apache.poi.ss.util.NumberToTextConverter;
+
+import java.math.BigDecimal;
+import java.util.ArrayList;
+
+/**
+ * Implementation of the DStdev function:
+ * Gets the standard deviation value of a column in an area with given conditions.
+ */
+public final class DStdev implements IDStarAlgorithm {
+    private final ArrayList<Double> values = new ArrayList<>();
+
+    @Override
+    public boolean processMatch(ValueEval eval) {
+        if (eval instanceof NumericValueEval) {
+            final double val = ((NumericValueEval)eval).getNumberValue();
+            values.add(val);
+        }
+        return true;
+    }
+
+    @Override
+    public ValueEval getResult() {
+        final double[] array = new double[values.size()];
+        int pos = 0;
+        for (Double d : values) {
+            array[pos++] = d;
+        }
+        final double stdev = StatsLib.stdev(array);
+        return new NumberEval(new BigDecimal(NumberToTextConverter.toText(stdev)).doubleValue());
+    }
+}
diff --git a/poi/src/test/java/org/apache/poi/ss/formula/functions/TestDStdev.java b/poi/src/test/java/org/apache/poi/ss/formula/functions/TestDStdev.java
new file mode 100644 (file)
index 0000000..48393dc
--- /dev/null
@@ -0,0 +1,62 @@
+
+/* ====================================================================
+   Licensed to the Apache Software Foundation (ASF) under one or more
+   contributor license agreements.  See the NOTICE file distributed with
+   this work for additional information regarding copyright ownership.
+   The ASF licenses this file to You under the Apache License, Version 2.0
+   (the "License"); you may not use this file except in compliance with
+   the License.  You may obtain a copy of the License at
+
+       http://www.apache.org/licenses/LICENSE-2.0
+
+   Unless required by applicable law or agreed to in writing, software
+   distributed under the License is distributed on an "AS IS" BASIS,
+   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+   See the License for the specific language governing permissions and
+   limitations under the License.
+==================================================================== */
+package org.apache.poi.ss.formula.functions;
+
+import org.apache.poi.hssf.usermodel.HSSFCell;
+import org.apache.poi.hssf.usermodel.HSSFFormulaEvaluator;
+import org.apache.poi.hssf.usermodel.HSSFSheet;
+import org.apache.poi.hssf.usermodel.HSSFWorkbook;
+import org.junit.jupiter.api.Test;
+
+import java.io.IOException;
+
+import static org.apache.poi.ss.util.Utils.addRow;
+import static org.apache.poi.ss.util.Utils.assertDouble;
+
+/**
+ * Testcase for function DSTDEV()
+ */
+public class TestDStdev {
+
+    //https://support.microsoft.com/en-us/office/dstdev-function-026b8c73-616d-4b5e-b072-241871c4ab96
+    @Test
+    void testMicrosoftExample1() throws IOException {
+        try (HSSFWorkbook wb = initWorkbook1()) {
+            HSSFFormulaEvaluator fe = new HSSFFormulaEvaluator(wb);
+            HSSFCell cell = wb.getSheetAt(0).getRow(0).createCell(12);
+            assertDouble(fe, cell, "DSTDEV(A5:E11, \"Yield\", A1:A3)", 2.96647939483827, 0.0000000001);
+        }
+    }
+
+    private HSSFWorkbook initWorkbook1() {
+        HSSFWorkbook wb = new HSSFWorkbook();
+        HSSFSheet sheet = wb.createSheet();
+        addRow(sheet, 0, "Tree", "Height", "Age", "Yield", "Profit", "Height");
+        addRow(sheet, 1, "=Apple", ">10", null, null, null, "<16");
+        addRow(sheet, 2, "=Pear");
+        addRow(sheet, 3);
+        addRow(sheet, 4, "Tree", "Height", "Age", "Yield", "Profit");
+        addRow(sheet, 5, "Apple", 18, 20, 14, 105);
+        addRow(sheet, 6, "Pear", 12, 12, 10, 96);
+        addRow(sheet, 7, "Cherry", 13, 14, 9, 105);
+        addRow(sheet, 8, "Apple", 14, 15, 10, 75);
+        addRow(sheet, 9, "Pear", 9, 8, 8, 77);
+        addRow(sheet, 10, "Apple", 8, 9, 6, 45);
+        return wb;
+    }
+}