1 // Copyright 2006-2007 all rights reserved; see LICENSE file for BSD-style license
3 package edu.berkeley.sbp;
4 import edu.berkeley.sbp.util.*;
5 import edu.berkeley.sbp.*;
6 import edu.berkeley.sbp.*;
9 import java.lang.reflect.*;
10 import java.lang.ref.*;
12 /** <font color=green>juxtaposition; zero or more adjacent Elements; can specify a rewriting</font> */
13 public abstract class Sequence implements Iterable<Element>, SequenceOrElement {
15 protected final Element[] elements;
17 boolean needed_or_hated = false;
18 boolean in_a_union = false;
20 final HashSet<Sequence> needs = new HashSet<Sequence>();
21 final HashSet<Sequence> hates = new HashSet<Sequence>();
23 // FIXME: these are ugly -- migrate into Grammar
24 HashMap<Sequence,Boolean> canNeed = new HashMap<Sequence,Boolean>();
25 HashMap<Sequence,Boolean> canKill = new HashMap<Sequence,Boolean>();
27 final Position firstp;
31 private static int global_sernum = 0;
32 private int sernum = global_sernum++;
34 int[] ret = new int[needs.size()];
36 for(Sequence s : needs) ret[i++] = s.sernum;
40 int[] ret = new int[hates.size()];
42 for(Sequence s : hates) ret[i++] = s.sernum;
47 // Static Constructors //////////////////////////////////////////////////////////////////////////////
49 /** create a sequence of one element */
50 public static Sequence create(Element e) { return create(new Element[] { e }, 0); }
52 /** create a sequence which drops the result of all but one of its element */
53 public static Sequence create(Element[] e, int which) {
54 return new Singleton(e, which); }
56 /** create a sequence which always evaluates to a constant result */
57 public static Sequence create(Object result, Element[] e) {
58 return new RewritingSequence(result, e, trues(e.length)); }
60 private static boolean[] trues(int length) {
61 boolean[] ret = new boolean[length];
62 for(int i=0; i<ret.length; i++) ret[i] = true;
67 * create a sequence (general form)
68 * @param head the head of the output tree
69 * @param e the elements to match
70 * @param drop only elements of <tt>e</tt> whose corresponding <tt>boolean</tt> in <tt>drops</tt>
71 * is <i>false</i> will be included in the output tree
72 * @param lifts which (if any) child trees to lift
74 public static Sequence create(Object head, Element[] e, boolean[] drop) {
75 return create(head, e, drop, new boolean[e.length]); }
76 public static Sequence create(Object head, Element[] e, boolean[] drop, boolean[] lifts) {
77 if (lifts==null) lifts = new boolean[e.length];
78 return new RewritingSequence(head, e, drop, lifts);
81 /** return a new sequence identical to this one, but with a positive conjunct <tt>s</tt> */
82 public Sequence and(Sequence s) {
84 throw new RuntimeException("you may not use a sequence as a conjunct if it belongs to a Union");
87 s.needed_or_hated=true;
91 /** return a new sequence identical to this one, but with a negative conjunct <tt>s</tt> */
92 public Sequence andnot(Sequence s) {
94 throw new RuntimeException("you may not use a sequence as a conjunct if it belongs to a Union");
97 s.needed_or_hated=true;
101 /** return a new sequence identical to this one, but with a follow-set restricted to <tt>a</tt> */
102 public Sequence followedBy(Atom a) { Sequence ret = dup(); ret.follow = a; return ret; }
104 ////////////////////////////////////////////////////////////////////////////////
106 abstract Sequence _clone();
107 private Sequence dup() {
108 Sequence ret = _clone();
109 for(Sequence s : needs) { ret.needs.add(s); }
110 for(Sequence s : hates) { ret.hates.add(s); }
115 Iterable<Sequence> needs() { return needs; }
116 Iterable<Sequence> hates() { return hates; }
118 Pos firstp() { return firstp; }
119 Pos lastp() { return firstp().last(); }
121 public Iterator<Element> iterator() { return new ArrayIterator<Element>(elements); }
122 protected Sequence(Element[] elements) {
123 this.elements = elements;
124 for(int i=0; i<elements.length; i++)
125 if (elements[i]==null)
126 throw new RuntimeException("cannot have nulls in a sequence: " + this);
127 this.firstp = new Position(this, 0, null);
130 abstract Forest epsilonForm(Input.Region loc);
132 protected abstract <T> Forest<T> postReduce(Input.Region loc, Forest<T>[] args, Position p);
135 // Position //////////////////////////////////////////////////////////////////////////////
137 static abstract class Pos implements IntegerMappable, Comparable<Pos>, Serializable {
140 public int ord() { return ord; }
142 final Forest[] holder;
144 Pos(int len) { this.holder = new Forest[len]; }
146 public abstract int provides();
147 public abstract int[] needs();
148 public abstract int[] hates();
149 public abstract boolean owner_needed_or_hated();
151 public abstract boolean isFirst();
152 public abstract boolean isLast();
153 public abstract Pos last();
154 public abstract Pos prev();
155 public abstract Pos next();
157 abstract Sequence owner();
158 abstract Element element();
160 public abstract int numPops();
161 public abstract <T> Forest<T> rewrite(Input.Region loc);
164 /** the imaginary position before or after an element of a sequence; corresponds to an "LR item" */
165 private static class Position extends Pos implements IntegerMappable {
167 public Pos getPos() {
168 return new DumbPos(elements.length, provides(), needs(), hates(), owner_needed_or_hated(), numPops(),
169 public int provides();
170 public int[] needs();
171 public int[] hates();
172 public boolean owner_needed_or_hated();
173 public int numPops();
174 public <T> Forest<T> rewrite(Input.Region loc)
178 public int numPops() { return pos; }
181 private final Position next;
182 private final Position prev;
183 private transient Sequence owner;
185 public int provides() { return owner().sernum; }
186 public int[] needs() { return owner().needs_int(); }
187 public int[] hates() { return owner().hates_int(); }
188 public boolean owner_needed_or_hated() { return owner().needed_or_hated; }
190 private Position(Sequence owner, int pos, Position prev) {
191 super(owner.elements.length);
194 this.next = pos==owner.elements.length ? null : new Position(owner, pos+1, this);
198 public int compareTo(Pos p) {
199 return ord - ((Position)p).ord;
202 public boolean isFirst() { return pos==0; }
203 public int pos() { return pos; }
205 /** the element immediately after this Position, or null if this is the last Position */
206 public Element element() { return pos>=owner().elements.length ? null : owner().elements[pos]; }
208 /** the element which produces the sequence to which this Position belongs */
209 public Sequence owner() { return owner; }
211 /** the next Position (the Position after <tt>this.element()</tt>) */
212 public Position next() { return next; }
214 /** true iff this Position is the last one in the sequence */
215 public boolean isLast() { return next()==null; }
216 public Position last() { return isLast() ? this : next().last(); }
217 public Position prev() { return prev; }
219 // Position /////////////////////////////////////////////////////////////////////////////////
221 public final <T> Forest<T> rewrite(Input.Region loc) {
222 if (isFirst()) owner().epsilonForm(loc);
223 for(int i=0; i<pos; i++) if (holder[i]==null) throw new Error("realbad " + i);
224 for(int i=pos; i<owner().elements.length; i++) {
225 if (holder[i]==null) holder[i] = ((Union)owner().elements[i]).epsilonForm(loc);
226 if (holder[i]==null) throw new Error("bad");
228 return owner().postReduce(loc, holder, this);
231 public String toString() {
232 StringBuffer ret = new StringBuffer();
234 for(Position p = (Position)owner().firstp(); p != null; p = p.next()) {
236 if (p==this) ret.append(" | ");
237 if (p.element()!=null) ret.append(p.element());
238 else ret.append(' ');
241 return ret.toString();
243 private final int idx = master_position_idx++;
244 public int toInt() { return idx; }
246 private static int master_position_idx = 0;
249 // toString //////////////////////////////////////////////////////////////////////////////
251 public String toString() { return toString(new StringBuffer(), false).toString(); }
252 StringBuffer toString(StringBuffer sb) { return toString(sb, true); }
253 StringBuffer toString(StringBuffer sb, boolean spacing) {
254 for(int i=0; i<elements.length; i++) {
255 sb.append(elements[i]+"");
258 if (follow != null) {
262 for(Sequence s : needs) {
266 for(Sequence s : hates) {
273 // Specialized Subclasses //////////////////////////////////////////////////////////////////////////////
275 static class Singleton extends Sequence {
276 private final int idx;
277 public Singleton(Element e) { this(new Element[] { e }, 0); }
278 public Singleton(Element[] e, int idx) { super(e); this.idx = idx; }
279 public <T> Forest<T> postReduce(Input.Region loc, Forest<T>[] args, Position p) { return args[idx]; }
280 Sequence _clone() { return new Singleton(elements,idx); }
281 Forest epsilonForm(Input.Region loc) {
282 return ((Union)elements[idx]).epsilonForm(loc);
286 static class RewritingSequence extends Sequence {
287 private final Object tag;
288 private final boolean[] drops;
289 private final boolean[] lifts;
290 Sequence _clone() { return new RewritingSequence(tag, elements, drops); }
291 public RewritingSequence(Object tag, Element[] e) { this(tag, e, null); }
292 public RewritingSequence(Object tag, Element[] e, boolean[] drops) { this(tag, e, drops, new boolean[e.length]); }
293 public RewritingSequence(Object tag, Element[] e, boolean[] drops, boolean[] lifts) {
295 if (tag==null) throw new Error();
297 this.drops = drops == null ? new boolean[e.length] : drops;
299 for(int i=0; i<this.drops.length; i++) if (!this.drops[i]) count++;
300 this.lifts = new boolean[count];
302 for(int i=0; i<this.drops.length; i++)
304 this.lifts[j++] = lifts[i];
306 public <T> Forest<T> postReduce(Input.Region loc, Forest<T>[] args, Position p) {
307 Forest<T>[] args2 = new Forest[lifts.length];
309 for(int i=0; i<args.length; i++) if (!drops[i]) args2[j++] = args[i];
310 return Forest.create(loc, (T)tag, args2, lifts);
312 public StringBuffer toString(StringBuffer sb, boolean spacing) {
313 int len = sb.length();
315 sb.append("\""+StringUtil.escapify(tag.toString(),"\"\r\n")+"\":: ");
316 super.toString(sb, spacing);
317 len = sb.length()-len;
318 if (spacing) for(int i=0; i<50-len; i++) sb.append(' ');
321 Forest epsilonForm(Input.Region loc) {
322 return Forest.create(loc, tag, new Forest[0], lifts);