1 /*
2  * Copyright 2016 Google Inc.
3  *
4  * Use of this source code is governed by a BSD-style license that can be
5  * found in the LICENSE file.
6  */
7 
8 #ifndef SKSL_CONTEXT
9 #define SKSL_CONTEXT
10 
11 #include "ir/SkSLType.h"
12 #include "ir/SkSLExpression.h"
13 
14 namespace SkSL {
15 
16 /**
17  * Contains compiler-wide objects, which currently means the core types.
18  */
19 class Context {
20 public:
Context()21     Context()
22     : fInvalid_Type(new Type("<INVALID>"))
23     , fVoid_Type(new Type("void"))
24     , fDouble_Type(new Type("double", Type::kFloat_NumberKind, 4))
25     , fDouble2_Type(new Type("double2", *fDouble_Type, 2))
26     , fDouble3_Type(new Type("double3", *fDouble_Type, 3))
27     , fDouble4_Type(new Type("double4", *fDouble_Type, 4))
28     , fFloat_Type(new Type("float", Type::kFloat_NumberKind, 3))
29     , fFloat2_Type(new Type("float2", *fFloat_Type, 2))
30     , fFloat3_Type(new Type("float3", *fFloat_Type, 3))
31     , fFloat4_Type(new Type("float4", *fFloat_Type, 4))
32     , fHalf_Type(new Type("half", Type::kFloat_NumberKind, 2))
33     , fHalf2_Type(new Type("half2", *fHalf_Type, 2))
34     , fHalf3_Type(new Type("half3", *fHalf_Type, 3))
35     , fHalf4_Type(new Type("half4", *fHalf_Type, 4))
36     , fUInt_Type(new Type("uint", Type::kUnsigned_NumberKind, 1))
37     , fUInt2_Type(new Type("uint2", *fUInt_Type, 2))
38     , fUInt3_Type(new Type("uint3", *fUInt_Type, 3))
39     , fUInt4_Type(new Type("uint4", *fUInt_Type, 4))
40     , fInt_Type(new Type("int", Type::kSigned_NumberKind, 1))
41     , fInt2_Type(new Type("int2", *fInt_Type, 2))
42     , fInt3_Type(new Type("int3", *fInt_Type, 3))
43     , fInt4_Type(new Type("int4", *fInt_Type, 4))
44     , fUShort_Type(new Type("ushort", Type::kUnsigned_NumberKind, 0))
45     , fUShort2_Type(new Type("ushort2", *fUShort_Type, 2))
46     , fUShort3_Type(new Type("ushort3", *fUShort_Type, 3))
47     , fUShort4_Type(new Type("ushort4", *fUShort_Type, 4))
48     , fShort_Type(new Type("short", Type::kSigned_NumberKind, 0))
49     , fShort2_Type(new Type("short2", *fShort_Type, 2))
50     , fShort3_Type(new Type("short3", *fShort_Type, 3))
51     , fShort4_Type(new Type("short4", *fShort_Type, 4))
52     , fBool_Type(new Type("bool", Type::kNonnumeric_NumberKind, -1))
53     , fBool2_Type(new Type("bool2", *fBool_Type, 2))
54     , fBool3_Type(new Type("bool3", *fBool_Type, 3))
55     , fBool4_Type(new Type("bool4", *fBool_Type, 4))
56     , fFloat2x2_Type(new Type("float2x2", *fFloat_Type, 2, 2))
57     , fFloat2x3_Type(new Type("float2x3", *fFloat_Type, 2, 3))
58     , fFloat2x4_Type(new Type("float2x4", *fFloat_Type, 2, 4))
59     , fFloat3x2_Type(new Type("float3x2", *fFloat_Type, 3, 2))
60     , fFloat3x3_Type(new Type("float3x3", *fFloat_Type, 3, 3))
61     , fFloat3x4_Type(new Type("float3x4", *fFloat_Type, 3, 4))
62     , fFloat4x2_Type(new Type("float4x2", *fFloat_Type, 4, 2))
63     , fFloat4x3_Type(new Type("float4x3", *fFloat_Type, 4, 3))
64     , fFloat4x4_Type(new Type("float4x4", *fFloat_Type, 4, 4))
65     , fHalf2x2_Type(new Type("half2x2", *fHalf_Type, 2, 2))
66     , fHalf2x3_Type(new Type("half2x3", *fHalf_Type, 2, 3))
67     , fHalf2x4_Type(new Type("half2x4", *fHalf_Type, 2, 4))
68     , fHalf3x2_Type(new Type("half3x2", *fHalf_Type, 3, 2))
69     , fHalf3x3_Type(new Type("half3x3", *fHalf_Type, 3, 3))
70     , fHalf3x4_Type(new Type("half3x4", *fHalf_Type, 3, 4))
71     , fHalf4x2_Type(new Type("half4x2", *fHalf_Type, 4, 2))
72     , fHalf4x3_Type(new Type("half4x3", *fHalf_Type, 4, 3))
73     , fHalf4x4_Type(new Type("half4x4", *fHalf_Type, 4, 4))
74     , fDouble2x2_Type(new Type("double2x2", *fDouble_Type, 2, 2))
75     , fDouble2x3_Type(new Type("double2x3", *fDouble_Type, 2, 3))
76     , fDouble2x4_Type(new Type("double2x4", *fDouble_Type, 2, 4))
77     , fDouble3x2_Type(new Type("double3x2", *fDouble_Type, 3, 2))
78     , fDouble3x3_Type(new Type("double3x3", *fDouble_Type, 3, 3))
79     , fDouble3x4_Type(new Type("double3x4", *fDouble_Type, 3, 4))
80     , fDouble4x2_Type(new Type("double4x2", *fDouble_Type, 4, 2))
81     , fDouble4x3_Type(new Type("double4x3", *fDouble_Type, 4, 3))
82     , fDouble4x4_Type(new Type("double4x4", *fDouble_Type, 4, 4))
83     , fSampler1D_Type(new Type("sampler1D", SpvDim1D, false, false, false, true))
84     , fSampler2D_Type(new Type("sampler2D", SpvDim2D, false, false, false, true))
85     , fSampler3D_Type(new Type("sampler3D", SpvDim3D, false, false, false, true))
86     , fSamplerExternalOES_Type(new Type("samplerExternalOES", SpvDim2D, false, false,
87                                         false, true))
88     , fSamplerCube_Type(new Type("samplerCube", SpvDimCube, false, false, false, true))
89     , fSampler2DRect_Type(new Type("sampler2DRect", SpvDimRect, false, false, false, true))
90     , fSampler1DArray_Type(new Type("sampler1DArray"))
91     , fSampler2DArray_Type(new Type("sampler2DArray"))
92     , fSamplerCubeArray_Type(new Type("samplerCubeArray"))
93     , fSamplerBuffer_Type(new Type("samplerBuffer", SpvDimBuffer, false, false, false,
94                                    true))
95     , fSampler2DMS_Type(new Type("sampler2DMS"))
96     , fSampler2DMSArray_Type(new Type("sampler2DMSArray"))
97     , fSampler1DShadow_Type(new Type("sampler1DShadow"))
98     , fSampler2DShadow_Type(new Type("sampler2DShadow"))
99     , fSamplerCubeShadow_Type(new Type("samplerCubeShadow"))
100     , fSampler2DRectShadow_Type(new Type("sampler2DRectShadow"))
101     , fSampler1DArrayShadow_Type(new Type("sampler1DArrayShadow"))
102     , fSampler2DArrayShadow_Type(new Type("sampler2DArrayShadow"))
103     , fSamplerCubeArrayShadow_Type(new Type("samplerCubeArrayShadow"))
104 
105     // Related to below FIXME, gsampler*s don't currently expand to cover integer case.
106     , fISampler2D_Type(new Type("isampler2D", SpvDim2D, false, false, false, true))
107 
108     // FIXME express these as "gimage2D" that expand to image2D, iimage2D, and uimage2D.
109     , fImage2D_Type(new Type("image2D", SpvDim2D, false, false, false, true))
110     , fIImage2D_Type(new Type("iimage2D", SpvDim2D, false, false, false, true))
111 
112     // FIXME express these as "gsubpassInput" that expand to subpassInput, isubpassInput,
113     // and usubpassInput.
114     , fSubpassInput_Type(new Type("subpassInput", SpvDimSubpassData, false, false,
115                                   false, false))
116     , fSubpassInputMS_Type(new Type("subpassInputMS", SpvDimSubpassData, false, false,
117                                     true, false))
118 
119     // FIXME figure out what we're supposed to do with the gsampler et al. types)
120     , fGSampler1D_Type(new Type("$gsampler1D", static_type(*fSampler1D_Type)))
121     , fGSampler2D_Type(new Type("$gsampler2D", static_type(*fSampler2D_Type)))
122     , fGSampler3D_Type(new Type("$gsampler3D", static_type(*fSampler3D_Type)))
123     , fGSamplerCube_Type(new Type("$gsamplerCube", static_type(*fSamplerCube_Type)))
124     , fGSampler2DRect_Type(new Type("$gsampler2DRect", static_type(*fSampler2DRect_Type)))
125     , fGSampler1DArray_Type(new Type("$gsampler1DArray",
126                                      static_type(*fSampler1DArray_Type)))
127     , fGSampler2DArray_Type(new Type("$gsampler2DArray",
128                                      static_type(*fSampler2DArray_Type)))
129     , fGSamplerCubeArray_Type(new Type("$gsamplerCubeArray",
130                                        static_type(*fSamplerCubeArray_Type)))
131     , fGSamplerBuffer_Type(new Type("$gsamplerBuffer", static_type(*fSamplerBuffer_Type)))
132     , fGSampler2DMS_Type(new Type("$gsampler2DMS", static_type(*fSampler2DMS_Type)))
133     , fGSampler2DMSArray_Type(new Type("$gsampler2DMSArray",
134                                        static_type(*fSampler2DMSArray_Type)))
135     , fGSampler2DArrayShadow_Type(new Type("$gsampler2DArrayShadow",
136                                            static_type(*fSampler2DArrayShadow_Type)))
137     , fGSamplerCubeArrayShadow_Type(new Type("$gsamplerCubeArrayShadow",
138                                              static_type(*fSamplerCubeArrayShadow_Type)))
139     , fGenType_Type(new Type("$genType", { fFloat_Type.get(), fFloat2_Type.get(),
140                                            fFloat3_Type.get(), fFloat4_Type.get() }))
141     , fGenHType_Type(new Type("$genHType", { fHalf_Type.get(), fHalf2_Type.get(),
142                                              fHalf3_Type.get(), fHalf4_Type.get() }))
143     , fGenDType_Type(new Type("$genDType", { fDouble_Type.get(), fDouble2_Type.get(),
144                                              fDouble3_Type.get(), fDouble4_Type.get() }))
145     , fGenIType_Type(new Type("$genIType", { fInt_Type.get(), fInt2_Type.get(),
146                                              fInt3_Type.get(), fInt4_Type.get() }))
147     , fGenUType_Type(new Type("$genUType", { fUInt_Type.get(), fUInt2_Type.get(),
148                                              fUInt3_Type.get(), fUInt4_Type.get() }))
149     , fGenBType_Type(new Type("$genBType", { fBool_Type.get(), fBool2_Type.get(),
150                                              fBool3_Type.get(), fBool4_Type.get() }))
151     , fMat_Type(new Type("$mat", { fFloat2x2_Type.get(),  fFloat2x3_Type.get(),
152                                    fFloat2x4_Type.get(),  fFloat3x2_Type.get(),
153                                    fFloat3x3_Type.get(),  fFloat3x4_Type.get(),
154                                    fFloat4x2_Type.get(),  fFloat4x3_Type.get(),
155                                    fFloat4x4_Type.get(),  fHalf2x2_Type.get(),
156                                    fHalf2x3_Type.get(),   fHalf2x4_Type.get(),
157                                    fHalf3x2_Type.get(),   fHalf3x3_Type.get(),
158                                    fHalf3x4_Type.get(),   fHalf4x2_Type.get(),
159                                    fHalf4x3_Type.get(),   fHalf4x4_Type.get(),
160                                    fDouble2x2_Type.get(), fDouble2x3_Type.get(),
161                                    fDouble2x4_Type.get(), fDouble3x2_Type.get(),
162                                    fDouble3x3_Type.get(), fDouble3x4_Type.get(),
163                                    fDouble4x2_Type.get(), fDouble4x3_Type.get(),
164                                    fDouble4x4_Type.get() }))
165     , fVec_Type(new Type("$vec", { fInvalid_Type.get(), fFloat2_Type.get(),
166                                            fFloat3_Type.get(), fFloat4_Type.get() }))
167     , fGVec_Type(new Type("$gvec"))
168     , fGVec2_Type(new Type("$gfloat2"))
169     , fGVec3_Type(new Type("$gfloat3"))
170     , fGVec4_Type(new Type("$gfloat4", static_type(*fFloat4_Type)))
171     , fHVec_Type(new Type("$hvec", { fInvalid_Type.get(), fHalf2_Type.get(),
172                                      fHalf3_Type.get(), fHalf4_Type.get() }))
173     , fDVec_Type(new Type("$dvec", { fInvalid_Type.get(), fDouble2_Type.get(),
174                                      fDouble3_Type.get(), fDouble4_Type.get() }))
175     , fIVec_Type(new Type("$ivec", { fInvalid_Type.get(), fInt2_Type.get(),
176                                      fInt3_Type.get(), fInt4_Type.get() }))
177     , fUVec_Type(new Type("$uvec", { fInvalid_Type.get(), fUInt2_Type.get(),
178                                      fUInt3_Type.get(), fUInt4_Type.get() }))
179     , fSVec_Type(new Type("$svec", { fInvalid_Type.get(), fShort2_Type.get(),
180                                      fShort3_Type.get(), fShort4_Type.get() }))
181     , fUSVec_Type(new Type("$usvec", { fInvalid_Type.get(), fUShort2_Type.get(),
182                                        fUShort3_Type.get(), fUShort4_Type.get() }))
183     , fBVec_Type(new Type("$bvec", { fInvalid_Type.get(), fBool2_Type.get(),
184                                      fBool3_Type.get(), fBool4_Type.get() }))
185     , fSkCaps_Type(new Type("$sk_Caps"))
186     , fSkArgs_Type(new Type("$sk_Args"))
187     , fFragmentProcessor_Type(new Type("fragmentProcessor"))
188     , fDefined_Expression(new Defined(*fInvalid_Type)) {}
189 
static_type(const Type & t)190     static std::vector<const Type*> static_type(const Type& t) {
191         return { &t, &t, &t, &t };
192     }
193 
194     const std::unique_ptr<Type> fInvalid_Type;
195     const std::unique_ptr<Type> fVoid_Type;
196 
197     const std::unique_ptr<Type> fDouble_Type;
198     const std::unique_ptr<Type> fDouble2_Type;
199     const std::unique_ptr<Type> fDouble3_Type;
200     const std::unique_ptr<Type> fDouble4_Type;
201 
202     const std::unique_ptr<Type> fFloat_Type;
203     const std::unique_ptr<Type> fFloat2_Type;
204     const std::unique_ptr<Type> fFloat3_Type;
205     const std::unique_ptr<Type> fFloat4_Type;
206 
207     const std::unique_ptr<Type> fHalf_Type;
208     const std::unique_ptr<Type> fHalf2_Type;
209     const std::unique_ptr<Type> fHalf3_Type;
210     const std::unique_ptr<Type> fHalf4_Type;
211 
212     const std::unique_ptr<Type> fUInt_Type;
213     const std::unique_ptr<Type> fUInt2_Type;
214     const std::unique_ptr<Type> fUInt3_Type;
215     const std::unique_ptr<Type> fUInt4_Type;
216 
217     const std::unique_ptr<Type> fInt_Type;
218     const std::unique_ptr<Type> fInt2_Type;
219     const std::unique_ptr<Type> fInt3_Type;
220     const std::unique_ptr<Type> fInt4_Type;
221 
222     const std::unique_ptr<Type> fUShort_Type;
223     const std::unique_ptr<Type> fUShort2_Type;
224     const std::unique_ptr<Type> fUShort3_Type;
225     const std::unique_ptr<Type> fUShort4_Type;
226 
227     const std::unique_ptr<Type> fShort_Type;
228     const std::unique_ptr<Type> fShort2_Type;
229     const std::unique_ptr<Type> fShort3_Type;
230     const std::unique_ptr<Type> fShort4_Type;
231 
232     const std::unique_ptr<Type> fBool_Type;
233     const std::unique_ptr<Type> fBool2_Type;
234     const std::unique_ptr<Type> fBool3_Type;
235     const std::unique_ptr<Type> fBool4_Type;
236 
237     const std::unique_ptr<Type> fFloat2x2_Type;
238     const std::unique_ptr<Type> fFloat2x3_Type;
239     const std::unique_ptr<Type> fFloat2x4_Type;
240     const std::unique_ptr<Type> fFloat3x2_Type;
241     const std::unique_ptr<Type> fFloat3x3_Type;
242     const std::unique_ptr<Type> fFloat3x4_Type;
243     const std::unique_ptr<Type> fFloat4x2_Type;
244     const std::unique_ptr<Type> fFloat4x3_Type;
245     const std::unique_ptr<Type> fFloat4x4_Type;
246 
247     const std::unique_ptr<Type> fHalf2x2_Type;
248     const std::unique_ptr<Type> fHalf2x3_Type;
249     const std::unique_ptr<Type> fHalf2x4_Type;
250     const std::unique_ptr<Type> fHalf3x2_Type;
251     const std::unique_ptr<Type> fHalf3x3_Type;
252     const std::unique_ptr<Type> fHalf3x4_Type;
253     const std::unique_ptr<Type> fHalf4x2_Type;
254     const std::unique_ptr<Type> fHalf4x3_Type;
255     const std::unique_ptr<Type> fHalf4x4_Type;
256 
257     const std::unique_ptr<Type> fDouble2x2_Type;
258     const std::unique_ptr<Type> fDouble2x3_Type;
259     const std::unique_ptr<Type> fDouble2x4_Type;
260     const std::unique_ptr<Type> fDouble3x2_Type;
261     const std::unique_ptr<Type> fDouble3x3_Type;
262     const std::unique_ptr<Type> fDouble3x4_Type;
263     const std::unique_ptr<Type> fDouble4x2_Type;
264     const std::unique_ptr<Type> fDouble4x3_Type;
265     const std::unique_ptr<Type> fDouble4x4_Type;
266 
267     const std::unique_ptr<Type> fSampler1D_Type;
268     const std::unique_ptr<Type> fSampler2D_Type;
269     const std::unique_ptr<Type> fSampler3D_Type;
270     const std::unique_ptr<Type> fSamplerExternalOES_Type;
271     const std::unique_ptr<Type> fSamplerCube_Type;
272     const std::unique_ptr<Type> fSampler2DRect_Type;
273     const std::unique_ptr<Type> fSampler1DArray_Type;
274     const std::unique_ptr<Type> fSampler2DArray_Type;
275     const std::unique_ptr<Type> fSamplerCubeArray_Type;
276     const std::unique_ptr<Type> fSamplerBuffer_Type;
277     const std::unique_ptr<Type> fSampler2DMS_Type;
278     const std::unique_ptr<Type> fSampler2DMSArray_Type;
279     const std::unique_ptr<Type> fSampler1DShadow_Type;
280     const std::unique_ptr<Type> fSampler2DShadow_Type;
281     const std::unique_ptr<Type> fSamplerCubeShadow_Type;
282     const std::unique_ptr<Type> fSampler2DRectShadow_Type;
283     const std::unique_ptr<Type> fSampler1DArrayShadow_Type;
284     const std::unique_ptr<Type> fSampler2DArrayShadow_Type;
285     const std::unique_ptr<Type> fSamplerCubeArrayShadow_Type;
286 
287     const std::unique_ptr<Type> fISampler2D_Type;
288 
289     const std::unique_ptr<Type> fImage2D_Type;
290     const std::unique_ptr<Type> fIImage2D_Type;
291 
292     const std::unique_ptr<Type> fSubpassInput_Type;
293     const std::unique_ptr<Type> fSubpassInputMS_Type;
294 
295     const std::unique_ptr<Type> fGSampler1D_Type;
296     const std::unique_ptr<Type> fGSampler2D_Type;
297     const std::unique_ptr<Type> fGSampler3D_Type;
298     const std::unique_ptr<Type> fGSamplerCube_Type;
299     const std::unique_ptr<Type> fGSampler2DRect_Type;
300     const std::unique_ptr<Type> fGSampler1DArray_Type;
301     const std::unique_ptr<Type> fGSampler2DArray_Type;
302     const std::unique_ptr<Type> fGSamplerCubeArray_Type;
303     const std::unique_ptr<Type> fGSamplerBuffer_Type;
304     const std::unique_ptr<Type> fGSampler2DMS_Type;
305     const std::unique_ptr<Type> fGSampler2DMSArray_Type;
306     const std::unique_ptr<Type> fGSampler2DArrayShadow_Type;
307     const std::unique_ptr<Type> fGSamplerCubeArrayShadow_Type;
308 
309     const std::unique_ptr<Type> fGenType_Type;
310     const std::unique_ptr<Type> fGenHType_Type;
311     const std::unique_ptr<Type> fGenDType_Type;
312     const std::unique_ptr<Type> fGenIType_Type;
313     const std::unique_ptr<Type> fGenUType_Type;
314     const std::unique_ptr<Type> fGenBType_Type;
315 
316     const std::unique_ptr<Type> fMat_Type;
317 
318     const std::unique_ptr<Type> fVec_Type;
319 
320     const std::unique_ptr<Type> fGVec_Type;
321     const std::unique_ptr<Type> fGVec2_Type;
322     const std::unique_ptr<Type> fGVec3_Type;
323     const std::unique_ptr<Type> fGVec4_Type;
324     const std::unique_ptr<Type> fHVec_Type;
325     const std::unique_ptr<Type> fDVec_Type;
326     const std::unique_ptr<Type> fIVec_Type;
327     const std::unique_ptr<Type> fUVec_Type;
328     const std::unique_ptr<Type> fSVec_Type;
329     const std::unique_ptr<Type> fUSVec_Type;
330 
331     const std::unique_ptr<Type> fBVec_Type;
332 
333     const std::unique_ptr<Type> fSkCaps_Type;
334     const std::unique_ptr<Type> fSkArgs_Type;
335     const std::unique_ptr<Type> fFragmentProcessor_Type;
336 
337     // dummy expression used to mark that a variable has a value during dataflow analysis (when it
338     // could have several different values, or the analyzer is otherwise unable to assign it a
339     // specific expression)
340     const std::unique_ptr<Expression> fDefined_Expression;
341 
342 private:
343     class Defined : public Expression {
344     public:
Defined(const Type & type)345         Defined(const Type& type)
346         : INHERITED(-1, kDefined_Kind, type) {}
347 
hasSideEffects()348         bool hasSideEffects() const override {
349             return false;
350         }
351 
description()352         String description() const override {
353             return "<defined>";
354         }
355 
356         typedef Expression INHERITED;
357     };
358 };
359 
360 } // namespace
361 
362 #endif
363