image_framework_ymj/include/open3d/3rdparty/math/vec4.h
2024-12-06 16:25:16 +08:00

133 lines
4.2 KiB
C++
Executable File

/*
* Copyright 2013 The Android Open Source Project
*
* 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.
*/
#ifndef MATH_VEC4_H_
#define MATH_VEC4_H_
#include <math/vec3.h>
#include <math/half.h>
#include <stdint.h>
#include <sys/types.h>
namespace filament {
namespace math {
// -------------------------------------------------------------------------------------
namespace details {
template<typename T>
class MATH_EMPTY_BASES TVec4 :
public TVecProductOperators<TVec4, T>,
public TVecAddOperators<TVec4, T>,
public TVecUnaryOperators<TVec4, T>,
public TVecComparisonOperators<TVec4, T>,
public TVecFunctions<TVec4, T> {
public:
typedef T value_type;
typedef T& reference;
typedef T const& const_reference;
typedef size_t size_type;
static constexpr size_t SIZE = 4;
union {
T v[SIZE] MATH_CONSTEXPR_INIT;
TVec2<T> xy, st, rg;
TVec3<T> xyz, stp, rgb;
struct {
union { T x, s, r; };
union {
TVec2<T> yz, tp, gb;
TVec3<T> yzw, tpq, gba;
struct {
union { T y, t, g; };
union {
TVec2<T> zw, pq, ba;
struct { T z, w; };
struct { T p, q; };
struct { T b, a; };
};
};
};
};
};
inline constexpr size_type size() const { return SIZE; }
// array access
inline constexpr T const& operator[](size_t i) const noexcept {
assert(i < SIZE);
return v[i];
}
inline constexpr T& operator[](size_t i) noexcept {
assert(i < SIZE);
return v[i];
}
// constructors
// default constructor
MATH_DEFAULT_CTOR_CONSTEXPR TVec4() noexcept MATH_DEFAULT_CTOR
// handles implicit conversion to a tvec4. must not be explicit.
template<typename A, typename = enable_if_arithmetic_t<A>>
constexpr TVec4(A v) noexcept : v{ T(v), T(v), T(v), T(v) } {}
template<typename A, typename B, typename C, typename D,
typename = enable_if_arithmetic_t<A, B, C, D>>
constexpr TVec4(A x, B y, C z, D w) noexcept : v{ T(x), T(y), T(z), T(w) } {}
template<typename A, typename B, typename C,
typename = enable_if_arithmetic_t<A, B, C>>
constexpr TVec4(const TVec2<A>& v, B z, C w) noexcept : v{ T(v[0]), T(v[1]), T(z), T(w) } {}
template<typename A, typename B, typename = enable_if_arithmetic_t<A, B>>
constexpr TVec4(const TVec2<A>& v, const TVec2<B>& w) noexcept : v{
T(v[0]), T(v[1]), T(w[0]), T(w[1]) } {}
template<typename A, typename B, typename = enable_if_arithmetic_t<A, B>>
constexpr TVec4(const TVec3<A>& v, B w) noexcept : v{ T(v[0]), T(v[1]), T(v[2]), T(w) } {}
template<typename A, typename = enable_if_arithmetic_t<A>>
constexpr TVec4(const TVec4<A>& v) noexcept : v{ T(v[0]), T(v[1]), T(v[2]), T(v[3]) } {}
};
} // namespace details
// ----------------------------------------------------------------------------------------
template<typename T, typename = details::enable_if_arithmetic_t<T>>
using vec4 = details::TVec4<T>;
using double4 = vec4<double>;
using float4 = vec4<float>;
using half4 = vec4<half>;
using int4 = vec4<int32_t>;
using uint4 = vec4<uint32_t>;
using short4 = vec4<int16_t>;
using ushort4 = vec4<uint16_t>;
using byte4 = vec4<int8_t>;
using ubyte4 = vec4<uint8_t>;
using bool4 = vec4<bool>;
// ----------------------------------------------------------------------------------------
} // namespace math
} // namespace filament
#endif // MATH_VEC4_H_