bundles/org.simantics.spreadsheet.graph/src/org/simantics/spreadsheet/graph/formula/CountifFormulaFunction.java

   1 package org.simantics.spreadsheet.graph.formula;\r
   2 \r
   3 import org.simantics.spreadsheet.graph.CellFormulaFunction;\r
   4 import org.simantics.spreadsheet.graph.CellValueVisitor;\r
   5 import org.simantics.spreadsheet.graph.SpreadsheetGraphUtils;\r
   6 import org.simantics.spreadsheet.graph.SpreadsheetMatrix;\r
   7 import org.simantics.spreadsheet.graph.parser.ast.AstArgList;\r
   8 import org.simantics.spreadsheet.graph.parser.ast.AstRange;\r
   9 \r
  10 public class CountifFormulaFunction implements CellFormulaFunction<Integer> {\r
  11 \r
  12     @Override\r
  13     public Integer evaluate(CellValueVisitor visitor, AstArgList args) {\r
  14         if (args.values.size() != 2) throw new IllegalStateException();\r
  15         \r
  16         AstRange range = (AstRange) args.values.get(0);\r
  17         SpreadsheetMatrix matrix = (SpreadsheetMatrix) range.accept(visitor);\r
  18         Object crit = null;\r
  19         try {\r
  20                 crit = args.values.get(1).accept(visitor);\r
  21         } catch (IllegalStateException e){\r
  22                 return 0;\r
  23         }\r
  24         \r
  25         String criteria = SpreadsheetGraphUtils.asString(crit);\r
  26         int result = 0;\r
  27         for (Object value : matrix.values) {\r
  28             if (SpreadsheetGraphUtils.matchCriteria(value, criteria))\r
  29                 result++;\r
  30         }\r
  31         return Integer.valueOf(result);\r
  32     }\r
  33 }\r