zicht / itertools

use Zicht\Itertools\lib\ChainIterator;

trait ChainTrait

{

    /**

     * Returns an iterator that returns elements from the first iterable

     * until it is exhausted, then proceeds to the next iterable, until

     * all the iterables are exhausted.

     *

     * Used for creating consecutive sequences as a single sequence.

     *

     * Note that the keys of the returned ChainIterator follow 0, 1, etc,

     * regardless of the keys given in the iterables.

     *

     * > iter\iterable([1, 2, 3], [4, 5, 6])->chain()

     * 1 2 3 4 5 6

     *

     * > iter\iterable('ABC', 'DEF')->chain()

     * A B C D E F

     *

     * @param array|string|\Iterator $iterable

     * @param array|string|\Iterator $iterable2

     * @return null|ChainIterator

     */

    public function chain(/* $iterable, $iterable2, ... */)

    {

        if ($this instanceof \Iterator) {

            $iterables = array_map(

                '\Zicht\Itertools\conversions\mixed_to_iterator',

                func_get_args()

            );

            $reflectorClass = new \ReflectionClass('\Zicht\Itertools\lib\ChainIterator');

            return $reflectorClass->newInstanceArgs(array_merge([$this], $iterables));

        }

        return null;

    }

}

use Zicht\Itertools\lib\ZipIterator;

trait ZipTrait

{

    /**

     * Returns an iterator where one or more iterables are zipped together

     *

     * This function returns a list of tuples, where the i-th tuple contains

     * the i-th element from each of the argument sequences or iterables.

     *

     * The returned list is truncated in length to the length of the

     * shortest argument sequence.

     *

     * > zip([1, 2, 3], ['a', 'b', 'c'])

     * [1, 'a'] [2, 'b'] [3, 'c']

     *

     * @param array|string|\Iterator $iterable

     * @param array|string|\Iterator $iterable2

     * @return ZipIterator

     */

    public function zip(/* $iterable, $iterable2, ... */)

    {

        if ($this instanceof \Iterator) {

            $iterables = array_map(

                '\Zicht\Itertools\conversions\mixed_to_iterator',

                func_get_args()

            );

            $reflectorClass = new \ReflectionClass('\Zicht\Itertools\lib\ZipIterator');

            return $reflectorClass->newInstanceArgs(array_merge([$this], $iterables));

        }

        return null;

    }

}