rsCpuCore.h revision b7d9c80c98fc96aa7c638e3124be24f13a6436b2
1/* 2 * Copyright (C) 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_CPU_CORE_H 18#define RSD_CPU_CORE_H 19 20#include "rsd_cpu.h" 21#include "rsSignal.h" 22#include "rsContext.h" 23#include "rsElement.h" 24#include "rsScriptC.h" 25 26namespace bcc { 27 class BCCContext; 28 class RSCompilerDriver; 29 class RSExecutable; 30} 31 32namespace android { 33namespace renderscript { 34 35 36typedef void (* InvokeFunc_t)(void); 37typedef void (* ForEachFunc_t)(void); 38typedef void (*WorkerCallback_t)(void *usr, uint32_t idx); 39 40class RsdCpuScriptImpl; 41class RsdCpuReferenceImpl; 42 43typedef struct ScriptTLSStructRec { 44 android::renderscript::Context * mContext; 45 const android::renderscript::Script * mScript; 46 RsdCpuScriptImpl *mImpl; 47} ScriptTLSStruct; 48 49typedef struct { 50 RsForEachStubParamStruct fep; 51 52 RsdCpuReferenceImpl *rsc; 53 RsdCpuScriptImpl *script; 54 55 ForEachFunc_t kernel; 56 uint32_t sig; 57 const Allocation * ain; 58 Allocation * aout; 59 60 uint32_t mSliceSize; 61 volatile int mSliceNum; 62 bool isThreadable; 63 64 uint32_t xStart; 65 uint32_t xEnd; 66 uint32_t yStart; 67 uint32_t yEnd; 68 uint32_t zStart; 69 uint32_t zEnd; 70 uint32_t arrayStart; 71 uint32_t arrayEnd; 72} MTLaunchStruct; 73 74 75 76 77class RsdCpuReferenceImpl : public RsdCpuReference { 78public: 79 virtual ~RsdCpuReferenceImpl(); 80 RsdCpuReferenceImpl(Context *); 81 82 void lockMutex(); 83 void unlockMutex(); 84 85 bool init(uint32_t version_major, uint32_t version_minor, sym_lookup_t, script_lookup_t); 86 virtual void setPriority(int32_t priority); 87 virtual void launchThreads(WorkerCallback_t cbk, void *data); 88 static void * helperThreadProc(void *vrsc); 89 RsdCpuScriptImpl * setTLS(RsdCpuScriptImpl *sc); 90 91 Context * getContext() {return mRSC;} 92 uint32_t getThreadCount() const { 93 return mWorkers.mCount + 1; 94 } 95 96 void launchThreads(const Allocation * ain, Allocation * aout, 97 const RsScriptCall *sc, MTLaunchStruct *mtls); 98 99 virtual CpuScript * createScript(const ScriptC *s, 100 char const *resName, char const *cacheDir, 101 uint8_t const *bitcode, size_t bitcodeSize, 102 uint32_t flags); 103 virtual CpuScript * createIntrinsic(const Script *s, 104 RsScriptIntrinsicID iid, Element *e); 105 virtual CpuScriptGroup * createScriptGroup(const ScriptGroup *sg); 106 107 const RsdCpuReference::CpuSymbol *symLookup(const char *); 108 109 RsdCpuReference::CpuScript * lookupScript(const Script *s) { 110 return mScriptLookupFn(mRSC, s); 111 } 112 113#ifndef RS_COMPATIBILITY_LIB 114 void setLinkRuntimeCallback( 115 bcc::RSLinkRuntimeCallback pLinkRuntimeCallback) { 116 mLinkRuntimeCallback = pLinkRuntimeCallback; 117 } 118 bcc::RSLinkRuntimeCallback getLinkRuntimeCallback() { 119 return mLinkRuntimeCallback; 120 } 121 122 void setSelectRTCallback(RSSelectRTCallback pSelectRTCallback) { 123 mSelectRTCallback = pSelectRTCallback; 124 } 125 RSSelectRTCallback getSelectRTCallback() { 126 return mSelectRTCallback; 127 } 128 129 virtual void setSetupCompilerCallback( 130 RSSetupCompilerCallback pSetupCompilerCallback) { 131 mSetupCompilerCallback = pSetupCompilerCallback; 132 } 133 virtual RSSetupCompilerCallback getSetupCompilerCallback() const { 134 return mSetupCompilerCallback; 135 } 136#endif 137 virtual bool getInForEach() { return mInForEach; } 138 139protected: 140 Context *mRSC; 141 uint32_t version_major; 142 uint32_t version_minor; 143 //bool mHasGraphics; 144 bool mInForEach; 145 146 struct Workers { 147 volatile int mRunningCount; 148 volatile int mLaunchCount; 149 uint32_t mCount; 150 pthread_t *mThreadId; 151 pid_t *mNativeThreadId; 152 Signal mCompleteSignal; 153 Signal *mLaunchSignals; 154 WorkerCallback_t mLaunchCallback; 155 void *mLaunchData; 156 }; 157 Workers mWorkers; 158 bool mExit; 159 sym_lookup_t mSymLookupFn; 160 script_lookup_t mScriptLookupFn; 161 162 ScriptTLSStruct mTlsStruct; 163 164#ifndef RS_COMPATIBILITY_LIB 165 bcc::RSLinkRuntimeCallback mLinkRuntimeCallback; 166 RSSelectRTCallback mSelectRTCallback; 167 RSSetupCompilerCallback mSetupCompilerCallback; 168#endif 169}; 170 171 172} 173} 174 175#endif 176