rsCpuScript.h revision dc0d8f7c0f1f43f25c34fbc04656ad578f6e953b
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_BCC_H 18#define RSD_BCC_H 19 20#include <rs_hal.h> 21#include <rsRuntime.h> 22 23#ifndef RS_COMPATIBILITY_LIB 24#include <bcc/ExecutionEngine/CompilerRTSymbolResolver.h> 25#include <bcc/ExecutionEngine/SymbolResolverProxy.h> 26#include <vector> 27#include <utility> 28#endif 29 30#include "rsCpuCore.h" 31 32namespace bcc { 33 class BCCContext; 34 class RSCompilerDriver; 35 class RSExecutable; 36} 37 38namespace bcinfo { 39 class MetadataExtractor; 40} 41 42namespace android { 43namespace renderscript { 44 45 46 47class RsdCpuScriptImpl : public RsdCpuReferenceImpl::CpuScript { 48public: 49 typedef void (*outer_foreach_t)( 50 const RsExpandKernelParams *, 51 uint32_t x1, uint32_t x2, 52 uint32_t outstep); 53 54 typedef void (* InvokeFunc_t)(void); 55 typedef void (* ForEachFunc_t)(void); 56 typedef int (* RootFunc_t)(void); 57#ifdef RS_COMPATIBILITY_LIB 58 typedef void (*WorkerCallback_t)(void *usr, uint32_t idx); 59#endif 60 61 bool init(char const *resName, char const *cacheDir, 62 uint8_t const *bitcode, size_t bitcodeSize, uint32_t flags, 63 char const *bccPluginName = nullptr); 64 virtual void populateScript(Script *); 65 66 virtual void invokeFunction(uint32_t slot, const void *params, size_t paramLength); 67 virtual int invokeRoot(); 68 virtual void preLaunch(uint32_t slot, const Allocation ** ains, 69 uint32_t inLen, Allocation * aout, const void * usr, 70 uint32_t usrLen, const RsScriptCall *sc); 71 virtual void postLaunch(uint32_t slot, const Allocation ** ains, 72 uint32_t inLen, Allocation * aout, 73 const void * usr, uint32_t usrLen, 74 const RsScriptCall *sc); 75 76 virtual void invokeForEach(uint32_t slot, 77 const Allocation ** ains, 78 uint32_t inLen, 79 Allocation* aout, 80 const void* usr, 81 uint32_t usrLen, 82 const RsScriptCall* sc); 83 84 virtual void invokeInit(); 85 virtual void invokeFreeChildren(); 86 87 virtual void setGlobalVar(uint32_t slot, const void *data, size_t dataLength); 88 virtual void getGlobalVar(uint32_t slot, void *data, size_t dataLength); 89 virtual void setGlobalVarWithElemDims(uint32_t slot, const void *data, size_t dataLength, 90 const Element *e, const uint32_t *dims, size_t dimLength); 91 virtual void setGlobalBind(uint32_t slot, Allocation *data); 92 virtual void setGlobalObj(uint32_t slot, ObjectBase *data); 93 94 95 virtual ~RsdCpuScriptImpl(); 96 RsdCpuScriptImpl(RsdCpuReferenceImpl *ctx, const Script *s); 97 98 const Script * getScript() {return mScript;} 99 100 void forEachMtlsSetup(const Allocation ** ains, uint32_t inLen, 101 Allocation * aout, const void * usr, uint32_t usrLen, 102 const RsScriptCall *sc, MTLaunchStruct *mtls); 103 104 virtual void forEachKernelSetup(uint32_t slot, MTLaunchStruct *mtls); 105 106 107 const RsdCpuReference::CpuSymbol * lookupSymbolMath(const char *sym); 108 static void * lookupRuntimeStub(void* pContext, char const* name); 109 110 virtual Allocation * getAllocationForPointer(const void *ptr) const; 111 bool storeRSInfoFromSO(); 112 113#ifndef RS_COMPATIBILITY_LIB 114 bool storeRSInfoFromObj(bcinfo::MetadataExtractor &bitcodeMetadata); 115 virtual void * getRSExecutable() { return mExecutable; } 116#endif 117 118protected: 119 RsdCpuReferenceImpl *mCtx; 120 const Script *mScript; 121 void *mScriptSO; 122 123#ifndef RS_COMPATIBILITY_LIB 124 // Returns the path to the core library we'll use. 125 const char* findCoreLib(const bcinfo::MetadataExtractor& bitCodeMetaData, const char* bitcode, 126 size_t bitcodeSize); 127 RootFunc_t mRoot; 128 RootFunc_t mRootExpand; 129 InvokeFunc_t mInit; 130 InvokeFunc_t mFreeChildren; 131 132 InvokeFunc_t *mInvokeFunctions; 133 ForEachFunc_t *mForEachFunctions; 134 void **mFieldAddress; 135 bool *mFieldIsObject; 136 uint32_t *mForEachSignatures; 137 size_t mExportedVariableCount; 138 size_t mExportedFunctionCount; 139 140 std::vector<std::pair<const char *, uint32_t> > mExportedForEachFuncList; 141 142 bcc::RSCompilerDriver *mCompilerDriver; 143 bcc::CompilerRTSymbolResolver mCompilerRuntime; 144 bcc::LookupFunctionSymbolResolver<void *> mRSRuntime; 145 bcc::SymbolResolverProxy mResolver; 146 bcc::RSExecutable *mExecutable; 147#else 148 RootFunc_t mRoot; 149 RootFunc_t mRootExpand; 150 InvokeFunc_t mInit; 151 InvokeFunc_t mFreeChildren; 152 InvokeFunc_t *mInvokeFunctions; 153 ForEachFunc_t *mForEachFunctions; 154 155 void **mFieldAddress; 156 bool *mFieldIsObject; 157 uint32_t *mForEachSignatures; 158 159 // for populate script 160 //int mVersionMajor; 161 //int mVersionMinor; 162 size_t mExportedVariableCount; 163 size_t mExportedFunctionCount; 164#endif 165 166 Allocation **mBoundAllocs; 167 void * mIntrinsicData; 168 bool mIsThreadable; 169}; 170 171Allocation * rsdScriptGetAllocationForPointer( 172 const Context *dc, 173 const Script *script, 174 const void *); 175 176 177 178} 179} 180 181#endif 182