Adopt spreadsheet changes made in Balas development

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

diff --git a/bundles/org.simantics.spreadsheet/src/org/simantics/spreadsheet/solver/formula/CountifFormulaFunction.java b/bundles/org.simantics.spreadsheet/src/org/simantics/spreadsheet/solver/formula/CountifFormulaFunction.java
new file mode 100644 (file)
index 0000000..82a3ac1
--- /dev/null
+++ b/bundles/org.simantics.spreadsheet/src/org/simantics/spreadsheet/solver/formula/CountifFormulaFunction.java
@@ -0,0 +1,31 @@
+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.AstRange;
+
+public class CountifFormulaFunction implements CellFormulaFunction<Integer> {
+
+    @Override
+    public Integer evaluate(CellValueVisitor visitor, AstArgList args) {
+        if (args.values.size() != 2) throw new IllegalStateException();
+
+        AstRange range = (AstRange) args.values.get(0);
+        SpreadsheetMatrix matrix = (SpreadsheetMatrix) range.accept(visitor);
+        Object crit = null;
+        try {
+            crit = args.values.get(1).accept(visitor);
+        } catch (IllegalStateException e){
+            return 0;
+        }
+
+        String criteria = Spreadsheets.asString(crit);
+        int result = 0;
+        for (Object value : matrix.values) {
+            if (Spreadsheets.matchCriteria(value, criteria))
+                result++;
+        }
+        return Integer.valueOf(result);
+    }
+}