refactor predecessor/successor into Node class
[sbp.git] / src / edu / berkeley / sbp / StateNode.java
1 // Copyright 2006-2007 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.Pos;
8 import edu.berkeley.sbp.Sequence.Pos;
9 import java.io.*;
10 import java.util.*;
11 import java.lang.reflect.*;
12
13 /** a node in the GSS */
14 final class StateNode
15     extends Node<ResultNode>
16     implements Invokable<Pos, ResultNode, Object>,
17                Iterable<ResultNode> {
18
19     /** which GSS.Phase this StateNode belongs to */
20     public GSS.Phase phase() { return phase; }
21     public Iterator<ResultNode> iterator() { return predecessors.iterator(); }
22     public Parser.Table.State state() { return state; }
23
24     boolean destroyed = false;
25
26     public void check() {
27         if (destroyed) return;
28         boolean dead = true;
29         // - all nodes must have a predecessor
30         // - non-doomed nodes must either:
31         //      - be on the frontier or
32         //      - have a non-doomed node closer to the frontier than themself
33         if (phase.isFrontier()) dead = false;
34         else for(ResultNode r : successors)
35                  if (state.doomed) { if (r.usedByAnyNode()) { dead = false; break; } }
36                  else              { if (r.usedByNonDoomedNode()) { dead = false; break; } }
37         dead |= predecessors.size()==0;
38         if (!dead) return;
39         destroyed = true;
40         if (phase() != null && phase().hash != null)
41             phase().hash.remove(state, predPhase);
42         while(successors.size()>0)
43             for(ResultNode r : successors) {
44                 successors.remove(r);
45                 r.removePred(this);
46                 break;
47             }
48         while(predecessors.size()>0)
49             for(ResultNode r : predecessors) {
50                 predecessors.remove(r);
51                 r.removeSucc(this);
52                 break;
53             }
54         predecessors = null;
55         successors = null;
56     }
57
58     //////////////////////////////////////////////////////////////////////
59
60     private final GSS.Phase phase;
61     private final GSS.Phase predPhase;
62     private final Parser.Table.State state;
63     private  boolean fromEmptyReduction;
64
65     public final void invoke(Pos r, ResultNode only, Object o) {
66         boolean emptyProductions = only==null;
67         if (emptyProductions != (r.numPops()==0)) return;
68         if (r.numPops()!=0)  reduce(r, r.numPops()-1, phase(), only);
69         else {
70             Input.Region region = phase().getLocation().createRegion(phase().getLocation());
71             phase().newNodeFromReduction(r.rewrite(region), r, this);
72         }
73     }
74
75     private void reduce(Pos r, int pos, GSS.Phase target, ResultNode only) {
76         for(ResultNode res : predecessors)
77             if (only == null || res == only)
78                 for(StateNode pred : res.getPreds())
79                     reduce2(r, pos, target, pred, res.getForest());
80     }
81
82     void reduce2(Pos r, int pos, GSS.Phase target, StateNode pred, Forest f) {
83         Forest[] holder = r.holder;
84         Forest old = pos >= holder.length ? null : holder[pos];
85         if (pos < holder.length) holder[pos] = f;
86         if (pos>0)  pred.reduce(r, pos-1, target, null);
87         else {
88             Input.Region region = pred.phase().getLocation().createRegion(target.getLocation());
89             new Reduction(pred, r, r.rewrite(region), target);
90         }
91         if (pos < holder.length) holder[pos] = old;
92     }
93
94     StateNode(GSS.Phase phase, Forest f, Pos reduction, StateNode pred, State state, boolean fromEmptyReduction) {
95         this(phase, new ResultNode(f, reduction, pred), state, fromEmptyReduction);
96     }
97     StateNode(GSS.Phase phase, ResultNode pred, State state, boolean fromEmptyReduction) {
98         this.phase = phase;
99         this.state = state;
100         this.fromEmptyReduction = fromEmptyReduction;
101         if (phase.hash.get(state, pred.phase()) != null) throw new Error("severe problem!");
102         this.predPhase = pred.phase();
103         phase.hash.put(state, pred.phase(), this);
104
105         predecessors.add(pred);
106         pred.addSucc(this);
107         if (!fromEmptyReduction)
108             state.invokeReductions(phase().getToken(), this, pred);
109
110         state.invokeEpsilonReductions(phase().token, this);
111     }
112
113     // Add/Remove Successors/Predecessors //////////////////////////////////////////////////////////////////////////////
114
115     public void removeSucc(ResultNode succ)   {
116         successors.remove(succ);
117         check();
118     }
119     public void removeResult(ResultNode result) {
120         predecessors.remove(result);
121         check();
122     }
123     public void addSucc(ResultNode succ)      {
124         successors.add(succ);
125     }
126     public void addResult(Forest f, Pos reduction, StateNode pred) {
127         for(ResultNode r : predecessors)
128             if (r.predecessorsContains(pred)) {
129                 r.merge(f);
130                 return;
131             }
132         ResultNode result = new ResultNode(f, reduction, pred);
133         predecessors.add(result);
134         result.addSucc(this);
135         if (!this.fromEmptyReduction) state.invokeReductions(phase().getToken(), this, result);
136     }
137 }