blob: 7065330e968c68cdbe6196a2bc508ea39815cbd6 [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 Capensc2534f42016-04-04 11:13:24 -0400139 if(src0.type != Shader::PARAMETER_VOID) s0 = fetchRegister(src0);
140 if(src1.type != Shader::PARAMETER_VOID) s1 = fetchRegister(src1);
141 if(src2.type != Shader::PARAMETER_VOID) s2 = fetchRegister(src2);
142 if(src3.type != Shader::PARAMETER_VOID) s3 = fetchRegister(src3);
143 if(src4.type != Shader::PARAMETER_VOID) s4 = fetchRegister(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;
Alexis Hetu0f448072016-03-18 10:56:08 -0400234 case Shader::OPCODE_ISGN: isgn(d, s0); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400235 case Shader::OPCODE_CRS: crs(d, s0, s1); break;
236 case Shader::OPCODE_FORWARD1: forward1(d, s0, s1, s2); break;
237 case Shader::OPCODE_FORWARD2: forward2(d, s0, s1, s2); break;
238 case Shader::OPCODE_FORWARD3: forward3(d, s0, s1, s2); break;
239 case Shader::OPCODE_FORWARD4: forward4(d, s0, s1, s2); break;
240 case Shader::OPCODE_REFLECT1: reflect1(d, s0, s1); break;
241 case Shader::OPCODE_REFLECT2: reflect2(d, s0, s1); break;
242 case Shader::OPCODE_REFLECT3: reflect3(d, s0, s1); break;
243 case Shader::OPCODE_REFLECT4: reflect4(d, s0, s1); break;
244 case Shader::OPCODE_REFRACT1: refract1(d, s0, s1, s2.x); break;
245 case Shader::OPCODE_REFRACT2: refract2(d, s0, s1, s2.x); break;
246 case Shader::OPCODE_REFRACT3: refract3(d, s0, s1, s2.x); break;
247 case Shader::OPCODE_REFRACT4: refract4(d, s0, s1, s2.x); break;
248 case Shader::OPCODE_NRM2: nrm2(d, s0, pp); break;
249 case Shader::OPCODE_NRM3: nrm3(d, s0, pp); break;
250 case Shader::OPCODE_NRM4: nrm4(d, s0, pp); break;
251 case Shader::OPCODE_ABS: abs(d, s0); break;
Alexis Hetu0f448072016-03-18 10:56:08 -0400252 case Shader::OPCODE_IABS: iabs(d, s0); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400253 case Shader::OPCODE_SINCOS: sincos(d, s0, pp); break;
254 case Shader::OPCODE_COS: cos(d, s0, pp); break;
255 case Shader::OPCODE_SIN: sin(d, s0, pp); break;
256 case Shader::OPCODE_TAN: tan(d, s0, pp); break;
257 case Shader::OPCODE_ACOS: acos(d, s0, pp); break;
258 case Shader::OPCODE_ASIN: asin(d, s0, pp); break;
259 case Shader::OPCODE_ATAN: atan(d, s0, pp); break;
260 case Shader::OPCODE_ATAN2: atan2(d, s0, s1, pp); break;
261 case Shader::OPCODE_COSH: cosh(d, s0, pp); break;
262 case Shader::OPCODE_SINH: sinh(d, s0, pp); break;
263 case Shader::OPCODE_TANH: tanh(d, s0, pp); break;
264 case Shader::OPCODE_ACOSH: acosh(d, s0, pp); break;
265 case Shader::OPCODE_ASINH: asinh(d, s0, pp); break;
266 case Shader::OPCODE_ATANH: atanh(d, s0, pp); break;
Nicolas Capens4f172c72016-01-13 08:34:30 -0500267 case Shader::OPCODE_M4X4: M4X4(d, s0, src1); break;
268 case Shader::OPCODE_M4X3: M4X3(d, s0, src1); break;
269 case Shader::OPCODE_M3X4: M3X4(d, s0, src1); break;
270 case Shader::OPCODE_M3X3: M3X3(d, s0, src1); break;
271 case Shader::OPCODE_M3X2: M3X2(d, s0, src1); break;
272 case Shader::OPCODE_TEX: TEXLD(d, s0, src1, project, bias); break;
Nicolas Capensc2534f42016-04-04 11:13:24 -0400273 case Shader::OPCODE_TEXLDD: TEXLDD(d, s0, src1, s2, s3, project); break;
274 case Shader::OPCODE_TEXLDL: TEXLDL(d, s0, src1, project); break;
Nicolas Capens4f172c72016-01-13 08:34:30 -0500275 case Shader::OPCODE_TEXSIZE: TEXSIZE(d, s0.x, src1); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400276 case Shader::OPCODE_TEXKILL: TEXKILL(cMask, d, dst.mask); break;
Nicolas Capens4f172c72016-01-13 08:34:30 -0500277 case Shader::OPCODE_TEXOFFSET: TEXOFFSET(d, s0, src1, s2, s3, project, bias); break;
278 case Shader::OPCODE_TEXLDLOFFSET: TEXLDL(d, s0, src1, s2, project, bias); break;
279 case Shader::OPCODE_TEXELFETCH: TEXELFETCH(d, s0, src1, s2); break;
280 case Shader::OPCODE_TEXELFETCHOFFSET: TEXELFETCH(d, s0, src1, s2, s3); break;
281 case Shader::OPCODE_TEXGRAD: TEXGRAD(d, s0, src1, s2, s3); break;
282 case Shader::OPCODE_TEXGRADOFFSET: TEXGRAD(d, s0, src1, s2, s3, s4); break;
283 case Shader::OPCODE_DISCARD: DISCARD(cMask, instruction); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400284 case Shader::OPCODE_DFDX: DFDX(d, s0); break;
285 case Shader::OPCODE_DFDY: DFDY(d, s0); break;
286 case Shader::OPCODE_FWIDTH: FWIDTH(d, s0); break;
Nicolas Capens4f172c72016-01-13 08:34:30 -0500287 case Shader::OPCODE_BREAK: BREAK(); break;
288 case Shader::OPCODE_BREAKC: BREAKC(s0, s1, control); break;
289 case Shader::OPCODE_BREAKP: BREAKP(src0); break;
290 case Shader::OPCODE_CONTINUE: CONTINUE(); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400291 case Shader::OPCODE_TEST: TEST(); break;
Nicolas Capens4f172c72016-01-13 08:34:30 -0500292 case Shader::OPCODE_CALL: CALL(dst.label, dst.callSite); break;
293 case Shader::OPCODE_CALLNZ: CALLNZ(dst.label, dst.callSite, src0); break;
294 case Shader::OPCODE_ELSE: ELSE(); break;
295 case Shader::OPCODE_ENDIF: ENDIF(); break;
296 case Shader::OPCODE_ENDLOOP: ENDLOOP(); break;
297 case Shader::OPCODE_ENDREP: ENDREP(); break;
298 case Shader::OPCODE_ENDWHILE: ENDWHILE(); break;
299 case Shader::OPCODE_IF: IF(src0); break;
300 case Shader::OPCODE_IFC: IFC(s0, s1, control); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400301 case Shader::OPCODE_LABEL: LABEL(dst.index); break;
Nicolas Capens4f172c72016-01-13 08:34:30 -0500302 case Shader::OPCODE_LOOP: LOOP(src1); break;
303 case Shader::OPCODE_REP: REP(src0); break;
304 case Shader::OPCODE_WHILE: WHILE(src0); break;
305 case Shader::OPCODE_RET: RET(); break;
306 case Shader::OPCODE_LEAVE: LEAVE(); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400307 case Shader::OPCODE_CMP: cmp(d, s0, s1, control); break;
308 case Shader::OPCODE_ALL: all(d.x, s0); break;
309 case Shader::OPCODE_ANY: any(d.x, s0); break;
310 case Shader::OPCODE_NOT: not(d, s0); break;
Alexis Hetuc4f2c292015-08-18 15:43:09 -0400311 case Shader::OPCODE_OR: or(d, s0, s1); break;
312 case Shader::OPCODE_XOR: xor(d, s0, s1); break;
313 case Shader::OPCODE_AND: and(d, s0, s1); break;
Alexis Hetu8d78cf72015-08-28 14:24:45 -0400314 case Shader::OPCODE_EQ: equal(d, s0, s1); break;
315 case Shader::OPCODE_NE: notEqual(d, s0, s1); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400316 case Shader::OPCODE_END: break;
317 default:
318 ASSERT(false);
319 }
320
321 if(dst.type != Shader::PARAMETER_VOID && dst.type != Shader::PARAMETER_LABEL && opcode != Shader::OPCODE_TEXKILL && opcode != Shader::OPCODE_NOP)
322 {
323 if(dst.integer)
324 {
325 switch(opcode)
326 {
327 case Shader::OPCODE_DIV:
328 if(dst.x) d.x = Trunc(d.x);
329 if(dst.y) d.y = Trunc(d.y);
330 if(dst.z) d.z = Trunc(d.z);
331 if(dst.w) d.w = Trunc(d.w);
332 break;
333 default:
334 break; // No truncation to integer required when arguments are integer
335 }
336 }
337
338 if(dst.saturate)
339 {
340 if(dst.x) d.x = Max(d.x, Float4(0.0f));
341 if(dst.y) d.y = Max(d.y, Float4(0.0f));
342 if(dst.z) d.z = Max(d.z, Float4(0.0f));
343 if(dst.w) d.w = Max(d.w, Float4(0.0f));
344
345 if(dst.x) d.x = Min(d.x, Float4(1.0f));
346 if(dst.y) d.y = Min(d.y, Float4(1.0f));
347 if(dst.z) d.z = Min(d.z, Float4(1.0f));
348 if(dst.w) d.w = Min(d.w, Float4(1.0f));
349 }
350
351 if(instruction->isPredicated())
352 {
353 Vector4f pDst; // FIXME: Rename
354
355 switch(dst.type)
356 {
357 case Shader::PARAMETER_TEMP:
358 if(dst.rel.type == Shader::PARAMETER_VOID)
359 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500360 if(dst.x) pDst.x = r[dst.index].x;
361 if(dst.y) pDst.y = r[dst.index].y;
362 if(dst.z) pDst.z = r[dst.index].z;
363 if(dst.w) pDst.w = r[dst.index].w;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400364 }
365 else
366 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500367 Int a = relativeAddress(dst);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400368
Nicolas Capens4f172c72016-01-13 08:34:30 -0500369 if(dst.x) pDst.x = r[dst.index + a].x;
370 if(dst.y) pDst.y = r[dst.index + a].y;
371 if(dst.z) pDst.z = r[dst.index + a].z;
372 if(dst.w) pDst.w = r[dst.index + a].w;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400373 }
374 break;
375 case Shader::PARAMETER_COLOROUT:
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500376 if(dst.rel.type == Shader::PARAMETER_VOID)
377 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500378 if(dst.x) pDst.x = oC[dst.index].x;
379 if(dst.y) pDst.y = oC[dst.index].y;
380 if(dst.z) pDst.z = oC[dst.index].z;
381 if(dst.w) pDst.w = oC[dst.index].w;
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500382 }
383 else
384 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500385 Int a = relativeAddress(dst) + dst.index;
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500386
Nicolas Capens4f172c72016-01-13 08:34:30 -0500387 if(dst.x) pDst.x = oC[a].x;
388 if(dst.y) pDst.y = oC[a].y;
389 if(dst.z) pDst.z = oC[a].z;
390 if(dst.w) pDst.w = oC[a].w;
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500391 }
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400392 break;
393 case Shader::PARAMETER_PREDICATE:
Nicolas Capens4f172c72016-01-13 08:34:30 -0500394 if(dst.x) pDst.x = p0.x;
395 if(dst.y) pDst.y = p0.y;
396 if(dst.z) pDst.z = p0.z;
397 if(dst.w) pDst.w = p0.w;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400398 break;
399 case Shader::PARAMETER_DEPTHOUT:
Nicolas Capens4f172c72016-01-13 08:34:30 -0500400 pDst.x = oDepth;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400401 break;
402 default:
403 ASSERT(false);
404 }
405
Nicolas Capens4f172c72016-01-13 08:34:30 -0500406 Int4 enable = enableMask(instruction);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400407
408 Int4 xEnable = enable;
409 Int4 yEnable = enable;
410 Int4 zEnable = enable;
411 Int4 wEnable = enable;
412
413 if(predicate)
414 {
415 unsigned char pSwizzle = instruction->predicateSwizzle;
416
Nicolas Capens4f172c72016-01-13 08:34:30 -0500417 Float4 xPredicate = p0[(pSwizzle >> 0) & 0x03];
418 Float4 yPredicate = p0[(pSwizzle >> 2) & 0x03];
419 Float4 zPredicate = p0[(pSwizzle >> 4) & 0x03];
420 Float4 wPredicate = p0[(pSwizzle >> 6) & 0x03];
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400421
422 if(!instruction->predicateNot)
423 {
424 if(dst.x) xEnable = xEnable & As<Int4>(xPredicate);
425 if(dst.y) yEnable = yEnable & As<Int4>(yPredicate);
426 if(dst.z) zEnable = zEnable & As<Int4>(zPredicate);
427 if(dst.w) wEnable = wEnable & As<Int4>(wPredicate);
428 }
429 else
430 {
431 if(dst.x) xEnable = xEnable & ~As<Int4>(xPredicate);
432 if(dst.y) yEnable = yEnable & ~As<Int4>(yPredicate);
433 if(dst.z) zEnable = zEnable & ~As<Int4>(zPredicate);
434 if(dst.w) wEnable = wEnable & ~As<Int4>(wPredicate);
435 }
436 }
437
438 if(dst.x) d.x = As<Float4>(As<Int4>(d.x) & xEnable);
439 if(dst.y) d.y = As<Float4>(As<Int4>(d.y) & yEnable);
440 if(dst.z) d.z = As<Float4>(As<Int4>(d.z) & zEnable);
441 if(dst.w) d.w = As<Float4>(As<Int4>(d.w) & wEnable);
442
443 if(dst.x) d.x = As<Float4>(As<Int4>(d.x) | (As<Int4>(pDst.x) & ~xEnable));
444 if(dst.y) d.y = As<Float4>(As<Int4>(d.y) | (As<Int4>(pDst.y) & ~yEnable));
445 if(dst.z) d.z = As<Float4>(As<Int4>(d.z) | (As<Int4>(pDst.z) & ~zEnable));
446 if(dst.w) d.w = As<Float4>(As<Int4>(d.w) | (As<Int4>(pDst.w) & ~wEnable));
447 }
448
449 switch(dst.type)
450 {
451 case Shader::PARAMETER_TEMP:
452 if(dst.rel.type == Shader::PARAMETER_VOID)
453 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500454 if(dst.x) r[dst.index].x = d.x;
455 if(dst.y) r[dst.index].y = d.y;
456 if(dst.z) r[dst.index].z = d.z;
457 if(dst.w) r[dst.index].w = d.w;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400458 }
459 else
460 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500461 Int a = relativeAddress(dst);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400462
Nicolas Capens4f172c72016-01-13 08:34:30 -0500463 if(dst.x) r[dst.index + a].x = d.x;
464 if(dst.y) r[dst.index + a].y = d.y;
465 if(dst.z) r[dst.index + a].z = d.z;
466 if(dst.w) r[dst.index + a].w = d.w;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400467 }
468 break;
469 case Shader::PARAMETER_COLOROUT:
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500470 if(dst.rel.type == Shader::PARAMETER_VOID)
471 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500472 if(dst.x) { oC[dst.index].x = d.x; }
473 if(dst.y) { oC[dst.index].y = d.y; }
474 if(dst.z) { oC[dst.index].z = d.z; }
475 if(dst.w) { oC[dst.index].w = d.w; }
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500476 }
477 else
478 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500479 Int a = relativeAddress(dst) + dst.index;
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500480
Nicolas Capens4f172c72016-01-13 08:34:30 -0500481 if(dst.x) { oC[a].x = d.x; }
482 if(dst.y) { oC[a].y = d.y; }
483 if(dst.z) { oC[a].z = d.z; }
484 if(dst.w) { oC[a].w = d.w; }
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500485 }
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400486 break;
487 case Shader::PARAMETER_PREDICATE:
Nicolas Capens4f172c72016-01-13 08:34:30 -0500488 if(dst.x) p0.x = d.x;
489 if(dst.y) p0.y = d.y;
490 if(dst.z) p0.z = d.z;
491 if(dst.w) p0.w = d.w;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400492 break;
493 case Shader::PARAMETER_DEPTHOUT:
Nicolas Capens4f172c72016-01-13 08:34:30 -0500494 oDepth = d.x;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400495 break;
496 default:
497 ASSERT(false);
498 }
499 }
500 }
501
502 if(currentLabel != -1)
503 {
504 Nucleus::setInsertBlock(returnBlock);
505 }
506
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500507 for(int i = 0; i < RENDERTARGETS; i++)
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400508 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500509 c[i] = oC[i];
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400510 }
511 }
Nicolas Capens5d961882016-01-01 23:18:14 -0500512
Nicolas Capens4f172c72016-01-13 08:34:30 -0500513 Bool PixelProgram::alphaTest(Int cMask[4])
Nicolas Capens5d961882016-01-01 23:18:14 -0500514 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500515 clampColor(c);
Nicolas Capens5d961882016-01-01 23:18:14 -0500516
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400517 if(!state.alphaTestActive())
518 {
519 return true;
520 }
521
522 Int aMask;
523
524 if(state.transparencyAntialiasing == TRANSPARENCY_NONE)
525 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500526 Short4 alpha = RoundShort4(c[0].w * Float4(0x1000));
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400527
Nicolas Capens4f172c72016-01-13 08:34:30 -0500528 PixelRoutine::alphaTest(aMask, alpha);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400529
530 for(unsigned int q = 0; q < state.multiSample; q++)
531 {
532 cMask[q] &= aMask;
533 }
534 }
535 else if(state.transparencyAntialiasing == TRANSPARENCY_ALPHA_TO_COVERAGE)
536 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500537 alphaToCoverage(cMask, c[0].w);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400538 }
539 else ASSERT(false);
540
541 Int pass = cMask[0];
542
543 for(unsigned int q = 1; q < state.multiSample; q++)
544 {
545 pass = pass | cMask[q];
546 }
547
Nicolas Capens5d961882016-01-01 23:18:14 -0500548 return pass != 0x0;
549 }
550
Nicolas Capens4f172c72016-01-13 08:34:30 -0500551 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 -0500552 {
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500553 for(int index = 0; index < RENDERTARGETS; index++)
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400554 {
555 if(!state.colorWriteActive(index))
556 {
557 continue;
Nicolas Capens5d961882016-01-01 23:18:14 -0500558 }
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400559
560 if(!postBlendSRGB && state.writeSRGB)
561 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500562 c[index].x = linearToSRGB(c[index].x);
563 c[index].y = linearToSRGB(c[index].y);
564 c[index].z = linearToSRGB(c[index].z);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400565 }
566
567 if(index == 0)
568 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500569 fogBlend(c[index], fog);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400570 }
571
572 switch(state.targetFormat[index])
573 {
574 case FORMAT_R5G6B5:
575 case FORMAT_X8R8G8B8:
576 case FORMAT_X8B8G8R8:
577 case FORMAT_A8R8G8B8:
578 case FORMAT_A8B8G8R8:
579 case FORMAT_A8:
580 case FORMAT_G16R16:
581 case FORMAT_A16B16G16R16:
582 for(unsigned int q = 0; q < state.multiSample; q++)
583 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500584 Pointer<Byte> buffer = cBuffer[index] + q * *Pointer<Int>(data + OFFSET(DrawData, colorSliceB[index]));
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400585 Vector4s color;
586
Nicolas Capens13c70cd2016-04-05 10:51:45 -0400587 if(state.targetFormat[index] == FORMAT_R5G6B5)
588 {
589 color.x = UShort4(c[index].x * Float4(0xFBFF), false);
590 color.y = UShort4(c[index].y * Float4(0xFDFF), false);
591 color.z = UShort4(c[index].z * Float4(0xFBFF), false);
592 color.w = UShort4(c[index].w * Float4(0xFFFF), false);
593 }
594 else
595 {
596 color.x = convertFixed16(c[index].x, false);
597 color.y = convertFixed16(c[index].y, false);
598 color.z = convertFixed16(c[index].z, false);
599 color.w = convertFixed16(c[index].w, false);
600 }
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400601
602 if(state.multiSampleMask & (1 << q))
603 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500604 alphaBlend(index, buffer, color, x);
605 logicOperation(index, buffer, color, x);
606 writeColor(index, buffer, x, color, sMask[q], zMask[q], cMask[q]);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400607 }
608 }
609 break;
610 case FORMAT_R32F:
611 case FORMAT_G32R32F:
612 case FORMAT_A32B32G32R32F:
Alexis Hetu1abb6382016-02-08 11:21:16 -0500613 case FORMAT_R32I:
614 case FORMAT_G32R32I:
615 case FORMAT_A32B32G32R32I:
616 case FORMAT_R32UI:
617 case FORMAT_G32R32UI:
618 case FORMAT_A32B32G32R32UI:
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400619 for(unsigned int q = 0; q < state.multiSample; q++)
620 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500621 Pointer<Byte> buffer = cBuffer[index] + q * *Pointer<Int>(data + OFFSET(DrawData, colorSliceB[index]));
622 Vector4f color = c[index];
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400623
624 if(state.multiSampleMask & (1 << q))
625 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500626 alphaBlend(index, buffer, color, x);
627 writeColor(index, buffer, x, color, sMask[q], zMask[q], cMask[q]);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400628 }
629 }
630 break;
631 default:
632 ASSERT(false);
633 }
634 }
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500635 }
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400636
Nicolas Capensc2534f42016-04-04 11:13:24 -0400637 void PixelProgram::sampleTexture(Vector4f &c, const Src &sampler, Float4 &u, Float4 &v, Float4 &w, Float4 &q, Vector4f &dsx, Vector4f &dsy, bool project, SamplerMethod method)
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400638 {
Nicolas Capensc2534f42016-04-04 11:13:24 -0400639 Vector4f tmp;
640
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400641 if(sampler.type == Shader::PARAMETER_SAMPLER && sampler.rel.type == Shader::PARAMETER_VOID)
642 {
Nicolas Capensc2534f42016-04-04 11:13:24 -0400643 sampleTexture(tmp, sampler.index, u, v, w, q, dsx, dsy, project, method);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400644 }
645 else
646 {
Nicolas Capensc2534f42016-04-04 11:13:24 -0400647 Int index = As<Int>(Float(fetchRegister(sampler).x.x));
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400648
649 for(int i = 0; i < TEXTURE_IMAGE_UNITS; i++)
650 {
651 if(shader->usesSampler(i))
652 {
653 If(index == i)
654 {
Nicolas Capensc2534f42016-04-04 11:13:24 -0400655 sampleTexture(tmp, i, u, v, w, q, dsx, dsy, project, method);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400656 // FIXME: When the sampler states are the same, we could use one sampler and just index the texture
657 }
658 }
659 }
660 }
Nicolas Capensc2534f42016-04-04 11:13:24 -0400661
662 c.x = tmp[(sampler.swizzle >> 0) & 0x3];
663 c.y = tmp[(sampler.swizzle >> 2) & 0x3];
664 c.z = tmp[(sampler.swizzle >> 4) & 0x3];
665 c.w = tmp[(sampler.swizzle >> 6) & 0x3];
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400666 }
667
Nicolas Capensc2534f42016-04-04 11:13:24 -0400668 void PixelProgram::sampleTexture(Vector4f &c, int stage, Float4 &u, Float4 &v, Float4 &w, Float4 &q, Vector4f &dsx, Vector4f &dsy, bool project, SamplerMethod method)
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400669 {
Nicolas Capensc2534f42016-04-04 11:13:24 -0400670 #if PERF_PROFILE
671 Long texTime = Ticks();
672 #endif
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400673
Nicolas Capens4f172c72016-01-13 08:34:30 -0500674 Pointer<Byte> texture = data + OFFSET(DrawData, mipmap) + stage * sizeof(Texture);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400675
676 if(!project)
677 {
Nicolas Capensc2534f42016-04-04 11:13:24 -0400678 sampler[stage]->sampleTexture(texture, c, u, v, w, q, dsx, dsy, method);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400679 }
680 else
681 {
682 Float4 rq = reciprocal(q);
683
684 Float4 u_q = u * rq;
685 Float4 v_q = v * rq;
686 Float4 w_q = w * rq;
687
Nicolas Capensc2534f42016-04-04 11:13:24 -0400688 sampler[stage]->sampleTexture(texture, c, u_q, v_q, w_q, q, dsx, dsy, method);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400689 }
690
Nicolas Capensc2534f42016-04-04 11:13:24 -0400691 #if PERF_PROFILE
692 cycles[PERF_TEX] += Ticks() - texTime;
693 #endif
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400694 }
695
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500696 void PixelProgram::clampColor(Vector4f oC[RENDERTARGETS])
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400697 {
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500698 for(int index = 0; index < RENDERTARGETS; index++)
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400699 {
700 if(!state.colorWriteActive(index) && !(index == 0 && state.alphaTestActive()))
701 {
702 continue;
703 }
704
705 switch(state.targetFormat[index])
706 {
707 case FORMAT_NULL:
708 break;
709 case FORMAT_R5G6B5:
710 case FORMAT_A8R8G8B8:
711 case FORMAT_A8B8G8R8:
712 case FORMAT_X8R8G8B8:
713 case FORMAT_X8B8G8R8:
714 case FORMAT_A8:
715 case FORMAT_G16R16:
716 case FORMAT_A16B16G16R16:
717 oC[index].x = Max(oC[index].x, Float4(0.0f)); oC[index].x = Min(oC[index].x, Float4(1.0f));
718 oC[index].y = Max(oC[index].y, Float4(0.0f)); oC[index].y = Min(oC[index].y, Float4(1.0f));
719 oC[index].z = Max(oC[index].z, Float4(0.0f)); oC[index].z = Min(oC[index].z, Float4(1.0f));
720 oC[index].w = Max(oC[index].w, Float4(0.0f)); oC[index].w = Min(oC[index].w, Float4(1.0f));
721 break;
722 case FORMAT_R32F:
723 case FORMAT_G32R32F:
724 case FORMAT_A32B32G32R32F:
Alexis Hetu1abb6382016-02-08 11:21:16 -0500725 case FORMAT_R32I:
726 case FORMAT_G32R32I:
727 case FORMAT_A32B32G32R32I:
728 case FORMAT_R32UI:
729 case FORMAT_G32R32UI:
730 case FORMAT_A32B32G32R32UI:
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400731 break;
732 default:
733 ASSERT(false);
734 }
735 }
736 }
737
Nicolas Capens4f172c72016-01-13 08:34:30 -0500738 Int4 PixelProgram::enableMask(const Shader::Instruction *instruction)
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400739 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500740 Int4 enable = instruction->analysisBranch ? Int4(enableStack[enableIndex]) : Int4(0xFFFFFFFF);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400741
742 if(!whileTest)
743 {
744 if(shader->containsBreakInstruction() && instruction->analysisBreak)
745 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500746 enable &= enableBreak;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400747 }
748
749 if(shader->containsContinueInstruction() && instruction->analysisContinue)
750 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500751 enable &= enableContinue;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400752 }
753
754 if(shader->containsLeaveInstruction() && instruction->analysisLeave)
755 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500756 enable &= enableLeave;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400757 }
758 }
759
760 return enable;
761 }
762
Nicolas Capensc2534f42016-04-04 11:13:24 -0400763 Vector4f PixelProgram::fetchRegister(const Src &src, unsigned int offset)
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400764 {
765 Vector4f reg;
Nicolas Capens5d961882016-01-01 23:18:14 -0500766 unsigned int i = src.index + offset;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400767
768 switch(src.type)
769 {
770 case Shader::PARAMETER_TEMP:
771 if(src.rel.type == Shader::PARAMETER_VOID)
772 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500773 reg = r[i];
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400774 }
775 else
776 {
Alexis Hetu2c2a7b22015-10-27 16:12:11 -0400777 Int a = relativeAddress(src, src.bufferIndex);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400778
Nicolas Capens4f172c72016-01-13 08:34:30 -0500779 reg = r[i + a];
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400780 }
781 break;
782 case Shader::PARAMETER_INPUT:
783 {
784 if(src.rel.type == Shader::PARAMETER_VOID) // Not relative
785 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500786 reg = v[i];
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400787 }
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400788 else
789 {
Alexis Hetu2c2a7b22015-10-27 16:12:11 -0400790 Int a = relativeAddress(src, src.bufferIndex);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400791
Nicolas Capens4f172c72016-01-13 08:34:30 -0500792 reg = v[i + a];
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400793 }
794 }
795 break;
796 case Shader::PARAMETER_CONST:
Nicolas Capens4f172c72016-01-13 08:34:30 -0500797 reg = readConstant(src, offset);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400798 break;
799 case Shader::PARAMETER_TEXTURE:
Nicolas Capens4f172c72016-01-13 08:34:30 -0500800 reg = v[2 + i];
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400801 break;
802 case Shader::PARAMETER_MISCTYPE:
Nicolas Capens4f172c72016-01-13 08:34:30 -0500803 if(src.index == 0) reg = vPos;
804 if(src.index == 1) reg = vFace;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400805 break;
806 case Shader::PARAMETER_SAMPLER:
807 if(src.rel.type == Shader::PARAMETER_VOID)
808 {
809 reg.x = As<Float4>(Int4(i));
810 }
811 else if(src.rel.type == Shader::PARAMETER_TEMP)
812 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500813 reg.x = As<Float4>(Int4(i) + As<Int4>(r[src.rel.index].x));
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400814 }
815 return reg;
816 case Shader::PARAMETER_PREDICATE: return reg; // Dummy
817 case Shader::PARAMETER_VOID: return reg; // Dummy
818 case Shader::PARAMETER_FLOAT4LITERAL:
819 reg.x = Float4(src.value[0]);
820 reg.y = Float4(src.value[1]);
821 reg.z = Float4(src.value[2]);
822 reg.w = Float4(src.value[3]);
823 break;
824 case Shader::PARAMETER_CONSTINT: return reg; // Dummy
825 case Shader::PARAMETER_CONSTBOOL: return reg; // Dummy
826 case Shader::PARAMETER_LOOP: return reg; // Dummy
827 case Shader::PARAMETER_COLOROUT:
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500828 if(src.rel.type == Shader::PARAMETER_VOID) // Not relative
829 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500830 reg = oC[i];
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500831 }
832 else
833 {
Alexis Hetu2c2a7b22015-10-27 16:12:11 -0400834 Int a = relativeAddress(src, src.bufferIndex);
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500835
Nicolas Capens4f172c72016-01-13 08:34:30 -0500836 reg = oC[i + a];
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500837 }
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400838 break;
839 case Shader::PARAMETER_DEPTHOUT:
Nicolas Capens4f172c72016-01-13 08:34:30 -0500840 reg.x = oDepth;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400841 break;
842 default:
843 ASSERT(false);
844 }
845
846 const Float4 &x = reg[(src.swizzle >> 0) & 0x3];
847 const Float4 &y = reg[(src.swizzle >> 2) & 0x3];
848 const Float4 &z = reg[(src.swizzle >> 4) & 0x3];
849 const Float4 &w = reg[(src.swizzle >> 6) & 0x3];
850
851 Vector4f mod;
852
853 switch(src.modifier)
854 {
855 case Shader::MODIFIER_NONE:
856 mod.x = x;
857 mod.y = y;
858 mod.z = z;
859 mod.w = w;
860 break;
861 case Shader::MODIFIER_NEGATE:
862 mod.x = -x;
863 mod.y = -y;
864 mod.z = -z;
865 mod.w = -w;
866 break;
867 case Shader::MODIFIER_ABS:
868 mod.x = Abs(x);
869 mod.y = Abs(y);
870 mod.z = Abs(z);
871 mod.w = Abs(w);
872 break;
873 case Shader::MODIFIER_ABS_NEGATE:
874 mod.x = -Abs(x);
875 mod.y = -Abs(y);
876 mod.z = -Abs(z);
877 mod.w = -Abs(w);
878 break;
879 case Shader::MODIFIER_NOT:
880 mod.x = As<Float4>(As<Int4>(x) ^ Int4(0xFFFFFFFF));
881 mod.y = As<Float4>(As<Int4>(y) ^ Int4(0xFFFFFFFF));
882 mod.z = As<Float4>(As<Int4>(z) ^ Int4(0xFFFFFFFF));
883 mod.w = As<Float4>(As<Int4>(w) ^ Int4(0xFFFFFFFF));
884 break;
885 default:
886 ASSERT(false);
887 }
888
889 return mod;
890 }
891
Alexis Hetu2c2a7b22015-10-27 16:12:11 -0400892 RValue<Pointer<Byte>> PixelProgram::uniformAddress(int bufferIndex, unsigned int index)
893 {
894 if(bufferIndex == -1)
895 {
896 return data + OFFSET(DrawData, ps.c[index]);
897 }
898 else
899 {
900 return *Pointer<Pointer<Byte>>(data + OFFSET(DrawData, ps.u[bufferIndex])) + index;
901 }
902 }
903
904 RValue<Pointer<Byte>> PixelProgram::uniformAddress(int bufferIndex, unsigned int index, Int& offset)
905 {
906 return uniformAddress(bufferIndex, index) + offset * sizeof(float4);
907 }
908
Nicolas Capens4f172c72016-01-13 08:34:30 -0500909 Vector4f PixelProgram::readConstant(const Src &src, unsigned int offset)
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400910 {
911 Vector4f c;
Nicolas Capens5d961882016-01-01 23:18:14 -0500912 unsigned int i = src.index + offset;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400913
914 if(src.rel.type == Shader::PARAMETER_VOID) // Not relative
915 {
Alexis Hetu2c2a7b22015-10-27 16:12:11 -0400916 c.x = c.y = c.z = c.w = *Pointer<Float4>(uniformAddress(src.bufferIndex, i));
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400917
918 c.x = c.x.xxxx;
919 c.y = c.y.yyyy;
920 c.z = c.z.zzzz;
921 c.w = c.w.wwww;
922
923 if(shader->containsDefineInstruction()) // Constant may be known at compile time
924 {
925 for(size_t j = 0; j < shader->getLength(); j++)
926 {
927 const Shader::Instruction &instruction = *shader->getInstruction(j);
928
929 if(instruction.opcode == Shader::OPCODE_DEF)
930 {
931 if(instruction.dst.index == i)
932 {
933 c.x = Float4(instruction.src[0].value[0]);
934 c.y = Float4(instruction.src[0].value[1]);
935 c.z = Float4(instruction.src[0].value[2]);
936 c.w = Float4(instruction.src[0].value[3]);
937
938 break;
939 }
940 }
941 }
942 }
943 }
944 else if(src.rel.type == Shader::PARAMETER_LOOP)
945 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500946 Int loopCounter = aL[loopDepth];
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400947
Alexis Hetu2c2a7b22015-10-27 16:12:11 -0400948 c.x = c.y = c.z = c.w = *Pointer<Float4>(uniformAddress(src.bufferIndex, i, loopCounter));
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400949
950 c.x = c.x.xxxx;
951 c.y = c.y.yyyy;
952 c.z = c.z.zzzz;
953 c.w = c.w.wwww;
954 }
955 else
956 {
Alexis Hetu2c2a7b22015-10-27 16:12:11 -0400957 Int a = relativeAddress(src, src.bufferIndex);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400958
Alexis Hetu2c2a7b22015-10-27 16:12:11 -0400959 c.x = c.y = c.z = c.w = *Pointer<Float4>(uniformAddress(src.bufferIndex, i, a));
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400960
961 c.x = c.x.xxxx;
962 c.y = c.y.yyyy;
963 c.z = c.z.zzzz;
964 c.w = c.w.wwww;
965 }
966
967 return c;
968 }
969
Alexis Hetu2c2a7b22015-10-27 16:12:11 -0400970 Int PixelProgram::relativeAddress(const Shader::Parameter &var, int bufferIndex)
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400971 {
972 ASSERT(var.rel.deterministic);
973
974 if(var.rel.type == Shader::PARAMETER_TEMP)
975 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500976 return As<Int>(Extract(r[var.rel.index].x, 0)) * var.rel.scale;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400977 }
978 else if(var.rel.type == Shader::PARAMETER_INPUT)
979 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500980 return As<Int>(Extract(v[var.rel.index].x, 0)) * var.rel.scale;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400981 }
982 else if(var.rel.type == Shader::PARAMETER_OUTPUT)
983 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500984 return As<Int>(Extract(oC[var.rel.index].x, 0)) * var.rel.scale;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400985 }
986 else if(var.rel.type == Shader::PARAMETER_CONST)
987 {
Alexis Hetu48be7352016-02-10 14:32:34 -0500988 return *Pointer<Int>(uniformAddress(bufferIndex, var.rel.index)) * var.rel.scale;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400989 }
Nicolas Capens907700d2016-01-20 17:09:28 -0500990 else if(var.rel.type == Shader::PARAMETER_LOOP)
991 {
992 return aL[loopDepth];
993 }
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400994 else ASSERT(false);
995
996 return 0;
997 }
998
999 Float4 PixelProgram::linearToSRGB(const Float4 &x) // Approximates x^(1.0/2.2)
1000 {
1001 Float4 sqrtx = Rcp_pp(RcpSqrt_pp(x));
1002 Float4 sRGB = sqrtx * Float4(1.14f) - x * Float4(0.14f);
1003
1004 return Min(Max(sRGB, Float4(0.0f)), Float4(1.0f));
1005 }
1006
Nicolas Capens4f172c72016-01-13 08:34:30 -05001007 void PixelProgram::M3X2(Vector4f &dst, Vector4f &src0, const Src &src1)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001008 {
Nicolas Capensc2534f42016-04-04 11:13:24 -04001009 Vector4f row0 = fetchRegister(src1, 0);
1010 Vector4f row1 = fetchRegister(src1, 1);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001011
1012 dst.x = dot3(src0, row0);
1013 dst.y = dot3(src0, row1);
1014 }
1015
Nicolas Capens4f172c72016-01-13 08:34:30 -05001016 void PixelProgram::M3X3(Vector4f &dst, Vector4f &src0, const Src &src1)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001017 {
Nicolas Capensc2534f42016-04-04 11:13:24 -04001018 Vector4f row0 = fetchRegister(src1, 0);
1019 Vector4f row1 = fetchRegister(src1, 1);
1020 Vector4f row2 = fetchRegister(src1, 2);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001021
1022 dst.x = dot3(src0, row0);
1023 dst.y = dot3(src0, row1);
1024 dst.z = dot3(src0, row2);
1025 }
1026
Nicolas Capens4f172c72016-01-13 08:34:30 -05001027 void PixelProgram::M3X4(Vector4f &dst, Vector4f &src0, const Src &src1)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001028 {
Nicolas Capensc2534f42016-04-04 11:13:24 -04001029 Vector4f row0 = fetchRegister(src1, 0);
1030 Vector4f row1 = fetchRegister(src1, 1);
1031 Vector4f row2 = fetchRegister(src1, 2);
1032 Vector4f row3 = fetchRegister(src1, 3);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001033
1034 dst.x = dot3(src0, row0);
1035 dst.y = dot3(src0, row1);
1036 dst.z = dot3(src0, row2);
1037 dst.w = dot3(src0, row3);
1038 }
1039
Nicolas Capens4f172c72016-01-13 08:34:30 -05001040 void PixelProgram::M4X3(Vector4f &dst, Vector4f &src0, const Src &src1)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001041 {
Nicolas Capensc2534f42016-04-04 11:13:24 -04001042 Vector4f row0 = fetchRegister(src1, 0);
1043 Vector4f row1 = fetchRegister(src1, 1);
1044 Vector4f row2 = fetchRegister(src1, 2);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001045
1046 dst.x = dot4(src0, row0);
1047 dst.y = dot4(src0, row1);
1048 dst.z = dot4(src0, row2);
1049 }
1050
Nicolas Capens4f172c72016-01-13 08:34:30 -05001051 void PixelProgram::M4X4(Vector4f &dst, Vector4f &src0, const Src &src1)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001052 {
Nicolas Capensc2534f42016-04-04 11:13:24 -04001053 Vector4f row0 = fetchRegister(src1, 0);
1054 Vector4f row1 = fetchRegister(src1, 1);
1055 Vector4f row2 = fetchRegister(src1, 2);
1056 Vector4f row3 = fetchRegister(src1, 3);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001057
1058 dst.x = dot4(src0, row0);
1059 dst.y = dot4(src0, row1);
1060 dst.z = dot4(src0, row2);
1061 dst.w = dot4(src0, row3);
1062 }
1063
Nicolas Capens4f172c72016-01-13 08:34:30 -05001064 void PixelProgram::TEXLD(Vector4f &dst, Vector4f &src0, const Src &src1, bool project, bool bias)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001065 {
Nicolas Capensc2534f42016-04-04 11:13:24 -04001066 sampleTexture(dst, src1, src0.x, src0.y, src0.z, src0.w, src0, src0, project, bias ? Bias : Implicit);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001067 }
1068
Nicolas Capens4f172c72016-01-13 08:34:30 -05001069 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 -04001070 {
1071 UNIMPLEMENTED();
1072 }
1073
Nicolas Capens4f172c72016-01-13 08:34:30 -05001074 void PixelProgram::TEXLDL(Vector4f &dst, Vector4f &src0, const Src &src1, Vector4f &offset, bool project, bool bias)
Alexis Hetu25d47fc2015-10-22 13:58:32 -04001075 {
1076 UNIMPLEMENTED();
1077 }
1078
Nicolas Capens4f172c72016-01-13 08:34:30 -05001079 void PixelProgram::TEXELFETCH(Vector4f &dst, Vector4f &src0, const Src& src1, Vector4f &src2)
Alexis Hetu25d47fc2015-10-22 13:58:32 -04001080 {
1081 UNIMPLEMENTED();
1082 }
1083
Nicolas Capens4f172c72016-01-13 08:34:30 -05001084 void PixelProgram::TEXELFETCH(Vector4f &dst, Vector4f &src0, const Src& src1, Vector4f &src2, Vector4f &offset)
Alexis Hetu25d47fc2015-10-22 13:58:32 -04001085 {
1086 UNIMPLEMENTED();
1087 }
1088
Nicolas Capens4f172c72016-01-13 08:34:30 -05001089 void PixelProgram::TEXGRAD(Vector4f &dst, Vector4f &src0, const Src& src1, Vector4f &src2, Vector4f &src3)
Alexis Hetu25d47fc2015-10-22 13:58:32 -04001090 {
1091 UNIMPLEMENTED();
1092 }
1093
Nicolas Capens4f172c72016-01-13 08:34:30 -05001094 void PixelProgram::TEXGRAD(Vector4f &dst, Vector4f &src0, const Src& src1, Vector4f &src2, Vector4f &src3, Vector4f &offset)
Alexis Hetu25d47fc2015-10-22 13:58:32 -04001095 {
1096 UNIMPLEMENTED();
1097 }
1098
Nicolas Capensc2534f42016-04-04 11:13:24 -04001099 void PixelProgram::TEXLDD(Vector4f &dst, Vector4f &src0, const Src &src1, Vector4f &src2, Vector4f &src3, bool project)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001100 {
Nicolas Capensc2534f42016-04-04 11:13:24 -04001101 sampleTexture(dst, src1, src0.x, src0.y, src0.z, src0.w, src2, src3, project, Grad);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001102 }
1103
Nicolas Capensc2534f42016-04-04 11:13:24 -04001104 void PixelProgram::TEXLDL(Vector4f &dst, Vector4f &src0, const Src &src1, bool project)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001105 {
Nicolas Capensc2534f42016-04-04 11:13:24 -04001106 sampleTexture(dst, src1, src0.x, src0.y, src0.z, src0.w, src0, src0, project, Lod);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001107 }
1108
Nicolas Capens4f172c72016-01-13 08:34:30 -05001109 void PixelProgram::TEXSIZE(Vector4f &dst, Float4 &lod, const Src &src1)
Alexis Hetu9bcb31d2015-07-22 17:03:26 -04001110 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001111 Pointer<Byte> textureMipmap = data + OFFSET(DrawData, mipmap) + src1.index * sizeof(Texture) + OFFSET(Texture, mipmap);
Alexis Hetu9bcb31d2015-07-22 17:03:26 -04001112 for(int i = 0; i < 4; ++i)
1113 {
1114 Pointer<Byte> mipmap = textureMipmap + (As<Int>(Extract(lod, i)) + Int(1)) * sizeof(Mipmap);
1115 dst.x = Insert(dst.x, As<Float>(Int(*Pointer<Short>(mipmap + OFFSET(Mipmap, width)))), i);
1116 dst.y = Insert(dst.y, As<Float>(Int(*Pointer<Short>(mipmap + OFFSET(Mipmap, height)))), i);
1117 dst.z = Insert(dst.z, As<Float>(Int(*Pointer<Short>(mipmap + OFFSET(Mipmap, depth)))), i);
1118 }
1119 }
1120
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001121 void PixelProgram::TEXKILL(Int cMask[4], Vector4f &src, unsigned char mask)
1122 {
1123 Int kill = -1;
1124
1125 if(mask & 0x1) kill &= SignMask(CmpNLT(src.x, Float4(0.0f)));
1126 if(mask & 0x2) kill &= SignMask(CmpNLT(src.y, Float4(0.0f)));
1127 if(mask & 0x4) kill &= SignMask(CmpNLT(src.z, Float4(0.0f)));
1128 if(mask & 0x8) kill &= SignMask(CmpNLT(src.w, Float4(0.0f)));
1129
1130 // FIXME: Dynamic branching affects TEXKILL?
1131 // if(shader->containsDynamicBranching())
1132 // {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001133 // kill = ~SignMask(enableMask());
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001134 // }
1135
1136 for(unsigned int q = 0; q < state.multiSample; q++)
1137 {
1138 cMask[q] &= kill;
1139 }
1140
1141 // FIXME: Branch to end of shader if all killed?
1142 }
1143
Nicolas Capens4f172c72016-01-13 08:34:30 -05001144 void PixelProgram::DISCARD(Int cMask[4], const Shader::Instruction *instruction)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001145 {
1146 Int kill = 0;
1147
1148 if(shader->containsDynamicBranching())
1149 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001150 kill = ~SignMask(enableMask(instruction));
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001151 }
1152
1153 for(unsigned int q = 0; q < state.multiSample; q++)
1154 {
1155 cMask[q] &= kill;
1156 }
1157
1158 // FIXME: Branch to end of shader if all killed?
1159 }
1160
1161 void PixelProgram::DFDX(Vector4f &dst, Vector4f &src)
1162 {
1163 dst.x = src.x.yyww - src.x.xxzz;
1164 dst.y = src.y.yyww - src.y.xxzz;
1165 dst.z = src.z.yyww - src.z.xxzz;
1166 dst.w = src.w.yyww - src.w.xxzz;
1167 }
1168
1169 void PixelProgram::DFDY(Vector4f &dst, Vector4f &src)
1170 {
1171 dst.x = src.x.zwzw - src.x.xyxy;
1172 dst.y = src.y.zwzw - src.y.xyxy;
1173 dst.z = src.z.zwzw - src.z.xyxy;
1174 dst.w = src.w.zwzw - src.w.xyxy;
1175 }
1176
1177 void PixelProgram::FWIDTH(Vector4f &dst, Vector4f &src)
1178 {
1179 // abs(dFdx(src)) + abs(dFdy(src));
1180 dst.x = Abs(src.x.yyww - src.x.xxzz) + Abs(src.x.zwzw - src.x.xyxy);
1181 dst.y = Abs(src.y.yyww - src.y.xxzz) + Abs(src.y.zwzw - src.y.xyxy);
1182 dst.z = Abs(src.z.yyww - src.z.xxzz) + Abs(src.z.zwzw - src.z.xyxy);
1183 dst.w = Abs(src.w.yyww - src.w.xxzz) + Abs(src.w.zwzw - src.w.xyxy);
1184 }
1185
Nicolas Capens4f172c72016-01-13 08:34:30 -05001186 void PixelProgram::BREAK()
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001187 {
1188 llvm::BasicBlock *deadBlock = Nucleus::createBasicBlock();
1189 llvm::BasicBlock *endBlock = loopRepEndBlock[loopRepDepth - 1];
1190
1191 if(breakDepth == 0)
1192 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001193 enableIndex = enableIndex - breakDepth;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001194 Nucleus::createBr(endBlock);
1195 }
1196 else
1197 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001198 enableBreak = enableBreak & ~enableStack[enableIndex];
1199 Bool allBreak = SignMask(enableBreak) == 0x0;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001200
Nicolas Capens4f172c72016-01-13 08:34:30 -05001201 enableIndex = enableIndex - breakDepth;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001202 branch(allBreak, endBlock, deadBlock);
1203 }
1204
1205 Nucleus::setInsertBlock(deadBlock);
Nicolas Capens4f172c72016-01-13 08:34:30 -05001206 enableIndex = enableIndex + breakDepth;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001207 }
1208
Nicolas Capens4f172c72016-01-13 08:34:30 -05001209 void PixelProgram::BREAKC(Vector4f &src0, Vector4f &src1, Control control)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001210 {
1211 Int4 condition;
1212
1213 switch(control)
1214 {
1215 case Shader::CONTROL_GT: condition = CmpNLE(src0.x, src1.x); break;
1216 case Shader::CONTROL_EQ: condition = CmpEQ(src0.x, src1.x); break;
1217 case Shader::CONTROL_GE: condition = CmpNLT(src0.x, src1.x); break;
1218 case Shader::CONTROL_LT: condition = CmpLT(src0.x, src1.x); break;
1219 case Shader::CONTROL_NE: condition = CmpNEQ(src0.x, src1.x); break;
1220 case Shader::CONTROL_LE: condition = CmpLE(src0.x, src1.x); break;
1221 default:
1222 ASSERT(false);
1223 }
1224
Nicolas Capens4f172c72016-01-13 08:34:30 -05001225 BREAK(condition);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001226 }
1227
Nicolas Capens4f172c72016-01-13 08:34:30 -05001228 void PixelProgram::BREAKP(const Src &predicateRegister) // FIXME: Factor out parts common with BREAKC
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001229 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001230 Int4 condition = As<Int4>(p0[predicateRegister.swizzle & 0x3]);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001231
1232 if(predicateRegister.modifier == Shader::MODIFIER_NOT)
1233 {
1234 condition = ~condition;
1235 }
1236
Nicolas Capens4f172c72016-01-13 08:34:30 -05001237 BREAK(condition);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001238 }
1239
Nicolas Capens4f172c72016-01-13 08:34:30 -05001240 void PixelProgram::BREAK(Int4 &condition)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001241 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001242 condition &= enableStack[enableIndex];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001243
1244 llvm::BasicBlock *continueBlock = Nucleus::createBasicBlock();
1245 llvm::BasicBlock *endBlock = loopRepEndBlock[loopRepDepth - 1];
1246
Nicolas Capens4f172c72016-01-13 08:34:30 -05001247 enableBreak = enableBreak & ~condition;
1248 Bool allBreak = SignMask(enableBreak) == 0x0;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001249
Nicolas Capens4f172c72016-01-13 08:34:30 -05001250 enableIndex = enableIndex - breakDepth;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001251 branch(allBreak, endBlock, continueBlock);
1252
1253 Nucleus::setInsertBlock(continueBlock);
Nicolas Capens4f172c72016-01-13 08:34:30 -05001254 enableIndex = enableIndex + breakDepth;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001255 }
1256
Nicolas Capens4f172c72016-01-13 08:34:30 -05001257 void PixelProgram::CONTINUE()
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001258 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001259 enableContinue = enableContinue & ~enableStack[enableIndex];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001260 }
1261
1262 void PixelProgram::TEST()
1263 {
1264 whileTest = true;
1265 }
1266
Nicolas Capens4f172c72016-01-13 08:34:30 -05001267 void PixelProgram::CALL(int labelIndex, int callSiteIndex)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001268 {
1269 if(!labelBlock[labelIndex])
1270 {
1271 labelBlock[labelIndex] = Nucleus::createBasicBlock();
1272 }
1273
1274 if(callRetBlock[labelIndex].size() > 1)
1275 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001276 callStack[stackIndex++] = UInt(callSiteIndex);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001277 }
1278
Nicolas Capens4f172c72016-01-13 08:34:30 -05001279 Int4 restoreLeave = enableLeave;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001280
1281 Nucleus::createBr(labelBlock[labelIndex]);
1282 Nucleus::setInsertBlock(callRetBlock[labelIndex][callSiteIndex]);
1283
Nicolas Capens4f172c72016-01-13 08:34:30 -05001284 enableLeave = restoreLeave;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001285 }
1286
Nicolas Capens4f172c72016-01-13 08:34:30 -05001287 void PixelProgram::CALLNZ(int labelIndex, int callSiteIndex, const Src &src)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001288 {
1289 if(src.type == Shader::PARAMETER_CONSTBOOL)
1290 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001291 CALLNZb(labelIndex, callSiteIndex, src);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001292 }
1293 else if(src.type == Shader::PARAMETER_PREDICATE)
1294 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001295 CALLNZp(labelIndex, callSiteIndex, src);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001296 }
1297 else ASSERT(false);
1298 }
1299
Nicolas Capens4f172c72016-01-13 08:34:30 -05001300 void PixelProgram::CALLNZb(int labelIndex, int callSiteIndex, const Src &boolRegister)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001301 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001302 Bool condition = (*Pointer<Byte>(data + OFFSET(DrawData, ps.b[boolRegister.index])) != Byte(0)); // FIXME
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001303
1304 if(boolRegister.modifier == Shader::MODIFIER_NOT)
1305 {
1306 condition = !condition;
1307 }
1308
1309 if(!labelBlock[labelIndex])
1310 {
1311 labelBlock[labelIndex] = Nucleus::createBasicBlock();
1312 }
1313
1314 if(callRetBlock[labelIndex].size() > 1)
1315 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001316 callStack[stackIndex++] = UInt(callSiteIndex);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001317 }
1318
Nicolas Capens4f172c72016-01-13 08:34:30 -05001319 Int4 restoreLeave = enableLeave;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001320
1321 branch(condition, labelBlock[labelIndex], callRetBlock[labelIndex][callSiteIndex]);
1322 Nucleus::setInsertBlock(callRetBlock[labelIndex][callSiteIndex]);
1323
Nicolas Capens4f172c72016-01-13 08:34:30 -05001324 enableLeave = restoreLeave;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001325 }
1326
Nicolas Capens4f172c72016-01-13 08:34:30 -05001327 void PixelProgram::CALLNZp(int labelIndex, int callSiteIndex, const Src &predicateRegister)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001328 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001329 Int4 condition = As<Int4>(p0[predicateRegister.swizzle & 0x3]);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001330
1331 if(predicateRegister.modifier == Shader::MODIFIER_NOT)
1332 {
1333 condition = ~condition;
1334 }
1335
Nicolas Capens4f172c72016-01-13 08:34:30 -05001336 condition &= enableStack[enableIndex];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001337
1338 if(!labelBlock[labelIndex])
1339 {
1340 labelBlock[labelIndex] = Nucleus::createBasicBlock();
1341 }
1342
1343 if(callRetBlock[labelIndex].size() > 1)
1344 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001345 callStack[stackIndex++] = UInt(callSiteIndex);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001346 }
1347
Nicolas Capens4f172c72016-01-13 08:34:30 -05001348 enableIndex++;
1349 enableStack[enableIndex] = condition;
1350 Int4 restoreLeave = enableLeave;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001351
1352 Bool notAllFalse = SignMask(condition) != 0;
1353 branch(notAllFalse, labelBlock[labelIndex], callRetBlock[labelIndex][callSiteIndex]);
1354 Nucleus::setInsertBlock(callRetBlock[labelIndex][callSiteIndex]);
1355
Nicolas Capens4f172c72016-01-13 08:34:30 -05001356 enableIndex--;
1357 enableLeave = restoreLeave;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001358 }
1359
Nicolas Capens4f172c72016-01-13 08:34:30 -05001360 void PixelProgram::ELSE()
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001361 {
1362 ifDepth--;
1363
1364 llvm::BasicBlock *falseBlock = ifFalseBlock[ifDepth];
1365 llvm::BasicBlock *endBlock = Nucleus::createBasicBlock();
1366
1367 if(isConditionalIf[ifDepth])
1368 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001369 Int4 condition = ~enableStack[enableIndex] & enableStack[enableIndex - 1];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001370 Bool notAllFalse = SignMask(condition) != 0;
1371
1372 branch(notAllFalse, falseBlock, endBlock);
1373
Nicolas Capens4f172c72016-01-13 08:34:30 -05001374 enableStack[enableIndex] = ~enableStack[enableIndex] & enableStack[enableIndex - 1];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001375 }
1376 else
1377 {
1378 Nucleus::createBr(endBlock);
1379 Nucleus::setInsertBlock(falseBlock);
1380 }
1381
1382 ifFalseBlock[ifDepth] = endBlock;
1383
1384 ifDepth++;
1385 }
1386
Nicolas Capens4f172c72016-01-13 08:34:30 -05001387 void PixelProgram::ENDIF()
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001388 {
1389 ifDepth--;
1390
1391 llvm::BasicBlock *endBlock = ifFalseBlock[ifDepth];
1392
1393 Nucleus::createBr(endBlock);
1394 Nucleus::setInsertBlock(endBlock);
1395
1396 if(isConditionalIf[ifDepth])
1397 {
1398 breakDepth--;
Nicolas Capens4f172c72016-01-13 08:34:30 -05001399 enableIndex--;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001400 }
1401 }
1402
Nicolas Capens4f172c72016-01-13 08:34:30 -05001403 void PixelProgram::ENDLOOP()
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001404 {
1405 loopRepDepth--;
1406
Nicolas Capens4f172c72016-01-13 08:34:30 -05001407 aL[loopDepth] = aL[loopDepth] + increment[loopDepth]; // FIXME: +=
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001408
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 loopDepth--;
1416 enableBreak = Int4(0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001417 }
1418
Nicolas Capens4f172c72016-01-13 08:34:30 -05001419 void PixelProgram::ENDREP()
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001420 {
1421 loopRepDepth--;
1422
1423 llvm::BasicBlock *testBlock = loopRepTestBlock[loopRepDepth];
1424 llvm::BasicBlock *endBlock = loopRepEndBlock[loopRepDepth];
1425
1426 Nucleus::createBr(testBlock);
1427 Nucleus::setInsertBlock(endBlock);
1428
Nicolas Capens4f172c72016-01-13 08:34:30 -05001429 loopDepth--;
1430 enableBreak = Int4(0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001431 }
1432
Nicolas Capens4f172c72016-01-13 08:34:30 -05001433 void PixelProgram::ENDWHILE()
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001434 {
1435 loopRepDepth--;
1436
1437 llvm::BasicBlock *testBlock = loopRepTestBlock[loopRepDepth];
1438 llvm::BasicBlock *endBlock = loopRepEndBlock[loopRepDepth];
1439
1440 Nucleus::createBr(testBlock);
1441 Nucleus::setInsertBlock(endBlock);
1442
Nicolas Capens4f172c72016-01-13 08:34:30 -05001443 enableIndex--;
1444 enableBreak = Int4(0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001445 whileTest = false;
1446 }
1447
Nicolas Capens4f172c72016-01-13 08:34:30 -05001448 void PixelProgram::IF(const Src &src)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001449 {
1450 if(src.type == Shader::PARAMETER_CONSTBOOL)
1451 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001452 IFb(src);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001453 }
1454 else if(src.type == Shader::PARAMETER_PREDICATE)
1455 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001456 IFp(src);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001457 }
1458 else
1459 {
Nicolas Capensc2534f42016-04-04 11:13:24 -04001460 Int4 condition = As<Int4>(fetchRegister(src).x);
Nicolas Capens4f172c72016-01-13 08:34:30 -05001461 IF(condition);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001462 }
1463 }
1464
Nicolas Capens4f172c72016-01-13 08:34:30 -05001465 void PixelProgram::IFb(const Src &boolRegister)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001466 {
1467 ASSERT(ifDepth < 24 + 4);
1468
Nicolas Capens4f172c72016-01-13 08:34:30 -05001469 Bool condition = (*Pointer<Byte>(data + OFFSET(DrawData, ps.b[boolRegister.index])) != Byte(0)); // FIXME
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001470
1471 if(boolRegister.modifier == Shader::MODIFIER_NOT)
1472 {
1473 condition = !condition;
1474 }
1475
1476 llvm::BasicBlock *trueBlock = Nucleus::createBasicBlock();
1477 llvm::BasicBlock *falseBlock = Nucleus::createBasicBlock();
1478
1479 branch(condition, trueBlock, falseBlock);
1480
1481 isConditionalIf[ifDepth] = false;
1482 ifFalseBlock[ifDepth] = falseBlock;
1483
1484 ifDepth++;
1485 }
1486
Nicolas Capens4f172c72016-01-13 08:34:30 -05001487 void PixelProgram::IFp(const Src &predicateRegister)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001488 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001489 Int4 condition = As<Int4>(p0[predicateRegister.swizzle & 0x3]);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001490
1491 if(predicateRegister.modifier == Shader::MODIFIER_NOT)
1492 {
1493 condition = ~condition;
1494 }
1495
Nicolas Capens4f172c72016-01-13 08:34:30 -05001496 IF(condition);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001497 }
1498
Nicolas Capens4f172c72016-01-13 08:34:30 -05001499 void PixelProgram::IFC(Vector4f &src0, Vector4f &src1, Control control)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001500 {
1501 Int4 condition;
1502
1503 switch(control)
1504 {
1505 case Shader::CONTROL_GT: condition = CmpNLE(src0.x, src1.x); break;
1506 case Shader::CONTROL_EQ: condition = CmpEQ(src0.x, src1.x); break;
1507 case Shader::CONTROL_GE: condition = CmpNLT(src0.x, src1.x); break;
1508 case Shader::CONTROL_LT: condition = CmpLT(src0.x, src1.x); break;
1509 case Shader::CONTROL_NE: condition = CmpNEQ(src0.x, src1.x); break;
1510 case Shader::CONTROL_LE: condition = CmpLE(src0.x, src1.x); break;
1511 default:
1512 ASSERT(false);
1513 }
1514
Nicolas Capens4f172c72016-01-13 08:34:30 -05001515 IF(condition);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001516 }
1517
Nicolas Capens4f172c72016-01-13 08:34:30 -05001518 void PixelProgram::IF(Int4 &condition)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001519 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001520 condition &= enableStack[enableIndex];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001521
Nicolas Capens4f172c72016-01-13 08:34:30 -05001522 enableIndex++;
1523 enableStack[enableIndex] = condition;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001524
1525 llvm::BasicBlock *trueBlock = Nucleus::createBasicBlock();
1526 llvm::BasicBlock *falseBlock = Nucleus::createBasicBlock();
1527
1528 Bool notAllFalse = SignMask(condition) != 0;
1529
1530 branch(notAllFalse, trueBlock, falseBlock);
1531
1532 isConditionalIf[ifDepth] = true;
1533 ifFalseBlock[ifDepth] = falseBlock;
1534
1535 ifDepth++;
1536 breakDepth++;
1537 }
1538
1539 void PixelProgram::LABEL(int labelIndex)
1540 {
1541 if(!labelBlock[labelIndex])
1542 {
1543 labelBlock[labelIndex] = Nucleus::createBasicBlock();
1544 }
1545
1546 Nucleus::setInsertBlock(labelBlock[labelIndex]);
1547 currentLabel = labelIndex;
1548 }
1549
Nicolas Capens4f172c72016-01-13 08:34:30 -05001550 void PixelProgram::LOOP(const Src &integerRegister)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001551 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001552 loopDepth++;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001553
Nicolas Capens4f172c72016-01-13 08:34:30 -05001554 iteration[loopDepth] = *Pointer<Int>(data + OFFSET(DrawData, ps.i[integerRegister.index][0]));
1555 aL[loopDepth] = *Pointer<Int>(data + OFFSET(DrawData, ps.i[integerRegister.index][1]));
1556 increment[loopDepth] = *Pointer<Int>(data + OFFSET(DrawData, ps.i[integerRegister.index][2]));
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001557
Nicolas Capens4f172c72016-01-13 08:34:30 -05001558 // If(increment[loopDepth] == 0)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001559 // {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001560 // increment[loopDepth] = 1;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001561 // }
1562
1563 llvm::BasicBlock *loopBlock = Nucleus::createBasicBlock();
1564 llvm::BasicBlock *testBlock = Nucleus::createBasicBlock();
1565 llvm::BasicBlock *endBlock = Nucleus::createBasicBlock();
1566
1567 loopRepTestBlock[loopRepDepth] = testBlock;
1568 loopRepEndBlock[loopRepDepth] = endBlock;
1569
1570 // FIXME: jump(testBlock)
1571 Nucleus::createBr(testBlock);
1572 Nucleus::setInsertBlock(testBlock);
1573
Nicolas Capens4f172c72016-01-13 08:34:30 -05001574 branch(iteration[loopDepth] > 0, loopBlock, endBlock);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001575 Nucleus::setInsertBlock(loopBlock);
1576
Nicolas Capens4f172c72016-01-13 08:34:30 -05001577 iteration[loopDepth] = iteration[loopDepth] - 1; // FIXME: --
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001578
1579 loopRepDepth++;
1580 breakDepth = 0;
1581 }
1582
Nicolas Capens4f172c72016-01-13 08:34:30 -05001583 void PixelProgram::REP(const Src &integerRegister)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001584 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001585 loopDepth++;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001586
Nicolas Capens4f172c72016-01-13 08:34:30 -05001587 iteration[loopDepth] = *Pointer<Int>(data + OFFSET(DrawData, ps.i[integerRegister.index][0]));
1588 aL[loopDepth] = aL[loopDepth - 1];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001589
1590 llvm::BasicBlock *loopBlock = Nucleus::createBasicBlock();
1591 llvm::BasicBlock *testBlock = Nucleus::createBasicBlock();
1592 llvm::BasicBlock *endBlock = Nucleus::createBasicBlock();
1593
1594 loopRepTestBlock[loopRepDepth] = testBlock;
1595 loopRepEndBlock[loopRepDepth] = endBlock;
1596
1597 // FIXME: jump(testBlock)
1598 Nucleus::createBr(testBlock);
1599 Nucleus::setInsertBlock(testBlock);
1600
Nicolas Capens4f172c72016-01-13 08:34:30 -05001601 branch(iteration[loopDepth] > 0, loopBlock, endBlock);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001602 Nucleus::setInsertBlock(loopBlock);
1603
Nicolas Capens4f172c72016-01-13 08:34:30 -05001604 iteration[loopDepth] = iteration[loopDepth] - 1; // FIXME: --
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001605
1606 loopRepDepth++;
1607 breakDepth = 0;
1608 }
1609
Nicolas Capens4f172c72016-01-13 08:34:30 -05001610 void PixelProgram::WHILE(const Src &temporaryRegister)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001611 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001612 enableIndex++;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001613
1614 llvm::BasicBlock *loopBlock = Nucleus::createBasicBlock();
1615 llvm::BasicBlock *testBlock = Nucleus::createBasicBlock();
1616 llvm::BasicBlock *endBlock = Nucleus::createBasicBlock();
1617
1618 loopRepTestBlock[loopRepDepth] = testBlock;
1619 loopRepEndBlock[loopRepDepth] = endBlock;
1620
Nicolas Capens4f172c72016-01-13 08:34:30 -05001621 Int4 restoreBreak = enableBreak;
1622 Int4 restoreContinue = enableContinue;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001623
1624 // FIXME: jump(testBlock)
1625 Nucleus::createBr(testBlock);
1626 Nucleus::setInsertBlock(testBlock);
Nicolas Capens4f172c72016-01-13 08:34:30 -05001627 enableContinue = restoreContinue;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001628
Nicolas Capensc2534f42016-04-04 11:13:24 -04001629 const Vector4f &src = fetchRegister(temporaryRegister);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001630 Int4 condition = As<Int4>(src.x);
Nicolas Capens4f172c72016-01-13 08:34:30 -05001631 condition &= enableStack[enableIndex - 1];
1632 enableStack[enableIndex] = condition;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001633
1634 Bool notAllFalse = SignMask(condition) != 0;
1635 branch(notAllFalse, loopBlock, endBlock);
1636
1637 Nucleus::setInsertBlock(endBlock);
Nicolas Capens4f172c72016-01-13 08:34:30 -05001638 enableBreak = restoreBreak;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001639
1640 Nucleus::setInsertBlock(loopBlock);
1641
1642 loopRepDepth++;
1643 breakDepth = 0;
1644 }
1645
Nicolas Capens4f172c72016-01-13 08:34:30 -05001646 void PixelProgram::RET()
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001647 {
1648 if(currentLabel == -1)
1649 {
1650 returnBlock = Nucleus::createBasicBlock();
1651 Nucleus::createBr(returnBlock);
1652 }
1653 else
1654 {
1655 llvm::BasicBlock *unreachableBlock = Nucleus::createBasicBlock();
1656
1657 if(callRetBlock[currentLabel].size() > 1) // Pop the return destination from the call stack
1658 {
1659 // FIXME: Encapsulate
Nicolas Capens4f172c72016-01-13 08:34:30 -05001660 UInt index = callStack[--stackIndex];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001661
1662 llvm::Value *value = index.loadValue();
1663 llvm::Value *switchInst = Nucleus::createSwitch(value, unreachableBlock, (int)callRetBlock[currentLabel].size());
1664
1665 for(unsigned int i = 0; i < callRetBlock[currentLabel].size(); i++)
1666 {
1667 Nucleus::addSwitchCase(switchInst, i, callRetBlock[currentLabel][i]);
1668 }
1669 }
1670 else if(callRetBlock[currentLabel].size() == 1) // Jump directly to the unique return destination
1671 {
1672 Nucleus::createBr(callRetBlock[currentLabel][0]);
1673 }
1674 else // Function isn't called
1675 {
1676 Nucleus::createBr(unreachableBlock);
1677 }
1678
1679 Nucleus::setInsertBlock(unreachableBlock);
1680 Nucleus::createUnreachable();
1681 }
1682 }
1683
Nicolas Capens4f172c72016-01-13 08:34:30 -05001684 void PixelProgram::LEAVE()
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001685 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001686 enableLeave = enableLeave & ~enableStack[enableIndex];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001687
1688 // FIXME: Return from function if all instances left
1689 // FIXME: Use enableLeave in other control-flow constructs
1690 }
Nicolas Capens5d961882016-01-01 23:18:14 -05001691}