blob: be24a006d7a4bf9e36491e79433df64507a7e6f5 [file] [log] [blame]
Nicolas Capens0bac2852016-05-07 06:09:58 -04001// Copyright 2016 The SwiftShader Authors. All Rights Reserved.
Nicolas Capens5d961882016-01-01 23:18:14 -05002//
Nicolas Capens0bac2852016-05-07 06:09:58 -04003// Licensed under the Apache License, Version 2.0 (the "License");
4// you may not use this file except in compliance with the License.
5// You may obtain a copy of the License at
Nicolas Capens5d961882016-01-01 23:18:14 -05006//
Nicolas Capens0bac2852016-05-07 06:09:58 -04007// http://www.apache.org/licenses/LICENSE-2.0
Nicolas Capens5d961882016-01-01 23:18:14 -05008//
Nicolas Capens0bac2852016-05-07 06:09:58 -04009// Unless required by applicable law or agreed to in writing, software
10// distributed under the License is distributed on an "AS IS" BASIS,
11// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12// See the License for the specific language governing permissions and
13// limitations under the License.
Nicolas Capens5d961882016-01-01 23:18:14 -050014
15#include "PixelProgram.hpp"
16#include "Primitive.hpp"
17#include "Renderer.hpp"
18#include "SamplerCore.hpp"
19
20namespace sw
21{
22 extern bool postBlendSRGB;
Alexis Hetuf2a8c372015-07-13 11:08:41 -040023 extern bool booleanFaceRegister;
24 extern bool halfIntegerCoordinates; // Pixel centers are not at integer coordinates
Nicolas Capens5d961882016-01-01 23:18:14 -050025 extern bool fullPixelPositionRegister;
26
Nicolas Capens4f172c72016-01-13 08:34:30 -050027 void PixelProgram::setBuiltins(Int &x, Int &y, Float4(&z)[4], Float4 &w)
Nicolas Capens5d961882016-01-01 23:18:14 -050028 {
Alexis Hetuf2a8c372015-07-13 11:08:41 -040029 if(shader->getVersion() >= 0x0300)
30 {
31 if(shader->vPosDeclared)
32 {
33 if(!halfIntegerCoordinates)
34 {
Nicolas Capens4f172c72016-01-13 08:34:30 -050035 vPos.x = Float4(Float(x)) + Float4(0, 1, 0, 1);
36 vPos.y = Float4(Float(y)) + Float4(0, 0, 1, 1);
Alexis Hetuf2a8c372015-07-13 11:08:41 -040037 }
38 else
39 {
Nicolas Capens4f172c72016-01-13 08:34:30 -050040 vPos.x = Float4(Float(x)) + Float4(0.5f, 1.5f, 0.5f, 1.5f);
41 vPos.y = Float4(Float(y)) + Float4(0.5f, 0.5f, 1.5f, 1.5f);
Alexis Hetuf2a8c372015-07-13 11:08:41 -040042 }
43
44 if(fullPixelPositionRegister)
45 {
Nicolas Capens4f172c72016-01-13 08:34:30 -050046 vPos.z = z[0]; // FIXME: Centroid?
47 vPos.w = w; // FIXME: Centroid?
Alexis Hetuf2a8c372015-07-13 11:08:41 -040048 }
49 }
50
51 if(shader->vFaceDeclared)
52 {
Nicolas Capens4f172c72016-01-13 08:34:30 -050053 Float4 area = *Pointer<Float>(primitive + OFFSET(Primitive, area));
Alexis Hetuf2a8c372015-07-13 11:08:41 -040054 Float4 face = booleanFaceRegister ? Float4(As<Float4>(CmpNLT(area, Float4(0.0f)))) : area;
55
Nicolas Capens4f172c72016-01-13 08:34:30 -050056 vFace.x = face;
57 vFace.y = face;
58 vFace.z = face;
59 vFace.w = face;
Alexis Hetuf2a8c372015-07-13 11:08:41 -040060 }
Nicolas Capens5d961882016-01-01 23:18:14 -050061 }
62 }
63
Nicolas Capens4f172c72016-01-13 08:34:30 -050064 void PixelProgram::applyShader(Int cMask[4])
Nicolas Capens5d961882016-01-01 23:18:14 -050065 {
Nicolas Capens4f172c72016-01-13 08:34:30 -050066 enableIndex = 0;
67 stackIndex = 0;
Alexis Hetuf2a8c372015-07-13 11:08:41 -040068
69 if(shader->containsLeaveInstruction())
70 {
Nicolas Capens4f172c72016-01-13 08:34:30 -050071 enableLeave = Int4(0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF);
Alexis Hetuf2a8c372015-07-13 11:08:41 -040072 }
73
Alexis Hetu1edcd8b2015-11-05 11:12:41 -050074 for(int i = 0; i < RENDERTARGETS; i++)
75 {
76 if(state.targetFormat[i] != FORMAT_NULL)
77 {
Nicolas Capens4f172c72016-01-13 08:34:30 -050078 oC[i] = Vector4f(0.0f, 0.0f, 0.0f, 0.0f);
Alexis Hetu1edcd8b2015-11-05 11:12:41 -050079 }
80 }
Alexis Hetuf2a8c372015-07-13 11:08:41 -040081
82 // Create all call site return blocks up front
83 for(size_t i = 0; i < shader->getLength(); i++)
84 {
85 const Shader::Instruction *instruction = shader->getInstruction(i);
86 Shader::Opcode opcode = instruction->opcode;
87
88 if(opcode == Shader::OPCODE_CALL || opcode == Shader::OPCODE_CALLNZ)
89 {
90 const Dst &dst = instruction->dst;
91
92 ASSERT(callRetBlock[dst.label].size() == dst.callSite);
93 callRetBlock[dst.label].push_back(Nucleus::createBasicBlock());
94 }
95 }
96
Nicolas Capens8b124c12016-04-18 14:09:37 -040097 bool broadcastColor0 = true;
98
Alexis Hetuf2a8c372015-07-13 11:08:41 -040099 for(size_t i = 0; i < shader->getLength(); i++)
100 {
101 const Shader::Instruction *instruction = shader->getInstruction(i);
102 Shader::Opcode opcode = instruction->opcode;
103
104 if(opcode == Shader::OPCODE_DCL || opcode == Shader::OPCODE_DEF || opcode == Shader::OPCODE_DEFI || opcode == Shader::OPCODE_DEFB)
105 {
106 continue;
107 }
108
109 const Dst &dst = instruction->dst;
110 const Src &src0 = instruction->src[0];
111 const Src &src1 = instruction->src[1];
112 const Src &src2 = instruction->src[2];
113 const Src &src3 = instruction->src[3];
Alexis Hetu25d47fc2015-10-22 13:58:32 -0400114 const Src &src4 = instruction->src[4];
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400115
116 bool predicate = instruction->predicate;
117 Control control = instruction->control;
118 bool pp = dst.partialPrecision;
119 bool project = instruction->project;
120 bool bias = instruction->bias;
121
122 Vector4f d;
123 Vector4f s0;
124 Vector4f s1;
125 Vector4f s2;
126 Vector4f s3;
Alexis Hetu25d47fc2015-10-22 13:58:32 -0400127 Vector4f s4;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400128
129 if(opcode == Shader::OPCODE_TEXKILL) // Takes destination as input
130 {
131 if(dst.type == Shader::PARAMETER_TEXTURE)
132 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500133 d.x = v[2 + dst.index].x;
134 d.y = v[2 + dst.index].y;
135 d.z = v[2 + dst.index].z;
136 d.w = v[2 + dst.index].w;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400137 }
138 else
139 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500140 d = r[dst.index];
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400141 }
142 }
143
Nicolas Capensc2534f42016-04-04 11:13:24 -0400144 if(src0.type != Shader::PARAMETER_VOID) s0 = fetchRegister(src0);
145 if(src1.type != Shader::PARAMETER_VOID) s1 = fetchRegister(src1);
146 if(src2.type != Shader::PARAMETER_VOID) s2 = fetchRegister(src2);
147 if(src3.type != Shader::PARAMETER_VOID) s3 = fetchRegister(src3);
148 if(src4.type != Shader::PARAMETER_VOID) s4 = fetchRegister(src4);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400149
150 switch(opcode)
151 {
152 case Shader::OPCODE_PS_2_0: break;
153 case Shader::OPCODE_PS_2_x: break;
154 case Shader::OPCODE_PS_3_0: break;
155 case Shader::OPCODE_DEF: break;
156 case Shader::OPCODE_DCL: break;
157 case Shader::OPCODE_NOP: break;
158 case Shader::OPCODE_MOV: mov(d, s0); break;
Alexis Hetu8d78cf72015-08-28 14:24:45 -0400159 case Shader::OPCODE_NEG: neg(d, s0); break;
160 case Shader::OPCODE_INEG: ineg(d, s0); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400161 case Shader::OPCODE_F2B: f2b(d, s0); break;
162 case Shader::OPCODE_B2F: b2f(d, s0); break;
Alexis Hetu8d78cf72015-08-28 14:24:45 -0400163 case Shader::OPCODE_F2I: f2i(d, s0); break;
164 case Shader::OPCODE_I2F: i2f(d, s0); break;
165 case Shader::OPCODE_F2U: f2u(d, s0); break;
166 case Shader::OPCODE_U2F: u2f(d, s0); break;
167 case Shader::OPCODE_I2B: i2b(d, s0); break;
168 case Shader::OPCODE_B2I: b2i(d, s0); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400169 case Shader::OPCODE_ADD: add(d, s0, s1); break;
Alexis Hetu8d78cf72015-08-28 14:24:45 -0400170 case Shader::OPCODE_IADD: iadd(d, s0, s1); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400171 case Shader::OPCODE_SUB: sub(d, s0, s1); break;
Alexis Hetu8d78cf72015-08-28 14:24:45 -0400172 case Shader::OPCODE_ISUB: isub(d, s0, s1); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400173 case Shader::OPCODE_MUL: mul(d, s0, s1); break;
Alexis Hetu8d78cf72015-08-28 14:24:45 -0400174 case Shader::OPCODE_IMUL: imul(d, s0, s1); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400175 case Shader::OPCODE_MAD: mad(d, s0, s1, s2); break;
Alexis Hetu8d78cf72015-08-28 14:24:45 -0400176 case Shader::OPCODE_IMAD: imad(d, s0, s1, s2); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400177 case Shader::OPCODE_DP1: dp1(d, s0, s1); break;
178 case Shader::OPCODE_DP2: dp2(d, s0, s1); break;
179 case Shader::OPCODE_DP2ADD: dp2add(d, s0, s1, s2); break;
180 case Shader::OPCODE_DP3: dp3(d, s0, s1); break;
181 case Shader::OPCODE_DP4: dp4(d, s0, s1); break;
Alexis Hetuc3d95f32015-09-23 12:27:32 -0400182 case Shader::OPCODE_DET2: det2(d, s0, s1); break;
183 case Shader::OPCODE_DET3: det3(d, s0, s1, s2); break;
184 case Shader::OPCODE_DET4: det4(d, s0, s1, s2, s3); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400185 case Shader::OPCODE_CMP0: cmp0(d, s0, s1, s2); break;
186 case Shader::OPCODE_ICMP: icmp(d, s0, s1, control); break;
Alexis Hetu8d78cf72015-08-28 14:24:45 -0400187 case Shader::OPCODE_UCMP: ucmp(d, s0, s1, control); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400188 case Shader::OPCODE_SELECT: select(d, s0, s1, s2); break;
189 case Shader::OPCODE_EXTRACT: extract(d.x, s0, s1.x); break;
190 case Shader::OPCODE_INSERT: insert(d, s0, s1.x, s2.x); break;
191 case Shader::OPCODE_FRC: frc(d, s0); break;
192 case Shader::OPCODE_TRUNC: trunc(d, s0); break;
193 case Shader::OPCODE_FLOOR: floor(d, s0); break;
194 case Shader::OPCODE_ROUND: round(d, s0); break;
195 case Shader::OPCODE_ROUNDEVEN: roundEven(d, s0); break;
196 case Shader::OPCODE_CEIL: ceil(d, s0); break;
197 case Shader::OPCODE_EXP2X: exp2x(d, s0, pp); break;
198 case Shader::OPCODE_EXP2: exp2(d, s0, pp); break;
199 case Shader::OPCODE_LOG2X: log2x(d, s0, pp); break;
200 case Shader::OPCODE_LOG2: log2(d, s0, pp); break;
201 case Shader::OPCODE_EXP: exp(d, s0, pp); break;
202 case Shader::OPCODE_LOG: log(d, s0, pp); break;
203 case Shader::OPCODE_RCPX: rcpx(d, s0, pp); break;
204 case Shader::OPCODE_DIV: div(d, s0, s1); break;
Alexis Hetu8d78cf72015-08-28 14:24:45 -0400205 case Shader::OPCODE_IDIV: idiv(d, s0, s1); break;
206 case Shader::OPCODE_UDIV: udiv(d, s0, s1); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400207 case Shader::OPCODE_MOD: mod(d, s0, s1); break;
Alexis Hetu8d78cf72015-08-28 14:24:45 -0400208 case Shader::OPCODE_IMOD: imod(d, s0, s1); break;
209 case Shader::OPCODE_UMOD: umod(d, s0, s1); break;
210 case Shader::OPCODE_SHL: shl(d, s0, s1); break;
211 case Shader::OPCODE_ISHR: ishr(d, s0, s1); break;
212 case Shader::OPCODE_USHR: ushr(d, s0, s1); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400213 case Shader::OPCODE_RSQX: rsqx(d, s0, pp); break;
214 case Shader::OPCODE_SQRT: sqrt(d, s0, pp); break;
215 case Shader::OPCODE_RSQ: rsq(d, s0, pp); break;
216 case Shader::OPCODE_LEN2: len2(d.x, s0, pp); break;
217 case Shader::OPCODE_LEN3: len3(d.x, s0, pp); break;
218 case Shader::OPCODE_LEN4: len4(d.x, s0, pp); break;
219 case Shader::OPCODE_DIST1: dist1(d.x, s0, s1, pp); break;
220 case Shader::OPCODE_DIST2: dist2(d.x, s0, s1, pp); break;
221 case Shader::OPCODE_DIST3: dist3(d.x, s0, s1, pp); break;
222 case Shader::OPCODE_DIST4: dist4(d.x, s0, s1, pp); break;
223 case Shader::OPCODE_MIN: min(d, s0, s1); break;
Alexis Hetu8d78cf72015-08-28 14:24:45 -0400224 case Shader::OPCODE_IMIN: imin(d, s0, s1); break;
225 case Shader::OPCODE_UMIN: umin(d, s0, s1); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400226 case Shader::OPCODE_MAX: max(d, s0, s1); break;
Alexis Hetu8d78cf72015-08-28 14:24:45 -0400227 case Shader::OPCODE_IMAX: imax(d, s0, s1); break;
228 case Shader::OPCODE_UMAX: umax(d, s0, s1); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400229 case Shader::OPCODE_LRP: lrp(d, s0, s1, s2); break;
230 case Shader::OPCODE_STEP: step(d, s0, s1); break;
231 case Shader::OPCODE_SMOOTH: smooth(d, s0, s1, s2); break;
Alexis Hetu8d78cf72015-08-28 14:24:45 -0400232 case Shader::OPCODE_FLOATBITSTOINT:
233 case Shader::OPCODE_FLOATBITSTOUINT:
234 case Shader::OPCODE_INTBITSTOFLOAT:
235 case Shader::OPCODE_UINTBITSTOFLOAT: d = s0; break;
Alexis Hetu9cde9742016-04-06 13:03:38 -0400236 case Shader::OPCODE_PACKSNORM2x16: packSnorm2x16(d, s0); break;
237 case Shader::OPCODE_PACKUNORM2x16: packUnorm2x16(d, s0); break;
Alexis Hetuffb35eb2016-04-06 18:05:00 -0400238 case Shader::OPCODE_PACKHALF2x16: packHalf2x16(d, s0); break;
Alexis Hetu9cde9742016-04-06 13:03:38 -0400239 case Shader::OPCODE_UNPACKSNORM2x16: unpackSnorm2x16(d, s0); break;
240 case Shader::OPCODE_UNPACKUNORM2x16: unpackUnorm2x16(d, s0); break;
Alexis Hetuffb35eb2016-04-06 18:05:00 -0400241 case Shader::OPCODE_UNPACKHALF2x16: unpackHalf2x16(d, s0); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400242 case Shader::OPCODE_POWX: powx(d, s0, s1, pp); break;
243 case Shader::OPCODE_POW: pow(d, s0, s1, pp); break;
244 case Shader::OPCODE_SGN: sgn(d, s0); break;
Alexis Hetu0f448072016-03-18 10:56:08 -0400245 case Shader::OPCODE_ISGN: isgn(d, s0); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400246 case Shader::OPCODE_CRS: crs(d, s0, s1); break;
247 case Shader::OPCODE_FORWARD1: forward1(d, s0, s1, s2); break;
248 case Shader::OPCODE_FORWARD2: forward2(d, s0, s1, s2); break;
249 case Shader::OPCODE_FORWARD3: forward3(d, s0, s1, s2); break;
250 case Shader::OPCODE_FORWARD4: forward4(d, s0, s1, s2); break;
251 case Shader::OPCODE_REFLECT1: reflect1(d, s0, s1); break;
252 case Shader::OPCODE_REFLECT2: reflect2(d, s0, s1); break;
253 case Shader::OPCODE_REFLECT3: reflect3(d, s0, s1); break;
254 case Shader::OPCODE_REFLECT4: reflect4(d, s0, s1); break;
255 case Shader::OPCODE_REFRACT1: refract1(d, s0, s1, s2.x); break;
256 case Shader::OPCODE_REFRACT2: refract2(d, s0, s1, s2.x); break;
257 case Shader::OPCODE_REFRACT3: refract3(d, s0, s1, s2.x); break;
258 case Shader::OPCODE_REFRACT4: refract4(d, s0, s1, s2.x); break;
259 case Shader::OPCODE_NRM2: nrm2(d, s0, pp); break;
260 case Shader::OPCODE_NRM3: nrm3(d, s0, pp); break;
261 case Shader::OPCODE_NRM4: nrm4(d, s0, pp); break;
262 case Shader::OPCODE_ABS: abs(d, s0); break;
Alexis Hetu0f448072016-03-18 10:56:08 -0400263 case Shader::OPCODE_IABS: iabs(d, s0); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400264 case Shader::OPCODE_SINCOS: sincos(d, s0, pp); break;
265 case Shader::OPCODE_COS: cos(d, s0, pp); break;
266 case Shader::OPCODE_SIN: sin(d, s0, pp); break;
267 case Shader::OPCODE_TAN: tan(d, s0, pp); break;
268 case Shader::OPCODE_ACOS: acos(d, s0, pp); break;
269 case Shader::OPCODE_ASIN: asin(d, s0, pp); break;
270 case Shader::OPCODE_ATAN: atan(d, s0, pp); break;
271 case Shader::OPCODE_ATAN2: atan2(d, s0, s1, pp); break;
272 case Shader::OPCODE_COSH: cosh(d, s0, pp); break;
273 case Shader::OPCODE_SINH: sinh(d, s0, pp); break;
274 case Shader::OPCODE_TANH: tanh(d, s0, pp); break;
275 case Shader::OPCODE_ACOSH: acosh(d, s0, pp); break;
276 case Shader::OPCODE_ASINH: asinh(d, s0, pp); break;
277 case Shader::OPCODE_ATANH: atanh(d, s0, pp); break;
Nicolas Capens4f172c72016-01-13 08:34:30 -0500278 case Shader::OPCODE_M4X4: M4X4(d, s0, src1); break;
279 case Shader::OPCODE_M4X3: M4X3(d, s0, src1); break;
280 case Shader::OPCODE_M3X4: M3X4(d, s0, src1); break;
281 case Shader::OPCODE_M3X3: M3X3(d, s0, src1); break;
282 case Shader::OPCODE_M3X2: M3X2(d, s0, src1); break;
283 case Shader::OPCODE_TEX: TEXLD(d, s0, src1, project, bias); break;
Nicolas Capensc2534f42016-04-04 11:13:24 -0400284 case Shader::OPCODE_TEXLDD: TEXLDD(d, s0, src1, s2, s3, project); break;
285 case Shader::OPCODE_TEXLDL: TEXLDL(d, s0, src1, project); break;
Nicolas Capens4f172c72016-01-13 08:34:30 -0500286 case Shader::OPCODE_TEXSIZE: TEXSIZE(d, s0.x, src1); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400287 case Shader::OPCODE_TEXKILL: TEXKILL(cMask, d, dst.mask); break;
Meng-Lin Wu2337a192016-06-01 13:54:07 -0400288 case Shader::OPCODE_TEXOFFSET: TEXOFFSET(d, s0, src1, s2, project, bias); break;
Nicolas Capens4f172c72016-01-13 08:34:30 -0500289 case Shader::OPCODE_TEXLDLOFFSET: TEXLDL(d, s0, src1, s2, project, bias); break;
290 case Shader::OPCODE_TEXELFETCH: TEXELFETCH(d, s0, src1, s2); break;
291 case Shader::OPCODE_TEXELFETCHOFFSET: TEXELFETCH(d, s0, src1, s2, s3); break;
292 case Shader::OPCODE_TEXGRAD: TEXGRAD(d, s0, src1, s2, s3); break;
293 case Shader::OPCODE_TEXGRADOFFSET: TEXGRAD(d, s0, src1, s2, s3, s4); break;
294 case Shader::OPCODE_DISCARD: DISCARD(cMask, instruction); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400295 case Shader::OPCODE_DFDX: DFDX(d, s0); break;
296 case Shader::OPCODE_DFDY: DFDY(d, s0); break;
297 case Shader::OPCODE_FWIDTH: FWIDTH(d, s0); break;
Nicolas Capens4f172c72016-01-13 08:34:30 -0500298 case Shader::OPCODE_BREAK: BREAK(); break;
299 case Shader::OPCODE_BREAKC: BREAKC(s0, s1, control); break;
300 case Shader::OPCODE_BREAKP: BREAKP(src0); break;
301 case Shader::OPCODE_CONTINUE: CONTINUE(); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400302 case Shader::OPCODE_TEST: TEST(); break;
Nicolas Capens4f172c72016-01-13 08:34:30 -0500303 case Shader::OPCODE_CALL: CALL(dst.label, dst.callSite); break;
304 case Shader::OPCODE_CALLNZ: CALLNZ(dst.label, dst.callSite, src0); break;
305 case Shader::OPCODE_ELSE: ELSE(); break;
306 case Shader::OPCODE_ENDIF: ENDIF(); break;
307 case Shader::OPCODE_ENDLOOP: ENDLOOP(); break;
308 case Shader::OPCODE_ENDREP: ENDREP(); break;
309 case Shader::OPCODE_ENDWHILE: ENDWHILE(); break;
Alexis Hetu9aa83a92016-05-02 17:34:46 -0400310 case Shader::OPCODE_ENDSWITCH: ENDSWITCH(); break;
Nicolas Capens4f172c72016-01-13 08:34:30 -0500311 case Shader::OPCODE_IF: IF(src0); break;
312 case Shader::OPCODE_IFC: IFC(s0, s1, control); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400313 case Shader::OPCODE_LABEL: LABEL(dst.index); break;
Nicolas Capens4f172c72016-01-13 08:34:30 -0500314 case Shader::OPCODE_LOOP: LOOP(src1); break;
315 case Shader::OPCODE_REP: REP(src0); break;
316 case Shader::OPCODE_WHILE: WHILE(src0); break;
Alexis Hetu9aa83a92016-05-02 17:34:46 -0400317 case Shader::OPCODE_SWITCH: SWITCH(); break;
Nicolas Capens4f172c72016-01-13 08:34:30 -0500318 case Shader::OPCODE_RET: RET(); break;
319 case Shader::OPCODE_LEAVE: LEAVE(); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400320 case Shader::OPCODE_CMP: cmp(d, s0, s1, control); break;
321 case Shader::OPCODE_ALL: all(d.x, s0); break;
322 case Shader::OPCODE_ANY: any(d.x, s0); break;
323 case Shader::OPCODE_NOT: not(d, s0); break;
Alexis Hetuc4f2c292015-08-18 15:43:09 -0400324 case Shader::OPCODE_OR: or(d, s0, s1); break;
325 case Shader::OPCODE_XOR: xor(d, s0, s1); break;
326 case Shader::OPCODE_AND: and(d, s0, s1); break;
Alexis Hetu8d78cf72015-08-28 14:24:45 -0400327 case Shader::OPCODE_EQ: equal(d, s0, s1); break;
328 case Shader::OPCODE_NE: notEqual(d, s0, s1); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400329 case Shader::OPCODE_END: break;
330 default:
331 ASSERT(false);
332 }
333
334 if(dst.type != Shader::PARAMETER_VOID && dst.type != Shader::PARAMETER_LABEL && opcode != Shader::OPCODE_TEXKILL && opcode != Shader::OPCODE_NOP)
335 {
336 if(dst.integer)
337 {
338 switch(opcode)
339 {
340 case Shader::OPCODE_DIV:
341 if(dst.x) d.x = Trunc(d.x);
342 if(dst.y) d.y = Trunc(d.y);
343 if(dst.z) d.z = Trunc(d.z);
344 if(dst.w) d.w = Trunc(d.w);
345 break;
346 default:
347 break; // No truncation to integer required when arguments are integer
348 }
349 }
350
351 if(dst.saturate)
352 {
353 if(dst.x) d.x = Max(d.x, Float4(0.0f));
354 if(dst.y) d.y = Max(d.y, Float4(0.0f));
355 if(dst.z) d.z = Max(d.z, Float4(0.0f));
356 if(dst.w) d.w = Max(d.w, Float4(0.0f));
357
358 if(dst.x) d.x = Min(d.x, Float4(1.0f));
359 if(dst.y) d.y = Min(d.y, Float4(1.0f));
360 if(dst.z) d.z = Min(d.z, Float4(1.0f));
361 if(dst.w) d.w = Min(d.w, Float4(1.0f));
362 }
363
364 if(instruction->isPredicated())
365 {
366 Vector4f pDst; // FIXME: Rename
367
368 switch(dst.type)
369 {
370 case Shader::PARAMETER_TEMP:
371 if(dst.rel.type == Shader::PARAMETER_VOID)
372 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500373 if(dst.x) pDst.x = r[dst.index].x;
374 if(dst.y) pDst.y = r[dst.index].y;
375 if(dst.z) pDst.z = r[dst.index].z;
376 if(dst.w) pDst.w = r[dst.index].w;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400377 }
378 else
379 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500380 Int a = relativeAddress(dst);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400381
Nicolas Capens4f172c72016-01-13 08:34:30 -0500382 if(dst.x) pDst.x = r[dst.index + a].x;
383 if(dst.y) pDst.y = r[dst.index + a].y;
384 if(dst.z) pDst.z = r[dst.index + a].z;
385 if(dst.w) pDst.w = r[dst.index + a].w;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400386 }
387 break;
388 case Shader::PARAMETER_COLOROUT:
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500389 if(dst.rel.type == Shader::PARAMETER_VOID)
390 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500391 if(dst.x) pDst.x = oC[dst.index].x;
392 if(dst.y) pDst.y = oC[dst.index].y;
393 if(dst.z) pDst.z = oC[dst.index].z;
394 if(dst.w) pDst.w = oC[dst.index].w;
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500395 }
396 else
397 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500398 Int a = relativeAddress(dst) + dst.index;
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500399
Nicolas Capens4f172c72016-01-13 08:34:30 -0500400 if(dst.x) pDst.x = oC[a].x;
401 if(dst.y) pDst.y = oC[a].y;
402 if(dst.z) pDst.z = oC[a].z;
403 if(dst.w) pDst.w = oC[a].w;
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500404 }
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400405 break;
406 case Shader::PARAMETER_PREDICATE:
Nicolas Capens4f172c72016-01-13 08:34:30 -0500407 if(dst.x) pDst.x = p0.x;
408 if(dst.y) pDst.y = p0.y;
409 if(dst.z) pDst.z = p0.z;
410 if(dst.w) pDst.w = p0.w;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400411 break;
412 case Shader::PARAMETER_DEPTHOUT:
Nicolas Capens4f172c72016-01-13 08:34:30 -0500413 pDst.x = oDepth;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400414 break;
415 default:
416 ASSERT(false);
417 }
418
Nicolas Capens4f172c72016-01-13 08:34:30 -0500419 Int4 enable = enableMask(instruction);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400420
421 Int4 xEnable = enable;
422 Int4 yEnable = enable;
423 Int4 zEnable = enable;
424 Int4 wEnable = enable;
425
426 if(predicate)
427 {
428 unsigned char pSwizzle = instruction->predicateSwizzle;
429
Nicolas Capens4f172c72016-01-13 08:34:30 -0500430 Float4 xPredicate = p0[(pSwizzle >> 0) & 0x03];
431 Float4 yPredicate = p0[(pSwizzle >> 2) & 0x03];
432 Float4 zPredicate = p0[(pSwizzle >> 4) & 0x03];
433 Float4 wPredicate = p0[(pSwizzle >> 6) & 0x03];
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400434
435 if(!instruction->predicateNot)
436 {
437 if(dst.x) xEnable = xEnable & As<Int4>(xPredicate);
438 if(dst.y) yEnable = yEnable & As<Int4>(yPredicate);
439 if(dst.z) zEnable = zEnable & As<Int4>(zPredicate);
440 if(dst.w) wEnable = wEnable & As<Int4>(wPredicate);
441 }
442 else
443 {
444 if(dst.x) xEnable = xEnable & ~As<Int4>(xPredicate);
445 if(dst.y) yEnable = yEnable & ~As<Int4>(yPredicate);
446 if(dst.z) zEnable = zEnable & ~As<Int4>(zPredicate);
447 if(dst.w) wEnable = wEnable & ~As<Int4>(wPredicate);
448 }
449 }
450
451 if(dst.x) d.x = As<Float4>(As<Int4>(d.x) & xEnable);
452 if(dst.y) d.y = As<Float4>(As<Int4>(d.y) & yEnable);
453 if(dst.z) d.z = As<Float4>(As<Int4>(d.z) & zEnable);
454 if(dst.w) d.w = As<Float4>(As<Int4>(d.w) & wEnable);
455
456 if(dst.x) d.x = As<Float4>(As<Int4>(d.x) | (As<Int4>(pDst.x) & ~xEnable));
457 if(dst.y) d.y = As<Float4>(As<Int4>(d.y) | (As<Int4>(pDst.y) & ~yEnable));
458 if(dst.z) d.z = As<Float4>(As<Int4>(d.z) | (As<Int4>(pDst.z) & ~zEnable));
459 if(dst.w) d.w = As<Float4>(As<Int4>(d.w) | (As<Int4>(pDst.w) & ~wEnable));
460 }
461
462 switch(dst.type)
463 {
464 case Shader::PARAMETER_TEMP:
465 if(dst.rel.type == Shader::PARAMETER_VOID)
466 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500467 if(dst.x) r[dst.index].x = d.x;
468 if(dst.y) r[dst.index].y = d.y;
469 if(dst.z) r[dst.index].z = d.z;
470 if(dst.w) r[dst.index].w = d.w;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400471 }
472 else
473 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500474 Int a = relativeAddress(dst);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400475
Nicolas Capens4f172c72016-01-13 08:34:30 -0500476 if(dst.x) r[dst.index + a].x = d.x;
477 if(dst.y) r[dst.index + a].y = d.y;
478 if(dst.z) r[dst.index + a].z = d.z;
479 if(dst.w) r[dst.index + a].w = d.w;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400480 }
481 break;
482 case Shader::PARAMETER_COLOROUT:
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500483 if(dst.rel.type == Shader::PARAMETER_VOID)
484 {
Nicolas Capens8b124c12016-04-18 14:09:37 -0400485 broadcastColor0 = (dst.index == 0) && broadcastColor0;
486
Nicolas Capens4f172c72016-01-13 08:34:30 -0500487 if(dst.x) { oC[dst.index].x = d.x; }
488 if(dst.y) { oC[dst.index].y = d.y; }
489 if(dst.z) { oC[dst.index].z = d.z; }
490 if(dst.w) { oC[dst.index].w = d.w; }
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500491 }
492 else
493 {
Nicolas Capens8b124c12016-04-18 14:09:37 -0400494 broadcastColor0 = false;
Nicolas Capens4f172c72016-01-13 08:34:30 -0500495 Int a = relativeAddress(dst) + dst.index;
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500496
Nicolas Capens4f172c72016-01-13 08:34:30 -0500497 if(dst.x) { oC[a].x = d.x; }
498 if(dst.y) { oC[a].y = d.y; }
499 if(dst.z) { oC[a].z = d.z; }
500 if(dst.w) { oC[a].w = d.w; }
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500501 }
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400502 break;
503 case Shader::PARAMETER_PREDICATE:
Nicolas Capens4f172c72016-01-13 08:34:30 -0500504 if(dst.x) p0.x = d.x;
505 if(dst.y) p0.y = d.y;
506 if(dst.z) p0.z = d.z;
507 if(dst.w) p0.w = d.w;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400508 break;
509 case Shader::PARAMETER_DEPTHOUT:
Nicolas Capens4f172c72016-01-13 08:34:30 -0500510 oDepth = d.x;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400511 break;
512 default:
513 ASSERT(false);
514 }
515 }
516 }
517
518 if(currentLabel != -1)
519 {
520 Nucleus::setInsertBlock(returnBlock);
521 }
522
Nicolas Capens8b124c12016-04-18 14:09:37 -0400523 if(broadcastColor0)
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400524 {
Nicolas Capens8b124c12016-04-18 14:09:37 -0400525 for(int i = 0; i < RENDERTARGETS; i++)
526 {
527 c[i] = oC[0];
528 }
529 }
530 else
531 {
532 for(int i = 0; i < RENDERTARGETS; i++)
533 {
534 c[i] = oC[i];
535 }
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400536 }
537 }
Nicolas Capens5d961882016-01-01 23:18:14 -0500538
Nicolas Capens4f172c72016-01-13 08:34:30 -0500539 Bool PixelProgram::alphaTest(Int cMask[4])
Nicolas Capens5d961882016-01-01 23:18:14 -0500540 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500541 clampColor(c);
Nicolas Capens5d961882016-01-01 23:18:14 -0500542
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400543 if(!state.alphaTestActive())
544 {
545 return true;
546 }
547
548 Int aMask;
549
550 if(state.transparencyAntialiasing == TRANSPARENCY_NONE)
551 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500552 Short4 alpha = RoundShort4(c[0].w * Float4(0x1000));
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400553
Nicolas Capens4f172c72016-01-13 08:34:30 -0500554 PixelRoutine::alphaTest(aMask, alpha);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400555
556 for(unsigned int q = 0; q < state.multiSample; q++)
557 {
558 cMask[q] &= aMask;
559 }
560 }
561 else if(state.transparencyAntialiasing == TRANSPARENCY_ALPHA_TO_COVERAGE)
562 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500563 alphaToCoverage(cMask, c[0].w);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400564 }
565 else ASSERT(false);
566
567 Int pass = cMask[0];
568
569 for(unsigned int q = 1; q < state.multiSample; q++)
570 {
571 pass = pass | cMask[q];
572 }
573
Nicolas Capens5d961882016-01-01 23:18:14 -0500574 return pass != 0x0;
575 }
576
Nicolas Capens4f172c72016-01-13 08:34:30 -0500577 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 -0500578 {
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500579 for(int index = 0; index < RENDERTARGETS; index++)
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400580 {
581 if(!state.colorWriteActive(index))
582 {
583 continue;
Nicolas Capens5d961882016-01-01 23:18:14 -0500584 }
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400585
Alexis Hetu049a1872016-04-25 16:59:58 -0400586 if(!postBlendSRGB && state.writeSRGB && !isSRGB(index))
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400587 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500588 c[index].x = linearToSRGB(c[index].x);
589 c[index].y = linearToSRGB(c[index].y);
590 c[index].z = linearToSRGB(c[index].z);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400591 }
592
593 if(index == 0)
594 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500595 fogBlend(c[index], fog);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400596 }
597
598 switch(state.targetFormat[index])
599 {
600 case FORMAT_R5G6B5:
601 case FORMAT_X8R8G8B8:
602 case FORMAT_X8B8G8R8:
603 case FORMAT_A8R8G8B8:
604 case FORMAT_A8B8G8R8:
Alexis Hetu049a1872016-04-25 16:59:58 -0400605 case FORMAT_SRGB8_X8:
606 case FORMAT_SRGB8_A8:
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400607 case FORMAT_A8:
608 case FORMAT_G16R16:
609 case FORMAT_A16B16G16R16:
610 for(unsigned int q = 0; q < state.multiSample; q++)
611 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500612 Pointer<Byte> buffer = cBuffer[index] + q * *Pointer<Int>(data + OFFSET(DrawData, colorSliceB[index]));
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400613 Vector4s color;
614
Nicolas Capens13c70cd2016-04-05 10:51:45 -0400615 if(state.targetFormat[index] == FORMAT_R5G6B5)
616 {
617 color.x = UShort4(c[index].x * Float4(0xFBFF), false);
618 color.y = UShort4(c[index].y * Float4(0xFDFF), false);
619 color.z = UShort4(c[index].z * Float4(0xFBFF), false);
620 color.w = UShort4(c[index].w * Float4(0xFFFF), false);
621 }
622 else
623 {
624 color.x = convertFixed16(c[index].x, false);
625 color.y = convertFixed16(c[index].y, false);
626 color.z = convertFixed16(c[index].z, false);
627 color.w = convertFixed16(c[index].w, false);
628 }
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400629
630 if(state.multiSampleMask & (1 << q))
631 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500632 alphaBlend(index, buffer, color, x);
633 logicOperation(index, buffer, color, x);
634 writeColor(index, buffer, x, color, sMask[q], zMask[q], cMask[q]);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400635 }
636 }
637 break;
638 case FORMAT_R32F:
639 case FORMAT_G32R32F:
Alexis Hetudbd1a8e2016-04-13 11:40:30 -0400640 case FORMAT_X32B32G32R32F:
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400641 case FORMAT_A32B32G32R32F:
Alexis Hetu1abb6382016-02-08 11:21:16 -0500642 case FORMAT_R32I:
643 case FORMAT_G32R32I:
644 case FORMAT_A32B32G32R32I:
645 case FORMAT_R32UI:
646 case FORMAT_G32R32UI:
647 case FORMAT_A32B32G32R32UI:
Alexis Hetubd7117d2016-06-02 10:35:59 -0400648 case FORMAT_R16I:
649 case FORMAT_G16R16I:
650 case FORMAT_A16B16G16R16I:
651 case FORMAT_R16UI:
652 case FORMAT_G16R16UI:
653 case FORMAT_A16B16G16R16UI:
654 case FORMAT_R8I:
655 case FORMAT_G8R8I:
656 case FORMAT_A8B8G8R8I:
657 case FORMAT_R8UI:
658 case FORMAT_G8R8UI:
659 case FORMAT_A8B8G8R8UI:
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400660 for(unsigned int q = 0; q < state.multiSample; q++)
661 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500662 Pointer<Byte> buffer = cBuffer[index] + q * *Pointer<Int>(data + OFFSET(DrawData, colorSliceB[index]));
663 Vector4f color = c[index];
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400664
665 if(state.multiSampleMask & (1 << q))
666 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500667 alphaBlend(index, buffer, color, x);
668 writeColor(index, buffer, x, color, sMask[q], zMask[q], cMask[q]);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400669 }
670 }
671 break;
672 default:
673 ASSERT(false);
674 }
675 }
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500676 }
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400677
Meng-Lin Wu2337a192016-06-01 13:54:07 -0400678 void PixelProgram::sampleTexture(Vector4f &c, const Src &sampler, Float4 &u, Float4 &v, Float4 &w, Float4 &q, Vector4f &dsx, Vector4f &dsy, Vector4f &offset, bool project, SamplerMethod method, bool hasOffset)
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400679 {
Nicolas Capensc2534f42016-04-04 11:13:24 -0400680 Vector4f tmp;
681
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400682 if(sampler.type == Shader::PARAMETER_SAMPLER && sampler.rel.type == Shader::PARAMETER_VOID)
683 {
Meng-Lin Wu2337a192016-06-01 13:54:07 -0400684 sampleTexture(tmp, sampler.index, u, v, w, q, dsx, dsy, offset, project, method, hasOffset);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400685 }
686 else
687 {
Nicolas Capensc2534f42016-04-04 11:13:24 -0400688 Int index = As<Int>(Float(fetchRegister(sampler).x.x));
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400689
690 for(int i = 0; i < TEXTURE_IMAGE_UNITS; i++)
691 {
692 if(shader->usesSampler(i))
693 {
694 If(index == i)
695 {
Meng-Lin Wu2337a192016-06-01 13:54:07 -0400696 sampleTexture(tmp, i, u, v, w, q, dsx, dsy, offset, project, method, hasOffset);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400697 // FIXME: When the sampler states are the same, we could use one sampler and just index the texture
698 }
699 }
700 }
701 }
Nicolas Capensc2534f42016-04-04 11:13:24 -0400702
703 c.x = tmp[(sampler.swizzle >> 0) & 0x3];
704 c.y = tmp[(sampler.swizzle >> 2) & 0x3];
705 c.z = tmp[(sampler.swizzle >> 4) & 0x3];
706 c.w = tmp[(sampler.swizzle >> 6) & 0x3];
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400707 }
708
Meng-Lin Wu2337a192016-06-01 13:54:07 -0400709 void PixelProgram::sampleTexture(Vector4f &c, int samplerIndex, Float4 &u, Float4 &v, Float4 &w, Float4 &q, Vector4f &dsx, Vector4f &dsy, Vector4f &offset, bool project, SamplerMethod method, bool hasOffset)
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400710 {
Nicolas Capensc2534f42016-04-04 11:13:24 -0400711 #if PERF_PROFILE
712 Long texTime = Ticks();
713 #endif
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400714
Meng-Lin Wu2337a192016-06-01 13:54:07 -0400715 Pointer<Byte> texture = data + OFFSET(DrawData, mipmap) + samplerIndex * sizeof(Texture);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400716
717 if(!project)
718 {
Meng-Lin Wu2337a192016-06-01 13:54:07 -0400719 sampler[samplerIndex]->sampleTexture(texture, c, u, v, w, q, dsx, dsy, offset, method, hasOffset);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400720 }
721 else
722 {
723 Float4 rq = reciprocal(q);
724
725 Float4 u_q = u * rq;
726 Float4 v_q = v * rq;
727 Float4 w_q = w * rq;
728
Meng-Lin Wu2337a192016-06-01 13:54:07 -0400729 sampler[samplerIndex]->sampleTexture(texture, c, u_q, v_q, w_q, q, dsx, dsy, offset, method, hasOffset);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400730 }
731
Nicolas Capensc2534f42016-04-04 11:13:24 -0400732 #if PERF_PROFILE
733 cycles[PERF_TEX] += Ticks() - texTime;
734 #endif
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400735 }
736
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500737 void PixelProgram::clampColor(Vector4f oC[RENDERTARGETS])
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400738 {
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500739 for(int index = 0; index < RENDERTARGETS; index++)
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400740 {
741 if(!state.colorWriteActive(index) && !(index == 0 && state.alphaTestActive()))
742 {
743 continue;
744 }
745
746 switch(state.targetFormat[index])
747 {
748 case FORMAT_NULL:
749 break;
750 case FORMAT_R5G6B5:
751 case FORMAT_A8R8G8B8:
752 case FORMAT_A8B8G8R8:
753 case FORMAT_X8R8G8B8:
754 case FORMAT_X8B8G8R8:
Alexis Hetu049a1872016-04-25 16:59:58 -0400755 case FORMAT_SRGB8_X8:
756 case FORMAT_SRGB8_A8:
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400757 case FORMAT_A8:
758 case FORMAT_G16R16:
759 case FORMAT_A16B16G16R16:
760 oC[index].x = Max(oC[index].x, Float4(0.0f)); oC[index].x = Min(oC[index].x, Float4(1.0f));
761 oC[index].y = Max(oC[index].y, Float4(0.0f)); oC[index].y = Min(oC[index].y, Float4(1.0f));
762 oC[index].z = Max(oC[index].z, Float4(0.0f)); oC[index].z = Min(oC[index].z, Float4(1.0f));
763 oC[index].w = Max(oC[index].w, Float4(0.0f)); oC[index].w = Min(oC[index].w, Float4(1.0f));
764 break;
765 case FORMAT_R32F:
766 case FORMAT_G32R32F:
Alexis Hetudbd1a8e2016-04-13 11:40:30 -0400767 case FORMAT_X32B32G32R32F:
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400768 case FORMAT_A32B32G32R32F:
Alexis Hetu1abb6382016-02-08 11:21:16 -0500769 case FORMAT_R32I:
770 case FORMAT_G32R32I:
771 case FORMAT_A32B32G32R32I:
772 case FORMAT_R32UI:
773 case FORMAT_G32R32UI:
774 case FORMAT_A32B32G32R32UI:
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400775 break;
776 default:
777 ASSERT(false);
778 }
779 }
780 }
781
Nicolas Capens4f172c72016-01-13 08:34:30 -0500782 Int4 PixelProgram::enableMask(const Shader::Instruction *instruction)
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400783 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500784 Int4 enable = instruction->analysisBranch ? Int4(enableStack[enableIndex]) : Int4(0xFFFFFFFF);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400785
786 if(!whileTest)
787 {
788 if(shader->containsBreakInstruction() && instruction->analysisBreak)
789 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500790 enable &= enableBreak;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400791 }
792
793 if(shader->containsContinueInstruction() && instruction->analysisContinue)
794 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500795 enable &= enableContinue;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400796 }
797
798 if(shader->containsLeaveInstruction() && instruction->analysisLeave)
799 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500800 enable &= enableLeave;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400801 }
802 }
803
804 return enable;
805 }
806
Nicolas Capensc2534f42016-04-04 11:13:24 -0400807 Vector4f PixelProgram::fetchRegister(const Src &src, unsigned int offset)
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400808 {
809 Vector4f reg;
Nicolas Capens5d961882016-01-01 23:18:14 -0500810 unsigned int i = src.index + offset;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400811
812 switch(src.type)
813 {
814 case Shader::PARAMETER_TEMP:
815 if(src.rel.type == Shader::PARAMETER_VOID)
816 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500817 reg = r[i];
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400818 }
819 else
820 {
Alexis Hetu2c2a7b22015-10-27 16:12:11 -0400821 Int a = relativeAddress(src, src.bufferIndex);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400822
Nicolas Capens4f172c72016-01-13 08:34:30 -0500823 reg = r[i + a];
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400824 }
825 break;
826 case Shader::PARAMETER_INPUT:
827 {
828 if(src.rel.type == Shader::PARAMETER_VOID) // Not relative
829 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500830 reg = v[i];
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400831 }
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400832 else
833 {
Alexis Hetu2c2a7b22015-10-27 16:12:11 -0400834 Int a = relativeAddress(src, src.bufferIndex);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400835
Nicolas Capens4f172c72016-01-13 08:34:30 -0500836 reg = v[i + a];
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400837 }
838 }
839 break;
840 case Shader::PARAMETER_CONST:
Nicolas Capens4f172c72016-01-13 08:34:30 -0500841 reg = readConstant(src, offset);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400842 break;
843 case Shader::PARAMETER_TEXTURE:
Nicolas Capens4f172c72016-01-13 08:34:30 -0500844 reg = v[2 + i];
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400845 break;
846 case Shader::PARAMETER_MISCTYPE:
Nicolas Capens4f172c72016-01-13 08:34:30 -0500847 if(src.index == 0) reg = vPos;
848 if(src.index == 1) reg = vFace;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400849 break;
850 case Shader::PARAMETER_SAMPLER:
851 if(src.rel.type == Shader::PARAMETER_VOID)
852 {
853 reg.x = As<Float4>(Int4(i));
854 }
855 else if(src.rel.type == Shader::PARAMETER_TEMP)
856 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500857 reg.x = As<Float4>(Int4(i) + As<Int4>(r[src.rel.index].x));
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400858 }
859 return reg;
860 case Shader::PARAMETER_PREDICATE: return reg; // Dummy
861 case Shader::PARAMETER_VOID: return reg; // Dummy
862 case Shader::PARAMETER_FLOAT4LITERAL:
863 reg.x = Float4(src.value[0]);
864 reg.y = Float4(src.value[1]);
865 reg.z = Float4(src.value[2]);
866 reg.w = Float4(src.value[3]);
867 break;
868 case Shader::PARAMETER_CONSTINT: return reg; // Dummy
869 case Shader::PARAMETER_CONSTBOOL: return reg; // Dummy
870 case Shader::PARAMETER_LOOP: return reg; // Dummy
871 case Shader::PARAMETER_COLOROUT:
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500872 if(src.rel.type == Shader::PARAMETER_VOID) // Not relative
873 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500874 reg = oC[i];
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500875 }
876 else
877 {
Alexis Hetu2c2a7b22015-10-27 16:12:11 -0400878 Int a = relativeAddress(src, src.bufferIndex);
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500879
Nicolas Capens4f172c72016-01-13 08:34:30 -0500880 reg = oC[i + a];
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500881 }
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400882 break;
883 case Shader::PARAMETER_DEPTHOUT:
Nicolas Capens4f172c72016-01-13 08:34:30 -0500884 reg.x = oDepth;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400885 break;
886 default:
887 ASSERT(false);
888 }
889
890 const Float4 &x = reg[(src.swizzle >> 0) & 0x3];
891 const Float4 &y = reg[(src.swizzle >> 2) & 0x3];
892 const Float4 &z = reg[(src.swizzle >> 4) & 0x3];
893 const Float4 &w = reg[(src.swizzle >> 6) & 0x3];
894
895 Vector4f mod;
896
897 switch(src.modifier)
898 {
899 case Shader::MODIFIER_NONE:
900 mod.x = x;
901 mod.y = y;
902 mod.z = z;
903 mod.w = w;
904 break;
905 case Shader::MODIFIER_NEGATE:
906 mod.x = -x;
907 mod.y = -y;
908 mod.z = -z;
909 mod.w = -w;
910 break;
911 case Shader::MODIFIER_ABS:
912 mod.x = Abs(x);
913 mod.y = Abs(y);
914 mod.z = Abs(z);
915 mod.w = Abs(w);
916 break;
917 case Shader::MODIFIER_ABS_NEGATE:
918 mod.x = -Abs(x);
919 mod.y = -Abs(y);
920 mod.z = -Abs(z);
921 mod.w = -Abs(w);
922 break;
923 case Shader::MODIFIER_NOT:
924 mod.x = As<Float4>(As<Int4>(x) ^ Int4(0xFFFFFFFF));
925 mod.y = As<Float4>(As<Int4>(y) ^ Int4(0xFFFFFFFF));
926 mod.z = As<Float4>(As<Int4>(z) ^ Int4(0xFFFFFFFF));
927 mod.w = As<Float4>(As<Int4>(w) ^ Int4(0xFFFFFFFF));
928 break;
929 default:
930 ASSERT(false);
931 }
932
933 return mod;
934 }
935
Alexis Hetu2c2a7b22015-10-27 16:12:11 -0400936 RValue<Pointer<Byte>> PixelProgram::uniformAddress(int bufferIndex, unsigned int index)
937 {
938 if(bufferIndex == -1)
939 {
940 return data + OFFSET(DrawData, ps.c[index]);
941 }
942 else
943 {
944 return *Pointer<Pointer<Byte>>(data + OFFSET(DrawData, ps.u[bufferIndex])) + index;
945 }
946 }
947
948 RValue<Pointer<Byte>> PixelProgram::uniformAddress(int bufferIndex, unsigned int index, Int& offset)
949 {
950 return uniformAddress(bufferIndex, index) + offset * sizeof(float4);
951 }
952
Nicolas Capens4f172c72016-01-13 08:34:30 -0500953 Vector4f PixelProgram::readConstant(const Src &src, unsigned int offset)
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400954 {
955 Vector4f c;
Nicolas Capens5d961882016-01-01 23:18:14 -0500956 unsigned int i = src.index + offset;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400957
958 if(src.rel.type == Shader::PARAMETER_VOID) // Not relative
959 {
Alexis Hetu2c2a7b22015-10-27 16:12:11 -0400960 c.x = c.y = c.z = c.w = *Pointer<Float4>(uniformAddress(src.bufferIndex, i));
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400961
962 c.x = c.x.xxxx;
963 c.y = c.y.yyyy;
964 c.z = c.z.zzzz;
965 c.w = c.w.wwww;
966
967 if(shader->containsDefineInstruction()) // Constant may be known at compile time
968 {
969 for(size_t j = 0; j < shader->getLength(); j++)
970 {
971 const Shader::Instruction &instruction = *shader->getInstruction(j);
972
973 if(instruction.opcode == Shader::OPCODE_DEF)
974 {
975 if(instruction.dst.index == i)
976 {
977 c.x = Float4(instruction.src[0].value[0]);
978 c.y = Float4(instruction.src[0].value[1]);
979 c.z = Float4(instruction.src[0].value[2]);
980 c.w = Float4(instruction.src[0].value[3]);
981
982 break;
983 }
984 }
985 }
986 }
987 }
988 else if(src.rel.type == Shader::PARAMETER_LOOP)
989 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500990 Int loopCounter = aL[loopDepth];
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400991
Alexis Hetu2c2a7b22015-10-27 16:12:11 -0400992 c.x = c.y = c.z = c.w = *Pointer<Float4>(uniformAddress(src.bufferIndex, i, loopCounter));
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400993
994 c.x = c.x.xxxx;
995 c.y = c.y.yyyy;
996 c.z = c.z.zzzz;
997 c.w = c.w.wwww;
998 }
999 else
1000 {
Alexis Hetu2c2a7b22015-10-27 16:12:11 -04001001 Int a = relativeAddress(src, src.bufferIndex);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001002
Alexis Hetu2c2a7b22015-10-27 16:12:11 -04001003 c.x = c.y = c.z = c.w = *Pointer<Float4>(uniformAddress(src.bufferIndex, i, a));
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001004
1005 c.x = c.x.xxxx;
1006 c.y = c.y.yyyy;
1007 c.z = c.z.zzzz;
1008 c.w = c.w.wwww;
1009 }
1010
1011 return c;
1012 }
1013
Alexis Hetu2c2a7b22015-10-27 16:12:11 -04001014 Int PixelProgram::relativeAddress(const Shader::Parameter &var, int bufferIndex)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001015 {
1016 ASSERT(var.rel.deterministic);
1017
1018 if(var.rel.type == Shader::PARAMETER_TEMP)
1019 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001020 return As<Int>(Extract(r[var.rel.index].x, 0)) * var.rel.scale;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001021 }
1022 else if(var.rel.type == Shader::PARAMETER_INPUT)
1023 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001024 return As<Int>(Extract(v[var.rel.index].x, 0)) * var.rel.scale;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001025 }
1026 else if(var.rel.type == Shader::PARAMETER_OUTPUT)
1027 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001028 return As<Int>(Extract(oC[var.rel.index].x, 0)) * var.rel.scale;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001029 }
1030 else if(var.rel.type == Shader::PARAMETER_CONST)
1031 {
Alexis Hetu48be7352016-02-10 14:32:34 -05001032 return *Pointer<Int>(uniformAddress(bufferIndex, var.rel.index)) * var.rel.scale;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001033 }
Nicolas Capens907700d2016-01-20 17:09:28 -05001034 else if(var.rel.type == Shader::PARAMETER_LOOP)
1035 {
1036 return aL[loopDepth];
1037 }
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001038 else ASSERT(false);
1039
1040 return 0;
1041 }
1042
1043 Float4 PixelProgram::linearToSRGB(const Float4 &x) // Approximates x^(1.0/2.2)
1044 {
1045 Float4 sqrtx = Rcp_pp(RcpSqrt_pp(x));
1046 Float4 sRGB = sqrtx * Float4(1.14f) - x * Float4(0.14f);
1047
1048 return Min(Max(sRGB, Float4(0.0f)), Float4(1.0f));
1049 }
1050
Nicolas Capens4f172c72016-01-13 08:34:30 -05001051 void PixelProgram::M3X2(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);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001055
1056 dst.x = dot3(src0, row0);
1057 dst.y = dot3(src0, row1);
1058 }
1059
Nicolas Capens4f172c72016-01-13 08:34:30 -05001060 void PixelProgram::M3X3(Vector4f &dst, Vector4f &src0, const Src &src1)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001061 {
Nicolas Capensc2534f42016-04-04 11:13:24 -04001062 Vector4f row0 = fetchRegister(src1, 0);
1063 Vector4f row1 = fetchRegister(src1, 1);
1064 Vector4f row2 = fetchRegister(src1, 2);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001065
1066 dst.x = dot3(src0, row0);
1067 dst.y = dot3(src0, row1);
1068 dst.z = dot3(src0, row2);
1069 }
1070
Nicolas Capens4f172c72016-01-13 08:34:30 -05001071 void PixelProgram::M3X4(Vector4f &dst, Vector4f &src0, const Src &src1)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001072 {
Nicolas Capensc2534f42016-04-04 11:13:24 -04001073 Vector4f row0 = fetchRegister(src1, 0);
1074 Vector4f row1 = fetchRegister(src1, 1);
1075 Vector4f row2 = fetchRegister(src1, 2);
1076 Vector4f row3 = fetchRegister(src1, 3);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001077
1078 dst.x = dot3(src0, row0);
1079 dst.y = dot3(src0, row1);
1080 dst.z = dot3(src0, row2);
1081 dst.w = dot3(src0, row3);
1082 }
1083
Nicolas Capens4f172c72016-01-13 08:34:30 -05001084 void PixelProgram::M4X3(Vector4f &dst, Vector4f &src0, const Src &src1)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001085 {
Nicolas Capensc2534f42016-04-04 11:13:24 -04001086 Vector4f row0 = fetchRegister(src1, 0);
1087 Vector4f row1 = fetchRegister(src1, 1);
1088 Vector4f row2 = fetchRegister(src1, 2);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001089
1090 dst.x = dot4(src0, row0);
1091 dst.y = dot4(src0, row1);
1092 dst.z = dot4(src0, row2);
1093 }
1094
Nicolas Capens4f172c72016-01-13 08:34:30 -05001095 void PixelProgram::M4X4(Vector4f &dst, Vector4f &src0, const Src &src1)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001096 {
Nicolas Capensc2534f42016-04-04 11:13:24 -04001097 Vector4f row0 = fetchRegister(src1, 0);
1098 Vector4f row1 = fetchRegister(src1, 1);
1099 Vector4f row2 = fetchRegister(src1, 2);
1100 Vector4f row3 = fetchRegister(src1, 3);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001101
1102 dst.x = dot4(src0, row0);
1103 dst.y = dot4(src0, row1);
1104 dst.z = dot4(src0, row2);
1105 dst.w = dot4(src0, row3);
1106 }
1107
Nicolas Capens4f172c72016-01-13 08:34:30 -05001108 void PixelProgram::TEXLD(Vector4f &dst, Vector4f &src0, const Src &src1, bool project, bool bias)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001109 {
Meng-Lin Wu2337a192016-06-01 13:54:07 -04001110 sampleTexture(dst, src1, src0.x, src0.y, src0.z, src0.w, src0, src0, src0, project, bias ? Bias : Implicit, false);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001111 }
1112
Meng-Lin Wu2337a192016-06-01 13:54:07 -04001113 void PixelProgram::TEXOFFSET(Vector4f &dst, Vector4f &src0, const Src &src1, Vector4f &src2, bool project, bool bias)
Alexis Hetu25d47fc2015-10-22 13:58:32 -04001114 {
Meng-Lin Wu2337a192016-06-01 13:54:07 -04001115 sampleTexture(dst, src1, src0.x, src0.y, src0.z, src0.w, src0, src0, src2, project, bias ? Bias : Implicit, true);
Alexis Hetu25d47fc2015-10-22 13:58:32 -04001116 }
1117
Nicolas Capens4f172c72016-01-13 08:34:30 -05001118 void PixelProgram::TEXLDL(Vector4f &dst, Vector4f &src0, const Src &src1, Vector4f &offset, bool project, bool bias)
Alexis Hetu25d47fc2015-10-22 13:58:32 -04001119 {
Meng-Lin Wu2337a192016-06-01 13:54:07 -04001120 sampleTexture(dst, src1, src0.x, src0.y, src0.z, src0.w, src0, src0, offset, project, Lod, true);
Alexis Hetu25d47fc2015-10-22 13:58:32 -04001121 }
1122
Nicolas Capens4f172c72016-01-13 08:34:30 -05001123 void PixelProgram::TEXELFETCH(Vector4f &dst, Vector4f &src0, const Src& src1, Vector4f &src2)
Alexis Hetu25d47fc2015-10-22 13:58:32 -04001124 {
1125 UNIMPLEMENTED();
1126 }
1127
Nicolas Capens4f172c72016-01-13 08:34:30 -05001128 void PixelProgram::TEXELFETCH(Vector4f &dst, Vector4f &src0, const Src& src1, Vector4f &src2, Vector4f &offset)
Alexis Hetu25d47fc2015-10-22 13:58:32 -04001129 {
1130 UNIMPLEMENTED();
1131 }
1132
Nicolas Capens4f172c72016-01-13 08:34:30 -05001133 void PixelProgram::TEXGRAD(Vector4f &dst, Vector4f &src0, const Src& src1, Vector4f &src2, Vector4f &src3)
Alexis Hetu25d47fc2015-10-22 13:58:32 -04001134 {
Meng-Lin Wu2337a192016-06-01 13:54:07 -04001135 sampleTexture(dst, src1, src0.x, src0.y, src0.z, src0.w, src2, src3, src0, false, Grad, false);
Alexis Hetu25d47fc2015-10-22 13:58:32 -04001136 }
1137
Nicolas Capens4f172c72016-01-13 08:34:30 -05001138 void PixelProgram::TEXGRAD(Vector4f &dst, Vector4f &src0, const Src& src1, Vector4f &src2, Vector4f &src3, Vector4f &offset)
Alexis Hetu25d47fc2015-10-22 13:58:32 -04001139 {
Meng-Lin Wu2337a192016-06-01 13:54:07 -04001140 sampleTexture(dst, src1, src0.x, src0.y, src0.z, src0.w, src2, src3, offset, false, Grad, true);
Alexis Hetu25d47fc2015-10-22 13:58:32 -04001141 }
1142
Nicolas Capensc2534f42016-04-04 11:13:24 -04001143 void PixelProgram::TEXLDD(Vector4f &dst, Vector4f &src0, const Src &src1, Vector4f &src2, Vector4f &src3, bool project)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001144 {
Meng-Lin Wu2337a192016-06-01 13:54:07 -04001145 sampleTexture(dst, src1, src0.x, src0.y, src0.z, src0.w, src2, src3, src0, project, Grad, false);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001146 }
1147
Nicolas Capensc2534f42016-04-04 11:13:24 -04001148 void PixelProgram::TEXLDL(Vector4f &dst, Vector4f &src0, const Src &src1, bool project)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001149 {
Meng-Lin Wu2337a192016-06-01 13:54:07 -04001150 sampleTexture(dst, src1, src0.x, src0.y, src0.z, src0.w, src0, src0, src0, project, Lod, false);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001151 }
1152
Nicolas Capens4f172c72016-01-13 08:34:30 -05001153 void PixelProgram::TEXSIZE(Vector4f &dst, Float4 &lod, const Src &src1)
Alexis Hetu9bcb31d2015-07-22 17:03:26 -04001154 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001155 Pointer<Byte> textureMipmap = data + OFFSET(DrawData, mipmap) + src1.index * sizeof(Texture) + OFFSET(Texture, mipmap);
Alexis Hetu9bcb31d2015-07-22 17:03:26 -04001156 for(int i = 0; i < 4; ++i)
1157 {
1158 Pointer<Byte> mipmap = textureMipmap + (As<Int>(Extract(lod, i)) + Int(1)) * sizeof(Mipmap);
1159 dst.x = Insert(dst.x, As<Float>(Int(*Pointer<Short>(mipmap + OFFSET(Mipmap, width)))), i);
1160 dst.y = Insert(dst.y, As<Float>(Int(*Pointer<Short>(mipmap + OFFSET(Mipmap, height)))), i);
1161 dst.z = Insert(dst.z, As<Float>(Int(*Pointer<Short>(mipmap + OFFSET(Mipmap, depth)))), i);
1162 }
1163 }
1164
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001165 void PixelProgram::TEXKILL(Int cMask[4], Vector4f &src, unsigned char mask)
1166 {
1167 Int kill = -1;
1168
1169 if(mask & 0x1) kill &= SignMask(CmpNLT(src.x, Float4(0.0f)));
1170 if(mask & 0x2) kill &= SignMask(CmpNLT(src.y, Float4(0.0f)));
1171 if(mask & 0x4) kill &= SignMask(CmpNLT(src.z, Float4(0.0f)));
1172 if(mask & 0x8) kill &= SignMask(CmpNLT(src.w, Float4(0.0f)));
1173
1174 // FIXME: Dynamic branching affects TEXKILL?
1175 // if(shader->containsDynamicBranching())
1176 // {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001177 // kill = ~SignMask(enableMask());
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
Nicolas Capens4f172c72016-01-13 08:34:30 -05001188 void PixelProgram::DISCARD(Int cMask[4], const Shader::Instruction *instruction)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001189 {
1190 Int kill = 0;
1191
1192 if(shader->containsDynamicBranching())
1193 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001194 kill = ~SignMask(enableMask(instruction));
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001195 }
1196
1197 for(unsigned int q = 0; q < state.multiSample; q++)
1198 {
1199 cMask[q] &= kill;
1200 }
1201
1202 // FIXME: Branch to end of shader if all killed?
1203 }
1204
1205 void PixelProgram::DFDX(Vector4f &dst, Vector4f &src)
1206 {
1207 dst.x = src.x.yyww - src.x.xxzz;
1208 dst.y = src.y.yyww - src.y.xxzz;
1209 dst.z = src.z.yyww - src.z.xxzz;
1210 dst.w = src.w.yyww - src.w.xxzz;
1211 }
1212
1213 void PixelProgram::DFDY(Vector4f &dst, Vector4f &src)
1214 {
1215 dst.x = src.x.zwzw - src.x.xyxy;
1216 dst.y = src.y.zwzw - src.y.xyxy;
1217 dst.z = src.z.zwzw - src.z.xyxy;
1218 dst.w = src.w.zwzw - src.w.xyxy;
1219 }
1220
1221 void PixelProgram::FWIDTH(Vector4f &dst, Vector4f &src)
1222 {
1223 // abs(dFdx(src)) + abs(dFdy(src));
1224 dst.x = Abs(src.x.yyww - src.x.xxzz) + Abs(src.x.zwzw - src.x.xyxy);
1225 dst.y = Abs(src.y.yyww - src.y.xxzz) + Abs(src.y.zwzw - src.y.xyxy);
1226 dst.z = Abs(src.z.yyww - src.z.xxzz) + Abs(src.z.zwzw - src.z.xyxy);
1227 dst.w = Abs(src.w.yyww - src.w.xxzz) + Abs(src.w.zwzw - src.w.xyxy);
1228 }
1229
Nicolas Capens4f172c72016-01-13 08:34:30 -05001230 void PixelProgram::BREAK()
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001231 {
1232 llvm::BasicBlock *deadBlock = Nucleus::createBasicBlock();
1233 llvm::BasicBlock *endBlock = loopRepEndBlock[loopRepDepth - 1];
1234
1235 if(breakDepth == 0)
1236 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001237 enableIndex = enableIndex - breakDepth;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001238 Nucleus::createBr(endBlock);
1239 }
1240 else
1241 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001242 enableBreak = enableBreak & ~enableStack[enableIndex];
1243 Bool allBreak = SignMask(enableBreak) == 0x0;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001244
Nicolas Capens4f172c72016-01-13 08:34:30 -05001245 enableIndex = enableIndex - breakDepth;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001246 branch(allBreak, endBlock, deadBlock);
1247 }
1248
1249 Nucleus::setInsertBlock(deadBlock);
Nicolas Capens4f172c72016-01-13 08:34:30 -05001250 enableIndex = enableIndex + breakDepth;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001251 }
1252
Nicolas Capens4f172c72016-01-13 08:34:30 -05001253 void PixelProgram::BREAKC(Vector4f &src0, Vector4f &src1, Control control)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001254 {
1255 Int4 condition;
1256
1257 switch(control)
1258 {
1259 case Shader::CONTROL_GT: condition = CmpNLE(src0.x, src1.x); break;
1260 case Shader::CONTROL_EQ: condition = CmpEQ(src0.x, src1.x); break;
1261 case Shader::CONTROL_GE: condition = CmpNLT(src0.x, src1.x); break;
1262 case Shader::CONTROL_LT: condition = CmpLT(src0.x, src1.x); break;
1263 case Shader::CONTROL_NE: condition = CmpNEQ(src0.x, src1.x); break;
1264 case Shader::CONTROL_LE: condition = CmpLE(src0.x, src1.x); break;
1265 default:
1266 ASSERT(false);
1267 }
1268
Nicolas Capens4f172c72016-01-13 08:34:30 -05001269 BREAK(condition);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001270 }
1271
Nicolas Capens4f172c72016-01-13 08:34:30 -05001272 void PixelProgram::BREAKP(const Src &predicateRegister) // FIXME: Factor out parts common with BREAKC
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001273 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001274 Int4 condition = As<Int4>(p0[predicateRegister.swizzle & 0x3]);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001275
1276 if(predicateRegister.modifier == Shader::MODIFIER_NOT)
1277 {
1278 condition = ~condition;
1279 }
1280
Nicolas Capens4f172c72016-01-13 08:34:30 -05001281 BREAK(condition);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001282 }
1283
Nicolas Capens4f172c72016-01-13 08:34:30 -05001284 void PixelProgram::BREAK(Int4 &condition)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001285 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001286 condition &= enableStack[enableIndex];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001287
1288 llvm::BasicBlock *continueBlock = Nucleus::createBasicBlock();
1289 llvm::BasicBlock *endBlock = loopRepEndBlock[loopRepDepth - 1];
1290
Nicolas Capens4f172c72016-01-13 08:34:30 -05001291 enableBreak = enableBreak & ~condition;
1292 Bool allBreak = SignMask(enableBreak) == 0x0;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001293
Nicolas Capens4f172c72016-01-13 08:34:30 -05001294 enableIndex = enableIndex - breakDepth;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001295 branch(allBreak, endBlock, continueBlock);
1296
1297 Nucleus::setInsertBlock(continueBlock);
Nicolas Capens4f172c72016-01-13 08:34:30 -05001298 enableIndex = enableIndex + breakDepth;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001299 }
1300
Nicolas Capens4f172c72016-01-13 08:34:30 -05001301 void PixelProgram::CONTINUE()
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001302 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001303 enableContinue = enableContinue & ~enableStack[enableIndex];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001304 }
1305
1306 void PixelProgram::TEST()
1307 {
1308 whileTest = true;
1309 }
1310
Nicolas Capens4f172c72016-01-13 08:34:30 -05001311 void PixelProgram::CALL(int labelIndex, int callSiteIndex)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001312 {
1313 if(!labelBlock[labelIndex])
1314 {
1315 labelBlock[labelIndex] = Nucleus::createBasicBlock();
1316 }
1317
1318 if(callRetBlock[labelIndex].size() > 1)
1319 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001320 callStack[stackIndex++] = UInt(callSiteIndex);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001321 }
1322
Nicolas Capens4f172c72016-01-13 08:34:30 -05001323 Int4 restoreLeave = enableLeave;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001324
1325 Nucleus::createBr(labelBlock[labelIndex]);
1326 Nucleus::setInsertBlock(callRetBlock[labelIndex][callSiteIndex]);
1327
Nicolas Capens4f172c72016-01-13 08:34:30 -05001328 enableLeave = restoreLeave;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001329 }
1330
Nicolas Capens4f172c72016-01-13 08:34:30 -05001331 void PixelProgram::CALLNZ(int labelIndex, int callSiteIndex, const Src &src)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001332 {
1333 if(src.type == Shader::PARAMETER_CONSTBOOL)
1334 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001335 CALLNZb(labelIndex, callSiteIndex, src);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001336 }
1337 else if(src.type == Shader::PARAMETER_PREDICATE)
1338 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001339 CALLNZp(labelIndex, callSiteIndex, src);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001340 }
1341 else ASSERT(false);
1342 }
1343
Nicolas Capens4f172c72016-01-13 08:34:30 -05001344 void PixelProgram::CALLNZb(int labelIndex, int callSiteIndex, const Src &boolRegister)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001345 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001346 Bool condition = (*Pointer<Byte>(data + OFFSET(DrawData, ps.b[boolRegister.index])) != Byte(0)); // FIXME
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001347
1348 if(boolRegister.modifier == Shader::MODIFIER_NOT)
1349 {
1350 condition = !condition;
1351 }
1352
1353 if(!labelBlock[labelIndex])
1354 {
1355 labelBlock[labelIndex] = Nucleus::createBasicBlock();
1356 }
1357
1358 if(callRetBlock[labelIndex].size() > 1)
1359 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001360 callStack[stackIndex++] = UInt(callSiteIndex);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001361 }
1362
Nicolas Capens4f172c72016-01-13 08:34:30 -05001363 Int4 restoreLeave = enableLeave;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001364
1365 branch(condition, labelBlock[labelIndex], callRetBlock[labelIndex][callSiteIndex]);
1366 Nucleus::setInsertBlock(callRetBlock[labelIndex][callSiteIndex]);
1367
Nicolas Capens4f172c72016-01-13 08:34:30 -05001368 enableLeave = restoreLeave;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001369 }
1370
Nicolas Capens4f172c72016-01-13 08:34:30 -05001371 void PixelProgram::CALLNZp(int labelIndex, int callSiteIndex, const Src &predicateRegister)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001372 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001373 Int4 condition = As<Int4>(p0[predicateRegister.swizzle & 0x3]);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001374
1375 if(predicateRegister.modifier == Shader::MODIFIER_NOT)
1376 {
1377 condition = ~condition;
1378 }
1379
Nicolas Capens4f172c72016-01-13 08:34:30 -05001380 condition &= enableStack[enableIndex];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001381
1382 if(!labelBlock[labelIndex])
1383 {
1384 labelBlock[labelIndex] = Nucleus::createBasicBlock();
1385 }
1386
1387 if(callRetBlock[labelIndex].size() > 1)
1388 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001389 callStack[stackIndex++] = UInt(callSiteIndex);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001390 }
1391
Nicolas Capens4f172c72016-01-13 08:34:30 -05001392 enableIndex++;
1393 enableStack[enableIndex] = condition;
1394 Int4 restoreLeave = enableLeave;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001395
1396 Bool notAllFalse = SignMask(condition) != 0;
1397 branch(notAllFalse, labelBlock[labelIndex], callRetBlock[labelIndex][callSiteIndex]);
1398 Nucleus::setInsertBlock(callRetBlock[labelIndex][callSiteIndex]);
1399
Nicolas Capens4f172c72016-01-13 08:34:30 -05001400 enableIndex--;
1401 enableLeave = restoreLeave;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001402 }
1403
Nicolas Capens4f172c72016-01-13 08:34:30 -05001404 void PixelProgram::ELSE()
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001405 {
1406 ifDepth--;
1407
1408 llvm::BasicBlock *falseBlock = ifFalseBlock[ifDepth];
1409 llvm::BasicBlock *endBlock = Nucleus::createBasicBlock();
1410
1411 if(isConditionalIf[ifDepth])
1412 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001413 Int4 condition = ~enableStack[enableIndex] & enableStack[enableIndex - 1];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001414 Bool notAllFalse = SignMask(condition) != 0;
1415
1416 branch(notAllFalse, falseBlock, endBlock);
1417
Nicolas Capens4f172c72016-01-13 08:34:30 -05001418 enableStack[enableIndex] = ~enableStack[enableIndex] & enableStack[enableIndex - 1];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001419 }
1420 else
1421 {
1422 Nucleus::createBr(endBlock);
1423 Nucleus::setInsertBlock(falseBlock);
1424 }
1425
1426 ifFalseBlock[ifDepth] = endBlock;
1427
1428 ifDepth++;
1429 }
1430
Nicolas Capens4f172c72016-01-13 08:34:30 -05001431 void PixelProgram::ENDIF()
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001432 {
1433 ifDepth--;
1434
1435 llvm::BasicBlock *endBlock = ifFalseBlock[ifDepth];
1436
1437 Nucleus::createBr(endBlock);
1438 Nucleus::setInsertBlock(endBlock);
1439
1440 if(isConditionalIf[ifDepth])
1441 {
1442 breakDepth--;
Nicolas Capens4f172c72016-01-13 08:34:30 -05001443 enableIndex--;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001444 }
1445 }
1446
Nicolas Capens4f172c72016-01-13 08:34:30 -05001447 void PixelProgram::ENDLOOP()
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001448 {
1449 loopRepDepth--;
1450
Nicolas Capens4f172c72016-01-13 08:34:30 -05001451 aL[loopDepth] = aL[loopDepth] + increment[loopDepth]; // FIXME: +=
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001452
1453 llvm::BasicBlock *testBlock = loopRepTestBlock[loopRepDepth];
1454 llvm::BasicBlock *endBlock = loopRepEndBlock[loopRepDepth];
1455
1456 Nucleus::createBr(testBlock);
1457 Nucleus::setInsertBlock(endBlock);
1458
Nicolas Capens4f172c72016-01-13 08:34:30 -05001459 loopDepth--;
1460 enableBreak = Int4(0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001461 }
1462
Nicolas Capens4f172c72016-01-13 08:34:30 -05001463 void PixelProgram::ENDREP()
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001464 {
1465 loopRepDepth--;
1466
1467 llvm::BasicBlock *testBlock = loopRepTestBlock[loopRepDepth];
1468 llvm::BasicBlock *endBlock = loopRepEndBlock[loopRepDepth];
1469
1470 Nucleus::createBr(testBlock);
1471 Nucleus::setInsertBlock(endBlock);
1472
Nicolas Capens4f172c72016-01-13 08:34:30 -05001473 loopDepth--;
1474 enableBreak = Int4(0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001475 }
1476
Nicolas Capens4f172c72016-01-13 08:34:30 -05001477 void PixelProgram::ENDWHILE()
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001478 {
1479 loopRepDepth--;
1480
1481 llvm::BasicBlock *testBlock = loopRepTestBlock[loopRepDepth];
1482 llvm::BasicBlock *endBlock = loopRepEndBlock[loopRepDepth];
1483
1484 Nucleus::createBr(testBlock);
1485 Nucleus::setInsertBlock(endBlock);
1486
Nicolas Capens4f172c72016-01-13 08:34:30 -05001487 enableIndex--;
1488 enableBreak = Int4(0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001489 whileTest = false;
1490 }
1491
Alexis Hetu9aa83a92016-05-02 17:34:46 -04001492 void PixelProgram::ENDSWITCH()
1493 {
1494 loopRepDepth--;
1495
1496 llvm::BasicBlock *endBlock = loopRepEndBlock[loopRepDepth];
1497
1498 Nucleus::createBr(loopRepEndBlock[loopRepDepth]);
1499 Nucleus::setInsertBlock(endBlock);
1500
1501 enableIndex--;
1502 enableBreak = Int4(0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF);
1503 }
1504
Nicolas Capens4f172c72016-01-13 08:34:30 -05001505 void PixelProgram::IF(const Src &src)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001506 {
1507 if(src.type == Shader::PARAMETER_CONSTBOOL)
1508 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001509 IFb(src);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001510 }
1511 else if(src.type == Shader::PARAMETER_PREDICATE)
1512 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001513 IFp(src);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001514 }
1515 else
1516 {
Nicolas Capensc2534f42016-04-04 11:13:24 -04001517 Int4 condition = As<Int4>(fetchRegister(src).x);
Nicolas Capens4f172c72016-01-13 08:34:30 -05001518 IF(condition);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001519 }
1520 }
1521
Nicolas Capens4f172c72016-01-13 08:34:30 -05001522 void PixelProgram::IFb(const Src &boolRegister)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001523 {
1524 ASSERT(ifDepth < 24 + 4);
1525
Nicolas Capens4f172c72016-01-13 08:34:30 -05001526 Bool condition = (*Pointer<Byte>(data + OFFSET(DrawData, ps.b[boolRegister.index])) != Byte(0)); // FIXME
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001527
1528 if(boolRegister.modifier == Shader::MODIFIER_NOT)
1529 {
1530 condition = !condition;
1531 }
1532
1533 llvm::BasicBlock *trueBlock = Nucleus::createBasicBlock();
1534 llvm::BasicBlock *falseBlock = Nucleus::createBasicBlock();
1535
1536 branch(condition, trueBlock, falseBlock);
1537
1538 isConditionalIf[ifDepth] = false;
1539 ifFalseBlock[ifDepth] = falseBlock;
1540
1541 ifDepth++;
1542 }
1543
Nicolas Capens4f172c72016-01-13 08:34:30 -05001544 void PixelProgram::IFp(const Src &predicateRegister)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001545 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001546 Int4 condition = As<Int4>(p0[predicateRegister.swizzle & 0x3]);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001547
1548 if(predicateRegister.modifier == Shader::MODIFIER_NOT)
1549 {
1550 condition = ~condition;
1551 }
1552
Nicolas Capens4f172c72016-01-13 08:34:30 -05001553 IF(condition);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001554 }
1555
Nicolas Capens4f172c72016-01-13 08:34:30 -05001556 void PixelProgram::IFC(Vector4f &src0, Vector4f &src1, Control control)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001557 {
1558 Int4 condition;
1559
1560 switch(control)
1561 {
1562 case Shader::CONTROL_GT: condition = CmpNLE(src0.x, src1.x); break;
1563 case Shader::CONTROL_EQ: condition = CmpEQ(src0.x, src1.x); break;
1564 case Shader::CONTROL_GE: condition = CmpNLT(src0.x, src1.x); break;
1565 case Shader::CONTROL_LT: condition = CmpLT(src0.x, src1.x); break;
1566 case Shader::CONTROL_NE: condition = CmpNEQ(src0.x, src1.x); break;
1567 case Shader::CONTROL_LE: condition = CmpLE(src0.x, src1.x); break;
1568 default:
1569 ASSERT(false);
1570 }
1571
Nicolas Capens4f172c72016-01-13 08:34:30 -05001572 IF(condition);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001573 }
1574
Nicolas Capens4f172c72016-01-13 08:34:30 -05001575 void PixelProgram::IF(Int4 &condition)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001576 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001577 condition &= enableStack[enableIndex];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001578
Nicolas Capens4f172c72016-01-13 08:34:30 -05001579 enableIndex++;
1580 enableStack[enableIndex] = condition;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001581
1582 llvm::BasicBlock *trueBlock = Nucleus::createBasicBlock();
1583 llvm::BasicBlock *falseBlock = Nucleus::createBasicBlock();
1584
1585 Bool notAllFalse = SignMask(condition) != 0;
1586
1587 branch(notAllFalse, trueBlock, falseBlock);
1588
1589 isConditionalIf[ifDepth] = true;
1590 ifFalseBlock[ifDepth] = falseBlock;
1591
1592 ifDepth++;
1593 breakDepth++;
1594 }
1595
1596 void PixelProgram::LABEL(int labelIndex)
1597 {
1598 if(!labelBlock[labelIndex])
1599 {
1600 labelBlock[labelIndex] = Nucleus::createBasicBlock();
1601 }
1602
1603 Nucleus::setInsertBlock(labelBlock[labelIndex]);
1604 currentLabel = labelIndex;
1605 }
1606
Nicolas Capens4f172c72016-01-13 08:34:30 -05001607 void PixelProgram::LOOP(const Src &integerRegister)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001608 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001609 loopDepth++;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001610
Nicolas Capens4f172c72016-01-13 08:34:30 -05001611 iteration[loopDepth] = *Pointer<Int>(data + OFFSET(DrawData, ps.i[integerRegister.index][0]));
1612 aL[loopDepth] = *Pointer<Int>(data + OFFSET(DrawData, ps.i[integerRegister.index][1]));
1613 increment[loopDepth] = *Pointer<Int>(data + OFFSET(DrawData, ps.i[integerRegister.index][2]));
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001614
Nicolas Capens4f172c72016-01-13 08:34:30 -05001615 // If(increment[loopDepth] == 0)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001616 // {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001617 // increment[loopDepth] = 1;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001618 // }
1619
1620 llvm::BasicBlock *loopBlock = Nucleus::createBasicBlock();
1621 llvm::BasicBlock *testBlock = Nucleus::createBasicBlock();
1622 llvm::BasicBlock *endBlock = Nucleus::createBasicBlock();
1623
1624 loopRepTestBlock[loopRepDepth] = testBlock;
1625 loopRepEndBlock[loopRepDepth] = endBlock;
1626
1627 // FIXME: jump(testBlock)
1628 Nucleus::createBr(testBlock);
1629 Nucleus::setInsertBlock(testBlock);
1630
Nicolas Capens4f172c72016-01-13 08:34:30 -05001631 branch(iteration[loopDepth] > 0, loopBlock, endBlock);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001632 Nucleus::setInsertBlock(loopBlock);
1633
Nicolas Capens4f172c72016-01-13 08:34:30 -05001634 iteration[loopDepth] = iteration[loopDepth] - 1; // FIXME: --
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001635
1636 loopRepDepth++;
1637 breakDepth = 0;
1638 }
1639
Nicolas Capens4f172c72016-01-13 08:34:30 -05001640 void PixelProgram::REP(const Src &integerRegister)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001641 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001642 loopDepth++;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001643
Nicolas Capens4f172c72016-01-13 08:34:30 -05001644 iteration[loopDepth] = *Pointer<Int>(data + OFFSET(DrawData, ps.i[integerRegister.index][0]));
1645 aL[loopDepth] = aL[loopDepth - 1];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001646
1647 llvm::BasicBlock *loopBlock = Nucleus::createBasicBlock();
1648 llvm::BasicBlock *testBlock = Nucleus::createBasicBlock();
1649 llvm::BasicBlock *endBlock = Nucleus::createBasicBlock();
1650
1651 loopRepTestBlock[loopRepDepth] = testBlock;
1652 loopRepEndBlock[loopRepDepth] = endBlock;
1653
1654 // FIXME: jump(testBlock)
1655 Nucleus::createBr(testBlock);
1656 Nucleus::setInsertBlock(testBlock);
1657
Nicolas Capens4f172c72016-01-13 08:34:30 -05001658 branch(iteration[loopDepth] > 0, loopBlock, endBlock);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001659 Nucleus::setInsertBlock(loopBlock);
1660
Nicolas Capens4f172c72016-01-13 08:34:30 -05001661 iteration[loopDepth] = iteration[loopDepth] - 1; // FIXME: --
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001662
1663 loopRepDepth++;
1664 breakDepth = 0;
1665 }
1666
Nicolas Capens4f172c72016-01-13 08:34:30 -05001667 void PixelProgram::WHILE(const Src &temporaryRegister)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001668 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001669 enableIndex++;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001670
1671 llvm::BasicBlock *loopBlock = Nucleus::createBasicBlock();
1672 llvm::BasicBlock *testBlock = Nucleus::createBasicBlock();
1673 llvm::BasicBlock *endBlock = Nucleus::createBasicBlock();
1674
1675 loopRepTestBlock[loopRepDepth] = testBlock;
1676 loopRepEndBlock[loopRepDepth] = endBlock;
1677
Nicolas Capens4f172c72016-01-13 08:34:30 -05001678 Int4 restoreBreak = enableBreak;
1679 Int4 restoreContinue = enableContinue;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001680
1681 // FIXME: jump(testBlock)
1682 Nucleus::createBr(testBlock);
1683 Nucleus::setInsertBlock(testBlock);
Nicolas Capens4f172c72016-01-13 08:34:30 -05001684 enableContinue = restoreContinue;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001685
Nicolas Capensc2534f42016-04-04 11:13:24 -04001686 const Vector4f &src = fetchRegister(temporaryRegister);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001687 Int4 condition = As<Int4>(src.x);
Nicolas Capens4f172c72016-01-13 08:34:30 -05001688 condition &= enableStack[enableIndex - 1];
Nicolas Capens2ff29482016-04-28 15:28:02 -04001689 if(shader->containsLeaveInstruction()) condition &= enableLeave;
Nicolas Capens4f172c72016-01-13 08:34:30 -05001690 enableStack[enableIndex] = condition;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001691
1692 Bool notAllFalse = SignMask(condition) != 0;
1693 branch(notAllFalse, loopBlock, endBlock);
1694
1695 Nucleus::setInsertBlock(endBlock);
Nicolas Capens4f172c72016-01-13 08:34:30 -05001696 enableBreak = restoreBreak;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001697
1698 Nucleus::setInsertBlock(loopBlock);
1699
1700 loopRepDepth++;
1701 breakDepth = 0;
1702 }
1703
Alexis Hetu9aa83a92016-05-02 17:34:46 -04001704 void PixelProgram::SWITCH()
1705 {
1706 enableIndex++;
1707 enableStack[enableIndex] = Int4(0xFFFFFFFF);
1708
1709 llvm::BasicBlock *endBlock = Nucleus::createBasicBlock();
1710
1711 loopRepTestBlock[loopRepDepth] = nullptr;
1712 loopRepEndBlock[loopRepDepth] = endBlock;
1713
1714 loopRepDepth++;
1715 breakDepth = 0;
1716 }
1717
Nicolas Capens4f172c72016-01-13 08:34:30 -05001718 void PixelProgram::RET()
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001719 {
1720 if(currentLabel == -1)
1721 {
1722 returnBlock = Nucleus::createBasicBlock();
1723 Nucleus::createBr(returnBlock);
1724 }
1725 else
1726 {
1727 llvm::BasicBlock *unreachableBlock = Nucleus::createBasicBlock();
1728
1729 if(callRetBlock[currentLabel].size() > 1) // Pop the return destination from the call stack
1730 {
1731 // FIXME: Encapsulate
Nicolas Capens4f172c72016-01-13 08:34:30 -05001732 UInt index = callStack[--stackIndex];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001733
1734 llvm::Value *value = index.loadValue();
1735 llvm::Value *switchInst = Nucleus::createSwitch(value, unreachableBlock, (int)callRetBlock[currentLabel].size());
1736
1737 for(unsigned int i = 0; i < callRetBlock[currentLabel].size(); i++)
1738 {
1739 Nucleus::addSwitchCase(switchInst, i, callRetBlock[currentLabel][i]);
1740 }
1741 }
1742 else if(callRetBlock[currentLabel].size() == 1) // Jump directly to the unique return destination
1743 {
1744 Nucleus::createBr(callRetBlock[currentLabel][0]);
1745 }
1746 else // Function isn't called
1747 {
1748 Nucleus::createBr(unreachableBlock);
1749 }
1750
1751 Nucleus::setInsertBlock(unreachableBlock);
1752 Nucleus::createUnreachable();
1753 }
1754 }
1755
Nicolas Capens4f172c72016-01-13 08:34:30 -05001756 void PixelProgram::LEAVE()
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001757 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001758 enableLeave = enableLeave & ~enableStack[enableIndex];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001759
1760 // FIXME: Return from function if all instances left
1761 // FIXME: Use enableLeave in other control-flow constructs
1762 }
Nicolas Capens5d961882016-01-01 23:18:14 -05001763}