blob: 2984b9edab9efe0c72b3d6269a1220ac57bcd74b [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
Nicolas Capens8b124c12016-04-18 14:09:37 -040094 bool broadcastColor0 = true;
95
Alexis Hetuf2a8c372015-07-13 11:08:41 -040096 for(size_t i = 0; i < shader->getLength(); i++)
97 {
98 const Shader::Instruction *instruction = shader->getInstruction(i);
99 Shader::Opcode opcode = instruction->opcode;
100
101 if(opcode == Shader::OPCODE_DCL || opcode == Shader::OPCODE_DEF || opcode == Shader::OPCODE_DEFI || opcode == Shader::OPCODE_DEFB)
102 {
103 continue;
104 }
105
106 const Dst &dst = instruction->dst;
107 const Src &src0 = instruction->src[0];
108 const Src &src1 = instruction->src[1];
109 const Src &src2 = instruction->src[2];
110 const Src &src3 = instruction->src[3];
Alexis Hetu25d47fc2015-10-22 13:58:32 -0400111 const Src &src4 = instruction->src[4];
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400112
113 bool predicate = instruction->predicate;
114 Control control = instruction->control;
115 bool pp = dst.partialPrecision;
116 bool project = instruction->project;
117 bool bias = instruction->bias;
118
119 Vector4f d;
120 Vector4f s0;
121 Vector4f s1;
122 Vector4f s2;
123 Vector4f s3;
Alexis Hetu25d47fc2015-10-22 13:58:32 -0400124 Vector4f s4;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400125
126 if(opcode == Shader::OPCODE_TEXKILL) // Takes destination as input
127 {
128 if(dst.type == Shader::PARAMETER_TEXTURE)
129 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500130 d.x = v[2 + dst.index].x;
131 d.y = v[2 + dst.index].y;
132 d.z = v[2 + dst.index].z;
133 d.w = v[2 + dst.index].w;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400134 }
135 else
136 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500137 d = r[dst.index];
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400138 }
139 }
140
Nicolas Capensc2534f42016-04-04 11:13:24 -0400141 if(src0.type != Shader::PARAMETER_VOID) s0 = fetchRegister(src0);
142 if(src1.type != Shader::PARAMETER_VOID) s1 = fetchRegister(src1);
143 if(src2.type != Shader::PARAMETER_VOID) s2 = fetchRegister(src2);
144 if(src3.type != Shader::PARAMETER_VOID) s3 = fetchRegister(src3);
145 if(src4.type != Shader::PARAMETER_VOID) s4 = fetchRegister(src4);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400146
147 switch(opcode)
148 {
149 case Shader::OPCODE_PS_2_0: break;
150 case Shader::OPCODE_PS_2_x: break;
151 case Shader::OPCODE_PS_3_0: break;
152 case Shader::OPCODE_DEF: break;
153 case Shader::OPCODE_DCL: break;
154 case Shader::OPCODE_NOP: break;
155 case Shader::OPCODE_MOV: mov(d, s0); break;
Alexis Hetu8d78cf72015-08-28 14:24:45 -0400156 case Shader::OPCODE_NEG: neg(d, s0); break;
157 case Shader::OPCODE_INEG: ineg(d, s0); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400158 case Shader::OPCODE_F2B: f2b(d, s0); break;
159 case Shader::OPCODE_B2F: b2f(d, s0); break;
Alexis Hetu8d78cf72015-08-28 14:24:45 -0400160 case Shader::OPCODE_F2I: f2i(d, s0); break;
161 case Shader::OPCODE_I2F: i2f(d, s0); break;
162 case Shader::OPCODE_F2U: f2u(d, s0); break;
163 case Shader::OPCODE_U2F: u2f(d, s0); break;
164 case Shader::OPCODE_I2B: i2b(d, s0); break;
165 case Shader::OPCODE_B2I: b2i(d, s0); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400166 case Shader::OPCODE_ADD: add(d, s0, s1); break;
Alexis Hetu8d78cf72015-08-28 14:24:45 -0400167 case Shader::OPCODE_IADD: iadd(d, s0, s1); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400168 case Shader::OPCODE_SUB: sub(d, s0, s1); break;
Alexis Hetu8d78cf72015-08-28 14:24:45 -0400169 case Shader::OPCODE_ISUB: isub(d, s0, s1); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400170 case Shader::OPCODE_MUL: mul(d, s0, s1); break;
Alexis Hetu8d78cf72015-08-28 14:24:45 -0400171 case Shader::OPCODE_IMUL: imul(d, s0, s1); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400172 case Shader::OPCODE_MAD: mad(d, s0, s1, s2); break;
Alexis Hetu8d78cf72015-08-28 14:24:45 -0400173 case Shader::OPCODE_IMAD: imad(d, s0, s1, s2); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400174 case Shader::OPCODE_DP1: dp1(d, s0, s1); break;
175 case Shader::OPCODE_DP2: dp2(d, s0, s1); break;
176 case Shader::OPCODE_DP2ADD: dp2add(d, s0, s1, s2); break;
177 case Shader::OPCODE_DP3: dp3(d, s0, s1); break;
178 case Shader::OPCODE_DP4: dp4(d, s0, s1); break;
Alexis Hetuc3d95f32015-09-23 12:27:32 -0400179 case Shader::OPCODE_DET2: det2(d, s0, s1); break;
180 case Shader::OPCODE_DET3: det3(d, s0, s1, s2); break;
181 case Shader::OPCODE_DET4: det4(d, s0, s1, s2, s3); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400182 case Shader::OPCODE_CMP0: cmp0(d, s0, s1, s2); break;
183 case Shader::OPCODE_ICMP: icmp(d, s0, s1, control); break;
Alexis Hetu8d78cf72015-08-28 14:24:45 -0400184 case Shader::OPCODE_UCMP: ucmp(d, s0, s1, control); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400185 case Shader::OPCODE_SELECT: select(d, s0, s1, s2); break;
186 case Shader::OPCODE_EXTRACT: extract(d.x, s0, s1.x); break;
187 case Shader::OPCODE_INSERT: insert(d, s0, s1.x, s2.x); break;
188 case Shader::OPCODE_FRC: frc(d, s0); break;
189 case Shader::OPCODE_TRUNC: trunc(d, s0); break;
190 case Shader::OPCODE_FLOOR: floor(d, s0); break;
191 case Shader::OPCODE_ROUND: round(d, s0); break;
192 case Shader::OPCODE_ROUNDEVEN: roundEven(d, s0); break;
193 case Shader::OPCODE_CEIL: ceil(d, s0); break;
194 case Shader::OPCODE_EXP2X: exp2x(d, s0, pp); break;
195 case Shader::OPCODE_EXP2: exp2(d, s0, pp); break;
196 case Shader::OPCODE_LOG2X: log2x(d, s0, pp); break;
197 case Shader::OPCODE_LOG2: log2(d, s0, pp); break;
198 case Shader::OPCODE_EXP: exp(d, s0, pp); break;
199 case Shader::OPCODE_LOG: log(d, s0, pp); break;
200 case Shader::OPCODE_RCPX: rcpx(d, s0, pp); break;
201 case Shader::OPCODE_DIV: div(d, s0, s1); break;
Alexis Hetu8d78cf72015-08-28 14:24:45 -0400202 case Shader::OPCODE_IDIV: idiv(d, s0, s1); break;
203 case Shader::OPCODE_UDIV: udiv(d, s0, s1); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400204 case Shader::OPCODE_MOD: mod(d, s0, s1); break;
Alexis Hetu8d78cf72015-08-28 14:24:45 -0400205 case Shader::OPCODE_IMOD: imod(d, s0, s1); break;
206 case Shader::OPCODE_UMOD: umod(d, s0, s1); break;
207 case Shader::OPCODE_SHL: shl(d, s0, s1); break;
208 case Shader::OPCODE_ISHR: ishr(d, s0, s1); break;
209 case Shader::OPCODE_USHR: ushr(d, s0, s1); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400210 case Shader::OPCODE_RSQX: rsqx(d, s0, pp); break;
211 case Shader::OPCODE_SQRT: sqrt(d, s0, pp); break;
212 case Shader::OPCODE_RSQ: rsq(d, s0, pp); break;
213 case Shader::OPCODE_LEN2: len2(d.x, s0, pp); break;
214 case Shader::OPCODE_LEN3: len3(d.x, s0, pp); break;
215 case Shader::OPCODE_LEN4: len4(d.x, s0, pp); break;
216 case Shader::OPCODE_DIST1: dist1(d.x, s0, s1, pp); break;
217 case Shader::OPCODE_DIST2: dist2(d.x, s0, s1, pp); break;
218 case Shader::OPCODE_DIST3: dist3(d.x, s0, s1, pp); break;
219 case Shader::OPCODE_DIST4: dist4(d.x, s0, s1, pp); break;
220 case Shader::OPCODE_MIN: min(d, s0, s1); break;
Alexis Hetu8d78cf72015-08-28 14:24:45 -0400221 case Shader::OPCODE_IMIN: imin(d, s0, s1); break;
222 case Shader::OPCODE_UMIN: umin(d, s0, s1); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400223 case Shader::OPCODE_MAX: max(d, s0, s1); break;
Alexis Hetu8d78cf72015-08-28 14:24:45 -0400224 case Shader::OPCODE_IMAX: imax(d, s0, s1); break;
225 case Shader::OPCODE_UMAX: umax(d, s0, s1); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400226 case Shader::OPCODE_LRP: lrp(d, s0, s1, s2); break;
227 case Shader::OPCODE_STEP: step(d, s0, s1); break;
228 case Shader::OPCODE_SMOOTH: smooth(d, s0, s1, s2); break;
Alexis Hetu8d78cf72015-08-28 14:24:45 -0400229 case Shader::OPCODE_FLOATBITSTOINT:
230 case Shader::OPCODE_FLOATBITSTOUINT:
231 case Shader::OPCODE_INTBITSTOFLOAT:
232 case Shader::OPCODE_UINTBITSTOFLOAT: d = s0; break;
Alexis Hetu9cde9742016-04-06 13:03:38 -0400233 case Shader::OPCODE_PACKSNORM2x16: packSnorm2x16(d, s0); break;
234 case Shader::OPCODE_PACKUNORM2x16: packUnorm2x16(d, s0); break;
Alexis Hetuffb35eb2016-04-06 18:05:00 -0400235 case Shader::OPCODE_PACKHALF2x16: packHalf2x16(d, s0); break;
Alexis Hetu9cde9742016-04-06 13:03:38 -0400236 case Shader::OPCODE_UNPACKSNORM2x16: unpackSnorm2x16(d, s0); break;
237 case Shader::OPCODE_UNPACKUNORM2x16: unpackUnorm2x16(d, s0); break;
Alexis Hetuffb35eb2016-04-06 18:05:00 -0400238 case Shader::OPCODE_UNPACKHALF2x16: unpackHalf2x16(d, s0); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400239 case Shader::OPCODE_POWX: powx(d, s0, s1, pp); break;
240 case Shader::OPCODE_POW: pow(d, s0, s1, pp); break;
241 case Shader::OPCODE_SGN: sgn(d, s0); break;
Alexis Hetu0f448072016-03-18 10:56:08 -0400242 case Shader::OPCODE_ISGN: isgn(d, s0); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400243 case Shader::OPCODE_CRS: crs(d, s0, s1); break;
244 case Shader::OPCODE_FORWARD1: forward1(d, s0, s1, s2); break;
245 case Shader::OPCODE_FORWARD2: forward2(d, s0, s1, s2); break;
246 case Shader::OPCODE_FORWARD3: forward3(d, s0, s1, s2); break;
247 case Shader::OPCODE_FORWARD4: forward4(d, s0, s1, s2); break;
248 case Shader::OPCODE_REFLECT1: reflect1(d, s0, s1); break;
249 case Shader::OPCODE_REFLECT2: reflect2(d, s0, s1); break;
250 case Shader::OPCODE_REFLECT3: reflect3(d, s0, s1); break;
251 case Shader::OPCODE_REFLECT4: reflect4(d, s0, s1); break;
252 case Shader::OPCODE_REFRACT1: refract1(d, s0, s1, s2.x); break;
253 case Shader::OPCODE_REFRACT2: refract2(d, s0, s1, s2.x); break;
254 case Shader::OPCODE_REFRACT3: refract3(d, s0, s1, s2.x); break;
255 case Shader::OPCODE_REFRACT4: refract4(d, s0, s1, s2.x); break;
256 case Shader::OPCODE_NRM2: nrm2(d, s0, pp); break;
257 case Shader::OPCODE_NRM3: nrm3(d, s0, pp); break;
258 case Shader::OPCODE_NRM4: nrm4(d, s0, pp); break;
259 case Shader::OPCODE_ABS: abs(d, s0); break;
Alexis Hetu0f448072016-03-18 10:56:08 -0400260 case Shader::OPCODE_IABS: iabs(d, s0); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400261 case Shader::OPCODE_SINCOS: sincos(d, s0, pp); break;
262 case Shader::OPCODE_COS: cos(d, s0, pp); break;
263 case Shader::OPCODE_SIN: sin(d, s0, pp); break;
264 case Shader::OPCODE_TAN: tan(d, s0, pp); break;
265 case Shader::OPCODE_ACOS: acos(d, s0, pp); break;
266 case Shader::OPCODE_ASIN: asin(d, s0, pp); break;
267 case Shader::OPCODE_ATAN: atan(d, s0, pp); break;
268 case Shader::OPCODE_ATAN2: atan2(d, s0, s1, pp); break;
269 case Shader::OPCODE_COSH: cosh(d, s0, pp); break;
270 case Shader::OPCODE_SINH: sinh(d, s0, pp); break;
271 case Shader::OPCODE_TANH: tanh(d, s0, pp); break;
272 case Shader::OPCODE_ACOSH: acosh(d, s0, pp); break;
273 case Shader::OPCODE_ASINH: asinh(d, s0, pp); break;
274 case Shader::OPCODE_ATANH: atanh(d, s0, pp); break;
Nicolas Capens4f172c72016-01-13 08:34:30 -0500275 case Shader::OPCODE_M4X4: M4X4(d, s0, src1); break;
276 case Shader::OPCODE_M4X3: M4X3(d, s0, src1); break;
277 case Shader::OPCODE_M3X4: M3X4(d, s0, src1); break;
278 case Shader::OPCODE_M3X3: M3X3(d, s0, src1); break;
279 case Shader::OPCODE_M3X2: M3X2(d, s0, src1); break;
280 case Shader::OPCODE_TEX: TEXLD(d, s0, src1, project, bias); break;
Nicolas Capensc2534f42016-04-04 11:13:24 -0400281 case Shader::OPCODE_TEXLDD: TEXLDD(d, s0, src1, s2, s3, project); break;
282 case Shader::OPCODE_TEXLDL: TEXLDL(d, s0, src1, project); break;
Nicolas Capens4f172c72016-01-13 08:34:30 -0500283 case Shader::OPCODE_TEXSIZE: TEXSIZE(d, s0.x, src1); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400284 case Shader::OPCODE_TEXKILL: TEXKILL(cMask, d, dst.mask); break;
Nicolas Capens4f172c72016-01-13 08:34:30 -0500285 case Shader::OPCODE_TEXOFFSET: TEXOFFSET(d, s0, src1, s2, s3, project, bias); break;
286 case Shader::OPCODE_TEXLDLOFFSET: TEXLDL(d, s0, src1, s2, project, bias); break;
287 case Shader::OPCODE_TEXELFETCH: TEXELFETCH(d, s0, src1, s2); break;
288 case Shader::OPCODE_TEXELFETCHOFFSET: TEXELFETCH(d, s0, src1, s2, s3); break;
289 case Shader::OPCODE_TEXGRAD: TEXGRAD(d, s0, src1, s2, s3); break;
290 case Shader::OPCODE_TEXGRADOFFSET: TEXGRAD(d, s0, src1, s2, s3, s4); break;
291 case Shader::OPCODE_DISCARD: DISCARD(cMask, instruction); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400292 case Shader::OPCODE_DFDX: DFDX(d, s0); break;
293 case Shader::OPCODE_DFDY: DFDY(d, s0); break;
294 case Shader::OPCODE_FWIDTH: FWIDTH(d, s0); break;
Nicolas Capens4f172c72016-01-13 08:34:30 -0500295 case Shader::OPCODE_BREAK: BREAK(); break;
296 case Shader::OPCODE_BREAKC: BREAKC(s0, s1, control); break;
297 case Shader::OPCODE_BREAKP: BREAKP(src0); break;
298 case Shader::OPCODE_CONTINUE: CONTINUE(); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400299 case Shader::OPCODE_TEST: TEST(); break;
Nicolas Capens4f172c72016-01-13 08:34:30 -0500300 case Shader::OPCODE_CALL: CALL(dst.label, dst.callSite); break;
301 case Shader::OPCODE_CALLNZ: CALLNZ(dst.label, dst.callSite, src0); break;
302 case Shader::OPCODE_ELSE: ELSE(); break;
303 case Shader::OPCODE_ENDIF: ENDIF(); break;
304 case Shader::OPCODE_ENDLOOP: ENDLOOP(); break;
305 case Shader::OPCODE_ENDREP: ENDREP(); break;
306 case Shader::OPCODE_ENDWHILE: ENDWHILE(); break;
307 case Shader::OPCODE_IF: IF(src0); break;
308 case Shader::OPCODE_IFC: IFC(s0, s1, control); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400309 case Shader::OPCODE_LABEL: LABEL(dst.index); break;
Nicolas Capens4f172c72016-01-13 08:34:30 -0500310 case Shader::OPCODE_LOOP: LOOP(src1); break;
311 case Shader::OPCODE_REP: REP(src0); break;
312 case Shader::OPCODE_WHILE: WHILE(src0); break;
313 case Shader::OPCODE_RET: RET(); break;
314 case Shader::OPCODE_LEAVE: LEAVE(); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400315 case Shader::OPCODE_CMP: cmp(d, s0, s1, control); break;
316 case Shader::OPCODE_ALL: all(d.x, s0); break;
317 case Shader::OPCODE_ANY: any(d.x, s0); break;
318 case Shader::OPCODE_NOT: not(d, s0); break;
Alexis Hetuc4f2c292015-08-18 15:43:09 -0400319 case Shader::OPCODE_OR: or(d, s0, s1); break;
320 case Shader::OPCODE_XOR: xor(d, s0, s1); break;
321 case Shader::OPCODE_AND: and(d, s0, s1); break;
Alexis Hetu8d78cf72015-08-28 14:24:45 -0400322 case Shader::OPCODE_EQ: equal(d, s0, s1); break;
323 case Shader::OPCODE_NE: notEqual(d, s0, s1); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400324 case Shader::OPCODE_END: break;
325 default:
326 ASSERT(false);
327 }
328
329 if(dst.type != Shader::PARAMETER_VOID && dst.type != Shader::PARAMETER_LABEL && opcode != Shader::OPCODE_TEXKILL && opcode != Shader::OPCODE_NOP)
330 {
331 if(dst.integer)
332 {
333 switch(opcode)
334 {
335 case Shader::OPCODE_DIV:
336 if(dst.x) d.x = Trunc(d.x);
337 if(dst.y) d.y = Trunc(d.y);
338 if(dst.z) d.z = Trunc(d.z);
339 if(dst.w) d.w = Trunc(d.w);
340 break;
341 default:
342 break; // No truncation to integer required when arguments are integer
343 }
344 }
345
346 if(dst.saturate)
347 {
348 if(dst.x) d.x = Max(d.x, Float4(0.0f));
349 if(dst.y) d.y = Max(d.y, Float4(0.0f));
350 if(dst.z) d.z = Max(d.z, Float4(0.0f));
351 if(dst.w) d.w = Max(d.w, Float4(0.0f));
352
353 if(dst.x) d.x = Min(d.x, Float4(1.0f));
354 if(dst.y) d.y = Min(d.y, Float4(1.0f));
355 if(dst.z) d.z = Min(d.z, Float4(1.0f));
356 if(dst.w) d.w = Min(d.w, Float4(1.0f));
357 }
358
359 if(instruction->isPredicated())
360 {
361 Vector4f pDst; // FIXME: Rename
362
363 switch(dst.type)
364 {
365 case Shader::PARAMETER_TEMP:
366 if(dst.rel.type == Shader::PARAMETER_VOID)
367 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500368 if(dst.x) pDst.x = r[dst.index].x;
369 if(dst.y) pDst.y = r[dst.index].y;
370 if(dst.z) pDst.z = r[dst.index].z;
371 if(dst.w) pDst.w = r[dst.index].w;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400372 }
373 else
374 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500375 Int a = relativeAddress(dst);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400376
Nicolas Capens4f172c72016-01-13 08:34:30 -0500377 if(dst.x) pDst.x = r[dst.index + a].x;
378 if(dst.y) pDst.y = r[dst.index + a].y;
379 if(dst.z) pDst.z = r[dst.index + a].z;
380 if(dst.w) pDst.w = r[dst.index + a].w;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400381 }
382 break;
383 case Shader::PARAMETER_COLOROUT:
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500384 if(dst.rel.type == Shader::PARAMETER_VOID)
385 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500386 if(dst.x) pDst.x = oC[dst.index].x;
387 if(dst.y) pDst.y = oC[dst.index].y;
388 if(dst.z) pDst.z = oC[dst.index].z;
389 if(dst.w) pDst.w = oC[dst.index].w;
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500390 }
391 else
392 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500393 Int a = relativeAddress(dst) + dst.index;
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500394
Nicolas Capens4f172c72016-01-13 08:34:30 -0500395 if(dst.x) pDst.x = oC[a].x;
396 if(dst.y) pDst.y = oC[a].y;
397 if(dst.z) pDst.z = oC[a].z;
398 if(dst.w) pDst.w = oC[a].w;
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500399 }
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400400 break;
401 case Shader::PARAMETER_PREDICATE:
Nicolas Capens4f172c72016-01-13 08:34:30 -0500402 if(dst.x) pDst.x = p0.x;
403 if(dst.y) pDst.y = p0.y;
404 if(dst.z) pDst.z = p0.z;
405 if(dst.w) pDst.w = p0.w;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400406 break;
407 case Shader::PARAMETER_DEPTHOUT:
Nicolas Capens4f172c72016-01-13 08:34:30 -0500408 pDst.x = oDepth;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400409 break;
410 default:
411 ASSERT(false);
412 }
413
Nicolas Capens4f172c72016-01-13 08:34:30 -0500414 Int4 enable = enableMask(instruction);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400415
416 Int4 xEnable = enable;
417 Int4 yEnable = enable;
418 Int4 zEnable = enable;
419 Int4 wEnable = enable;
420
421 if(predicate)
422 {
423 unsigned char pSwizzle = instruction->predicateSwizzle;
424
Nicolas Capens4f172c72016-01-13 08:34:30 -0500425 Float4 xPredicate = p0[(pSwizzle >> 0) & 0x03];
426 Float4 yPredicate = p0[(pSwizzle >> 2) & 0x03];
427 Float4 zPredicate = p0[(pSwizzle >> 4) & 0x03];
428 Float4 wPredicate = p0[(pSwizzle >> 6) & 0x03];
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400429
430 if(!instruction->predicateNot)
431 {
432 if(dst.x) xEnable = xEnable & As<Int4>(xPredicate);
433 if(dst.y) yEnable = yEnable & As<Int4>(yPredicate);
434 if(dst.z) zEnable = zEnable & As<Int4>(zPredicate);
435 if(dst.w) wEnable = wEnable & As<Int4>(wPredicate);
436 }
437 else
438 {
439 if(dst.x) xEnable = xEnable & ~As<Int4>(xPredicate);
440 if(dst.y) yEnable = yEnable & ~As<Int4>(yPredicate);
441 if(dst.z) zEnable = zEnable & ~As<Int4>(zPredicate);
442 if(dst.w) wEnable = wEnable & ~As<Int4>(wPredicate);
443 }
444 }
445
446 if(dst.x) d.x = As<Float4>(As<Int4>(d.x) & xEnable);
447 if(dst.y) d.y = As<Float4>(As<Int4>(d.y) & yEnable);
448 if(dst.z) d.z = As<Float4>(As<Int4>(d.z) & zEnable);
449 if(dst.w) d.w = As<Float4>(As<Int4>(d.w) & wEnable);
450
451 if(dst.x) d.x = As<Float4>(As<Int4>(d.x) | (As<Int4>(pDst.x) & ~xEnable));
452 if(dst.y) d.y = As<Float4>(As<Int4>(d.y) | (As<Int4>(pDst.y) & ~yEnable));
453 if(dst.z) d.z = As<Float4>(As<Int4>(d.z) | (As<Int4>(pDst.z) & ~zEnable));
454 if(dst.w) d.w = As<Float4>(As<Int4>(d.w) | (As<Int4>(pDst.w) & ~wEnable));
455 }
456
457 switch(dst.type)
458 {
459 case Shader::PARAMETER_TEMP:
460 if(dst.rel.type == Shader::PARAMETER_VOID)
461 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500462 if(dst.x) r[dst.index].x = d.x;
463 if(dst.y) r[dst.index].y = d.y;
464 if(dst.z) r[dst.index].z = d.z;
465 if(dst.w) r[dst.index].w = d.w;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400466 }
467 else
468 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500469 Int a = relativeAddress(dst);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400470
Nicolas Capens4f172c72016-01-13 08:34:30 -0500471 if(dst.x) r[dst.index + a].x = d.x;
472 if(dst.y) r[dst.index + a].y = d.y;
473 if(dst.z) r[dst.index + a].z = d.z;
474 if(dst.w) r[dst.index + a].w = d.w;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400475 }
476 break;
477 case Shader::PARAMETER_COLOROUT:
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500478 if(dst.rel.type == Shader::PARAMETER_VOID)
479 {
Nicolas Capens8b124c12016-04-18 14:09:37 -0400480 broadcastColor0 = (dst.index == 0) && broadcastColor0;
481
Nicolas Capens4f172c72016-01-13 08:34:30 -0500482 if(dst.x) { oC[dst.index].x = d.x; }
483 if(dst.y) { oC[dst.index].y = d.y; }
484 if(dst.z) { oC[dst.index].z = d.z; }
485 if(dst.w) { oC[dst.index].w = d.w; }
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500486 }
487 else
488 {
Nicolas Capens8b124c12016-04-18 14:09:37 -0400489 broadcastColor0 = false;
Nicolas Capens4f172c72016-01-13 08:34:30 -0500490 Int a = relativeAddress(dst) + dst.index;
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500491
Nicolas Capens4f172c72016-01-13 08:34:30 -0500492 if(dst.x) { oC[a].x = d.x; }
493 if(dst.y) { oC[a].y = d.y; }
494 if(dst.z) { oC[a].z = d.z; }
495 if(dst.w) { oC[a].w = d.w; }
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500496 }
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400497 break;
498 case Shader::PARAMETER_PREDICATE:
Nicolas Capens4f172c72016-01-13 08:34:30 -0500499 if(dst.x) p0.x = d.x;
500 if(dst.y) p0.y = d.y;
501 if(dst.z) p0.z = d.z;
502 if(dst.w) p0.w = d.w;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400503 break;
504 case Shader::PARAMETER_DEPTHOUT:
Nicolas Capens4f172c72016-01-13 08:34:30 -0500505 oDepth = d.x;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400506 break;
507 default:
508 ASSERT(false);
509 }
510 }
511 }
512
513 if(currentLabel != -1)
514 {
515 Nucleus::setInsertBlock(returnBlock);
516 }
517
Nicolas Capens8b124c12016-04-18 14:09:37 -0400518 if(broadcastColor0)
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400519 {
Nicolas Capens8b124c12016-04-18 14:09:37 -0400520 for(int i = 0; i < RENDERTARGETS; i++)
521 {
522 c[i] = oC[0];
523 }
524 }
525 else
526 {
527 for(int i = 0; i < RENDERTARGETS; i++)
528 {
529 c[i] = oC[i];
530 }
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400531 }
532 }
Nicolas Capens5d961882016-01-01 23:18:14 -0500533
Nicolas Capens4f172c72016-01-13 08:34:30 -0500534 Bool PixelProgram::alphaTest(Int cMask[4])
Nicolas Capens5d961882016-01-01 23:18:14 -0500535 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500536 clampColor(c);
Nicolas Capens5d961882016-01-01 23:18:14 -0500537
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400538 if(!state.alphaTestActive())
539 {
540 return true;
541 }
542
543 Int aMask;
544
545 if(state.transparencyAntialiasing == TRANSPARENCY_NONE)
546 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500547 Short4 alpha = RoundShort4(c[0].w * Float4(0x1000));
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400548
Nicolas Capens4f172c72016-01-13 08:34:30 -0500549 PixelRoutine::alphaTest(aMask, alpha);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400550
551 for(unsigned int q = 0; q < state.multiSample; q++)
552 {
553 cMask[q] &= aMask;
554 }
555 }
556 else if(state.transparencyAntialiasing == TRANSPARENCY_ALPHA_TO_COVERAGE)
557 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500558 alphaToCoverage(cMask, c[0].w);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400559 }
560 else ASSERT(false);
561
562 Int pass = cMask[0];
563
564 for(unsigned int q = 1; q < state.multiSample; q++)
565 {
566 pass = pass | cMask[q];
567 }
568
Nicolas Capens5d961882016-01-01 23:18:14 -0500569 return pass != 0x0;
570 }
571
Nicolas Capens4f172c72016-01-13 08:34:30 -0500572 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 -0500573 {
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500574 for(int index = 0; index < RENDERTARGETS; index++)
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400575 {
576 if(!state.colorWriteActive(index))
577 {
578 continue;
Nicolas Capens5d961882016-01-01 23:18:14 -0500579 }
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400580
Alexis Hetu049a1872016-04-25 16:59:58 -0400581 if(!postBlendSRGB && state.writeSRGB && !isSRGB(index))
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400582 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500583 c[index].x = linearToSRGB(c[index].x);
584 c[index].y = linearToSRGB(c[index].y);
585 c[index].z = linearToSRGB(c[index].z);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400586 }
587
588 if(index == 0)
589 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500590 fogBlend(c[index], fog);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400591 }
592
593 switch(state.targetFormat[index])
594 {
595 case FORMAT_R5G6B5:
596 case FORMAT_X8R8G8B8:
597 case FORMAT_X8B8G8R8:
598 case FORMAT_A8R8G8B8:
599 case FORMAT_A8B8G8R8:
Alexis Hetu049a1872016-04-25 16:59:58 -0400600 case FORMAT_SRGB8_X8:
601 case FORMAT_SRGB8_A8:
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400602 case FORMAT_A8:
603 case FORMAT_G16R16:
604 case FORMAT_A16B16G16R16:
605 for(unsigned int q = 0; q < state.multiSample; q++)
606 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500607 Pointer<Byte> buffer = cBuffer[index] + q * *Pointer<Int>(data + OFFSET(DrawData, colorSliceB[index]));
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400608 Vector4s color;
609
Nicolas Capens13c70cd2016-04-05 10:51:45 -0400610 if(state.targetFormat[index] == FORMAT_R5G6B5)
611 {
612 color.x = UShort4(c[index].x * Float4(0xFBFF), false);
613 color.y = UShort4(c[index].y * Float4(0xFDFF), false);
614 color.z = UShort4(c[index].z * Float4(0xFBFF), false);
615 color.w = UShort4(c[index].w * Float4(0xFFFF), false);
616 }
617 else
618 {
619 color.x = convertFixed16(c[index].x, false);
620 color.y = convertFixed16(c[index].y, false);
621 color.z = convertFixed16(c[index].z, false);
622 color.w = convertFixed16(c[index].w, false);
623 }
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400624
625 if(state.multiSampleMask & (1 << q))
626 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500627 alphaBlend(index, buffer, color, x);
628 logicOperation(index, buffer, color, x);
629 writeColor(index, buffer, x, color, sMask[q], zMask[q], cMask[q]);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400630 }
631 }
632 break;
633 case FORMAT_R32F:
634 case FORMAT_G32R32F:
Alexis Hetudbd1a8e2016-04-13 11:40:30 -0400635 case FORMAT_X32B32G32R32F:
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400636 case FORMAT_A32B32G32R32F:
Alexis Hetu1abb6382016-02-08 11:21:16 -0500637 case FORMAT_R32I:
638 case FORMAT_G32R32I:
639 case FORMAT_A32B32G32R32I:
640 case FORMAT_R32UI:
641 case FORMAT_G32R32UI:
642 case FORMAT_A32B32G32R32UI:
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400643 for(unsigned int q = 0; q < state.multiSample; q++)
644 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500645 Pointer<Byte> buffer = cBuffer[index] + q * *Pointer<Int>(data + OFFSET(DrawData, colorSliceB[index]));
646 Vector4f color = c[index];
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400647
648 if(state.multiSampleMask & (1 << q))
649 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500650 alphaBlend(index, buffer, color, x);
651 writeColor(index, buffer, x, color, sMask[q], zMask[q], cMask[q]);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400652 }
653 }
654 break;
655 default:
656 ASSERT(false);
657 }
658 }
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500659 }
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400660
Nicolas Capensc2534f42016-04-04 11:13:24 -0400661 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 -0400662 {
Nicolas Capensc2534f42016-04-04 11:13:24 -0400663 Vector4f tmp;
664
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400665 if(sampler.type == Shader::PARAMETER_SAMPLER && sampler.rel.type == Shader::PARAMETER_VOID)
666 {
Nicolas Capensc2534f42016-04-04 11:13:24 -0400667 sampleTexture(tmp, sampler.index, u, v, w, q, dsx, dsy, project, method);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400668 }
669 else
670 {
Nicolas Capensc2534f42016-04-04 11:13:24 -0400671 Int index = As<Int>(Float(fetchRegister(sampler).x.x));
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400672
673 for(int i = 0; i < TEXTURE_IMAGE_UNITS; i++)
674 {
675 if(shader->usesSampler(i))
676 {
677 If(index == i)
678 {
Nicolas Capensc2534f42016-04-04 11:13:24 -0400679 sampleTexture(tmp, i, u, v, w, q, dsx, dsy, project, method);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400680 // FIXME: When the sampler states are the same, we could use one sampler and just index the texture
681 }
682 }
683 }
684 }
Nicolas Capensc2534f42016-04-04 11:13:24 -0400685
686 c.x = tmp[(sampler.swizzle >> 0) & 0x3];
687 c.y = tmp[(sampler.swizzle >> 2) & 0x3];
688 c.z = tmp[(sampler.swizzle >> 4) & 0x3];
689 c.w = tmp[(sampler.swizzle >> 6) & 0x3];
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400690 }
691
Nicolas Capensc2534f42016-04-04 11:13:24 -0400692 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 -0400693 {
Nicolas Capensc2534f42016-04-04 11:13:24 -0400694 #if PERF_PROFILE
695 Long texTime = Ticks();
696 #endif
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400697
Nicolas Capens4f172c72016-01-13 08:34:30 -0500698 Pointer<Byte> texture = data + OFFSET(DrawData, mipmap) + stage * sizeof(Texture);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400699
700 if(!project)
701 {
Nicolas Capensc2534f42016-04-04 11:13:24 -0400702 sampler[stage]->sampleTexture(texture, c, u, v, w, q, dsx, dsy, method);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400703 }
704 else
705 {
706 Float4 rq = reciprocal(q);
707
708 Float4 u_q = u * rq;
709 Float4 v_q = v * rq;
710 Float4 w_q = w * rq;
711
Nicolas Capensc2534f42016-04-04 11:13:24 -0400712 sampler[stage]->sampleTexture(texture, c, u_q, v_q, w_q, q, dsx, dsy, method);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400713 }
714
Nicolas Capensc2534f42016-04-04 11:13:24 -0400715 #if PERF_PROFILE
716 cycles[PERF_TEX] += Ticks() - texTime;
717 #endif
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400718 }
719
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500720 void PixelProgram::clampColor(Vector4f oC[RENDERTARGETS])
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400721 {
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500722 for(int index = 0; index < RENDERTARGETS; index++)
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400723 {
724 if(!state.colorWriteActive(index) && !(index == 0 && state.alphaTestActive()))
725 {
726 continue;
727 }
728
729 switch(state.targetFormat[index])
730 {
731 case FORMAT_NULL:
732 break;
733 case FORMAT_R5G6B5:
734 case FORMAT_A8R8G8B8:
735 case FORMAT_A8B8G8R8:
736 case FORMAT_X8R8G8B8:
737 case FORMAT_X8B8G8R8:
Alexis Hetu049a1872016-04-25 16:59:58 -0400738 case FORMAT_SRGB8_X8:
739 case FORMAT_SRGB8_A8:
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400740 case FORMAT_A8:
741 case FORMAT_G16R16:
742 case FORMAT_A16B16G16R16:
743 oC[index].x = Max(oC[index].x, Float4(0.0f)); oC[index].x = Min(oC[index].x, Float4(1.0f));
744 oC[index].y = Max(oC[index].y, Float4(0.0f)); oC[index].y = Min(oC[index].y, Float4(1.0f));
745 oC[index].z = Max(oC[index].z, Float4(0.0f)); oC[index].z = Min(oC[index].z, Float4(1.0f));
746 oC[index].w = Max(oC[index].w, Float4(0.0f)); oC[index].w = Min(oC[index].w, Float4(1.0f));
747 break;
748 case FORMAT_R32F:
749 case FORMAT_G32R32F:
Alexis Hetudbd1a8e2016-04-13 11:40:30 -0400750 case FORMAT_X32B32G32R32F:
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400751 case FORMAT_A32B32G32R32F:
Alexis Hetu1abb6382016-02-08 11:21:16 -0500752 case FORMAT_R32I:
753 case FORMAT_G32R32I:
754 case FORMAT_A32B32G32R32I:
755 case FORMAT_R32UI:
756 case FORMAT_G32R32UI:
757 case FORMAT_A32B32G32R32UI:
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400758 break;
759 default:
760 ASSERT(false);
761 }
762 }
763 }
764
Nicolas Capens4f172c72016-01-13 08:34:30 -0500765 Int4 PixelProgram::enableMask(const Shader::Instruction *instruction)
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400766 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500767 Int4 enable = instruction->analysisBranch ? Int4(enableStack[enableIndex]) : Int4(0xFFFFFFFF);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400768
769 if(!whileTest)
770 {
771 if(shader->containsBreakInstruction() && instruction->analysisBreak)
772 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500773 enable &= enableBreak;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400774 }
775
776 if(shader->containsContinueInstruction() && instruction->analysisContinue)
777 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500778 enable &= enableContinue;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400779 }
780
781 if(shader->containsLeaveInstruction() && instruction->analysisLeave)
782 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500783 enable &= enableLeave;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400784 }
785 }
786
787 return enable;
788 }
789
Nicolas Capensc2534f42016-04-04 11:13:24 -0400790 Vector4f PixelProgram::fetchRegister(const Src &src, unsigned int offset)
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400791 {
792 Vector4f reg;
Nicolas Capens5d961882016-01-01 23:18:14 -0500793 unsigned int i = src.index + offset;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400794
795 switch(src.type)
796 {
797 case Shader::PARAMETER_TEMP:
798 if(src.rel.type == Shader::PARAMETER_VOID)
799 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500800 reg = r[i];
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400801 }
802 else
803 {
Alexis Hetu2c2a7b22015-10-27 16:12:11 -0400804 Int a = relativeAddress(src, src.bufferIndex);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400805
Nicolas Capens4f172c72016-01-13 08:34:30 -0500806 reg = r[i + a];
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400807 }
808 break;
809 case Shader::PARAMETER_INPUT:
810 {
811 if(src.rel.type == Shader::PARAMETER_VOID) // Not relative
812 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500813 reg = v[i];
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400814 }
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400815 else
816 {
Alexis Hetu2c2a7b22015-10-27 16:12:11 -0400817 Int a = relativeAddress(src, src.bufferIndex);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400818
Nicolas Capens4f172c72016-01-13 08:34:30 -0500819 reg = v[i + a];
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400820 }
821 }
822 break;
823 case Shader::PARAMETER_CONST:
Nicolas Capens4f172c72016-01-13 08:34:30 -0500824 reg = readConstant(src, offset);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400825 break;
826 case Shader::PARAMETER_TEXTURE:
Nicolas Capens4f172c72016-01-13 08:34:30 -0500827 reg = v[2 + i];
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400828 break;
829 case Shader::PARAMETER_MISCTYPE:
Nicolas Capens4f172c72016-01-13 08:34:30 -0500830 if(src.index == 0) reg = vPos;
831 if(src.index == 1) reg = vFace;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400832 break;
833 case Shader::PARAMETER_SAMPLER:
834 if(src.rel.type == Shader::PARAMETER_VOID)
835 {
836 reg.x = As<Float4>(Int4(i));
837 }
838 else if(src.rel.type == Shader::PARAMETER_TEMP)
839 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500840 reg.x = As<Float4>(Int4(i) + As<Int4>(r[src.rel.index].x));
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400841 }
842 return reg;
843 case Shader::PARAMETER_PREDICATE: return reg; // Dummy
844 case Shader::PARAMETER_VOID: return reg; // Dummy
845 case Shader::PARAMETER_FLOAT4LITERAL:
846 reg.x = Float4(src.value[0]);
847 reg.y = Float4(src.value[1]);
848 reg.z = Float4(src.value[2]);
849 reg.w = Float4(src.value[3]);
850 break;
851 case Shader::PARAMETER_CONSTINT: return reg; // Dummy
852 case Shader::PARAMETER_CONSTBOOL: return reg; // Dummy
853 case Shader::PARAMETER_LOOP: return reg; // Dummy
854 case Shader::PARAMETER_COLOROUT:
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500855 if(src.rel.type == Shader::PARAMETER_VOID) // Not relative
856 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500857 reg = oC[i];
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500858 }
859 else
860 {
Alexis Hetu2c2a7b22015-10-27 16:12:11 -0400861 Int a = relativeAddress(src, src.bufferIndex);
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500862
Nicolas Capens4f172c72016-01-13 08:34:30 -0500863 reg = oC[i + a];
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500864 }
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400865 break;
866 case Shader::PARAMETER_DEPTHOUT:
Nicolas Capens4f172c72016-01-13 08:34:30 -0500867 reg.x = oDepth;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400868 break;
869 default:
870 ASSERT(false);
871 }
872
873 const Float4 &x = reg[(src.swizzle >> 0) & 0x3];
874 const Float4 &y = reg[(src.swizzle >> 2) & 0x3];
875 const Float4 &z = reg[(src.swizzle >> 4) & 0x3];
876 const Float4 &w = reg[(src.swizzle >> 6) & 0x3];
877
878 Vector4f mod;
879
880 switch(src.modifier)
881 {
882 case Shader::MODIFIER_NONE:
883 mod.x = x;
884 mod.y = y;
885 mod.z = z;
886 mod.w = w;
887 break;
888 case Shader::MODIFIER_NEGATE:
889 mod.x = -x;
890 mod.y = -y;
891 mod.z = -z;
892 mod.w = -w;
893 break;
894 case Shader::MODIFIER_ABS:
895 mod.x = Abs(x);
896 mod.y = Abs(y);
897 mod.z = Abs(z);
898 mod.w = Abs(w);
899 break;
900 case Shader::MODIFIER_ABS_NEGATE:
901 mod.x = -Abs(x);
902 mod.y = -Abs(y);
903 mod.z = -Abs(z);
904 mod.w = -Abs(w);
905 break;
906 case Shader::MODIFIER_NOT:
907 mod.x = As<Float4>(As<Int4>(x) ^ Int4(0xFFFFFFFF));
908 mod.y = As<Float4>(As<Int4>(y) ^ Int4(0xFFFFFFFF));
909 mod.z = As<Float4>(As<Int4>(z) ^ Int4(0xFFFFFFFF));
910 mod.w = As<Float4>(As<Int4>(w) ^ Int4(0xFFFFFFFF));
911 break;
912 default:
913 ASSERT(false);
914 }
915
916 return mod;
917 }
918
Alexis Hetu2c2a7b22015-10-27 16:12:11 -0400919 RValue<Pointer<Byte>> PixelProgram::uniformAddress(int bufferIndex, unsigned int index)
920 {
921 if(bufferIndex == -1)
922 {
923 return data + OFFSET(DrawData, ps.c[index]);
924 }
925 else
926 {
927 return *Pointer<Pointer<Byte>>(data + OFFSET(DrawData, ps.u[bufferIndex])) + index;
928 }
929 }
930
931 RValue<Pointer<Byte>> PixelProgram::uniformAddress(int bufferIndex, unsigned int index, Int& offset)
932 {
933 return uniformAddress(bufferIndex, index) + offset * sizeof(float4);
934 }
935
Nicolas Capens4f172c72016-01-13 08:34:30 -0500936 Vector4f PixelProgram::readConstant(const Src &src, unsigned int offset)
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400937 {
938 Vector4f c;
Nicolas Capens5d961882016-01-01 23:18:14 -0500939 unsigned int i = src.index + offset;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400940
941 if(src.rel.type == Shader::PARAMETER_VOID) // Not relative
942 {
Alexis Hetu2c2a7b22015-10-27 16:12:11 -0400943 c.x = c.y = c.z = c.w = *Pointer<Float4>(uniformAddress(src.bufferIndex, i));
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400944
945 c.x = c.x.xxxx;
946 c.y = c.y.yyyy;
947 c.z = c.z.zzzz;
948 c.w = c.w.wwww;
949
950 if(shader->containsDefineInstruction()) // Constant may be known at compile time
951 {
952 for(size_t j = 0; j < shader->getLength(); j++)
953 {
954 const Shader::Instruction &instruction = *shader->getInstruction(j);
955
956 if(instruction.opcode == Shader::OPCODE_DEF)
957 {
958 if(instruction.dst.index == i)
959 {
960 c.x = Float4(instruction.src[0].value[0]);
961 c.y = Float4(instruction.src[0].value[1]);
962 c.z = Float4(instruction.src[0].value[2]);
963 c.w = Float4(instruction.src[0].value[3]);
964
965 break;
966 }
967 }
968 }
969 }
970 }
971 else if(src.rel.type == Shader::PARAMETER_LOOP)
972 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500973 Int loopCounter = aL[loopDepth];
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400974
Alexis Hetu2c2a7b22015-10-27 16:12:11 -0400975 c.x = c.y = c.z = c.w = *Pointer<Float4>(uniformAddress(src.bufferIndex, i, loopCounter));
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400976
977 c.x = c.x.xxxx;
978 c.y = c.y.yyyy;
979 c.z = c.z.zzzz;
980 c.w = c.w.wwww;
981 }
982 else
983 {
Alexis Hetu2c2a7b22015-10-27 16:12:11 -0400984 Int a = relativeAddress(src, src.bufferIndex);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400985
Alexis Hetu2c2a7b22015-10-27 16:12:11 -0400986 c.x = c.y = c.z = c.w = *Pointer<Float4>(uniformAddress(src.bufferIndex, i, a));
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400987
988 c.x = c.x.xxxx;
989 c.y = c.y.yyyy;
990 c.z = c.z.zzzz;
991 c.w = c.w.wwww;
992 }
993
994 return c;
995 }
996
Alexis Hetu2c2a7b22015-10-27 16:12:11 -0400997 Int PixelProgram::relativeAddress(const Shader::Parameter &var, int bufferIndex)
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400998 {
999 ASSERT(var.rel.deterministic);
1000
1001 if(var.rel.type == Shader::PARAMETER_TEMP)
1002 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001003 return As<Int>(Extract(r[var.rel.index].x, 0)) * var.rel.scale;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001004 }
1005 else if(var.rel.type == Shader::PARAMETER_INPUT)
1006 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001007 return As<Int>(Extract(v[var.rel.index].x, 0)) * var.rel.scale;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001008 }
1009 else if(var.rel.type == Shader::PARAMETER_OUTPUT)
1010 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001011 return As<Int>(Extract(oC[var.rel.index].x, 0)) * var.rel.scale;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001012 }
1013 else if(var.rel.type == Shader::PARAMETER_CONST)
1014 {
Alexis Hetu48be7352016-02-10 14:32:34 -05001015 return *Pointer<Int>(uniformAddress(bufferIndex, var.rel.index)) * var.rel.scale;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001016 }
Nicolas Capens907700d2016-01-20 17:09:28 -05001017 else if(var.rel.type == Shader::PARAMETER_LOOP)
1018 {
1019 return aL[loopDepth];
1020 }
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001021 else ASSERT(false);
1022
1023 return 0;
1024 }
1025
1026 Float4 PixelProgram::linearToSRGB(const Float4 &x) // Approximates x^(1.0/2.2)
1027 {
1028 Float4 sqrtx = Rcp_pp(RcpSqrt_pp(x));
1029 Float4 sRGB = sqrtx * Float4(1.14f) - x * Float4(0.14f);
1030
1031 return Min(Max(sRGB, Float4(0.0f)), Float4(1.0f));
1032 }
1033
Nicolas Capens4f172c72016-01-13 08:34:30 -05001034 void PixelProgram::M3X2(Vector4f &dst, Vector4f &src0, const Src &src1)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001035 {
Nicolas Capensc2534f42016-04-04 11:13:24 -04001036 Vector4f row0 = fetchRegister(src1, 0);
1037 Vector4f row1 = fetchRegister(src1, 1);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001038
1039 dst.x = dot3(src0, row0);
1040 dst.y = dot3(src0, row1);
1041 }
1042
Nicolas Capens4f172c72016-01-13 08:34:30 -05001043 void PixelProgram::M3X3(Vector4f &dst, Vector4f &src0, const Src &src1)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001044 {
Nicolas Capensc2534f42016-04-04 11:13:24 -04001045 Vector4f row0 = fetchRegister(src1, 0);
1046 Vector4f row1 = fetchRegister(src1, 1);
1047 Vector4f row2 = fetchRegister(src1, 2);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001048
1049 dst.x = dot3(src0, row0);
1050 dst.y = dot3(src0, row1);
1051 dst.z = dot3(src0, row2);
1052 }
1053
Nicolas Capens4f172c72016-01-13 08:34:30 -05001054 void PixelProgram::M3X4(Vector4f &dst, Vector4f &src0, const Src &src1)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001055 {
Nicolas Capensc2534f42016-04-04 11:13:24 -04001056 Vector4f row0 = fetchRegister(src1, 0);
1057 Vector4f row1 = fetchRegister(src1, 1);
1058 Vector4f row2 = fetchRegister(src1, 2);
1059 Vector4f row3 = fetchRegister(src1, 3);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001060
1061 dst.x = dot3(src0, row0);
1062 dst.y = dot3(src0, row1);
1063 dst.z = dot3(src0, row2);
1064 dst.w = dot3(src0, row3);
1065 }
1066
Nicolas Capens4f172c72016-01-13 08:34:30 -05001067 void PixelProgram::M4X3(Vector4f &dst, Vector4f &src0, const Src &src1)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001068 {
Nicolas Capensc2534f42016-04-04 11:13:24 -04001069 Vector4f row0 = fetchRegister(src1, 0);
1070 Vector4f row1 = fetchRegister(src1, 1);
1071 Vector4f row2 = fetchRegister(src1, 2);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001072
1073 dst.x = dot4(src0, row0);
1074 dst.y = dot4(src0, row1);
1075 dst.z = dot4(src0, row2);
1076 }
1077
Nicolas Capens4f172c72016-01-13 08:34:30 -05001078 void PixelProgram::M4X4(Vector4f &dst, Vector4f &src0, const Src &src1)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001079 {
Nicolas Capensc2534f42016-04-04 11:13:24 -04001080 Vector4f row0 = fetchRegister(src1, 0);
1081 Vector4f row1 = fetchRegister(src1, 1);
1082 Vector4f row2 = fetchRegister(src1, 2);
1083 Vector4f row3 = fetchRegister(src1, 3);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001084
1085 dst.x = dot4(src0, row0);
1086 dst.y = dot4(src0, row1);
1087 dst.z = dot4(src0, row2);
1088 dst.w = dot4(src0, row3);
1089 }
1090
Nicolas Capens4f172c72016-01-13 08:34:30 -05001091 void PixelProgram::TEXLD(Vector4f &dst, Vector4f &src0, const Src &src1, bool project, bool bias)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001092 {
Nicolas Capensc2534f42016-04-04 11:13:24 -04001093 sampleTexture(dst, src1, src0.x, src0.y, src0.z, src0.w, src0, src0, project, bias ? Bias : Implicit);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001094 }
1095
Nicolas Capens4f172c72016-01-13 08:34:30 -05001096 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 -04001097 {
1098 UNIMPLEMENTED();
1099 }
1100
Nicolas Capens4f172c72016-01-13 08:34:30 -05001101 void PixelProgram::TEXLDL(Vector4f &dst, Vector4f &src0, const Src &src1, Vector4f &offset, bool project, bool bias)
Alexis Hetu25d47fc2015-10-22 13:58:32 -04001102 {
1103 UNIMPLEMENTED();
1104 }
1105
Nicolas Capens4f172c72016-01-13 08:34:30 -05001106 void PixelProgram::TEXELFETCH(Vector4f &dst, Vector4f &src0, const Src& src1, Vector4f &src2)
Alexis Hetu25d47fc2015-10-22 13:58:32 -04001107 {
1108 UNIMPLEMENTED();
1109 }
1110
Nicolas Capens4f172c72016-01-13 08:34:30 -05001111 void PixelProgram::TEXELFETCH(Vector4f &dst, Vector4f &src0, const Src& src1, Vector4f &src2, Vector4f &offset)
Alexis Hetu25d47fc2015-10-22 13:58:32 -04001112 {
1113 UNIMPLEMENTED();
1114 }
1115
Nicolas Capens4f172c72016-01-13 08:34:30 -05001116 void PixelProgram::TEXGRAD(Vector4f &dst, Vector4f &src0, const Src& src1, Vector4f &src2, Vector4f &src3)
Alexis Hetu25d47fc2015-10-22 13:58:32 -04001117 {
1118 UNIMPLEMENTED();
1119 }
1120
Nicolas Capens4f172c72016-01-13 08:34:30 -05001121 void PixelProgram::TEXGRAD(Vector4f &dst, Vector4f &src0, const Src& src1, Vector4f &src2, Vector4f &src3, Vector4f &offset)
Alexis Hetu25d47fc2015-10-22 13:58:32 -04001122 {
1123 UNIMPLEMENTED();
1124 }
1125
Nicolas Capensc2534f42016-04-04 11:13:24 -04001126 void PixelProgram::TEXLDD(Vector4f &dst, Vector4f &src0, const Src &src1, Vector4f &src2, Vector4f &src3, bool project)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001127 {
Nicolas Capensc2534f42016-04-04 11:13:24 -04001128 sampleTexture(dst, src1, src0.x, src0.y, src0.z, src0.w, src2, src3, project, Grad);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001129 }
1130
Nicolas Capensc2534f42016-04-04 11:13:24 -04001131 void PixelProgram::TEXLDL(Vector4f &dst, Vector4f &src0, const Src &src1, bool project)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001132 {
Nicolas Capensc2534f42016-04-04 11:13:24 -04001133 sampleTexture(dst, src1, src0.x, src0.y, src0.z, src0.w, src0, src0, project, Lod);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001134 }
1135
Nicolas Capens4f172c72016-01-13 08:34:30 -05001136 void PixelProgram::TEXSIZE(Vector4f &dst, Float4 &lod, const Src &src1)
Alexis Hetu9bcb31d2015-07-22 17:03:26 -04001137 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001138 Pointer<Byte> textureMipmap = data + OFFSET(DrawData, mipmap) + src1.index * sizeof(Texture) + OFFSET(Texture, mipmap);
Alexis Hetu9bcb31d2015-07-22 17:03:26 -04001139 for(int i = 0; i < 4; ++i)
1140 {
1141 Pointer<Byte> mipmap = textureMipmap + (As<Int>(Extract(lod, i)) + Int(1)) * sizeof(Mipmap);
1142 dst.x = Insert(dst.x, As<Float>(Int(*Pointer<Short>(mipmap + OFFSET(Mipmap, width)))), i);
1143 dst.y = Insert(dst.y, As<Float>(Int(*Pointer<Short>(mipmap + OFFSET(Mipmap, height)))), i);
1144 dst.z = Insert(dst.z, As<Float>(Int(*Pointer<Short>(mipmap + OFFSET(Mipmap, depth)))), i);
1145 }
1146 }
1147
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001148 void PixelProgram::TEXKILL(Int cMask[4], Vector4f &src, unsigned char mask)
1149 {
1150 Int kill = -1;
1151
1152 if(mask & 0x1) kill &= SignMask(CmpNLT(src.x, Float4(0.0f)));
1153 if(mask & 0x2) kill &= SignMask(CmpNLT(src.y, Float4(0.0f)));
1154 if(mask & 0x4) kill &= SignMask(CmpNLT(src.z, Float4(0.0f)));
1155 if(mask & 0x8) kill &= SignMask(CmpNLT(src.w, Float4(0.0f)));
1156
1157 // FIXME: Dynamic branching affects TEXKILL?
1158 // if(shader->containsDynamicBranching())
1159 // {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001160 // kill = ~SignMask(enableMask());
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001161 // }
1162
1163 for(unsigned int q = 0; q < state.multiSample; q++)
1164 {
1165 cMask[q] &= kill;
1166 }
1167
1168 // FIXME: Branch to end of shader if all killed?
1169 }
1170
Nicolas Capens4f172c72016-01-13 08:34:30 -05001171 void PixelProgram::DISCARD(Int cMask[4], const Shader::Instruction *instruction)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001172 {
1173 Int kill = 0;
1174
1175 if(shader->containsDynamicBranching())
1176 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001177 kill = ~SignMask(enableMask(instruction));
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001178 }
1179
1180 for(unsigned int q = 0; q < state.multiSample; q++)
1181 {
1182 cMask[q] &= kill;
1183 }
1184
1185 // FIXME: Branch to end of shader if all killed?
1186 }
1187
1188 void PixelProgram::DFDX(Vector4f &dst, Vector4f &src)
1189 {
1190 dst.x = src.x.yyww - src.x.xxzz;
1191 dst.y = src.y.yyww - src.y.xxzz;
1192 dst.z = src.z.yyww - src.z.xxzz;
1193 dst.w = src.w.yyww - src.w.xxzz;
1194 }
1195
1196 void PixelProgram::DFDY(Vector4f &dst, Vector4f &src)
1197 {
1198 dst.x = src.x.zwzw - src.x.xyxy;
1199 dst.y = src.y.zwzw - src.y.xyxy;
1200 dst.z = src.z.zwzw - src.z.xyxy;
1201 dst.w = src.w.zwzw - src.w.xyxy;
1202 }
1203
1204 void PixelProgram::FWIDTH(Vector4f &dst, Vector4f &src)
1205 {
1206 // abs(dFdx(src)) + abs(dFdy(src));
1207 dst.x = Abs(src.x.yyww - src.x.xxzz) + Abs(src.x.zwzw - src.x.xyxy);
1208 dst.y = Abs(src.y.yyww - src.y.xxzz) + Abs(src.y.zwzw - src.y.xyxy);
1209 dst.z = Abs(src.z.yyww - src.z.xxzz) + Abs(src.z.zwzw - src.z.xyxy);
1210 dst.w = Abs(src.w.yyww - src.w.xxzz) + Abs(src.w.zwzw - src.w.xyxy);
1211 }
1212
Nicolas Capens4f172c72016-01-13 08:34:30 -05001213 void PixelProgram::BREAK()
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001214 {
1215 llvm::BasicBlock *deadBlock = Nucleus::createBasicBlock();
1216 llvm::BasicBlock *endBlock = loopRepEndBlock[loopRepDepth - 1];
1217
1218 if(breakDepth == 0)
1219 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001220 enableIndex = enableIndex - breakDepth;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001221 Nucleus::createBr(endBlock);
1222 }
1223 else
1224 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001225 enableBreak = enableBreak & ~enableStack[enableIndex];
1226 Bool allBreak = SignMask(enableBreak) == 0x0;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001227
Nicolas Capens4f172c72016-01-13 08:34:30 -05001228 enableIndex = enableIndex - breakDepth;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001229 branch(allBreak, endBlock, deadBlock);
1230 }
1231
1232 Nucleus::setInsertBlock(deadBlock);
Nicolas Capens4f172c72016-01-13 08:34:30 -05001233 enableIndex = enableIndex + breakDepth;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001234 }
1235
Nicolas Capens4f172c72016-01-13 08:34:30 -05001236 void PixelProgram::BREAKC(Vector4f &src0, Vector4f &src1, Control control)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001237 {
1238 Int4 condition;
1239
1240 switch(control)
1241 {
1242 case Shader::CONTROL_GT: condition = CmpNLE(src0.x, src1.x); break;
1243 case Shader::CONTROL_EQ: condition = CmpEQ(src0.x, src1.x); break;
1244 case Shader::CONTROL_GE: condition = CmpNLT(src0.x, src1.x); break;
1245 case Shader::CONTROL_LT: condition = CmpLT(src0.x, src1.x); break;
1246 case Shader::CONTROL_NE: condition = CmpNEQ(src0.x, src1.x); break;
1247 case Shader::CONTROL_LE: condition = CmpLE(src0.x, src1.x); break;
1248 default:
1249 ASSERT(false);
1250 }
1251
Nicolas Capens4f172c72016-01-13 08:34:30 -05001252 BREAK(condition);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001253 }
1254
Nicolas Capens4f172c72016-01-13 08:34:30 -05001255 void PixelProgram::BREAKP(const Src &predicateRegister) // FIXME: Factor out parts common with BREAKC
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001256 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001257 Int4 condition = As<Int4>(p0[predicateRegister.swizzle & 0x3]);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001258
1259 if(predicateRegister.modifier == Shader::MODIFIER_NOT)
1260 {
1261 condition = ~condition;
1262 }
1263
Nicolas Capens4f172c72016-01-13 08:34:30 -05001264 BREAK(condition);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001265 }
1266
Nicolas Capens4f172c72016-01-13 08:34:30 -05001267 void PixelProgram::BREAK(Int4 &condition)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001268 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001269 condition &= enableStack[enableIndex];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001270
1271 llvm::BasicBlock *continueBlock = Nucleus::createBasicBlock();
1272 llvm::BasicBlock *endBlock = loopRepEndBlock[loopRepDepth - 1];
1273
Nicolas Capens4f172c72016-01-13 08:34:30 -05001274 enableBreak = enableBreak & ~condition;
1275 Bool allBreak = SignMask(enableBreak) == 0x0;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001276
Nicolas Capens4f172c72016-01-13 08:34:30 -05001277 enableIndex = enableIndex - breakDepth;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001278 branch(allBreak, endBlock, continueBlock);
1279
1280 Nucleus::setInsertBlock(continueBlock);
Nicolas Capens4f172c72016-01-13 08:34:30 -05001281 enableIndex = enableIndex + breakDepth;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001282 }
1283
Nicolas Capens4f172c72016-01-13 08:34:30 -05001284 void PixelProgram::CONTINUE()
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001285 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001286 enableContinue = enableContinue & ~enableStack[enableIndex];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001287 }
1288
1289 void PixelProgram::TEST()
1290 {
1291 whileTest = true;
1292 }
1293
Nicolas Capens4f172c72016-01-13 08:34:30 -05001294 void PixelProgram::CALL(int labelIndex, int callSiteIndex)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001295 {
1296 if(!labelBlock[labelIndex])
1297 {
1298 labelBlock[labelIndex] = Nucleus::createBasicBlock();
1299 }
1300
1301 if(callRetBlock[labelIndex].size() > 1)
1302 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001303 callStack[stackIndex++] = UInt(callSiteIndex);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001304 }
1305
Nicolas Capens4f172c72016-01-13 08:34:30 -05001306 Int4 restoreLeave = enableLeave;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001307
1308 Nucleus::createBr(labelBlock[labelIndex]);
1309 Nucleus::setInsertBlock(callRetBlock[labelIndex][callSiteIndex]);
1310
Nicolas Capens4f172c72016-01-13 08:34:30 -05001311 enableLeave = restoreLeave;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001312 }
1313
Nicolas Capens4f172c72016-01-13 08:34:30 -05001314 void PixelProgram::CALLNZ(int labelIndex, int callSiteIndex, const Src &src)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001315 {
1316 if(src.type == Shader::PARAMETER_CONSTBOOL)
1317 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001318 CALLNZb(labelIndex, callSiteIndex, src);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001319 }
1320 else if(src.type == Shader::PARAMETER_PREDICATE)
1321 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001322 CALLNZp(labelIndex, callSiteIndex, src);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001323 }
1324 else ASSERT(false);
1325 }
1326
Nicolas Capens4f172c72016-01-13 08:34:30 -05001327 void PixelProgram::CALLNZb(int labelIndex, int callSiteIndex, const Src &boolRegister)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001328 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001329 Bool condition = (*Pointer<Byte>(data + OFFSET(DrawData, ps.b[boolRegister.index])) != Byte(0)); // FIXME
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001330
1331 if(boolRegister.modifier == Shader::MODIFIER_NOT)
1332 {
1333 condition = !condition;
1334 }
1335
1336 if(!labelBlock[labelIndex])
1337 {
1338 labelBlock[labelIndex] = Nucleus::createBasicBlock();
1339 }
1340
1341 if(callRetBlock[labelIndex].size() > 1)
1342 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001343 callStack[stackIndex++] = UInt(callSiteIndex);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001344 }
1345
Nicolas Capens4f172c72016-01-13 08:34:30 -05001346 Int4 restoreLeave = enableLeave;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001347
1348 branch(condition, labelBlock[labelIndex], callRetBlock[labelIndex][callSiteIndex]);
1349 Nucleus::setInsertBlock(callRetBlock[labelIndex][callSiteIndex]);
1350
Nicolas Capens4f172c72016-01-13 08:34:30 -05001351 enableLeave = restoreLeave;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001352 }
1353
Nicolas Capens4f172c72016-01-13 08:34:30 -05001354 void PixelProgram::CALLNZp(int labelIndex, int callSiteIndex, const Src &predicateRegister)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001355 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001356 Int4 condition = As<Int4>(p0[predicateRegister.swizzle & 0x3]);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001357
1358 if(predicateRegister.modifier == Shader::MODIFIER_NOT)
1359 {
1360 condition = ~condition;
1361 }
1362
Nicolas Capens4f172c72016-01-13 08:34:30 -05001363 condition &= enableStack[enableIndex];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001364
1365 if(!labelBlock[labelIndex])
1366 {
1367 labelBlock[labelIndex] = Nucleus::createBasicBlock();
1368 }
1369
1370 if(callRetBlock[labelIndex].size() > 1)
1371 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001372 callStack[stackIndex++] = UInt(callSiteIndex);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001373 }
1374
Nicolas Capens4f172c72016-01-13 08:34:30 -05001375 enableIndex++;
1376 enableStack[enableIndex] = condition;
1377 Int4 restoreLeave = enableLeave;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001378
1379 Bool notAllFalse = SignMask(condition) != 0;
1380 branch(notAllFalse, labelBlock[labelIndex], callRetBlock[labelIndex][callSiteIndex]);
1381 Nucleus::setInsertBlock(callRetBlock[labelIndex][callSiteIndex]);
1382
Nicolas Capens4f172c72016-01-13 08:34:30 -05001383 enableIndex--;
1384 enableLeave = restoreLeave;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001385 }
1386
Nicolas Capens4f172c72016-01-13 08:34:30 -05001387 void PixelProgram::ELSE()
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001388 {
1389 ifDepth--;
1390
1391 llvm::BasicBlock *falseBlock = ifFalseBlock[ifDepth];
1392 llvm::BasicBlock *endBlock = Nucleus::createBasicBlock();
1393
1394 if(isConditionalIf[ifDepth])
1395 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001396 Int4 condition = ~enableStack[enableIndex] & enableStack[enableIndex - 1];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001397 Bool notAllFalse = SignMask(condition) != 0;
1398
1399 branch(notAllFalse, falseBlock, endBlock);
1400
Nicolas Capens4f172c72016-01-13 08:34:30 -05001401 enableStack[enableIndex] = ~enableStack[enableIndex] & enableStack[enableIndex - 1];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001402 }
1403 else
1404 {
1405 Nucleus::createBr(endBlock);
1406 Nucleus::setInsertBlock(falseBlock);
1407 }
1408
1409 ifFalseBlock[ifDepth] = endBlock;
1410
1411 ifDepth++;
1412 }
1413
Nicolas Capens4f172c72016-01-13 08:34:30 -05001414 void PixelProgram::ENDIF()
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001415 {
1416 ifDepth--;
1417
1418 llvm::BasicBlock *endBlock = ifFalseBlock[ifDepth];
1419
1420 Nucleus::createBr(endBlock);
1421 Nucleus::setInsertBlock(endBlock);
1422
1423 if(isConditionalIf[ifDepth])
1424 {
1425 breakDepth--;
Nicolas Capens4f172c72016-01-13 08:34:30 -05001426 enableIndex--;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001427 }
1428 }
1429
Nicolas Capens4f172c72016-01-13 08:34:30 -05001430 void PixelProgram::ENDLOOP()
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001431 {
1432 loopRepDepth--;
1433
Nicolas Capens4f172c72016-01-13 08:34:30 -05001434 aL[loopDepth] = aL[loopDepth] + increment[loopDepth]; // FIXME: +=
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001435
1436 llvm::BasicBlock *testBlock = loopRepTestBlock[loopRepDepth];
1437 llvm::BasicBlock *endBlock = loopRepEndBlock[loopRepDepth];
1438
1439 Nucleus::createBr(testBlock);
1440 Nucleus::setInsertBlock(endBlock);
1441
Nicolas Capens4f172c72016-01-13 08:34:30 -05001442 loopDepth--;
1443 enableBreak = Int4(0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001444 }
1445
Nicolas Capens4f172c72016-01-13 08:34:30 -05001446 void PixelProgram::ENDREP()
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001447 {
1448 loopRepDepth--;
1449
1450 llvm::BasicBlock *testBlock = loopRepTestBlock[loopRepDepth];
1451 llvm::BasicBlock *endBlock = loopRepEndBlock[loopRepDepth];
1452
1453 Nucleus::createBr(testBlock);
1454 Nucleus::setInsertBlock(endBlock);
1455
Nicolas Capens4f172c72016-01-13 08:34:30 -05001456 loopDepth--;
1457 enableBreak = Int4(0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001458 }
1459
Nicolas Capens4f172c72016-01-13 08:34:30 -05001460 void PixelProgram::ENDWHILE()
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001461 {
1462 loopRepDepth--;
1463
1464 llvm::BasicBlock *testBlock = loopRepTestBlock[loopRepDepth];
1465 llvm::BasicBlock *endBlock = loopRepEndBlock[loopRepDepth];
1466
1467 Nucleus::createBr(testBlock);
1468 Nucleus::setInsertBlock(endBlock);
1469
Nicolas Capens4f172c72016-01-13 08:34:30 -05001470 enableIndex--;
1471 enableBreak = Int4(0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001472 whileTest = false;
1473 }
1474
Nicolas Capens4f172c72016-01-13 08:34:30 -05001475 void PixelProgram::IF(const Src &src)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001476 {
1477 if(src.type == Shader::PARAMETER_CONSTBOOL)
1478 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001479 IFb(src);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001480 }
1481 else if(src.type == Shader::PARAMETER_PREDICATE)
1482 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001483 IFp(src);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001484 }
1485 else
1486 {
Nicolas Capensc2534f42016-04-04 11:13:24 -04001487 Int4 condition = As<Int4>(fetchRegister(src).x);
Nicolas Capens4f172c72016-01-13 08:34:30 -05001488 IF(condition);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001489 }
1490 }
1491
Nicolas Capens4f172c72016-01-13 08:34:30 -05001492 void PixelProgram::IFb(const Src &boolRegister)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001493 {
1494 ASSERT(ifDepth < 24 + 4);
1495
Nicolas Capens4f172c72016-01-13 08:34:30 -05001496 Bool condition = (*Pointer<Byte>(data + OFFSET(DrawData, ps.b[boolRegister.index])) != Byte(0)); // FIXME
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001497
1498 if(boolRegister.modifier == Shader::MODIFIER_NOT)
1499 {
1500 condition = !condition;
1501 }
1502
1503 llvm::BasicBlock *trueBlock = Nucleus::createBasicBlock();
1504 llvm::BasicBlock *falseBlock = Nucleus::createBasicBlock();
1505
1506 branch(condition, trueBlock, falseBlock);
1507
1508 isConditionalIf[ifDepth] = false;
1509 ifFalseBlock[ifDepth] = falseBlock;
1510
1511 ifDepth++;
1512 }
1513
Nicolas Capens4f172c72016-01-13 08:34:30 -05001514 void PixelProgram::IFp(const Src &predicateRegister)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001515 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001516 Int4 condition = As<Int4>(p0[predicateRegister.swizzle & 0x3]);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001517
1518 if(predicateRegister.modifier == Shader::MODIFIER_NOT)
1519 {
1520 condition = ~condition;
1521 }
1522
Nicolas Capens4f172c72016-01-13 08:34:30 -05001523 IF(condition);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001524 }
1525
Nicolas Capens4f172c72016-01-13 08:34:30 -05001526 void PixelProgram::IFC(Vector4f &src0, Vector4f &src1, Control control)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001527 {
1528 Int4 condition;
1529
1530 switch(control)
1531 {
1532 case Shader::CONTROL_GT: condition = CmpNLE(src0.x, src1.x); break;
1533 case Shader::CONTROL_EQ: condition = CmpEQ(src0.x, src1.x); break;
1534 case Shader::CONTROL_GE: condition = CmpNLT(src0.x, src1.x); break;
1535 case Shader::CONTROL_LT: condition = CmpLT(src0.x, src1.x); break;
1536 case Shader::CONTROL_NE: condition = CmpNEQ(src0.x, src1.x); break;
1537 case Shader::CONTROL_LE: condition = CmpLE(src0.x, src1.x); break;
1538 default:
1539 ASSERT(false);
1540 }
1541
Nicolas Capens4f172c72016-01-13 08:34:30 -05001542 IF(condition);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001543 }
1544
Nicolas Capens4f172c72016-01-13 08:34:30 -05001545 void PixelProgram::IF(Int4 &condition)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001546 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001547 condition &= enableStack[enableIndex];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001548
Nicolas Capens4f172c72016-01-13 08:34:30 -05001549 enableIndex++;
1550 enableStack[enableIndex] = condition;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001551
1552 llvm::BasicBlock *trueBlock = Nucleus::createBasicBlock();
1553 llvm::BasicBlock *falseBlock = Nucleus::createBasicBlock();
1554
1555 Bool notAllFalse = SignMask(condition) != 0;
1556
1557 branch(notAllFalse, trueBlock, falseBlock);
1558
1559 isConditionalIf[ifDepth] = true;
1560 ifFalseBlock[ifDepth] = falseBlock;
1561
1562 ifDepth++;
1563 breakDepth++;
1564 }
1565
1566 void PixelProgram::LABEL(int labelIndex)
1567 {
1568 if(!labelBlock[labelIndex])
1569 {
1570 labelBlock[labelIndex] = Nucleus::createBasicBlock();
1571 }
1572
1573 Nucleus::setInsertBlock(labelBlock[labelIndex]);
1574 currentLabel = labelIndex;
1575 }
1576
Nicolas Capens4f172c72016-01-13 08:34:30 -05001577 void PixelProgram::LOOP(const Src &integerRegister)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001578 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001579 loopDepth++;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001580
Nicolas Capens4f172c72016-01-13 08:34:30 -05001581 iteration[loopDepth] = *Pointer<Int>(data + OFFSET(DrawData, ps.i[integerRegister.index][0]));
1582 aL[loopDepth] = *Pointer<Int>(data + OFFSET(DrawData, ps.i[integerRegister.index][1]));
1583 increment[loopDepth] = *Pointer<Int>(data + OFFSET(DrawData, ps.i[integerRegister.index][2]));
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001584
Nicolas Capens4f172c72016-01-13 08:34:30 -05001585 // If(increment[loopDepth] == 0)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001586 // {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001587 // increment[loopDepth] = 1;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001588 // }
1589
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::REP(const Src &integerRegister)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001611 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001612 loopDepth++;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001613
Nicolas Capens4f172c72016-01-13 08:34:30 -05001614 iteration[loopDepth] = *Pointer<Int>(data + OFFSET(DrawData, ps.i[integerRegister.index][0]));
1615 aL[loopDepth] = aL[loopDepth - 1];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001616
1617 llvm::BasicBlock *loopBlock = Nucleus::createBasicBlock();
1618 llvm::BasicBlock *testBlock = Nucleus::createBasicBlock();
1619 llvm::BasicBlock *endBlock = Nucleus::createBasicBlock();
1620
1621 loopRepTestBlock[loopRepDepth] = testBlock;
1622 loopRepEndBlock[loopRepDepth] = endBlock;
1623
1624 // FIXME: jump(testBlock)
1625 Nucleus::createBr(testBlock);
1626 Nucleus::setInsertBlock(testBlock);
1627
Nicolas Capens4f172c72016-01-13 08:34:30 -05001628 branch(iteration[loopDepth] > 0, loopBlock, endBlock);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001629 Nucleus::setInsertBlock(loopBlock);
1630
Nicolas Capens4f172c72016-01-13 08:34:30 -05001631 iteration[loopDepth] = iteration[loopDepth] - 1; // FIXME: --
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001632
1633 loopRepDepth++;
1634 breakDepth = 0;
1635 }
1636
Nicolas Capens4f172c72016-01-13 08:34:30 -05001637 void PixelProgram::WHILE(const Src &temporaryRegister)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001638 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001639 enableIndex++;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001640
1641 llvm::BasicBlock *loopBlock = Nucleus::createBasicBlock();
1642 llvm::BasicBlock *testBlock = Nucleus::createBasicBlock();
1643 llvm::BasicBlock *endBlock = Nucleus::createBasicBlock();
1644
1645 loopRepTestBlock[loopRepDepth] = testBlock;
1646 loopRepEndBlock[loopRepDepth] = endBlock;
1647
Nicolas Capens4f172c72016-01-13 08:34:30 -05001648 Int4 restoreBreak = enableBreak;
1649 Int4 restoreContinue = enableContinue;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001650
1651 // FIXME: jump(testBlock)
1652 Nucleus::createBr(testBlock);
1653 Nucleus::setInsertBlock(testBlock);
Nicolas Capens4f172c72016-01-13 08:34:30 -05001654 enableContinue = restoreContinue;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001655
Nicolas Capensc2534f42016-04-04 11:13:24 -04001656 const Vector4f &src = fetchRegister(temporaryRegister);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001657 Int4 condition = As<Int4>(src.x);
Nicolas Capens4f172c72016-01-13 08:34:30 -05001658 condition &= enableStack[enableIndex - 1];
Nicolas Capens2ff29482016-04-28 15:28:02 -04001659 if(shader->containsLeaveInstruction()) condition &= enableLeave;
Nicolas Capens4f172c72016-01-13 08:34:30 -05001660 enableStack[enableIndex] = condition;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001661
1662 Bool notAllFalse = SignMask(condition) != 0;
1663 branch(notAllFalse, loopBlock, endBlock);
1664
1665 Nucleus::setInsertBlock(endBlock);
Nicolas Capens4f172c72016-01-13 08:34:30 -05001666 enableBreak = restoreBreak;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001667
1668 Nucleus::setInsertBlock(loopBlock);
1669
1670 loopRepDepth++;
1671 breakDepth = 0;
1672 }
1673
Nicolas Capens4f172c72016-01-13 08:34:30 -05001674 void PixelProgram::RET()
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001675 {
1676 if(currentLabel == -1)
1677 {
1678 returnBlock = Nucleus::createBasicBlock();
1679 Nucleus::createBr(returnBlock);
1680 }
1681 else
1682 {
1683 llvm::BasicBlock *unreachableBlock = Nucleus::createBasicBlock();
1684
1685 if(callRetBlock[currentLabel].size() > 1) // Pop the return destination from the call stack
1686 {
1687 // FIXME: Encapsulate
Nicolas Capens4f172c72016-01-13 08:34:30 -05001688 UInt index = callStack[--stackIndex];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001689
1690 llvm::Value *value = index.loadValue();
1691 llvm::Value *switchInst = Nucleus::createSwitch(value, unreachableBlock, (int)callRetBlock[currentLabel].size());
1692
1693 for(unsigned int i = 0; i < callRetBlock[currentLabel].size(); i++)
1694 {
1695 Nucleus::addSwitchCase(switchInst, i, callRetBlock[currentLabel][i]);
1696 }
1697 }
1698 else if(callRetBlock[currentLabel].size() == 1) // Jump directly to the unique return destination
1699 {
1700 Nucleus::createBr(callRetBlock[currentLabel][0]);
1701 }
1702 else // Function isn't called
1703 {
1704 Nucleus::createBr(unreachableBlock);
1705 }
1706
1707 Nucleus::setInsertBlock(unreachableBlock);
1708 Nucleus::createUnreachable();
1709 }
1710 }
1711
Nicolas Capens4f172c72016-01-13 08:34:30 -05001712 void PixelProgram::LEAVE()
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001713 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001714 enableLeave = enableLeave & ~enableStack[enableIndex];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001715
1716 // FIXME: Return from function if all instances left
1717 // FIXME: Use enableLeave in other control-flow constructs
1718 }
Nicolas Capens5d961882016-01-01 23:18:14 -05001719}