rsCpuScript.h revision f37121300217d3b39ab66dd9c8881bcbcad932df
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, 52709a0978ae141198018ca9769f8d96292a8928e6Jason Sams uint32_t instep, uint32_t outstep); 53110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams#ifdef RS_COMPATIBILITY_LIB 54110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams typedef void (* InvokeFunc_t)(void); 55110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams typedef void (* ForEachFunc_t)(void); 56110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams typedef int (* RootFunc_t)(void); 57110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams typedef void (*WorkerCallback_t)(void *usr, uint32_t idx); 58110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams#endif 59709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 60709a0978ae141198018ca9769f8d96292a8928e6Jason Sams bool init(char const *resName, char const *cacheDir, 61005113297b19ed256b6db9d6bc293ed9266899fcStephen Hines uint8_t const *bitcode, size_t bitcodeSize, uint32_t flags, 62005113297b19ed256b6db9d6bc293ed9266899fcStephen Hines char const *bccPluginName = NULL); 63709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void populateScript(Script *); 64709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 65709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void invokeFunction(uint32_t slot, const void *params, size_t paramLength); 66709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual int invokeRoot(); 67f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes virtual void preLaunch(uint32_t slot, const Allocation ** ains, 68f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes uint32_t inLen, Allocation * aout, const void * usr, 6917e3cdc24776d8fdbf1ce16287b9b4dcd516708fJason Sams uint32_t usrLen, const RsScriptCall *sc); 70f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes virtual void postLaunch(uint32_t slot, const Allocation ** ains, 71f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes uint32_t inLen, Allocation * aout, 72f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes const void * usr, uint32_t usrLen, 73f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes const RsScriptCall *sc); 74f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes 75709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void invokeForEach(uint32_t slot, 76f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes const Allocation ** ains, 77f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes uint32_t inLen, 78f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes Allocation* aout, 79f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes const void* usr, 80f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes uint32_t usrLen, 81f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes const RsScriptCall* sc); 82f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes 83709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void invokeInit(); 84709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void invokeFreeChildren(); 85709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 86709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void setGlobalVar(uint32_t slot, const void *data, size_t dataLength); 879c64239ebbfa4170190ede812e69150035e008e0Tim Murray virtual void getGlobalVar(uint32_t slot, void *data, size_t dataLength); 88709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void setGlobalVarWithElemDims(uint32_t slot, const void *data, size_t dataLength, 89ac8d146a41f18afad5314ac8af440d6aedbe20bfStephen Hines const Element *e, const uint32_t *dims, size_t dimLength); 90709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void setGlobalBind(uint32_t slot, Allocation *data); 91709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void setGlobalObj(uint32_t slot, ObjectBase *data); 92709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 93709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 94709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual ~RsdCpuScriptImpl(); 95709a0978ae141198018ca9769f8d96292a8928e6Jason Sams RsdCpuScriptImpl(RsdCpuReferenceImpl *ctx, const Script *s); 96709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 97709a0978ae141198018ca9769f8d96292a8928e6Jason Sams const Script * getScript() {return mScript;} 98709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 994b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes void forEachMtlsSetup(const Allocation ** ains, uint32_t inLen, 1004b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes Allocation * aout, const void * usr, uint32_t usrLen, 1014b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes const RsScriptCall *sc, MTLaunchStruct *mtls); 1024b3c34e6833e39bc89c2128002806b654b8e623dChris Wailes 103709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void forEachKernelSetup(uint32_t slot, MTLaunchStruct *mtls); 104709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 105709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 106709a0978ae141198018ca9769f8d96292a8928e6Jason Sams const RsdCpuReference::CpuSymbol * lookupSymbolMath(const char *sym); 107709a0978ae141198018ca9769f8d96292a8928e6Jason Sams static void * lookupRuntimeStub(void* pContext, char const* name); 108709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 109709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual Allocation * getAllocationForPointer(const void *ptr) const; 110709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 111cadfac411e6690e39de36c4f9e94deb9b7d2d08eJason Sams#ifndef RS_COMPATIBILITY_LIB 112f218bf115af4ae4fd79adbb8842608b308a4cf07Stephen Hines virtual void * getRSExecutable() { return mExecutable; } 113cadfac411e6690e39de36c4f9e94deb9b7d2d08eJason Sams#endif 114709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 115709a0978ae141198018ca9769f8d96292a8928e6Jason Samsprotected: 116709a0978ae141198018ca9769f8d96292a8928e6Jason Sams RsdCpuReferenceImpl *mCtx; 117709a0978ae141198018ca9769f8d96292a8928e6Jason Sams const Script *mScript; 118709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 119110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams#ifndef RS_COMPATIBILITY_LIB 12040e35cdbe217ec8bf9fc3c69873c7d62fc14158fJean-Luc Brouillet // Returns the path to the core library we'll use. 12140e35cdbe217ec8bf9fc3c69873c7d62fc14158fJean-Luc Brouillet const char* findCoreLib(const bcinfo::MetadataExtractor& bitCodeMetaData, const char* bitcode, 1229ab5094dd32352b33e251e540934f6e814c5fa5bJean-Luc Brouillet size_t bitcodeSize); 123709a0978ae141198018ca9769f8d96292a8928e6Jason Sams int (*mRoot)(); 124709a0978ae141198018ca9769f8d96292a8928e6Jason Sams int (*mRootExpand)(); 125709a0978ae141198018ca9769f8d96292a8928e6Jason Sams void (*mInit)(); 126709a0978ae141198018ca9769f8d96292a8928e6Jason Sams void (*mFreeChildren)(); 127709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 12829809d1f95d4cd4cbc6b2f9384b3321759691e13Tim Murray std::vector<std::pair<const char *, uint32_t> > mExportedForEachFuncList; 12929809d1f95d4cd4cbc6b2f9384b3321759691e13Tim Murray 130709a0978ae141198018ca9769f8d96292a8928e6Jason Sams bcc::BCCContext *mCompilerContext; 131709a0978ae141198018ca9769f8d96292a8928e6Jason Sams bcc::RSCompilerDriver *mCompilerDriver; 13225e3af55a43faddced1a9931574dfdc3cc8ad8fdStephen Hines bcc::CompilerRTSymbolResolver mCompilerRuntime; 13325e3af55a43faddced1a9931574dfdc3cc8ad8fdStephen Hines bcc::LookupFunctionSymbolResolver<void *> mRSRuntime; 13425e3af55a43faddced1a9931574dfdc3cc8ad8fdStephen Hines bcc::SymbolResolverProxy mResolver; 135709a0978ae141198018ca9769f8d96292a8928e6Jason Sams bcc::RSExecutable *mExecutable; 136110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams#else 137110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams void *mScriptSO; 138110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams RootFunc_t mRoot; 139110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams RootFunc_t mRootExpand; 140110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams InvokeFunc_t mInit; 141110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams InvokeFunc_t mFreeChildren; 142110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams InvokeFunc_t *mInvokeFunctions; 143110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams ForEachFunc_t *mForEachFunctions; 144110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams 145110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams void **mFieldAddress; 146110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams bool *mFieldIsObject; 147110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams uint32_t *mForEachSignatures; 148110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams 149110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams // for populate script 150110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams //int mVersionMajor; 151110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams //int mVersionMinor; 152110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams size_t mExportedVariableCount; 153110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams size_t mExportedFunctionCount; 154110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams#endif 155709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 156709a0978ae141198018ca9769f8d96292a8928e6Jason Sams Allocation **mBoundAllocs; 157709a0978ae141198018ca9769f8d96292a8928e6Jason Sams void * mIntrinsicData; 158709a0978ae141198018ca9769f8d96292a8928e6Jason Sams bool mIsThreadable; 159709a0978ae141198018ca9769f8d96292a8928e6Jason Sams}; 160709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 161709a0978ae141198018ca9769f8d96292a8928e6Jason SamsAllocation * rsdScriptGetAllocationForPointer( 162709a0978ae141198018ca9769f8d96292a8928e6Jason Sams const Context *dc, 163709a0978ae141198018ca9769f8d96292a8928e6Jason Sams const Script *script, 164709a0978ae141198018ca9769f8d96292a8928e6Jason Sams const void *); 165709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 166709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 167709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 168709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 169709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 170709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 171709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#endif 172