Jason Sams | 1b6a088 | 2012-03-12 15:07:58 -0700 | [diff] [blame] | 1 | /* |
| 2 | * Copyright 2012, The Android Open Source Project |
| 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 | |
Jean-Luc Brouillet | efcff10 | 2014-06-03 16:13:51 -0700 | [diff] [blame] | 17 | #ifndef _FRAMEWORKS_COMPILE_SLANG_SLANG_RS_REFLECTION_CPP_H_ // NOLINT |
Jason Sams | 1b6a088 | 2012-03-12 15:07:58 -0700 | [diff] [blame] | 18 | #define _FRAMEWORKS_COMPILE_SLANG_SLANG_RS_REFLECTION_CPP_H_ |
| 19 | |
Jean-Luc Brouillet | 1f9d121 | 2014-06-02 20:27:29 -0700 | [diff] [blame] | 20 | #include "slang_rs_reflect_utils.h" |
Jason Sams | 1b6a088 | 2012-03-12 15:07:58 -0700 | [diff] [blame] | 21 | |
Stephen Hines | 003ac66 | 2013-08-21 00:37:51 -0700 | [diff] [blame] | 22 | #include <set> |
| 23 | #include <string> |
| 24 | |
Jean-Luc Brouillet | 2ce118e | 2014-05-27 17:41:22 -0700 | [diff] [blame] | 25 | #define RS_EXPORT_VAR_PREFIX "mExportVar_" |
Stephen Hines | 8070683 | 2013-08-28 18:08:57 -0700 | [diff] [blame] | 26 | |
Jason Sams | 1b6a088 | 2012-03-12 15:07:58 -0700 | [diff] [blame] | 27 | namespace slang { |
| 28 | |
Jean-Luc Brouillet | efcff10 | 2014-06-03 16:13:51 -0700 | [diff] [blame] | 29 | class RSReflectionCpp { |
| 30 | public: |
Jean-Luc Brouillet | 59f22c3 | 2014-06-04 14:53:48 -0700 | [diff] [blame] | 31 | RSReflectionCpp(const RSContext *Context, const std::string &OutputDirectory, |
| 32 | const std::string &RSSourceFileName, |
| 33 | const std::string &BitCodeFileName); |
Stephen Hines | 02a9826 | 2012-11-14 12:40:26 -0800 | [diff] [blame] | 34 | virtual ~RSReflectionCpp(); |
| 35 | |
Jean-Luc Brouillet | 59f22c3 | 2014-06-04 14:53:48 -0700 | [diff] [blame] | 36 | bool reflect(); |
Jason Sams | 1b6a088 | 2012-03-12 15:07:58 -0700 | [diff] [blame] | 37 | |
Jean-Luc Brouillet | efcff10 | 2014-06-03 16:13:51 -0700 | [diff] [blame] | 38 | private: |
Matt Wala | 1c6b927 | 2015-08-03 14:06:07 -0700 | [diff] [blame] | 39 | struct Argument { |
| 40 | std::string Type; |
| 41 | std::string Name; |
| 42 | std::string DefaultValue; |
| 43 | Argument(std::string Type, std::string Name, std::string DefaultValue = "") |
| 44 | : Type(Type), Name(Name), DefaultValue(DefaultValue) {} |
| 45 | }; |
| 46 | typedef std::vector<Argument> ArgumentList; |
Jean-Luc Brouillet | efcff10 | 2014-06-03 16:13:51 -0700 | [diff] [blame] | 47 | |
| 48 | // Information coming from the compiler about the code we're reflecting. |
| 49 | const RSContext *mRSContext; |
| 50 | |
| 51 | // Path to the *.rs file for which we're generating C++ code. |
| 52 | std::string mRSSourceFilePath; |
| 53 | // Path to the file that contains the byte code generated from the *.rs file. |
| 54 | std::string mBitCodeFilePath; |
| 55 | // The directory where we'll generate the C++ files. |
| 56 | std::string mOutputDirectory; |
| 57 | // A cleaned up version of the *.rs file name that can be used in generating |
| 58 | // C++ identifiers. |
| 59 | std::string mCleanedRSFileName; |
| 60 | // The name of the generated C++ class. |
| 61 | std::string mClassName; |
| 62 | |
| 63 | // TODO document |
Stephen Hines | 7dd6da2 | 2012-11-15 19:56:03 -0800 | [diff] [blame] | 64 | unsigned int mNextExportVarSlot; |
| 65 | unsigned int mNextExportFuncSlot; |
| 66 | unsigned int mNextExportForEachSlot; |
| 67 | |
Jean-Luc Brouillet | efcff10 | 2014-06-03 16:13:51 -0700 | [diff] [blame] | 68 | // Generated RS Elements for type-checking code. |
| 69 | std::set<std::string> mTypesToCheck; |
| 70 | |
Stephen Hines | 7dd6da2 | 2012-11-15 19:56:03 -0800 | [diff] [blame] | 71 | inline void clear() { |
| 72 | mNextExportVarSlot = 0; |
| 73 | mNextExportFuncSlot = 0; |
| 74 | mNextExportForEachSlot = 0; |
Stephen Hines | 003ac66 | 2013-08-21 00:37:51 -0700 | [diff] [blame] | 75 | mTypesToCheck.clear(); |
Stephen Hines | 7dd6da2 | 2012-11-15 19:56:03 -0800 | [diff] [blame] | 76 | } |
| 77 | |
Jean-Luc Brouillet | 1f9d121 | 2014-06-02 20:27:29 -0700 | [diff] [blame] | 78 | // The file we are currently generating, either the header or the |
| 79 | // implementation file. |
| 80 | GeneratedFile mOut; |
| 81 | |
Jean-Luc Brouillet | efcff10 | 2014-06-03 16:13:51 -0700 | [diff] [blame] | 82 | void genInitValue(const clang::APValue &Val, bool asBool = false); |
| 83 | static const char *getVectorAccessor(unsigned index); |
| 84 | |
Jean-Luc Brouillet | 2ce118e | 2014-05-27 17:41:22 -0700 | [diff] [blame] | 85 | inline unsigned int getNextExportVarSlot() { return mNextExportVarSlot++; } |
Stephen Hines | 7dd6da2 | 2012-11-15 19:56:03 -0800 | [diff] [blame] | 86 | |
Jean-Luc Brouillet | 2ce118e | 2014-05-27 17:41:22 -0700 | [diff] [blame] | 87 | inline unsigned int getNextExportFuncSlot() { return mNextExportFuncSlot++; } |
Stephen Hines | 7dd6da2 | 2012-11-15 19:56:03 -0800 | [diff] [blame] | 88 | |
| 89 | inline unsigned int getNextExportForEachSlot() { |
| 90 | return mNextExportForEachSlot++; |
| 91 | } |
| 92 | |
Jean-Luc Brouillet | eb8b99e | 2014-06-03 20:59:01 -0700 | [diff] [blame] | 93 | bool writeHeaderFile(); |
| 94 | bool writeImplementationFile(); |
Matt Wala | 1c6b927 | 2015-08-03 14:06:07 -0700 | [diff] [blame] | 95 | |
| 96 | // Write out signatures both in the header and implementation. |
Jean-Luc Brouillet | 1f9d121 | 2014-06-02 20:27:29 -0700 | [diff] [blame] | 97 | void makeFunctionSignature(bool isDefinition, const RSExportFunc *ef); |
Matt Wala | 1c6b927 | 2015-08-03 14:06:07 -0700 | [diff] [blame] | 98 | |
Jean-Luc Brouillet | eb8b99e | 2014-06-03 20:59:01 -0700 | [diff] [blame] | 99 | bool genEncodedBitCode(); |
Jean-Luc Brouillet | 1cea271 | 2014-06-05 13:54:11 -0700 | [diff] [blame] | 100 | void genFieldsToStoreExportVariableValues(); |
| 101 | void genTypeInstancesUsedInForEach(); |
| 102 | void genFieldsForAllocationTypeVerification(); |
Matt Wala | 1c6b927 | 2015-08-03 14:06:07 -0700 | [diff] [blame] | 103 | |
| 104 | // Write out the code for the getters and setters. |
Jean-Luc Brouillet | 1cea271 | 2014-06-05 13:54:11 -0700 | [diff] [blame] | 105 | void genExportVariablesGetterAndSetter(); |
Matt Wala | 1c6b927 | 2015-08-03 14:06:07 -0700 | [diff] [blame] | 106 | |
| 107 | // Write out the code for the declaration of the kernel entry points. |
Jean-Luc Brouillet | 1cea271 | 2014-06-05 13:54:11 -0700 | [diff] [blame] | 108 | void genForEachDeclarations(); |
| 109 | void genExportFunctionDeclarations(); |
Jason Sams | 1b6a088 | 2012-03-12 15:07:58 -0700 | [diff] [blame] | 110 | |
Matt Wala | 1c6b927 | 2015-08-03 14:06:07 -0700 | [diff] [blame] | 111 | // Write out code for the definitions of the kernel entry points. |
| 112 | void genExportForEachBodies(); |
Matt Wala | 1c6b927 | 2015-08-03 14:06:07 -0700 | [diff] [blame] | 113 | void genExportFunctionBodies(); |
| 114 | |
Stephen Hines | 02a9826 | 2012-11-14 12:40:26 -0800 | [diff] [blame] | 115 | bool startScriptHeader(); |
Stephen Hines | 7dd6da2 | 2012-11-15 19:56:03 -0800 | [diff] [blame] | 116 | |
Stephen Hines | 8070683 | 2013-08-28 18:08:57 -0700 | [diff] [blame] | 117 | // Write out code for an export variable initialization. |
Jean-Luc Brouillet | 2ce118e | 2014-05-27 17:41:22 -0700 | [diff] [blame] | 118 | void genInitExportVariable(const RSExportType *ET, const std::string &VarName, |
Stephen Hines | 8070683 | 2013-08-28 18:08:57 -0700 | [diff] [blame] | 119 | const clang::APValue &Val); |
| 120 | void genZeroInitExportVariable(const std::string &VarName); |
| 121 | void genInitBoolExportVariable(const std::string &VarName, |
| 122 | const clang::APValue &Val); |
| 123 | void genInitPrimitiveExportVariable(const std::string &VarName, |
| 124 | const clang::APValue &Val); |
| 125 | |
Stephen Hines | 7dd6da2 | 2012-11-15 19:56:03 -0800 | [diff] [blame] | 126 | // Produce an argument string of the form "T1 t, T2 u, T3 v". |
Jean-Luc Brouillet | eb8b99e | 2014-06-03 20:59:01 -0700 | [diff] [blame] | 127 | void genArguments(const ArgumentList &Args, int Offset); |
Stephen Hines | 7dd6da2 | 2012-11-15 19:56:03 -0800 | [diff] [blame] | 128 | |
Stephen Hines | 7dd6da2 | 2012-11-15 19:56:03 -0800 | [diff] [blame] | 129 | void genPointerTypeExportVariable(const RSExportVar *EV); |
Stephen Hines | 7dd6da2 | 2012-11-15 19:56:03 -0800 | [diff] [blame] | 130 | void genMatrixTypeExportVariable(const RSExportVar *EV); |
Stephen Hines | 7dd6da2 | 2012-11-15 19:56:03 -0800 | [diff] [blame] | 131 | void genRecordTypeExportVariable(const RSExportVar *EV); |
| 132 | |
Jean-Luc Brouillet | 1cea271 | 2014-06-05 13:54:11 -0700 | [diff] [blame] | 133 | void genGetterAndSetter(const RSExportPrimitiveType *EPT, const RSExportVar* EV); |
| 134 | void genGetterAndSetter(const RSExportVectorType *EVT, const RSExportVar* EV); |
| 135 | void genGetterAndSetter(const RSExportConstantArrayType *AT, const RSExportVar* EV); |
| 136 | void genGetterAndSetter(const RSExportRecordType *ERT, const RSExportVar *EV); |
| 137 | |
Stephen Hines | 7dd6da2 | 2012-11-15 19:56:03 -0800 | [diff] [blame] | 138 | // Write out a local FieldPacker (if necessary). |
Jean-Luc Brouillet | 2ce118e | 2014-05-27 17:41:22 -0700 | [diff] [blame] | 139 | bool genCreateFieldPacker(const RSExportType *T, const char *FieldPackerName); |
Stephen Hines | 7dd6da2 | 2012-11-15 19:56:03 -0800 | [diff] [blame] | 140 | |
| 141 | // Populate (write) the FieldPacker with add() operations. |
Jean-Luc Brouillet | 2ce118e | 2014-05-27 17:41:22 -0700 | [diff] [blame] | 142 | void genPackVarOfType(const RSExportType *ET, const char *VarName, |
Stephen Hines | 7dd6da2 | 2012-11-15 19:56:03 -0800 | [diff] [blame] | 143 | const char *FieldPackerName); |
Stephen Hines | 003ac66 | 2013-08-21 00:37:51 -0700 | [diff] [blame] | 144 | |
| 145 | // Generate a runtime type check for VarName. |
| 146 | void genTypeCheck(const RSExportType *ET, const char *VarName); |
| 147 | |
Matt Wala | 1c6b927 | 2015-08-03 14:06:07 -0700 | [diff] [blame] | 148 | // Generate a type instance for a given type. |
Stephen Hines | 003ac66 | 2013-08-21 00:37:51 -0700 | [diff] [blame] | 149 | void genTypeInstanceFromPointer(const RSExportType *ET); |
| 150 | void genTypeInstance(const RSExportType *ET); |
| 151 | |
Jean-Luc Brouillet | 2ce118e | 2014-05-27 17:41:22 -0700 | [diff] [blame] | 152 | }; // class RSReflectionCpp |
Jason Sams | 1b6a088 | 2012-03-12 15:07:58 -0700 | [diff] [blame] | 153 | |
Jean-Luc Brouillet | 2ce118e | 2014-05-27 17:41:22 -0700 | [diff] [blame] | 154 | } // namespace slang |
Jason Sams | 1b6a088 | 2012-03-12 15:07:58 -0700 | [diff] [blame] | 155 | |
Jean-Luc Brouillet | 2ce118e | 2014-05-27 17:41:22 -0700 | [diff] [blame] | 156 | #endif // _FRAMEWORKS_COMPILE_SLANG_SLANG_RS_REFLECTION_CPP_H_ NOLINT |