From 1ff98e82bcbca64fcc20a9d63dfee4dfc2b3a3fa Mon Sep 17 00:00:00 2001
From: bdemsky <bdemsky>
Date: Thu, 31 Aug 2006 00:59:31 +0000
Subject: [PATCH] add constraint check nodes to flat representation

---
 Robust/src/IR/Flat/BuildFlat.java       | 16 +++++++++++++++-
 Robust/src/IR/Flat/FKind.java           |  1 +
 Robust/src/IR/Flat/FlatCheckNode.java   | 19 +++++++++++++++++++
 Robust/src/IR/Tree/ConstraintCheck.java |  9 +++++++++
 Robust/src/IR/Tree/SemanticCheck.java   | 22 ++++++++++++++++++++++
 5 files changed, 66 insertions(+), 1 deletion(-)
 create mode 100644 Robust/src/IR/Flat/FlatCheckNode.java

diff --git a/Robust/src/IR/Flat/BuildFlat.java b/Robust/src/IR/Flat/BuildFlat.java
index 1b39c025..1828f9dd 100644
--- a/Robust/src/IR/Flat/BuildFlat.java
+++ b/Robust/src/IR/Flat/BuildFlat.java
@@ -735,7 +735,21 @@ public class BuildFlat {
     private NodePair flattenTaskExitNode(TaskExitNode ten) {
 	FlatFlagActionNode ffan=new FlatFlagActionNode(true);
 	updateFlagActionNode(ffan, ten.getFlagEffects());
-	return new NodePair(ffan, ffan);
+	NodePair fcn=flattenConstraintCheck(ten.getChecks());
+	ffan.addNext(fcn.getBegin());
+	return new NodePair(ffan, fcn.getEnd());
+    }
+
+    private NodePair flattenConstraintCheck(Vector ccs) {
+	FlatNode begin=new FlatNop();
+	FlatNode last=begin;
+	for(int i=0;i<ccs.size();i++) {
+	    ConstraintCheck cc=(ConstraintCheck) ccs.get(i);
+	    FlatCheckNode fcn=new FlatCheckNode(getTempforVar(cc.getVar()), cc.getSpec());
+	    last.addNext(fcn);
+	    last=fcn;
+	}
+	return new NodePair(begin,last);
     }
 
     private NodePair flattenSubBlockNode(SubBlockNode sbn) {
diff --git a/Robust/src/IR/Flat/FKind.java b/Robust/src/IR/Flat/FKind.java
index 304d7eed..a348540a 100644
--- a/Robust/src/IR/Flat/FKind.java
+++ b/Robust/src/IR/Flat/FKind.java
@@ -14,4 +14,5 @@ public class FKind {
     public static final int FlatSetElementNode=11;
     public static final int FlatElementNode=12;
     public static final int FlatFlagActionNode=13;
+    public static final int FlatCheckNode=14;
 }
diff --git a/Robust/src/IR/Flat/FlatCheckNode.java b/Robust/src/IR/Flat/FlatCheckNode.java
new file mode 100644
index 00000000..146890c1
--- /dev/null
+++ b/Robust/src/IR/Flat/FlatCheckNode.java
@@ -0,0 +1,19 @@
+package IR.Flat;
+
+public class FlatCheckNode extends FlatNode {
+    TempDescriptor td;
+    String spec;
+
+    public FlatCheckNode(TempDescriptor td, String spec) {
+	this.td=td;
+	this.spec=spec;
+    }
+
+    public int kind() {
+        return FKind.FlatCheckNode;
+    }
+    
+    public TempDescriptor [] readsTemps() {
+	return new TempDescriptor[] {td};
+    }
+}
diff --git a/Robust/src/IR/Tree/ConstraintCheck.java b/Robust/src/IR/Tree/ConstraintCheck.java
index 6af53608..3560e68f 100644
--- a/Robust/src/IR/Tree/ConstraintCheck.java
+++ b/Robust/src/IR/Tree/ConstraintCheck.java
@@ -5,12 +5,21 @@ import IR.*;
 public class ConstraintCheck {
     String varname;
     String specname;
+    VarDescriptor vd;
 
     public ConstraintCheck(String varname, String specname) {
 	this.varname=varname;
 	this.specname=specname;
     }
 
+    public void setVar(VarDescriptor vd) {
+	this.vd=vd;
+    }
+
+    public VarDescriptor getVar() {
+	return vd;
+    }
+
     public String getVarName() {
 	return varname;
     }
diff --git a/Robust/src/IR/Tree/SemanticCheck.java b/Robust/src/IR/Tree/SemanticCheck.java
index 04b115b8..a4abfbd4 100644
--- a/Robust/src/IR/Tree/SemanticCheck.java
+++ b/Robust/src/IR/Tree/SemanticCheck.java
@@ -78,6 +78,27 @@ public class SemanticCheck {
 	checkTypeDescriptor(fd.getType());
     }
 
+    public void checkConstraintCheck(TaskDescriptor td, Vector ccs) {
+	if (ccs==null)
+	    return; /* No constraint checks to check */
+	for(int i=0;i<ccs.size();i++) {
+	    ConstraintCheck cc=(ConstraintCheck) ccs.get(i);
+	    String varname=cc.getVarName();
+
+	    //Make sure the variable is declared as a parameter to the task
+	    VarDescriptor vd=(VarDescriptor)td.getParameterTable().get(varname);
+	    if (vd==null)
+		throw new Error("Parameter "+varname+" in Constraint Check not declared");
+	    cc.setVar(vd);
+
+	    //Make sure it correspods to a class
+	    TypeDescriptor type_d=vd.getType();
+	    if (!type_d.isClass())
+		throw new Error("Cannot have non-object argument for flag_effect");
+
+	}
+    }
+
     public void checkFlagEffects(TaskDescriptor td, Vector vfe) {
 	if (vfe==null)
 	    return; /* No flag effects to check */
@@ -257,6 +278,7 @@ public class SemanticCheck {
 	if (md instanceof MethodDescriptor)
 	    throw new Error("Illegal taskexit appears in Method: "+md.getSymbol());
 	checkFlagEffects((TaskDescriptor)md, ten.getFlagEffects());
+	checkConstraintCheck((TaskDescriptor) md, ten.getChecks());
     }
 
     void checkIfStatementNode(Descriptor md, SymbolTable nametable, IfStatementNode isn) {
-- 
2.34.1