mirror of
https://github.com/drewcassidy/quicktex.git
synced 2024-09-13 06:37:34 +00:00
Rework ranges library
Better matches the standard library, and iterators moved to their own file
This commit is contained in:
parent
9b3c1d0ca3
commit
c92d58d115
@ -5,6 +5,7 @@ ColumnLimit: 120
|
|||||||
AllowShortBlocksOnASingleLine: Always
|
AllowShortBlocksOnASingleLine: Always
|
||||||
AllowShortFunctionsOnASingleLine: All
|
AllowShortFunctionsOnASingleLine: All
|
||||||
AlwaysBreakTemplateDeclarations: MultiLine
|
AlwaysBreakTemplateDeclarations: MultiLine
|
||||||
|
#RequiresClausePositionStyle: SingleLine # requires Clang 15 :(
|
||||||
#AlignConsecutiveDeclarations: true
|
#AlignConsecutiveDeclarations: true
|
||||||
---
|
---
|
||||||
|
|
||||||
|
@ -24,6 +24,7 @@
|
|||||||
#include <numeric>
|
#include <numeric>
|
||||||
#include <xsimd/xsimd.hpp>
|
#include <xsimd/xsimd.hpp>
|
||||||
|
|
||||||
|
#include "util/iterator.h"
|
||||||
#include "util/map.h"
|
#include "util/map.h"
|
||||||
#include "util/math.h"
|
#include "util/math.h"
|
||||||
#include "util/ranges.h"
|
#include "util/ranges.h"
|
||||||
@ -417,10 +418,10 @@ class Matrix : public VecBase<std::conditional_t<N == 1, T, VecBase<T, T, N>>, T
|
|||||||
}
|
}
|
||||||
|
|
||||||
protected:
|
protected:
|
||||||
class column_iterator : public index_iterator_base<column_iterator> {
|
class column_iterator : public index_iterator_base<column_iterator, column_type> {
|
||||||
public:
|
public:
|
||||||
using value_type = column_type;
|
using value_type = column_type;
|
||||||
using base = index_iterator_base<column_iterator>;
|
using base = index_iterator_base<column_iterator, column_type>;
|
||||||
|
|
||||||
column_iterator(const Matrix *matrix = nullptr, int index = 0) : base(index), _matrix(matrix){};
|
column_iterator(const Matrix *matrix = nullptr, int index = 0) : base(index), _matrix(matrix){};
|
||||||
|
|
||||||
@ -435,10 +436,10 @@ class Matrix : public VecBase<std::conditional_t<N == 1, T, VecBase<T, T, N>>, T
|
|||||||
const Matrix *_matrix;
|
const Matrix *_matrix;
|
||||||
};
|
};
|
||||||
|
|
||||||
template <typename V> class linear_iterator : public index_iterator_base<linear_iterator<V>> {
|
template <typename V> class linear_iterator : public index_iterator_base<linear_iterator<V>, T> {
|
||||||
public:
|
public:
|
||||||
using value_type = column_type;
|
using value_type = T;
|
||||||
using base = index_iterator_base<linear_iterator<V>>;
|
using base = index_iterator_base<linear_iterator<V>, T>;
|
||||||
|
|
||||||
linear_iterator(V *matrix = nullptr, int index = 0) : base(index), _matrix(matrix){};
|
linear_iterator(V *matrix = nullptr, int index = 0) : base(index), _matrix(matrix){};
|
||||||
|
|
||||||
|
@ -26,6 +26,7 @@
|
|||||||
#include <numeric>
|
#include <numeric>
|
||||||
#include <type_traits>
|
#include <type_traits>
|
||||||
|
|
||||||
|
#include "iterator.h"
|
||||||
#include "util/math.h"
|
#include "util/math.h"
|
||||||
#include "util/ranges.h"
|
#include "util/ranges.h"
|
||||||
|
|
||||||
@ -125,7 +126,7 @@ size_t unpack_into(P packed, OI begin, OI end, WI widths, bool little_endian = t
|
|||||||
template <typename P, typename OR, typename WR>
|
template <typename P, typename OR, typename WR>
|
||||||
requires std::unsigned_integral<P> && range<OR> && range<WR>
|
requires std::unsigned_integral<P> && range<OR> && range<WR>
|
||||||
size_t unpack_into(P packed, OR &dest, const WR &widths, bool little_endian = true) {
|
size_t unpack_into(P packed, OR &dest, const WR &widths, bool little_endian = true) {
|
||||||
assert(distance(widths) == distance(dest));
|
assert(size(widths) == size(dest));
|
||||||
return unpack_into(packed, dest.begin(), dest.end(), widths.begin(), little_endian);
|
return unpack_into(packed, dest.begin(), dest.end(), widths.begin(), little_endian);
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -201,7 +202,7 @@ std::array<U, N> unpack(P packed, const std::array<size_t, N> &widths, bool litt
|
|||||||
template <typename U, size_t N, typename P, typename WR>
|
template <typename U, size_t N, typename P, typename WR>
|
||||||
requires std::unsigned_integral<P> && range<WR>
|
requires std::unsigned_integral<P> && range<WR>
|
||||||
std::array<U, N> unpack(P packed, const WR &widths, bool little_endian = true) {
|
std::array<U, N> unpack(P packed, const WR &widths, bool little_endian = true) {
|
||||||
assert(distance(widths) == N);
|
assert(size(widths) == N);
|
||||||
return unpack<U, N>(packed, widths.begin(), little_endian);
|
return unpack<U, N>(packed, widths.begin(), little_endian);
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -273,7 +274,7 @@ inline constexpr P pack(II start, II end, WI widths, bool little_endian = true)
|
|||||||
template <typename P, typename IR, typename WR>
|
template <typename P, typename IR, typename WR>
|
||||||
requires std::unsigned_integral<P> && range<IR> && range<WR>
|
requires std::unsigned_integral<P> && range<IR> && range<WR>
|
||||||
inline constexpr P pack(IR r, WR widths, bool little_endian = true) {
|
inline constexpr P pack(IR r, WR widths, bool little_endian = true) {
|
||||||
assert(distance(widths) == distance(r));
|
assert(size(widths) == size(r));
|
||||||
return pack<P>(r.begin(), r.end(), widths.start(), little_endian);
|
return pack<P>(r.begin(), r.end(), widths.start(), little_endian);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -65,4 +65,4 @@ constexpr inline auto operator^=(E& a, E b) noexcept -> E& {
|
|||||||
a = a ^ b;
|
a = a ^ b;
|
||||||
return a;
|
return a;
|
||||||
}
|
}
|
||||||
} // namespace quicktex::util
|
} // namespace quicktex
|
146
quicktex/util/iterator.h
Normal file
146
quicktex/util/iterator.h
Normal file
@ -0,0 +1,146 @@
|
|||||||
|
/* Quicktex Texture Compression Library
|
||||||
|
Copyright (C) 2021 Andrew Cassidy <drewcassidy@me.com>
|
||||||
|
Partially derived from rgbcx.h written by Richard Geldreich <richgel99@gmail.com>
|
||||||
|
and licenced under the public domain
|
||||||
|
|
||||||
|
This program is free software: you can redistribute it and/or modify
|
||||||
|
it under the terms of the GNU Lesser General Public License as published by
|
||||||
|
the Free Software Foundation, either version 3 of the License, or
|
||||||
|
(at your option) any later version.
|
||||||
|
|
||||||
|
This program is distributed in the hope that it will be useful,
|
||||||
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
GNU Lesser General Public License for more details.
|
||||||
|
|
||||||
|
You should have received a copy of the GNU Lesser General Public License
|
||||||
|
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
namespace quicktex {
|
||||||
|
|
||||||
|
namespace detail {
|
||||||
|
template <class R> using subs_value_t = std::remove_reference_t<decltype(std::declval<R &>()[0])>;
|
||||||
|
}
|
||||||
|
|
||||||
|
template <typename D, typename T> class index_iterator_base {
|
||||||
|
public:
|
||||||
|
using value_type = T;
|
||||||
|
using size_type = int;
|
||||||
|
using difference_type = int;
|
||||||
|
|
||||||
|
D &operator++() {
|
||||||
|
_index++;
|
||||||
|
return static_cast<D &>(*this);
|
||||||
|
}
|
||||||
|
D operator++(int) {
|
||||||
|
D old = static_cast<D &>(*this);
|
||||||
|
_index++;
|
||||||
|
return old;
|
||||||
|
}
|
||||||
|
D &operator--() {
|
||||||
|
_index--;
|
||||||
|
return static_cast<D &>(*this);
|
||||||
|
}
|
||||||
|
D operator--(int) {
|
||||||
|
D old = static_cast<D &>(*this);
|
||||||
|
_index--;
|
||||||
|
return old;
|
||||||
|
}
|
||||||
|
|
||||||
|
D operator+(difference_type rhs) const {
|
||||||
|
D d = static_cast<const D &>(*this);
|
||||||
|
d._index += rhs;
|
||||||
|
return d;
|
||||||
|
}
|
||||||
|
|
||||||
|
D operator-(difference_type rhs) const {
|
||||||
|
D d = static_cast<const D &>(*this);
|
||||||
|
d._index -= rhs;
|
||||||
|
return d;
|
||||||
|
}
|
||||||
|
|
||||||
|
D &operator+=(difference_type rhs) {
|
||||||
|
*this = *this + rhs;
|
||||||
|
return *this;
|
||||||
|
}
|
||||||
|
|
||||||
|
D &operator-=(difference_type rhs) {
|
||||||
|
*this = *this - rhs;
|
||||||
|
return *this;
|
||||||
|
}
|
||||||
|
|
||||||
|
difference_type operator-(const D &rhs) const { return (difference_type)_index - rhs._index; }
|
||||||
|
|
||||||
|
friend D operator+(difference_type lhs, const D &rhs) { return rhs + lhs; }
|
||||||
|
|
||||||
|
friend auto operator<=>(const D &lhs, const D &rhs) { return lhs._index <=> rhs._index; }
|
||||||
|
|
||||||
|
T &operator[](difference_type i) { return *(static_cast<D &>(*this) + i); }
|
||||||
|
T &operator[](difference_type i) const { return *(static_cast<const D &>(*this) + i); }
|
||||||
|
|
||||||
|
protected:
|
||||||
|
int _index;
|
||||||
|
|
||||||
|
private:
|
||||||
|
friend D;
|
||||||
|
index_iterator_base(size_t index = 0) : _index(index) {}
|
||||||
|
};
|
||||||
|
|
||||||
|
template <typename R>
|
||||||
|
requires requires(const R &r) { r[0]; }
|
||||||
|
class index_iterator : public index_iterator_base<index_iterator<R>, detail::subs_value_t<R>> {
|
||||||
|
public:
|
||||||
|
using base = index_iterator_base<index_iterator<R>, detail::subs_value_t<R>>;
|
||||||
|
using typename base::difference_type;
|
||||||
|
using typename base::size_type;
|
||||||
|
using typename base::value_type;
|
||||||
|
|
||||||
|
index_iterator() : base(0), _range(nullptr) {}
|
||||||
|
index_iterator(R &range, int index) : base(index), _range(&range) {}
|
||||||
|
|
||||||
|
value_type &operator*() const {
|
||||||
|
assert(_range != nullptr);
|
||||||
|
assert(this->_index >= 0);
|
||||||
|
assert(this->_index < (size_type)_range->size());
|
||||||
|
return (*_range)[this->_index];
|
||||||
|
}
|
||||||
|
value_type *operator->() const { return &(this->operator*()); }
|
||||||
|
|
||||||
|
friend bool operator==(const index_iterator &lhs, const index_iterator &rhs) {
|
||||||
|
return (lhs._range == rhs._range) && (lhs._index == rhs._index);
|
||||||
|
}
|
||||||
|
|
||||||
|
private:
|
||||||
|
R *_range;
|
||||||
|
};
|
||||||
|
|
||||||
|
template <typename T> class const_iterator : public index_iterator_base<const_iterator<T>, const T> {
|
||||||
|
public:
|
||||||
|
using base = index_iterator_base<const_iterator<T>, const T>;
|
||||||
|
using typename base::difference_type;
|
||||||
|
using typename base::size_type;
|
||||||
|
using typename base::value_type;
|
||||||
|
|
||||||
|
const_iterator() : base(0), _value(T{}) {}
|
||||||
|
const_iterator(T value, int index = 0) : base(index), _value(value) {}
|
||||||
|
|
||||||
|
value_type &operator*() const { return _value; }
|
||||||
|
value_type *operator->() const { return &_value; }
|
||||||
|
|
||||||
|
friend bool operator==(const const_iterator &lhs, const const_iterator &rhs) {
|
||||||
|
return (lhs._value == rhs._value) && (lhs._index == rhs._index);
|
||||||
|
}
|
||||||
|
|
||||||
|
private:
|
||||||
|
T _value;
|
||||||
|
};
|
||||||
|
|
||||||
|
// const_iterator is guaranteed to be a random access iterator. it is not writable for obvious reasons
|
||||||
|
static_assert(std::random_access_iterator<const_iterator<int>>);
|
||||||
|
|
||||||
|
// index_iterator satisfied forward_iterator
|
||||||
|
static_assert(std::random_access_iterator<index_iterator<std::array<int, 4>>>);
|
||||||
|
} // namespace quicktex
|
@ -30,7 +30,7 @@ namespace quicktex {
|
|||||||
namespace detail {
|
namespace detail {
|
||||||
|
|
||||||
template <typename T>
|
template <typename T>
|
||||||
concept simdable = subscriptable_range<T> && std::contiguous_iterator<decltype(std::declval<T>().begin())> &&
|
concept simdable = random_access_range<T> && std::contiguous_iterator<decltype(std::declval<T>().begin())> &&
|
||||||
std::is_arithmetic_v<range_value_t<T>>;
|
std::is_arithmetic_v<range_value_t<T>>;
|
||||||
template <typename T, bool serial = false> struct chunker_impl {};
|
template <typename T, bool serial = false> struct chunker_impl {};
|
||||||
|
|
||||||
@ -74,7 +74,7 @@ struct chunker_impl<T, serial> {
|
|||||||
};
|
};
|
||||||
|
|
||||||
template <typename T, bool serial>
|
template <typename T, bool serial>
|
||||||
requires subscriptable_range<T> && (!simdable<T> || serial)
|
requires random_access_range<T> && (!simdable<T> || serial)
|
||||||
struct chunker_impl<T, serial> {
|
struct chunker_impl<T, serial> {
|
||||||
// range with data that cant be SIMDed
|
// range with data that cant be SIMDed
|
||||||
static constexpr size_t steps = 1;
|
static constexpr size_t steps = 1;
|
||||||
|
@ -29,171 +29,46 @@
|
|||||||
#include <numeric>
|
#include <numeric>
|
||||||
#include <string>
|
#include <string>
|
||||||
#include <type_traits>
|
#include <type_traits>
|
||||||
#include <vector>
|
|
||||||
|
|
||||||
namespace quicktex {
|
namespace quicktex {
|
||||||
|
|
||||||
// std::ranges::range is not usable by default in libc++ 13
|
// std::ranges is not usable by default in libc++ 13
|
||||||
template <class T>
|
template <class T>
|
||||||
concept range = requires(T &t) {
|
concept range = requires(T &t) {
|
||||||
t.begin();
|
t.begin();
|
||||||
t.end();
|
t.end();
|
||||||
};
|
};
|
||||||
|
|
||||||
template <class T>
|
using std::size;
|
||||||
concept sized = requires(T &t) { std::size(t); };
|
template <range T> constexpr auto size(const T &range) { return std::distance(range.begin(), range.end()); }
|
||||||
|
|
||||||
template <class T>
|
template <class T>
|
||||||
concept sized_range = range<T> && sized<T>;
|
concept sized_range = range<T> && requires(T &t) { size(t); };
|
||||||
|
|
||||||
template <typename T>
|
template <class R> using iterator_t = decltype(std::declval<R &>().begin());
|
||||||
concept const_subscriptable = requires(T &t) {
|
template <class R> using sentinel_t = decltype(std::declval<R &>().end());
|
||||||
t[0];
|
template <class R> using range_size_t = decltype(size(std::declval<R &>()));
|
||||||
std::size(t);
|
template <class R> using range_difference_t = std::iter_difference_t<iterator_t<R>>;
|
||||||
};
|
template <class R> using range_value_t = std::iter_value_t<iterator_t<R>>;
|
||||||
|
template <class R> using range_reference_t = std::iter_reference_t<iterator_t<R>>;
|
||||||
|
template <class R> using range_rvalue_reference_t = std::iter_rvalue_reference_t<iterator_t<R>>;
|
||||||
|
|
||||||
template <class T>
|
template <class R>
|
||||||
concept subscriptable = const_subscriptable<T> && requires(T &t) {
|
concept input_range = range<R> && std::input_iterator<iterator_t<R>>;
|
||||||
t[0] = std::declval<decltype(t[0])>(); // subscript is assignable
|
|
||||||
};
|
|
||||||
|
|
||||||
template <typename T>
|
template <class R, typename T>
|
||||||
concept subscriptable_range = sized_range<T> && subscriptable<T>;
|
concept output_range = range<R> && (std::output_iterator<iterator_t<R>, T>);
|
||||||
|
|
||||||
template <typename T> struct range_value { using type = void; };
|
template <class R>
|
||||||
|
concept forward_range = range<R> && std::forward_iterator<iterator_t<R>>;
|
||||||
|
|
||||||
template <typename T>
|
template <class R>
|
||||||
requires requires(T &t) { std::begin(t); }
|
concept bidirectional_range = range<R> && std::bidirectional_iterator<iterator_t<R>>;
|
||||||
struct range_value<T> {
|
|
||||||
using type = std::iter_value_t<decltype((std::declval<T>()).begin())>;
|
|
||||||
};
|
|
||||||
|
|
||||||
template <typename T> using range_value_t = typename range_value<T>::type;
|
template <class R>
|
||||||
|
concept random_access_range = range<R> && std::random_access_iterator<iterator_t<R>>;
|
||||||
|
|
||||||
// some quick inline checks
|
template <class R>
|
||||||
static_assert(const_subscriptable<const std::array<int, 4>>); // const array can be subscripted
|
concept contiguous_range = range<R> && std::contiguous_iterator<iterator_t<R>>;
|
||||||
static_assert(!subscriptable<const std::array<int, 4>>); // const array cannot be assigned to
|
|
||||||
static_assert(subscriptable_range<std::array<int, 4>>); // array is subscriptable, sized, and has begin() and end()
|
|
||||||
static_assert(sized_range<std::initializer_list<int>>); // initializer list is a range and has size()
|
|
||||||
static_assert(std::same_as<range_value_t<int>, void>);
|
|
||||||
|
|
||||||
template <class T>
|
|
||||||
requires range<T>
|
|
||||||
size_t distance(T range) {
|
|
||||||
return std::distance(range.begin(), range.end());
|
|
||||||
}
|
|
||||||
|
|
||||||
template <typename D> class index_iterator_base {
|
|
||||||
public:
|
|
||||||
typedef long long difference_type;
|
|
||||||
|
|
||||||
D &operator++() {
|
|
||||||
_index++;
|
|
||||||
return static_cast<D &>(*this);
|
|
||||||
}
|
|
||||||
D operator++(int) {
|
|
||||||
D old = static_cast<D &>(*this);
|
|
||||||
_index++;
|
|
||||||
return old;
|
|
||||||
}
|
|
||||||
D &operator--() {
|
|
||||||
_index--;
|
|
||||||
return static_cast<D &>(*this);
|
|
||||||
}
|
|
||||||
D operator--(int) {
|
|
||||||
D old = static_cast<D &>(*this);
|
|
||||||
_index--;
|
|
||||||
return old;
|
|
||||||
}
|
|
||||||
|
|
||||||
D operator+(difference_type rhs) const {
|
|
||||||
D d = static_cast<const D &>(*this);
|
|
||||||
d._index += rhs;
|
|
||||||
return d;
|
|
||||||
}
|
|
||||||
|
|
||||||
D operator-(difference_type rhs) const {
|
|
||||||
D d = static_cast<const D &>(*this);
|
|
||||||
d._index -= rhs;
|
|
||||||
return d;
|
|
||||||
}
|
|
||||||
|
|
||||||
D &operator+=(difference_type rhs) {
|
|
||||||
*this = *this + rhs;
|
|
||||||
return *this;
|
|
||||||
}
|
|
||||||
|
|
||||||
D &operator-=(difference_type rhs) {
|
|
||||||
*this = *this - rhs;
|
|
||||||
return *this;
|
|
||||||
}
|
|
||||||
|
|
||||||
difference_type operator-(const D &rhs) const { return (difference_type)_index - rhs._index; }
|
|
||||||
|
|
||||||
friend D operator+(difference_type lhs, const D &rhs) { return rhs + lhs; }
|
|
||||||
|
|
||||||
friend auto operator<=>(const D &lhs, const D &rhs) { return lhs._index <=> rhs._index; }
|
|
||||||
|
|
||||||
auto &operator[](difference_type i) { return *(static_cast<D &>(*this) + i); }
|
|
||||||
auto operator[](difference_type i) const { return *(static_cast<const D &>(*this) + i); }
|
|
||||||
|
|
||||||
protected:
|
|
||||||
size_t _index;
|
|
||||||
|
|
||||||
private:
|
|
||||||
friend D;
|
|
||||||
index_iterator_base(size_t index = 0) : _index(index) {}
|
|
||||||
};
|
|
||||||
|
|
||||||
template <typename T> class const_iterator : public index_iterator_base<const_iterator<T>> {
|
|
||||||
public:
|
|
||||||
typedef index_iterator_base<const_iterator<T>> base;
|
|
||||||
typedef long long difference_type;
|
|
||||||
typedef T value_type;
|
|
||||||
|
|
||||||
const_iterator() : base(0), _value(T{}) {}
|
|
||||||
const_iterator(T value, size_t index = 0) : base(index), _value(value) {}
|
|
||||||
|
|
||||||
T operator*() const { return _value; }
|
|
||||||
const T *operator->() const { return &_value; }
|
|
||||||
|
|
||||||
friend bool operator==(const const_iterator &lhs, const const_iterator &rhs) {
|
|
||||||
return (lhs._value == rhs._value) && (lhs._index == rhs._index);
|
|
||||||
}
|
|
||||||
|
|
||||||
private:
|
|
||||||
T _value;
|
|
||||||
};
|
|
||||||
|
|
||||||
// const_iterator is guaranteed to be a random access iterator. it is not writable for obvious reasons
|
|
||||||
static_assert(std::random_access_iterator<const_iterator<int>>);
|
|
||||||
|
|
||||||
template <typename R>
|
|
||||||
requires subscriptable<R>
|
|
||||||
class index_iterator : public index_iterator_base<index_iterator<R>> {
|
|
||||||
public:
|
|
||||||
using base = index_iterator_base<index_iterator<R>>;
|
|
||||||
using difference_type = long long;
|
|
||||||
using value_type = range_value_t<R>;
|
|
||||||
|
|
||||||
index_iterator() : base(0), _range(nullptr) {}
|
|
||||||
index_iterator(R &range, size_t index) : base(index), _range(&range) {}
|
|
||||||
|
|
||||||
auto operator*() const {
|
|
||||||
// if we have the information, do a bounds check
|
|
||||||
if constexpr (sized<R>) { assert(this->_index < std::size(*_range)); }
|
|
||||||
return (*_range)[this->_index];
|
|
||||||
}
|
|
||||||
auto *operator->() const { return &((*_range)[this->_index]); }
|
|
||||||
|
|
||||||
friend bool operator==(const index_iterator &lhs, const index_iterator &rhs) {
|
|
||||||
return (lhs._range == rhs._range) && (lhs._index == rhs._index);
|
|
||||||
}
|
|
||||||
|
|
||||||
private:
|
|
||||||
R *_range;
|
|
||||||
};
|
|
||||||
|
|
||||||
// index_iterator satisfied forward_iterator
|
|
||||||
static_assert(std::random_access_iterator<index_iterator<std::array<int, 4>>>);
|
|
||||||
} // namespace quicktex
|
} // namespace quicktex
|
Loading…
Reference in New Issue
Block a user