CoCalc -- vec3ia

GitHub Repository: godotengine/godot
Path: blob/master/thirdparty/embree/common/math/vec3ia_sycl.h
⁹⁹¹² views
1
// Copyright 2009-2021 Intel Corporation
2
// SPDX-License-Identifier: Apache-2.0
3

4
#pragma once
5

6
#include "../sys/alloc.h"
7
#include "emath.h"
8
#include "../simd/sse.h"
9

10
namespace embree
11
{
12
  ////////////////////////////////////////////////////////////////////////////////
13
  /// SSE Vec3ia Type
14
  ////////////////////////////////////////////////////////////////////////////////
15

16
  struct __aligned(16) Vec3ia
17
  {
18
    ALIGNED_STRUCT_(16);
19

20
    struct { int x,y,z; };
21

22
    typedef int Scalar;
23
    enum { N = 3 };
24

25
    ////////////////////////////////////////////////////////////////////////////////
26
    /// Constructors, Assignment & Cast Operators
27
    ////////////////////////////////////////////////////////////////////////////////
28

29
    __forceinline Vec3ia( ) {}
30
    //__forceinline Vec3ia( const __m128i a ) : m128(a) {}
31

32
    __forceinline Vec3ia( const Vec3ia& other ) : x(other.x), y(other.y), z(other.z) {}
33
    __forceinline Vec3ia& operator =(const Vec3ia& other) { x = other.x; y = other.y; z = other.z; return *this; }
34

35
    __forceinline explicit Vec3ia( const int a ) : x(a), y(a), z(a) {}
36
    __forceinline          Vec3ia( const int x, const int y, const int z) : x(x), y(y), z(z) {}
37
    //__forceinline explicit Vec3ia( const __m128 a ) : m128(_mm_cvtps_epi32(a)) {}
38
    __forceinline explicit Vec3ia(const vint4& a) : x(a[0]), y(a[1]), z(a[2]) {}
39

40
    __forceinline explicit Vec3ia( const Vec3fa& a );
41

42
    //__forceinline operator const __m128i&() const { return m128; }
43
    //__forceinline operator       __m128i&()       { return m128; }
44
    __forceinline operator vint4() const { return vint4(x,y,z,z); }
45

46
    ////////////////////////////////////////////////////////////////////////////////
47
    /// Constants
48
    ////////////////////////////////////////////////////////////////////////////////
49

50
    __forceinline Vec3ia( ZeroTy   ) : x(0), y(0), z(0) {}
51
    __forceinline Vec3ia( OneTy    ) : x(1), y(1), z(1) {}
52
    __forceinline Vec3ia( PosInfTy ) : x(0x7FFFFFFF), y(0x7FFFFFFF), z(0x7FFFFFFF) {}
53
    __forceinline Vec3ia( NegInfTy ) : x(0x80000000), y(0x80000000), z(0x80000000) {}
54

55
    ////////////////////////////////////////////////////////////////////////////////
56
    /// Array Access
57
    ////////////////////////////////////////////////////////////////////////////////
58

59
    __forceinline const int& operator []( const size_t index ) const { assert(index < 3); return (&x)[index]; }
60
    __forceinline       int& operator []( const size_t index )       { assert(index < 3); return (&x)[index]; }
61
  };
62

63

64
  ////////////////////////////////////////////////////////////////////////////////
65
  /// Unary Operators
66
  ////////////////////////////////////////////////////////////////////////////////
67

68
  __forceinline Vec3ia operator +( const Vec3ia& a ) { return Vec3ia(+a.x,+a.y,+a.z); }
69
  __forceinline Vec3ia operator -( const Vec3ia& a ) { return Vec3ia(-a.x,-a.y,-a.z); }
70
  __forceinline Vec3ia abs       ( const Vec3ia& a ) { return Vec3ia(sycl::abs(a.x),sycl::abs(a.y),sycl::abs(a.z)); }
71

72
  ////////////////////////////////////////////////////////////////////////////////
73
  /// Binary Operators
74
  ////////////////////////////////////////////////////////////////////////////////
75

76
  __forceinline Vec3ia operator +( const Vec3ia& a, const Vec3ia& b ) { return Vec3ia(a.x+b.x, a.y+b.y, a.z+b.z); }
77
  __forceinline Vec3ia operator +( const Vec3ia& a, const int     b ) { return a+Vec3ia(b); }
78
  __forceinline Vec3ia operator +( const int     a, const Vec3ia& b ) { return Vec3ia(a)+b; }
79

80
  __forceinline Vec3ia operator -( const Vec3ia& a, const Vec3ia& b ) { return Vec3ia(a.x-b.x, a.y-b.y, a.z-b.z); }
81
  __forceinline Vec3ia operator -( const Vec3ia& a, const int     b ) { return a-Vec3ia(b); }
82
  __forceinline Vec3ia operator -( const int     a, const Vec3ia& b ) { return Vec3ia(a)-b; }
83

84
  __forceinline Vec3ia operator *( const Vec3ia& a, const Vec3ia& b ) { return Vec3ia(a.x*b.x, a.y*b.y, a.z*b.z); }
85
  __forceinline Vec3ia operator *( const Vec3ia& a, const int     b ) { return a * Vec3ia(b); }
86
  __forceinline Vec3ia operator *( const int     a, const Vec3ia& b ) { return Vec3ia(a) * b; }
87

88
  __forceinline Vec3ia operator &( const Vec3ia& a, const Vec3ia& b ) { return Vec3ia(a.x&b.x, a.y&b.y, a.z&b.z); }
89
  __forceinline Vec3ia operator &( const Vec3ia& a, const int     b ) { return a & Vec3ia(b); }
90
  __forceinline Vec3ia operator &( const int     a, const Vec3ia& b ) { return Vec3ia(a) & b; }
91

92
  __forceinline Vec3ia operator |( const Vec3ia& a, const Vec3ia& b ) { return Vec3ia(a.x|b.x, a.y|b.y, a.z|b.z); }
93
  __forceinline Vec3ia operator |( const Vec3ia& a, const int     b ) { return a | Vec3ia(b); }
94
  __forceinline Vec3ia operator |( const int     a, const Vec3ia& b ) { return Vec3ia(a) | b; }
95

96
  __forceinline Vec3ia operator ^( const Vec3ia& a, const Vec3ia& b ) { return Vec3ia(a.x^b.x, a.y^b.y, a.z^b.z); }
97
  __forceinline Vec3ia operator ^( const Vec3ia& a, const int     b ) { return a ^ Vec3ia(b); }
98
  __forceinline Vec3ia operator ^( const int     a, const Vec3ia& b ) { return Vec3ia(a) ^ b; }
99

100
  __forceinline Vec3ia operator <<( const Vec3ia& a, const int n ) { return Vec3ia(a.x<<n, a.y<<n, a.z<<n); }
101
  __forceinline Vec3ia operator >>( const Vec3ia& a, const int n ) { return Vec3ia(a.x>>n, a.y>>n, a.z>>n); }
102

103
  __forceinline Vec3ia sll ( const Vec3ia& a, const int b ) { return Vec3ia(a.x<<b, a.y<<b, a.z<<b); }
104
  __forceinline Vec3ia sra ( const Vec3ia& a, const int b ) { return Vec3ia(a.x>>b, a.y>>b, a.z>>b); }
105
  __forceinline Vec3ia srl ( const Vec3ia& a, const int b ) { return Vec3ia(unsigned(a.x)>>b, unsigned(a.y)>>b, unsigned(a.z)>>b); }
106

107
  ////////////////////////////////////////////////////////////////////////////////
108
  /// Assignment Operators
109
  ////////////////////////////////////////////////////////////////////////////////
110

111
  __forceinline Vec3ia& operator +=( Vec3ia& a, const Vec3ia& b ) { return a = a + b; }
112
  __forceinline Vec3ia& operator +=( Vec3ia& a, const int&   b ) { return a = a + b; }
113
  
114
  __forceinline Vec3ia& operator -=( Vec3ia& a, const Vec3ia& b ) { return a = a - b; }
115
  __forceinline Vec3ia& operator -=( Vec3ia& a, const int&   b ) { return a = a - b; }
116
  
117
  __forceinline Vec3ia& operator *=( Vec3ia& a, const Vec3ia& b ) { return a = a * b; }
118
  __forceinline Vec3ia& operator *=( Vec3ia& a, const int&    b ) { return a = a * b; }
119
  
120
  __forceinline Vec3ia& operator &=( Vec3ia& a, const Vec3ia& b ) { return a = a & b; }
121
  __forceinline Vec3ia& operator &=( Vec3ia& a, const int&    b ) { return a = a & b; }
122
  
123
  __forceinline Vec3ia& operator |=( Vec3ia& a, const Vec3ia& b ) { return a = a | b; }
124
  __forceinline Vec3ia& operator |=( Vec3ia& a, const int&    b ) { return a = a | b; }
125
  
126
  __forceinline Vec3ia& operator <<=( Vec3ia& a, const int& b ) { return a = a << b; }
127
  __forceinline Vec3ia& operator >>=( Vec3ia& a, const int& b ) { return a = a >> b; }
128

129
  ////////////////////////////////////////////////////////////////////////////////
130
  /// Reductions
131
  ////////////////////////////////////////////////////////////////////////////////
132

133
  __forceinline int reduce_add(const Vec3ia& v) { return v.x+v.y+v.z; }
134
  __forceinline int reduce_mul(const Vec3ia& v) { return v.x*v.y*v.z; }
135
  __forceinline int reduce_min(const Vec3ia& v) { return sycl::min(sycl::min(v.x,v.y),v.z); }
136
  __forceinline int reduce_max(const Vec3ia& v) { return sycl::max(sycl::max(v.x,v.y),v.z); }
137

138
  ////////////////////////////////////////////////////////////////////////////////
139
  /// Comparison Operators
140
  ////////////////////////////////////////////////////////////////////////////////
141

142
  __forceinline bool operator ==( const Vec3ia& a, const Vec3ia& b ) { return a.x == b.x & a.y == b.y & a.z == b.z; }
143
  __forceinline bool operator !=( const Vec3ia& a, const Vec3ia& b ) { return a.x != b.x & a.y != b.y & a.z != b.z; }
144

145
/*
146
  __forceinline bool operator < ( const Vec3ia& a, const Vec3ia& b ) {
147
    if (a.x != b.x) return a.x < b.x;
148
    if (a.y != b.y) return a.y < b.y;
149
    if (a.z != b.z) return a.z < b.z;
150
    return false;
151
  }
152
*/
153
  __forceinline Vec3ba eq_mask( const Vec3ia& a, const Vec3ia& b ) { return Vec3ba(a.x == b.x, a.y == b.y, a.z == b.z); }
154
  __forceinline Vec3ba lt_mask( const Vec3ia& a, const Vec3ia& b ) { return Vec3ba(a.x <  b.x, a.y <  b.y, a.z <  b.z); }
155
  __forceinline Vec3ba gt_mask( const Vec3ia& a, const Vec3ia& b ) { return Vec3ba(a.x >  b.x, a.y >  b.y, a.z >  b.z); }
156

157
  ////////////////////////////////////////////////////////////////////////////////
158
  /// Select
159
  ////////////////////////////////////////////////////////////////////////////////
160
  
161
  __forceinline Vec3ia select( const Vec3ba& m, const Vec3ia& t, const Vec3ia& f ) {
162
    const int x = m.x ? t.x : f.x;
163
    const int y = m.y ? t.y : f.y;
164
    const int z = m.z ? t.z : f.z;
165
    return Vec3ia(x,y,z);
166
  }
167
  
168
  __forceinline Vec3ia min( const Vec3ia& a, const Vec3ia& b ) { return Vec3ia(sycl::min(a.x,b.x), sycl::min(a.y,b.y), sycl::min(a.z,b.z)); }
169
  __forceinline Vec3ia max( const Vec3ia& a, const Vec3ia& b ) { return Vec3ia(sycl::max(a.x,b.x), sycl::max(a.y,b.y), sycl::max(a.z,b.z)); }
170
  
171
  ////////////////////////////////////////////////////////////////////////////////
172
  /// Output Operators
173
  ////////////////////////////////////////////////////////////////////////////////
174
  
175
  inline embree_ostream operator<<(embree_ostream cout, const Vec3ia& a) {
176
    return cout;
177
  }
178
}
179

180
Product

Resources

Company
