Zonr Chang | c383a50 | 2010-10-12 01:52:08 +0800 | [diff] [blame] | 1 | /* |
Stephen Hines | 9999ec3 | 2012-02-10 18:22:14 -0800 | [diff] [blame] | 2 | * Copyright 2010-2012, The Android Open Source Project |
Zonr Chang | c383a50 | 2010-10-12 01:52:08 +0800 | [diff] [blame] | 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 | |
zonr | 6315f76 | 2010-10-05 15:35:14 +0800 | [diff] [blame] | 17 | #include "slang_rs_context.h" |
| 18 | |
Stephen Hines | e639eb5 | 2010-11-08 19:27:20 -0800 | [diff] [blame] | 19 | #include <string> |
| 20 | |
| 21 | #include "clang/AST/ASTContext.h" |
| 22 | #include "clang/AST/Decl.h" |
| 23 | #include "clang/AST/DeclBase.h" |
Logan | be27482 | 2011-02-16 22:02:54 +0800 | [diff] [blame] | 24 | #include "clang/AST/Mangle.h" |
Stephen Hines | e639eb5 | 2010-11-08 19:27:20 -0800 | [diff] [blame] | 25 | #include "clang/AST/Type.h" |
| 26 | |
| 27 | #include "clang/Basic/Linkage.h" |
| 28 | #include "clang/Basic/TargetInfo.h" |
| 29 | |
zonr | 6315f76 | 2010-10-05 15:35:14 +0800 | [diff] [blame] | 30 | #include "llvm/LLVMContext.h" |
| 31 | #include "llvm/Target/TargetData.h" |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 32 | |
zonr | 6315f76 | 2010-10-05 15:35:14 +0800 | [diff] [blame] | 33 | #include "slang.h" |
Stephen Hines | 6e6578a | 2011-02-07 18:05:48 -0800 | [diff] [blame] | 34 | #include "slang_assert.h" |
Stephen Hines | 593a894 | 2011-05-10 15:29:50 -0700 | [diff] [blame] | 35 | #include "slang_rs_export_foreach.h" |
zonr | 6315f76 | 2010-10-05 15:35:14 +0800 | [diff] [blame] | 36 | #include "slang_rs_export_func.h" |
| 37 | #include "slang_rs_export_type.h" |
Stephen Hines | e639eb5 | 2010-11-08 19:27:20 -0800 | [diff] [blame] | 38 | #include "slang_rs_export_var.h" |
| 39 | #include "slang_rs_exportable.h" |
zonr | 6315f76 | 2010-10-05 15:35:14 +0800 | [diff] [blame] | 40 | #include "slang_rs_pragma_handler.h" |
Stephen Hines | e639eb5 | 2010-11-08 19:27:20 -0800 | [diff] [blame] | 41 | #include "slang_rs_reflection.h" |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 42 | |
Stephen Hines | e639eb5 | 2010-11-08 19:27:20 -0800 | [diff] [blame] | 43 | namespace slang { |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 44 | |
Stephen Hines | 9e5b503 | 2010-11-03 13:19:14 -0700 | [diff] [blame] | 45 | RSContext::RSContext(clang::Preprocessor &PP, |
| 46 | clang::ASTContext &Ctx, |
Stephen Hines | 3fd0a94 | 2011-01-18 12:27:39 -0800 | [diff] [blame] | 47 | const clang::TargetInfo &Target, |
Stephen Hines | 4cc67fc | 2011-01-31 16:48:57 -0800 | [diff] [blame] | 48 | PragmaList *Pragmas, |
Stephen Hines | 4a4bf92 | 2011-08-18 17:20:33 -0700 | [diff] [blame] | 49 | unsigned int TargetAPI, |
Stephen Hines | 4cc67fc | 2011-01-31 16:48:57 -0800 | [diff] [blame] | 50 | std::vector<std::string> *GeneratedFileNames) |
zonr | 6315f76 | 2010-10-05 15:35:14 +0800 | [diff] [blame] | 51 | : mPP(PP), |
| 52 | mCtx(Ctx), |
| 53 | mTarget(Target), |
Stephen Hines | 3fd0a94 | 2011-01-18 12:27:39 -0800 | [diff] [blame] | 54 | mPragmas(Pragmas), |
Stephen Hines | 4a4bf92 | 2011-08-18 17:20:33 -0700 | [diff] [blame] | 55 | mTargetAPI(TargetAPI), |
Stephen Hines | 4cc67fc | 2011-01-31 16:48:57 -0800 | [diff] [blame] | 56 | mGeneratedFileNames(GeneratedFileNames), |
zonr | 6315f76 | 2010-10-05 15:35:14 +0800 | [diff] [blame] | 57 | mTargetData(NULL), |
| 58 | mLLVMContext(llvm::getGlobalContext()), |
Stephen Hines | 96ab06c | 2011-01-05 15:29:26 -0800 | [diff] [blame] | 59 | mLicenseNote(NULL), |
Shih-wei Liao | 3fa286b | 2011-02-10 11:04:44 -0800 | [diff] [blame] | 60 | version(0), |
Logan | be27482 | 2011-02-16 22:02:54 +0800 | [diff] [blame] | 61 | mMangleCtx(Ctx.createMangleContext()) { |
Stephen Hines | 6e6578a | 2011-02-07 18:05:48 -0800 | [diff] [blame] | 62 | slangAssert(mGeneratedFileNames && "Must supply GeneratedFileNames"); |
Stephen Hines | 4cc67fc | 2011-01-31 16:48:57 -0800 | [diff] [blame] | 63 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 64 | // For #pragma rs export_type |
Stephen Hines | 9e5b503 | 2010-11-03 13:19:14 -0700 | [diff] [blame] | 65 | PP.AddPragmaHandler( |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 66 | "rs", RSPragmaHandler::CreatePragmaExportTypeHandler(this)); |
Victor Hsieh | d8a0d18 | 2010-07-07 19:22:33 +0800 | [diff] [blame] | 67 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 68 | // For #pragma rs java_package_name |
Stephen Hines | 9e5b503 | 2010-11-03 13:19:14 -0700 | [diff] [blame] | 69 | PP.AddPragmaHandler( |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 70 | "rs", RSPragmaHandler::CreatePragmaJavaPackageNameHandler(this)); |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 71 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 72 | // For #pragma rs set_reflect_license |
Stephen Hines | 9e5b503 | 2010-11-03 13:19:14 -0700 | [diff] [blame] | 73 | PP.AddPragmaHandler( |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 74 | "rs", RSPragmaHandler::CreatePragmaReflectLicenseHandler(this)); |
| 75 | |
Stephen Hines | 96ab06c | 2011-01-05 15:29:26 -0800 | [diff] [blame] | 76 | // For #pragma version |
| 77 | PP.AddPragmaHandler(RSPragmaHandler::CreatePragmaVersionHandler(this)); |
| 78 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 79 | // Prepare target data |
Stephen Hines | 9e5b503 | 2010-11-03 13:19:14 -0700 | [diff] [blame] | 80 | mTargetData = new llvm::TargetData(Target.getTargetDescription()); |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 81 | |
| 82 | return; |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 83 | } |
| 84 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 85 | bool RSContext::processExportVar(const clang::VarDecl *VD) { |
Stephen Hines | 6e6578a | 2011-02-07 18:05:48 -0800 | [diff] [blame] | 86 | slangAssert(!VD->getName().empty() && "Variable name should not be empty"); |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 87 | |
zonr | 6315f76 | 2010-10-05 15:35:14 +0800 | [diff] [blame] | 88 | // TODO(zonr): some check on variable |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 89 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 90 | RSExportType *ET = RSExportType::CreateFromDecl(this, VD); |
| 91 | if (!ET) |
| 92 | return false; |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 93 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 94 | RSExportVar *EV = new RSExportVar(this, VD, ET); |
| 95 | if (EV == NULL) |
| 96 | return false; |
| 97 | else |
| 98 | mExportVars.push_back(EV); |
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 | return true; |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 101 | } |
| 102 | |
zonr | 6315f76 | 2010-10-05 15:35:14 +0800 | [diff] [blame] | 103 | bool RSContext::processExportFunc(const clang::FunctionDecl *FD) { |
Stephen Hines | 6e6578a | 2011-02-07 18:05:48 -0800 | [diff] [blame] | 104 | slangAssert(!FD->getName().empty() && "Function name should not be empty"); |
Shih-wei Liao | 537446c | 2010-06-11 16:05:55 -0700 | [diff] [blame] | 105 | |
Stephen Hines | 3fbe68a | 2010-11-17 17:28:59 -0800 | [diff] [blame] | 106 | if (!FD->isThisDeclarationADefinition()) { |
| 107 | return true; |
| 108 | } |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 109 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 110 | if (FD->getStorageClass() != clang::SC_None) { |
| 111 | fprintf(stderr, "RSContext::processExportFunc : cannot export extern or " |
| 112 | "static function '%s'\n", FD->getName().str().c_str()); |
| 113 | return false; |
| 114 | } |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 115 | |
Stephen Hines | 9999ec3 | 2012-02-10 18:22:14 -0800 | [diff] [blame] | 116 | if (RSExportForEach::isSpecialRSFunc(mTargetAPI, FD)) { |
| 117 | // Do not reflect specialized functions like init, dtor, or graphics root. |
| 118 | return RSExportForEach::validateSpecialFuncDecl(mTargetAPI, |
| 119 | getDiagnostics(), FD); |
| 120 | } else if (RSExportForEach::isRSForEachFunc(mTargetAPI, FD)) { |
Stephen Hines | 593a894 | 2011-05-10 15:29:50 -0700 | [diff] [blame] | 121 | RSExportForEach *EFE = RSExportForEach::Create(this, FD); |
| 122 | if (EFE == NULL) |
| 123 | return false; |
| 124 | else |
| 125 | mExportForEach.push_back(EFE); |
| 126 | return true; |
Stephen Hines | 3fbe68a | 2010-11-17 17:28:59 -0800 | [diff] [blame] | 127 | } |
| 128 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 129 | RSExportFunc *EF = RSExportFunc::Create(this, FD); |
| 130 | if (EF == NULL) |
| 131 | return false; |
| 132 | else |
| 133 | mExportFuncs.push_back(EF); |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 134 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 135 | return true; |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 136 | } |
| 137 | |
| 138 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 139 | bool RSContext::processExportType(const llvm::StringRef &Name) { |
Stephen Hines | 9e5b503 | 2010-11-03 13:19:14 -0700 | [diff] [blame] | 140 | clang::TranslationUnitDecl *TUDecl = mCtx.getTranslationUnitDecl(); |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 141 | |
Stephen Hines | 6e6578a | 2011-02-07 18:05:48 -0800 | [diff] [blame] | 142 | slangAssert(TUDecl != NULL && "Translation unit declaration (top-level " |
| 143 | "declaration) is null object"); |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 144 | |
Stephen Hines | 9e5b503 | 2010-11-03 13:19:14 -0700 | [diff] [blame] | 145 | const clang::IdentifierInfo *II = mPP.getIdentifierInfo(Name); |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 146 | if (II == NULL) |
zonr | 6315f76 | 2010-10-05 15:35:14 +0800 | [diff] [blame] | 147 | // TODO(zonr): alert identifier @Name mark as an exportable type cannot be |
| 148 | // found |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 149 | return false; |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 150 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 151 | clang::DeclContext::lookup_const_result R = TUDecl->lookup(II); |
| 152 | RSExportType *ET = NULL; |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 153 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 154 | for (clang::DeclContext::lookup_const_iterator I = R.first, E = R.second; |
| 155 | I != E; |
| 156 | I++) { |
| 157 | clang::NamedDecl *const ND = *I; |
| 158 | const clang::Type *T = NULL; |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 159 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 160 | switch (ND->getKind()) { |
| 161 | case clang::Decl::Typedef: { |
| 162 | T = static_cast<const clang::TypedefDecl*>( |
| 163 | ND)->getCanonicalDecl()->getUnderlyingType().getTypePtr(); |
| 164 | break; |
| 165 | } |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 166 | case clang::Decl::Record: { |
| 167 | T = static_cast<const clang::RecordDecl*>(ND)->getTypeForDecl(); |
| 168 | break; |
| 169 | } |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 170 | default: { |
| 171 | // unsupported, skip |
| 172 | break; |
Shih-wei Liao | ac91815 | 2010-08-29 02:43:24 -0700 | [diff] [blame] | 173 | } |
| 174 | } |
Shih-wei Liao | 537446c | 2010-06-11 16:05:55 -0700 | [diff] [blame] | 175 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 176 | if (T != NULL) |
| 177 | ET = RSExportType::Create(this, T); |
| 178 | } |
| 179 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 180 | return (ET != NULL); |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 181 | } |
| 182 | |
Stephen Hines | c17e198 | 2012-02-22 12:30:45 -0800 | [diff] [blame] | 183 | |
| 184 | // Possibly re-order ForEach exports (maybe generating a dummy "root" function). |
| 185 | // We require "root" to be listed as slot 0 of our exported compute kernels, |
| 186 | // so this only needs to be created if we have other non-root kernels. |
| 187 | void RSContext::cleanupForEach() { |
| 188 | bool foundNonRoot = false; |
| 189 | ExportForEachList::iterator begin = mExportForEach.begin(); |
| 190 | |
| 191 | for (ExportForEachList::iterator I = begin, E = mExportForEach.end(); |
| 192 | I != E; |
| 193 | I++) { |
| 194 | RSExportForEach *EFE = *I; |
| 195 | if (!EFE->getName().compare("root")) { |
| 196 | if (I == begin) { |
| 197 | // Nothing to do, since it is the first function |
| 198 | return; |
| 199 | } |
| 200 | |
| 201 | mExportForEach.erase(I); |
| 202 | mExportForEach.push_front(EFE); |
| 203 | return; |
| 204 | } else { |
| 205 | foundNonRoot = true; |
| 206 | } |
| 207 | } |
| 208 | |
| 209 | // If we found a non-root kernel, but no root() function, we need to add a |
| 210 | // dummy version (so that script->script calls of rsForEach don't behave |
| 211 | // erratically). |
| 212 | if (foundNonRoot) { |
| 213 | RSExportForEach *DummyRoot = RSExportForEach::CreateDummyRoot(this); |
| 214 | mExportForEach.push_front(DummyRoot); |
| 215 | } |
| 216 | } |
| 217 | |
| 218 | |
Stephen Hines | c808a99 | 2010-11-29 17:20:42 -0800 | [diff] [blame] | 219 | bool RSContext::processExport() { |
Stephen Hines | b6809ed | 2011-05-09 13:05:04 -0700 | [diff] [blame] | 220 | bool valid = true; |
Stephen Hines | 593a894 | 2011-05-10 15:29:50 -0700 | [diff] [blame] | 221 | |
| 222 | if (getDiagnostics()->hasErrorOccurred()) { |
| 223 | return false; |
| 224 | } |
| 225 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 226 | // Export variable |
Stephen Hines | 9e5b503 | 2010-11-03 13:19:14 -0700 | [diff] [blame] | 227 | clang::TranslationUnitDecl *TUDecl = mCtx.getTranslationUnitDecl(); |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 228 | for (clang::DeclContext::decl_iterator DI = TUDecl->decls_begin(), |
| 229 | DE = TUDecl->decls_end(); |
Stephen Hines | c97a333 | 2010-11-30 15:31:08 -0800 | [diff] [blame] | 230 | DI != DE; |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 231 | DI++) { |
| 232 | if (DI->getKind() == clang::Decl::Var) { |
| 233 | clang::VarDecl *VD = (clang::VarDecl*) (*DI); |
Stephen Hines | 2d2512c | 2010-11-18 18:54:21 -0800 | [diff] [blame] | 234 | if (VD->getLinkage() == clang::ExternalLinkage) { |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 235 | if (!processExportVar(VD)) { |
Stephen Hines | b6809ed | 2011-05-09 13:05:04 -0700 | [diff] [blame] | 236 | valid = false; |
Ying Wang | 3f8b44d | 2010-09-04 01:17:01 -0700 | [diff] [blame] | 237 | } |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 238 | } |
| 239 | } else if (DI->getKind() == clang::Decl::Function) { |
| 240 | // Export functions |
| 241 | clang::FunctionDecl *FD = (clang::FunctionDecl*) (*DI); |
Stephen Hines | 2d2512c | 2010-11-18 18:54:21 -0800 | [diff] [blame] | 242 | if (FD->getLinkage() == clang::ExternalLinkage) { |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 243 | if (!processExportFunc(FD)) { |
Stephen Hines | b6809ed | 2011-05-09 13:05:04 -0700 | [diff] [blame] | 244 | valid = false; |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 245 | } |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 246 | } |
Ying Wang | 3f8b44d | 2010-09-04 01:17:01 -0700 | [diff] [blame] | 247 | } |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 248 | } |
Ying Wang | 3f8b44d | 2010-09-04 01:17:01 -0700 | [diff] [blame] | 249 | |
Stephen Hines | c17e198 | 2012-02-22 12:30:45 -0800 | [diff] [blame] | 250 | if (valid) { |
| 251 | cleanupForEach(); |
| 252 | } |
| 253 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 254 | // Finally, export type forcely set to be exported by user |
| 255 | for (NeedExportTypeSet::const_iterator EI = mNeedExportTypes.begin(), |
| 256 | EE = mNeedExportTypes.end(); |
| 257 | EI != EE; |
| 258 | EI++) { |
| 259 | if (!processExportType(EI->getKey())) { |
Stephen Hines | b6809ed | 2011-05-09 13:05:04 -0700 | [diff] [blame] | 260 | valid = false; |
Ying Wang | 3f8b44d | 2010-09-04 01:17:01 -0700 | [diff] [blame] | 261 | } |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 262 | } |
Shih-wei Liao | 537446c | 2010-06-11 16:05:55 -0700 | [diff] [blame] | 263 | |
Stephen Hines | b6809ed | 2011-05-09 13:05:04 -0700 | [diff] [blame] | 264 | return valid; |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 265 | } |
| 266 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 267 | bool RSContext::insertExportType(const llvm::StringRef &TypeName, |
| 268 | RSExportType *ET) { |
| 269 | ExportTypeMap::value_type *NewItem = |
| 270 | ExportTypeMap::value_type::Create(TypeName.begin(), |
| 271 | TypeName.end(), |
| 272 | mExportTypes.getAllocator(), |
| 273 | ET); |
Shih-wei Liao | 6de8927 | 2010-07-15 15:26:20 -0700 | [diff] [blame] | 274 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 275 | if (mExportTypes.insert(NewItem)) { |
Shih-wei Liao | 6de8927 | 2010-07-15 15:26:20 -0700 | [diff] [blame] | 276 | return true; |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 277 | } else { |
| 278 | free(NewItem); |
| 279 | return false; |
| 280 | } |
| 281 | } |
| 282 | |
Shih-wei Liao | b81c6a4 | 2010-10-10 14:15:00 -0700 | [diff] [blame] | 283 | bool RSContext::reflectToJava(const std::string &OutputPathBase, |
| 284 | const std::string &OutputPackageName, |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 285 | const std::string &InputFileName, |
| 286 | const std::string &OutputBCFileName, |
Shih-wei Liao | b81c6a4 | 2010-10-10 14:15:00 -0700 | [diff] [blame] | 287 | std::string *RealPackageName) { |
| 288 | if (RealPackageName != NULL) |
| 289 | RealPackageName->clear(); |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 290 | |
Shih-wei Liao | b81c6a4 | 2010-10-10 14:15:00 -0700 | [diff] [blame] | 291 | const std::string &PackageName = |
| 292 | ((OutputPackageName.empty()) ? mReflectJavaPackageName : |
| 293 | OutputPackageName); |
Stephen Hines | 35f5b39 | 2010-11-22 16:39:16 -0800 | [diff] [blame] | 294 | if (PackageName.empty()) { |
Stephen Hines | 7aff4a0 | 2011-12-08 18:34:27 -0800 | [diff] [blame] | 295 | clang::DiagnosticsEngine *DiagEngine = getDiagnostics(); |
| 296 | const clang::SourceManager *SM = getSourceManager(); |
| 297 | DiagEngine->Report( |
| 298 | SM->getLocForEndOfFile(SM->getMainFileID()), |
| 299 | DiagEngine->getCustomDiagID(clang::DiagnosticsEngine::Error, |
| 300 | "missing \"#pragma rs " |
| 301 | "java_package_name(com.foo.bar)\" " |
| 302 | "in source file")); |
Stephen Hines | 35f5b39 | 2010-11-22 16:39:16 -0800 | [diff] [blame] | 303 | return false; |
| 304 | } |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 305 | |
| 306 | // Copy back the really applied package name |
Shih-wei Liao | b81c6a4 | 2010-10-10 14:15:00 -0700 | [diff] [blame] | 307 | RealPackageName->assign(PackageName); |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 308 | |
Stephen Hines | 4cc67fc | 2011-01-31 16:48:57 -0800 | [diff] [blame] | 309 | RSReflection *R = new RSReflection(this, mGeneratedFileNames); |
Shih-wei Liao | b81c6a4 | 2010-10-10 14:15:00 -0700 | [diff] [blame] | 310 | bool ret = R->reflect(OutputPathBase, PackageName, |
| 311 | InputFileName, OutputBCFileName); |
zonr | 6315f76 | 2010-10-05 15:35:14 +0800 | [diff] [blame] | 312 | if (!ret) |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 313 | fprintf(stderr, "RSContext::reflectToJava : failed to do reflection " |
| 314 | "(%s)\n", R->getLastError()); |
| 315 | delete R; |
| 316 | return ret; |
| 317 | } |
| 318 | |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 319 | RSContext::~RSContext() { |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 320 | delete mLicenseNote; |
| 321 | delete mTargetData; |
Zonr Chang | a41ce1d | 2010-10-06 02:23:12 +0800 | [diff] [blame] | 322 | for (ExportableList::iterator I = mExportables.begin(), |
| 323 | E = mExportables.end(); |
| 324 | I != E; |
| 325 | I++) { |
Zonr Chang | 641558f | 2010-10-12 21:07:06 +0800 | [diff] [blame] | 326 | if (!(*I)->isKeep()) |
| 327 | delete *I; |
Zonr Chang | a41ce1d | 2010-10-06 02:23:12 +0800 | [diff] [blame] | 328 | } |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 329 | } |
Stephen Hines | e639eb5 | 2010-11-08 19:27:20 -0800 | [diff] [blame] | 330 | |
| 331 | } // namespace slang |