X-Git-Url: http://git.megacz.com/?p=sbp.git;a=blobdiff_plain;f=src%2Fedu%2Fberkeley%2Fsbp%2FUnion.java;h=1843a4620cb5dd2ca96b92636ce4c332883230e9;hp=5ff00d669a165256cd7c8835f0a698c02a756624;hb=b9228a7c1bd98e514e6a5c0483c2e539ee777992;hpb=62be1c1ae2ac13086b508e34eeffdb7e536d5b61

diff --git a/src/edu/berkeley/sbp/Union.java b/src/edu/berkeley/sbp/Union.java
index 5ff00d6..1843a46 100644
--- a/src/edu/berkeley/sbp/Union.java
+++ b/src/edu/berkeley/sbp/Union.java
@@ -19,13 +19,14 @@ import java.lang.ref.*;
  */
 public class Union extends Element implements Iterable<Sequence> {
 
-    /*private*/ final String name;
-    /*private*/ final boolean synthetic;
+    private final String  name;
+    private final boolean synthetic;
     private boolean viewed = false;
 
     private final List<Sequence> alternatives = new ArrayList<Sequence>();
 
     public Union(String name) { this(name, false); }
+    public Union(String name, Sequence s) { this(name, false); add(s); }
 
     /**
      *  Since every cycle in a non-degenerate grammar contains at
@@ -53,14 +54,17 @@ public class Union extends Element implements Iterable<Sequence> {
     }
 
     /** adds an alternative */
-    public void add(Sequence s) {
-        /*
-          FIXME
+    public Union add(Sequence s) {
         if (viewed)
-            throw new RuntimeException("attempt to add a Sequence to a Union that has already been examined:\n  "+this);
-        */
-        if (alternatives.contains(s)) return;
+            throw new RuntimeException("once Union.contains() or Union.iterator() has been invoked, "+
+                                       "you may not add any more Sequences to it\n  "+
+                                       "  union in question: " + this);
+        if (s.needed_or_hated)
+            throw new RuntimeException("you may not add a conjunct directly to a Union");
+        s.in_a_union = true;
+        if (alternatives.contains(s)) return this;
         alternatives.add(s);
+        return this;
     }
 
     /** adds a one-element sequence */
@@ -71,28 +75,28 @@ public class Union extends Element implements Iterable<Sequence> {
 
     // Epsilon Form //////////////////////////////////////////////////////////////////////////////
 
-    // FIXME
-    //private Forest.Many epsilonForm = null;
+    private Forest.Many epsilonForm = null;
     Forest epsilonForm(Input.Region loc) {
-        //if (epsilonForm != null) return epsilonForm;
+        // FIXME: this is pretty ugly...
+        viewed = true;
+        if (epsilonForm != null) return epsilonForm;
         Forest.Many epsilonForm = new Forest.Many();
-        for(Sequence s : this) {
-            // FIXME FIXME FIXME
-            if (new Walk.Cache().possiblyEpsilon(s))
+        Cache cache = new Cache(null, null);
+        for(Sequence s : this)
+            if (cache.possiblyEpsilon(s))
                 epsilonForm.merge(s.epsilonForm(loc));
-        }
         return epsilonForm;
     }
 
 
     // Display //////////////////////////////////////////////////////////////////////////////
 
-    public String getName() {
-        if (name != null) return name;
-        return "(anon_union)";
-    }
+    boolean isSynthetic() { return synthetic; }
+    String getName()      { return name==null ? "(anon_union)" : name; }
+
     public String toString() {
-        viewed = true;
+        // technically this should be turned on, but we don't make a big deal
+        //viewed = true;
         if (name != null) return name;
         StringBuffer sb = new StringBuffer();
         sb.append("(");
@@ -103,7 +107,8 @@ public class Union extends Element implements Iterable<Sequence> {
 
     /** display this union in long/expanded form */
     public StringBuffer toString(StringBuffer sb) {
-        viewed = true;
+        // technically this should be turned on, but we don't make a big deal
+        //viewed = true;
         if (synthetic) return sb;
         boolean first = true;
         String before = StringUtil.pad(15, getName()) + " = ";