532 lines
18 KiB
C++
532 lines
18 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 2012-2015 Danny Robson <danny@nerdcruft.net>
|
|
*/
|
|
|
|
#ifndef __UTIL_COORDS_OPS
|
|
#define __UTIL_COORDS_OPS
|
|
|
|
#include "../preprocessor.hpp"
|
|
#include "../maths.hpp"
|
|
#include "../types/bits.hpp"
|
|
|
|
#include <cstdlib>
|
|
#include <cmath>
|
|
|
|
namespace util {
|
|
// forward declerations for traits
|
|
template <size_t,typename> struct point;
|
|
template <size_t,typename> struct extent;
|
|
template <size_t,typename> struct vector;
|
|
template <size_t,typename> struct colour;
|
|
|
|
///////////////////////////////////////////////////////////////////////
|
|
// operation traits
|
|
namespace coord {
|
|
template <
|
|
template <size_t,typename> class A,
|
|
template <size_t,typename> class B
|
|
>
|
|
struct traits { };
|
|
|
|
//-------------------------------------------------------------------------
|
|
template <> struct traits<colour,colour> { template <size_t S, typename T> using result = colour<S,T>; };
|
|
template <> struct traits<extent,extent> { template <size_t S, typename T> using result = extent<S,T>; };
|
|
template <> struct traits<extent,vector> { template <size_t S, typename T> using result = extent<S,T>; };
|
|
template <> struct traits<point,extent> { template <size_t S, typename T> using result = point <S,T>; };
|
|
template <> struct traits<point,vector> { template <size_t S, typename T> using result = point <S,T>; };
|
|
template <> struct traits<vector,point> { template <size_t S, typename T> using result = point <S,T>; };
|
|
template <> struct traits<vector,vector> { template <size_t S, typename T> using result = vector<S,T>; };
|
|
}
|
|
|
|
template <template <size_t,typename> class> struct is_coord : std::false_type { };
|
|
|
|
template <> struct is_coord<point> : std::true_type { };
|
|
template <> struct is_coord<extent> : std::true_type { };
|
|
template <> struct is_coord<vector> : std::true_type { };
|
|
template <> struct is_coord<colour> : std::true_type { };
|
|
|
|
///////////////////////////////////////////////////////////////////////////
|
|
// vector operators
|
|
#define ELEMENT_OP(OP) \
|
|
template < \
|
|
size_t S, \
|
|
typename T, \
|
|
typename U, \
|
|
template <size_t,typename> class A, \
|
|
template <size_t,typename> class B, \
|
|
typename = typename std::enable_if< \
|
|
is_coord<A>::value && is_coord<B>::value, \
|
|
void \
|
|
>::type \
|
|
> \
|
|
auto \
|
|
operator OP (A<S,T> a, B<S,U> b) \
|
|
{ \
|
|
typename coord::traits<A,B>::template result< \
|
|
S,typename std::common_type<T,U>::type \
|
|
> out; \
|
|
for (size_t i = 0; i < S; ++i) \
|
|
out[i] = a[i] OP b[i]; \
|
|
return out; \
|
|
} \
|
|
\
|
|
template < \
|
|
size_t S, \
|
|
typename T, \
|
|
typename U, \
|
|
template <size_t,typename> class A, \
|
|
template <size_t,typename> class B, \
|
|
typename = typename std::enable_if< \
|
|
is_coord<A>::value && \
|
|
is_coord<B>::value && \
|
|
std::is_same< \
|
|
typename std::common_type<T,U>::type, \
|
|
T \
|
|
>::value, \
|
|
void \
|
|
>::type \
|
|
> \
|
|
auto& \
|
|
operator PASTE(OP,=) (A<S,T>& a, B<S,U> b) \
|
|
{ \
|
|
for (size_t i = 0; i < S; ++i) \
|
|
a[i] PASTE(OP,=) b[i]; \
|
|
return a; \
|
|
}
|
|
|
|
ELEMENT_OP(+)
|
|
ELEMENT_OP(-)
|
|
ELEMENT_OP(*)
|
|
ELEMENT_OP(/)
|
|
ELEMENT_OP(%)
|
|
#undef ELEMENT_OP
|
|
|
|
///////////////////////////////////////////////////////////////////////////
|
|
// scalar operators
|
|
#define SCALAR_OP(OP) \
|
|
template < \
|
|
size_t S, \
|
|
typename T, \
|
|
typename U, \
|
|
template <size_t,typename> class K, \
|
|
typename = typename std::enable_if< \
|
|
is_coord<K>::value,void \
|
|
>::type \
|
|
> \
|
|
auto \
|
|
operator OP (U u, K<S,T> k) \
|
|
{ \
|
|
K<S,typename std::common_type<T,U>::type> out; \
|
|
\
|
|
for (size_t i = 0; i < S; ++i) \
|
|
out[i] = u OP k[i]; \
|
|
return out; \
|
|
} \
|
|
\
|
|
template < \
|
|
size_t S, \
|
|
typename T, \
|
|
typename U, \
|
|
template <size_t,typename> class K, \
|
|
typename = typename std::enable_if< \
|
|
is_coord<K>::value,void \
|
|
>::type \
|
|
> \
|
|
auto \
|
|
operator OP (K<S,T> k, U u) \
|
|
{ \
|
|
K<S,typename std::common_type<T,U>::type> out; \
|
|
\
|
|
for (size_t i = 0; i < S; ++i) \
|
|
out[i] = k[i] OP u; \
|
|
return out; \
|
|
}
|
|
|
|
SCALAR_OP(+)
|
|
SCALAR_OP(-)
|
|
SCALAR_OP(*)
|
|
SCALAR_OP(/)
|
|
SCALAR_OP(%)
|
|
|
|
#undef SCALAR_OP
|
|
|
|
|
|
//-------------------------------------------------------------------------
|
|
// scalar assignment operators.
|
|
//
|
|
// we must check the operands/results do not need casting to store in the
|
|
// destination type to avoid silent errors accumulating.
|
|
#define SCALAR_OP(OP) \
|
|
template < \
|
|
size_t S, \
|
|
typename T, \
|
|
typename U, \
|
|
template <size_t,typename> class K, \
|
|
typename = typename std::enable_if< \
|
|
is_coord<K>::value, \
|
|
void \
|
|
>::type \
|
|
> \
|
|
typename std::enable_if< \
|
|
std::is_same< \
|
|
T, \
|
|
typename std::common_type<T,U>::type>::value, \
|
|
K<S,T> \
|
|
>::type& \
|
|
operator OP (K<S,T> &k, U u) \
|
|
{ \
|
|
for (size_t i = 0; i < S; ++i) \
|
|
k[i] OP u; \
|
|
\
|
|
return k; \
|
|
}
|
|
|
|
SCALAR_OP(+=)
|
|
SCALAR_OP(-=)
|
|
SCALAR_OP(*=)
|
|
SCALAR_OP(/=)
|
|
SCALAR_OP(%=)
|
|
#undef SCALAR_OP
|
|
|
|
//-------------------------------------------------------------------------
|
|
// negation
|
|
template <
|
|
size_t S,
|
|
typename T,
|
|
template<size_t,typename> class K
|
|
>
|
|
typename std::enable_if<
|
|
std::is_signed<T>::value,
|
|
K<S,T>
|
|
>::type
|
|
operator- (K<S,T> k)
|
|
{
|
|
for (auto &v: k)
|
|
v = -v;
|
|
return k;
|
|
}
|
|
|
|
|
|
///////////////////////////////////////////////////////////////////////////
|
|
// unary operators
|
|
|
|
#define UNARY_OP(OP) \
|
|
template < \
|
|
size_t S, \
|
|
typename T, \
|
|
template <size_t,typename> class K \
|
|
> \
|
|
auto \
|
|
operator OP (K<S,T> k) \
|
|
{ \
|
|
K<S,decltype(OP std::declval<T> ())> out; \
|
|
\
|
|
for (size_t i = 0; i < S; ++i) \
|
|
out[i] = OP k[i]; \
|
|
\
|
|
return k; \
|
|
}
|
|
|
|
UNARY_OP(!)
|
|
UNARY_OP(~)
|
|
|
|
#undef UNARY_OP
|
|
|
|
|
|
///////////////////////////////////////////////////////////////////////////
|
|
// logic operators
|
|
|
|
/// elementwise equality operator
|
|
template <
|
|
size_t S,
|
|
typename T,
|
|
template <size_t,typename> class K
|
|
>
|
|
bool
|
|
operator== (K<S,T> a, K<S,T> b)
|
|
{
|
|
bool (*predicate)(const T&, const T&) = almost_equal;
|
|
|
|
return std::equal (std::begin (a),
|
|
std::end (a),
|
|
std::begin (b),
|
|
predicate);
|
|
}
|
|
|
|
///------------------------------------------------------------------------
|
|
/// elementwise inquality operator
|
|
template <
|
|
size_t S,
|
|
typename T,
|
|
template <size_t,typename> class K
|
|
>
|
|
bool
|
|
operator!= (K<S,T> a, K<S,T> b)
|
|
{
|
|
return !(a == b);
|
|
}
|
|
|
|
|
|
///////////////////////////////////////////////////////////////////////////
|
|
// special operators
|
|
|
|
/// point-point subtraction giving a vector difference
|
|
template <
|
|
size_t S,
|
|
typename T,
|
|
typename U
|
|
>
|
|
vector<S,typename std::common_type<T,U>::type>
|
|
operator- (point<S,T> a, point<S,U> b)
|
|
{
|
|
vector<S,typename std::common_type<T,U>::type> out;
|
|
for (size_t i = 0; i < S; ++i)
|
|
out[i] = a[i] - b[i];
|
|
return out;
|
|
}
|
|
|
|
|
|
//-------------------------------------------------------------------------
|
|
template <
|
|
size_t S,
|
|
typename T,
|
|
typename U
|
|
>
|
|
vector<S,typename std::common_type<T,U>::type>
|
|
operator- (U u, point<S,T> p)
|
|
{
|
|
return point<S,U> {u} - p;
|
|
}
|
|
|
|
|
|
//-------------------------------------------------------------------------
|
|
template <
|
|
size_t S,
|
|
typename T
|
|
>
|
|
T dot (const T (&a)[S], const T (&b)[S])
|
|
{
|
|
T sum = 0;
|
|
for (size_t i = 0; i < S; ++i)
|
|
sum += a[i] * b[i];
|
|
return sum;
|
|
}
|
|
|
|
template <
|
|
size_t S,
|
|
typename T,
|
|
template <size_t,typename> class A,
|
|
template <size_t,typename> class B
|
|
>
|
|
typename std::enable_if<
|
|
is_coord<A>::value && is_coord<B>::value,
|
|
T
|
|
>::type
|
|
dot (A<S,T> a, B<S,T> b)
|
|
{
|
|
return dot<S,T> (a.data, b.data);
|
|
}
|
|
|
|
|
|
template <
|
|
size_t S,
|
|
typename T,
|
|
template <size_t,typename> class K
|
|
>
|
|
typename std::enable_if<is_coord<K>::value,T>::type
|
|
dot (K<S,T> a, const T (&b)[S])
|
|
{
|
|
return dot<S,T> (a.data, b);
|
|
}
|
|
|
|
template <
|
|
size_t S,
|
|
typename T,
|
|
template <size_t,typename> class K
|
|
>
|
|
typename std::enable_if<
|
|
is_coord<K>::value,
|
|
T
|
|
>::type
|
|
dot (const T (&a)[S], K<S,T> b)
|
|
{
|
|
return dot<S,T> (a, b.data);
|
|
}
|
|
|
|
|
|
//-------------------------------------------------------------------------
|
|
template <
|
|
size_t S,
|
|
typename T,
|
|
template <size_t,typename> class K
|
|
>
|
|
K<S,T>
|
|
abs (K<S,T> k)
|
|
{
|
|
for (auto &v: k)
|
|
v = std::abs (v);
|
|
return k;
|
|
}
|
|
|
|
//-------------------------------------------------------------------------
|
|
template <
|
|
size_t S,
|
|
typename T,
|
|
template <size_t,typename> class K
|
|
>
|
|
K<S,T>
|
|
pow (K<S,T> k)
|
|
{
|
|
for (auto &v: k)
|
|
v = pow (v);
|
|
return k;
|
|
}
|
|
|
|
|
|
|
|
///////////////////////////////////////////////////////////////////////////
|
|
// logical element operators
|
|
|
|
/// return a coord type containing the max element at each offset
|
|
template <
|
|
size_t S,
|
|
typename T,
|
|
template <size_t,typename> class K
|
|
>
|
|
K<S,T>
|
|
min (K<S,T> a, K<S,T> b)
|
|
{
|
|
K<S,T> out;
|
|
for (size_t i = 0; i < S; ++i)
|
|
out[i] = min (a[i], b[i]);
|
|
return out;
|
|
}
|
|
|
|
|
|
///------------------------------------------------------------------------
|
|
// /return a coord type containing the max element at each offset
|
|
template <
|
|
size_t S,
|
|
typename T,
|
|
template <size_t,typename> class K
|
|
>
|
|
K<S,T>
|
|
max (K<S,T> a, K<S,T> b)
|
|
{
|
|
K<S,T> out;
|
|
for (size_t i = 0; i < S; ++i)
|
|
out[i] = max (a[i], b[i]);
|
|
return out;
|
|
}
|
|
|
|
|
|
///------------------------------------------------------------------------
|
|
template <size_t S, typename T, template<size_t,typename> class K>
|
|
T
|
|
min (K<S,T> k)
|
|
{ return *std::min_element (k.begin (), k.end ()); }
|
|
|
|
|
|
template <size_t S, typename T, template<size_t,typename> class K>
|
|
T
|
|
max (K<S,T> k)
|
|
{ return *std::max_element (k.begin (), k.end ()); }
|
|
|
|
|
|
//-------------------------------------------------------------------------
|
|
#define VECTOR_OP(OP) \
|
|
template < \
|
|
size_t S, \
|
|
typename T, \
|
|
typename U, \
|
|
template <size_t,typename> class A, \
|
|
template <size_t,typename> class B \
|
|
> \
|
|
vector<S,bool> \
|
|
operator OP (const A<S,T> a, const B<S,U> b) \
|
|
{ \
|
|
vector<S,bool> out; \
|
|
for (size_t i = 0; i < S; ++i) \
|
|
out[i] = a[i] OP b[i]; \
|
|
return out; \
|
|
}
|
|
|
|
VECTOR_OP(<)
|
|
VECTOR_OP(>)
|
|
VECTOR_OP(<=)
|
|
VECTOR_OP(>=)
|
|
|
|
#undef VECTOR_OP
|
|
|
|
|
|
#define SCALAR_OP(OP) \
|
|
template < \
|
|
size_t S, \
|
|
typename T, \
|
|
typename U, \
|
|
template <size_t,typename> class K \
|
|
> \
|
|
vector<S,bool> \
|
|
operator OP (const K<S,T> k, const U u) \
|
|
{ \
|
|
vector<S,bool> out; \
|
|
for (size_t i = 0; i < S; ++i) \
|
|
out[i] = k[i] OP u; \
|
|
return out; \
|
|
}
|
|
|
|
SCALAR_OP(<)
|
|
SCALAR_OP(>)
|
|
SCALAR_OP(<=)
|
|
SCALAR_OP(>=)
|
|
|
|
#undef SCALAR_OP
|
|
|
|
|
|
//-------------------------------------------------------------------------
|
|
template <size_t S, template <size_t,typename> class K>
|
|
bool
|
|
any (const K<S,bool> k)
|
|
{
|
|
return std::any_of (k.begin (), k.end (), identity<bool>);
|
|
}
|
|
|
|
//-------------------------------------------------------------------------
|
|
template <size_t S, template <size_t,typename> class K>
|
|
bool
|
|
all (const K<S,bool> k)
|
|
{
|
|
return std::all_of (k.begin (), k.end (), identity<bool>);
|
|
}
|
|
|
|
|
|
///////////////////////////////////////////////////////////////////////////
|
|
template <size_t S, typename T, template<size_t,typename> class K>
|
|
typename std::enable_if<
|
|
std::is_floating_point<T>::value,
|
|
K<S,T>
|
|
>::type
|
|
floor (K<S,T> k)
|
|
{
|
|
T (*floor_func)(T) = std::floor;
|
|
|
|
K<S,T> v;
|
|
std::transform (k.begin (), k.end (), v.begin (), floor_func);
|
|
return v;
|
|
}
|
|
}
|
|
|
|
#endif
|