deferred_enum 0.0.1 → 0.0.2

data/deferred_enum.gemspec

@@ -10,7 +10,7 @@ Gem::Specification.new do |s|
   s.homepage    = "https://github.com/take-five/deferred_enum"
   s.summary     = %q{Introduces lazy computations to Enumerable}
   s.description = File.read(File.expand_path('../README.rdoc', __FILE__))
-  s.date        = Time.now.strftime('%Y-%m-%d')
+  s.date        = "2011-12-15"
 
   s.rubyforge_project = "deferred_enum"
 

data/lib/deferred_enum/map.rb

@@ -0,0 +1,12 @@
+class DeferredEnumerator::Map < DeferredEnumerator #:nodoc:all
+  def initialize(obj, map = nil)
+    @_map = map
+    super(obj)
+  end
+
+  def each
+    return super unless block_given?
+
+    super { |e| yield(@_map ? @_map.call(e) : e) }
+  end
+end

data/lib/deferred_enum/select.rb

@@ -0,0 +1,12 @@
+class DeferredEnumerator::Select < DeferredEnumerator #:nodoc:all
+  def initialize(obj, filter = nil)
+    @_filter = filter
+    super(obj)
+  end
+
+  def each
+    return super unless block_given?
+
+    super { |e| yield(e) if !@_filter || @_filter.call(e) }
+  end
+end

data/lib/deferred_enum/version.rb

@@ -1,3 +1,3 @@
 module DeferredEnum
-  VERSION = "0.0.1"
+  VERSION = "0.0.2"
 end

data/lib/deferred_enumerable.rb

@@ -27,32 +27,26 @@ module DeferredEnumerable
   end
 
   # Returns a new enumerator with the results of running block once for every element in enum.
-  def collect # :yields: obj
+  def collect(&block) # :yields: obj
     return self unless block_given?
 
-    DeferredEnumerator.new do |yielder|
-      each { |entry| yielder << yield(entry) }
-    end
+    DeferredEnumerator::Map.new(self, block)
   end
   alias map collect
 
   # Returns a enumerator containing all elements of enum for which block is not false
-  def select # :yields: obj
+  def select(&block) # :yields: obj
     return self unless block_given?
 
-    DeferredEnumerator.new do |yielder|
-      each { |entry| yielder << entry if yield(entry) }
-    end
+    DeferredEnumerator::Select.new(self, block)
   end
   alias find_all select
 
   # Returns a enumerator containing all elements of enum for which block is false
-  def reject # :yields: obj
+  def reject(&block) # :yields: obj
     return self unless block_given?
 
-    DeferredEnumerator.new do |yielder|
-      each { |entry| yielder << entry unless yield(entry) }
-    end
+    select { |e| !block.call(e) }
   end
 
   # Returns a new enumerator that is a one-dimensional flattening of this enumerator (recursively).
@@ -125,7 +119,7 @@ module DeferredEnumerable
     raise TypeError, 'Integer (> 0) expected' unless n.is_a?(Fixnum) && n > 0
 
     DeferredEnumerator.new do |yielder|
-      each { |entry| yielder << entry if (n -= 1) >= 0}
+      each { |entry| yielder << entry if (n -= 1) >= 0 }
     end
   end
 

data/lib/deferred_enumerator.rb

@@ -3,4 +3,7 @@ require "deferred_enum/enumerable"
 
 class DeferredEnumerator < Enumerator
   include ::DeferredEnumerable
+
+  require "deferred_enum/map"
+  require "deferred_enum/select"
 end

metadata

@@ -5,8 +5,8 @@ version: !ruby/object:Gem::Version
   segments: 
   - 0
   - 0
-  - 1
-  version: 0.0.1
+  - 2
+  version: 0.0.2
 platform: ruby
 authors: 
 - Alexey Mikhaylov
@@ -14,7 +14,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2011-12-11 00:00:00 +06:00
+date: 2011-12-15 00:00:00 +06:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency 
@@ -82,6 +82,8 @@ files:
 - deferred_enum.gemspec
 - lib/deferred_enum.rb
 - lib/deferred_enum/enumerable.rb
+- lib/deferred_enum/map.rb
+- lib/deferred_enum/select.rb
 - lib/deferred_enum/version.rb
 - lib/deferred_enumerable.rb
 - lib/deferred_enumerator.rb