slang_rs_context.h revision cc1b9699446aea20773e4c3c6ff5759fedd8ab51
1/* 2 * Copyright 2010-2012, The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17#ifndef _FRAMEWORKS_COMPILE_SLANG_SLANG_RS_CONTEXT_H_ // NOLINT 18#define _FRAMEWORKS_COMPILE_SLANG_SLANG_RS_CONTEXT_H_ 19 20#include <cstdio> 21#include <list> 22#include <map> 23#include <string> 24 25#include "clang/Lex/Preprocessor.h" 26#include "clang/AST/Mangle.h" 27 28#include "llvm/ADT/OwningPtr.h" 29#include "llvm/ADT/StringSet.h" 30#include "llvm/ADT/StringMap.h" 31 32#include "slang_pragma_recorder.h" 33 34namespace llvm { 35 class LLVMContext; 36 class DataLayout; 37} // namespace llvm 38 39namespace clang { 40 class VarDecl; 41 class ASTContext; 42 class TargetInfo; 43 class FunctionDecl; 44 class SourceManager; 45} // namespace clang 46 47namespace slang { 48 class RSExportable; 49 class RSExportVar; 50 class RSExportFunc; 51 class RSExportForEach; 52 class RSExportType; 53 54class RSContext { 55 typedef llvm::StringSet<> NeedExportVarSet; 56 typedef llvm::StringSet<> NeedExportFuncSet; 57 typedef llvm::StringSet<> NeedExportTypeSet; 58 59 public: 60 typedef std::list<RSExportable*> ExportableList; 61 typedef std::list<RSExportVar*> ExportVarList; 62 typedef std::list<RSExportFunc*> ExportFuncList; 63 typedef std::list<RSExportForEach*> ExportForEachList; 64 typedef llvm::StringMap<RSExportType*> ExportTypeMap; 65 66 private: 67 clang::Preprocessor &mPP; 68 clang::ASTContext &mCtx; 69 PragmaList *mPragmas; 70 unsigned int mTargetAPI; 71 std::vector<std::string> *mGeneratedFileNames; 72 73 llvm::DataLayout *mDataLayout; 74 llvm::LLVMContext &mLLVMContext; 75 76 ExportableList mExportables; 77 78 NeedExportTypeSet mNeedExportTypes; 79 80 std::string *mLicenseNote; 81 std::string mReflectJavaPackageName; 82 std::string mReflectJavaPathName; 83 84 std::string mRSPackageName; 85 86 int version; 87 88 bool mIsCompatLib; 89 90 llvm::OwningPtr<clang::MangleContext> mMangleCtx; 91 92 bool processExportVar(const clang::VarDecl *VD); 93 bool processExportFunc(const clang::FunctionDecl *FD); 94 bool processExportType(const llvm::StringRef &Name); 95 96 void cleanupForEach(); 97 98 ExportVarList mExportVars; 99 ExportFuncList mExportFuncs; 100 ExportForEachList mExportForEach; 101 ExportTypeMap mExportTypes; 102 103 public: 104 RSContext(clang::Preprocessor &PP, 105 clang::ASTContext &Ctx, 106 const clang::TargetInfo &Target, 107 PragmaList *Pragmas, 108 unsigned int TargetAPI, 109 std::vector<std::string> *GeneratedFileNames); 110 111 inline clang::Preprocessor &getPreprocessor() const { return mPP; } 112 inline clang::ASTContext &getASTContext() const { return mCtx; } 113 inline clang::MangleContext &getMangleContext() const { 114 return *mMangleCtx; 115 } 116 inline const llvm::DataLayout *getDataLayout() const { return mDataLayout; } 117 inline llvm::LLVMContext &getLLVMContext() const { return mLLVMContext; } 118 inline const clang::SourceManager *getSourceManager() const { 119 return &mPP.getSourceManager(); 120 } 121 inline clang::DiagnosticsEngine *getDiagnostics() const { 122 return &mPP.getDiagnostics(); 123 } 124 inline unsigned int getTargetAPI() const { 125 return mTargetAPI; 126 } 127 128 inline void setLicenseNote(const std::string &S) { 129 mLicenseNote = new std::string(S); 130 } 131 inline const std::string *getLicenseNote() const { return mLicenseNote; } 132 133 inline void addExportType(const std::string &S) { 134 mNeedExportTypes.insert(S); 135 } 136 137 inline void setReflectJavaPackageName(const std::string &S) { 138 mReflectJavaPackageName = S; 139 } 140 inline const std::string &getReflectJavaPackageName() const { 141 return mReflectJavaPackageName; 142 } 143 144 inline void setRSPackageName(const std::string &S) { 145 mRSPackageName = S; 146 // If we are not targeting the actual Android Renderscript classes, 147 // we should reflect code that works with the compatibility library. 148 if (mRSPackageName.compare("android.renderscript") != 0) { 149 mIsCompatLib = true; 150 } 151 } 152 inline const std::string &getRSPackageName() const { 153 return mRSPackageName; 154 } 155 156 bool processExport(); 157 inline void newExportable(RSExportable *E) { 158 if (E != NULL) 159 mExportables.push_back(E); 160 } 161 typedef ExportableList::iterator exportable_iterator; 162 exportable_iterator exportable_begin() { 163 return mExportables.begin(); 164 } 165 exportable_iterator exportable_end() { 166 return mExportables.end(); 167 } 168 169 typedef ExportVarList::const_iterator const_export_var_iterator; 170 const_export_var_iterator export_vars_begin() const { 171 return mExportVars.begin(); 172 } 173 const_export_var_iterator export_vars_end() const { 174 return mExportVars.end(); 175 } 176 inline bool hasExportVar() const { 177 return !mExportVars.empty(); 178 } 179 180 typedef ExportFuncList::const_iterator const_export_func_iterator; 181 const_export_func_iterator export_funcs_begin() const { 182 return mExportFuncs.begin(); 183 } 184 const_export_func_iterator export_funcs_end() const { 185 return mExportFuncs.end(); 186 } 187 inline bool hasExportFunc() const { return !mExportFuncs.empty(); } 188 189 typedef ExportForEachList::const_iterator const_export_foreach_iterator; 190 const_export_foreach_iterator export_foreach_begin() const { 191 return mExportForEach.begin(); 192 } 193 const_export_foreach_iterator export_foreach_end() const { 194 return mExportForEach.end(); 195 } 196 inline bool hasExportForEach() const { return !mExportForEach.empty(); } 197 198 typedef ExportTypeMap::iterator export_type_iterator; 199 typedef ExportTypeMap::const_iterator const_export_type_iterator; 200 export_type_iterator export_types_begin() { return mExportTypes.begin(); } 201 export_type_iterator export_types_end() { return mExportTypes.end(); } 202 const_export_type_iterator export_types_begin() const { 203 return mExportTypes.begin(); 204 } 205 const_export_type_iterator export_types_end() const { 206 return mExportTypes.end(); 207 } 208 inline bool hasExportType() const { return !mExportTypes.empty(); } 209 export_type_iterator findExportType(const llvm::StringRef &TypeName) { 210 return mExportTypes.find(TypeName); 211 } 212 const_export_type_iterator findExportType(const llvm::StringRef &TypeName) 213 const { 214 return mExportTypes.find(TypeName); 215 } 216 217 // Insert the specified Typename/Type pair into the map. If the key already 218 // exists in the map, return false and ignore the request, otherwise insert it 219 // and return true. 220 bool insertExportType(const llvm::StringRef &TypeName, RSExportType *Type); 221 222 int getVersion() const { return version; } 223 void setVersion(int v) { 224 version = v; 225 } 226 227 bool isCompatLib() const { return mIsCompatLib; } 228 229 void addPragma(const std::string &T, const std::string &V) { 230 mPragmas->push_back(make_pair(T, V)); 231 } 232 233 // Report an error or a warning to the user. 234 template <unsigned N> 235 clang::DiagnosticBuilder Report(clang::DiagnosticsEngine::Level Level, 236 const char (&Message)[N]) { 237 clang::DiagnosticsEngine *DiagEngine = getDiagnostics(); 238 return DiagEngine->Report(DiagEngine->getCustomDiagID(Level, Message)); 239} 240 241 template <unsigned N> 242 clang::DiagnosticBuilder Report(clang::DiagnosticsEngine::Level Level, 243 const clang::SourceLocation Loc, 244 const char (&Message)[N]) { 245 clang::DiagnosticsEngine *DiagEngine = getDiagnostics(); 246 const clang::SourceManager *SM = getSourceManager(); 247 return DiagEngine->Report(clang::FullSourceLoc(Loc, *SM), 248 DiagEngine->getCustomDiagID(Level, Message)); 249} 250 251 // Utility functions to report errors and warnings to make the calling code 252 // easier to read. 253 template <unsigned N> 254 clang::DiagnosticBuilder ReportError(const char (&Message)[N]) { 255 return Report<N>(clang::DiagnosticsEngine::Error, Message); 256 } 257 258 template <unsigned N> 259 clang::DiagnosticBuilder ReportError(const clang::SourceLocation Loc, 260 const char (&Message)[N]) { 261 return Report<N>(clang::DiagnosticsEngine::Error, Loc, Message); 262 } 263 264 template <unsigned N> 265 clang::DiagnosticBuilder ReportWarning(const char (&Message)[N]) { 266 return Report<N>(clang::DiagnosticsEngine::Warning, Message); 267 } 268 269 template <unsigned N> 270 clang::DiagnosticBuilder ReportWarning(const clang::SourceLocation Loc, 271 const char (&Message)[N]) { 272 return Report<N>(clang::DiagnosticsEngine::Warning, Loc, Message); 273 } 274 275 ~RSContext(); 276}; 277 278} // namespace slang 279 280#endif // _FRAMEWORKS_COMPILE_SLANG_SLANG_RS_CONTEXT_H_ NOLINT 281