rsCpuScript.h revision dc0d8f7c0f1f43f25c34fbc04656ad578f6e953b
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 2425e3af55a43faddced1a9931574dfdc3cc8ad8fdStephen Hines#include <bcc/ExecutionEngine/CompilerRTSymbolResolver.h> 2525e3af55a43faddced1a9931574dfdc3cc8ad8fdStephen Hines#include <bcc/ExecutionEngine/SymbolResolverProxy.h> 2629809d1f95d4cd4cbc6b2f9384b3321759691e13Tim Murray#include <vector> 2729809d1f95d4cd4cbc6b2f9384b3321759691e13Tim Murray#include <utility> 2825e3af55a43faddced1a9931574dfdc3cc8ad8fdStephen Hines#endif 2925e3af55a43faddced1a9931574dfdc3cc8ad8fdStephen Hines 30709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#include "rsCpuCore.h" 31709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 32709a0978ae141198018ca9769f8d96292a8928e6Jason Samsnamespace bcc { 33709a0978ae141198018ca9769f8d96292a8928e6Jason Sams class BCCContext; 34709a0978ae141198018ca9769f8d96292a8928e6Jason Sams class RSCompilerDriver; 35709a0978ae141198018ca9769f8d96292a8928e6Jason Sams class RSExecutable; 36709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 37709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 389ab5094dd32352b33e251e540934f6e814c5fa5bJean-Luc Brouilletnamespace bcinfo { 399ab5094dd32352b33e251e540934f6e814c5fa5bJean-Luc Brouillet class MetadataExtractor; 409ab5094dd32352b33e251e540934f6e814c5fa5bJean-Luc Brouillet} 419ab5094dd32352b33e251e540934f6e814c5fa5bJean-Luc Brouillet 42709a0978ae141198018ca9769f8d96292a8928e6Jason Samsnamespace android { 43709a0978ae141198018ca9769f8d96292a8928e6Jason Samsnamespace renderscript { 44709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 45709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 46709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 47709a0978ae141198018ca9769f8d96292a8928e6Jason Samsclass RsdCpuScriptImpl : public RsdCpuReferenceImpl::CpuScript { 48709a0978ae141198018ca9769f8d96292a8928e6Jason Samspublic: 49709a0978ae141198018ca9769f8d96292a8928e6Jason Sams typedef void (*outer_foreach_t)( 5080ef693674f69c0343c41564e30f80e7fb513b60Chris Wailes const RsExpandKernelParams *, 51709a0978ae141198018ca9769f8d96292a8928e6Jason Sams uint32_t x1, uint32_t x2, 529ed79105cc6a8dbfaf959875249f36022cc2c798Chris Wailes uint32_t outstep); 53dc0d8f7c0f1f43f25c34fbc04656ad578f6e953bPirama Arumuga Nainar 54110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams typedef void (* InvokeFunc_t)(void); 55110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams typedef void (* ForEachFunc_t)(void); 56110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams typedef int (* RootFunc_t)(void); 57dc0d8f7c0f1f43f25c34fbc04656ad578f6e953bPirama Arumuga Nainar#ifdef RS_COMPATIBILITY_LIB 58110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams typedef void (*WorkerCallback_t)(void *usr, uint32_t idx); 59110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams#endif 60709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 61709a0978ae141198018ca9769f8d96292a8928e6Jason Sams bool init(char const *resName, char const *cacheDir, 62005113297b19ed256b6db9d6bc293ed9266899fcStephen Hines uint8_t const *bitcode, size_t bitcodeSize, uint32_t flags, 6344bef6fba6244292b751387f3d6c31cca96c28adChris Wailes char const *bccPluginName = nullptr); 64709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void populateScript(Script *); 65709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 66709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void invokeFunction(uint32_t slot, const void *params, size_t paramLength); 67709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual int invokeRoot(); 68f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes virtual void preLaunch(uint32_t slot, const Allocation ** ains, 69f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes uint32_t inLen, Allocation * aout, const void * usr, 7017e3cdc24776d8fdbf1ce16287b9b4dcd516708fJason Sams uint32_t usrLen, const RsScriptCall *sc); 71f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes virtual void postLaunch(uint32_t slot, const Allocation ** ains, 72f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes uint32_t inLen, Allocation * aout, 73f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes const void * usr, uint32_t usrLen, 74f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes const RsScriptCall *sc); 75f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes 76709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void invokeForEach(uint32_t slot, 77f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes const Allocation ** ains, 78f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes uint32_t inLen, 79f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes Allocation* aout, 80f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes const void* usr, 81f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes uint32_t usrLen, 82f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes const RsScriptCall* sc); 83f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes 84709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void invokeInit(); 85709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void invokeFreeChildren(); 86709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 87709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void setGlobalVar(uint32_t slot, const void *data, size_t dataLength); 889c64239ebbfa4170190ede812e69150035e008e0Tim Murray virtual void getGlobalVar(uint32_t slot, void *data, size_t dataLength); 89709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void setGlobalVarWithElemDims(uint32_t slot, const void *data, size_t dataLength, 90ac8d146a41f18afad5314ac8af440d6aedbe20bfStephen Hines const Element *e, const uint32_t *dims, size_t dimLength); 91709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void setGlobalBind(uint32_t slot, Allocation *data); 92709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void setGlobalObj(uint32_t slot, ObjectBase *data); 93709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 94709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 95709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual ~RsdCpuScriptImpl(); 96709a0978ae141198018ca9769f8d96292a8928e6Jason Sams RsdCpuScriptImpl(RsdCpuReferenceImpl *ctx, const Script *s); 97709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 98709a0978ae141198018ca9769f8d96292a8928e6Jason Sams const Script * getScript() {return mScript;} 99709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 1004b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes void forEachMtlsSetup(const Allocation ** ains, uint32_t inLen, 1014b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes Allocation * aout, const void * usr, uint32_t usrLen, 1024b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes const RsScriptCall *sc, MTLaunchStruct *mtls); 1034b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes 104709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void forEachKernelSetup(uint32_t slot, MTLaunchStruct *mtls); 105709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 106709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 107709a0978ae141198018ca9769f8d96292a8928e6Jason Sams const RsdCpuReference::CpuSymbol * lookupSymbolMath(const char *sym); 108709a0978ae141198018ca9769f8d96292a8928e6Jason Sams static void * lookupRuntimeStub(void* pContext, char const* name); 109709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 110709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual Allocation * getAllocationForPointer(const void *ptr) const; 111dc0d8f7c0f1f43f25c34fbc04656ad578f6e953bPirama Arumuga Nainar bool storeRSInfoFromSO(); 112709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 113cadfac411e6690e39de36c4f9e94deb9b7d2d08eJason Sams#ifndef RS_COMPATIBILITY_LIB 114dc0d8f7c0f1f43f25c34fbc04656ad578f6e953bPirama Arumuga Nainar bool storeRSInfoFromObj(bcinfo::MetadataExtractor &bitcodeMetadata); 115f218bf115af4ae4fd79adbb8842608b308a4cf07Stephen Hines virtual void * getRSExecutable() { return mExecutable; } 116cadfac411e6690e39de36c4f9e94deb9b7d2d08eJason Sams#endif 117709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 118709a0978ae141198018ca9769f8d96292a8928e6Jason Samsprotected: 119709a0978ae141198018ca9769f8d96292a8928e6Jason Sams RsdCpuReferenceImpl *mCtx; 120709a0978ae141198018ca9769f8d96292a8928e6Jason Sams const Script *mScript; 121dc0d8f7c0f1f43f25c34fbc04656ad578f6e953bPirama Arumuga Nainar void *mScriptSO; 122709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 123110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams#ifndef RS_COMPATIBILITY_LIB 12440e35cdbe217ec8bf9fc3c69873c7d62fc14158fJean-Luc Brouillet // Returns the path to the core library we'll use. 12540e35cdbe217ec8bf9fc3c69873c7d62fc14158fJean-Luc Brouillet const char* findCoreLib(const bcinfo::MetadataExtractor& bitCodeMetaData, const char* bitcode, 1269ab5094dd32352b33e251e540934f6e814c5fa5bJean-Luc Brouillet size_t bitcodeSize); 127dc0d8f7c0f1f43f25c34fbc04656ad578f6e953bPirama Arumuga Nainar RootFunc_t mRoot; 128dc0d8f7c0f1f43f25c34fbc04656ad578f6e953bPirama Arumuga Nainar RootFunc_t mRootExpand; 129dc0d8f7c0f1f43f25c34fbc04656ad578f6e953bPirama Arumuga Nainar InvokeFunc_t mInit; 130dc0d8f7c0f1f43f25c34fbc04656ad578f6e953bPirama Arumuga Nainar InvokeFunc_t mFreeChildren; 131dc0d8f7c0f1f43f25c34fbc04656ad578f6e953bPirama Arumuga Nainar 132dc0d8f7c0f1f43f25c34fbc04656ad578f6e953bPirama Arumuga Nainar InvokeFunc_t *mInvokeFunctions; 133dc0d8f7c0f1f43f25c34fbc04656ad578f6e953bPirama Arumuga Nainar ForEachFunc_t *mForEachFunctions; 134dc0d8f7c0f1f43f25c34fbc04656ad578f6e953bPirama Arumuga Nainar void **mFieldAddress; 135dc0d8f7c0f1f43f25c34fbc04656ad578f6e953bPirama Arumuga Nainar bool *mFieldIsObject; 136dc0d8f7c0f1f43f25c34fbc04656ad578f6e953bPirama Arumuga Nainar uint32_t *mForEachSignatures; 137dc0d8f7c0f1f43f25c34fbc04656ad578f6e953bPirama Arumuga Nainar size_t mExportedVariableCount; 138dc0d8f7c0f1f43f25c34fbc04656ad578f6e953bPirama Arumuga Nainar size_t mExportedFunctionCount; 139709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 14029809d1f95d4cd4cbc6b2f9384b3321759691e13Tim Murray std::vector<std::pair<const char *, uint32_t> > mExportedForEachFuncList; 14129809d1f95d4cd4cbc6b2f9384b3321759691e13Tim Murray 142709a0978ae141198018ca9769f8d96292a8928e6Jason Sams bcc::RSCompilerDriver *mCompilerDriver; 14325e3af55a43faddced1a9931574dfdc3cc8ad8fdStephen Hines bcc::CompilerRTSymbolResolver mCompilerRuntime; 14425e3af55a43faddced1a9931574dfdc3cc8ad8fdStephen Hines bcc::LookupFunctionSymbolResolver<void *> mRSRuntime; 14525e3af55a43faddced1a9931574dfdc3cc8ad8fdStephen Hines bcc::SymbolResolverProxy mResolver; 146709a0978ae141198018ca9769f8d96292a8928e6Jason Sams bcc::RSExecutable *mExecutable; 147110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams#else 148110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams RootFunc_t mRoot; 149110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams RootFunc_t mRootExpand; 150110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams InvokeFunc_t mInit; 151110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams InvokeFunc_t mFreeChildren; 152110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams InvokeFunc_t *mInvokeFunctions; 153110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams ForEachFunc_t *mForEachFunctions; 154110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams 155110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams void **mFieldAddress; 156110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams bool *mFieldIsObject; 157110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams uint32_t *mForEachSignatures; 158110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams 159110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams // for populate script 160110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams //int mVersionMajor; 161110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams //int mVersionMinor; 162110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams size_t mExportedVariableCount; 163110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams size_t mExportedFunctionCount; 164110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams#endif 165709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 166709a0978ae141198018ca9769f8d96292a8928e6Jason Sams Allocation **mBoundAllocs; 167709a0978ae141198018ca9769f8d96292a8928e6Jason Sams void * mIntrinsicData; 168709a0978ae141198018ca9769f8d96292a8928e6Jason Sams bool mIsThreadable; 169709a0978ae141198018ca9769f8d96292a8928e6Jason Sams}; 170709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 171709a0978ae141198018ca9769f8d96292a8928e6Jason SamsAllocation * rsdScriptGetAllocationForPointer( 172709a0978ae141198018ca9769f8d96292a8928e6Jason Sams const Context *dc, 173709a0978ae141198018ca9769f8d96292a8928e6Jason Sams const Script *script, 174709a0978ae141198018ca9769f8d96292a8928e6Jason Sams const void *); 175709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 176709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 177709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 178709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 179709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 180709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 181709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#endif 182