Iter.php

<?php

namespace MathPHP\Util;

use MathPHP\Exception;

/**
 * @internal
 */
class Iter
{
    /**
     * Zip - Make an iterator that aggregates items from multiple iterators
     * Similar to Python's zip function
     * @internal
     *
     * @param iterable ...$iterables
     *
     * @return \MultipleIterator
     */
    public static function zip(iterable ...$iterables): \MultipleIterator
    {
        $zippedIterator = new \MultipleIterator();
        foreach ($iterables as $iterable) {
            $zippedIterator->attachIterator(self::makeIterator($iterable));
        }

        return $zippedIterator;
    }

    /**
     * @param iterable $iterable
     *
     * @return \Iterator|\IteratorIterator|\ArrayIterator
     */
    public static function makeIterator(iterable $iterable): \Iterator
    {
        switch (true) {
            case $iterable instanceof \Iterator:
                return $iterable;

            case $iterable instanceof \Traversable:
                return new \IteratorIterator($iterable);

            case \is_array($iterable):
                return new \ArrayIterator($iterable);
        }

        throw new \LogicException(\gettype($iterable) . ' type is not an expected iterable type (Iterator|Traversable|array)');
    }
}