blob: f5f189e7a4bc8c028ec5ef4dc4f7b7734ec681a9 [file] [log] [blame]
Nicolas Capens0bac2852016-05-07 06:09:58 -04001// Copyright 2016 The SwiftShader Authors. All Rights Reserved.
2//
3// Licensed under the Apache License, Version 2.0 (the "License");
4// you may not use this file except in compliance with the License.
5// You may obtain a copy of the License at
6//
7// http://www.apache.org/licenses/LICENSE-2.0
8//
9// Unless required by applicable law or agreed to in writing, software
10// distributed under the License is distributed on an "AS IS" BASIS,
11// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12// See the License for the specific language governing permissions and
13// limitations under the License.
14
Nicolas Capens48461502018-08-06 14:20:45 -040015#ifndef rr_Reactor_hpp
16#define rr_Reactor_hpp
Nicolas Capensd022e412016-09-26 13:30:14 -040017
Nicolas Capens0bac2852016-05-07 06:09:58 -040018#include "Nucleus.hpp"
Nicolas Capensc9991d42021-06-16 00:46:28 -040019#include "Pragma.hpp"
Nicolas Capens0bac2852016-05-07 06:09:58 -040020#include "Routine.hpp"
Nicolas Capens2e653e52022-02-23 13:30:13 -050021#include "Swizzle.hpp"
Ben Clayton351be422019-04-30 12:26:57 +010022#include "Traits.hpp"
Nicolas Capens0bac2852016-05-07 06:09:58 -040023
Sean Risser19e30802022-06-01 10:58:10 -040024#include <array>
Nicolas Capens4dd1eff2017-08-04 09:33:04 -040025#include <cassert>
Nicolas Capens3edbc042022-03-01 16:02:18 -050026#include <cmath>
Nicolas Capens3bbc5e12016-09-27 10:49:52 -040027#include <cstddef>
Chris Forbes878d4b02019-01-21 10:48:35 -080028#include <cstdio>
Nicolas Capense5720882020-01-13 14:10:04 -050029#include <limits>
Ben Clayton169872e2019-02-27 23:58:35 +000030#include <tuple>
Antonio Maioranof14f6c42020-11-03 16:34:35 -050031#include <unordered_map>
Ben Clayton1bc7ee92019-02-14 18:43:22 +000032
Ben Claytonac07ed82019-03-26 14:17:41 +000033#ifdef ENABLE_RR_DEBUG_INFO
Ben Clayton713b8d32019-12-17 20:37:56 +000034// Functions used for generating JIT debug info.
35// See docs/ReactorDebugInfo.md for more information.
36namespace rr {
37// Update the current source location for debug.
38void EmitDebugLocation();
39// Bind value to its symbolic name taken from the backtrace.
40void EmitDebugVariable(class Value *value);
41// Flush any pending variable bindings before the line ends.
42void FlushDebug();
43} // namespace rr
44# define RR_DEBUG_INFO_UPDATE_LOC() rr::EmitDebugLocation()
45# define RR_DEBUG_INFO_EMIT_VAR(value) rr::EmitDebugVariable(value)
46# define RR_DEBUG_INFO_FLUSH() rr::FlushDebug()
Ben Claytonac07ed82019-03-26 14:17:41 +000047#else
Ben Clayton713b8d32019-12-17 20:37:56 +000048# define RR_DEBUG_INFO_UPDATE_LOC()
49# define RR_DEBUG_INFO_EMIT_VAR(value)
50# define RR_DEBUG_INFO_FLUSH()
51#endif // ENABLE_RR_DEBUG_INFO
Ben Claytonac07ed82019-03-26 14:17:41 +000052
Antonio Maiorano8cbee412020-06-10 15:59:20 -040053#ifdef ENABLE_RR_PRINT
54namespace rr {
55int DebugPrintf(const char *format, ...);
56}
57#endif
58
Nicolas Capens259ce702020-11-18 11:32:05 -050059// A Clang extension to determine compiler features.
60// We use it to detect Sanitizer builds (e.g. -fsanitize=memory).
61#ifndef __has_feature
62# define __has_feature(x) 0
63#endif
64
Nicolas Capens157ba262019-12-10 17:49:14 -050065namespace rr {
66
Nicolas Capens70505b42022-01-31 22:29:48 -050067struct Caps
Nicolas Capens0bac2852016-05-07 06:09:58 -040068{
Nicolas Capens70505b42022-01-31 22:29:48 -050069 static std::string backendName();
70 static bool coroutinesSupported(); // Support for rr::Coroutine<F>
71 static bool fmaIsFast(); // rr::FMA() is faster than `x * y + z`
Nicolas Capens157ba262019-12-10 17:49:14 -050072};
Nicolas Capens157ba262019-12-10 17:49:14 -050073
74class Bool;
75class Byte;
76class SByte;
77class Byte4;
78class SByte4;
79class Byte8;
80class SByte8;
81class Byte16;
82class SByte16;
83class Short;
84class UShort;
85class Short2;
86class UShort2;
87class Short4;
88class UShort4;
89class Short8;
90class UShort8;
91class Int;
92class UInt;
93class Int2;
94class UInt2;
95class Int4;
96class UInt4;
97class Long;
98class Half;
99class Float;
100class Float2;
101class Float4;
102
Nicolas Capens44f94692022-06-20 23:15:46 -0400103namespace SIMD {
104class Int;
105class UInt;
106class Float;
107} // namespace SIMD
108
Nicolas Capens2e653e52022-02-23 13:30:13 -0500109template<>
110struct Scalar<Float4>
111{
112 using Type = Float;
113};
114
115template<>
116struct Scalar<Int4>
117{
118 using Type = Int;
119};
120
121template<>
122struct Scalar<UInt4>
123{
124 using Type = UInt;
125};
126
Nicolas Capens44f94692022-06-20 23:15:46 -0400127template<>
128struct Scalar<SIMD::Float>
129{
130 using Type = Float;
131};
132
133template<>
134struct Scalar<SIMD::Int>
135{
136 using Type = Int;
137};
138
139template<>
140struct Scalar<SIMD::UInt>
141{
142 using Type = UInt;
143};
144
Nicolas Capens157ba262019-12-10 17:49:14 -0500145class Void
146{
147public:
Nicolas Capens519cf222020-05-08 15:27:19 -0400148 static Type *type();
Nicolas Capens157ba262019-12-10 17:49:14 -0500149};
Ben Claytonc7904162019-04-17 17:35:48 -0400150
Nicolas Capens157ba262019-12-10 17:49:14 -0500151template<class T>
152class RValue;
Nicolas Capensd022e412016-09-26 13:30:14 -0400153
Nicolas Capens157ba262019-12-10 17:49:14 -0500154template<class T>
155class Pointer;
156
157class Variable
158{
159 friend class Nucleus;
Nicolas Capens157ba262019-12-10 17:49:14 -0500160
Antonio Maiorano84c61e12020-12-02 12:06:05 -0500161 Variable() = delete;
Ben Clayton713b8d32019-12-17 20:37:56 +0000162 Variable &operator=(const Variable &) = delete;
Nicolas Capens157ba262019-12-10 17:49:14 -0500163
164public:
165 void materialize() const;
166
167 Value *loadValue() const;
168 Value *storeValue(Value *value) const;
169
170 Value *getBaseAddress() const;
171 Value *getElementPointer(Value *index, bool unsignedIndex) const;
172
Antonio Maioranobae138d2020-12-02 14:25:10 -0500173 Type *getType() const { return type; }
Antonio Maiorano84c61e12020-12-02 12:06:05 -0500174 int getArraySize() const { return arraySize; }
Nicolas Capens67cdce92020-05-01 21:37:20 -0400175
Antonio Maiorano6c839a62020-11-06 11:19:24 -0500176 // This function is only public for testing purposes, as it affects performance.
177 // It is not considered part of Reactor's public API.
178 static void materializeAll();
179
Nicolas Capens157ba262019-12-10 17:49:14 -0500180protected:
Antonio Maioranobae138d2020-12-02 14:25:10 -0500181 Variable(Type *type, int arraySize);
Ben Clayton713b8d32019-12-17 20:37:56 +0000182 Variable(const Variable &) = default;
Nicolas Capens157ba262019-12-10 17:49:14 -0500183
Nicolas Capens67cdce92020-05-01 21:37:20 -0400184 virtual ~Variable();
Nicolas Capens157ba262019-12-10 17:49:14 -0500185
186private:
Nicolas Capens157ba262019-12-10 17:49:14 -0500187 static void killUnmaterialized();
188
Antonio Maioranof14f6c42020-11-03 16:34:35 -0500189 // Set of variables that do not have a stack location yet.
190 class UnmaterializedVariables
191 {
192 public:
193 void add(const Variable *v);
194 void remove(const Variable *v);
195 void clear();
196 void materializeAll();
197
198 private:
199 int counter = 0;
200 std::unordered_map<const Variable *, int> variables;
201 };
202
Nicolas Capens7d6b5912020-04-28 15:57:57 -0400203 // This has to be a raw pointer because glibc 2.17 doesn't support __cxa_thread_atexit_impl
204 // for destructing objects at exit. See crbug.com/1074222
Antonio Maioranof14f6c42020-11-03 16:34:35 -0500205 static thread_local UnmaterializedVariables *unmaterializedVariables;
Nicolas Capens157ba262019-12-10 17:49:14 -0500206
Antonio Maioranobae138d2020-12-02 14:25:10 -0500207 Type *const type;
Antonio Maiorano84c61e12020-12-02 12:06:05 -0500208 const int arraySize;
Nicolas Capens157ba262019-12-10 17:49:14 -0500209 mutable Value *rvalue = nullptr;
210 mutable Value *address = nullptr;
211};
212
213template<class T>
214class LValue : public Variable
215{
216public:
Antonio Maiorano84c61e12020-12-02 12:06:05 -0500217 LValue(int arraySize = 0);
Nicolas Capens157ba262019-12-10 17:49:14 -0500218
219 RValue<Pointer<T>> operator&();
220
Nicolas Capens5f77c5e2020-05-01 22:51:11 -0400221 RValue<T> load() const
222 {
Nicolas Capens3b655b62020-05-13 15:23:54 -0400223 return RValue<T>(this->loadValue());
Nicolas Capens5f77c5e2020-05-01 22:51:11 -0400224 }
225
226 RValue<T> store(RValue<T> rvalue) const
227 {
Nicolas Capens3b655b62020-05-13 15:23:54 -0400228 this->storeValue(rvalue.value());
Nicolas Capens5f77c5e2020-05-01 22:51:11 -0400229
230 return rvalue;
231 }
232
Nicolas Capens157ba262019-12-10 17:49:14 -0500233 // self() returns the this pointer to this LValue<T> object.
234 // This function exists because operator&() is overloaded.
Ben Clayton713b8d32019-12-17 20:37:56 +0000235 inline LValue<T> *self() { return this; }
Nicolas Capens157ba262019-12-10 17:49:14 -0500236};
Nicolas Capensd022e412016-09-26 13:30:14 -0400237
Nicolas Capens157ba262019-12-10 17:49:14 -0500238template<class T>
239class Reference
240{
241public:
242 using reference_underlying_type = T;
Nicolas Capensd022e412016-09-26 13:30:14 -0400243
Nicolas Capens157ba262019-12-10 17:49:14 -0500244 explicit Reference(Value *pointer, int alignment = 1);
Nicolas Capens990e4b22021-06-17 22:26:36 -0400245 Reference(const Reference<T> &ref) = default;
Nicolas Capensd022e412016-09-26 13:30:14 -0400246
Nicolas Capens157ba262019-12-10 17:49:14 -0500247 RValue<T> operator=(RValue<T> rhs) const;
248 RValue<T> operator=(const Reference<T> &ref) const;
Nicolas Capens5da8d8d2019-03-27 14:45:34 -0400249
Nicolas Capens157ba262019-12-10 17:49:14 -0500250 RValue<T> operator+=(RValue<T> rhs) const;
Nicolas Capens5da8d8d2019-03-27 14:45:34 -0400251
Nicolas Capens157ba262019-12-10 17:49:14 -0500252 RValue<Pointer<T>> operator&() const { return RValue<Pointer<T>>(address); }
Nicolas Capens0192d152019-03-27 14:46:07 -0400253
Nicolas Capens157ba262019-12-10 17:49:14 -0500254 Value *loadValue() const;
Nicolas Capensb4e4f112020-05-01 23:06:41 -0400255 RValue<T> load() const;
Nicolas Capens157ba262019-12-10 17:49:14 -0500256 int getAlignment() const;
Nicolas Capens0192d152019-03-27 14:46:07 -0400257
Nicolas Capens157ba262019-12-10 17:49:14 -0500258private:
Nicolas Capens990e4b22021-06-17 22:26:36 -0400259 Value *const address;
Nicolas Capens5da8d8d2019-03-27 14:45:34 -0400260
Nicolas Capens157ba262019-12-10 17:49:14 -0500261 const int alignment;
262};
Nicolas Capens0192d152019-03-27 14:46:07 -0400263
Nicolas Capens157ba262019-12-10 17:49:14 -0500264template<class T>
265struct BoolLiteral
266{
Nicolas Capens3edbc042022-03-01 16:02:18 -0500267 struct Type;
Nicolas Capens157ba262019-12-10 17:49:14 -0500268};
Nicolas Capens0192d152019-03-27 14:46:07 -0400269
Nicolas Capens157ba262019-12-10 17:49:14 -0500270template<>
271struct BoolLiteral<Bool>
272{
Nicolas Capens3edbc042022-03-01 16:02:18 -0500273 using Type = bool;
Nicolas Capens157ba262019-12-10 17:49:14 -0500274};
Ben Clayton83dd4522019-06-27 10:37:00 +0100275
Nicolas Capens157ba262019-12-10 17:49:14 -0500276template<class T>
277struct IntLiteral
278{
Nicolas Capens3edbc042022-03-01 16:02:18 -0500279 struct Type;
Nicolas Capens157ba262019-12-10 17:49:14 -0500280};
Nicolas Capens0192d152019-03-27 14:46:07 -0400281
Nicolas Capens157ba262019-12-10 17:49:14 -0500282template<>
283struct IntLiteral<Int>
284{
Nicolas Capens3edbc042022-03-01 16:02:18 -0500285 using Type = int;
Nicolas Capens157ba262019-12-10 17:49:14 -0500286};
Nicolas Capens0192d152019-03-27 14:46:07 -0400287
Nicolas Capens157ba262019-12-10 17:49:14 -0500288template<>
289struct IntLiteral<UInt>
290{
Nicolas Capens3edbc042022-03-01 16:02:18 -0500291 using Type = unsigned int;
Nicolas Capens157ba262019-12-10 17:49:14 -0500292};
Nicolas Capens297d26e2016-11-18 12:52:17 -0500293
Daniele Vettorelf908b182022-02-09 17:38:08 +0000294template<class T>
295struct LongLiteral
296{
Nicolas Capens3edbc042022-03-01 16:02:18 -0500297 struct Type;
Daniele Vettorelf908b182022-02-09 17:38:08 +0000298};
299
Nicolas Capens157ba262019-12-10 17:49:14 -0500300template<>
Daniele Vettorelf908b182022-02-09 17:38:08 +0000301struct LongLiteral<Long>
Nicolas Capens157ba262019-12-10 17:49:14 -0500302{
Nicolas Capens3edbc042022-03-01 16:02:18 -0500303 using Type = int64_t;
Nicolas Capens157ba262019-12-10 17:49:14 -0500304};
Nicolas Capensd022e412016-09-26 13:30:14 -0400305
Nicolas Capens157ba262019-12-10 17:49:14 -0500306template<class T>
307struct FloatLiteral
308{
Nicolas Capens3edbc042022-03-01 16:02:18 -0500309 struct Type;
Nicolas Capens157ba262019-12-10 17:49:14 -0500310};
Nicolas Capens297d26e2016-11-18 12:52:17 -0500311
Nicolas Capens157ba262019-12-10 17:49:14 -0500312template<>
313struct FloatLiteral<Float>
314{
Nicolas Capens3edbc042022-03-01 16:02:18 -0500315 using Type = float;
316};
317
318template<class T>
319struct BroadcastLiteral
320{
321 struct Type;
322};
323
324template<>
325struct BroadcastLiteral<Int4>
326{
327 using Type = int;
328};
329
330template<>
331struct BroadcastLiteral<UInt4>
332{
333 using Type = unsigned int;
334};
335
336template<>
337struct BroadcastLiteral<Float4>
338{
339 using Type = float;
Nicolas Capens157ba262019-12-10 17:49:14 -0500340};
Ben Clayton0953d9b2019-06-25 20:57:06 +0100341
Nicolas Capens44f94692022-06-20 23:15:46 -0400342template<>
343struct BroadcastLiteral<SIMD::Int>
344{
345 using Type = int;
346};
347
348template<>
349struct BroadcastLiteral<SIMD::UInt>
350{
351 using Type = unsigned int;
352};
353
354template<>
355struct BroadcastLiteral<SIMD::Float>
356{
357 using Type = float;
358};
359
Nicolas Capens157ba262019-12-10 17:49:14 -0500360template<class T>
361class RValue
362{
363public:
364 using rvalue_underlying_type = T;
Nicolas Capensd022e412016-09-26 13:30:14 -0400365
Nicolas Capens157ba262019-12-10 17:49:14 -0500366 explicit RValue(Value *rvalue);
Nicolas Capensd022e412016-09-26 13:30:14 -0400367
Nicolas Capens157ba262019-12-10 17:49:14 -0500368 RValue(const RValue<T> &rvalue);
Nicolas Capens157ba262019-12-10 17:49:14 -0500369 RValue(const T &lvalue);
Nicolas Capens3edbc042022-03-01 16:02:18 -0500370 RValue(typename BoolLiteral<T>::Type b);
371 RValue(typename IntLiteral<T>::Type i);
372 RValue(typename LongLiteral<T>::Type i);
373 RValue(typename FloatLiteral<T>::Type f);
374 RValue(typename BroadcastLiteral<T>::Type x);
Nicolas Capens157ba262019-12-10 17:49:14 -0500375 RValue(const Reference<T> &rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -0400376
Nicolas Capens67cdce92020-05-01 21:37:20 -0400377 // Rvalues cannot be assigned to: "(a + b) = c;"
Ben Clayton713b8d32019-12-17 20:37:56 +0000378 RValue<T> &operator=(const RValue<T> &) = delete;
Nicolas Capensd022e412016-09-26 13:30:14 -0400379
Nicolas Capensb6e8c3f2020-05-01 23:28:37 -0400380 Value *value() const { return val; }
381
Nicolas Capens442e25b2022-06-22 12:02:52 -0400382 static int element_count() { return T::element_count(); }
383
Nicolas Capensb6e8c3f2020-05-01 23:28:37 -0400384private:
385 Value *const val;
Nicolas Capens157ba262019-12-10 17:49:14 -0500386};
Nicolas Capensd022e412016-09-26 13:30:14 -0400387
Nicolas Capens157ba262019-12-10 17:49:14 -0500388template<typename T>
Nicolas Capensb6e8c3f2020-05-01 23:28:37 -0400389class Argument
Nicolas Capens157ba262019-12-10 17:49:14 -0500390{
Nicolas Capensb6e8c3f2020-05-01 23:28:37 -0400391public:
392 explicit Argument(Value *val)
393 : val(val)
Ben Clayton713b8d32019-12-17 20:37:56 +0000394 {}
Nicolas Capensd022e412016-09-26 13:30:14 -0400395
Nicolas Capensb6e8c3f2020-05-01 23:28:37 -0400396 RValue<T> rvalue() const { return RValue<T>(val); }
397
398private:
399 Value *const val;
Nicolas Capens157ba262019-12-10 17:49:14 -0500400};
Nicolas Capensd022e412016-09-26 13:30:14 -0400401
Nicolas Capens157ba262019-12-10 17:49:14 -0500402class Bool : public LValue<Bool>
403{
404public:
405 Bool(Argument<Bool> argument);
Nicolas Capensd022e412016-09-26 13:30:14 -0400406
Nicolas Capens157ba262019-12-10 17:49:14 -0500407 Bool() = default;
408 Bool(bool x);
409 Bool(RValue<Bool> rhs);
410 Bool(const Bool &rhs);
411 Bool(const Reference<Bool> &rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -0400412
Ben Clayton713b8d32019-12-17 20:37:56 +0000413 // RValue<Bool> operator=(bool rhs); // FIXME: Implement
Nicolas Capens157ba262019-12-10 17:49:14 -0500414 RValue<Bool> operator=(RValue<Bool> rhs);
415 RValue<Bool> operator=(const Bool &rhs);
416 RValue<Bool> operator=(const Reference<Bool> &rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -0400417
Nicolas Capens519cf222020-05-08 15:27:19 -0400418 static Type *type();
Nicolas Capens157ba262019-12-10 17:49:14 -0500419};
Nicolas Capensd022e412016-09-26 13:30:14 -0400420
Nicolas Capens157ba262019-12-10 17:49:14 -0500421RValue<Bool> operator!(RValue<Bool> val);
422RValue<Bool> operator&&(RValue<Bool> lhs, RValue<Bool> rhs);
423RValue<Bool> operator||(RValue<Bool> lhs, RValue<Bool> rhs);
424RValue<Bool> operator!=(RValue<Bool> lhs, RValue<Bool> rhs);
425RValue<Bool> operator==(RValue<Bool> lhs, RValue<Bool> rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -0400426
Nicolas Capens157ba262019-12-10 17:49:14 -0500427class Byte : public LValue<Byte>
428{
429public:
430 Byte(Argument<Byte> argument);
Nicolas Capensd022e412016-09-26 13:30:14 -0400431
Nicolas Capens157ba262019-12-10 17:49:14 -0500432 explicit Byte(RValue<Int> cast);
433 explicit Byte(RValue<UInt> cast);
434 explicit Byte(RValue<UShort> cast);
Nicolas Capensd022e412016-09-26 13:30:14 -0400435
Nicolas Capens157ba262019-12-10 17:49:14 -0500436 Byte() = default;
437 Byte(int x);
438 Byte(unsigned char x);
439 Byte(RValue<Byte> rhs);
440 Byte(const Byte &rhs);
441 Byte(const Reference<Byte> &rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -0400442
Ben Clayton713b8d32019-12-17 20:37:56 +0000443 // RValue<Byte> operator=(unsigned char rhs); // FIXME: Implement
Nicolas Capens157ba262019-12-10 17:49:14 -0500444 RValue<Byte> operator=(RValue<Byte> rhs);
445 RValue<Byte> operator=(const Byte &rhs);
446 RValue<Byte> operator=(const Reference<Byte> &rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -0400447
Nicolas Capens519cf222020-05-08 15:27:19 -0400448 static Type *type();
Nicolas Capens157ba262019-12-10 17:49:14 -0500449};
Nicolas Capensd022e412016-09-26 13:30:14 -0400450
Nicolas Capens157ba262019-12-10 17:49:14 -0500451RValue<Byte> operator+(RValue<Byte> lhs, RValue<Byte> rhs);
452RValue<Byte> operator-(RValue<Byte> lhs, RValue<Byte> rhs);
453RValue<Byte> operator*(RValue<Byte> lhs, RValue<Byte> rhs);
454RValue<Byte> operator/(RValue<Byte> lhs, RValue<Byte> rhs);
455RValue<Byte> operator%(RValue<Byte> lhs, RValue<Byte> rhs);
456RValue<Byte> operator&(RValue<Byte> lhs, RValue<Byte> rhs);
457RValue<Byte> operator|(RValue<Byte> lhs, RValue<Byte> rhs);
458RValue<Byte> operator^(RValue<Byte> lhs, RValue<Byte> rhs);
459RValue<Byte> operator<<(RValue<Byte> lhs, RValue<Byte> rhs);
460RValue<Byte> operator>>(RValue<Byte> lhs, RValue<Byte> rhs);
461RValue<Byte> operator+=(Byte &lhs, RValue<Byte> rhs);
462RValue<Byte> operator-=(Byte &lhs, RValue<Byte> rhs);
463RValue<Byte> operator*=(Byte &lhs, RValue<Byte> rhs);
464RValue<Byte> operator/=(Byte &lhs, RValue<Byte> rhs);
465RValue<Byte> operator%=(Byte &lhs, RValue<Byte> rhs);
466RValue<Byte> operator&=(Byte &lhs, RValue<Byte> rhs);
467RValue<Byte> operator|=(Byte &lhs, RValue<Byte> rhs);
468RValue<Byte> operator^=(Byte &lhs, RValue<Byte> rhs);
469RValue<Byte> operator<<=(Byte &lhs, RValue<Byte> rhs);
470RValue<Byte> operator>>=(Byte &lhs, RValue<Byte> rhs);
471RValue<Byte> operator+(RValue<Byte> val);
472RValue<Byte> operator-(RValue<Byte> val);
473RValue<Byte> operator~(RValue<Byte> val);
Ben Clayton713b8d32019-12-17 20:37:56 +0000474RValue<Byte> operator++(Byte &val, int); // Post-increment
475const Byte &operator++(Byte &val); // Pre-increment
476RValue<Byte> operator--(Byte &val, int); // Post-decrement
477const Byte &operator--(Byte &val); // Pre-decrement
Nicolas Capens157ba262019-12-10 17:49:14 -0500478RValue<Bool> operator<(RValue<Byte> lhs, RValue<Byte> rhs);
479RValue<Bool> operator<=(RValue<Byte> lhs, RValue<Byte> rhs);
480RValue<Bool> operator>(RValue<Byte> lhs, RValue<Byte> rhs);
481RValue<Bool> operator>=(RValue<Byte> lhs, RValue<Byte> rhs);
482RValue<Bool> operator!=(RValue<Byte> lhs, RValue<Byte> rhs);
483RValue<Bool> operator==(RValue<Byte> lhs, RValue<Byte> rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -0400484
Nicolas Capens157ba262019-12-10 17:49:14 -0500485class SByte : public LValue<SByte>
486{
487public:
488 SByte(Argument<SByte> argument);
Nicolas Capensd022e412016-09-26 13:30:14 -0400489
Nicolas Capens157ba262019-12-10 17:49:14 -0500490 explicit SByte(RValue<Int> cast);
491 explicit SByte(RValue<Short> cast);
Nicolas Capensd022e412016-09-26 13:30:14 -0400492
Nicolas Capens157ba262019-12-10 17:49:14 -0500493 SByte() = default;
494 SByte(signed char x);
495 SByte(RValue<SByte> rhs);
496 SByte(const SByte &rhs);
497 SByte(const Reference<SByte> &rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -0400498
Ben Clayton713b8d32019-12-17 20:37:56 +0000499 // RValue<SByte> operator=(signed char rhs); // FIXME: Implement
Nicolas Capens157ba262019-12-10 17:49:14 -0500500 RValue<SByte> operator=(RValue<SByte> rhs);
501 RValue<SByte> operator=(const SByte &rhs);
502 RValue<SByte> operator=(const Reference<SByte> &rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -0400503
Nicolas Capens519cf222020-05-08 15:27:19 -0400504 static Type *type();
Nicolas Capens157ba262019-12-10 17:49:14 -0500505};
Nicolas Capensd022e412016-09-26 13:30:14 -0400506
Nicolas Capens157ba262019-12-10 17:49:14 -0500507RValue<SByte> operator+(RValue<SByte> lhs, RValue<SByte> rhs);
508RValue<SByte> operator-(RValue<SByte> lhs, RValue<SByte> rhs);
509RValue<SByte> operator*(RValue<SByte> lhs, RValue<SByte> rhs);
510RValue<SByte> operator/(RValue<SByte> lhs, RValue<SByte> rhs);
511RValue<SByte> operator%(RValue<SByte> lhs, RValue<SByte> rhs);
512RValue<SByte> operator&(RValue<SByte> lhs, RValue<SByte> rhs);
513RValue<SByte> operator|(RValue<SByte> lhs, RValue<SByte> rhs);
514RValue<SByte> operator^(RValue<SByte> lhs, RValue<SByte> rhs);
515RValue<SByte> operator<<(RValue<SByte> lhs, RValue<SByte> rhs);
516RValue<SByte> operator>>(RValue<SByte> lhs, RValue<SByte> rhs);
517RValue<SByte> operator+=(SByte &lhs, RValue<SByte> rhs);
518RValue<SByte> operator-=(SByte &lhs, RValue<SByte> rhs);
519RValue<SByte> operator*=(SByte &lhs, RValue<SByte> rhs);
520RValue<SByte> operator/=(SByte &lhs, RValue<SByte> rhs);
521RValue<SByte> operator%=(SByte &lhs, RValue<SByte> rhs);
522RValue<SByte> operator&=(SByte &lhs, RValue<SByte> rhs);
523RValue<SByte> operator|=(SByte &lhs, RValue<SByte> rhs);
524RValue<SByte> operator^=(SByte &lhs, RValue<SByte> rhs);
525RValue<SByte> operator<<=(SByte &lhs, RValue<SByte> rhs);
526RValue<SByte> operator>>=(SByte &lhs, RValue<SByte> rhs);
527RValue<SByte> operator+(RValue<SByte> val);
528RValue<SByte> operator-(RValue<SByte> val);
529RValue<SByte> operator~(RValue<SByte> val);
Ben Clayton713b8d32019-12-17 20:37:56 +0000530RValue<SByte> operator++(SByte &val, int); // Post-increment
531const SByte &operator++(SByte &val); // Pre-increment
532RValue<SByte> operator--(SByte &val, int); // Post-decrement
533const SByte &operator--(SByte &val); // Pre-decrement
Nicolas Capens157ba262019-12-10 17:49:14 -0500534RValue<Bool> operator<(RValue<SByte> lhs, RValue<SByte> rhs);
535RValue<Bool> operator<=(RValue<SByte> lhs, RValue<SByte> rhs);
536RValue<Bool> operator>(RValue<SByte> lhs, RValue<SByte> rhs);
537RValue<Bool> operator>=(RValue<SByte> lhs, RValue<SByte> rhs);
538RValue<Bool> operator!=(RValue<SByte> lhs, RValue<SByte> rhs);
539RValue<Bool> operator==(RValue<SByte> lhs, RValue<SByte> rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -0400540
Nicolas Capens157ba262019-12-10 17:49:14 -0500541class Short : public LValue<Short>
542{
543public:
544 Short(Argument<Short> argument);
Nicolas Capensd022e412016-09-26 13:30:14 -0400545
Nicolas Capens157ba262019-12-10 17:49:14 -0500546 explicit Short(RValue<Int> cast);
Nicolas Capensd022e412016-09-26 13:30:14 -0400547
Nicolas Capens157ba262019-12-10 17:49:14 -0500548 Short() = default;
549 Short(short x);
550 Short(RValue<Short> rhs);
551 Short(const Short &rhs);
552 Short(const Reference<Short> &rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -0400553
Ben Clayton713b8d32019-12-17 20:37:56 +0000554 // RValue<Short> operator=(short rhs); // FIXME: Implement
Nicolas Capens157ba262019-12-10 17:49:14 -0500555 RValue<Short> operator=(RValue<Short> rhs);
556 RValue<Short> operator=(const Short &rhs);
557 RValue<Short> operator=(const Reference<Short> &rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -0400558
Nicolas Capens519cf222020-05-08 15:27:19 -0400559 static Type *type();
Nicolas Capens157ba262019-12-10 17:49:14 -0500560};
Nicolas Capensd022e412016-09-26 13:30:14 -0400561
Nicolas Capens157ba262019-12-10 17:49:14 -0500562RValue<Short> operator+(RValue<Short> lhs, RValue<Short> rhs);
563RValue<Short> operator-(RValue<Short> lhs, RValue<Short> rhs);
564RValue<Short> operator*(RValue<Short> lhs, RValue<Short> rhs);
565RValue<Short> operator/(RValue<Short> lhs, RValue<Short> rhs);
566RValue<Short> operator%(RValue<Short> lhs, RValue<Short> rhs);
567RValue<Short> operator&(RValue<Short> lhs, RValue<Short> rhs);
568RValue<Short> operator|(RValue<Short> lhs, RValue<Short> rhs);
569RValue<Short> operator^(RValue<Short> lhs, RValue<Short> rhs);
570RValue<Short> operator<<(RValue<Short> lhs, RValue<Short> rhs);
571RValue<Short> operator>>(RValue<Short> lhs, RValue<Short> rhs);
572RValue<Short> operator+=(Short &lhs, RValue<Short> rhs);
573RValue<Short> operator-=(Short &lhs, RValue<Short> rhs);
574RValue<Short> operator*=(Short &lhs, RValue<Short> rhs);
575RValue<Short> operator/=(Short &lhs, RValue<Short> rhs);
576RValue<Short> operator%=(Short &lhs, RValue<Short> rhs);
577RValue<Short> operator&=(Short &lhs, RValue<Short> rhs);
578RValue<Short> operator|=(Short &lhs, RValue<Short> rhs);
579RValue<Short> operator^=(Short &lhs, RValue<Short> rhs);
580RValue<Short> operator<<=(Short &lhs, RValue<Short> rhs);
581RValue<Short> operator>>=(Short &lhs, RValue<Short> rhs);
582RValue<Short> operator+(RValue<Short> val);
583RValue<Short> operator-(RValue<Short> val);
584RValue<Short> operator~(RValue<Short> val);
Ben Clayton713b8d32019-12-17 20:37:56 +0000585RValue<Short> operator++(Short &val, int); // Post-increment
586const Short &operator++(Short &val); // Pre-increment
587RValue<Short> operator--(Short &val, int); // Post-decrement
588const Short &operator--(Short &val); // Pre-decrement
Nicolas Capens157ba262019-12-10 17:49:14 -0500589RValue<Bool> operator<(RValue<Short> lhs, RValue<Short> rhs);
590RValue<Bool> operator<=(RValue<Short> lhs, RValue<Short> rhs);
591RValue<Bool> operator>(RValue<Short> lhs, RValue<Short> rhs);
592RValue<Bool> operator>=(RValue<Short> lhs, RValue<Short> rhs);
593RValue<Bool> operator!=(RValue<Short> lhs, RValue<Short> rhs);
594RValue<Bool> operator==(RValue<Short> lhs, RValue<Short> rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -0400595
Nicolas Capens157ba262019-12-10 17:49:14 -0500596class UShort : public LValue<UShort>
597{
598public:
599 UShort(Argument<UShort> argument);
Nicolas Capensd022e412016-09-26 13:30:14 -0400600
Nicolas Capens157ba262019-12-10 17:49:14 -0500601 explicit UShort(RValue<UInt> cast);
602 explicit UShort(RValue<Int> cast);
Jason Macnak341ad7e2022-03-16 18:17:57 -0700603 explicit UShort(RValue<Byte> cast);
Nicolas Capensd022e412016-09-26 13:30:14 -0400604
Nicolas Capens157ba262019-12-10 17:49:14 -0500605 UShort() = default;
606 UShort(unsigned short x);
607 UShort(RValue<UShort> rhs);
608 UShort(const UShort &rhs);
609 UShort(const Reference<UShort> &rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -0400610
Ben Clayton713b8d32019-12-17 20:37:56 +0000611 // RValue<UShort> operator=(unsigned short rhs); // FIXME: Implement
Nicolas Capens157ba262019-12-10 17:49:14 -0500612 RValue<UShort> operator=(RValue<UShort> rhs);
613 RValue<UShort> operator=(const UShort &rhs);
614 RValue<UShort> operator=(const Reference<UShort> &rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -0400615
Nicolas Capens519cf222020-05-08 15:27:19 -0400616 static Type *type();
Nicolas Capens157ba262019-12-10 17:49:14 -0500617};
Nicolas Capensd022e412016-09-26 13:30:14 -0400618
Nicolas Capens157ba262019-12-10 17:49:14 -0500619RValue<UShort> operator+(RValue<UShort> lhs, RValue<UShort> rhs);
620RValue<UShort> operator-(RValue<UShort> lhs, RValue<UShort> rhs);
621RValue<UShort> operator*(RValue<UShort> lhs, RValue<UShort> rhs);
622RValue<UShort> operator/(RValue<UShort> lhs, RValue<UShort> rhs);
623RValue<UShort> operator%(RValue<UShort> lhs, RValue<UShort> rhs);
624RValue<UShort> operator&(RValue<UShort> lhs, RValue<UShort> rhs);
625RValue<UShort> operator|(RValue<UShort> lhs, RValue<UShort> rhs);
626RValue<UShort> operator^(RValue<UShort> lhs, RValue<UShort> rhs);
627RValue<UShort> operator<<(RValue<UShort> lhs, RValue<UShort> rhs);
628RValue<UShort> operator>>(RValue<UShort> lhs, RValue<UShort> rhs);
629RValue<UShort> operator+=(UShort &lhs, RValue<UShort> rhs);
630RValue<UShort> operator-=(UShort &lhs, RValue<UShort> rhs);
631RValue<UShort> operator*=(UShort &lhs, RValue<UShort> rhs);
632RValue<UShort> operator/=(UShort &lhs, RValue<UShort> rhs);
633RValue<UShort> operator%=(UShort &lhs, RValue<UShort> rhs);
634RValue<UShort> operator&=(UShort &lhs, RValue<UShort> rhs);
635RValue<UShort> operator|=(UShort &lhs, RValue<UShort> rhs);
636RValue<UShort> operator^=(UShort &lhs, RValue<UShort> rhs);
637RValue<UShort> operator<<=(UShort &lhs, RValue<UShort> rhs);
638RValue<UShort> operator>>=(UShort &lhs, RValue<UShort> rhs);
639RValue<UShort> operator+(RValue<UShort> val);
640RValue<UShort> operator-(RValue<UShort> val);
641RValue<UShort> operator~(RValue<UShort> val);
Ben Clayton713b8d32019-12-17 20:37:56 +0000642RValue<UShort> operator++(UShort &val, int); // Post-increment
643const UShort &operator++(UShort &val); // Pre-increment
644RValue<UShort> operator--(UShort &val, int); // Post-decrement
645const UShort &operator--(UShort &val); // Pre-decrement
Nicolas Capens157ba262019-12-10 17:49:14 -0500646RValue<Bool> operator<(RValue<UShort> lhs, RValue<UShort> rhs);
647RValue<Bool> operator<=(RValue<UShort> lhs, RValue<UShort> rhs);
648RValue<Bool> operator>(RValue<UShort> lhs, RValue<UShort> rhs);
649RValue<Bool> operator>=(RValue<UShort> lhs, RValue<UShort> rhs);
650RValue<Bool> operator!=(RValue<UShort> lhs, RValue<UShort> rhs);
651RValue<Bool> operator==(RValue<UShort> lhs, RValue<UShort> rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -0400652
Nicolas Capens157ba262019-12-10 17:49:14 -0500653class Byte4 : public LValue<Byte4>
654{
655public:
656 explicit Byte4(RValue<Byte8> cast);
Nicolas Capens133b87d2020-01-25 16:26:28 -0500657 explicit Byte4(RValue<UShort4> cast);
658 explicit Byte4(RValue<Short4> cast);
659 explicit Byte4(RValue<UInt4> cast);
660 explicit Byte4(RValue<Int4> cast);
Nicolas Capens16b5f152016-10-13 13:39:01 -0400661
Nicolas Capens157ba262019-12-10 17:49:14 -0500662 Byte4() = default;
Ben Clayton713b8d32019-12-17 20:37:56 +0000663 // Byte4(int x, int y, int z, int w);
Nicolas Capens133b87d2020-01-25 16:26:28 -0500664 Byte4(RValue<Byte4> rhs);
665 Byte4(const Byte4 &rhs);
Nicolas Capens157ba262019-12-10 17:49:14 -0500666 Byte4(const Reference<Byte4> &rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -0400667
Nicolas Capens133b87d2020-01-25 16:26:28 -0500668 RValue<Byte4> operator=(RValue<Byte4> rhs);
669 RValue<Byte4> operator=(const Byte4 &rhs);
Ben Clayton713b8d32019-12-17 20:37:56 +0000670 // RValue<Byte4> operator=(const Reference<Byte4> &rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -0400671
Nicolas Capens519cf222020-05-08 15:27:19 -0400672 static Type *type();
Nicolas Capens442e25b2022-06-22 12:02:52 -0400673 static int element_count() { return 4; }
Nicolas Capens157ba262019-12-10 17:49:14 -0500674};
Nicolas Capensd022e412016-09-26 13:30:14 -0400675
Jason Macnak341ad7e2022-03-16 18:17:57 -0700676RValue<Byte4> Insert(RValue<Byte4> val, RValue<Byte> element, int i);
677
Nicolas Capensd022e412016-09-26 13:30:14 -0400678// RValue<Byte4> operator+(RValue<Byte4> lhs, RValue<Byte4> rhs);
679// RValue<Byte4> operator-(RValue<Byte4> lhs, RValue<Byte4> rhs);
680// RValue<Byte4> operator*(RValue<Byte4> lhs, RValue<Byte4> rhs);
681// RValue<Byte4> operator/(RValue<Byte4> lhs, RValue<Byte4> rhs);
682// RValue<Byte4> operator%(RValue<Byte4> lhs, RValue<Byte4> rhs);
683// RValue<Byte4> operator&(RValue<Byte4> lhs, RValue<Byte4> rhs);
684// RValue<Byte4> operator|(RValue<Byte4> lhs, RValue<Byte4> rhs);
685// RValue<Byte4> operator^(RValue<Byte4> lhs, RValue<Byte4> rhs);
686// RValue<Byte4> operator<<(RValue<Byte4> lhs, RValue<Byte4> rhs);
687// RValue<Byte4> operator>>(RValue<Byte4> lhs, RValue<Byte4> rhs);
Nicolas Capens96d4e092016-11-18 14:22:38 -0500688// RValue<Byte4> operator+=(Byte4 &lhs, RValue<Byte4> rhs);
689// RValue<Byte4> operator-=(Byte4 &lhs, RValue<Byte4> rhs);
690// RValue<Byte4> operator*=(Byte4 &lhs, RValue<Byte4> rhs);
691// RValue<Byte4> operator/=(Byte4 &lhs, RValue<Byte4> rhs);
692// RValue<Byte4> operator%=(Byte4 &lhs, RValue<Byte4> rhs);
693// RValue<Byte4> operator&=(Byte4 &lhs, RValue<Byte4> rhs);
694// RValue<Byte4> operator|=(Byte4 &lhs, RValue<Byte4> rhs);
695// RValue<Byte4> operator^=(Byte4 &lhs, RValue<Byte4> rhs);
696// RValue<Byte4> operator<<=(Byte4 &lhs, RValue<Byte4> rhs);
697// RValue<Byte4> operator>>=(Byte4 &lhs, RValue<Byte4> rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -0400698// RValue<Byte4> operator+(RValue<Byte4> val);
699// RValue<Byte4> operator-(RValue<Byte4> val);
700// RValue<Byte4> operator~(RValue<Byte4> val);
Nicolas Capens96d4e092016-11-18 14:22:38 -0500701// RValue<Byte4> operator++(Byte4 &val, int); // Post-increment
702// const Byte4 &operator++(Byte4 &val); // Pre-increment
703// RValue<Byte4> operator--(Byte4 &val, int); // Post-decrement
704// const Byte4 &operator--(Byte4 &val); // Pre-decrement
Nicolas Capensd022e412016-09-26 13:30:14 -0400705
Nicolas Capens157ba262019-12-10 17:49:14 -0500706class SByte4 : public LValue<SByte4>
707{
708public:
709 SByte4() = default;
Ben Clayton713b8d32019-12-17 20:37:56 +0000710 // SByte4(int x, int y, int z, int w);
711 // SByte4(RValue<SByte4> rhs);
712 // SByte4(const SByte4 &rhs);
713 // SByte4(const Reference<SByte4> &rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -0400714
Ben Clayton713b8d32019-12-17 20:37:56 +0000715 // RValue<SByte4> operator=(RValue<SByte4> rhs);
716 // RValue<SByte4> operator=(const SByte4 &rhs);
717 // RValue<SByte4> operator=(const Reference<SByte4> &rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -0400718
Nicolas Capens519cf222020-05-08 15:27:19 -0400719 static Type *type();
Nicolas Capens442e25b2022-06-22 12:02:52 -0400720 static int element_count() { return 4; }
Nicolas Capens157ba262019-12-10 17:49:14 -0500721};
Nicolas Capensd022e412016-09-26 13:30:14 -0400722
723// RValue<SByte4> operator+(RValue<SByte4> lhs, RValue<SByte4> rhs);
724// RValue<SByte4> operator-(RValue<SByte4> lhs, RValue<SByte4> rhs);
725// RValue<SByte4> operator*(RValue<SByte4> lhs, RValue<SByte4> rhs);
726// RValue<SByte4> operator/(RValue<SByte4> lhs, RValue<SByte4> rhs);
727// RValue<SByte4> operator%(RValue<SByte4> lhs, RValue<SByte4> rhs);
728// RValue<SByte4> operator&(RValue<SByte4> lhs, RValue<SByte4> rhs);
729// RValue<SByte4> operator|(RValue<SByte4> lhs, RValue<SByte4> rhs);
730// RValue<SByte4> operator^(RValue<SByte4> lhs, RValue<SByte4> rhs);
731// RValue<SByte4> operator<<(RValue<SByte4> lhs, RValue<SByte4> rhs);
732// RValue<SByte4> operator>>(RValue<SByte4> lhs, RValue<SByte4> rhs);
Nicolas Capens96d4e092016-11-18 14:22:38 -0500733// RValue<SByte4> operator+=(SByte4 &lhs, RValue<SByte4> rhs);
734// RValue<SByte4> operator-=(SByte4 &lhs, RValue<SByte4> rhs);
735// RValue<SByte4> operator*=(SByte4 &lhs, RValue<SByte4> rhs);
736// RValue<SByte4> operator/=(SByte4 &lhs, RValue<SByte4> rhs);
737// RValue<SByte4> operator%=(SByte4 &lhs, RValue<SByte4> rhs);
738// RValue<SByte4> operator&=(SByte4 &lhs, RValue<SByte4> rhs);
739// RValue<SByte4> operator|=(SByte4 &lhs, RValue<SByte4> rhs);
740// RValue<SByte4> operator^=(SByte4 &lhs, RValue<SByte4> rhs);
741// RValue<SByte4> operator<<=(SByte4 &lhs, RValue<SByte4> rhs);
742// RValue<SByte4> operator>>=(SByte4 &lhs, RValue<SByte4> rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -0400743// RValue<SByte4> operator+(RValue<SByte4> val);
744// RValue<SByte4> operator-(RValue<SByte4> val);
745// RValue<SByte4> operator~(RValue<SByte4> val);
Nicolas Capens96d4e092016-11-18 14:22:38 -0500746// RValue<SByte4> operator++(SByte4 &val, int); // Post-increment
747// const SByte4 &operator++(SByte4 &val); // Pre-increment
748// RValue<SByte4> operator--(SByte4 &val, int); // Post-decrement
749// const SByte4 &operator--(SByte4 &val); // Pre-decrement
Nicolas Capensd022e412016-09-26 13:30:14 -0400750
Nicolas Capens157ba262019-12-10 17:49:14 -0500751class Byte8 : public LValue<Byte8>
752{
753public:
754 Byte8() = default;
755 Byte8(uint8_t x0, uint8_t x1, uint8_t x2, uint8_t x3, uint8_t x4, uint8_t x5, uint8_t x6, uint8_t x7);
756 Byte8(RValue<Byte8> rhs);
757 Byte8(const Byte8 &rhs);
758 Byte8(const Reference<Byte8> &rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -0400759
Nicolas Capens157ba262019-12-10 17:49:14 -0500760 RValue<Byte8> operator=(RValue<Byte8> rhs);
761 RValue<Byte8> operator=(const Byte8 &rhs);
762 RValue<Byte8> operator=(const Reference<Byte8> &rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -0400763
Nicolas Capens519cf222020-05-08 15:27:19 -0400764 static Type *type();
Nicolas Capens442e25b2022-06-22 12:02:52 -0400765 static int element_count() { return 8; }
Nicolas Capens157ba262019-12-10 17:49:14 -0500766};
Nicolas Capensd022e412016-09-26 13:30:14 -0400767
Nicolas Capens157ba262019-12-10 17:49:14 -0500768RValue<Byte8> operator+(RValue<Byte8> lhs, RValue<Byte8> rhs);
769RValue<Byte8> operator-(RValue<Byte8> lhs, RValue<Byte8> rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -0400770// RValue<Byte8> operator*(RValue<Byte8> lhs, RValue<Byte8> rhs);
771// RValue<Byte8> operator/(RValue<Byte8> lhs, RValue<Byte8> rhs);
772// RValue<Byte8> operator%(RValue<Byte8> lhs, RValue<Byte8> rhs);
Nicolas Capens157ba262019-12-10 17:49:14 -0500773RValue<Byte8> operator&(RValue<Byte8> lhs, RValue<Byte8> rhs);
774RValue<Byte8> operator|(RValue<Byte8> lhs, RValue<Byte8> rhs);
775RValue<Byte8> operator^(RValue<Byte8> lhs, RValue<Byte8> rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -0400776// RValue<Byte8> operator<<(RValue<Byte8> lhs, RValue<Byte8> rhs);
777// RValue<Byte8> operator>>(RValue<Byte8> lhs, RValue<Byte8> rhs);
Nicolas Capens157ba262019-12-10 17:49:14 -0500778RValue<Byte8> operator+=(Byte8 &lhs, RValue<Byte8> rhs);
779RValue<Byte8> operator-=(Byte8 &lhs, RValue<Byte8> rhs);
Nicolas Capens96d4e092016-11-18 14:22:38 -0500780// RValue<Byte8> operator*=(Byte8 &lhs, RValue<Byte8> rhs);
781// RValue<Byte8> operator/=(Byte8 &lhs, RValue<Byte8> rhs);
782// RValue<Byte8> operator%=(Byte8 &lhs, RValue<Byte8> rhs);
Nicolas Capens157ba262019-12-10 17:49:14 -0500783RValue<Byte8> operator&=(Byte8 &lhs, RValue<Byte8> rhs);
784RValue<Byte8> operator|=(Byte8 &lhs, RValue<Byte8> rhs);
785RValue<Byte8> operator^=(Byte8 &lhs, RValue<Byte8> rhs);
Nicolas Capens96d4e092016-11-18 14:22:38 -0500786// RValue<Byte8> operator<<=(Byte8 &lhs, RValue<Byte8> rhs);
787// RValue<Byte8> operator>>=(Byte8 &lhs, RValue<Byte8> rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -0400788// RValue<Byte8> operator+(RValue<Byte8> val);
789// RValue<Byte8> operator-(RValue<Byte8> val);
Nicolas Capens157ba262019-12-10 17:49:14 -0500790RValue<Byte8> operator~(RValue<Byte8> val);
Nicolas Capens96d4e092016-11-18 14:22:38 -0500791// RValue<Byte8> operator++(Byte8 &val, int); // Post-increment
792// const Byte8 &operator++(Byte8 &val); // Pre-increment
793// RValue<Byte8> operator--(Byte8 &val, int); // Post-decrement
794// const Byte8 &operator--(Byte8 &val); // Pre-decrement
Nicolas Capensd022e412016-09-26 13:30:14 -0400795
Nicolas Capens157ba262019-12-10 17:49:14 -0500796RValue<Byte8> AddSat(RValue<Byte8> x, RValue<Byte8> y);
797RValue<Byte8> SubSat(RValue<Byte8> x, RValue<Byte8> y);
798RValue<Short4> Unpack(RValue<Byte4> x);
799RValue<Short4> Unpack(RValue<Byte4> x, RValue<Byte4> y);
800RValue<Short4> UnpackLow(RValue<Byte8> x, RValue<Byte8> y);
801RValue<Short4> UnpackHigh(RValue<Byte8> x, RValue<Byte8> y);
802RValue<Int> SignMask(RValue<Byte8> x);
Nicolas Capensd022e412016-09-26 13:30:14 -0400803// RValue<Byte8> CmpGT(RValue<Byte8> x, RValue<Byte8> y);
Nicolas Capens157ba262019-12-10 17:49:14 -0500804RValue<Byte8> CmpEQ(RValue<Byte8> x, RValue<Byte8> y);
Nicolas Capens133b87d2020-01-25 16:26:28 -0500805RValue<Byte8> Swizzle(RValue<Byte8> x, uint32_t select);
Nicolas Capensd022e412016-09-26 13:30:14 -0400806
Nicolas Capens157ba262019-12-10 17:49:14 -0500807class SByte8 : public LValue<SByte8>
808{
809public:
810 SByte8() = default;
811 SByte8(uint8_t x0, uint8_t x1, uint8_t x2, uint8_t x3, uint8_t x4, uint8_t x5, uint8_t x6, uint8_t x7);
812 SByte8(RValue<SByte8> rhs);
813 SByte8(const SByte8 &rhs);
814 SByte8(const Reference<SByte8> &rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -0400815
Nicolas Capens157ba262019-12-10 17:49:14 -0500816 RValue<SByte8> operator=(RValue<SByte8> rhs);
817 RValue<SByte8> operator=(const SByte8 &rhs);
818 RValue<SByte8> operator=(const Reference<SByte8> &rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -0400819
Nicolas Capens519cf222020-05-08 15:27:19 -0400820 static Type *type();
Nicolas Capens442e25b2022-06-22 12:02:52 -0400821 static int element_count() { return 8; }
Nicolas Capens157ba262019-12-10 17:49:14 -0500822};
Nicolas Capensd022e412016-09-26 13:30:14 -0400823
Nicolas Capens157ba262019-12-10 17:49:14 -0500824RValue<SByte8> operator+(RValue<SByte8> lhs, RValue<SByte8> rhs);
825RValue<SByte8> operator-(RValue<SByte8> lhs, RValue<SByte8> rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -0400826// RValue<SByte8> operator*(RValue<SByte8> lhs, RValue<SByte8> rhs);
827// RValue<SByte8> operator/(RValue<SByte8> lhs, RValue<SByte8> rhs);
828// RValue<SByte8> operator%(RValue<SByte8> lhs, RValue<SByte8> rhs);
Nicolas Capens157ba262019-12-10 17:49:14 -0500829RValue<SByte8> operator&(RValue<SByte8> lhs, RValue<SByte8> rhs);
830RValue<SByte8> operator|(RValue<SByte8> lhs, RValue<SByte8> rhs);
831RValue<SByte8> operator^(RValue<SByte8> lhs, RValue<SByte8> rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -0400832// RValue<SByte8> operator<<(RValue<SByte8> lhs, RValue<SByte8> rhs);
833// RValue<SByte8> operator>>(RValue<SByte8> lhs, RValue<SByte8> rhs);
Nicolas Capens157ba262019-12-10 17:49:14 -0500834RValue<SByte8> operator+=(SByte8 &lhs, RValue<SByte8> rhs);
835RValue<SByte8> operator-=(SByte8 &lhs, RValue<SByte8> rhs);
Nicolas Capens96d4e092016-11-18 14:22:38 -0500836// RValue<SByte8> operator*=(SByte8 &lhs, RValue<SByte8> rhs);
837// RValue<SByte8> operator/=(SByte8 &lhs, RValue<SByte8> rhs);
838// RValue<SByte8> operator%=(SByte8 &lhs, RValue<SByte8> rhs);
Nicolas Capens157ba262019-12-10 17:49:14 -0500839RValue<SByte8> operator&=(SByte8 &lhs, RValue<SByte8> rhs);
840RValue<SByte8> operator|=(SByte8 &lhs, RValue<SByte8> rhs);
841RValue<SByte8> operator^=(SByte8 &lhs, RValue<SByte8> rhs);
Nicolas Capens96d4e092016-11-18 14:22:38 -0500842// RValue<SByte8> operator<<=(SByte8 &lhs, RValue<SByte8> rhs);
843// RValue<SByte8> operator>>=(SByte8 &lhs, RValue<SByte8> rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -0400844// RValue<SByte8> operator+(RValue<SByte8> val);
845// RValue<SByte8> operator-(RValue<SByte8> val);
Nicolas Capens157ba262019-12-10 17:49:14 -0500846RValue<SByte8> operator~(RValue<SByte8> val);
Nicolas Capens96d4e092016-11-18 14:22:38 -0500847// RValue<SByte8> operator++(SByte8 &val, int); // Post-increment
848// const SByte8 &operator++(SByte8 &val); // Pre-increment
849// RValue<SByte8> operator--(SByte8 &val, int); // Post-decrement
850// const SByte8 &operator--(SByte8 &val); // Pre-decrement
Nicolas Capensd022e412016-09-26 13:30:14 -0400851
Nicolas Capens157ba262019-12-10 17:49:14 -0500852RValue<SByte8> AddSat(RValue<SByte8> x, RValue<SByte8> y);
853RValue<SByte8> SubSat(RValue<SByte8> x, RValue<SByte8> y);
854RValue<Short4> UnpackLow(RValue<SByte8> x, RValue<SByte8> y);
855RValue<Short4> UnpackHigh(RValue<SByte8> x, RValue<SByte8> y);
856RValue<Int> SignMask(RValue<SByte8> x);
857RValue<Byte8> CmpGT(RValue<SByte8> x, RValue<SByte8> y);
858RValue<Byte8> CmpEQ(RValue<SByte8> x, RValue<SByte8> y);
Nicolas Capensd022e412016-09-26 13:30:14 -0400859
Nicolas Capens157ba262019-12-10 17:49:14 -0500860class Byte16 : public LValue<Byte16>
861{
862public:
863 Byte16() = default;
Nicolas Capens157ba262019-12-10 17:49:14 -0500864 Byte16(RValue<Byte16> rhs);
865 Byte16(const Byte16 &rhs);
866 Byte16(const Reference<Byte16> &rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -0400867
Nicolas Capens157ba262019-12-10 17:49:14 -0500868 RValue<Byte16> operator=(RValue<Byte16> rhs);
869 RValue<Byte16> operator=(const Byte16 &rhs);
870 RValue<Byte16> operator=(const Reference<Byte16> &rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -0400871
Nicolas Capens519cf222020-05-08 15:27:19 -0400872 static Type *type();
Nicolas Capens442e25b2022-06-22 12:02:52 -0400873 static int element_count() { return 16; }
Nicolas Capens157ba262019-12-10 17:49:14 -0500874};
Nicolas Capensd022e412016-09-26 13:30:14 -0400875
876// RValue<Byte16> operator+(RValue<Byte16> lhs, RValue<Byte16> rhs);
877// RValue<Byte16> operator-(RValue<Byte16> lhs, RValue<Byte16> rhs);
878// RValue<Byte16> operator*(RValue<Byte16> lhs, RValue<Byte16> rhs);
879// RValue<Byte16> operator/(RValue<Byte16> lhs, RValue<Byte16> rhs);
880// RValue<Byte16> operator%(RValue<Byte16> lhs, RValue<Byte16> rhs);
881// RValue<Byte16> operator&(RValue<Byte16> lhs, RValue<Byte16> rhs);
882// RValue<Byte16> operator|(RValue<Byte16> lhs, RValue<Byte16> rhs);
883// RValue<Byte16> operator^(RValue<Byte16> lhs, RValue<Byte16> rhs);
884// RValue<Byte16> operator<<(RValue<Byte16> lhs, RValue<Byte16> rhs);
885// RValue<Byte16> operator>>(RValue<Byte16> lhs, RValue<Byte16> rhs);
Nicolas Capens96d4e092016-11-18 14:22:38 -0500886// RValue<Byte16> operator+=(Byte16 &lhs, RValue<Byte16> rhs);
887// RValue<Byte16> operator-=(Byte16 &lhs, RValue<Byte16> rhs);
888// RValue<Byte16> operator*=(Byte16 &lhs, RValue<Byte16> rhs);
889// RValue<Byte16> operator/=(Byte16 &lhs, RValue<Byte16> rhs);
890// RValue<Byte16> operator%=(Byte16 &lhs, RValue<Byte16> rhs);
891// RValue<Byte16> operator&=(Byte16 &lhs, RValue<Byte16> rhs);
892// RValue<Byte16> operator|=(Byte16 &lhs, RValue<Byte16> rhs);
893// RValue<Byte16> operator^=(Byte16 &lhs, RValue<Byte16> rhs);
894// RValue<Byte16> operator<<=(Byte16 &lhs, RValue<Byte16> rhs);
895// RValue<Byte16> operator>>=(Byte16 &lhs, RValue<Byte16> rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -0400896// RValue<Byte16> operator+(RValue<Byte16> val);
897// RValue<Byte16> operator-(RValue<Byte16> val);
898// RValue<Byte16> operator~(RValue<Byte16> val);
Nicolas Capens96d4e092016-11-18 14:22:38 -0500899// RValue<Byte16> operator++(Byte16 &val, int); // Post-increment
900// const Byte16 &operator++(Byte16 &val); // Pre-increment
901// RValue<Byte16> operator--(Byte16 &val, int); // Post-decrement
902// const Byte16 &operator--(Byte16 &val); // Pre-decrement
Nicolas Capens133b87d2020-01-25 16:26:28 -0500903RValue<Byte16> Swizzle(RValue<Byte16> x, uint64_t select);
Nicolas Capensd022e412016-09-26 13:30:14 -0400904
Nicolas Capens157ba262019-12-10 17:49:14 -0500905class SByte16 : public LValue<SByte16>
906{
907public:
908 SByte16() = default;
Ben Clayton713b8d32019-12-17 20:37:56 +0000909 // SByte16(int x, int y, int z, int w);
910 // SByte16(RValue<SByte16> rhs);
911 // SByte16(const SByte16 &rhs);
912 // SByte16(const Reference<SByte16> &rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -0400913
Ben Clayton713b8d32019-12-17 20:37:56 +0000914 // RValue<SByte16> operator=(RValue<SByte16> rhs);
915 // RValue<SByte16> operator=(const SByte16 &rhs);
916 // RValue<SByte16> operator=(const Reference<SByte16> &rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -0400917
Nicolas Capens519cf222020-05-08 15:27:19 -0400918 static Type *type();
Nicolas Capens442e25b2022-06-22 12:02:52 -0400919 static int element_count() { return 16; }
Nicolas Capens157ba262019-12-10 17:49:14 -0500920};
Nicolas Capensd022e412016-09-26 13:30:14 -0400921
922// RValue<SByte16> operator+(RValue<SByte16> lhs, RValue<SByte16> rhs);
923// RValue<SByte16> operator-(RValue<SByte16> lhs, RValue<SByte16> rhs);
924// RValue<SByte16> operator*(RValue<SByte16> lhs, RValue<SByte16> rhs);
925// RValue<SByte16> operator/(RValue<SByte16> lhs, RValue<SByte16> rhs);
926// RValue<SByte16> operator%(RValue<SByte16> lhs, RValue<SByte16> rhs);
927// RValue<SByte16> operator&(RValue<SByte16> lhs, RValue<SByte16> rhs);
928// RValue<SByte16> operator|(RValue<SByte16> lhs, RValue<SByte16> rhs);
929// RValue<SByte16> operator^(RValue<SByte16> lhs, RValue<SByte16> rhs);
930// RValue<SByte16> operator<<(RValue<SByte16> lhs, RValue<SByte16> rhs);
931// RValue<SByte16> operator>>(RValue<SByte16> lhs, RValue<SByte16> rhs);
Nicolas Capens96d4e092016-11-18 14:22:38 -0500932// RValue<SByte16> operator+=(SByte16 &lhs, RValue<SByte16> rhs);
933// RValue<SByte16> operator-=(SByte16 &lhs, RValue<SByte16> rhs);
934// RValue<SByte16> operator*=(SByte16 &lhs, RValue<SByte16> rhs);
935// RValue<SByte16> operator/=(SByte16 &lhs, RValue<SByte16> rhs);
936// RValue<SByte16> operator%=(SByte16 &lhs, RValue<SByte16> rhs);
937// RValue<SByte16> operator&=(SByte16 &lhs, RValue<SByte16> rhs);
938// RValue<SByte16> operator|=(SByte16 &lhs, RValue<SByte16> rhs);
939// RValue<SByte16> operator^=(SByte16 &lhs, RValue<SByte16> rhs);
940// RValue<SByte16> operator<<=(SByte16 &lhs, RValue<SByte16> rhs);
941// RValue<SByte16> operator>>=(SByte16 &lhs, RValue<SByte16> rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -0400942// RValue<SByte16> operator+(RValue<SByte16> val);
943// RValue<SByte16> operator-(RValue<SByte16> val);
944// RValue<SByte16> operator~(RValue<SByte16> val);
Nicolas Capens96d4e092016-11-18 14:22:38 -0500945// RValue<SByte16> operator++(SByte16 &val, int); // Post-increment
946// const SByte16 &operator++(SByte16 &val); // Pre-increment
947// RValue<SByte16> operator--(SByte16 &val, int); // Post-decrement
948// const SByte16 &operator--(SByte16 &val); // Pre-decrement
Nicolas Capensd022e412016-09-26 13:30:14 -0400949
Nicolas Capens157ba262019-12-10 17:49:14 -0500950class Short2 : public LValue<Short2>
951{
952public:
953 explicit Short2(RValue<Short4> cast);
Nicolas Capens16b5f152016-10-13 13:39:01 -0400954
Nicolas Capens519cf222020-05-08 15:27:19 -0400955 static Type *type();
Nicolas Capens442e25b2022-06-22 12:02:52 -0400956 static int element_count() { return 2; }
Nicolas Capens157ba262019-12-10 17:49:14 -0500957};
Nicolas Capens16b5f152016-10-13 13:39:01 -0400958
Nicolas Capens157ba262019-12-10 17:49:14 -0500959class UShort2 : public LValue<UShort2>
960{
961public:
962 explicit UShort2(RValue<UShort4> cast);
Nicolas Capens16b5f152016-10-13 13:39:01 -0400963
Nicolas Capens519cf222020-05-08 15:27:19 -0400964 static Type *type();
Nicolas Capens442e25b2022-06-22 12:02:52 -0400965 static int element_count() { return 2; }
Nicolas Capens157ba262019-12-10 17:49:14 -0500966};
Nicolas Capens16b5f152016-10-13 13:39:01 -0400967
Nicolas Capens157ba262019-12-10 17:49:14 -0500968class Short4 : public LValue<Short4>
969{
970public:
971 explicit Short4(RValue<Int> cast);
972 explicit Short4(RValue<Int4> cast);
Nicolas Capens71196862022-01-06 21:31:57 -0500973 explicit Short4(RValue<UInt4> cast);
Ben Clayton713b8d32019-12-17 20:37:56 +0000974 // explicit Short4(RValue<Float> cast);
Nicolas Capens157ba262019-12-10 17:49:14 -0500975 explicit Short4(RValue<Float4> cast);
Nicolas Capensd022e412016-09-26 13:30:14 -0400976
Nicolas Capens157ba262019-12-10 17:49:14 -0500977 Short4() = default;
978 Short4(short xyzw);
979 Short4(short x, short y, short z, short w);
980 Short4(RValue<Short4> rhs);
981 Short4(const Short4 &rhs);
982 Short4(const Reference<Short4> &rhs);
983 Short4(RValue<UShort4> rhs);
984 Short4(const UShort4 &rhs);
985 Short4(const Reference<UShort4> &rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -0400986
Nicolas Capens157ba262019-12-10 17:49:14 -0500987 RValue<Short4> operator=(RValue<Short4> rhs);
988 RValue<Short4> operator=(const Short4 &rhs);
989 RValue<Short4> operator=(const Reference<Short4> &rhs);
990 RValue<Short4> operator=(RValue<UShort4> rhs);
991 RValue<Short4> operator=(const UShort4 &rhs);
992 RValue<Short4> operator=(const Reference<UShort4> &rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -0400993
Nicolas Capens519cf222020-05-08 15:27:19 -0400994 static Type *type();
Nicolas Capens442e25b2022-06-22 12:02:52 -0400995 static int element_count() { return 4; }
Nicolas Capens157ba262019-12-10 17:49:14 -0500996};
Nicolas Capensd022e412016-09-26 13:30:14 -0400997
Nicolas Capens157ba262019-12-10 17:49:14 -0500998RValue<Short4> operator+(RValue<Short4> lhs, RValue<Short4> rhs);
999RValue<Short4> operator-(RValue<Short4> lhs, RValue<Short4> rhs);
1000RValue<Short4> operator*(RValue<Short4> lhs, RValue<Short4> rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -04001001// RValue<Short4> operator/(RValue<Short4> lhs, RValue<Short4> rhs);
1002// RValue<Short4> operator%(RValue<Short4> lhs, RValue<Short4> rhs);
Nicolas Capens157ba262019-12-10 17:49:14 -05001003RValue<Short4> operator&(RValue<Short4> lhs, RValue<Short4> rhs);
1004RValue<Short4> operator|(RValue<Short4> lhs, RValue<Short4> rhs);
1005RValue<Short4> operator^(RValue<Short4> lhs, RValue<Short4> rhs);
1006RValue<Short4> operator<<(RValue<Short4> lhs, unsigned char rhs);
1007RValue<Short4> operator>>(RValue<Short4> lhs, unsigned char rhs);
1008RValue<Short4> operator+=(Short4 &lhs, RValue<Short4> rhs);
1009RValue<Short4> operator-=(Short4 &lhs, RValue<Short4> rhs);
1010RValue<Short4> operator*=(Short4 &lhs, RValue<Short4> rhs);
Nicolas Capens96d4e092016-11-18 14:22:38 -05001011// RValue<Short4> operator/=(Short4 &lhs, RValue<Short4> rhs);
1012// RValue<Short4> operator%=(Short4 &lhs, RValue<Short4> rhs);
Nicolas Capens157ba262019-12-10 17:49:14 -05001013RValue<Short4> operator&=(Short4 &lhs, RValue<Short4> rhs);
1014RValue<Short4> operator|=(Short4 &lhs, RValue<Short4> rhs);
1015RValue<Short4> operator^=(Short4 &lhs, RValue<Short4> rhs);
1016RValue<Short4> operator<<=(Short4 &lhs, unsigned char rhs);
1017RValue<Short4> operator>>=(Short4 &lhs, unsigned char rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -04001018// RValue<Short4> operator+(RValue<Short4> val);
Nicolas Capens157ba262019-12-10 17:49:14 -05001019RValue<Short4> operator-(RValue<Short4> val);
1020RValue<Short4> operator~(RValue<Short4> val);
Nicolas Capens96d4e092016-11-18 14:22:38 -05001021// RValue<Short4> operator++(Short4 &val, int); // Post-increment
1022// const Short4 &operator++(Short4 &val); // Pre-increment
1023// RValue<Short4> operator--(Short4 &val, int); // Post-decrement
1024// const Short4 &operator--(Short4 &val); // Pre-decrement
Nicolas Capensd022e412016-09-26 13:30:14 -04001025// RValue<Bool> operator<(RValue<Short4> lhs, RValue<Short4> rhs);
1026// RValue<Bool> operator<=(RValue<Short4> lhs, RValue<Short4> rhs);
1027// RValue<Bool> operator>(RValue<Short4> lhs, RValue<Short4> rhs);
1028// RValue<Bool> operator>=(RValue<Short4> lhs, RValue<Short4> rhs);
1029// RValue<Bool> operator!=(RValue<Short4> lhs, RValue<Short4> rhs);
1030// RValue<Bool> operator==(RValue<Short4> lhs, RValue<Short4> rhs);
1031
Nicolas Capens157ba262019-12-10 17:49:14 -05001032RValue<Short4> RoundShort4(RValue<Float4> cast);
1033RValue<Short4> Max(RValue<Short4> x, RValue<Short4> y);
1034RValue<Short4> Min(RValue<Short4> x, RValue<Short4> y);
1035RValue<Short4> AddSat(RValue<Short4> x, RValue<Short4> y);
1036RValue<Short4> SubSat(RValue<Short4> x, RValue<Short4> y);
1037RValue<Short4> MulHigh(RValue<Short4> x, RValue<Short4> y);
1038RValue<Int2> MulAdd(RValue<Short4> x, RValue<Short4> y);
1039RValue<SByte8> PackSigned(RValue<Short4> x, RValue<Short4> y);
1040RValue<Byte8> PackUnsigned(RValue<Short4> x, RValue<Short4> y);
1041RValue<Int2> UnpackLow(RValue<Short4> x, RValue<Short4> y);
1042RValue<Int2> UnpackHigh(RValue<Short4> x, RValue<Short4> y);
1043RValue<Short4> Swizzle(RValue<Short4> x, uint16_t select);
1044RValue<Short4> Insert(RValue<Short4> val, RValue<Short> element, int i);
1045RValue<Short> Extract(RValue<Short4> val, int i);
1046RValue<Short4> CmpGT(RValue<Short4> x, RValue<Short4> y);
1047RValue<Short4> CmpEQ(RValue<Short4> x, RValue<Short4> y);
Nicolas Capensd022e412016-09-26 13:30:14 -04001048
Nicolas Capens157ba262019-12-10 17:49:14 -05001049class UShort4 : public LValue<UShort4>
1050{
1051public:
Nicolas Capens71196862022-01-06 21:31:57 -05001052 explicit UShort4(RValue<UInt4> cast);
Nicolas Capens157ba262019-12-10 17:49:14 -05001053 explicit UShort4(RValue<Int4> cast);
1054 explicit UShort4(RValue<Float4> cast, bool saturate = false);
Nicolas Capensd022e412016-09-26 13:30:14 -04001055
Nicolas Capens157ba262019-12-10 17:49:14 -05001056 UShort4() = default;
1057 UShort4(unsigned short xyzw);
1058 UShort4(unsigned short x, unsigned short y, unsigned short z, unsigned short w);
1059 UShort4(RValue<UShort4> rhs);
1060 UShort4(const UShort4 &rhs);
1061 UShort4(const Reference<UShort4> &rhs);
1062 UShort4(RValue<Short4> rhs);
1063 UShort4(const Short4 &rhs);
1064 UShort4(const Reference<Short4> &rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -04001065
Nicolas Capens157ba262019-12-10 17:49:14 -05001066 RValue<UShort4> operator=(RValue<UShort4> rhs);
1067 RValue<UShort4> operator=(const UShort4 &rhs);
1068 RValue<UShort4> operator=(const Reference<UShort4> &rhs);
1069 RValue<UShort4> operator=(RValue<Short4> rhs);
1070 RValue<UShort4> operator=(const Short4 &rhs);
1071 RValue<UShort4> operator=(const Reference<Short4> &rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -04001072
Nicolas Capens519cf222020-05-08 15:27:19 -04001073 static Type *type();
Nicolas Capens442e25b2022-06-22 12:02:52 -04001074 static int element_count() { return 4; }
Nicolas Capens157ba262019-12-10 17:49:14 -05001075};
Nicolas Capensd022e412016-09-26 13:30:14 -04001076
Nicolas Capens157ba262019-12-10 17:49:14 -05001077RValue<UShort4> operator+(RValue<UShort4> lhs, RValue<UShort4> rhs);
1078RValue<UShort4> operator-(RValue<UShort4> lhs, RValue<UShort4> rhs);
1079RValue<UShort4> operator*(RValue<UShort4> lhs, RValue<UShort4> rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -04001080// RValue<UShort4> operator/(RValue<UShort4> lhs, RValue<UShort4> rhs);
1081// RValue<UShort4> operator%(RValue<UShort4> lhs, RValue<UShort4> rhs);
Nicolas Capens157ba262019-12-10 17:49:14 -05001082RValue<UShort4> operator&(RValue<UShort4> lhs, RValue<UShort4> rhs);
1083RValue<UShort4> operator|(RValue<UShort4> lhs, RValue<UShort4> rhs);
1084RValue<UShort4> operator^(RValue<UShort4> lhs, RValue<UShort4> rhs);
1085RValue<UShort4> operator<<(RValue<UShort4> lhs, unsigned char rhs);
1086RValue<UShort4> operator>>(RValue<UShort4> lhs, unsigned char rhs);
Nicolas Capens96d4e092016-11-18 14:22:38 -05001087// RValue<UShort4> operator+=(UShort4 &lhs, RValue<UShort4> rhs);
1088// RValue<UShort4> operator-=(UShort4 &lhs, RValue<UShort4> rhs);
1089// RValue<UShort4> operator*=(UShort4 &lhs, RValue<UShort4> rhs);
1090// RValue<UShort4> operator/=(UShort4 &lhs, RValue<UShort4> rhs);
1091// RValue<UShort4> operator%=(UShort4 &lhs, RValue<UShort4> rhs);
1092// RValue<UShort4> operator&=(UShort4 &lhs, RValue<UShort4> rhs);
1093// RValue<UShort4> operator|=(UShort4 &lhs, RValue<UShort4> rhs);
1094// RValue<UShort4> operator^=(UShort4 &lhs, RValue<UShort4> rhs);
Nicolas Capens157ba262019-12-10 17:49:14 -05001095RValue<UShort4> operator<<=(UShort4 &lhs, unsigned char rhs);
1096RValue<UShort4> operator>>=(UShort4 &lhs, unsigned char rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -04001097// RValue<UShort4> operator+(RValue<UShort4> val);
1098// RValue<UShort4> operator-(RValue<UShort4> val);
Nicolas Capens157ba262019-12-10 17:49:14 -05001099RValue<UShort4> operator~(RValue<UShort4> val);
Nicolas Capens96d4e092016-11-18 14:22:38 -05001100// RValue<UShort4> operator++(UShort4 &val, int); // Post-increment
1101// const UShort4 &operator++(UShort4 &val); // Pre-increment
1102// RValue<UShort4> operator--(UShort4 &val, int); // Post-decrement
1103// const UShort4 &operator--(UShort4 &val); // Pre-decrement
Nicolas Capensd022e412016-09-26 13:30:14 -04001104
Jason Macnak0587e072022-02-11 16:49:02 -08001105RValue<UShort4> Insert(RValue<UShort4> val, RValue<UShort> element, int i);
Nicolas Capens157ba262019-12-10 17:49:14 -05001106RValue<UShort4> Max(RValue<UShort4> x, RValue<UShort4> y);
1107RValue<UShort4> Min(RValue<UShort4> x, RValue<UShort4> y);
1108RValue<UShort4> AddSat(RValue<UShort4> x, RValue<UShort4> y);
1109RValue<UShort4> SubSat(RValue<UShort4> x, RValue<UShort4> y);
1110RValue<UShort4> MulHigh(RValue<UShort4> x, RValue<UShort4> y);
1111RValue<UShort4> Average(RValue<UShort4> x, RValue<UShort4> y);
Nicolas Capensd022e412016-09-26 13:30:14 -04001112
Nicolas Capens157ba262019-12-10 17:49:14 -05001113class Short8 : public LValue<Short8>
1114{
1115public:
1116 Short8() = default;
1117 Short8(short c);
1118 Short8(short c0, short c1, short c2, short c3, short c4, short c5, short c6, short c7);
1119 Short8(RValue<Short8> rhs);
Ben Clayton713b8d32019-12-17 20:37:56 +00001120 // Short8(const Short8 &rhs);
Nicolas Capens157ba262019-12-10 17:49:14 -05001121 Short8(const Reference<Short8> &rhs);
1122 Short8(RValue<Short4> lo, RValue<Short4> hi);
Nicolas Capensd022e412016-09-26 13:30:14 -04001123
Nicolas Capens157ba262019-12-10 17:49:14 -05001124 RValue<Short8> operator=(RValue<Short8> rhs);
1125 RValue<Short8> operator=(const Short8 &rhs);
1126 RValue<Short8> operator=(const Reference<Short8> &rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -04001127
Nicolas Capens519cf222020-05-08 15:27:19 -04001128 static Type *type();
Nicolas Capens442e25b2022-06-22 12:02:52 -04001129 static int element_count() { return 8; }
Nicolas Capens157ba262019-12-10 17:49:14 -05001130};
Nicolas Capensd022e412016-09-26 13:30:14 -04001131
Nicolas Capens157ba262019-12-10 17:49:14 -05001132RValue<Short8> operator+(RValue<Short8> lhs, RValue<Short8> rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -04001133// RValue<Short8> operator-(RValue<Short8> lhs, RValue<Short8> rhs);
1134// RValue<Short8> operator*(RValue<Short8> lhs, RValue<Short8> rhs);
1135// RValue<Short8> operator/(RValue<Short8> lhs, RValue<Short8> rhs);
1136// RValue<Short8> operator%(RValue<Short8> lhs, RValue<Short8> rhs);
Nicolas Capens157ba262019-12-10 17:49:14 -05001137RValue<Short8> operator&(RValue<Short8> lhs, RValue<Short8> rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -04001138// RValue<Short8> operator|(RValue<Short8> lhs, RValue<Short8> rhs);
1139// RValue<Short8> operator^(RValue<Short8> lhs, RValue<Short8> rhs);
Nicolas Capens157ba262019-12-10 17:49:14 -05001140RValue<Short8> operator<<(RValue<Short8> lhs, unsigned char rhs);
1141RValue<Short8> operator>>(RValue<Short8> lhs, unsigned char rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -04001142// RValue<Short8> operator<<(RValue<Short8> lhs, RValue<Short8> rhs);
1143// RValue<Short8> operator>>(RValue<Short8> lhs, RValue<Short8> rhs);
Nicolas Capens96d4e092016-11-18 14:22:38 -05001144// RValue<Short8> operator+=(Short8 &lhs, RValue<Short8> rhs);
1145// RValue<Short8> operator-=(Short8 &lhs, RValue<Short8> rhs);
1146// RValue<Short8> operator*=(Short8 &lhs, RValue<Short8> rhs);
1147// RValue<Short8> operator/=(Short8 &lhs, RValue<Short8> rhs);
1148// RValue<Short8> operator%=(Short8 &lhs, RValue<Short8> rhs);
1149// RValue<Short8> operator&=(Short8 &lhs, RValue<Short8> rhs);
1150// RValue<Short8> operator|=(Short8 &lhs, RValue<Short8> rhs);
1151// RValue<Short8> operator^=(Short8 &lhs, RValue<Short8> rhs);
1152// RValue<Short8> operator<<=(Short8 &lhs, RValue<Short8> rhs);
1153// RValue<Short8> operator>>=(Short8 &lhs, RValue<Short8> rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -04001154// RValue<Short8> operator+(RValue<Short8> val);
1155// RValue<Short8> operator-(RValue<Short8> val);
1156// RValue<Short8> operator~(RValue<Short8> val);
Nicolas Capens96d4e092016-11-18 14:22:38 -05001157// RValue<Short8> operator++(Short8 &val, int); // Post-increment
1158// const Short8 &operator++(Short8 &val); // Pre-increment
1159// RValue<Short8> operator--(Short8 &val, int); // Post-decrement
1160// const Short8 &operator--(Short8 &val); // Pre-decrement
Nicolas Capensd022e412016-09-26 13:30:14 -04001161// RValue<Bool> operator<(RValue<Short8> lhs, RValue<Short8> rhs);
1162// RValue<Bool> operator<=(RValue<Short8> lhs, RValue<Short8> rhs);
1163// RValue<Bool> operator>(RValue<Short8> lhs, RValue<Short8> rhs);
1164// RValue<Bool> operator>=(RValue<Short8> lhs, RValue<Short8> rhs);
1165// RValue<Bool> operator!=(RValue<Short8> lhs, RValue<Short8> rhs);
1166// RValue<Bool> operator==(RValue<Short8> lhs, RValue<Short8> rhs);
1167
Nicolas Capens157ba262019-12-10 17:49:14 -05001168RValue<Short8> MulHigh(RValue<Short8> x, RValue<Short8> y);
1169RValue<Int4> MulAdd(RValue<Short8> x, RValue<Short8> y);
Nicolas Capensd022e412016-09-26 13:30:14 -04001170
Nicolas Capens157ba262019-12-10 17:49:14 -05001171class UShort8 : public LValue<UShort8>
1172{
1173public:
1174 UShort8() = default;
1175 UShort8(unsigned short c);
1176 UShort8(unsigned short c0, unsigned short c1, unsigned short c2, unsigned short c3, unsigned short c4, unsigned short c5, unsigned short c6, unsigned short c7);
1177 UShort8(RValue<UShort8> rhs);
Ben Clayton713b8d32019-12-17 20:37:56 +00001178 // UShort8(const UShort8 &rhs);
Nicolas Capens157ba262019-12-10 17:49:14 -05001179 UShort8(const Reference<UShort8> &rhs);
1180 UShort8(RValue<UShort4> lo, RValue<UShort4> hi);
Nicolas Capensd022e412016-09-26 13:30:14 -04001181
Nicolas Capens157ba262019-12-10 17:49:14 -05001182 RValue<UShort8> operator=(RValue<UShort8> rhs);
1183 RValue<UShort8> operator=(const UShort8 &rhs);
1184 RValue<UShort8> operator=(const Reference<UShort8> &rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -04001185
Nicolas Capens519cf222020-05-08 15:27:19 -04001186 static Type *type();
Nicolas Capens442e25b2022-06-22 12:02:52 -04001187 static int element_count() { return 8; }
Nicolas Capens157ba262019-12-10 17:49:14 -05001188};
Nicolas Capensd022e412016-09-26 13:30:14 -04001189
Nicolas Capens157ba262019-12-10 17:49:14 -05001190RValue<UShort8> operator+(RValue<UShort8> lhs, RValue<UShort8> rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -04001191// RValue<UShort8> operator-(RValue<UShort8> lhs, RValue<UShort8> rhs);
Nicolas Capens157ba262019-12-10 17:49:14 -05001192RValue<UShort8> operator*(RValue<UShort8> lhs, RValue<UShort8> rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -04001193// RValue<UShort8> operator/(RValue<UShort8> lhs, RValue<UShort8> rhs);
1194// RValue<UShort8> operator%(RValue<UShort8> lhs, RValue<UShort8> rhs);
Nicolas Capens157ba262019-12-10 17:49:14 -05001195RValue<UShort8> operator&(RValue<UShort8> lhs, RValue<UShort8> rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -04001196// RValue<UShort8> operator|(RValue<UShort8> lhs, RValue<UShort8> rhs);
1197// RValue<UShort8> operator^(RValue<UShort8> lhs, RValue<UShort8> rhs);
Nicolas Capens157ba262019-12-10 17:49:14 -05001198RValue<UShort8> operator<<(RValue<UShort8> lhs, unsigned char rhs);
1199RValue<UShort8> operator>>(RValue<UShort8> lhs, unsigned char rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -04001200// RValue<UShort8> operator<<(RValue<UShort8> lhs, RValue<UShort8> rhs);
1201// RValue<UShort8> operator>>(RValue<UShort8> lhs, RValue<UShort8> rhs);
Nicolas Capens157ba262019-12-10 17:49:14 -05001202RValue<UShort8> operator+=(UShort8 &lhs, RValue<UShort8> rhs);
Nicolas Capens96d4e092016-11-18 14:22:38 -05001203// RValue<UShort8> operator-=(UShort8 &lhs, RValue<UShort8> rhs);
1204// RValue<UShort8> operator*=(UShort8 &lhs, RValue<UShort8> rhs);
1205// RValue<UShort8> operator/=(UShort8 &lhs, RValue<UShort8> rhs);
1206// RValue<UShort8> operator%=(UShort8 &lhs, RValue<UShort8> rhs);
1207// RValue<UShort8> operator&=(UShort8 &lhs, RValue<UShort8> rhs);
1208// RValue<UShort8> operator|=(UShort8 &lhs, RValue<UShort8> rhs);
1209// RValue<UShort8> operator^=(UShort8 &lhs, RValue<UShort8> rhs);
1210// RValue<UShort8> operator<<=(UShort8 &lhs, RValue<UShort8> rhs);
1211// RValue<UShort8> operator>>=(UShort8 &lhs, RValue<UShort8> rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -04001212// RValue<UShort8> operator+(RValue<UShort8> val);
1213// RValue<UShort8> operator-(RValue<UShort8> val);
Nicolas Capens157ba262019-12-10 17:49:14 -05001214RValue<UShort8> operator~(RValue<UShort8> val);
Nicolas Capens96d4e092016-11-18 14:22:38 -05001215// RValue<UShort8> operator++(UShort8 &val, int); // Post-increment
1216// const UShort8 &operator++(UShort8 &val); // Pre-increment
1217// RValue<UShort8> operator--(UShort8 &val, int); // Post-decrement
1218// const UShort8 &operator--(UShort8 &val); // Pre-decrement
Nicolas Capensd022e412016-09-26 13:30:14 -04001219// RValue<Bool> operator<(RValue<UShort8> lhs, RValue<UShort8> rhs);
1220// RValue<Bool> operator<=(RValue<UShort8> lhs, RValue<UShort8> rhs);
1221// RValue<Bool> operator>(RValue<UShort8> lhs, RValue<UShort8> rhs);
1222// RValue<Bool> operator>=(RValue<UShort8> lhs, RValue<UShort8> rhs);
1223// RValue<Bool> operator!=(RValue<UShort8> lhs, RValue<UShort8> rhs);
1224// RValue<Bool> operator==(RValue<UShort8> lhs, RValue<UShort8> rhs);
1225
Nicolas Capens133b87d2020-01-25 16:26:28 -05001226RValue<UShort8> Swizzle(RValue<UShort8> x, uint32_t select);
Nicolas Capens157ba262019-12-10 17:49:14 -05001227RValue<UShort8> MulHigh(RValue<UShort8> x, RValue<UShort8> y);
Nicolas Capensd022e412016-09-26 13:30:14 -04001228
Nicolas Capens157ba262019-12-10 17:49:14 -05001229class Int : public LValue<Int>
1230{
1231public:
1232 Int(Argument<Int> argument);
Nicolas Capensd022e412016-09-26 13:30:14 -04001233
Nicolas Capens157ba262019-12-10 17:49:14 -05001234 explicit Int(RValue<Byte> cast);
1235 explicit Int(RValue<SByte> cast);
1236 explicit Int(RValue<Short> cast);
1237 explicit Int(RValue<UShort> cast);
1238 explicit Int(RValue<Int2> cast);
1239 explicit Int(RValue<Long> cast);
1240 explicit Int(RValue<Float> cast);
Nicolas Capensd022e412016-09-26 13:30:14 -04001241
Nicolas Capens157ba262019-12-10 17:49:14 -05001242 Int() = default;
1243 Int(int x);
1244 Int(RValue<Int> rhs);
1245 Int(RValue<UInt> rhs);
1246 Int(const Int &rhs);
1247 Int(const UInt &rhs);
1248 Int(const Reference<Int> &rhs);
1249 Int(const Reference<UInt> &rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -04001250
Nicolas Capens6e9eafd2022-02-23 13:09:57 -05001251 template<int T>
1252 Int(const SwizzleMask1<Int4, T> &rhs);
1253
Nicolas Capens157ba262019-12-10 17:49:14 -05001254 RValue<Int> operator=(int rhs);
1255 RValue<Int> operator=(RValue<Int> rhs);
1256 RValue<Int> operator=(RValue<UInt> rhs);
1257 RValue<Int> operator=(const Int &rhs);
1258 RValue<Int> operator=(const UInt &rhs);
1259 RValue<Int> operator=(const Reference<Int> &rhs);
1260 RValue<Int> operator=(const Reference<UInt> &rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -04001261
Nicolas Capens6e9eafd2022-02-23 13:09:57 -05001262 template<int T>
1263 RValue<Int> operator=(const SwizzleMask1<Int4, T> &rhs);
1264
Nicolas Capens519cf222020-05-08 15:27:19 -04001265 static Type *type();
Nicolas Capens157ba262019-12-10 17:49:14 -05001266};
Nicolas Capensd022e412016-09-26 13:30:14 -04001267
Nicolas Capens157ba262019-12-10 17:49:14 -05001268RValue<Int> operator+(RValue<Int> lhs, RValue<Int> rhs);
1269RValue<Int> operator-(RValue<Int> lhs, RValue<Int> rhs);
1270RValue<Int> operator*(RValue<Int> lhs, RValue<Int> rhs);
1271RValue<Int> operator/(RValue<Int> lhs, RValue<Int> rhs);
1272RValue<Int> operator%(RValue<Int> lhs, RValue<Int> rhs);
1273RValue<Int> operator&(RValue<Int> lhs, RValue<Int> rhs);
1274RValue<Int> operator|(RValue<Int> lhs, RValue<Int> rhs);
1275RValue<Int> operator^(RValue<Int> lhs, RValue<Int> rhs);
1276RValue<Int> operator<<(RValue<Int> lhs, RValue<Int> rhs);
1277RValue<Int> operator>>(RValue<Int> lhs, RValue<Int> rhs);
1278RValue<Int> operator+=(Int &lhs, RValue<Int> rhs);
1279RValue<Int> operator-=(Int &lhs, RValue<Int> rhs);
1280RValue<Int> operator*=(Int &lhs, RValue<Int> rhs);
1281RValue<Int> operator/=(Int &lhs, RValue<Int> rhs);
1282RValue<Int> operator%=(Int &lhs, RValue<Int> rhs);
1283RValue<Int> operator&=(Int &lhs, RValue<Int> rhs);
1284RValue<Int> operator|=(Int &lhs, RValue<Int> rhs);
1285RValue<Int> operator^=(Int &lhs, RValue<Int> rhs);
1286RValue<Int> operator<<=(Int &lhs, RValue<Int> rhs);
1287RValue<Int> operator>>=(Int &lhs, RValue<Int> rhs);
1288RValue<Int> operator+(RValue<Int> val);
1289RValue<Int> operator-(RValue<Int> val);
1290RValue<Int> operator~(RValue<Int> val);
Ben Clayton713b8d32019-12-17 20:37:56 +00001291RValue<Int> operator++(Int &val, int); // Post-increment
1292const Int &operator++(Int &val); // Pre-increment
1293RValue<Int> operator--(Int &val, int); // Post-decrement
1294const Int &operator--(Int &val); // Pre-decrement
Nicolas Capens157ba262019-12-10 17:49:14 -05001295RValue<Bool> operator<(RValue<Int> lhs, RValue<Int> rhs);
1296RValue<Bool> operator<=(RValue<Int> lhs, RValue<Int> rhs);
1297RValue<Bool> operator>(RValue<Int> lhs, RValue<Int> rhs);
1298RValue<Bool> operator>=(RValue<Int> lhs, RValue<Int> rhs);
1299RValue<Bool> operator!=(RValue<Int> lhs, RValue<Int> rhs);
1300RValue<Bool> operator==(RValue<Int> lhs, RValue<Int> rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -04001301
Nicolas Capens157ba262019-12-10 17:49:14 -05001302RValue<Int> Max(RValue<Int> x, RValue<Int> y);
1303RValue<Int> Min(RValue<Int> x, RValue<Int> y);
1304RValue<Int> Clamp(RValue<Int> x, RValue<Int> min, RValue<Int> max);
1305RValue<Int> RoundInt(RValue<Float> cast);
Nicolas Capensd022e412016-09-26 13:30:14 -04001306
Nicolas Capens157ba262019-12-10 17:49:14 -05001307class Long : public LValue<Long>
1308{
1309public:
Ben Clayton713b8d32019-12-17 20:37:56 +00001310 // Long(Argument<Long> argument);
Nicolas Capensd022e412016-09-26 13:30:14 -04001311
Ben Clayton713b8d32019-12-17 20:37:56 +00001312 // explicit Long(RValue<Short> cast);
1313 // explicit Long(RValue<UShort> cast);
Nicolas Capens157ba262019-12-10 17:49:14 -05001314 explicit Long(RValue<Int> cast);
1315 explicit Long(RValue<UInt> cast);
Ben Clayton713b8d32019-12-17 20:37:56 +00001316 // explicit Long(RValue<Float> cast);
Nicolas Capensd022e412016-09-26 13:30:14 -04001317
Nicolas Capens157ba262019-12-10 17:49:14 -05001318 Long() = default;
Ben Clayton713b8d32019-12-17 20:37:56 +00001319 // Long(qword x);
Nicolas Capens157ba262019-12-10 17:49:14 -05001320 Long(RValue<Long> rhs);
Ben Clayton713b8d32019-12-17 20:37:56 +00001321 // Long(RValue<ULong> rhs);
1322 // Long(const Long &rhs);
1323 // Long(const Reference<Long> &rhs);
1324 // Long(const ULong &rhs);
1325 // Long(const Reference<ULong> &rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -04001326
Nicolas Capens157ba262019-12-10 17:49:14 -05001327 RValue<Long> operator=(int64_t rhs);
1328 RValue<Long> operator=(RValue<Long> rhs);
Ben Clayton713b8d32019-12-17 20:37:56 +00001329 // RValue<Long> operator=(RValue<ULong> rhs);
Nicolas Capens157ba262019-12-10 17:49:14 -05001330 RValue<Long> operator=(const Long &rhs);
1331 RValue<Long> operator=(const Reference<Long> &rhs);
Ben Clayton713b8d32019-12-17 20:37:56 +00001332 // RValue<Long> operator=(const ULong &rhs);
1333 // RValue<Long> operator=(const Reference<ULong> &rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -04001334
Nicolas Capens519cf222020-05-08 15:27:19 -04001335 static Type *type();
Nicolas Capens157ba262019-12-10 17:49:14 -05001336};
Nicolas Capensd022e412016-09-26 13:30:14 -04001337
Nicolas Capens157ba262019-12-10 17:49:14 -05001338RValue<Long> operator+(RValue<Long> lhs, RValue<Long> rhs);
1339RValue<Long> operator-(RValue<Long> lhs, RValue<Long> rhs);
1340RValue<Long> operator*(RValue<Long> lhs, RValue<Long> rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -04001341// RValue<Long> operator/(RValue<Long> lhs, RValue<Long> rhs);
1342// RValue<Long> operator%(RValue<Long> lhs, RValue<Long> rhs);
1343// RValue<Long> operator&(RValue<Long> lhs, RValue<Long> rhs);
1344// RValue<Long> operator|(RValue<Long> lhs, RValue<Long> rhs);
1345// RValue<Long> operator^(RValue<Long> lhs, RValue<Long> rhs);
1346// RValue<Long> operator<<(RValue<Long> lhs, RValue<Long> rhs);
Nicolas Capens157ba262019-12-10 17:49:14 -05001347RValue<Long> operator>>(RValue<Long> lhs, RValue<Long> rhs);
1348RValue<Long> operator+=(Long &lhs, RValue<Long> rhs);
1349RValue<Long> operator-=(Long &lhs, RValue<Long> rhs);
Nicolas Capens96d4e092016-11-18 14:22:38 -05001350// RValue<Long> operator*=(Long &lhs, RValue<Long> rhs);
1351// RValue<Long> operator/=(Long &lhs, RValue<Long> rhs);
1352// RValue<Long> operator%=(Long &lhs, RValue<Long> rhs);
1353// RValue<Long> operator&=(Long &lhs, RValue<Long> rhs);
1354// RValue<Long> operator|=(Long &lhs, RValue<Long> rhs);
1355// RValue<Long> operator^=(Long &lhs, RValue<Long> rhs);
1356// RValue<Long> operator<<=(Long &lhs, RValue<Long> rhs);
1357// RValue<Long> operator>>=(Long &lhs, RValue<Long> rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -04001358// RValue<Long> operator+(RValue<Long> val);
1359// RValue<Long> operator-(RValue<Long> val);
1360// RValue<Long> operator~(RValue<Long> val);
Nicolas Capens96d4e092016-11-18 14:22:38 -05001361// RValue<Long> operator++(Long &val, int); // Post-increment
1362// const Long &operator++(Long &val); // Pre-increment
1363// RValue<Long> operator--(Long &val, int); // Post-decrement
1364// const Long &operator--(Long &val); // Pre-decrement
Nicolas Capensd022e412016-09-26 13:30:14 -04001365// RValue<Bool> operator<(RValue<Long> lhs, RValue<Long> rhs);
1366// RValue<Bool> operator<=(RValue<Long> lhs, RValue<Long> rhs);
1367// RValue<Bool> operator>(RValue<Long> lhs, RValue<Long> rhs);
1368// RValue<Bool> operator>=(RValue<Long> lhs, RValue<Long> rhs);
1369// RValue<Bool> operator!=(RValue<Long> lhs, RValue<Long> rhs);
1370// RValue<Bool> operator==(RValue<Long> lhs, RValue<Long> rhs);
1371
1372// RValue<Long> RoundLong(RValue<Float> cast);
Ben Clayton713b8d32019-12-17 20:37:56 +00001373RValue<Long> AddAtomic(RValue<Pointer<Long>> x, RValue<Long> y);
Nicolas Capensd022e412016-09-26 13:30:14 -04001374
Nicolas Capens157ba262019-12-10 17:49:14 -05001375class UInt : public LValue<UInt>
1376{
1377public:
1378 UInt(Argument<UInt> argument);
Nicolas Capensd022e412016-09-26 13:30:14 -04001379
Nicolas Capens157ba262019-12-10 17:49:14 -05001380 explicit UInt(RValue<UShort> cast);
1381 explicit UInt(RValue<Long> cast);
1382 explicit UInt(RValue<Float> cast);
Nicolas Capensd022e412016-09-26 13:30:14 -04001383
Nicolas Capens157ba262019-12-10 17:49:14 -05001384 UInt() = default;
1385 UInt(int x);
1386 UInt(unsigned int x);
1387 UInt(RValue<UInt> rhs);
1388 UInt(RValue<Int> rhs);
1389 UInt(const UInt &rhs);
1390 UInt(const Int &rhs);
1391 UInt(const Reference<UInt> &rhs);
1392 UInt(const Reference<Int> &rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -04001393
Nicolas Capens157ba262019-12-10 17:49:14 -05001394 RValue<UInt> operator=(unsigned int rhs);
1395 RValue<UInt> operator=(RValue<UInt> rhs);
1396 RValue<UInt> operator=(RValue<Int> rhs);
1397 RValue<UInt> operator=(const UInt &rhs);
1398 RValue<UInt> operator=(const Int &rhs);
1399 RValue<UInt> operator=(const Reference<UInt> &rhs);
1400 RValue<UInt> operator=(const Reference<Int> &rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -04001401
Nicolas Capens519cf222020-05-08 15:27:19 -04001402 static Type *type();
Nicolas Capens157ba262019-12-10 17:49:14 -05001403};
Nicolas Capensd022e412016-09-26 13:30:14 -04001404
Nicolas Capens157ba262019-12-10 17:49:14 -05001405RValue<UInt> operator+(RValue<UInt> lhs, RValue<UInt> rhs);
1406RValue<UInt> operator-(RValue<UInt> lhs, RValue<UInt> rhs);
1407RValue<UInt> operator*(RValue<UInt> lhs, RValue<UInt> rhs);
1408RValue<UInt> operator/(RValue<UInt> lhs, RValue<UInt> rhs);
1409RValue<UInt> operator%(RValue<UInt> lhs, RValue<UInt> rhs);
1410RValue<UInt> operator&(RValue<UInt> lhs, RValue<UInt> rhs);
1411RValue<UInt> operator|(RValue<UInt> lhs, RValue<UInt> rhs);
1412RValue<UInt> operator^(RValue<UInt> lhs, RValue<UInt> rhs);
1413RValue<UInt> operator<<(RValue<UInt> lhs, RValue<UInt> rhs);
1414RValue<UInt> operator>>(RValue<UInt> lhs, RValue<UInt> rhs);
1415RValue<UInt> operator+=(UInt &lhs, RValue<UInt> rhs);
1416RValue<UInt> operator-=(UInt &lhs, RValue<UInt> rhs);
1417RValue<UInt> operator*=(UInt &lhs, RValue<UInt> rhs);
1418RValue<UInt> operator/=(UInt &lhs, RValue<UInt> rhs);
1419RValue<UInt> operator%=(UInt &lhs, RValue<UInt> rhs);
1420RValue<UInt> operator&=(UInt &lhs, RValue<UInt> rhs);
1421RValue<UInt> operator|=(UInt &lhs, RValue<UInt> rhs);
1422RValue<UInt> operator^=(UInt &lhs, RValue<UInt> rhs);
1423RValue<UInt> operator<<=(UInt &lhs, RValue<UInt> rhs);
1424RValue<UInt> operator>>=(UInt &lhs, RValue<UInt> rhs);
1425RValue<UInt> operator+(RValue<UInt> val);
1426RValue<UInt> operator-(RValue<UInt> val);
1427RValue<UInt> operator~(RValue<UInt> val);
Ben Clayton713b8d32019-12-17 20:37:56 +00001428RValue<UInt> operator++(UInt &val, int); // Post-increment
1429const UInt &operator++(UInt &val); // Pre-increment
1430RValue<UInt> operator--(UInt &val, int); // Post-decrement
1431const UInt &operator--(UInt &val); // Pre-decrement
Nicolas Capens157ba262019-12-10 17:49:14 -05001432RValue<Bool> operator<(RValue<UInt> lhs, RValue<UInt> rhs);
1433RValue<Bool> operator<=(RValue<UInt> lhs, RValue<UInt> rhs);
1434RValue<Bool> operator>(RValue<UInt> lhs, RValue<UInt> rhs);
1435RValue<Bool> operator>=(RValue<UInt> lhs, RValue<UInt> rhs);
1436RValue<Bool> operator!=(RValue<UInt> lhs, RValue<UInt> rhs);
1437RValue<Bool> operator==(RValue<UInt> lhs, RValue<UInt> rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -04001438
Nicolas Capens157ba262019-12-10 17:49:14 -05001439RValue<UInt> Max(RValue<UInt> x, RValue<UInt> y);
1440RValue<UInt> Min(RValue<UInt> x, RValue<UInt> y);
1441RValue<UInt> Clamp(RValue<UInt> x, RValue<UInt> min, RValue<UInt> max);
Chris Forbes17813932019-04-18 11:45:54 -07001442
Nicolas Capens157ba262019-12-10 17:49:14 -05001443RValue<UInt> AddAtomic(RValue<Pointer<UInt>> x, RValue<UInt> y, std::memory_order memoryOrder);
1444RValue<UInt> SubAtomic(RValue<Pointer<UInt>> x, RValue<UInt> y, std::memory_order memoryOrder);
1445RValue<UInt> AndAtomic(RValue<Pointer<UInt>> x, RValue<UInt> y, std::memory_order memoryOrder);
1446RValue<UInt> OrAtomic(RValue<Pointer<UInt>> x, RValue<UInt> y, std::memory_order memoryOrder);
1447RValue<UInt> XorAtomic(RValue<Pointer<UInt>> x, RValue<UInt> y, std::memory_order memoryOrder);
1448RValue<Int> MinAtomic(RValue<Pointer<Int>> x, RValue<Int> y, std::memory_order memoryOrder);
1449RValue<Int> MaxAtomic(RValue<Pointer<Int>> x, RValue<Int> y, std::memory_order memoryOrder);
1450RValue<UInt> MinAtomic(RValue<Pointer<UInt>> x, RValue<UInt> y, std::memory_order memoryOrder);
1451RValue<UInt> MaxAtomic(RValue<Pointer<UInt>> x, RValue<UInt> y, std::memory_order memoryOrder);
1452RValue<UInt> ExchangeAtomic(RValue<Pointer<UInt>> x, RValue<UInt> y, std::memory_order memoryOrder);
1453RValue<UInt> CompareExchangeAtomic(RValue<Pointer<UInt>> x, RValue<UInt> y, RValue<UInt> compare, std::memory_order memoryOrderEqual, std::memory_order memoryOrderUnequal);
Chris Forbes17813932019-04-18 11:45:54 -07001454
Nicolas Capensd022e412016-09-26 13:30:14 -04001455// RValue<UInt> RoundUInt(RValue<Float> cast);
1456
Nicolas Capens157ba262019-12-10 17:49:14 -05001457class Int2 : public LValue<Int2>
1458{
1459public:
Ben Clayton713b8d32019-12-17 20:37:56 +00001460 // explicit Int2(RValue<Int> cast);
Nicolas Capens157ba262019-12-10 17:49:14 -05001461 explicit Int2(RValue<Int4> cast);
Nicolas Capensd022e412016-09-26 13:30:14 -04001462
Nicolas Capens157ba262019-12-10 17:49:14 -05001463 Int2() = default;
1464 Int2(int x, int y);
1465 Int2(RValue<Int2> rhs);
1466 Int2(const Int2 &rhs);
1467 Int2(const Reference<Int2> &rhs);
1468 Int2(RValue<Int> lo, RValue<Int> hi);
Nicolas Capensd022e412016-09-26 13:30:14 -04001469
Nicolas Capens157ba262019-12-10 17:49:14 -05001470 RValue<Int2> operator=(RValue<Int2> rhs);
1471 RValue<Int2> operator=(const Int2 &rhs);
1472 RValue<Int2> operator=(const Reference<Int2> &rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -04001473
Nicolas Capens519cf222020-05-08 15:27:19 -04001474 static Type *type();
Nicolas Capens442e25b2022-06-22 12:02:52 -04001475 static int element_count() { return 2; }
Nicolas Capens157ba262019-12-10 17:49:14 -05001476};
Nicolas Capensd022e412016-09-26 13:30:14 -04001477
Nicolas Capens157ba262019-12-10 17:49:14 -05001478RValue<Int2> operator+(RValue<Int2> lhs, RValue<Int2> rhs);
1479RValue<Int2> operator-(RValue<Int2> lhs, RValue<Int2> rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -04001480// RValue<Int2> operator*(RValue<Int2> lhs, RValue<Int2> rhs);
1481// RValue<Int2> operator/(RValue<Int2> lhs, RValue<Int2> rhs);
1482// RValue<Int2> operator%(RValue<Int2> lhs, RValue<Int2> rhs);
Nicolas Capens157ba262019-12-10 17:49:14 -05001483RValue<Int2> operator&(RValue<Int2> lhs, RValue<Int2> rhs);
1484RValue<Int2> operator|(RValue<Int2> lhs, RValue<Int2> rhs);
1485RValue<Int2> operator^(RValue<Int2> lhs, RValue<Int2> rhs);
1486RValue<Int2> operator<<(RValue<Int2> lhs, unsigned char rhs);
1487RValue<Int2> operator>>(RValue<Int2> lhs, unsigned char rhs);
1488RValue<Int2> operator+=(Int2 &lhs, RValue<Int2> rhs);
1489RValue<Int2> operator-=(Int2 &lhs, RValue<Int2> rhs);
Nicolas Capens96d4e092016-11-18 14:22:38 -05001490// RValue<Int2> operator*=(Int2 &lhs, RValue<Int2> rhs);
1491// RValue<Int2> operator/=(Int2 &lhs, RValue<Int2> rhs);
1492// RValue<Int2> operator%=(Int2 &lhs, RValue<Int2> rhs);
Nicolas Capens157ba262019-12-10 17:49:14 -05001493RValue<Int2> operator&=(Int2 &lhs, RValue<Int2> rhs);
1494RValue<Int2> operator|=(Int2 &lhs, RValue<Int2> rhs);
1495RValue<Int2> operator^=(Int2 &lhs, RValue<Int2> rhs);
1496RValue<Int2> operator<<=(Int2 &lhs, unsigned char rhs);
1497RValue<Int2> operator>>=(Int2 &lhs, unsigned char rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -04001498// RValue<Int2> operator+(RValue<Int2> val);
1499// RValue<Int2> operator-(RValue<Int2> val);
Nicolas Capens157ba262019-12-10 17:49:14 -05001500RValue<Int2> operator~(RValue<Int2> val);
Nicolas Capens96d4e092016-11-18 14:22:38 -05001501// RValue<Int2> operator++(Int2 &val, int); // Post-increment
1502// const Int2 &operator++(Int2 &val); // Pre-increment
1503// RValue<Int2> operator--(Int2 &val, int); // Post-decrement
1504// const Int2 &operator--(Int2 &val); // Pre-decrement
Nicolas Capensd022e412016-09-26 13:30:14 -04001505// RValue<Bool> operator<(RValue<Int2> lhs, RValue<Int2> rhs);
1506// RValue<Bool> operator<=(RValue<Int2> lhs, RValue<Int2> rhs);
1507// RValue<Bool> operator>(RValue<Int2> lhs, RValue<Int2> rhs);
1508// RValue<Bool> operator>=(RValue<Int2> lhs, RValue<Int2> rhs);
1509// RValue<Bool> operator!=(RValue<Int2> lhs, RValue<Int2> rhs);
1510// RValue<Bool> operator==(RValue<Int2> lhs, RValue<Int2> rhs);
1511
1512// RValue<Int2> RoundInt(RValue<Float4> cast);
Nicolas Capens157ba262019-12-10 17:49:14 -05001513RValue<Short4> UnpackLow(RValue<Int2> x, RValue<Int2> y);
1514RValue<Short4> UnpackHigh(RValue<Int2> x, RValue<Int2> y);
1515RValue<Int> Extract(RValue<Int2> val, int i);
1516RValue<Int2> Insert(RValue<Int2> val, RValue<Int> element, int i);
Nicolas Capensd022e412016-09-26 13:30:14 -04001517
Nicolas Capens157ba262019-12-10 17:49:14 -05001518class UInt2 : public LValue<UInt2>
1519{
1520public:
1521 UInt2() = default;
1522 UInt2(unsigned int x, unsigned int y);
1523 UInt2(RValue<UInt2> rhs);
1524 UInt2(const UInt2 &rhs);
1525 UInt2(const Reference<UInt2> &rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -04001526
Nicolas Capens157ba262019-12-10 17:49:14 -05001527 RValue<UInt2> operator=(RValue<UInt2> rhs);
1528 RValue<UInt2> operator=(const UInt2 &rhs);
1529 RValue<UInt2> operator=(const Reference<UInt2> &rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -04001530
Nicolas Capens519cf222020-05-08 15:27:19 -04001531 static Type *type();
Nicolas Capens442e25b2022-06-22 12:02:52 -04001532 static int element_count() { return 2; }
Nicolas Capens157ba262019-12-10 17:49:14 -05001533};
Nicolas Capensd022e412016-09-26 13:30:14 -04001534
Nicolas Capens157ba262019-12-10 17:49:14 -05001535RValue<UInt2> operator+(RValue<UInt2> lhs, RValue<UInt2> rhs);
1536RValue<UInt2> operator-(RValue<UInt2> lhs, RValue<UInt2> rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -04001537// RValue<UInt2> operator*(RValue<UInt2> lhs, RValue<UInt2> rhs);
1538// RValue<UInt2> operator/(RValue<UInt2> lhs, RValue<UInt2> rhs);
1539// RValue<UInt2> operator%(RValue<UInt2> lhs, RValue<UInt2> rhs);
Nicolas Capens157ba262019-12-10 17:49:14 -05001540RValue<UInt2> operator&(RValue<UInt2> lhs, RValue<UInt2> rhs);
1541RValue<UInt2> operator|(RValue<UInt2> lhs, RValue<UInt2> rhs);
1542RValue<UInt2> operator^(RValue<UInt2> lhs, RValue<UInt2> rhs);
1543RValue<UInt2> operator<<(RValue<UInt2> lhs, unsigned char rhs);
1544RValue<UInt2> operator>>(RValue<UInt2> lhs, unsigned char rhs);
1545RValue<UInt2> operator+=(UInt2 &lhs, RValue<UInt2> rhs);
1546RValue<UInt2> operator-=(UInt2 &lhs, RValue<UInt2> rhs);
Nicolas Capens96d4e092016-11-18 14:22:38 -05001547// RValue<UInt2> operator*=(UInt2 &lhs, RValue<UInt2> rhs);
1548// RValue<UInt2> operator/=(UInt2 &lhs, RValue<UInt2> rhs);
1549// RValue<UInt2> operator%=(UInt2 &lhs, RValue<UInt2> rhs);
Nicolas Capens157ba262019-12-10 17:49:14 -05001550RValue<UInt2> operator&=(UInt2 &lhs, RValue<UInt2> rhs);
1551RValue<UInt2> operator|=(UInt2 &lhs, RValue<UInt2> rhs);
1552RValue<UInt2> operator^=(UInt2 &lhs, RValue<UInt2> rhs);
1553RValue<UInt2> operator<<=(UInt2 &lhs, unsigned char rhs);
1554RValue<UInt2> operator>>=(UInt2 &lhs, unsigned char rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -04001555// RValue<UInt2> operator+(RValue<UInt2> val);
1556// RValue<UInt2> operator-(RValue<UInt2> val);
Nicolas Capens157ba262019-12-10 17:49:14 -05001557RValue<UInt2> operator~(RValue<UInt2> val);
Nicolas Capens96d4e092016-11-18 14:22:38 -05001558// RValue<UInt2> operator++(UInt2 &val, int); // Post-increment
1559// const UInt2 &operator++(UInt2 &val); // Pre-increment
1560// RValue<UInt2> operator--(UInt2 &val, int); // Post-decrement
1561// const UInt2 &operator--(UInt2 &val); // Pre-decrement
Nicolas Capensd022e412016-09-26 13:30:14 -04001562// RValue<Bool> operator<(RValue<UInt2> lhs, RValue<UInt2> rhs);
1563// RValue<Bool> operator<=(RValue<UInt2> lhs, RValue<UInt2> rhs);
1564// RValue<Bool> operator>(RValue<UInt2> lhs, RValue<UInt2> rhs);
1565// RValue<Bool> operator>=(RValue<UInt2> lhs, RValue<UInt2> rhs);
1566// RValue<Bool> operator!=(RValue<UInt2> lhs, RValue<UInt2> rhs);
1567// RValue<Bool> operator==(RValue<UInt2> lhs, RValue<UInt2> rhs);
1568
1569// RValue<UInt2> RoundInt(RValue<Float4> cast);
Nicolas Capens157ba262019-12-10 17:49:14 -05001570RValue<UInt> Extract(RValue<UInt2> val, int i);
1571RValue<UInt2> Insert(RValue<UInt2> val, RValue<UInt> element, int i);
Nicolas Capensd022e412016-09-26 13:30:14 -04001572
Nicolas Capens157ba262019-12-10 17:49:14 -05001573class Int4 : public LValue<Int4>, public XYZW<Int4>
1574{
1575public:
1576 explicit Int4(RValue<Byte4> cast);
1577 explicit Int4(RValue<SByte4> cast);
1578 explicit Int4(RValue<Float4> cast);
1579 explicit Int4(RValue<Short4> cast);
1580 explicit Int4(RValue<UShort4> cast);
Nicolas Capenscb986762017-01-20 11:34:37 -05001581
Nicolas Capens157ba262019-12-10 17:49:14 -05001582 Int4();
1583 Int4(int xyzw);
1584 Int4(int x, int yzw);
1585 Int4(int x, int y, int zw);
1586 Int4(int x, int y, int z, int w);
1587 Int4(RValue<Int4> rhs);
1588 Int4(const Int4 &rhs);
1589 Int4(const Reference<Int4> &rhs);
1590 Int4(RValue<UInt4> rhs);
1591 Int4(const UInt4 &rhs);
1592 Int4(const Reference<UInt4> &rhs);
1593 Int4(RValue<Int2> lo, RValue<Int2> hi);
1594 Int4(RValue<Int> rhs);
1595 Int4(const Int &rhs);
1596 Int4(const Reference<Int> &rhs);
Nicolas Capenscb986762017-01-20 11:34:37 -05001597
Nicolas Capens0eb3b102022-06-15 16:49:17 -04001598 template<int T>
1599 Int4(const SwizzleMask1<Int4, T> &rhs);
1600
Nicolas Capens82d425d2022-03-02 16:35:20 -05001601 RValue<Int4> operator=(int broadcast);
Nicolas Capens157ba262019-12-10 17:49:14 -05001602 RValue<Int4> operator=(RValue<Int4> rhs);
1603 RValue<Int4> operator=(const Int4 &rhs);
1604 RValue<Int4> operator=(const Reference<Int4> &rhs);
Nicolas Capenscb986762017-01-20 11:34:37 -05001605
Nicolas Capens519cf222020-05-08 15:27:19 -04001606 static Type *type();
Nicolas Capens442e25b2022-06-22 12:02:52 -04001607 static int element_count() { return 4; }
Nicolas Capenscb986762017-01-20 11:34:37 -05001608
Nicolas Capens157ba262019-12-10 17:49:14 -05001609private:
1610 void constant(int x, int y, int z, int w);
1611};
Nicolas Capenscb986762017-01-20 11:34:37 -05001612
Nicolas Capens157ba262019-12-10 17:49:14 -05001613RValue<Int4> operator+(RValue<Int4> lhs, RValue<Int4> rhs);
1614RValue<Int4> operator-(RValue<Int4> lhs, RValue<Int4> rhs);
1615RValue<Int4> operator*(RValue<Int4> lhs, RValue<Int4> rhs);
1616RValue<Int4> operator/(RValue<Int4> lhs, RValue<Int4> rhs);
1617RValue<Int4> operator%(RValue<Int4> lhs, RValue<Int4> rhs);
1618RValue<Int4> operator&(RValue<Int4> lhs, RValue<Int4> rhs);
1619RValue<Int4> operator|(RValue<Int4> lhs, RValue<Int4> rhs);
1620RValue<Int4> operator^(RValue<Int4> lhs, RValue<Int4> rhs);
1621RValue<Int4> operator<<(RValue<Int4> lhs, unsigned char rhs);
1622RValue<Int4> operator>>(RValue<Int4> lhs, unsigned char rhs);
1623RValue<Int4> operator<<(RValue<Int4> lhs, RValue<Int4> rhs);
1624RValue<Int4> operator>>(RValue<Int4> lhs, RValue<Int4> rhs);
1625RValue<Int4> operator+=(Int4 &lhs, RValue<Int4> rhs);
1626RValue<Int4> operator-=(Int4 &lhs, RValue<Int4> rhs);
1627RValue<Int4> operator*=(Int4 &lhs, RValue<Int4> rhs);
Nicolas Capens96d4e092016-11-18 14:22:38 -05001628// RValue<Int4> operator/=(Int4 &lhs, RValue<Int4> rhs);
1629// RValue<Int4> operator%=(Int4 &lhs, RValue<Int4> rhs);
Nicolas Capens157ba262019-12-10 17:49:14 -05001630RValue<Int4> operator&=(Int4 &lhs, RValue<Int4> rhs);
1631RValue<Int4> operator|=(Int4 &lhs, RValue<Int4> rhs);
1632RValue<Int4> operator^=(Int4 &lhs, RValue<Int4> rhs);
1633RValue<Int4> operator<<=(Int4 &lhs, unsigned char rhs);
1634RValue<Int4> operator>>=(Int4 &lhs, unsigned char rhs);
1635RValue<Int4> operator+(RValue<Int4> val);
1636RValue<Int4> operator-(RValue<Int4> val);
1637RValue<Int4> operator~(RValue<Int4> val);
Nicolas Capens96d4e092016-11-18 14:22:38 -05001638// RValue<Int4> operator++(Int4 &val, int); // Post-increment
1639// const Int4 &operator++(Int4 &val); // Pre-increment
1640// RValue<Int4> operator--(Int4 &val, int); // Post-decrement
1641// const Int4 &operator--(Int4 &val); // Pre-decrement
Nicolas Capensd022e412016-09-26 13:30:14 -04001642// RValue<Bool> operator<(RValue<Int4> lhs, RValue<Int4> rhs);
1643// RValue<Bool> operator<=(RValue<Int4> lhs, RValue<Int4> rhs);
1644// RValue<Bool> operator>(RValue<Int4> lhs, RValue<Int4> rhs);
1645// RValue<Bool> operator>=(RValue<Int4> lhs, RValue<Int4> rhs);
1646// RValue<Bool> operator!=(RValue<Int4> lhs, RValue<Int4> rhs);
1647// RValue<Bool> operator==(RValue<Int4> lhs, RValue<Int4> rhs);
1648
Nicolas Capens157ba262019-12-10 17:49:14 -05001649RValue<Int4> CmpEQ(RValue<Int4> x, RValue<Int4> y);
1650RValue<Int4> CmpLT(RValue<Int4> x, RValue<Int4> y);
1651RValue<Int4> CmpLE(RValue<Int4> x, RValue<Int4> y);
1652RValue<Int4> CmpNEQ(RValue<Int4> x, RValue<Int4> y);
1653RValue<Int4> CmpNLT(RValue<Int4> x, RValue<Int4> y);
1654RValue<Int4> CmpNLE(RValue<Int4> x, RValue<Int4> y);
Ben Clayton713b8d32019-12-17 20:37:56 +00001655inline RValue<Int4> CmpGT(RValue<Int4> x, RValue<Int4> y)
1656{
1657 return CmpNLE(x, y);
1658}
1659inline RValue<Int4> CmpGE(RValue<Int4> x, RValue<Int4> y)
1660{
1661 return CmpNLT(x, y);
1662}
Nicolas Capens44f94692022-06-20 23:15:46 -04001663RValue<Int4> Abs(RValue<Int4> x);
Nicolas Capens157ba262019-12-10 17:49:14 -05001664RValue<Int4> Max(RValue<Int4> x, RValue<Int4> y);
1665RValue<Int4> Min(RValue<Int4> x, RValue<Int4> y);
Nicolas Capenseeb81842021-01-12 17:44:40 -05001666// Convert to nearest integer. If a converted value is outside of the integer
1667// range, the returned result is undefined.
Nicolas Capens157ba262019-12-10 17:49:14 -05001668RValue<Int4> RoundInt(RValue<Float4> cast);
Nicolas Capenseeb81842021-01-12 17:44:40 -05001669// Rounds to the nearest integer, but clamps very large values to an
1670// implementation-dependent range.
1671// Specifically, on x86, values larger than 2147483583.0 are converted to
1672// 2147483583 (0x7FFFFFBF) instead of producing 0x80000000.
1673RValue<Int4> RoundIntClamped(RValue<Float4> cast);
Nicolas Capens157ba262019-12-10 17:49:14 -05001674RValue<Short8> PackSigned(RValue<Int4> x, RValue<Int4> y);
1675RValue<UShort8> PackUnsigned(RValue<Int4> x, RValue<Int4> y);
1676RValue<Int> Extract(RValue<Int4> val, int i);
1677RValue<Int4> Insert(RValue<Int4> val, RValue<Int> element, int i);
1678RValue<Int> SignMask(RValue<Int4> x);
1679RValue<Int4> Swizzle(RValue<Int4> x, uint16_t select);
1680RValue<Int4> Shuffle(RValue<Int4> x, RValue<Int4> y, uint16_t select);
1681RValue<Int4> MulHigh(RValue<Int4> x, RValue<Int4> y);
Nicolas Capensd022e412016-09-26 13:30:14 -04001682
Nicolas Capens157ba262019-12-10 17:49:14 -05001683class UInt4 : public LValue<UInt4>, public XYZW<UInt4>
1684{
1685public:
1686 explicit UInt4(RValue<Float4> cast);
Nicolas Capensd022e412016-09-26 13:30:14 -04001687
Nicolas Capens157ba262019-12-10 17:49:14 -05001688 UInt4();
1689 UInt4(int xyzw);
1690 UInt4(int x, int yzw);
1691 UInt4(int x, int y, int zw);
1692 UInt4(int x, int y, int z, int w);
1693 UInt4(RValue<UInt4> rhs);
1694 UInt4(const UInt4 &rhs);
1695 UInt4(const Reference<UInt4> &rhs);
1696 UInt4(RValue<Int4> rhs);
1697 UInt4(const Int4 &rhs);
1698 UInt4(const Reference<Int4> &rhs);
1699 UInt4(RValue<UInt2> lo, RValue<UInt2> hi);
1700 UInt4(RValue<UInt> rhs);
1701 UInt4(const UInt &rhs);
1702 UInt4(const Reference<UInt> &rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -04001703
Nicolas Capens157ba262019-12-10 17:49:14 -05001704 RValue<UInt4> operator=(RValue<UInt4> rhs);
1705 RValue<UInt4> operator=(const UInt4 &rhs);
1706 RValue<UInt4> operator=(const Reference<UInt4> &rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -04001707
Nicolas Capens519cf222020-05-08 15:27:19 -04001708 static Type *type();
Nicolas Capens442e25b2022-06-22 12:02:52 -04001709 static int element_count() { return 4; }
Nicolas Capensd022e412016-09-26 13:30:14 -04001710
Nicolas Capens157ba262019-12-10 17:49:14 -05001711private:
1712 void constant(int x, int y, int z, int w);
1713};
Nicolas Capensd022e412016-09-26 13:30:14 -04001714
Nicolas Capens157ba262019-12-10 17:49:14 -05001715RValue<UInt4> operator+(RValue<UInt4> lhs, RValue<UInt4> rhs);
1716RValue<UInt4> operator-(RValue<UInt4> lhs, RValue<UInt4> rhs);
1717RValue<UInt4> operator*(RValue<UInt4> lhs, RValue<UInt4> rhs);
1718RValue<UInt4> operator/(RValue<UInt4> lhs, RValue<UInt4> rhs);
1719RValue<UInt4> operator%(RValue<UInt4> lhs, RValue<UInt4> rhs);
1720RValue<UInt4> operator&(RValue<UInt4> lhs, RValue<UInt4> rhs);
1721RValue<UInt4> operator|(RValue<UInt4> lhs, RValue<UInt4> rhs);
1722RValue<UInt4> operator^(RValue<UInt4> lhs, RValue<UInt4> rhs);
1723RValue<UInt4> operator<<(RValue<UInt4> lhs, unsigned char rhs);
1724RValue<UInt4> operator>>(RValue<UInt4> lhs, unsigned char rhs);
1725RValue<UInt4> operator<<(RValue<UInt4> lhs, RValue<UInt4> rhs);
1726RValue<UInt4> operator>>(RValue<UInt4> lhs, RValue<UInt4> rhs);
1727RValue<UInt4> operator+=(UInt4 &lhs, RValue<UInt4> rhs);
1728RValue<UInt4> operator-=(UInt4 &lhs, RValue<UInt4> rhs);
1729RValue<UInt4> operator*=(UInt4 &lhs, RValue<UInt4> rhs);
Nicolas Capens96d4e092016-11-18 14:22:38 -05001730// RValue<UInt4> operator/=(UInt4 &lhs, RValue<UInt4> rhs);
1731// RValue<UInt4> operator%=(UInt4 &lhs, RValue<UInt4> rhs);
Nicolas Capens157ba262019-12-10 17:49:14 -05001732RValue<UInt4> operator&=(UInt4 &lhs, RValue<UInt4> rhs);
1733RValue<UInt4> operator|=(UInt4 &lhs, RValue<UInt4> rhs);
1734RValue<UInt4> operator^=(UInt4 &lhs, RValue<UInt4> rhs);
1735RValue<UInt4> operator<<=(UInt4 &lhs, unsigned char rhs);
1736RValue<UInt4> operator>>=(UInt4 &lhs, unsigned char rhs);
1737RValue<UInt4> operator+(RValue<UInt4> val);
1738RValue<UInt4> operator-(RValue<UInt4> val);
1739RValue<UInt4> operator~(RValue<UInt4> val);
Nicolas Capens96d4e092016-11-18 14:22:38 -05001740// RValue<UInt4> operator++(UInt4 &val, int); // Post-increment
1741// const UInt4 &operator++(UInt4 &val); // Pre-increment
1742// RValue<UInt4> operator--(UInt4 &val, int); // Post-decrement
1743// const UInt4 &operator--(UInt4 &val); // Pre-decrement
Nicolas Capensd022e412016-09-26 13:30:14 -04001744// RValue<Bool> operator<(RValue<UInt4> lhs, RValue<UInt4> rhs);
1745// RValue<Bool> operator<=(RValue<UInt4> lhs, RValue<UInt4> rhs);
1746// RValue<Bool> operator>(RValue<UInt4> lhs, RValue<UInt4> rhs);
1747// RValue<Bool> operator>=(RValue<UInt4> lhs, RValue<UInt4> rhs);
1748// RValue<Bool> operator!=(RValue<UInt4> lhs, RValue<UInt4> rhs);
1749// RValue<Bool> operator==(RValue<UInt4> lhs, RValue<UInt4> rhs);
1750
Nicolas Capens157ba262019-12-10 17:49:14 -05001751RValue<UInt4> CmpEQ(RValue<UInt4> x, RValue<UInt4> y);
1752RValue<UInt4> CmpLT(RValue<UInt4> x, RValue<UInt4> y);
1753RValue<UInt4> CmpLE(RValue<UInt4> x, RValue<UInt4> y);
1754RValue<UInt4> CmpNEQ(RValue<UInt4> x, RValue<UInt4> y);
1755RValue<UInt4> CmpNLT(RValue<UInt4> x, RValue<UInt4> y);
1756RValue<UInt4> CmpNLE(RValue<UInt4> x, RValue<UInt4> y);
Ben Clayton713b8d32019-12-17 20:37:56 +00001757inline RValue<UInt4> CmpGT(RValue<UInt4> x, RValue<UInt4> y)
1758{
1759 return CmpNLE(x, y);
1760}
1761inline RValue<UInt4> CmpGE(RValue<UInt4> x, RValue<UInt4> y)
1762{
1763 return CmpNLT(x, y);
1764}
Nicolas Capens157ba262019-12-10 17:49:14 -05001765RValue<UInt4> Max(RValue<UInt4> x, RValue<UInt4> y);
1766RValue<UInt4> Min(RValue<UInt4> x, RValue<UInt4> y);
1767RValue<UInt4> MulHigh(RValue<UInt4> x, RValue<UInt4> y);
1768RValue<UInt> Extract(RValue<UInt4> val, int i);
1769RValue<UInt4> Insert(RValue<UInt4> val, RValue<UInt> element, int i);
Nicolas Capensd022e412016-09-26 13:30:14 -04001770// RValue<UInt4> RoundInt(RValue<Float4> cast);
Nicolas Capens157ba262019-12-10 17:49:14 -05001771RValue<UInt4> Swizzle(RValue<UInt4> x, uint16_t select);
1772RValue<UInt4> Shuffle(RValue<UInt4> x, RValue<UInt4> y, uint16_t select);
Nicolas Capensd022e412016-09-26 13:30:14 -04001773
Nicolas Capens157ba262019-12-10 17:49:14 -05001774class Half : public LValue<Half>
1775{
1776public:
1777 explicit Half(RValue<Float> cast);
Alexis Hetu734e2572018-12-20 14:00:49 -05001778
Nicolas Capens519cf222020-05-08 15:27:19 -04001779 static Type *type();
Nicolas Capens157ba262019-12-10 17:49:14 -05001780};
Alexis Hetu734e2572018-12-20 14:00:49 -05001781
Nicolas Capens157ba262019-12-10 17:49:14 -05001782class Float : public LValue<Float>
1783{
1784public:
1785 explicit Float(RValue<Int> cast);
1786 explicit Float(RValue<UInt> cast);
1787 explicit Float(RValue<Half> cast);
Nicolas Capensd022e412016-09-26 13:30:14 -04001788
Nicolas Capens157ba262019-12-10 17:49:14 -05001789 Float() = default;
1790 Float(float x);
1791 Float(RValue<Float> rhs);
1792 Float(const Float &rhs);
1793 Float(const Reference<Float> &rhs);
1794 Float(Argument<Float> argument);
Nicolas Capensd022e412016-09-26 13:30:14 -04001795
Nicolas Capens157ba262019-12-10 17:49:14 -05001796 template<int T>
1797 Float(const SwizzleMask1<Float4, T> &rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -04001798
Nicolas Capens0aca3ca2020-09-19 23:59:08 -04001799 RValue<Float> operator=(float rhs);
Nicolas Capens157ba262019-12-10 17:49:14 -05001800 RValue<Float> operator=(RValue<Float> rhs);
1801 RValue<Float> operator=(const Float &rhs);
1802 RValue<Float> operator=(const Reference<Float> &rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -04001803
Nicolas Capens157ba262019-12-10 17:49:14 -05001804 template<int T>
1805 RValue<Float> operator=(const SwizzleMask1<Float4, T> &rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -04001806
Nicolas Capense5720882020-01-13 14:10:04 -05001807 static Float infinity();
1808
Nicolas Capens519cf222020-05-08 15:27:19 -04001809 static Type *type();
Nicolas Capens157ba262019-12-10 17:49:14 -05001810};
Nicolas Capensd022e412016-09-26 13:30:14 -04001811
Nicolas Capens157ba262019-12-10 17:49:14 -05001812RValue<Float> operator+(RValue<Float> lhs, RValue<Float> rhs);
1813RValue<Float> operator-(RValue<Float> lhs, RValue<Float> rhs);
1814RValue<Float> operator*(RValue<Float> lhs, RValue<Float> rhs);
1815RValue<Float> operator/(RValue<Float> lhs, RValue<Float> rhs);
1816RValue<Float> operator+=(Float &lhs, RValue<Float> rhs);
1817RValue<Float> operator-=(Float &lhs, RValue<Float> rhs);
1818RValue<Float> operator*=(Float &lhs, RValue<Float> rhs);
1819RValue<Float> operator/=(Float &lhs, RValue<Float> rhs);
1820RValue<Float> operator+(RValue<Float> val);
1821RValue<Float> operator-(RValue<Float> val);
1822RValue<Bool> operator<(RValue<Float> lhs, RValue<Float> rhs);
1823RValue<Bool> operator<=(RValue<Float> lhs, RValue<Float> rhs);
1824RValue<Bool> operator>(RValue<Float> lhs, RValue<Float> rhs);
1825RValue<Bool> operator>=(RValue<Float> lhs, RValue<Float> rhs);
1826RValue<Bool> operator!=(RValue<Float> lhs, RValue<Float> rhs);
1827RValue<Bool> operator==(RValue<Float> lhs, RValue<Float> rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -04001828
Nicolas Capens157ba262019-12-10 17:49:14 -05001829RValue<Float> Abs(RValue<Float> x);
1830RValue<Float> Max(RValue<Float> x, RValue<Float> y);
1831RValue<Float> Min(RValue<Float> x, RValue<Float> y);
Nicolas Capensd04f3f52022-02-05 01:19:14 -05001832RValue<Float> Rcp(RValue<Float> x, bool relaxedPrecision, bool exactAtPow2 = false);
1833RValue<Float> RcpSqrt(RValue<Float> x, bool relaxedPrecision);
Nicolas Capens157ba262019-12-10 17:49:14 -05001834RValue<Float> Sqrt(RValue<Float> x);
Nicolas Capens88ac3672019-08-01 13:22:34 -04001835
1836// RValue<Int4> IsInf(RValue<Float> x);
1837// RValue<Int4> IsNan(RValue<Float> x);
Nicolas Capens157ba262019-12-10 17:49:14 -05001838RValue<Float> Round(RValue<Float> x);
1839RValue<Float> Trunc(RValue<Float> x);
1840RValue<Float> Frac(RValue<Float> x);
1841RValue<Float> Floor(RValue<Float> x);
1842RValue<Float> Ceil(RValue<Float> x);
Nicolas Capens88ac3672019-08-01 13:22:34 -04001843
Nicolas Capens157ba262019-12-10 17:49:14 -05001844// Trigonometric functions
1845// TODO: Currently unimplemented for Subzero.
Nicolas Capens88ac3672019-08-01 13:22:34 -04001846// RValue<Float> Sin(RValue<Float> x);
1847// RValue<Float> Cos(RValue<Float> x);
1848// RValue<Float> Tan(RValue<Float> x);
1849// RValue<Float> Asin(RValue<Float> x);
1850// RValue<Float> Acos(RValue<Float> x);
1851// RValue<Float> Atan(RValue<Float> x);
1852// RValue<Float> Sinh(RValue<Float> x);
1853// RValue<Float> Cosh(RValue<Float> x);
1854// RValue<Float> Tanh(RValue<Float> x);
1855// RValue<Float> Asinh(RValue<Float> x);
1856// RValue<Float> Acosh(RValue<Float> x);
1857// RValue<Float> Atanh(RValue<Float> x);
1858// RValue<Float> Atan2(RValue<Float> x, RValue<Float> y);
1859
Nicolas Capens157ba262019-12-10 17:49:14 -05001860// Exponential functions
1861// TODO: Currently unimplemented for Subzero.
Nicolas Capens88ac3672019-08-01 13:22:34 -04001862// RValue<Float> Pow(RValue<Float> x, RValue<Float> y);
1863// RValue<Float> Exp(RValue<Float> x);
1864// RValue<Float> Log(RValue<Float> x);
Nicolas Capens157ba262019-12-10 17:49:14 -05001865RValue<Float> Exp2(RValue<Float> x);
1866RValue<Float> Log2(RValue<Float> x);
Nicolas Capensd022e412016-09-26 13:30:14 -04001867
Nicolas Capens157ba262019-12-10 17:49:14 -05001868class Float2 : public LValue<Float2>
1869{
1870public:
Ben Clayton713b8d32019-12-17 20:37:56 +00001871 // explicit Float2(RValue<Byte2> cast);
1872 // explicit Float2(RValue<Short2> cast);
1873 // explicit Float2(RValue<UShort2> cast);
1874 // explicit Float2(RValue<Int2> cast);
1875 // explicit Float2(RValue<UInt2> cast);
Nicolas Capens157ba262019-12-10 17:49:14 -05001876 explicit Float2(RValue<Float4> cast);
Nicolas Capensd022e412016-09-26 13:30:14 -04001877
Nicolas Capens157ba262019-12-10 17:49:14 -05001878 Float2() = default;
Ben Clayton713b8d32019-12-17 20:37:56 +00001879 // Float2(float x, float y);
1880 // Float2(RValue<Float2> rhs);
1881 // Float2(const Float2 &rhs);
1882 // Float2(const Reference<Float2> &rhs);
1883 // Float2(RValue<Float> rhs);
1884 // Float2(const Float &rhs);
1885 // Float2(const Reference<Float> &rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -04001886
Ben Clayton713b8d32019-12-17 20:37:56 +00001887 // template<int T>
1888 // Float2(const SwizzleMask1<T> &rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -04001889
Nicolas Capens82d425d2022-03-02 16:35:20 -05001890 // RValue<Float2> operator=(float broadcast);
Ben Clayton713b8d32019-12-17 20:37:56 +00001891 // RValue<Float2> operator=(RValue<Float2> rhs);
1892 // RValue<Float2> operator=(const Float2 &rhs);
1893 // RValue<Float2> operator=(const Reference<Float2> &rhs);
1894 // RValue<Float2> operator=(RValue<Float> rhs);
1895 // RValue<Float2> operator=(const Float &rhs);
1896 // RValue<Float2> operator=(const Reference<Float> &rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -04001897
Ben Clayton713b8d32019-12-17 20:37:56 +00001898 // template<int T>
1899 // RValue<Float2> operator=(const SwizzleMask1<T> &rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -04001900
Nicolas Capens519cf222020-05-08 15:27:19 -04001901 static Type *type();
Nicolas Capens442e25b2022-06-22 12:02:52 -04001902 static int element_count() { return 2; }
Nicolas Capens157ba262019-12-10 17:49:14 -05001903};
Nicolas Capensd022e412016-09-26 13:30:14 -04001904
1905// RValue<Float2> operator+(RValue<Float2> lhs, RValue<Float2> rhs);
1906// RValue<Float2> operator-(RValue<Float2> lhs, RValue<Float2> rhs);
1907// RValue<Float2> operator*(RValue<Float2> lhs, RValue<Float2> rhs);
1908// RValue<Float2> operator/(RValue<Float2> lhs, RValue<Float2> rhs);
1909// RValue<Float2> operator%(RValue<Float2> lhs, RValue<Float2> rhs);
Nicolas Capens96d4e092016-11-18 14:22:38 -05001910// RValue<Float2> operator+=(Float2 &lhs, RValue<Float2> rhs);
1911// RValue<Float2> operator-=(Float2 &lhs, RValue<Float2> rhs);
1912// RValue<Float2> operator*=(Float2 &lhs, RValue<Float2> rhs);
1913// RValue<Float2> operator/=(Float2 &lhs, RValue<Float2> rhs);
1914// RValue<Float2> operator%=(Float2 &lhs, RValue<Float2> rhs);
Nicolas Capensd022e412016-09-26 13:30:14 -04001915// RValue<Float2> operator+(RValue<Float2> val);
1916// RValue<Float2> operator-(RValue<Float2> val);
1917
1918// RValue<Float2> Abs(RValue<Float2> x);
1919// RValue<Float2> Max(RValue<Float2> x, RValue<Float2> y);
1920// RValue<Float2> Min(RValue<Float2> x, RValue<Float2> y);
Ben Clayton8701dc42019-12-05 21:27:03 +00001921// RValue<Float2> Swizzle(RValue<Float2> x, uint16_t select);
1922// RValue<Float2> Mask(Float2 &lhs, RValue<Float2> rhs, uint16_t select);
Nicolas Capensd022e412016-09-26 13:30:14 -04001923
Nicolas Capens157ba262019-12-10 17:49:14 -05001924class Float4 : public LValue<Float4>, public XYZW<Float4>
1925{
1926public:
1927 explicit Float4(RValue<Byte4> cast);
1928 explicit Float4(RValue<SByte4> cast);
1929 explicit Float4(RValue<Short4> cast);
1930 explicit Float4(RValue<UShort4> cast);
1931 explicit Float4(RValue<Int4> cast);
1932 explicit Float4(RValue<UInt4> cast);
1933
1934 Float4();
1935 Float4(float xyzw);
1936 Float4(float x, float yzw);
1937 Float4(float x, float y, float zw);
1938 Float4(float x, float y, float z, float w);
1939 Float4(RValue<Float4> rhs);
1940 Float4(const Float4 &rhs);
1941 Float4(const Reference<Float4> &rhs);
1942 Float4(RValue<Float> rhs);
1943 Float4(const Float &rhs);
1944 Float4(const Reference<Float> &rhs);
1945
1946 template<int T>
1947 Float4(const SwizzleMask1<Float4, T> &rhs);
1948 template<int T>
1949 Float4(const Swizzle4<Float4, T> &rhs);
1950 template<int X, int Y>
1951 Float4(const Swizzle2<Float4, X> &x, const Swizzle2<Float4, Y> &y);
1952 template<int X, int Y>
1953 Float4(const SwizzleMask2<Float4, X> &x, const Swizzle2<Float4, Y> &y);
1954 template<int X, int Y>
1955 Float4(const Swizzle2<Float4, X> &x, const SwizzleMask2<Float4, Y> &y);
1956 template<int X, int Y>
1957 Float4(const SwizzleMask2<Float4, X> &x, const SwizzleMask2<Float4, Y> &y);
Nicolas Capens419b7d72020-10-02 16:15:34 -04001958 Float4(RValue<Float2> lo, RValue<Float2> hi);
Nicolas Capens157ba262019-12-10 17:49:14 -05001959
Nicolas Capens82d425d2022-03-02 16:35:20 -05001960 RValue<Float4> operator=(float broadcast);
Nicolas Capens157ba262019-12-10 17:49:14 -05001961 RValue<Float4> operator=(RValue<Float4> rhs);
1962 RValue<Float4> operator=(const Float4 &rhs);
1963 RValue<Float4> operator=(const Reference<Float4> &rhs);
1964 RValue<Float4> operator=(RValue<Float> rhs);
1965 RValue<Float4> operator=(const Float &rhs);
1966 RValue<Float4> operator=(const Reference<Float> &rhs);
1967
1968 template<int T>
1969 RValue<Float4> operator=(const SwizzleMask1<Float4, T> &rhs);
1970 template<int T>
1971 RValue<Float4> operator=(const Swizzle4<Float4, T> &rhs);
1972
Nicolas Capense5720882020-01-13 14:10:04 -05001973 static Float4 infinity();
1974
Nicolas Capens519cf222020-05-08 15:27:19 -04001975 static Type *type();
Nicolas Capens442e25b2022-06-22 12:02:52 -04001976 static int element_count() { return 4; }
Ben Clayton713b8d32019-12-17 20:37:56 +00001977
Nicolas Capens157ba262019-12-10 17:49:14 -05001978private:
1979 void constant(float x, float y, float z, float w);
Nicolas Capens157ba262019-12-10 17:49:14 -05001980};
1981
1982RValue<Float4> operator+(RValue<Float4> lhs, RValue<Float4> rhs);
1983RValue<Float4> operator-(RValue<Float4> lhs, RValue<Float4> rhs);
1984RValue<Float4> operator*(RValue<Float4> lhs, RValue<Float4> rhs);
1985RValue<Float4> operator/(RValue<Float4> lhs, RValue<Float4> rhs);
1986RValue<Float4> operator%(RValue<Float4> lhs, RValue<Float4> rhs);
1987RValue<Float4> operator+=(Float4 &lhs, RValue<Float4> rhs);
1988RValue<Float4> operator-=(Float4 &lhs, RValue<Float4> rhs);
1989RValue<Float4> operator*=(Float4 &lhs, RValue<Float4> rhs);
1990RValue<Float4> operator/=(Float4 &lhs, RValue<Float4> rhs);
1991RValue<Float4> operator%=(Float4 &lhs, RValue<Float4> rhs);
1992RValue<Float4> operator+(RValue<Float4> val);
1993RValue<Float4> operator-(RValue<Float4> val);
1994
Nicolas Capensbc74bc22022-01-26 10:47:00 -05001995// Computes `x * y + z`, which may be fused into one operation to produce a higher-precision result.
1996RValue<Float4> MulAdd(RValue<Float4> x, RValue<Float4> y, RValue<Float4> z);
Nicolas Capens75d79f22022-01-31 17:46:26 -05001997// Computes a fused `x * y + z` operation. Caps::fmaIsFast indicates whether it emits an FMA instruction.
1998RValue<Float4> FMA(RValue<Float4> x, RValue<Float4> y, RValue<Float4> z);
Nicolas Capensbc74bc22022-01-26 10:47:00 -05001999
Nicolas Capens157ba262019-12-10 17:49:14 -05002000RValue<Float4> Abs(RValue<Float4> x);
2001RValue<Float4> Max(RValue<Float4> x, RValue<Float4> y);
2002RValue<Float4> Min(RValue<Float4> x, RValue<Float4> y);
Antonio Maioranod1561872020-12-14 14:03:53 -05002003
Nicolas Capensd04f3f52022-02-05 01:19:14 -05002004RValue<Float4> Rcp(RValue<Float4> x, bool relaxedPrecision, bool exactAtPow2 = false);
2005RValue<Float4> RcpSqrt(RValue<Float4> x, bool relaxedPrecision);
Nicolas Capens157ba262019-12-10 17:49:14 -05002006RValue<Float4> Sqrt(RValue<Float4> x);
2007RValue<Float4> Insert(RValue<Float4> val, RValue<Float> element, int i);
2008RValue<Float> Extract(RValue<Float4> x, int i);
2009RValue<Float4> Swizzle(RValue<Float4> x, uint16_t select);
2010RValue<Float4> Shuffle(RValue<Float4> x, RValue<Float4> y, uint16_t select);
2011RValue<Float4> ShuffleLowHigh(RValue<Float4> x, RValue<Float4> y, uint16_t imm);
2012RValue<Float4> UnpackLow(RValue<Float4> x, RValue<Float4> y);
2013RValue<Float4> UnpackHigh(RValue<Float4> x, RValue<Float4> y);
2014RValue<Float4> Mask(Float4 &lhs, RValue<Float4> rhs, uint16_t select);
2015RValue<Int> SignMask(RValue<Float4> x);
2016
2017// Ordered comparison functions
2018RValue<Int4> CmpEQ(RValue<Float4> x, RValue<Float4> y);
2019RValue<Int4> CmpLT(RValue<Float4> x, RValue<Float4> y);
2020RValue<Int4> CmpLE(RValue<Float4> x, RValue<Float4> y);
2021RValue<Int4> CmpNEQ(RValue<Float4> x, RValue<Float4> y);
2022RValue<Int4> CmpNLT(RValue<Float4> x, RValue<Float4> y);
2023RValue<Int4> CmpNLE(RValue<Float4> x, RValue<Float4> y);
Ben Clayton713b8d32019-12-17 20:37:56 +00002024inline RValue<Int4> CmpGT(RValue<Float4> x, RValue<Float4> y)
2025{
2026 return CmpNLE(x, y);
2027}
2028inline RValue<Int4> CmpGE(RValue<Float4> x, RValue<Float4> y)
2029{
2030 return CmpNLT(x, y);
2031}
Nicolas Capens157ba262019-12-10 17:49:14 -05002032
2033// Unordered comparison functions
2034RValue<Int4> CmpUEQ(RValue<Float4> x, RValue<Float4> y);
2035RValue<Int4> CmpULT(RValue<Float4> x, RValue<Float4> y);
2036RValue<Int4> CmpULE(RValue<Float4> x, RValue<Float4> y);
2037RValue<Int4> CmpUNEQ(RValue<Float4> x, RValue<Float4> y);
2038RValue<Int4> CmpUNLT(RValue<Float4> x, RValue<Float4> y);
2039RValue<Int4> CmpUNLE(RValue<Float4> x, RValue<Float4> y);
Ben Clayton713b8d32019-12-17 20:37:56 +00002040inline RValue<Int4> CmpUGT(RValue<Float4> x, RValue<Float4> y)
2041{
2042 return CmpUNLE(x, y);
2043}
2044inline RValue<Int4> CmpUGE(RValue<Float4> x, RValue<Float4> y)
2045{
2046 return CmpUNLT(x, y);
2047}
Nicolas Capens157ba262019-12-10 17:49:14 -05002048
2049RValue<Int4> IsInf(RValue<Float4> x);
2050RValue<Int4> IsNan(RValue<Float4> x);
2051RValue<Float4> Round(RValue<Float4> x);
2052RValue<Float4> Trunc(RValue<Float4> x);
2053RValue<Float4> Frac(RValue<Float4> x);
2054RValue<Float4> Floor(RValue<Float4> x);
2055RValue<Float4> Ceil(RValue<Float4> x);
2056
2057// Trigonometric functions
Nicolas Capens157ba262019-12-10 17:49:14 -05002058RValue<Float4> Sin(RValue<Float4> x);
2059RValue<Float4> Cos(RValue<Float4> x);
2060RValue<Float4> Tan(RValue<Float4> x);
Nicolas Capensd04f3f52022-02-05 01:19:14 -05002061RValue<Float4> Asin(RValue<Float4> x);
2062RValue<Float4> Acos(RValue<Float4> x);
Nicolas Capens157ba262019-12-10 17:49:14 -05002063RValue<Float4> Atan(RValue<Float4> x);
2064RValue<Float4> Sinh(RValue<Float4> x);
2065RValue<Float4> Cosh(RValue<Float4> x);
2066RValue<Float4> Tanh(RValue<Float4> x);
2067RValue<Float4> Asinh(RValue<Float4> x);
2068RValue<Float4> Acosh(RValue<Float4> x);
2069RValue<Float4> Atanh(RValue<Float4> x);
2070RValue<Float4> Atan2(RValue<Float4> x, RValue<Float4> y);
2071
2072// Exponential functions
Nicolas Capens157ba262019-12-10 17:49:14 -05002073RValue<Float4> Pow(RValue<Float4> x, RValue<Float4> y);
2074RValue<Float4> Exp(RValue<Float4> x);
2075RValue<Float4> Log(RValue<Float4> x);
2076RValue<Float4> Exp2(RValue<Float4> x);
2077RValue<Float4> Log2(RValue<Float4> x);
2078
Nicolas Capens442e25b2022-06-22 12:02:52 -04002079// Call a unary C function on each element of a vector type.
2080template<typename Func, typename T>
2081inline RValue<T> ScalarizeCall(Func func, const RValue<T> &x)
2082{
2083 T result;
2084 for(int i = 0; i < T::element_count(); i++)
2085 {
2086 result = Insert(result, Call(func, Extract(x, i)), i);
2087 }
2088
2089 return result;
2090}
2091
2092// Call a binary C function on each element of a vector type.
2093template<typename Func, typename T>
2094inline RValue<T> ScalarizeCall(Func func, const RValue<T> &x, const RValue<T> &y)
2095{
2096 T result;
2097 for(int i = 0; i < T::element_count(); i++)
2098 {
2099 result = Insert(result, Call(func, Extract(x, i), Extract(y, i)), i);
2100 }
2101
2102 return result;
2103}
2104
2105// Call a ternary C function on each element of a vector type.
2106template<typename Func, typename T>
2107inline RValue<T> ScalarizeCall(Func func, const RValue<T> &x, const RValue<T> &y, const RValue<T> &z)
2108{
2109 T result;
2110 for(int i = 0; i < T::element_count(); i++)
2111 {
2112 result = Insert(result, Call(func, Extract(x, i), Extract(y, i), Extract(z, i)), i);
2113 }
2114
2115 return result;
2116}
2117
2118// Invoke a unary lambda expression on each element of a vector type.
2119template<typename Func, typename T>
2120inline RValue<T> Scalarize(Func func, const RValue<T> &x)
2121{
2122 T result;
2123 for(int i = 0; i < T::element_count(); i++)
2124 {
2125 result = Insert(result, func(Extract(x, i)), i);
2126 }
2127
2128 return result;
2129}
2130
2131// Invoke a binary lambda expression on each element of a vector type.
2132template<typename Func, typename T>
2133inline RValue<T> Scalarize(Func func, const RValue<T> &x, const RValue<T> &y)
2134{
2135 T result;
2136 for(int i = 0; i < T::element_count(); i++)
2137 {
2138 result = Insert(result, func(Extract(x, i), Extract(y, i)), i);
2139 }
2140
2141 return result;
2142}
2143
2144// Invoke a ternary lambda expression on each element of a vector type.
2145template<typename Func, typename T>
2146inline RValue<T> Scalarize(Func func, const RValue<T> &x, const RValue<T> &y, const RValue<T> &z)
2147{
2148 T result;
2149 for(int i = 0; i < T::element_count(); i++)
2150 {
2151 result = Insert(result, func(Extract(x, i), Extract(y, i), Extract(z, i)), i);
2152 }
2153
2154 return result;
2155}
2156
Nicolas Capens157ba262019-12-10 17:49:14 -05002157// Bit Manipulation functions.
2158// TODO: Currently unimplemented for Subzero.
2159
2160// Count leading zeros.
2161// Returns 32 when: !isZeroUndef && x == 0.
2162// Returns an undefined value when: isZeroUndef && x == 0.
2163RValue<UInt> Ctlz(RValue<UInt> x, bool isZeroUndef);
2164RValue<UInt4> Ctlz(RValue<UInt4> x, bool isZeroUndef);
2165
2166// Count trailing zeros.
2167// Returns 32 when: !isZeroUndef && x == 0.
2168// Returns an undefined value when: isZeroUndef && x == 0.
2169RValue<UInt> Cttz(RValue<UInt> x, bool isZeroUndef);
2170RValue<UInt4> Cttz(RValue<UInt4> x, bool isZeroUndef);
2171
2172template<class T>
2173class Pointer : public LValue<Pointer<T>>
2174{
2175public:
2176 template<class S>
Ben Clayton713b8d32019-12-17 20:37:56 +00002177 Pointer(RValue<Pointer<S>> pointerS, int alignment = 1)
2178 : alignment(alignment)
Nicolas Capensa25311a2017-01-16 17:19:00 -05002179 {
Nicolas Capensb6e8c3f2020-05-01 23:28:37 -04002180 Value *pointerT = Nucleus::createBitCast(pointerS.value(), Nucleus::getPointerType(T::type()));
Nicolas Capens3b655b62020-05-13 15:23:54 -04002181 this->storeValue(pointerT);
Ben Clayton204a4102019-07-31 13:17:47 +01002182 }
2183
Nicolas Capens157ba262019-12-10 17:49:14 -05002184 template<class S>
Ben Clayton713b8d32019-12-17 20:37:56 +00002185 Pointer(const Pointer<S> &pointer, int alignment = 1)
2186 : alignment(alignment)
Nicolas Capens86509d92019-03-21 13:23:50 -04002187 {
Nicolas Capens157ba262019-12-10 17:49:14 -05002188 Value *pointerS = pointer.loadValue();
Nicolas Capens519cf222020-05-08 15:27:19 -04002189 Value *pointerT = Nucleus::createBitCast(pointerS, Nucleus::getPointerType(T::type()));
Nicolas Capens3b655b62020-05-13 15:23:54 -04002190 this->storeValue(pointerT);
Nicolas Capens86509d92019-03-21 13:23:50 -04002191 }
2192
Nicolas Capens157ba262019-12-10 17:49:14 -05002193 Pointer(Argument<Pointer<T>> argument);
Ben Clayton97035bd2019-04-16 11:35:38 -04002194
Nicolas Capens157ba262019-12-10 17:49:14 -05002195 Pointer();
2196 Pointer(RValue<Pointer<T>> rhs);
2197 Pointer(const Pointer<T> &rhs);
2198 Pointer(const Reference<Pointer<T>> &rhs);
2199 Pointer(std::nullptr_t);
Ben Claytoncb2ebc92019-06-20 00:18:03 +01002200
Nicolas Capens157ba262019-12-10 17:49:14 -05002201 RValue<Pointer<T>> operator=(RValue<Pointer<T>> rhs);
2202 RValue<Pointer<T>> operator=(const Pointer<T> &rhs);
2203 RValue<Pointer<T>> operator=(const Reference<Pointer<T>> &rhs);
2204 RValue<Pointer<T>> operator=(std::nullptr_t);
Ben Clayton0fc611f2019-04-18 11:23:27 -04002205
Nicolas Capensef72cb42021-11-12 13:57:14 -05002206 Reference<T> operator*() const;
2207 Reference<T> operator[](int index) const;
2208 Reference<T> operator[](unsigned int index) const;
2209 Reference<T> operator[](RValue<Int> index) const;
2210 Reference<T> operator[](RValue<UInt> index) const;
Nicolas Capens86509d92019-03-21 13:23:50 -04002211
Nicolas Capens519cf222020-05-08 15:27:19 -04002212 static Type *type();
Ben Clayton97035bd2019-04-16 11:35:38 -04002213
Nicolas Capens157ba262019-12-10 17:49:14 -05002214private:
2215 const int alignment;
2216};
Ben Clayton97035bd2019-04-16 11:35:38 -04002217
Nicolas Capens157ba262019-12-10 17:49:14 -05002218RValue<Pointer<Byte>> operator+(RValue<Pointer<Byte>> lhs, int offset);
2219RValue<Pointer<Byte>> operator+(RValue<Pointer<Byte>> lhs, RValue<Int> offset);
2220RValue<Pointer<Byte>> operator+(RValue<Pointer<Byte>> lhs, RValue<UInt> offset);
2221RValue<Pointer<Byte>> operator+=(Pointer<Byte> &lhs, int offset);
2222RValue<Pointer<Byte>> operator+=(Pointer<Byte> &lhs, RValue<Int> offset);
2223RValue<Pointer<Byte>> operator+=(Pointer<Byte> &lhs, RValue<UInt> offset);
Ben Clayton97035bd2019-04-16 11:35:38 -04002224
Nicolas Capens157ba262019-12-10 17:49:14 -05002225RValue<Pointer<Byte>> operator-(RValue<Pointer<Byte>> lhs, int offset);
2226RValue<Pointer<Byte>> operator-(RValue<Pointer<Byte>> lhs, RValue<Int> offset);
2227RValue<Pointer<Byte>> operator-(RValue<Pointer<Byte>> lhs, RValue<UInt> offset);
2228RValue<Pointer<Byte>> operator-=(Pointer<Byte> &lhs, int offset);
2229RValue<Pointer<Byte>> operator-=(Pointer<Byte> &lhs, RValue<Int> offset);
2230RValue<Pointer<Byte>> operator-=(Pointer<Byte> &lhs, RValue<UInt> offset);
Nicolas Capensd022e412016-09-26 13:30:14 -04002231
Ben Clayton713b8d32019-12-17 20:37:56 +00002232template<typename T>
Nicolas Capens157ba262019-12-10 17:49:14 -05002233RValue<Bool> operator==(const Pointer<T> &lhs, const Pointer<T> &rhs)
2234{
Nicolas Capens8603b122021-02-12 16:14:18 -05002235 return RValue<Bool>(Nucleus::createICmpEQ(lhs.loadValue(), rhs.loadValue()));
2236}
2237
2238template<typename T>
2239RValue<Bool> operator!=(const Pointer<T> &lhs, const Pointer<T> &rhs)
2240{
2241 return RValue<Bool>(Nucleus::createICmpNE(lhs.loadValue(), rhs.loadValue()));
Nicolas Capens157ba262019-12-10 17:49:14 -05002242}
Ben Clayton0953d9b2019-06-25 20:57:06 +01002243
Nicolas Capens157ba262019-12-10 17:49:14 -05002244template<typename T>
2245RValue<T> Load(RValue<Pointer<T>> pointer, unsigned int alignment, bool atomic, std::memory_order memoryOrder)
2246{
Nicolas Capensb6e8c3f2020-05-01 23:28:37 -04002247 return RValue<T>(Nucleus::createLoad(pointer.value(), T::type(), false, alignment, atomic, memoryOrder));
Nicolas Capens157ba262019-12-10 17:49:14 -05002248}
2249
2250template<typename T>
2251RValue<T> Load(Pointer<T> pointer, unsigned int alignment, bool atomic, std::memory_order memoryOrder)
2252{
2253 return Load(RValue<Pointer<T>>(pointer), alignment, atomic, memoryOrder);
2254}
2255
2256// TODO: Use SIMD to template these.
Nicolas Capense4b77942021-08-03 17:09:41 -04002257// TODO(b/155867273): These can be undeprecated if implemented for Subzero.
2258[[deprecated]] RValue<Float4> MaskedLoad(RValue<Pointer<Float4>> base, RValue<Int4> mask, unsigned int alignment, bool zeroMaskedLanes = false);
2259[[deprecated]] RValue<Int4> MaskedLoad(RValue<Pointer<Int4>> base, RValue<Int4> mask, unsigned int alignment, bool zeroMaskedLanes = false);
2260[[deprecated]] void MaskedStore(RValue<Pointer<Float4>> base, RValue<Float4> val, RValue<Int4> mask, unsigned int alignment);
2261[[deprecated]] void MaskedStore(RValue<Pointer<Int4>> base, RValue<Int4> val, RValue<Int4> mask, unsigned int alignment);
Nicolas Capens157ba262019-12-10 17:49:14 -05002262
Nicolas Capens157ba262019-12-10 17:49:14 -05002263template<typename T>
2264void Store(RValue<T> value, RValue<Pointer<T>> pointer, unsigned int alignment, bool atomic, std::memory_order memoryOrder)
2265{
Nicolas Capensb6e8c3f2020-05-01 23:28:37 -04002266 Nucleus::createStore(value.value(), pointer.value(), T::type(), false, alignment, atomic, memoryOrder);
Nicolas Capens157ba262019-12-10 17:49:14 -05002267}
2268
2269template<typename T>
2270void Store(RValue<T> value, Pointer<T> pointer, unsigned int alignment, bool atomic, std::memory_order memoryOrder)
2271{
2272 Store(value, RValue<Pointer<T>>(pointer), alignment, atomic, memoryOrder);
2273}
2274
2275template<typename T>
2276void Store(T value, Pointer<T> pointer, unsigned int alignment, bool atomic, std::memory_order memoryOrder)
2277{
2278 Store(RValue<T>(value), RValue<Pointer<T>>(pointer), alignment, atomic, memoryOrder);
2279}
2280
Sean Risser19e30802022-06-01 10:58:10 -04002281enum class OutOfBoundsBehavior
2282{
2283 Nullify, // Loads become zero, stores are elided.
2284 RobustBufferAccess, // As defined by the Vulkan spec (in short: access anywhere within bounds, or zeroing).
2285 UndefinedValue, // Only for load operations. Not secure. No program termination.
2286 UndefinedBehavior, // Program may terminate.
2287};
2288
Sean Risser19e30802022-06-01 10:58:10 -04002289RValue<Bool> AnyTrue(const RValue<Int4> &bools);
2290RValue<Bool> AnyFalse(const RValue<Int4> &bools);
2291RValue<Bool> AllTrue(const RValue<Int4> &bools);
2292RValue<Bool> AllFalse(const RValue<Int4> &bools);
2293
2294RValue<Bool> Divergent(const RValue<Int4> &ints);
2295RValue<Bool> Divergent(const RValue<Float4> &floats);
2296RValue<Bool> Uniform(const RValue<Int4> &ints);
2297RValue<Bool> Uniform(const RValue<Float4> &floats);
2298
Nicolas Capens157ba262019-12-10 17:49:14 -05002299// Fence adds a memory barrier that enforces ordering constraints on memory
2300// operations. memoryOrder can only be one of:
2301// std::memory_order_acquire, std::memory_order_release,
2302// std::memory_order_acq_rel, or std::memory_order_seq_cst.
2303void Fence(std::memory_order memoryOrder);
2304
2305template<class T, int S = 1>
2306class Array : public LValue<T>
2307{
2308public:
2309 Array(int size = S);
2310
2311 Reference<T> operator[](int index);
2312 Reference<T> operator[](unsigned int index);
2313 Reference<T> operator[](RValue<Int> index);
2314 Reference<T> operator[](RValue<UInt> index);
2315
2316 // self() returns the this pointer to this Array object.
2317 // This function exists because operator&() is overloaded by LValue<T>.
Ben Clayton713b8d32019-12-17 20:37:56 +00002318 inline Array *self() { return this; }
Nicolas Capens157ba262019-12-10 17:49:14 -05002319};
Nicolas Capensd022e412016-09-26 13:30:14 -04002320
Nicolas Capens96d4e092016-11-18 14:22:38 -05002321// RValue<Array<T>> operator++(Array<T> &val, int); // Post-increment
2322// const Array<T> &operator++(Array<T> &val); // Pre-increment
2323// RValue<Array<T>> operator--(Array<T> &val, int); // Post-decrement
2324// const Array<T> &operator--(Array<T> &val); // Pre-decrement
Nicolas Capensd022e412016-09-26 13:30:14 -04002325
Nicolas Capens157ba262019-12-10 17:49:14 -05002326void branch(RValue<Bool> cmp, BasicBlock *bodyBB, BasicBlock *endBB);
Nicolas Capensd022e412016-09-26 13:30:14 -04002327
Nicolas Capens157ba262019-12-10 17:49:14 -05002328// ValueOf returns a rr::Value* for the given C-type, RValue<T>, LValue<T>
2329// or Reference<T>.
Ben Clayton713b8d32019-12-17 20:37:56 +00002330template<typename T>
2331inline Value *ValueOf(const T &v)
Nicolas Capens157ba262019-12-10 17:49:14 -05002332{
2333 return ReactorType<T>::cast(v).loadValue();
Nicolas Capensd022e412016-09-26 13:30:14 -04002334}
2335
Nicolas Capens157ba262019-12-10 17:49:14 -05002336void Return();
2337
2338template<class T>
2339void Return(const T &ret)
Nicolas Capensd022e412016-09-26 13:30:14 -04002340{
Ben Clayton713b8d32019-12-17 20:37:56 +00002341 static_assert(CanBeUsedAsReturn<ReactorTypeT<T>>::value, "Unsupported type for Return()");
Nicolas Capens157ba262019-12-10 17:49:14 -05002342 Nucleus::createRet(ValueOf<T>(ret));
2343 // Place any unreachable instructions in an unreferenced block.
2344 Nucleus::setInsertBlock(Nucleus::createBasicBlock());
2345}
2346
2347// Generic template, leave undefined!
2348template<typename FunctionType>
2349class Function;
2350
2351// Specialized for function types
2352template<typename Return, typename... Arguments>
2353class Function<Return(Arguments...)>
2354{
2355 // Static assert that the function signature is valid.
2356 static_assert(sizeof(AssertFunctionSignatureIsValid<Return(Arguments...)>) >= 0, "Invalid function signature");
2357
2358public:
2359 Function();
2360
Nicolas Capens157ba262019-12-10 17:49:14 -05002361 template<int index>
2362 Argument<typename std::tuple_element<index, std::tuple<Arguments...>>::type> Arg() const
Nicolas Capens22479eb2016-09-28 22:34:26 -04002363 {
Nicolas Capens157ba262019-12-10 17:49:14 -05002364 Value *arg = Nucleus::getArgument(index);
2365 return Argument<typename std::tuple_element<index, std::tuple<Arguments...>>::type>(arg);
2366 }
2367
2368 std::shared_ptr<Routine> operator()(const char *name, ...);
Nicolas Capens157ba262019-12-10 17:49:14 -05002369
2370protected:
Nicolas Capens3d26cfc2021-01-22 16:51:00 -05002371 std::unique_ptr<Nucleus> core;
Ben Clayton713b8d32019-12-17 20:37:56 +00002372 std::vector<Type *> arguments;
Nicolas Capens157ba262019-12-10 17:49:14 -05002373};
2374
2375template<typename Return>
2376class Function<Return()> : public Function<Return(Void)>
2377{
2378};
2379
2380// FunctionT accepts a C-style function type template argument, allowing it to return a type-safe RoutineT wrapper
2381template<typename FunctionType>
2382class FunctionT;
2383
2384template<typename Return, typename... Arguments>
2385class FunctionT<Return(Arguments...)> : public Function<CToReactorT<Return>(CToReactorT<Arguments>...)>
2386{
2387public:
2388 // Type of base class
2389 using BaseType = Function<CToReactorT<Return>(CToReactorT<Arguments>...)>;
2390
2391 // Function type, e.g. void(int,float)
2392 using CFunctionType = Return(Arguments...);
2393
2394 // Reactor function type, e.g. Void(Int, Float)
2395 using ReactorFunctionType = CToReactorT<Return>(CToReactorT<Arguments>...);
2396
2397 // Returned RoutineT type
2398 using RoutineType = RoutineT<CFunctionType>;
2399
2400 // Hide base implementations of operator()
2401
Antonio Maiorano9ff80662020-11-27 13:51:35 -05002402 template<typename... VarArgs>
2403 RoutineType operator()(const char *name, VarArgs... varArgs)
Nicolas Capens157ba262019-12-10 17:49:14 -05002404 {
Antonio Maiorano9ff80662020-11-27 13:51:35 -05002405 return RoutineType(BaseType::operator()(name, std::forward<VarArgs>(varArgs)...));
Nicolas Capens157ba262019-12-10 17:49:14 -05002406 }
Nicolas Capens157ba262019-12-10 17:49:14 -05002407};
2408
2409RValue<Long> Ticks();
2410
2411} // namespace rr
2412
2413/* Inline implementations */
2414
2415namespace rr {
2416
2417template<class T>
Antonio Maiorano84c61e12020-12-02 12:06:05 -05002418LValue<T>::LValue(int arraySize)
Antonio Maioranobae138d2020-12-02 14:25:10 -05002419 : Variable(T::type(), arraySize)
Nicolas Capens157ba262019-12-10 17:49:14 -05002420{
Ben Claytonac07ed82019-03-26 14:17:41 +00002421#ifdef ENABLE_RR_DEBUG_INFO
Nicolas Capens157ba262019-12-10 17:49:14 -05002422 materialize();
Ben Clayton713b8d32019-12-17 20:37:56 +00002423#endif // ENABLE_RR_DEBUG_INFO
Nicolas Capens157ba262019-12-10 17:49:14 -05002424}
Nicolas Capens22479eb2016-09-28 22:34:26 -04002425
Nicolas Capens157ba262019-12-10 17:49:14 -05002426template<class T>
2427RValue<Pointer<T>> LValue<T>::operator&()
2428{
Nicolas Capens3b655b62020-05-13 15:23:54 -04002429 return RValue<Pointer<T>>(this->getBaseAddress());
Nicolas Capens157ba262019-12-10 17:49:14 -05002430}
Nicolas Capensd022e412016-09-26 13:30:14 -04002431
Nicolas Capens157ba262019-12-10 17:49:14 -05002432template<class T>
Ben Clayton713b8d32019-12-17 20:37:56 +00002433Reference<T>::Reference(Value *pointer, int alignment)
Nicolas Capens990e4b22021-06-17 22:26:36 -04002434 : address(pointer)
2435 , alignment(alignment)
Nicolas Capens157ba262019-12-10 17:49:14 -05002436{
Nicolas Capens157ba262019-12-10 17:49:14 -05002437}
Nicolas Capensd022e412016-09-26 13:30:14 -04002438
Nicolas Capens157ba262019-12-10 17:49:14 -05002439template<class T>
2440RValue<T> Reference<T>::operator=(RValue<T> rhs) const
2441{
Nicolas Capensb6e8c3f2020-05-01 23:28:37 -04002442 Nucleus::createStore(rhs.value(), address, T::type(), false, alignment);
Nicolas Capensd022e412016-09-26 13:30:14 -04002443
Nicolas Capens157ba262019-12-10 17:49:14 -05002444 return rhs;
2445}
2446
2447template<class T>
2448RValue<T> Reference<T>::operator=(const Reference<T> &ref) const
2449{
Nicolas Capens519cf222020-05-08 15:27:19 -04002450 Value *tmp = Nucleus::createLoad(ref.address, T::type(), false, ref.alignment);
2451 Nucleus::createStore(tmp, address, T::type(), false, alignment);
Nicolas Capens157ba262019-12-10 17:49:14 -05002452
2453 return RValue<T>(tmp);
2454}
2455
2456template<class T>
2457RValue<T> Reference<T>::operator+=(RValue<T> rhs) const
2458{
2459 return *this = *this + rhs;
2460}
2461
2462template<class T>
2463Value *Reference<T>::loadValue() const
2464{
Nicolas Capens519cf222020-05-08 15:27:19 -04002465 return Nucleus::createLoad(address, T::type(), false, alignment);
Nicolas Capens157ba262019-12-10 17:49:14 -05002466}
2467
2468template<class T>
Nicolas Capensb4e4f112020-05-01 23:06:41 -04002469RValue<T> Reference<T>::load() const
2470{
2471 return RValue<T>(loadValue());
2472}
2473
2474template<class T>
Nicolas Capens157ba262019-12-10 17:49:14 -05002475int Reference<T>::getAlignment() const
2476{
2477 return alignment;
2478}
Nicolas Capensd022e412016-09-26 13:30:14 -04002479
Nicolas Capens157ba262019-12-10 17:49:14 -05002480template<class T>
Ben Clayton713b8d32019-12-17 20:37:56 +00002481RValue<T>::RValue(const RValue<T> &rvalue)
Nicolas Capensb6e8c3f2020-05-01 23:28:37 -04002482 : val(rvalue.val)
Nicolas Capens157ba262019-12-10 17:49:14 -05002483{
Nicolas Capensb6e8c3f2020-05-01 23:28:37 -04002484 RR_DEBUG_INFO_EMIT_VAR(val);
Nicolas Capens157ba262019-12-10 17:49:14 -05002485}
Ben Claytonac07ed82019-03-26 14:17:41 +00002486
Nicolas Capens157ba262019-12-10 17:49:14 -05002487template<class T>
Nicolas Capensb6e8c3f2020-05-01 23:28:37 -04002488RValue<T>::RValue(Value *value)
2489 : val(value)
Nicolas Capens157ba262019-12-10 17:49:14 -05002490{
Nicolas Capensb6e8c3f2020-05-01 23:28:37 -04002491 assert(Nucleus::createBitCast(value, T::type()) == value); // Run-time type should match T, so bitcast is no-op.
2492 RR_DEBUG_INFO_EMIT_VAR(val);
Nicolas Capens157ba262019-12-10 17:49:14 -05002493}
Nicolas Capensd022e412016-09-26 13:30:14 -04002494
Nicolas Capens157ba262019-12-10 17:49:14 -05002495template<class T>
2496RValue<T>::RValue(const T &lvalue)
Nicolas Capensb6e8c3f2020-05-01 23:28:37 -04002497 : val(lvalue.loadValue())
Nicolas Capens157ba262019-12-10 17:49:14 -05002498{
Nicolas Capensb6e8c3f2020-05-01 23:28:37 -04002499 RR_DEBUG_INFO_EMIT_VAR(val);
Nicolas Capens157ba262019-12-10 17:49:14 -05002500}
Nicolas Capensd022e412016-09-26 13:30:14 -04002501
Nicolas Capens3edbc042022-03-01 16:02:18 -05002502template<>
2503inline RValue<Bool>::RValue(bool b)
2504 : val(Nucleus::createConstantBool(b))
Nicolas Capens157ba262019-12-10 17:49:14 -05002505{
Nicolas Capensb6e8c3f2020-05-01 23:28:37 -04002506 RR_DEBUG_INFO_EMIT_VAR(val);
Nicolas Capens157ba262019-12-10 17:49:14 -05002507}
Ben Clayton35e90e22019-03-15 10:06:06 +00002508
Nicolas Capens157ba262019-12-10 17:49:14 -05002509template<class T>
Nicolas Capens3edbc042022-03-01 16:02:18 -05002510RValue<T>::RValue(typename IntLiteral<T>::Type i)
Nicolas Capensb6e8c3f2020-05-01 23:28:37 -04002511 : val(Nucleus::createConstantInt(i))
Nicolas Capens157ba262019-12-10 17:49:14 -05002512{
Nicolas Capensb6e8c3f2020-05-01 23:28:37 -04002513 RR_DEBUG_INFO_EMIT_VAR(val);
Nicolas Capens157ba262019-12-10 17:49:14 -05002514}
Nicolas Capensd022e412016-09-26 13:30:14 -04002515
Nicolas Capens3edbc042022-03-01 16:02:18 -05002516template<>
2517inline RValue<Long>::RValue(int64_t i)
Daniele Vettorelf908b182022-02-09 17:38:08 +00002518 : val(Nucleus::createConstantLong(i))
2519{
2520 RR_DEBUG_INFO_EMIT_VAR(val);
2521}
2522
Nicolas Capens3edbc042022-03-01 16:02:18 -05002523template<>
2524inline RValue<Float>::RValue(float f)
Nicolas Capensb6e8c3f2020-05-01 23:28:37 -04002525 : val(Nucleus::createConstantFloat(f))
Nicolas Capens157ba262019-12-10 17:49:14 -05002526{
Nicolas Capensb6e8c3f2020-05-01 23:28:37 -04002527 RR_DEBUG_INFO_EMIT_VAR(val);
Nicolas Capens157ba262019-12-10 17:49:14 -05002528}
Nicolas Capensd022e412016-09-26 13:30:14 -04002529
Nicolas Capens44f94692022-06-20 23:15:46 -04002530inline Value *broadcast(int i, Type *type)
Nicolas Capens3edbc042022-03-01 16:02:18 -05002531{
Nicolas Capens4e7d3102022-06-21 01:42:18 -04002532 std::vector<int64_t> constantVector = { i };
Nicolas Capens44f94692022-06-20 23:15:46 -04002533 return Nucleus::createConstantVector(constantVector, type);
Nicolas Capens3edbc042022-03-01 16:02:18 -05002534}
2535
2536template<>
2537inline RValue<Int4>::RValue(int i)
Nicolas Capens44f94692022-06-20 23:15:46 -04002538 : val(broadcast(i, Int4::type()))
Nicolas Capens3edbc042022-03-01 16:02:18 -05002539{
2540 RR_DEBUG_INFO_EMIT_VAR(val);
2541}
2542
Nicolas Capens3edbc042022-03-01 16:02:18 -05002543template<>
2544inline RValue<UInt4>::RValue(unsigned int i)
Nicolas Capens44f94692022-06-20 23:15:46 -04002545 : val(broadcast(int(i), UInt4::type()))
Nicolas Capens3edbc042022-03-01 16:02:18 -05002546{
2547 RR_DEBUG_INFO_EMIT_VAR(val);
2548}
2549
Nicolas Capens44f94692022-06-20 23:15:46 -04002550inline Value *broadcast(float f, Type *type)
Nicolas Capens3edbc042022-03-01 16:02:18 -05002551{
2552 // See Float(float) constructor for the rationale behind this assert.
2553 assert(std::isfinite(f));
2554
Nicolas Capens4e7d3102022-06-21 01:42:18 -04002555 std::vector<double> constantVector = { f };
Nicolas Capens44f94692022-06-20 23:15:46 -04002556 return Nucleus::createConstantVector(constantVector, type);
Nicolas Capens3edbc042022-03-01 16:02:18 -05002557}
2558
2559template<>
2560inline RValue<Float4>::RValue(float f)
Nicolas Capens44f94692022-06-20 23:15:46 -04002561 : val(broadcast(f, Float4::type()))
Nicolas Capens3edbc042022-03-01 16:02:18 -05002562{
2563 RR_DEBUG_INFO_EMIT_VAR(val);
2564}
2565
Nicolas Capens157ba262019-12-10 17:49:14 -05002566template<class T>
2567RValue<T>::RValue(const Reference<T> &ref)
Nicolas Capensb6e8c3f2020-05-01 23:28:37 -04002568 : val(ref.loadValue())
Nicolas Capens157ba262019-12-10 17:49:14 -05002569{
Nicolas Capensb6e8c3f2020-05-01 23:28:37 -04002570 RR_DEBUG_INFO_EMIT_VAR(val);
Nicolas Capens157ba262019-12-10 17:49:14 -05002571}
Nicolas Capensd022e412016-09-26 13:30:14 -04002572
Nicolas Capens157ba262019-12-10 17:49:14 -05002573template<class Vector4, int T>
2574Swizzle2<Vector4, T>::operator RValue<Vector4>() const
2575{
2576 RR_DEBUG_INFO_UPDATE_LOC();
2577 Value *vector = parent->loadValue();
Nicolas Capensd022e412016-09-26 13:30:14 -04002578
Nicolas Capens157ba262019-12-10 17:49:14 -05002579 return Swizzle(RValue<Vector4>(vector), T);
2580}
Nicolas Capensd022e412016-09-26 13:30:14 -04002581
Nicolas Capens157ba262019-12-10 17:49:14 -05002582template<class Vector4, int T>
2583Swizzle4<Vector4, T>::operator RValue<Vector4>() const
2584{
2585 RR_DEBUG_INFO_UPDATE_LOC();
2586 Value *vector = parent->loadValue();
Nicolas Capensd022e412016-09-26 13:30:14 -04002587
Nicolas Capens157ba262019-12-10 17:49:14 -05002588 return Swizzle(RValue<Vector4>(vector), T);
2589}
Nicolas Capensd022e412016-09-26 13:30:14 -04002590
Nicolas Capens157ba262019-12-10 17:49:14 -05002591template<class Vector4, int T>
2592SwizzleMask4<Vector4, T>::operator RValue<Vector4>() const
2593{
2594 RR_DEBUG_INFO_UPDATE_LOC();
2595 Value *vector = parent->loadValue();
Nicolas Capensd022e412016-09-26 13:30:14 -04002596
Nicolas Capens157ba262019-12-10 17:49:14 -05002597 return Swizzle(RValue<Vector4>(vector), T);
2598}
Nicolas Capensd022e412016-09-26 13:30:14 -04002599
Nicolas Capens157ba262019-12-10 17:49:14 -05002600template<class Vector4, int T>
2601RValue<Vector4> SwizzleMask4<Vector4, T>::operator=(RValue<Vector4> rhs)
2602{
2603 RR_DEBUG_INFO_UPDATE_LOC();
2604 return Mask(*parent, rhs, T);
2605}
Nicolas Capensd022e412016-09-26 13:30:14 -04002606
Nicolas Capens157ba262019-12-10 17:49:14 -05002607template<class Vector4, int T>
2608RValue<Vector4> SwizzleMask4<Vector4, T>::operator=(RValue<typename Scalar<Vector4>::Type> rhs)
2609{
2610 RR_DEBUG_INFO_UPDATE_LOC();
2611 return Mask(*parent, Vector4(rhs), T);
2612}
Nicolas Capensd022e412016-09-26 13:30:14 -04002613
Nicolas Capens157ba262019-12-10 17:49:14 -05002614template<class Vector4, int T>
Ben Clayton713b8d32019-12-17 20:37:56 +00002615SwizzleMask1<Vector4, T>::operator RValue<typename Scalar<Vector4>::Type>() const // FIXME: Call a non-template function
Nicolas Capens157ba262019-12-10 17:49:14 -05002616{
2617 RR_DEBUG_INFO_UPDATE_LOC();
2618 return Extract(*parent, T & 0x3);
2619}
Nicolas Capensd022e412016-09-26 13:30:14 -04002620
Nicolas Capens157ba262019-12-10 17:49:14 -05002621template<class Vector4, int T>
2622SwizzleMask1<Vector4, T>::operator RValue<Vector4>() const
2623{
2624 RR_DEBUG_INFO_UPDATE_LOC();
2625 Value *vector = parent->loadValue();
Nicolas Capensd022e412016-09-26 13:30:14 -04002626
Nicolas Capens157ba262019-12-10 17:49:14 -05002627 return Swizzle(RValue<Vector4>(vector), T);
2628}
Nicolas Capensd022e412016-09-26 13:30:14 -04002629
Nicolas Capens157ba262019-12-10 17:49:14 -05002630template<class Vector4, int T>
2631RValue<Vector4> SwizzleMask1<Vector4, T>::operator=(float x)
2632{
2633 RR_DEBUG_INFO_UPDATE_LOC();
2634 return *parent = Insert(*parent, Float(x), T & 0x3);
2635}
Nicolas Capensd022e412016-09-26 13:30:14 -04002636
Nicolas Capens157ba262019-12-10 17:49:14 -05002637template<class Vector4, int T>
2638RValue<Vector4> SwizzleMask1<Vector4, T>::operator=(RValue<Vector4> rhs)
2639{
2640 RR_DEBUG_INFO_UPDATE_LOC();
2641 return Mask(*parent, Float4(rhs), T);
2642}
Nicolas Capensd022e412016-09-26 13:30:14 -04002643
Nicolas Capens157ba262019-12-10 17:49:14 -05002644template<class Vector4, int T>
Ben Clayton713b8d32019-12-17 20:37:56 +00002645RValue<Vector4> SwizzleMask1<Vector4, T>::operator=(RValue<typename Scalar<Vector4>::Type> rhs) // FIXME: Call a non-template function
Nicolas Capens157ba262019-12-10 17:49:14 -05002646{
2647 RR_DEBUG_INFO_UPDATE_LOC();
2648 return *parent = Insert(*parent, rhs, T & 0x3);
2649}
Nicolas Capensd022e412016-09-26 13:30:14 -04002650
Nicolas Capens157ba262019-12-10 17:49:14 -05002651template<class Vector4, int T>
2652SwizzleMask2<Vector4, T>::operator RValue<Vector4>() const
2653{
2654 RR_DEBUG_INFO_UPDATE_LOC();
2655 Value *vector = parent->loadValue();
Nicolas Capensd022e412016-09-26 13:30:14 -04002656
Nicolas Capens157ba262019-12-10 17:49:14 -05002657 return Swizzle(RValue<Float4>(vector), T);
2658}
Nicolas Capensd022e412016-09-26 13:30:14 -04002659
Nicolas Capens157ba262019-12-10 17:49:14 -05002660template<class Vector4, int T>
2661RValue<Vector4> SwizzleMask2<Vector4, T>::operator=(RValue<Vector4> rhs)
2662{
2663 RR_DEBUG_INFO_UPDATE_LOC();
2664 return Mask(*parent, Float4(rhs), T);
2665}
Nicolas Capensd022e412016-09-26 13:30:14 -04002666
Nicolas Capens157ba262019-12-10 17:49:14 -05002667template<int T>
Nicolas Capens6e9eafd2022-02-23 13:09:57 -05002668Int::Int(const SwizzleMask1<Int4, T> &rhs)
2669{
2670 *this = rhs.operator RValue<Int>();
2671}
2672
2673template<int T>
2674RValue<Int> Int::operator=(const SwizzleMask1<Int4, T> &rhs)
2675{
2676 return *this = rhs.operator RValue<Int>();
2677}
2678
2679template<int T>
Nicolas Capens157ba262019-12-10 17:49:14 -05002680Float::Float(const SwizzleMask1<Float4, T> &rhs)
2681{
2682 *this = rhs.operator RValue<Float>();
2683}
Nicolas Capensd022e412016-09-26 13:30:14 -04002684
Nicolas Capens157ba262019-12-10 17:49:14 -05002685template<int T>
2686RValue<Float> Float::operator=(const SwizzleMask1<Float4, T> &rhs)
2687{
2688 return *this = rhs.operator RValue<Float>();
2689}
Nicolas Capensd022e412016-09-26 13:30:14 -04002690
Nicolas Capens157ba262019-12-10 17:49:14 -05002691template<int T>
Nicolas Capens0eb3b102022-06-15 16:49:17 -04002692Int4::Int4(const SwizzleMask1<Int4, T> &rhs)
2693 : XYZW(this)
2694{
2695 *this = rhs.operator RValue<Int4>();
2696}
2697
2698template<int T>
Ben Clayton713b8d32019-12-17 20:37:56 +00002699Float4::Float4(const SwizzleMask1<Float4, T> &rhs)
2700 : XYZW(this)
Nicolas Capens157ba262019-12-10 17:49:14 -05002701{
2702 *this = rhs.operator RValue<Float4>();
2703}
Nicolas Capensd022e412016-09-26 13:30:14 -04002704
Nicolas Capens157ba262019-12-10 17:49:14 -05002705template<int T>
Ben Clayton713b8d32019-12-17 20:37:56 +00002706Float4::Float4(const Swizzle4<Float4, T> &rhs)
2707 : XYZW(this)
Nicolas Capens157ba262019-12-10 17:49:14 -05002708{
2709 *this = rhs.operator RValue<Float4>();
2710}
Nicolas Capensd022e412016-09-26 13:30:14 -04002711
Nicolas Capens157ba262019-12-10 17:49:14 -05002712template<int X, int Y>
Ben Clayton713b8d32019-12-17 20:37:56 +00002713Float4::Float4(const Swizzle2<Float4, X> &x, const Swizzle2<Float4, Y> &y)
2714 : XYZW(this)
Nicolas Capens157ba262019-12-10 17:49:14 -05002715{
2716 RR_DEBUG_INFO_UPDATE_LOC();
2717 *this = ShuffleLowHigh(*x.parent, *y.parent, (uint16_t(X) & 0xFF00u) | (uint16_t(Y >> 8) & 0x00FFu));
2718}
Nicolas Capensd022e412016-09-26 13:30:14 -04002719
Nicolas Capens157ba262019-12-10 17:49:14 -05002720template<int X, int Y>
Ben Clayton713b8d32019-12-17 20:37:56 +00002721Float4::Float4(const SwizzleMask2<Float4, X> &x, const Swizzle2<Float4, Y> &y)
2722 : XYZW(this)
Nicolas Capens157ba262019-12-10 17:49:14 -05002723{
2724 RR_DEBUG_INFO_UPDATE_LOC();
2725 *this = ShuffleLowHigh(*x.parent, *y.parent, (uint16_t(X) & 0xFF00u) | (uint16_t(Y >> 8) & 0x00FFu));
2726}
Nicolas Capensd022e412016-09-26 13:30:14 -04002727
Nicolas Capens157ba262019-12-10 17:49:14 -05002728template<int X, int Y>
Ben Clayton713b8d32019-12-17 20:37:56 +00002729Float4::Float4(const Swizzle2<Float4, X> &x, const SwizzleMask2<Float4, Y> &y)
2730 : XYZW(this)
Nicolas Capens157ba262019-12-10 17:49:14 -05002731{
2732 RR_DEBUG_INFO_UPDATE_LOC();
2733 *this = ShuffleLowHigh(*x.parent, *y.parent, (uint16_t(X) & 0xFF00u) | (uint16_t(Y >> 8) & 0x00FFu));
2734}
Nicolas Capensd022e412016-09-26 13:30:14 -04002735
Nicolas Capens157ba262019-12-10 17:49:14 -05002736template<int X, int Y>
Ben Clayton713b8d32019-12-17 20:37:56 +00002737Float4::Float4(const SwizzleMask2<Float4, X> &x, const SwizzleMask2<Float4, Y> &y)
2738 : XYZW(this)
Nicolas Capens157ba262019-12-10 17:49:14 -05002739{
2740 RR_DEBUG_INFO_UPDATE_LOC();
2741 *this = ShuffleLowHigh(*x.parent, *y.parent, (uint16_t(X) & 0xFF00u) | (uint16_t(Y >> 8) & 0x00FFu));
2742}
Nicolas Capensd022e412016-09-26 13:30:14 -04002743
Nicolas Capens157ba262019-12-10 17:49:14 -05002744template<int T>
2745RValue<Float4> Float4::operator=(const SwizzleMask1<Float4, T> &rhs)
2746{
2747 return *this = rhs.operator RValue<Float4>();
2748}
Nicolas Capensd022e412016-09-26 13:30:14 -04002749
Nicolas Capens157ba262019-12-10 17:49:14 -05002750template<int T>
2751RValue<Float4> Float4::operator=(const Swizzle4<Float4, T> &rhs)
2752{
2753 return *this = rhs.operator RValue<Float4>();
2754}
Nicolas Capensd022e412016-09-26 13:30:14 -04002755
Nicolas Capens157ba262019-12-10 17:49:14 -05002756// Returns a reactor pointer to the fixed-address ptr.
Nicolas Capens3d7faaa2022-10-04 14:48:57 -04002757RValue<Pointer<Byte>> ConstantPointer(const void *ptr);
Ben Claytonb7eb3a82019-11-19 00:43:50 +00002758
Nicolas Capens157ba262019-12-10 17:49:14 -05002759// Returns a reactor pointer to an immutable copy of the data of size bytes.
Nicolas Capens3d7faaa2022-10-04 14:48:57 -04002760RValue<Pointer<Byte>> ConstantData(const void *data, size_t size);
Ben Claytonb7eb3a82019-11-19 00:43:50 +00002761
Nicolas Capens157ba262019-12-10 17:49:14 -05002762template<class T>
Ben Clayton713b8d32019-12-17 20:37:56 +00002763Pointer<T>::Pointer(Argument<Pointer<T>> argument)
2764 : alignment(1)
Nicolas Capens157ba262019-12-10 17:49:14 -05002765{
Nicolas Capens3b655b62020-05-13 15:23:54 -04002766 this->store(argument.rvalue());
Nicolas Capens157ba262019-12-10 17:49:14 -05002767}
Nicolas Capensd022e412016-09-26 13:30:14 -04002768
Nicolas Capens157ba262019-12-10 17:49:14 -05002769template<class T>
Ben Clayton713b8d32019-12-17 20:37:56 +00002770Pointer<T>::Pointer()
2771 : alignment(1)
2772{}
Nicolas Capensd022e412016-09-26 13:30:14 -04002773
Nicolas Capens157ba262019-12-10 17:49:14 -05002774template<class T>
Ben Clayton713b8d32019-12-17 20:37:56 +00002775Pointer<T>::Pointer(RValue<Pointer<T>> rhs)
2776 : alignment(1)
Nicolas Capens157ba262019-12-10 17:49:14 -05002777{
Nicolas Capens3b655b62020-05-13 15:23:54 -04002778 this->store(rhs);
Nicolas Capens157ba262019-12-10 17:49:14 -05002779}
Nicolas Capensd022e412016-09-26 13:30:14 -04002780
Nicolas Capens157ba262019-12-10 17:49:14 -05002781template<class T>
Ben Clayton713b8d32019-12-17 20:37:56 +00002782Pointer<T>::Pointer(const Pointer<T> &rhs)
2783 : alignment(rhs.alignment)
Nicolas Capens157ba262019-12-10 17:49:14 -05002784{
Nicolas Capens3b655b62020-05-13 15:23:54 -04002785 this->store(rhs.load());
Nicolas Capens157ba262019-12-10 17:49:14 -05002786}
Nicolas Capensd022e412016-09-26 13:30:14 -04002787
Nicolas Capens157ba262019-12-10 17:49:14 -05002788template<class T>
Ben Clayton713b8d32019-12-17 20:37:56 +00002789Pointer<T>::Pointer(const Reference<Pointer<T>> &rhs)
2790 : alignment(rhs.getAlignment())
Nicolas Capens157ba262019-12-10 17:49:14 -05002791{
Nicolas Capens3b655b62020-05-13 15:23:54 -04002792 this->store(rhs.load());
Nicolas Capens157ba262019-12-10 17:49:14 -05002793}
Nicolas Capensd022e412016-09-26 13:30:14 -04002794
Nicolas Capens157ba262019-12-10 17:49:14 -05002795template<class T>
Ben Clayton713b8d32019-12-17 20:37:56 +00002796Pointer<T>::Pointer(std::nullptr_t)
2797 : alignment(1)
Nicolas Capens157ba262019-12-10 17:49:14 -05002798{
Nicolas Capens519cf222020-05-08 15:27:19 -04002799 Value *value = Nucleus::createNullPointer(T::type());
Nicolas Capens3b655b62020-05-13 15:23:54 -04002800 this->storeValue(value);
Nicolas Capens157ba262019-12-10 17:49:14 -05002801}
Ben Clayton0697da02019-08-01 13:35:48 +01002802
Nicolas Capens157ba262019-12-10 17:49:14 -05002803template<class T>
2804RValue<Pointer<T>> Pointer<T>::operator=(RValue<Pointer<T>> rhs)
2805{
Nicolas Capens3b655b62020-05-13 15:23:54 -04002806 return this->store(rhs);
Nicolas Capens157ba262019-12-10 17:49:14 -05002807}
Nicolas Capensd022e412016-09-26 13:30:14 -04002808
Nicolas Capens157ba262019-12-10 17:49:14 -05002809template<class T>
2810RValue<Pointer<T>> Pointer<T>::operator=(const Pointer<T> &rhs)
2811{
Nicolas Capens3b655b62020-05-13 15:23:54 -04002812 return this->store(rhs.load());
Nicolas Capens157ba262019-12-10 17:49:14 -05002813}
Nicolas Capensd022e412016-09-26 13:30:14 -04002814
Nicolas Capens157ba262019-12-10 17:49:14 -05002815template<class T>
2816RValue<Pointer<T>> Pointer<T>::operator=(const Reference<Pointer<T>> &rhs)
2817{
Nicolas Capens3b655b62020-05-13 15:23:54 -04002818 return this->store(rhs.load());
Nicolas Capens157ba262019-12-10 17:49:14 -05002819}
Nicolas Capensd022e412016-09-26 13:30:14 -04002820
Nicolas Capens157ba262019-12-10 17:49:14 -05002821template<class T>
2822RValue<Pointer<T>> Pointer<T>::operator=(std::nullptr_t)
2823{
Nicolas Capens519cf222020-05-08 15:27:19 -04002824 Value *value = Nucleus::createNullPointer(T::type());
Nicolas Capens3b655b62020-05-13 15:23:54 -04002825 this->storeValue(value);
Ben Clayton0697da02019-08-01 13:35:48 +01002826
Nicolas Capens157ba262019-12-10 17:49:14 -05002827 return RValue<Pointer<T>>(this);
2828}
Ben Clayton0697da02019-08-01 13:35:48 +01002829
Nicolas Capens157ba262019-12-10 17:49:14 -05002830template<class T>
Nicolas Capensef72cb42021-11-12 13:57:14 -05002831Reference<T> Pointer<T>::operator*() const
Nicolas Capens157ba262019-12-10 17:49:14 -05002832{
Nicolas Capens3b655b62020-05-13 15:23:54 -04002833 return Reference<T>(this->loadValue(), alignment);
Nicolas Capens157ba262019-12-10 17:49:14 -05002834}
Nicolas Capensd022e412016-09-26 13:30:14 -04002835
Nicolas Capens157ba262019-12-10 17:49:14 -05002836template<class T>
Nicolas Capensef72cb42021-11-12 13:57:14 -05002837Reference<T> Pointer<T>::operator[](int index) const
Nicolas Capens157ba262019-12-10 17:49:14 -05002838{
2839 RR_DEBUG_INFO_UPDATE_LOC();
Nicolas Capens3b655b62020-05-13 15:23:54 -04002840 Value *element = Nucleus::createGEP(this->loadValue(), T::type(), Nucleus::createConstantInt(index), false);
Nicolas Capensd294def2017-01-26 17:44:37 -08002841
Nicolas Capens157ba262019-12-10 17:49:14 -05002842 return Reference<T>(element, alignment);
2843}
Nicolas Capensd294def2017-01-26 17:44:37 -08002844
Nicolas Capens157ba262019-12-10 17:49:14 -05002845template<class T>
Nicolas Capensef72cb42021-11-12 13:57:14 -05002846Reference<T> Pointer<T>::operator[](unsigned int index) const
Nicolas Capens157ba262019-12-10 17:49:14 -05002847{
2848 RR_DEBUG_INFO_UPDATE_LOC();
Nicolas Capens3b655b62020-05-13 15:23:54 -04002849 Value *element = Nucleus::createGEP(this->loadValue(), T::type(), Nucleus::createConstantInt(index), true);
Nicolas Capensd022e412016-09-26 13:30:14 -04002850
Nicolas Capens157ba262019-12-10 17:49:14 -05002851 return Reference<T>(element, alignment);
2852}
Nicolas Capensd022e412016-09-26 13:30:14 -04002853
Nicolas Capens157ba262019-12-10 17:49:14 -05002854template<class T>
Nicolas Capensef72cb42021-11-12 13:57:14 -05002855Reference<T> Pointer<T>::operator[](RValue<Int> index) const
Nicolas Capens157ba262019-12-10 17:49:14 -05002856{
2857 RR_DEBUG_INFO_UPDATE_LOC();
Nicolas Capens3b655b62020-05-13 15:23:54 -04002858 Value *element = Nucleus::createGEP(this->loadValue(), T::type(), index.value(), false);
Nicolas Capensd294def2017-01-26 17:44:37 -08002859
Nicolas Capens157ba262019-12-10 17:49:14 -05002860 return Reference<T>(element, alignment);
2861}
Nicolas Capensd294def2017-01-26 17:44:37 -08002862
Nicolas Capens157ba262019-12-10 17:49:14 -05002863template<class T>
Nicolas Capensef72cb42021-11-12 13:57:14 -05002864Reference<T> Pointer<T>::operator[](RValue<UInt> index) const
Nicolas Capens157ba262019-12-10 17:49:14 -05002865{
2866 RR_DEBUG_INFO_UPDATE_LOC();
Nicolas Capens3b655b62020-05-13 15:23:54 -04002867 Value *element = Nucleus::createGEP(this->loadValue(), T::type(), index.value(), true);
Nicolas Capensd022e412016-09-26 13:30:14 -04002868
Nicolas Capens157ba262019-12-10 17:49:14 -05002869 return Reference<T>(element, alignment);
2870}
Nicolas Capensd022e412016-09-26 13:30:14 -04002871
Nicolas Capens157ba262019-12-10 17:49:14 -05002872template<class T>
Nicolas Capens519cf222020-05-08 15:27:19 -04002873Type *Pointer<T>::type()
Nicolas Capens157ba262019-12-10 17:49:14 -05002874{
Nicolas Capens519cf222020-05-08 15:27:19 -04002875 return Nucleus::getPointerType(T::type());
Nicolas Capens157ba262019-12-10 17:49:14 -05002876}
Nicolas Capensd022e412016-09-26 13:30:14 -04002877
Nicolas Capens157ba262019-12-10 17:49:14 -05002878template<class T, int S>
Ben Clayton713b8d32019-12-17 20:37:56 +00002879Array<T, S>::Array(int size)
Antonio Maiorano84c61e12020-12-02 12:06:05 -05002880 : LValue<T>(size)
Nicolas Capens157ba262019-12-10 17:49:14 -05002881{
2882}
Nicolas Capensd022e412016-09-26 13:30:14 -04002883
Nicolas Capens157ba262019-12-10 17:49:14 -05002884template<class T, int S>
2885Reference<T> Array<T, S>::operator[](int index)
2886{
Antonio Maiorano84c61e12020-12-02 12:06:05 -05002887 assert(index < Variable::getArraySize());
Nicolas Capens3b655b62020-05-13 15:23:54 -04002888 Value *element = this->getElementPointer(Nucleus::createConstantInt(index), false);
Nicolas Capensd294def2017-01-26 17:44:37 -08002889
Nicolas Capens157ba262019-12-10 17:49:14 -05002890 return Reference<T>(element);
2891}
Nicolas Capensd294def2017-01-26 17:44:37 -08002892
Nicolas Capens157ba262019-12-10 17:49:14 -05002893template<class T, int S>
2894Reference<T> Array<T, S>::operator[](unsigned int index)
2895{
Antonio Maiorano84c61e12020-12-02 12:06:05 -05002896 assert(index < static_cast<unsigned int>(Variable::getArraySize()));
Nicolas Capens3b655b62020-05-13 15:23:54 -04002897 Value *element = this->getElementPointer(Nucleus::createConstantInt(index), true);
Nicolas Capensd022e412016-09-26 13:30:14 -04002898
Nicolas Capens157ba262019-12-10 17:49:14 -05002899 return Reference<T>(element);
2900}
Nicolas Capensd022e412016-09-26 13:30:14 -04002901
Nicolas Capens157ba262019-12-10 17:49:14 -05002902template<class T, int S>
2903Reference<T> Array<T, S>::operator[](RValue<Int> index)
2904{
Nicolas Capens3b655b62020-05-13 15:23:54 -04002905 Value *element = this->getElementPointer(index.value(), false);
Nicolas Capensd294def2017-01-26 17:44:37 -08002906
Nicolas Capens157ba262019-12-10 17:49:14 -05002907 return Reference<T>(element);
2908}
Nicolas Capensd294def2017-01-26 17:44:37 -08002909
Nicolas Capens157ba262019-12-10 17:49:14 -05002910template<class T, int S>
2911Reference<T> Array<T, S>::operator[](RValue<UInt> index)
2912{
Nicolas Capens3b655b62020-05-13 15:23:54 -04002913 Value *element = this->getElementPointer(index.value(), true);
Nicolas Capensd022e412016-09-26 13:30:14 -04002914
Nicolas Capens157ba262019-12-10 17:49:14 -05002915 return Reference<T>(element);
2916}
Nicolas Capensd022e412016-09-26 13:30:14 -04002917
2918// template<class T>
Nicolas Capens96d4e092016-11-18 14:22:38 -05002919// RValue<Array<T>> operator++(Array<T> &val, int)
Nicolas Capensd022e412016-09-26 13:30:14 -04002920// {
2921// // FIXME: Requires storing the address of the array
2922// }
2923
2924// template<class T>
Nicolas Capens96d4e092016-11-18 14:22:38 -05002925// const Array<T> &operator++(Array<T> &val)
Nicolas Capensd022e412016-09-26 13:30:14 -04002926// {
2927// // FIXME: Requires storing the address of the array
2928// }
2929
2930// template<class T>
Nicolas Capens96d4e092016-11-18 14:22:38 -05002931// RValue<Array<T>> operator--(Array<T> &val, int)
Nicolas Capensd022e412016-09-26 13:30:14 -04002932// {
2933// // FIXME: Requires storing the address of the array
2934// }
2935
2936// template<class T>
Nicolas Capens96d4e092016-11-18 14:22:38 -05002937// const Array<T> &operator--(Array<T> &val)
Nicolas Capensd022e412016-09-26 13:30:14 -04002938// {
2939// // FIXME: Requires storing the address of the array
2940// }
2941
Nicolas Capens157ba262019-12-10 17:49:14 -05002942template<class T>
2943RValue<T> IfThenElse(RValue<Bool> condition, RValue<T> ifTrue, RValue<T> ifFalse)
2944{
2945 RR_DEBUG_INFO_UPDATE_LOC();
Nicolas Capensb6e8c3f2020-05-01 23:28:37 -04002946 return RValue<T>(Nucleus::createSelect(condition.value(), ifTrue.value(), ifFalse.value()));
Nicolas Capens157ba262019-12-10 17:49:14 -05002947}
2948
2949template<class T>
2950RValue<T> IfThenElse(RValue<Bool> condition, const T &ifTrue, RValue<T> ifFalse)
2951{
2952 RR_DEBUG_INFO_UPDATE_LOC();
2953 Value *trueValue = ifTrue.loadValue();
2954
Nicolas Capensb6e8c3f2020-05-01 23:28:37 -04002955 return RValue<T>(Nucleus::createSelect(condition.value(), trueValue, ifFalse.value()));
Nicolas Capens157ba262019-12-10 17:49:14 -05002956}
2957
2958template<class T>
2959RValue<T> IfThenElse(RValue<Bool> condition, RValue<T> ifTrue, const T &ifFalse)
2960{
2961 RR_DEBUG_INFO_UPDATE_LOC();
2962 Value *falseValue = ifFalse.loadValue();
2963
Nicolas Capensb6e8c3f2020-05-01 23:28:37 -04002964 return RValue<T>(Nucleus::createSelect(condition.value(), ifTrue.value(), falseValue));
Nicolas Capens157ba262019-12-10 17:49:14 -05002965}
2966
2967template<class T>
2968RValue<T> IfThenElse(RValue<Bool> condition, const T &ifTrue, const T &ifFalse)
2969{
2970 RR_DEBUG_INFO_UPDATE_LOC();
2971 Value *trueValue = ifTrue.loadValue();
2972 Value *falseValue = ifFalse.loadValue();
2973
Nicolas Capensb6e8c3f2020-05-01 23:28:37 -04002974 return RValue<T>(Nucleus::createSelect(condition.value(), trueValue, falseValue));
Nicolas Capens157ba262019-12-10 17:49:14 -05002975}
2976
2977template<typename Return, typename... Arguments>
2978Function<Return(Arguments...)>::Function()
Nicolas Capens3d26cfc2021-01-22 16:51:00 -05002979 : core(new Nucleus())
Nicolas Capens157ba262019-12-10 17:49:14 -05002980{
Nicolas Capens519cf222020-05-08 15:27:19 -04002981 Type *types[] = { Arguments::type()... };
Nicolas Capens157ba262019-12-10 17:49:14 -05002982 for(Type *type : types)
Nicolas Capensd022e412016-09-26 13:30:14 -04002983 {
Nicolas Capens519cf222020-05-08 15:27:19 -04002984 if(type != Void::type())
Nicolas Capensd022e412016-09-26 13:30:14 -04002985 {
Nicolas Capens157ba262019-12-10 17:49:14 -05002986 arguments.push_back(type);
Nicolas Capensd022e412016-09-26 13:30:14 -04002987 }
Nicolas Capensd022e412016-09-26 13:30:14 -04002988 }
2989
Nicolas Capens519cf222020-05-08 15:27:19 -04002990 Nucleus::createFunction(Return::type(), arguments);
Nicolas Capens157ba262019-12-10 17:49:14 -05002991}
2992
2993template<typename Return, typename... Arguments>
Nicolas Capens157ba262019-12-10 17:49:14 -05002994std::shared_ptr<Routine> Function<Return(Arguments...)>::operator()(const char *name, ...)
2995{
2996 char fullName[1024 + 1];
2997
2998 va_list vararg;
2999 va_start(vararg, name);
3000 vsnprintf(fullName, 1024, name, vararg);
3001 va_end(vararg);
3002
Nicolas Capens79d4c6c2022-04-22 17:20:26 -04003003 auto routine = core->acquireRoutine(fullName);
Nicolas Capens3d26cfc2021-01-22 16:51:00 -05003004 core.reset(nullptr);
3005
3006 return routine;
Nicolas Capens157ba262019-12-10 17:49:14 -05003007}
3008
3009template<class T, class S>
3010RValue<T> ReinterpretCast(RValue<S> val)
3011{
3012 RR_DEBUG_INFO_UPDATE_LOC();
Nicolas Capensb6e8c3f2020-05-01 23:28:37 -04003013 return RValue<T>(Nucleus::createBitCast(val.value(), T::type()));
Nicolas Capens157ba262019-12-10 17:49:14 -05003014}
3015
3016template<class T, class S>
3017RValue<T> ReinterpretCast(const LValue<S> &var)
3018{
3019 RR_DEBUG_INFO_UPDATE_LOC();
3020 Value *val = var.loadValue();
3021
Nicolas Capens519cf222020-05-08 15:27:19 -04003022 return RValue<T>(Nucleus::createBitCast(val, T::type()));
Nicolas Capens157ba262019-12-10 17:49:14 -05003023}
3024
3025template<class T, class S>
3026RValue<T> ReinterpretCast(const Reference<S> &var)
3027{
3028 return ReinterpretCast<T>(RValue<S>(var));
3029}
3030
3031template<class T>
3032RValue<T> As(Value *val)
3033{
3034 RR_DEBUG_INFO_UPDATE_LOC();
Nicolas Capens519cf222020-05-08 15:27:19 -04003035 return RValue<T>(Nucleus::createBitCast(val, T::type()));
Nicolas Capens157ba262019-12-10 17:49:14 -05003036}
3037
3038template<class T, class S>
3039RValue<T> As(RValue<S> val)
3040{
3041 return ReinterpretCast<T>(val);
3042}
3043
3044template<class T, class S>
3045RValue<T> As(const LValue<S> &var)
3046{
3047 return ReinterpretCast<T>(var);
3048}
3049
3050template<class T, class S>
3051RValue<T> As(const Reference<S> &val)
3052{
3053 return ReinterpretCast<T>(val);
3054}
3055
3056// Calls the function pointer fptr with the given arguments, return type
3057// and parameter types. Returns the call's return value if the function has
3058// a non-void return type.
Ben Clayton713b8d32019-12-17 20:37:56 +00003059Value *Call(RValue<Pointer<Byte>> fptr, Type *retTy, std::initializer_list<Value *> args, std::initializer_list<Type *> paramTys);
Nicolas Capens157ba262019-12-10 17:49:14 -05003060
Ben Clayton713b8d32019-12-17 20:37:56 +00003061template<typename F>
3062class CallHelper
3063{};
Nicolas Capens157ba262019-12-10 17:49:14 -05003064
Ben Clayton713b8d32019-12-17 20:37:56 +00003065template<typename Return, typename... Arguments>
Nicolas Capens157ba262019-12-10 17:49:14 -05003066class CallHelper<Return(Arguments...)>
3067{
3068public:
3069 using RReturn = CToReactorT<Return>;
3070
3071 static inline RReturn Call(Return(fptr)(Arguments...), CToReactorT<Arguments>... args)
Nicolas Capensd022e412016-09-26 13:30:14 -04003072 {
Nicolas Capens157ba262019-12-10 17:49:14 -05003073 return RValue<RReturn>(rr::Call(
Ben Clayton713b8d32019-12-17 20:37:56 +00003074 ConstantPointer(reinterpret_cast<void *>(fptr)),
Nicolas Capens519cf222020-05-08 15:27:19 -04003075 RReturn::type(),
Ben Clayton713b8d32019-12-17 20:37:56 +00003076 { ValueOf(args)... },
Nicolas Capens519cf222020-05-08 15:27:19 -04003077 { CToReactorT<Arguments>::type()... }));
Nicolas Capensd022e412016-09-26 13:30:14 -04003078 }
3079
Nicolas Capens157ba262019-12-10 17:49:14 -05003080 static inline RReturn Call(Pointer<Byte> fptr, CToReactorT<Arguments>... args)
Nicolas Capensd022e412016-09-26 13:30:14 -04003081 {
Nicolas Capens157ba262019-12-10 17:49:14 -05003082 return RValue<RReturn>(rr::Call(
Ben Clayton713b8d32019-12-17 20:37:56 +00003083 fptr,
Nicolas Capens519cf222020-05-08 15:27:19 -04003084 RReturn::type(),
Ben Clayton713b8d32019-12-17 20:37:56 +00003085 { ValueOf(args)... },
Nicolas Capens519cf222020-05-08 15:27:19 -04003086 { CToReactorT<Arguments>::type()... }));
Nicolas Capens157ba262019-12-10 17:49:14 -05003087 }
3088};
Nicolas Capensd022e412016-09-26 13:30:14 -04003089
Ben Clayton713b8d32019-12-17 20:37:56 +00003090template<typename... Arguments>
Nicolas Capens157ba262019-12-10 17:49:14 -05003091class CallHelper<void(Arguments...)>
3092{
3093public:
3094 static inline void Call(void(fptr)(Arguments...), CToReactorT<Arguments>... args)
3095 {
Ben Clayton713b8d32019-12-17 20:37:56 +00003096 rr::Call(ConstantPointer(reinterpret_cast<void *>(fptr)),
Nicolas Capens519cf222020-05-08 15:27:19 -04003097 Void::type(),
Ben Clayton713b8d32019-12-17 20:37:56 +00003098 { ValueOf(args)... },
Nicolas Capens519cf222020-05-08 15:27:19 -04003099 { CToReactorT<Arguments>::type()... });
Ben Clayton68cfc782019-06-29 12:31:08 +01003100 }
3101
Nicolas Capens157ba262019-12-10 17:49:14 -05003102 static inline void Call(Pointer<Byte> fptr, CToReactorT<Arguments>... args)
Ben Clayton68cfc782019-06-29 12:31:08 +01003103 {
Nicolas Capens157ba262019-12-10 17:49:14 -05003104 rr::Call(fptr,
Nicolas Capens519cf222020-05-08 15:27:19 -04003105 Void::type(),
Ben Clayton713b8d32019-12-17 20:37:56 +00003106 { ValueOf(args)... },
Nicolas Capens519cf222020-05-08 15:27:19 -04003107 { CToReactorT<Arguments>::type()... });
Nicolas Capensd022e412016-09-26 13:30:14 -04003108 }
Nicolas Capens157ba262019-12-10 17:49:14 -05003109};
Nicolas Capensd022e412016-09-26 13:30:14 -04003110
Ben Clayton713b8d32019-12-17 20:37:56 +00003111template<typename T>
3112inline ReactorTypeT<T> CastToReactor(const T &v)
3113{
3114 return ReactorType<T>::cast(v);
3115}
Nicolas Capens157ba262019-12-10 17:49:14 -05003116
3117// Calls the static function pointer fptr with the given arguments args.
Ben Clayton713b8d32019-12-17 20:37:56 +00003118template<typename Return, typename... CArgs, typename... RArgs>
Nicolas Capens0eb3b102022-06-15 16:49:17 -04003119inline CToReactorT<Return> Call(Return(fptr)(CArgs...), RArgs &&...args)
Nicolas Capens157ba262019-12-10 17:49:14 -05003120{
3121 return CallHelper<Return(CArgs...)>::Call(fptr, CastToReactor(std::forward<RArgs>(args))...);
3122}
3123
3124// Calls the static function pointer fptr with the given arguments args.
3125// Overload for calling functions with void return type.
Ben Clayton713b8d32019-12-17 20:37:56 +00003126template<typename... CArgs, typename... RArgs>
Nicolas Capens0eb3b102022-06-15 16:49:17 -04003127inline void Call(void(fptr)(CArgs...), RArgs &&...args)
Nicolas Capens157ba262019-12-10 17:49:14 -05003128{
3129 CallHelper<void(CArgs...)>::Call(fptr, CastToReactor(std::forward<RArgs>(args))...);
3130}
3131
3132// Calls the member function pointer fptr with the given arguments args.
3133// object can be a Class*, or a Pointer<Byte>.
Ben Clayton713b8d32019-12-17 20:37:56 +00003134template<typename Return, typename Class, typename C, typename... CArgs, typename... RArgs>
Nicolas Capens0eb3b102022-06-15 16:49:17 -04003135inline CToReactorT<Return> Call(Return (Class::*fptr)(CArgs...), C &&object, RArgs &&...args)
Nicolas Capens157ba262019-12-10 17:49:14 -05003136{
Ben Clayton713b8d32019-12-17 20:37:56 +00003137 using Helper = CallHelper<Return(Class *, void *, CArgs...)>;
Nicolas Capens157ba262019-12-10 17:49:14 -05003138 using fptrTy = decltype(fptr);
3139
3140 struct Static
Nicolas Capensd022e412016-09-26 13:30:14 -04003141 {
Ben Clayton713b8d32019-12-17 20:37:56 +00003142 static inline Return Call(Class *object, void *fptrptr, CArgs... args)
Ben Claytond853c122019-04-16 17:51:49 -04003143 {
Ben Clayton713b8d32019-12-17 20:37:56 +00003144 auto fptr = *reinterpret_cast<fptrTy *>(fptrptr);
Nicolas Capens157ba262019-12-10 17:49:14 -05003145 return (object->*fptr)(std::forward<CArgs>(args)...);
Ben Claytond853c122019-04-16 17:51:49 -04003146 }
3147 };
3148
Nicolas Capens157ba262019-12-10 17:49:14 -05003149 return Helper::Call(&Static::Call,
3150 CastToReactor(object),
3151 ConstantData(&fptr, sizeof(fptr)),
3152 CastToReactor(std::forward<RArgs>(args))...);
3153}
Ben Claytond853c122019-04-16 17:51:49 -04003154
Nicolas Capens157ba262019-12-10 17:49:14 -05003155// Calls the member function pointer fptr with the given arguments args.
3156// Overload for calling functions with void return type.
3157// object can be a Class*, or a Pointer<Byte>.
Ben Clayton713b8d32019-12-17 20:37:56 +00003158template<typename Class, typename C, typename... CArgs, typename... RArgs>
Nicolas Capens0eb3b102022-06-15 16:49:17 -04003159inline void Call(void (Class::*fptr)(CArgs...), C &&object, RArgs &&...args)
Nicolas Capens157ba262019-12-10 17:49:14 -05003160{
Ben Clayton713b8d32019-12-17 20:37:56 +00003161 using Helper = CallHelper<void(Class *, void *, CArgs...)>;
Nicolas Capens157ba262019-12-10 17:49:14 -05003162 using fptrTy = decltype(fptr);
3163
3164 struct Static
3165 {
Ben Clayton713b8d32019-12-17 20:37:56 +00003166 static inline void Call(Class *object, void *fptrptr, CArgs... args)
Ben Claytond853c122019-04-16 17:51:49 -04003167 {
Ben Clayton713b8d32019-12-17 20:37:56 +00003168 auto fptr = *reinterpret_cast<fptrTy *>(fptrptr);
Nicolas Capens157ba262019-12-10 17:49:14 -05003169 (object->*fptr)(std::forward<CArgs>(args)...);
Ben Claytond853c122019-04-16 17:51:49 -04003170 }
3171 };
3172
Nicolas Capens157ba262019-12-10 17:49:14 -05003173 Helper::Call(&Static::Call,
3174 CastToReactor(object),
3175 ConstantData(&fptr, sizeof(fptr)),
3176 CastToReactor(std::forward<RArgs>(args))...);
3177}
Ben Clayton51f08312019-11-08 14:39:26 +00003178
Nicolas Capens22f14a82020-12-16 15:41:22 -05003179// NonVoidFunction<F> and VoidFunction<F> are helper classes which define ReturnType
3180// when F matches a non-void fuction signature or void function signature, respectively,
3181// as the function's return type.
3182template<typename F>
3183struct NonVoidFunction
3184{};
3185
3186template<typename Return, typename... Arguments>
3187struct NonVoidFunction<Return(Arguments...)>
3188{
3189 using ReturnType = Return;
3190};
3191
3192template<typename... Arguments>
3193struct NonVoidFunction<void(Arguments...)>
3194{
3195};
3196
3197template<typename F>
3198using NonVoidFunctionReturnType = typename NonVoidFunction<F>::ReturnType;
3199
3200template<typename F>
3201struct VoidFunction
3202{};
3203
3204template<typename... Arguments>
3205struct VoidFunction<void(Arguments...)>
3206{
3207 using ReturnType = void;
3208};
3209
3210template<typename F>
3211using VoidFunctionReturnType = typename VoidFunction<F>::ReturnType;
3212
3213// Calls the Reactor function pointer fptr with the signature FUNCTION_SIGNATURE and arguments.
3214// Overload for calling functions with non-void return type.
Ben Clayton713b8d32019-12-17 20:37:56 +00003215template<typename FUNCTION_SIGNATURE, typename... RArgs>
Nicolas Capens0eb3b102022-06-15 16:49:17 -04003216inline CToReactorT<NonVoidFunctionReturnType<FUNCTION_SIGNATURE>> Call(Pointer<Byte> fptr, RArgs &&...args)
Nicolas Capens22f14a82020-12-16 15:41:22 -05003217{
3218 return CallHelper<FUNCTION_SIGNATURE>::Call(fptr, CastToReactor(std::forward<RArgs>(args))...);
3219}
3220
3221// Calls the Reactor function pointer fptr with the signature FUNCTION_SIGNATURE and arguments.
3222// Overload for calling functions with void return type.
3223template<typename FUNCTION_SIGNATURE, typename... RArgs>
Nicolas Capens0eb3b102022-06-15 16:49:17 -04003224inline VoidFunctionReturnType<FUNCTION_SIGNATURE> Call(Pointer<Byte> fptr, RArgs &&...args)
Nicolas Capens157ba262019-12-10 17:49:14 -05003225{
3226 CallHelper<FUNCTION_SIGNATURE>::Call(fptr, CastToReactor(std::forward<RArgs>(args))...);
3227}
3228
3229// Breakpoint emits an instruction that will cause the application to trap.
3230// This can be used to stop an attached debugger at the given call.
3231void Breakpoint();
3232
3233class ForData
3234{
3235public:
Ben Clayton713b8d32019-12-17 20:37:56 +00003236 ForData(bool init)
3237 : loopOnce(init)
Ben Claytond853c122019-04-16 17:51:49 -04003238 {
Ben Claytond853c122019-04-16 17:51:49 -04003239 }
3240
Nicolas Capens157ba262019-12-10 17:49:14 -05003241 operator bool()
Antonio Maiorano7363cd22019-10-11 15:23:22 -04003242 {
Nicolas Capens157ba262019-12-10 17:49:14 -05003243 return loopOnce;
Antonio Maiorano7363cd22019-10-11 15:23:22 -04003244 }
3245
Nicolas Capens157ba262019-12-10 17:49:14 -05003246 bool operator=(bool value)
Ben Claytonb7eb3a82019-11-19 00:43:50 +00003247 {
Nicolas Capens157ba262019-12-10 17:49:14 -05003248 return loopOnce = value;
Ben Claytonb7eb3a82019-11-19 00:43:50 +00003249 }
3250
Nicolas Capens157ba262019-12-10 17:49:14 -05003251 bool setup()
Ben Claytonb7eb3a82019-11-19 00:43:50 +00003252 {
Nicolas Capens157ba262019-12-10 17:49:14 -05003253 RR_DEBUG_INFO_FLUSH();
3254 if(Nucleus::getInsertBlock() != endBB)
Nicolas Capens37ed9082016-11-16 17:40:48 -05003255 {
Nicolas Capens157ba262019-12-10 17:49:14 -05003256 testBB = Nucleus::createBasicBlock();
Nicolas Capens37ed9082016-11-16 17:40:48 -05003257
Nicolas Capens157ba262019-12-10 17:49:14 -05003258 Nucleus::createBr(testBB);
3259 Nucleus::setInsertBlock(testBB);
Nicolas Capens37ed9082016-11-16 17:40:48 -05003260
3261 return true;
3262 }
3263
Nicolas Capens157ba262019-12-10 17:49:14 -05003264 return false;
Nicolas Capens0bac2852016-05-07 06:09:58 -04003265 }
3266
Nicolas Capens157ba262019-12-10 17:49:14 -05003267 bool test(RValue<Bool> cmp)
3268 {
3269 BasicBlock *bodyBB = Nucleus::createBasicBlock();
3270 endBB = Nucleus::createBasicBlock();
Nicolas Capens0bac2852016-05-07 06:09:58 -04003271
Nicolas Capensb6e8c3f2020-05-01 23:28:37 -04003272 Nucleus::createCondBr(cmp.value(), bodyBB, endBB);
Nicolas Capens157ba262019-12-10 17:49:14 -05003273 Nucleus::setInsertBlock(bodyBB);
Nicolas Capens37ed9082016-11-16 17:40:48 -05003274
Nicolas Capens157ba262019-12-10 17:49:14 -05003275 return true;
3276 }
3277
3278 void end()
3279 {
3280 Nucleus::createBr(testBB);
3281 Nucleus::setInsertBlock(endBB);
3282 }
3283
3284private:
3285 BasicBlock *testBB = nullptr;
3286 BasicBlock *endBB = nullptr;
3287 bool loopOnce = true;
3288};
3289
3290class IfElseData
3291{
3292public:
Ben Clayton713b8d32019-12-17 20:37:56 +00003293 IfElseData(RValue<Bool> cmp)
Nicolas Capens157ba262019-12-10 17:49:14 -05003294 {
Nicolas Capens157ba262019-12-10 17:49:14 -05003295 trueBB = Nucleus::createBasicBlock();
Nicolas Capensd85a7a22021-02-18 15:23:40 -05003296 falseBB = Nucleus::createBasicBlock();
3297 endBB = falseBB; // Until we encounter an Else statement, these are the same.
Nicolas Capens157ba262019-12-10 17:49:14 -05003298
Nicolas Capensd85a7a22021-02-18 15:23:40 -05003299 Nucleus::createCondBr(cmp.value(), trueBB, falseBB);
Nicolas Capens157ba262019-12-10 17:49:14 -05003300 Nucleus::setInsertBlock(trueBB);
3301 }
3302
3303 ~IfElseData()
3304 {
3305 Nucleus::createBr(endBB);
Nicolas Capens157ba262019-12-10 17:49:14 -05003306 Nucleus::setInsertBlock(endBB);
3307 }
3308
3309 operator int()
3310 {
3311 return iteration;
3312 }
3313
3314 IfElseData &operator++()
3315 {
3316 ++iteration;
3317
3318 return *this;
3319 }
3320
3321 void elseClause()
3322 {
Nicolas Capensd85a7a22021-02-18 15:23:40 -05003323 endBB = Nucleus::createBasicBlock();
Nicolas Capens157ba262019-12-10 17:49:14 -05003324 Nucleus::createBr(endBB);
3325
Nicolas Capens157ba262019-12-10 17:49:14 -05003326 Nucleus::setInsertBlock(falseBB);
3327 }
3328
3329private:
Nicolas Capensd85a7a22021-02-18 15:23:40 -05003330 BasicBlock *trueBB = nullptr;
3331 BasicBlock *falseBB = nullptr;
3332 BasicBlock *endBB = nullptr;
3333 int iteration = 0;
Nicolas Capens157ba262019-12-10 17:49:14 -05003334};
3335
Ben Clayton713b8d32019-12-17 20:37:56 +00003336#define For(init, cond, inc) \
3337 for(ForData for__ = true; for__; for__ = false) \
3338 for(init; for__.setup() && for__.test(cond); inc, for__.end())
Nicolas Capens157ba262019-12-10 17:49:14 -05003339
3340#define While(cond) For((void)0, cond, (void)0)
3341
Ben Clayton713b8d32019-12-17 20:37:56 +00003342#define Do \
3343 { \
3344 BasicBlock *body__ = Nucleus::createBasicBlock(); \
3345 Nucleus::createBr(body__); \
3346 Nucleus::setInsertBlock(body__);
Nicolas Capens157ba262019-12-10 17:49:14 -05003347
Nicolas Capensb6e8c3f2020-05-01 23:28:37 -04003348#define Until(cond) \
3349 BasicBlock *end__ = Nucleus::createBasicBlock(); \
3350 Nucleus::createCondBr((cond).value(), end__, body__); \
3351 Nucleus::setInsertBlock(end__); \
3352 } \
3353 do \
3354 { \
Ben Clayton713b8d32019-12-17 20:37:56 +00003355 } while(false) // Require a semi-colon at the end of the Until()
Nicolas Capensd022e412016-09-26 13:30:14 -04003356
Ben Clayton713b8d32019-12-17 20:37:56 +00003357enum
3358{
3359 IF_BLOCK__,
3360 ELSE_CLAUSE__,
3361 ELSE_BLOCK__,
3362 IFELSE_NUM__
3363};
Nicolas Capens157ba262019-12-10 17:49:14 -05003364
Nicolas Capens782dbeb2020-10-30 14:17:00 -04003365#define If(cond) \
3366 for(IfElseData ifElse__{ cond }; ifElse__ < IFELSE_NUM__; ++ifElse__) \
Ben Clayton713b8d32019-12-17 20:37:56 +00003367 if(ifElse__ == IF_BLOCK__)
Nicolas Capens157ba262019-12-10 17:49:14 -05003368
Ben Clayton713b8d32019-12-17 20:37:56 +00003369#define Else \
3370 else if(ifElse__ == ELSE_CLAUSE__) \
3371 { \
3372 ifElse__.elseClause(); \
3373 } \
3374 else // ELSE_BLOCK__
Nicolas Capens157ba262019-12-10 17:49:14 -05003375
Nicolas Capensd0851362020-04-30 15:59:30 -04003376// The OFFSET macro is a generalization of the offsetof() macro defined in <cstddef>.
3377// It allows e.g. getting the offset of array elements, even when indexed dynamically.
3378// We cast the address '32' and subtract it again, because null-dereference is undefined behavior.
3379#define OFFSET(s, m) ((int)(size_t) & reinterpret_cast<const volatile char &>((((s *)32)->m)) - 32)
3380
Nicolas Capens157ba262019-12-10 17:49:14 -05003381} // namespace rr
3382
Ben Clayton7e11f462019-11-18 16:00:31 +00003383#include "Traits.inl"
3384
Ben Clayton713b8d32019-12-17 20:37:56 +00003385#endif // rr_Reactor_hpp