blob: 9fec0b3e53f1818b8c416ac28f62e9a6ec2b66d6 [file] [log] [blame]
Nicolas Capens5d961882016-01-01 23:18:14 -05001// SwiftShader Software Renderer
2//
3// Copyright(c) 2015 Google Inc.
4//
5// All rights reserved. No part of this software may be copied, distributed, transmitted,
6// transcribed, stored in a retrieval system, translated into any human or computer
7// language by any means, or disclosed to third parties without the explicit written
8// agreement of Google Inc. Without such an agreement, no rights or licenses, express
9// or implied, including but not limited to any patent rights, are granted to you.
10//
11
12#include "PixelProgram.hpp"
13#include "Primitive.hpp"
14#include "Renderer.hpp"
15#include "SamplerCore.hpp"
16
17namespace sw
18{
19 extern bool postBlendSRGB;
Alexis Hetuf2a8c372015-07-13 11:08:41 -040020 extern bool booleanFaceRegister;
21 extern bool halfIntegerCoordinates; // Pixel centers are not at integer coordinates
Nicolas Capens5d961882016-01-01 23:18:14 -050022 extern bool fullPixelPositionRegister;
23
Nicolas Capens4f172c72016-01-13 08:34:30 -050024 void PixelProgram::setBuiltins(Int &x, Int &y, Float4(&z)[4], Float4 &w)
Nicolas Capens5d961882016-01-01 23:18:14 -050025 {
Alexis Hetuf2a8c372015-07-13 11:08:41 -040026 if(shader->getVersion() >= 0x0300)
27 {
28 if(shader->vPosDeclared)
29 {
30 if(!halfIntegerCoordinates)
31 {
Nicolas Capens4f172c72016-01-13 08:34:30 -050032 vPos.x = Float4(Float(x)) + Float4(0, 1, 0, 1);
33 vPos.y = Float4(Float(y)) + Float4(0, 0, 1, 1);
Alexis Hetuf2a8c372015-07-13 11:08:41 -040034 }
35 else
36 {
Nicolas Capens4f172c72016-01-13 08:34:30 -050037 vPos.x = Float4(Float(x)) + Float4(0.5f, 1.5f, 0.5f, 1.5f);
38 vPos.y = Float4(Float(y)) + Float4(0.5f, 0.5f, 1.5f, 1.5f);
Alexis Hetuf2a8c372015-07-13 11:08:41 -040039 }
40
41 if(fullPixelPositionRegister)
42 {
Nicolas Capens4f172c72016-01-13 08:34:30 -050043 vPos.z = z[0]; // FIXME: Centroid?
44 vPos.w = w; // FIXME: Centroid?
Alexis Hetuf2a8c372015-07-13 11:08:41 -040045 }
46 }
47
48 if(shader->vFaceDeclared)
49 {
Nicolas Capens4f172c72016-01-13 08:34:30 -050050 Float4 area = *Pointer<Float>(primitive + OFFSET(Primitive, area));
Alexis Hetuf2a8c372015-07-13 11:08:41 -040051 Float4 face = booleanFaceRegister ? Float4(As<Float4>(CmpNLT(area, Float4(0.0f)))) : area;
52
Nicolas Capens4f172c72016-01-13 08:34:30 -050053 vFace.x = face;
54 vFace.y = face;
55 vFace.z = face;
56 vFace.w = face;
Alexis Hetuf2a8c372015-07-13 11:08:41 -040057 }
Nicolas Capens5d961882016-01-01 23:18:14 -050058 }
59 }
60
Nicolas Capens4f172c72016-01-13 08:34:30 -050061 void PixelProgram::applyShader(Int cMask[4])
Nicolas Capens5d961882016-01-01 23:18:14 -050062 {
Nicolas Capens4f172c72016-01-13 08:34:30 -050063 enableIndex = 0;
64 stackIndex = 0;
Alexis Hetuf2a8c372015-07-13 11:08:41 -040065
66 if(shader->containsLeaveInstruction())
67 {
Nicolas Capens4f172c72016-01-13 08:34:30 -050068 enableLeave = Int4(0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF);
Alexis Hetuf2a8c372015-07-13 11:08:41 -040069 }
70
Alexis Hetu1edcd8b2015-11-05 11:12:41 -050071 for(int i = 0; i < RENDERTARGETS; i++)
72 {
73 if(state.targetFormat[i] != FORMAT_NULL)
74 {
Nicolas Capens4f172c72016-01-13 08:34:30 -050075 oC[i] = Vector4f(0.0f, 0.0f, 0.0f, 0.0f);
Alexis Hetu1edcd8b2015-11-05 11:12:41 -050076 }
77 }
Alexis Hetuf2a8c372015-07-13 11:08:41 -040078
79 // Create all call site return blocks up front
80 for(size_t i = 0; i < shader->getLength(); i++)
81 {
82 const Shader::Instruction *instruction = shader->getInstruction(i);
83 Shader::Opcode opcode = instruction->opcode;
84
85 if(opcode == Shader::OPCODE_CALL || opcode == Shader::OPCODE_CALLNZ)
86 {
87 const Dst &dst = instruction->dst;
88
89 ASSERT(callRetBlock[dst.label].size() == dst.callSite);
90 callRetBlock[dst.label].push_back(Nucleus::createBasicBlock());
91 }
92 }
93
94 for(size_t i = 0; i < shader->getLength(); i++)
95 {
96 const Shader::Instruction *instruction = shader->getInstruction(i);
97 Shader::Opcode opcode = instruction->opcode;
98
99 if(opcode == Shader::OPCODE_DCL || opcode == Shader::OPCODE_DEF || opcode == Shader::OPCODE_DEFI || opcode == Shader::OPCODE_DEFB)
100 {
101 continue;
102 }
103
104 const Dst &dst = instruction->dst;
105 const Src &src0 = instruction->src[0];
106 const Src &src1 = instruction->src[1];
107 const Src &src2 = instruction->src[2];
108 const Src &src3 = instruction->src[3];
Alexis Hetu25d47fc2015-10-22 13:58:32 -0400109 const Src &src4 = instruction->src[4];
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400110
111 bool predicate = instruction->predicate;
112 Control control = instruction->control;
113 bool pp = dst.partialPrecision;
114 bool project = instruction->project;
115 bool bias = instruction->bias;
116
117 Vector4f d;
118 Vector4f s0;
119 Vector4f s1;
120 Vector4f s2;
121 Vector4f s3;
Alexis Hetu25d47fc2015-10-22 13:58:32 -0400122 Vector4f s4;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400123
124 if(opcode == Shader::OPCODE_TEXKILL) // Takes destination as input
125 {
126 if(dst.type == Shader::PARAMETER_TEXTURE)
127 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500128 d.x = v[2 + dst.index].x;
129 d.y = v[2 + dst.index].y;
130 d.z = v[2 + dst.index].z;
131 d.w = v[2 + dst.index].w;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400132 }
133 else
134 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500135 d = r[dst.index];
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400136 }
137 }
138
Nicolas Capens4f172c72016-01-13 08:34:30 -0500139 if(src0.type != Shader::PARAMETER_VOID) s0 = fetchRegisterF(src0);
140 if(src1.type != Shader::PARAMETER_VOID) s1 = fetchRegisterF(src1);
141 if(src2.type != Shader::PARAMETER_VOID) s2 = fetchRegisterF(src2);
142 if(src3.type != Shader::PARAMETER_VOID) s3 = fetchRegisterF(src3);
143 if(src4.type != Shader::PARAMETER_VOID) s4 = fetchRegisterF(src4);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400144
145 switch(opcode)
146 {
147 case Shader::OPCODE_PS_2_0: break;
148 case Shader::OPCODE_PS_2_x: break;
149 case Shader::OPCODE_PS_3_0: break;
150 case Shader::OPCODE_DEF: break;
151 case Shader::OPCODE_DCL: break;
152 case Shader::OPCODE_NOP: break;
153 case Shader::OPCODE_MOV: mov(d, s0); break;
Alexis Hetu8d78cf72015-08-28 14:24:45 -0400154 case Shader::OPCODE_NEG: neg(d, s0); break;
155 case Shader::OPCODE_INEG: ineg(d, s0); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400156 case Shader::OPCODE_F2B: f2b(d, s0); break;
157 case Shader::OPCODE_B2F: b2f(d, s0); break;
Alexis Hetu8d78cf72015-08-28 14:24:45 -0400158 case Shader::OPCODE_F2I: f2i(d, s0); break;
159 case Shader::OPCODE_I2F: i2f(d, s0); break;
160 case Shader::OPCODE_F2U: f2u(d, s0); break;
161 case Shader::OPCODE_U2F: u2f(d, s0); break;
162 case Shader::OPCODE_I2B: i2b(d, s0); break;
163 case Shader::OPCODE_B2I: b2i(d, s0); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400164 case Shader::OPCODE_ADD: add(d, s0, s1); break;
Alexis Hetu8d78cf72015-08-28 14:24:45 -0400165 case Shader::OPCODE_IADD: iadd(d, s0, s1); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400166 case Shader::OPCODE_SUB: sub(d, s0, s1); break;
Alexis Hetu8d78cf72015-08-28 14:24:45 -0400167 case Shader::OPCODE_ISUB: isub(d, s0, s1); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400168 case Shader::OPCODE_MUL: mul(d, s0, s1); break;
Alexis Hetu8d78cf72015-08-28 14:24:45 -0400169 case Shader::OPCODE_IMUL: imul(d, s0, s1); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400170 case Shader::OPCODE_MAD: mad(d, s0, s1, s2); break;
Alexis Hetu8d78cf72015-08-28 14:24:45 -0400171 case Shader::OPCODE_IMAD: imad(d, s0, s1, s2); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400172 case Shader::OPCODE_DP1: dp1(d, s0, s1); break;
173 case Shader::OPCODE_DP2: dp2(d, s0, s1); break;
174 case Shader::OPCODE_DP2ADD: dp2add(d, s0, s1, s2); break;
175 case Shader::OPCODE_DP3: dp3(d, s0, s1); break;
176 case Shader::OPCODE_DP4: dp4(d, s0, s1); break;
Alexis Hetuc3d95f32015-09-23 12:27:32 -0400177 case Shader::OPCODE_DET2: det2(d, s0, s1); break;
178 case Shader::OPCODE_DET3: det3(d, s0, s1, s2); break;
179 case Shader::OPCODE_DET4: det4(d, s0, s1, s2, s3); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400180 case Shader::OPCODE_CMP0: cmp0(d, s0, s1, s2); break;
181 case Shader::OPCODE_ICMP: icmp(d, s0, s1, control); break;
Alexis Hetu8d78cf72015-08-28 14:24:45 -0400182 case Shader::OPCODE_UCMP: ucmp(d, s0, s1, control); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400183 case Shader::OPCODE_SELECT: select(d, s0, s1, s2); break;
184 case Shader::OPCODE_EXTRACT: extract(d.x, s0, s1.x); break;
185 case Shader::OPCODE_INSERT: insert(d, s0, s1.x, s2.x); break;
186 case Shader::OPCODE_FRC: frc(d, s0); break;
187 case Shader::OPCODE_TRUNC: trunc(d, s0); break;
188 case Shader::OPCODE_FLOOR: floor(d, s0); break;
189 case Shader::OPCODE_ROUND: round(d, s0); break;
190 case Shader::OPCODE_ROUNDEVEN: roundEven(d, s0); break;
191 case Shader::OPCODE_CEIL: ceil(d, s0); break;
192 case Shader::OPCODE_EXP2X: exp2x(d, s0, pp); break;
193 case Shader::OPCODE_EXP2: exp2(d, s0, pp); break;
194 case Shader::OPCODE_LOG2X: log2x(d, s0, pp); break;
195 case Shader::OPCODE_LOG2: log2(d, s0, pp); break;
196 case Shader::OPCODE_EXP: exp(d, s0, pp); break;
197 case Shader::OPCODE_LOG: log(d, s0, pp); break;
198 case Shader::OPCODE_RCPX: rcpx(d, s0, pp); break;
199 case Shader::OPCODE_DIV: div(d, s0, s1); break;
Alexis Hetu8d78cf72015-08-28 14:24:45 -0400200 case Shader::OPCODE_IDIV: idiv(d, s0, s1); break;
201 case Shader::OPCODE_UDIV: udiv(d, s0, s1); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400202 case Shader::OPCODE_MOD: mod(d, s0, s1); break;
Alexis Hetu8d78cf72015-08-28 14:24:45 -0400203 case Shader::OPCODE_IMOD: imod(d, s0, s1); break;
204 case Shader::OPCODE_UMOD: umod(d, s0, s1); break;
205 case Shader::OPCODE_SHL: shl(d, s0, s1); break;
206 case Shader::OPCODE_ISHR: ishr(d, s0, s1); break;
207 case Shader::OPCODE_USHR: ushr(d, s0, s1); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400208 case Shader::OPCODE_RSQX: rsqx(d, s0, pp); break;
209 case Shader::OPCODE_SQRT: sqrt(d, s0, pp); break;
210 case Shader::OPCODE_RSQ: rsq(d, s0, pp); break;
211 case Shader::OPCODE_LEN2: len2(d.x, s0, pp); break;
212 case Shader::OPCODE_LEN3: len3(d.x, s0, pp); break;
213 case Shader::OPCODE_LEN4: len4(d.x, s0, pp); break;
214 case Shader::OPCODE_DIST1: dist1(d.x, s0, s1, pp); break;
215 case Shader::OPCODE_DIST2: dist2(d.x, s0, s1, pp); break;
216 case Shader::OPCODE_DIST3: dist3(d.x, s0, s1, pp); break;
217 case Shader::OPCODE_DIST4: dist4(d.x, s0, s1, pp); break;
218 case Shader::OPCODE_MIN: min(d, s0, s1); break;
Alexis Hetu8d78cf72015-08-28 14:24:45 -0400219 case Shader::OPCODE_IMIN: imin(d, s0, s1); break;
220 case Shader::OPCODE_UMIN: umin(d, s0, s1); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400221 case Shader::OPCODE_MAX: max(d, s0, s1); break;
Alexis Hetu8d78cf72015-08-28 14:24:45 -0400222 case Shader::OPCODE_IMAX: imax(d, s0, s1); break;
223 case Shader::OPCODE_UMAX: umax(d, s0, s1); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400224 case Shader::OPCODE_LRP: lrp(d, s0, s1, s2); break;
225 case Shader::OPCODE_STEP: step(d, s0, s1); break;
226 case Shader::OPCODE_SMOOTH: smooth(d, s0, s1, s2); break;
Alexis Hetu8d78cf72015-08-28 14:24:45 -0400227 case Shader::OPCODE_FLOATBITSTOINT:
228 case Shader::OPCODE_FLOATBITSTOUINT:
229 case Shader::OPCODE_INTBITSTOFLOAT:
230 case Shader::OPCODE_UINTBITSTOFLOAT: d = s0; break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400231 case Shader::OPCODE_POWX: powx(d, s0, s1, pp); break;
232 case Shader::OPCODE_POW: pow(d, s0, s1, pp); break;
233 case Shader::OPCODE_SGN: sgn(d, s0); break;
234 case Shader::OPCODE_CRS: crs(d, s0, s1); break;
235 case Shader::OPCODE_FORWARD1: forward1(d, s0, s1, s2); break;
236 case Shader::OPCODE_FORWARD2: forward2(d, s0, s1, s2); break;
237 case Shader::OPCODE_FORWARD3: forward3(d, s0, s1, s2); break;
238 case Shader::OPCODE_FORWARD4: forward4(d, s0, s1, s2); break;
239 case Shader::OPCODE_REFLECT1: reflect1(d, s0, s1); break;
240 case Shader::OPCODE_REFLECT2: reflect2(d, s0, s1); break;
241 case Shader::OPCODE_REFLECT3: reflect3(d, s0, s1); break;
242 case Shader::OPCODE_REFLECT4: reflect4(d, s0, s1); break;
243 case Shader::OPCODE_REFRACT1: refract1(d, s0, s1, s2.x); break;
244 case Shader::OPCODE_REFRACT2: refract2(d, s0, s1, s2.x); break;
245 case Shader::OPCODE_REFRACT3: refract3(d, s0, s1, s2.x); break;
246 case Shader::OPCODE_REFRACT4: refract4(d, s0, s1, s2.x); break;
247 case Shader::OPCODE_NRM2: nrm2(d, s0, pp); break;
248 case Shader::OPCODE_NRM3: nrm3(d, s0, pp); break;
249 case Shader::OPCODE_NRM4: nrm4(d, s0, pp); break;
250 case Shader::OPCODE_ABS: abs(d, s0); break;
251 case Shader::OPCODE_SINCOS: sincos(d, s0, pp); break;
252 case Shader::OPCODE_COS: cos(d, s0, pp); break;
253 case Shader::OPCODE_SIN: sin(d, s0, pp); break;
254 case Shader::OPCODE_TAN: tan(d, s0, pp); break;
255 case Shader::OPCODE_ACOS: acos(d, s0, pp); break;
256 case Shader::OPCODE_ASIN: asin(d, s0, pp); break;
257 case Shader::OPCODE_ATAN: atan(d, s0, pp); break;
258 case Shader::OPCODE_ATAN2: atan2(d, s0, s1, pp); break;
259 case Shader::OPCODE_COSH: cosh(d, s0, pp); break;
260 case Shader::OPCODE_SINH: sinh(d, s0, pp); break;
261 case Shader::OPCODE_TANH: tanh(d, s0, pp); break;
262 case Shader::OPCODE_ACOSH: acosh(d, s0, pp); break;
263 case Shader::OPCODE_ASINH: asinh(d, s0, pp); break;
264 case Shader::OPCODE_ATANH: atanh(d, s0, pp); break;
Nicolas Capens4f172c72016-01-13 08:34:30 -0500265 case Shader::OPCODE_M4X4: M4X4(d, s0, src1); break;
266 case Shader::OPCODE_M4X3: M4X3(d, s0, src1); break;
267 case Shader::OPCODE_M3X4: M3X4(d, s0, src1); break;
268 case Shader::OPCODE_M3X3: M3X3(d, s0, src1); break;
269 case Shader::OPCODE_M3X2: M3X2(d, s0, src1); break;
270 case Shader::OPCODE_TEX: TEXLD(d, s0, src1, project, bias); break;
271 case Shader::OPCODE_TEXLDD: TEXLDD(d, s0, src1, s2, s3, project, bias); break;
272 case Shader::OPCODE_TEXLDL: TEXLDL(d, s0, src1, project, bias); break;
273 case Shader::OPCODE_TEXSIZE: TEXSIZE(d, s0.x, src1); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400274 case Shader::OPCODE_TEXKILL: TEXKILL(cMask, d, dst.mask); break;
Nicolas Capens4f172c72016-01-13 08:34:30 -0500275 case Shader::OPCODE_TEXOFFSET: TEXOFFSET(d, s0, src1, s2, s3, project, bias); break;
276 case Shader::OPCODE_TEXLDLOFFSET: TEXLDL(d, s0, src1, s2, project, bias); break;
277 case Shader::OPCODE_TEXELFETCH: TEXELFETCH(d, s0, src1, s2); break;
278 case Shader::OPCODE_TEXELFETCHOFFSET: TEXELFETCH(d, s0, src1, s2, s3); break;
279 case Shader::OPCODE_TEXGRAD: TEXGRAD(d, s0, src1, s2, s3); break;
280 case Shader::OPCODE_TEXGRADOFFSET: TEXGRAD(d, s0, src1, s2, s3, s4); break;
281 case Shader::OPCODE_DISCARD: DISCARD(cMask, instruction); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400282 case Shader::OPCODE_DFDX: DFDX(d, s0); break;
283 case Shader::OPCODE_DFDY: DFDY(d, s0); break;
284 case Shader::OPCODE_FWIDTH: FWIDTH(d, s0); break;
Nicolas Capens4f172c72016-01-13 08:34:30 -0500285 case Shader::OPCODE_BREAK: BREAK(); break;
286 case Shader::OPCODE_BREAKC: BREAKC(s0, s1, control); break;
287 case Shader::OPCODE_BREAKP: BREAKP(src0); break;
288 case Shader::OPCODE_CONTINUE: CONTINUE(); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400289 case Shader::OPCODE_TEST: TEST(); break;
Nicolas Capens4f172c72016-01-13 08:34:30 -0500290 case Shader::OPCODE_CALL: CALL(dst.label, dst.callSite); break;
291 case Shader::OPCODE_CALLNZ: CALLNZ(dst.label, dst.callSite, src0); break;
292 case Shader::OPCODE_ELSE: ELSE(); break;
293 case Shader::OPCODE_ENDIF: ENDIF(); break;
294 case Shader::OPCODE_ENDLOOP: ENDLOOP(); break;
295 case Shader::OPCODE_ENDREP: ENDREP(); break;
296 case Shader::OPCODE_ENDWHILE: ENDWHILE(); break;
297 case Shader::OPCODE_IF: IF(src0); break;
298 case Shader::OPCODE_IFC: IFC(s0, s1, control); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400299 case Shader::OPCODE_LABEL: LABEL(dst.index); break;
Nicolas Capens4f172c72016-01-13 08:34:30 -0500300 case Shader::OPCODE_LOOP: LOOP(src1); break;
301 case Shader::OPCODE_REP: REP(src0); break;
302 case Shader::OPCODE_WHILE: WHILE(src0); break;
303 case Shader::OPCODE_RET: RET(); break;
304 case Shader::OPCODE_LEAVE: LEAVE(); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400305 case Shader::OPCODE_CMP: cmp(d, s0, s1, control); break;
306 case Shader::OPCODE_ALL: all(d.x, s0); break;
307 case Shader::OPCODE_ANY: any(d.x, s0); break;
308 case Shader::OPCODE_NOT: not(d, s0); break;
Alexis Hetuc4f2c292015-08-18 15:43:09 -0400309 case Shader::OPCODE_OR: or(d, s0, s1); break;
310 case Shader::OPCODE_XOR: xor(d, s0, s1); break;
311 case Shader::OPCODE_AND: and(d, s0, s1); break;
Alexis Hetu8d78cf72015-08-28 14:24:45 -0400312 case Shader::OPCODE_EQ: equal(d, s0, s1); break;
313 case Shader::OPCODE_NE: notEqual(d, s0, s1); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400314 case Shader::OPCODE_END: break;
315 default:
316 ASSERT(false);
317 }
318
319 if(dst.type != Shader::PARAMETER_VOID && dst.type != Shader::PARAMETER_LABEL && opcode != Shader::OPCODE_TEXKILL && opcode != Shader::OPCODE_NOP)
320 {
321 if(dst.integer)
322 {
323 switch(opcode)
324 {
325 case Shader::OPCODE_DIV:
326 if(dst.x) d.x = Trunc(d.x);
327 if(dst.y) d.y = Trunc(d.y);
328 if(dst.z) d.z = Trunc(d.z);
329 if(dst.w) d.w = Trunc(d.w);
330 break;
331 default:
332 break; // No truncation to integer required when arguments are integer
333 }
334 }
335
336 if(dst.saturate)
337 {
338 if(dst.x) d.x = Max(d.x, Float4(0.0f));
339 if(dst.y) d.y = Max(d.y, Float4(0.0f));
340 if(dst.z) d.z = Max(d.z, Float4(0.0f));
341 if(dst.w) d.w = Max(d.w, Float4(0.0f));
342
343 if(dst.x) d.x = Min(d.x, Float4(1.0f));
344 if(dst.y) d.y = Min(d.y, Float4(1.0f));
345 if(dst.z) d.z = Min(d.z, Float4(1.0f));
346 if(dst.w) d.w = Min(d.w, Float4(1.0f));
347 }
348
349 if(instruction->isPredicated())
350 {
351 Vector4f pDst; // FIXME: Rename
352
353 switch(dst.type)
354 {
355 case Shader::PARAMETER_TEMP:
356 if(dst.rel.type == Shader::PARAMETER_VOID)
357 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500358 if(dst.x) pDst.x = r[dst.index].x;
359 if(dst.y) pDst.y = r[dst.index].y;
360 if(dst.z) pDst.z = r[dst.index].z;
361 if(dst.w) pDst.w = r[dst.index].w;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400362 }
363 else
364 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500365 Int a = relativeAddress(dst);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400366
Nicolas Capens4f172c72016-01-13 08:34:30 -0500367 if(dst.x) pDst.x = r[dst.index + a].x;
368 if(dst.y) pDst.y = r[dst.index + a].y;
369 if(dst.z) pDst.z = r[dst.index + a].z;
370 if(dst.w) pDst.w = r[dst.index + a].w;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400371 }
372 break;
373 case Shader::PARAMETER_COLOROUT:
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500374 if(dst.rel.type == Shader::PARAMETER_VOID)
375 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500376 if(dst.x) pDst.x = oC[dst.index].x;
377 if(dst.y) pDst.y = oC[dst.index].y;
378 if(dst.z) pDst.z = oC[dst.index].z;
379 if(dst.w) pDst.w = oC[dst.index].w;
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500380 }
381 else
382 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500383 Int a = relativeAddress(dst) + dst.index;
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500384
Nicolas Capens4f172c72016-01-13 08:34:30 -0500385 if(dst.x) pDst.x = oC[a].x;
386 if(dst.y) pDst.y = oC[a].y;
387 if(dst.z) pDst.z = oC[a].z;
388 if(dst.w) pDst.w = oC[a].w;
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500389 }
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400390 break;
391 case Shader::PARAMETER_PREDICATE:
Nicolas Capens4f172c72016-01-13 08:34:30 -0500392 if(dst.x) pDst.x = p0.x;
393 if(dst.y) pDst.y = p0.y;
394 if(dst.z) pDst.z = p0.z;
395 if(dst.w) pDst.w = p0.w;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400396 break;
397 case Shader::PARAMETER_DEPTHOUT:
Nicolas Capens4f172c72016-01-13 08:34:30 -0500398 pDst.x = oDepth;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400399 break;
400 default:
401 ASSERT(false);
402 }
403
Nicolas Capens4f172c72016-01-13 08:34:30 -0500404 Int4 enable = enableMask(instruction);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400405
406 Int4 xEnable = enable;
407 Int4 yEnable = enable;
408 Int4 zEnable = enable;
409 Int4 wEnable = enable;
410
411 if(predicate)
412 {
413 unsigned char pSwizzle = instruction->predicateSwizzle;
414
Nicolas Capens4f172c72016-01-13 08:34:30 -0500415 Float4 xPredicate = p0[(pSwizzle >> 0) & 0x03];
416 Float4 yPredicate = p0[(pSwizzle >> 2) & 0x03];
417 Float4 zPredicate = p0[(pSwizzle >> 4) & 0x03];
418 Float4 wPredicate = p0[(pSwizzle >> 6) & 0x03];
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400419
420 if(!instruction->predicateNot)
421 {
422 if(dst.x) xEnable = xEnable & As<Int4>(xPredicate);
423 if(dst.y) yEnable = yEnable & As<Int4>(yPredicate);
424 if(dst.z) zEnable = zEnable & As<Int4>(zPredicate);
425 if(dst.w) wEnable = wEnable & As<Int4>(wPredicate);
426 }
427 else
428 {
429 if(dst.x) xEnable = xEnable & ~As<Int4>(xPredicate);
430 if(dst.y) yEnable = yEnable & ~As<Int4>(yPredicate);
431 if(dst.z) zEnable = zEnable & ~As<Int4>(zPredicate);
432 if(dst.w) wEnable = wEnable & ~As<Int4>(wPredicate);
433 }
434 }
435
436 if(dst.x) d.x = As<Float4>(As<Int4>(d.x) & xEnable);
437 if(dst.y) d.y = As<Float4>(As<Int4>(d.y) & yEnable);
438 if(dst.z) d.z = As<Float4>(As<Int4>(d.z) & zEnable);
439 if(dst.w) d.w = As<Float4>(As<Int4>(d.w) & wEnable);
440
441 if(dst.x) d.x = As<Float4>(As<Int4>(d.x) | (As<Int4>(pDst.x) & ~xEnable));
442 if(dst.y) d.y = As<Float4>(As<Int4>(d.y) | (As<Int4>(pDst.y) & ~yEnable));
443 if(dst.z) d.z = As<Float4>(As<Int4>(d.z) | (As<Int4>(pDst.z) & ~zEnable));
444 if(dst.w) d.w = As<Float4>(As<Int4>(d.w) | (As<Int4>(pDst.w) & ~wEnable));
445 }
446
447 switch(dst.type)
448 {
449 case Shader::PARAMETER_TEMP:
450 if(dst.rel.type == Shader::PARAMETER_VOID)
451 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500452 if(dst.x) r[dst.index].x = d.x;
453 if(dst.y) r[dst.index].y = d.y;
454 if(dst.z) r[dst.index].z = d.z;
455 if(dst.w) r[dst.index].w = d.w;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400456 }
457 else
458 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500459 Int a = relativeAddress(dst);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400460
Nicolas Capens4f172c72016-01-13 08:34:30 -0500461 if(dst.x) r[dst.index + a].x = d.x;
462 if(dst.y) r[dst.index + a].y = d.y;
463 if(dst.z) r[dst.index + a].z = d.z;
464 if(dst.w) r[dst.index + a].w = d.w;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400465 }
466 break;
467 case Shader::PARAMETER_COLOROUT:
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500468 if(dst.rel.type == Shader::PARAMETER_VOID)
469 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500470 if(dst.x) { oC[dst.index].x = d.x; }
471 if(dst.y) { oC[dst.index].y = d.y; }
472 if(dst.z) { oC[dst.index].z = d.z; }
473 if(dst.w) { oC[dst.index].w = d.w; }
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500474 }
475 else
476 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500477 Int a = relativeAddress(dst) + dst.index;
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500478
Nicolas Capens4f172c72016-01-13 08:34:30 -0500479 if(dst.x) { oC[a].x = d.x; }
480 if(dst.y) { oC[a].y = d.y; }
481 if(dst.z) { oC[a].z = d.z; }
482 if(dst.w) { oC[a].w = d.w; }
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500483 }
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400484 break;
485 case Shader::PARAMETER_PREDICATE:
Nicolas Capens4f172c72016-01-13 08:34:30 -0500486 if(dst.x) p0.x = d.x;
487 if(dst.y) p0.y = d.y;
488 if(dst.z) p0.z = d.z;
489 if(dst.w) p0.w = d.w;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400490 break;
491 case Shader::PARAMETER_DEPTHOUT:
Nicolas Capens4f172c72016-01-13 08:34:30 -0500492 oDepth = d.x;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400493 break;
494 default:
495 ASSERT(false);
496 }
497 }
498 }
499
500 if(currentLabel != -1)
501 {
502 Nucleus::setInsertBlock(returnBlock);
503 }
504
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500505 for(int i = 0; i < RENDERTARGETS; i++)
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400506 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500507 c[i] = oC[i];
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400508 }
509 }
Nicolas Capens5d961882016-01-01 23:18:14 -0500510
Nicolas Capens4f172c72016-01-13 08:34:30 -0500511 Bool PixelProgram::alphaTest(Int cMask[4])
Nicolas Capens5d961882016-01-01 23:18:14 -0500512 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500513 clampColor(c);
Nicolas Capens5d961882016-01-01 23:18:14 -0500514
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400515 if(!state.alphaTestActive())
516 {
517 return true;
518 }
519
520 Int aMask;
521
522 if(state.transparencyAntialiasing == TRANSPARENCY_NONE)
523 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500524 Short4 alpha = RoundShort4(c[0].w * Float4(0x1000));
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400525
Nicolas Capens4f172c72016-01-13 08:34:30 -0500526 PixelRoutine::alphaTest(aMask, alpha);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400527
528 for(unsigned int q = 0; q < state.multiSample; q++)
529 {
530 cMask[q] &= aMask;
531 }
532 }
533 else if(state.transparencyAntialiasing == TRANSPARENCY_ALPHA_TO_COVERAGE)
534 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500535 alphaToCoverage(cMask, c[0].w);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400536 }
537 else ASSERT(false);
538
539 Int pass = cMask[0];
540
541 for(unsigned int q = 1; q < state.multiSample; q++)
542 {
543 pass = pass | cMask[q];
544 }
545
Nicolas Capens5d961882016-01-01 23:18:14 -0500546 return pass != 0x0;
547 }
548
Nicolas Capens4f172c72016-01-13 08:34:30 -0500549 void PixelProgram::rasterOperation(Float4 &fog, Pointer<Byte> cBuffer[4], Int &x, Int sMask[4], Int zMask[4], Int cMask[4])
Nicolas Capens5d961882016-01-01 23:18:14 -0500550 {
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500551 for(int index = 0; index < RENDERTARGETS; index++)
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400552 {
553 if(!state.colorWriteActive(index))
554 {
555 continue;
Nicolas Capens5d961882016-01-01 23:18:14 -0500556 }
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400557
558 if(!postBlendSRGB && state.writeSRGB)
559 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500560 c[index].x = linearToSRGB(c[index].x);
561 c[index].y = linearToSRGB(c[index].y);
562 c[index].z = linearToSRGB(c[index].z);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400563 }
564
565 if(index == 0)
566 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500567 fogBlend(c[index], fog);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400568 }
569
570 switch(state.targetFormat[index])
571 {
572 case FORMAT_R5G6B5:
573 case FORMAT_X8R8G8B8:
574 case FORMAT_X8B8G8R8:
575 case FORMAT_A8R8G8B8:
576 case FORMAT_A8B8G8R8:
577 case FORMAT_A8:
578 case FORMAT_G16R16:
579 case FORMAT_A16B16G16R16:
580 for(unsigned int q = 0; q < state.multiSample; q++)
581 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500582 Pointer<Byte> buffer = cBuffer[index] + q * *Pointer<Int>(data + OFFSET(DrawData, colorSliceB[index]));
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400583 Vector4s color;
584
Nicolas Capens4f172c72016-01-13 08:34:30 -0500585 color.x = convertFixed16(c[index].x, false);
586 color.y = convertFixed16(c[index].y, false);
587 color.z = convertFixed16(c[index].z, false);
588 color.w = convertFixed16(c[index].w, false);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400589
590 if(state.multiSampleMask & (1 << q))
591 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500592 alphaBlend(index, buffer, color, x);
593 logicOperation(index, buffer, color, x);
594 writeColor(index, buffer, x, color, sMask[q], zMask[q], cMask[q]);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400595 }
596 }
597 break;
598 case FORMAT_R32F:
599 case FORMAT_G32R32F:
600 case FORMAT_A32B32G32R32F:
601 for(unsigned int q = 0; q < state.multiSample; q++)
602 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500603 Pointer<Byte> buffer = cBuffer[index] + q * *Pointer<Int>(data + OFFSET(DrawData, colorSliceB[index]));
604 Vector4f color = c[index];
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400605
606 if(state.multiSampleMask & (1 << q))
607 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500608 alphaBlend(index, buffer, color, x);
609 writeColor(index, buffer, x, color, sMask[q], zMask[q], cMask[q]);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400610 }
611 }
612 break;
613 default:
614 ASSERT(false);
615 }
616 }
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500617 }
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400618
Nicolas Capens4f172c72016-01-13 08:34:30 -0500619 void PixelProgram::sampleTexture(Vector4f &c, const Src &sampler, Float4 &u, Float4 &v, Float4 &w, Float4 &q, Vector4f &dsx, Vector4f &dsy, bool project, bool bias, bool gradients, bool lodProvided)
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400620 {
621 if(sampler.type == Shader::PARAMETER_SAMPLER && sampler.rel.type == Shader::PARAMETER_VOID)
622 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500623 sampleTexture(c, sampler.index, u, v, w, q, dsx, dsy, project, bias, gradients, lodProvided);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400624 }
625 else
626 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500627 Int index = As<Int>(Float(fetchRegisterF(sampler).x.x));
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400628
629 for(int i = 0; i < TEXTURE_IMAGE_UNITS; i++)
630 {
631 if(shader->usesSampler(i))
632 {
633 If(index == i)
634 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500635 sampleTexture(c, i, u, v, w, q, dsx, dsy, project, bias, gradients, lodProvided);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400636 // FIXME: When the sampler states are the same, we could use one sampler and just index the texture
637 }
638 }
639 }
640 }
641 }
642
Nicolas Capens4f172c72016-01-13 08:34:30 -0500643 void PixelProgram::sampleTexture(Vector4f &c, int stage, Float4 &u, Float4 &v, Float4 &w, Float4 &q, Vector4f &dsx, Vector4f &dsy, bool project, bool bias, bool gradients, bool lodProvided)
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400644 {
645#if PERF_PROFILE
646 Long texTime = Ticks();
647#endif
648
Nicolas Capens4f172c72016-01-13 08:34:30 -0500649 Pointer<Byte> texture = data + OFFSET(DrawData, mipmap) + stage * sizeof(Texture);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400650
651 if(!project)
652 {
653 sampler[stage]->sampleTexture(texture, c, u, v, w, q, dsx, dsy, bias, gradients, lodProvided);
654 }
655 else
656 {
657 Float4 rq = reciprocal(q);
658
659 Float4 u_q = u * rq;
660 Float4 v_q = v * rq;
661 Float4 w_q = w * rq;
662
663 sampler[stage]->sampleTexture(texture, c, u_q, v_q, w_q, q, dsx, dsy, bias, gradients, lodProvided);
664 }
665
666#if PERF_PROFILE
Nicolas Capens4f172c72016-01-13 08:34:30 -0500667 cycles[PERF_TEX] += Ticks() - texTime;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400668#endif
669 }
670
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500671 void PixelProgram::clampColor(Vector4f oC[RENDERTARGETS])
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400672 {
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500673 for(int index = 0; index < RENDERTARGETS; index++)
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400674 {
675 if(!state.colorWriteActive(index) && !(index == 0 && state.alphaTestActive()))
676 {
677 continue;
678 }
679
680 switch(state.targetFormat[index])
681 {
682 case FORMAT_NULL:
683 break;
684 case FORMAT_R5G6B5:
685 case FORMAT_A8R8G8B8:
686 case FORMAT_A8B8G8R8:
687 case FORMAT_X8R8G8B8:
688 case FORMAT_X8B8G8R8:
689 case FORMAT_A8:
690 case FORMAT_G16R16:
691 case FORMAT_A16B16G16R16:
692 oC[index].x = Max(oC[index].x, Float4(0.0f)); oC[index].x = Min(oC[index].x, Float4(1.0f));
693 oC[index].y = Max(oC[index].y, Float4(0.0f)); oC[index].y = Min(oC[index].y, Float4(1.0f));
694 oC[index].z = Max(oC[index].z, Float4(0.0f)); oC[index].z = Min(oC[index].z, Float4(1.0f));
695 oC[index].w = Max(oC[index].w, Float4(0.0f)); oC[index].w = Min(oC[index].w, Float4(1.0f));
696 break;
697 case FORMAT_R32F:
698 case FORMAT_G32R32F:
699 case FORMAT_A32B32G32R32F:
700 break;
701 default:
702 ASSERT(false);
703 }
704 }
705 }
706
Nicolas Capens4f172c72016-01-13 08:34:30 -0500707 Int4 PixelProgram::enableMask(const Shader::Instruction *instruction)
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400708 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500709 Int4 enable = instruction->analysisBranch ? Int4(enableStack[enableIndex]) : Int4(0xFFFFFFFF);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400710
711 if(!whileTest)
712 {
713 if(shader->containsBreakInstruction() && instruction->analysisBreak)
714 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500715 enable &= enableBreak;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400716 }
717
718 if(shader->containsContinueInstruction() && instruction->analysisContinue)
719 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500720 enable &= enableContinue;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400721 }
722
723 if(shader->containsLeaveInstruction() && instruction->analysisLeave)
724 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500725 enable &= enableLeave;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400726 }
727 }
728
729 return enable;
730 }
731
Nicolas Capens4f172c72016-01-13 08:34:30 -0500732 Vector4f PixelProgram::fetchRegisterF(const Src &src, unsigned int offset)
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400733 {
734 Vector4f reg;
Nicolas Capens5d961882016-01-01 23:18:14 -0500735 unsigned int i = src.index + offset;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400736
737 switch(src.type)
738 {
739 case Shader::PARAMETER_TEMP:
740 if(src.rel.type == Shader::PARAMETER_VOID)
741 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500742 reg = r[i];
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400743 }
744 else
745 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500746 Int a = relativeAddress(src);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400747
Nicolas Capens4f172c72016-01-13 08:34:30 -0500748 reg = r[i + a];
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400749 }
750 break;
751 case Shader::PARAMETER_INPUT:
752 {
753 if(src.rel.type == Shader::PARAMETER_VOID) // Not relative
754 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500755 reg = v[i];
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400756 }
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400757 else
758 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500759 Int a = relativeAddress(src);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400760
Nicolas Capens4f172c72016-01-13 08:34:30 -0500761 reg = v[i + a];
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400762 }
763 }
764 break;
765 case Shader::PARAMETER_CONST:
Nicolas Capens4f172c72016-01-13 08:34:30 -0500766 reg = readConstant(src, offset);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400767 break;
768 case Shader::PARAMETER_TEXTURE:
Nicolas Capens4f172c72016-01-13 08:34:30 -0500769 reg = v[2 + i];
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400770 break;
771 case Shader::PARAMETER_MISCTYPE:
Nicolas Capens4f172c72016-01-13 08:34:30 -0500772 if(src.index == 0) reg = vPos;
773 if(src.index == 1) reg = vFace;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400774 break;
775 case Shader::PARAMETER_SAMPLER:
776 if(src.rel.type == Shader::PARAMETER_VOID)
777 {
778 reg.x = As<Float4>(Int4(i));
779 }
780 else if(src.rel.type == Shader::PARAMETER_TEMP)
781 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500782 reg.x = As<Float4>(Int4(i) + As<Int4>(r[src.rel.index].x));
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400783 }
784 return reg;
785 case Shader::PARAMETER_PREDICATE: return reg; // Dummy
786 case Shader::PARAMETER_VOID: return reg; // Dummy
787 case Shader::PARAMETER_FLOAT4LITERAL:
788 reg.x = Float4(src.value[0]);
789 reg.y = Float4(src.value[1]);
790 reg.z = Float4(src.value[2]);
791 reg.w = Float4(src.value[3]);
792 break;
793 case Shader::PARAMETER_CONSTINT: return reg; // Dummy
794 case Shader::PARAMETER_CONSTBOOL: return reg; // Dummy
795 case Shader::PARAMETER_LOOP: return reg; // Dummy
796 case Shader::PARAMETER_COLOROUT:
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500797 if(src.rel.type == Shader::PARAMETER_VOID) // Not relative
798 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500799 reg = oC[i];
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500800 }
801 else
802 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500803 Int a = relativeAddress(src);
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500804
Nicolas Capens4f172c72016-01-13 08:34:30 -0500805 reg = oC[i + a];
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500806 }
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400807 break;
808 case Shader::PARAMETER_DEPTHOUT:
Nicolas Capens4f172c72016-01-13 08:34:30 -0500809 reg.x = oDepth;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400810 break;
811 default:
812 ASSERT(false);
813 }
814
815 const Float4 &x = reg[(src.swizzle >> 0) & 0x3];
816 const Float4 &y = reg[(src.swizzle >> 2) & 0x3];
817 const Float4 &z = reg[(src.swizzle >> 4) & 0x3];
818 const Float4 &w = reg[(src.swizzle >> 6) & 0x3];
819
820 Vector4f mod;
821
822 switch(src.modifier)
823 {
824 case Shader::MODIFIER_NONE:
825 mod.x = x;
826 mod.y = y;
827 mod.z = z;
828 mod.w = w;
829 break;
830 case Shader::MODIFIER_NEGATE:
831 mod.x = -x;
832 mod.y = -y;
833 mod.z = -z;
834 mod.w = -w;
835 break;
836 case Shader::MODIFIER_ABS:
837 mod.x = Abs(x);
838 mod.y = Abs(y);
839 mod.z = Abs(z);
840 mod.w = Abs(w);
841 break;
842 case Shader::MODIFIER_ABS_NEGATE:
843 mod.x = -Abs(x);
844 mod.y = -Abs(y);
845 mod.z = -Abs(z);
846 mod.w = -Abs(w);
847 break;
848 case Shader::MODIFIER_NOT:
849 mod.x = As<Float4>(As<Int4>(x) ^ Int4(0xFFFFFFFF));
850 mod.y = As<Float4>(As<Int4>(y) ^ Int4(0xFFFFFFFF));
851 mod.z = As<Float4>(As<Int4>(z) ^ Int4(0xFFFFFFFF));
852 mod.w = As<Float4>(As<Int4>(w) ^ Int4(0xFFFFFFFF));
853 break;
854 default:
855 ASSERT(false);
856 }
857
858 return mod;
859 }
860
Nicolas Capens4f172c72016-01-13 08:34:30 -0500861 Vector4f PixelProgram::readConstant(const Src &src, unsigned int offset)
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400862 {
863 Vector4f c;
Nicolas Capens5d961882016-01-01 23:18:14 -0500864 unsigned int i = src.index + offset;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400865
866 if(src.rel.type == Shader::PARAMETER_VOID) // Not relative
867 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500868 c.x = c.y = c.z = c.w = *Pointer<Float4>(data + OFFSET(DrawData, ps.c[i]));
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400869
870 c.x = c.x.xxxx;
871 c.y = c.y.yyyy;
872 c.z = c.z.zzzz;
873 c.w = c.w.wwww;
874
875 if(shader->containsDefineInstruction()) // Constant may be known at compile time
876 {
877 for(size_t j = 0; j < shader->getLength(); j++)
878 {
879 const Shader::Instruction &instruction = *shader->getInstruction(j);
880
881 if(instruction.opcode == Shader::OPCODE_DEF)
882 {
883 if(instruction.dst.index == i)
884 {
885 c.x = Float4(instruction.src[0].value[0]);
886 c.y = Float4(instruction.src[0].value[1]);
887 c.z = Float4(instruction.src[0].value[2]);
888 c.w = Float4(instruction.src[0].value[3]);
889
890 break;
891 }
892 }
893 }
894 }
895 }
896 else if(src.rel.type == Shader::PARAMETER_LOOP)
897 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500898 Int loopCounter = aL[loopDepth];
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400899
Nicolas Capens4f172c72016-01-13 08:34:30 -0500900 c.x = c.y = c.z = c.w = *Pointer<Float4>(data + OFFSET(DrawData, ps.c[i]) + loopCounter * 16);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400901
902 c.x = c.x.xxxx;
903 c.y = c.y.yyyy;
904 c.z = c.z.zzzz;
905 c.w = c.w.wwww;
906 }
907 else
908 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500909 Int a = relativeAddress(src);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400910
Nicolas Capens4f172c72016-01-13 08:34:30 -0500911 c.x = c.y = c.z = c.w = *Pointer<Float4>(data + OFFSET(DrawData, ps.c[i]) + a * 16);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400912
913 c.x = c.x.xxxx;
914 c.y = c.y.yyyy;
915 c.z = c.z.zzzz;
916 c.w = c.w.wwww;
917 }
918
919 return c;
920 }
921
Nicolas Capens4f172c72016-01-13 08:34:30 -0500922 Int PixelProgram::relativeAddress(const Shader::Parameter &var)
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400923 {
924 ASSERT(var.rel.deterministic);
925
926 if(var.rel.type == Shader::PARAMETER_TEMP)
927 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500928 return As<Int>(Extract(r[var.rel.index].x, 0)) * var.rel.scale;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400929 }
930 else if(var.rel.type == Shader::PARAMETER_INPUT)
931 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500932 return As<Int>(Extract(v[var.rel.index].x, 0)) * var.rel.scale;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400933 }
934 else if(var.rel.type == Shader::PARAMETER_OUTPUT)
935 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500936 return As<Int>(Extract(oC[var.rel.index].x, 0)) * var.rel.scale;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400937 }
938 else if(var.rel.type == Shader::PARAMETER_CONST)
939 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500940 RValue<Int4> c = *Pointer<Int4>(data + OFFSET(DrawData, ps.c[var.rel.index]));
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400941
Alexis Hetu02a2bb82015-08-20 14:10:33 -0400942 return Extract(c, 0) * var.rel.scale;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400943 }
Nicolas Capens907700d2016-01-20 17:09:28 -0500944 else if(var.rel.type == Shader::PARAMETER_LOOP)
945 {
946 return aL[loopDepth];
947 }
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400948 else ASSERT(false);
949
950 return 0;
951 }
952
953 Float4 PixelProgram::linearToSRGB(const Float4 &x) // Approximates x^(1.0/2.2)
954 {
955 Float4 sqrtx = Rcp_pp(RcpSqrt_pp(x));
956 Float4 sRGB = sqrtx * Float4(1.14f) - x * Float4(0.14f);
957
958 return Min(Max(sRGB, Float4(0.0f)), Float4(1.0f));
959 }
960
Nicolas Capens4f172c72016-01-13 08:34:30 -0500961 void PixelProgram::M3X2(Vector4f &dst, Vector4f &src0, const Src &src1)
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400962 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500963 Vector4f row0 = fetchRegisterF(src1, 0);
964 Vector4f row1 = fetchRegisterF(src1, 1);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400965
966 dst.x = dot3(src0, row0);
967 dst.y = dot3(src0, row1);
968 }
969
Nicolas Capens4f172c72016-01-13 08:34:30 -0500970 void PixelProgram::M3X3(Vector4f &dst, Vector4f &src0, const Src &src1)
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400971 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500972 Vector4f row0 = fetchRegisterF(src1, 0);
973 Vector4f row1 = fetchRegisterF(src1, 1);
974 Vector4f row2 = fetchRegisterF(src1, 2);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400975
976 dst.x = dot3(src0, row0);
977 dst.y = dot3(src0, row1);
978 dst.z = dot3(src0, row2);
979 }
980
Nicolas Capens4f172c72016-01-13 08:34:30 -0500981 void PixelProgram::M3X4(Vector4f &dst, Vector4f &src0, const Src &src1)
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400982 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500983 Vector4f row0 = fetchRegisterF(src1, 0);
984 Vector4f row1 = fetchRegisterF(src1, 1);
985 Vector4f row2 = fetchRegisterF(src1, 2);
986 Vector4f row3 = fetchRegisterF(src1, 3);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400987
988 dst.x = dot3(src0, row0);
989 dst.y = dot3(src0, row1);
990 dst.z = dot3(src0, row2);
991 dst.w = dot3(src0, row3);
992 }
993
Nicolas Capens4f172c72016-01-13 08:34:30 -0500994 void PixelProgram::M4X3(Vector4f &dst, Vector4f &src0, const Src &src1)
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400995 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500996 Vector4f row0 = fetchRegisterF(src1, 0);
997 Vector4f row1 = fetchRegisterF(src1, 1);
998 Vector4f row2 = fetchRegisterF(src1, 2);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400999
1000 dst.x = dot4(src0, row0);
1001 dst.y = dot4(src0, row1);
1002 dst.z = dot4(src0, row2);
1003 }
1004
Nicolas Capens4f172c72016-01-13 08:34:30 -05001005 void PixelProgram::M4X4(Vector4f &dst, Vector4f &src0, const Src &src1)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001006 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001007 Vector4f row0 = fetchRegisterF(src1, 0);
1008 Vector4f row1 = fetchRegisterF(src1, 1);
1009 Vector4f row2 = fetchRegisterF(src1, 2);
1010 Vector4f row3 = fetchRegisterF(src1, 3);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001011
1012 dst.x = dot4(src0, row0);
1013 dst.y = dot4(src0, row1);
1014 dst.z = dot4(src0, row2);
1015 dst.w = dot4(src0, row3);
1016 }
1017
Nicolas Capens4f172c72016-01-13 08:34:30 -05001018 void PixelProgram::TEXLD(Vector4f &dst, Vector4f &src0, const Src &src1, bool project, bool bias)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001019 {
1020 Vector4f tmp;
Nicolas Capens4f172c72016-01-13 08:34:30 -05001021 sampleTexture(tmp, src1, src0.x, src0.y, src0.z, src0.w, src0, src0, project, bias);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001022
1023 dst.x = tmp[(src1.swizzle >> 0) & 0x3];
1024 dst.y = tmp[(src1.swizzle >> 2) & 0x3];
1025 dst.z = tmp[(src1.swizzle >> 4) & 0x3];
1026 dst.w = tmp[(src1.swizzle >> 6) & 0x3];
1027 }
1028
Nicolas Capens4f172c72016-01-13 08:34:30 -05001029 void PixelProgram::TEXOFFSET(Vector4f &dst, Vector4f &src0, const Src& src1, Vector4f &src2, Vector4f &src3, bool project, bool bias)
Alexis Hetu25d47fc2015-10-22 13:58:32 -04001030 {
1031 UNIMPLEMENTED();
1032 }
1033
Nicolas Capens4f172c72016-01-13 08:34:30 -05001034 void PixelProgram::TEXLDL(Vector4f &dst, Vector4f &src0, const Src &src1, Vector4f &offset, bool project, bool bias)
Alexis Hetu25d47fc2015-10-22 13:58:32 -04001035 {
1036 UNIMPLEMENTED();
1037 }
1038
Nicolas Capens4f172c72016-01-13 08:34:30 -05001039 void PixelProgram::TEXELFETCH(Vector4f &dst, Vector4f &src0, const Src& src1, Vector4f &src2)
Alexis Hetu25d47fc2015-10-22 13:58:32 -04001040 {
1041 UNIMPLEMENTED();
1042 }
1043
Nicolas Capens4f172c72016-01-13 08:34:30 -05001044 void PixelProgram::TEXELFETCH(Vector4f &dst, Vector4f &src0, const Src& src1, Vector4f &src2, Vector4f &offset)
Alexis Hetu25d47fc2015-10-22 13:58:32 -04001045 {
1046 UNIMPLEMENTED();
1047 }
1048
Nicolas Capens4f172c72016-01-13 08:34:30 -05001049 void PixelProgram::TEXGRAD(Vector4f &dst, Vector4f &src0, const Src& src1, Vector4f &src2, Vector4f &src3)
Alexis Hetu25d47fc2015-10-22 13:58:32 -04001050 {
1051 UNIMPLEMENTED();
1052 }
1053
Nicolas Capens4f172c72016-01-13 08:34:30 -05001054 void PixelProgram::TEXGRAD(Vector4f &dst, Vector4f &src0, const Src& src1, Vector4f &src2, Vector4f &src3, Vector4f &offset)
Alexis Hetu25d47fc2015-10-22 13:58:32 -04001055 {
1056 UNIMPLEMENTED();
1057 }
1058
Nicolas Capens4f172c72016-01-13 08:34:30 -05001059 void PixelProgram::TEXLDD(Vector4f &dst, Vector4f &src0, const Src &src1, Vector4f &src2, Vector4f &src3, bool project, bool bias)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001060 {
1061 Vector4f tmp;
Nicolas Capens4f172c72016-01-13 08:34:30 -05001062 sampleTexture(tmp, src1, src0.x, src0.y, src0.z, src0.w, src2, src3, project, bias, true);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001063
1064 dst.x = tmp[(src1.swizzle >> 0) & 0x3];
1065 dst.y = tmp[(src1.swizzle >> 2) & 0x3];
1066 dst.z = tmp[(src1.swizzle >> 4) & 0x3];
1067 dst.w = tmp[(src1.swizzle >> 6) & 0x3];
1068 }
1069
Nicolas Capens4f172c72016-01-13 08:34:30 -05001070 void PixelProgram::TEXLDL(Vector4f &dst, Vector4f &src0, const Src &src1, bool project, bool bias)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001071 {
1072 Vector4f tmp;
Nicolas Capens4f172c72016-01-13 08:34:30 -05001073 sampleTexture(tmp, src1, src0.x, src0.y, src0.z, src0.w, src0, src0, project, bias, false, true);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001074
1075 dst.x = tmp[(src1.swizzle >> 0) & 0x3];
1076 dst.y = tmp[(src1.swizzle >> 2) & 0x3];
1077 dst.z = tmp[(src1.swizzle >> 4) & 0x3];
1078 dst.w = tmp[(src1.swizzle >> 6) & 0x3];
1079 }
1080
Nicolas Capens4f172c72016-01-13 08:34:30 -05001081 void PixelProgram::TEXSIZE(Vector4f &dst, Float4 &lod, const Src &src1)
Alexis Hetu9bcb31d2015-07-22 17:03:26 -04001082 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001083 Pointer<Byte> textureMipmap = data + OFFSET(DrawData, mipmap) + src1.index * sizeof(Texture) + OFFSET(Texture, mipmap);
Alexis Hetu9bcb31d2015-07-22 17:03:26 -04001084 for(int i = 0; i < 4; ++i)
1085 {
1086 Pointer<Byte> mipmap = textureMipmap + (As<Int>(Extract(lod, i)) + Int(1)) * sizeof(Mipmap);
1087 dst.x = Insert(dst.x, As<Float>(Int(*Pointer<Short>(mipmap + OFFSET(Mipmap, width)))), i);
1088 dst.y = Insert(dst.y, As<Float>(Int(*Pointer<Short>(mipmap + OFFSET(Mipmap, height)))), i);
1089 dst.z = Insert(dst.z, As<Float>(Int(*Pointer<Short>(mipmap + OFFSET(Mipmap, depth)))), i);
1090 }
1091 }
1092
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001093 void PixelProgram::TEXKILL(Int cMask[4], Vector4f &src, unsigned char mask)
1094 {
1095 Int kill = -1;
1096
1097 if(mask & 0x1) kill &= SignMask(CmpNLT(src.x, Float4(0.0f)));
1098 if(mask & 0x2) kill &= SignMask(CmpNLT(src.y, Float4(0.0f)));
1099 if(mask & 0x4) kill &= SignMask(CmpNLT(src.z, Float4(0.0f)));
1100 if(mask & 0x8) kill &= SignMask(CmpNLT(src.w, Float4(0.0f)));
1101
1102 // FIXME: Dynamic branching affects TEXKILL?
1103 // if(shader->containsDynamicBranching())
1104 // {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001105 // kill = ~SignMask(enableMask());
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001106 // }
1107
1108 for(unsigned int q = 0; q < state.multiSample; q++)
1109 {
1110 cMask[q] &= kill;
1111 }
1112
1113 // FIXME: Branch to end of shader if all killed?
1114 }
1115
Nicolas Capens4f172c72016-01-13 08:34:30 -05001116 void PixelProgram::DISCARD(Int cMask[4], const Shader::Instruction *instruction)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001117 {
1118 Int kill = 0;
1119
1120 if(shader->containsDynamicBranching())
1121 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001122 kill = ~SignMask(enableMask(instruction));
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001123 }
1124
1125 for(unsigned int q = 0; q < state.multiSample; q++)
1126 {
1127 cMask[q] &= kill;
1128 }
1129
1130 // FIXME: Branch to end of shader if all killed?
1131 }
1132
1133 void PixelProgram::DFDX(Vector4f &dst, Vector4f &src)
1134 {
1135 dst.x = src.x.yyww - src.x.xxzz;
1136 dst.y = src.y.yyww - src.y.xxzz;
1137 dst.z = src.z.yyww - src.z.xxzz;
1138 dst.w = src.w.yyww - src.w.xxzz;
1139 }
1140
1141 void PixelProgram::DFDY(Vector4f &dst, Vector4f &src)
1142 {
1143 dst.x = src.x.zwzw - src.x.xyxy;
1144 dst.y = src.y.zwzw - src.y.xyxy;
1145 dst.z = src.z.zwzw - src.z.xyxy;
1146 dst.w = src.w.zwzw - src.w.xyxy;
1147 }
1148
1149 void PixelProgram::FWIDTH(Vector4f &dst, Vector4f &src)
1150 {
1151 // abs(dFdx(src)) + abs(dFdy(src));
1152 dst.x = Abs(src.x.yyww - src.x.xxzz) + Abs(src.x.zwzw - src.x.xyxy);
1153 dst.y = Abs(src.y.yyww - src.y.xxzz) + Abs(src.y.zwzw - src.y.xyxy);
1154 dst.z = Abs(src.z.yyww - src.z.xxzz) + Abs(src.z.zwzw - src.z.xyxy);
1155 dst.w = Abs(src.w.yyww - src.w.xxzz) + Abs(src.w.zwzw - src.w.xyxy);
1156 }
1157
Nicolas Capens4f172c72016-01-13 08:34:30 -05001158 void PixelProgram::BREAK()
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001159 {
1160 llvm::BasicBlock *deadBlock = Nucleus::createBasicBlock();
1161 llvm::BasicBlock *endBlock = loopRepEndBlock[loopRepDepth - 1];
1162
1163 if(breakDepth == 0)
1164 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001165 enableIndex = enableIndex - breakDepth;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001166 Nucleus::createBr(endBlock);
1167 }
1168 else
1169 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001170 enableBreak = enableBreak & ~enableStack[enableIndex];
1171 Bool allBreak = SignMask(enableBreak) == 0x0;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001172
Nicolas Capens4f172c72016-01-13 08:34:30 -05001173 enableIndex = enableIndex - breakDepth;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001174 branch(allBreak, endBlock, deadBlock);
1175 }
1176
1177 Nucleus::setInsertBlock(deadBlock);
Nicolas Capens4f172c72016-01-13 08:34:30 -05001178 enableIndex = enableIndex + breakDepth;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001179 }
1180
Nicolas Capens4f172c72016-01-13 08:34:30 -05001181 void PixelProgram::BREAKC(Vector4f &src0, Vector4f &src1, Control control)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001182 {
1183 Int4 condition;
1184
1185 switch(control)
1186 {
1187 case Shader::CONTROL_GT: condition = CmpNLE(src0.x, src1.x); break;
1188 case Shader::CONTROL_EQ: condition = CmpEQ(src0.x, src1.x); break;
1189 case Shader::CONTROL_GE: condition = CmpNLT(src0.x, src1.x); break;
1190 case Shader::CONTROL_LT: condition = CmpLT(src0.x, src1.x); break;
1191 case Shader::CONTROL_NE: condition = CmpNEQ(src0.x, src1.x); break;
1192 case Shader::CONTROL_LE: condition = CmpLE(src0.x, src1.x); break;
1193 default:
1194 ASSERT(false);
1195 }
1196
Nicolas Capens4f172c72016-01-13 08:34:30 -05001197 BREAK(condition);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001198 }
1199
Nicolas Capens4f172c72016-01-13 08:34:30 -05001200 void PixelProgram::BREAKP(const Src &predicateRegister) // FIXME: Factor out parts common with BREAKC
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001201 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001202 Int4 condition = As<Int4>(p0[predicateRegister.swizzle & 0x3]);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001203
1204 if(predicateRegister.modifier == Shader::MODIFIER_NOT)
1205 {
1206 condition = ~condition;
1207 }
1208
Nicolas Capens4f172c72016-01-13 08:34:30 -05001209 BREAK(condition);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001210 }
1211
Nicolas Capens4f172c72016-01-13 08:34:30 -05001212 void PixelProgram::BREAK(Int4 &condition)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001213 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001214 condition &= enableStack[enableIndex];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001215
1216 llvm::BasicBlock *continueBlock = Nucleus::createBasicBlock();
1217 llvm::BasicBlock *endBlock = loopRepEndBlock[loopRepDepth - 1];
1218
Nicolas Capens4f172c72016-01-13 08:34:30 -05001219 enableBreak = enableBreak & ~condition;
1220 Bool allBreak = SignMask(enableBreak) == 0x0;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001221
Nicolas Capens4f172c72016-01-13 08:34:30 -05001222 enableIndex = enableIndex - breakDepth;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001223 branch(allBreak, endBlock, continueBlock);
1224
1225 Nucleus::setInsertBlock(continueBlock);
Nicolas Capens4f172c72016-01-13 08:34:30 -05001226 enableIndex = enableIndex + breakDepth;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001227 }
1228
Nicolas Capens4f172c72016-01-13 08:34:30 -05001229 void PixelProgram::CONTINUE()
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001230 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001231 enableContinue = enableContinue & ~enableStack[enableIndex];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001232 }
1233
1234 void PixelProgram::TEST()
1235 {
1236 whileTest = true;
1237 }
1238
Nicolas Capens4f172c72016-01-13 08:34:30 -05001239 void PixelProgram::CALL(int labelIndex, int callSiteIndex)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001240 {
1241 if(!labelBlock[labelIndex])
1242 {
1243 labelBlock[labelIndex] = Nucleus::createBasicBlock();
1244 }
1245
1246 if(callRetBlock[labelIndex].size() > 1)
1247 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001248 callStack[stackIndex++] = UInt(callSiteIndex);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001249 }
1250
Nicolas Capens4f172c72016-01-13 08:34:30 -05001251 Int4 restoreLeave = enableLeave;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001252
1253 Nucleus::createBr(labelBlock[labelIndex]);
1254 Nucleus::setInsertBlock(callRetBlock[labelIndex][callSiteIndex]);
1255
Nicolas Capens4f172c72016-01-13 08:34:30 -05001256 enableLeave = restoreLeave;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001257 }
1258
Nicolas Capens4f172c72016-01-13 08:34:30 -05001259 void PixelProgram::CALLNZ(int labelIndex, int callSiteIndex, const Src &src)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001260 {
1261 if(src.type == Shader::PARAMETER_CONSTBOOL)
1262 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001263 CALLNZb(labelIndex, callSiteIndex, src);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001264 }
1265 else if(src.type == Shader::PARAMETER_PREDICATE)
1266 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001267 CALLNZp(labelIndex, callSiteIndex, src);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001268 }
1269 else ASSERT(false);
1270 }
1271
Nicolas Capens4f172c72016-01-13 08:34:30 -05001272 void PixelProgram::CALLNZb(int labelIndex, int callSiteIndex, const Src &boolRegister)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001273 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001274 Bool condition = (*Pointer<Byte>(data + OFFSET(DrawData, ps.b[boolRegister.index])) != Byte(0)); // FIXME
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001275
1276 if(boolRegister.modifier == Shader::MODIFIER_NOT)
1277 {
1278 condition = !condition;
1279 }
1280
1281 if(!labelBlock[labelIndex])
1282 {
1283 labelBlock[labelIndex] = Nucleus::createBasicBlock();
1284 }
1285
1286 if(callRetBlock[labelIndex].size() > 1)
1287 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001288 callStack[stackIndex++] = UInt(callSiteIndex);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001289 }
1290
Nicolas Capens4f172c72016-01-13 08:34:30 -05001291 Int4 restoreLeave = enableLeave;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001292
1293 branch(condition, labelBlock[labelIndex], callRetBlock[labelIndex][callSiteIndex]);
1294 Nucleus::setInsertBlock(callRetBlock[labelIndex][callSiteIndex]);
1295
Nicolas Capens4f172c72016-01-13 08:34:30 -05001296 enableLeave = restoreLeave;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001297 }
1298
Nicolas Capens4f172c72016-01-13 08:34:30 -05001299 void PixelProgram::CALLNZp(int labelIndex, int callSiteIndex, const Src &predicateRegister)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001300 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001301 Int4 condition = As<Int4>(p0[predicateRegister.swizzle & 0x3]);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001302
1303 if(predicateRegister.modifier == Shader::MODIFIER_NOT)
1304 {
1305 condition = ~condition;
1306 }
1307
Nicolas Capens4f172c72016-01-13 08:34:30 -05001308 condition &= enableStack[enableIndex];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001309
1310 if(!labelBlock[labelIndex])
1311 {
1312 labelBlock[labelIndex] = Nucleus::createBasicBlock();
1313 }
1314
1315 if(callRetBlock[labelIndex].size() > 1)
1316 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001317 callStack[stackIndex++] = UInt(callSiteIndex);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001318 }
1319
Nicolas Capens4f172c72016-01-13 08:34:30 -05001320 enableIndex++;
1321 enableStack[enableIndex] = condition;
1322 Int4 restoreLeave = enableLeave;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001323
1324 Bool notAllFalse = SignMask(condition) != 0;
1325 branch(notAllFalse, labelBlock[labelIndex], callRetBlock[labelIndex][callSiteIndex]);
1326 Nucleus::setInsertBlock(callRetBlock[labelIndex][callSiteIndex]);
1327
Nicolas Capens4f172c72016-01-13 08:34:30 -05001328 enableIndex--;
1329 enableLeave = restoreLeave;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001330 }
1331
Nicolas Capens4f172c72016-01-13 08:34:30 -05001332 void PixelProgram::ELSE()
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001333 {
1334 ifDepth--;
1335
1336 llvm::BasicBlock *falseBlock = ifFalseBlock[ifDepth];
1337 llvm::BasicBlock *endBlock = Nucleus::createBasicBlock();
1338
1339 if(isConditionalIf[ifDepth])
1340 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001341 Int4 condition = ~enableStack[enableIndex] & enableStack[enableIndex - 1];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001342 Bool notAllFalse = SignMask(condition) != 0;
1343
1344 branch(notAllFalse, falseBlock, endBlock);
1345
Nicolas Capens4f172c72016-01-13 08:34:30 -05001346 enableStack[enableIndex] = ~enableStack[enableIndex] & enableStack[enableIndex - 1];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001347 }
1348 else
1349 {
1350 Nucleus::createBr(endBlock);
1351 Nucleus::setInsertBlock(falseBlock);
1352 }
1353
1354 ifFalseBlock[ifDepth] = endBlock;
1355
1356 ifDepth++;
1357 }
1358
Nicolas Capens4f172c72016-01-13 08:34:30 -05001359 void PixelProgram::ENDIF()
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001360 {
1361 ifDepth--;
1362
1363 llvm::BasicBlock *endBlock = ifFalseBlock[ifDepth];
1364
1365 Nucleus::createBr(endBlock);
1366 Nucleus::setInsertBlock(endBlock);
1367
1368 if(isConditionalIf[ifDepth])
1369 {
1370 breakDepth--;
Nicolas Capens4f172c72016-01-13 08:34:30 -05001371 enableIndex--;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001372 }
1373 }
1374
Nicolas Capens4f172c72016-01-13 08:34:30 -05001375 void PixelProgram::ENDLOOP()
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001376 {
1377 loopRepDepth--;
1378
Nicolas Capens4f172c72016-01-13 08:34:30 -05001379 aL[loopDepth] = aL[loopDepth] + increment[loopDepth]; // FIXME: +=
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001380
1381 llvm::BasicBlock *testBlock = loopRepTestBlock[loopRepDepth];
1382 llvm::BasicBlock *endBlock = loopRepEndBlock[loopRepDepth];
1383
1384 Nucleus::createBr(testBlock);
1385 Nucleus::setInsertBlock(endBlock);
1386
Nicolas Capens4f172c72016-01-13 08:34:30 -05001387 loopDepth--;
1388 enableBreak = Int4(0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001389 }
1390
Nicolas Capens4f172c72016-01-13 08:34:30 -05001391 void PixelProgram::ENDREP()
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001392 {
1393 loopRepDepth--;
1394
1395 llvm::BasicBlock *testBlock = loopRepTestBlock[loopRepDepth];
1396 llvm::BasicBlock *endBlock = loopRepEndBlock[loopRepDepth];
1397
1398 Nucleus::createBr(testBlock);
1399 Nucleus::setInsertBlock(endBlock);
1400
Nicolas Capens4f172c72016-01-13 08:34:30 -05001401 loopDepth--;
1402 enableBreak = Int4(0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001403 }
1404
Nicolas Capens4f172c72016-01-13 08:34:30 -05001405 void PixelProgram::ENDWHILE()
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001406 {
1407 loopRepDepth--;
1408
1409 llvm::BasicBlock *testBlock = loopRepTestBlock[loopRepDepth];
1410 llvm::BasicBlock *endBlock = loopRepEndBlock[loopRepDepth];
1411
1412 Nucleus::createBr(testBlock);
1413 Nucleus::setInsertBlock(endBlock);
1414
Nicolas Capens4f172c72016-01-13 08:34:30 -05001415 enableIndex--;
1416 enableBreak = Int4(0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001417 whileTest = false;
1418 }
1419
Nicolas Capens4f172c72016-01-13 08:34:30 -05001420 void PixelProgram::IF(const Src &src)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001421 {
1422 if(src.type == Shader::PARAMETER_CONSTBOOL)
1423 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001424 IFb(src);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001425 }
1426 else if(src.type == Shader::PARAMETER_PREDICATE)
1427 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001428 IFp(src);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001429 }
1430 else
1431 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001432 Int4 condition = As<Int4>(fetchRegisterF(src).x);
1433 IF(condition);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001434 }
1435 }
1436
Nicolas Capens4f172c72016-01-13 08:34:30 -05001437 void PixelProgram::IFb(const Src &boolRegister)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001438 {
1439 ASSERT(ifDepth < 24 + 4);
1440
Nicolas Capens4f172c72016-01-13 08:34:30 -05001441 Bool condition = (*Pointer<Byte>(data + OFFSET(DrawData, ps.b[boolRegister.index])) != Byte(0)); // FIXME
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001442
1443 if(boolRegister.modifier == Shader::MODIFIER_NOT)
1444 {
1445 condition = !condition;
1446 }
1447
1448 llvm::BasicBlock *trueBlock = Nucleus::createBasicBlock();
1449 llvm::BasicBlock *falseBlock = Nucleus::createBasicBlock();
1450
1451 branch(condition, trueBlock, falseBlock);
1452
1453 isConditionalIf[ifDepth] = false;
1454 ifFalseBlock[ifDepth] = falseBlock;
1455
1456 ifDepth++;
1457 }
1458
Nicolas Capens4f172c72016-01-13 08:34:30 -05001459 void PixelProgram::IFp(const Src &predicateRegister)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001460 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001461 Int4 condition = As<Int4>(p0[predicateRegister.swizzle & 0x3]);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001462
1463 if(predicateRegister.modifier == Shader::MODIFIER_NOT)
1464 {
1465 condition = ~condition;
1466 }
1467
Nicolas Capens4f172c72016-01-13 08:34:30 -05001468 IF(condition);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001469 }
1470
Nicolas Capens4f172c72016-01-13 08:34:30 -05001471 void PixelProgram::IFC(Vector4f &src0, Vector4f &src1, Control control)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001472 {
1473 Int4 condition;
1474
1475 switch(control)
1476 {
1477 case Shader::CONTROL_GT: condition = CmpNLE(src0.x, src1.x); break;
1478 case Shader::CONTROL_EQ: condition = CmpEQ(src0.x, src1.x); break;
1479 case Shader::CONTROL_GE: condition = CmpNLT(src0.x, src1.x); break;
1480 case Shader::CONTROL_LT: condition = CmpLT(src0.x, src1.x); break;
1481 case Shader::CONTROL_NE: condition = CmpNEQ(src0.x, src1.x); break;
1482 case Shader::CONTROL_LE: condition = CmpLE(src0.x, src1.x); break;
1483 default:
1484 ASSERT(false);
1485 }
1486
Nicolas Capens4f172c72016-01-13 08:34:30 -05001487 IF(condition);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001488 }
1489
Nicolas Capens4f172c72016-01-13 08:34:30 -05001490 void PixelProgram::IF(Int4 &condition)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001491 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001492 condition &= enableStack[enableIndex];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001493
Nicolas Capens4f172c72016-01-13 08:34:30 -05001494 enableIndex++;
1495 enableStack[enableIndex] = condition;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001496
1497 llvm::BasicBlock *trueBlock = Nucleus::createBasicBlock();
1498 llvm::BasicBlock *falseBlock = Nucleus::createBasicBlock();
1499
1500 Bool notAllFalse = SignMask(condition) != 0;
1501
1502 branch(notAllFalse, trueBlock, falseBlock);
1503
1504 isConditionalIf[ifDepth] = true;
1505 ifFalseBlock[ifDepth] = falseBlock;
1506
1507 ifDepth++;
1508 breakDepth++;
1509 }
1510
1511 void PixelProgram::LABEL(int labelIndex)
1512 {
1513 if(!labelBlock[labelIndex])
1514 {
1515 labelBlock[labelIndex] = Nucleus::createBasicBlock();
1516 }
1517
1518 Nucleus::setInsertBlock(labelBlock[labelIndex]);
1519 currentLabel = labelIndex;
1520 }
1521
Nicolas Capens4f172c72016-01-13 08:34:30 -05001522 void PixelProgram::LOOP(const Src &integerRegister)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001523 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001524 loopDepth++;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001525
Nicolas Capens4f172c72016-01-13 08:34:30 -05001526 iteration[loopDepth] = *Pointer<Int>(data + OFFSET(DrawData, ps.i[integerRegister.index][0]));
1527 aL[loopDepth] = *Pointer<Int>(data + OFFSET(DrawData, ps.i[integerRegister.index][1]));
1528 increment[loopDepth] = *Pointer<Int>(data + OFFSET(DrawData, ps.i[integerRegister.index][2]));
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001529
Nicolas Capens4f172c72016-01-13 08:34:30 -05001530 // If(increment[loopDepth] == 0)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001531 // {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001532 // increment[loopDepth] = 1;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001533 // }
1534
1535 llvm::BasicBlock *loopBlock = Nucleus::createBasicBlock();
1536 llvm::BasicBlock *testBlock = Nucleus::createBasicBlock();
1537 llvm::BasicBlock *endBlock = Nucleus::createBasicBlock();
1538
1539 loopRepTestBlock[loopRepDepth] = testBlock;
1540 loopRepEndBlock[loopRepDepth] = endBlock;
1541
1542 // FIXME: jump(testBlock)
1543 Nucleus::createBr(testBlock);
1544 Nucleus::setInsertBlock(testBlock);
1545
Nicolas Capens4f172c72016-01-13 08:34:30 -05001546 branch(iteration[loopDepth] > 0, loopBlock, endBlock);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001547 Nucleus::setInsertBlock(loopBlock);
1548
Nicolas Capens4f172c72016-01-13 08:34:30 -05001549 iteration[loopDepth] = iteration[loopDepth] - 1; // FIXME: --
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001550
1551 loopRepDepth++;
1552 breakDepth = 0;
1553 }
1554
Nicolas Capens4f172c72016-01-13 08:34:30 -05001555 void PixelProgram::REP(const Src &integerRegister)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001556 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001557 loopDepth++;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001558
Nicolas Capens4f172c72016-01-13 08:34:30 -05001559 iteration[loopDepth] = *Pointer<Int>(data + OFFSET(DrawData, ps.i[integerRegister.index][0]));
1560 aL[loopDepth] = aL[loopDepth - 1];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001561
1562 llvm::BasicBlock *loopBlock = Nucleus::createBasicBlock();
1563 llvm::BasicBlock *testBlock = Nucleus::createBasicBlock();
1564 llvm::BasicBlock *endBlock = Nucleus::createBasicBlock();
1565
1566 loopRepTestBlock[loopRepDepth] = testBlock;
1567 loopRepEndBlock[loopRepDepth] = endBlock;
1568
1569 // FIXME: jump(testBlock)
1570 Nucleus::createBr(testBlock);
1571 Nucleus::setInsertBlock(testBlock);
1572
Nicolas Capens4f172c72016-01-13 08:34:30 -05001573 branch(iteration[loopDepth] > 0, loopBlock, endBlock);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001574 Nucleus::setInsertBlock(loopBlock);
1575
Nicolas Capens4f172c72016-01-13 08:34:30 -05001576 iteration[loopDepth] = iteration[loopDepth] - 1; // FIXME: --
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001577
1578 loopRepDepth++;
1579 breakDepth = 0;
1580 }
1581
Nicolas Capens4f172c72016-01-13 08:34:30 -05001582 void PixelProgram::WHILE(const Src &temporaryRegister)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001583 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001584 enableIndex++;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001585
1586 llvm::BasicBlock *loopBlock = Nucleus::createBasicBlock();
1587 llvm::BasicBlock *testBlock = Nucleus::createBasicBlock();
1588 llvm::BasicBlock *endBlock = Nucleus::createBasicBlock();
1589
1590 loopRepTestBlock[loopRepDepth] = testBlock;
1591 loopRepEndBlock[loopRepDepth] = endBlock;
1592
Nicolas Capens4f172c72016-01-13 08:34:30 -05001593 Int4 restoreBreak = enableBreak;
1594 Int4 restoreContinue = enableContinue;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001595
1596 // FIXME: jump(testBlock)
1597 Nucleus::createBr(testBlock);
1598 Nucleus::setInsertBlock(testBlock);
Nicolas Capens4f172c72016-01-13 08:34:30 -05001599 enableContinue = restoreContinue;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001600
Nicolas Capens4f172c72016-01-13 08:34:30 -05001601 const Vector4f &src = fetchRegisterF(temporaryRegister);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001602 Int4 condition = As<Int4>(src.x);
Nicolas Capens4f172c72016-01-13 08:34:30 -05001603 condition &= enableStack[enableIndex - 1];
1604 enableStack[enableIndex] = condition;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001605
1606 Bool notAllFalse = SignMask(condition) != 0;
1607 branch(notAllFalse, loopBlock, endBlock);
1608
1609 Nucleus::setInsertBlock(endBlock);
Nicolas Capens4f172c72016-01-13 08:34:30 -05001610 enableBreak = restoreBreak;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001611
1612 Nucleus::setInsertBlock(loopBlock);
1613
1614 loopRepDepth++;
1615 breakDepth = 0;
1616 }
1617
Nicolas Capens4f172c72016-01-13 08:34:30 -05001618 void PixelProgram::RET()
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001619 {
1620 if(currentLabel == -1)
1621 {
1622 returnBlock = Nucleus::createBasicBlock();
1623 Nucleus::createBr(returnBlock);
1624 }
1625 else
1626 {
1627 llvm::BasicBlock *unreachableBlock = Nucleus::createBasicBlock();
1628
1629 if(callRetBlock[currentLabel].size() > 1) // Pop the return destination from the call stack
1630 {
1631 // FIXME: Encapsulate
Nicolas Capens4f172c72016-01-13 08:34:30 -05001632 UInt index = callStack[--stackIndex];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001633
1634 llvm::Value *value = index.loadValue();
1635 llvm::Value *switchInst = Nucleus::createSwitch(value, unreachableBlock, (int)callRetBlock[currentLabel].size());
1636
1637 for(unsigned int i = 0; i < callRetBlock[currentLabel].size(); i++)
1638 {
1639 Nucleus::addSwitchCase(switchInst, i, callRetBlock[currentLabel][i]);
1640 }
1641 }
1642 else if(callRetBlock[currentLabel].size() == 1) // Jump directly to the unique return destination
1643 {
1644 Nucleus::createBr(callRetBlock[currentLabel][0]);
1645 }
1646 else // Function isn't called
1647 {
1648 Nucleus::createBr(unreachableBlock);
1649 }
1650
1651 Nucleus::setInsertBlock(unreachableBlock);
1652 Nucleus::createUnreachable();
1653 }
1654 }
1655
Nicolas Capens4f172c72016-01-13 08:34:30 -05001656 void PixelProgram::LEAVE()
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001657 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001658 enableLeave = enableLeave & ~enableStack[enableIndex];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001659
1660 // FIXME: Return from function if all instances left
1661 // FIXME: Use enableLeave in other control-flow constructs
1662 }
Nicolas Capens5d961882016-01-01 23:18:14 -05001663}