Make NFA.toString() return the current state's toString() as it's mor…

…e useful.
drguildo · Sep 28, 2015 · 435b75e · 435b75e
1 parent 7e0aaa3
commit 435b75e
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/src/io/sjm/automata/NFA.java b/src/io/sjm/automata/NFA.java
@@ -1,9 +1,9 @@
 package io.sjm.automata;
 
-import java.util.Set;
-
 import io.sjm.stdlib.datastructures.Sets;
 
+import java.util.Set;
+
 public class NFA<T> {
   private Set<T> currentStates;
   private Set<T> acceptStates;
@@ -27,7 +27,12 @@ public void readString(String s) {
     s.chars().forEach(c -> readCharacter((char) c));
   }
 
-  private Set<T> currentStates() {
+  public Set<T> currentStates() {
     return rulebook.followFreeMoves(currentStates);
   }
+
+  @Override
+  public String toString() {
+    return currentStates().toString();
+  }
 }