GSS: use PriorityQueue, remove ugly debugging garbage
[sbp.git] / src / edu / berkeley / sbp / GSS.java
1 // Copyright 2006 all rights reserved; see LICENSE file for BSD-style license
2
3 package edu.berkeley.sbp;
4 import edu.berkeley.sbp.*;
5 import edu.berkeley.sbp.util.*;
6 import edu.berkeley.sbp.Parser.Table.*;
7 import edu.berkeley.sbp.Sequence.Position;
8 import java.io.*;
9 import java.util.*;
10 import java.lang.reflect.*;
11
12 /** implements Tomita's Graph Structured Stack */
13 class GSS {
14
15     Input input;
16     public GSS(Input input) { this.input = input; }
17     public Input getInput() { return input; }
18
19     // FIXME: right now, these are the performance bottleneck
20     HashMapBag<Integer,Sequence>       performed       = new HashMapBag<Integer,Sequence>();
21
22     /** FIXME */
23     Forest.Many finalResult;
24
25     /** corresponds to a positions <i>between tokens</i> the input stream; same as Tomita's U_i's */
26     class Phase<Tok> implements Invokable<State, Result, Object>, IntegerMappable, GraphViz.ToGraphViz, Iterable<Node> {
27
28         public PriorityQueue<Reduction> reductionQueue = new PriorityQueue<Reduction>();
29
30         public void invoke(State st, Result result, Object o) {
31             //shifts++;
32             good |= next.newNode(result, st, false);
33         }
34
35         /** the token immediately after this phase */
36         final Tok token;
37         final int pos;
38
39         public IntPairMap<Node> hash;  /* ALLOC */
40         private boolean good;
41         private Phase next = null;
42         private Phase prev;
43         private Input.Location location;
44         private Input.Location nextLocation;
45         private Input.Location prevLocation;
46         
47         private Forest forest;
48
49         public Phase(Phase prev, Phase previous, Tok token, Input.Location location,
50                      Input.Location nextLocation, Forest forest) throws ParseFailed {
51             this.prevLocation = prev==null ? location : prev.getLocation();
52             this.prev = prev;
53             this.forest = forest;
54             this.pos = previous==null ? 0 : previous.pos+1;
55             this.token = token;
56             this.location = location;
57             this.nextLocation = nextLocation;
58             performed.clear();
59             hash = new IntPairMap<Node>();
60             good = false;
61             finalResult = null;
62             if (prev != null) prev.shift(this, forest);
63         }
64
65         public boolean isFrontier() { return next==null; }
66         public boolean isDone() throws ParseFailed {
67             if (token != null) return false;
68             if (token==null && finalResult==null)
69                 ParseFailed.error("unexpected end of file", this);
70             return true;
71         }
72
73         public Input.Location getPrevLocation() { return prevLocation; }
74         public Input.Location getLocation() { return location; }
75         public Input.Region   getRegion() { return getPrevLocation().createRegion(getLocation()); }
76         public Input.Location getNextLocation() { return nextLocation; }
77
78         /** perform all shift operations, adding promoted nodes to <tt>next</tt> */
79         public void shift(Phase next, Forest result) throws ParseFailed {
80             // this massively improves GC performance
81             if (prev!=null) prev.hash = null;
82             this.next = next;
83             for(Node n : hash.values()) {
84                 if (token == null && n.state().isAccepting()) {
85                     if (finalResult==null) finalResult = new Forest.Many();
86                     for(Result r : n)
87                         finalResult.merge(r.getForest());
88                 }
89                 if (token == null) continue;
90                 n.state().invokeShifts(token, this, new Result(result, n, null), null);
91             }
92             if (!good && token!=null) ParseFailed.error("unexpected character", this);
93             if (token==null && finalResult==null) ParseFailed.error("unexpected end of file", this);
94         }
95
96         /** perform all reduction operations */
97         public void reduce() throws ParseFailed {
98             while(!reductionQueue.isEmpty())
99                 reductionQueue.poll().perform();
100         }
101
102         public void newNodeFromReduction(Result result, State state, boolean fromEmptyReduction, Position reduction) {
103             int pos = result.phase().pos;
104             Sequence owner = reduction.owner();
105             for(Sequence s : owner.hates)
106                 if (performed.contains(pos, s))
107                     return;
108             for(Sequence s : owner.needs)
109                 if (!performed.contains(pos, s))
110                     return;
111             if (owner.needed_or_hated && !performed.contains(pos, owner))
112                 performed.add(pos, owner);
113             if (state!=null)
114                 newNode(result, state, fromEmptyReduction);
115         }
116
117         /** add a new node (merging with existing nodes if possible)
118          *  @param parent             the parent of the new node
119          *  @param result             the SPPF result corresponding to the new node
120          *  @param state              the state that the new node is in
121          *  @param fromEmptyReduction true iff this node is being created as a result of a reduction of length zero (see GRMLR paper)
122          *  @param start              the earliest part of the input contributing to this node (used to make merging decisions)
123          */
124         public boolean newNode(Result result, State state, boolean fromEmptyReduction) {
125             Node p = hash.get(state, result.phase());
126             if (p != null) { p.merge(result); return true; }
127             do {
128                 if (token != null && state.canShift(token)) break;
129                 if (state.isAccepting()) break;
130                 if (token==null) break;
131                 if (!state.canReduce(token)) return false;
132             } while(false);
133             new Node(Phase.this, result, state, fromEmptyReduction);  // ALLOC
134             return true;
135         }
136
137         public int toInt() { return pos+1; }
138         public int size() { return hash==null ? 0 : hash.size(); }
139         public int pos() { return pos; }
140         public Tok getToken() { return token; }
141         public Iterator<Node> iterator() { return hash.iterator(); }
142         public GSS getGSS() { return GSS.this; }
143
144         // GraphViz //////////////////////////////////////////////////////////////////////////////
145
146         public GraphViz.Node toGraphViz(GraphViz gv) {
147             if (gv.hasNode(this)) return gv.createNode(this);
148             GraphViz.Group g = gv.createGroup(this);
149             g.label = "Phase " + pos;
150             g.color = "gray";
151             g.cluster = true;
152             return g;
153         }
154         public boolean isTransparent() { return false; }
155         public boolean isHidden() { return false; }
156
157     }
158
159 }