Class: Enumerator::Lazy

Inherits:
Enumerator show all
Defined in:
opal/opal/corelib/enumerator.rb

Defined Under Namespace

Classes: StopLazyError

Instance Method Summary collapse

Methods inherited from Enumerator

#each, for, #size, #with_index

Methods included from Enumerable

#all?, #any?, #chunk, #chunk_while, #count, #cycle, #detect, #each_cons, #each_entry, #each_slice, #each_with_index, #each_with_object, #entries, #enumerator_size, #find_index, #first, #grep_v, #group_by, #include?, #inject, #max, #max_by, #min, #min_by, #minmax, #minmax_by, #none?, #one?, #partition, #reverse_each, #slice_after, #slice_before, #slice_when, #sort, #sort_by, #sum, #uniq, #zip

Constructor Details

#initialize(object, size = nil, &block) ⇒ Lazy

Returns a new instance of Lazy



149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
# File 'opal/opal/corelib/enumerator.rb', line 149

def initialize(object, size = nil, &block)
  unless block_given?
    raise ArgumentError, 'tried to call lazy new without a block'
  end

  @enumerator = object

  super size do |yielder, *each_args|
    object.each(*each_args) do |*args|
      %x{
        args.unshift(#{yielder});

        Opal.yieldX(block, args);
      }
    end
  rescue Exception
    nil
  end
end

Instance Method Details

#collect(&block) ⇒ Object Also known as: map



175
176
177
178
179
180
181
182
183
184
185
186
187
# File 'opal/opal/corelib/enumerator.rb', line 175

def collect(&block)
  unless block
    raise ArgumentError, 'tried to call lazy map without a block'
  end

  Lazy.new(self, enumerator_size) do |enum, *args|
    %x{
      var value = Opal.yieldX(block, args);

      #{enum.yield `value`};
    }
  end
end

#collect_concat(&block) ⇒ Object Also known as: flat_map



189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
# File 'opal/opal/corelib/enumerator.rb', line 189

def collect_concat(&block)
  unless block
    raise ArgumentError, 'tried to call lazy map without a block'
  end

  Lazy.new(self, nil) do |enum, *args|
    %x{
      var value = Opal.yieldX(block, args);

      if (#{`value`.respond_to? :force} && #{`value`.respond_to? :each}) {
        #{`value`.each { |v| enum.yield v }}
      }
      else {
        var array = #{Opal.try_convert `value`, Array, :to_ary};

        if (array === nil) {
          #{enum.yield `value`};
        }
        else {
          #{`value`.each { |v| enum.yield v }};
        }
      }
    }
  end
end

#drop(n) ⇒ Object



215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
# File 'opal/opal/corelib/enumerator.rb', line 215

def drop(n)
  n = Opal.coerce_to n, Integer, :to_int

  if n < 0
    raise ArgumentError, 'attempt to drop negative size'
  end

  current_size = enumerator_size
  set_size     = if Integer === current_size
                   n < current_size ? n : current_size
                 else
                   current_size
                 end

  dropped = 0
  Lazy.new(self, set_size) do |enum, *args|
    if dropped < n
      dropped += 1
    else
      enum.yield(*args)
    end
  end
end

#drop_while(&block) ⇒ Object



239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
# File 'opal/opal/corelib/enumerator.rb', line 239

def drop_while(&block)
  unless block
    raise ArgumentError, 'tried to call lazy drop_while without a block'
  end

  succeeding = true
  Lazy.new(self, nil) do |enum, *args|
    if succeeding
      %x{
        var value = Opal.yieldX(block, args);

        if (#{Opal.falsy?(`value`)}) {
          succeeding = false;

          #{enum.yield(*args)};
        }
      }
    else
      enum.yield(*args)
    end
  end
end

#enum_for(method = :each, *args, &block) ⇒ Object Also known as: to_enum



262
263
264
# File 'opal/opal/corelib/enumerator.rb', line 262

def enum_for(method = :each, *args, &block)
  self.class.for(self, method, *args, &block)
end

#find_all(&block) ⇒ Object Also known as: select



266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
# File 'opal/opal/corelib/enumerator.rb', line 266

def find_all(&block)
  unless block
    raise ArgumentError, 'tried to call lazy select without a block'
  end

  Lazy.new(self, nil) do |enum, *args|
    %x{
      var value = Opal.yieldX(block, args);

      if (#{Opal.truthy?(`value`)}) {
        #{enum.yield(*args)};
      }
    }
  end
end

#grep(pattern, &block) ⇒ Object



284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
# File 'opal/opal/corelib/enumerator.rb', line 284

def grep(pattern, &block)
  if block
    Lazy.new(self, nil) do |enum, *args|
      %x{
        var param = #{Opal.destructure(args)},
            value = #{pattern === `param`};

        if (#{Opal.truthy?(`value`)}) {
          value = Opal.yield1(block, param);

          #{enum.yield `Opal.yield1(block, param)`};
        }
      }
    end
  else
    Lazy.new(self, nil) do |enum, *args|
      %x{
        var param = #{Opal.destructure(args)},
            value = #{pattern === `param`};

        if (#{Opal.truthy?(`value`)}) {
          #{enum.yield `param`};
        }
      }
    end
  end
end

#inspectObject



378
379
380
# File 'opal/opal/corelib/enumerator.rb', line 378

def inspect
  "#<#{self.class}: #{@enumerator.inspect}>"
end

#lazyObject



171
172
173
# File 'opal/opal/corelib/enumerator.rb', line 171

def lazy
  self
end

#reject(&block) ⇒ Object



316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
# File 'opal/opal/corelib/enumerator.rb', line 316

def reject(&block)
  unless block
    raise ArgumentError, 'tried to call lazy reject without a block'
  end

  Lazy.new(self, nil) do |enum, *args|
    %x{
      var value = Opal.yieldX(block, args);

      if (#{Opal.falsy?(`value`)}) {
        #{enum.yield(*args)};
      }
    }
  end
end

#take(n) ⇒ Object



332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
# File 'opal/opal/corelib/enumerator.rb', line 332

def take(n)
  n = Opal.coerce_to n, Integer, :to_int

  if n < 0
    raise ArgumentError, 'attempt to take negative size'
  end

  current_size = enumerator_size
  set_size     = if Integer === current_size
                   n < current_size ? n : current_size
                 else
                   current_size
                 end

  taken = 0
  Lazy.new(self, set_size) do |enum, *args|
    if taken < n
      enum.yield(*args)
      taken += 1
    else
      raise StopLazyError
    end
  end
end

#take_while(&block) ⇒ Object



357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
# File 'opal/opal/corelib/enumerator.rb', line 357

def take_while(&block)
  unless block
    raise ArgumentError, 'tried to call lazy take_while without a block'
  end

  Lazy.new(self, nil) do |enum, *args|
    %x{
      var value = Opal.yieldX(block, args);

      if (#{Opal.truthy?(`value`)}) {
        #{enum.yield(*args)};
      }
      else {
        #{raise StopLazyError};
      }
    }
  end
end