1 /*
2  * Copyright (C) 2011-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 
17 #ifndef RSD_CPU_SCRIPT_H
18 #define RSD_CPU_SCRIPT_H
19 
20 #include <rs_hal.h>
21 #include <rsRuntime.h>
22 
23 #ifndef RS_COMPATIBILITY_LIB
24 #include <utility>
25 #endif
26 
27 #include "rsCpuCore.h"
28 
29 #include <string>
30 
31 namespace bcinfo {
32     class MetadataExtractor;
33 } // namespace bcinfo
34 
35 namespace android {
36 namespace renderscript {
37 
38 class ScriptExecutable;
39 
40 class RsdCpuScriptImpl : public RsdCpuReferenceImpl::CpuScript {
41 public:
42     bool init(char const *resName, char const *cacheDir,
43               uint8_t const *bitcode, size_t bitcodeSize, uint32_t flags,
44               char const *bccPluginName = nullptr);
45     void populateScript(Script *) override;
46 
47     void invokeFunction(uint32_t slot, const void *params, size_t paramLength) override;
48     int invokeRoot() override;
49     virtual void preLaunch(uint32_t slot, const Allocation ** ains,
50                            uint32_t inLen, Allocation * aout, const void * usr,
51                            uint32_t usrLen, const RsScriptCall *sc);
52     virtual void postLaunch(uint32_t slot, const Allocation ** ains,
53                             uint32_t inLen, Allocation * aout,
54                             const void * usr, uint32_t usrLen,
55                             const RsScriptCall *sc);
56 
57     void invokeForEach(uint32_t slot,
58                        const Allocation ** ains,
59                        uint32_t inLen,
60                        Allocation* aout,
61                        const void* usr,
62                        uint32_t usrLen,
63                        const RsScriptCall* sc) override;
64 
65     void invokeReduce(uint32_t slot,
66                       const Allocation ** ains, uint32_t inLen,
67                       Allocation* aout,
68                       const RsScriptCall* sc) override;
69 
70     void invokeInit() override;
71     void invokeFreeChildren() override;
72 
73     void setGlobalVar(uint32_t slot, const void *data, size_t dataLength) override;
74     void getGlobalVar(uint32_t slot, void *data, size_t dataLength) override;
75     void setGlobalVarWithElemDims(uint32_t slot, const void *data, size_t dataLength,
76                                   const Element *e, const uint32_t *dims,
77                                   size_t dimLength) override;
78     void setGlobalBind(uint32_t slot, Allocation *data) override;
79     void setGlobalObj(uint32_t slot, ObjectBase *data) override;
80 
81     const char* getFieldName(uint32_t slot) const;
82 
83     ~RsdCpuScriptImpl() override;
84     RsdCpuScriptImpl(RsdCpuReferenceImpl *ctx, const Script *s);
85 
getScript()86     const Script * getScript() {return mScript;}
87 
88     bool forEachMtlsSetup(const Allocation ** ains, uint32_t inLen,
89                           Allocation * aout, const void * usr, uint32_t usrLen,
90                           const RsScriptCall *sc, MTLaunchStructForEach *mtls);
91 
92     virtual void forEachKernelSetup(uint32_t slot, MTLaunchStructForEach *mtls);
93 
94     // Build an MTLaunchStruct suitable for launching a general reduce-style kernel.
95     bool reduceMtlsSetup(const Allocation ** ains, uint32_t inLen, const Allocation *aout,
96                          const RsScriptCall *sc, MTLaunchStructReduce *mtls);
97     // Finalize an MTLaunchStruct for launching a general reduce-style kernel.
98     virtual void reduceKernelSetup(uint32_t slot, MTLaunchStructReduce *mtls);
99 
100     const RsdCpuReference::CpuSymbol * lookupSymbolMath(const char *sym);
101     static void * lookupRuntimeStub(void* pContext, char const* name);
102 
103     Allocation * getAllocationForPointer(const void *ptr) const override;
104     bool storeRSInfoFromSO();
105 
106     int getGlobalEntries() const override;
107     const char * getGlobalName(int i) const override;
108     const void * getGlobalAddress(int i) const override;
109     size_t getGlobalSize(int i) const override;
110     uint32_t getGlobalProperties(int i) const override;
111 
112 protected:
113     RsdCpuReferenceImpl *mCtx;
114     const Script *mScript;
115     void *mScriptSO;
116 
117 #ifndef RS_COMPATIBILITY_LIB
118     // Returns the path to the core library we'll use.
119     const char* findCoreLib(const bcinfo::MetadataExtractor& bitCodeMetaData, const char* bitcode,
120                             size_t bitcodeSize);
121 #endif
122 
123     RootFunc_t mRoot;
124     RootFunc_t mRootExpand;
125     InitOrDtorFunc_t mInit;
126     InitOrDtorFunc_t mFreeChildren;
127     ScriptExecutable* mScriptExec;
128 
129     Allocation **mBoundAllocs;
130     void * mIntrinsicData;
131     bool mIsThreadable;
132 
133 public:
134     static const char* BCC_EXE_PATH;
getBitcodeFilePath()135     const char* getBitcodeFilePath() const { return mBitcodeFilePath.c_str(); }
136 
137 private:
138     bool setUpMtlsDimensions(MTLaunchStructCommon *mtls,
139                              const RsLaunchDimensions &baseDim,
140                              const RsScriptCall *sc);
141 
142     std::string mBitcodeFilePath;
143     uint32_t mBuildChecksum;
144     bool mChecksumNeeded;
145 };
146 
147 Allocation * rsdScriptGetAllocationForPointer(
148                         const Context *dc,
149                         const Script *script,
150                         const void *);
151 
152 uint32_t constructBuildChecksum(uint8_t const *bitcode, size_t bitcodeSize,
153                                 const char *commandLine,
154                                 const char ** bccFiles, size_t numFiles);
155 
156 } // namespace renderscript
157 
158 #ifdef __LP64__
159 #define SYSLIBPATH "/system/lib64"
160 #define SYSLIBPATH_BC "/system/lib64"
161 #define SYSLIBPATH_VENDOR "/system/vendor/lib64"
162 #elif defined(BUILD_ARM_FOR_X86) && defined(__arm__)
163 #define SYSLIBPATH "/system/lib/arm"
164 #define SYSLIBPATH_BC "/system/lib"
165 #define SYSLIBPATH_VENDOR "/system/vendor/lib/arm"
166 #else
167 #define SYSLIBPATH "/system/lib"
168 #define SYSLIBPATH_BC "/system/lib"
169 #define SYSLIBPATH_VENDOR "/system/vendor/lib"
170 #endif
171 
172 } // namespace android
173 
174 namespace {
175 
is_force_recompile()176 inline bool is_force_recompile() {
177   char buf[PROP_VALUE_MAX];
178 
179   // Re-compile if floating point precision has been overridden.
180   android::renderscript::property_get("debug.rs.precision", buf, "");
181   if (buf[0] != '\0') {
182     return true;
183   }
184 
185   // Re-compile if debug.rs.forcerecompile is set.
186   android::renderscript::property_get("debug.rs.forcerecompile", buf, "0");
187   if ((::strcmp(buf, "1") == 0) || (::strcmp(buf, "true") == 0)) {
188     return true;
189   } else {
190     return false;
191   }
192 }
193 
getVndkSysLibPath()194 inline std::string getVndkSysLibPath() {
195   char buf[PROP_VALUE_MAX];
196   android::renderscript::property_get("ro.vndk.version", buf, "");
197   std::string vndk_path = "/apex/com.android.vndk.v" + std::string(buf) + "/lib";
198   #ifdef __LP64__
199   vndk_path += "64";
200   #endif
201   return vndk_path;
202 }
203 
204 }  // anonymous namespace
205 
206 #endif  // RSD_CPU_SCRIPT_H
207