Refactor case element generation

Case elements are now responsible for generating themselves.

1 files changed, 9 insertions, 0 deletions

diff --git a/src/main/java/bjc/rgens/parser/elements/RangeCaseElement.java b/src/main/java/bjc/rgens/parser/elements/RangeCaseElement.java
index d98bc61..7cebde9 100755
--- a/src/main/java/bjc/rgens/parser/elements/RangeCaseElement.java
+++ b/src/main/java/bjc/rgens/parser/elements/RangeCaseElement.java
@@ -1,5 +1,7 @@
 package bjc.rgens.parser.elements;
 
+import bjc.rgens.parser.GenerationState;
+
 public class RangeCaseElement extends CaseElement {
 	public final int begin;
 	public final int end;
@@ -11,6 +13,13 @@ public class RangeCaseElement extends CaseElement {
 		end = en;
 	}
 
+	public void generate(GenerationState state) {
+		int val = state.rnd.nextInt(end - begin);
+		val += begin;
+
+		state.contents.append(val);
+	}
+
 	@Override
 	public int hashCode() {
 		final int prime = 31;