blob: 4384ec682092ffb414aaea0835ca1bb3c41721f2 [file] [log] [blame]
Jason Sams1b937f52010-06-09 14:26:16 -07001#ifndef __RS_CL_RSH__
2#define __RS_CL_RSH__
3
Stephen Hinesf6a28c62011-01-18 16:53:19 -08004#ifdef BCC_PREPARE_BC
5#define _RS_STATIC extern
6#else
7#define _RS_STATIC static
8#endif
Jason Sams1b937f52010-06-09 14:26:16 -07009
Stephen Hinesf6a28c62011-01-18 16:53:19 -080010#define M_PI 3.14159265358979323846264338327950288f /* pi */
Jason Sams1b937f52010-06-09 14:26:16 -070011
12// Conversions
13#define CVT_FUNC_2(typeout, typein) \
Stephen Hinesf6a28c62011-01-18 16:53:19 -080014_RS_STATIC typeout##2 __attribute__((overloadable)) convert_##typeout##2(typein##2 v) { \
Jason Sams5e9811f2010-06-21 17:42:41 -070015 typeout##2 r = {(typeout)v.x, (typeout)v.y}; \
Jason Sams1b937f52010-06-09 14:26:16 -070016 return r; \
17} \
Stephen Hinesf6a28c62011-01-18 16:53:19 -080018_RS_STATIC typeout##3 __attribute__((overloadable)) convert_##typeout##3(typein##3 v) { \
Jason Sams5e9811f2010-06-21 17:42:41 -070019 typeout##3 r = {(typeout)v.x, (typeout)v.y, (typeout)v.z}; \
Jason Sams1b937f52010-06-09 14:26:16 -070020 return r; \
21} \
Stephen Hinesf6a28c62011-01-18 16:53:19 -080022_RS_STATIC typeout##4 __attribute__((overloadable)) convert_##typeout##4(typein##4 v) { \
Jason Sams5e9811f2010-06-21 17:42:41 -070023 typeout##4 r = {(typeout)v.x, (typeout)v.y, (typeout)v.z, (typeout)v.w}; \
Jason Sams1b937f52010-06-09 14:26:16 -070024 return r; \
25}
26
27#define CVT_FUNC(type) CVT_FUNC_2(type, uchar) \
28 CVT_FUNC_2(type, char) \
29 CVT_FUNC_2(type, ushort) \
30 CVT_FUNC_2(type, short) \
Jason Sams1b937f52010-06-09 14:26:16 -070031 CVT_FUNC_2(type, uint) \
Stephen Hines47416532011-01-11 14:38:18 -080032 CVT_FUNC_2(type, int) \
Jason Sams1b937f52010-06-09 14:26:16 -070033 CVT_FUNC_2(type, float)
34
35CVT_FUNC(char)
36CVT_FUNC(uchar)
37CVT_FUNC(short)
38CVT_FUNC(ushort)
39CVT_FUNC(int)
40CVT_FUNC(uint)
41CVT_FUNC(float)
42
43
44
Jason Sams22fa3712010-05-19 17:22:57 -070045// Float ops, 6.11.2
46
Jason Sams1b937f52010-06-09 14:26:16 -070047#define DEF_FUNC_1(fnc) \
Stephen Hinesf6a28c62011-01-18 16:53:19 -080048_RS_STATIC float2 __attribute__((overloadable)) fnc(float2 v) { \
Jason Sams1b937f52010-06-09 14:26:16 -070049 float2 r; \
50 r.x = fnc(v.x); \
51 r.y = fnc(v.y); \
52 return r; \
53} \
Stephen Hinesf6a28c62011-01-18 16:53:19 -080054_RS_STATIC float3 __attribute__((overloadable)) fnc(float3 v) { \
Jason Sams1b937f52010-06-09 14:26:16 -070055 float3 r; \
56 r.x = fnc(v.x); \
57 r.y = fnc(v.y); \
58 r.z = fnc(v.z); \
59 return r; \
60} \
Stephen Hinesf6a28c62011-01-18 16:53:19 -080061_RS_STATIC float4 __attribute__((overloadable)) fnc(float4 v) { \
Jason Sams1b937f52010-06-09 14:26:16 -070062 float4 r; \
63 r.x = fnc(v.x); \
64 r.y = fnc(v.y); \
65 r.z = fnc(v.z); \
66 r.w = fnc(v.w); \
67 return r; \
68}
69
Stephen Hines47416532011-01-11 14:38:18 -080070#define DEF_FUNC_1_RI(fnc) \
Stephen Hinesf6a28c62011-01-18 16:53:19 -080071_RS_STATIC int2 __attribute__((overloadable)) fnc(float2 v) { \
Stephen Hines47416532011-01-11 14:38:18 -080072 int2 r; \
73 r.x = fnc(v.x); \
74 r.y = fnc(v.y); \
75 return r; \
76} \
Stephen Hinesf6a28c62011-01-18 16:53:19 -080077_RS_STATIC int3 __attribute__((overloadable)) fnc(float3 v) { \
Stephen Hines47416532011-01-11 14:38:18 -080078 int3 r; \
79 r.x = fnc(v.x); \
80 r.y = fnc(v.y); \
81 r.z = fnc(v.z); \
82 return r; \
83} \
Stephen Hinesf6a28c62011-01-18 16:53:19 -080084_RS_STATIC int4 __attribute__((overloadable)) fnc(float4 v) { \
Stephen Hines47416532011-01-11 14:38:18 -080085 int4 r; \
86 r.x = fnc(v.x); \
87 r.y = fnc(v.y); \
88 r.z = fnc(v.z); \
89 r.w = fnc(v.w); \
90 return r; \
91}
92
Jason Sams1b937f52010-06-09 14:26:16 -070093#define DEF_FUNC_2(fnc) \
Stephen Hinesf6a28c62011-01-18 16:53:19 -080094_RS_STATIC float2 __attribute__((overloadable)) fnc(float2 v1, float2 v2) { \
Jason Sams1b937f52010-06-09 14:26:16 -070095 float2 r; \
96 r.x = fnc(v1.x, v2.x); \
97 r.y = fnc(v1.y, v2.y); \
98 return r; \
99} \
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800100_RS_STATIC float3 __attribute__((overloadable)) fnc(float3 v1, float3 v2) { \
Jason Sams1b937f52010-06-09 14:26:16 -0700101 float3 r; \
102 r.x = fnc(v1.x, v2.x); \
103 r.y = fnc(v1.y, v2.y); \
104 r.z = fnc(v1.z, v2.z); \
105 return r; \
106} \
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800107_RS_STATIC float4 __attribute__((overloadable)) fnc(float4 v1, float4 v2) { \
Jason Sams1b937f52010-06-09 14:26:16 -0700108 float4 r; \
109 r.x = fnc(v1.x, v2.x); \
110 r.y = fnc(v1.y, v2.y); \
111 r.z = fnc(v1.z, v2.z); \
112 r.w = fnc(v1.w, v2.z); \
113 return r; \
114}
115
116#define DEF_FUNC_2F(fnc) \
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800117_RS_STATIC float2 __attribute__((overloadable)) fnc(float2 v1, float v2) { \
Jason Sams1b937f52010-06-09 14:26:16 -0700118 float2 r; \
119 r.x = fnc(v1.x, v2); \
120 r.y = fnc(v1.y, v2); \
121 return r; \
122} \
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800123_RS_STATIC float3 __attribute__((overloadable)) fnc(float3 v1, float v2) { \
Jason Sams1b937f52010-06-09 14:26:16 -0700124 float3 r; \
125 r.x = fnc(v1.x, v2); \
126 r.y = fnc(v1.y, v2); \
127 r.z = fnc(v1.z, v2); \
128 return r; \
129} \
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800130_RS_STATIC float4 __attribute__((overloadable)) fnc(float4 v1, float v2) { \
Jason Sams1b937f52010-06-09 14:26:16 -0700131 float4 r; \
132 r.x = fnc(v1.x, v2); \
133 r.y = fnc(v1.y, v2); \
134 r.z = fnc(v1.z, v2); \
135 r.w = fnc(v1.w, v2); \
136 return r; \
137}
138
139
Jason Sams22fa3712010-05-19 17:22:57 -0700140extern float __attribute__((overloadable)) acos(float);
Jason Sams1b937f52010-06-09 14:26:16 -0700141DEF_FUNC_1(acos)
Jason Sams22fa3712010-05-19 17:22:57 -0700142
143extern float __attribute__((overloadable)) acosh(float);
Jason Sams1b937f52010-06-09 14:26:16 -0700144DEF_FUNC_1(acosh)
Jason Sams22fa3712010-05-19 17:22:57 -0700145
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800146_RS_STATIC float __attribute__((overloadable)) acospi(float v) {
Jason Sams1b937f52010-06-09 14:26:16 -0700147 return acos(v) / M_PI;
148}
149DEF_FUNC_1(acospi)
Jason Sams22fa3712010-05-19 17:22:57 -0700150
151extern float __attribute__((overloadable)) asin(float);
Jason Sams1b937f52010-06-09 14:26:16 -0700152DEF_FUNC_1(asin)
Jason Sams22fa3712010-05-19 17:22:57 -0700153
154extern float __attribute__((overloadable)) asinh(float);
Jason Sams1b937f52010-06-09 14:26:16 -0700155DEF_FUNC_1(asinh)
Jason Sams22fa3712010-05-19 17:22:57 -0700156
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800157_RS_STATIC float __attribute__((overloadable)) asinpi(float v) {
Jason Sams1b937f52010-06-09 14:26:16 -0700158 return asin(v) / M_PI;
159}
160DEF_FUNC_1(asinpi)
Jason Sams22fa3712010-05-19 17:22:57 -0700161
162extern float __attribute__((overloadable)) atan(float);
Jason Sams1b937f52010-06-09 14:26:16 -0700163DEF_FUNC_1(atan)
Jason Sams22fa3712010-05-19 17:22:57 -0700164
165extern float __attribute__((overloadable)) atan2(float, float);
Jason Sams1b937f52010-06-09 14:26:16 -0700166DEF_FUNC_2(atan2)
Jason Sams22fa3712010-05-19 17:22:57 -0700167
168extern float __attribute__((overloadable)) atanh(float);
Jason Sams1b937f52010-06-09 14:26:16 -0700169DEF_FUNC_1(atanh)
Jason Sams22fa3712010-05-19 17:22:57 -0700170
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800171_RS_STATIC float __attribute__((overloadable)) atanpi(float v) {
Jason Sams1b937f52010-06-09 14:26:16 -0700172 return atan(v) / M_PI;
173}
174DEF_FUNC_1(atanpi)
Jason Sams22fa3712010-05-19 17:22:57 -0700175
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800176_RS_STATIC float __attribute__((overloadable)) atan2pi(float y, float x) {
Jason Sams1b937f52010-06-09 14:26:16 -0700177 return atan2(y, x) / M_PI;
178}
179DEF_FUNC_2(atan2pi)
Jason Sams22fa3712010-05-19 17:22:57 -0700180
181extern float __attribute__((overloadable)) cbrt(float);
Jason Sams1b937f52010-06-09 14:26:16 -0700182DEF_FUNC_1(cbrt)
Jason Sams22fa3712010-05-19 17:22:57 -0700183
184extern float __attribute__((overloadable)) ceil(float);
Jason Sams1b937f52010-06-09 14:26:16 -0700185DEF_FUNC_1(ceil)
Jason Sams22fa3712010-05-19 17:22:57 -0700186
187extern float __attribute__((overloadable)) copysign(float, float);
Jason Sams1b937f52010-06-09 14:26:16 -0700188DEF_FUNC_2(copysign)
Jason Sams22fa3712010-05-19 17:22:57 -0700189
190extern float __attribute__((overloadable)) cos(float);
Jason Sams1b937f52010-06-09 14:26:16 -0700191DEF_FUNC_1(cos)
Jason Sams22fa3712010-05-19 17:22:57 -0700192
193extern float __attribute__((overloadable)) cosh(float);
Jason Sams1b937f52010-06-09 14:26:16 -0700194DEF_FUNC_1(cosh)
Jason Sams22fa3712010-05-19 17:22:57 -0700195
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800196_RS_STATIC float __attribute__((overloadable)) cospi(float v) {
Jason Sams1b937f52010-06-09 14:26:16 -0700197 return cos(v * M_PI);
198}
199DEF_FUNC_1(cospi)
Jason Sams22fa3712010-05-19 17:22:57 -0700200
201extern float __attribute__((overloadable)) erfc(float);
Jason Sams1b937f52010-06-09 14:26:16 -0700202DEF_FUNC_1(erfc)
Jason Sams22fa3712010-05-19 17:22:57 -0700203
204extern float __attribute__((overloadable)) erf(float);
Jason Sams1b937f52010-06-09 14:26:16 -0700205DEF_FUNC_1(erf)
Jason Sams22fa3712010-05-19 17:22:57 -0700206
207extern float __attribute__((overloadable)) exp(float);
Jason Sams1b937f52010-06-09 14:26:16 -0700208DEF_FUNC_1(exp)
Jason Sams22fa3712010-05-19 17:22:57 -0700209
210extern float __attribute__((overloadable)) exp2(float);
Jason Sams1b937f52010-06-09 14:26:16 -0700211DEF_FUNC_1(exp2)
Jason Sams22fa3712010-05-19 17:22:57 -0700212
Jason Sams1b937f52010-06-09 14:26:16 -0700213extern float __attribute__((overloadable)) pow(float, float);
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800214_RS_STATIC float __attribute__((overloadable)) exp10(float v) {
Jason Sams1b937f52010-06-09 14:26:16 -0700215 return pow(10.f, v);
216}
217DEF_FUNC_1(exp10)
Jason Sams22fa3712010-05-19 17:22:57 -0700218
219extern float __attribute__((overloadable)) expm1(float);
Jason Sams1b937f52010-06-09 14:26:16 -0700220DEF_FUNC_1(expm1)
Jason Sams22fa3712010-05-19 17:22:57 -0700221
222extern float __attribute__((overloadable)) fabs(float);
Jason Sams1b937f52010-06-09 14:26:16 -0700223DEF_FUNC_1(fabs)
Jason Sams22fa3712010-05-19 17:22:57 -0700224
225extern float __attribute__((overloadable)) fdim(float, float);
Jason Sams1b937f52010-06-09 14:26:16 -0700226DEF_FUNC_2(fdim)
Jason Sams22fa3712010-05-19 17:22:57 -0700227
228extern float __attribute__((overloadable)) floor(float);
Jason Sams1b937f52010-06-09 14:26:16 -0700229DEF_FUNC_1(floor)
Jason Sams22fa3712010-05-19 17:22:57 -0700230
231extern float __attribute__((overloadable)) fma(float, float, float);
232extern float2 __attribute__((overloadable)) fma(float2, float2, float2);
233extern float3 __attribute__((overloadable)) fma(float3, float3, float3);
234extern float4 __attribute__((overloadable)) fma(float4, float4, float4);
Jason Sams22fa3712010-05-19 17:22:57 -0700235
236extern float __attribute__((overloadable)) fmax(float, float);
Jason Sams1b937f52010-06-09 14:26:16 -0700237DEF_FUNC_2(fmax);
238DEF_FUNC_2F(fmax);
Jason Sams22fa3712010-05-19 17:22:57 -0700239
240extern float __attribute__((overloadable)) fmin(float, float);
Jason Sams1b937f52010-06-09 14:26:16 -0700241DEF_FUNC_2(fmin);
242DEF_FUNC_2F(fmin);
Jason Sams22fa3712010-05-19 17:22:57 -0700243
244extern float __attribute__((overloadable)) fmod(float, float);
Jason Sams1b937f52010-06-09 14:26:16 -0700245DEF_FUNC_2(fmod)
Jason Sams22fa3712010-05-19 17:22:57 -0700246
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800247_RS_STATIC float __attribute__((overloadable)) fract(float v, float *iptr) {
Jason Sams1b937f52010-06-09 14:26:16 -0700248 int i = (int)floor(v);
249 iptr[0] = i;
250 return fmin(v - i, 0x1.fffffep-1f);
251}
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800252_RS_STATIC float2 __attribute__((overloadable)) fract(float2 v, float2 *iptr) {
Jason Sams1b937f52010-06-09 14:26:16 -0700253 float t[2];
254 float2 r;
255 r.x = fract(v.x, &t[0]);
256 r.y = fract(v.y, &t[1]);
257 iptr[0] = t[0];
258 iptr[1] = t[1];
259 return r;
260}
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800261_RS_STATIC float3 __attribute__((overloadable)) fract(float3 v, float3 *iptr) {
Jason Sams1b937f52010-06-09 14:26:16 -0700262 float t[3];
263 float3 r;
264 r.x = fract(v.x, &t[0]);
265 r.y = fract(v.y, &t[1]);
266 r.z = fract(v.z, &t[2]);
267 iptr[0] = t[0];
268 iptr[1] = t[1];
269 iptr[2] = t[2];
270 return r;
271}
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800272_RS_STATIC float4 __attribute__((overloadable)) fract(float4 v, float4 *iptr) {
Jason Sams1b937f52010-06-09 14:26:16 -0700273 float t[4];
274 float4 r;
275 r.x = fract(v.x, &t[0]);
276 r.y = fract(v.y, &t[1]);
277 r.z = fract(v.z, &t[2]);
278 r.w = fract(v.w, &t[3]);
279 iptr[0] = t[0];
280 iptr[1] = t[1];
281 iptr[2] = t[2];
282 iptr[3] = t[3];
283 return r;
284}
Jason Sams22fa3712010-05-19 17:22:57 -0700285
286extern float __attribute__((overloadable)) frexp(float, float *);
287extern float2 __attribute__((overloadable)) frexp(float2, float2 *);
288extern float3 __attribute__((overloadable)) frexp(float3, float3 *);
289extern float4 __attribute__((overloadable)) frexp(float4, float4 *);
Jason Sams22fa3712010-05-19 17:22:57 -0700290
291extern float __attribute__((overloadable)) hypot(float, float);
Jason Sams1b937f52010-06-09 14:26:16 -0700292DEF_FUNC_2(hypot)
Jason Sams22fa3712010-05-19 17:22:57 -0700293
294extern int __attribute__((overloadable)) ilogb(float);
Stephen Hines47416532011-01-11 14:38:18 -0800295DEF_FUNC_1_RI(ilogb)
Jason Sams22fa3712010-05-19 17:22:57 -0700296
297extern float __attribute__((overloadable)) ldexp(float, int);
298extern float2 __attribute__((overloadable)) ldexp(float2, int2);
299extern float3 __attribute__((overloadable)) ldexp(float3, int3);
300extern float4 __attribute__((overloadable)) ldexp(float4, int4);
Jason Sams22fa3712010-05-19 17:22:57 -0700301extern float2 __attribute__((overloadable)) ldexp(float2, int);
302extern float3 __attribute__((overloadable)) ldexp(float3, int);
303extern float4 __attribute__((overloadable)) ldexp(float4, int);
Jason Sams22fa3712010-05-19 17:22:57 -0700304
305extern float __attribute__((overloadable)) lgamma(float);
Jason Sams1b937f52010-06-09 14:26:16 -0700306DEF_FUNC_1(lgamma)
Jason Sams22fa3712010-05-19 17:22:57 -0700307extern float __attribute__((overloadable)) lgamma(float, float *);
308extern float2 __attribute__((overloadable)) lgamma(float2, float2 *);
309extern float3 __attribute__((overloadable)) lgamma(float3, float3 *);
310extern float4 __attribute__((overloadable)) lgamma(float4, float4 *);
Jason Sams22fa3712010-05-19 17:22:57 -0700311
312extern float __attribute__((overloadable)) log(float);
Jason Sams1b937f52010-06-09 14:26:16 -0700313DEF_FUNC_1(log)
Jason Sams22fa3712010-05-19 17:22:57 -0700314
Jason Sams22fa3712010-05-19 17:22:57 -0700315
316extern float __attribute__((overloadable)) log10(float);
Jason Sams1b937f52010-06-09 14:26:16 -0700317DEF_FUNC_1(log10)
318
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800319_RS_STATIC float __attribute__((overloadable)) log2(float v) {
Jason Sams1b937f52010-06-09 14:26:16 -0700320 return log10(v) / log10(2.f);
321}
322DEF_FUNC_1(log2)
Jason Sams22fa3712010-05-19 17:22:57 -0700323
324extern float __attribute__((overloadable)) log1p(float);
Jason Sams1b937f52010-06-09 14:26:16 -0700325DEF_FUNC_1(log1p)
Jason Sams22fa3712010-05-19 17:22:57 -0700326
327extern float __attribute__((overloadable)) logb(float);
Jason Sams1b937f52010-06-09 14:26:16 -0700328DEF_FUNC_1(logb)
Jason Sams22fa3712010-05-19 17:22:57 -0700329
330extern float __attribute__((overloadable)) mad(float, float, float);
331extern float2 __attribute__((overloadable)) mad(float2, float2, float2);
332extern float3 __attribute__((overloadable)) mad(float3, float3, float3);
333extern float4 __attribute__((overloadable)) mad(float4, float4, float4);
Jason Sams22fa3712010-05-19 17:22:57 -0700334
335extern float __attribute__((overloadable)) modf(float, float *);
336extern float2 __attribute__((overloadable)) modf(float2, float2 *);
337extern float3 __attribute__((overloadable)) modf(float3, float3 *);
338extern float4 __attribute__((overloadable)) modf(float4, float4 *);
Jason Sams22fa3712010-05-19 17:22:57 -0700339
Jason Sams1b937f52010-06-09 14:26:16 -0700340//extern float __attribute__((overloadable)) nan(uint);
Jason Sams22fa3712010-05-19 17:22:57 -0700341
342extern float __attribute__((overloadable)) nextafter(float, float);
Jason Sams1b937f52010-06-09 14:26:16 -0700343DEF_FUNC_2(nextafter)
Jason Sams22fa3712010-05-19 17:22:57 -0700344
Jason Sams1b937f52010-06-09 14:26:16 -0700345DEF_FUNC_2(pow)
Jason Sams22fa3712010-05-19 17:22:57 -0700346
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800347_RS_STATIC float __attribute__((overloadable)) pown(float v, int p) {
Jason Sams1b937f52010-06-09 14:26:16 -0700348 return pow(v, (float)p);
349}
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800350_RS_STATIC float2 __attribute__((overloadable)) pown(float2 v, int2 p) {
Jason Sams1b937f52010-06-09 14:26:16 -0700351 return pow(v, (float2)p);
352}
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800353_RS_STATIC float3 __attribute__((overloadable)) pown(float3 v, int3 p) {
Jason Sams1b937f52010-06-09 14:26:16 -0700354 return pow(v, (float3)p);
355}
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800356_RS_STATIC float4 __attribute__((overloadable)) pown(float4 v, int4 p) {
Jason Sams1b937f52010-06-09 14:26:16 -0700357 return pow(v, (float4)p);
358}
Jason Sams22fa3712010-05-19 17:22:57 -0700359
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800360_RS_STATIC float __attribute__((overloadable)) powr(float v, float p) {
Jason Sams1b937f52010-06-09 14:26:16 -0700361 return pow(v, p);
362}
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800363_RS_STATIC float2 __attribute__((overloadable)) powr(float2 v, float2 p) {
Jason Sams1b937f52010-06-09 14:26:16 -0700364 return pow(v, p);
365}
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800366_RS_STATIC float3 __attribute__((overloadable)) powr(float3 v, float3 p) {
Jason Sams1b937f52010-06-09 14:26:16 -0700367 return pow(v, p);
368}
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800369_RS_STATIC float4 __attribute__((overloadable)) powr(float4 v, float4 p) {
Jason Sams1b937f52010-06-09 14:26:16 -0700370 return pow(v, p);
371}
Jason Sams22fa3712010-05-19 17:22:57 -0700372
373extern float __attribute__((overloadable)) remainder(float, float);
Jason Sams1b937f52010-06-09 14:26:16 -0700374DEF_FUNC_2(remainder)
Jason Sams22fa3712010-05-19 17:22:57 -0700375
Stephen Hines47416532011-01-11 14:38:18 -0800376extern float __attribute__((overloadable)) remquo(float, float, int *);
377extern float2 __attribute__((overloadable)) remquo(float2, float2, int2 *);
378extern float3 __attribute__((overloadable)) remquo(float3, float3, int3 *);
379extern float4 __attribute__((overloadable)) remquo(float4, float4, int4 *);
Jason Sams22fa3712010-05-19 17:22:57 -0700380
381extern float __attribute__((overloadable)) rint(float);
Jason Sams1b937f52010-06-09 14:26:16 -0700382DEF_FUNC_1(rint)
Jason Sams22fa3712010-05-19 17:22:57 -0700383
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800384_RS_STATIC float __attribute__((overloadable)) rootn(float v, int r) {
Jason Sams1b937f52010-06-09 14:26:16 -0700385 return pow(v, 1.f / r);
386}
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800387_RS_STATIC float2 __attribute__((overloadable)) rootn(float2 v, int2 r) {
Jason Sams1b937f52010-06-09 14:26:16 -0700388 float2 t = {1.f / r.x, 1.f / r.y};
389 return pow(v, t);
390}
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800391_RS_STATIC float3 __attribute__((overloadable)) rootn(float3 v, int3 r) {
Jason Sams1b937f52010-06-09 14:26:16 -0700392 float3 t = {1.f / r.x, 1.f / r.y, 1.f / r.z};
393 return pow(v, t);
394}
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800395_RS_STATIC float4 __attribute__((overloadable)) rootn(float4 v, int4 r) {
Jason Sams1b937f52010-06-09 14:26:16 -0700396 float4 t = {1.f / r.x, 1.f / r.y, 1.f / r.z, 1.f / r.w};
397 return pow(v, t);
398}
Jason Sams22fa3712010-05-19 17:22:57 -0700399
400extern float __attribute__((overloadable)) round(float);
Jason Sams1b937f52010-06-09 14:26:16 -0700401DEF_FUNC_1(round)
Jason Sams22fa3712010-05-19 17:22:57 -0700402
403extern float __attribute__((overloadable)) sqrt(float);
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800404_RS_STATIC float __attribute__((overloadable)) rsqrt(float v) {
Jason Sams1b937f52010-06-09 14:26:16 -0700405 return 1.f / sqrt(v);
406}
Stephen Hines47416532011-01-11 14:38:18 -0800407DEF_FUNC_1(rsqrt)
Jason Sams1b937f52010-06-09 14:26:16 -0700408
409extern float __attribute__((overloadable)) sin(float);
410DEF_FUNC_1(sin)
411
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800412_RS_STATIC float __attribute__((overloadable)) sincos(float v, float *cosptr) {
Jason Sams1b937f52010-06-09 14:26:16 -0700413 *cosptr = cos(v);
414 return sin(v);
415}
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800416_RS_STATIC float2 __attribute__((overloadable)) sincos(float2 v, float2 *cosptr) {
Jason Sams1b937f52010-06-09 14:26:16 -0700417 *cosptr = cos(v);
418 return sin(v);
419}
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800420_RS_STATIC float3 __attribute__((overloadable)) sincos(float3 v, float3 *cosptr) {
Jason Sams1b937f52010-06-09 14:26:16 -0700421 *cosptr = cos(v);
422 return sin(v);
423}
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800424_RS_STATIC float4 __attribute__((overloadable)) sincos(float4 v, float4 *cosptr) {
Jason Sams1b937f52010-06-09 14:26:16 -0700425 *cosptr = cos(v);
426 return sin(v);
427}
428
429extern float __attribute__((overloadable)) sinh(float);
430DEF_FUNC_1(sinh)
431
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800432_RS_STATIC float __attribute__((overloadable)) sinpi(float v) {
Jason Sams1b937f52010-06-09 14:26:16 -0700433 return sin(v * M_PI);
434}
435DEF_FUNC_1(sinpi)
436
437DEF_FUNC_1(sqrt)
Jason Sams22fa3712010-05-19 17:22:57 -0700438
439extern float __attribute__((overloadable)) tan(float);
Jason Sams1b937f52010-06-09 14:26:16 -0700440DEF_FUNC_1(tan)
Jason Sams22fa3712010-05-19 17:22:57 -0700441
442extern float __attribute__((overloadable)) tanh(float);
Jason Sams1b937f52010-06-09 14:26:16 -0700443DEF_FUNC_1(tanh)
Jason Sams22fa3712010-05-19 17:22:57 -0700444
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800445_RS_STATIC float __attribute__((overloadable)) tanpi(float v) {
Jason Sams1b937f52010-06-09 14:26:16 -0700446 return tan(v * M_PI);
447}
448DEF_FUNC_1(tanpi)
Jason Sams22fa3712010-05-19 17:22:57 -0700449
450extern float __attribute__((overloadable)) tgamma(float);
Jason Sams1b937f52010-06-09 14:26:16 -0700451DEF_FUNC_1(tgamma)
Jason Sams22fa3712010-05-19 17:22:57 -0700452
453extern float __attribute__((overloadable)) trunc(float);
Jason Sams1b937f52010-06-09 14:26:16 -0700454DEF_FUNC_1(trunc)
Jason Sams22fa3712010-05-19 17:22:57 -0700455
456// Int ops (partial), 6.11.3
Jason Sams22fa3712010-05-19 17:22:57 -0700457
Stephen Hinesb902e4c2011-01-14 18:49:01 -0800458#define DEF_RIFUNC_1(typeout, typein, fnc) \
459extern typeout __attribute__((overloadable)) fnc(typein); \
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800460_RS_STATIC typeout##2 __attribute__((overloadable)) fnc(typein##2 v) { \
Stephen Hinesb902e4c2011-01-14 18:49:01 -0800461 typeout##2 r; \
462 r.x = fnc(v.x); \
463 r.y = fnc(v.y); \
464 return r; \
465} \
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800466_RS_STATIC typeout##3 __attribute__((overloadable)) fnc(typein##3 v) { \
Stephen Hinesb902e4c2011-01-14 18:49:01 -0800467 typeout##3 r; \
468 r.x = fnc(v.x); \
469 r.y = fnc(v.y); \
470 r.z = fnc(v.z); \
471 return r; \
472} \
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800473_RS_STATIC typeout##4 __attribute__((overloadable)) fnc(typein##4 v) { \
Stephen Hinesb902e4c2011-01-14 18:49:01 -0800474 typeout##4 r; \
475 r.x = fnc(v.x); \
476 r.y = fnc(v.y); \
477 r.z = fnc(v.z); \
478 r.w = fnc(v.w); \
479 return r; \
Jason Sams1b937f52010-06-09 14:26:16 -0700480}
Jason Sams22fa3712010-05-19 17:22:57 -0700481
Stephen Hinesb902e4c2011-01-14 18:49:01 -0800482#define DEF_UIFUNC_1(fnc) \
483DEF_RIFUNC_1(uchar, char, fnc) \
484DEF_RIFUNC_1(ushort, short, fnc) \
485DEF_RIFUNC_1(uint, int, fnc)
Jason Sams22fa3712010-05-19 17:22:57 -0700486
Stephen Hinesb902e4c2011-01-14 18:49:01 -0800487#define DEF_IFUNC_1(fnc) \
488DEF_RIFUNC_1(uchar, uchar, fnc) \
489DEF_RIFUNC_1(char, char, fnc) \
490DEF_RIFUNC_1(ushort, ushort, fnc) \
491DEF_RIFUNC_1(short, short, fnc) \
492DEF_RIFUNC_1(uint, uint, fnc) \
493DEF_RIFUNC_1(int, int, fnc)
Jason Sams22fa3712010-05-19 17:22:57 -0700494
Stephen Hinesb902e4c2011-01-14 18:49:01 -0800495#define DEF_RIFUNC_2(type, fnc, body) \
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800496_RS_STATIC type __attribute__((overloadable)) fnc(type v1, type v2) { \
Stephen Hinesb902e4c2011-01-14 18:49:01 -0800497 return body; \
498} \
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800499_RS_STATIC type##2 __attribute__((overloadable)) fnc(type##2 v1, type##2 v2) { \
Stephen Hinesb902e4c2011-01-14 18:49:01 -0800500 type##2 r; \
501 r.x = fnc(v1.x, v2.x); \
502 r.y = fnc(v1.y, v2.y); \
503 return r; \
504} \
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800505_RS_STATIC type##3 __attribute__((overloadable)) fnc(type##3 v1, type##3 v2) { \
Stephen Hinesb902e4c2011-01-14 18:49:01 -0800506 type##3 r; \
507 r.x = fnc(v1.x, v2.x); \
508 r.y = fnc(v1.y, v2.y); \
509 r.z = fnc(v1.z, v2.z); \
510 return r; \
511} \
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800512_RS_STATIC type##4 __attribute__((overloadable)) fnc(type##4 v1, type##4 v2) { \
Stephen Hinesb902e4c2011-01-14 18:49:01 -0800513 type##4 r; \
514 r.x = fnc(v1.x, v2.x); \
515 r.y = fnc(v1.y, v2.y); \
516 r.z = fnc(v1.z, v2.z); \
517 r.w = fnc(v1.w, v2.w); \
518 return r; \
519} \
Jason Sams22fa3712010-05-19 17:22:57 -0700520
Stephen Hinesb902e4c2011-01-14 18:49:01 -0800521#define DEF_IFUNC_2(fnc, body) \
522DEF_RIFUNC_2(uchar, fnc, body) \
523DEF_RIFUNC_2(char, fnc, body) \
524DEF_RIFUNC_2(ushort, fnc, body) \
525DEF_RIFUNC_2(short, fnc, body) \
526DEF_RIFUNC_2(uint, fnc, body) \
527DEF_RIFUNC_2(int, fnc, body) \
528DEF_RIFUNC_2(float, fnc, body)
529
530DEF_UIFUNC_1(abs)
531DEF_IFUNC_1(clz)
532
533DEF_IFUNC_2(min, (v1 < v2 ? v1 : v2))
534DEF_FUNC_2F(min)
535
536DEF_IFUNC_2(max, (v1 > v2 ? v1 : v2))
537DEF_FUNC_2F(max)
Jason Sams22fa3712010-05-19 17:22:57 -0700538
539// 6.11.4
540
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800541_RS_STATIC float __attribute__((overloadable)) clamp(float amount, float low, float high) {
Jason Sams1b937f52010-06-09 14:26:16 -0700542 return amount < low ? low : (amount > high ? high : amount);
543}
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800544_RS_STATIC float2 __attribute__((overloadable)) clamp(float2 amount, float2 low, float2 high) {
Jason Sams1b937f52010-06-09 14:26:16 -0700545 float2 r;
546 r.x = amount.x < low.x ? low.x : (amount.x > high.x ? high.x : amount.x);
547 r.y = amount.y < low.y ? low.y : (amount.y > high.y ? high.y : amount.y);
548 return r;
549}
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800550_RS_STATIC float3 __attribute__((overloadable)) clamp(float3 amount, float3 low, float3 high) {
Jason Sams1b937f52010-06-09 14:26:16 -0700551 float3 r;
552 r.x = amount.x < low.x ? low.x : (amount.x > high.x ? high.x : amount.x);
553 r.y = amount.y < low.y ? low.y : (amount.y > high.y ? high.y : amount.y);
554 r.z = amount.z < low.z ? low.z : (amount.z > high.z ? high.z : amount.z);
555 return r;
556}
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800557_RS_STATIC float4 __attribute__((overloadable)) clamp(float4 amount, float4 low, float4 high) {
Jason Sams1b937f52010-06-09 14:26:16 -0700558 float4 r;
559 r.x = amount.x < low.x ? low.x : (amount.x > high.x ? high.x : amount.x);
560 r.y = amount.y < low.y ? low.y : (amount.y > high.y ? high.y : amount.y);
561 r.z = amount.z < low.z ? low.z : (amount.z > high.z ? high.z : amount.z);
562 r.w = amount.w < low.w ? low.w : (amount.w > high.w ? high.w : amount.w);
563 return r;
564}
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800565_RS_STATIC float2 __attribute__((overloadable)) clamp(float2 amount, float low, float high) {
Jason Sams1b937f52010-06-09 14:26:16 -0700566 float2 r;
567 r.x = amount.x < low ? low : (amount.x > high ? high : amount.x);
568 r.y = amount.y < low ? low : (amount.y > high ? high : amount.y);
569 return r;
570}
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800571_RS_STATIC float3 __attribute__((overloadable)) clamp(float3 amount, float low, float high) {
Jason Sams1b937f52010-06-09 14:26:16 -0700572 float3 r;
573 r.x = amount.x < low ? low : (amount.x > high ? high : amount.x);
574 r.y = amount.y < low ? low : (amount.y > high ? high : amount.y);
575 r.z = amount.z < low ? low : (amount.z > high ? high : amount.z);
576 return r;
577}
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800578_RS_STATIC float4 __attribute__((overloadable)) clamp(float4 amount, float low, float high) {
Jason Sams1b937f52010-06-09 14:26:16 -0700579 float4 r;
580 r.x = amount.x < low ? low : (amount.x > high ? high : amount.x);
581 r.y = amount.y < low ? low : (amount.y > high ? high : amount.y);
582 r.z = amount.z < low ? low : (amount.z > high ? high : amount.z);
583 r.w = amount.w < low ? low : (amount.w > high ? high : amount.w);
584 return r;
585}
Jason Sams22fa3712010-05-19 17:22:57 -0700586
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800587_RS_STATIC float __attribute__((overloadable)) degrees(float radians) {
Jason Sams1b937f52010-06-09 14:26:16 -0700588 return radians * (180.f / M_PI);
589}
590DEF_FUNC_1(degrees)
Jason Sams22fa3712010-05-19 17:22:57 -0700591
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800592_RS_STATIC float __attribute__((overloadable)) mix(float start, float stop, float amount) {
Jason Sams1b937f52010-06-09 14:26:16 -0700593 return start + (stop - start) * amount;
594}
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800595_RS_STATIC float2 __attribute__((overloadable)) mix(float2 start, float2 stop, float2 amount) {
Jason Sams1b937f52010-06-09 14:26:16 -0700596 return start + (stop - start) * amount;
597}
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800598_RS_STATIC float3 __attribute__((overloadable)) mix(float3 start, float3 stop, float3 amount) {
Jason Sams1b937f52010-06-09 14:26:16 -0700599 return start + (stop - start) * amount;
600}
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800601_RS_STATIC float4 __attribute__((overloadable)) mix(float4 start, float4 stop, float4 amount) {
Jason Sams1b937f52010-06-09 14:26:16 -0700602 return start + (stop - start) * amount;
603}
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800604_RS_STATIC float2 __attribute__((overloadable)) mix(float2 start, float2 stop, float amount) {
Jason Sams1b937f52010-06-09 14:26:16 -0700605 return start + (stop - start) * amount;
606}
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800607_RS_STATIC float3 __attribute__((overloadable)) mix(float3 start, float3 stop, float amount) {
Jason Sams1b937f52010-06-09 14:26:16 -0700608 return start + (stop - start) * amount;
609}
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800610_RS_STATIC float4 __attribute__((overloadable)) mix(float4 start, float4 stop, float amount) {
Jason Sams1b937f52010-06-09 14:26:16 -0700611 return start + (stop - start) * amount;
612}
Jason Sams22fa3712010-05-19 17:22:57 -0700613
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800614_RS_STATIC float __attribute__((overloadable)) radians(float degrees) {
Jason Sams1b937f52010-06-09 14:26:16 -0700615 return degrees * (M_PI / 180.f);
616}
617DEF_FUNC_1(radians)
Jason Sams22fa3712010-05-19 17:22:57 -0700618
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800619_RS_STATIC float __attribute__((overloadable)) step(float edge, float v) {
Jason Sams1b937f52010-06-09 14:26:16 -0700620 return (v < edge) ? 0.f : 1.f;
621}
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800622_RS_STATIC float2 __attribute__((overloadable)) step(float2 edge, float2 v) {
Jason Sams1b937f52010-06-09 14:26:16 -0700623 float2 r;
624 r.x = (v.x < edge.x) ? 0.f : 1.f;
625 r.y = (v.y < edge.y) ? 0.f : 1.f;
626 return r;
627}
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800628_RS_STATIC float3 __attribute__((overloadable)) step(float3 edge, float3 v) {
Jason Sams1b937f52010-06-09 14:26:16 -0700629 float3 r;
630 r.x = (v.x < edge.x) ? 0.f : 1.f;
631 r.y = (v.y < edge.y) ? 0.f : 1.f;
632 r.z = (v.z < edge.z) ? 0.f : 1.f;
633 return r;
634}
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800635_RS_STATIC float4 __attribute__((overloadable)) step(float4 edge, float4 v) {
Jason Sams1b937f52010-06-09 14:26:16 -0700636 float4 r;
637 r.x = (v.x < edge.x) ? 0.f : 1.f;
638 r.y = (v.y < edge.y) ? 0.f : 1.f;
639 r.z = (v.z < edge.z) ? 0.f : 1.f;
640 r.w = (v.w < edge.w) ? 0.f : 1.f;
641 return r;
642}
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800643_RS_STATIC float2 __attribute__((overloadable)) step(float2 edge, float v) {
Jason Sams1b937f52010-06-09 14:26:16 -0700644 float2 r;
645 r.x = (v < edge.x) ? 0.f : 1.f;
646 r.y = (v < edge.y) ? 0.f : 1.f;
647 return r;
648}
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800649_RS_STATIC float3 __attribute__((overloadable)) step(float3 edge, float v) {
Jason Sams1b937f52010-06-09 14:26:16 -0700650 float3 r;
651 r.x = (v < edge.x) ? 0.f : 1.f;
652 r.y = (v < edge.y) ? 0.f : 1.f;
653 r.z = (v < edge.z) ? 0.f : 1.f;
654 return r;
655}
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800656_RS_STATIC float4 __attribute__((overloadable)) step(float4 edge, float v) {
Jason Sams1b937f52010-06-09 14:26:16 -0700657 float4 r;
658 r.x = (v < edge.x) ? 0.f : 1.f;
659 r.y = (v < edge.y) ? 0.f : 1.f;
660 r.z = (v < edge.z) ? 0.f : 1.f;
661 r.w = (v < edge.w) ? 0.f : 1.f;
662 return r;
663}
Jason Sams22fa3712010-05-19 17:22:57 -0700664
665extern float __attribute__((overloadable)) smoothstep(float, float, float);
666extern float2 __attribute__((overloadable)) smoothstep(float2, float2, float2);
667extern float3 __attribute__((overloadable)) smoothstep(float3, float3, float3);
668extern float4 __attribute__((overloadable)) smoothstep(float4, float4, float4);
Jason Sams22fa3712010-05-19 17:22:57 -0700669extern float2 __attribute__((overloadable)) smoothstep(float, float, float2);
670extern float3 __attribute__((overloadable)) smoothstep(float, float, float3);
671extern float4 __attribute__((overloadable)) smoothstep(float, float, float4);
Jason Sams22fa3712010-05-19 17:22:57 -0700672
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800673_RS_STATIC float __attribute__((overloadable)) sign(float v) {
Jason Sams1b937f52010-06-09 14:26:16 -0700674 if (v > 0) return 1.f;
675 if (v < 0) return -1.f;
676 return v;
677}
678DEF_FUNC_1(sign)
Jason Sams22fa3712010-05-19 17:22:57 -0700679
680// 6.11.5
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800681_RS_STATIC float3 __attribute__((overloadable)) cross(float3 lhs, float3 rhs) {
Jason Sams1b937f52010-06-09 14:26:16 -0700682 float3 r;
683 r.x = lhs.y * rhs.z - lhs.z * rhs.y;
684 r.y = lhs.z * rhs.x - lhs.x * rhs.z;
685 r.z = lhs.x * rhs.y - lhs.y * rhs.x;
686 return r;
687}
Jason Sams22fa3712010-05-19 17:22:57 -0700688
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800689_RS_STATIC float4 __attribute__((overloadable)) cross(float4 lhs, float4 rhs) {
Jason Sams1b937f52010-06-09 14:26:16 -0700690 float4 r;
691 r.x = lhs.y * rhs.z - lhs.z * rhs.y;
692 r.y = lhs.z * rhs.x - lhs.x * rhs.z;
693 r.z = lhs.x * rhs.y - lhs.y * rhs.x;
694 r.w = 0.f;
695 return r;
696}
Jason Sams22fa3712010-05-19 17:22:57 -0700697
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800698_RS_STATIC float __attribute__((overloadable)) dot(float lhs, float rhs) {
Jason Sams1b937f52010-06-09 14:26:16 -0700699 return lhs * rhs;
700}
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800701_RS_STATIC float __attribute__((overloadable)) dot(float2 lhs, float2 rhs) {
Jason Sams1b937f52010-06-09 14:26:16 -0700702 return lhs.x*rhs.x + lhs.y*rhs.y;
703}
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800704_RS_STATIC float __attribute__((overloadable)) dot(float3 lhs, float3 rhs) {
Jason Sams1b937f52010-06-09 14:26:16 -0700705 return lhs.x*rhs.x + lhs.y*rhs.y + lhs.z*rhs.z;
706}
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800707_RS_STATIC float __attribute__((overloadable)) dot(float4 lhs, float4 rhs) {
Jason Sams1b937f52010-06-09 14:26:16 -0700708 return lhs.x*rhs.x + lhs.y*rhs.y + lhs.z*rhs.z + lhs.w*rhs.w;
709}
Jason Sams22fa3712010-05-19 17:22:57 -0700710
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800711_RS_STATIC float __attribute__((overloadable)) length(float v) {
Jason Sams1b937f52010-06-09 14:26:16 -0700712 return v;
713}
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800714_RS_STATIC float __attribute__((overloadable)) length(float2 v) {
Jason Sams1b937f52010-06-09 14:26:16 -0700715 return sqrt(v.x*v.x + v.y*v.y);
716}
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800717_RS_STATIC float __attribute__((overloadable)) length(float3 v) {
Jason Sams1b937f52010-06-09 14:26:16 -0700718 return sqrt(v.x*v.x + v.y*v.y + v.z*v.z);
719}
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800720_RS_STATIC float __attribute__((overloadable)) length(float4 v) {
Jason Sams1b937f52010-06-09 14:26:16 -0700721 return sqrt(v.x*v.x + v.y*v.y + v.z*v.z + v.w*v.w);
722}
Jason Sams22fa3712010-05-19 17:22:57 -0700723
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800724_RS_STATIC float __attribute__((overloadable)) distance(float lhs, float rhs) {
Jason Sams1b937f52010-06-09 14:26:16 -0700725 return length(lhs - rhs);
726}
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800727_RS_STATIC float __attribute__((overloadable)) distance(float2 lhs, float2 rhs) {
Jason Sams1b937f52010-06-09 14:26:16 -0700728 return length(lhs - rhs);
729}
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800730_RS_STATIC float __attribute__((overloadable)) distance(float3 lhs, float3 rhs) {
Jason Sams1b937f52010-06-09 14:26:16 -0700731 return length(lhs - rhs);
732}
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800733_RS_STATIC float __attribute__((overloadable)) distance(float4 lhs, float4 rhs) {
Jason Sams1b937f52010-06-09 14:26:16 -0700734 return length(lhs - rhs);
735}
Jason Sams22fa3712010-05-19 17:22:57 -0700736
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800737_RS_STATIC float __attribute__((overloadable)) normalize(float v) {
Jason Sams1b937f52010-06-09 14:26:16 -0700738 return 1.f;
739}
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800740_RS_STATIC float2 __attribute__((overloadable)) normalize(float2 v) {
Jason Sams1b937f52010-06-09 14:26:16 -0700741 return v / length(v);
742}
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800743_RS_STATIC float3 __attribute__((overloadable)) normalize(float3 v) {
Jason Sams1b937f52010-06-09 14:26:16 -0700744 return v / length(v);
745}
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800746_RS_STATIC float4 __attribute__((overloadable)) normalize(float4 v) {
Jason Sams1b937f52010-06-09 14:26:16 -0700747 return v / length(v);
748}
749
Stephen Hines47416532011-01-11 14:38:18 -0800750#undef CVT_FUNC
751#undef CVT_FUNC_2
752#undef DEF_FUNC_1
753#undef DEF_FUNC_1_RI
754#undef DEF_FUNC_2
755#undef DEF_FUNC_2F
Stephen Hinesb902e4c2011-01-14 18:49:01 -0800756#undef DEF_RIFUNC_1
757#undef DEF_UIFUNC_1
758#undef DEF_IFUNC_1
759#undef DEF_RIFUNC_2
760#undef DEF_IFUNC_2
Stephen Hinesf6a28c62011-01-18 16:53:19 -0800761#undef _RS_STATIC
Jason Sams1b937f52010-06-09 14:26:16 -0700762
763#endif