From d51784a250522ed0c416f4e9aa868b5ea43c8c18 Mon Sep 17 00:00:00 2001
From: parrt <parrt@antlr.org>
Date: Sat, 23 Jul 2011 12:58:09 -0800
Subject: [PATCH] got lexer actions working again

[git-p4: depot-paths = "//depot/code/antlr4/main/": change = 8901]
---
 .../v4/tool/templates/codegen/Java/Java.stg   | 30 +++++++------------
 .../v4/codegen/OutputModelController.java     | 22 ++++++++++++++
 .../src/org/antlr/v4/codegen/model/Lexer.java | 15 +++-------
 .../org/antlr/v4/codegen/model/Parser.java    | 11 -------
 4 files changed, 36 insertions(+), 42 deletions(-)

diff --git a/tool/resources/org/antlr/v4/tool/templates/codegen/Java/Java.stg b/tool/resources/org/antlr/v4/tool/templates/codegen/Java/Java.stg
index 094541992..c689dc2a6 100644
--- a/tool/resources/org/antlr/v4/tool/templates/codegen/Java/Java.stg
+++ b/tool/resources/org/antlr/v4/tool/templates/codegen/Java/Java.stg
@@ -58,10 +58,16 @@ public class <parser.name> extends Parser {
 	@Override
 	public ATN getATN() { return _ATN; }
 
+	<dumpActions(parser, actionFuncs, sempredFuncs)>
+    <atn>
+}
+>>
+
+dumpActions(recog, actionFuncs, sempredFuncs) ::= <<
 <if(actionFuncs)>
 	public void _action(RuleContext _localctx, int ruleIndex, int predIndex) {
 		switch ( ruleIndex ) {
-		    <parser.actionFuncs:{f|
+		    <recog.actionFuncs:{f|
 case <f.ruleIndex> : <f.name>_action((<f.ctxType>)_localctx, predIndex);}; separator="\n">
     	}
 	}
@@ -70,31 +76,15 @@ case <f.ruleIndex> : <f.name>_action((<f.ctxType>)_localctx, predIndex);}; separ
 <if(sempredFuncs)>
 	public boolean _sempred(RuleContext _localctx, int ruleIndex, int predIndex) {
 		switch ( ruleIndex ) {
-		    <parser.sempredFuncs:{f|
+		    <recog.sempredFuncs:{f|
 case <f.ruleIndex> : return <f.name>_sempred((<f.ctxType>)_localctx, predIndex);}; separator="\n">
     	}
 		return true;
 	}
     <sempredFuncs; separator="\n">
 <endif>
-
-    <atn>
-}
 >>
 
-/*
-dumpActions(actions) ::= <<
-<if(actions)>
-	public void action(int ruleIndex, int actionIndex) {
-		switch ( actionIndex ) {
-		    <actions:{index|
-case <index> : <actions.(index)> break;}; separator="\n">
-    	}
-	}
-<endif>
->>
-*/
-
 ctor(p) ::= <<
 public <p.name>(TokenStream input) {
     super(input);
@@ -553,7 +543,7 @@ import org.antlr.v4.runtime.misc.*;
 <lexer>
 >>
 
-Lexer(lexer, atn, actions, sempreds) ::= <<
+Lexer(lexer, atn, actionFuncs, sempredFuncs) ::= <<
 public class <lexer.name> extends Lexer {
     public static final int
         <lexer.tokens:{k | <k>=<lexer.tokens.(k)>}; separator=", ", wrap, anchor>;
@@ -585,7 +575,7 @@ public class <lexer.name> extends Lexer {
 
     <lexer.namedActions.members>
 
-<!    <dumpActions(actions)> !>
+	<dumpActions(lexer, actionFuncs, sempredFuncs)>
     <atn>
 }
 >>
diff --git a/tool/src/org/antlr/v4/codegen/OutputModelController.java b/tool/src/org/antlr/v4/codegen/OutputModelController.java
index 4b134361f..9d7b9e46e 100644
--- a/tool/src/org/antlr/v4/codegen/OutputModelController.java
+++ b/tool/src/org/antlr/v4/codegen/OutputModelController.java
@@ -93,6 +93,28 @@ public class OutputModelController {
 		LexerFile file = lexerFile(gen.getRecognizerFileName());
 		setRoot(file);
 		file.lexer = lexer(file);
+
+		Grammar g = delegate.getGrammar();
+		for (Rule r : g.rules.values()) {
+			String ctxType = gen.target.getRuleFunctionContextStructName(r);
+			for (ActionAST a : r.actions) {
+				if ( a instanceof PredAST ) {
+					PredAST p = (PredAST)a;
+					RuleSempredFunction rsf = new RuleSempredFunction(delegate, r, ctxType);
+					file.lexer.sempredFuncs.add(rsf);
+					rsf.actions.put(g.sempreds.get(p), new Action(delegate, p));
+				}
+				else if ( a.getType()==ANTLRParser.ACTION ||
+						  a.getType()==ANTLRParser.FORCED_ACTION )
+				{
+					// lexer sees {{...}} and {..} as same; neither are done until accept
+					RuleActionFunction raf = new RuleActionFunction(delegate, r, ctxType);
+					file.lexer.actionFuncs.add(raf);
+					raf.actions.put(g.actions.get(a), new ForcedAction(delegate, a));
+				}
+			}
+		}
+
 		return file;
 	}
 
diff --git a/tool/src/org/antlr/v4/codegen/model/Lexer.java b/tool/src/org/antlr/v4/codegen/model/Lexer.java
index ab25928ea..c30f69d0f 100644
--- a/tool/src/org/antlr/v4/codegen/model/Lexer.java
+++ b/tool/src/org/antlr/v4/codegen/model/Lexer.java
@@ -43,8 +43,10 @@ public class Lexer extends OutputModelObject {
 	public Collection<String> modes;
 
 	@ModelElement public SerializedATN atn;
-	@ModelElement public LinkedHashMap<Integer, Action> actions;
-	@ModelElement public LinkedHashMap<Integer, Action> sempreds;
+	@ModelElement public List<RuleActionFunction> actionFuncs =
+		new ArrayList<RuleActionFunction>();
+	@ModelElement public List<RuleSempredFunction> sempredFuncs =
+		new ArrayList<RuleSempredFunction>();
 
 	public Lexer(OutputModelFactory factory, LexerFile file) {
 		this.factory = factory;
@@ -63,15 +65,6 @@ public class Lexer extends OutputModelObject {
 
 		tokenNames = g.getTokenDisplayNames();
 		ruleNames = g.rules.keySet();
-
-		sempreds = new LinkedHashMap<Integer, Action>();
-		for (PredAST p : g.sempreds.keySet()) {
-			sempreds.put(g.sempreds.get(p), new Action(factory, p));
-		}
-		actions = new LinkedHashMap<Integer, Action>();
-		for (ActionAST a : g.actions.keySet()) {
-			actions.put(g.actions.get(a), new Action(factory, a));
-		}
 	}
 
 }
diff --git a/tool/src/org/antlr/v4/codegen/model/Parser.java b/tool/src/org/antlr/v4/codegen/model/Parser.java
index bf9e556b6..c46384484 100644
--- a/tool/src/org/antlr/v4/codegen/model/Parser.java
+++ b/tool/src/org/antlr/v4/codegen/model/Parser.java
@@ -62,16 +62,5 @@ public class Parser extends OutputModelObject {
 		tokenNames = g.getTokenDisplayNames();
 		ruleNames = g.rules.keySet();
 		atn = new SerializedATN(factory, g.atn);
-
-		/*
-		sempreds = new LinkedHashMap<Integer, Action>();
-		for (PredAST p : g.sempreds.keySet()) {
-			sempreds.put(g.sempreds.get(p), new Action(factory, p));
-		}
-		actions = new LinkedHashMap<Integer, ForcedAction>();
-		for (ActionAST a : g.actions.keySet()) {
-			actions.put(g.actions.get(a), new ForcedAction(factory, a));
-		}
-		*/
 	}
 }