OLD | NEW |
1 /* | 1 /* |
2 * Copyright 2016 Google Inc. | 2 * Copyright 2016 Google Inc. |
3 * | 3 * |
4 * Use of this source code is governed by a BSD-style license that can be | 4 * Use of this source code is governed by a BSD-style license that can be |
5 * found in the LICENSE file. | 5 * found in the LICENSE file. |
6 */ | 6 */ |
7 | 7 |
8 #ifndef SKSL_INDEX | 8 #ifndef SKSL_INDEX |
9 #define SKSL_INDEX | 9 #define SKSL_INDEX |
10 | 10 |
11 #include "SkSLExpression.h" | 11 #include "SkSLExpression.h" |
12 #include "SkSLUtil.h" | 12 #include "SkSLUtil.h" |
13 | 13 |
14 namespace SkSL { | 14 namespace SkSL { |
15 | 15 |
16 /** | 16 /** |
17 * Given a type, returns the type that will result from extracting an array valu
e from it. | 17 * Given a type, returns the type that will result from extracting an array valu
e from it. |
18 */ | 18 */ |
19 static const Type& index_type(const Context& context, const Type& type) { | 19 static const Type& index_type(const Context& context, const Type& type) { |
20 if (type.kind() == Type::kMatrix_Kind) { | 20 if (type.kind() == Type::kMatrix_Kind) { |
21 if (type.componentType() == *context.fFloat_Type) { | 21 if (type.componentType() == *context.fFloat_Type) { |
22 switch (type.columns()) { | 22 switch (type.rows()) { |
23 case 2: return *context.fVec2_Type; | 23 case 2: return *context.fVec2_Type; |
24 case 3: return *context.fVec3_Type; | 24 case 3: return *context.fVec3_Type; |
25 case 4: return *context.fVec4_Type; | 25 case 4: return *context.fVec4_Type; |
26 default: ASSERT(false); | 26 default: ASSERT(false); |
27 } | 27 } |
28 } else { | 28 } else { |
29 ASSERT(type.componentType() == *context.fDouble_Type); | 29 ASSERT(type.componentType() == *context.fDouble_Type); |
30 switch (type.columns()) { | 30 switch (type.rows()) { |
31 case 2: return *context.fDVec2_Type; | 31 case 2: return *context.fDVec2_Type; |
32 case 3: return *context.fDVec3_Type; | 32 case 3: return *context.fDVec3_Type; |
33 case 4: return *context.fDVec4_Type; | 33 case 4: return *context.fDVec4_Type; |
34 default: ASSERT(false); | 34 default: ASSERT(false); |
35 } | 35 } |
36 } | 36 } |
37 } | 37 } |
38 return type.componentType(); | 38 return type.componentType(); |
39 } | 39 } |
40 | 40 |
41 /** | 41 /** |
42 * An expression which extracts a value from an array or matrix, as in 'm[2]'. | 42 * An expression which extracts a value from an array or matrix, as in 'm[2]'. |
43 */ | 43 */ |
44 struct IndexExpression : public Expression { | 44 struct IndexExpression : public Expression { |
45 IndexExpression(const Context& context, std::unique_ptr<Expression> base, | 45 IndexExpression(const Context& context, std::unique_ptr<Expression> base, |
46 std::unique_ptr<Expression> index) | 46 std::unique_ptr<Expression> index) |
47 : INHERITED(base->fPosition, kIndex_Kind, index_type(context, base->fType)) | 47 : INHERITED(base->fPosition, kIndex_Kind, index_type(context, base->fType)) |
48 , fBase(std::move(base)) | 48 , fBase(std::move(base)) |
49 , fIndex(std::move(index)) { | 49 , fIndex(std::move(index)) { |
50 ASSERT(fIndex->fType == *context.fInt_Type); | 50 ASSERT(fIndex->fType == *context.fInt_Type || fIndex->fType == *context.
fUInt_Type); |
51 } | 51 } |
52 | 52 |
53 std::string description() const override { | 53 std::string description() const override { |
54 return fBase->description() + "[" + fIndex->description() + "]"; | 54 return fBase->description() + "[" + fIndex->description() + "]"; |
55 } | 55 } |
56 | 56 |
57 const std::unique_ptr<Expression> fBase; | 57 const std::unique_ptr<Expression> fBase; |
58 const std::unique_ptr<Expression> fIndex; | 58 const std::unique_ptr<Expression> fIndex; |
59 | 59 |
60 typedef Expression INHERITED; | 60 typedef Expression INHERITED; |
61 }; | 61 }; |
62 | 62 |
63 } // namespace | 63 } // namespace |
64 | 64 |
65 #endif | 65 #endif |
OLD | NEW |