From 6655b3471d5e8e1a6bf757d115fbd9f3c23c5ad0 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Fri, 25 Mar 2016 16:55:43 +0000
Subject: Add lazyMap implementation

---
 .../mammoth/internal/util/MammothIterables.java    | 25 ++++++++++++++++++++++
 1 file changed, 25 insertions(+)

(limited to 'src/main/java/org/zwobble/mammoth/internal/util/MammothIterables.java')

diff --git a/src/main/java/org/zwobble/mammoth/internal/util/MammothIterables.java b/src/main/java/org/zwobble/mammoth/internal/util/MammothIterables.java
index 4335dc1..0280ac9 100644
--- a/src/main/java/org/zwobble/mammoth/internal/util/MammothIterables.java
+++ b/src/main/java/org/zwobble/mammoth/internal/util/MammothIterables.java
@@ -2,7 +2,9 @@ package org.zwobble.mammoth.internal.util;
 
 import com.google.common.collect.Iterables;
 
+import java.util.Iterator;
 import java.util.Optional;
+import java.util.function.Function;
 import java.util.function.Predicate;
 
 public class MammothIterables {
@@ -18,4 +20,27 @@ public class MammothIterables {
         }
         return Optional.empty();
     }
+
+    public static <T, R> Iterable<R> lazyMap(Iterable<T> iterable, Function<T, R> function) {
+        return new Iterable<R>() {
+            @Override
+            public Iterator<R> iterator() {
+                return map(iterable.iterator(), function);
+            }
+        };
+    }
+
+    private static <T, R> Iterator<R> map(Iterator<T> iterator, Function<T, R> function) {
+        return new Iterator<R>() {
+            @Override
+            public boolean hasNext() {
+                return iterator.hasNext();
+            }
+
+            @Override
+            public R next() {
+                return function.apply(iterator.next());
+            }
+        };
+    }
 }
-- 
cgit v1.2.3