diff options
Diffstat (limited to 'tool/src/org/antlr/v4/tool/LeftRecursiveRule.java')
-rw-r--r-- | tool/src/org/antlr/v4/tool/LeftRecursiveRule.java | 175 |
1 files changed, 175 insertions, 0 deletions
diff --git a/tool/src/org/antlr/v4/tool/LeftRecursiveRule.java b/tool/src/org/antlr/v4/tool/LeftRecursiveRule.java new file mode 100644 index 0000000..1d3b990 --- /dev/null +++ b/tool/src/org/antlr/v4/tool/LeftRecursiveRule.java @@ -0,0 +1,175 @@ +/* + * [The "BSD license"] + * Copyright (c) 2012 Terence Parr + * Copyright (c) 2012 Sam Harwell + * All rights reserved. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions + * are met: + * + * 1. Redistributions of source code must retain the above copyright + * notice, this list of conditions and the following disclaimer. + * 2. Redistributions in binary form must reproduce the above copyright + * notice, this list of conditions and the following disclaimer in the + * documentation and/or other materials provided with the distribution. + * 3. The name of the author may not be used to endorse or promote products + * derived from this software without specific prior written permission. + * + * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR + * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES + * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. + * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, + * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT + * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, + * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY + * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT + * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF + * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. + */ + +package org.antlr.v4.tool; + +import org.antlr.v4.analysis.LeftRecursiveRuleAltInfo; +import org.antlr.v4.misc.OrderedHashMap; +import org.antlr.v4.runtime.misc.Pair; +import org.antlr.v4.tool.ast.AltAST; +import org.antlr.v4.tool.ast.GrammarAST; +import org.antlr.v4.tool.ast.RuleAST; + +import java.util.ArrayList; +import java.util.HashMap; +import java.util.List; +import java.util.Map; + +public class LeftRecursiveRule extends Rule { + public List<LeftRecursiveRuleAltInfo> recPrimaryAlts; + public OrderedHashMap<Integer, LeftRecursiveRuleAltInfo> recOpAlts; + public RuleAST originalAST; + + /** Did we delete any labels on direct left-recur refs? Points at ID of ^(= ID el) */ + public List<Pair<GrammarAST,String>> leftRecursiveRuleRefLabels = + new ArrayList<Pair<GrammarAST,String>>(); + + public LeftRecursiveRule(Grammar g, String name, RuleAST ast) { + super(g, name, ast, 1); + originalAST = ast; + alt = new Alternative[numberOfAlts+1]; // always just one + for (int i=1; i<=numberOfAlts; i++) alt[i] = new Alternative(this, i); + } + + @Override + public boolean hasAltSpecificContexts() { + return super.hasAltSpecificContexts() || getAltLabels()!=null; + } + + @Override + public int getOriginalNumberOfAlts() { + int n = 0; + if ( recPrimaryAlts!=null ) n += recPrimaryAlts.size(); + if ( recOpAlts!=null ) n += recOpAlts.size(); + return n; + } + + public RuleAST getOriginalAST() { + return originalAST; + } + + @Override + public List<AltAST> getUnlabeledAltASTs() { + List<AltAST> alts = new ArrayList<AltAST>(); + for (LeftRecursiveRuleAltInfo altInfo : recPrimaryAlts) { + if (altInfo.altLabel == null) alts.add(altInfo.originalAltAST); + } + for (int i = 0; i < recOpAlts.size(); i++) { + LeftRecursiveRuleAltInfo altInfo = recOpAlts.getElement(i); + if ( altInfo.altLabel==null ) alts.add(altInfo.originalAltAST); + } + if ( alts.isEmpty() ) return null; + return alts; + } + + /** Return an array that maps predicted alt from primary decision + * to original alt of rule. For following rule, return [0, 2, 4] + * + e : e '*' e + | INT + | e '+' e + | ID + ; + + * That maps predicted alt 1 to original alt 2 and predicted 2 to alt 4. + * + * @since 4.5.1 + */ + public int[] getPrimaryAlts() { + if ( recPrimaryAlts.size()==0 ) return null; + int[] alts = new int[recPrimaryAlts.size()+1]; + for (int i = 0; i < recPrimaryAlts.size(); i++) { // recPrimaryAlts is a List not Map like recOpAlts + LeftRecursiveRuleAltInfo altInfo = recPrimaryAlts.get(i); + alts[i+1] = altInfo.altNum; + } + return alts; + } + + /** Return an array that maps predicted alt from recursive op decision + * to original alt of rule. For following rule, return [0, 1, 3] + * + e : e '*' e + | INT + | e '+' e + | ID + ; + + * That maps predicted alt 1 to original alt 1 and predicted 2 to alt 3. + * + * @since 4.5.1 + */ + public int[] getRecursiveOpAlts() { + if ( recOpAlts.size()==0 ) return null; + int[] alts = new int[recOpAlts.size()+1]; + int alt = 1; + for (LeftRecursiveRuleAltInfo altInfo : recOpAlts.values()) { + alts[alt] = altInfo.altNum; + alt++; // recOpAlts has alts possibly with gaps + } + return alts; + } + + /** Get -> labels from those alts we deleted for left-recursive rules. */ + @Override + public Map<String, List<Pair<Integer, AltAST>>> getAltLabels() { + Map<String, List<Pair<Integer, AltAST>>> labels = new HashMap<String, List<Pair<Integer, AltAST>>>(); + Map<String, List<Pair<Integer, AltAST>>> normalAltLabels = super.getAltLabels(); + if ( normalAltLabels!=null ) labels.putAll(normalAltLabels); + if ( recPrimaryAlts!=null ) { + for (LeftRecursiveRuleAltInfo altInfo : recPrimaryAlts) { + if (altInfo.altLabel != null) { + List<Pair<Integer, AltAST>> pairs = labels.get(altInfo.altLabel); + if (pairs == null) { + pairs = new ArrayList<Pair<Integer, AltAST>>(); + labels.put(altInfo.altLabel, pairs); + } + + pairs.add(new Pair<Integer, AltAST>(altInfo.altNum, altInfo.originalAltAST)); + } + } + } + if ( recOpAlts!=null ) { + for (int i = 0; i < recOpAlts.size(); i++) { + LeftRecursiveRuleAltInfo altInfo = recOpAlts.getElement(i); + if ( altInfo.altLabel!=null ) { + List<Pair<Integer, AltAST>> pairs = labels.get(altInfo.altLabel); + if (pairs == null) { + pairs = new ArrayList<Pair<Integer, AltAST>>(); + labels.put(altInfo.altLabel, pairs); + } + + pairs.add(new Pair<Integer, AltAST>(altInfo.altNum, altInfo.originalAltAST)); + } + } + } + if ( labels.isEmpty() ) return null; + return labels; + } +} |