b7e7e39703f54e35c426427f6e804f076cc18648
[sbp.git] / src / edu / berkeley / sbp / Repeat.java
1 package edu.berkeley.sbp;
2 import edu.berkeley.sbp.util.*;
3 import edu.berkeley.sbp.*;
4 import edu.berkeley.sbp.*;
5 import java.io.*;
6 import java.util.*;
7 import java.lang.reflect.*;
8 import java.lang.ref.*;
9
10 /** currently this class exports only static methods to create repetitions; there are no public instance methods or constructors */
11 class Repeat extends Union {
12
13     public Repeat(final Element e, boolean zeroOkay, boolean manyOkay, Object tag) {
14         this(e, zeroOkay, manyOkay, null, false, tag); }
15     public Repeat(final Element e, boolean zeroOkay, boolean manyOkay, final Element separator, Object tag) {
16         this(e, zeroOkay, manyOkay, separator, false, tag); }
17     protected Repeat(final Element e, boolean zeroOkay, boolean manyOkay, final Element separator, boolean maximal, Object tag) {
18         super(e+(!manyOkay ? "?" : (zeroOkay ? (maximal ? "**" : "*") : (maximal ? "++" : "+")))+(separator==null?"":("/"+separator)), true);
19         if (zeroOkay) {
20             add(new Sequence.Constant.Empty());
21             if (manyOkay) add(new Sequence.Singleton(Sequence.many1(e, separator)));
22             else          add(new Sequence.Singleton(e));
23         } else {
24             add(new Sequence.RewritingSequence(tag, new Element[] { e }, null));
25             if (separator==null)
26                 add(new Sequence.Unwrap(new Element[] { e,                 Repeat.this }, tag));
27             else
28                 add(new Sequence.Unwrap(new Element[] { e, separator,      Repeat.this }, tag, new boolean[] { false, true, false }));
29         }
30     }
31
32     public static class Maximal extends Repeat {
33         public Maximal(final Element e, boolean zeroOkay, boolean manyOkay, final Atom separator, Object tag) {
34             super(e, zeroOkay, manyOkay, separator, true, tag);
35             if (zeroOkay && separator != null)
36                 throw new RuntimeException("cannot create a maximal repetition of zero or more items with a separator (yet): " + this);
37             for(Sequence s : this)
38                 s.follow = new Invert(separator);
39         }
40         public Maximal(final Atom e, boolean zeroOkay, boolean manyOkay, Object tag) {
41             super(e, zeroOkay, manyOkay, null, true, tag);
42             for(Sequence s : this)
43                 s.follow = new Invert(e);
44         }
45     }
46
47
48     /** an atom which tracks the inverse of some other atom */
49     private static class Invert<T extends Input> extends Atom<T> {
50         private final Atom<T> a;
51         public Invert(Atom<T> a) { this.a = a; }
52         public Topology<T> top() { return a.complement(); }
53         public String toString() { return "~"+a; }
54     }
55 }