blob: 013dcca6e859e7830c1834115f42a77b2a1d4848 [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"
Nicolas Capens708c24b2017-10-26 13:07:10 -040016
Nicolas Capens5d961882016-01-01 23:18:14 -050017#include "SamplerCore.hpp"
Nicolas Capens708c24b2017-10-26 13:07:10 -040018#include "Renderer/Primitive.hpp"
19#include "Renderer/Renderer.hpp"
Nicolas Capens5d961882016-01-01 23:18:14 -050020
21namespace sw
22{
23 extern bool postBlendSRGB;
Alexis Hetuf2a8c372015-07-13 11:08:41 -040024 extern bool booleanFaceRegister;
25 extern bool halfIntegerCoordinates; // Pixel centers are not at integer coordinates
Nicolas Capens5d961882016-01-01 23:18:14 -050026 extern bool fullPixelPositionRegister;
27
Nicolas Capens4f172c72016-01-13 08:34:30 -050028 void PixelProgram::setBuiltins(Int &x, Int &y, Float4(&z)[4], Float4 &w)
Nicolas Capens5d961882016-01-01 23:18:14 -050029 {
Alexis Hetu53ad4af2017-12-06 14:49:07 -050030 if(shader->getShaderModel() >= 0x0300)
Alexis Hetuf2a8c372015-07-13 11:08:41 -040031 {
Alexis Hetu02ad0aa2016-08-02 11:18:14 -040032 if(shader->isVPosDeclared())
Alexis Hetuf2a8c372015-07-13 11:08:41 -040033 {
34 if(!halfIntegerCoordinates)
35 {
Nicolas Capens4f172c72016-01-13 08:34:30 -050036 vPos.x = Float4(Float(x)) + Float4(0, 1, 0, 1);
37 vPos.y = Float4(Float(y)) + Float4(0, 0, 1, 1);
Alexis Hetuf2a8c372015-07-13 11:08:41 -040038 }
39 else
40 {
Nicolas Capens4f172c72016-01-13 08:34:30 -050041 vPos.x = Float4(Float(x)) + Float4(0.5f, 1.5f, 0.5f, 1.5f);
42 vPos.y = Float4(Float(y)) + Float4(0.5f, 0.5f, 1.5f, 1.5f);
Alexis Hetuf2a8c372015-07-13 11:08:41 -040043 }
44
45 if(fullPixelPositionRegister)
46 {
Nicolas Capens4f172c72016-01-13 08:34:30 -050047 vPos.z = z[0]; // FIXME: Centroid?
48 vPos.w = w; // FIXME: Centroid?
Alexis Hetuf2a8c372015-07-13 11:08:41 -040049 }
50 }
51
Alexis Hetu02ad0aa2016-08-02 11:18:14 -040052 if(shader->isVFaceDeclared())
Alexis Hetuf2a8c372015-07-13 11:08:41 -040053 {
Nicolas Capens4f172c72016-01-13 08:34:30 -050054 Float4 area = *Pointer<Float>(primitive + OFFSET(Primitive, area));
Alexis Hetuf2a8c372015-07-13 11:08:41 -040055 Float4 face = booleanFaceRegister ? Float4(As<Float4>(CmpNLT(area, Float4(0.0f)))) : area;
56
Nicolas Capens4f172c72016-01-13 08:34:30 -050057 vFace.x = face;
58 vFace.y = face;
59 vFace.z = face;
60 vFace.w = face;
Alexis Hetuf2a8c372015-07-13 11:08:41 -040061 }
Nicolas Capens5d961882016-01-01 23:18:14 -050062 }
63 }
64
Nicolas Capens4f172c72016-01-13 08:34:30 -050065 void PixelProgram::applyShader(Int cMask[4])
Nicolas Capens5d961882016-01-01 23:18:14 -050066 {
Nicolas Capens4f172c72016-01-13 08:34:30 -050067 enableIndex = 0;
68 stackIndex = 0;
Alexis Hetuf2a8c372015-07-13 11:08:41 -040069
70 if(shader->containsLeaveInstruction())
71 {
Nicolas Capens4f172c72016-01-13 08:34:30 -050072 enableLeave = Int4(0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF);
Alexis Hetuf2a8c372015-07-13 11:08:41 -040073 }
74
Alexis Hetu1edcd8b2015-11-05 11:12:41 -050075 for(int i = 0; i < RENDERTARGETS; i++)
76 {
77 if(state.targetFormat[i] != FORMAT_NULL)
78 {
Nicolas Capens4f172c72016-01-13 08:34:30 -050079 oC[i] = Vector4f(0.0f, 0.0f, 0.0f, 0.0f);
Alexis Hetu1edcd8b2015-11-05 11:12:41 -050080 }
81 }
Alexis Hetuf2a8c372015-07-13 11:08:41 -040082
83 // Create all call site return blocks up front
84 for(size_t i = 0; i < shader->getLength(); i++)
85 {
86 const Shader::Instruction *instruction = shader->getInstruction(i);
87 Shader::Opcode opcode = instruction->opcode;
88
89 if(opcode == Shader::OPCODE_CALL || opcode == Shader::OPCODE_CALLNZ)
90 {
91 const Dst &dst = instruction->dst;
92
93 ASSERT(callRetBlock[dst.label].size() == dst.callSite);
94 callRetBlock[dst.label].push_back(Nucleus::createBasicBlock());
95 }
96 }
97
Nicolas Capens8b124c12016-04-18 14:09:37 -040098 bool broadcastColor0 = true;
99
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400100 for(size_t i = 0; i < shader->getLength(); i++)
101 {
102 const Shader::Instruction *instruction = shader->getInstruction(i);
103 Shader::Opcode opcode = instruction->opcode;
104
105 if(opcode == Shader::OPCODE_DCL || opcode == Shader::OPCODE_DEF || opcode == Shader::OPCODE_DEFI || opcode == Shader::OPCODE_DEFB)
106 {
107 continue;
108 }
109
110 const Dst &dst = instruction->dst;
111 const Src &src0 = instruction->src[0];
112 const Src &src1 = instruction->src[1];
113 const Src &src2 = instruction->src[2];
114 const Src &src3 = instruction->src[3];
Alexis Hetu25d47fc2015-10-22 13:58:32 -0400115 const Src &src4 = instruction->src[4];
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400116
117 bool predicate = instruction->predicate;
118 Control control = instruction->control;
119 bool pp = dst.partialPrecision;
120 bool project = instruction->project;
121 bool bias = instruction->bias;
122
123 Vector4f d;
124 Vector4f s0;
125 Vector4f s1;
126 Vector4f s2;
127 Vector4f s3;
Alexis Hetu25d47fc2015-10-22 13:58:32 -0400128 Vector4f s4;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400129
130 if(opcode == Shader::OPCODE_TEXKILL) // Takes destination as input
131 {
132 if(dst.type == Shader::PARAMETER_TEXTURE)
133 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500134 d.x = v[2 + dst.index].x;
135 d.y = v[2 + dst.index].y;
136 d.z = v[2 + dst.index].z;
137 d.w = v[2 + dst.index].w;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400138 }
139 else
140 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500141 d = r[dst.index];
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400142 }
143 }
144
Nicolas Capensc2534f42016-04-04 11:13:24 -0400145 if(src0.type != Shader::PARAMETER_VOID) s0 = fetchRegister(src0);
146 if(src1.type != Shader::PARAMETER_VOID) s1 = fetchRegister(src1);
147 if(src2.type != Shader::PARAMETER_VOID) s2 = fetchRegister(src2);
148 if(src3.type != Shader::PARAMETER_VOID) s3 = fetchRegister(src3);
149 if(src4.type != Shader::PARAMETER_VOID) s4 = fetchRegister(src4);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400150
151 switch(opcode)
152 {
153 case Shader::OPCODE_PS_2_0: break;
154 case Shader::OPCODE_PS_2_x: break;
155 case Shader::OPCODE_PS_3_0: break;
156 case Shader::OPCODE_DEF: break;
157 case Shader::OPCODE_DCL: break;
158 case Shader::OPCODE_NOP: break;
159 case Shader::OPCODE_MOV: mov(d, s0); break;
Alexis Hetu8d78cf72015-08-28 14:24:45 -0400160 case Shader::OPCODE_NEG: neg(d, s0); break;
161 case Shader::OPCODE_INEG: ineg(d, s0); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400162 case Shader::OPCODE_F2B: f2b(d, s0); break;
163 case Shader::OPCODE_B2F: b2f(d, s0); break;
Alexis Hetu8d78cf72015-08-28 14:24:45 -0400164 case Shader::OPCODE_F2I: f2i(d, s0); break;
165 case Shader::OPCODE_I2F: i2f(d, s0); break;
166 case Shader::OPCODE_F2U: f2u(d, s0); break;
167 case Shader::OPCODE_U2F: u2f(d, s0); break;
168 case Shader::OPCODE_I2B: i2b(d, s0); break;
169 case Shader::OPCODE_B2I: b2i(d, s0); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400170 case Shader::OPCODE_ADD: add(d, s0, s1); break;
Alexis Hetu8d78cf72015-08-28 14:24:45 -0400171 case Shader::OPCODE_IADD: iadd(d, s0, s1); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400172 case Shader::OPCODE_SUB: sub(d, s0, s1); break;
Alexis Hetu8d78cf72015-08-28 14:24:45 -0400173 case Shader::OPCODE_ISUB: isub(d, s0, s1); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400174 case Shader::OPCODE_MUL: mul(d, s0, s1); break;
Alexis Hetu8d78cf72015-08-28 14:24:45 -0400175 case Shader::OPCODE_IMUL: imul(d, s0, s1); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400176 case Shader::OPCODE_MAD: mad(d, s0, s1, s2); break;
Alexis Hetu8d78cf72015-08-28 14:24:45 -0400177 case Shader::OPCODE_IMAD: imad(d, s0, s1, s2); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400178 case Shader::OPCODE_DP1: dp1(d, s0, s1); break;
179 case Shader::OPCODE_DP2: dp2(d, s0, s1); break;
180 case Shader::OPCODE_DP2ADD: dp2add(d, s0, s1, s2); break;
181 case Shader::OPCODE_DP3: dp3(d, s0, s1); break;
182 case Shader::OPCODE_DP4: dp4(d, s0, s1); break;
Alexis Hetuc3d95f32015-09-23 12:27:32 -0400183 case Shader::OPCODE_DET2: det2(d, s0, s1); break;
184 case Shader::OPCODE_DET3: det3(d, s0, s1, s2); break;
185 case Shader::OPCODE_DET4: det4(d, s0, s1, s2, s3); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400186 case Shader::OPCODE_CMP0: cmp0(d, s0, s1, s2); break;
187 case Shader::OPCODE_ICMP: icmp(d, s0, s1, control); break;
Alexis Hetu8d78cf72015-08-28 14:24:45 -0400188 case Shader::OPCODE_UCMP: ucmp(d, s0, s1, control); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400189 case Shader::OPCODE_SELECT: select(d, s0, s1, s2); break;
190 case Shader::OPCODE_EXTRACT: extract(d.x, s0, s1.x); break;
191 case Shader::OPCODE_INSERT: insert(d, s0, s1.x, s2.x); break;
192 case Shader::OPCODE_FRC: frc(d, s0); break;
193 case Shader::OPCODE_TRUNC: trunc(d, s0); break;
194 case Shader::OPCODE_FLOOR: floor(d, s0); break;
195 case Shader::OPCODE_ROUND: round(d, s0); break;
196 case Shader::OPCODE_ROUNDEVEN: roundEven(d, s0); break;
197 case Shader::OPCODE_CEIL: ceil(d, s0); break;
198 case Shader::OPCODE_EXP2X: exp2x(d, s0, pp); break;
199 case Shader::OPCODE_EXP2: exp2(d, s0, pp); break;
200 case Shader::OPCODE_LOG2X: log2x(d, s0, pp); break;
201 case Shader::OPCODE_LOG2: log2(d, s0, pp); break;
202 case Shader::OPCODE_EXP: exp(d, s0, pp); break;
203 case Shader::OPCODE_LOG: log(d, s0, pp); break;
204 case Shader::OPCODE_RCPX: rcpx(d, s0, pp); break;
205 case Shader::OPCODE_DIV: div(d, s0, s1); break;
Alexis Hetu8d78cf72015-08-28 14:24:45 -0400206 case Shader::OPCODE_IDIV: idiv(d, s0, s1); break;
207 case Shader::OPCODE_UDIV: udiv(d, s0, s1); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400208 case Shader::OPCODE_MOD: mod(d, s0, s1); break;
Alexis Hetu8d78cf72015-08-28 14:24:45 -0400209 case Shader::OPCODE_IMOD: imod(d, s0, s1); break;
210 case Shader::OPCODE_UMOD: umod(d, s0, s1); break;
211 case Shader::OPCODE_SHL: shl(d, s0, s1); break;
212 case Shader::OPCODE_ISHR: ishr(d, s0, s1); break;
213 case Shader::OPCODE_USHR: ushr(d, s0, s1); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400214 case Shader::OPCODE_RSQX: rsqx(d, s0, pp); break;
215 case Shader::OPCODE_SQRT: sqrt(d, s0, pp); break;
216 case Shader::OPCODE_RSQ: rsq(d, s0, pp); break;
217 case Shader::OPCODE_LEN2: len2(d.x, s0, pp); break;
218 case Shader::OPCODE_LEN3: len3(d.x, s0, pp); break;
219 case Shader::OPCODE_LEN4: len4(d.x, s0, pp); break;
220 case Shader::OPCODE_DIST1: dist1(d.x, s0, s1, pp); break;
221 case Shader::OPCODE_DIST2: dist2(d.x, s0, s1, pp); break;
222 case Shader::OPCODE_DIST3: dist3(d.x, s0, s1, pp); break;
223 case Shader::OPCODE_DIST4: dist4(d.x, s0, s1, pp); break;
224 case Shader::OPCODE_MIN: min(d, s0, s1); break;
Alexis Hetu8d78cf72015-08-28 14:24:45 -0400225 case Shader::OPCODE_IMIN: imin(d, s0, s1); break;
226 case Shader::OPCODE_UMIN: umin(d, s0, s1); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400227 case Shader::OPCODE_MAX: max(d, s0, s1); break;
Alexis Hetu8d78cf72015-08-28 14:24:45 -0400228 case Shader::OPCODE_IMAX: imax(d, s0, s1); break;
229 case Shader::OPCODE_UMAX: umax(d, s0, s1); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400230 case Shader::OPCODE_LRP: lrp(d, s0, s1, s2); break;
231 case Shader::OPCODE_STEP: step(d, s0, s1); break;
232 case Shader::OPCODE_SMOOTH: smooth(d, s0, s1, s2); break;
Alexis Hetu8ef6d102017-11-09 15:49:09 -0500233 case Shader::OPCODE_ISINF: isinf(d, s0); break;
234 case Shader::OPCODE_ISNAN: isnan(d, s0); break;
Alexis Hetu8d78cf72015-08-28 14:24:45 -0400235 case Shader::OPCODE_FLOATBITSTOINT:
236 case Shader::OPCODE_FLOATBITSTOUINT:
237 case Shader::OPCODE_INTBITSTOFLOAT:
238 case Shader::OPCODE_UINTBITSTOFLOAT: d = s0; break;
Alexis Hetu9cde9742016-04-06 13:03:38 -0400239 case Shader::OPCODE_PACKSNORM2x16: packSnorm2x16(d, s0); break;
240 case Shader::OPCODE_PACKUNORM2x16: packUnorm2x16(d, s0); break;
Alexis Hetuffb35eb2016-04-06 18:05:00 -0400241 case Shader::OPCODE_PACKHALF2x16: packHalf2x16(d, s0); break;
Alexis Hetu9cde9742016-04-06 13:03:38 -0400242 case Shader::OPCODE_UNPACKSNORM2x16: unpackSnorm2x16(d, s0); break;
243 case Shader::OPCODE_UNPACKUNORM2x16: unpackUnorm2x16(d, s0); break;
Alexis Hetuffb35eb2016-04-06 18:05:00 -0400244 case Shader::OPCODE_UNPACKHALF2x16: unpackHalf2x16(d, s0); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400245 case Shader::OPCODE_POWX: powx(d, s0, s1, pp); break;
246 case Shader::OPCODE_POW: pow(d, s0, s1, pp); break;
247 case Shader::OPCODE_SGN: sgn(d, s0); break;
Alexis Hetu0f448072016-03-18 10:56:08 -0400248 case Shader::OPCODE_ISGN: isgn(d, s0); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400249 case Shader::OPCODE_CRS: crs(d, s0, s1); break;
250 case Shader::OPCODE_FORWARD1: forward1(d, s0, s1, s2); break;
251 case Shader::OPCODE_FORWARD2: forward2(d, s0, s1, s2); break;
252 case Shader::OPCODE_FORWARD3: forward3(d, s0, s1, s2); break;
253 case Shader::OPCODE_FORWARD4: forward4(d, s0, s1, s2); break;
254 case Shader::OPCODE_REFLECT1: reflect1(d, s0, s1); break;
255 case Shader::OPCODE_REFLECT2: reflect2(d, s0, s1); break;
256 case Shader::OPCODE_REFLECT3: reflect3(d, s0, s1); break;
257 case Shader::OPCODE_REFLECT4: reflect4(d, s0, s1); break;
258 case Shader::OPCODE_REFRACT1: refract1(d, s0, s1, s2.x); break;
259 case Shader::OPCODE_REFRACT2: refract2(d, s0, s1, s2.x); break;
260 case Shader::OPCODE_REFRACT3: refract3(d, s0, s1, s2.x); break;
261 case Shader::OPCODE_REFRACT4: refract4(d, s0, s1, s2.x); break;
262 case Shader::OPCODE_NRM2: nrm2(d, s0, pp); break;
263 case Shader::OPCODE_NRM3: nrm3(d, s0, pp); break;
264 case Shader::OPCODE_NRM4: nrm4(d, s0, pp); break;
265 case Shader::OPCODE_ABS: abs(d, s0); break;
Alexis Hetu0f448072016-03-18 10:56:08 -0400266 case Shader::OPCODE_IABS: iabs(d, s0); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400267 case Shader::OPCODE_SINCOS: sincos(d, s0, pp); break;
268 case Shader::OPCODE_COS: cos(d, s0, pp); break;
269 case Shader::OPCODE_SIN: sin(d, s0, pp); break;
270 case Shader::OPCODE_TAN: tan(d, s0, pp); break;
271 case Shader::OPCODE_ACOS: acos(d, s0, pp); break;
272 case Shader::OPCODE_ASIN: asin(d, s0, pp); break;
273 case Shader::OPCODE_ATAN: atan(d, s0, pp); break;
274 case Shader::OPCODE_ATAN2: atan2(d, s0, s1, pp); break;
275 case Shader::OPCODE_COSH: cosh(d, s0, pp); break;
276 case Shader::OPCODE_SINH: sinh(d, s0, pp); break;
277 case Shader::OPCODE_TANH: tanh(d, s0, pp); break;
278 case Shader::OPCODE_ACOSH: acosh(d, s0, pp); break;
279 case Shader::OPCODE_ASINH: asinh(d, s0, pp); break;
280 case Shader::OPCODE_ATANH: atanh(d, s0, pp); break;
Nicolas Capens4f172c72016-01-13 08:34:30 -0500281 case Shader::OPCODE_M4X4: M4X4(d, s0, src1); break;
282 case Shader::OPCODE_M4X3: M4X3(d, s0, src1); break;
283 case Shader::OPCODE_M3X4: M3X4(d, s0, src1); break;
284 case Shader::OPCODE_M3X3: M3X3(d, s0, src1); break;
285 case Shader::OPCODE_M3X2: M3X2(d, s0, src1); break;
Nicolas Capensa0b57832017-11-07 13:07:53 -0500286 case Shader::OPCODE_TEX: TEX(d, s0, src1, project, bias); break;
287 case Shader::OPCODE_TEXLDD: TEXGRAD(d, s0, src1, s2, s3); break;
288 case Shader::OPCODE_TEXLDL: TEXLOD(d, s0, src1, s0.w); break;
289 case Shader::OPCODE_TEXLOD: TEXLOD(d, s0, src1, s2.x); break;
Nicolas Capens4f172c72016-01-13 08:34:30 -0500290 case Shader::OPCODE_TEXSIZE: TEXSIZE(d, s0.x, src1); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400291 case Shader::OPCODE_TEXKILL: TEXKILL(cMask, d, dst.mask); break;
Nicolas Capensa0b57832017-11-07 13:07:53 -0500292 case Shader::OPCODE_TEXOFFSET: TEXOFFSET(d, s0, src1, s2); break;
293 case Shader::OPCODE_TEXLODOFFSET: TEXLODOFFSET(d, s0, src1, s2, s3.x); break;
294 case Shader::OPCODE_TEXELFETCH: TEXELFETCH(d, s0, src1, s2.x); break;
295 case Shader::OPCODE_TEXELFETCHOFFSET: TEXELFETCHOFFSET(d, s0, src1, s2, s3.x); break;
Nicolas Capens4f172c72016-01-13 08:34:30 -0500296 case Shader::OPCODE_TEXGRAD: TEXGRAD(d, s0, src1, s2, s3); break;
Nicolas Capensa0b57832017-11-07 13:07:53 -0500297 case Shader::OPCODE_TEXGRADOFFSET: TEXGRADOFFSET(d, s0, src1, s2, s3, s4); break;
298 case Shader::OPCODE_TEXBIAS: TEXBIAS(d, s0, src1, s2.x); break;
299 case Shader::OPCODE_TEXOFFSETBIAS: TEXOFFSETBIAS(d, s0, src1, s2, s3.x); break;
Nicolas Capens4f172c72016-01-13 08:34:30 -0500300 case Shader::OPCODE_DISCARD: DISCARD(cMask, instruction); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400301 case Shader::OPCODE_DFDX: DFDX(d, s0); break;
302 case Shader::OPCODE_DFDY: DFDY(d, s0); break;
303 case Shader::OPCODE_FWIDTH: FWIDTH(d, s0); break;
Nicolas Capens4f172c72016-01-13 08:34:30 -0500304 case Shader::OPCODE_BREAK: BREAK(); break;
305 case Shader::OPCODE_BREAKC: BREAKC(s0, s1, control); break;
306 case Shader::OPCODE_BREAKP: BREAKP(src0); break;
307 case Shader::OPCODE_CONTINUE: CONTINUE(); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400308 case Shader::OPCODE_TEST: TEST(); break;
Nicolas Capens4f172c72016-01-13 08:34:30 -0500309 case Shader::OPCODE_CALL: CALL(dst.label, dst.callSite); break;
310 case Shader::OPCODE_CALLNZ: CALLNZ(dst.label, dst.callSite, src0); break;
311 case Shader::OPCODE_ELSE: ELSE(); break;
312 case Shader::OPCODE_ENDIF: ENDIF(); break;
313 case Shader::OPCODE_ENDLOOP: ENDLOOP(); break;
314 case Shader::OPCODE_ENDREP: ENDREP(); break;
315 case Shader::OPCODE_ENDWHILE: ENDWHILE(); break;
Alexis Hetu9aa83a92016-05-02 17:34:46 -0400316 case Shader::OPCODE_ENDSWITCH: ENDSWITCH(); break;
Nicolas Capens4f172c72016-01-13 08:34:30 -0500317 case Shader::OPCODE_IF: IF(src0); break;
318 case Shader::OPCODE_IFC: IFC(s0, s1, control); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400319 case Shader::OPCODE_LABEL: LABEL(dst.index); break;
Nicolas Capens4f172c72016-01-13 08:34:30 -0500320 case Shader::OPCODE_LOOP: LOOP(src1); break;
321 case Shader::OPCODE_REP: REP(src0); break;
322 case Shader::OPCODE_WHILE: WHILE(src0); break;
Alexis Hetu9aa83a92016-05-02 17:34:46 -0400323 case Shader::OPCODE_SWITCH: SWITCH(); break;
Nicolas Capens4f172c72016-01-13 08:34:30 -0500324 case Shader::OPCODE_RET: RET(); break;
325 case Shader::OPCODE_LEAVE: LEAVE(); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400326 case Shader::OPCODE_CMP: cmp(d, s0, s1, control); break;
327 case Shader::OPCODE_ALL: all(d.x, s0); break;
328 case Shader::OPCODE_ANY: any(d.x, s0); break;
Alexis Hetu24f454e2016-08-31 17:22:13 -0400329 case Shader::OPCODE_NOT: bitwise_not(d, s0); break;
330 case Shader::OPCODE_OR: bitwise_or(d, s0, s1); break;
331 case Shader::OPCODE_XOR: bitwise_xor(d, s0, s1); break;
332 case Shader::OPCODE_AND: bitwise_and(d, s0, s1); break;
Alexis Hetu8d78cf72015-08-28 14:24:45 -0400333 case Shader::OPCODE_EQ: equal(d, s0, s1); break;
334 case Shader::OPCODE_NE: notEqual(d, s0, s1); break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400335 case Shader::OPCODE_END: break;
336 default:
337 ASSERT(false);
338 }
339
340 if(dst.type != Shader::PARAMETER_VOID && dst.type != Shader::PARAMETER_LABEL && opcode != Shader::OPCODE_TEXKILL && opcode != Shader::OPCODE_NOP)
341 {
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400342 if(dst.saturate)
343 {
344 if(dst.x) d.x = Max(d.x, Float4(0.0f));
345 if(dst.y) d.y = Max(d.y, Float4(0.0f));
346 if(dst.z) d.z = Max(d.z, Float4(0.0f));
347 if(dst.w) d.w = Max(d.w, Float4(0.0f));
348
349 if(dst.x) d.x = Min(d.x, Float4(1.0f));
350 if(dst.y) d.y = Min(d.y, Float4(1.0f));
351 if(dst.z) d.z = Min(d.z, Float4(1.0f));
352 if(dst.w) d.w = Min(d.w, Float4(1.0f));
353 }
354
355 if(instruction->isPredicated())
356 {
357 Vector4f pDst; // FIXME: Rename
358
359 switch(dst.type)
360 {
361 case Shader::PARAMETER_TEMP:
362 if(dst.rel.type == Shader::PARAMETER_VOID)
363 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500364 if(dst.x) pDst.x = r[dst.index].x;
365 if(dst.y) pDst.y = r[dst.index].y;
366 if(dst.z) pDst.z = r[dst.index].z;
367 if(dst.w) pDst.w = r[dst.index].w;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400368 }
369 else
370 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500371 Int a = relativeAddress(dst);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400372
Nicolas Capens4f172c72016-01-13 08:34:30 -0500373 if(dst.x) pDst.x = r[dst.index + a].x;
374 if(dst.y) pDst.y = r[dst.index + a].y;
375 if(dst.z) pDst.z = r[dst.index + a].z;
376 if(dst.w) pDst.w = r[dst.index + a].w;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400377 }
378 break;
379 case Shader::PARAMETER_COLOROUT:
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500380 if(dst.rel.type == Shader::PARAMETER_VOID)
381 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500382 if(dst.x) pDst.x = oC[dst.index].x;
383 if(dst.y) pDst.y = oC[dst.index].y;
384 if(dst.z) pDst.z = oC[dst.index].z;
385 if(dst.w) pDst.w = oC[dst.index].w;
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500386 }
387 else
388 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500389 Int a = relativeAddress(dst) + dst.index;
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500390
Nicolas Capens4f172c72016-01-13 08:34:30 -0500391 if(dst.x) pDst.x = oC[a].x;
392 if(dst.y) pDst.y = oC[a].y;
393 if(dst.z) pDst.z = oC[a].z;
394 if(dst.w) pDst.w = oC[a].w;
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500395 }
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400396 break;
397 case Shader::PARAMETER_PREDICATE:
Nicolas Capens4f172c72016-01-13 08:34:30 -0500398 if(dst.x) pDst.x = p0.x;
399 if(dst.y) pDst.y = p0.y;
400 if(dst.z) pDst.z = p0.z;
401 if(dst.w) pDst.w = p0.w;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400402 break;
403 case Shader::PARAMETER_DEPTHOUT:
Nicolas Capens4f172c72016-01-13 08:34:30 -0500404 pDst.x = oDepth;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400405 break;
406 default:
407 ASSERT(false);
408 }
409
Nicolas Capens4f172c72016-01-13 08:34:30 -0500410 Int4 enable = enableMask(instruction);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400411
412 Int4 xEnable = enable;
413 Int4 yEnable = enable;
414 Int4 zEnable = enable;
415 Int4 wEnable = enable;
416
417 if(predicate)
418 {
419 unsigned char pSwizzle = instruction->predicateSwizzle;
420
Nicolas Capens4f172c72016-01-13 08:34:30 -0500421 Float4 xPredicate = p0[(pSwizzle >> 0) & 0x03];
422 Float4 yPredicate = p0[(pSwizzle >> 2) & 0x03];
423 Float4 zPredicate = p0[(pSwizzle >> 4) & 0x03];
424 Float4 wPredicate = p0[(pSwizzle >> 6) & 0x03];
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400425
426 if(!instruction->predicateNot)
427 {
428 if(dst.x) xEnable = xEnable & As<Int4>(xPredicate);
429 if(dst.y) yEnable = yEnable & As<Int4>(yPredicate);
430 if(dst.z) zEnable = zEnable & As<Int4>(zPredicate);
431 if(dst.w) wEnable = wEnable & As<Int4>(wPredicate);
432 }
433 else
434 {
435 if(dst.x) xEnable = xEnable & ~As<Int4>(xPredicate);
436 if(dst.y) yEnable = yEnable & ~As<Int4>(yPredicate);
437 if(dst.z) zEnable = zEnable & ~As<Int4>(zPredicate);
438 if(dst.w) wEnable = wEnable & ~As<Int4>(wPredicate);
439 }
440 }
441
442 if(dst.x) d.x = As<Float4>(As<Int4>(d.x) & xEnable);
443 if(dst.y) d.y = As<Float4>(As<Int4>(d.y) & yEnable);
444 if(dst.z) d.z = As<Float4>(As<Int4>(d.z) & zEnable);
445 if(dst.w) d.w = As<Float4>(As<Int4>(d.w) & wEnable);
446
447 if(dst.x) d.x = As<Float4>(As<Int4>(d.x) | (As<Int4>(pDst.x) & ~xEnable));
448 if(dst.y) d.y = As<Float4>(As<Int4>(d.y) | (As<Int4>(pDst.y) & ~yEnable));
449 if(dst.z) d.z = As<Float4>(As<Int4>(d.z) | (As<Int4>(pDst.z) & ~zEnable));
450 if(dst.w) d.w = As<Float4>(As<Int4>(d.w) | (As<Int4>(pDst.w) & ~wEnable));
451 }
452
453 switch(dst.type)
454 {
455 case Shader::PARAMETER_TEMP:
456 if(dst.rel.type == Shader::PARAMETER_VOID)
457 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500458 if(dst.x) r[dst.index].x = d.x;
459 if(dst.y) r[dst.index].y = d.y;
460 if(dst.z) r[dst.index].z = d.z;
461 if(dst.w) r[dst.index].w = d.w;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400462 }
463 else
464 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500465 Int a = relativeAddress(dst);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400466
Nicolas Capens4f172c72016-01-13 08:34:30 -0500467 if(dst.x) r[dst.index + a].x = d.x;
468 if(dst.y) r[dst.index + a].y = d.y;
469 if(dst.z) r[dst.index + a].z = d.z;
470 if(dst.w) r[dst.index + a].w = d.w;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400471 }
472 break;
473 case Shader::PARAMETER_COLOROUT:
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500474 if(dst.rel.type == Shader::PARAMETER_VOID)
475 {
Nicolas Capens8b124c12016-04-18 14:09:37 -0400476 broadcastColor0 = (dst.index == 0) && broadcastColor0;
477
Nicolas Capens4f172c72016-01-13 08:34:30 -0500478 if(dst.x) { oC[dst.index].x = d.x; }
479 if(dst.y) { oC[dst.index].y = d.y; }
480 if(dst.z) { oC[dst.index].z = d.z; }
481 if(dst.w) { oC[dst.index].w = d.w; }
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500482 }
483 else
484 {
Nicolas Capens8b124c12016-04-18 14:09:37 -0400485 broadcastColor0 = false;
Nicolas Capens4f172c72016-01-13 08:34:30 -0500486 Int a = relativeAddress(dst) + dst.index;
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500487
Nicolas Capens4f172c72016-01-13 08:34:30 -0500488 if(dst.x) { oC[a].x = d.x; }
489 if(dst.y) { oC[a].y = d.y; }
490 if(dst.z) { oC[a].z = d.z; }
491 if(dst.w) { oC[a].w = d.w; }
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500492 }
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400493 break;
494 case Shader::PARAMETER_PREDICATE:
Nicolas Capens4f172c72016-01-13 08:34:30 -0500495 if(dst.x) p0.x = d.x;
496 if(dst.y) p0.y = d.y;
497 if(dst.z) p0.z = d.z;
498 if(dst.w) p0.w = d.w;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400499 break;
500 case Shader::PARAMETER_DEPTHOUT:
Nicolas Capens4f172c72016-01-13 08:34:30 -0500501 oDepth = d.x;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400502 break;
503 default:
504 ASSERT(false);
505 }
506 }
507 }
508
509 if(currentLabel != -1)
510 {
511 Nucleus::setInsertBlock(returnBlock);
512 }
513
Nicolas Capens8b124c12016-04-18 14:09:37 -0400514 if(broadcastColor0)
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400515 {
Nicolas Capens8b124c12016-04-18 14:09:37 -0400516 for(int i = 0; i < RENDERTARGETS; i++)
517 {
518 c[i] = oC[0];
519 }
520 }
521 else
522 {
523 for(int i = 0; i < RENDERTARGETS; i++)
524 {
525 c[i] = oC[i];
526 }
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400527 }
Nicolas Capens7bc079c2018-01-02 15:55:11 -0500528
529 clampColor(c);
530
531 if(state.depthOverride)
532 {
533 oDepth = Min(Max(oDepth, Float4(0.0f)), Float4(1.0f));
534 }
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400535 }
Nicolas Capens5d961882016-01-01 23:18:14 -0500536
Nicolas Capens4f172c72016-01-13 08:34:30 -0500537 Bool PixelProgram::alphaTest(Int cMask[4])
Nicolas Capens5d961882016-01-01 23:18:14 -0500538 {
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400539 if(!state.alphaTestActive())
540 {
541 return true;
542 }
543
544 Int aMask;
545
546 if(state.transparencyAntialiasing == TRANSPARENCY_NONE)
547 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500548 Short4 alpha = RoundShort4(c[0].w * Float4(0x1000));
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400549
Nicolas Capens4f172c72016-01-13 08:34:30 -0500550 PixelRoutine::alphaTest(aMask, alpha);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400551
552 for(unsigned int q = 0; q < state.multiSample; q++)
553 {
554 cMask[q] &= aMask;
555 }
556 }
557 else if(state.transparencyAntialiasing == TRANSPARENCY_ALPHA_TO_COVERAGE)
558 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500559 alphaToCoverage(cMask, c[0].w);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400560 }
561 else ASSERT(false);
562
563 Int pass = cMask[0];
564
565 for(unsigned int q = 1; q < state.multiSample; q++)
566 {
567 pass = pass | cMask[q];
568 }
569
Nicolas Capens5d961882016-01-01 23:18:14 -0500570 return pass != 0x0;
571 }
572
Nicolas Capens4f172c72016-01-13 08:34:30 -0500573 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 -0500574 {
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500575 for(int index = 0; index < RENDERTARGETS; index++)
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400576 {
577 if(!state.colorWriteActive(index))
578 {
579 continue;
Nicolas Capens5d961882016-01-01 23:18:14 -0500580 }
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400581
Alexis Hetu049a1872016-04-25 16:59:58 -0400582 if(!postBlendSRGB && state.writeSRGB && !isSRGB(index))
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400583 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500584 c[index].x = linearToSRGB(c[index].x);
585 c[index].y = linearToSRGB(c[index].y);
586 c[index].z = linearToSRGB(c[index].z);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400587 }
588
589 if(index == 0)
590 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500591 fogBlend(c[index], fog);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400592 }
593
594 switch(state.targetFormat[index])
595 {
596 case FORMAT_R5G6B5:
597 case FORMAT_X8R8G8B8:
598 case FORMAT_X8B8G8R8:
599 case FORMAT_A8R8G8B8:
600 case FORMAT_A8B8G8R8:
Alexis Hetu049a1872016-04-25 16:59:58 -0400601 case FORMAT_SRGB8_X8:
602 case FORMAT_SRGB8_A8:
Alexis Hetu143dfc72016-09-13 18:41:27 -0400603 case FORMAT_G8R8:
Alexis Hetu0cff3cb2016-05-04 16:23:50 -0400604 case FORMAT_R8:
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400605 case FORMAT_A8:
606 case FORMAT_G16R16:
607 case FORMAT_A16B16G16R16:
608 for(unsigned int q = 0; q < state.multiSample; q++)
609 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500610 Pointer<Byte> buffer = cBuffer[index] + q * *Pointer<Int>(data + OFFSET(DrawData, colorSliceB[index]));
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400611 Vector4s color;
612
Nicolas Capens13c70cd2016-04-05 10:51:45 -0400613 if(state.targetFormat[index] == FORMAT_R5G6B5)
614 {
615 color.x = UShort4(c[index].x * Float4(0xFBFF), false);
616 color.y = UShort4(c[index].y * Float4(0xFDFF), false);
617 color.z = UShort4(c[index].z * Float4(0xFBFF), false);
618 color.w = UShort4(c[index].w * Float4(0xFFFF), false);
619 }
620 else
621 {
622 color.x = convertFixed16(c[index].x, false);
623 color.y = convertFixed16(c[index].y, false);
624 color.z = convertFixed16(c[index].z, false);
625 color.w = convertFixed16(c[index].w, false);
626 }
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400627
628 if(state.multiSampleMask & (1 << q))
629 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500630 alphaBlend(index, buffer, color, x);
631 logicOperation(index, buffer, color, x);
632 writeColor(index, buffer, x, color, sMask[q], zMask[q], cMask[q]);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400633 }
634 }
635 break;
636 case FORMAT_R32F:
637 case FORMAT_G32R32F:
Alexis Hetudbd1a8e2016-04-13 11:40:30 -0400638 case FORMAT_X32B32G32R32F:
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400639 case FORMAT_A32B32G32R32F:
Nicolas Capens67fdd832017-12-21 11:20:54 -0500640 case FORMAT_X32B32G32R32F_UNSIGNED:
Alexis Hetu1abb6382016-02-08 11:21:16 -0500641 case FORMAT_R32I:
642 case FORMAT_G32R32I:
643 case FORMAT_A32B32G32R32I:
644 case FORMAT_R32UI:
645 case FORMAT_G32R32UI:
646 case FORMAT_A32B32G32R32UI:
Alexis Hetubd7117d2016-06-02 10:35:59 -0400647 case FORMAT_R16I:
648 case FORMAT_G16R16I:
649 case FORMAT_A16B16G16R16I:
650 case FORMAT_R16UI:
651 case FORMAT_G16R16UI:
652 case FORMAT_A16B16G16R16UI:
653 case FORMAT_R8I:
654 case FORMAT_G8R8I:
655 case FORMAT_A8B8G8R8I:
656 case FORMAT_R8UI:
657 case FORMAT_G8R8UI:
658 case FORMAT_A8B8G8R8UI:
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400659 for(unsigned int q = 0; q < state.multiSample; q++)
660 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500661 Pointer<Byte> buffer = cBuffer[index] + q * *Pointer<Int>(data + OFFSET(DrawData, colorSliceB[index]));
662 Vector4f color = c[index];
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400663
664 if(state.multiSampleMask & (1 << q))
665 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500666 alphaBlend(index, buffer, color, x);
667 writeColor(index, buffer, x, color, sMask[q], zMask[q], cMask[q]);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400668 }
669 }
670 break;
671 default:
672 ASSERT(false);
673 }
674 }
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500675 }
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400676
Nicolas Capensa0b57832017-11-07 13:07:53 -0500677 Vector4f PixelProgram::sampleTexture(const Src &sampler, Vector4f &uvwq, Float4 &bias, Vector4f &dsx, Vector4f &dsy, Vector4f &offset, SamplerFunction function)
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400678 {
Nicolas Capensc2534f42016-04-04 11:13:24 -0400679 Vector4f tmp;
680
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400681 if(sampler.type == Shader::PARAMETER_SAMPLER && sampler.rel.type == Shader::PARAMETER_VOID)
682 {
Nicolas Capensa0b57832017-11-07 13:07:53 -0500683 tmp = sampleTexture(sampler.index, uvwq, bias, dsx, dsy, offset, function);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400684 }
685 else
686 {
Nicolas Capensc2534f42016-04-04 11:13:24 -0400687 Int index = As<Int>(Float(fetchRegister(sampler).x.x));
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400688
689 for(int i = 0; i < TEXTURE_IMAGE_UNITS; i++)
690 {
691 if(shader->usesSampler(i))
692 {
693 If(index == i)
694 {
Nicolas Capensa0b57832017-11-07 13:07:53 -0500695 tmp = sampleTexture(i, uvwq, bias, dsx, dsy, offset, function);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400696 // FIXME: When the sampler states are the same, we could use one sampler and just index the texture
697 }
698 }
699 }
700 }
Nicolas Capensc2534f42016-04-04 11:13:24 -0400701
Nicolas Capens89a218b2017-11-07 13:05:20 -0500702 Vector4f c;
Nicolas Capensc2534f42016-04-04 11:13:24 -0400703 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];
Nicolas Capens89a218b2017-11-07 13:05:20 -0500707
708 return c;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400709 }
710
Nicolas Capensa0b57832017-11-07 13:07:53 -0500711 Vector4f PixelProgram::sampleTexture(int samplerIndex, Vector4f &uvwq, Float4 &bias, Vector4f &dsx, Vector4f &dsy, Vector4f &offset, SamplerFunction function)
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400712 {
Nicolas Capensc2534f42016-04-04 11:13:24 -0400713 #if PERF_PROFILE
714 Long texTime = Ticks();
715 #endif
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400716
Meng-Lin Wu2337a192016-06-01 13:54:07 -0400717 Pointer<Byte> texture = data + OFFSET(DrawData, mipmap) + samplerIndex * sizeof(Texture);
Nicolas Capensa0b57832017-11-07 13:07:53 -0500718 Vector4f c = SamplerCore(constants, state.sampler[samplerIndex]).sampleTexture(texture, uvwq.x, uvwq.y, uvwq.z, uvwq.w, bias, dsx, dsy, offset, function);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400719
Nicolas Capensc2534f42016-04-04 11:13:24 -0400720 #if PERF_PROFILE
721 cycles[PERF_TEX] += Ticks() - texTime;
722 #endif
Nicolas Capens89a218b2017-11-07 13:05:20 -0500723
724 return c;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400725 }
726
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500727 void PixelProgram::clampColor(Vector4f oC[RENDERTARGETS])
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400728 {
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500729 for(int index = 0; index < RENDERTARGETS; index++)
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400730 {
731 if(!state.colorWriteActive(index) && !(index == 0 && state.alphaTestActive()))
732 {
733 continue;
734 }
735
736 switch(state.targetFormat[index])
737 {
738 case FORMAT_NULL:
739 break;
740 case FORMAT_R5G6B5:
741 case FORMAT_A8R8G8B8:
742 case FORMAT_A8B8G8R8:
743 case FORMAT_X8R8G8B8:
744 case FORMAT_X8B8G8R8:
Alexis Hetu049a1872016-04-25 16:59:58 -0400745 case FORMAT_SRGB8_X8:
746 case FORMAT_SRGB8_A8:
Alexis Hetu143dfc72016-09-13 18:41:27 -0400747 case FORMAT_G8R8:
Alexis Hetu0cff3cb2016-05-04 16:23:50 -0400748 case FORMAT_R8:
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400749 case FORMAT_A8:
750 case FORMAT_G16R16:
751 case FORMAT_A16B16G16R16:
752 oC[index].x = Max(oC[index].x, Float4(0.0f)); oC[index].x = Min(oC[index].x, Float4(1.0f));
753 oC[index].y = Max(oC[index].y, Float4(0.0f)); oC[index].y = Min(oC[index].y, Float4(1.0f));
754 oC[index].z = Max(oC[index].z, Float4(0.0f)); oC[index].z = Min(oC[index].z, Float4(1.0f));
755 oC[index].w = Max(oC[index].w, Float4(0.0f)); oC[index].w = Min(oC[index].w, Float4(1.0f));
756 break;
757 case FORMAT_R32F:
758 case FORMAT_G32R32F:
Alexis Hetudbd1a8e2016-04-13 11:40:30 -0400759 case FORMAT_X32B32G32R32F:
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400760 case FORMAT_A32B32G32R32F:
Alexis Hetu1abb6382016-02-08 11:21:16 -0500761 case FORMAT_R32I:
762 case FORMAT_G32R32I:
763 case FORMAT_A32B32G32R32I:
764 case FORMAT_R32UI:
765 case FORMAT_G32R32UI:
766 case FORMAT_A32B32G32R32UI:
Alexis Hetu8e7675d2016-09-27 21:06:00 -0400767 case FORMAT_R16I:
768 case FORMAT_G16R16I:
769 case FORMAT_A16B16G16R16I:
770 case FORMAT_R16UI:
771 case FORMAT_G16R16UI:
772 case FORMAT_A16B16G16R16UI:
773 case FORMAT_R8I:
774 case FORMAT_G8R8I:
775 case FORMAT_A8B8G8R8I:
776 case FORMAT_R8UI:
777 case FORMAT_G8R8UI:
778 case FORMAT_A8B8G8R8UI:
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400779 break;
Nicolas Capens67fdd832017-12-21 11:20:54 -0500780 case FORMAT_X32B32G32R32F_UNSIGNED:
781 oC[index].x = Max(oC[index].x, Float4(0.0f));
782 oC[index].y = Max(oC[index].y, Float4(0.0f));
783 oC[index].z = Max(oC[index].z, Float4(0.0f));
784 oC[index].w = Max(oC[index].w, Float4(0.0f));
785 break;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400786 default:
787 ASSERT(false);
788 }
789 }
790 }
791
Nicolas Capens4f172c72016-01-13 08:34:30 -0500792 Int4 PixelProgram::enableMask(const Shader::Instruction *instruction)
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400793 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500794 Int4 enable = instruction->analysisBranch ? Int4(enableStack[enableIndex]) : Int4(0xFFFFFFFF);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400795
796 if(!whileTest)
797 {
798 if(shader->containsBreakInstruction() && instruction->analysisBreak)
799 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500800 enable &= enableBreak;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400801 }
802
803 if(shader->containsContinueInstruction() && instruction->analysisContinue)
804 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500805 enable &= enableContinue;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400806 }
807
808 if(shader->containsLeaveInstruction() && instruction->analysisLeave)
809 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500810 enable &= enableLeave;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400811 }
812 }
813
814 return enable;
815 }
816
Nicolas Capensc2534f42016-04-04 11:13:24 -0400817 Vector4f PixelProgram::fetchRegister(const Src &src, unsigned int offset)
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400818 {
819 Vector4f reg;
Nicolas Capens5d961882016-01-01 23:18:14 -0500820 unsigned int i = src.index + offset;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400821
822 switch(src.type)
823 {
824 case Shader::PARAMETER_TEMP:
825 if(src.rel.type == Shader::PARAMETER_VOID)
826 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500827 reg = r[i];
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400828 }
829 else
830 {
Alexis Hetu2c2a7b22015-10-27 16:12:11 -0400831 Int a = relativeAddress(src, src.bufferIndex);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400832
Nicolas Capens4f172c72016-01-13 08:34:30 -0500833 reg = r[i + a];
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400834 }
835 break;
836 case Shader::PARAMETER_INPUT:
837 {
838 if(src.rel.type == Shader::PARAMETER_VOID) // Not relative
839 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500840 reg = v[i];
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400841 }
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400842 else
843 {
Alexis Hetu2c2a7b22015-10-27 16:12:11 -0400844 Int a = relativeAddress(src, src.bufferIndex);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400845
Nicolas Capens4f172c72016-01-13 08:34:30 -0500846 reg = v[i + a];
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400847 }
848 }
849 break;
850 case Shader::PARAMETER_CONST:
Nicolas Capens4f172c72016-01-13 08:34:30 -0500851 reg = readConstant(src, offset);
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400852 break;
853 case Shader::PARAMETER_TEXTURE:
Nicolas Capens4f172c72016-01-13 08:34:30 -0500854 reg = v[2 + i];
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400855 break;
856 case Shader::PARAMETER_MISCTYPE:
Alexis Hetu877ddfc2017-07-25 17:48:00 -0400857 if(src.index == Shader::VPosIndex) reg = vPos;
858 if(src.index == Shader::VFaceIndex) reg = vFace;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400859 break;
860 case Shader::PARAMETER_SAMPLER:
861 if(src.rel.type == Shader::PARAMETER_VOID)
862 {
863 reg.x = As<Float4>(Int4(i));
864 }
865 else if(src.rel.type == Shader::PARAMETER_TEMP)
866 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500867 reg.x = As<Float4>(Int4(i) + As<Int4>(r[src.rel.index].x));
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400868 }
869 return reg;
870 case Shader::PARAMETER_PREDICATE: return reg; // Dummy
871 case Shader::PARAMETER_VOID: return reg; // Dummy
872 case Shader::PARAMETER_FLOAT4LITERAL:
873 reg.x = Float4(src.value[0]);
874 reg.y = Float4(src.value[1]);
875 reg.z = Float4(src.value[2]);
876 reg.w = Float4(src.value[3]);
877 break;
878 case Shader::PARAMETER_CONSTINT: return reg; // Dummy
879 case Shader::PARAMETER_CONSTBOOL: return reg; // Dummy
880 case Shader::PARAMETER_LOOP: return reg; // Dummy
881 case Shader::PARAMETER_COLOROUT:
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500882 if(src.rel.type == Shader::PARAMETER_VOID) // Not relative
883 {
Nicolas Capens4f172c72016-01-13 08:34:30 -0500884 reg = oC[i];
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500885 }
886 else
887 {
Alexis Hetu2c2a7b22015-10-27 16:12:11 -0400888 Int a = relativeAddress(src, src.bufferIndex);
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500889
Nicolas Capens4f172c72016-01-13 08:34:30 -0500890 reg = oC[i + a];
Alexis Hetu1edcd8b2015-11-05 11:12:41 -0500891 }
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400892 break;
893 case Shader::PARAMETER_DEPTHOUT:
Nicolas Capens4f172c72016-01-13 08:34:30 -0500894 reg.x = oDepth;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400895 break;
896 default:
897 ASSERT(false);
898 }
899
900 const Float4 &x = reg[(src.swizzle >> 0) & 0x3];
901 const Float4 &y = reg[(src.swizzle >> 2) & 0x3];
902 const Float4 &z = reg[(src.swizzle >> 4) & 0x3];
903 const Float4 &w = reg[(src.swizzle >> 6) & 0x3];
904
905 Vector4f mod;
906
907 switch(src.modifier)
908 {
909 case Shader::MODIFIER_NONE:
910 mod.x = x;
911 mod.y = y;
912 mod.z = z;
913 mod.w = w;
914 break;
915 case Shader::MODIFIER_NEGATE:
916 mod.x = -x;
917 mod.y = -y;
918 mod.z = -z;
919 mod.w = -w;
920 break;
921 case Shader::MODIFIER_ABS:
922 mod.x = Abs(x);
923 mod.y = Abs(y);
924 mod.z = Abs(z);
925 mod.w = Abs(w);
926 break;
927 case Shader::MODIFIER_ABS_NEGATE:
928 mod.x = -Abs(x);
929 mod.y = -Abs(y);
930 mod.z = -Abs(z);
931 mod.w = -Abs(w);
932 break;
933 case Shader::MODIFIER_NOT:
934 mod.x = As<Float4>(As<Int4>(x) ^ Int4(0xFFFFFFFF));
935 mod.y = As<Float4>(As<Int4>(y) ^ Int4(0xFFFFFFFF));
936 mod.z = As<Float4>(As<Int4>(z) ^ Int4(0xFFFFFFFF));
937 mod.w = As<Float4>(As<Int4>(w) ^ Int4(0xFFFFFFFF));
938 break;
939 default:
940 ASSERT(false);
941 }
942
943 return mod;
944 }
945
Alexis Hetu2c2a7b22015-10-27 16:12:11 -0400946 RValue<Pointer<Byte>> PixelProgram::uniformAddress(int bufferIndex, unsigned int index)
947 {
948 if(bufferIndex == -1)
949 {
950 return data + OFFSET(DrawData, ps.c[index]);
951 }
952 else
953 {
954 return *Pointer<Pointer<Byte>>(data + OFFSET(DrawData, ps.u[bufferIndex])) + index;
955 }
956 }
957
958 RValue<Pointer<Byte>> PixelProgram::uniformAddress(int bufferIndex, unsigned int index, Int& offset)
959 {
960 return uniformAddress(bufferIndex, index) + offset * sizeof(float4);
961 }
962
Nicolas Capens4f172c72016-01-13 08:34:30 -0500963 Vector4f PixelProgram::readConstant(const Src &src, unsigned int offset)
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400964 {
965 Vector4f c;
Nicolas Capens5d961882016-01-01 23:18:14 -0500966 unsigned int i = src.index + offset;
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400967
968 if(src.rel.type == Shader::PARAMETER_VOID) // Not relative
969 {
Alexis Hetu2c2a7b22015-10-27 16:12:11 -0400970 c.x = c.y = c.z = c.w = *Pointer<Float4>(uniformAddress(src.bufferIndex, i));
Alexis Hetuf2a8c372015-07-13 11:08:41 -0400971
972 c.x = c.x.xxxx;
973 c.y = c.y.yyyy;
974 c.z = c.z.zzzz;
975 c.w = c.w.wwww;
976
977 if(shader->containsDefineInstruction()) // Constant may be known at compile time
978 {
979 for(size_t j = 0; j < shader->getLength(); j++)
980 {
981 const Shader::Instruction &instruction = *shader->getInstruction(j);
982
983 if(instruction.opcode == Shader::OPCODE_DEF)
984 {
985 if(instruction.dst.index == i)
986 {
987 c.x = Float4(instruction.src[0].value[0]);
988 c.y = Float4(instruction.src[0].value[1]);
989 c.z = Float4(instruction.src[0].value[2]);
990 c.w = Float4(instruction.src[0].value[3]);
991
992 break;
993 }
994 }
995 }
996 }
997 }
998 else if(src.rel.type == Shader::PARAMETER_LOOP)
999 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001000 Int loopCounter = aL[loopDepth];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001001
Alexis Hetu2c2a7b22015-10-27 16:12:11 -04001002 c.x = c.y = c.z = c.w = *Pointer<Float4>(uniformAddress(src.bufferIndex, i, loopCounter));
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001003
1004 c.x = c.x.xxxx;
1005 c.y = c.y.yyyy;
1006 c.z = c.z.zzzz;
1007 c.w = c.w.wwww;
1008 }
1009 else
1010 {
Alexis Hetu2c2a7b22015-10-27 16:12:11 -04001011 Int a = relativeAddress(src, src.bufferIndex);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001012
Alexis Hetu2c2a7b22015-10-27 16:12:11 -04001013 c.x = c.y = c.z = c.w = *Pointer<Float4>(uniformAddress(src.bufferIndex, i, a));
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001014
1015 c.x = c.x.xxxx;
1016 c.y = c.y.yyyy;
1017 c.z = c.z.zzzz;
1018 c.w = c.w.wwww;
1019 }
1020
1021 return c;
1022 }
1023
Alexis Hetu2c2a7b22015-10-27 16:12:11 -04001024 Int PixelProgram::relativeAddress(const Shader::Parameter &var, int bufferIndex)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001025 {
1026 ASSERT(var.rel.deterministic);
1027
1028 if(var.rel.type == Shader::PARAMETER_TEMP)
1029 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001030 return As<Int>(Extract(r[var.rel.index].x, 0)) * var.rel.scale;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001031 }
1032 else if(var.rel.type == Shader::PARAMETER_INPUT)
1033 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001034 return As<Int>(Extract(v[var.rel.index].x, 0)) * var.rel.scale;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001035 }
1036 else if(var.rel.type == Shader::PARAMETER_OUTPUT)
1037 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001038 return As<Int>(Extract(oC[var.rel.index].x, 0)) * var.rel.scale;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001039 }
1040 else if(var.rel.type == Shader::PARAMETER_CONST)
1041 {
Alexis Hetu48be7352016-02-10 14:32:34 -05001042 return *Pointer<Int>(uniformAddress(bufferIndex, var.rel.index)) * var.rel.scale;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001043 }
Nicolas Capens907700d2016-01-20 17:09:28 -05001044 else if(var.rel.type == Shader::PARAMETER_LOOP)
1045 {
1046 return aL[loopDepth];
1047 }
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001048 else ASSERT(false);
1049
1050 return 0;
1051 }
1052
1053 Float4 PixelProgram::linearToSRGB(const Float4 &x) // Approximates x^(1.0/2.2)
1054 {
1055 Float4 sqrtx = Rcp_pp(RcpSqrt_pp(x));
1056 Float4 sRGB = sqrtx * Float4(1.14f) - x * Float4(0.14f);
1057
1058 return Min(Max(sRGB, Float4(0.0f)), Float4(1.0f));
1059 }
1060
Nicolas Capens4f172c72016-01-13 08:34:30 -05001061 void PixelProgram::M3X2(Vector4f &dst, Vector4f &src0, const Src &src1)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001062 {
Nicolas Capensc2534f42016-04-04 11:13:24 -04001063 Vector4f row0 = fetchRegister(src1, 0);
1064 Vector4f row1 = fetchRegister(src1, 1);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001065
1066 dst.x = dot3(src0, row0);
1067 dst.y = dot3(src0, row1);
1068 }
1069
Nicolas Capens4f172c72016-01-13 08:34:30 -05001070 void PixelProgram::M3X3(Vector4f &dst, Vector4f &src0, const Src &src1)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001071 {
Nicolas Capensc2534f42016-04-04 11:13:24 -04001072 Vector4f row0 = fetchRegister(src1, 0);
1073 Vector4f row1 = fetchRegister(src1, 1);
1074 Vector4f row2 = fetchRegister(src1, 2);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001075
1076 dst.x = dot3(src0, row0);
1077 dst.y = dot3(src0, row1);
1078 dst.z = dot3(src0, row2);
1079 }
1080
Nicolas Capens4f172c72016-01-13 08:34:30 -05001081 void PixelProgram::M3X4(Vector4f &dst, Vector4f &src0, const Src &src1)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001082 {
Nicolas Capensc2534f42016-04-04 11:13:24 -04001083 Vector4f row0 = fetchRegister(src1, 0);
1084 Vector4f row1 = fetchRegister(src1, 1);
1085 Vector4f row2 = fetchRegister(src1, 2);
1086 Vector4f row3 = fetchRegister(src1, 3);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001087
1088 dst.x = dot3(src0, row0);
1089 dst.y = dot3(src0, row1);
1090 dst.z = dot3(src0, row2);
1091 dst.w = dot3(src0, row3);
1092 }
1093
Nicolas Capens4f172c72016-01-13 08:34:30 -05001094 void PixelProgram::M4X3(Vector4f &dst, Vector4f &src0, const Src &src1)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001095 {
Nicolas Capensc2534f42016-04-04 11:13:24 -04001096 Vector4f row0 = fetchRegister(src1, 0);
1097 Vector4f row1 = fetchRegister(src1, 1);
1098 Vector4f row2 = fetchRegister(src1, 2);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001099
1100 dst.x = dot4(src0, row0);
1101 dst.y = dot4(src0, row1);
1102 dst.z = dot4(src0, row2);
1103 }
1104
Nicolas Capens4f172c72016-01-13 08:34:30 -05001105 void PixelProgram::M4X4(Vector4f &dst, Vector4f &src0, const Src &src1)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001106 {
Nicolas Capensc2534f42016-04-04 11:13:24 -04001107 Vector4f row0 = fetchRegister(src1, 0);
1108 Vector4f row1 = fetchRegister(src1, 1);
1109 Vector4f row2 = fetchRegister(src1, 2);
1110 Vector4f row3 = fetchRegister(src1, 3);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001111
1112 dst.x = dot4(src0, row0);
1113 dst.y = dot4(src0, row1);
1114 dst.z = dot4(src0, row2);
1115 dst.w = dot4(src0, row3);
1116 }
1117
Nicolas Capensa0b57832017-11-07 13:07:53 -05001118 void PixelProgram::TEX(Vector4f &dst, Vector4f &src0, const Src &src1, bool project, bool bias)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001119 {
Nicolas Capens0484c792016-06-13 22:02:36 -04001120 if(project)
1121 {
1122 Vector4f proj;
1123 Float4 rw = reciprocal(src0.w);
1124 proj.x = src0.x * rw;
1125 proj.y = src0.y * rw;
1126 proj.z = src0.z * rw;
1127
Nicolas Capensa0b57832017-11-07 13:07:53 -05001128 dst = sampleTexture(src1, proj, src0.x, (src0), (src0), (src0), Implicit);
Nicolas Capens0484c792016-06-13 22:02:36 -04001129 }
1130 else
1131 {
Nicolas Capensa0b57832017-11-07 13:07:53 -05001132 dst = sampleTexture(src1, src0, src0.x, (src0), (src0), (src0), bias ? Bias : Implicit);
Nicolas Capens0484c792016-06-13 22:02:36 -04001133 }
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001134 }
1135
Nicolas Capensa0b57832017-11-07 13:07:53 -05001136 void PixelProgram::TEXOFFSET(Vector4f &dst, Vector4f &src0, const Src &src1, Vector4f &offset)
Alexis Hetu25d47fc2015-10-22 13:58:32 -04001137 {
Nicolas Capensa0b57832017-11-07 13:07:53 -05001138 dst = sampleTexture(src1, src0, (src0.x), (src0), (src0), offset, {Implicit, Offset});
Alexis Hetu25d47fc2015-10-22 13:58:32 -04001139 }
1140
Nicolas Capensa0b57832017-11-07 13:07:53 -05001141 void PixelProgram::TEXLODOFFSET(Vector4f &dst, Vector4f &src0, const Src &src1, Vector4f &offset, Float4 &lod)
Alexis Hetu25d47fc2015-10-22 13:58:32 -04001142 {
Nicolas Capensa0b57832017-11-07 13:07:53 -05001143 dst = sampleTexture(src1, src0, lod, (src0), (src0), offset, {Lod, Offset});
Alexis Hetu25d47fc2015-10-22 13:58:32 -04001144 }
1145
Nicolas Capensa0b57832017-11-07 13:07:53 -05001146 void PixelProgram::TEXBIAS(Vector4f &dst, Vector4f &src0, const Src &src1, Float4 &bias)
Alexis Hetu25d47fc2015-10-22 13:58:32 -04001147 {
Nicolas Capensa0b57832017-11-07 13:07:53 -05001148 dst = sampleTexture(src1, src0, bias, (src0), (src0), (src0), Bias);
1149 }
Nicolas Capens67fdd832017-12-21 11:20:54 -05001150
Nicolas Capensa0b57832017-11-07 13:07:53 -05001151 void PixelProgram::TEXOFFSETBIAS(Vector4f &dst, Vector4f &src0, const Src &src1, Vector4f &offset, Float4 &bias)
1152 {
1153 dst = sampleTexture(src1, src0, bias, (src0), (src0), offset, {Bias, Offset});
Alexis Hetu25d47fc2015-10-22 13:58:32 -04001154 }
1155
Nicolas Capensa0b57832017-11-07 13:07:53 -05001156 void PixelProgram::TEXELFETCH(Vector4f &dst, Vector4f &src0, const Src& src1, Float4 &lod)
Alexis Hetu25d47fc2015-10-22 13:58:32 -04001157 {
Nicolas Capensa0b57832017-11-07 13:07:53 -05001158 dst = sampleTexture(src1, src0, lod, (src0), (src0), (src0), Fetch);
Alexis Hetu25d47fc2015-10-22 13:58:32 -04001159 }
1160
Nicolas Capensa0b57832017-11-07 13:07:53 -05001161 void PixelProgram::TEXELFETCHOFFSET(Vector4f &dst, Vector4f &src0, const Src& src1, Vector4f &offset, Float4 &lod)
Alexis Hetu25d47fc2015-10-22 13:58:32 -04001162 {
Nicolas Capensa0b57832017-11-07 13:07:53 -05001163 dst = sampleTexture(src1, src0, lod, (src0), (src0), offset, {Fetch, Offset});
Alexis Hetu25d47fc2015-10-22 13:58:32 -04001164 }
1165
Nicolas Capensa0b57832017-11-07 13:07:53 -05001166 void PixelProgram::TEXGRAD(Vector4f &dst, Vector4f &src0, const Src& src1, Vector4f &dsx, Vector4f &dsy)
Alexis Hetu25d47fc2015-10-22 13:58:32 -04001167 {
Nicolas Capensa0b57832017-11-07 13:07:53 -05001168 dst = sampleTexture(src1, src0, (src0.x), dsx, dsy, (src0), Grad);
Alexis Hetu25d47fc2015-10-22 13:58:32 -04001169 }
1170
Nicolas Capensa0b57832017-11-07 13:07:53 -05001171 void PixelProgram::TEXGRADOFFSET(Vector4f &dst, Vector4f &src0, const Src& src1, Vector4f &dsx, Vector4f &dsy, Vector4f &offset)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001172 {
Nicolas Capensa0b57832017-11-07 13:07:53 -05001173 dst = sampleTexture(src1, src0, (src0.x), dsx, dsy, offset, {Grad, Offset});
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001174 }
1175
Nicolas Capensa0b57832017-11-07 13:07:53 -05001176 void PixelProgram::TEXLOD(Vector4f &dst, Vector4f &src0, const Src &src1, Float4 &lod)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001177 {
Nicolas Capensa0b57832017-11-07 13:07:53 -05001178 dst = sampleTexture(src1, src0, lod, (src0), (src0), (src0), Lod);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001179 }
1180
Nicolas Capens4f172c72016-01-13 08:34:30 -05001181 void PixelProgram::TEXSIZE(Vector4f &dst, Float4 &lod, const Src &src1)
Alexis Hetu9bcb31d2015-07-22 17:03:26 -04001182 {
Alexis Hetu95ac1872016-06-06 13:26:52 -04001183 Pointer<Byte> texture = data + OFFSET(DrawData, mipmap) + src1.index * sizeof(Texture);
Nicolas Capens89a218b2017-11-07 13:05:20 -05001184 dst = SamplerCore::textureSize(texture, lod);
Alexis Hetu9bcb31d2015-07-22 17:03:26 -04001185 }
1186
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001187 void PixelProgram::TEXKILL(Int cMask[4], Vector4f &src, unsigned char mask)
1188 {
1189 Int kill = -1;
1190
1191 if(mask & 0x1) kill &= SignMask(CmpNLT(src.x, Float4(0.0f)));
1192 if(mask & 0x2) kill &= SignMask(CmpNLT(src.y, Float4(0.0f)));
1193 if(mask & 0x4) kill &= SignMask(CmpNLT(src.z, Float4(0.0f)));
1194 if(mask & 0x8) kill &= SignMask(CmpNLT(src.w, Float4(0.0f)));
1195
1196 // FIXME: Dynamic branching affects TEXKILL?
1197 // if(shader->containsDynamicBranching())
1198 // {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001199 // kill = ~SignMask(enableMask());
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001200 // }
1201
1202 for(unsigned int q = 0; q < state.multiSample; q++)
1203 {
1204 cMask[q] &= kill;
1205 }
1206
1207 // FIXME: Branch to end of shader if all killed?
1208 }
1209
Nicolas Capens4f172c72016-01-13 08:34:30 -05001210 void PixelProgram::DISCARD(Int cMask[4], const Shader::Instruction *instruction)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001211 {
1212 Int kill = 0;
1213
1214 if(shader->containsDynamicBranching())
1215 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001216 kill = ~SignMask(enableMask(instruction));
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001217 }
1218
1219 for(unsigned int q = 0; q < state.multiSample; q++)
1220 {
1221 cMask[q] &= kill;
1222 }
1223
1224 // FIXME: Branch to end of shader if all killed?
1225 }
1226
1227 void PixelProgram::DFDX(Vector4f &dst, Vector4f &src)
1228 {
1229 dst.x = src.x.yyww - src.x.xxzz;
1230 dst.y = src.y.yyww - src.y.xxzz;
1231 dst.z = src.z.yyww - src.z.xxzz;
1232 dst.w = src.w.yyww - src.w.xxzz;
1233 }
1234
1235 void PixelProgram::DFDY(Vector4f &dst, Vector4f &src)
1236 {
1237 dst.x = src.x.zwzw - src.x.xyxy;
1238 dst.y = src.y.zwzw - src.y.xyxy;
1239 dst.z = src.z.zwzw - src.z.xyxy;
1240 dst.w = src.w.zwzw - src.w.xyxy;
1241 }
1242
1243 void PixelProgram::FWIDTH(Vector4f &dst, Vector4f &src)
1244 {
1245 // abs(dFdx(src)) + abs(dFdy(src));
1246 dst.x = Abs(src.x.yyww - src.x.xxzz) + Abs(src.x.zwzw - src.x.xyxy);
1247 dst.y = Abs(src.y.yyww - src.y.xxzz) + Abs(src.y.zwzw - src.y.xyxy);
1248 dst.z = Abs(src.z.yyww - src.z.xxzz) + Abs(src.z.zwzw - src.z.xyxy);
1249 dst.w = Abs(src.w.yyww - src.w.xxzz) + Abs(src.w.zwzw - src.w.xyxy);
1250 }
1251
Nicolas Capens4f172c72016-01-13 08:34:30 -05001252 void PixelProgram::BREAK()
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001253 {
Nicolas Capensc8b67a42016-09-25 15:02:52 -04001254 BasicBlock *deadBlock = Nucleus::createBasicBlock();
1255 BasicBlock *endBlock = loopRepEndBlock[loopRepDepth - 1];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001256
1257 if(breakDepth == 0)
1258 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001259 enableIndex = enableIndex - breakDepth;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001260 Nucleus::createBr(endBlock);
1261 }
1262 else
1263 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001264 enableBreak = enableBreak & ~enableStack[enableIndex];
1265 Bool allBreak = SignMask(enableBreak) == 0x0;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001266
Nicolas Capens4f172c72016-01-13 08:34:30 -05001267 enableIndex = enableIndex - breakDepth;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001268 branch(allBreak, endBlock, deadBlock);
1269 }
1270
1271 Nucleus::setInsertBlock(deadBlock);
Nicolas Capens4f172c72016-01-13 08:34:30 -05001272 enableIndex = enableIndex + breakDepth;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001273 }
1274
Nicolas Capens4f172c72016-01-13 08:34:30 -05001275 void PixelProgram::BREAKC(Vector4f &src0, Vector4f &src1, Control control)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001276 {
1277 Int4 condition;
1278
1279 switch(control)
1280 {
1281 case Shader::CONTROL_GT: condition = CmpNLE(src0.x, src1.x); break;
1282 case Shader::CONTROL_EQ: condition = CmpEQ(src0.x, src1.x); break;
1283 case Shader::CONTROL_GE: condition = CmpNLT(src0.x, src1.x); break;
1284 case Shader::CONTROL_LT: condition = CmpLT(src0.x, src1.x); break;
1285 case Shader::CONTROL_NE: condition = CmpNEQ(src0.x, src1.x); break;
1286 case Shader::CONTROL_LE: condition = CmpLE(src0.x, src1.x); break;
1287 default:
1288 ASSERT(false);
1289 }
1290
Nicolas Capens4f172c72016-01-13 08:34:30 -05001291 BREAK(condition);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001292 }
1293
Nicolas Capens4f172c72016-01-13 08:34:30 -05001294 void PixelProgram::BREAKP(const Src &predicateRegister) // FIXME: Factor out parts common with BREAKC
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001295 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001296 Int4 condition = As<Int4>(p0[predicateRegister.swizzle & 0x3]);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001297
1298 if(predicateRegister.modifier == Shader::MODIFIER_NOT)
1299 {
1300 condition = ~condition;
1301 }
1302
Nicolas Capens4f172c72016-01-13 08:34:30 -05001303 BREAK(condition);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001304 }
1305
Nicolas Capens4f172c72016-01-13 08:34:30 -05001306 void PixelProgram::BREAK(Int4 &condition)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001307 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001308 condition &= enableStack[enableIndex];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001309
Nicolas Capensc8b67a42016-09-25 15:02:52 -04001310 BasicBlock *continueBlock = Nucleus::createBasicBlock();
1311 BasicBlock *endBlock = loopRepEndBlock[loopRepDepth - 1];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001312
Nicolas Capens4f172c72016-01-13 08:34:30 -05001313 enableBreak = enableBreak & ~condition;
1314 Bool allBreak = SignMask(enableBreak) == 0x0;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001315
Nicolas Capens4f172c72016-01-13 08:34:30 -05001316 enableIndex = enableIndex - breakDepth;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001317 branch(allBreak, endBlock, continueBlock);
1318
1319 Nucleus::setInsertBlock(continueBlock);
Nicolas Capens4f172c72016-01-13 08:34:30 -05001320 enableIndex = enableIndex + breakDepth;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001321 }
1322
Nicolas Capens4f172c72016-01-13 08:34:30 -05001323 void PixelProgram::CONTINUE()
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001324 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001325 enableContinue = enableContinue & ~enableStack[enableIndex];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001326 }
1327
1328 void PixelProgram::TEST()
1329 {
1330 whileTest = true;
1331 }
1332
Nicolas Capens4f172c72016-01-13 08:34:30 -05001333 void PixelProgram::CALL(int labelIndex, int callSiteIndex)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001334 {
1335 if(!labelBlock[labelIndex])
1336 {
1337 labelBlock[labelIndex] = Nucleus::createBasicBlock();
1338 }
1339
1340 if(callRetBlock[labelIndex].size() > 1)
1341 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001342 callStack[stackIndex++] = UInt(callSiteIndex);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001343 }
1344
Nicolas Capens4f172c72016-01-13 08:34:30 -05001345 Int4 restoreLeave = enableLeave;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001346
1347 Nucleus::createBr(labelBlock[labelIndex]);
1348 Nucleus::setInsertBlock(callRetBlock[labelIndex][callSiteIndex]);
1349
Nicolas Capens4f172c72016-01-13 08:34:30 -05001350 enableLeave = restoreLeave;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001351 }
1352
Nicolas Capens4f172c72016-01-13 08:34:30 -05001353 void PixelProgram::CALLNZ(int labelIndex, int callSiteIndex, const Src &src)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001354 {
1355 if(src.type == Shader::PARAMETER_CONSTBOOL)
1356 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001357 CALLNZb(labelIndex, callSiteIndex, src);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001358 }
1359 else if(src.type == Shader::PARAMETER_PREDICATE)
1360 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001361 CALLNZp(labelIndex, callSiteIndex, src);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001362 }
1363 else ASSERT(false);
1364 }
1365
Nicolas Capens4f172c72016-01-13 08:34:30 -05001366 void PixelProgram::CALLNZb(int labelIndex, int callSiteIndex, const Src &boolRegister)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001367 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001368 Bool condition = (*Pointer<Byte>(data + OFFSET(DrawData, ps.b[boolRegister.index])) != Byte(0)); // FIXME
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001369
1370 if(boolRegister.modifier == Shader::MODIFIER_NOT)
1371 {
1372 condition = !condition;
1373 }
1374
1375 if(!labelBlock[labelIndex])
1376 {
1377 labelBlock[labelIndex] = Nucleus::createBasicBlock();
1378 }
1379
1380 if(callRetBlock[labelIndex].size() > 1)
1381 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001382 callStack[stackIndex++] = UInt(callSiteIndex);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001383 }
1384
Nicolas Capens4f172c72016-01-13 08:34:30 -05001385 Int4 restoreLeave = enableLeave;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001386
1387 branch(condition, labelBlock[labelIndex], callRetBlock[labelIndex][callSiteIndex]);
1388 Nucleus::setInsertBlock(callRetBlock[labelIndex][callSiteIndex]);
1389
Nicolas Capens4f172c72016-01-13 08:34:30 -05001390 enableLeave = restoreLeave;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001391 }
1392
Nicolas Capens4f172c72016-01-13 08:34:30 -05001393 void PixelProgram::CALLNZp(int labelIndex, int callSiteIndex, const Src &predicateRegister)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001394 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001395 Int4 condition = As<Int4>(p0[predicateRegister.swizzle & 0x3]);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001396
1397 if(predicateRegister.modifier == Shader::MODIFIER_NOT)
1398 {
1399 condition = ~condition;
1400 }
1401
Nicolas Capens4f172c72016-01-13 08:34:30 -05001402 condition &= enableStack[enableIndex];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001403
1404 if(!labelBlock[labelIndex])
1405 {
1406 labelBlock[labelIndex] = Nucleus::createBasicBlock();
1407 }
1408
1409 if(callRetBlock[labelIndex].size() > 1)
1410 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001411 callStack[stackIndex++] = UInt(callSiteIndex);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001412 }
1413
Nicolas Capens4f172c72016-01-13 08:34:30 -05001414 enableIndex++;
1415 enableStack[enableIndex] = condition;
1416 Int4 restoreLeave = enableLeave;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001417
1418 Bool notAllFalse = SignMask(condition) != 0;
1419 branch(notAllFalse, labelBlock[labelIndex], callRetBlock[labelIndex][callSiteIndex]);
1420 Nucleus::setInsertBlock(callRetBlock[labelIndex][callSiteIndex]);
1421
Nicolas Capens4f172c72016-01-13 08:34:30 -05001422 enableIndex--;
1423 enableLeave = restoreLeave;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001424 }
1425
Nicolas Capens4f172c72016-01-13 08:34:30 -05001426 void PixelProgram::ELSE()
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001427 {
1428 ifDepth--;
1429
Nicolas Capensc8b67a42016-09-25 15:02:52 -04001430 BasicBlock *falseBlock = ifFalseBlock[ifDepth];
1431 BasicBlock *endBlock = Nucleus::createBasicBlock();
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001432
1433 if(isConditionalIf[ifDepth])
1434 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001435 Int4 condition = ~enableStack[enableIndex] & enableStack[enableIndex - 1];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001436 Bool notAllFalse = SignMask(condition) != 0;
1437
1438 branch(notAllFalse, falseBlock, endBlock);
1439
Nicolas Capens4f172c72016-01-13 08:34:30 -05001440 enableStack[enableIndex] = ~enableStack[enableIndex] & enableStack[enableIndex - 1];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001441 }
1442 else
1443 {
1444 Nucleus::createBr(endBlock);
1445 Nucleus::setInsertBlock(falseBlock);
1446 }
1447
1448 ifFalseBlock[ifDepth] = endBlock;
1449
1450 ifDepth++;
1451 }
1452
Nicolas Capens4f172c72016-01-13 08:34:30 -05001453 void PixelProgram::ENDIF()
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001454 {
1455 ifDepth--;
1456
Nicolas Capensc8b67a42016-09-25 15:02:52 -04001457 BasicBlock *endBlock = ifFalseBlock[ifDepth];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001458
1459 Nucleus::createBr(endBlock);
1460 Nucleus::setInsertBlock(endBlock);
1461
1462 if(isConditionalIf[ifDepth])
1463 {
1464 breakDepth--;
Nicolas Capens4f172c72016-01-13 08:34:30 -05001465 enableIndex--;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001466 }
1467 }
1468
Nicolas Capens4f172c72016-01-13 08:34:30 -05001469 void PixelProgram::ENDLOOP()
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001470 {
1471 loopRepDepth--;
1472
Nicolas Capens4f172c72016-01-13 08:34:30 -05001473 aL[loopDepth] = aL[loopDepth] + increment[loopDepth]; // FIXME: +=
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001474
Nicolas Capensc8b67a42016-09-25 15:02:52 -04001475 BasicBlock *testBlock = loopRepTestBlock[loopRepDepth];
1476 BasicBlock *endBlock = loopRepEndBlock[loopRepDepth];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001477
1478 Nucleus::createBr(testBlock);
1479 Nucleus::setInsertBlock(endBlock);
1480
Nicolas Capens4f172c72016-01-13 08:34:30 -05001481 loopDepth--;
1482 enableBreak = Int4(0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001483 }
1484
Nicolas Capens4f172c72016-01-13 08:34:30 -05001485 void PixelProgram::ENDREP()
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001486 {
1487 loopRepDepth--;
1488
Nicolas Capensc8b67a42016-09-25 15:02:52 -04001489 BasicBlock *testBlock = loopRepTestBlock[loopRepDepth];
1490 BasicBlock *endBlock = loopRepEndBlock[loopRepDepth];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001491
1492 Nucleus::createBr(testBlock);
1493 Nucleus::setInsertBlock(endBlock);
1494
Nicolas Capens4f172c72016-01-13 08:34:30 -05001495 loopDepth--;
1496 enableBreak = Int4(0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001497 }
1498
Nicolas Capens4f172c72016-01-13 08:34:30 -05001499 void PixelProgram::ENDWHILE()
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001500 {
1501 loopRepDepth--;
1502
Nicolas Capensc8b67a42016-09-25 15:02:52 -04001503 BasicBlock *testBlock = loopRepTestBlock[loopRepDepth];
1504 BasicBlock *endBlock = loopRepEndBlock[loopRepDepth];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001505
1506 Nucleus::createBr(testBlock);
1507 Nucleus::setInsertBlock(endBlock);
1508
Nicolas Capens4f172c72016-01-13 08:34:30 -05001509 enableIndex--;
1510 enableBreak = Int4(0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001511 whileTest = false;
1512 }
1513
Alexis Hetu9aa83a92016-05-02 17:34:46 -04001514 void PixelProgram::ENDSWITCH()
1515 {
1516 loopRepDepth--;
1517
Nicolas Capensc8b67a42016-09-25 15:02:52 -04001518 BasicBlock *endBlock = loopRepEndBlock[loopRepDepth];
Alexis Hetu9aa83a92016-05-02 17:34:46 -04001519
Nicolas Capensac6d5052018-01-05 15:34:00 -05001520 Nucleus::createBr(endBlock);
Alexis Hetu9aa83a92016-05-02 17:34:46 -04001521 Nucleus::setInsertBlock(endBlock);
1522
Alexis Hetu9aa83a92016-05-02 17:34:46 -04001523 enableBreak = Int4(0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF);
1524 }
1525
Nicolas Capens4f172c72016-01-13 08:34:30 -05001526 void PixelProgram::IF(const Src &src)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001527 {
1528 if(src.type == Shader::PARAMETER_CONSTBOOL)
1529 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001530 IFb(src);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001531 }
1532 else if(src.type == Shader::PARAMETER_PREDICATE)
1533 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001534 IFp(src);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001535 }
1536 else
1537 {
Nicolas Capensc2534f42016-04-04 11:13:24 -04001538 Int4 condition = As<Int4>(fetchRegister(src).x);
Nicolas Capens4f172c72016-01-13 08:34:30 -05001539 IF(condition);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001540 }
1541 }
1542
Nicolas Capens4f172c72016-01-13 08:34:30 -05001543 void PixelProgram::IFb(const Src &boolRegister)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001544 {
1545 ASSERT(ifDepth < 24 + 4);
1546
Nicolas Capens4f172c72016-01-13 08:34:30 -05001547 Bool condition = (*Pointer<Byte>(data + OFFSET(DrawData, ps.b[boolRegister.index])) != Byte(0)); // FIXME
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001548
1549 if(boolRegister.modifier == Shader::MODIFIER_NOT)
1550 {
1551 condition = !condition;
1552 }
1553
Nicolas Capensc8b67a42016-09-25 15:02:52 -04001554 BasicBlock *trueBlock = Nucleus::createBasicBlock();
1555 BasicBlock *falseBlock = Nucleus::createBasicBlock();
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001556
1557 branch(condition, trueBlock, falseBlock);
1558
1559 isConditionalIf[ifDepth] = false;
1560 ifFalseBlock[ifDepth] = falseBlock;
1561
1562 ifDepth++;
1563 }
1564
Nicolas Capens4f172c72016-01-13 08:34:30 -05001565 void PixelProgram::IFp(const Src &predicateRegister)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001566 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001567 Int4 condition = As<Int4>(p0[predicateRegister.swizzle & 0x3]);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001568
1569 if(predicateRegister.modifier == Shader::MODIFIER_NOT)
1570 {
1571 condition = ~condition;
1572 }
1573
Nicolas Capens4f172c72016-01-13 08:34:30 -05001574 IF(condition);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001575 }
1576
Nicolas Capens4f172c72016-01-13 08:34:30 -05001577 void PixelProgram::IFC(Vector4f &src0, Vector4f &src1, Control control)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001578 {
1579 Int4 condition;
1580
1581 switch(control)
1582 {
1583 case Shader::CONTROL_GT: condition = CmpNLE(src0.x, src1.x); break;
1584 case Shader::CONTROL_EQ: condition = CmpEQ(src0.x, src1.x); break;
1585 case Shader::CONTROL_GE: condition = CmpNLT(src0.x, src1.x); break;
1586 case Shader::CONTROL_LT: condition = CmpLT(src0.x, src1.x); break;
1587 case Shader::CONTROL_NE: condition = CmpNEQ(src0.x, src1.x); break;
1588 case Shader::CONTROL_LE: condition = CmpLE(src0.x, src1.x); break;
1589 default:
1590 ASSERT(false);
1591 }
1592
Nicolas Capens4f172c72016-01-13 08:34:30 -05001593 IF(condition);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001594 }
1595
Nicolas Capens4f172c72016-01-13 08:34:30 -05001596 void PixelProgram::IF(Int4 &condition)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001597 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001598 condition &= enableStack[enableIndex];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001599
Nicolas Capens4f172c72016-01-13 08:34:30 -05001600 enableIndex++;
1601 enableStack[enableIndex] = condition;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001602
Nicolas Capensc8b67a42016-09-25 15:02:52 -04001603 BasicBlock *trueBlock = Nucleus::createBasicBlock();
1604 BasicBlock *falseBlock = Nucleus::createBasicBlock();
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001605
1606 Bool notAllFalse = SignMask(condition) != 0;
1607
1608 branch(notAllFalse, trueBlock, falseBlock);
1609
1610 isConditionalIf[ifDepth] = true;
1611 ifFalseBlock[ifDepth] = falseBlock;
1612
1613 ifDepth++;
1614 breakDepth++;
1615 }
1616
1617 void PixelProgram::LABEL(int labelIndex)
1618 {
1619 if(!labelBlock[labelIndex])
1620 {
1621 labelBlock[labelIndex] = Nucleus::createBasicBlock();
1622 }
1623
1624 Nucleus::setInsertBlock(labelBlock[labelIndex]);
1625 currentLabel = labelIndex;
1626 }
1627
Nicolas Capens4f172c72016-01-13 08:34:30 -05001628 void PixelProgram::LOOP(const Src &integerRegister)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001629 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001630 loopDepth++;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001631
Nicolas Capens4f172c72016-01-13 08:34:30 -05001632 iteration[loopDepth] = *Pointer<Int>(data + OFFSET(DrawData, ps.i[integerRegister.index][0]));
1633 aL[loopDepth] = *Pointer<Int>(data + OFFSET(DrawData, ps.i[integerRegister.index][1]));
1634 increment[loopDepth] = *Pointer<Int>(data + OFFSET(DrawData, ps.i[integerRegister.index][2]));
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001635
Nicolas Capens4f172c72016-01-13 08:34:30 -05001636 // If(increment[loopDepth] == 0)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001637 // {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001638 // increment[loopDepth] = 1;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001639 // }
1640
Nicolas Capensc8b67a42016-09-25 15:02:52 -04001641 BasicBlock *loopBlock = Nucleus::createBasicBlock();
1642 BasicBlock *testBlock = Nucleus::createBasicBlock();
1643 BasicBlock *endBlock = Nucleus::createBasicBlock();
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001644
1645 loopRepTestBlock[loopRepDepth] = testBlock;
1646 loopRepEndBlock[loopRepDepth] = endBlock;
1647
1648 // FIXME: jump(testBlock)
1649 Nucleus::createBr(testBlock);
1650 Nucleus::setInsertBlock(testBlock);
1651
Nicolas Capens4f172c72016-01-13 08:34:30 -05001652 branch(iteration[loopDepth] > 0, loopBlock, endBlock);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001653 Nucleus::setInsertBlock(loopBlock);
1654
Nicolas Capens4f172c72016-01-13 08:34:30 -05001655 iteration[loopDepth] = iteration[loopDepth] - 1; // FIXME: --
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001656
1657 loopRepDepth++;
1658 breakDepth = 0;
1659 }
1660
Nicolas Capens4f172c72016-01-13 08:34:30 -05001661 void PixelProgram::REP(const Src &integerRegister)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001662 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001663 loopDepth++;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001664
Nicolas Capens4f172c72016-01-13 08:34:30 -05001665 iteration[loopDepth] = *Pointer<Int>(data + OFFSET(DrawData, ps.i[integerRegister.index][0]));
1666 aL[loopDepth] = aL[loopDepth - 1];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001667
Nicolas Capensc8b67a42016-09-25 15:02:52 -04001668 BasicBlock *loopBlock = Nucleus::createBasicBlock();
1669 BasicBlock *testBlock = Nucleus::createBasicBlock();
1670 BasicBlock *endBlock = Nucleus::createBasicBlock();
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001671
1672 loopRepTestBlock[loopRepDepth] = testBlock;
1673 loopRepEndBlock[loopRepDepth] = endBlock;
1674
1675 // FIXME: jump(testBlock)
1676 Nucleus::createBr(testBlock);
1677 Nucleus::setInsertBlock(testBlock);
1678
Nicolas Capens4f172c72016-01-13 08:34:30 -05001679 branch(iteration[loopDepth] > 0, loopBlock, endBlock);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001680 Nucleus::setInsertBlock(loopBlock);
1681
Nicolas Capens4f172c72016-01-13 08:34:30 -05001682 iteration[loopDepth] = iteration[loopDepth] - 1; // FIXME: --
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001683
1684 loopRepDepth++;
1685 breakDepth = 0;
1686 }
1687
Nicolas Capens4f172c72016-01-13 08:34:30 -05001688 void PixelProgram::WHILE(const Src &temporaryRegister)
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001689 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001690 enableIndex++;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001691
Nicolas Capensc8b67a42016-09-25 15:02:52 -04001692 BasicBlock *loopBlock = Nucleus::createBasicBlock();
1693 BasicBlock *testBlock = Nucleus::createBasicBlock();
1694 BasicBlock *endBlock = Nucleus::createBasicBlock();
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001695
1696 loopRepTestBlock[loopRepDepth] = testBlock;
1697 loopRepEndBlock[loopRepDepth] = endBlock;
1698
Nicolas Capens4f172c72016-01-13 08:34:30 -05001699 Int4 restoreBreak = enableBreak;
1700 Int4 restoreContinue = enableContinue;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001701
1702 // FIXME: jump(testBlock)
1703 Nucleus::createBr(testBlock);
1704 Nucleus::setInsertBlock(testBlock);
Nicolas Capens4f172c72016-01-13 08:34:30 -05001705 enableContinue = restoreContinue;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001706
Nicolas Capensc2534f42016-04-04 11:13:24 -04001707 const Vector4f &src = fetchRegister(temporaryRegister);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001708 Int4 condition = As<Int4>(src.x);
Nicolas Capens4f172c72016-01-13 08:34:30 -05001709 condition &= enableStack[enableIndex - 1];
Nicolas Capens2ff29482016-04-28 15:28:02 -04001710 if(shader->containsLeaveInstruction()) condition &= enableLeave;
Nicolas Capens4f172c72016-01-13 08:34:30 -05001711 enableStack[enableIndex] = condition;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001712
1713 Bool notAllFalse = SignMask(condition) != 0;
1714 branch(notAllFalse, loopBlock, endBlock);
1715
1716 Nucleus::setInsertBlock(endBlock);
Nicolas Capens4f172c72016-01-13 08:34:30 -05001717 enableBreak = restoreBreak;
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001718
1719 Nucleus::setInsertBlock(loopBlock);
1720
1721 loopRepDepth++;
1722 breakDepth = 0;
1723 }
1724
Alexis Hetu9aa83a92016-05-02 17:34:46 -04001725 void PixelProgram::SWITCH()
1726 {
Nicolas Capensc8b67a42016-09-25 15:02:52 -04001727 BasicBlock *endBlock = Nucleus::createBasicBlock();
Alexis Hetu9aa83a92016-05-02 17:34:46 -04001728
1729 loopRepTestBlock[loopRepDepth] = nullptr;
1730 loopRepEndBlock[loopRepDepth] = endBlock;
1731
1732 loopRepDepth++;
1733 breakDepth = 0;
1734 }
1735
Nicolas Capens4f172c72016-01-13 08:34:30 -05001736 void PixelProgram::RET()
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001737 {
1738 if(currentLabel == -1)
1739 {
1740 returnBlock = Nucleus::createBasicBlock();
1741 Nucleus::createBr(returnBlock);
1742 }
1743 else
1744 {
Nicolas Capensc8b67a42016-09-25 15:02:52 -04001745 BasicBlock *unreachableBlock = Nucleus::createBasicBlock();
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001746
1747 if(callRetBlock[currentLabel].size() > 1) // Pop the return destination from the call stack
1748 {
1749 // FIXME: Encapsulate
Nicolas Capens4f172c72016-01-13 08:34:30 -05001750 UInt index = callStack[--stackIndex];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001751
Nicolas Capens19336542016-09-26 10:32:29 -04001752 Value *value = index.loadValue();
Nicolas Capensb98fe5c2016-11-09 12:24:06 -05001753 SwitchCases *switchCases = Nucleus::createSwitch(value, unreachableBlock, (int)callRetBlock[currentLabel].size());
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001754
1755 for(unsigned int i = 0; i < callRetBlock[currentLabel].size(); i++)
1756 {
Nicolas Capensb98fe5c2016-11-09 12:24:06 -05001757 Nucleus::addSwitchCase(switchCases, i, callRetBlock[currentLabel][i]);
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001758 }
1759 }
1760 else if(callRetBlock[currentLabel].size() == 1) // Jump directly to the unique return destination
1761 {
1762 Nucleus::createBr(callRetBlock[currentLabel][0]);
1763 }
1764 else // Function isn't called
1765 {
1766 Nucleus::createBr(unreachableBlock);
1767 }
1768
1769 Nucleus::setInsertBlock(unreachableBlock);
1770 Nucleus::createUnreachable();
1771 }
1772 }
1773
Nicolas Capens4f172c72016-01-13 08:34:30 -05001774 void PixelProgram::LEAVE()
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001775 {
Nicolas Capens4f172c72016-01-13 08:34:30 -05001776 enableLeave = enableLeave & ~enableStack[enableIndex];
Alexis Hetuf2a8c372015-07-13 11:08:41 -04001777
1778 // FIXME: Return from function if all instances left
1779 // FIXME: Use enableLeave in other control-flow constructs
1780 }
Nicolas Capens5d961882016-01-01 23:18:14 -05001781}