如何为 Range 创建自定义迭代器

Question

我想创建 Range 的子类，以便指定 1 以外的步长，这样我就可以执行以下操作：

>> a = RangeWithStepSize.new(-1, 2, 0.5).each {|x| puts(x)}
-1.0
-0.5
0.0
0.5
1.0
1.5
2.0
=> -1..2

我第一次尝试实现不起作用：

 class RangeWithStepSize < Range

  attr_reader :step_size

  def initialize(start_v, end_v, step_size, exclusive = false)
    super(start_v, end_v, exclusive)
    @step_size = step_size
  end

  def each
    self.step(step_size).each
  end

end

>> a = RangeWithStepSize.new(-1, 2, 0.5).each {|x| puts(x)}
=> #<Enumerator: [-1.0, -0.5, 0.0, 0.5, 1.0, 1.5, 2.0]:each>

看起来 RangeWithStepSize#each 返回一个有效的枚举器，但它没有枚举。知道为什么吗？

这可能是相关的，但我注意到没有块的 Range#step 不会返回文档中指定的枚举器；它返回一个数组：

>> Range.new(-1, 2).step(0.5).class
=> Array

数组是可枚举的，但它不是枚举器。这是一个文档错误吗？

澄清

我想制作一个封装步长的 Range 版本，这样我就可以这样做：

a = RangeWithStepSize(-1, 2, 0.5)
b = RangeWithStepSize(-1, 2, 0.25)

...所以枚举 a 会产生一个步长大小为 0.5，b 产生 0.25。

Answer 1

你知道你可以做到这一点，对吧？这里不需要继承。

(-1..2).step(0.5) do |x|
    puts x
end

您的代码只需进行一些小的调整即可工作：

class RangeWithStepSize < Range

  attr_reader :step_size

  def initialize(start_v, end_v, step_size, exclusive = false)
    super(start_v, end_v, exclusive)
    @step_size = step_size
  end

  def each (&block)
    self.step(step_size).each(&block)
  end

end