TokenStreamSelector.java
上传用户:afrynkmhm
上传日期:2007-01-06
资源大小:1262k
文件大小:3k
- package antlr;
- /* ANTLR Translator Generator
- * Project led by Terence Parr at http://www.jGuru.com
- * Software rights: http://www.antlr.org/RIGHTS.html
- *
- * $Id: //depot/code/org.antlr/release/antlr-2.7.0/antlr/TokenStreamSelector.java#1 $
- */
- import java.util.Hashtable;
- import antlr.collections.impl.LList;
- import antlr.collections.Stack;
- import java.io.IOException;
- /** A token stream MUX (multiplexor) knows about n token streams
- * and can multiplex them onto the same channel for use by token
- * stream consumer like a parser. This is a way to have multiple
- * lexers break up the same input stream for a single parser.
- * Or, you can have multiple instances of the same lexer handle
- * multiple input streams; this works great for includes.
- */
- public class TokenStreamSelector implements TokenStream {
- /** The set of inputs to the MUX */
- protected Hashtable inputStreamNames;
- /** The currently-selected token stream input */
- protected TokenStream input;
- /** Used to track stack of input streams */
- protected Stack streamStack = new LList();
- public TokenStreamSelector() {
- super();
- inputStreamNames = new Hashtable();
- }
- public void addInputStream(TokenStream stream, String key) {
- inputStreamNames.put(key, stream);
- }
- /** Return the stream from tokens are being pulled at
- * the moment.
- */
- public TokenStream getCurrentStream() {
- return input;
- }
- public TokenStream getStream(String sname) {
- TokenStream stream = (TokenStream)inputStreamNames.get(sname);
- if ( stream==null ) {
- throw new IllegalArgumentException("TokenStream "+sname+" not found");
- }
- return stream;
- }
- public Token nextToken() throws TokenStreamException {
- // return input.nextToken();
- // keep looking for a token until you don't
- // get a retry exception.
- for (;;) {
- try {
- return input.nextToken();
- }
- catch (TokenStreamRetryException r) {
- // just retry "forever"
- }
- }
- }
- public TokenStream pop() {
- TokenStream stream = (TokenStream) streamStack.pop();
- select(stream);
- return stream;
- }
- public void push(TokenStream stream) {
- streamStack.push(input); // save current stream
- select(stream);
- }
- public void push(String sname) {
- streamStack.push(input);
- select(sname);
- }
- /** Abort recognition of current Token and try again.
- * A stream can push a new stream (for include files
- * for example, and then retry(), which will cause
- * the current stream to abort back to this.nextToken().
- * this.nextToken() then asks for a token from the
- * current stream, which is the new "substream."
- */
- public void retry() throws TokenStreamRetryException {
- throw new TokenStreamRetryException();
- }
- /** Set the stream without pushing old stream */
- public void select(TokenStream stream) {
- input = stream;
- }
- public void select(String sname) throws IllegalArgumentException {
- input = getStream(sname);
- }
- }