slang_rs_context.cpp revision 0a813a3ef2a82f19d7eab9e23ae8493197143803
1c383a500aa59423264811be3874461bf8adbfea0Zonr Chang/* 29999ec3aa0c4d7a6befd3a300dc07f0cea91cb6cStephen Hines * Copyright 2010-2012, The Android Open Source Project 3c383a500aa59423264811be3874461bf8adbfea0Zonr Chang * 4c383a500aa59423264811be3874461bf8adbfea0Zonr Chang * Licensed under the Apache License, Version 2.0 (the "License"); 5c383a500aa59423264811be3874461bf8adbfea0Zonr Chang * you may not use this file except in compliance with the License. 6c383a500aa59423264811be3874461bf8adbfea0Zonr Chang * You may obtain a copy of the License at 7c383a500aa59423264811be3874461bf8adbfea0Zonr Chang * 8c383a500aa59423264811be3874461bf8adbfea0Zonr Chang * http://www.apache.org/licenses/LICENSE-2.0 9c383a500aa59423264811be3874461bf8adbfea0Zonr Chang * 10c383a500aa59423264811be3874461bf8adbfea0Zonr Chang * Unless required by applicable law or agreed to in writing, software 11c383a500aa59423264811be3874461bf8adbfea0Zonr Chang * distributed under the License is distributed on an "AS IS" BASIS, 12c383a500aa59423264811be3874461bf8adbfea0Zonr Chang * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13c383a500aa59423264811be3874461bf8adbfea0Zonr Chang * See the License for the specific language governing permissions and 14c383a500aa59423264811be3874461bf8adbfea0Zonr Chang * limitations under the License. 15c383a500aa59423264811be3874461bf8adbfea0Zonr Chang */ 16c383a500aa59423264811be3874461bf8adbfea0Zonr Chang 176315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr#include "slang_rs_context.h" 186315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr 19e639eb5caa2c386b4a60659a4929e8a6141a2cbeStephen Hines#include <string> 2041ebf534161bb67f6207a070c1f6a895dc853408Zonr Chang 21e639eb5caa2c386b4a60659a4929e8a6141a2cbeStephen Hines#include "clang/AST/ASTContext.h" 229ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao#include "clang/AST/Decl.h" 239ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao#include "clang/AST/DeclBase.h" 24be27482cdeaf08576bc39b72a15d35d13014a636Logan#include "clang/AST/Mangle.h" 25e639eb5caa2c386b4a60659a4929e8a6141a2cbeStephen Hines#include "clang/AST/Type.h" 26e639eb5caa2c386b4a60659a4929e8a6141a2cbeStephen Hines 27e639eb5caa2c386b4a60659a4929e8a6141a2cbeStephen Hines#include "clang/Basic/Linkage.h" 28e639eb5caa2c386b4a60659a4929e8a6141a2cbeStephen Hines#include "clang/Basic/TargetInfo.h" 2941ebf534161bb67f6207a070c1f6a895dc853408Zonr Chang 30e639eb5caa2c386b4a60659a4929e8a6141a2cbeStephen Hines#include "llvm/LLVMContext.h" 31e639eb5caa2c386b4a60659a4929e8a6141a2cbeStephen Hines#include "llvm/Target/TargetData.h" 32e639eb5caa2c386b4a60659a4929e8a6141a2cbeStephen Hines 336315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr#include "slang.h" 346e6578a360497f78a181e63d7783422a9c9bfb15Stephen Hines#include "slang_assert.h" 35593a894650e81be54173106ec266f0311cebebd3Stephen Hines#include "slang_rs_export_foreach.h" 366315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr#include "slang_rs_export_func.h" 376315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr#include "slang_rs_export_type.h" 38e639eb5caa2c386b4a60659a4929e8a6141a2cbeStephen Hines#include "slang_rs_export_var.h" 39e639eb5caa2c386b4a60659a4929e8a6141a2cbeStephen Hines#include "slang_rs_exportable.h" 406315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr#include "slang_rs_pragma_handler.h" 41e639eb5caa2c386b4a60659a4929e8a6141a2cbeStephen Hines#include "slang_rs_reflection.h" 429ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao 43e639eb5caa2c386b4a60659a4929e8a6141a2cbeStephen Hinesnamespace slang { 449ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao 459e5b503349719144f63ccb7c62ee9c291a7d83b8Stephen HinesRSContext::RSContext(clang::Preprocessor &PP, 469e5b503349719144f63ccb7c62ee9c291a7d83b8Stephen Hines clang::ASTContext &Ctx, 473fd0a94a5cf1656569b1aea07043cc63939dcb46Stephen Hines const clang::TargetInfo &Target, 484cc67fce91f43215d61b2695746eab102a3db516Stephen Hines PragmaList *Pragmas, 494a4bf92a8add68629a7e6e59ef81c3c3fe603a75Stephen Hines unsigned int TargetAPI, 504cc67fce91f43215d61b2695746eab102a3db516Stephen Hines std::vector<std::string> *GeneratedFileNames) 516315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr : mPP(PP), 526315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr mCtx(Ctx), 536315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr mTarget(Target), 543fd0a94a5cf1656569b1aea07043cc63939dcb46Stephen Hines mPragmas(Pragmas), 554a4bf92a8add68629a7e6e59ef81c3c3fe603a75Stephen Hines mTargetAPI(TargetAPI), 564cc67fce91f43215d61b2695746eab102a3db516Stephen Hines mGeneratedFileNames(GeneratedFileNames), 576315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr mTargetData(NULL), 586315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr mLLVMContext(llvm::getGlobalContext()), 5996ab06cbe40b2d73c0eb614f814cd761d8962b6bStephen Hines mLicenseNote(NULL), 600a813a3ef2a82f19d7eab9e23ae8493197143803Stephen Hines mRSPackageName("android.renderscript"), 613fa286b4c2f110c6be2bbfac9c715bb1ec880338Shih-wei Liao version(0), 62be27482cdeaf08576bc39b72a15d35d13014a636Logan mMangleCtx(Ctx.createMangleContext()) { 636e6578a360497f78a181e63d7783422a9c9bfb15Stephen Hines slangAssert(mGeneratedFileNames && "Must supply GeneratedFileNames"); 644cc67fce91f43215d61b2695746eab102a3db516Stephen Hines 659ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao // For #pragma rs export_type 669e5b503349719144f63ccb7c62ee9c291a7d83b8Stephen Hines PP.AddPragmaHandler( 679ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao "rs", RSPragmaHandler::CreatePragmaExportTypeHandler(this)); 68462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 699ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao // For #pragma rs java_package_name 709e5b503349719144f63ccb7c62ee9c291a7d83b8Stephen Hines PP.AddPragmaHandler( 719ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao "rs", RSPragmaHandler::CreatePragmaJavaPackageNameHandler(this)); 729ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao 739ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao // For #pragma rs set_reflect_license 749e5b503349719144f63ccb7c62ee9c291a7d83b8Stephen Hines PP.AddPragmaHandler( 759ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao "rs", RSPragmaHandler::CreatePragmaReflectLicenseHandler(this)); 769ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao 7796ab06cbe40b2d73c0eb614f814cd761d8962b6bStephen Hines // For #pragma version 7896ab06cbe40b2d73c0eb614f814cd761d8962b6bStephen Hines PP.AddPragmaHandler(RSPragmaHandler::CreatePragmaVersionHandler(this)); 7996ab06cbe40b2d73c0eb614f814cd761d8962b6bStephen Hines 809ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao // Prepare target data 819e5b503349719144f63ccb7c62ee9c291a7d83b8Stephen Hines mTargetData = new llvm::TargetData(Target.getTargetDescription()); 829ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao 839ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao return; 84462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao} 85462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 869ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liaobool RSContext::processExportVar(const clang::VarDecl *VD) { 876e6578a360497f78a181e63d7783422a9c9bfb15Stephen Hines slangAssert(!VD->getName().empty() && "Variable name should not be empty"); 88537446c9542fdda458920876ed8a020a97ddf7a4Shih-wei Liao 896315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr // TODO(zonr): some check on variable 90462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 919ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao RSExportType *ET = RSExportType::CreateFromDecl(this, VD); 929ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao if (!ET) 939ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao return false; 94462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 959ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao RSExportVar *EV = new RSExportVar(this, VD, ET); 969ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao if (EV == NULL) 979ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao return false; 989ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao else 999ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao mExportVars.push_back(EV); 100462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 1019ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao return true; 102462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao} 103462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 1046315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonrbool RSContext::processExportFunc(const clang::FunctionDecl *FD) { 1056e6578a360497f78a181e63d7783422a9c9bfb15Stephen Hines slangAssert(!FD->getName().empty() && "Function name should not be empty"); 106462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 1073fbe68a4ec20fec25f8a40191437bbc02d00f591Stephen Hines if (!FD->isThisDeclarationADefinition()) { 1083fbe68a4ec20fec25f8a40191437bbc02d00f591Stephen Hines return true; 1093fbe68a4ec20fec25f8a40191437bbc02d00f591Stephen Hines } 110462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 1119ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao if (FD->getStorageClass() != clang::SC_None) { 1129ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao fprintf(stderr, "RSContext::processExportFunc : cannot export extern or " 1139ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao "static function '%s'\n", FD->getName().str().c_str()); 1149ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao return false; 1159ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao } 116462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 1179999ec3aa0c4d7a6befd3a300dc07f0cea91cb6cStephen Hines if (RSExportForEach::isSpecialRSFunc(mTargetAPI, FD)) { 1189999ec3aa0c4d7a6befd3a300dc07f0cea91cb6cStephen Hines // Do not reflect specialized functions like init, dtor, or graphics root. 1199999ec3aa0c4d7a6befd3a300dc07f0cea91cb6cStephen Hines return RSExportForEach::validateSpecialFuncDecl(mTargetAPI, 1209999ec3aa0c4d7a6befd3a300dc07f0cea91cb6cStephen Hines getDiagnostics(), FD); 1219999ec3aa0c4d7a6befd3a300dc07f0cea91cb6cStephen Hines } else if (RSExportForEach::isRSForEachFunc(mTargetAPI, FD)) { 122593a894650e81be54173106ec266f0311cebebd3Stephen Hines RSExportForEach *EFE = RSExportForEach::Create(this, FD); 123593a894650e81be54173106ec266f0311cebebd3Stephen Hines if (EFE == NULL) 124593a894650e81be54173106ec266f0311cebebd3Stephen Hines return false; 125593a894650e81be54173106ec266f0311cebebd3Stephen Hines else 126593a894650e81be54173106ec266f0311cebebd3Stephen Hines mExportForEach.push_back(EFE); 127593a894650e81be54173106ec266f0311cebebd3Stephen Hines return true; 1283fbe68a4ec20fec25f8a40191437bbc02d00f591Stephen Hines } 1293fbe68a4ec20fec25f8a40191437bbc02d00f591Stephen Hines 1309ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao RSExportFunc *EF = RSExportFunc::Create(this, FD); 1319ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao if (EF == NULL) 1329ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao return false; 1339ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao else 1349ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao mExportFuncs.push_back(EF); 135462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 1369ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao return true; 1379ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao} 138462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 139462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 1409ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liaobool RSContext::processExportType(const llvm::StringRef &Name) { 1419e5b503349719144f63ccb7c62ee9c291a7d83b8Stephen Hines clang::TranslationUnitDecl *TUDecl = mCtx.getTranslationUnitDecl(); 142462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 1436e6578a360497f78a181e63d7783422a9c9bfb15Stephen Hines slangAssert(TUDecl != NULL && "Translation unit declaration (top-level " 1446e6578a360497f78a181e63d7783422a9c9bfb15Stephen Hines "declaration) is null object"); 145f52a620440fa62257dfdcf2583f0f9df5b855c76Shih-wei Liao 1469e5b503349719144f63ccb7c62ee9c291a7d83b8Stephen Hines const clang::IdentifierInfo *II = mPP.getIdentifierInfo(Name); 1479ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao if (II == NULL) 1486315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr // TODO(zonr): alert identifier @Name mark as an exportable type cannot be 1496315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr // found 1509ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao return false; 151f52a620440fa62257dfdcf2583f0f9df5b855c76Shih-wei Liao 1529ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao clang::DeclContext::lookup_const_result R = TUDecl->lookup(II); 1539ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao RSExportType *ET = NULL; 154f52a620440fa62257dfdcf2583f0f9df5b855c76Shih-wei Liao 1559ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao for (clang::DeclContext::lookup_const_iterator I = R.first, E = R.second; 1569ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao I != E; 1579ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao I++) { 1589ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao clang::NamedDecl *const ND = *I; 1599ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao const clang::Type *T = NULL; 160462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 1619ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao switch (ND->getKind()) { 1629ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao case clang::Decl::Typedef: { 1639ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao T = static_cast<const clang::TypedefDecl*>( 1649ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao ND)->getCanonicalDecl()->getUnderlyingType().getTypePtr(); 1659ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao break; 1669ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao } 1679ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao case clang::Decl::Record: { 1689ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao T = static_cast<const clang::RecordDecl*>(ND)->getTypeForDecl(); 1699ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao break; 1709ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao } 1719ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao default: { 1729ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao // unsupported, skip 1739ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao break; 174ac91815eb4918b9307ec876f34f2b60adf28c78bShih-wei Liao } 175ac91815eb4918b9307ec876f34f2b60adf28c78bShih-wei Liao } 176537446c9542fdda458920876ed8a020a97ddf7a4Shih-wei Liao 1779ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao if (T != NULL) 1789ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao ET = RSExportType::Create(this, T); 1799ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao } 180462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 1819ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao return (ET != NULL); 182462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao} 183462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 184c17e198ffcd37bfc57e3add1f6eee952ae2a2eabStephen Hines 185c17e198ffcd37bfc57e3add1f6eee952ae2a2eabStephen Hines// Possibly re-order ForEach exports (maybe generating a dummy "root" function). 186c17e198ffcd37bfc57e3add1f6eee952ae2a2eabStephen Hines// We require "root" to be listed as slot 0 of our exported compute kernels, 187c17e198ffcd37bfc57e3add1f6eee952ae2a2eabStephen Hines// so this only needs to be created if we have other non-root kernels. 188c17e198ffcd37bfc57e3add1f6eee952ae2a2eabStephen Hinesvoid RSContext::cleanupForEach() { 189c17e198ffcd37bfc57e3add1f6eee952ae2a2eabStephen Hines bool foundNonRoot = false; 190c17e198ffcd37bfc57e3add1f6eee952ae2a2eabStephen Hines ExportForEachList::iterator begin = mExportForEach.begin(); 191c17e198ffcd37bfc57e3add1f6eee952ae2a2eabStephen Hines 192c17e198ffcd37bfc57e3add1f6eee952ae2a2eabStephen Hines for (ExportForEachList::iterator I = begin, E = mExportForEach.end(); 193c17e198ffcd37bfc57e3add1f6eee952ae2a2eabStephen Hines I != E; 194c17e198ffcd37bfc57e3add1f6eee952ae2a2eabStephen Hines I++) { 195c17e198ffcd37bfc57e3add1f6eee952ae2a2eabStephen Hines RSExportForEach *EFE = *I; 196c17e198ffcd37bfc57e3add1f6eee952ae2a2eabStephen Hines if (!EFE->getName().compare("root")) { 197c17e198ffcd37bfc57e3add1f6eee952ae2a2eabStephen Hines if (I == begin) { 198c17e198ffcd37bfc57e3add1f6eee952ae2a2eabStephen Hines // Nothing to do, since it is the first function 199c17e198ffcd37bfc57e3add1f6eee952ae2a2eabStephen Hines return; 200c17e198ffcd37bfc57e3add1f6eee952ae2a2eabStephen Hines } 201c17e198ffcd37bfc57e3add1f6eee952ae2a2eabStephen Hines 202c17e198ffcd37bfc57e3add1f6eee952ae2a2eabStephen Hines mExportForEach.erase(I); 203c17e198ffcd37bfc57e3add1f6eee952ae2a2eabStephen Hines mExportForEach.push_front(EFE); 204c17e198ffcd37bfc57e3add1f6eee952ae2a2eabStephen Hines return; 205c17e198ffcd37bfc57e3add1f6eee952ae2a2eabStephen Hines } else { 206c17e198ffcd37bfc57e3add1f6eee952ae2a2eabStephen Hines foundNonRoot = true; 207c17e198ffcd37bfc57e3add1f6eee952ae2a2eabStephen Hines } 208c17e198ffcd37bfc57e3add1f6eee952ae2a2eabStephen Hines } 209c17e198ffcd37bfc57e3add1f6eee952ae2a2eabStephen Hines 210c17e198ffcd37bfc57e3add1f6eee952ae2a2eabStephen Hines // If we found a non-root kernel, but no root() function, we need to add a 211c17e198ffcd37bfc57e3add1f6eee952ae2a2eabStephen Hines // dummy version (so that script->script calls of rsForEach don't behave 212c17e198ffcd37bfc57e3add1f6eee952ae2a2eabStephen Hines // erratically). 213c17e198ffcd37bfc57e3add1f6eee952ae2a2eabStephen Hines if (foundNonRoot) { 214c17e198ffcd37bfc57e3add1f6eee952ae2a2eabStephen Hines RSExportForEach *DummyRoot = RSExportForEach::CreateDummyRoot(this); 215c17e198ffcd37bfc57e3add1f6eee952ae2a2eabStephen Hines mExportForEach.push_front(DummyRoot); 216c17e198ffcd37bfc57e3add1f6eee952ae2a2eabStephen Hines } 217c17e198ffcd37bfc57e3add1f6eee952ae2a2eabStephen Hines} 218c17e198ffcd37bfc57e3add1f6eee952ae2a2eabStephen Hines 219c17e198ffcd37bfc57e3add1f6eee952ae2a2eabStephen Hines 220c808a99831115928b4648f4c8b86dc682594217aStephen Hinesbool RSContext::processExport() { 221b6809edd6ee8b90982425fb8ad24e867708b46fcStephen Hines bool valid = true; 222593a894650e81be54173106ec266f0311cebebd3Stephen Hines 223593a894650e81be54173106ec266f0311cebebd3Stephen Hines if (getDiagnostics()->hasErrorOccurred()) { 224593a894650e81be54173106ec266f0311cebebd3Stephen Hines return false; 225593a894650e81be54173106ec266f0311cebebd3Stephen Hines } 226593a894650e81be54173106ec266f0311cebebd3Stephen Hines 2279ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao // Export variable 2289e5b503349719144f63ccb7c62ee9c291a7d83b8Stephen Hines clang::TranslationUnitDecl *TUDecl = mCtx.getTranslationUnitDecl(); 2299ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao for (clang::DeclContext::decl_iterator DI = TUDecl->decls_begin(), 2309ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao DE = TUDecl->decls_end(); 231c97a333bc84ce8c28c96d07734cbded75c914639Stephen Hines DI != DE; 2329ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao DI++) { 2339ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao if (DI->getKind() == clang::Decl::Var) { 2349ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao clang::VarDecl *VD = (clang::VarDecl*) (*DI); 2352d2512c5703bb238b7935ab5228beff6563f2f94Stephen Hines if (VD->getLinkage() == clang::ExternalLinkage) { 2369ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao if (!processExportVar(VD)) { 237b6809edd6ee8b90982425fb8ad24e867708b46fcStephen Hines valid = false; 2389ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao } 2399ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao } 2409ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao } else if (DI->getKind() == clang::Decl::Function) { 2419ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao // Export functions 2429ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao clang::FunctionDecl *FD = (clang::FunctionDecl*) (*DI); 2432d2512c5703bb238b7935ab5228beff6563f2f94Stephen Hines if (FD->getLinkage() == clang::ExternalLinkage) { 2449ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao if (!processExportFunc(FD)) { 245b6809edd6ee8b90982425fb8ad24e867708b46fcStephen Hines valid = false; 2469ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao } 2479ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao } 2483f8b44dba57685b437cecc208f2a20a4ed93ed36Ying Wang } 2499ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao } 2509ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao 251c17e198ffcd37bfc57e3add1f6eee952ae2a2eabStephen Hines if (valid) { 252c17e198ffcd37bfc57e3add1f6eee952ae2a2eabStephen Hines cleanupForEach(); 253c17e198ffcd37bfc57e3add1f6eee952ae2a2eabStephen Hines } 254c17e198ffcd37bfc57e3add1f6eee952ae2a2eabStephen Hines 2559ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao // Finally, export type forcely set to be exported by user 2569ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao for (NeedExportTypeSet::const_iterator EI = mNeedExportTypes.begin(), 2579ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao EE = mNeedExportTypes.end(); 2589ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao EI != EE; 2599ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao EI++) { 2609ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao if (!processExportType(EI->getKey())) { 261b6809edd6ee8b90982425fb8ad24e867708b46fcStephen Hines valid = false; 2623f8b44dba57685b437cecc208f2a20a4ed93ed36Ying Wang } 2639ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao } 2649ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao 265b6809edd6ee8b90982425fb8ad24e867708b46fcStephen Hines return valid; 2669ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao} 267537446c9542fdda458920876ed8a020a97ddf7a4Shih-wei Liao 2689ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liaobool RSContext::insertExportType(const llvm::StringRef &TypeName, 2699ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao RSExportType *ET) { 2709ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao ExportTypeMap::value_type *NewItem = 2719ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao ExportTypeMap::value_type::Create(TypeName.begin(), 2729ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao TypeName.end(), 2739ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao mExportTypes.getAllocator(), 2749ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao ET); 2759ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao 2769ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao if (mExportTypes.insert(NewItem)) { 2779ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao return true; 2789ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao } else { 2799ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao free(NewItem); 2809ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao return false; 2819ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao } 282462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao} 283462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 284b81c6a4cbd9c08e0b20ea4fbc615b416ac1bc9ecShih-wei Liaobool RSContext::reflectToJava(const std::string &OutputPathBase, 285b81c6a4cbd9c08e0b20ea4fbc615b416ac1bc9ecShih-wei Liao const std::string &OutputPackageName, 2860a813a3ef2a82f19d7eab9e23ae8493197143803Stephen Hines const std::string &RSPackageName, 2879ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao const std::string &InputFileName, 2889ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao const std::string &OutputBCFileName, 289b81c6a4cbd9c08e0b20ea4fbc615b416ac1bc9ecShih-wei Liao std::string *RealPackageName) { 290b81c6a4cbd9c08e0b20ea4fbc615b416ac1bc9ecShih-wei Liao if (RealPackageName != NULL) 291b81c6a4cbd9c08e0b20ea4fbc615b416ac1bc9ecShih-wei Liao RealPackageName->clear(); 292b81c6a4cbd9c08e0b20ea4fbc615b416ac1bc9ecShih-wei Liao 293b81c6a4cbd9c08e0b20ea4fbc615b416ac1bc9ecShih-wei Liao const std::string &PackageName = 294b81c6a4cbd9c08e0b20ea4fbc615b416ac1bc9ecShih-wei Liao ((OutputPackageName.empty()) ? mReflectJavaPackageName : 295b81c6a4cbd9c08e0b20ea4fbc615b416ac1bc9ecShih-wei Liao OutputPackageName); 29635f5b39f0490953f1fe13ef803b43e3ced9a01d9Stephen Hines if (PackageName.empty()) { 2977aff4a0a124209fdf93ecbcd7aed701d39ba094bStephen Hines clang::DiagnosticsEngine *DiagEngine = getDiagnostics(); 2987aff4a0a124209fdf93ecbcd7aed701d39ba094bStephen Hines const clang::SourceManager *SM = getSourceManager(); 2997aff4a0a124209fdf93ecbcd7aed701d39ba094bStephen Hines DiagEngine->Report( 3007aff4a0a124209fdf93ecbcd7aed701d39ba094bStephen Hines SM->getLocForEndOfFile(SM->getMainFileID()), 3017aff4a0a124209fdf93ecbcd7aed701d39ba094bStephen Hines DiagEngine->getCustomDiagID(clang::DiagnosticsEngine::Error, 3027aff4a0a124209fdf93ecbcd7aed701d39ba094bStephen Hines "missing \"#pragma rs " 3037aff4a0a124209fdf93ecbcd7aed701d39ba094bStephen Hines "java_package_name(com.foo.bar)\" " 3047aff4a0a124209fdf93ecbcd7aed701d39ba094bStephen Hines "in source file")); 30535f5b39f0490953f1fe13ef803b43e3ced9a01d9Stephen Hines return false; 30635f5b39f0490953f1fe13ef803b43e3ced9a01d9Stephen Hines } 3079ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao 3089ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao // Copy back the really applied package name 309b81c6a4cbd9c08e0b20ea4fbc615b416ac1bc9ecShih-wei Liao RealPackageName->assign(PackageName); 3109ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao 3110a813a3ef2a82f19d7eab9e23ae8493197143803Stephen Hines if (!RSPackageName.empty()) { 3120a813a3ef2a82f19d7eab9e23ae8493197143803Stephen Hines mRSPackageName = RSPackageName; 3130a813a3ef2a82f19d7eab9e23ae8493197143803Stephen Hines } 3140a813a3ef2a82f19d7eab9e23ae8493197143803Stephen Hines 3154cc67fce91f43215d61b2695746eab102a3db516Stephen Hines RSReflection *R = new RSReflection(this, mGeneratedFileNames); 3160a813a3ef2a82f19d7eab9e23ae8493197143803Stephen Hines bool ret = R->reflect(OutputPathBase, PackageName, mRSPackageName, 317b81c6a4cbd9c08e0b20ea4fbc615b416ac1bc9ecShih-wei Liao InputFileName, OutputBCFileName); 3186315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr if (!ret) 3199ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao fprintf(stderr, "RSContext::reflectToJava : failed to do reflection " 3209ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao "(%s)\n", R->getLastError()); 3219ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao delete R; 3229ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao return ret; 3239ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao} 3246de89272b00a31f2a73e2f56edf9cc511df46265Shih-wei Liao 325462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei LiaoRSContext::~RSContext() { 3269ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao delete mLicenseNote; 3279ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao delete mTargetData; 328a41ce1d98094da84643995d40d71c529905123fcZonr Chang for (ExportableList::iterator I = mExportables.begin(), 329a41ce1d98094da84643995d40d71c529905123fcZonr Chang E = mExportables.end(); 330a41ce1d98094da84643995d40d71c529905123fcZonr Chang I != E; 331a41ce1d98094da84643995d40d71c529905123fcZonr Chang I++) { 332641558f02fe6ce0ee3ae5076eb366c25e2ad5903Zonr Chang if (!(*I)->isKeep()) 333641558f02fe6ce0ee3ae5076eb366c25e2ad5903Zonr Chang delete *I; 334a41ce1d98094da84643995d40d71c529905123fcZonr Chang } 335462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao} 336e639eb5caa2c386b4a60659a4929e8a6141a2cbeStephen Hines 337e639eb5caa2c386b4a60659a4929e8a6141a2cbeStephen Hines} // namespace slang 338