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 17e639eb5caa2c386b4a60659a4929e8a6141a2cbeStephen Hines#ifndef _FRAMEWORKS_COMPILE_SLANG_SLANG_BACKEND_H_ // NOLINT 18e639eb5caa2c386b4a60659a4929e8a6141a2cbeStephen Hines#define _FRAMEWORKS_COMPILE_SLANG_SLANG_BACKEND_H_ 19e639eb5caa2c386b4a60659a4929e8a6141a2cbeStephen Hines 20e639eb5caa2c386b4a60659a4929e8a6141a2cbeStephen Hines#include "clang/AST/ASTConsumer.h" 21462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 22c706907a8041faaa882f9bd87f1d1c1669023a62Stephen Hines#include "llvm/IR/LegacyPassManager.h" 23462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 2421cc01860b95cad7ae60c686e511e8f4ae034e39Pirama Arumuga Nainar#include "llvm/Support/raw_ostream.h" 25462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 263a9ca1f0d6bd8f12c2bb2adea51f95c255996180Zonr Chang#include "slang.h" 2734e6205f1d5ceb90a73192cee8b215fa0b7ab306David Gross#include "slang_pragma_list.h" 288024ed54c23c08534434da14d3be99c3efcc5754Jean-Luc Brouillet#include "slang_rs_check_ast.h" 29fb40ee2a90f37967bf4a40a18dec7f60e5c580d8Yang Ni#include "slang_rs_foreach_lowering.h" 308024ed54c23c08534434da14d3be99c3efcc5754Jean-Luc Brouillet#include "slang_rs_object_ref_count.h" 314cc499d6e5ec602309501873449c938af61170b2Stephen Hines#include "slang_version.h" 326315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr 33462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liaonamespace llvm { 3421cc01860b95cad7ae60c686e511e8f4ae034e39Pirama Arumuga Nainar class buffer_ostream; 356315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr class LLVMContext; 366315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr class NamedMDNode; 376315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr class Module; 389ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao} 39462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 40462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liaonamespace clang { 418024ed54c23c08534434da14d3be99c3efcc5754Jean-Luc Brouillet class ASTConsumer; 428024ed54c23c08534434da14d3be99c3efcc5754Jean-Luc Brouillet class ASTContext; 433a9ca1f0d6bd8f12c2bb2adea51f95c255996180Zonr Chang class CodeGenOptions; 446315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr class CodeGenerator; 456315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr class DeclGroupRef; 468024ed54c23c08534434da14d3be99c3efcc5754Jean-Luc Brouillet class DiagnosticsEngine; 478024ed54c23c08534434da14d3be99c3efcc5754Jean-Luc Brouillet class FunctionDecl; 488f093e05e28046b6fc74175b66a06152f72e0c66Pirama Arumuga Nainar class HeaderSearchOptions; 498f093e05e28046b6fc74175b66a06152f72e0c66Pirama Arumuga Nainar class PreprocessorOptions; 506315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr class TagDecl; 518024ed54c23c08534434da14d3be99c3efcc5754Jean-Luc Brouillet class TargetOptions; 526315f76e3cc6ff2d012d1183a0b030d4ff0dc808zonr class VarDecl; 539ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao} 54462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 55462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liaonamespace slang { 56462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 578024ed54c23c08534434da14d3be99c3efcc5754Jean-Luc Brouilletclass RSContext; 588024ed54c23c08534434da14d3be99c3efcc5754Jean-Luc Brouillet 599ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liaoclass Backend : public clang::ASTConsumer { 609ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao private: 619ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao const clang::TargetOptions &mTargetOpts; 62462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 6368fc02ca4a7235e2981be5eee4ad968a9d3928c0Zonr Chang llvm::Module *mpModule; 6468fc02ca4a7235e2981be5eee4ad968a9d3928c0Zonr Chang 659ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao // Output stream 669ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao llvm::raw_ostream *mpOS; 673a9ca1f0d6bd8f12c2bb2adea51f95c255996180Zonr Chang Slang::OutputType mOT; 686b22674f4ef0a6c689c589830f1c44f443520785Kirk Stewart 699ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao // This helps us translate Clang AST using into LLVM IR 709ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao clang::CodeGenerator *mGen; 71462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 729ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao // Passes 73462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 749ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao // Passes apply on function scope in a translation unit 75c706907a8041faaa882f9bd87f1d1c1669023a62Stephen Hines llvm::legacy::FunctionPassManager *mPerFunctionPasses; 769ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao // Passes apply on module scope 77c706907a8041faaa882f9bd87f1d1c1669023a62Stephen Hines llvm::legacy::PassManager *mPerModulePasses; 789ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao // Passes for code emission 79c706907a8041faaa882f9bd87f1d1c1669023a62Stephen Hines llvm::legacy::FunctionPassManager *mCodeGenPasses; 80462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 8121cc01860b95cad7ae60c686e511e8f4ae034e39Pirama Arumuga Nainar llvm::buffer_ostream mBufferOutStream; 82462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 833a9ca1f0d6bd8f12c2bb2adea51f95c255996180Zonr Chang void CreateFunctionPasses(); 843a9ca1f0d6bd8f12c2bb2adea51f95c255996180Zonr Chang void CreateModulePasses(); 859ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao bool CreateCodeGenPasses(); 869ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao 878024ed54c23c08534434da14d3be99c3efcc5754Jean-Luc Brouillet RSContext *mContext; 888024ed54c23c08534434da14d3be99c3efcc5754Jean-Luc Brouillet 898024ed54c23c08534434da14d3be99c3efcc5754Jean-Luc Brouillet clang::SourceManager &mSourceMgr; 908024ed54c23c08534434da14d3be99c3efcc5754Jean-Luc Brouillet 912770d0e31ef3b14cd51ca07273240ad0995dc5cdDavid Gross bool mASTPrint; 922770d0e31ef3b14cd51ca07273240ad0995dc5cdDavid Gross 938024ed54c23c08534434da14d3be99c3efcc5754Jean-Luc Brouillet bool mAllowRSPrefix; 948024ed54c23c08534434da14d3be99c3efcc5754Jean-Luc Brouillet 958024ed54c23c08534434da14d3be99c3efcc5754Jean-Luc Brouillet bool mIsFilterscript; 968024ed54c23c08534434da14d3be99c3efcc5754Jean-Luc Brouillet 978024ed54c23c08534434da14d3be99c3efcc5754Jean-Luc Brouillet llvm::NamedMDNode *mExportVarMetadata; 988024ed54c23c08534434da14d3be99c3efcc5754Jean-Luc Brouillet llvm::NamedMDNode *mExportFuncMetadata; 998024ed54c23c08534434da14d3be99c3efcc5754Jean-Luc Brouillet llvm::NamedMDNode *mExportForEachNameMetadata; 1008024ed54c23c08534434da14d3be99c3efcc5754Jean-Luc Brouillet llvm::NamedMDNode *mExportForEachSignatureMetadata; 1018ee018bdb53fc5abdc430878e2bec0c0c0aa560fDavid Gross llvm::NamedMDNode *mExportReduceMetadata; 1028024ed54c23c08534434da14d3be99c3efcc5754Jean-Luc Brouillet llvm::NamedMDNode *mExportTypeMetadata; 1038024ed54c23c08534434da14d3be99c3efcc5754Jean-Luc Brouillet llvm::NamedMDNode *mRSObjectSlotsMetadata; 1048024ed54c23c08534434da14d3be99c3efcc5754Jean-Luc Brouillet 1058024ed54c23c08534434da14d3be99c3efcc5754Jean-Luc Brouillet RSObjectRefCount mRefCount; 1068024ed54c23c08534434da14d3be99c3efcc5754Jean-Luc Brouillet 1078024ed54c23c08534434da14d3be99c3efcc5754Jean-Luc Brouillet RSCheckAST mASTChecker; 1088024ed54c23c08534434da14d3be99c3efcc5754Jean-Luc Brouillet 109fb40ee2a90f37967bf4a40a18dec7f60e5c580d8Yang Ni RSForEachLowering mForEachHandler; 110fb40ee2a90f37967bf4a40a18dec7f60e5c580d8Yang Ni 1118024ed54c23c08534434da14d3be99c3efcc5754Jean-Luc Brouillet void AnnotateFunction(clang::FunctionDecl *FD); 1128024ed54c23c08534434da14d3be99c3efcc5754Jean-Luc Brouillet 1138024ed54c23c08534434da14d3be99c3efcc5754Jean-Luc Brouillet void dumpExportVarInfo(llvm::Module *M); 1148024ed54c23c08534434da14d3be99c3efcc5754Jean-Luc Brouillet void dumpExportFunctionInfo(llvm::Module *M); 1158024ed54c23c08534434da14d3be99c3efcc5754Jean-Luc Brouillet void dumpExportForEachInfo(llvm::Module *M); 1168ee018bdb53fc5abdc430878e2bec0c0c0aa560fDavid Gross void dumpExportReduceInfo(llvm::Module *M); 1178024ed54c23c08534434da14d3be99c3efcc5754Jean-Luc Brouillet void dumpExportTypeInfo(llvm::Module *M); 1188024ed54c23c08534434da14d3be99c3efcc5754Jean-Luc Brouillet 1199319dfc974a82794d46e9f474f316590f480b976Yang Ni // Translates any rsForEach() or rsForEachWithOptions() calls inside the body 1209319dfc974a82794d46e9f474f316590f480b976Yang Ni // of FD to lower-level runtime calls to rsForEachInternal(), if FD is not a 1219319dfc974a82794d46e9f474f316590f480b976Yang Ni // kernel function itself, as indicated by isKernel being false. If isKernel 1229319dfc974a82794d46e9f474f316590f480b976Yang Ni // is true, reports an error on any calls to rsForEach() or 1239319dfc974a82794d46e9f474f316590f480b976Yang Ni // rsForEachWithOptions(). 1249319dfc974a82794d46e9f474f316590f480b976Yang Ni void LowerRSForEachCall(clang::FunctionDecl* FD, bool isKernel); 125fb40ee2a90f37967bf4a40a18dec7f60e5c580d8Yang Ni 1269ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao protected: 1279ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao llvm::LLVMContext &mLLVMContext; 1289207a2e495c8363606861e4f034504ec5c153dabLogan Chien clang::DiagnosticsEngine &mDiagEngine; 129c460b37ffb50819a32c2a8967754b6f784b28263mkopec const clang::CodeGenOptions &mCodeGenOpts; 1309ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao 1313fd0a94a5cf1656569b1aea07043cc63939dcb46Stephen Hines PragmaList *mPragmas; 1329ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao 1338024ed54c23c08534434da14d3be99c3efcc5754Jean-Luc Brouillet unsigned int getTargetAPI() const { return mContext->getTargetAPI(); } 1348024ed54c23c08534434da14d3be99c3efcc5754Jean-Luc Brouillet 1358024ed54c23c08534434da14d3be99c3efcc5754Jean-Luc Brouillet // TODO These are no longer virtual from base. Look into merging into caller. 1364cc499d6e5ec602309501873449c938af61170b2Stephen Hines 1378785d056b8912325b7e2c9c6c7adbd13b90964c4Zonr Chang // This handler will be invoked before Clang translates @Ctx to LLVM IR. This 13868fc02ca4a7235e2981be5eee4ad968a9d3928c0Zonr Chang // give you an opportunity to modified the IR in AST level (scope information, 13968fc02ca4a7235e2981be5eee4ad968a9d3928c0Zonr Chang // unoptimized IR, etc.). After the return from this method, slang will start 14068fc02ca4a7235e2981be5eee4ad968a9d3928c0Zonr Chang // translate @Ctx into LLVM IR. One should not operate on @Ctx afterwards 14168fc02ca4a7235e2981be5eee4ad968a9d3928c0Zonr Chang // since the changes applied on that never reflects to the LLVM module used 14268fc02ca4a7235e2981be5eee4ad968a9d3928c0Zonr Chang // in the final codegen. 1438024ed54c23c08534434da14d3be99c3efcc5754Jean-Luc Brouillet void HandleTranslationUnitPre(clang::ASTContext &Ctx); 14468fc02ca4a7235e2981be5eee4ad968a9d3928c0Zonr Chang 14568fc02ca4a7235e2981be5eee4ad968a9d3928c0Zonr Chang // This handler will be invoked when Clang have converted AST tree to LLVM IR. 14668fc02ca4a7235e2981be5eee4ad968a9d3928c0Zonr Chang // The @M contains the resulting LLVM IR tree. After the return from this 14768fc02ca4a7235e2981be5eee4ad968a9d3928c0Zonr Chang // method, slang will start doing optimization and code generation for @M. 1488024ed54c23c08534434da14d3be99c3efcc5754Jean-Luc Brouillet void HandleTranslationUnitPost(llvm::Module *M); 1499ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao 1509ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao public: 1518024ed54c23c08534434da14d3be99c3efcc5754Jean-Luc Brouillet Backend(RSContext *Context, 1528024ed54c23c08534434da14d3be99c3efcc5754Jean-Luc Brouillet clang::DiagnosticsEngine *DiagEngine, 1532770d0e31ef3b14cd51ca07273240ad0995dc5cdDavid Gross const RSCCOptions &Opts, 1548f093e05e28046b6fc74175b66a06152f72e0c66Pirama Arumuga Nainar const clang::HeaderSearchOptions &HeaderSearchOpts, 1558f093e05e28046b6fc74175b66a06152f72e0c66Pirama Arumuga Nainar const clang::PreprocessorOptions &PreprocessorOpts, 1568024ed54c23c08534434da14d3be99c3efcc5754Jean-Luc Brouillet const clang::CodeGenOptions &CodeGenOpts, 1578024ed54c23c08534434da14d3be99c3efcc5754Jean-Luc Brouillet const clang::TargetOptions &TargetOpts, 1588024ed54c23c08534434da14d3be99c3efcc5754Jean-Luc Brouillet PragmaList *Pragmas, 1598024ed54c23c08534434da14d3be99c3efcc5754Jean-Luc Brouillet llvm::raw_ostream *OS, 1608024ed54c23c08534434da14d3be99c3efcc5754Jean-Luc Brouillet Slang::OutputType OT, 1618024ed54c23c08534434da14d3be99c3efcc5754Jean-Luc Brouillet clang::SourceManager &SourceMgr, 1628024ed54c23c08534434da14d3be99c3efcc5754Jean-Luc Brouillet bool AllowRSPrefix, 1638024ed54c23c08534434da14d3be99c3efcc5754Jean-Luc Brouillet bool IsFilterscript); 1648024ed54c23c08534434da14d3be99c3efcc5754Jean-Luc Brouillet 1658024ed54c23c08534434da14d3be99c3efcc5754Jean-Luc Brouillet virtual ~Backend(); 1669ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao 1679ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao // Initialize - This is called to initialize the consumer, providing the 1689ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao // ASTContext. 1698024ed54c23c08534434da14d3be99c3efcc5754Jean-Luc Brouillet void Initialize(clang::ASTContext &Ctx) override; 1708024ed54c23c08534434da14d3be99c3efcc5754Jean-Luc Brouillet 1718024ed54c23c08534434da14d3be99c3efcc5754Jean-Luc Brouillet // TODO Clean up what should be private, protected 1728024ed54c23c08534434da14d3be99c3efcc5754Jean-Luc Brouillet // TODO Also clean up the include files 1739ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao 1749ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao // HandleTopLevelDecl - Handle the specified top-level declaration. This is 1759ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao // called by the parser to process every top-level Decl*. Note that D can be 1769ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao // the head of a chain of Decls (e.g. for `int a, b` the chain will have two 1779ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao // elements). Use Decl::getNextDeclarator() to walk the chain. 1788024ed54c23c08534434da14d3be99c3efcc5754Jean-Luc Brouillet bool HandleTopLevelDecl(clang::DeclGroupRef D) override; 1799ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao 1809ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao // HandleTranslationUnit - This method is called when the ASTs for entire 1819ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao // translation unit have been parsed. 1828024ed54c23c08534434da14d3be99c3efcc5754Jean-Luc Brouillet void HandleTranslationUnit(clang::ASTContext &Ctx) override; 1839ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao 1849ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao // HandleTagDeclDefinition - This callback is invoked each time a TagDecl 1859ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao // (e.g. struct, union, enum, class) is completed. This allows the client to 1869ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao // hack on the type, which can occur at any point in the file (because these 1879ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao // can be defined in declspecs). 1888024ed54c23c08534434da14d3be99c3efcc5754Jean-Luc Brouillet void HandleTagDeclDefinition(clang::TagDecl *D) override; 1899ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao 1909ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao // CompleteTentativeDefinition - Callback invoked at the end of a translation 1919ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao // unit to notify the consumer that the given tentative definition should be 1929ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao // completed. 1938024ed54c23c08534434da14d3be99c3efcc5754Jean-Luc Brouillet void CompleteTentativeDefinition(clang::VarDecl *D) override; 194462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao}; 195462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 1969ef2f785e0cc490af678dfd685995dec787321ffShih-wei Liao} // namespace slang 197462aefd62cc646d2ff753c1d003ef3cd7bbea26Shih-wei Liao 198e639eb5caa2c386b4a60659a4929e8a6141a2cbeStephen Hines#endif // _FRAMEWORKS_COMPILE_SLANG_SLANG_BACKEND_H_ NOLINT 199