Logan | 1f028c0 | 2010-11-27 01:02:48 +0800 | [diff] [blame] | 1 | /* |
Stephen Hines | db16918 | 2012-01-05 18:46:36 -0800 | [diff] [blame] | 2 | * Copyright 2010-2012, The Android Open Source Project |
Logan | 1f028c0 | 2010-11-27 01:02:48 +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 Chang | c72c4dd | 2012-04-12 15:38:53 +0800 | [diff] [blame^] | 17 | #ifndef BCC_COMPILER_H |
| 18 | #define BCC_COMPILER_H |
Logan | 1f028c0 | 2010-11-27 01:02:48 +0800 | [diff] [blame] | 19 | |
Logan | 1f028c0 | 2010-11-27 01:02:48 +0800 | [diff] [blame] | 20 | namespace llvm { |
Logan | 1f028c0 | 2010-11-27 01:02:48 +0800 | [diff] [blame] | 21 | |
Zonr Chang | ade9277 | 2012-04-13 15:58:24 +0800 | [diff] [blame] | 22 | class raw_ostream; |
| 23 | class PassManager; |
| 24 | class TargetData; |
| 25 | class TargetMachine; |
| 26 | |
| 27 | } // end namespace llvm |
Logan | 1f028c0 | 2010-11-27 01:02:48 +0800 | [diff] [blame] | 28 | |
| 29 | namespace bcc { |
| 30 | |
Zonr Chang | ade9277 | 2012-04-13 15:58:24 +0800 | [diff] [blame] | 31 | class CompilerConfig; |
| 32 | class OutputFile; |
| 33 | class Script; |
Logan | 1f028c0 | 2010-11-27 01:02:48 +0800 | [diff] [blame] | 34 | |
Zonr Chang | ade9277 | 2012-04-13 15:58:24 +0800 | [diff] [blame] | 35 | //===----------------------------------------------------------------------===// |
| 36 | // Design of Compiler |
| 37 | //===----------------------------------------------------------------------===// |
| 38 | // 1. A compiler instance can be constructed provided an "initial config." |
| 39 | // 2. A compiler can later be re-configured using config(). |
| 40 | // 3. Once config() is invoked, it'll re-create TargetMachine instance (i.e., |
| 41 | // mTarget) according to the configuration supplied. TargetMachine instance |
| 42 | // is *shared* across the different calls to compile() before the next call |
| 43 | // to config(). |
| 44 | // 4. Once a compiler instance is created, you can use the compile() service |
| 45 | // to compile the file over and over again. Each call uses TargetMachine |
| 46 | // instance to construct the compilation passes. |
| 47 | class Compiler { |
| 48 | public: |
| 49 | enum ErrorCode { |
| 50 | kSuccess, |
Logan | 1f028c0 | 2010-11-27 01:02:48 +0800 | [diff] [blame] | 51 | |
Zonr Chang | ade9277 | 2012-04-13 15:58:24 +0800 | [diff] [blame] | 52 | kInvalidConfigNoTarget, |
| 53 | kErrCreateTargetMachine, |
| 54 | kErrSwitchTargetMachine, |
| 55 | kErrNoTargetMachine, |
| 56 | kErrTargetDataNoMemory, |
| 57 | kErrMaterialization, |
| 58 | kErrInvalidOutputFileState, |
| 59 | kErrPrepareOutput, |
| 60 | kPrepareCodeGenPass, |
Logan | 1f028c0 | 2010-11-27 01:02:48 +0800 | [diff] [blame] | 61 | |
Zonr Chang | ade9277 | 2012-04-13 15:58:24 +0800 | [diff] [blame] | 62 | kErrHookBeforeAddLTOPasses, |
| 63 | kErrHookAfterAddLTOPasses, |
| 64 | kErrHookBeforeExecuteLTOPasses, |
| 65 | kErrHookAfterExecuteLTOPasses, |
Logan | 1f028c0 | 2010-11-27 01:02:48 +0800 | [diff] [blame] | 66 | |
Zonr Chang | ade9277 | 2012-04-13 15:58:24 +0800 | [diff] [blame] | 67 | kErrHookBeforeAddCodeGenPasses, |
| 68 | kErrHookAfterAddCodeGenPasses, |
| 69 | kErrHookBeforeExecuteCodeGenPasses, |
| 70 | kErrHookAfterExecuteCodeGenPasses, |
Logan | 1f028c0 | 2010-11-27 01:02:48 +0800 | [diff] [blame] | 71 | |
Zonr Chang | ade9277 | 2012-04-13 15:58:24 +0800 | [diff] [blame] | 72 | kMaxErrorCode, |
| 73 | }; |
Logan | 1f028c0 | 2010-11-27 01:02:48 +0800 | [diff] [blame] | 74 | |
Zonr Chang | ade9277 | 2012-04-13 15:58:24 +0800 | [diff] [blame] | 75 | static const char *GetErrorString(enum ErrorCode pErrCode); |
Logan | 1f028c0 | 2010-11-27 01:02:48 +0800 | [diff] [blame] | 76 | |
Zonr Chang | ade9277 | 2012-04-13 15:58:24 +0800 | [diff] [blame] | 77 | private: |
| 78 | llvm::TargetMachine *mTarget; |
| 79 | // LTO is enabled by default. |
| 80 | bool mEnableLTO; |
Logan | 2a6dc82 | 2011-01-06 04:05:20 +0800 | [diff] [blame] | 81 | |
Zonr Chang | ade9277 | 2012-04-13 15:58:24 +0800 | [diff] [blame] | 82 | enum ErrorCode runLTO(Script &pScript); |
| 83 | enum ErrorCode runCodeGen(Script &pScript, llvm::raw_ostream &pResult); |
Logan | 1f028c0 | 2010-11-27 01:02:48 +0800 | [diff] [blame] | 84 | |
Zonr Chang | ade9277 | 2012-04-13 15:58:24 +0800 | [diff] [blame] | 85 | public: |
| 86 | Compiler(); |
| 87 | Compiler(const CompilerConfig &pConfig); |
Logan Chien | da5e0c3 | 2011-06-13 03:47:21 +0800 | [diff] [blame] | 88 | |
Zonr Chang | ade9277 | 2012-04-13 15:58:24 +0800 | [diff] [blame] | 89 | enum ErrorCode config(const CompilerConfig &pConfig); |
Logan | 1f028c0 | 2010-11-27 01:02:48 +0800 | [diff] [blame] | 90 | |
Zonr Chang | ade9277 | 2012-04-13 15:58:24 +0800 | [diff] [blame] | 91 | // Compile a script and output the result to a LLVM stream. |
| 92 | enum ErrorCode compile(Script &pScript, llvm::raw_ostream &pResult); |
Logan | 1f028c0 | 2010-11-27 01:02:48 +0800 | [diff] [blame] | 93 | |
Zonr Chang | ade9277 | 2012-04-13 15:58:24 +0800 | [diff] [blame] | 94 | // Compile a script and output the result to a file. |
| 95 | enum ErrorCode compile(Script &pScript, OutputFile &pResult); |
Logan | 1f028c0 | 2010-11-27 01:02:48 +0800 | [diff] [blame] | 96 | |
Zonr Chang | ade9277 | 2012-04-13 15:58:24 +0800 | [diff] [blame] | 97 | void enableLTO(bool pEnable = true) |
| 98 | { mEnableLTO = pEnable; } |
Logan | 1f028c0 | 2010-11-27 01:02:48 +0800 | [diff] [blame] | 99 | |
Zonr Chang | ade9277 | 2012-04-13 15:58:24 +0800 | [diff] [blame] | 100 | virtual ~Compiler(); |
Logan | ecf4cbd | 2011-01-06 05:34:11 +0800 | [diff] [blame] | 101 | |
Zonr Chang | ade9277 | 2012-04-13 15:58:24 +0800 | [diff] [blame] | 102 | protected: |
| 103 | //===--------------------------------------------------------------------===// |
| 104 | // Plugin callbacks for sub-class. |
| 105 | //===--------------------------------------------------------------------===// |
| 106 | // Called before adding first pass to code-generation passes. |
| 107 | virtual bool beforeAddLTOPasses(Script &pScript, llvm::PassManager &pPM) |
| 108 | { return true; } |
Logan Chien | 9347e0b | 2011-07-07 19:51:47 +0800 | [diff] [blame] | 109 | |
Zonr Chang | ade9277 | 2012-04-13 15:58:24 +0800 | [diff] [blame] | 110 | // Called after adding last pass to code-generation passes. |
| 111 | virtual bool afterAddLTOPasses(Script &pScript, llvm::PassManager &pPM) |
| 112 | { return true; } |
Shih-wei Liao | 4deffde | 2012-01-17 20:38:17 +0800 | [diff] [blame] | 113 | |
Zonr Chang | ade9277 | 2012-04-13 15:58:24 +0800 | [diff] [blame] | 114 | // Called before executing code-generation passes. |
| 115 | virtual bool beforeExecuteLTOPasses(Script &pScript, |
| 116 | llvm::PassManager &pPM) |
| 117 | { return true; } |
Logan | 1f028c0 | 2010-11-27 01:02:48 +0800 | [diff] [blame] | 118 | |
Zonr Chang | ade9277 | 2012-04-13 15:58:24 +0800 | [diff] [blame] | 119 | // Called after executing code-generation passes. |
| 120 | virtual bool afterExecuteLTOPasses(Script &pScript) |
| 121 | { return true; } |
Shih-wei Liao | 5e3e0ce | 2011-06-17 13:59:46 -0700 | [diff] [blame] | 122 | |
Zonr Chang | ade9277 | 2012-04-13 15:58:24 +0800 | [diff] [blame] | 123 | // Called before adding first pass to code-generation passes. |
| 124 | virtual bool beforeAddCodeGenPasses(Script &pScript, llvm::PassManager &pPM) |
| 125 | { return true; } |
Logan Chien | da5e0c3 | 2011-06-13 03:47:21 +0800 | [diff] [blame] | 126 | |
Zonr Chang | ade9277 | 2012-04-13 15:58:24 +0800 | [diff] [blame] | 127 | // Called after adding last pass to code-generation passes. |
| 128 | virtual bool afterAddCodeGenPasses(Script &pScript, llvm::PassManager &pPM) |
| 129 | { return true; } |
Logan | 1f028c0 | 2010-11-27 01:02:48 +0800 | [diff] [blame] | 130 | |
Zonr Chang | ade9277 | 2012-04-13 15:58:24 +0800 | [diff] [blame] | 131 | // Called before executing code-generation passes. |
| 132 | virtual bool beforeExecuteCodeGenPasses(Script &pScript, |
| 133 | llvm::PassManager &pPM) |
| 134 | { return true; } |
Logan | 1f028c0 | 2010-11-27 01:02:48 +0800 | [diff] [blame] | 135 | |
Zonr Chang | ade9277 | 2012-04-13 15:58:24 +0800 | [diff] [blame] | 136 | // Called after executing code-generation passes. |
| 137 | virtual bool afterExecuteCodeGenPasses(Script &pScript) |
| 138 | { return true; } |
| 139 | }; |
Logan | 1f028c0 | 2010-11-27 01:02:48 +0800 | [diff] [blame] | 140 | |
Zonr Chang | ade9277 | 2012-04-13 15:58:24 +0800 | [diff] [blame] | 141 | } // end namespace bcc |
Logan | 1f028c0 | 2010-11-27 01:02:48 +0800 | [diff] [blame] | 142 | |
Zonr Chang | c72c4dd | 2012-04-12 15:38:53 +0800 | [diff] [blame^] | 143 | #endif // BCC_COMPILER_H |