rsCpuScript.h revision 2abfcc6d129fe3defddef4540aa95cc445c03a7a
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 <utility> 25#endif 26 27#include "rsCpuCore.h" 28 29namespace bcc { 30 class BCCContext; 31 class RSCompilerDriver; 32} 33 34namespace bcinfo { 35 class MetadataExtractor; 36} 37 38namespace android { 39namespace renderscript { 40 41class ScriptExecutable; 42 43class RsdCpuScriptImpl : public RsdCpuReferenceImpl::CpuScript { 44public: 45 typedef void (*outer_foreach_t)( 46 const RsExpandKernelParams *, 47 uint32_t x1, uint32_t x2, 48 uint32_t outstep); 49 50 typedef void (* InvokeFunc_t)(void); 51 typedef void (* ForEachFunc_t)(void); 52 typedef int (* RootFunc_t)(void); 53#ifdef RS_COMPATIBILITY_LIB 54 typedef void (*WorkerCallback_t)(void *usr, uint32_t idx); 55#endif 56 57 bool init(char const *resName, char const *cacheDir, 58 uint8_t const *bitcode, size_t bitcodeSize, uint32_t flags, 59 char const *bccPluginName = nullptr); 60 virtual void populateScript(Script *); 61 62 virtual void invokeFunction(uint32_t slot, const void *params, size_t paramLength); 63 virtual int invokeRoot(); 64 virtual void preLaunch(uint32_t slot, const Allocation ** ains, 65 uint32_t inLen, Allocation * aout, const void * usr, 66 uint32_t usrLen, const RsScriptCall *sc); 67 virtual void postLaunch(uint32_t slot, const Allocation ** ains, 68 uint32_t inLen, Allocation * aout, 69 const void * usr, uint32_t usrLen, 70 const RsScriptCall *sc); 71 72 virtual void invokeForEach(uint32_t slot, 73 const Allocation ** ains, 74 uint32_t inLen, 75 Allocation* aout, 76 const void* usr, 77 uint32_t usrLen, 78 const RsScriptCall* sc); 79 80 virtual void invokeInit(); 81 virtual void invokeFreeChildren(); 82 83 virtual void setGlobalVar(uint32_t slot, const void *data, size_t dataLength); 84 virtual void getGlobalVar(uint32_t slot, void *data, size_t dataLength); 85 virtual void setGlobalVarWithElemDims(uint32_t slot, const void *data, size_t dataLength, 86 const Element *e, const uint32_t *dims, size_t dimLength); 87 virtual void setGlobalBind(uint32_t slot, Allocation *data); 88 virtual void setGlobalObj(uint32_t slot, ObjectBase *data); 89 90 91 virtual ~RsdCpuScriptImpl(); 92 RsdCpuScriptImpl(RsdCpuReferenceImpl *ctx, const Script *s); 93 94 const Script * getScript() {return mScript;} 95 96 bool forEachMtlsSetup(const Allocation ** ains, uint32_t inLen, 97 Allocation * aout, const void * usr, uint32_t usrLen, 98 const RsScriptCall *sc, MTLaunchStruct *mtls); 99 100 virtual void forEachKernelSetup(uint32_t slot, MTLaunchStruct *mtls); 101 102 103 const RsdCpuReference::CpuSymbol * lookupSymbolMath(const char *sym); 104 static void * lookupRuntimeStub(void* pContext, char const* name); 105 106 virtual Allocation * getAllocationForPointer(const void *ptr) const; 107 bool storeRSInfoFromSO(); 108 109protected: 110 RsdCpuReferenceImpl *mCtx; 111 const Script *mScript; 112 void *mScriptSO; 113 114#ifndef RS_COMPATIBILITY_LIB 115 // Returns the path to the core library we'll use. 116 const char* findCoreLib(const bcinfo::MetadataExtractor& bitCodeMetaData, const char* bitcode, 117 size_t bitcodeSize); 118 119 bcc::RSCompilerDriver *mCompilerDriver; 120#endif 121 122 RootFunc_t mRoot; 123 RootFunc_t mRootExpand; 124 InvokeFunc_t mInit; 125 InvokeFunc_t mFreeChildren; 126 ScriptExecutable* mScriptExec; 127 128 Allocation **mBoundAllocs; 129 void * mIntrinsicData; 130 bool mIsThreadable; 131 132public: 133 static const char* BCC_EXE_PATH; 134 const char* getBitcodeFilePath() const { return mBitcodeFilePath.string(); } 135 136private: 137 String8 mBitcodeFilePath; 138}; 139 140Allocation * rsdScriptGetAllocationForPointer( 141 const Context *dc, 142 const Script *script, 143 const void *); 144 145} 146 147#ifdef __LP64__ 148#define SYSLIBPATH "/system/lib64" 149#else 150#define SYSLIBPATH "/system/lib" 151#endif 152 153} 154 155#endif 156