zonr | 6315f76 | 2010-10-05 15:35:14 +0800 | [diff] [blame] | 1 | #ifndef _SLANG_COMPILER_RS_CONTEXT_H |
| 2 | #define _SLANG_COMPILER_RS_CONTEXT_H |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 3 | |
| 4 | #include <map> |
| 5 | #include <list> |
| 6 | #include <string> |
| 7 | #include <cstdio> |
| 8 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 9 | #include "llvm/ADT/StringSet.h" |
| 10 | #include "llvm/ADT/StringMap.h" |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 11 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 12 | #include "clang/Lex/Preprocessor.h" |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 13 | |
| 14 | namespace llvm { |
zonr | 6315f76 | 2010-10-05 15:35:14 +0800 | [diff] [blame] | 15 | class LLVMContext; |
| 16 | class TargetData; |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 17 | } // namespace llvm |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 18 | |
| 19 | namespace clang { |
zonr | 6315f76 | 2010-10-05 15:35:14 +0800 | [diff] [blame] | 20 | class VarDecl; |
| 21 | class ASTContext; |
| 22 | class TargetInfo; |
| 23 | class FunctionDecl; |
| 24 | class SourceManager; |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 25 | } // namespace clang |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 26 | |
| 27 | namespace slang { |
Zonr Chang | a41ce1d | 2010-10-06 02:23:12 +0800 | [diff] [blame] | 28 | class RSExportable; |
| 29 | class RSExportVar; |
| 30 | class RSExportFunc; |
| 31 | class RSExportType; |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 32 | |
| 33 | class RSContext { |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 34 | typedef llvm::StringSet<> NeedExportVarSet; |
| 35 | typedef llvm::StringSet<> NeedExportFuncSet; |
| 36 | typedef llvm::StringSet<> NeedExportTypeSet; |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 37 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 38 | public: |
Zonr Chang | a41ce1d | 2010-10-06 02:23:12 +0800 | [diff] [blame] | 39 | typedef std::list<RSExportable*> ExportableList; |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 40 | typedef std::list<RSExportVar*> ExportVarList; |
| 41 | typedef std::list<RSExportFunc*> ExportFuncList; |
| 42 | typedef llvm::StringMap<RSExportType*> ExportTypeMap; |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 43 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 44 | private: |
| 45 | clang::Preprocessor *mPP; |
| 46 | clang::ASTContext *mCtx; |
| 47 | const clang::TargetInfo *mTarget; |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 48 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 49 | llvm::TargetData *mTargetData; |
| 50 | llvm::LLVMContext &mLLVMContext; |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 51 | |
Zonr Chang | a41ce1d | 2010-10-06 02:23:12 +0800 | [diff] [blame] | 52 | ExportableList mExportables; |
| 53 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 54 | // Record the variables/types/elements annotated in #pragma to be exported |
| 55 | NeedExportVarSet mNeedExportVars; |
| 56 | NeedExportFuncSet mNeedExportFuncs; |
| 57 | NeedExportTypeSet mNeedExportTypes; |
| 58 | bool mExportAllNonStaticVars; |
| 59 | bool mExportAllNonStaticFuncs; |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 60 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 61 | std::string *mLicenseNote; |
| 62 | std::string mReflectJavaPackageName; |
| 63 | std::string mReflectJavaPathName; |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 64 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 65 | bool processExportVar(const clang::VarDecl *VD); |
| 66 | bool processExportFunc(const clang::FunctionDecl *FD); |
| 67 | bool processExportType(const llvm::StringRef &Name); |
Shih-wei Liao | 537446c | 2010-06-11 16:05:55 -0700 | [diff] [blame] | 68 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 69 | ExportVarList mExportVars; |
| 70 | ExportFuncList mExportFuncs; |
| 71 | ExportTypeMap mExportTypes; |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 72 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 73 | public: |
| 74 | RSContext(clang::Preprocessor *PP, |
| 75 | clang::ASTContext *Ctx, |
| 76 | const clang::TargetInfo *Target); |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 77 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 78 | inline clang::Preprocessor *getPreprocessor() const { return mPP; } |
| 79 | inline clang::ASTContext *getASTContext() const { return mCtx; } |
| 80 | inline const llvm::TargetData *getTargetData() const { return mTargetData; } |
| 81 | inline llvm::LLVMContext &getLLVMContext() const { return mLLVMContext; } |
| 82 | inline const clang::SourceManager *getSourceManager() const { |
| 83 | return &mPP->getSourceManager(); |
| 84 | } |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 85 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 86 | inline void setLicenseNote(const std::string &S) { |
| 87 | mLicenseNote = new std::string(S); |
| 88 | } |
| 89 | inline const std::string *getLicenseNote() const { return mLicenseNote; } |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 90 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 91 | inline void addExportVar(const std::string &S) { |
| 92 | mNeedExportVars.insert(S); |
| 93 | return; |
| 94 | } |
| 95 | inline void addExportFunc(const std::string &S) { |
| 96 | mNeedExportFuncs.insert(S); |
| 97 | return; |
| 98 | } |
| 99 | inline void addExportType(const std::string &S) { |
| 100 | mNeedExportTypes.insert(S); |
| 101 | return; |
| 102 | } |
Victor Hsieh | d8a0d18 | 2010-07-07 19:22:33 +0800 | [diff] [blame] | 103 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 104 | inline void setExportAllNonStaticVars(bool flag) { |
| 105 | mExportAllNonStaticVars = flag; |
| 106 | } |
| 107 | inline void setExportAllNonStaticFuncs(bool flag) { |
| 108 | mExportAllNonStaticFuncs = flag; |
| 109 | } |
| 110 | inline void setReflectJavaPackageName(const std::string &S) { |
| 111 | mReflectJavaPackageName = S; |
| 112 | return; |
| 113 | } |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 114 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 115 | void processExport(); |
Zonr Chang | a41ce1d | 2010-10-06 02:23:12 +0800 | [diff] [blame] | 116 | inline void newExportable(RSExportable *E) { |
| 117 | if (E != NULL) |
| 118 | mExportables.push_back(E); |
| 119 | } |
Shih-wei Liao | 537446c | 2010-06-11 16:05:55 -0700 | [diff] [blame] | 120 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 121 | typedef ExportVarList::const_iterator const_export_var_iterator; |
| 122 | const_export_var_iterator export_vars_begin() const { |
| 123 | return mExportVars.begin(); |
| 124 | } |
| 125 | const_export_var_iterator export_vars_end() const { |
| 126 | return mExportVars.end(); |
| 127 | } |
| 128 | inline bool hasExportVar() const { |
| 129 | return !mExportVars.empty(); |
| 130 | } |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 131 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 132 | typedef ExportFuncList::const_iterator const_export_func_iterator; |
| 133 | const_export_func_iterator export_funcs_begin() const { |
| 134 | return mExportFuncs.begin(); |
| 135 | } |
| 136 | const_export_func_iterator export_funcs_end() const { |
| 137 | return mExportFuncs.end(); |
| 138 | } |
| 139 | inline bool hasExportFunc() const { return !mExportFuncs.empty(); } |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 140 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 141 | typedef ExportTypeMap::iterator export_type_iterator; |
| 142 | typedef ExportTypeMap::const_iterator const_export_type_iterator; |
| 143 | export_type_iterator export_types_begin() { return mExportTypes.begin(); } |
| 144 | export_type_iterator export_types_end() { return mExportTypes.end(); } |
| 145 | const_export_type_iterator export_types_begin() const { |
| 146 | return mExportTypes.begin(); |
| 147 | } |
| 148 | const_export_type_iterator export_types_end() const { |
| 149 | return mExportTypes.end(); |
| 150 | } |
| 151 | inline bool hasExportType() const { return !mExportTypes.empty(); } |
| 152 | export_type_iterator findExportType(const llvm::StringRef &TypeName) { |
| 153 | return mExportTypes.find(TypeName); |
| 154 | } |
| 155 | const_export_type_iterator findExportType(const llvm::StringRef &TypeName) |
| 156 | const { |
| 157 | return mExportTypes.find(TypeName); |
| 158 | } |
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 | // Insert the specified Typename/Type pair into the map. If the key already |
| 161 | // exists in the map, return false and ignore the request, otherwise insert it |
| 162 | // and return true. |
| 163 | bool insertExportType(const llvm::StringRef &TypeName, RSExportType *Type); |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 164 | |
Shih-wei Liao | b81c6a4 | 2010-10-10 14:15:00 -0700 | [diff] [blame^] | 165 | bool reflectToJava(const std::string &OutputPathBase, |
| 166 | const std::string &OutputPackageName, |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 167 | const std::string &InputFileName, |
| 168 | const std::string &OutputBCFileName, |
Shih-wei Liao | b81c6a4 | 2010-10-10 14:15:00 -0700 | [diff] [blame^] | 169 | std::string *RealPackageName); |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 170 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 171 | ~RSContext(); |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 172 | }; |
| 173 | |
Shih-wei Liao | 9ef2f78 | 2010-10-01 12:31:37 -0700 | [diff] [blame] | 174 | } // namespace slang |
Shih-wei Liao | 462aefd | 2010-06-04 15:32:04 -0700 | [diff] [blame] | 175 | |
zonr | 6315f76 | 2010-10-05 15:35:14 +0800 | [diff] [blame] | 176 | #endif // _SLANG_COMPILER_RS_CONTEXT_H |