xenum 0.1.1 → 0.1.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 6bbf9d98e7579cf718f304fdde5a13459d5c3b225383d48d20fba38bc25d6f59
-  data.tar.gz: 5d448b96d97c4438fbb491ebfe89e8aee222977c0df42edb568972ef41cb1ccb
+  metadata.gz: 6702efc29813e62dabd6f74d9c7bc8067b191a79f3b416f6a1336ba20e6497d4
+  data.tar.gz: 94676ca2107d67c7680d2e6b963e67c5e19f7ab8386c9808e96c751dcafe34ee
 SHA512:
-  metadata.gz: be579d94a4ce9c11aac9e2be34c2542e92e6c6fef2d7624e662cbfcc5f8f277262d58094bb5b245d44d6e6949a85811faf081f1ffed2592dd3bdc8e62d907565
-  data.tar.gz: 6bdabc0c8e78e6dd35ee33d7af734cc15b86bf66e6d40e26b14804900d211dae1ddbd53c8cc86904cd2c90e723bafb55a3d36083551c84154f5feb4f379a497e
+  metadata.gz: f82eac1237b52b16d7950bcaee008811a71ad905e966141893a13a51d04aa8ea9590d47590099f2eb204ec5d2df1a3b75a26db1e3ad5d0a6c159c56bf84776f2
+  data.tar.gz: 71df4dec9373651f55a6b5f9abbd4b74f0778b6be2726de4a69735886c0b5528c4e47a46f9d65fb7119b4c3ffa0ee655ce26b092273dfd1ec4b9eedb2bd23b81

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    xenum (0.1.0)
+    xenum (0.1.3)
 
 GEM
   remote: https://rubygems.org/

data/README.md

@@ -30,6 +30,18 @@ works like `Array#flatten`, but for all `Enumerable`, and it is lazy
 # [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14]
 ```
 
+### lazy_insert
+
+works like `Array#insert`, but not limited to `Array`
+
+```ruby
+10.times.lazy_insert(3, 'a', 'b').to_a
+# [0, 1, 2, "a", "b", 3, 4, 5, 6, 7, 8, 9]
+
+10.times.lazy_insert(-3, 'a', 'b').to_a
+# [0, 1, 2, 3, 4, 5, 6, 7, "a", "b", 8, 9]
+```
+
 ### lazy_product
 
 works like `Array#product`, but not limited to `Array`
@@ -55,4 +67,8 @@ e.take(5)
 
 [(6..8), [3,4,5], 3.times].reduce(&:merge_sort).to_a
 # [0, 1, 2, 3, 4, 5, 6, 7, 8]
+
+# same result but faster
+(6..8).merge_sort([3,4,5], 3.times).to_a
+# [0, 1, 2, 3, 4, 5, 6, 7, 8]
 ```

data/lib/xenum/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Xenum
-  VERSION = "0.1.1"
+  VERSION = "0.1.3"
 end

data/lib/xenum.rb

@@ -26,6 +26,16 @@ module Enumerable
     end
   end
 
+  def lazy_insert(index, *objs)
+    return self if objs.empty?
+
+    if index >= 0
+      lazy_insert_pos(index, objs)
+    else
+      lazy_insert_neg(index, objs)
+    end
+  end
+
   def lazy_product(*enums)
     if enums.empty?
       return Enumerator.new do |yielder|
@@ -45,7 +55,24 @@ module Enumerable
     end
   end
 
-  def merge_sort(that)
+  def merge_sort(*enums)
+    case enums.size
+    when 0
+      self
+    when 1
+      _merge_sort(enums[0])
+    else
+      enums2 = enums.pop((enums.size / 2) - 1)
+      enum2 = enums.pop
+      e2 = enum2.merge_sort(*enums2)
+      e1 = merge_sort(*enums)
+      e1.merge_sort(e2)
+    end
+  end
+
+  private
+
+  def _merge_sort(that)
     this = Enumerator === self ? self : self.to_enum
     that = Enumerator === that ? that : that.to_enum
 
@@ -92,4 +119,46 @@ module Enumerable
       end
     end
   end
+
+  def lazy_insert_neg(index, objs)
+    these = Enumerator === self ? self : self.to_enum
+    queue = []
+
+    (index.abs - 1).times do
+      begin
+        queue << these.next
+      rescue StopIteration
+        raise IndexError, "index #{index} too small; minimum: #{-(queue.size + 1)}"
+      end
+    end
+
+    Enumerator.new do |yielder|
+      loop do
+        begin
+          queue << these.next
+          yielder << queue.shift
+        rescue StopIteration
+          break
+        end
+      end
+      objs.each{ |obj| yielder << obj }
+      queue.each{ |this| yielder << this }
+    end
+  end
+
+  def lazy_insert_pos(index, objs)
+    these = self
+    Enumerator.new do |yielder|
+      current_index = -1
+      these.each do |this|
+        current_index += 1
+        objs.each{ |obj| yielder << obj } if current_index == index
+        yielder << this
+      end
+      if current_index < index
+        ((current_index + 1)...index).each{ yielder << nil }
+        objs.each{ |obj| yielder << obj }
+      end
+    end
+  end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: xenum
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.3
 platform: ruby
 authors:
 - ken
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2024-03-15 00:00:00.000000000 Z
+date: 2024-03-19 00:00:00.000000000 Z
 dependencies: []
 description:
 email: