Top Description Inners Fields Constructors Methods
org.antlr.v4.runtime

public Class TokenStreamRewriter

extends Object
Class Inheritance
Imports
org.antlr.v4.runtime.misc.Interval, java.util.ArrayList, .HashMap, .List, .Map

Useful for rewriting out a buffered input token stream after doing some augmentation or other manipulations on it.

You can insert stuff, replace, and delete chunks. Note that the operations are done lazily--only if you convert the buffer to a String with TokenStream#getText(). This is very efficient because you are not moving data around all the time. As the buffer of tokens is converted to strings, the getText() method(s) scan the input token stream and check to see if there is an operation at the current index. If so, the operation is done and then normal String rendering continues on the buffer. This is like having multiple Turing machine instruction streams (programs) operating on a single input tape. :)

This rewriter makes no modifications to the token stream. It does not ask the stream to fill itself up nor does it advance the input cursor. The token stream TokenStream#index() will return the same value before and after any getText() call.

The rewriter only works on tokens that you have in the buffer and ignores the current input cursor. If you are buffering tokens on-demand, calling getText() halfway through the input will only do rewrites for those tokens in the first half of the file.

Since the operations are done lazily at getText-time, operations do not screw up the token index values. That is, an insert operation at token index i does not change the index values for tokens i+1..n-1.

Because operations never actually alter the buffer, you may always get the original token stream back without undoing anything. Since the instructions are queued up, you can easily simulate transactions and roll back any changes if there is an error just by removing instructions. For example,

CharStream input = new ANTLRFileStream("input");
TLexer lex = new TLexer(input);
CommonTokenStream tokens = new CommonTokenStream(lex);
T parser = new T(tokens);
TokenStreamRewriter rewriter = new TokenStreamRewriter(tokens);
parser.startRule();

Then in the rules, you can execute (assuming rewriter is visible):

Token t,u;
...
rewriter.insertAfter(t, "text to put after t");}
rewriter.insertAfter(u, "text after u");}
System.out.println(rewriter.getText());

You can also have multiple "instruction streams" and get multiple rewrites from a single pass over the input. Just name the instruction streams and use that name again when printing the buffer. This could be useful for generating a C file and also its header file--all from the same buffer:

rewriter.insertAfter("pass1", t, "text to put after t");}
rewriter.insertAfter("pass2", u, "text after u");}
System.out.println(rewriter.getText("pass1"));
System.out.println(rewriter.getText("pass2"));

If you don't use named rewrite streams, a "default" stream is used as the first example shows.

Nested and Inner Type Summary

Modifier and TypeClass and Description
pack-priv class
TokenStreamRewriter.InsertAfterOp

Distinguish between insert after/before to do the "insert afters" first and then the "insert befores" at same index.

pack-priv class
pack-priv class
TokenStreamRewriter.ReplaceOp

I'm going to try replacing range from x..y with (y-x)+1 ReplaceOp instructions.

public class

Field Summary

Modifier and TypeField and Description
public static final String
protected final Map<String, Integer>
lastRewriteTokenIndexes

Map String (program name) → Integer index

public static final int
public static final int
protected final Map<String, List<TokenStreamRewriter.RewriteOperation>>
programs

You may have multiple, named streams of rewrite operations.

protected final TokenStream
tokens

Our source stream

Constructor Summary

AccessConstructor and Description
public

Method Summary

Modifier and TypeMethod and Description
protected String
public void
delete(int index)

public void
delete(int from, int to)

public void
delete(Token indexT)

public void
delete(Token from, Token to)

public void
delete(String programName, int from, int to)

public void
delete(String programName, Token from, Token to)

public void
public void
deleteProgram(String programName)

Reset the program so that no instructions exist

protected <T extends TokenStreamRewriter.RewriteOperation> List<? extends T>
getKindOfOps(List<? extends TokenStreamRewriter.RewriteOperation> rewrites, Class<T> kind, int before)

Get all operations before an index of a particular kind

public int
protected int
protected List<TokenStreamRewriter.RewriteOperation>
public String
getText()

Return the text from the original tokens altered per the instructions given to this rewriter.

public String
getText(String programName)

Return the text from the original tokens altered per the instructions given to this rewriter in programName.

public String
getText(Interval interval)

Return the text associated with the tokens in the interval from the original token stream but with the alterations given to this rewriter.

public String
getText(String programName, Interval interval)

public final TokenStream
private List<TokenStreamRewriter.RewriteOperation>
public void
public void
insertAfter(int index, Object text)

public void
insertAfter(String programName, Token t, Object text)

public void
insertAfter(String programName, int index, Object text)

public void
public void
insertBefore(int index, Object text)

public void
insertBefore(String programName, Token t, Object text)

public void
insertBefore(String programName, int index, Object text)

protected Map<Integer, TokenStreamRewriter.RewriteOperation>
reduceToSingleOperationPerIndex(List<TokenStreamRewriter.RewriteOperation> rewrites)

We need to combine operations and report invalid operations (like overlapping replaces that are not completed nested).

public void
replace(int index, Object text)

public void
replace(int from, int to, Object text)

public void
replace(Token indexT, Object text)

public void
replace(Token from, Token to, Object text)

public void
replace(String programName, int from, int to, Object text)

public void
replace(String programName, Token from, Token to, Object text)

public void
rollback(int instructionIndex)

public void
rollback(String programName, int instructionIndex)

Rollback the instruction stream for a program so that the indicated instruction (via instructionIndex) is no longer in the stream.

protected void
setLastRewriteTokenIndex(String programName, int i)

Inherited from java.lang.Object:
cloneequalsfinalizegetClasshashCodenotifynotifyAlltoStringwaitwaitwait

Field Detail

DEFAULT_PROGRAM_NAMEback to summary
public static final String DEFAULT_PROGRAM_NAME
lastRewriteTokenIndexesback to summary
protected final Map<String, Integer> lastRewriteTokenIndexes

Map String (program name) → Integer index

MIN_TOKEN_INDEXback to summary
public static final int MIN_TOKEN_INDEX
PROGRAM_INIT_SIZEback to summary
public static final int PROGRAM_INIT_SIZE
programsback to summary
protected final Map<String, List<TokenStreamRewriter.RewriteOperation>> programs

You may have multiple, named streams of rewrite operations. I'm calling these things "programs." Maps String (name) → rewrite (List)

tokensback to summary
protected final TokenStream tokens

Our source stream

Constructor Detail

TokenStreamRewriterback to summary
public TokenStreamRewriter(TokenStream tokens)

Method Detail

catOpTextback to summary
protected String catOpText(Object a, Object b)
deleteback to summary
public void delete(int index)
deleteback to summary
public void delete(int from, int to)
deleteback to summary
public void delete(Token indexT)
deleteback to summary
public void delete(Token from, Token to)
deleteback to summary
public void delete(String programName, int from, int to)
deleteback to summary
public void delete(String programName, Token from, Token to)
deleteProgramback to summary
public void deleteProgram()
deleteProgramback to summary
public void deleteProgram(String programName)

Reset the program so that no instructions exist

getKindOfOpsback to summary
protected <T extends TokenStreamRewriter.RewriteOperation> List<? extends T> getKindOfOps(List<? extends TokenStreamRewriter.RewriteOperation> rewrites, Class<T> kind, int before)

Get all operations before an index of a particular kind

getLastRewriteTokenIndexback to summary
public int getLastRewriteTokenIndex()
getLastRewriteTokenIndexback to summary
protected int getLastRewriteTokenIndex(String programName)
getProgramback to summary
protected List<TokenStreamRewriter.RewriteOperation> getProgram(String name)
getTextback to summary
public String getText()

Return the text from the original tokens altered per the instructions given to this rewriter.

getTextback to summary
public String getText(String programName)

Return the text from the original tokens altered per the instructions given to this rewriter in programName.

getTextback to summary
public String getText(Interval interval)

Return the text associated with the tokens in the interval from the original token stream but with the alterations given to this rewriter. The interval refers to the indexes in the original token stream. We do not alter the token stream in any way, so the indexes and intervals are still consistent. Includes any operations done to the first and last token in the interval. So, if you did an insertBefore on the first token, you would get that insertion. The same is true if you do an insertAfter the stop token.

getTextback to summary
public String getText(String programName, Interval interval)
getTokenStreamback to summary
public final TokenStream getTokenStream()
initializeProgramback to summary
private List<TokenStreamRewriter.RewriteOperation> initializeProgram(String name)
insertAfterback to summary
public void insertAfter(Token t, Object text)
insertAfterback to summary
public void insertAfter(int index, Object text)
insertAfterback to summary
public void insertAfter(String programName, Token t, Object text)
insertAfterback to summary
public void insertAfter(String programName, int index, Object text)
insertBeforeback to summary
public void insertBefore(Token t, Object text)
insertBeforeback to summary
public void insertBefore(int index, Object text)
insertBeforeback to summary
public void insertBefore(String programName, Token t, Object text)
insertBeforeback to summary
public void insertBefore(String programName, int index, Object text)
reduceToSingleOperationPerIndexback to summary
protected Map<Integer, TokenStreamRewriter.RewriteOperation> reduceToSingleOperationPerIndex(List<TokenStreamRewriter.RewriteOperation> rewrites)

We need to combine operations and report invalid operations (like overlapping replaces that are not completed nested). Inserts to same index need to be combined etc... Here are the cases: I.i.u I.j.v leave alone, nonoverlapping I.i.u I.i.v combine: Iivu R.i-j.u R.x-y.v | i-j in x-y delete first R R.i-j.u R.i-j.v delete first R R.i-j.u R.x-y.v | x-y in i-j ERROR R.i-j.u R.x-y.v | boundaries overlap ERROR Delete special case of replace (text==null): D.i-j.u D.x-y.v | boundaries overlap combine to max(min)..max(right) I.i.u R.x-y.v | i in (x+1)-y delete I (since insert before we're not deleting i) I.i.u R.x-y.v | i not in (x+1)-y leave alone, nonoverlapping R.x-y.v I.i.u | i in x-y ERROR R.x-y.v I.x.u R.x-y.uv (combine, delete I) R.x-y.v I.i.u | i not in x-y leave alone, nonoverlapping I.i.u = insert u before op @ index i R.x-y.u = replace x-y indexed tokens with u First we need to examine replaces. For any replace op: 1. wipe out any insertions before op within that range. 2. Drop any replace op before that is contained completely within that range. 3. Throw exception upon boundary overlap with any previous replace. Then we can deal with inserts: 1. for any inserts to same index, combine even if not adjacent. 2. for any prior replace with same left boundary, combine this insert with replace and delete this replace. 3. throw exception if index in same range as previous replace Don't actually delete; make op null in list. Easier to walk list. Later we can throw as we add to index → op map. Note that I.2 R.2-2 will wipe out I.2 even though, technically, the inserted stuff would be before the replace range. But, if you add tokens in front of a method body '{' and then delete the method body, I think the stuff before the '{' you added should disappear too. Return a map from token index to operation.

replaceback to summary
public void replace(int index, Object text)
replaceback to summary
public void replace(int from, int to, Object text)
replaceback to summary
public void replace(Token indexT, Object text)
replaceback to summary
public void replace(Token from, Token to, Object text)
replaceback to summary
public void replace(String programName, int from, int to, Object text)
replaceback to summary
public void replace(String programName, Token from, Token to, Object text)
rollbackback to summary
public void rollback(int instructionIndex)
rollbackback to summary
public void rollback(String programName, int instructionIndex)

Rollback the instruction stream for a program so that the indicated instruction (via instructionIndex) is no longer in the stream.

setLastRewriteTokenIndexback to summary
protected void setLastRewriteTokenIndex(String programName, int i)
org.antlr.v4.runtime back to summary

pack-priv Class TokenStreamRewriter.InsertAfterOp

extends InsertBeforeOp
Class Inheritance

Distinguish between insert after/before to do the "insert afters" first and then the "insert befores" at same index. Implementation of "insert after" is "insert before index+1".

Constructor Summary

AccessConstructor and Description
public
InsertAfterOp(int index, Object text)

Method Summary

Inherited from org.antlr.v4.runtime.TokenStreamRewriter.InsertBeforeOp:
execute

Constructor Detail

InsertAfterOpback to summary
public InsertAfterOp(int index, Object text)
org.antlr.v4.runtime back to summary

pack-priv Class TokenStreamRewriter.InsertBeforeOp

extends RewriteOperation
Class Inheritance
Known Direct Subclasses
org.antlr.v4.runtime.TokenStreamRewriter.InsertAfterOp

Field Summary

Inherited from org.antlr.v4.runtime.TokenStreamRewriter.RewriteOperation:
indexinstructionIndextext

Constructor Summary

AccessConstructor and Description
public
InsertBeforeOp(int index, Object text)

Method Summary

Modifier and TypeMethod and Description
public int
execute(StringBuilder buf)

Overrides org.antlr.v4.runtime.TokenStreamRewriter.RewriteOperation.execute.

Execute the rewrite operation by possibly adding to the buffer.
Inherited from org.antlr.v4.runtime.TokenStreamRewriter.RewriteOperation:
toString

Constructor Detail

InsertBeforeOpback to summary
public InsertBeforeOp(int index, Object text)

Method Detail

executeback to summary
public int execute(StringBuilder buf)

Overrides org.antlr.v4.runtime.TokenStreamRewriter.RewriteOperation.execute.

Doc from org.antlr.v4.runtime.TokenStreamRewriter.RewriteOperation.execute.

Execute the rewrite operation by possibly adding to the buffer. Return the index of the next token to operate on.

Annotations
@Override
org.antlr.v4.runtime back to summary

pack-priv Class TokenStreamRewriter.ReplaceOp

extends RewriteOperation
Class Inheritance

I'm going to try replacing range from x..y with (y-x)+1 ReplaceOp instructions.

Field Summary

Modifier and TypeField and Description
protected int
Inherited from org.antlr.v4.runtime.TokenStreamRewriter.RewriteOperation:
indexinstructionIndextext

Constructor Summary

AccessConstructor and Description
public
ReplaceOp(int from, int to, Object text)

Method Summary

Modifier and TypeMethod and Description
public int
execute(StringBuilder buf)

Overrides org.antlr.v4.runtime.TokenStreamRewriter.RewriteOperation.execute.

Execute the rewrite operation by possibly adding to the buffer.
public String
toString()

Overrides org.antlr.v4.runtime.TokenStreamRewriter.RewriteOperation.toString.

Returns a string representation of the object.

Field Detail

lastIndexback to summary
protected int lastIndex

Constructor Detail

ReplaceOpback to summary
public ReplaceOp(int from, int to, Object text)

Method Detail

executeback to summary
public int execute(StringBuilder buf)

Overrides org.antlr.v4.runtime.TokenStreamRewriter.RewriteOperation.execute.

Doc from org.antlr.v4.runtime.TokenStreamRewriter.RewriteOperation.execute.

Execute the rewrite operation by possibly adding to the buffer. Return the index of the next token to operate on.

Annotations
@Override
toStringback to summary
public String toString()

Overrides org.antlr.v4.runtime.TokenStreamRewriter.RewriteOperation.toString.

Doc from java.lang.Object.toString.

Returns a string representation of the object.

Returns:String

a string representation of the object.

Annotations
@Override
org.antlr.v4.runtime back to summary

public Class TokenStreamRewriter.RewriteOperation

extends Object
Class Inheritance
Known Direct Subclasses
org.antlr.v4.runtime.TokenStreamRewriter.InsertBeforeOp, org.antlr.v4.runtime.TokenStreamRewriter.ReplaceOp

Field Summary

Modifier and TypeField and Description
protected int
index

Token buffer index.

protected int
instructionIndex

What index into rewrites List are we?

protected Object

Constructor Summary

AccessConstructor and Description
protected
RewriteOperation(int index)

protected
RewriteOperation(int index, Object text)

Method Summary

Modifier and TypeMethod and Description
public int
execute(StringBuilder buf)

Execute the rewrite operation by possibly adding to the buffer.

public String
toString()

Overrides java.lang.Object.toString.

Returns a string representation of the object.
Inherited from java.lang.Object:
cloneequalsfinalizegetClasshashCodenotifynotifyAllwaitwaitwait

Field Detail

indexback to summary
protected int index

Token buffer index.

instructionIndexback to summary
protected int instructionIndex

What index into rewrites List are we?

textback to summary
protected Object text

Constructor Detail

RewriteOperationback to summary
protected RewriteOperation(int index)
RewriteOperationback to summary
protected RewriteOperation(int index, Object text)

Method Detail

executeback to summary
public int execute(StringBuilder buf)

Execute the rewrite operation by possibly adding to the buffer. Return the index of the next token to operate on.

toStringback to summary
public String toString()

Overrides java.lang.Object.toString.

Doc from java.lang.Object.toString.

Returns a string representation of the object.

Returns:String

a string representation of the object.

Annotations
@Override