Adopt spreadsheet changes made in Balas development

[simantics/platform.git] / bundles / org.simantics.spreadsheet / src / org / simantics / spreadsheet / solver / formula / GeomeanFormulaFunction.java

diff --git a/bundles/org.simantics.spreadsheet/src/org/simantics/spreadsheet/solver/formula/GeomeanFormulaFunction.java b/bundles/org.simantics.spreadsheet/src/org/simantics/spreadsheet/solver/formula/GeomeanFormulaFunction.java
new file mode 100644 (file)
index 0000000..25f3f58
--- /dev/null
+++ b/bundles/org.simantics.spreadsheet/src/org/simantics/spreadsheet/solver/formula/GeomeanFormulaFunction.java
@@ -0,0 +1,45 @@
+package org.simantics.spreadsheet.solver.formula;
+
+import org.simantics.spreadsheet.SpreadsheetMatrix;
+import org.simantics.spreadsheet.Spreadsheets;
+import org.simantics.spreadsheet.solver.formula.parser.ast.AstArgList;
+import org.simantics.spreadsheet.solver.formula.parser.ast.AstValue;
+
+public class GeomeanFormulaFunction implements CellFormulaFunction<Object> {
+
+    @Override
+    public Object evaluate(CellValueVisitor visitor, AstArgList args) {
+        if (args.values.size() == 0)
+            throw new IllegalStateException();
+        double result = 1.0;
+        double count = 0.0;
+        for (AstValue value : args.values){
+            Object r = value.accept(visitor);
+            if (r instanceof SpreadsheetMatrix) {
+                Object v = ((SpreadsheetMatrix) r).productWithFormulaError();
+                if(v instanceof String) return v;
+                double dval = ((Number)v).doubleValue();
+                if(dval!=0.0){
+                    result = result*dval;
+                    count += ((SpreadsheetMatrix) r).countOfActualDoubleValues();
+                }
+            } else {
+                FormulaError2 error = FormulaError2.forObject(r);
+                if(error!=null) return error.getString();
+
+                Number vNum = Spreadsheets.asValidNumber(r);
+                Double v = null;
+                if(vNum!=null) v = vNum.doubleValue();
+
+                if(v!=null){
+                    if(v<=0) return FormulaError2.NUM.getString();
+                    result = result*v;
+                    count++;
+                }
+            }
+        }
+        if(result==0.0 || count==0.0) return FormulaError2.NUM.getString();
+        return Double.valueOf(Math.pow(result, (1.0/count)));
+    }
+
+}