Class data_iterator

cpp/nd/iterator.hpp:17–72

class array;

class data_iterator
{
public:
    data_iterator(std::shared_ptr<void> owner, std::span<const uint8_t> data, icm::shape shape, dtype dt)
        : owner_(std::move(owner))
        , data_(data)
        , shape_(std::move(shape))
        , dtype_(dt)
    {
        ASSERT(std::accumulate(shape_.begin(), shape_.end(), 1, std::multiplies{}) * dtype_bytes(dtype_) ==
               data.size());
    }

    array operator*() const;

    void operator++()
    {
        data_ = std::span<const uint8_t>(data_.data() + data_.size(), data_.size());
    }

    void operator--()
    {
        data_ = std::span<const uint8_t>(data_.data() - data_.size(), data_.size());
    }

    void operator+=(std::ptrdiff_t n)
    {
        data_ = std::span<const uint8_t>(data_.data() + n * data_.size(), data_.size());
    }

    void operator-=(std::ptrdiff_t n)
    {
        data_ = std::span<const uint8_t>(data_.data() - n * data_.size(), data_.size());
    }

    std::ptrdiff_t operator-(const data_iterator& other) const
    {
        return (data_.data() - other.data_.data()) / data_.size();
    }

    bool operator==(const data_iterator& other) const
    {
        return data_.data() == other.data_.data();
    }

    bool operator!=(const data_iterator& other) const
    {
        return data_.data() != other.data_.data();
    }

private:
    std::shared_ptr<void> owner_;
    std::span<const uint8_t> data_;
    icm::shape shape_;
    dtype dtype_;
};

using dynamic_vector_iterator = std::vector<array>::const_iterator;