blob: 2513edf0511b69a5e67b91aab6405a802aaeca1d [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 Hetu0cff3cb2016-05-04 16:23:50 -0400607 case FORMAT_R8:
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400608 case FORMAT_A8:
609 case FORMAT_G16R16:
610 case FORMAT_A16B16G16R16:
611 for(unsigned int q = 0; q < state.multiSample; q++)
612 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500613 Pointer<Byte> buffer = cBuffer[index] + q * *Pointer<Int>(data + OFFSET(DrawData, colorSliceB[index]));
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400614 Vector4s color;
615
Nicolas Capens13c70cd2016-04-05 10:51:45 -0400616 if(state.targetFormat[index] == FORMAT_R5G6B5)
617 {
618 color.x = UShort4(c[index].x * Float4(0xFBFF), false);
619 color.y = UShort4(c[index].y * Float4(0xFDFF), false);
620 color.z = UShort4(c[index].z * Float4(0xFBFF), false);
621 color.w = UShort4(c[index].w * Float4(0xFFFF), false);
622 }
623 else
624 {
625 color.x = convertFixed16(c[index].x, false);
626 color.y = convertFixed16(c[index].y, false);
627 color.z = convertFixed16(c[index].z, false);
628 color.w = convertFixed16(c[index].w, false);
629 }
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400630
631 if(state.multiSampleMask & (1 << q))
632 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500633 alphaBlend(index, buffer, color, x);
634 logicOperation(index, buffer, color, x);
635 writeColor(index, buffer, x, color, sMask[q], zMask[q], cMask[q]);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400636 }
637 }
638 break;
639 case FORMAT_R32F:
640 case FORMAT_G32R32F:
Alexis Hetudbd1a8e2016-04-13 11:40:30 -0400641 case FORMAT_X32B32G32R32F:
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400642 case FORMAT_A32B32G32R32F:
Alexis Hetu1abb6382016-02-08 11:21:16 -0500643 case FORMAT_R32I:
644 case FORMAT_G32R32I:
645 case FORMAT_A32B32G32R32I:
646 case FORMAT_R32UI:
647 case FORMAT_G32R32UI:
648 case FORMAT_A32B32G32R32UI:
Alexis Hetubd7117d2016-06-02 10:35:59 -0400649 case FORMAT_R16I:
650 case FORMAT_G16R16I:
651 case FORMAT_A16B16G16R16I:
652 case FORMAT_R16UI:
653 case FORMAT_G16R16UI:
654 case FORMAT_A16B16G16R16UI:
655 case FORMAT_R8I:
656 case FORMAT_G8R8I:
657 case FORMAT_A8B8G8R8I:
658 case FORMAT_R8UI:
659 case FORMAT_G8R8UI:
660 case FORMAT_A8B8G8R8UI:
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400661 for(unsigned int q = 0; q < state.multiSample; q++)
662 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500663 Pointer<Byte> buffer = cBuffer[index] + q * *Pointer<Int>(data + OFFSET(DrawData, colorSliceB[index]));
664 Vector4f color = c[index];
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400665
666 if(state.multiSampleMask & (1 << q))
667 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500668 alphaBlend(index, buffer, color, x);
669 writeColor(index, buffer, x, color, sMask[q], zMask[q], cMask[q]);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400670 }
671 }
672 break;
673 default:
674 ASSERT(false);
675 }
676 }
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500677 }
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400678
Meng-Lin Wu2fce5822016-06-07 16:15:12 -0400679 void PixelProgram::sampleTexture(Vector4f &c, const Src &sampler, Float4 &u, Float4 &v, Float4 &w, Float4 &q, Vector4f &dsx, Vector4f &dsy, Vector4f &offset, SamplerMethod method, unsigned int options)
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400680 {
Nicolas Capensc2534f42016-04-04 11:13:24 -0400681 Vector4f tmp;
682
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400683 if(sampler.type == Shader::PARAMETER_SAMPLER && sampler.rel.type == Shader::PARAMETER_VOID)
684 {
Meng-Lin Wu2fce5822016-06-07 16:15:12 -0400685 sampleTexture(tmp, sampler.index, u, v, w, q, dsx, dsy, offset, method, options);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400686 }
687 else
688 {
Nicolas Capensc2534f42016-04-04 11:13:24 -0400689 Int index = As<Int>(Float(fetchRegister(sampler).x.x));
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400690
691 for(int i = 0; i < TEXTURE_IMAGE_UNITS; i++)
692 {
693 if(shader->usesSampler(i))
694 {
695 If(index == i)
696 {
Meng-Lin Wu2fce5822016-06-07 16:15:12 -0400697 sampleTexture(tmp, i, u, v, w, q, dsx, dsy, offset, method, options);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400698 // FIXME: When the sampler states are the same, we could use one sampler and just index the texture
699 }
700 }
701 }
702 }
Nicolas Capensc2534f42016-04-04 11:13:24 -0400703
704 c.x = tmp[(sampler.swizzle >> 0) & 0x3];
705 c.y = tmp[(sampler.swizzle >> 2) & 0x3];
706 c.z = tmp[(sampler.swizzle >> 4) & 0x3];
707 c.w = tmp[(sampler.swizzle >> 6) & 0x3];
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400708 }
709
Meng-Lin Wu2fce5822016-06-07 16:15:12 -0400710 void PixelProgram::sampleTexture(Vector4f &c, int samplerIndex, Float4 &u, Float4 &v, Float4 &w, Float4 &q, Vector4f &dsx, Vector4f &dsy, Vector4f &offset, SamplerMethod method, unsigned int options)
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400711 {
Nicolas Capensc2534f42016-04-04 11:13:24 -0400712 #if PERF_PROFILE
713 Long texTime = Ticks();
714 #endif
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400715
Meng-Lin Wu2337a192016-06-01 13:54:07 -0400716 Pointer<Byte> texture = data + OFFSET(DrawData, mipmap) + samplerIndex * sizeof(Texture);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400717
Meng-Lin Wu2fce5822016-06-07 16:15:12 -0400718 if(!(options & Project))
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400719 {
Meng-Lin Wu2fce5822016-06-07 16:15:12 -0400720 sampler[samplerIndex]->sampleTexture(texture, c, u, v, w, q, dsx, dsy, offset, options, method);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400721 }
722 else
723 {
724 Float4 rq = reciprocal(q);
725
726 Float4 u_q = u * rq;
727 Float4 v_q = v * rq;
728 Float4 w_q = w * rq;
729
Meng-Lin Wu2fce5822016-06-07 16:15:12 -0400730 sampler[samplerIndex]->sampleTexture(texture, c, u_q, v_q, w_q, q, dsx, dsy, offset, options, method);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400731 }
732
Nicolas Capensc2534f42016-04-04 11:13:24 -0400733 #if PERF_PROFILE
734 cycles[PERF_TEX] += Ticks() - texTime;
735 #endif
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400736 }
737
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500738 void PixelProgram::clampColor(Vector4f oC[RENDERTARGETS])
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400739 {
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500740 for(int index = 0; index < RENDERTARGETS; index++)
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400741 {
742 if(!state.colorWriteActive(index) && !(index == 0 && state.alphaTestActive()))
743 {
744 continue;
745 }
746
747 switch(state.targetFormat[index])
748 {
749 case FORMAT_NULL:
750 break;
751 case FORMAT_R5G6B5:
752 case FORMAT_A8R8G8B8:
753 case FORMAT_A8B8G8R8:
754 case FORMAT_X8R8G8B8:
755 case FORMAT_X8B8G8R8:
Alexis Hetu049a1872016-04-25 16:59:58 -0400756 case FORMAT_SRGB8_X8:
757 case FORMAT_SRGB8_A8:
Alexis Hetu0cff3cb2016-05-04 16:23:50 -0400758 case FORMAT_R8:
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400759 case FORMAT_A8:
760 case FORMAT_G16R16:
761 case FORMAT_A16B16G16R16:
762 oC[index].x = Max(oC[index].x, Float4(0.0f)); oC[index].x = Min(oC[index].x, Float4(1.0f));
763 oC[index].y = Max(oC[index].y, Float4(0.0f)); oC[index].y = Min(oC[index].y, Float4(1.0f));
764 oC[index].z = Max(oC[index].z, Float4(0.0f)); oC[index].z = Min(oC[index].z, Float4(1.0f));
765 oC[index].w = Max(oC[index].w, Float4(0.0f)); oC[index].w = Min(oC[index].w, Float4(1.0f));
766 break;
767 case FORMAT_R32F:
768 case FORMAT_G32R32F:
Alexis Hetudbd1a8e2016-04-13 11:40:30 -0400769 case FORMAT_X32B32G32R32F:
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400770 case FORMAT_A32B32G32R32F:
Alexis Hetu1abb6382016-02-08 11:21:16 -0500771 case FORMAT_R32I:
772 case FORMAT_G32R32I:
773 case FORMAT_A32B32G32R32I:
774 case FORMAT_R32UI:
775 case FORMAT_G32R32UI:
776 case FORMAT_A32B32G32R32UI:
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400777 break;
778 default:
779 ASSERT(false);
780 }
781 }
782 }
783
Nicolas Capens4f172c72016-01-13 08:34:30 -0500784 Int4 PixelProgram::enableMask(const Shader::Instruction *instruction)
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400785 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500786 Int4 enable = instruction->analysisBranch ? Int4(enableStack[enableIndex]) : Int4(0xFFFFFFFF);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400787
788 if(!whileTest)
789 {
790 if(shader->containsBreakInstruction() && instruction->analysisBreak)
791 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500792 enable &= enableBreak;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400793 }
794
795 if(shader->containsContinueInstruction() && instruction->analysisContinue)
796 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500797 enable &= enableContinue;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400798 }
799
800 if(shader->containsLeaveInstruction() && instruction->analysisLeave)
801 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500802 enable &= enableLeave;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400803 }
804 }
805
806 return enable;
807 }
808
Nicolas Capensc2534f42016-04-04 11:13:24 -0400809 Vector4f PixelProgram::fetchRegister(const Src &src, unsigned int offset)
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400810 {
811 Vector4f reg;
Nicolas Capens5d961882016-01-01 23:18:14 -0500812 unsigned int i = src.index + offset;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400813
814 switch(src.type)
815 {
816 case Shader::PARAMETER_TEMP:
817 if(src.rel.type == Shader::PARAMETER_VOID)
818 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500819 reg = r[i];
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400820 }
821 else
822 {
Alexis Hetu2c2a7b22015-10-27 16:12:11 -0400823 Int a = relativeAddress(src, src.bufferIndex);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400824
Nicolas Capens4f172c72016-01-13 08:34:30 -0500825 reg = r[i + a];
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400826 }
827 break;
828 case Shader::PARAMETER_INPUT:
829 {
830 if(src.rel.type == Shader::PARAMETER_VOID) // Not relative
831 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500832 reg = v[i];
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400833 }
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400834 else
835 {
Alexis Hetu2c2a7b22015-10-27 16:12:11 -0400836 Int a = relativeAddress(src, src.bufferIndex);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400837
Nicolas Capens4f172c72016-01-13 08:34:30 -0500838 reg = v[i + a];
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400839 }
840 }
841 break;
842 case Shader::PARAMETER_CONST:
Nicolas Capens4f172c72016-01-13 08:34:30 -0500843 reg = readConstant(src, offset);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400844 break;
845 case Shader::PARAMETER_TEXTURE:
Nicolas Capens4f172c72016-01-13 08:34:30 -0500846 reg = v[2 + i];
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400847 break;
848 case Shader::PARAMETER_MISCTYPE:
Nicolas Capens4f172c72016-01-13 08:34:30 -0500849 if(src.index == 0) reg = vPos;
850 if(src.index == 1) reg = vFace;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400851 break;
852 case Shader::PARAMETER_SAMPLER:
853 if(src.rel.type == Shader::PARAMETER_VOID)
854 {
855 reg.x = As<Float4>(Int4(i));
856 }
857 else if(src.rel.type == Shader::PARAMETER_TEMP)
858 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500859 reg.x = As<Float4>(Int4(i) + As<Int4>(r[src.rel.index].x));
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400860 }
861 return reg;
862 case Shader::PARAMETER_PREDICATE: return reg; // Dummy
863 case Shader::PARAMETER_VOID: return reg; // Dummy
864 case Shader::PARAMETER_FLOAT4LITERAL:
865 reg.x = Float4(src.value[0]);
866 reg.y = Float4(src.value[1]);
867 reg.z = Float4(src.value[2]);
868 reg.w = Float4(src.value[3]);
869 break;
870 case Shader::PARAMETER_CONSTINT: return reg; // Dummy
871 case Shader::PARAMETER_CONSTBOOL: return reg; // Dummy
872 case Shader::PARAMETER_LOOP: return reg; // Dummy
873 case Shader::PARAMETER_COLOROUT:
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500874 if(src.rel.type == Shader::PARAMETER_VOID) // Not relative
875 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500876 reg = oC[i];
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500877 }
878 else
879 {
Alexis Hetu2c2a7b22015-10-27 16:12:11 -0400880 Int a = relativeAddress(src, src.bufferIndex);
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500881
Nicolas Capens4f172c72016-01-13 08:34:30 -0500882 reg = oC[i + a];
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500883 }
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400884 break;
885 case Shader::PARAMETER_DEPTHOUT:
Nicolas Capens4f172c72016-01-13 08:34:30 -0500886 reg.x = oDepth;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400887 break;
888 default:
889 ASSERT(false);
890 }
891
892 const Float4 &x = reg[(src.swizzle >> 0) & 0x3];
893 const Float4 &y = reg[(src.swizzle >> 2) & 0x3];
894 const Float4 &z = reg[(src.swizzle >> 4) & 0x3];
895 const Float4 &w = reg[(src.swizzle >> 6) & 0x3];
896
897 Vector4f mod;
898
899 switch(src.modifier)
900 {
901 case Shader::MODIFIER_NONE:
902 mod.x = x;
903 mod.y = y;
904 mod.z = z;
905 mod.w = w;
906 break;
907 case Shader::MODIFIER_NEGATE:
908 mod.x = -x;
909 mod.y = -y;
910 mod.z = -z;
911 mod.w = -w;
912 break;
913 case Shader::MODIFIER_ABS:
914 mod.x = Abs(x);
915 mod.y = Abs(y);
916 mod.z = Abs(z);
917 mod.w = Abs(w);
918 break;
919 case Shader::MODIFIER_ABS_NEGATE:
920 mod.x = -Abs(x);
921 mod.y = -Abs(y);
922 mod.z = -Abs(z);
923 mod.w = -Abs(w);
924 break;
925 case Shader::MODIFIER_NOT:
926 mod.x = As<Float4>(As<Int4>(x) ^ Int4(0xFFFFFFFF));
927 mod.y = As<Float4>(As<Int4>(y) ^ Int4(0xFFFFFFFF));
928 mod.z = As<Float4>(As<Int4>(z) ^ Int4(0xFFFFFFFF));
929 mod.w = As<Float4>(As<Int4>(w) ^ Int4(0xFFFFFFFF));
930 break;
931 default:
932 ASSERT(false);
933 }
934
935 return mod;
936 }
937
Alexis Hetu2c2a7b22015-10-27 16:12:11 -0400938 RValue<Pointer<Byte>> PixelProgram::uniformAddress(int bufferIndex, unsigned int index)
939 {
940 if(bufferIndex == -1)
941 {
942 return data + OFFSET(DrawData, ps.c[index]);
943 }
944 else
945 {
946 return *Pointer<Pointer<Byte>>(data + OFFSET(DrawData, ps.u[bufferIndex])) + index;
947 }
948 }
949
950 RValue<Pointer<Byte>> PixelProgram::uniformAddress(int bufferIndex, unsigned int index, Int& offset)
951 {
952 return uniformAddress(bufferIndex, index) + offset * sizeof(float4);
953 }
954
Nicolas Capens4f172c72016-01-13 08:34:30 -0500955 Vector4f PixelProgram::readConstant(const Src &src, unsigned int offset)
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400956 {
957 Vector4f c;
Nicolas Capens5d961882016-01-01 23:18:14 -0500958 unsigned int i = src.index + offset;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400959
960 if(src.rel.type == Shader::PARAMETER_VOID) // Not relative
961 {
Alexis Hetu2c2a7b22015-10-27 16:12:11 -0400962 c.x = c.y = c.z = c.w = *Pointer<Float4>(uniformAddress(src.bufferIndex, i));
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400963
964 c.x = c.x.xxxx;
965 c.y = c.y.yyyy;
966 c.z = c.z.zzzz;
967 c.w = c.w.wwww;
968
969 if(shader->containsDefineInstruction()) // Constant may be known at compile time
970 {
971 for(size_t j = 0; j < shader->getLength(); j++)
972 {
973 const Shader::Instruction &instruction = *shader->getInstruction(j);
974
975 if(instruction.opcode == Shader::OPCODE_DEF)
976 {
977 if(instruction.dst.index == i)
978 {
979 c.x = Float4(instruction.src[0].value[0]);
980 c.y = Float4(instruction.src[0].value[1]);
981 c.z = Float4(instruction.src[0].value[2]);
982 c.w = Float4(instruction.src[0].value[3]);
983
984 break;
985 }
986 }
987 }
988 }
989 }
990 else if(src.rel.type == Shader::PARAMETER_LOOP)
991 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500992 Int loopCounter = aL[loopDepth];
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400993
Alexis Hetu2c2a7b22015-10-27 16:12:11 -0400994 c.x = c.y = c.z = c.w = *Pointer<Float4>(uniformAddress(src.bufferIndex, i, loopCounter));
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400995
996 c.x = c.x.xxxx;
997 c.y = c.y.yyyy;
998 c.z = c.z.zzzz;
999 c.w = c.w.wwww;
1000 }
1001 else
1002 {
Alexis Hetu2c2a7b22015-10-27 16:12:11 -04001003 Int a = relativeAddress(src, src.bufferIndex);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001004
Alexis Hetu2c2a7b22015-10-27 16:12:11 -04001005 c.x = c.y = c.z = c.w = *Pointer<Float4>(uniformAddress(src.bufferIndex, i, a));
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001006
1007 c.x = c.x.xxxx;
1008 c.y = c.y.yyyy;
1009 c.z = c.z.zzzz;
1010 c.w = c.w.wwww;
1011 }
1012
1013 return c;
1014 }
1015
Alexis Hetu2c2a7b22015-10-27 16:12:11 -04001016 Int PixelProgram::relativeAddress(const Shader::Parameter &var, int bufferIndex)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001017 {
1018 ASSERT(var.rel.deterministic);
1019
1020 if(var.rel.type == Shader::PARAMETER_TEMP)
1021 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001022 return As<Int>(Extract(r[var.rel.index].x, 0)) * var.rel.scale;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001023 }
1024 else if(var.rel.type == Shader::PARAMETER_INPUT)
1025 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001026 return As<Int>(Extract(v[var.rel.index].x, 0)) * var.rel.scale;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001027 }
1028 else if(var.rel.type == Shader::PARAMETER_OUTPUT)
1029 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001030 return As<Int>(Extract(oC[var.rel.index].x, 0)) * var.rel.scale;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001031 }
1032 else if(var.rel.type == Shader::PARAMETER_CONST)
1033 {
Alexis Hetu48be7352016-02-10 14:32:34 -05001034 return *Pointer<Int>(uniformAddress(bufferIndex, var.rel.index)) * var.rel.scale;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001035 }
Nicolas Capens907700d2016-01-20 17:09:28 -05001036 else if(var.rel.type == Shader::PARAMETER_LOOP)
1037 {
1038 return aL[loopDepth];
1039 }
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001040 else ASSERT(false);
1041
1042 return 0;
1043 }
1044
1045 Float4 PixelProgram::linearToSRGB(const Float4 &x) // Approximates x^(1.0/2.2)
1046 {
1047 Float4 sqrtx = Rcp_pp(RcpSqrt_pp(x));
1048 Float4 sRGB = sqrtx * Float4(1.14f) - x * Float4(0.14f);
1049
1050 return Min(Max(sRGB, Float4(0.0f)), Float4(1.0f));
1051 }
1052
Nicolas Capens4f172c72016-01-13 08:34:30 -05001053 void PixelProgram::M3X2(Vector4f &dst, Vector4f &src0, const Src &src1)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001054 {
Nicolas Capensc2534f42016-04-04 11:13:24 -04001055 Vector4f row0 = fetchRegister(src1, 0);
1056 Vector4f row1 = fetchRegister(src1, 1);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001057
1058 dst.x = dot3(src0, row0);
1059 dst.y = dot3(src0, row1);
1060 }
1061
Nicolas Capens4f172c72016-01-13 08:34:30 -05001062 void PixelProgram::M3X3(Vector4f &dst, Vector4f &src0, const Src &src1)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001063 {
Nicolas Capensc2534f42016-04-04 11:13:24 -04001064 Vector4f row0 = fetchRegister(src1, 0);
1065 Vector4f row1 = fetchRegister(src1, 1);
1066 Vector4f row2 = fetchRegister(src1, 2);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001067
1068 dst.x = dot3(src0, row0);
1069 dst.y = dot3(src0, row1);
1070 dst.z = dot3(src0, row2);
1071 }
1072
Nicolas Capens4f172c72016-01-13 08:34:30 -05001073 void PixelProgram::M3X4(Vector4f &dst, Vector4f &src0, const Src &src1)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001074 {
Nicolas Capensc2534f42016-04-04 11:13:24 -04001075 Vector4f row0 = fetchRegister(src1, 0);
1076 Vector4f row1 = fetchRegister(src1, 1);
1077 Vector4f row2 = fetchRegister(src1, 2);
1078 Vector4f row3 = fetchRegister(src1, 3);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001079
1080 dst.x = dot3(src0, row0);
1081 dst.y = dot3(src0, row1);
1082 dst.z = dot3(src0, row2);
1083 dst.w = dot3(src0, row3);
1084 }
1085
Nicolas Capens4f172c72016-01-13 08:34:30 -05001086 void PixelProgram::M4X3(Vector4f &dst, Vector4f &src0, const Src &src1)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001087 {
Nicolas Capensc2534f42016-04-04 11:13:24 -04001088 Vector4f row0 = fetchRegister(src1, 0);
1089 Vector4f row1 = fetchRegister(src1, 1);
1090 Vector4f row2 = fetchRegister(src1, 2);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001091
1092 dst.x = dot4(src0, row0);
1093 dst.y = dot4(src0, row1);
1094 dst.z = dot4(src0, row2);
1095 }
1096
Nicolas Capens4f172c72016-01-13 08:34:30 -05001097 void PixelProgram::M4X4(Vector4f &dst, Vector4f &src0, const Src &src1)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001098 {
Nicolas Capensc2534f42016-04-04 11:13:24 -04001099 Vector4f row0 = fetchRegister(src1, 0);
1100 Vector4f row1 = fetchRegister(src1, 1);
1101 Vector4f row2 = fetchRegister(src1, 2);
1102 Vector4f row3 = fetchRegister(src1, 3);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001103
1104 dst.x = dot4(src0, row0);
1105 dst.y = dot4(src0, row1);
1106 dst.z = dot4(src0, row2);
1107 dst.w = dot4(src0, row3);
1108 }
1109
Nicolas Capens4f172c72016-01-13 08:34:30 -05001110 void PixelProgram::TEXLD(Vector4f &dst, Vector4f &src0, const Src &src1, bool project, bool bias)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001111 {
Meng-Lin Wu2fce5822016-06-07 16:15:12 -04001112 sampleTexture(dst, src1, src0.x, src0.y, src0.z, src0.w, src0, src0, src0, bias ? Bias : Implicit, project ? Project : None);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001113 }
1114
Meng-Lin Wu2337a192016-06-01 13:54:07 -04001115 void PixelProgram::TEXOFFSET(Vector4f &dst, Vector4f &src0, const Src &src1, Vector4f &src2, bool project, bool bias)
Alexis Hetu25d47fc2015-10-22 13:58:32 -04001116 {
Meng-Lin Wu2fce5822016-06-07 16:15:12 -04001117 sampleTexture(dst, src1, src0.x, src0.y, src0.z, src0.w, src0, src0, src2, bias ? Bias : Implicit, project ? (Project | Offset) : Offset);
Alexis Hetu25d47fc2015-10-22 13:58:32 -04001118 }
1119
Nicolas Capens4f172c72016-01-13 08:34:30 -05001120 void PixelProgram::TEXLDL(Vector4f &dst, Vector4f &src0, const Src &src1, Vector4f &offset, bool project, bool bias)
Alexis Hetu25d47fc2015-10-22 13:58:32 -04001121 {
Meng-Lin Wu2fce5822016-06-07 16:15:12 -04001122 sampleTexture(dst, src1, src0.x, src0.y, src0.z, src0.w, src0, src0, offset, Lod, project ? (Project | Offset) : Offset);
Alexis Hetu25d47fc2015-10-22 13:58:32 -04001123 }
1124
Nicolas Capens4f172c72016-01-13 08:34:30 -05001125 void PixelProgram::TEXELFETCH(Vector4f &dst, Vector4f &src0, const Src& src1, Vector4f &src2)
Alexis Hetu25d47fc2015-10-22 13:58:32 -04001126 {
Meng-Lin Wu2fce5822016-06-07 16:15:12 -04001127 sampleTexture(dst, src1, src0.x, src0.y, src0.z, Float4(As<Int4>(src2.x)), src0, src0, src0, Lod, Fetch);
Alexis Hetu25d47fc2015-10-22 13:58:32 -04001128 }
1129
Nicolas Capens4f172c72016-01-13 08:34:30 -05001130 void PixelProgram::TEXELFETCH(Vector4f &dst, Vector4f &src0, const Src& src1, Vector4f &src2, Vector4f &offset)
Alexis Hetu25d47fc2015-10-22 13:58:32 -04001131 {
Meng-Lin Wu2fce5822016-06-07 16:15:12 -04001132 sampleTexture(dst, src1, src0.x, src0.y, src0.z, Float4(As<Int4>(src2.x)), src0, src0, offset, Lod, Fetch | Offset);
Alexis Hetu25d47fc2015-10-22 13:58:32 -04001133 }
1134
Nicolas Capens4f172c72016-01-13 08:34:30 -05001135 void PixelProgram::TEXGRAD(Vector4f &dst, Vector4f &src0, const Src& src1, Vector4f &src2, Vector4f &src3)
Alexis Hetu25d47fc2015-10-22 13:58:32 -04001136 {
Meng-Lin Wu2fce5822016-06-07 16:15:12 -04001137 sampleTexture(dst, src1, src0.x, src0.y, src0.z, src0.w, src2, src3, src0, Grad, None);
Alexis Hetu25d47fc2015-10-22 13:58:32 -04001138 }
1139
Nicolas Capens4f172c72016-01-13 08:34:30 -05001140 void PixelProgram::TEXGRAD(Vector4f &dst, Vector4f &src0, const Src& src1, Vector4f &src2, Vector4f &src3, Vector4f &offset)
Alexis Hetu25d47fc2015-10-22 13:58:32 -04001141 {
Meng-Lin Wu2fce5822016-06-07 16:15:12 -04001142 sampleTexture(dst, src1, src0.x, src0.y, src0.z, src0.w, src2, src3, offset, Grad, Offset);
Alexis Hetu25d47fc2015-10-22 13:58:32 -04001143 }
1144
Nicolas Capensc2534f42016-04-04 11:13:24 -04001145 void PixelProgram::TEXLDD(Vector4f &dst, Vector4f &src0, const Src &src1, Vector4f &src2, Vector4f &src3, bool project)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001146 {
Meng-Lin Wu2fce5822016-06-07 16:15:12 -04001147 sampleTexture(dst, src1, src0.x, src0.y, src0.z, src0.w, src2, src3, src0, Grad, project ? Project : None);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001148 }
1149
Nicolas Capensc2534f42016-04-04 11:13:24 -04001150 void PixelProgram::TEXLDL(Vector4f &dst, Vector4f &src0, const Src &src1, bool project)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001151 {
Meng-Lin Wu2fce5822016-06-07 16:15:12 -04001152 sampleTexture(dst, src1, src0.x, src0.y, src0.z, src0.w, src0, src0, src0, Lod, project ? Project : None);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001153 }
1154
Nicolas Capens4f172c72016-01-13 08:34:30 -05001155 void PixelProgram::TEXSIZE(Vector4f &dst, Float4 &lod, const Src &src1)
Alexis Hetu9bcb31d2015-07-22 17:03:26 -04001156 {
Alexis Hetu95ac1872016-06-06 13:26:52 -04001157 Pointer<Byte> texture = data + OFFSET(DrawData, mipmap) + src1.index * sizeof(Texture);
1158 sampler[src1.index]->textureSize(texture, dst, lod);
Alexis Hetu9bcb31d2015-07-22 17:03:26 -04001159 }
1160
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001161 void PixelProgram::TEXKILL(Int cMask[4], Vector4f &src, unsigned char mask)
1162 {
1163 Int kill = -1;
1164
1165 if(mask & 0x1) kill &= SignMask(CmpNLT(src.x, Float4(0.0f)));
1166 if(mask & 0x2) kill &= SignMask(CmpNLT(src.y, Float4(0.0f)));
1167 if(mask & 0x4) kill &= SignMask(CmpNLT(src.z, Float4(0.0f)));
1168 if(mask & 0x8) kill &= SignMask(CmpNLT(src.w, Float4(0.0f)));
1169
1170 // FIXME: Dynamic branching affects TEXKILL?
1171 // if(shader->containsDynamicBranching())
1172 // {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001173 // kill = ~SignMask(enableMask());
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001174 // }
1175
1176 for(unsigned int q = 0; q < state.multiSample; q++)
1177 {
1178 cMask[q] &= kill;
1179 }
1180
1181 // FIXME: Branch to end of shader if all killed?
1182 }
1183
Nicolas Capens4f172c72016-01-13 08:34:30 -05001184 void PixelProgram::DISCARD(Int cMask[4], const Shader::Instruction *instruction)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001185 {
1186 Int kill = 0;
1187
1188 if(shader->containsDynamicBranching())
1189 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001190 kill = ~SignMask(enableMask(instruction));
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001191 }
1192
1193 for(unsigned int q = 0; q < state.multiSample; q++)
1194 {
1195 cMask[q] &= kill;
1196 }
1197
1198 // FIXME: Branch to end of shader if all killed?
1199 }
1200
1201 void PixelProgram::DFDX(Vector4f &dst, Vector4f &src)
1202 {
1203 dst.x = src.x.yyww - src.x.xxzz;
1204 dst.y = src.y.yyww - src.y.xxzz;
1205 dst.z = src.z.yyww - src.z.xxzz;
1206 dst.w = src.w.yyww - src.w.xxzz;
1207 }
1208
1209 void PixelProgram::DFDY(Vector4f &dst, Vector4f &src)
1210 {
1211 dst.x = src.x.zwzw - src.x.xyxy;
1212 dst.y = src.y.zwzw - src.y.xyxy;
1213 dst.z = src.z.zwzw - src.z.xyxy;
1214 dst.w = src.w.zwzw - src.w.xyxy;
1215 }
1216
1217 void PixelProgram::FWIDTH(Vector4f &dst, Vector4f &src)
1218 {
1219 // abs(dFdx(src)) + abs(dFdy(src));
1220 dst.x = Abs(src.x.yyww - src.x.xxzz) + Abs(src.x.zwzw - src.x.xyxy);
1221 dst.y = Abs(src.y.yyww - src.y.xxzz) + Abs(src.y.zwzw - src.y.xyxy);
1222 dst.z = Abs(src.z.yyww - src.z.xxzz) + Abs(src.z.zwzw - src.z.xyxy);
1223 dst.w = Abs(src.w.yyww - src.w.xxzz) + Abs(src.w.zwzw - src.w.xyxy);
1224 }
1225
Nicolas Capens4f172c72016-01-13 08:34:30 -05001226 void PixelProgram::BREAK()
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001227 {
1228 llvm::BasicBlock *deadBlock = Nucleus::createBasicBlock();
1229 llvm::BasicBlock *endBlock = loopRepEndBlock[loopRepDepth - 1];
1230
1231 if(breakDepth == 0)
1232 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001233 enableIndex = enableIndex - breakDepth;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001234 Nucleus::createBr(endBlock);
1235 }
1236 else
1237 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001238 enableBreak = enableBreak & ~enableStack[enableIndex];
1239 Bool allBreak = SignMask(enableBreak) == 0x0;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001240
Nicolas Capens4f172c72016-01-13 08:34:30 -05001241 enableIndex = enableIndex - breakDepth;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001242 branch(allBreak, endBlock, deadBlock);
1243 }
1244
1245 Nucleus::setInsertBlock(deadBlock);
Nicolas Capens4f172c72016-01-13 08:34:30 -05001246 enableIndex = enableIndex + breakDepth;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001247 }
1248
Nicolas Capens4f172c72016-01-13 08:34:30 -05001249 void PixelProgram::BREAKC(Vector4f &src0, Vector4f &src1, Control control)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001250 {
1251 Int4 condition;
1252
1253 switch(control)
1254 {
1255 case Shader::CONTROL_GT: condition = CmpNLE(src0.x, src1.x); break;
1256 case Shader::CONTROL_EQ: condition = CmpEQ(src0.x, src1.x); break;
1257 case Shader::CONTROL_GE: condition = CmpNLT(src0.x, src1.x); break;
1258 case Shader::CONTROL_LT: condition = CmpLT(src0.x, src1.x); break;
1259 case Shader::CONTROL_NE: condition = CmpNEQ(src0.x, src1.x); break;
1260 case Shader::CONTROL_LE: condition = CmpLE(src0.x, src1.x); break;
1261 default:
1262 ASSERT(false);
1263 }
1264
Nicolas Capens4f172c72016-01-13 08:34:30 -05001265 BREAK(condition);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001266 }
1267
Nicolas Capens4f172c72016-01-13 08:34:30 -05001268 void PixelProgram::BREAKP(const Src &predicateRegister) // FIXME: Factor out parts common with BREAKC
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001269 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001270 Int4 condition = As<Int4>(p0[predicateRegister.swizzle & 0x3]);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001271
1272 if(predicateRegister.modifier == Shader::MODIFIER_NOT)
1273 {
1274 condition = ~condition;
1275 }
1276
Nicolas Capens4f172c72016-01-13 08:34:30 -05001277 BREAK(condition);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001278 }
1279
Nicolas Capens4f172c72016-01-13 08:34:30 -05001280 void PixelProgram::BREAK(Int4 &condition)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001281 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001282 condition &= enableStack[enableIndex];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001283
1284 llvm::BasicBlock *continueBlock = Nucleus::createBasicBlock();
1285 llvm::BasicBlock *endBlock = loopRepEndBlock[loopRepDepth - 1];
1286
Nicolas Capens4f172c72016-01-13 08:34:30 -05001287 enableBreak = enableBreak & ~condition;
1288 Bool allBreak = SignMask(enableBreak) == 0x0;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001289
Nicolas Capens4f172c72016-01-13 08:34:30 -05001290 enableIndex = enableIndex - breakDepth;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001291 branch(allBreak, endBlock, continueBlock);
1292
1293 Nucleus::setInsertBlock(continueBlock);
Nicolas Capens4f172c72016-01-13 08:34:30 -05001294 enableIndex = enableIndex + breakDepth;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001295 }
1296
Nicolas Capens4f172c72016-01-13 08:34:30 -05001297 void PixelProgram::CONTINUE()
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001298 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001299 enableContinue = enableContinue & ~enableStack[enableIndex];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001300 }
1301
1302 void PixelProgram::TEST()
1303 {
1304 whileTest = true;
1305 }
1306
Nicolas Capens4f172c72016-01-13 08:34:30 -05001307 void PixelProgram::CALL(int labelIndex, int callSiteIndex)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001308 {
1309 if(!labelBlock[labelIndex])
1310 {
1311 labelBlock[labelIndex] = Nucleus::createBasicBlock();
1312 }
1313
1314 if(callRetBlock[labelIndex].size() > 1)
1315 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001316 callStack[stackIndex++] = UInt(callSiteIndex);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001317 }
1318
Nicolas Capens4f172c72016-01-13 08:34:30 -05001319 Int4 restoreLeave = enableLeave;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001320
1321 Nucleus::createBr(labelBlock[labelIndex]);
1322 Nucleus::setInsertBlock(callRetBlock[labelIndex][callSiteIndex]);
1323
Nicolas Capens4f172c72016-01-13 08:34:30 -05001324 enableLeave = restoreLeave;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001325 }
1326
Nicolas Capens4f172c72016-01-13 08:34:30 -05001327 void PixelProgram::CALLNZ(int labelIndex, int callSiteIndex, const Src &src)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001328 {
1329 if(src.type == Shader::PARAMETER_CONSTBOOL)
1330 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001331 CALLNZb(labelIndex, callSiteIndex, src);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001332 }
1333 else if(src.type == Shader::PARAMETER_PREDICATE)
1334 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001335 CALLNZp(labelIndex, callSiteIndex, src);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001336 }
1337 else ASSERT(false);
1338 }
1339
Nicolas Capens4f172c72016-01-13 08:34:30 -05001340 void PixelProgram::CALLNZb(int labelIndex, int callSiteIndex, const Src &boolRegister)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001341 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001342 Bool condition = (*Pointer<Byte>(data + OFFSET(DrawData, ps.b[boolRegister.index])) != Byte(0)); // FIXME
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001343
1344 if(boolRegister.modifier == Shader::MODIFIER_NOT)
1345 {
1346 condition = !condition;
1347 }
1348
1349 if(!labelBlock[labelIndex])
1350 {
1351 labelBlock[labelIndex] = Nucleus::createBasicBlock();
1352 }
1353
1354 if(callRetBlock[labelIndex].size() > 1)
1355 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001356 callStack[stackIndex++] = UInt(callSiteIndex);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001357 }
1358
Nicolas Capens4f172c72016-01-13 08:34:30 -05001359 Int4 restoreLeave = enableLeave;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001360
1361 branch(condition, labelBlock[labelIndex], callRetBlock[labelIndex][callSiteIndex]);
1362 Nucleus::setInsertBlock(callRetBlock[labelIndex][callSiteIndex]);
1363
Nicolas Capens4f172c72016-01-13 08:34:30 -05001364 enableLeave = restoreLeave;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001365 }
1366
Nicolas Capens4f172c72016-01-13 08:34:30 -05001367 void PixelProgram::CALLNZp(int labelIndex, int callSiteIndex, const Src &predicateRegister)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001368 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001369 Int4 condition = As<Int4>(p0[predicateRegister.swizzle & 0x3]);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001370
1371 if(predicateRegister.modifier == Shader::MODIFIER_NOT)
1372 {
1373 condition = ~condition;
1374 }
1375
Nicolas Capens4f172c72016-01-13 08:34:30 -05001376 condition &= enableStack[enableIndex];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001377
1378 if(!labelBlock[labelIndex])
1379 {
1380 labelBlock[labelIndex] = Nucleus::createBasicBlock();
1381 }
1382
1383 if(callRetBlock[labelIndex].size() > 1)
1384 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001385 callStack[stackIndex++] = UInt(callSiteIndex);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001386 }
1387
Nicolas Capens4f172c72016-01-13 08:34:30 -05001388 enableIndex++;
1389 enableStack[enableIndex] = condition;
1390 Int4 restoreLeave = enableLeave;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001391
1392 Bool notAllFalse = SignMask(condition) != 0;
1393 branch(notAllFalse, labelBlock[labelIndex], callRetBlock[labelIndex][callSiteIndex]);
1394 Nucleus::setInsertBlock(callRetBlock[labelIndex][callSiteIndex]);
1395
Nicolas Capens4f172c72016-01-13 08:34:30 -05001396 enableIndex--;
1397 enableLeave = restoreLeave;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001398 }
1399
Nicolas Capens4f172c72016-01-13 08:34:30 -05001400 void PixelProgram::ELSE()
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001401 {
1402 ifDepth--;
1403
1404 llvm::BasicBlock *falseBlock = ifFalseBlock[ifDepth];
1405 llvm::BasicBlock *endBlock = Nucleus::createBasicBlock();
1406
1407 if(isConditionalIf[ifDepth])
1408 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001409 Int4 condition = ~enableStack[enableIndex] & enableStack[enableIndex - 1];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001410 Bool notAllFalse = SignMask(condition) != 0;
1411
1412 branch(notAllFalse, falseBlock, endBlock);
1413
Nicolas Capens4f172c72016-01-13 08:34:30 -05001414 enableStack[enableIndex] = ~enableStack[enableIndex] & enableStack[enableIndex - 1];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001415 }
1416 else
1417 {
1418 Nucleus::createBr(endBlock);
1419 Nucleus::setInsertBlock(falseBlock);
1420 }
1421
1422 ifFalseBlock[ifDepth] = endBlock;
1423
1424 ifDepth++;
1425 }
1426
Nicolas Capens4f172c72016-01-13 08:34:30 -05001427 void PixelProgram::ENDIF()
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001428 {
1429 ifDepth--;
1430
1431 llvm::BasicBlock *endBlock = ifFalseBlock[ifDepth];
1432
1433 Nucleus::createBr(endBlock);
1434 Nucleus::setInsertBlock(endBlock);
1435
1436 if(isConditionalIf[ifDepth])
1437 {
1438 breakDepth--;
Nicolas Capens4f172c72016-01-13 08:34:30 -05001439 enableIndex--;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001440 }
1441 }
1442
Nicolas Capens4f172c72016-01-13 08:34:30 -05001443 void PixelProgram::ENDLOOP()
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001444 {
1445 loopRepDepth--;
1446
Nicolas Capens4f172c72016-01-13 08:34:30 -05001447 aL[loopDepth] = aL[loopDepth] + increment[loopDepth]; // FIXME: +=
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001448
1449 llvm::BasicBlock *testBlock = loopRepTestBlock[loopRepDepth];
1450 llvm::BasicBlock *endBlock = loopRepEndBlock[loopRepDepth];
1451
1452 Nucleus::createBr(testBlock);
1453 Nucleus::setInsertBlock(endBlock);
1454
Nicolas Capens4f172c72016-01-13 08:34:30 -05001455 loopDepth--;
1456 enableBreak = Int4(0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001457 }
1458
Nicolas Capens4f172c72016-01-13 08:34:30 -05001459 void PixelProgram::ENDREP()
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001460 {
1461 loopRepDepth--;
1462
1463 llvm::BasicBlock *testBlock = loopRepTestBlock[loopRepDepth];
1464 llvm::BasicBlock *endBlock = loopRepEndBlock[loopRepDepth];
1465
1466 Nucleus::createBr(testBlock);
1467 Nucleus::setInsertBlock(endBlock);
1468
Nicolas Capens4f172c72016-01-13 08:34:30 -05001469 loopDepth--;
1470 enableBreak = Int4(0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001471 }
1472
Nicolas Capens4f172c72016-01-13 08:34:30 -05001473 void PixelProgram::ENDWHILE()
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001474 {
1475 loopRepDepth--;
1476
1477 llvm::BasicBlock *testBlock = loopRepTestBlock[loopRepDepth];
1478 llvm::BasicBlock *endBlock = loopRepEndBlock[loopRepDepth];
1479
1480 Nucleus::createBr(testBlock);
1481 Nucleus::setInsertBlock(endBlock);
1482
Nicolas Capens4f172c72016-01-13 08:34:30 -05001483 enableIndex--;
1484 enableBreak = Int4(0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001485 whileTest = false;
1486 }
1487
Alexis Hetu9aa83a92016-05-02 17:34:46 -04001488 void PixelProgram::ENDSWITCH()
1489 {
1490 loopRepDepth--;
1491
1492 llvm::BasicBlock *endBlock = loopRepEndBlock[loopRepDepth];
1493
1494 Nucleus::createBr(loopRepEndBlock[loopRepDepth]);
1495 Nucleus::setInsertBlock(endBlock);
1496
1497 enableIndex--;
1498 enableBreak = Int4(0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF);
1499 }
1500
Nicolas Capens4f172c72016-01-13 08:34:30 -05001501 void PixelProgram::IF(const Src &src)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001502 {
1503 if(src.type == Shader::PARAMETER_CONSTBOOL)
1504 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001505 IFb(src);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001506 }
1507 else if(src.type == Shader::PARAMETER_PREDICATE)
1508 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001509 IFp(src);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001510 }
1511 else
1512 {
Nicolas Capensc2534f42016-04-04 11:13:24 -04001513 Int4 condition = As<Int4>(fetchRegister(src).x);
Nicolas Capens4f172c72016-01-13 08:34:30 -05001514 IF(condition);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001515 }
1516 }
1517
Nicolas Capens4f172c72016-01-13 08:34:30 -05001518 void PixelProgram::IFb(const Src &boolRegister)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001519 {
1520 ASSERT(ifDepth < 24 + 4);
1521
Nicolas Capens4f172c72016-01-13 08:34:30 -05001522 Bool condition = (*Pointer<Byte>(data + OFFSET(DrawData, ps.b[boolRegister.index])) != Byte(0)); // FIXME
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001523
1524 if(boolRegister.modifier == Shader::MODIFIER_NOT)
1525 {
1526 condition = !condition;
1527 }
1528
1529 llvm::BasicBlock *trueBlock = Nucleus::createBasicBlock();
1530 llvm::BasicBlock *falseBlock = Nucleus::createBasicBlock();
1531
1532 branch(condition, trueBlock, falseBlock);
1533
1534 isConditionalIf[ifDepth] = false;
1535 ifFalseBlock[ifDepth] = falseBlock;
1536
1537 ifDepth++;
1538 }
1539
Nicolas Capens4f172c72016-01-13 08:34:30 -05001540 void PixelProgram::IFp(const Src &predicateRegister)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001541 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001542 Int4 condition = As<Int4>(p0[predicateRegister.swizzle & 0x3]);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001543
1544 if(predicateRegister.modifier == Shader::MODIFIER_NOT)
1545 {
1546 condition = ~condition;
1547 }
1548
Nicolas Capens4f172c72016-01-13 08:34:30 -05001549 IF(condition);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001550 }
1551
Nicolas Capens4f172c72016-01-13 08:34:30 -05001552 void PixelProgram::IFC(Vector4f &src0, Vector4f &src1, Control control)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001553 {
1554 Int4 condition;
1555
1556 switch(control)
1557 {
1558 case Shader::CONTROL_GT: condition = CmpNLE(src0.x, src1.x); break;
1559 case Shader::CONTROL_EQ: condition = CmpEQ(src0.x, src1.x); break;
1560 case Shader::CONTROL_GE: condition = CmpNLT(src0.x, src1.x); break;
1561 case Shader::CONTROL_LT: condition = CmpLT(src0.x, src1.x); break;
1562 case Shader::CONTROL_NE: condition = CmpNEQ(src0.x, src1.x); break;
1563 case Shader::CONTROL_LE: condition = CmpLE(src0.x, src1.x); break;
1564 default:
1565 ASSERT(false);
1566 }
1567
Nicolas Capens4f172c72016-01-13 08:34:30 -05001568 IF(condition);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001569 }
1570
Nicolas Capens4f172c72016-01-13 08:34:30 -05001571 void PixelProgram::IF(Int4 &condition)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001572 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001573 condition &= enableStack[enableIndex];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001574
Nicolas Capens4f172c72016-01-13 08:34:30 -05001575 enableIndex++;
1576 enableStack[enableIndex] = condition;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001577
1578 llvm::BasicBlock *trueBlock = Nucleus::createBasicBlock();
1579 llvm::BasicBlock *falseBlock = Nucleus::createBasicBlock();
1580
1581 Bool notAllFalse = SignMask(condition) != 0;
1582
1583 branch(notAllFalse, trueBlock, falseBlock);
1584
1585 isConditionalIf[ifDepth] = true;
1586 ifFalseBlock[ifDepth] = falseBlock;
1587
1588 ifDepth++;
1589 breakDepth++;
1590 }
1591
1592 void PixelProgram::LABEL(int labelIndex)
1593 {
1594 if(!labelBlock[labelIndex])
1595 {
1596 labelBlock[labelIndex] = Nucleus::createBasicBlock();
1597 }
1598
1599 Nucleus::setInsertBlock(labelBlock[labelIndex]);
1600 currentLabel = labelIndex;
1601 }
1602
Nicolas Capens4f172c72016-01-13 08:34:30 -05001603 void PixelProgram::LOOP(const Src &integerRegister)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001604 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001605 loopDepth++;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001606
Nicolas Capens4f172c72016-01-13 08:34:30 -05001607 iteration[loopDepth] = *Pointer<Int>(data + OFFSET(DrawData, ps.i[integerRegister.index][0]));
1608 aL[loopDepth] = *Pointer<Int>(data + OFFSET(DrawData, ps.i[integerRegister.index][1]));
1609 increment[loopDepth] = *Pointer<Int>(data + OFFSET(DrawData, ps.i[integerRegister.index][2]));
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001610
Nicolas Capens4f172c72016-01-13 08:34:30 -05001611 // If(increment[loopDepth] == 0)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001612 // {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001613 // increment[loopDepth] = 1;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001614 // }
1615
1616 llvm::BasicBlock *loopBlock = Nucleus::createBasicBlock();
1617 llvm::BasicBlock *testBlock = Nucleus::createBasicBlock();
1618 llvm::BasicBlock *endBlock = Nucleus::createBasicBlock();
1619
1620 loopRepTestBlock[loopRepDepth] = testBlock;
1621 loopRepEndBlock[loopRepDepth] = endBlock;
1622
1623 // FIXME: jump(testBlock)
1624 Nucleus::createBr(testBlock);
1625 Nucleus::setInsertBlock(testBlock);
1626
Nicolas Capens4f172c72016-01-13 08:34:30 -05001627 branch(iteration[loopDepth] > 0, loopBlock, endBlock);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001628 Nucleus::setInsertBlock(loopBlock);
1629
Nicolas Capens4f172c72016-01-13 08:34:30 -05001630 iteration[loopDepth] = iteration[loopDepth] - 1; // FIXME: --
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001631
1632 loopRepDepth++;
1633 breakDepth = 0;
1634 }
1635
Nicolas Capens4f172c72016-01-13 08:34:30 -05001636 void PixelProgram::REP(const Src &integerRegister)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001637 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001638 loopDepth++;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001639
Nicolas Capens4f172c72016-01-13 08:34:30 -05001640 iteration[loopDepth] = *Pointer<Int>(data + OFFSET(DrawData, ps.i[integerRegister.index][0]));
1641 aL[loopDepth] = aL[loopDepth - 1];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001642
1643 llvm::BasicBlock *loopBlock = Nucleus::createBasicBlock();
1644 llvm::BasicBlock *testBlock = Nucleus::createBasicBlock();
1645 llvm::BasicBlock *endBlock = Nucleus::createBasicBlock();
1646
1647 loopRepTestBlock[loopRepDepth] = testBlock;
1648 loopRepEndBlock[loopRepDepth] = endBlock;
1649
1650 // FIXME: jump(testBlock)
1651 Nucleus::createBr(testBlock);
1652 Nucleus::setInsertBlock(testBlock);
1653
Nicolas Capens4f172c72016-01-13 08:34:30 -05001654 branch(iteration[loopDepth] > 0, loopBlock, endBlock);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001655 Nucleus::setInsertBlock(loopBlock);
1656
Nicolas Capens4f172c72016-01-13 08:34:30 -05001657 iteration[loopDepth] = iteration[loopDepth] - 1; // FIXME: --
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001658
1659 loopRepDepth++;
1660 breakDepth = 0;
1661 }
1662
Nicolas Capens4f172c72016-01-13 08:34:30 -05001663 void PixelProgram::WHILE(const Src &temporaryRegister)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001664 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001665 enableIndex++;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001666
1667 llvm::BasicBlock *loopBlock = Nucleus::createBasicBlock();
1668 llvm::BasicBlock *testBlock = Nucleus::createBasicBlock();
1669 llvm::BasicBlock *endBlock = Nucleus::createBasicBlock();
1670
1671 loopRepTestBlock[loopRepDepth] = testBlock;
1672 loopRepEndBlock[loopRepDepth] = endBlock;
1673
Nicolas Capens4f172c72016-01-13 08:34:30 -05001674 Int4 restoreBreak = enableBreak;
1675 Int4 restoreContinue = enableContinue;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001676
1677 // FIXME: jump(testBlock)
1678 Nucleus::createBr(testBlock);
1679 Nucleus::setInsertBlock(testBlock);
Nicolas Capens4f172c72016-01-13 08:34:30 -05001680 enableContinue = restoreContinue;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001681
Nicolas Capensc2534f42016-04-04 11:13:24 -04001682 const Vector4f &src = fetchRegister(temporaryRegister);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001683 Int4 condition = As<Int4>(src.x);
Nicolas Capens4f172c72016-01-13 08:34:30 -05001684 condition &= enableStack[enableIndex - 1];
Nicolas Capens2ff29482016-04-28 15:28:02 -04001685 if(shader->containsLeaveInstruction()) condition &= enableLeave;
Nicolas Capens4f172c72016-01-13 08:34:30 -05001686 enableStack[enableIndex] = condition;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001687
1688 Bool notAllFalse = SignMask(condition) != 0;
1689 branch(notAllFalse, loopBlock, endBlock);
1690
1691 Nucleus::setInsertBlock(endBlock);
Nicolas Capens4f172c72016-01-13 08:34:30 -05001692 enableBreak = restoreBreak;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001693
1694 Nucleus::setInsertBlock(loopBlock);
1695
1696 loopRepDepth++;
1697 breakDepth = 0;
1698 }
1699
Alexis Hetu9aa83a92016-05-02 17:34:46 -04001700 void PixelProgram::SWITCH()
1701 {
1702 enableIndex++;
1703 enableStack[enableIndex] = Int4(0xFFFFFFFF);
1704
1705 llvm::BasicBlock *endBlock = Nucleus::createBasicBlock();
1706
1707 loopRepTestBlock[loopRepDepth] = nullptr;
1708 loopRepEndBlock[loopRepDepth] = endBlock;
1709
1710 loopRepDepth++;
1711 breakDepth = 0;
1712 }
1713
Nicolas Capens4f172c72016-01-13 08:34:30 -05001714 void PixelProgram::RET()
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001715 {
1716 if(currentLabel == -1)
1717 {
1718 returnBlock = Nucleus::createBasicBlock();
1719 Nucleus::createBr(returnBlock);
1720 }
1721 else
1722 {
1723 llvm::BasicBlock *unreachableBlock = Nucleus::createBasicBlock();
1724
1725 if(callRetBlock[currentLabel].size() > 1) // Pop the return destination from the call stack
1726 {
1727 // FIXME: Encapsulate
Nicolas Capens4f172c72016-01-13 08:34:30 -05001728 UInt index = callStack[--stackIndex];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001729
1730 llvm::Value *value = index.loadValue();
1731 llvm::Value *switchInst = Nucleus::createSwitch(value, unreachableBlock, (int)callRetBlock[currentLabel].size());
1732
1733 for(unsigned int i = 0; i < callRetBlock[currentLabel].size(); i++)
1734 {
1735 Nucleus::addSwitchCase(switchInst, i, callRetBlock[currentLabel][i]);
1736 }
1737 }
1738 else if(callRetBlock[currentLabel].size() == 1) // Jump directly to the unique return destination
1739 {
1740 Nucleus::createBr(callRetBlock[currentLabel][0]);
1741 }
1742 else // Function isn't called
1743 {
1744 Nucleus::createBr(unreachableBlock);
1745 }
1746
1747 Nucleus::setInsertBlock(unreachableBlock);
1748 Nucleus::createUnreachable();
1749 }
1750 }
1751
Nicolas Capens4f172c72016-01-13 08:34:30 -05001752 void PixelProgram::LEAVE()
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001753 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001754 enableLeave = enableLeave & ~enableStack[enableIndex];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001755
1756 // FIXME: Return from function if all instances left
1757 // FIXME: Use enableLeave in other control-flow constructs
1758 }
Nicolas Capens5d961882016-01-01 23:18:14 -05001759}