rsCpuScript.h revision d9bae689c1b8c3f2ed1a5f2b374dc9393584b8dd
1709a0978ae141198018ca9769f8d96292a8928e6Jason Sams/* 2709a0978ae141198018ca9769f8d96292a8928e6Jason Sams * Copyright (C) 2011-2012 The Android Open Source Project 3709a0978ae141198018ca9769f8d96292a8928e6Jason Sams * 4709a0978ae141198018ca9769f8d96292a8928e6Jason Sams * Licensed under the Apache License, Version 2.0 (the "License"); 5709a0978ae141198018ca9769f8d96292a8928e6Jason Sams * you may not use this file except in compliance with the License. 6709a0978ae141198018ca9769f8d96292a8928e6Jason Sams * You may obtain a copy of the License at 7709a0978ae141198018ca9769f8d96292a8928e6Jason Sams * 8709a0978ae141198018ca9769f8d96292a8928e6Jason Sams * http://www.apache.org/licenses/LICENSE-2.0 9709a0978ae141198018ca9769f8d96292a8928e6Jason Sams * 10709a0978ae141198018ca9769f8d96292a8928e6Jason Sams * Unless required by applicable law or agreed to in writing, software 11709a0978ae141198018ca9769f8d96292a8928e6Jason Sams * distributed under the License is distributed on an "AS IS" BASIS, 12709a0978ae141198018ca9769f8d96292a8928e6Jason Sams * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13709a0978ae141198018ca9769f8d96292a8928e6Jason Sams * See the License for the specific language governing permissions and 14709a0978ae141198018ca9769f8d96292a8928e6Jason Sams * limitations under the License. 15709a0978ae141198018ca9769f8d96292a8928e6Jason Sams */ 16709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 17709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#ifndef RSD_BCC_H 18709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#define RSD_BCC_H 19709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 20709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#include <rs_hal.h> 21709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#include <rsRuntime.h> 22709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 2325e3af55a43faddced1a9931574dfdc3cc8ad8fdStephen Hines#ifndef RS_COMPATIBILITY_LIB 2429809d1f95d4cd4cbc6b2f9384b3321759691e13Tim Murray#include <utility> 2525e3af55a43faddced1a9931574dfdc3cc8ad8fdStephen Hines#endif 2625e3af55a43faddced1a9931574dfdc3cc8ad8fdStephen Hines 27709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#include "rsCpuCore.h" 28709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 29d9bae689c1b8c3f2ed1a5f2b374dc9393584b8ddYang Ni#include <vector> 30d9bae689c1b8c3f2ed1a5f2b374dc9393584b8ddYang Ni 31709a0978ae141198018ca9769f8d96292a8928e6Jason Samsnamespace bcc { 32709a0978ae141198018ca9769f8d96292a8928e6Jason Sams class BCCContext; 33709a0978ae141198018ca9769f8d96292a8928e6Jason Sams class RSCompilerDriver; 34709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 35709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 369ab5094dd32352b33e251e540934f6e814c5fa5bJean-Luc Brouilletnamespace bcinfo { 379ab5094dd32352b33e251e540934f6e814c5fa5bJean-Luc Brouillet class MetadataExtractor; 389ab5094dd32352b33e251e540934f6e814c5fa5bJean-Luc Brouillet} 399ab5094dd32352b33e251e540934f6e814c5fa5bJean-Luc Brouillet 40709a0978ae141198018ca9769f8d96292a8928e6Jason Samsnamespace android { 41709a0978ae141198018ca9769f8d96292a8928e6Jason Samsnamespace renderscript { 42709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 43d9bae689c1b8c3f2ed1a5f2b374dc9393584b8ddYang Niclass ScriptExecutable { 44d9bae689c1b8c3f2ed1a5f2b374dc9393584b8ddYang Ni public: 45d9bae689c1b8c3f2ed1a5f2b374dc9393584b8ddYang Ni ScriptExecutable(Context* RSContext, 46d9bae689c1b8c3f2ed1a5f2b374dc9393584b8ddYang Ni std::vector<void*>& fieldAddress, 47d9bae689c1b8c3f2ed1a5f2b374dc9393584b8ddYang Ni std::vector<bool>& fieldIsObject, 48d9bae689c1b8c3f2ed1a5f2b374dc9393584b8ddYang Ni std::vector<InvokeFunc_t>& invokeFunctions, 49d9bae689c1b8c3f2ed1a5f2b374dc9393584b8ddYang Ni std::vector<ForEachFunc_t>& forEachFunctions, 50d9bae689c1b8c3f2ed1a5f2b374dc9393584b8ddYang Ni std::vector<uint32_t>& forEachSignatures) : mRS(RSContext) { 51d9bae689c1b8c3f2ed1a5f2b374dc9393584b8ddYang Ni mFieldAddress.swap(fieldAddress); 52d9bae689c1b8c3f2ed1a5f2b374dc9393584b8ddYang Ni mFieldIsObject.swap(fieldIsObject); 53d9bae689c1b8c3f2ed1a5f2b374dc9393584b8ddYang Ni mInvokeFunctions.swap(invokeFunctions); 54d9bae689c1b8c3f2ed1a5f2b374dc9393584b8ddYang Ni mForEachFunctions.swap(forEachFunctions); 55d9bae689c1b8c3f2ed1a5f2b374dc9393584b8ddYang Ni mForEachSignatures.swap(forEachSignatures); 56d9bae689c1b8c3f2ed1a5f2b374dc9393584b8ddYang Ni } 57d9bae689c1b8c3f2ed1a5f2b374dc9393584b8ddYang Ni 58d9bae689c1b8c3f2ed1a5f2b374dc9393584b8ddYang Ni ~ScriptExecutable() { 59d9bae689c1b8c3f2ed1a5f2b374dc9393584b8ddYang Ni for (size_t i = 0; i < mFieldAddress.size(); ++i) { 60d9bae689c1b8c3f2ed1a5f2b374dc9393584b8ddYang Ni if (mFieldIsObject[i]) { 61d9bae689c1b8c3f2ed1a5f2b374dc9393584b8ddYang Ni if (mFieldAddress[i] != nullptr) { 62d9bae689c1b8c3f2ed1a5f2b374dc9393584b8ddYang Ni rs_object_base *obj_addr = 63d9bae689c1b8c3f2ed1a5f2b374dc9393584b8ddYang Ni reinterpret_cast<rs_object_base *>(mFieldAddress[i]); 64d9bae689c1b8c3f2ed1a5f2b374dc9393584b8ddYang Ni rsrClearObject(mRS, obj_addr); 65d9bae689c1b8c3f2ed1a5f2b374dc9393584b8ddYang Ni } 66d9bae689c1b8c3f2ed1a5f2b374dc9393584b8ddYang Ni } 67d9bae689c1b8c3f2ed1a5f2b374dc9393584b8ddYang Ni } 68d9bae689c1b8c3f2ed1a5f2b374dc9393584b8ddYang Ni } 69d9bae689c1b8c3f2ed1a5f2b374dc9393584b8ddYang Ni 70d9bae689c1b8c3f2ed1a5f2b374dc9393584b8ddYang Ni static ScriptExecutable* 71d9bae689c1b8c3f2ed1a5f2b374dc9393584b8ddYang Ni createFromSharedObject(Context* RSContext, void* sharedObj); 72d9bae689c1b8c3f2ed1a5f2b374dc9393584b8ddYang Ni 73d9bae689c1b8c3f2ed1a5f2b374dc9393584b8ddYang Ni size_t getExportedVariableCount() const { return mFieldAddress.size(); } 74d9bae689c1b8c3f2ed1a5f2b374dc9393584b8ddYang Ni size_t getExportedFunctionCount() const { return mInvokeFunctions.size(); } 75d9bae689c1b8c3f2ed1a5f2b374dc9393584b8ddYang Ni size_t getExportedForEachCount() const { return mForEachFunctions.size(); } 76d9bae689c1b8c3f2ed1a5f2b374dc9393584b8ddYang Ni 77d9bae689c1b8c3f2ed1a5f2b374dc9393584b8ddYang Ni void* getFieldAddress(int slot) const { return mFieldAddress[slot]; } 78d9bae689c1b8c3f2ed1a5f2b374dc9393584b8ddYang Ni bool getFieldIsObject(int slot) const { return mFieldIsObject[slot]; } 79d9bae689c1b8c3f2ed1a5f2b374dc9393584b8ddYang Ni InvokeFunc_t getInvokeFunction(int slot) const { return mInvokeFunctions[slot]; } 80d9bae689c1b8c3f2ed1a5f2b374dc9393584b8ddYang Ni ForEachFunc_t getForEachFunction(int slot) const { return mForEachFunctions[slot]; } 81d9bae689c1b8c3f2ed1a5f2b374dc9393584b8ddYang Ni uint32_t getForEachSignature(int slot) const { return mForEachSignatures[slot]; } 82d9bae689c1b8c3f2ed1a5f2b374dc9393584b8ddYang Ni 83d9bae689c1b8c3f2ed1a5f2b374dc9393584b8ddYang Ni private: 84d9bae689c1b8c3f2ed1a5f2b374dc9393584b8ddYang Ni std::vector<void*> mFieldAddress; 85d9bae689c1b8c3f2ed1a5f2b374dc9393584b8ddYang Ni std::vector<bool> mFieldIsObject; 86d9bae689c1b8c3f2ed1a5f2b374dc9393584b8ddYang Ni std::vector<InvokeFunc_t> mInvokeFunctions; 87d9bae689c1b8c3f2ed1a5f2b374dc9393584b8ddYang Ni std::vector<ForEachFunc_t> mForEachFunctions; 88d9bae689c1b8c3f2ed1a5f2b374dc9393584b8ddYang Ni std::vector<uint32_t> mForEachSignatures; 89d9bae689c1b8c3f2ed1a5f2b374dc9393584b8ddYang Ni 90d9bae689c1b8c3f2ed1a5f2b374dc9393584b8ddYang Ni Context* mRS; 91d9bae689c1b8c3f2ed1a5f2b374dc9393584b8ddYang Ni}; 92709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 93709a0978ae141198018ca9769f8d96292a8928e6Jason Samsclass RsdCpuScriptImpl : public RsdCpuReferenceImpl::CpuScript { 94709a0978ae141198018ca9769f8d96292a8928e6Jason Samspublic: 95709a0978ae141198018ca9769f8d96292a8928e6Jason Sams typedef void (*outer_foreach_t)( 9680ef693674f69c0343c41564e30f80e7fb513b60Chris Wailes const RsExpandKernelParams *, 97709a0978ae141198018ca9769f8d96292a8928e6Jason Sams uint32_t x1, uint32_t x2, 989ed79105cc6a8dbfaf959875249f36022cc2c798Chris Wailes uint32_t outstep); 99dc0d8f7c0f1f43f25c34fbc04656ad578f6e953bPirama Arumuga Nainar 100110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams typedef void (* InvokeFunc_t)(void); 101110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams typedef void (* ForEachFunc_t)(void); 102110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams typedef int (* RootFunc_t)(void); 103dc0d8f7c0f1f43f25c34fbc04656ad578f6e953bPirama Arumuga Nainar#ifdef RS_COMPATIBILITY_LIB 104110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams typedef void (*WorkerCallback_t)(void *usr, uint32_t idx); 105110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams#endif 106709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 107709a0978ae141198018ca9769f8d96292a8928e6Jason Sams bool init(char const *resName, char const *cacheDir, 108005113297b19ed256b6db9d6bc293ed9266899fcStephen Hines uint8_t const *bitcode, size_t bitcodeSize, uint32_t flags, 10944bef6fba6244292b751387f3d6c31cca96c28adChris Wailes char const *bccPluginName = nullptr); 110709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void populateScript(Script *); 111709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 112709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void invokeFunction(uint32_t slot, const void *params, size_t paramLength); 113709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual int invokeRoot(); 114f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes virtual void preLaunch(uint32_t slot, const Allocation ** ains, 115f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes uint32_t inLen, Allocation * aout, const void * usr, 11617e3cdc24776d8fdbf1ce16287b9b4dcd516708fJason Sams uint32_t usrLen, const RsScriptCall *sc); 117f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes virtual void postLaunch(uint32_t slot, const Allocation ** ains, 118f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes uint32_t inLen, Allocation * aout, 119f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes const void * usr, uint32_t usrLen, 120f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes const RsScriptCall *sc); 121f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes 122709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void invokeForEach(uint32_t slot, 123f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes const Allocation ** ains, 124f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes uint32_t inLen, 125f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes Allocation* aout, 126f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes const void* usr, 127f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes uint32_t usrLen, 128f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes const RsScriptCall* sc); 129f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes 130709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void invokeInit(); 131709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void invokeFreeChildren(); 132709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 133709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void setGlobalVar(uint32_t slot, const void *data, size_t dataLength); 1349c64239ebbfa4170190ede812e69150035e008e0Tim Murray virtual void getGlobalVar(uint32_t slot, void *data, size_t dataLength); 135709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void setGlobalVarWithElemDims(uint32_t slot, const void *data, size_t dataLength, 136ac8d146a41f18afad5314ac8af440d6aedbe20bfStephen Hines const Element *e, const uint32_t *dims, size_t dimLength); 137709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void setGlobalBind(uint32_t slot, Allocation *data); 138709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void setGlobalObj(uint32_t slot, ObjectBase *data); 139709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 140709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 141709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual ~RsdCpuScriptImpl(); 142709a0978ae141198018ca9769f8d96292a8928e6Jason Sams RsdCpuScriptImpl(RsdCpuReferenceImpl *ctx, const Script *s); 143709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 144709a0978ae141198018ca9769f8d96292a8928e6Jason Sams const Script * getScript() {return mScript;} 145709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 1464b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes void forEachMtlsSetup(const Allocation ** ains, uint32_t inLen, 1474b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes Allocation * aout, const void * usr, uint32_t usrLen, 1484b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes const RsScriptCall *sc, MTLaunchStruct *mtls); 1494b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes 150709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void forEachKernelSetup(uint32_t slot, MTLaunchStruct *mtls); 151709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 152709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 153709a0978ae141198018ca9769f8d96292a8928e6Jason Sams const RsdCpuReference::CpuSymbol * lookupSymbolMath(const char *sym); 154709a0978ae141198018ca9769f8d96292a8928e6Jason Sams static void * lookupRuntimeStub(void* pContext, char const* name); 155709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 156709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual Allocation * getAllocationForPointer(const void *ptr) const; 157dc0d8f7c0f1f43f25c34fbc04656ad578f6e953bPirama Arumuga Nainar bool storeRSInfoFromSO(); 158709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 159709a0978ae141198018ca9769f8d96292a8928e6Jason Samsprotected: 160709a0978ae141198018ca9769f8d96292a8928e6Jason Sams RsdCpuReferenceImpl *mCtx; 161709a0978ae141198018ca9769f8d96292a8928e6Jason Sams const Script *mScript; 162dc0d8f7c0f1f43f25c34fbc04656ad578f6e953bPirama Arumuga Nainar void *mScriptSO; 163709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 164110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams#ifndef RS_COMPATIBILITY_LIB 16540e35cdbe217ec8bf9fc3c69873c7d62fc14158fJean-Luc Brouillet // Returns the path to the core library we'll use. 16640e35cdbe217ec8bf9fc3c69873c7d62fc14158fJean-Luc Brouillet const char* findCoreLib(const bcinfo::MetadataExtractor& bitCodeMetaData, const char* bitcode, 1679ab5094dd32352b33e251e540934f6e814c5fa5bJean-Luc Brouillet size_t bitcodeSize); 16829809d1f95d4cd4cbc6b2f9384b3321759691e13Tim Murray 169709a0978ae141198018ca9769f8d96292a8928e6Jason Sams bcc::RSCompilerDriver *mCompilerDriver; 17045e753a46e587c69b3b0d0c5138e88715a24a29aStephen Hines#endif 17145e753a46e587c69b3b0d0c5138e88715a24a29aStephen Hines 172110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams RootFunc_t mRoot; 173110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams RootFunc_t mRootExpand; 174110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams InvokeFunc_t mInit; 175110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams InvokeFunc_t mFreeChildren; 176d9bae689c1b8c3f2ed1a5f2b374dc9393584b8ddYang Ni ScriptExecutable* mScriptExec; 177709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 178709a0978ae141198018ca9769f8d96292a8928e6Jason Sams Allocation **mBoundAllocs; 179709a0978ae141198018ca9769f8d96292a8928e6Jason Sams void * mIntrinsicData; 180709a0978ae141198018ca9769f8d96292a8928e6Jason Sams bool mIsThreadable; 181709a0978ae141198018ca9769f8d96292a8928e6Jason Sams}; 182709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 183709a0978ae141198018ca9769f8d96292a8928e6Jason SamsAllocation * rsdScriptGetAllocationForPointer( 184709a0978ae141198018ca9769f8d96292a8928e6Jason Sams const Context *dc, 185709a0978ae141198018ca9769f8d96292a8928e6Jason Sams const Script *script, 186709a0978ae141198018ca9769f8d96292a8928e6Jason Sams const void *); 187709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 188709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 189709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 190709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 191d9bae689c1b8c3f2ed1a5f2b374dc9393584b8ddYang Ni 192709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 193709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 194709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#endif 195