libcruft-util/view.hpp

521 lines
16 KiB
C++

/*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*
* Copyright 2015-2017 Danny Robson <danny@nerdcruft.net>
*/
#ifndef CRUFT_UTIL_VIEW_HPP
#define CRUFT_UTIL_VIEW_HPP
#include "cast.hpp"
#include "debug.hpp"
#include "types/traits.hpp"
#include <cstdlib>
#include <ostream>
#include <string>
#include <cstring>
#include <stdexcept>
#include <iterator>
namespace util {
template <typename BeginT, typename EndT = BeginT>
struct view {
public:
//---------------------------------------------------------------------
using value_type = typename std::iterator_traits<
remove_restrict_t<BeginT>
>::value_type;
//---------------------------------------------------------------------
constexpr
view (const BeginT &first, const EndT &last) noexcept:
m_begin (first),
m_end (last)
{ ; }
//---------------------------------------------------------------------
template <typename CountT, typename = std::enable_if_t<std::is_integral_v<CountT>,void>>
view (
const BeginT &_begin,
CountT _size
):
view (_begin, _begin + _size)
{ ; }
//---------------------------------------------------------------------
template <
typename ValueT,
typename = std::enable_if_t<
std::is_same_v<BeginT, const ValueT*> &&
std::is_same_v<EndT, const ValueT*>
>
>
view (const view<ValueT*,ValueT*> &rhs):
m_begin (rhs.begin ()),
m_end (rhs.end ())
{ ; }
//---------------------------------------------------------------------
// explicitly cater for the char array case so that we don't
// accidentally include the trailing null in the data.
template <std::size_t N>
view (const char (&value)[N]):
view (std::begin (value), std::begin (value) + N - 1)
{
static_assert (N > 0);
}
//---------------------------------------------------------------------
template <std::size_t N, typename ValueT>
view (const ValueT(&value)[N]):
view (std::begin (value), std::end (value))
{ ; }
//---------------------------------------------------------------------
constexpr
view (const view &rhs) noexcept:
view (rhs.m_begin, rhs.m_end)
{ ; }
//---------------------------------------------------------------------
// technically we could get away without explicitly defining a move
// constructor here, but by nulling rhs we can more easily use this
// class as a base for unique owning pointers without exposing the
// begin/end data members to them directly.
constexpr view (view &&rhs) noexcept:
view (std::move (rhs.m_begin), std::move (rhs.m_end))
{ ; }
//---------------------------------------------------------------------
// allow null construction of views where IteratorT is constructible
// from nullptr_t
//
// ideally we would avoid exposing this as it promotes use of nulls but
// it simplifies construction of views that are data members of classes
// when we may not immediately know the values we should contain.
constexpr view (std::nullptr_t) noexcept:
view {nullptr,nullptr}
{ ; }
//---------------------------------------------------------------------
template <typename CharT, typename Traits, typename Allocator>
view (std::basic_string<CharT,Traits,Allocator> &val):
view (std::data (val), std::data (val) + std::size (val))
{ ; }
//---------------------------------------------------------------------
// non-contigous containers should use their begin/end iterators
// directly
template <
typename ContainerT,
typename std::enable_if_t<!is_contiguous_v<ContainerT>,ContainerT*> = nullptr
>
constexpr explicit
view (
ContainerT &klass
):
view (std::begin (klass), std::end (klass))
{ ; }
//---------------------------------------------------------------------
// contiguous containers are often used with pointers to their contents
// for other operations so we directly construct them using pointers to
// their data for user convenience.
template <
typename ContainerT,
typename std::enable_if_t<is_contiguous_v<ContainerT>,ContainerT*> = nullptr
>
constexpr explicit
view (
ContainerT &klass
):
view (std::data (klass), std::data (klass) + std::size (klass))
{ ; }
//---------------------------------------------------------------------
view&
operator= (const view &rhs) noexcept
{
m_begin = rhs.m_begin;
m_end = rhs.m_end;
return *this;
}
//---------------------------------------------------------------------
view&
operator= (view &&rhs) noexcept
{
m_begin = rhs.m_begin;
m_end = rhs.m_end;
rhs.m_begin = BeginT{};
rhs.m_end = EndT{};
return *this;
};
///////////////////////////////////////////////////////////////////////
constexpr BeginT begin (void) noexcept { return m_begin; }
constexpr EndT end (void) noexcept { return m_end; }
//---------------------------------------------------------------------
constexpr const BeginT begin (void) const noexcept { return cbegin (); }
constexpr const EndT end (void) const noexcept { return cend (); }
//---------------------------------------------------------------------
constexpr const BeginT cbegin (void) const noexcept { return m_begin; }
constexpr const EndT cend (void) const noexcept { return m_end; }
//---------------------------------------------------------------------
auto data (void) { return begin (); }
auto data (void) const { return begin (); }
///////////////////////////////////////////////////////////////////////
constexpr bool
empty (void) const noexcept
{
return m_begin == m_end;
}
//---------------------------------------------------------------------
constexpr auto
size (void) const noexcept
{
return sign_cast<std::size_t> (std::distance (m_begin, m_end));
}
//---------------------------------------------------------------------
[[gnu::warn_unused_result]] constexpr auto
redim (int count) const
{
CHECK_GT (count, 0);
if (count > size ())
throw std::invalid_argument ("redim to higher size not allowed");
return view { m_begin, m_begin + count };
};
//---------------------------------------------------------------------
util::view<BeginT,EndT>
operator- (util::view<BeginT,BeginT> prefix) const
{
CHECK_EQ (prefix.begin (), begin ());
return { prefix.end (), end () };
}
///////////////////////////////////////////////////////////////////////
template <
typename ValueT,
typename = std::enable_if_t<
std::is_pointer_v<BeginT> &&
sizeof (*std::declval<BeginT> ()) == sizeof (ValueT)
,
void
>
>
view<ValueT*>
cast (void) const
{
return {
reinterpret_cast<ValueT*> (m_begin),
reinterpret_cast<ValueT*> (m_end)
};
}
///////////////////////////////////////////////////////////////////////
constexpr auto&
operator[] (size_t idx)& noexcept
{
auto it = begin ();
std::advance (it, idx);
return *it;
}
//---------------------------------------------------------------------
constexpr auto&
operator[] (size_t idx) const& noexcept
{
auto it = begin ();
std::advance (it, idx);
return *it;
}
private:
///////////////////////////////////////////////////////////////////////
BeginT m_begin;
EndT m_end;
};
//-------------------------------------------------------------------------
template <typename ValueT, std::size_t N>
view (const ValueT(&)[N]) -> view<const ValueT*,const ValueT*>;
//-------------------------------------------------------------------------
template <
typename IteratorT,
typename SizeT,
typename = std::enable_if_t<
std::is_integral_v<SizeT>,void
>
>
view (IteratorT, SizeT) -> view<IteratorT,IteratorT>;
//-------------------------------------------------------------------------
template <
typename ContainerT,
typename = std::enable_if_t<is_contiguous_v<ContainerT>, void>
>
view (ContainerT&) -> view<
typename ContainerT::value_type*,
typename ContainerT::value_type*
>;
//-------------------------------------------------------------------------
template <
typename ContainerT,
typename = std::enable_if_t<!is_contiguous_v<ContainerT>, void>
>
view (ContainerT&) -> view<
decltype (std::begin (std::declval<ContainerT> ())),
decltype (std::end (std::declval<ContainerT> ()))
>;
///////////////////////////////////////////////////////////////////////////
template <typename ValueT, size_t N>
auto
make_view (const ValueT (&arr)[N])
{
return util::view<const ValueT*> (arr + 0, arr + N);
}
//-------------------------------------------------------------------------
template <typename ContainerT>
auto
make_view (ContainerT &t)
{
return util::view<decltype(std::begin (t))> { std::begin (t), std::end (t) };
}
//-------------------------------------------------------------------------
template <typename ContainerT>
auto
make_view (const ContainerT &t)
{
return util::view<decltype(std::cbegin (t))> { std::cbegin (t), std::cend (t) };
}
//-------------------------------------------------------------------------
// disable the possibility of creating a view to a temporary. note that
// this only works if an lval version has already been defined otherwise
// universal reference rules will capture both lval and rval here.
template <typename ContainerT>
auto
make_view (ContainerT&&) = delete;
///////////////////////////////////////////////////////////////////////////
template <typename ContainerT>
auto
make_cview (const ContainerT &t)
{
return make_view (t);
//return util::view<decltype(std::cbegin (t))> { std::cbegin (t), std::cend (t) };
}
//-------------------------------------------------------------------------
template <typename BeginT, typename EndT>
auto
make_view (BeginT first, EndT last)
{
return view<BeginT, EndT> {first, last};
}
//-------------------------------------------------------------------------
template <typename ValueT>
auto
make_cview (ValueT *first, ValueT *last)
{
return view<const ValueT*> {first, last};
}
///////////////////////////////////////////////////////////////////////////
inline
view<const char*> make_view (const char *str)
{
return { str, str + strlen (str) };
}
//-------------------------------------------------------------------------
inline
view<char*> make_view (char *str)
{
return { str, str + strlen (str) };
}
//-------------------------------------------------------------------------
template <typename CharT, typename TraitsT, typename AllocT>
view<const CharT*>
make_view (const std::basic_string<CharT,TraitsT,AllocT> &str)
{
return {
std::data (str),
std::data (str) + std::size (str)
};
}
//-------------------------------------------------------------------------
template <typename CharT, typename TraitsT, typename AllocT>
view<CharT*>
make_view (std::basic_string<CharT,TraitsT,AllocT> &str)
{
return {
std::data (str),
std::data (str) + std::size (str)
};
}
//-------------------------------------------------------------------------
template <typename CharT, typename TraitsT, typename AllocT>
view<const CharT*>
make_view (const std::basic_string<CharT,TraitsT,AllocT>&&) = delete;
//-------------------------------------------------------------------------
template <typename CharT, typename TraitsT, typename AllocT>
view<CharT*>
make_view (std::basic_string<CharT,TraitsT,AllocT>&&) = delete;
///////////////////////////////////////////////////////////////////////////
template <typename BeginT, typename EndT>
constexpr bool
operator== (const view<BeginT> &a, const view<EndT> &b)
{
return a.size () == b.size () &&
std::equal (std::begin (a), std::end (a), std::begin (b));
}
//-------------------------------------------------------------------------
// defer equality to the view/view operator by way of make_view
template <typename IteratorT, typename ValueT>
constexpr bool
operator== (const view<IteratorT> &a, const ValueT &b)
{
return a == make_view (b);
}
//-------------------------------------------------------------------------
// reverse the arguments and forward to the above operator. we formumlate
// equality this way to avoid implementing the operator twice for each
// weird case.
template <typename IteratorT, typename ValueT>
constexpr bool
operator== (const ValueT &a, const view<IteratorT> &b)
{
return b == a;
}
///////////////////////////////////////////////////////////////////////////
template <typename IteratorT, typename ValueT>
constexpr bool
operator!= (const util::view<IteratorT> &a, const ValueT &b)
{
return !(a == b);
}
//-------------------------------------------------------------------------
template <typename IteratorT, typename ValueT>
constexpr bool
operator!= (const ValueT &a, const util::view<IteratorT> &b)
{
return !(a == b);
}
///////////////////////////////////////////////////////////////////////////
template <typename IteratorT>
std::ostream&
operator<< (std::ostream &os, view<IteratorT> val)
{
std::copy (
std::cbegin (val),
std::cend (val),
std::ostream_iterator<typename decltype(val)::value_type> (os)
);
return os;
}
///////////////////////////////////////////////////////////////////////////
/// a basic stringlike comparison operator that behaves as
/// std::string::compare would.
///
/// provided so that the common case of stringlike views can be used in a
/// std::map and similar without a great deal of work.
inline bool
operator< (util::view<const char*> a, util::view<const char*> b)
{
const auto la = std::size (a);
const auto lb = std::size (b);
const auto res = strncmp (
std::data (a),
std::data (b),
util::min (la, lb)
);
return res < 0 || (res == 0 && la < lb);
}
}
#endif