• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 /*
2  * Copyright 2010-2012, The Android Open Source Project
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *     http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 
17 #include "slang_rs_export_type.h"
18 
19 #include <list>
20 #include <vector>
21 
22 #include "clang/AST/ASTContext.h"
23 #include "clang/AST/Attr.h"
24 #include "clang/AST/RecordLayout.h"
25 
26 #include "llvm/ADT/StringExtras.h"
27 #include "llvm/IR/DataLayout.h"
28 #include "llvm/IR/DerivedTypes.h"
29 #include "llvm/IR/Type.h"
30 
31 #include "slang_assert.h"
32 #include "slang_rs_context.h"
33 #include "slang_rs_export_element.h"
34 #include "slang_version.h"
35 
36 #define CHECK_PARENT_EQUALITY(ParentClass, E) \
37   if (!ParentClass::equals(E))                \
38     return false;
39 
40 namespace slang {
41 
42 namespace {
43 
44 /* For the data types we support, their category, names, and size (in bits).
45  *
46  * IMPORTANT: The data types in this table should be at the same index
47  * as specified by the corresponding DataType enum.
48  */
49 static RSReflectionType gReflectionTypes[] = {
50     {PrimitiveDataType, "FLOAT_16", "F16", 16, "half", "half", "Half", "Half", false},
51     {PrimitiveDataType, "FLOAT_32", "F32", 32, "float", "float", "Float", "Float", false},
52     {PrimitiveDataType, "FLOAT_64", "F64", 64, "double", "double", "Double", "Double",false},
53     {PrimitiveDataType, "SIGNED_8", "I8", 8, "int8_t", "byte", "Byte", "Byte", false},
54     {PrimitiveDataType, "SIGNED_16", "I16", 16, "int16_t", "short", "Short", "Short", false},
55     {PrimitiveDataType, "SIGNED_32", "I32", 32, "int32_t", "int", "Int", "Int", false},
56     {PrimitiveDataType, "SIGNED_64", "I64", 64, "int64_t", "long", "Long", "Long", false},
57     {PrimitiveDataType, "UNSIGNED_8", "U8", 8, "uint8_t", "short", "UByte", "Short", true},
58     {PrimitiveDataType, "UNSIGNED_16", "U16", 16, "uint16_t", "int", "UShort", "Int", true},
59     {PrimitiveDataType, "UNSIGNED_32", "U32", 32, "uint32_t", "long", "UInt", "Long", true},
60     {PrimitiveDataType, "UNSIGNED_64", "U64", 64, "uint64_t", "long", "ULong", "Long", false},
61 
62     {PrimitiveDataType, "BOOLEAN", "BOOLEAN", 8, "bool", "boolean", NULL, NULL, false},
63 
64     {PrimitiveDataType, "UNSIGNED_5_6_5", NULL, 16, NULL, NULL, NULL, NULL, false},
65     {PrimitiveDataType, "UNSIGNED_5_5_5_1", NULL, 16, NULL, NULL, NULL, NULL, false},
66     {PrimitiveDataType, "UNSIGNED_4_4_4_4", NULL, 16, NULL, NULL, NULL, NULL, false},
67 
68     {MatrixDataType, "MATRIX_2X2", NULL, 4*32, "rsMatrix_2x2", "Matrix2f", NULL, NULL, false},
69     {MatrixDataType, "MATRIX_3X3", NULL, 9*32, "rsMatrix_3x3", "Matrix3f", NULL, NULL, false},
70     {MatrixDataType, "MATRIX_4X4", NULL, 16*32, "rsMatrix_4x4", "Matrix4f", NULL, NULL, false},
71 
72     // RS object types are 32 bits in 32-bit RS, but 256 bits in 64-bit RS.
73     // This is handled specially by the GetSizeInBits() method.
74     {ObjectDataType, "RS_ELEMENT", "ELEMENT", 32, "Element", "Element", NULL, NULL, false},
75     {ObjectDataType, "RS_TYPE", "TYPE", 32, "Type", "Type", NULL, NULL, false},
76     {ObjectDataType, "RS_ALLOCATION", "ALLOCATION", 32, "Allocation", "Allocation", NULL, NULL, false},
77     {ObjectDataType, "RS_SAMPLER", "SAMPLER", 32, "Sampler", "Sampler", NULL, NULL, false},
78     {ObjectDataType, "RS_SCRIPT", "SCRIPT", 32, "Script", "Script", NULL, NULL, false},
79     {ObjectDataType, "RS_MESH", "MESH", 32, "Mesh", "Mesh", NULL, NULL, false},
80     {ObjectDataType, "RS_PATH", "PATH", 32, "Path", "Path", NULL, NULL, false},
81 
82     {ObjectDataType, "RS_PROGRAM_FRAGMENT", "PROGRAM_FRAGMENT", 32, "ProgramFragment", "ProgramFragment", NULL, NULL, false},
83     {ObjectDataType, "RS_PROGRAM_VERTEX", "PROGRAM_VERTEX", 32, "ProgramVertex", "ProgramVertex", NULL, NULL, false},
84     {ObjectDataType, "RS_PROGRAM_RASTER", "PROGRAM_RASTER", 32, "ProgramRaster", "ProgramRaster", NULL, NULL, false},
85     {ObjectDataType, "RS_PROGRAM_STORE", "PROGRAM_STORE", 32, "ProgramStore", "ProgramStore", NULL, NULL, false},
86     {ObjectDataType, "RS_FONT", "FONT", 32, "Font", "Font", NULL, NULL, false}
87 };
88 
89 const int kMaxVectorSize = 4;
90 
91 struct BuiltinInfo {
92   clang::BuiltinType::Kind builtinTypeKind;
93   DataType type;
94   /* TODO If we return std::string instead of llvm::StringRef, we could build
95    * the name instead of duplicating the entries.
96    */
97   const char *cname[kMaxVectorSize];
98 };
99 
100 
101 BuiltinInfo BuiltinInfoTable[] = {
102     {clang::BuiltinType::Bool, DataTypeBoolean,
103      {"bool", "bool2", "bool3", "bool4"}},
104     {clang::BuiltinType::Char_U, DataTypeUnsigned8,
105      {"uchar", "uchar2", "uchar3", "uchar4"}},
106     {clang::BuiltinType::UChar, DataTypeUnsigned8,
107      {"uchar", "uchar2", "uchar3", "uchar4"}},
108     {clang::BuiltinType::Char16, DataTypeSigned16,
109      {"short", "short2", "short3", "short4"}},
110     {clang::BuiltinType::Char32, DataTypeSigned32,
111      {"int", "int2", "int3", "int4"}},
112     {clang::BuiltinType::UShort, DataTypeUnsigned16,
113      {"ushort", "ushort2", "ushort3", "ushort4"}},
114     {clang::BuiltinType::UInt, DataTypeUnsigned32,
115      {"uint", "uint2", "uint3", "uint4"}},
116     {clang::BuiltinType::ULong, DataTypeUnsigned64,
117      {"ulong", "ulong2", "ulong3", "ulong4"}},
118     {clang::BuiltinType::ULongLong, DataTypeUnsigned64,
119      {"ulong", "ulong2", "ulong3", "ulong4"}},
120 
121     {clang::BuiltinType::Char_S, DataTypeSigned8,
122      {"char", "char2", "char3", "char4"}},
123     {clang::BuiltinType::SChar, DataTypeSigned8,
124      {"char", "char2", "char3", "char4"}},
125     {clang::BuiltinType::Short, DataTypeSigned16,
126      {"short", "short2", "short3", "short4"}},
127     {clang::BuiltinType::Int, DataTypeSigned32,
128      {"int", "int2", "int3", "int4"}},
129     {clang::BuiltinType::Long, DataTypeSigned64,
130      {"long", "long2", "long3", "long4"}},
131     {clang::BuiltinType::LongLong, DataTypeSigned64,
132      {"long", "long2", "long3", "long4"}},
133     {clang::BuiltinType::Float, DataTypeFloat32,
134      {"float", "float2", "float3", "float4"}},
135     {clang::BuiltinType::Double, DataTypeFloat64,
136      {"double", "double2", "double3", "double4"}},
137 };
138 const int BuiltinInfoTableCount = sizeof(BuiltinInfoTable) / sizeof(BuiltinInfoTable[0]);
139 
140 struct NameAndPrimitiveType {
141   const char *name;
142   DataType dataType;
143 };
144 
145 static NameAndPrimitiveType MatrixAndObjectDataTypes[] = {
146     {"rs_matrix2x2", DataTypeRSMatrix2x2},
147     {"rs_matrix3x3", DataTypeRSMatrix3x3},
148     {"rs_matrix4x4", DataTypeRSMatrix4x4},
149     {"rs_element", DataTypeRSElement},
150     {"rs_type", DataTypeRSType},
151     {"rs_allocation", DataTypeRSAllocation},
152     {"rs_sampler", DataTypeRSSampler},
153     {"rs_script", DataTypeRSScript},
154     {"rs_mesh", DataTypeRSMesh},
155     {"rs_path", DataTypeRSPath},
156     {"rs_program_fragment", DataTypeRSProgramFragment},
157     {"rs_program_vertex", DataTypeRSProgramVertex},
158     {"rs_program_raster", DataTypeRSProgramRaster},
159     {"rs_program_store", DataTypeRSProgramStore},
160     {"rs_font", DataTypeRSFont},
161 };
162 
163 const int MatrixAndObjectDataTypesCount =
164     sizeof(MatrixAndObjectDataTypes) / sizeof(MatrixAndObjectDataTypes[0]);
165 
166 static const clang::Type *TypeExportableHelper(
167     const clang::Type *T,
168     llvm::SmallPtrSet<const clang::Type*, 8>& SPS,
169     slang::RSContext *Context,
170     const clang::VarDecl *VD,
171     const clang::RecordDecl *TopLevelRecord);
172 
173 template <unsigned N>
ReportTypeError(slang::RSContext * Context,const clang::NamedDecl * ND,const clang::RecordDecl * TopLevelRecord,const char (& Message)[N],unsigned int TargetAPI=0)174 static void ReportTypeError(slang::RSContext *Context,
175                             const clang::NamedDecl *ND,
176                             const clang::RecordDecl *TopLevelRecord,
177                             const char (&Message)[N],
178                             unsigned int TargetAPI = 0) {
179   // Attempt to use the type declaration first (if we have one).
180   // Fall back to the variable definition, if we are looking at something
181   // like an array declaration that can't be exported.
182   if (TopLevelRecord) {
183     Context->ReportError(TopLevelRecord->getLocation(), Message)
184         << TopLevelRecord->getName() << TargetAPI;
185   } else if (ND) {
186     Context->ReportError(ND->getLocation(), Message) << ND->getName()
187                                                      << TargetAPI;
188   } else {
189     slangAssert(false && "Variables should be validated before exporting");
190   }
191 }
192 
ConstantArrayTypeExportableHelper(const clang::ConstantArrayType * CAT,llvm::SmallPtrSet<const clang::Type *,8> & SPS,slang::RSContext * Context,const clang::VarDecl * VD,const clang::RecordDecl * TopLevelRecord)193 static const clang::Type *ConstantArrayTypeExportableHelper(
194     const clang::ConstantArrayType *CAT,
195     llvm::SmallPtrSet<const clang::Type*, 8>& SPS,
196     slang::RSContext *Context,
197     const clang::VarDecl *VD,
198     const clang::RecordDecl *TopLevelRecord) {
199   // Check element type
200   const clang::Type *ElementType = GetConstantArrayElementType(CAT);
201   if (ElementType->isArrayType()) {
202     ReportTypeError(Context, VD, TopLevelRecord,
203                     "multidimensional arrays cannot be exported: '%0'");
204     return NULL;
205   } else if (ElementType->isExtVectorType()) {
206     const clang::ExtVectorType *EVT =
207         static_cast<const clang::ExtVectorType*>(ElementType);
208     unsigned numElements = EVT->getNumElements();
209 
210     const clang::Type *BaseElementType = GetExtVectorElementType(EVT);
211     if (!RSExportPrimitiveType::IsPrimitiveType(BaseElementType)) {
212       ReportTypeError(Context, VD, TopLevelRecord,
213         "vectors of non-primitive types cannot be exported: '%0'");
214       return NULL;
215     }
216 
217     if (numElements == 3 && CAT->getSize() != 1) {
218       ReportTypeError(Context, VD, TopLevelRecord,
219         "arrays of width 3 vector types cannot be exported: '%0'");
220       return NULL;
221     }
222   }
223 
224   if (TypeExportableHelper(ElementType, SPS, Context, VD,
225                            TopLevelRecord) == NULL) {
226     return NULL;
227   } else {
228     return CAT;
229   }
230 }
231 
FindBuiltinType(clang::BuiltinType::Kind builtinTypeKind)232 BuiltinInfo *FindBuiltinType(clang::BuiltinType::Kind builtinTypeKind) {
233   for (int i = 0; i < BuiltinInfoTableCount; i++) {
234     if (builtinTypeKind == BuiltinInfoTable[i].builtinTypeKind) {
235       return &BuiltinInfoTable[i];
236     }
237   }
238   return NULL;
239 }
240 
TypeExportableHelper(clang::Type const * T,llvm::SmallPtrSet<clang::Type const *,8> & SPS,slang::RSContext * Context,clang::VarDecl const * VD,clang::RecordDecl const * TopLevelRecord)241 static const clang::Type *TypeExportableHelper(
242     clang::Type const *T,
243     llvm::SmallPtrSet<clang::Type const *, 8> &SPS,
244     slang::RSContext *Context,
245     clang::VarDecl const *VD,
246     clang::RecordDecl const *TopLevelRecord) {
247   // Normalize first
248   if ((T = GetCanonicalType(T)) == NULL)
249     return NULL;
250 
251   if (SPS.count(T))
252     return T;
253 
254   const clang::Type *CTI = T->getCanonicalTypeInternal().getTypePtr();
255 
256   switch (T->getTypeClass()) {
257     case clang::Type::Builtin: {
258       const clang::BuiltinType *BT = static_cast<const clang::BuiltinType*>(CTI);
259       return FindBuiltinType(BT->getKind()) == NULL ? NULL : T;
260     }
261     case clang::Type::Record: {
262       if (RSExportPrimitiveType::GetRSSpecificType(T) != DataTypeUnknown) {
263         return T;  // RS object type, no further checks are needed
264       }
265 
266       // Check internal struct
267       if (T->isUnionType()) {
268         ReportTypeError(Context, VD, T->getAsUnionType()->getDecl(),
269                         "unions cannot be exported: '%0'");
270         return NULL;
271       } else if (!T->isStructureType()) {
272         slangAssert(false && "Unknown type cannot be exported");
273         return NULL;
274       }
275 
276       clang::RecordDecl *RD = T->getAsStructureType()->getDecl();
277       if (RD != NULL) {
278         RD = RD->getDefinition();
279         if (RD == NULL) {
280           ReportTypeError(Context, NULL, T->getAsStructureType()->getDecl(),
281                           "struct is not defined in this module");
282           return NULL;
283         }
284       }
285 
286       if (!TopLevelRecord) {
287         TopLevelRecord = RD;
288       }
289       if (RD->getName().empty()) {
290         ReportTypeError(Context, NULL, RD,
291                         "anonymous structures cannot be exported");
292         return NULL;
293       }
294 
295       // Fast check
296       if (RD->hasFlexibleArrayMember() || RD->hasObjectMember())
297         return NULL;
298 
299       // Insert myself into checking set
300       SPS.insert(T);
301 
302       // Check all element
303       for (clang::RecordDecl::field_iterator FI = RD->field_begin(),
304                FE = RD->field_end();
305            FI != FE;
306            FI++) {
307         const clang::FieldDecl *FD = *FI;
308         const clang::Type *FT = RSExportType::GetTypeOfDecl(FD);
309         FT = GetCanonicalType(FT);
310 
311         if (!TypeExportableHelper(FT, SPS, Context, VD, TopLevelRecord)) {
312           return NULL;
313         }
314 
315         // We don't support bit fields yet
316         //
317         // TODO(zonr/srhines): allow bit fields of size 8, 16, 32
318         if (FD->isBitField()) {
319           Context->ReportError(
320               FD->getLocation(),
321               "bit fields are not able to be exported: '%0.%1'")
322               << RD->getName() << FD->getName();
323           return NULL;
324         }
325       }
326 
327       return T;
328     }
329     case clang::Type::Pointer: {
330       if (TopLevelRecord) {
331         ReportTypeError(Context, VD, TopLevelRecord,
332             "structures containing pointers cannot be exported: '%0'");
333         return NULL;
334       }
335 
336       const clang::PointerType *PT = static_cast<const clang::PointerType*>(CTI);
337       const clang::Type *PointeeType = GetPointeeType(PT);
338 
339       if (PointeeType->getTypeClass() == clang::Type::Pointer) {
340         ReportTypeError(Context, VD, TopLevelRecord,
341             "multiple levels of pointers cannot be exported: '%0'");
342         return NULL;
343       }
344       // We don't support pointer with array-type pointee or unsupported pointee
345       // type
346       if (PointeeType->isArrayType() ||
347           (TypeExportableHelper(PointeeType, SPS, Context, VD,
348                                 TopLevelRecord) == NULL))
349         return NULL;
350       else
351         return T;
352     }
353     case clang::Type::ExtVector: {
354       const clang::ExtVectorType *EVT =
355               static_cast<const clang::ExtVectorType*>(CTI);
356       // Only vector with size 2, 3 and 4 are supported.
357       if (EVT->getNumElements() < 2 || EVT->getNumElements() > 4)
358         return NULL;
359 
360       // Check base element type
361       const clang::Type *ElementType = GetExtVectorElementType(EVT);
362 
363       if ((ElementType->getTypeClass() != clang::Type::Builtin) ||
364           (TypeExportableHelper(ElementType, SPS, Context, VD,
365                                 TopLevelRecord) == NULL))
366         return NULL;
367       else
368         return T;
369     }
370     case clang::Type::ConstantArray: {
371       const clang::ConstantArrayType *CAT =
372               static_cast<const clang::ConstantArrayType*>(CTI);
373 
374       return ConstantArrayTypeExportableHelper(CAT, SPS, Context, VD,
375                                                TopLevelRecord);
376     }
377     case clang::Type::Enum: {
378       // FIXME: We currently convert enums to integers, rather than reflecting
379       // a more complete (and nicer type-safe Java version).
380       return Context->getASTContext().IntTy.getTypePtr();
381     }
382     default: {
383       slangAssert(false && "Unknown type cannot be validated");
384       return NULL;
385     }
386   }
387 }
388 
389 // Return the type that can be used to create RSExportType, will always return
390 // the canonical type
391 // If the Type T is not exportable, this function returns NULL. DiagEngine is
392 // used to generate proper Clang diagnostic messages when a
393 // non-exportable type is detected. TopLevelRecord is used to capture the
394 // highest struct (in the case of a nested hierarchy) for detecting other
395 // types that cannot be exported (mostly pointers within a struct).
TypeExportable(const clang::Type * T,slang::RSContext * Context,const clang::VarDecl * VD)396 static const clang::Type *TypeExportable(const clang::Type *T,
397                                          slang::RSContext *Context,
398                                          const clang::VarDecl *VD) {
399   llvm::SmallPtrSet<const clang::Type*, 8> SPS =
400       llvm::SmallPtrSet<const clang::Type*, 8>();
401 
402   return TypeExportableHelper(T, SPS, Context, VD, NULL);
403 }
404 
ValidateRSObjectInVarDecl(slang::RSContext * Context,clang::VarDecl * VD,bool InCompositeType,unsigned int TargetAPI)405 static bool ValidateRSObjectInVarDecl(slang::RSContext *Context,
406                                       clang::VarDecl *VD, bool InCompositeType,
407                                       unsigned int TargetAPI) {
408   if (TargetAPI < SLANG_JB_TARGET_API) {
409     // Only if we are already in a composite type (like an array or structure).
410     if (InCompositeType) {
411       // Only if we are actually exported (i.e. non-static).
412       if (VD->hasLinkage() &&
413           (VD->getFormalLinkage() == clang::ExternalLinkage)) {
414         // Only if we are not a pointer to an object.
415         const clang::Type *T = GetCanonicalType(VD->getType().getTypePtr());
416         if (T->getTypeClass() != clang::Type::Pointer) {
417           ReportTypeError(Context, VD, NULL,
418                           "arrays/structures containing RS object types "
419                           "cannot be exported in target API < %1: '%0'",
420                           SLANG_JB_TARGET_API);
421           return false;
422         }
423       }
424     }
425   }
426 
427   return true;
428 }
429 
430 // Helper function for ValidateType(). We do a recursive descent on the
431 // type hierarchy to ensure that we can properly export/handle the
432 // declaration.
433 // \return true if the variable declaration is valid,
434 //         false if it is invalid (along with proper diagnostics).
435 //
436 // C - ASTContext (for diagnostics + builtin types).
437 // T - sub-type that we are validating.
438 // ND - (optional) top-level named declaration that we are validating.
439 // SPS - set of types we have already seen/validated.
440 // InCompositeType - true if we are within an outer composite type.
441 // UnionDecl - set if we are in a sub-type of a union.
442 // TargetAPI - target SDK API level.
443 // IsFilterscript - whether or not we are compiling for Filterscript
ValidateTypeHelper(slang::RSContext * Context,clang::ASTContext & C,const clang::Type * & T,clang::NamedDecl * ND,clang::SourceLocation Loc,llvm::SmallPtrSet<const clang::Type *,8> & SPS,bool InCompositeType,clang::RecordDecl * UnionDecl,unsigned int TargetAPI,bool IsFilterscript)444 static bool ValidateTypeHelper(
445     slang::RSContext *Context,
446     clang::ASTContext &C,
447     const clang::Type *&T,
448     clang::NamedDecl *ND,
449     clang::SourceLocation Loc,
450     llvm::SmallPtrSet<const clang::Type*, 8>& SPS,
451     bool InCompositeType,
452     clang::RecordDecl *UnionDecl,
453     unsigned int TargetAPI,
454     bool IsFilterscript) {
455   if ((T = GetCanonicalType(T)) == NULL)
456     return true;
457 
458   if (SPS.count(T))
459     return true;
460 
461   const clang::Type *CTI = T->getCanonicalTypeInternal().getTypePtr();
462 
463   switch (T->getTypeClass()) {
464     case clang::Type::Record: {
465       if (RSExportPrimitiveType::IsRSObjectType(T)) {
466         clang::VarDecl *VD = (ND ? llvm::dyn_cast<clang::VarDecl>(ND) : NULL);
467         if (VD && !ValidateRSObjectInVarDecl(Context, VD, InCompositeType,
468                                              TargetAPI)) {
469           return false;
470         }
471       }
472 
473       if (RSExportPrimitiveType::GetRSSpecificType(T) != DataTypeUnknown) {
474         if (!UnionDecl) {
475           return true;
476         } else if (RSExportPrimitiveType::IsRSObjectType(T)) {
477           ReportTypeError(Context, NULL, UnionDecl,
478               "unions containing RS object types are not allowed");
479           return false;
480         }
481       }
482 
483       clang::RecordDecl *RD = NULL;
484 
485       // Check internal struct
486       if (T->isUnionType()) {
487         RD = T->getAsUnionType()->getDecl();
488         UnionDecl = RD;
489       } else if (T->isStructureType()) {
490         RD = T->getAsStructureType()->getDecl();
491       } else {
492         slangAssert(false && "Unknown type cannot be exported");
493         return false;
494       }
495 
496       if (RD != NULL) {
497         RD = RD->getDefinition();
498         if (RD == NULL) {
499           // FIXME
500           return true;
501         }
502       }
503 
504       // Fast check
505       if (RD->hasFlexibleArrayMember() || RD->hasObjectMember())
506         return false;
507 
508       // Insert myself into checking set
509       SPS.insert(T);
510 
511       // Check all elements
512       for (clang::RecordDecl::field_iterator FI = RD->field_begin(),
513                FE = RD->field_end();
514            FI != FE;
515            FI++) {
516         const clang::FieldDecl *FD = *FI;
517         const clang::Type *FT = RSExportType::GetTypeOfDecl(FD);
518         FT = GetCanonicalType(FT);
519 
520         if (!ValidateTypeHelper(Context, C, FT, ND, Loc, SPS, true, UnionDecl,
521                                 TargetAPI, IsFilterscript)) {
522           return false;
523         }
524       }
525 
526       return true;
527     }
528 
529     case clang::Type::Builtin: {
530       if (IsFilterscript) {
531         clang::QualType QT = T->getCanonicalTypeInternal();
532         if (QT == C.DoubleTy ||
533             QT == C.LongDoubleTy ||
534             QT == C.LongTy ||
535             QT == C.LongLongTy) {
536           if (ND) {
537             Context->ReportError(
538                 Loc,
539                 "Builtin types > 32 bits in size are forbidden in "
540                 "Filterscript: '%0'")
541                 << ND->getName();
542           } else {
543             Context->ReportError(
544                 Loc,
545                 "Builtin types > 32 bits in size are forbidden in "
546                 "Filterscript");
547           }
548           return false;
549         }
550       }
551       break;
552     }
553 
554     case clang::Type::Pointer: {
555       if (IsFilterscript) {
556         if (ND) {
557           Context->ReportError(Loc,
558                                "Pointers are forbidden in Filterscript: '%0'")
559               << ND->getName();
560           return false;
561         } else {
562           // TODO(srhines): Find a better way to handle expressions (i.e. no
563           // NamedDecl) involving pointers in FS that should be allowed.
564           // An example would be calls to library functions like
565           // rsMatrixMultiply() that take rs_matrixNxN * types.
566         }
567       }
568 
569       const clang::PointerType *PT = static_cast<const clang::PointerType*>(CTI);
570       const clang::Type *PointeeType = GetPointeeType(PT);
571 
572       return ValidateTypeHelper(Context, C, PointeeType, ND, Loc, SPS,
573                                 InCompositeType, UnionDecl, TargetAPI,
574                                 IsFilterscript);
575     }
576 
577     case clang::Type::ExtVector: {
578       const clang::ExtVectorType *EVT =
579               static_cast<const clang::ExtVectorType*>(CTI);
580       const clang::Type *ElementType = GetExtVectorElementType(EVT);
581       if (TargetAPI < SLANG_ICS_TARGET_API &&
582           InCompositeType &&
583           EVT->getNumElements() == 3 &&
584           ND &&
585           ND->getFormalLinkage() == clang::ExternalLinkage) {
586         ReportTypeError(Context, ND, NULL,
587                         "structs containing vectors of dimension 3 cannot "
588                         "be exported at this API level: '%0'");
589         return false;
590       }
591       return ValidateTypeHelper(Context, C, ElementType, ND, Loc, SPS, true,
592                                 UnionDecl, TargetAPI, IsFilterscript);
593     }
594 
595     case clang::Type::ConstantArray: {
596       const clang::ConstantArrayType *CAT = static_cast<const clang::ConstantArrayType*>(CTI);
597       const clang::Type *ElementType = GetConstantArrayElementType(CAT);
598       return ValidateTypeHelper(Context, C, ElementType, ND, Loc, SPS, true,
599                                 UnionDecl, TargetAPI, IsFilterscript);
600     }
601 
602     default: {
603       break;
604     }
605   }
606 
607   return true;
608 }
609 
610 }  // namespace
611 
CreateDummyName(const char * type,const std::string & name)612 std::string CreateDummyName(const char *type, const std::string &name) {
613   std::stringstream S;
614   S << "<" << type;
615   if (!name.empty()) {
616     S << ":" << name;
617   }
618   S << ">";
619   return S.str();
620 }
621 
622 /****************************** RSExportType ******************************/
NormalizeType(const clang::Type * & T,llvm::StringRef & TypeName,RSContext * Context,const clang::VarDecl * VD)623 bool RSExportType::NormalizeType(const clang::Type *&T,
624                                  llvm::StringRef &TypeName,
625                                  RSContext *Context,
626                                  const clang::VarDecl *VD) {
627   if ((T = TypeExportable(T, Context, VD)) == NULL) {
628     return false;
629   }
630   // Get type name
631   TypeName = RSExportType::GetTypeName(T);
632   if (Context && TypeName.empty()) {
633     if (VD) {
634       Context->ReportError(VD->getLocation(),
635                            "anonymous types cannot be exported");
636     } else {
637       Context->ReportError("anonymous types cannot be exported");
638     }
639     return false;
640   }
641 
642   return true;
643 }
644 
ValidateType(slang::RSContext * Context,clang::ASTContext & C,clang::QualType QT,clang::NamedDecl * ND,clang::SourceLocation Loc,unsigned int TargetAPI,bool IsFilterscript)645 bool RSExportType::ValidateType(slang::RSContext *Context, clang::ASTContext &C,
646                                 clang::QualType QT, clang::NamedDecl *ND,
647                                 clang::SourceLocation Loc,
648                                 unsigned int TargetAPI, bool IsFilterscript) {
649   const clang::Type *T = QT.getTypePtr();
650   llvm::SmallPtrSet<const clang::Type*, 8> SPS =
651       llvm::SmallPtrSet<const clang::Type*, 8>();
652 
653   return ValidateTypeHelper(Context, C, T, ND, Loc, SPS, false, NULL, TargetAPI,
654                             IsFilterscript);
655   return true;
656 }
657 
ValidateVarDecl(slang::RSContext * Context,clang::VarDecl * VD,unsigned int TargetAPI,bool IsFilterscript)658 bool RSExportType::ValidateVarDecl(slang::RSContext *Context,
659                                    clang::VarDecl *VD, unsigned int TargetAPI,
660                                    bool IsFilterscript) {
661   return ValidateType(Context, VD->getASTContext(), VD->getType(), VD,
662                       VD->getLocation(), TargetAPI, IsFilterscript);
663 }
664 
665 const clang::Type
GetTypeOfDecl(const clang::DeclaratorDecl * DD)666 *RSExportType::GetTypeOfDecl(const clang::DeclaratorDecl *DD) {
667   if (DD) {
668     clang::QualType T = DD->getType();
669 
670     if (T.isNull())
671       return NULL;
672     else
673       return T.getTypePtr();
674   }
675   return NULL;
676 }
677 
GetTypeName(const clang::Type * T)678 llvm::StringRef RSExportType::GetTypeName(const clang::Type* T) {
679   T = GetCanonicalType(T);
680   if (T == NULL)
681     return llvm::StringRef();
682 
683   const clang::Type *CTI = T->getCanonicalTypeInternal().getTypePtr();
684 
685   switch (T->getTypeClass()) {
686     case clang::Type::Builtin: {
687       const clang::BuiltinType *BT = static_cast<const clang::BuiltinType*>(CTI);
688       BuiltinInfo *info = FindBuiltinType(BT->getKind());
689       if (info != NULL) {
690         return info->cname[0];
691       }
692       slangAssert(false && "Unknown data type of the builtin");
693       break;
694     }
695     case clang::Type::Record: {
696       clang::RecordDecl *RD;
697       if (T->isStructureType()) {
698         RD = T->getAsStructureType()->getDecl();
699       } else {
700         break;
701       }
702 
703       llvm::StringRef Name = RD->getName();
704       if (Name.empty()) {
705         if (RD->getTypedefNameForAnonDecl() != NULL) {
706           Name = RD->getTypedefNameForAnonDecl()->getName();
707         }
708 
709         if (Name.empty()) {
710           // Try to find a name from redeclaration (i.e. typedef)
711           for (clang::TagDecl::redecl_iterator RI = RD->redecls_begin(),
712                    RE = RD->redecls_end();
713                RI != RE;
714                RI++) {
715             slangAssert(*RI != NULL && "cannot be NULL object");
716 
717             Name = (*RI)->getName();
718             if (!Name.empty())
719               break;
720           }
721         }
722       }
723       return Name;
724     }
725     case clang::Type::Pointer: {
726       // "*" plus pointee name
727       const clang::PointerType *P = static_cast<const clang::PointerType*>(CTI);
728       const clang::Type *PT = GetPointeeType(P);
729       llvm::StringRef PointeeName;
730       if (NormalizeType(PT, PointeeName, NULL, NULL)) {
731         char *Name = new char[ 1 /* * */ + PointeeName.size() + 1 ];
732         Name[0] = '*';
733         memcpy(Name + 1, PointeeName.data(), PointeeName.size());
734         Name[PointeeName.size() + 1] = '\0';
735         return Name;
736       }
737       break;
738     }
739     case clang::Type::ExtVector: {
740       const clang::ExtVectorType *EVT =
741               static_cast<const clang::ExtVectorType*>(CTI);
742       return RSExportVectorType::GetTypeName(EVT);
743       break;
744     }
745     case clang::Type::ConstantArray : {
746       // Construct name for a constant array is too complicated.
747       return "<ConstantArray>";
748     }
749     default: {
750       break;
751     }
752   }
753 
754   return llvm::StringRef();
755 }
756 
757 
Create(RSContext * Context,const clang::Type * T,const llvm::StringRef & TypeName)758 RSExportType *RSExportType::Create(RSContext *Context,
759                                    const clang::Type *T,
760                                    const llvm::StringRef &TypeName) {
761   // Lookup the context to see whether the type was processed before.
762   // Newly created RSExportType will insert into context
763   // in RSExportType::RSExportType()
764   RSContext::export_type_iterator ETI = Context->findExportType(TypeName);
765 
766   if (ETI != Context->export_types_end())
767     return ETI->second;
768 
769   const clang::Type *CTI = T->getCanonicalTypeInternal().getTypePtr();
770 
771   RSExportType *ET = NULL;
772   switch (T->getTypeClass()) {
773     case clang::Type::Record: {
774       DataType dt = RSExportPrimitiveType::GetRSSpecificType(TypeName);
775       switch (dt) {
776         case DataTypeUnknown: {
777           // User-defined types
778           ET = RSExportRecordType::Create(Context,
779                                           T->getAsStructureType(),
780                                           TypeName);
781           break;
782         }
783         case DataTypeRSMatrix2x2: {
784           // 2 x 2 Matrix type
785           ET = RSExportMatrixType::Create(Context,
786                                           T->getAsStructureType(),
787                                           TypeName,
788                                           2);
789           break;
790         }
791         case DataTypeRSMatrix3x3: {
792           // 3 x 3 Matrix type
793           ET = RSExportMatrixType::Create(Context,
794                                           T->getAsStructureType(),
795                                           TypeName,
796                                           3);
797           break;
798         }
799         case DataTypeRSMatrix4x4: {
800           // 4 x 4 Matrix type
801           ET = RSExportMatrixType::Create(Context,
802                                           T->getAsStructureType(),
803                                           TypeName,
804                                           4);
805           break;
806         }
807         default: {
808           // Others are primitive types
809           ET = RSExportPrimitiveType::Create(Context, T, TypeName);
810           break;
811         }
812       }
813       break;
814     }
815     case clang::Type::Builtin: {
816       ET = RSExportPrimitiveType::Create(Context, T, TypeName);
817       break;
818     }
819     case clang::Type::Pointer: {
820       ET = RSExportPointerType::Create(Context,
821                                        static_cast<const clang::PointerType*>(CTI),
822                                        TypeName);
823       // FIXME: free the name (allocated in RSExportType::GetTypeName)
824       delete [] TypeName.data();
825       break;
826     }
827     case clang::Type::ExtVector: {
828       ET = RSExportVectorType::Create(Context,
829                                       static_cast<const clang::ExtVectorType*>(CTI),
830                                       TypeName);
831       break;
832     }
833     case clang::Type::ConstantArray: {
834       ET = RSExportConstantArrayType::Create(
835               Context,
836               static_cast<const clang::ConstantArrayType*>(CTI));
837       break;
838     }
839     default: {
840       Context->ReportError("unknown type cannot be exported: '%0'")
841           << T->getTypeClassName();
842       break;
843     }
844   }
845 
846   return ET;
847 }
848 
Create(RSContext * Context,const clang::Type * T)849 RSExportType *RSExportType::Create(RSContext *Context, const clang::Type *T) {
850   llvm::StringRef TypeName;
851   if (NormalizeType(T, TypeName, Context, NULL)) {
852     return Create(Context, T, TypeName);
853   } else {
854     return NULL;
855   }
856 }
857 
CreateFromDecl(RSContext * Context,const clang::VarDecl * VD)858 RSExportType *RSExportType::CreateFromDecl(RSContext *Context,
859                                            const clang::VarDecl *VD) {
860   return RSExportType::Create(Context, GetTypeOfDecl(VD));
861 }
862 
getStoreSize() const863 size_t RSExportType::getStoreSize() const {
864   return getRSContext()->getDataLayout()->getTypeStoreSize(getLLVMType());
865 }
866 
getAllocSize() const867 size_t RSExportType::getAllocSize() const {
868     return getRSContext()->getDataLayout()->getTypeAllocSize(getLLVMType());
869 }
870 
RSExportType(RSContext * Context,ExportClass Class,const llvm::StringRef & Name)871 RSExportType::RSExportType(RSContext *Context,
872                            ExportClass Class,
873                            const llvm::StringRef &Name)
874     : RSExportable(Context, RSExportable::EX_TYPE),
875       mClass(Class),
876       // Make a copy on Name since memory stored @Name is either allocated in
877       // ASTContext or allocated in GetTypeName which will be destroyed later.
878       mName(Name.data(), Name.size()),
879       mLLVMType(NULL) {
880   // Don't cache the type whose name start with '<'. Those type failed to
881   // get their name since constructing their name in GetTypeName() requiring
882   // complicated work.
883   if (!IsDummyName(Name)) {
884     // TODO(zonr): Need to check whether the insertion is successful or not.
885     Context->insertExportType(llvm::StringRef(Name), this);
886   }
887 
888 }
889 
keep()890 bool RSExportType::keep() {
891   if (!RSExportable::keep())
892     return false;
893   // Invalidate converted LLVM type.
894   mLLVMType = NULL;
895   return true;
896 }
897 
equals(const RSExportable * E) const898 bool RSExportType::equals(const RSExportable *E) const {
899   CHECK_PARENT_EQUALITY(RSExportable, E);
900   return (static_cast<const RSExportType*>(E)->getClass() == getClass());
901 }
902 
~RSExportType()903 RSExportType::~RSExportType() {
904 }
905 
906 /************************** RSExportPrimitiveType **************************/
907 llvm::ManagedStatic<RSExportPrimitiveType::RSSpecificTypeMapTy>
908 RSExportPrimitiveType::RSSpecificTypeMap;
909 
IsPrimitiveType(const clang::Type * T)910 bool RSExportPrimitiveType::IsPrimitiveType(const clang::Type *T) {
911   if ((T != NULL) && (T->getTypeClass() == clang::Type::Builtin))
912     return true;
913   else
914     return false;
915 }
916 
917 DataType
GetRSSpecificType(const llvm::StringRef & TypeName)918 RSExportPrimitiveType::GetRSSpecificType(const llvm::StringRef &TypeName) {
919   if (TypeName.empty())
920     return DataTypeUnknown;
921 
922   if (RSSpecificTypeMap->empty()) {
923     for (int i = 0; i < MatrixAndObjectDataTypesCount; i++) {
924       RSSpecificTypeMap->GetOrCreateValue(MatrixAndObjectDataTypes[i].name,
925                                           MatrixAndObjectDataTypes[i].dataType);
926     }
927   }
928 
929   RSSpecificTypeMapTy::const_iterator I = RSSpecificTypeMap->find(TypeName);
930   if (I == RSSpecificTypeMap->end())
931     return DataTypeUnknown;
932   else
933     return I->getValue();
934 }
935 
GetRSSpecificType(const clang::Type * T)936 DataType RSExportPrimitiveType::GetRSSpecificType(const clang::Type *T) {
937   T = GetCanonicalType(T);
938   if ((T == NULL) || (T->getTypeClass() != clang::Type::Record))
939     return DataTypeUnknown;
940 
941   return GetRSSpecificType( RSExportType::GetTypeName(T) );
942 }
943 
IsRSMatrixType(DataType DT)944 bool RSExportPrimitiveType::IsRSMatrixType(DataType DT) {
945     if (DT < 0 || DT >= DataTypeMax) {
946         return false;
947     }
948     return gReflectionTypes[DT].category == MatrixDataType;
949 }
950 
IsRSObjectType(DataType DT)951 bool RSExportPrimitiveType::IsRSObjectType(DataType DT) {
952     if (DT < 0 || DT >= DataTypeMax) {
953         return false;
954     }
955     return gReflectionTypes[DT].category == ObjectDataType;
956 }
957 
IsStructureTypeWithRSObject(const clang::Type * T)958 bool RSExportPrimitiveType::IsStructureTypeWithRSObject(const clang::Type *T) {
959   bool RSObjectTypeSeen = false;
960   while (T && T->isArrayType()) {
961     T = T->getArrayElementTypeNoTypeQual();
962   }
963 
964   const clang::RecordType *RT = T->getAsStructureType();
965   if (!RT) {
966     return false;
967   }
968 
969   const clang::RecordDecl *RD = RT->getDecl();
970   if (RD) {
971     RD = RD->getDefinition();
972   }
973   if (!RD) {
974     return false;
975   }
976 
977   for (clang::RecordDecl::field_iterator FI = RD->field_begin(),
978          FE = RD->field_end();
979        FI != FE;
980        FI++) {
981     // We just look through all field declarations to see if we find a
982     // declaration for an RS object type (or an array of one).
983     const clang::FieldDecl *FD = *FI;
984     const clang::Type *FT = RSExportType::GetTypeOfDecl(FD);
985     while (FT && FT->isArrayType()) {
986       FT = FT->getArrayElementTypeNoTypeQual();
987     }
988 
989     DataType DT = GetRSSpecificType(FT);
990     if (IsRSObjectType(DT)) {
991       // RS object types definitely need to be zero-initialized
992       RSObjectTypeSeen = true;
993     } else {
994       switch (DT) {
995         case DataTypeRSMatrix2x2:
996         case DataTypeRSMatrix3x3:
997         case DataTypeRSMatrix4x4:
998           // Matrix types should get zero-initialized as well
999           RSObjectTypeSeen = true;
1000           break;
1001         default:
1002           // Ignore all other primitive types
1003           break;
1004       }
1005       while (FT && FT->isArrayType()) {
1006         FT = FT->getArrayElementTypeNoTypeQual();
1007       }
1008       if (FT->isStructureType()) {
1009         // Recursively handle structs of structs (even though these can't
1010         // be exported, it is possible for a user to have them internally).
1011         RSObjectTypeSeen |= IsStructureTypeWithRSObject(FT);
1012       }
1013     }
1014   }
1015 
1016   return RSObjectTypeSeen;
1017 }
1018 
GetSizeInBits(const RSExportPrimitiveType * EPT)1019 size_t RSExportPrimitiveType::GetSizeInBits(const RSExportPrimitiveType *EPT) {
1020   int type = EPT->getType();
1021   slangAssert((type > DataTypeUnknown && type < DataTypeMax) &&
1022               "RSExportPrimitiveType::GetSizeInBits : unknown data type");
1023   // All RS object types are 256 bits in 64-bit RS.
1024   if (EPT->isRSObjectType() && EPT->getRSContext()->is64Bit()) {
1025     return 256;
1026   }
1027   return gReflectionTypes[type].size_in_bits;
1028 }
1029 
1030 DataType
GetDataType(RSContext * Context,const clang::Type * T)1031 RSExportPrimitiveType::GetDataType(RSContext *Context, const clang::Type *T) {
1032   if (T == NULL)
1033     return DataTypeUnknown;
1034 
1035   switch (T->getTypeClass()) {
1036     case clang::Type::Builtin: {
1037       const clang::BuiltinType *BT =
1038               static_cast<const clang::BuiltinType*>(T->getCanonicalTypeInternal().getTypePtr());
1039       BuiltinInfo *info = FindBuiltinType(BT->getKind());
1040       if (info != NULL) {
1041         return info->type;
1042       }
1043       // The size of type WChar depend on platform so we abandon the support
1044       // to them.
1045       Context->ReportError("built-in type cannot be exported: '%0'")
1046           << T->getTypeClassName();
1047       break;
1048     }
1049     case clang::Type::Record: {
1050       // must be RS object type
1051       return RSExportPrimitiveType::GetRSSpecificType(T);
1052     }
1053     default: {
1054       Context->ReportError("primitive type cannot be exported: '%0'")
1055           << T->getTypeClassName();
1056       break;
1057     }
1058   }
1059 
1060   return DataTypeUnknown;
1061 }
1062 
1063 RSExportPrimitiveType
Create(RSContext * Context,const clang::Type * T,const llvm::StringRef & TypeName,bool Normalized)1064 *RSExportPrimitiveType::Create(RSContext *Context,
1065                                const clang::Type *T,
1066                                const llvm::StringRef &TypeName,
1067                                bool Normalized) {
1068   DataType DT = GetDataType(Context, T);
1069 
1070   if ((DT == DataTypeUnknown) || TypeName.empty())
1071     return NULL;
1072   else
1073     return new RSExportPrimitiveType(Context, ExportClassPrimitive, TypeName,
1074                                      DT, Normalized);
1075 }
1076 
Create(RSContext * Context,const clang::Type * T)1077 RSExportPrimitiveType *RSExportPrimitiveType::Create(RSContext *Context,
1078                                                      const clang::Type *T) {
1079   llvm::StringRef TypeName;
1080   if (RSExportType::NormalizeType(T, TypeName, Context, NULL)
1081       && IsPrimitiveType(T)) {
1082     return Create(Context, T, TypeName);
1083   } else {
1084     return NULL;
1085   }
1086 }
1087 
convertToLLVMType() const1088 llvm::Type *RSExportPrimitiveType::convertToLLVMType() const {
1089   llvm::LLVMContext &C = getRSContext()->getLLVMContext();
1090 
1091   if (isRSObjectType()) {
1092     // struct {
1093     //   int *p;
1094     // } __attribute__((packed, aligned(pointer_size)))
1095     //
1096     // which is
1097     //
1098     // <{ [1 x i32] }> in LLVM
1099     //
1100     std::vector<llvm::Type *> Elements;
1101     if (getRSContext()->is64Bit()) {
1102       // 64-bit path
1103       Elements.push_back(llvm::ArrayType::get(llvm::Type::getInt64Ty(C), 4));
1104       return llvm::StructType::get(C, Elements, true);
1105     } else {
1106       // 32-bit legacy path
1107       Elements.push_back(llvm::ArrayType::get(llvm::Type::getInt32Ty(C), 1));
1108       return llvm::StructType::get(C, Elements, true);
1109     }
1110   }
1111 
1112   switch (mType) {
1113     case DataTypeFloat32: {
1114       return llvm::Type::getFloatTy(C);
1115       break;
1116     }
1117     case DataTypeFloat64: {
1118       return llvm::Type::getDoubleTy(C);
1119       break;
1120     }
1121     case DataTypeBoolean: {
1122       return llvm::Type::getInt1Ty(C);
1123       break;
1124     }
1125     case DataTypeSigned8:
1126     case DataTypeUnsigned8: {
1127       return llvm::Type::getInt8Ty(C);
1128       break;
1129     }
1130     case DataTypeSigned16:
1131     case DataTypeUnsigned16:
1132     case DataTypeUnsigned565:
1133     case DataTypeUnsigned5551:
1134     case DataTypeUnsigned4444: {
1135       return llvm::Type::getInt16Ty(C);
1136       break;
1137     }
1138     case DataTypeSigned32:
1139     case DataTypeUnsigned32: {
1140       return llvm::Type::getInt32Ty(C);
1141       break;
1142     }
1143     case DataTypeSigned64:
1144     case DataTypeUnsigned64: {
1145       return llvm::Type::getInt64Ty(C);
1146       break;
1147     }
1148     default: {
1149       slangAssert(false && "Unknown data type");
1150     }
1151   }
1152 
1153   return NULL;
1154 }
1155 
equals(const RSExportable * E) const1156 bool RSExportPrimitiveType::equals(const RSExportable *E) const {
1157   CHECK_PARENT_EQUALITY(RSExportType, E);
1158   return (static_cast<const RSExportPrimitiveType*>(E)->getType() == getType());
1159 }
1160 
getRSReflectionType(DataType DT)1161 RSReflectionType *RSExportPrimitiveType::getRSReflectionType(DataType DT) {
1162   if (DT > DataTypeUnknown && DT < DataTypeMax) {
1163     return &gReflectionTypes[DT];
1164   } else {
1165     return NULL;
1166   }
1167 }
1168 
1169 /**************************** RSExportPointerType ****************************/
1170 
1171 RSExportPointerType
Create(RSContext * Context,const clang::PointerType * PT,const llvm::StringRef & TypeName)1172 *RSExportPointerType::Create(RSContext *Context,
1173                              const clang::PointerType *PT,
1174                              const llvm::StringRef &TypeName) {
1175   const clang::Type *PointeeType = GetPointeeType(PT);
1176   const RSExportType *PointeeET;
1177 
1178   if (PointeeType->getTypeClass() != clang::Type::Pointer) {
1179     PointeeET = RSExportType::Create(Context, PointeeType);
1180   } else {
1181     // Double or higher dimension of pointer, export as int*
1182     PointeeET = RSExportPrimitiveType::Create(Context,
1183                     Context->getASTContext().IntTy.getTypePtr());
1184   }
1185 
1186   if (PointeeET == NULL) {
1187     // Error diagnostic is emitted for corresponding pointee type
1188     return NULL;
1189   }
1190 
1191   return new RSExportPointerType(Context, TypeName, PointeeET);
1192 }
1193 
convertToLLVMType() const1194 llvm::Type *RSExportPointerType::convertToLLVMType() const {
1195   llvm::Type *PointeeType = mPointeeType->getLLVMType();
1196   return llvm::PointerType::getUnqual(PointeeType);
1197 }
1198 
keep()1199 bool RSExportPointerType::keep() {
1200   if (!RSExportType::keep())
1201     return false;
1202   const_cast<RSExportType*>(mPointeeType)->keep();
1203   return true;
1204 }
1205 
equals(const RSExportable * E) const1206 bool RSExportPointerType::equals(const RSExportable *E) const {
1207   CHECK_PARENT_EQUALITY(RSExportType, E);
1208   return (static_cast<const RSExportPointerType*>(E)
1209               ->getPointeeType()->equals(getPointeeType()));
1210 }
1211 
1212 /***************************** RSExportVectorType *****************************/
1213 llvm::StringRef
GetTypeName(const clang::ExtVectorType * EVT)1214 RSExportVectorType::GetTypeName(const clang::ExtVectorType *EVT) {
1215   const clang::Type *ElementType = GetExtVectorElementType(EVT);
1216   llvm::StringRef name;
1217 
1218   if ((ElementType->getTypeClass() != clang::Type::Builtin))
1219     return name;
1220 
1221   const clang::BuiltinType *BT =
1222           static_cast<const clang::BuiltinType*>(
1223               ElementType->getCanonicalTypeInternal().getTypePtr());
1224 
1225   if ((EVT->getNumElements() < 1) ||
1226       (EVT->getNumElements() > 4))
1227     return name;
1228 
1229   BuiltinInfo *info = FindBuiltinType(BT->getKind());
1230   if (info != NULL) {
1231     int I = EVT->getNumElements() - 1;
1232     if (I < kMaxVectorSize) {
1233       name = info->cname[I];
1234     } else {
1235       slangAssert(false && "Max vector is 4");
1236     }
1237   }
1238   return name;
1239 }
1240 
Create(RSContext * Context,const clang::ExtVectorType * EVT,const llvm::StringRef & TypeName,bool Normalized)1241 RSExportVectorType *RSExportVectorType::Create(RSContext *Context,
1242                                                const clang::ExtVectorType *EVT,
1243                                                const llvm::StringRef &TypeName,
1244                                                bool Normalized) {
1245   slangAssert(EVT != NULL && EVT->getTypeClass() == clang::Type::ExtVector);
1246 
1247   const clang::Type *ElementType = GetExtVectorElementType(EVT);
1248   DataType DT = RSExportPrimitiveType::GetDataType(Context, ElementType);
1249 
1250   if (DT != DataTypeUnknown)
1251     return new RSExportVectorType(Context,
1252                                   TypeName,
1253                                   DT,
1254                                   Normalized,
1255                                   EVT->getNumElements());
1256   else
1257     return NULL;
1258 }
1259 
convertToLLVMType() const1260 llvm::Type *RSExportVectorType::convertToLLVMType() const {
1261   llvm::Type *ElementType = RSExportPrimitiveType::convertToLLVMType();
1262   return llvm::VectorType::get(ElementType, getNumElement());
1263 }
1264 
equals(const RSExportable * E) const1265 bool RSExportVectorType::equals(const RSExportable *E) const {
1266   CHECK_PARENT_EQUALITY(RSExportPrimitiveType, E);
1267   return (static_cast<const RSExportVectorType*>(E)->getNumElement()
1268               == getNumElement());
1269 }
1270 
1271 /***************************** RSExportMatrixType *****************************/
Create(RSContext * Context,const clang::RecordType * RT,const llvm::StringRef & TypeName,unsigned Dim)1272 RSExportMatrixType *RSExportMatrixType::Create(RSContext *Context,
1273                                                const clang::RecordType *RT,
1274                                                const llvm::StringRef &TypeName,
1275                                                unsigned Dim) {
1276   slangAssert((RT != NULL) && (RT->getTypeClass() == clang::Type::Record));
1277   slangAssert((Dim > 1) && "Invalid dimension of matrix");
1278 
1279   // Check whether the struct rs_matrix is in our expected form (but assume it's
1280   // correct if we're not sure whether it's correct or not)
1281   const clang::RecordDecl* RD = RT->getDecl();
1282   RD = RD->getDefinition();
1283   if (RD != NULL) {
1284     // Find definition, perform further examination
1285     if (RD->field_empty()) {
1286       Context->ReportError(
1287           RD->getLocation(),
1288           "invalid matrix struct: must have 1 field for saving values: '%0'")
1289           << RD->getName();
1290       return NULL;
1291     }
1292 
1293     clang::RecordDecl::field_iterator FIT = RD->field_begin();
1294     const clang::FieldDecl *FD = *FIT;
1295     const clang::Type *FT = RSExportType::GetTypeOfDecl(FD);
1296     if ((FT == NULL) || (FT->getTypeClass() != clang::Type::ConstantArray)) {
1297       Context->ReportError(RD->getLocation(),
1298                            "invalid matrix struct: first field should"
1299                            " be an array with constant size: '%0'")
1300           << RD->getName();
1301       return NULL;
1302     }
1303     const clang::ConstantArrayType *CAT =
1304       static_cast<const clang::ConstantArrayType *>(FT);
1305     const clang::Type *ElementType = GetConstantArrayElementType(CAT);
1306     if ((ElementType == NULL) ||
1307         (ElementType->getTypeClass() != clang::Type::Builtin) ||
1308         (static_cast<const clang::BuiltinType *>(ElementType)->getKind() !=
1309          clang::BuiltinType::Float)) {
1310       Context->ReportError(RD->getLocation(),
1311                            "invalid matrix struct: first field "
1312                            "should be a float array: '%0'")
1313           << RD->getName();
1314       return NULL;
1315     }
1316 
1317     if (CAT->getSize() != Dim * Dim) {
1318       Context->ReportError(RD->getLocation(),
1319                            "invalid matrix struct: first field "
1320                            "should be an array with size %0: '%1'")
1321           << (Dim * Dim) << (RD->getName());
1322       return NULL;
1323     }
1324 
1325     FIT++;
1326     if (FIT != RD->field_end()) {
1327       Context->ReportError(RD->getLocation(),
1328                            "invalid matrix struct: must have "
1329                            "exactly 1 field: '%0'")
1330           << RD->getName();
1331       return NULL;
1332     }
1333   }
1334 
1335   return new RSExportMatrixType(Context, TypeName, Dim);
1336 }
1337 
convertToLLVMType() const1338 llvm::Type *RSExportMatrixType::convertToLLVMType() const {
1339   // Construct LLVM type:
1340   // struct {
1341   //  float X[mDim * mDim];
1342   // }
1343 
1344   llvm::LLVMContext &C = getRSContext()->getLLVMContext();
1345   llvm::ArrayType *X = llvm::ArrayType::get(llvm::Type::getFloatTy(C),
1346                                             mDim * mDim);
1347   return llvm::StructType::get(C, X, false);
1348 }
1349 
equals(const RSExportable * E) const1350 bool RSExportMatrixType::equals(const RSExportable *E) const {
1351   CHECK_PARENT_EQUALITY(RSExportType, E);
1352   return (static_cast<const RSExportMatrixType*>(E)->getDim() == getDim());
1353 }
1354 
1355 /************************* RSExportConstantArrayType *************************/
1356 RSExportConstantArrayType
Create(RSContext * Context,const clang::ConstantArrayType * CAT)1357 *RSExportConstantArrayType::Create(RSContext *Context,
1358                                    const clang::ConstantArrayType *CAT) {
1359   slangAssert(CAT != NULL && CAT->getTypeClass() == clang::Type::ConstantArray);
1360 
1361   slangAssert((CAT->getSize().getActiveBits() < 32) && "array too large");
1362 
1363   unsigned Size = static_cast<unsigned>(CAT->getSize().getZExtValue());
1364   slangAssert((Size > 0) && "Constant array should have size greater than 0");
1365 
1366   const clang::Type *ElementType = GetConstantArrayElementType(CAT);
1367   RSExportType *ElementET = RSExportType::Create(Context, ElementType);
1368 
1369   if (ElementET == NULL) {
1370     return NULL;
1371   }
1372 
1373   return new RSExportConstantArrayType(Context,
1374                                        ElementET,
1375                                        Size);
1376 }
1377 
convertToLLVMType() const1378 llvm::Type *RSExportConstantArrayType::convertToLLVMType() const {
1379   return llvm::ArrayType::get(mElementType->getLLVMType(), getSize());
1380 }
1381 
keep()1382 bool RSExportConstantArrayType::keep() {
1383   if (!RSExportType::keep())
1384     return false;
1385   const_cast<RSExportType*>(mElementType)->keep();
1386   return true;
1387 }
1388 
equals(const RSExportable * E) const1389 bool RSExportConstantArrayType::equals(const RSExportable *E) const {
1390   CHECK_PARENT_EQUALITY(RSExportType, E);
1391   const RSExportConstantArrayType *RHS =
1392       static_cast<const RSExportConstantArrayType*>(E);
1393   return ((getSize() == RHS->getSize()) &&
1394           (getElementType()->equals(RHS->getElementType())));
1395 }
1396 
1397 /**************************** RSExportRecordType ****************************/
Create(RSContext * Context,const clang::RecordType * RT,const llvm::StringRef & TypeName,bool mIsArtificial)1398 RSExportRecordType *RSExportRecordType::Create(RSContext *Context,
1399                                                const clang::RecordType *RT,
1400                                                const llvm::StringRef &TypeName,
1401                                                bool mIsArtificial) {
1402   slangAssert(RT != NULL && RT->getTypeClass() == clang::Type::Record);
1403 
1404   const clang::RecordDecl *RD = RT->getDecl();
1405   slangAssert(RD->isStruct());
1406 
1407   RD = RD->getDefinition();
1408   if (RD == NULL) {
1409     slangAssert(false && "struct is not defined in this module");
1410     return NULL;
1411   }
1412 
1413   // Struct layout construct by clang. We rely on this for obtaining the
1414   // alloc size of a struct and offset of every field in that struct.
1415   const clang::ASTRecordLayout *RL =
1416       &Context->getASTContext().getASTRecordLayout(RD);
1417   slangAssert((RL != NULL) &&
1418       "Failed to retrieve the struct layout from Clang.");
1419 
1420   RSExportRecordType *ERT =
1421       new RSExportRecordType(Context,
1422                              TypeName,
1423                              RD->hasAttr<clang::PackedAttr>(),
1424                              mIsArtificial,
1425                              RL->getDataSize().getQuantity(),
1426                              RL->getSize().getQuantity());
1427   unsigned int Index = 0;
1428 
1429   for (clang::RecordDecl::field_iterator FI = RD->field_begin(),
1430            FE = RD->field_end();
1431        FI != FE;
1432        FI++, Index++) {
1433 
1434     // FIXME: All fields should be primitive type
1435     slangAssert(FI->getKind() == clang::Decl::Field);
1436     clang::FieldDecl *FD = *FI;
1437 
1438     if (FD->isBitField()) {
1439       return NULL;
1440     }
1441 
1442     // Type
1443     RSExportType *ET = RSExportElement::CreateFromDecl(Context, FD);
1444 
1445     if (ET != NULL) {
1446       ERT->mFields.push_back(
1447           new Field(ET, FD->getName(), ERT,
1448                     static_cast<size_t>(RL->getFieldOffset(Index) >> 3)));
1449     } else {
1450       Context->ReportError(RD->getLocation(),
1451                            "field type cannot be exported: '%0.%1'")
1452           << RD->getName() << FD->getName();
1453       return NULL;
1454     }
1455   }
1456 
1457   return ERT;
1458 }
1459 
convertToLLVMType() const1460 llvm::Type *RSExportRecordType::convertToLLVMType() const {
1461   // Create an opaque type since struct may reference itself recursively.
1462 
1463   // TODO(sliao): LLVM took out the OpaqueType. Any other to migrate to?
1464   std::vector<llvm::Type*> FieldTypes;
1465 
1466   for (const_field_iterator FI = fields_begin(), FE = fields_end();
1467        FI != FE;
1468        FI++) {
1469     const Field *F = *FI;
1470     const RSExportType *FET = F->getType();
1471 
1472     FieldTypes.push_back(FET->getLLVMType());
1473   }
1474 
1475   llvm::StructType *ST = llvm::StructType::get(getRSContext()->getLLVMContext(),
1476                                                FieldTypes,
1477                                                mIsPacked);
1478   if (ST != NULL) {
1479     return ST;
1480   } else {
1481     return NULL;
1482   }
1483 }
1484 
keep()1485 bool RSExportRecordType::keep() {
1486   if (!RSExportType::keep())
1487     return false;
1488   for (std::list<const Field*>::iterator I = mFields.begin(),
1489           E = mFields.end();
1490        I != E;
1491        I++) {
1492     const_cast<RSExportType*>((*I)->getType())->keep();
1493   }
1494   return true;
1495 }
1496 
equals(const RSExportable * E) const1497 bool RSExportRecordType::equals(const RSExportable *E) const {
1498   CHECK_PARENT_EQUALITY(RSExportType, E);
1499 
1500   const RSExportRecordType *ERT = static_cast<const RSExportRecordType*>(E);
1501 
1502   if (ERT->getFields().size() != getFields().size())
1503     return false;
1504 
1505   const_field_iterator AI = fields_begin(), BI = ERT->fields_begin();
1506 
1507   for (unsigned i = 0, e = getFields().size(); i != e; i++) {
1508     if (!(*AI)->getType()->equals((*BI)->getType()))
1509       return false;
1510     AI++;
1511     BI++;
1512   }
1513 
1514   return true;
1515 }
1516 
convertToRTD(RSReflectionTypeData * rtd) const1517 void RSExportType::convertToRTD(RSReflectionTypeData *rtd) const {
1518     memset(rtd, 0, sizeof(*rtd));
1519     rtd->vecSize = 1;
1520 
1521     switch(getClass()) {
1522     case RSExportType::ExportClassPrimitive: {
1523             const RSExportPrimitiveType *EPT = static_cast<const RSExportPrimitiveType*>(this);
1524             rtd->type = RSExportPrimitiveType::getRSReflectionType(EPT);
1525             return;
1526         }
1527     case RSExportType::ExportClassPointer: {
1528             const RSExportPointerType *EPT = static_cast<const RSExportPointerType*>(this);
1529             const RSExportType *PointeeType = EPT->getPointeeType();
1530             PointeeType->convertToRTD(rtd);
1531             rtd->isPointer = true;
1532             return;
1533         }
1534     case RSExportType::ExportClassVector: {
1535             const RSExportVectorType *EVT = static_cast<const RSExportVectorType*>(this);
1536             rtd->type = EVT->getRSReflectionType(EVT);
1537             rtd->vecSize = EVT->getNumElement();
1538             return;
1539         }
1540     case RSExportType::ExportClassMatrix: {
1541             const RSExportMatrixType *EMT = static_cast<const RSExportMatrixType*>(this);
1542             unsigned Dim = EMT->getDim();
1543             slangAssert((Dim >= 2) && (Dim <= 4));
1544             rtd->type = &gReflectionTypes[15 + Dim-2];
1545             return;
1546         }
1547     case RSExportType::ExportClassConstantArray: {
1548             const RSExportConstantArrayType* CAT =
1549               static_cast<const RSExportConstantArrayType*>(this);
1550             CAT->getElementType()->convertToRTD(rtd);
1551             rtd->arraySize = CAT->getSize();
1552             return;
1553         }
1554     case RSExportType::ExportClassRecord: {
1555             slangAssert(!"RSExportType::ExportClassRecord not implemented");
1556             return;// RS_TYPE_CLASS_NAME_PREFIX + ET->getName() + ".Item";
1557         }
1558     default: {
1559             slangAssert(false && "Unknown class of type");
1560         }
1561     }
1562 }
1563 
1564 
1565 }  // namespace slang
1566