From c25450f6288748782ad60cc7c4b8e0e0bdc52c1c Mon Sep 17 00:00:00 2001
From: Carlo Zancanaro <carlo@zancanaro.id.au>
Date: Sun, 31 May 2015 13:53:08 +1000
Subject: Better code, based off ParentRunner among other things

---
 src/main/java/au/id/zancanaro/Generators.java | 15 +++++++++++----
 1 file changed, 11 insertions(+), 4 deletions(-)

(limited to 'src/main/java/au/id/zancanaro/Generators.java')

diff --git a/src/main/java/au/id/zancanaro/Generators.java b/src/main/java/au/id/zancanaro/Generators.java
index f065d1d..5b025fa 100644
--- a/src/main/java/au/id/zancanaro/Generators.java
+++ b/src/main/java/au/id/zancanaro/Generators.java
@@ -1,14 +1,21 @@
 package au.id.zancanaro;
 
 import java.util.Iterator;
+import java.util.function.Predicate;
 
 public class Generators {
-    @SafeVarargs
-    public static <T> Generator<T[]> arrayGenerator(Generator<? extends T>... generators) {
-        return Generator.tuple((Generator<T>[]) generators);
+    public static <T> Generator<T> suchThat(Generator<T> gen, Predicate<T> pred) {
+        return (random, size) -> {
+            RoseTree<T> result = gen.generate(random, size);
+            if (pred.test(result.getValue())) {
+                return result.filter(pred);
+            } else {
+                return suchThat(gen, pred).generate(random, size);
+            }
+        };
     }
 
-    public static Generator<Integer> integerGenerator() {
+    public static Generator<Integer> integer() {
         return (random, size) -> {
             int value = random.nextInt(size);
             return new RoseTree<>(value, intShrinkingIterable(value));
-- 
cgit v1.2.3