MRBuffer.h

Go to the documentation of this file.

#pragma once
 
#include "MRMacros.h"
#include "MRNoDefInit.h"
#include "MRId.h"
#include <cassert>
#include <concepts>
#include <memory>
#include <type_traits>
 
namespace MR
{
 
 
template <typename T>
struct ZeroOnMove
{
    T val = 0;
    constexpr ZeroOnMove() noexcept {}
    constexpr ZeroOnMove( const ZeroOnMove & ) noexcept = delete;
    constexpr ZeroOnMove( ZeroOnMove && z ) noexcept : val( z.val ) { z.val = 0; }
    constexpr ZeroOnMove& operator =( const ZeroOnMove & ) noexcept = delete;
    constexpr ZeroOnMove& operator =( ZeroOnMove && z ) noexcept { val = z.val; z.val = 0; return * this; }
};
 
template <typename T>
struct NoCtor;
 
template <typename T>
concept Trivial = std::is_trivially_constructible_v<T>;

template <Trivial T>
struct NoCtor<T>
{
    using type = T;
};

template <std::constructible_from<NoInit> T>
struct NoCtor<T>
{
    using type = NoDefInit<T>;
};

template <typename V, typename I>
class Buffer
{
public:
    using T = typename NoCtor<V>::type;
    using reference = T&;
    using const_reference = const T&;
    using iterator = T*;
    using const_iterator = const T*;
 
    Buffer() = default;
    explicit Buffer( size_t size ) { resize( size ); }
 
    [[nodiscard]] auto capacity() const { return capacity_.val; }
    [[nodiscard]] auto size() const { return size_.val; }
    [[nodiscard]] bool empty() const { return size_.val == 0; }
 
    void clear() { data_.reset(); capacity_ = {}; size_ = {}; }
 
    void resize( size_t newSize )
    {
        if ( size_.val == newSize )
            return;
        if ( newSize > capacity_.val )
        {
#if __cpp_lib_smart_ptr_for_overwrite >= 202002L
            data_ = std::make_unique_for_overwrite<T[]>( capacity_.val = newSize );
#else
            data_.reset( new T[capacity_.val = newSize] );
#endif
        }
        size_.val = newSize;
    }
 
    [[nodiscard]] const_reference operator[]( I i ) const MR_LIFETIMEBOUND
    {
        assert( i < size_.val );
        return data_[i];
    }
    [[nodiscard]] reference operator[]( I i ) MR_LIFETIMEBOUND
    {
        assert( i < size_.val );
        return data_[i];
    }
 
    [[nodiscard]] auto data() MR_LIFETIMEBOUND { return data_.get(); }
    [[nodiscard]] auto data() const MR_LIFETIMEBOUND { return data_.get(); }

    [[nodiscard]] I beginId() const { return I{ size_t(0) }; }

    [[nodiscard]] I backId() const { assert( !empty() ); return I{ size() - 1 }; }

    [[nodiscard]] I endId() const { return I{ size() }; }

    [[nodiscard]] size_t heapBytes() const { return capacity() * sizeof(T); }
 
    [[nodiscard]] friend auto begin( const Buffer & a ) { return a.data(); }
    [[nodiscard]] friend auto begin( Buffer & a ) { return a.data(); }
    [[nodiscard]] friend auto end( const Buffer & a ) { return a.data() + a.size(); }
    [[nodiscard]] friend auto end( Buffer & a ) { return a.data() + a.size(); }
 
private:
    std::unique_ptr<T[]> data_;
    ZeroOnMove<size_t> capacity_, size_;
};

template <typename T, typename I>
inline T getAt( const Buffer<T, I> & bmap MR_LIFETIMEBOUND_NESTED, I key, T def = {} )
{
    return key ? T{bmap[key]} : def;
}

template <typename T, typename I>
struct BMap
{
    Buffer<T, I> b;
    size_t tsize = 0; 
};

struct PackMapping
{
    UndirectedEdgeBMap e;
    FaceBMap f;
    VertBMap v;
};

template <typename T>
BMap<T, T> compose( const BMap<T, T> & a, const BMap<T, T> & b )
{
    BMap<T, T> res;
    res.b.resize( b.b.size() );
    res.tsize = a.tsize;
    for ( T x( 0 ); x < b.b.size(); ++x )
    {
        auto bx = b.b[x];
        if ( bx < a.b.size() ) 
            res.b[x] = a.b[bx];
    }
    return res;
}
 
}