Package org.antlr.v4.runtime.tree.xpath

Source Code of org.antlr.v4.runtime.tree.xpath.XPathTokenElement

package org.antlr.v4.runtime.tree.xpath;

import org.antlr.v4.runtime.tree.ParseTree;
import org.antlr.v4.runtime.tree.TerminalNode;
import org.antlr.v4.runtime.tree.Tree;
import org.antlr.v4.runtime.tree.Trees;

import java.util.ArrayList;
import java.util.Collection;
import java.util.List;

public class XPathTokenElement extends XPathElement {
  protected int tokenType;
  public XPathTokenElement(String tokenName, int tokenType) {
    super(tokenName);
    this.tokenType = tokenType;
  }

  @Override
  public Collection<ParseTree> evaluate(ParseTree t) {
    // return all children of t that match nodeName
    List<ParseTree> nodes = new ArrayList<ParseTree>();
    for (Tree c : Trees.getChildren(t)) {
      if ( c instanceof TerminalNode ) {
        TerminalNode tnode = (TerminalNode)c;
        if ( (tnode.getSymbol().getType() == tokenType && !invert) ||
           (tnode.getSymbol().getType() != tokenType && invert) )
        {
          nodes.add(tnode);
        }
      }
    }
    return nodes;
  }
}
TOP

Related Classes of org.antlr.v4.runtime.tree.xpath.XPathTokenElement

TOP
Copyright © 2018 www.massapi.com. All rights reserved.
All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.