1 /* 2 * Copyright 2015, The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17 #ifndef BCC_RS_UTILS_H 18 #define BCC_RS_UTILS_H 19 20 #include "rsDefines.h" 21 22 #include <llvm/IR/Type.h> 23 #include <llvm/IR/DerivedTypes.h> 24 25 namespace { 26 27 static inline llvm::StringRef getUnsuffixedStructName(const llvm::StructType *T) { 28 // Get just the object type name with no suffix. 29 size_t LastDot = T->getName().rfind('.'); 30 if (LastDot == strlen("struct")) { 31 // If we get back to just the "struct" part, we know that we had a 32 // raw typename (i.e. struct.rs_element with no ".[0-9]+" suffix on it. 33 // In that case, we will want to create our slice such that it contains 34 // the entire name. 35 LastDot = T->getName().size(); 36 } 37 return T->getStructName().slice(0, LastDot); 38 } 39 40 const char kAllocationTypeName[] = "struct.rs_allocation"; 41 const char kElementTypeName[] = "struct.rs_element"; 42 const char kSamplerTypeName[] = "struct.rs_sampler"; 43 const char kScriptTypeName[] = "struct.rs_script"; 44 const char kTypeTypeName[] = "struct.rs_type"; 45 46 // Returns the RsDataType for a given input LLVM type. 47 // This is only used to distinguish the associated RS object types (i.e. 48 // rs_allocation, rs_element, rs_sampler, rs_script, and rs_type). 49 // All other types are reported back as RS_TYPE_NONE, since no special 50 // handling would be necessary. 51 static inline enum RsDataType getRsDataTypeForType(const llvm::Type *T) { 52 if (T->isStructTy()) { 53 const llvm::StringRef StructName = getUnsuffixedStructName(llvm::dyn_cast<const llvm::StructType>(T)); 54 if (StructName.equals(kAllocationTypeName)) { 55 return RS_TYPE_ALLOCATION; 56 } else if (StructName.equals(kElementTypeName)) { 57 return RS_TYPE_ELEMENT; 58 } else if (StructName.equals(kSamplerTypeName)) { 59 return RS_TYPE_SAMPLER; 60 } else if (StructName.equals(kScriptTypeName)) { 61 return RS_TYPE_SCRIPT; 62 } else if (StructName.equals(kTypeTypeName)) { 63 return RS_TYPE_TYPE; 64 } 65 } 66 return RS_TYPE_NONE; 67 } 68 69 // Returns true if the input type is one of our RenderScript object types 70 // (allocation, element, sampler, script, type) and false if it is not. 71 static inline bool isRsObjectType(const llvm::Type *T) { 72 return getRsDataTypeForType(T) != RS_TYPE_NONE; 73 } 74 75 } // end namespace 76 77 #endif // BCC_RS_UTILS_H 78
