X-Git-Url: https://gerrit.simantics.org/r/gitweb?a=blobdiff_plain;f=bundles%2Forg.simantics.spreadsheet.graph%2Fsrc%2Forg%2Fsimantics%2Fspreadsheet%2Fgraph%2Fformula%2FIfFormulaFunction.java;fp=bundles%2Forg.simantics.spreadsheet.graph%2Fsrc%2Forg%2Fsimantics%2Fspreadsheet%2Fgraph%2Fformula%2FIfFormulaFunction.java;h=c770f34bfff20d397c38642252ac8e543d65132e;hb=969bd23cab98a79ca9101af33334000879fb60c5;hp=0000000000000000000000000000000000000000;hpb=866dba5cd5a3929bbeae85991796acb212338a08;p=simantics%2Fplatform.git

diff --git a/bundles/org.simantics.spreadsheet.graph/src/org/simantics/spreadsheet/graph/formula/IfFormulaFunction.java b/bundles/org.simantics.spreadsheet.graph/src/org/simantics/spreadsheet/graph/formula/IfFormulaFunction.java
new file mode 100644
index 000000000..c770f34bf
--- /dev/null
+++ b/bundles/org.simantics.spreadsheet.graph/src/org/simantics/spreadsheet/graph/formula/IfFormulaFunction.java
@@ -0,0 +1,42 @@
+package org.simantics.spreadsheet.graph.formula;
+
+import org.simantics.spreadsheet.graph.CellFormulaFunction;
+import org.simantics.spreadsheet.graph.CellValueVisitor;
+import org.simantics.spreadsheet.graph.SpreadsheetGraphUtils;
+import org.simantics.spreadsheet.graph.parser.ast.AstArgList;
+import org.simantics.spreadsheet.graph.parser.ast.AstNothing;
+import org.simantics.spreadsheet.graph.parser.ast.AstValue;
+
+public class IfFormulaFunction implements CellFormulaFunction<Object> {
+
+    @Override
+    public Object evaluate(CellValueVisitor visitor, AstArgList args) {
+        if (args.values.size() != 3) throw new IllegalStateException();
+        
+        Object condition = args.values.get(0).accept(visitor);
+        AstValue ifTrueResult = args.values.get(1);
+        AstValue ifFalseResult = args.values.get(2);
+        
+        FormulaError2 error = FormulaError2.forObject(condition);
+		if(error!=null) return error.getString();
+
+        if (SpreadsheetGraphUtils.asBoolean(condition)) {
+            if(ifTrueResult==null || ifTrueResult instanceof AstNothing)
+            	return 0;
+            try {
+            	return ifTrueResult.accept(visitor);
+            } catch (IllegalStateException e){
+            	return FormulaError2.NAME.getString();
+            }
+        }
+        else {
+            if(ifFalseResult==null || ifFalseResult instanceof AstNothing)
+            	return 0;
+            try {
+            	return ifFalseResult.accept(visitor);
+            } catch (IllegalStateException e){
+            	return FormulaError2.NAME.getString();
+            }
+        }
+    }
+}
\ No newline at end of file