Zonr Chang | c383a50 | 2010-10-12 01:52:08 +0800 | [diff] [blame] | 1 | /* |
| 2 | * Copyright 2010, 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 | |
Stephen Hines | e639eb5 | 2010-11-08 19:27:20 -0800 | [diff] [blame] | 17 | #ifndef _FRAMEWORKS_COMPILE_SLANG_SLANG_RS_EXPORT_TYPE_H_ // NOLINT |
| 18 | #define _FRAMEWORKS_COMPILE_SLANG_SLANG_RS_EXPORT_TYPE_H_ |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 19 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 20 | #include <list> |
Stephen Hines | e639eb5 | 2010-11-08 19:27:20 -0800 | [diff] [blame] | 21 | #include <set> |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 22 | #include <string> |
| 23 | |
zonr | 6315f76 | 2010-10-05 15:35:14 +0800 | [diff] [blame] | 24 | #include "clang/AST/Decl.h" |
Stephen Hines | e639eb5 | 2010-11-08 19:27:20 -0800 | [diff] [blame] | 25 | #include "clang/AST/Type.h" |
| 26 | |
| 27 | #include "llvm/ADT/SmallPtrSet.h" |
| 28 | #include "llvm/ADT/StringMap.h" |
| 29 | #include "llvm/ADT/StringRef.h" |
| 30 | |
| 31 | #include "llvm/Support/ManagedStatic.h" |
zonr | 6315f76 | 2010-10-05 15:35:14 +0800 | [diff] [blame] | 32 | |
Zonr Chang | a41ce1d | 2010-10-06 02:23:12 +0800 | [diff] [blame] | 33 | #include "slang_rs_exportable.h" |
| 34 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 35 | #define GET_CANONICAL_TYPE(T) \ |
Zonr Chang | 2e1dba6 | 2010-10-05 22:20:11 +0800 | [diff] [blame] | 36 | (((T) == NULL) ? NULL : (T)->getCanonicalTypeInternal().getTypePtr()) |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 37 | #define UNSAFE_CAST_TYPE(TT, T) \ |
Zonr Chang | 2e1dba6 | 2010-10-05 22:20:11 +0800 | [diff] [blame] | 38 | static_cast<TT*>(T->getCanonicalTypeInternal().getTypePtr()) |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 39 | #define GET_EXT_VECTOR_ELEMENT_TYPE(T) \ |
Zonr Chang | 2e1dba6 | 2010-10-05 22:20:11 +0800 | [diff] [blame] | 40 | (((T) == NULL) ? NULL : \ |
| 41 | GET_CANONICAL_TYPE((T)->getElementType().getTypePtr())) |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 42 | #define GET_POINTEE_TYPE(T) \ |
Zonr Chang | 2e1dba6 | 2010-10-05 22:20:11 +0800 | [diff] [blame] | 43 | (((T) == NULL) ? NULL : \ |
| 44 | GET_CANONICAL_TYPE((T)->getPointeeType().getTypePtr())) |
| 45 | #define GET_CONSTANT_ARRAY_ELEMENT_TYPE(T) \ |
| 46 | (((T) == NULL) ? NULL : \ |
| 47 | GET_CANONICAL_TYPE((T)->getElementType().getTypePtr())) |
Zonr Chang | 0da0a7d | 2010-10-05 21:26:37 +0800 | [diff] [blame] | 48 | #define DUMMY_RS_TYPE_NAME_PREFIX "<" |
| 49 | #define DUMMY_RS_TYPE_NAME_POSTFIX ">" |
Zonr Chang | 2e1dba6 | 2010-10-05 22:20:11 +0800 | [diff] [blame] | 50 | #define DUMMY_TYPE_NAME_FOR_RS_CONSTANT_ARRAY_TYPE \ |
| 51 | DUMMY_RS_TYPE_NAME_PREFIX"ConstantArray"DUMMY_RS_TYPE_NAME_POSTFIX |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 52 | |
Zonr Chang | a7a828d | 2010-10-22 03:44:27 +0800 | [diff] [blame] | 53 | union RSType; |
| 54 | |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 55 | namespace llvm { |
zonr | 6315f76 | 2010-10-05 15:35:14 +0800 | [diff] [blame] | 56 | class Type; |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 57 | } // namespace llvm |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 58 | |
| 59 | namespace slang { |
| 60 | |
zonr | 6315f76 | 2010-10-05 15:35:14 +0800 | [diff] [blame] | 61 | class RSContext; |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 62 | |
Zonr Chang | a41ce1d | 2010-10-06 02:23:12 +0800 | [diff] [blame] | 63 | class RSExportType : public RSExportable { |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 64 | friend class RSExportElement; |
| 65 | public: |
| 66 | typedef enum { |
| 67 | ExportClassPrimitive, |
| 68 | ExportClassPointer, |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 69 | ExportClassVector, |
Zonr Chang | 92b344a | 2010-10-05 20:39:03 +0800 | [diff] [blame] | 70 | ExportClassMatrix, |
Zonr Chang | 2e1dba6 | 2010-10-05 22:20:11 +0800 | [diff] [blame] | 71 | ExportClassConstantArray, |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 72 | ExportClassRecord |
| 73 | } ExportClass; |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 74 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 75 | private: |
Zonr Chang | 6b6320a | 2010-10-05 22:42:01 +0800 | [diff] [blame] | 76 | ExportClass mClass; |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 77 | std::string mName; |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 78 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 79 | // Cache the result after calling convertToLLVMType() at the first time |
| 80 | mutable const llvm::Type *mLLVMType; |
Zonr Chang | a7a828d | 2010-10-22 03:44:27 +0800 | [diff] [blame] | 81 | // Cache the result after calling convertToSpecType() at the first time |
| 82 | mutable union RSType *mSpecType; |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 83 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 84 | protected: |
Zonr Chang | 6b6320a | 2010-10-05 22:42:01 +0800 | [diff] [blame] | 85 | RSExportType(RSContext *Context, |
| 86 | ExportClass Class, |
| 87 | const llvm::StringRef &Name); |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 88 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 89 | // Let's make it private since there're some prerequisites to call this |
| 90 | // function. |
| 91 | // |
Stephen Hines | e5e6443 | 2010-12-02 18:48:20 -0800 | [diff] [blame] | 92 | // @T was normalized by calling RSExportType::NormalizeType(). |
zonr | 6315f76 | 2010-10-05 15:35:14 +0800 | [diff] [blame] | 93 | // @TypeName was retrieve from RSExportType::GetTypeName() before calling |
| 94 | // this. |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 95 | // |
| 96 | static RSExportType *Create(RSContext *Context, |
| 97 | const clang::Type *T, |
zonr | 6315f76 | 2010-10-05 15:35:14 +0800 | [diff] [blame] | 98 | const llvm::StringRef &TypeName); |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 99 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 100 | static llvm::StringRef GetTypeName(const clang::Type *T); |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 101 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 102 | // This function convert the RSExportType to LLVM type. Actually, it should be |
| 103 | // "convert Clang type to LLVM type." However, clang doesn't make this API |
| 104 | // (lib/CodeGen/CodeGenTypes.h) public, we need to do by ourselves. |
| 105 | // |
| 106 | // Once we can get LLVM type, we can use LLVM to get alignment information, |
| 107 | // allocation size of a given type and structure layout that LLVM used |
| 108 | // (all of these information are target dependent) without dealing with these |
| 109 | // by ourselves. |
| 110 | virtual const llvm::Type *convertToLLVMType() const = 0; |
Stephen Hines | e639eb5 | 2010-11-08 19:27:20 -0800 | [diff] [blame] | 111 | // Record type may recursively reference its type definition. We need a |
Zonr Chang | 3cd3dd3 | 2010-10-22 02:11:35 +0800 | [diff] [blame] | 112 | // temporary type setup before the type construction gets done. |
| 113 | inline void setAbstractLLVMType(const llvm::Type *LLVMType) const { |
| 114 | mLLVMType = LLVMType; |
| 115 | } |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 116 | |
Zonr Chang | a7a828d | 2010-10-22 03:44:27 +0800 | [diff] [blame] | 117 | virtual union RSType *convertToSpecType() const = 0; |
| 118 | inline void setSpecTypeTemporarily(union RSType *SpecType) const { |
| 119 | mSpecType = SpecType; |
| 120 | } |
| 121 | |
| 122 | virtual ~RSExportType(); |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 123 | public: |
Stephen Hines | e5e6443 | 2010-12-02 18:48:20 -0800 | [diff] [blame] | 124 | // This function additionally verifies that the Type T is exportable. |
| 125 | // If it is not, this function returns false. Otherwise it returns true. |
| 126 | static bool NormalizeType(const clang::Type *&T, |
| 127 | llvm::StringRef &TypeName, |
| 128 | clang::Diagnostic *Diags, |
Stephen Hines | dd6206b | 2010-12-09 19:39:22 -0800 | [diff] [blame] | 129 | clang::SourceManager *SM, |
| 130 | const clang::VarDecl *VD); |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 131 | // @T may not be normalized |
| 132 | static RSExportType *Create(RSContext *Context, const clang::Type *T); |
| 133 | static RSExportType *CreateFromDecl(RSContext *Context, |
| 134 | const clang::VarDecl *VD); |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 135 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 136 | static const clang::Type *GetTypeOfDecl(const clang::DeclaratorDecl *DD); |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 137 | |
Zonr Chang | 6b6320a | 2010-10-05 22:42:01 +0800 | [diff] [blame] | 138 | inline ExportClass getClass() const { return mClass; } |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 139 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 140 | inline const llvm::Type *getLLVMType() const { |
| 141 | if (mLLVMType == NULL) |
| 142 | mLLVMType = convertToLLVMType(); |
| 143 | return mLLVMType; |
| 144 | } |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 145 | |
Zonr Chang | a7a828d | 2010-10-22 03:44:27 +0800 | [diff] [blame] | 146 | inline const union RSType *getSpecType() const { |
| 147 | if (mSpecType == NULL) |
| 148 | mSpecType = convertToSpecType(); |
| 149 | return mSpecType; |
| 150 | } |
| 151 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 152 | // Return the number of bits necessary to hold the specified RSExportType |
| 153 | static size_t GetTypeStoreSize(const RSExportType *ET); |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 154 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 155 | // The size of allocation of specified RSExportType (alignment considered) |
| 156 | static size_t GetTypeAllocSize(const RSExportType *ET); |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 157 | |
Zonr Chang | 641558f | 2010-10-12 21:07:06 +0800 | [diff] [blame] | 158 | inline const std::string &getName() const { return mName; } |
| 159 | |
Zonr Chang | 3cd3dd3 | 2010-10-22 02:11:35 +0800 | [diff] [blame] | 160 | virtual bool keep(); |
Zonr Chang | 641558f | 2010-10-12 21:07:06 +0800 | [diff] [blame] | 161 | virtual bool equals(const RSExportable *E) const; |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 162 | }; // RSExportType |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 163 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 164 | // Primitive types |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 165 | class RSExportPrimitiveType : public RSExportType { |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 166 | friend class RSExportType; |
| 167 | friend class RSExportElement; |
| 168 | public: |
| 169 | // From graphics/java/android/renderscript/Element.java: Element.DataType |
| 170 | typedef enum { |
| 171 | DataTypeUnknown = -1, |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 172 | |
Zonr Chang | b1771ef | 2010-10-22 18:03:46 +0800 | [diff] [blame] | 173 | #define ENUM_PRIMITIVE_DATA_TYPE_RANGE(begin_type, end_type) \ |
| 174 | FirstPrimitiveType = DataType ## begin_type, \ |
| 175 | LastPrimitiveType = DataType ## end_type, |
| 176 | |
| 177 | #define ENUM_RS_MATRIX_DATA_TYPE_RANGE(begin_type, end_type) \ |
| 178 | FirstRSMatrixType = DataType ## begin_type, \ |
| 179 | LastRSMatrixType = DataType ## end_type, |
| 180 | |
| 181 | #define ENUM_RS_OBJECT_DATA_TYPE_RANGE(begin_type, end_type) \ |
| 182 | FirstRSObjectType = DataType ## begin_type, \ |
| 183 | LastRSObjectType = DataType ## end_type, |
| 184 | |
Zonr Chang | a65ec16 | 2010-10-17 01:53:05 +0800 | [diff] [blame] | 185 | #define ENUM_RS_DATA_TYPE(type, cname, bits) \ |
| 186 | DataType ## type, |
Zonr Chang | b1771ef | 2010-10-22 18:03:46 +0800 | [diff] [blame] | 187 | |
Zonr Chang | a65ec16 | 2010-10-17 01:53:05 +0800 | [diff] [blame] | 188 | #include "RSDataTypeEnums.inc" |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 189 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 190 | DataTypeMax |
| 191 | } DataType; |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 192 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 193 | // From graphics/java/android/renderscript/Element.java: Element.DataKind |
| 194 | typedef enum { |
Zonr Chang | a65ec16 | 2010-10-17 01:53:05 +0800 | [diff] [blame] | 195 | DataKindUnknown = -1 |
| 196 | #define ENUM_RS_DATA_KIND(kind) \ |
| 197 | , DataKind ## kind |
| 198 | #include "RSDataKindEnums.inc" |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 199 | } DataKind; |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 200 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 201 | private: |
Zonr Chang | a65ec16 | 2010-10-17 01:53:05 +0800 | [diff] [blame] | 202 | // NOTE: There's no any instance of RSExportPrimitiveType which mType |
| 203 | // is of the value DataTypeRSMatrix*. DataTypeRSMatrix* enumeration here is |
| 204 | // only for RSExportPrimitiveType::GetRSObjectType to *recognize* the struct |
| 205 | // rs_matrix{2x2, 3x3, 4x4}. These matrix type are represented as |
| 206 | // RSExportMatrixType. |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 207 | DataType mType; |
| 208 | DataKind mKind; |
| 209 | bool mNormalized; |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 210 | |
Zonr Chang | b1771ef | 2010-10-22 18:03:46 +0800 | [diff] [blame] | 211 | typedef llvm::StringMap<DataType> RSSpecificTypeMapTy; |
| 212 | static llvm::ManagedStatic<RSSpecificTypeMapTy> RSSpecificTypeMap; |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 213 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 214 | static llvm::Type *RSObjectLLVMType; |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 215 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 216 | static const size_t SizeOfDataTypeInBits[]; |
Stephen Hines | e5e6443 | 2010-12-02 18:48:20 -0800 | [diff] [blame] | 217 | // @T was normalized by calling RSExportType::NormalizeType() before calling |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 218 | // this. |
| 219 | // @TypeName was retrieved from RSExportType::GetTypeName() before calling |
| 220 | // this |
| 221 | static RSExportPrimitiveType *Create(RSContext *Context, |
| 222 | const clang::Type *T, |
| 223 | const llvm::StringRef &TypeName, |
| 224 | DataKind DK = DataKindUser, |
| 225 | bool Normalized = false); |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 226 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 227 | protected: |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 228 | RSExportPrimitiveType(RSContext *Context, |
Zonr Chang | 6b6320a | 2010-10-05 22:42:01 +0800 | [diff] [blame] | 229 | // for derived class to set their type class |
| 230 | ExportClass Class, |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 231 | const llvm::StringRef &Name, |
| 232 | DataType DT, |
| 233 | DataKind DK, |
zonr | 6315f76 | 2010-10-05 15:35:14 +0800 | [diff] [blame] | 234 | bool Normalized) |
Zonr Chang | 6b6320a | 2010-10-05 22:42:01 +0800 | [diff] [blame] | 235 | : RSExportType(Context, Class, Name), |
zonr | 6315f76 | 2010-10-05 15:35:14 +0800 | [diff] [blame] | 236 | mType(DT), |
| 237 | mKind(DK), |
| 238 | mNormalized(Normalized) { |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 239 | return; |
| 240 | } |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 241 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 242 | virtual const llvm::Type *convertToLLVMType() const; |
Zonr Chang | a7a828d | 2010-10-22 03:44:27 +0800 | [diff] [blame] | 243 | virtual union RSType *convertToSpecType() const; |
Stephen Hines | 2ef9bc0 | 2010-12-13 18:33:23 -0800 | [diff] [blame] | 244 | |
| 245 | static DataType GetDataType(RSContext *Context, const clang::Type *T); |
| 246 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 247 | public: |
Stephen Hines | dd6206b | 2010-12-09 19:39:22 -0800 | [diff] [blame] | 248 | // T is normalized by calling RSExportType::NormalizeType() before |
| 249 | // calling this |
| 250 | static bool IsPrimitiveType(const clang::Type *T); |
| 251 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 252 | // @T may not be normalized |
| 253 | static RSExportPrimitiveType *Create(RSContext *Context, |
| 254 | const clang::Type *T, |
| 255 | DataKind DK = DataKindUser); |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 256 | |
Zonr Chang | b1771ef | 2010-10-22 18:03:46 +0800 | [diff] [blame] | 257 | static DataType GetRSSpecificType(const llvm::StringRef &TypeName); |
| 258 | static DataType GetRSSpecificType(const clang::Type *T); |
| 259 | |
| 260 | static bool IsRSMatrixType(DataType DT); |
| 261 | static bool IsRSObjectType(DataType DT); |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 262 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 263 | static size_t GetSizeInBits(const RSExportPrimitiveType *EPT); |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 264 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 265 | inline DataType getType() const { return mType; } |
| 266 | inline DataKind getKind() const { return mKind; } |
| 267 | inline bool isRSObjectType() const { |
| 268 | return ((mType >= DataTypeRSElement) && (mType < DataTypeMax)); |
| 269 | } |
Zonr Chang | 641558f | 2010-10-12 21:07:06 +0800 | [diff] [blame] | 270 | |
| 271 | virtual bool equals(const RSExportable *E) const; |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 272 | }; // RSExportPrimitiveType |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 273 | |
| 274 | |
| 275 | class RSExportPointerType : public RSExportType { |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 276 | friend class RSExportType; |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 277 | friend class RSExportFunc; |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 278 | private: |
| 279 | const RSExportType *mPointeeType; |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 280 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 281 | RSExportPointerType(RSContext *Context, |
| 282 | const llvm::StringRef &Name, |
zonr | 6315f76 | 2010-10-05 15:35:14 +0800 | [diff] [blame] | 283 | const RSExportType *PointeeType) |
Zonr Chang | 6b6320a | 2010-10-05 22:42:01 +0800 | [diff] [blame] | 284 | : RSExportType(Context, ExportClassPointer, Name), |
zonr | 6315f76 | 2010-10-05 15:35:14 +0800 | [diff] [blame] | 285 | mPointeeType(PointeeType) { |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 286 | return; |
| 287 | } |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 288 | |
Stephen Hines | e5e6443 | 2010-12-02 18:48:20 -0800 | [diff] [blame] | 289 | // @PT was normalized by calling RSExportType::NormalizeType() before calling |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 290 | // this. |
| 291 | static RSExportPointerType *Create(RSContext *Context, |
| 292 | const clang::PointerType *PT, |
| 293 | const llvm::StringRef &TypeName); |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 294 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 295 | virtual const llvm::Type *convertToLLVMType() const; |
Zonr Chang | a7a828d | 2010-10-22 03:44:27 +0800 | [diff] [blame] | 296 | virtual union RSType *convertToSpecType() const; |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 297 | public: |
Zonr Chang | 3cd3dd3 | 2010-10-22 02:11:35 +0800 | [diff] [blame] | 298 | virtual bool keep(); |
Zonr Chang | 641558f | 2010-10-12 21:07:06 +0800 | [diff] [blame] | 299 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 300 | inline const RSExportType *getPointeeType() const { return mPointeeType; } |
Zonr Chang | 641558f | 2010-10-12 21:07:06 +0800 | [diff] [blame] | 301 | |
| 302 | virtual bool equals(const RSExportable *E) const; |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 303 | }; // RSExportPointerType |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 304 | |
| 305 | |
| 306 | class RSExportVectorType : public RSExportPrimitiveType { |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 307 | friend class RSExportType; |
| 308 | friend class RSExportElement; |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 309 | private: |
Zonr Chang | 92b344a | 2010-10-05 20:39:03 +0800 | [diff] [blame] | 310 | unsigned mNumElement; // number of element |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 311 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 312 | RSExportVectorType(RSContext *Context, |
| 313 | const llvm::StringRef &Name, |
| 314 | DataType DT, |
| 315 | DataKind DK, |
| 316 | bool Normalized, |
Zonr Chang | 92b344a | 2010-10-05 20:39:03 +0800 | [diff] [blame] | 317 | unsigned NumElement) |
Zonr Chang | 6b6320a | 2010-10-05 22:42:01 +0800 | [diff] [blame] | 318 | : RSExportPrimitiveType(Context, ExportClassVector, Name, |
| 319 | DT, DK, Normalized), |
zonr | 6315f76 | 2010-10-05 15:35:14 +0800 | [diff] [blame] | 320 | mNumElement(NumElement) { |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 321 | return; |
| 322 | } |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 323 | |
Stephen Hines | e5e6443 | 2010-12-02 18:48:20 -0800 | [diff] [blame] | 324 | // @EVT was normalized by calling RSExportType::NormalizeType() before |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 325 | // calling this. |
| 326 | static RSExportVectorType *Create(RSContext *Context, |
| 327 | const clang::ExtVectorType *EVT, |
| 328 | const llvm::StringRef &TypeName, |
| 329 | DataKind DK = DataKindUser, |
| 330 | bool Normalized = false); |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 331 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 332 | virtual const llvm::Type *convertToLLVMType() const; |
Zonr Chang | a7a828d | 2010-10-22 03:44:27 +0800 | [diff] [blame] | 333 | virtual union RSType *convertToSpecType() const; |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 334 | public: |
| 335 | static llvm::StringRef GetTypeName(const clang::ExtVectorType *EVT); |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 336 | |
Zonr Chang | 92b344a | 2010-10-05 20:39:03 +0800 | [diff] [blame] | 337 | inline unsigned getNumElement() const { return mNumElement; } |
Zonr Chang | 641558f | 2010-10-12 21:07:06 +0800 | [diff] [blame] | 338 | |
| 339 | virtual bool equals(const RSExportable *E) const; |
Zonr Chang | 92b344a | 2010-10-05 20:39:03 +0800 | [diff] [blame] | 340 | }; |
| 341 | |
| 342 | // Only *square* *float* matrix is supported by now. |
| 343 | // |
| 344 | // struct rs_matrix{2x2,3x3,4x4, ..., NxN} should be defined as the following |
| 345 | // form *exactly*: |
| 346 | // typedef struct { |
| 347 | // float m[{NxN}]; |
| 348 | // } rs_matrixNxN; |
| 349 | // |
| 350 | // where mDim will be N. |
| 351 | class RSExportMatrixType : public RSExportType { |
| 352 | friend class RSExportType; |
| 353 | private: |
Zonr Chang | 2e1dba6 | 2010-10-05 22:20:11 +0800 | [diff] [blame] | 354 | unsigned mDim; // dimension |
Zonr Chang | 92b344a | 2010-10-05 20:39:03 +0800 | [diff] [blame] | 355 | |
| 356 | RSExportMatrixType(RSContext *Context, |
| 357 | const llvm::StringRef &Name, |
| 358 | unsigned Dim) |
Zonr Chang | 6b6320a | 2010-10-05 22:42:01 +0800 | [diff] [blame] | 359 | : RSExportType(Context, ExportClassMatrix, Name), |
Zonr Chang | 92b344a | 2010-10-05 20:39:03 +0800 | [diff] [blame] | 360 | mDim(Dim) { |
| 361 | return; |
| 362 | } |
| 363 | |
| 364 | virtual const llvm::Type *convertToLLVMType() const; |
Zonr Chang | a7a828d | 2010-10-22 03:44:27 +0800 | [diff] [blame] | 365 | virtual union RSType *convertToSpecType() const; |
Zonr Chang | 92b344a | 2010-10-05 20:39:03 +0800 | [diff] [blame] | 366 | public: |
Stephen Hines | e5e6443 | 2010-12-02 18:48:20 -0800 | [diff] [blame] | 367 | // @RT was normalized by calling RSExportType::NormalizeType() before |
Zonr Chang | 92b344a | 2010-10-05 20:39:03 +0800 | [diff] [blame] | 368 | // calling this. |
| 369 | static RSExportMatrixType *Create(RSContext *Context, |
| 370 | const clang::RecordType *RT, |
| 371 | const llvm::StringRef &TypeName, |
| 372 | unsigned Dim); |
| 373 | |
| 374 | inline unsigned getDim() const { return mDim; } |
Zonr Chang | 641558f | 2010-10-12 21:07:06 +0800 | [diff] [blame] | 375 | |
| 376 | virtual bool equals(const RSExportable *E) const; |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 377 | }; |
| 378 | |
Zonr Chang | 2e1dba6 | 2010-10-05 22:20:11 +0800 | [diff] [blame] | 379 | class RSExportConstantArrayType : public RSExportType { |
| 380 | friend class RSExportType; |
| 381 | private: |
| 382 | const RSExportType *mElementType; // Array element type |
| 383 | unsigned mSize; // Array size |
| 384 | |
| 385 | RSExportConstantArrayType(RSContext *Context, |
| 386 | const RSExportType *ElementType, |
| 387 | unsigned Size) |
| 388 | : RSExportType(Context, |
Zonr Chang | 6b6320a | 2010-10-05 22:42:01 +0800 | [diff] [blame] | 389 | ExportClassConstantArray, |
Zonr Chang | 2e1dba6 | 2010-10-05 22:20:11 +0800 | [diff] [blame] | 390 | DUMMY_TYPE_NAME_FOR_RS_CONSTANT_ARRAY_TYPE), |
| 391 | mElementType(ElementType), |
| 392 | mSize(Size) { |
| 393 | return; |
| 394 | } |
| 395 | |
Stephen Hines | e5e6443 | 2010-12-02 18:48:20 -0800 | [diff] [blame] | 396 | // @CAT was normalized by calling RSExportType::NormalizeType() before |
Zonr Chang | 2e1dba6 | 2010-10-05 22:20:11 +0800 | [diff] [blame] | 397 | // calling this. |
| 398 | static RSExportConstantArrayType *Create(RSContext *Context, |
| 399 | const clang::ConstantArrayType *CAT); |
| 400 | |
| 401 | virtual const llvm::Type *convertToLLVMType() const; |
Zonr Chang | a7a828d | 2010-10-22 03:44:27 +0800 | [diff] [blame] | 402 | virtual union RSType *convertToSpecType() const; |
Zonr Chang | 2e1dba6 | 2010-10-05 22:20:11 +0800 | [diff] [blame] | 403 | public: |
Zonr Chang | 2e1dba6 | 2010-10-05 22:20:11 +0800 | [diff] [blame] | 404 | inline unsigned getSize() const { return mSize; } |
| 405 | inline const RSExportType *getElementType() const { return mElementType; } |
Zonr Chang | 641558f | 2010-10-12 21:07:06 +0800 | [diff] [blame] | 406 | |
Zonr Chang | 3cd3dd3 | 2010-10-22 02:11:35 +0800 | [diff] [blame] | 407 | virtual bool keep(); |
Zonr Chang | 641558f | 2010-10-12 21:07:06 +0800 | [diff] [blame] | 408 | virtual bool equals(const RSExportable *E) const; |
Zonr Chang | 2e1dba6 | 2010-10-05 22:20:11 +0800 | [diff] [blame] | 409 | }; |
| 410 | |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 411 | class RSExportRecordType : public RSExportType { |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 412 | friend class RSExportType; |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 413 | public: |
| 414 | class Field { |
| 415 | private: |
| 416 | const RSExportType *mType; |
| 417 | // Field name |
| 418 | std::string mName; |
| 419 | // Link to the struct that contain this field |
| 420 | const RSExportRecordType *mParent; |
Zonr Chang | 0da0a7d | 2010-10-05 21:26:37 +0800 | [diff] [blame] | 421 | // Offset in the container |
| 422 | size_t mOffset; |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 423 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 424 | public: |
| 425 | Field(const RSExportType *T, |
| 426 | const llvm::StringRef &Name, |
| 427 | const RSExportRecordType *Parent, |
Zonr Chang | 0da0a7d | 2010-10-05 21:26:37 +0800 | [diff] [blame] | 428 | size_t Offset) |
zonr | 6315f76 | 2010-10-05 15:35:14 +0800 | [diff] [blame] | 429 | : mType(T), |
| 430 | mName(Name.data(), Name.size()), |
| 431 | mParent(Parent), |
Zonr Chang | 0da0a7d | 2010-10-05 21:26:37 +0800 | [diff] [blame] | 432 | mOffset(Offset) { |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 433 | return; |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 434 | } |
| 435 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 436 | inline const RSExportRecordType *getParent() const { return mParent; } |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 437 | inline const RSExportType *getType() const { return mType; } |
| 438 | inline const std::string &getName() const { return mName; } |
Zonr Chang | 0da0a7d | 2010-10-05 21:26:37 +0800 | [diff] [blame] | 439 | inline size_t getOffsetInParent() const { return mOffset; } |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 440 | }; |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 441 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 442 | typedef std::list<const Field*>::const_iterator const_field_iterator; |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 443 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 444 | inline const_field_iterator fields_begin() const { |
| 445 | return this->mFields.begin(); |
| 446 | } |
| 447 | inline const_field_iterator fields_end() const { |
| 448 | return this->mFields.end(); |
| 449 | } |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 450 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 451 | private: |
| 452 | std::list<const Field*> mFields; |
| 453 | bool mIsPacked; |
| 454 | // Artificial export struct type is not exported by user (and thus it won't |
| 455 | // get reflected) |
| 456 | bool mIsArtificial; |
Zonr Chang | 0da0a7d | 2010-10-05 21:26:37 +0800 | [diff] [blame] | 457 | size_t mAllocSize; |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 458 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 459 | RSExportRecordType(RSContext *Context, |
| 460 | const llvm::StringRef &Name, |
| 461 | bool IsPacked, |
Zonr Chang | 0da0a7d | 2010-10-05 21:26:37 +0800 | [diff] [blame] | 462 | bool IsArtificial, |
| 463 | size_t AllocSize) |
Zonr Chang | 6b6320a | 2010-10-05 22:42:01 +0800 | [diff] [blame] | 464 | : RSExportType(Context, ExportClassRecord, Name), |
zonr | 6315f76 | 2010-10-05 15:35:14 +0800 | [diff] [blame] | 465 | mIsPacked(IsPacked), |
Zonr Chang | 0da0a7d | 2010-10-05 21:26:37 +0800 | [diff] [blame] | 466 | mIsArtificial(IsArtificial), |
| 467 | mAllocSize(AllocSize) { |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 468 | return; |
| 469 | } |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 470 | |
Stephen Hines | e5e6443 | 2010-12-02 18:48:20 -0800 | [diff] [blame] | 471 | // @RT was normalized by calling RSExportType::NormalizeType() before calling |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 472 | // this. |
| 473 | // @TypeName was retrieved from RSExportType::GetTypeName() before calling |
| 474 | // this. |
| 475 | static RSExportRecordType *Create(RSContext *Context, |
| 476 | const clang::RecordType *RT, |
| 477 | const llvm::StringRef &TypeName, |
| 478 | bool mIsArtificial = false); |
| 479 | |
| 480 | virtual const llvm::Type *convertToLLVMType() const; |
Zonr Chang | a7a828d | 2010-10-22 03:44:27 +0800 | [diff] [blame] | 481 | virtual union RSType *convertToSpecType() const; |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 482 | public: |
Zonr Chang | 0da0a7d | 2010-10-05 21:26:37 +0800 | [diff] [blame] | 483 | inline const std::list<const Field*>& getFields() const { return mFields; } |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 484 | inline bool isPacked() const { return mIsPacked; } |
| 485 | inline bool isArtificial() const { return mIsArtificial; } |
Zonr Chang | 0da0a7d | 2010-10-05 21:26:37 +0800 | [diff] [blame] | 486 | inline size_t getAllocSize() const { return mAllocSize; } |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 487 | |
Zonr Chang | 3cd3dd3 | 2010-10-22 02:11:35 +0800 | [diff] [blame] | 488 | virtual bool keep(); |
Zonr Chang | 641558f | 2010-10-12 21:07:06 +0800 | [diff] [blame] | 489 | virtual bool equals(const RSExportable *E) const; |
| 490 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 491 | ~RSExportRecordType() { |
| 492 | for (std::list<const Field*>::iterator I = mFields.begin(), |
| 493 | E = mFields.end(); |
| 494 | I != E; |
| 495 | I++) |
| 496 | if (*I != NULL) |
| 497 | delete *I; |
| 498 | return; |
| 499 | } |
| 500 | }; // RSExportRecordType |
| 501 | |
| 502 | } // namespace slang |
| 503 | |
Stephen Hines | e639eb5 | 2010-11-08 19:27:20 -0800 | [diff] [blame] | 504 | #endif // _FRAMEWORKS_COMPILE_SLANG_SLANG_RS_EXPORT_TYPE_H_ NOLINT |