Merge 36e4352 into dc817d7

.php-cs-fixer.dist.php

@@ -57,6 +57,7 @@
         ],
         'php_unit_test_case_static_method_calls' => false,
         'yoda_style' => true,
+        'random_api_migration' => false,
     ])
     ->setFinder(
         PhpCsFixer\Finder::create()

src/Standard.php

@@ -37,6 +37,8 @@
 use Iterator;
 use function iterator_to_array;
 use IteratorAggregate;
+use function mt_getrandmax;
+use function mt_rand;
 use Traversable;
 
 /**
@@ -470,8 +472,9 @@ private static function skip(Iterator $input, int $skip): Generator
      */
     private static function take(Generator $input, int $take): Generator
     {
-        foreach ($input as $key => $value) {
-            yield $key => $value;
+        while ($input->valid()) {
+            yield $input->key() => $input->current();
+            $input->next();
 
             // Stop once taken enough.
             if (0 === --$take) {
@@ -578,4 +581,117 @@ private static function toIterators(iterable ...$inputs): array
             return new ArrayIterator($input);
         }, $inputs);
     }
+
+    /**
+     * Reservoir sampling method with an optional weighting function. Uses the most optimal algorithm.
+     *
+     * @see https://en.wikipedia.org/wiki/Reservoir_sampling
+     *
+     * @param int       $size       The desired sample size
+     * @param ?callable $weightFunc The optional weighting function
+     */
+    public function reservoir(int $size, ?callable $weightFunc = null): array
+    {
+        if (null === $this->pipeline) {
+            return [];
+        }
+
+        if ($size <= 0) {
+            return [];
+        }
+
+        // Algorithms below assume inputs are non-rewindable
+        $this->pipeline = self::makeNonRewindable($this->pipeline);
+
+        $result = null === $weightFunc ?
+            self::reservoirRandom($this->pipeline, $size) :
+            self::reservoirWeighted($this->pipeline, $size, $weightFunc);
+
+        return iterator_to_array($result, true);
+    }
+
+    private static function drainValues(Generator $input): Generator
+    {
+        while ($input->valid()) {
+            yield $input->current();
+            $input->next();
+        }
+    }
+
+    /**
+     * Simple and slow algorithm, commonly known as Algorithm R.
+     *
+     * @see https://en.wikipedia.org/wiki/Reservoir_sampling#Simple_algorithm
+     * @psalm-param positive-int $size
+     */
+    private static function reservoirRandom(Generator $input, int $size): Generator
+    {
+        // Take an initial sample (AKA fill the reservoir array)
+        foreach (self::take($input, $size) as $output) {
+            yield $output;
+        }
+
+        // Return if there's nothing more to fetch
+        if (!$input->valid()) {
+            return;
+        }
+
+        $counter = $size;
+
+        // Produce replacement elements with gradually decreasing probability
+        foreach (self::drainValues($input) as $value) {
+            $key = mt_rand(0, $counter);
+
+            if ($key < $size) {
+                yield $key => $value;
+            }
+
+            ++$counter;
+        }
+    }
+
+    /**
+     * Weighted random sampling.
+     *
+     * @see https://en.wikipedia.org/wiki/Reservoir_sampling#Algorithm_A-Chao
+     * @psalm-param positive-int $size
+     */
+    private static function reservoirWeighted(Generator $input, int $size, callable $weightFunc): Generator
+    {
+        $sum = 0.0;
+
+        // Take an initial sample (AKA fill the reservoir array)
+        foreach (self::take($input, $size) as $output) {
+            yield $output;
+            $sum += $weightFunc($output);
+        }
+
+        // Return if there's nothing more to fetch
+        if (!$input->valid()) {
+            return;
+        }
+
+        foreach (self::drainValues($input) as $value) {
+            $weight = $weightFunc($value);
+            $sum += $weight;
+
+            // probability for this item
+            $probability = $weight / $sum;
+
+            // @infection-ignore-all
+            if (self::random() <= $probability) {
+                yield mt_rand(0, $size - 1) => $value;
+            }
+        }
+    }
+
+    /**
+     * Returns a pseudorandom value between zero (inclusive) and one (exclusive).
+     *
+     * @infection-ignore-all
+     */
+    private static function random(): float
+    {
+        return mt_rand(0, mt_getrandmax() - 1) / mt_getrandmax();
+    }
 }

tests/ReservoirTest.php

@@ -0,0 +1,183 @@
+<?php
+/**
+ * Copyright 2017, 2018 Alexey Kopytko <alexey@kopytko.com>
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+declare(strict_types=1);
+
+namespace Tests\Pipeline;
+
+use function abs;
+use ArrayIterator;
+use IteratorIterator;
+use function mt_rand;
+use function mt_srand;
+use function ord;
+use PHPUnit\Framework\TestCase;
+use function Pipeline\map;
+use Pipeline\Standard;
+use function Pipeline\take;
+use function range;
+use function sin;
+
+/**
+ * @covers \Pipeline\Standard
+ *
+ * @internal
+ */
+final class ReservoirTest extends TestCase
+{
+    protected function setUp(): void
+    {
+        mt_srand(0);
+    }
+
+    public function testRandomSeed(): void
+    {
+        $this->assertSame(
+            [20, 17, 11, 13, 18, 13],
+            take(range(0, 5))->map(function () {
+                return mt_rand(10, 20);
+            })->toArray()
+        );
+    }
+
+    public function testNoop(): void
+    {
+        $pipeline = new Standard();
+        $this->assertSame([], $pipeline->reservoir(1000));
+    }
+
+    public function provideInputs(): iterable
+    {
+        yield 'no change expected' => [['a', 'b', 'c'], 3, ['a', 'b', 'c']];
+
+        yield [['a', 'b', 'c'], -1, []];
+
+        yield [['a', 'b', 'c'], 0, []];
+
+        yield [['a', 'b', 'c'], 1, ['c']];
+
+        yield [['a', 'b', 'c'], 2, ['a', 'b']];
+
+        yield [['a', 'b', 'c'], 4, ['a', 'b', 'c']];
+
+        yield [['a', 'b', 'c', 'd', 'e', 'f'], 2, ['f', 'b']];
+
+        yield [['a', 'b', 'c', 'd', 'e', 'f'], 3, ['d', 'b', 'c']];
+
+        yield [['a', 'b', 'c', 'd', 'e', 'f'], 4, ['a', 'b', 'c', 'f']];
+
+        yield 'a thousand of numbers' => [range(0, 1000), 10, [
+            838,
+            96,
+            381,
+            971,
+            87,
+            715,
+            589,
+            168,
+            693,
+            366,
+        ]];
+    }
+
+    /**
+     * @dataProvider provideInputs
+     */
+    public function testSampleFromGenerator(array $input, int $size, array $expected): void
+    {
+        $this->assertSame($expected, map(static function () use ($input) {
+            yield from $input;
+        })->reservoir($size));
+    }
+
+    /**
+     * @dataProvider provideInputs
+     */
+    public function testSampleFromArray(array $input, int $size, array $expected): void
+    {
+        $this->assertSame($expected, take($input)->reservoir($size));
+    }
+
+    /**
+     * @dataProvider provideInputs
+     */
+    public function testSampleFromIterator(array $input, int $size, array $expected): void
+    {
+        $input = new IteratorIterator(new ArrayIterator($input));
+
+        $this->assertSame($expected, take($input)->reservoir($size));
+    }
+
+    public function provideWeightedInputs(): iterable
+    {
+        $weightFn = static function (string $input): float {
+            return abs(sin(ord($input[0])));
+        };
+
+        yield 'no change expected' => [['a', 'b', 'c'], 3, $weightFn, ['a', 'b', 'c']];
+
+        yield [['a', 'b', 'c'], -1, $weightFn, []];
+
+        yield [['a', 'b', 'c'], 0, $weightFn, []];
+
+        yield [['a', 'b', 'c'], 1, $weightFn, ['c']];
+
+        yield [['a', 'b', 'c'], 2, $weightFn, ['a', 'c']];
+
+        yield [['a', 'b', 'c'], 4, $weightFn, ['a', 'b', 'c']];
+
+        $weightFnInt = static function (int $input): float {
+            return abs(sin($input / 1000));
+        };
+
+        yield 'a thousand of numbers' => [range(0, 1000), 5, $weightFnInt, [
+            437,
+            1,
+            358,
+            240,
+            197,
+        ]];
+    }
+
+    /**
+     * @dataProvider provideWeightedInputs
+     */
+    public function testWeightedSampleFromGenerator(array $input, int $size, callable $weightFn, array $expected): void
+    {
+        $this->assertSame($expected, map(static function () use ($input) {
+            yield from $input;
+        })->reservoir($size, $weightFn));
+    }
+
+    /**
+     * @dataProvider provideWeightedInputs
+     */
+    public function testWeightedSampleFromArray(array $input, int $size, callable $weightFn, array $expected): void
+    {
+        $this->assertSame($expected, take($input)->reservoir($size, $weightFn));
+    }
+
+    /**
+     * @dataProvider provideWeightedInputs
+     */
+    public function testWeightedSampleFromIterator(array $input, int $size, callable $weightFn, array $expected): void
+    {
+        $input = new IteratorIterator(new ArrayIterator($input));
+
+        $this->assertSame($expected, take($input)->reservoir($size, $weightFn));
+    }
+}