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.
| Modifier and Type | Class and Description |
|---|---|
| pack-priv class | TokenStreamRewriter.
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.
I'm going to try replacing range from x..y with (y-x)+1 ReplaceOp instructions. |
| public class |
| Modifier and Type | Field and Description |
|---|---|
| public static final String | |
| protected final Map | lastRewriteTokenIndexes
Map String (program name) → Integer index |
| public static final int | |
| public static final int | |
| protected final Map | programs
You may have multiple, named streams of rewrite operations. |
| protected final TokenStream | tokens
Our source stream |
| Access | Constructor and Description |
|---|---|
| public |
| Modifier and Type | Method and Description |
|---|---|
| protected String | |
| public void | |
| public void | |
| public void | |
| public void | |
| public void | |
| public void | |
| public void | |
| public void | |
| protected <T extends TokenStreamRewriter. | getKindOfOps(List<? extends TokenStreamRewriter.
Get all operations before an index of a particular kind |
| public int | |
| protected int | |
| protected List | |
| public String | getText()
Return the text from the original tokens altered per the instructions given to this rewriter. |
| public String | |
| public String | |
| public String | |
| public final TokenStream | |
| private List | |
| public void | |
| public void | |
| public void | |
| public void | |
| public void | |
| public void | |
| public void | |
| public void | |
| protected Map | reduceToSingleOperationPerIndex(List<TokenStreamRewriter.
We need to combine operations and report invalid operations (like overlapping replaces that are not completed nested). |
| public void | |
| public void | |
| public void | |
| public void | |
| public void | |
| public void | |
| public void | |
| public void | |
| protected void |
| DEFAULT_PROGRAM_NAME | back to summary |
|---|---|
| public static final String DEFAULT_PROGRAM_NAME | |
| lastRewriteTokenIndexes | back to summary |
|---|---|
| protected final Map<String, Integer> lastRewriteTokenIndexes Map String (program name) → Integer index | |
| MIN_TOKEN_INDEX | back to summary |
|---|---|
| public static final int MIN_TOKEN_INDEX | |
| PROGRAM_INIT_SIZE | back to summary |
|---|---|
| public static final int PROGRAM_INIT_SIZE | |
| programs | back to summary |
|---|---|
| protected final Map<String, List<TokenStreamRewriter. You may have multiple, named streams of rewrite operations. I'm calling these things "programs." Maps String (name) → rewrite (List) | |
| tokens | back to summary |
|---|---|
| protected final TokenStream tokens Our source stream | |
| TokenStreamRewriter | back to summary |
|---|---|
| public TokenStreamRewriter(TokenStream tokens) | |
| catOpText | back to summary |
|---|---|
| protected String catOpText(Object a, Object b) | |
| delete | back to summary |
|---|---|
| public void delete(int index) | |
| delete | back to summary |
|---|---|
| public void delete(int from, int to) | |
| delete | back to summary |
|---|---|
| public void delete(Token indexT) | |
| delete | back to summary |
|---|---|
| public void delete(Token from, Token to) | |
| delete | back to summary |
|---|---|
| public void delete(String programName, int from, int to) | |
| delete | back to summary |
|---|---|
| public void delete(String programName, Token from, Token to) | |
| deleteProgram | back to summary |
|---|---|
| public void deleteProgram() | |
| deleteProgram | back to summary |
|---|---|
| public void deleteProgram(String programName) Reset the program so that no instructions exist | |
| getKindOfOps | back to summary |
|---|---|
| protected <T extends TokenStreamRewriter. Get all operations before an index of a particular kind | |
| getLastRewriteTokenIndex | back to summary |
|---|---|
| public int getLastRewriteTokenIndex() | |
| getLastRewriteTokenIndex | back to summary |
|---|---|
| protected int getLastRewriteTokenIndex(String programName) | |
| getProgram | back to summary |
|---|---|
| protected List | |
| getText | back to summary |
|---|---|
| public String getText() Return the text from the original tokens altered per the instructions given to this rewriter. | |
| getText | back to summary |
|---|---|
| public String getText(String programName) Return the text from the original tokens altered per the instructions given to this rewriter in programName. | |
| getText | back 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. | |
| getText | back to summary |
|---|---|
| public String getText(String programName, Interval interval) | |
| getTokenStream | back to summary |
|---|---|
| public final TokenStream getTokenStream() | |
| initializeProgram | back to summary |
|---|---|
| private List | |
| insertAfter | back to summary |
|---|---|
| public void insertAfter(Token t, Object text) | |
| insertAfter | back to summary |
|---|---|
| public void insertAfter(int index, Object text) | |
| insertAfter | back to summary |
|---|---|
| public void insertAfter(String programName, Token t, Object text) | |
| insertAfter | back to summary |
|---|---|
| public void insertAfter(String programName, int index, Object text) | |
| insertBefore | back to summary |
|---|---|
| public void insertBefore(Token t, Object text) | |
| insertBefore | back to summary |
|---|---|
| public void insertBefore(int index, Object text) | |
| insertBefore | back to summary |
|---|---|
| public void insertBefore(String programName, Token t, Object text) | |
| insertBefore | back to summary |
|---|---|
| public void insertBefore(String programName, int index, Object text) | |
| reduceToSingleOperationPerIndex | back to summary |
|---|---|
| protected Map 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. | |
| replace | back to summary |
|---|---|
| public void replace(int index, Object text) | |
| replace | back to summary |
|---|---|
| public void replace(int from, int to, Object text) | |
| replace | back to summary |
|---|---|
| public void replace(Token indexT, Object text) | |
| replace | back to summary |
|---|---|
| public void replace(Token from, Token to, Object text) | |
| replace | back to summary |
|---|---|
| public void replace(String programName, int from, int to, Object text) | |
| replace | back to summary |
|---|---|
| public void replace(String programName, Token from, Token to, Object text) | |
| rollback | back to summary |
|---|---|
| public void rollback(int instructionIndex) | |
| rollback | back 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. | |
| setLastRewriteTokenIndex | back to summary |
|---|---|
| protected void setLastRewriteTokenIndex(String programName, int i) | |
| Access | Constructor and Description |
|---|---|
| public |
| InsertAfterOp | back to summary |
|---|---|
| public InsertAfterOp(int index, Object text) | |
| Access | Constructor and Description |
|---|---|
| public |
| Modifier and Type | Method and Description |
|---|---|
| public int | execute(StringBuilder buf)
Overrides org. |
| InsertBeforeOp | back to summary |
|---|---|
| public InsertBeforeOp(int index, Object text) | |
| execute | back to summary |
|---|---|
| public int execute(StringBuilder buf) Overrides org. Doc from org. Execute the rewrite operation by possibly adding to the buffer. Return the index of the next token to operate on.
| |
| Modifier and Type | Field and Description |
|---|---|
| protected int |
| Access | Constructor and Description |
|---|---|
| public |
| Modifier and Type | Method and Description |
|---|---|
| public int | execute(StringBuilder buf)
Overrides org. |
| public String | toString()
Overrides org. |
| lastIndex | back to summary |
|---|---|
| protected int lastIndex | |
| ReplaceOp | back to summary |
|---|---|
| public ReplaceOp(int from, int to, Object text) | |
| execute | back to summary |
|---|---|
| public int execute(StringBuilder buf) Overrides org. Doc from org. Execute the rewrite operation by possibly adding to the buffer. Return the index of the next token to operate on.
| |
| toString | back to summary |
|---|---|
| public String toString() Overrides org. Doc from java. Returns a string representation of the object. | |
| Modifier and Type | Field and Description |
|---|---|
| protected int | index
Token buffer index. |
| protected int | instructionIndex
What index into rewrites List are we? |
| protected Object |
| Access | Constructor and Description |
|---|---|
| protected | |
| protected |
| Modifier and Type | Method and Description |
|---|---|
| public int | |
| public String |
| index | back to summary |
|---|---|
| protected int index Token buffer index. | |
| instructionIndex | back to summary |
|---|---|
| protected int instructionIndex What index into rewrites List are we? | |
| text | back to summary |
|---|---|
| protected Object text | |
| RewriteOperation | back to summary |
|---|---|
| protected RewriteOperation(int index) | |
| RewriteOperation | back to summary |
|---|---|
| protected RewriteOperation(int index, Object text) | |
| execute | back 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. | |
| toString | back to summary |
|---|---|
| public String toString() Overrides java. Doc from java. Returns a string representation of the object. | |