rsCpuScript.h revision 45e753a46e587c69b3b0d0c5138e88715a24a29a
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 <vector> 2529809d1f95d4cd4cbc6b2f9384b3321759691e13Tim Murray#include <utility> 2625e3af55a43faddced1a9931574dfdc3cc8ad8fdStephen Hines#endif 2725e3af55a43faddced1a9931574dfdc3cc8ad8fdStephen Hines 28709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#include "rsCpuCore.h" 29709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 30709a0978ae141198018ca9769f8d96292a8928e6Jason Samsnamespace bcc { 31709a0978ae141198018ca9769f8d96292a8928e6Jason Sams class BCCContext; 32709a0978ae141198018ca9769f8d96292a8928e6Jason Sams class RSCompilerDriver; 33709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 34709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 359ab5094dd32352b33e251e540934f6e814c5fa5bJean-Luc Brouilletnamespace bcinfo { 369ab5094dd32352b33e251e540934f6e814c5fa5bJean-Luc Brouillet class MetadataExtractor; 379ab5094dd32352b33e251e540934f6e814c5fa5bJean-Luc Brouillet} 389ab5094dd32352b33e251e540934f6e814c5fa5bJean-Luc Brouillet 39709a0978ae141198018ca9769f8d96292a8928e6Jason Samsnamespace android { 40709a0978ae141198018ca9769f8d96292a8928e6Jason Samsnamespace renderscript { 41709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 42709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 43709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 44709a0978ae141198018ca9769f8d96292a8928e6Jason Samsclass RsdCpuScriptImpl : public RsdCpuReferenceImpl::CpuScript { 45709a0978ae141198018ca9769f8d96292a8928e6Jason Samspublic: 46709a0978ae141198018ca9769f8d96292a8928e6Jason Sams typedef void (*outer_foreach_t)( 4780ef693674f69c0343c41564e30f80e7fb513b60Chris Wailes const RsExpandKernelParams *, 48709a0978ae141198018ca9769f8d96292a8928e6Jason Sams uint32_t x1, uint32_t x2, 499ed79105cc6a8dbfaf959875249f36022cc2c798Chris Wailes uint32_t outstep); 50dc0d8f7c0f1f43f25c34fbc04656ad578f6e953bPirama Arumuga Nainar 51110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams typedef void (* InvokeFunc_t)(void); 52110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams typedef void (* ForEachFunc_t)(void); 53110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams typedef int (* RootFunc_t)(void); 54dc0d8f7c0f1f43f25c34fbc04656ad578f6e953bPirama Arumuga Nainar#ifdef RS_COMPATIBILITY_LIB 55110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams typedef void (*WorkerCallback_t)(void *usr, uint32_t idx); 56110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams#endif 57709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 58709a0978ae141198018ca9769f8d96292a8928e6Jason Sams bool init(char const *resName, char const *cacheDir, 59005113297b19ed256b6db9d6bc293ed9266899fcStephen Hines uint8_t const *bitcode, size_t bitcodeSize, uint32_t flags, 6044bef6fba6244292b751387f3d6c31cca96c28adChris Wailes char const *bccPluginName = nullptr); 61709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void populateScript(Script *); 62709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 63709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void invokeFunction(uint32_t slot, const void *params, size_t paramLength); 64709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual int invokeRoot(); 65f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes virtual void preLaunch(uint32_t slot, const Allocation ** ains, 66f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes uint32_t inLen, Allocation * aout, const void * usr, 6717e3cdc24776d8fdbf1ce16287b9b4dcd516708fJason Sams uint32_t usrLen, const RsScriptCall *sc); 68f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes virtual void postLaunch(uint32_t slot, const Allocation ** ains, 69f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes uint32_t inLen, Allocation * aout, 70f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes const void * usr, uint32_t usrLen, 71f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes const RsScriptCall *sc); 72f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes 73709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void invokeForEach(uint32_t slot, 74f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes const Allocation ** ains, 75f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes uint32_t inLen, 76f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes Allocation* aout, 77f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes const void* usr, 78f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes uint32_t usrLen, 79f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes const RsScriptCall* sc); 80f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes 81709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void invokeInit(); 82709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void invokeFreeChildren(); 83709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 84709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void setGlobalVar(uint32_t slot, const void *data, size_t dataLength); 859c64239ebbfa4170190ede812e69150035e008e0Tim Murray virtual void getGlobalVar(uint32_t slot, void *data, size_t dataLength); 86709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void setGlobalVarWithElemDims(uint32_t slot, const void *data, size_t dataLength, 87ac8d146a41f18afad5314ac8af440d6aedbe20bfStephen Hines const Element *e, const uint32_t *dims, size_t dimLength); 88709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void setGlobalBind(uint32_t slot, Allocation *data); 89709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void setGlobalObj(uint32_t slot, ObjectBase *data); 90709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 91709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 92709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual ~RsdCpuScriptImpl(); 93709a0978ae141198018ca9769f8d96292a8928e6Jason Sams RsdCpuScriptImpl(RsdCpuReferenceImpl *ctx, const Script *s); 94709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 95709a0978ae141198018ca9769f8d96292a8928e6Jason Sams const Script * getScript() {return mScript;} 96709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 974b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes void forEachMtlsSetup(const Allocation ** ains, uint32_t inLen, 984b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes Allocation * aout, const void * usr, uint32_t usrLen, 994b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes const RsScriptCall *sc, MTLaunchStruct *mtls); 1004b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes 101709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void forEachKernelSetup(uint32_t slot, MTLaunchStruct *mtls); 102709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 103709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 104709a0978ae141198018ca9769f8d96292a8928e6Jason Sams const RsdCpuReference::CpuSymbol * lookupSymbolMath(const char *sym); 105709a0978ae141198018ca9769f8d96292a8928e6Jason Sams static void * lookupRuntimeStub(void* pContext, char const* name); 106709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 107709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual Allocation * getAllocationForPointer(const void *ptr) const; 108dc0d8f7c0f1f43f25c34fbc04656ad578f6e953bPirama Arumuga Nainar bool storeRSInfoFromSO(); 109709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 110709a0978ae141198018ca9769f8d96292a8928e6Jason Samsprotected: 111709a0978ae141198018ca9769f8d96292a8928e6Jason Sams RsdCpuReferenceImpl *mCtx; 112709a0978ae141198018ca9769f8d96292a8928e6Jason Sams const Script *mScript; 113dc0d8f7c0f1f43f25c34fbc04656ad578f6e953bPirama Arumuga Nainar void *mScriptSO; 114709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 115110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams#ifndef RS_COMPATIBILITY_LIB 11640e35cdbe217ec8bf9fc3c69873c7d62fc14158fJean-Luc Brouillet // Returns the path to the core library we'll use. 11740e35cdbe217ec8bf9fc3c69873c7d62fc14158fJean-Luc Brouillet const char* findCoreLib(const bcinfo::MetadataExtractor& bitCodeMetaData, const char* bitcode, 1189ab5094dd32352b33e251e540934f6e814c5fa5bJean-Luc Brouillet size_t bitcodeSize); 11929809d1f95d4cd4cbc6b2f9384b3321759691e13Tim Murray 120709a0978ae141198018ca9769f8d96292a8928e6Jason Sams bcc::RSCompilerDriver *mCompilerDriver; 12145e753a46e587c69b3b0d0c5138e88715a24a29aStephen Hines#endif 12245e753a46e587c69b3b0d0c5138e88715a24a29aStephen Hines 123110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams RootFunc_t mRoot; 124110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams RootFunc_t mRootExpand; 125110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams InvokeFunc_t mInit; 126110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams InvokeFunc_t mFreeChildren; 127110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams InvokeFunc_t *mInvokeFunctions; 128110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams ForEachFunc_t *mForEachFunctions; 129110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams 130110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams void **mFieldAddress; 131110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams bool *mFieldIsObject; 132110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams uint32_t *mForEachSignatures; 133110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams 134110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams size_t mExportedVariableCount; 135110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams size_t mExportedFunctionCount; 136709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 137709a0978ae141198018ca9769f8d96292a8928e6Jason Sams Allocation **mBoundAllocs; 138709a0978ae141198018ca9769f8d96292a8928e6Jason Sams void * mIntrinsicData; 139709a0978ae141198018ca9769f8d96292a8928e6Jason Sams bool mIsThreadable; 140709a0978ae141198018ca9769f8d96292a8928e6Jason Sams}; 141709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 142709a0978ae141198018ca9769f8d96292a8928e6Jason SamsAllocation * rsdScriptGetAllocationForPointer( 143709a0978ae141198018ca9769f8d96292a8928e6Jason Sams const Context *dc, 144709a0978ae141198018ca9769f8d96292a8928e6Jason Sams const Script *script, 145709a0978ae141198018ca9769f8d96292a8928e6Jason Sams const void *); 146709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 147709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 148709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 149709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 150709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 151709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 152709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#endif 153