slang_rs_export_type.cpp revision f8149d9e5a3795e9952717ee6346789a134c55c7
1462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao#include <vector> 2462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 3462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao#include "slang_rs_context.hpp" 4462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao#include "slang_rs_export_type.hpp" 5462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao#include "slang_rs_export_element.hpp" 6462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 7462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao#include "llvm/Type.h" 8462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao#include "llvm/DerivedTypes.h" 9462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 10462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao#include "llvm/ADT/StringExtras.h" /* for function llvm::utostr_32() */ 11462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao#include "llvm/Target/TargetData.h" /* for class llvm::TargetData and class llvm::StructLayout */ 12462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 130a3f20ec28ed6f5ae1ed5d61f6b6e3e577f7f5d1Shih-wei Liao#include "clang/AST/RecordLayout.h" 140a3f20ec28ed6f5ae1ed5d61f6b6e3e577f7f5d1Shih-wei Liao 15462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liaonamespace slang { 16462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 17462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao/****************************** RSExportType ******************************/ 18462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liaobool RSExportType::NormalizeType(const Type*& T, llvm::StringRef& TypeName) { 19462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao llvm::SmallPtrSet<const Type*, 8> SPS = llvm::SmallPtrSet<const Type*, 8>(); 20462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 211f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao if((T = RSExportType::TypeExportable(T, SPS)) == NULL) 22462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao /* TODO: warning the user: type not exportable */ 23462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return false; 24462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 25462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao /* Get type name */ 26462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao TypeName = RSExportType::GetTypeName(T); 27462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao if(TypeName.empty()) 28f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao /* TODO: warning the user: the type is unnamed */ 29462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return false; 30462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 31462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return true; 32462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao} 33462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 34462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liaoconst Type* RSExportType::GetTypeOfDecl(const DeclaratorDecl* DD) { 35462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao if(DD && DD->getTypeSourceInfo()) { 36462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao QualType T = DD->getTypeSourceInfo()->getType(); 37462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao if(T.isNull()) 38462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return NULL; 39462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao else 40462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return T.getTypePtr(); 41462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao } 42462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao} 43462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 44462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liaollvm::StringRef RSExportType::GetTypeName(const Type* T) { 45462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao T = GET_CANONICAL_TYPE(T); 46462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao if(T == NULL) 47462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return llvm::StringRef(); 48462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 49462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao switch(T->getTypeClass()) { 50462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao case Type::Builtin: 51462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao { 52462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao const BuiltinType* BT = UNSAFE_CAST_TYPE(BuiltinType, T); 53462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 54462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao switch(BT->getKind()) { 55462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao#define SLANG_RS_SUPPORT_BUILTIN_TYPE(builtin_type, type) \ 56462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao case builtin_type: \ 57462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao /* Compiler is smart enough to optimize following *big if branches* since they all become "constant comparison" after macro expansion */ \ 58462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao if(type == RSExportPrimitiveType::DataTypeFloat32) return "float"; \ 59462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao else if(type == RSExportPrimitiveType::DataTypeUnsigned8) return "uchar"; \ 60462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao else if(type == RSExportPrimitiveType::DataTypeUnsigned16) return "ushort"; \ 61462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao else if(type == RSExportPrimitiveType::DataTypeUnsigned32) return "uint"; \ 62462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao else if(type == RSExportPrimitiveType::DataTypeSigned8) return "char"; \ 63462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao else if(type == RSExportPrimitiveType::DataTypeSigned16) return "short"; \ 64462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao else if(type == RSExportPrimitiveType::DataTypeSigned32) return "int"; \ 651f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao else if(type == RSExportPrimitiveType::DataTypeBool) return "bool"; \ 66462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao else assert(false && "Unknow data type of supported builtin"); \ 67462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao break; 68462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao#include "slang_rs_export_type_support.inc" 69462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 70462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao default: assert(false && "Unknow data type of the builtin"); break; 71462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao } 72462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao } 73462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao break; 74462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 75462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao case Type::Record: 76462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao { 77462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao const RecordDecl* RD = T->getAsStructureType()->getDecl(); 78462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao llvm::StringRef Name = RD->getName(); 79462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao if(Name.empty()) { 801f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao if(RD->getTypedefForAnonDecl() != NULL) 81462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao Name = RD->getTypedefForAnonDecl()->getName(); 82462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 83462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao if(Name.empty()) 84462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao /* Try to find a name from redeclaration (i.e. typedef) */ 85462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao for(TagDecl::redecl_iterator RI = RD->redecls_begin(); 86462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao RI != RD->redecls_end(); 87462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao RI++) 88462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao { 89462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao assert(*RI != NULL && "cannot be NULL object"); 90462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 91462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao Name = (*RI)->getName(); 92462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao if(!Name.empty()) 93462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao break; 94462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao } 95462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao } 96462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return Name; 97462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao } 98462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao break; 99462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 100462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao case Type::Pointer: 101462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao { 102462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao /* "*" plus pointee name */ 103462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao const Type* PT = GET_POINTEE_TYPE(T); 104462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao llvm::StringRef PointeeName; 105462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao if(NormalizeType(PT, PointeeName)) { 1064d7dc59dd4d82aeeb7d77d1cfc048944e6e2bdbaShih-wei Liao char* Name = new char[ 1 /* * */ + PointeeName.size() + 1 ]; 107462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao Name[0] = '*'; 108462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao memcpy(Name + 1, PointeeName.data(), PointeeName.size()); 1094d7dc59dd4d82aeeb7d77d1cfc048944e6e2bdbaShih-wei Liao Name[PointeeName.size() + 1] = '\0'; 110462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return Name; 111462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao } 112462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao } 113462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao break; 114462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 115f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao case Type::ConstantArray: 116f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao { 117f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao const ConstantArrayType* ECT = UNSAFE_CAST_TYPE(ConstantArrayType, T); 118f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao return RSExportConstantArrayType::GetTypeName(ECT); 119f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao } 120f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao break; 121f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao 122462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao case Type::ExtVector: 123462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao { 124462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao const ExtVectorType* EVT = UNSAFE_CAST_TYPE(ExtVectorType, T); 125462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return RSExportVectorType::GetTypeName(EVT); 126462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao } 127462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao break; 128462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 129462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao default: 130462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao break; 131462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao } 132462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 133462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return llvm::StringRef(); 134462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao} 135462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 136462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liaoconst Type* RSExportType::TypeExportable(const Type* T, llvm::SmallPtrSet<const Type*, 8>& SPS) { 137462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao /* Normailize first */ 138462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao if((T = GET_CANONICAL_TYPE(T)) == NULL) 139462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return NULL; 140462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 141462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao if(SPS.count(T)) 142462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return T; 1431f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao 144462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao switch(T->getTypeClass()) { 145462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao case Type::Builtin: 146462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao { 147462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao const BuiltinType* BT = UNSAFE_CAST_TYPE(BuiltinType, T); 148462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 149462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao switch(BT->getKind()) { 150462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao#define SLANG_RS_SUPPORT_BUILTIN_TYPE(builtin_type, type) \ 151462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao case builtin_type: 152462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao#include "slang_rs_export_type_support.inc" 153462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return T; 154462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao break; 155462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 156462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao default: 157462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return NULL; 158462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao break; 159462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao } 160462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao } 161462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao break; 162462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 163462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao case Type::Record: 164462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao { 165462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao if(RSExportPrimitiveType::GetRSObjectType(T) != RSExportPrimitiveType::DataTypeUnknown) 166462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return T; /* RS object type, no further checks are needed */ 167462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 168462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao /* Check internal struct */ 169462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao const RecordDecl* RD = T->getAsStructureType()->getDecl(); 170462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao if(RD != NULL) 171462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao RD = RD->getDefinition(); 172462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 173462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao /* Fast check */ 1741f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao if(RD->hasFlexibleArrayMember() || RD->hasObjectMember()) 175462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return NULL; 1761f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao 177462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao /* Insert myself into checking set */ 178462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao SPS.insert(T); 179462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 180462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao /* Check all element */ 181462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao for(RecordDecl::field_iterator F = RD->field_begin(); 182462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao F != RD->field_end(); 183462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao F++) 184462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao { 185462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao const Type* FT = GetTypeOfDecl(*F); 186462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao FT = GET_CANONICAL_TYPE(FT); 187462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 1881f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao if(!TypeExportable(FT, SPS)) 189462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao /* TODO: warning: unsupported field type */ 190462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return NULL; 191462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao } 192462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 193462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return T; 194462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao } 195462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao break; 196462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 197462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao case Type::Pointer: 198462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao { 199462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao const PointerType* PT = UNSAFE_CAST_TYPE(PointerType, T); 200462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao const Type* PointeeType = GET_POINTEE_TYPE(PT); 201462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 202462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao if((PointeeType->getTypeClass() != Type::Pointer) && (TypeExportable(PointeeType, SPS) == NULL) ) 203462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return NULL; 204462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return T; 205462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao } 206462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao break; 207462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 208f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao case Type::ConstantArray: 209f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao { 210f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao const ConstantArrayType* ECT = UNSAFE_CAST_TYPE(ConstantArrayType, T); 211f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao /* Check element numbers */ 212f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao // if (ECT->getNumElements() != 4 && ECT->getNumElements() != 9 && ECT->getNumElements() != 16) { /* only support 2x2, 3x3, 4x4 arrays */ 213f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao // return NULL; 214f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao 215f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao /* Check base element type */ 216f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao const Type* ElementType = GET_CONSTANT_ARRAY_ELEMENT_TYPE(ECT); 217f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao 218f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao if((ElementType->getTypeClass() != Type::Builtin) || (TypeExportable(ElementType, SPS) == NULL)) 219f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao return NULL; 220f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao return T; 221f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao } 222f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao break; 223f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao 224462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao case Type::ExtVector: 225462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao { 226462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao const ExtVectorType* EVT = UNSAFE_CAST_TYPE(ExtVectorType, T); 227462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao /* Check element numbers */ 228462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao if(EVT->getNumElements() < 2 || EVT->getNumElements() > 4) /* only support vector with size 2, 3 and 4 */ 229462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return NULL; 230462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 231462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao /* Check base element type */ 232462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao const Type* ElementType = GET_EXT_VECTOR_ELEMENT_TYPE(EVT); 233462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 234462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao if((ElementType->getTypeClass() != Type::Builtin) || (TypeExportable(ElementType, SPS) == NULL)) 235462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return NULL; 236462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return T; 237462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao } 238462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao break; 239462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 240462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao default: 241462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return NULL; 242462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao break; 243462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao } 244462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao} 245462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 246462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei LiaoRSExportType* RSExportType::Create(RSContext* Context, const Type* T, const llvm::StringRef& TypeName) { 2471f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao /* 2481f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao * Lookup the context to see whether the type was processed before. 2491f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao * Newly create RSExportType will insert into context in RSExportType::RSExportType() 250462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao */ 251462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao RSContext::export_type_iterator ETI = Context->findExportType(TypeName); 252462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 2531f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao if(ETI != Context->export_types_end()) 254462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return ETI->second; 255462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 256462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao RSExportType* ET = NULL; 257462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao switch(T->getTypeClass()) { 258f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao case Type::Record: { 259f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao RSExportPrimitiveType::DataType dt = RSExportPrimitiveType::GetRSObjectType(TypeName); 260f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao switch (dt) { 261f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao case RSExportPrimitiveType::DataTypeUnknown: // User-defined types 262462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao ET = RSExportRecordType::Create(Context, T->getAsStructureType(), TypeName); 263f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao break; 264f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao case RSExportPrimitiveType::DataTypeRSMatrix2x2: 265f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao case RSExportPrimitiveType::DataTypeRSMatrix3x3: 266f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao case RSExportPrimitiveType::DataTypeRSMatrix4x4: { 267f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao const clang::RecordType* RT = static_cast<const RecordType*> (T); 268f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao const RecordDecl* RD = RT->getDecl(); 269f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao RD = RD->getDefinition(); 270f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao RecordDecl::field_iterator fit = RD->field_begin(); 271f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao FieldDecl* FD = *fit; 272f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao const Type* FT = RSExportType::GetTypeOfDecl(FD); 273f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao ET = RSExportConstantArrayType::Create(Context, static_cast<const ConstantArrayType*> (FT), TypeName); 274f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao break; 275f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao } 276f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao default: 277f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao ET = RSExportPrimitiveType::Create(Context, T, TypeName); 278f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao break; 279f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao } 280f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao break; 281f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao } 282462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao case Type::Builtin: 283462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao ET = RSExportPrimitiveType::Create(Context, T, TypeName); 284f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao break; 285462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 286462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao case Type::Pointer: 287462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao ET = RSExportPointerType::Create(Context, UNSAFE_CAST_TYPE(PointerType, T), TypeName); 288462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao /* free the name (allocated in RSExportType::GetTypeName) */ 289462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao delete [] TypeName.data(); 290f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao break; 291f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao 292f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao case Type::ConstantArray: 293f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao ET = RSExportConstantArrayType::Create(Context, UNSAFE_CAST_TYPE(ConstantArrayType, T), TypeName); 294f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao break; 295462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 296462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao case Type::ExtVector: 297462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao ET = RSExportVectorType::Create(Context, UNSAFE_CAST_TYPE(ExtVectorType, T), TypeName); 298f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao break; 299462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 300462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao default: 301462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao /* TODO: warning: type is not exportable */ 302462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao printf("RSExportType::Create : type '%s' is not exportable\n", T->getTypeClassName()); 303f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao break; 304462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao } 305462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 306462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return ET; 307462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao} 3081f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao 309462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei LiaoRSExportType* RSExportType::Create(RSContext* Context, const Type* T) { 310462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao llvm::StringRef TypeName; 311462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao if(NormalizeType(T, TypeName)) 312462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return Create(Context, T, TypeName); 313462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao else 314462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return NULL; 315462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao} 316462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 317462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei LiaoRSExportType* RSExportType::CreateFromDecl(RSContext* Context, const VarDecl* VD) { 318462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return RSExportType::Create(Context, GetTypeOfDecl(VD)); 319462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao} 320462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 321462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liaosize_t RSExportType::GetTypeStoreSize(const RSExportType* ET) { 322462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return ET->getRSContext()->getTargetData()->getTypeStoreSize(ET->getLLVMType()); 323462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao} 324462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 325462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liaosize_t RSExportType::GetTypeAllocSize(const RSExportType* ET) { 3260a3f20ec28ed6f5ae1ed5d61f6b6e3e577f7f5d1Shih-wei Liao if(ET->getClass() == RSExportType::ExportClassRecord) 3270a3f20ec28ed6f5ae1ed5d61f6b6e3e577f7f5d1Shih-wei Liao return static_cast<const RSExportRecordType*>(ET)->getAllocSize(); 3280a3f20ec28ed6f5ae1ed5d61f6b6e3e577f7f5d1Shih-wei Liao else 3290a3f20ec28ed6f5ae1ed5d61f6b6e3e577f7f5d1Shih-wei Liao return ET->getRSContext()->getTargetData()->getTypeAllocSize(ET->getLLVMType()); 330462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao} 331462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 3321f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei LiaoRSExportType::RSExportType(RSContext* Context, const llvm::StringRef& Name) : 3331f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao mContext(Context), 3341f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao mName(Name.data(), Name.size()), /* make a copy on Name since data of @Name which is stored in ASTContext will be destroyed later */ 335462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao mLLVMType(NULL) 3361f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao{ 337462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao /* TODO: need to check whether the insertion is successful or not */ 338462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao Context->insertExportType(Name, this); 3391f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao return; 340462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao} 341462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 342462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao/****************************** RSExportPrimitiveType ******************************/ 343462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei LiaoRSExportPrimitiveType::RSObjectTypeMapTy* RSExportPrimitiveType::RSObjectTypeMap = NULL; 344462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liaollvm::Type* RSExportPrimitiveType::RSObjectLLVMType = NULL; 345462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 346462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liaobool RSExportPrimitiveType::IsPrimitiveType(const Type* T) { 347462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao if((T != NULL) && (T->getTypeClass() == Type::Builtin)) 348462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return true; 349462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao else 350462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return false; 351462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao} 352462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 353462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei LiaoRSExportPrimitiveType::DataType RSExportPrimitiveType::GetRSObjectType(const llvm::StringRef& TypeName) { 354462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao if(TypeName.empty()) 355462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return DataTypeUnknown; 356462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 357462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao if(RSObjectTypeMap == NULL) { 358462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao RSObjectTypeMap = new RSObjectTypeMapTy(16); 359462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 360462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao#define USE_ELEMENT_DATA_TYPE 361462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao#define DEF_RS_OBJECT_TYPE(type, name) \ 362462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao RSObjectTypeMap->GetOrCreateValue(name, GET_ELEMENT_DATA_TYPE(type)); 363462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao#include "slang_rs_export_element_support.inc" 364462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao } 365462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 366462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao RSObjectTypeMapTy::const_iterator I = RSObjectTypeMap->find(TypeName); 367462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao if(I == RSObjectTypeMap->end()) 368462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return DataTypeUnknown; 369462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao else 370462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return I->getValue(); 371462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao} 372462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 373462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei LiaoRSExportPrimitiveType::DataType RSExportPrimitiveType::GetRSObjectType(const Type* T) { 374462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao T = GET_CANONICAL_TYPE(T); 375462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao if((T == NULL) || (T->getTypeClass() != Type::Record)) 376462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return DataTypeUnknown; 377462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 378462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return GetRSObjectType( RSExportType::GetTypeName(T) ); 379462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao} 380462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 3811f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liaoconst size_t RSExportPrimitiveType::SizeOfDataTypeInBits[RSExportPrimitiveType::DataTypeMax + 1] = { 382462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 0, 3831f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao 16, /* DataTypeFloat16 */ 3841f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao 32, /* DataTypeFloat32 */ 3851f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao 64, /* DataTypeFloat64 */ 3861f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao 8, /* DataTypeSigned8 */ 3871f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao 16, /* DataTypeSigned16 */ 3881f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao 32, /* DataTypeSigned32 */ 3891f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao 64, /* DataTypeSigned64 */ 3901f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao 8, /* DataTypeUnsigned8 */ 3911f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao 16, /* DataTypeUnsigned16 */ 3921f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao 32, /* DataTypeUnsigned32 */ 3931f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao 64, /* DataTypeUnSigned64 */ 3941f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao 3951f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao 16, /* DataTypeUnsigned565 */ 3961f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao 16, /* DataTypeUnsigned5551 */ 3971f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao 16, /* DataTypeUnsigned4444 */ 3981f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao 3991f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao 1, /* DataTypeBool */ 4001f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao 4011f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao 32, /* DataTypeRSElement */ 4021f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao 32, /* DataTypeRSType */ 4031f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao 32, /* DataTypeRSAllocation */ 4041f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao 32, /* DataTypeRSSampler */ 4051f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao 32, /* DataTypeRSScript */ 4061f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao 32, /* DataTypeRSMesh */ 4071f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao 32, /* DataTypeRSProgramFragment */ 4081f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao 32, /* DataTypeRSProgramVertex */ 4091f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao 32, /* DataTypeRSProgramRaster */ 4101f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao 32, /* DataTypeRSProgramStore */ 4111f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao 32, /* DataTypeRSFont */ 412116d3d2fcbde4970b07263deeeabc341ff8d76e9Shih-wei Liao 128, /* DataTypeRSMatrix2x2 */ 413116d3d2fcbde4970b07263deeeabc341ff8d76e9Shih-wei Liao 288, /* DataTypeRSMatrix3x3 */ 414116d3d2fcbde4970b07263deeeabc341ff8d76e9Shih-wei Liao 512, /* DataTypeRSMatrix4x4 */ 415462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 0 416462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao}; 417462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 4181f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liaosize_t RSExportPrimitiveType::GetSizeInBits(const RSExportPrimitiveType* EPT) { 4191f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao assert(((EPT->getType() >= DataTypeFloat32) && (EPT->getType() < DataTypeMax)) && "RSExportPrimitiveType::GetSizeInBits : unknown data type"); 4201f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao return SizeOfDataTypeInBits[ static_cast<int>(EPT->getType()) ]; 421462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao} 422462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 423462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei LiaoRSExportPrimitiveType::DataType RSExportPrimitiveType::GetDataType(const Type* T) { 424462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao if(T == NULL) 425462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return DataTypeUnknown; 426462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 427462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao switch(T->getTypeClass()) { 428462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao case Type::Builtin: 429462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao { 430462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao const BuiltinType* BT = UNSAFE_CAST_TYPE(BuiltinType, T); 431462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao switch(BT->getKind()) { 432462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao#define SLANG_RS_SUPPORT_BUILTIN_TYPE(builtin_type, type) \ 433462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao case builtin_type: \ 434462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return type; \ 435462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao break; 436462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao#include "slang_rs_export_type_support.inc" 437462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 438462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao /* The size of types Long, ULong and WChar depend on platform so we abandon the support to them */ 439462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao /* Type of its size exceeds 32 bits (e.g. int64_t, double, etc.) does not support */ 440462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 441462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao default: 442462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao /* TODO: warning the user: unsupported type */ 443462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao printf("RSExportPrimitiveType::GetDataType : built-in type has no corresponding data type for built-in type"); 444462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao break; 445462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao } 446462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao } 447462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao break; 448462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 449462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao case Type::Record: 450462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao /* must be RS object type */ 451462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return RSExportPrimitiveType::GetRSObjectType(T); 452462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao break; 453462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 454462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao default: 455462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao printf("RSExportPrimitiveType::GetDataType : type '%s' is not supported primitive type", T->getTypeClassName()); 456462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao break; 457462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao } 458462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 459462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return DataTypeUnknown; 460462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao} 461462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 462462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei LiaoRSExportPrimitiveType* RSExportPrimitiveType::Create(RSContext* Context, const Type* T, const llvm::StringRef& TypeName, DataKind DK, bool Normalized) { 463462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao DataType DT = GetDataType(T); 464462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 465462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao if((DT == DataTypeUnknown) || TypeName.empty()) 466462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return NULL; 467462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao else 468462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return new RSExportPrimitiveType(Context, TypeName, DT, DK, Normalized); 469462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao} 470462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 471462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei LiaoRSExportPrimitiveType* RSExportPrimitiveType::Create(RSContext* Context, const Type* T, DataKind DK) { 472462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao llvm::StringRef TypeName; 473462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao if(RSExportType::NormalizeType(T, TypeName) && IsPrimitiveType(T)) 474462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return Create(Context, T, TypeName, DK); 475462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao else 476462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return NULL; 477462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao} 478462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 4791f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei LiaoRSExportType::ExportClass RSExportPrimitiveType::getClass() const { 4801f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao return RSExportType::ExportClassPrimitive; 481462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao} 482462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 483462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liaoconst llvm::Type* RSExportPrimitiveType::convertToLLVMType() const { 484462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao llvm::LLVMContext& C = getRSContext()->getLLVMContext(); 485462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 486462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao if(isRSObjectType()) { 487462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao /* struct { int* p; } __attribute__((packed, aligned(pointer_size))) which is <{ [1 x i32] }> in LLVM */ 488462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao if(RSObjectLLVMType == NULL) { 489462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao std::vector<const llvm::Type*> Elements; 490462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao Elements.push_back( llvm::ArrayType::get(llvm::Type::getInt32Ty(C), 1) ); 491462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao RSObjectLLVMType = llvm::StructType::get(C, Elements, true); 492462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao } 493462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return RSObjectLLVMType; 494462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao } 495462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 496462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao switch(mType) { 4971f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao case DataTypeFloat32: 4981f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao return llvm::Type::getFloatTy(C); 499462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao break; 500462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 501462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao case DataTypeSigned8: 502462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao case DataTypeUnsigned8: 5031f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao return llvm::Type::getInt8Ty(C); 504462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao break; 505462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 506462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao case DataTypeSigned16: 507462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao case DataTypeUnsigned16: 508462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao case DataTypeUnsigned565: 509462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao case DataTypeUnsigned5551: 510462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao case DataTypeUnsigned4444: 5111f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao return llvm::Type::getInt16Ty(C); 512462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao break; 513462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 514462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao case DataTypeSigned32: 515462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao case DataTypeUnsigned32: 5161f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao return llvm::Type::getInt32Ty(C); 5171f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao break; 5181f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao 5191f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao case DataTypeBool: 5201f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao return llvm::Type::getInt1Ty(C); 521462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao break; 522462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 523462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao default: 524462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao assert(false && "Unknown data type"); 525462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao break; 526462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao } 527462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 528462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return NULL; 529462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao} 530462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 531462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao/****************************** RSExportPointerType ******************************/ 532462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 533462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liaoconst Type* RSExportPointerType::IntegerType = NULL; 534462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 535462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei LiaoRSExportPointerType* RSExportPointerType::Create(RSContext* Context, const PointerType* PT, const llvm::StringRef& TypeName) { 536462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao const Type* PointeeType = GET_POINTEE_TYPE(PT); 537462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao const RSExportType* PointeeET; 538462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 539462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao if(PointeeType->getTypeClass() != Type::Pointer) { 540462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao PointeeET = RSExportType::Create(Context, PointeeType); 541462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao } else { 542462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao /* Double or higher dimension of pointer, export as int* */ 543462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao assert(IntegerType != NULL && "Built-in integer type is not set"); 544462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao PointeeET = RSExportPrimitiveType::Create(Context, IntegerType); 545462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao } 546462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 547462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao if(PointeeET == NULL) { 548462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao printf("Failed to create type for pointee"); 549462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return NULL; 550462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao } 551462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 552462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return new RSExportPointerType(Context, TypeName, PointeeET); 553462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao} 554462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 5551f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei LiaoRSExportType::ExportClass RSExportPointerType::getClass() const { 556462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return RSExportType::ExportClassPointer; 557462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao} 558462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 559462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liaoconst llvm::Type* RSExportPointerType::convertToLLVMType() const { 560462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao const llvm::Type* PointeeType = mPointeeType->getLLVMType(); 561462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return llvm::PointerType::getUnqual(PointeeType); 562462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao} 563462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 564f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao/****************************** RSExportConstantArrayType ******************************/ 565f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liaollvm::StringRef RSExportConstantArrayType::GetTypeName(const ConstantArrayType* CT) { 566f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao llvm::APInt i = CT->getSize(); 567f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao if (i == 4) { 568f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao return llvm::StringRef("rs_matrix2x2"); 569f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao } else if (i == 9) { 570f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao return llvm::StringRef("rs_matrix3x3"); 571f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao } else if (i == 16) { 572f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao return llvm::StringRef("rs_matrix4x4"); 573f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao } 574f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao return llvm::StringRef(); 575f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao} 576f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao 577f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei LiaoRSExportConstantArrayType* RSExportConstantArrayType::Create(RSContext* Context, const ConstantArrayType* CT, const llvm::StringRef& TypeName, DataKind DK, bool Normalized) { 578f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao assert(CT != NULL && CT->getTypeClass() == Type::ConstantArray); 579f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao 580f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao const Type* ElementType = GET_CONSTANT_ARRAY_ELEMENT_TYPE(CT); 581f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao // RSExportPrimitiveType::DataType DT = RSExportPrimitiveType::GetDataType(ElementType); 582f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao int64_t siz = CT->getSize().getSExtValue(); 583f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao RSExportPrimitiveType::DataType DT; 584f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao if (siz == 4) { 585f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao DT = RSExportPrimitiveType::DataTypeRSMatrix2x2; 586f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao } else if (siz == 9) { 587f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao DT = RSExportPrimitiveType::DataTypeRSMatrix3x3; 588f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao } else if (siz == 16) { 589f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao DT = RSExportPrimitiveType::DataTypeRSMatrix4x4; 590f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao } else { 591f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao printf("RSExportConstantArrayType::Create : unsupported base element type\n"); 592f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao } 593f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao 594f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao return new RSExportConstantArrayType(Context, TypeName, DT, DK, Normalized, siz); 595f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao} 596f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao 597f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei LiaoRSExportType::ExportClass RSExportConstantArrayType::getClass() const { 598f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao return RSExportType::ExportClassConstantArray; 599f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao} 600f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao 601f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liaoconst llvm::Type* RSExportConstantArrayType::convertToLLVMType() const { 602f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao llvm::LLVMContext& C = getRSContext()->getLLVMContext(); 603f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao return llvm::ArrayType::get(llvm::Type::getFloatTy(C), mNumElement); 604f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao} 605f8149d9e5a3795e9952717ee6346789a134c55c7Shih-wei Liao 606462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao/****************************** RSExportVectorType ******************************/ 6071f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liaoconst char* RSExportVectorType::VectorTypeNameStore[][3] = { 608462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao /* 0 */ { "char2", "char3", "char4" }, 609462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao /* 1 */ { "uchar2", "uchar3", "uchar4" }, 610462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao /* 2 */ { "short2", "short3", "short4" }, 611462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao /* 3 */ { "ushort2", "ushort3", "ushort4" }, 612462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao /* 4 */ { "int2", "int3", "int4" }, 613462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao /* 5 */ { "uint2", "uint3", "uint4" }, 614462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao /* 6 */ { "float2", "float3", "float4" }, 6151f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao}; 616462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 617462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liaollvm::StringRef RSExportVectorType::GetTypeName(const ExtVectorType* EVT) { 618462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao const Type* ElementType = GET_EXT_VECTOR_ELEMENT_TYPE(EVT); 619462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 620462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao if((ElementType->getTypeClass() != Type::Builtin)) 621462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return llvm::StringRef(); 622462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 623462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao const BuiltinType* BT = UNSAFE_CAST_TYPE(BuiltinType, ElementType); 624462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao const char** BaseElement = NULL; 625462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 626462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao switch(BT->getKind()) { 627462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao#define SLANG_RS_SUPPORT_BUILTIN_TYPE(builtin_type, type) \ 628462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao case builtin_type: \ 629462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao /* Compiler is smart enough to optimize following *big if branches* since they all become "constant comparison" after macro expansion */ \ 630462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao if(type == RSExportPrimitiveType::DataTypeSigned8) BaseElement = VectorTypeNameStore[0]; \ 631462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao else if(type == RSExportPrimitiveType::DataTypeUnsigned8) BaseElement = VectorTypeNameStore[1]; \ 632462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao else if(type == RSExportPrimitiveType::DataTypeSigned16) BaseElement = VectorTypeNameStore[2]; \ 633462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao else if(type == RSExportPrimitiveType::DataTypeUnsigned16) BaseElement = VectorTypeNameStore[3]; \ 634462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao else if(type == RSExportPrimitiveType::DataTypeSigned32) BaseElement = VectorTypeNameStore[4]; \ 635462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao else if(type == RSExportPrimitiveType::DataTypeUnsigned32) BaseElement = VectorTypeNameStore[5]; \ 636462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao else if(type == RSExportPrimitiveType::DataTypeFloat32) BaseElement = VectorTypeNameStore[6]; \ 6371f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao else if(type == RSExportPrimitiveType::DataTypeBool) BaseElement = VectorTypeNameStore[0]; \ 638462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao break; 639462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao#include "slang_rs_export_type_support.inc" 640462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 641462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao default: return llvm::StringRef(); break; 642462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao } 643462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 644462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao if((BaseElement != NULL) && (EVT->getNumElements() > 1) && (EVT->getNumElements() <= 4)) 645462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return BaseElement[EVT->getNumElements() - 2]; 646462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao else 647462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return llvm::StringRef(); 648462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao} 649462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 650462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei LiaoRSExportVectorType* RSExportVectorType::Create(RSContext* Context, const ExtVectorType* EVT, const llvm::StringRef& TypeName, DataKind DK, bool Normalized) { 651462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao assert(EVT != NULL && EVT->getTypeClass() == Type::ExtVector); 652462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 653462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao const Type* ElementType = GET_EXT_VECTOR_ELEMENT_TYPE(EVT); 654462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao RSExportPrimitiveType::DataType DT = RSExportPrimitiveType::GetDataType(ElementType); 655462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 6561f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao if(DT != RSExportPrimitiveType::DataTypeUnknown) 657462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return new RSExportVectorType(Context, TypeName, DT, DK, Normalized, EVT->getNumElements()); 658462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao else 659462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao printf("RSExportVectorType::Create : unsupported base element type\n"); 660462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao} 661462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 6621f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei LiaoRSExportType::ExportClass RSExportVectorType::getClass() const { 6631f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao return RSExportType::ExportClassVector; 664462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao} 665462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 666462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liaoconst llvm::Type* RSExportVectorType::convertToLLVMType() const { 667462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao const llvm::Type* ElementType = RSExportPrimitiveType::convertToLLVMType(); 668462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return llvm::VectorType::get(ElementType, getNumElement()); 669462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao} 670462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 671462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao/****************************** RSExportRecordType ******************************/ 672462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei LiaoRSExportRecordType* RSExportRecordType::Create(RSContext* Context, const RecordType* RT, const llvm::StringRef& TypeName, bool mIsArtificial) { 673462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao assert(RT != NULL && RT->getTypeClass() == Type::Record); 674462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 675462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao const RecordDecl* RD = RT->getDecl(); 676462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao assert(RD->isStruct()); 677462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 678462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao RD = RD->getDefinition(); 679462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao if(RD == NULL) { 680462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao /* TODO: warning: actual struct definition isn't declared in this moudle */ 681462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao printf("RSExportRecordType::Create : this struct is not defined in this module."); 682462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return NULL; 683462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao } 684462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 685462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao RSExportRecordType* ERT = new RSExportRecordType(Context, TypeName, RD->hasAttr<PackedAttr>(), mIsArtificial); 686462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao unsigned int Index = 0; 687462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 688462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao for(RecordDecl::field_iterator fit = RD->field_begin(); 689462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao fit != RD->field_end(); 690462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao fit++, Index++) 691462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao { 692462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao#define FAILED_CREATE_FIELD(err) do { \ 693462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao if(*err) \ 694462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao printf("RSExportRecordType::Create : failed to create field (%s)\n", err); \ 695462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao delete ERT; \ 696462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return NULL; \ 697462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao } while(false) 698462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao /* FIXME: All fields should be primitive type */ 699462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao assert((*fit)->getKind() == Decl::Field); 700462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao FieldDecl* FD = *fit; 701462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 702462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao /* We don't support bit field TODO: allow bitfield with size 8, 16, 32 */ 703462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao if(FD->isBitField()) 704462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao FAILED_CREATE_FIELD("bit field is not supported"); 705462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 706462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao /* Type */ 707462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao RSExportType* ET = RSExportElement::CreateFromDecl(Context, FD); 708462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 709462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao if(ET != NULL) 710462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao ERT->mFields.push_back( new Field(ET, FD->getName(), ERT, Index) ); 711462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao else 712462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao FAILED_CREATE_FIELD(FD->getName().str().c_str()); 713462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao#undef FAILED_CREATE_FIELD 714462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao } 715462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 7160a3f20ec28ed6f5ae1ed5d61f6b6e3e577f7f5d1Shih-wei Liao 7170a3f20ec28ed6f5ae1ed5d61f6b6e3e577f7f5d1Shih-wei Liao const ASTRecordLayout &ASTRL = Context->getASTContext()->getASTRecordLayout(RD); 7180a3f20ec28ed6f5ae1ed5d61f6b6e3e577f7f5d1Shih-wei Liao ERT->AllocSize = (ASTRL.getSize() > ASTRL.getDataSize()) ? (ASTRL.getSize() >> 3) : (ASTRL.getDataSize() >> 3); 7190a3f20ec28ed6f5ae1ed5d61f6b6e3e577f7f5d1Shih-wei Liao 720462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return ERT; 721462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao} 722462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 723462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei LiaoRSExportType::ExportClass RSExportRecordType::getClass() const { 724462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return RSExportType::ExportClassRecord; 725462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao} 726462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 727462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liaoconst llvm::Type* RSExportRecordType::convertToLLVMType() const { 728462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao std::vector<const llvm::Type*> FieldTypes; 729462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 730462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao for(const_field_iterator FI = fields_begin(); 731462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao FI != fields_end(); 732462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao FI++) 733462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao { 734462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao const Field* F = *FI; 735462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao const RSExportType* FET = F->getType(); 7361f0d88fbff28e4e2dd563d93c8fe0047381c09ccShih-wei Liao 737462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao FieldTypes.push_back(FET->getLLVMType()); 738462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao } 739462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 740462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return llvm::StructType::get(getRSContext()->getLLVMContext(), FieldTypes, mIsPacked); 741462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao} 742462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 743462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao/****************************** RSExportRecordType::Field ******************************/ 744462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liaosize_t RSExportRecordType::Field::getOffsetInParent() const { 745462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao /* Struct layout obtains below will be cached by LLVM */ 746462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao const llvm::StructLayout* SL = mParent->getRSContext()->getTargetData()->getStructLayout(static_cast<const llvm::StructType*>(mParent->getLLVMType())); 747462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao return SL->getElementOffset(mIndex); 748462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao} 749462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 750462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao} /* namespace slang */ 751