Make variable fields private

This makes the two maps in GenerationState private. Use the
defineVar/findVar or defineRuleVar/findRuleVar instead.

This is another change on the way to doing autovivify

5 files changed, 5 insertions, 12 deletions

diff --git a/src/main/java/bjc/rgens/parser/elements/ExpVariableCaseElement.java b/src/main/java/bjc/rgens/parser/elements/ExpVariableCaseElement.java
index e58d073..1606e59 100755
--- a/src/main/java/bjc/rgens/parser/elements/ExpVariableCaseElement.java
+++ b/src/main/java/bjc/rgens/parser/elements/ExpVariableCaseElement.java
@@ -35,6 +35,6 @@ public class ExpVariableCaseElement extends VariableDefCaseElement {
 			throw new GrammarException(msg);
 		}
 
-		state.vars.put(varName, newState.contents.toString());
+		state.defineVar(varName, newState.contents.toString());
 	}
 }
diff --git a/src/main/java/bjc/rgens/parser/elements/LitVariableCaseElement.java b/src/main/java/bjc/rgens/parser/elements/LitVariableCaseElement.java
index 3c6ba98..c5e5c6f 100755
--- a/src/main/java/bjc/rgens/parser/elements/LitVariableCaseElement.java
+++ b/src/main/java/bjc/rgens/parser/elements/LitVariableCaseElement.java
@@ -8,6 +8,6 @@ public class LitVariableCaseElement extends VariableDefCaseElement {
 	}
 
 	public void generate(GenerationState state) {
-		state.vars.put(varName, varDef);
+		state.defineVar(varName, varDef);
 	}
 }
diff --git a/src/main/java/bjc/rgens/parser/elements/RuleVariableCaseElement.java b/src/main/java/bjc/rgens/parser/elements/RuleVariableCaseElement.java
index d11915e..a7fdbec 100644
--- a/src/main/java/bjc/rgens/parser/elements/RuleVariableCaseElement.java
+++ b/src/main/java/bjc/rgens/parser/elements/RuleVariableCaseElement.java
@@ -30,7 +30,7 @@ public class RuleVariableCaseElement extends VariableDefCaseElement {
 			rl = rl.exhaust();
 		}
 
-		state.rlVars.put(varName, rl);
+		state.defineRuleVar(varName, rl);
 
 		if(exhaust) {
 			fine("Defined exhausted rulevar '%s' ('%s')", varName, varDef);
diff --git a/src/main/java/bjc/rgens/parser/elements/vars/ARefVariableElement.java b/src/main/java/bjc/rgens/parser/elements/vars/ARefVariableElement.java
index a4bb730..5e8fe05 100644
--- a/src/main/java/bjc/rgens/parser/elements/vars/ARefVariableElement.java
+++ b/src/main/java/bjc/rgens/parser/elements/vars/ARefVariableElement.java
@@ -14,11 +14,7 @@ public class ARefVariableElement extends VariableElement {
 	}
 	
 	public void generate(GenerationState state) {
-		if(!state.rlVars.containsKey(value)) {
-			throw new GrammarException("No rule variable named " + value);
-		}
-
-		Rule rl = state.rlVars.get(value);
+		Rule rl = state.findRuleVar(value, state);
 
 		GenerationState newState = state.newBuf();
 
diff --git a/src/main/java/bjc/rgens/parser/elements/vars/VRefVariableElement.java b/src/main/java/bjc/rgens/parser/elements/vars/VRefVariableElement.java
index b19f785..a16503f 100644
--- a/src/main/java/bjc/rgens/parser/elements/vars/VRefVariableElement.java
+++ b/src/main/java/bjc/rgens/parser/elements/vars/VRefVariableElement.java
@@ -13,11 +13,8 @@ public class VRefVariableElement extends VariableElement {
 	}
 
 	public void generate(GenerationState state) {
-		if (!state.vars.containsKey(nam)) {
-			throw new GrammarException(String.format("No variable '%s' defined", nam));
-		}
+		String strang = state.findVar(nam, state);
 
-		String strang = state.vars.get(nam);
 		if(forbidSpaces && strang.contains(" ")) {
 			throw new GrammarException(String.format("Cannot include variable %s w/ spaces in body in rule name", nam));
 		}