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, #each_with_index, 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, #filter_map, #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, #tally, #to_h, #uniq, #zip

Constructor Details

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

Returns a new instance of Lazy.



158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
# File 'opal/opal/corelib/enumerator.rb', line 158

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



184
185
186
187
188
189
190
191
192
193
194
195
196
# File 'opal/opal/corelib/enumerator.rb', line 184

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



198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
# File 'opal/opal/corelib/enumerator.rb', line 198

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



224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
# File 'opal/opal/corelib/enumerator.rb', line 224

def drop(n)
  n = `$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



248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
# File 'opal/opal/corelib/enumerator.rb', line 248

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 ($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



271
272
273
# File 'opal/opal/corelib/enumerator.rb', line 271

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

#filterObject



275
# File 'opal/opal/corelib/enumerator.rb', line 275

alias filter find_all

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



277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
# File 'opal/opal/corelib/enumerator.rb', line 277

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 ($truthy(value)) {
        #{enum.yield(*args)};
      }
    }
  end
end

#grep(pattern, &block) ⇒ Object



295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
# File 'opal/opal/corelib/enumerator.rb', line 295

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

        if ($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 ($truthy(value)) {
          #{enum.yield `param`};
        }
      }
    end
  end
end

#inspectObject



389
390
391
# File 'opal/opal/corelib/enumerator.rb', line 389

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

#lazyObject



180
181
182
# File 'opal/opal/corelib/enumerator.rb', line 180

def lazy
  self
end

#reject(&block) ⇒ Object



327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
# File 'opal/opal/corelib/enumerator.rb', line 327

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 ($falsy(value)) {
        #{enum.yield(*args)};
      }
    }
  end
end

#take(n) ⇒ Object



343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
# File 'opal/opal/corelib/enumerator.rb', line 343

def take(n)
  n = `$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



368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
# File 'opal/opal/corelib/enumerator.rb', line 368

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 ($truthy(value)) {
        #{enum.yield(*args)};
      }
      else {
        #{raise StopLazyError};
      }
    }
  end
end