MRMesh/MRVector.h

Go to the documentation of this file.

#pragma once
 
#include "MRMacros.h"
#include "MRMeshFwd.h"
#include <cassert>
#include <vector>
 
namespace MR
{
 
template <typename T, typename I>
class Vector
{
public:
    using reference = typename std::vector<T>::reference;
    using const_reference = typename std::vector<T>::const_reference;
    using iterator = typename std::vector<T>::iterator;
    using const_iterator = typename std::vector<T>::const_iterator;
 
    Vector() = default;
    explicit Vector( size_t size ) MR_REQUIRES_IF_SUPPORTED( std::default_initializable<T> ) : vec_( size ) { }
    explicit Vector( size_t size, const T & val ) : vec_( size, val ) { }
    Vector( std::vector<T> && vec ) : vec_( std::move( vec ) ) { }
    template< class InputIt >
    Vector( InputIt first, InputIt last ) : vec_( first, last ) { }
    Vector( std::initializer_list<T> init ) : vec_( init ) { }
 
    [[nodiscard]] bool operator == ( const Vector & b ) const MR_REQUIRES_IF_SUPPORTED( std::equality_comparable<T> ) { return vec_ == b.vec_; }
    [[nodiscard]] bool operator != ( const Vector & b ) const MR_REQUIRES_IF_SUPPORTED( std::equality_comparable<T> ) { return vec_ != b.vec_; }
 
    void clear() { vec_.clear(); }
    [[nodiscard]] bool empty() const { return vec_.empty(); }
 
    [[nodiscard]] std::size_t size() const { return vec_.size(); }
 
    void resize( size_t newSize ) MR_REQUIRES_IF_SUPPORTED( std::movable<T> && std::default_initializable<T> ) { vec_.resize( newSize ); }
    void resize( size_t newSize, const T & t ) MR_REQUIRES_IF_SUPPORTED( std::movable<T> ) { vec_.resize( newSize, t ); }
 
    // resizes the vector skipping initialization of its elements (more precisely initializing them using ( noInit ) constructor )
    void resizeNoInit( size_t targetSize ) MR_REQUIRES_IF_SUPPORTED( std::constructible_from<T, NoInit> )
    {
        // allocate enough memory
        reserve( targetSize );
        // resize without memory access
        while ( size() < targetSize )
            emplace_back( noInit );
        // in case initial size was larger
        resize( targetSize );
    }
 
    [[nodiscard]] std::size_t capacity() const { return vec_.capacity(); }
    void reserve( size_t capacity ) { vec_.reserve( capacity ); }
 
    [[nodiscard]] const_reference operator[]( I i ) const
    {
        assert( i < vec_.size() );
        return vec_[i];
    }
    [[nodiscard]] reference operator[]( I i )
    {
        assert( i < vec_.size() );
        return vec_[i];
    }
 
    void resizeWithReserve( size_t newSize, const T & value = T() )
    {
        auto reserved = vec_.capacity();
        if ( reserved > 0 && newSize > reserved )
        {
            while ( newSize > reserved )
                reserved <<= 1;
            vec_.reserve( reserved );
        }
        vec_.resize( newSize, value );
    }
 
    void autoResizeSet( I pos, size_t len, T val )
    {
        assert( pos );
        const int p{ pos };
        if ( const auto sz = size(); p + len > sz )
        {
            resizeWithReserve( p + len, val );
            if ( p >= sz )
                return;
            len = sz - p;
        }
        for ( size_t i = 0; i < len; ++i )
            vec_[ p + i ] = val;
    }
    void autoResizeSet( I i, T val ) { autoResizeSet( i, 1, val ); }
 
    [[nodiscard]] reference autoResizeAt( I i ) MR_REQUIRES_IF_SUPPORTED( std::default_initializable<T> )
    {
        if ( i + 1 > size() )
            resizeWithReserve( i + 1 );
        return vec_[i];
    }
 
    void push_back( const T & t ) { vec_.push_back( t ); }
    void push_back( T && t ) { vec_.push_back( std::move( t ) ); }
    void pop_back() { vec_.pop_back(); }
 
    template<typename... Args>
    T& emplace_back( Args&&... args ) { return vec_.emplace_back( std::forward<Args>(args)... ); }
 
    [[nodiscard]] const_reference front() const { return vec_.front(); }
    [[nodiscard]]       reference front()       { return vec_.front(); }
    [[nodiscard]] const_reference  back() const { return vec_.back(); }
    [[nodiscard]]       reference  back()       { return vec_.back(); }
    [[nodiscard]] I beginId() const { return I( size_t(0) ); }
    [[nodiscard]] I backId() const { assert( !vec_.empty() ); return I( vec_.size() - 1 ); }
    [[nodiscard]] I endId() const { return I( vec_.size() ); }
 
    [[nodiscard]] T* data() { return vec_.data(); }
    [[nodiscard]] const T* data() const { return vec_.data(); }
 
    void swap( Vector & b ) { vec_.swap( b.vec_ ); }
 
    [[nodiscard]] size_t heapBytes() const { return capacity() * sizeof(T); }
 
    std::vector<T> vec_;
};
 
template <typename T, typename I>
[[nodiscard]] inline auto begin( const Vector<T, I> & a )
    { return a.vec_.begin(); }
 
template <typename T, typename I>
[[nodiscard]] inline auto begin( Vector<T, I> & a )
    { return a.vec_.begin(); }
 
template <typename T, typename I>
[[nodiscard]] inline auto end( const Vector<T, I> & a )
    { return a.vec_.end(); }
 
template <typename T, typename I>
[[nodiscard]] inline auto end( Vector<T, I> & a )
    { return a.vec_.end(); }
 
template <typename T, typename I>
[[nodiscard]] inline T getAt( const Vector<T, I> & a, I id, T def = {} )
{
    return ( id && id < a.size() ) ? a[id] : def;
}
 
} // namespace MR