contiguous_iterator.hpp

Go to the documentation of this file.

// SPDX-License-Identifier: BSD-3-Clause
/* Copyright 2018-2021, Intel Corporation */
 
#ifndef LIBPMEMOBJ_CPP_CONTIGUOUS_ITERATOR_HPP
#define LIBPMEMOBJ_CPP_CONTIGUOUS_ITERATOR_HPP
 
#include <algorithm>
#include <cassert>
#include <functional>
 
#include <libpmemobj++/detail/common.hpp>
 
namespace pmem
{
 
namespace detail
{
 
template <typename Iterator, typename Reference, typename Pointer>
struct contiguous_iterator {
    constexpr contiguous_iterator(Pointer begin) : ptr(begin)
    {
    }
 
    Reference operator*() const
    {
        return *ptr;
    }
 
    Pointer operator->() const
    {
        return ptr;
    }
 
    Iterator &
    operator++()
    {
        static_cast<Iterator *>(this)->change_by(1);
        return *static_cast<Iterator *>(this);
    }
 
    Iterator
    operator++(int)
    {
        Iterator tmp(*static_cast<Iterator *>(this));
        static_cast<Iterator *>(this)->change_by(1);
        return tmp;
    }
 
    Iterator &
    operator--()
    {
        static_cast<Iterator *>(this)->change_by(-1);
        return *static_cast<Iterator *>(this);
    }
 
    Iterator
    operator--(int)
    {
        Iterator tmp(*static_cast<Iterator *>(this));
        static_cast<Iterator *>(this)->change_by(-1);
        return tmp;
    }
 
    Iterator &
    operator+=(std::ptrdiff_t n)
    {
        static_cast<Iterator *>(this)->change_by(n);
        return *static_cast<Iterator *>(this);
    }
 
    Iterator &
    operator-=(std::ptrdiff_t n)
    {
        static_cast<Iterator *>(this)->change_by(-n);
        return *static_cast<Iterator *>(this);
    }
 
    Iterator
    operator+(std::ptrdiff_t n) const
    {
        Iterator tmp(*static_cast<const Iterator *>(this));
        tmp += n;
        return tmp;
    }
 
    Iterator
    operator-(std::ptrdiff_t n) const
    {
        Iterator tmp(*static_cast<const Iterator *>(this));
        tmp -= n;
        return tmp;
    }
 
    friend std::ptrdiff_t
    operator-(const Iterator &lhs, const Iterator &rhs)
    {
        return lhs.ptr - rhs.ptr;
    }
 
    Reference operator[](std::ptrdiff_t n)
    {
        return ptr[n];
    }
 
    Pointer
    get_ptr() const
    {
        return ptr;
    }
 
protected:
    void
    change_by(std::ptrdiff_t n)
    {
        ptr += n;
    }
 
    Pointer ptr;
};
 
template <typename T>
struct range_snapshotting_iterator
    : public contiguous_iterator<range_snapshotting_iterator<T>, T &, T *> {
    using iterator_category = std::random_access_iterator_tag;
    using value_type = T;
    using difference_type = std::ptrdiff_t;
    using reference = T &;
    using pointer = T *;
    using base_type = contiguous_iterator<range_snapshotting_iterator<T>,
                          reference, pointer>;
 
    range_snapshotting_iterator(pointer ptr = nullptr,
                    pointer data = nullptr,
                    std::size_t size = 0,
                    std::size_t snapshot_size = 1)
        : base_type(ptr),
          data(data),
          size(size),
          snapshot_size(snapshot_size)
    {
        assert(data <= ptr);
 
        if (snapshot_size && ptr)
            snapshot_range(ptr);
    }
 
    operator const T *() const
    {
        return this->ptr;
    }
 
    reference operator[](std::ptrdiff_t n)
    {
        detail::conditional_add_to_tx(&this->ptr[n], 1,
                          POBJ_XADD_ASSUME_INITIALIZED);
        return base_type::operator[](n);
    }
 
    friend void
    swap(range_snapshotting_iterator &lhs, range_snapshotting_iterator &rhs)
    {
        std::swap(lhs.ptr, rhs.ptr);
        std::swap(lhs.data, rhs.data);
        std::swap(lhs.size, rhs.size);
        std::swap(lhs.snapshot_size, rhs.snapshot_size);
    }
 
    template <typename Iterator, typename Reference, typename Pointer>
    friend struct contiguous_iterator;
 
protected:
    void
    change_by(std::ptrdiff_t n)
    {
        conditional_snapshot_range(this->ptr, n);
        base_type::change_by(n);
    }
 
private:
    /*
     * Conditionally snapshot range of length snapshot_size,
     * which contain address equal to ptr + diff.
     */
    void
    conditional_snapshot_range(pointer ptr, difference_type diff)
    {
        if (snapshot_size == 0)
            return;
 
        auto new_ptr = ptr + diff;
 
        /* if new pointer is outside of the array */
        if (new_ptr < data || new_ptr >= data + size)
            return;
 
        /* if new pointer is in the same range */
        if (static_cast<std::size_t>(ptr - data) / snapshot_size ==
            static_cast<std::size_t>(new_ptr - data) / snapshot_size)
            return;
 
        snapshot_range(new_ptr);
    }
 
    void
    snapshot_range(pointer ptr)
    {
        /* align index to snapshot_size */
        auto range_begin =
            ptr - static_cast<uint64_t>(ptr - data) % snapshot_size;
        auto range_size = snapshot_size;
 
        if (range_begin + range_size > data + size)
            range_size = static_cast<uint64_t>(data + size -
                               range_begin);
#ifndef NDEBUG
        verify_range(range_begin, range_size);
#endif
 
        detail::conditional_add_to_tx(range_begin, range_size,
                          POBJ_XADD_ASSUME_INITIALIZED);
    }
 
#ifndef NDEBUG
    void
    verify_range(pointer range_begin, uint64_t range_size)
    {
        auto range_offset = static_cast<uint64_t>(range_begin - data);
 
        assert(range_begin >= data);
        assert(range_offset % snapshot_size == 0);
        assert((range_offset + range_size) % snapshot_size == 0 ||
               range_begin + range_size == data + size);
    }
#endif
 
    pointer data;
    std::size_t size;
    std::size_t snapshot_size;
};
 
template <typename T>
struct basic_contiguous_iterator
    : public contiguous_iterator<basic_contiguous_iterator<T>, T &, T *> {
    using iterator_category = std::random_access_iterator_tag;
    using value_type = T;
    using difference_type = std::ptrdiff_t;
    using reference = T &;
    using pointer = T *;
    using base_type = contiguous_iterator<basic_contiguous_iterator<T>,
                          reference, pointer>;
 
    basic_contiguous_iterator(pointer ptr = nullptr) : base_type(ptr)
    {
    }
 
    operator const T *() const
    {
        return this->ptr;
    }
 
    reference operator*() const
    {
        detail::conditional_add_to_tx(this->ptr, 1,
                          POBJ_XADD_ASSUME_INITIALIZED);
        return base_type::operator*();
    }
 
    pointer operator->() const
    {
        detail::conditional_add_to_tx(this->ptr, 1,
                          POBJ_XADD_ASSUME_INITIALIZED);
        return base_type::operator->();
    }
 
    reference operator[](std::ptrdiff_t n)
    {
        detail::conditional_add_to_tx(&this->ptr[n], 1,
                          POBJ_XADD_ASSUME_INITIALIZED);
        return base_type::operator[](n);
    }
 
    friend void
    swap(basic_contiguous_iterator &lhs, basic_contiguous_iterator &rhs)
    {
        std::swap(lhs.ptr, rhs.ptr);
    }
};
 
} /* namespace detail */
 
} /* namespace pmem */
 
#endif /* LIBPMEMOBJ_CPP_CONTIGUOUS_ITERATOR_HPP */