Просмотр архива ROTOR 14.1.1


<?php declare(strict_types=1);

/**
 * This file is part of the Nette Framework (https://nette.org)
 * Copyright (c) 2004 David Grudl (https://davidgrudl.com)
 */

namespace NetteIterators;

use Nette;


/**
 * Enhanced caching iterator with first/last/counter tracking.
 *
 * @template TKey
 * @template TValue
 * @extends CachingIterator<TKey, TValue, Iterator<TKey, TValue>>
 * @property-read bool $first
 * @property-read bool $last
 * @property-read bool $empty
 * @property-read bool $odd
 * @property-read bool $even
 * @property-read int $counter
 * @property-read TKey $nextKey
 * @property-read TValue $nextValue
 */
class CachingIterator extends CachingIterator implements Countable
{
    use NetteSmartObject;

    private int $counter = 0;


    /** @param  iterable<TKey, TValue>|stdClass  $iterable */
    public function __construct(iterable|stdClass $iterable)
    {
        $iterable = $iterable instanceof stdClass
            ? new ArrayIterator((array) $iterable)
            : NetteUtilsIterables::toIterator($iterable);
        parent::__construct($iterable, 0);
    }


    /**
     * Is the current element the first one?
     */
    public function isFirst(?int $gridWidth = null): bool
    {
        return $this->counter === 1 || ($gridWidth && $this->counter !== 0 && (($this->counter - 1) % $gridWidth) === 0);
    }


    /**
     * Is the current element the last one?
     */
    public function isLast(?int $gridWidth = null): bool
    {
        return !$this->hasNext() || ($gridWidth && ($this->counter % $gridWidth) === 0);
    }


    public function isEmpty(): bool
    {
        return $this->counter === 0;
    }


    public function isOdd(): bool
    {
        return $this->counter % 2 === 1;
    }


    public function isEven(): bool
    {
        return $this->counter % 2 === 0;
    }


    public function getCounter(): int
    {
        return $this->counter;
    }


    public function count(): int
    {
        $inner = $this->getInnerIterator();
        if ($inner instanceof Countable) {
            return $inner->count();

        } else {
            throw new NetteNotSupportedException('Iterator is not countable.');
        }
    }


    /**
     * Forwards to the next element.
     */
    public function next(): void
    {
        parent::next();
        if (parent::valid()) {
            $this->counter++;
        }
    }


    /**
     * Rewinds the Iterator.
     */
    public function rewind(): void
    {
        parent::rewind();
        $this->counter = parent::valid() ? 1 : 0;
    }


    /** @return TKey */
    public function getNextKey(): mixed
    {
        return $this->getInnerIterator()->key();
    }


    /** @return TValue */
    public function getNextValue(): mixed
    {
        return $this->getInnerIterator()->current();
    }
}