slang_rs_backend.cpp revision 1bdd4978caabcdc9489bdcb7f1cd6087340699e8
1c383a500aa59423264811be3874461bf8adbfea0Zonr Chang/* 2c383a500aa59423264811be3874461bf8adbfea0Zonr Chang * Copyright 2010, 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_backend.h" 18462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 196315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr#include <vector> 206315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr#include <string> 21462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 226315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr#include "llvm/Metadata.h" 239ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao#include "llvm/Constant.h" 249ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao#include "llvm/Constants.h" 259ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao#include "llvm/Module.h" 269ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao#include "llvm/Function.h" 279ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao#include "llvm/DerivedTypes.h" 28462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 296315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr#include "llvm/Support/IRBuilder.h" 306315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr 316315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr#include "llvm/ADT/Twine.h" 326315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr#include "llvm/ADT/StringExtras.h" 336315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr 34592a954aae4cb946970b557e94afd5ee453fd57eZonr Chang#include "slang_rs.h" 356315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr#include "slang_rs_context.h" 367f2f385d3129ba4de82ef56adb9cb4385fd33001Zonr Chang#include "slang_rs_metadata.h" 376315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr#include "slang_rs_export_var.h" 386315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr#include "slang_rs_export_func.h" 396315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr#include "slang_rs_export_type.h" 40462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 419ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liaousing namespace slang; 42462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 439ef2f785e0cc490af678dfd685995dec787321ffShih-wei LiaoRSBackend::RSBackend(RSContext *Context, 449ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao clang::Diagnostic &Diags, 459ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao const clang::CodeGenOptions &CodeGenOpts, 469ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao const clang::TargetOptions &TargetOpts, 479ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao const PragmaList &Pragmas, 489ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao llvm::raw_ostream *OS, 493a9ca1f0d6bd8f12c2bb2adea51f95c255996180Zonr Chang Slang::OutputType OT, 509ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao clang::SourceManager &SourceMgr, 516315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr bool AllowRSPrefix) 526315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr : Backend(Diags, 536315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr CodeGenOpts, 546315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr TargetOpts, 556315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr Pragmas, 566315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr OS, 573a9ca1f0d6bd8f12c2bb2adea51f95c255996180Zonr Chang OT), 586315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr mContext(Context), 593a9ca1f0d6bd8f12c2bb2adea51f95c255996180Zonr Chang mSourceMgr(SourceMgr), 603a9ca1f0d6bd8f12c2bb2adea51f95c255996180Zonr Chang mAllowRSPrefix(AllowRSPrefix), 616315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr mExportVarMetadata(NULL), 626315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr mExportFuncMetadata(NULL), 636315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr mExportTypeMetadata(NULL) { 649ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao return; 65462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao} 66462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 67cfae0f350e4e0d8f7b4c71780b3f74f58fa23afbStephen Hines// 1) Add zero initialization of local RS object types 68cfae0f350e4e0d8f7b4c71780b3f74f58fa23afbStephen Hinesvoid RSBackend::AnnotateFunction(clang::FunctionDecl *FD) { 69cfae0f350e4e0d8f7b4c71780b3f74f58fa23afbStephen Hines if (FD && 70cfae0f350e4e0d8f7b4c71780b3f74f58fa23afbStephen Hines FD->hasBody() && 71cfae0f350e4e0d8f7b4c71780b3f74f58fa23afbStephen Hines !SlangRS::IsFunctionInRSHeaderFile(FD, mSourceMgr)) { 721bdd4978caabcdc9489bdcb7f1cd6087340699e8Stephen Hines mRefCount.Init(mContext->getASTContext()); 734b32ffdfc1ac766f8932e7effbcdf7484e804a8eStephen Hines mRefCount.Visit(FD->getBody()); 74cfae0f350e4e0d8f7b4c71780b3f74f58fa23afbStephen Hines } 75cfae0f350e4e0d8f7b4c71780b3f74f58fa23afbStephen Hines return; 76cfae0f350e4e0d8f7b4c71780b3f74f58fa23afbStephen Hines} 77cfae0f350e4e0d8f7b4c71780b3f74f58fa23afbStephen Hines 78cfae0f350e4e0d8f7b4c71780b3f74f58fa23afbStephen Hinesvoid RSBackend::HandleTopLevelDecl(clang::DeclGroupRef D) { 79cfae0f350e4e0d8f7b4c71780b3f74f58fa23afbStephen Hines // Disallow user-defined functions with prefix "rs" 80cfae0f350e4e0d8f7b4c71780b3f74f58fa23afbStephen Hines if (!mAllowRSPrefix) { 81cfae0f350e4e0d8f7b4c71780b3f74f58fa23afbStephen Hines // Iterate all function declarations in the program. 82cfae0f350e4e0d8f7b4c71780b3f74f58fa23afbStephen Hines for (clang::DeclGroupRef::iterator I = D.begin(), E = D.end(); 83cfae0f350e4e0d8f7b4c71780b3f74f58fa23afbStephen Hines I != E; I++) { 84cfae0f350e4e0d8f7b4c71780b3f74f58fa23afbStephen Hines clang::FunctionDecl *FD = dyn_cast<clang::FunctionDecl>(*I); 85cfae0f350e4e0d8f7b4c71780b3f74f58fa23afbStephen Hines if (FD == NULL) 86cfae0f350e4e0d8f7b4c71780b3f74f58fa23afbStephen Hines continue; 87cfae0f350e4e0d8f7b4c71780b3f74f58fa23afbStephen Hines if (!FD->getName().startswith("rs")) // Check prefix 88cfae0f350e4e0d8f7b4c71780b3f74f58fa23afbStephen Hines continue; 89cfae0f350e4e0d8f7b4c71780b3f74f58fa23afbStephen Hines if (!SlangRS::IsFunctionInRSHeaderFile(FD, mSourceMgr)) 90cfae0f350e4e0d8f7b4c71780b3f74f58fa23afbStephen Hines mDiags.Report(clang::FullSourceLoc(FD->getLocation(), mSourceMgr), 91cfae0f350e4e0d8f7b4c71780b3f74f58fa23afbStephen Hines mDiags.getCustomDiagID(clang::Diagnostic::Error, 92cfae0f350e4e0d8f7b4c71780b3f74f58fa23afbStephen Hines "invalid function name prefix, " 93cfae0f350e4e0d8f7b4c71780b3f74f58fa23afbStephen Hines "\"rs\" is reserved: '%0'")) 94cfae0f350e4e0d8f7b4c71780b3f74f58fa23afbStephen Hines << FD->getName(); 95cfae0f350e4e0d8f7b4c71780b3f74f58fa23afbStephen Hines } 96cfae0f350e4e0d8f7b4c71780b3f74f58fa23afbStephen Hines } 97cfae0f350e4e0d8f7b4c71780b3f74f58fa23afbStephen Hines 98cfae0f350e4e0d8f7b4c71780b3f74f58fa23afbStephen Hines // Process any non-static function declarations 99cfae0f350e4e0d8f7b4c71780b3f74f58fa23afbStephen Hines for (clang::DeclGroupRef::iterator I = D.begin(), E = D.end(); I != E; I++) { 100cfae0f350e4e0d8f7b4c71780b3f74f58fa23afbStephen Hines AnnotateFunction(dyn_cast<clang::FunctionDecl>(*I)); 101cfae0f350e4e0d8f7b4c71780b3f74f58fa23afbStephen Hines } 102cfae0f350e4e0d8f7b4c71780b3f74f58fa23afbStephen Hines 103cfae0f350e4e0d8f7b4c71780b3f74f58fa23afbStephen Hines Backend::HandleTopLevelDecl(D); 104cfae0f350e4e0d8f7b4c71780b3f74f58fa23afbStephen Hines return; 105cfae0f350e4e0d8f7b4c71780b3f74f58fa23afbStephen Hines} 106cfae0f350e4e0d8f7b4c71780b3f74f58fa23afbStephen Hines 107fcda2352b9e140529f8f3c89f05b10a70c0048b2Stephen Hinesvoid RSBackend::HandleTranslationUnitPre(clang::ASTContext& C) { 108fcda2352b9e140529f8f3c89f05b10a70c0048b2Stephen Hines clang::TranslationUnitDecl *TUDecl = C.getTranslationUnitDecl(); 109fcda2352b9e140529f8f3c89f05b10a70c0048b2Stephen Hines 110cfae0f350e4e0d8f7b4c71780b3f74f58fa23afbStephen Hines // Process any static function declarations 111fcda2352b9e140529f8f3c89f05b10a70c0048b2Stephen Hines for (clang::DeclContext::decl_iterator I = TUDecl->decls_begin(), 112fcda2352b9e140529f8f3c89f05b10a70c0048b2Stephen Hines E = TUDecl->decls_end(); I != E; I++) { 113fcda2352b9e140529f8f3c89f05b10a70c0048b2Stephen Hines if ((I->getKind() >= clang::Decl::firstFunction) && 114fcda2352b9e140529f8f3c89f05b10a70c0048b2Stephen Hines (I->getKind() <= clang::Decl::lastFunction)) { 115cfae0f350e4e0d8f7b4c71780b3f74f58fa23afbStephen Hines AnnotateFunction(static_cast<clang::FunctionDecl*>(*I)); 116fcda2352b9e140529f8f3c89f05b10a70c0048b2Stephen Hines } 117fcda2352b9e140529f8f3c89f05b10a70c0048b2Stephen Hines } 118fcda2352b9e140529f8f3c89f05b10a70c0048b2Stephen Hines 119fcda2352b9e140529f8f3c89f05b10a70c0048b2Stephen Hines return; 120fcda2352b9e140529f8f3c89f05b10a70c0048b2Stephen Hines} 121462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 122fcda2352b9e140529f8f3c89f05b10a70c0048b2Stephen Hines/////////////////////////////////////////////////////////////////////////////// 12368fc02ca4a7235e2981be5eee4ad968a9d3928c0Zonr Changvoid RSBackend::HandleTranslationUnitPost(llvm::Module *M) { 1249ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao mContext->processExport(); 1259ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao 1269ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao // Dump export variable info 1279ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao if (mContext->hasExportVar()) { 1289ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao if (mExportVarMetadata == NULL) 12968fc02ca4a7235e2981be5eee4ad968a9d3928c0Zonr Chang mExportVarMetadata = M->getOrInsertNamedMetadata(RS_EXPORT_VAR_MN); 1309ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao 1319ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao llvm::SmallVector<llvm::Value*, 2> ExportVarInfo; 1329ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao 1336315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr for (RSContext::const_export_var_iterator I = mContext->export_vars_begin(), 1346315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr E = mContext->export_vars_end(); 1356315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr I != E; 1366315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr I++) { 1376315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr const RSExportVar *EV = *I; 1386315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr const RSExportType *ET = EV->getType(); 1399ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao 1409ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao // Variable name 1419ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao ExportVarInfo.push_back( 1429ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao llvm::MDString::get(mLLVMContext, EV->getName().c_str())); 1439ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao 1449ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao // Type name 145a65ec168e41e3ee9c6e8ac04cde694bbbfc2590aZonr Chang switch (ET->getClass()) { 146a65ec168e41e3ee9c6e8ac04cde694bbbfc2590aZonr Chang case RSExportType::ExportClassPrimitive: { 147a65ec168e41e3ee9c6e8ac04cde694bbbfc2590aZonr Chang ExportVarInfo.push_back( 148a65ec168e41e3ee9c6e8ac04cde694bbbfc2590aZonr Chang llvm::MDString::get( 1499ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao mLLVMContext, llvm::utostr_32( 150a65ec168e41e3ee9c6e8ac04cde694bbbfc2590aZonr Chang static_cast<const RSExportPrimitiveType*>(ET)->getType()))); 151a65ec168e41e3ee9c6e8ac04cde694bbbfc2590aZonr Chang break; 152a65ec168e41e3ee9c6e8ac04cde694bbbfc2590aZonr Chang } 153a65ec168e41e3ee9c6e8ac04cde694bbbfc2590aZonr Chang case RSExportType::ExportClassPointer: { 154a65ec168e41e3ee9c6e8ac04cde694bbbfc2590aZonr Chang ExportVarInfo.push_back( 155a65ec168e41e3ee9c6e8ac04cde694bbbfc2590aZonr Chang llvm::MDString::get( 1569ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao mLLVMContext, ("*" + static_cast<const RSExportPointerType*>(ET) 157a65ec168e41e3ee9c6e8ac04cde694bbbfc2590aZonr Chang ->getPointeeType()->getName()).c_str())); 158a65ec168e41e3ee9c6e8ac04cde694bbbfc2590aZonr Chang break; 159a65ec168e41e3ee9c6e8ac04cde694bbbfc2590aZonr Chang } 160a65ec168e41e3ee9c6e8ac04cde694bbbfc2590aZonr Chang case RSExportType::ExportClassMatrix: { 161a65ec168e41e3ee9c6e8ac04cde694bbbfc2590aZonr Chang ExportVarInfo.push_back( 162a65ec168e41e3ee9c6e8ac04cde694bbbfc2590aZonr Chang llvm::MDString::get( 163a65ec168e41e3ee9c6e8ac04cde694bbbfc2590aZonr Chang mLLVMContext, llvm::utostr_32( 164a65ec168e41e3ee9c6e8ac04cde694bbbfc2590aZonr Chang RSExportPrimitiveType::DataTypeRSMatrix2x2 + 165a65ec168e41e3ee9c6e8ac04cde694bbbfc2590aZonr Chang static_cast<const RSExportMatrixType*>(ET)->getDim() - 2))); 166a65ec168e41e3ee9c6e8ac04cde694bbbfc2590aZonr Chang break; 167a65ec168e41e3ee9c6e8ac04cde694bbbfc2590aZonr Chang } 168a65ec168e41e3ee9c6e8ac04cde694bbbfc2590aZonr Chang case RSExportType::ExportClassVector: 169a65ec168e41e3ee9c6e8ac04cde694bbbfc2590aZonr Chang case RSExportType::ExportClassConstantArray: 170a65ec168e41e3ee9c6e8ac04cde694bbbfc2590aZonr Chang case RSExportType::ExportClassRecord: { 171a65ec168e41e3ee9c6e8ac04cde694bbbfc2590aZonr Chang ExportVarInfo.push_back( 172a65ec168e41e3ee9c6e8ac04cde694bbbfc2590aZonr Chang llvm::MDString::get(mLLVMContext, 173a65ec168e41e3ee9c6e8ac04cde694bbbfc2590aZonr Chang EV->getType()->getName().c_str())); 174a65ec168e41e3ee9c6e8ac04cde694bbbfc2590aZonr Chang break; 175a65ec168e41e3ee9c6e8ac04cde694bbbfc2590aZonr Chang } 176a65ec168e41e3ee9c6e8ac04cde694bbbfc2590aZonr Chang } 1779ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao 1789ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao mExportVarMetadata->addOperand( 1799ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao llvm::MDNode::get(mLLVMContext, 1809ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao ExportVarInfo.data(), 1819ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao ExportVarInfo.size()) ); 1829ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao 1839ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao ExportVarInfo.clear(); 184462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao } 1859ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao } 1869ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao 1879ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao // Dump export function info 1889ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao if (mContext->hasExportFunc()) { 1899ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao if (mExportFuncMetadata == NULL) 1909ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao mExportFuncMetadata = 19168fc02ca4a7235e2981be5eee4ad968a9d3928c0Zonr Chang M->getOrInsertNamedMetadata(RS_EXPORT_FUNC_MN); 1929ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao 1939ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao llvm::SmallVector<llvm::Value*, 1> ExportFuncInfo; 1949ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao 1956315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr for (RSContext::const_export_func_iterator 1966315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr I = mContext->export_funcs_begin(), 1976315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr E = mContext->export_funcs_end(); 1989ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao I != E; 1999ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao I++) { 2006315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr const RSExportFunc *EF = *I; 2019ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao 2029ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao // Function name 2036315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr if (!EF->hasParam()) { 2046315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr ExportFuncInfo.push_back(llvm::MDString::get(mLLVMContext, 2056315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr EF->getName().c_str())); 2066315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr } else { 20768fc02ca4a7235e2981be5eee4ad968a9d3928c0Zonr Chang llvm::Function *F = M->getFunction(EF->getName()); 2089ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao llvm::Function *HelperFunction; 2099ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao const std::string HelperFunctionName(".helper_" + EF->getName()); 2109ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao 2119ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao assert(F && "Function marked as exported disappeared in Bitcode"); 2129ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao 2139ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao // Create helper function 214462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao { 2150da0a7dc51c25943fe31d0bfccbdfee326a3199cZonr Chang llvm::StructType *HelperFunctionParameterTy = NULL; 2160da0a7dc51c25943fe31d0bfccbdfee326a3199cZonr Chang 2170da0a7dc51c25943fe31d0bfccbdfee326a3199cZonr Chang if (!F->getArgumentList().empty()) { 2180da0a7dc51c25943fe31d0bfccbdfee326a3199cZonr Chang std::vector<const llvm::Type*> HelperFunctionParameterTys; 2190da0a7dc51c25943fe31d0bfccbdfee326a3199cZonr Chang for (llvm::Function::arg_iterator AI = F->arg_begin(), 2200da0a7dc51c25943fe31d0bfccbdfee326a3199cZonr Chang AE = F->arg_end(); AI != AE; AI++) 2210da0a7dc51c25943fe31d0bfccbdfee326a3199cZonr Chang HelperFunctionParameterTys.push_back(AI->getType()); 2220da0a7dc51c25943fe31d0bfccbdfee326a3199cZonr Chang 2230da0a7dc51c25943fe31d0bfccbdfee326a3199cZonr Chang HelperFunctionParameterTy = 2240da0a7dc51c25943fe31d0bfccbdfee326a3199cZonr Chang llvm::StructType::get(mLLVMContext, HelperFunctionParameterTys); 2250da0a7dc51c25943fe31d0bfccbdfee326a3199cZonr Chang } 2260da0a7dc51c25943fe31d0bfccbdfee326a3199cZonr Chang 2270da0a7dc51c25943fe31d0bfccbdfee326a3199cZonr Chang if (!EF->checkParameterPacketType(HelperFunctionParameterTy)) { 2280da0a7dc51c25943fe31d0bfccbdfee326a3199cZonr Chang fprintf(stderr, "Failed to export function %s: parameter type " 2290da0a7dc51c25943fe31d0bfccbdfee326a3199cZonr Chang "mismatch during creation of helper function.\n", 2300da0a7dc51c25943fe31d0bfccbdfee326a3199cZonr Chang EF->getName().c_str()); 2310da0a7dc51c25943fe31d0bfccbdfee326a3199cZonr Chang 2320da0a7dc51c25943fe31d0bfccbdfee326a3199cZonr Chang const RSExportRecordType *Expected = EF->getParamPacketType(); 2330da0a7dc51c25943fe31d0bfccbdfee326a3199cZonr Chang if (Expected) { 2340da0a7dc51c25943fe31d0bfccbdfee326a3199cZonr Chang fprintf(stderr, "Expected:\n"); 2350da0a7dc51c25943fe31d0bfccbdfee326a3199cZonr Chang Expected->getLLVMType()->dump(); 2360da0a7dc51c25943fe31d0bfccbdfee326a3199cZonr Chang } 2370da0a7dc51c25943fe31d0bfccbdfee326a3199cZonr Chang if (HelperFunctionParameterTy) { 2380da0a7dc51c25943fe31d0bfccbdfee326a3199cZonr Chang fprintf(stderr, "Got:\n"); 2390da0a7dc51c25943fe31d0bfccbdfee326a3199cZonr Chang HelperFunctionParameterTy->dump(); 2400da0a7dc51c25943fe31d0bfccbdfee326a3199cZonr Chang } 2410da0a7dc51c25943fe31d0bfccbdfee326a3199cZonr Chang } 2420da0a7dc51c25943fe31d0bfccbdfee326a3199cZonr Chang 2439ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao std::vector<const llvm::Type*> Params; 2440da0a7dc51c25943fe31d0bfccbdfee326a3199cZonr Chang if (HelperFunctionParameterTy) { 2450da0a7dc51c25943fe31d0bfccbdfee326a3199cZonr Chang llvm::PointerType *HelperFunctionParameterTyP = 2460da0a7dc51c25943fe31d0bfccbdfee326a3199cZonr Chang llvm::PointerType::getUnqual(HelperFunctionParameterTy); 2470da0a7dc51c25943fe31d0bfccbdfee326a3199cZonr Chang Params.push_back(HelperFunctionParameterTyP); 2480da0a7dc51c25943fe31d0bfccbdfee326a3199cZonr Chang } 2499ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao 2500da0a7dc51c25943fe31d0bfccbdfee326a3199cZonr Chang llvm::FunctionType * HelperFunctionType = 2510da0a7dc51c25943fe31d0bfccbdfee326a3199cZonr Chang llvm::FunctionType::get(F->getReturnType(), 2520da0a7dc51c25943fe31d0bfccbdfee326a3199cZonr Chang Params, 2530da0a7dc51c25943fe31d0bfccbdfee326a3199cZonr Chang /* IsVarArgs = */false); 2549ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao 2559ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao HelperFunction = 2569ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao llvm::Function::Create(HelperFunctionType, 2579ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao llvm::GlobalValue::ExternalLinkage, 2589ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao HelperFunctionName, 25968fc02ca4a7235e2981be5eee4ad968a9d3928c0Zonr Chang M); 2609ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao 2619ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao HelperFunction->addFnAttr(llvm::Attribute::NoInline); 2629ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao HelperFunction->setCallingConv(F->getCallingConv()); 2639ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao 2649ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao // Create helper function body 2659ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao { 2669ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao llvm::Argument *HelperFunctionParameter = 2679ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao &(*HelperFunction->arg_begin()); 2689ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao llvm::BasicBlock *BB = 2699ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao llvm::BasicBlock::Create(mLLVMContext, "entry", HelperFunction); 2709ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao llvm::IRBuilder<> *IB = new llvm::IRBuilder<>(BB); 2719ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao llvm::SmallVector<llvm::Value*, 6> Params; 2729ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao llvm::Value *Idx[2]; 2739ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao 2749ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao Idx[0] = 2759ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao llvm::ConstantInt::get(llvm::Type::getInt32Ty(mLLVMContext), 0); 2769ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao 2779ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao // getelementptr and load instruction for all elements in 2789ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao // parameter .p 2790da0a7dc51c25943fe31d0bfccbdfee326a3199cZonr Chang for (size_t i = 0; i < EF->getNumParameters(); i++) { 2809ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao // getelementptr 2819ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao Idx[1] = 2829ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao llvm::ConstantInt::get( 2839ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao llvm::Type::getInt32Ty(mLLVMContext), i); 2849ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao llvm::Value *Ptr = IB->CreateInBoundsGEP(HelperFunctionParameter, 2859ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao Idx, 2869ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao Idx + 2); 2879ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao 2889ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao // load 2899ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao llvm::Value *V = IB->CreateLoad(Ptr); 2909ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao Params.push_back(V); 2919ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao } 292462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 2939ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao // Call and pass the all elements as paramter to F 2949ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao llvm::CallInst *CI = IB->CreateCall(F, 2959ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao Params.data(), 2969ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao Params.data() + Params.size()); 2971ebc0ca6ffa7effb875883d18205ed4943ab8fc2Shih-wei Liao 2989ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao CI->setCallingConv(F->getCallingConv()); 2991ebc0ca6ffa7effb875883d18205ed4943ab8fc2Shih-wei Liao 3009ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao if (F->getReturnType() == llvm::Type::getVoidTy(mLLVMContext)) 3019ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao IB->CreateRetVoid(); 3029ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao else 3039ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao IB->CreateRet(CI); 304462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 3059ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao delete IB; 3069ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao } 3079ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao } 308462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 3099ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao ExportFuncInfo.push_back( 3109ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao llvm::MDString::get(mLLVMContext, HelperFunctionName.c_str())); 3119ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao } 312462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 3139ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao mExportFuncMetadata->addOperand( 3149ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao llvm::MDNode::get(mLLVMContext, 3159ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao ExportFuncInfo.data(), 3169ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao ExportFuncInfo.size())); 317462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 3189ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao ExportFuncInfo.clear(); 3199ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao } 3209ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao } 3219ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao 3229ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao // Dump export type info 3239ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao if (mContext->hasExportType()) { 3249ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao llvm::SmallVector<llvm::Value*, 1> ExportTypeInfo; 3259ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao 3266315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr for (RSContext::const_export_type_iterator 3276315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr I = mContext->export_types_begin(), 3286315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr E = mContext->export_types_end(); 3299ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao I != E; 3309ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao I++) { 3319ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao // First, dump type name list to export 3329ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao const RSExportType *ET = I->getValue(); 3339ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao 3349ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao ExportTypeInfo.clear(); 3359ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao // Type name 3369ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao ExportTypeInfo.push_back( 3379ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao llvm::MDString::get(mLLVMContext, ET->getName().c_str())); 3389ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao 3399ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao if (ET->getClass() == RSExportType::ExportClassRecord) { 3409ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao const RSExportRecordType *ERT = 3419ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao static_cast<const RSExportRecordType*>(ET); 3429ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao 3439ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao if (mExportTypeMetadata == NULL) 3449ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao mExportTypeMetadata = 34568fc02ca4a7235e2981be5eee4ad968a9d3928c0Zonr Chang M->getOrInsertNamedMetadata(RS_EXPORT_TYPE_MN); 3469ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao 3479ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao mExportTypeMetadata->addOperand( 3489ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao llvm::MDNode::get(mLLVMContext, 3499ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao ExportTypeInfo.data(), 3509ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao ExportTypeInfo.size())); 3519ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao 3529ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao // Now, export struct field information to %[struct name] 3539ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao std::string StructInfoMetadataName("%"); 3549ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao StructInfoMetadataName.append(ET->getName()); 3559ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao llvm::NamedMDNode *StructInfoMetadata = 35668fc02ca4a7235e2981be5eee4ad968a9d3928c0Zonr Chang M->getOrInsertNamedMetadata(StructInfoMetadataName); 3579ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao llvm::SmallVector<llvm::Value*, 3> FieldInfo; 3589ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao 3599ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao assert(StructInfoMetadata->getNumOperands() == 0 && 3609ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao "Metadata with same name was created before"); 3616315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr for (RSExportRecordType::const_field_iterator FI = ERT->fields_begin(), 3626315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr FE = ERT->fields_end(); 3636315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr FI != FE; 3649ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao FI++) { 3659ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao const RSExportRecordType::Field *F = *FI; 3669ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao 3679ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao // 1. field name 3689ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao FieldInfo.push_back(llvm::MDString::get(mLLVMContext, 3699ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao F->getName().c_str())); 3709ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao 3719ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao // 2. field type name 3729ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao FieldInfo.push_back( 3739ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao llvm::MDString::get(mLLVMContext, 3749ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao F->getType()->getName().c_str())); 3759ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao 3769ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao // 3. field kind 3779ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao switch (F->getType()->getClass()) { 3789ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao case RSExportType::ExportClassPrimitive: 3799ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao case RSExportType::ExportClassVector: { 3809ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao const RSExportPrimitiveType *EPT = 3819ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao static_cast<const RSExportPrimitiveType*>(F->getType()); 3829ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao FieldInfo.push_back( 3839ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao llvm::MDString::get(mLLVMContext, 3849ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao llvm::itostr(EPT->getKind()))); 3859ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao break; 386462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao } 387462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 3889ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao default: { 3899ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao FieldInfo.push_back( 3909ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao llvm::MDString::get(mLLVMContext, 3919ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao llvm::itostr( 3926315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr RSExportPrimitiveType::DataKindUser))); 3939ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao break; 3949ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao } 3959ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao } 396462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 3976315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr StructInfoMetadata->addOperand(llvm::MDNode::get(mLLVMContext, 3986315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr FieldInfo.data(), 3996315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr FieldInfo.size())); 400462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 4019ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao FieldInfo.clear(); 402462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao } 4039ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao } // ET->getClass() == RSExportType::ExportClassRecord 404462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao } 4059ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao } 406462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 4079ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao return; 408462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao} 409462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 410462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei LiaoRSBackend::~RSBackend() { 4119ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao return; 412462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao} 413