13447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein/* 23447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein [The "BSD license"] 33447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein Copyright (c) 2005-2009 Terence Parr 43447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein All rights reserved. 53447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein 63447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein Redistribution and use in source and binary forms, with or without 73447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein modification, are permitted provided that the following conditions 83447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein are met: 93447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein 1. Redistributions of source code must retain the above copyright 103447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein notice, this list of conditions and the following disclaimer. 113447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein 2. Redistributions in binary form must reproduce the above copyright 123447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein notice, this list of conditions and the following disclaimer in the 133447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein documentation and/or other materials provided with the distribution. 143447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein 3. The name of the author may not be used to endorse or promote products 153447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein derived from this software without specific prior written permission. 163447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein 173447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR 183447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 193447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 203447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, 213447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT 223447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 233447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 243447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 253447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 263447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 273447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein */ 283447a5916aa62f44de24cc441fc9987116ddff52Andrew Sappersteinpackage org.antlr.runtime.tree; 293447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein 303447a5916aa62f44de24cc441fc9987116ddff52Andrew Sappersteinimport org.antlr.runtime.Token; 313447a5916aa62f44de24cc441fc9987116ddff52Andrew Sappersteinimport org.antlr.runtime.CommonToken; 323447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein 333447a5916aa62f44de24cc441fc9987116ddff52Andrew Sappersteinpublic class TreePatternParser { 343447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein protected TreePatternLexer tokenizer; 353447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein protected int ttype; 363447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein protected TreeWizard wizard; 373447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein protected TreeAdaptor adaptor; 383447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein 393447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein public TreePatternParser(TreePatternLexer tokenizer, TreeWizard wizard, TreeAdaptor adaptor) { 403447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein this.tokenizer = tokenizer; 413447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein this.wizard = wizard; 423447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein this.adaptor = adaptor; 433447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein ttype = tokenizer.nextToken(); // kickstart 443447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein } 453447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein 463447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein public Object pattern() { 473447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein if ( ttype==TreePatternLexer.BEGIN ) { 483447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein return parseTree(); 493447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein } 503447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein else if ( ttype==TreePatternLexer.ID ) { 513447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein Object node = parseNode(); 523447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein if ( ttype==TreePatternLexer.EOF ) { 533447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein return node; 543447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein } 553447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein return null; // extra junk on end 563447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein } 573447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein return null; 583447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein } 593447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein 603447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein public Object parseTree() { 613447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein if ( ttype != TreePatternLexer.BEGIN ) { 623447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein throw new RuntimeException("no BEGIN"); 633447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein } 643447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein ttype = tokenizer.nextToken(); 653447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein Object root = parseNode(); 663447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein if ( root==null ) { 673447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein return null; 683447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein } 693447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein while ( ttype==TreePatternLexer.BEGIN || 703447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein ttype==TreePatternLexer.ID || 713447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein ttype==TreePatternLexer.PERCENT || 723447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein ttype==TreePatternLexer.DOT ) 733447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein { 743447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein if ( ttype==TreePatternLexer.BEGIN ) { 753447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein Object subtree = parseTree(); 763447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein adaptor.addChild(root, subtree); 773447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein } 783447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein else { 793447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein Object child = parseNode(); 803447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein if ( child==null ) { 813447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein return null; 823447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein } 833447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein adaptor.addChild(root, child); 843447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein } 853447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein } 863447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein if ( ttype != TreePatternLexer.END ) { 873447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein throw new RuntimeException("no END"); 883447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein } 893447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein ttype = tokenizer.nextToken(); 903447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein return root; 913447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein } 923447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein 933447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein public Object parseNode() { 943447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein // "%label:" prefix 953447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein String label = null; 963447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein if ( ttype == TreePatternLexer.PERCENT ) { 973447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein ttype = tokenizer.nextToken(); 983447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein if ( ttype != TreePatternLexer.ID ) { 993447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein return null; 1003447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein } 1013447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein label = tokenizer.sval.toString(); 1023447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein ttype = tokenizer.nextToken(); 1033447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein if ( ttype != TreePatternLexer.COLON ) { 1043447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein return null; 1053447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein } 1063447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein ttype = tokenizer.nextToken(); // move to ID following colon 1073447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein } 1083447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein 1093447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein // Wildcard? 1103447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein if ( ttype == TreePatternLexer.DOT ) { 1113447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein ttype = tokenizer.nextToken(); 1123447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein Token wildcardPayload = new CommonToken(0, "."); 1133447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein TreeWizard.TreePattern node = 1143447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein new TreeWizard.WildcardTreePattern(wildcardPayload); 1153447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein if ( label!=null ) { 1163447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein node.label = label; 1173447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein } 1183447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein return node; 1193447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein } 1203447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein 1213447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein // "ID" or "ID[arg]" 1223447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein if ( ttype != TreePatternLexer.ID ) { 1233447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein return null; 1243447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein } 1253447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein String tokenName = tokenizer.sval.toString(); 1263447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein ttype = tokenizer.nextToken(); 1273447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein if ( tokenName.equals("nil") ) { 1283447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein return adaptor.nil(); 1293447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein } 1303447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein String text = tokenName; 1313447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein // check for arg 1323447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein String arg = null; 1333447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein if ( ttype == TreePatternLexer.ARG ) { 1343447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein arg = tokenizer.sval.toString(); 1353447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein text = arg; 1363447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein ttype = tokenizer.nextToken(); 1373447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein } 1383447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein 1393447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein // create node 1403447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein int treeNodeType = wizard.getTokenType(tokenName); 1413447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein if ( treeNodeType==Token.INVALID_TOKEN_TYPE ) { 1423447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein return null; 1433447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein } 1443447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein Object node; 1453447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein node = adaptor.create(treeNodeType, text); 1463447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein if ( label!=null && node.getClass()==TreeWizard.TreePattern.class ) { 1473447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein ((TreeWizard.TreePattern)node).label = label; 1483447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein } 1493447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein if ( arg!=null && node.getClass()==TreeWizard.TreePattern.class ) { 1503447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein ((TreeWizard.TreePattern)node).hasTextArg = true; 1513447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein } 1523447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein return node; 1533447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein } 1543447a5916aa62f44de24cc441fc9987116ddff52Andrew Sapperstein} 155