rsCpuCore.cpp revision f218bf115af4ae4fd79adbb8842608b308a4cf07
1709a0978ae141198018ca9769f8d96292a8928e6Jason Sams/* 2709a0978ae141198018ca9769f8d96292a8928e6Jason Sams * Copyright (C) 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#include "rsCpuCore.h" 18709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#include "rsCpuScript.h" 19709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#include "rsCpuScriptGroup.h" 20709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 21709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#include <malloc.h> 22709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#include "rsContext.h" 23709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 24709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#include <sys/types.h> 25709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#include <sys/resource.h> 26709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#include <sched.h> 27709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#include <cutils/properties.h> 28709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#include <sys/syscall.h> 29709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#include <string.h> 30709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#include "utils/StopWatch.h" 31709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 32709a0978ae141198018ca9769f8d96292a8928e6Jason Samsusing namespace android; 33709a0978ae141198018ca9769f8d96292a8928e6Jason Samsusing namespace android::renderscript; 34709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 35709a0978ae141198018ca9769f8d96292a8928e6Jason Samstypedef void (*outer_foreach_t)( 36709a0978ae141198018ca9769f8d96292a8928e6Jason Sams const android::renderscript::RsForEachStubParamStruct *, 37709a0978ae141198018ca9769f8d96292a8928e6Jason Sams uint32_t x1, uint32_t x2, 38709a0978ae141198018ca9769f8d96292a8928e6Jason Sams uint32_t instep, uint32_t outstep); 39709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 40709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 41709a0978ae141198018ca9769f8d96292a8928e6Jason Samsstatic pthread_key_t gThreadTLSKey = 0; 42709a0978ae141198018ca9769f8d96292a8928e6Jason Samsstatic uint32_t gThreadTLSKeyCount = 0; 43709a0978ae141198018ca9769f8d96292a8928e6Jason Samsstatic pthread_mutex_t gInitMutex = PTHREAD_MUTEX_INITIALIZER; 44709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 45709a0978ae141198018ca9769f8d96292a8928e6Jason SamsRsdCpuReference::~RsdCpuReference() { 46709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 47709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 48709a0978ae141198018ca9769f8d96292a8928e6Jason SamsRsdCpuReference * RsdCpuReference::create(Context *rsc, uint32_t version_major, 49f218bf115af4ae4fd79adbb8842608b308a4cf07Stephen Hines uint32_t version_minor, sym_lookup_t lfn, script_lookup_t slfn, 50f218bf115af4ae4fd79adbb8842608b308a4cf07Stephen Hines bcc::RSLinkRuntimeCallback pLinkRuntimeCallback) { 51709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 52709a0978ae141198018ca9769f8d96292a8928e6Jason Sams RsdCpuReferenceImpl *cpu = new RsdCpuReferenceImpl(rsc); 53709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if (!cpu) { 54709a0978ae141198018ca9769f8d96292a8928e6Jason Sams return NULL; 55709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 56709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if (!cpu->init(version_major, version_minor, lfn, slfn)) { 57709a0978ae141198018ca9769f8d96292a8928e6Jason Sams delete cpu; 58709a0978ae141198018ca9769f8d96292a8928e6Jason Sams return NULL; 59709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 60f218bf115af4ae4fd79adbb8842608b308a4cf07Stephen Hines 61f218bf115af4ae4fd79adbb8842608b308a4cf07Stephen Hines cpu->setLinkRuntimeCallback(pLinkRuntimeCallback); 62f218bf115af4ae4fd79adbb8842608b308a4cf07Stephen Hines 63709a0978ae141198018ca9769f8d96292a8928e6Jason Sams return cpu; 64709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 65709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 66709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 67709a0978ae141198018ca9769f8d96292a8928e6Jason SamsContext * RsdCpuReference::getTlsContext() { 68709a0978ae141198018ca9769f8d96292a8928e6Jason Sams ScriptTLSStruct * tls = (ScriptTLSStruct *)pthread_getspecific(gThreadTLSKey); 69709a0978ae141198018ca9769f8d96292a8928e6Jason Sams return tls->mContext; 70709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 71709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 72709a0978ae141198018ca9769f8d96292a8928e6Jason Samsconst Script * RsdCpuReference::getTlsScript() { 73709a0978ae141198018ca9769f8d96292a8928e6Jason Sams ScriptTLSStruct * tls = (ScriptTLSStruct *)pthread_getspecific(gThreadTLSKey); 74709a0978ae141198018ca9769f8d96292a8928e6Jason Sams return tls->mScript; 75709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 76709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 77f218bf115af4ae4fd79adbb8842608b308a4cf07Stephen Hinespthread_key_t RsdCpuReference::getThreadTLSKey(){ return gThreadTLSKey; } 78709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 79709a0978ae141198018ca9769f8d96292a8928e6Jason Sams//////////////////////////////////////////////////////////// 80709a0978ae141198018ca9769f8d96292a8928e6Jason Sams/// 81709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 82709a0978ae141198018ca9769f8d96292a8928e6Jason SamsRsdCpuReferenceImpl::RsdCpuReferenceImpl(Context *rsc) { 83709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mRSC = rsc; 84709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 85709a0978ae141198018ca9769f8d96292a8928e6Jason Sams version_major = 0; 86709a0978ae141198018ca9769f8d96292a8928e6Jason Sams version_minor = 0; 87709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mInForEach = false; 88709a0978ae141198018ca9769f8d96292a8928e6Jason Sams memset(&mWorkers, 0, sizeof(mWorkers)); 89709a0978ae141198018ca9769f8d96292a8928e6Jason Sams memset(&mTlsStruct, 0, sizeof(mTlsStruct)); 90709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mExit = false; 91f218bf115af4ae4fd79adbb8842608b308a4cf07Stephen Hines mLinkRuntimeCallback = NULL; 92709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 93709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 94709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 95709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 96709a0978ae141198018ca9769f8d96292a8928e6Jason Samsvoid * RsdCpuReferenceImpl::helperThreadProc(void *vrsc) { 97709a0978ae141198018ca9769f8d96292a8928e6Jason Sams RsdCpuReferenceImpl *dc = (RsdCpuReferenceImpl *)vrsc; 98709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 99709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 100709a0978ae141198018ca9769f8d96292a8928e6Jason Sams uint32_t idx = (uint32_t)android_atomic_inc(&dc->mWorkers.mLaunchCount); 101709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 102709a0978ae141198018ca9769f8d96292a8928e6Jason Sams //ALOGV("RS helperThread starting %p idx=%i", dc, idx); 103709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 104709a0978ae141198018ca9769f8d96292a8928e6Jason Sams dc->mWorkers.mLaunchSignals[idx].init(); 105709a0978ae141198018ca9769f8d96292a8928e6Jason Sams dc->mWorkers.mNativeThreadId[idx] = gettid(); 106709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 107709a0978ae141198018ca9769f8d96292a8928e6Jason Sams memset(&dc->mTlsStruct, 0, sizeof(dc->mTlsStruct)); 108709a0978ae141198018ca9769f8d96292a8928e6Jason Sams int status = pthread_setspecific(gThreadTLSKey, &dc->mTlsStruct); 109709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if (status) { 110709a0978ae141198018ca9769f8d96292a8928e6Jason Sams ALOGE("pthread_setspecific %i", status); 111709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 112709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 113709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#if 0 114709a0978ae141198018ca9769f8d96292a8928e6Jason Sams typedef struct {uint64_t bits[1024 / 64]; } cpu_set_t; 115709a0978ae141198018ca9769f8d96292a8928e6Jason Sams cpu_set_t cpuset; 116709a0978ae141198018ca9769f8d96292a8928e6Jason Sams memset(&cpuset, 0, sizeof(cpuset)); 117709a0978ae141198018ca9769f8d96292a8928e6Jason Sams cpuset.bits[idx / 64] |= 1ULL << (idx % 64); 118709a0978ae141198018ca9769f8d96292a8928e6Jason Sams int ret = syscall(241, rsc->mWorkers.mNativeThreadId[idx], 119709a0978ae141198018ca9769f8d96292a8928e6Jason Sams sizeof(cpuset), &cpuset); 120709a0978ae141198018ca9769f8d96292a8928e6Jason Sams ALOGE("SETAFFINITY ret = %i %s", ret, EGLUtils::strerror(ret)); 121709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#endif 122709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 123709a0978ae141198018ca9769f8d96292a8928e6Jason Sams while (!dc->mExit) { 124709a0978ae141198018ca9769f8d96292a8928e6Jason Sams dc->mWorkers.mLaunchSignals[idx].wait(); 125709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if (dc->mWorkers.mLaunchCallback) { 126709a0978ae141198018ca9769f8d96292a8928e6Jason Sams // idx +1 is used because the calling thread is always worker 0. 127709a0978ae141198018ca9769f8d96292a8928e6Jason Sams dc->mWorkers.mLaunchCallback(dc->mWorkers.mLaunchData, idx+1); 128709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 129709a0978ae141198018ca9769f8d96292a8928e6Jason Sams android_atomic_dec(&dc->mWorkers.mRunningCount); 130709a0978ae141198018ca9769f8d96292a8928e6Jason Sams dc->mWorkers.mCompleteSignal.set(); 131709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 132709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 133709a0978ae141198018ca9769f8d96292a8928e6Jason Sams //ALOGV("RS helperThread exited %p idx=%i", dc, idx); 134709a0978ae141198018ca9769f8d96292a8928e6Jason Sams return NULL; 135709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 136709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 137709a0978ae141198018ca9769f8d96292a8928e6Jason Samsvoid RsdCpuReferenceImpl::launchThreads(WorkerCallback_t cbk, void *data) { 138709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mWorkers.mLaunchData = data; 139709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mWorkers.mLaunchCallback = cbk; 1404d252d6e807b89764dad123ac845df298c52ca97Tim Murray 1414d252d6e807b89764dad123ac845df298c52ca97Tim Murray // fast path for very small launches 1424d252d6e807b89764dad123ac845df298c52ca97Tim Murray MTLaunchStruct *mtls = (MTLaunchStruct *)data; 1434d252d6e807b89764dad123ac845df298c52ca97Tim Murray if (mtls && mtls->fep.dimY <= 1 && mtls->xEnd <= mtls->xStart + mtls->mSliceSize) { 1444d252d6e807b89764dad123ac845df298c52ca97Tim Murray if (mWorkers.mLaunchCallback) { 1454d252d6e807b89764dad123ac845df298c52ca97Tim Murray mWorkers.mLaunchCallback(mWorkers.mLaunchData, 0); 1464d252d6e807b89764dad123ac845df298c52ca97Tim Murray } 1474d252d6e807b89764dad123ac845df298c52ca97Tim Murray return; 1484d252d6e807b89764dad123ac845df298c52ca97Tim Murray } 1494d252d6e807b89764dad123ac845df298c52ca97Tim Murray 150709a0978ae141198018ca9769f8d96292a8928e6Jason Sams android_atomic_release_store(mWorkers.mCount, &mWorkers.mRunningCount); 151709a0978ae141198018ca9769f8d96292a8928e6Jason Sams for (uint32_t ct = 0; ct < mWorkers.mCount; ct++) { 152709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mWorkers.mLaunchSignals[ct].set(); 153709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 154709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 155709a0978ae141198018ca9769f8d96292a8928e6Jason Sams // We use the calling thread as one of the workers so we can start without 156709a0978ae141198018ca9769f8d96292a8928e6Jason Sams // the delay of the thread wakeup. 157709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if (mWorkers.mLaunchCallback) { 1584d252d6e807b89764dad123ac845df298c52ca97Tim Murray mWorkers.mLaunchCallback(mWorkers.mLaunchData, 0); 159709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 160709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 161709a0978ae141198018ca9769f8d96292a8928e6Jason Sams while (android_atomic_acquire_load(&mWorkers.mRunningCount) != 0) { 162709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mWorkers.mCompleteSignal.wait(); 163709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 164709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 165709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 166709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 167709a0978ae141198018ca9769f8d96292a8928e6Jason Samsvoid RsdCpuReferenceImpl::lockMutex() { 168709a0978ae141198018ca9769f8d96292a8928e6Jason Sams pthread_mutex_lock(&gInitMutex); 169709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 170709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 171709a0978ae141198018ca9769f8d96292a8928e6Jason Samsvoid RsdCpuReferenceImpl::unlockMutex() { 172709a0978ae141198018ca9769f8d96292a8928e6Jason Sams pthread_mutex_unlock(&gInitMutex); 173709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 174709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 175709a0978ae141198018ca9769f8d96292a8928e6Jason Samsbool RsdCpuReferenceImpl::init(uint32_t version_major, uint32_t version_minor, 176709a0978ae141198018ca9769f8d96292a8928e6Jason Sams sym_lookup_t lfn, script_lookup_t slfn) { 177709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 178709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mSymLookupFn = lfn; 179709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mScriptLookupFn = slfn; 180709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 181709a0978ae141198018ca9769f8d96292a8928e6Jason Sams lockMutex(); 182709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if (!gThreadTLSKeyCount) { 183709a0978ae141198018ca9769f8d96292a8928e6Jason Sams int status = pthread_key_create(&gThreadTLSKey, NULL); 184709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if (status) { 185709a0978ae141198018ca9769f8d96292a8928e6Jason Sams ALOGE("Failed to init thread tls key."); 186709a0978ae141198018ca9769f8d96292a8928e6Jason Sams unlockMutex(); 187709a0978ae141198018ca9769f8d96292a8928e6Jason Sams return false; 188709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 189709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 190709a0978ae141198018ca9769f8d96292a8928e6Jason Sams gThreadTLSKeyCount++; 191709a0978ae141198018ca9769f8d96292a8928e6Jason Sams unlockMutex(); 192709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 193709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mTlsStruct.mContext = mRSC; 194709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mTlsStruct.mScript = NULL; 195709a0978ae141198018ca9769f8d96292a8928e6Jason Sams int status = pthread_setspecific(gThreadTLSKey, &mTlsStruct); 196709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if (status) { 197709a0978ae141198018ca9769f8d96292a8928e6Jason Sams ALOGE("pthread_setspecific %i", status); 198709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 199709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 200709a0978ae141198018ca9769f8d96292a8928e6Jason Sams int cpu = sysconf(_SC_NPROCESSORS_ONLN); 201709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if(mRSC->props.mDebugMaxThreads) { 202709a0978ae141198018ca9769f8d96292a8928e6Jason Sams cpu = mRSC->props.mDebugMaxThreads; 203709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 204709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if (cpu < 2) { 205709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mWorkers.mCount = 0; 206709a0978ae141198018ca9769f8d96292a8928e6Jason Sams return true; 207709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 208709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 209709a0978ae141198018ca9769f8d96292a8928e6Jason Sams // Subtract one from the cpu count because we also use the command thread as a worker. 210709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mWorkers.mCount = (uint32_t)(cpu - 1); 211709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 212709a0978ae141198018ca9769f8d96292a8928e6Jason Sams ALOGV("%p Launching thread(s), CPUs %i", mRSC, mWorkers.mCount); 213709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 214709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mWorkers.mThreadId = (pthread_t *) calloc(mWorkers.mCount, sizeof(pthread_t)); 215709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mWorkers.mNativeThreadId = (pid_t *) calloc(mWorkers.mCount, sizeof(pid_t)); 216709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mWorkers.mLaunchSignals = new Signal[mWorkers.mCount]; 217709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mWorkers.mLaunchCallback = NULL; 218709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 219709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mWorkers.mCompleteSignal.init(); 220709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 221709a0978ae141198018ca9769f8d96292a8928e6Jason Sams android_atomic_release_store(mWorkers.mCount, &mWorkers.mRunningCount); 222709a0978ae141198018ca9769f8d96292a8928e6Jason Sams android_atomic_release_store(0, &mWorkers.mLaunchCount); 223709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 224709a0978ae141198018ca9769f8d96292a8928e6Jason Sams pthread_attr_t threadAttr; 225709a0978ae141198018ca9769f8d96292a8928e6Jason Sams status = pthread_attr_init(&threadAttr); 226709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if (status) { 227709a0978ae141198018ca9769f8d96292a8928e6Jason Sams ALOGE("Failed to init thread attribute."); 228709a0978ae141198018ca9769f8d96292a8928e6Jason Sams return false; 229709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 230709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 231709a0978ae141198018ca9769f8d96292a8928e6Jason Sams for (uint32_t ct=0; ct < mWorkers.mCount; ct++) { 232709a0978ae141198018ca9769f8d96292a8928e6Jason Sams status = pthread_create(&mWorkers.mThreadId[ct], &threadAttr, helperThreadProc, this); 233709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if (status) { 234709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mWorkers.mCount = ct; 235709a0978ae141198018ca9769f8d96292a8928e6Jason Sams ALOGE("Created fewer than expected number of RS threads."); 236709a0978ae141198018ca9769f8d96292a8928e6Jason Sams break; 237709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 238709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 239709a0978ae141198018ca9769f8d96292a8928e6Jason Sams while (android_atomic_acquire_load(&mWorkers.mRunningCount) != 0) { 240709a0978ae141198018ca9769f8d96292a8928e6Jason Sams usleep(100); 241709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 242709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 243709a0978ae141198018ca9769f8d96292a8928e6Jason Sams pthread_attr_destroy(&threadAttr); 244709a0978ae141198018ca9769f8d96292a8928e6Jason Sams return true; 245709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 246709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 247709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 248709a0978ae141198018ca9769f8d96292a8928e6Jason Samsvoid RsdCpuReferenceImpl::setPriority(int32_t priority) { 249709a0978ae141198018ca9769f8d96292a8928e6Jason Sams for (uint32_t ct=0; ct < mWorkers.mCount; ct++) { 250709a0978ae141198018ca9769f8d96292a8928e6Jason Sams setpriority(PRIO_PROCESS, mWorkers.mNativeThreadId[ct], priority); 251709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 252709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 253709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 254709a0978ae141198018ca9769f8d96292a8928e6Jason SamsRsdCpuReferenceImpl::~RsdCpuReferenceImpl() { 255709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mExit = true; 256709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mWorkers.mLaunchData = NULL; 257709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mWorkers.mLaunchCallback = NULL; 258709a0978ae141198018ca9769f8d96292a8928e6Jason Sams android_atomic_release_store(mWorkers.mCount, &mWorkers.mRunningCount); 259709a0978ae141198018ca9769f8d96292a8928e6Jason Sams for (uint32_t ct = 0; ct < mWorkers.mCount; ct++) { 260709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mWorkers.mLaunchSignals[ct].set(); 261709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 262709a0978ae141198018ca9769f8d96292a8928e6Jason Sams void *res; 263709a0978ae141198018ca9769f8d96292a8928e6Jason Sams for (uint32_t ct = 0; ct < mWorkers.mCount; ct++) { 264709a0978ae141198018ca9769f8d96292a8928e6Jason Sams pthread_join(mWorkers.mThreadId[ct], &res); 265709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 266709a0978ae141198018ca9769f8d96292a8928e6Jason Sams rsAssert(android_atomic_acquire_load(&mWorkers.mRunningCount) == 0); 267709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 268709a0978ae141198018ca9769f8d96292a8928e6Jason Sams // Global structure cleanup. 269709a0978ae141198018ca9769f8d96292a8928e6Jason Sams lockMutex(); 270709a0978ae141198018ca9769f8d96292a8928e6Jason Sams --gThreadTLSKeyCount; 271709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if (!gThreadTLSKeyCount) { 272709a0978ae141198018ca9769f8d96292a8928e6Jason Sams pthread_key_delete(gThreadTLSKey); 273709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 274709a0978ae141198018ca9769f8d96292a8928e6Jason Sams unlockMutex(); 275709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 276709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 277709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 278709a0978ae141198018ca9769f8d96292a8928e6Jason Samstypedef void (*rs_t)(const void *, void *, const void *, uint32_t, uint32_t, uint32_t, uint32_t); 279709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 280709a0978ae141198018ca9769f8d96292a8928e6Jason Samsstatic void wc_xy(void *usr, uint32_t idx) { 281709a0978ae141198018ca9769f8d96292a8928e6Jason Sams MTLaunchStruct *mtls = (MTLaunchStruct *)usr; 282709a0978ae141198018ca9769f8d96292a8928e6Jason Sams RsForEachStubParamStruct p; 283709a0978ae141198018ca9769f8d96292a8928e6Jason Sams memcpy(&p, &mtls->fep, sizeof(p)); 284709a0978ae141198018ca9769f8d96292a8928e6Jason Sams p.lid = idx; 285709a0978ae141198018ca9769f8d96292a8928e6Jason Sams uint32_t sig = mtls->sig; 286709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 287709a0978ae141198018ca9769f8d96292a8928e6Jason Sams outer_foreach_t fn = (outer_foreach_t) mtls->kernel; 288709a0978ae141198018ca9769f8d96292a8928e6Jason Sams while (1) { 289709a0978ae141198018ca9769f8d96292a8928e6Jason Sams uint32_t slice = (uint32_t)android_atomic_inc(&mtls->mSliceNum); 290709a0978ae141198018ca9769f8d96292a8928e6Jason Sams uint32_t yStart = mtls->yStart + slice * mtls->mSliceSize; 291709a0978ae141198018ca9769f8d96292a8928e6Jason Sams uint32_t yEnd = yStart + mtls->mSliceSize; 292709a0978ae141198018ca9769f8d96292a8928e6Jason Sams yEnd = rsMin(yEnd, mtls->yEnd); 293709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if (yEnd <= yStart) { 294709a0978ae141198018ca9769f8d96292a8928e6Jason Sams return; 295709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 296709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 297709a0978ae141198018ca9769f8d96292a8928e6Jason Sams //ALOGE("usr idx %i, x %i,%i y %i,%i", idx, mtls->xStart, mtls->xEnd, yStart, yEnd); 298709a0978ae141198018ca9769f8d96292a8928e6Jason Sams //ALOGE("usr ptr in %p, out %p", mtls->fep.ptrIn, mtls->fep.ptrOut); 299709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 300709a0978ae141198018ca9769f8d96292a8928e6Jason Sams for (p.y = yStart; p.y < yEnd; p.y++) { 301709a0978ae141198018ca9769f8d96292a8928e6Jason Sams p.out = mtls->fep.ptrOut + (mtls->fep.yStrideOut * p.y) + 302709a0978ae141198018ca9769f8d96292a8928e6Jason Sams (mtls->fep.eStrideOut * mtls->xStart); 303709a0978ae141198018ca9769f8d96292a8928e6Jason Sams p.in = mtls->fep.ptrIn + (mtls->fep.yStrideIn * p.y) + 304709a0978ae141198018ca9769f8d96292a8928e6Jason Sams (mtls->fep.eStrideIn * mtls->xStart); 305709a0978ae141198018ca9769f8d96292a8928e6Jason Sams fn(&p, mtls->xStart, mtls->xEnd, mtls->fep.eStrideIn, mtls->fep.eStrideOut); 306709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 307709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 308709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 309709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 310709a0978ae141198018ca9769f8d96292a8928e6Jason Samsstatic void wc_x(void *usr, uint32_t idx) { 311709a0978ae141198018ca9769f8d96292a8928e6Jason Sams MTLaunchStruct *mtls = (MTLaunchStruct *)usr; 312709a0978ae141198018ca9769f8d96292a8928e6Jason Sams RsForEachStubParamStruct p; 313709a0978ae141198018ca9769f8d96292a8928e6Jason Sams memcpy(&p, &mtls->fep, sizeof(p)); 314709a0978ae141198018ca9769f8d96292a8928e6Jason Sams p.lid = idx; 315709a0978ae141198018ca9769f8d96292a8928e6Jason Sams uint32_t sig = mtls->sig; 316709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 317709a0978ae141198018ca9769f8d96292a8928e6Jason Sams outer_foreach_t fn = (outer_foreach_t) mtls->kernel; 318709a0978ae141198018ca9769f8d96292a8928e6Jason Sams while (1) { 319709a0978ae141198018ca9769f8d96292a8928e6Jason Sams uint32_t slice = (uint32_t)android_atomic_inc(&mtls->mSliceNum); 320709a0978ae141198018ca9769f8d96292a8928e6Jason Sams uint32_t xStart = mtls->xStart + slice * mtls->mSliceSize; 321709a0978ae141198018ca9769f8d96292a8928e6Jason Sams uint32_t xEnd = xStart + mtls->mSliceSize; 322709a0978ae141198018ca9769f8d96292a8928e6Jason Sams xEnd = rsMin(xEnd, mtls->xEnd); 323709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if (xEnd <= xStart) { 324709a0978ae141198018ca9769f8d96292a8928e6Jason Sams return; 325709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 326709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 327709a0978ae141198018ca9769f8d96292a8928e6Jason Sams //ALOGE("usr slice %i idx %i, x %i,%i", slice, idx, xStart, xEnd); 328709a0978ae141198018ca9769f8d96292a8928e6Jason Sams //ALOGE("usr ptr in %p, out %p", mtls->fep.ptrIn, mtls->fep.ptrOut); 329709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 330709a0978ae141198018ca9769f8d96292a8928e6Jason Sams p.out = mtls->fep.ptrOut + (mtls->fep.eStrideOut * xStart); 331709a0978ae141198018ca9769f8d96292a8928e6Jason Sams p.in = mtls->fep.ptrIn + (mtls->fep.eStrideIn * xStart); 332709a0978ae141198018ca9769f8d96292a8928e6Jason Sams fn(&p, xStart, xEnd, mtls->fep.eStrideIn, mtls->fep.eStrideOut); 333709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 334709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 335709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 336709a0978ae141198018ca9769f8d96292a8928e6Jason Samsvoid RsdCpuReferenceImpl::launchThreads(const Allocation * ain, Allocation * aout, 337709a0978ae141198018ca9769f8d96292a8928e6Jason Sams const RsScriptCall *sc, MTLaunchStruct *mtls) { 338709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 339709a0978ae141198018ca9769f8d96292a8928e6Jason Sams //android::StopWatch kernel_time("kernel time"); 340709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 341709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if ((mWorkers.mCount >= 1) && mtls->isThreadable && !mInForEach) { 342709a0978ae141198018ca9769f8d96292a8928e6Jason Sams const size_t targetByteChunk = 16 * 1024; 343709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mInForEach = true; 344709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if (mtls->fep.dimY > 1) { 345709a0978ae141198018ca9769f8d96292a8928e6Jason Sams uint32_t s1 = mtls->fep.dimY / ((mWorkers.mCount + 1) * 4); 346709a0978ae141198018ca9769f8d96292a8928e6Jason Sams uint32_t s2 = 0; 347709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 348709a0978ae141198018ca9769f8d96292a8928e6Jason Sams // This chooses our slice size to rate limit atomic ops to 349709a0978ae141198018ca9769f8d96292a8928e6Jason Sams // one per 16k bytes of reads/writes. 350709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if (mtls->fep.yStrideOut) { 351709a0978ae141198018ca9769f8d96292a8928e6Jason Sams s2 = targetByteChunk / mtls->fep.yStrideOut; 352709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } else { 353709a0978ae141198018ca9769f8d96292a8928e6Jason Sams s2 = targetByteChunk / mtls->fep.yStrideIn; 354709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 355709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mtls->mSliceSize = rsMin(s1, s2); 356709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 357709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if(mtls->mSliceSize < 1) { 358709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mtls->mSliceSize = 1; 359709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 360709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 361709a0978ae141198018ca9769f8d96292a8928e6Jason Sams // mtls->mSliceSize = 2; 362709a0978ae141198018ca9769f8d96292a8928e6Jason Sams launchThreads(wc_xy, mtls); 363709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } else { 364709a0978ae141198018ca9769f8d96292a8928e6Jason Sams uint32_t s1 = mtls->fep.dimX / ((mWorkers.mCount + 1) * 4); 365709a0978ae141198018ca9769f8d96292a8928e6Jason Sams uint32_t s2 = 0; 366709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 367709a0978ae141198018ca9769f8d96292a8928e6Jason Sams // This chooses our slice size to rate limit atomic ops to 368709a0978ae141198018ca9769f8d96292a8928e6Jason Sams // one per 16k bytes of reads/writes. 369709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if (mtls->fep.eStrideOut) { 370709a0978ae141198018ca9769f8d96292a8928e6Jason Sams s2 = targetByteChunk / mtls->fep.eStrideOut; 371709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } else { 372709a0978ae141198018ca9769f8d96292a8928e6Jason Sams s2 = targetByteChunk / mtls->fep.eStrideIn; 373709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 374709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mtls->mSliceSize = rsMin(s1, s2); 375709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 376709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if(mtls->mSliceSize < 1) { 377709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mtls->mSliceSize = 1; 378709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 379709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 380709a0978ae141198018ca9769f8d96292a8928e6Jason Sams launchThreads(wc_x, mtls); 381709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 382709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mInForEach = false; 383709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 384709a0978ae141198018ca9769f8d96292a8928e6Jason Sams //ALOGE("launch 1"); 385709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } else { 386709a0978ae141198018ca9769f8d96292a8928e6Jason Sams RsForEachStubParamStruct p; 387709a0978ae141198018ca9769f8d96292a8928e6Jason Sams memcpy(&p, &mtls->fep, sizeof(p)); 388709a0978ae141198018ca9769f8d96292a8928e6Jason Sams uint32_t sig = mtls->sig; 389709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 390709a0978ae141198018ca9769f8d96292a8928e6Jason Sams //ALOGE("launch 3"); 391709a0978ae141198018ca9769f8d96292a8928e6Jason Sams outer_foreach_t fn = (outer_foreach_t) mtls->kernel; 392709a0978ae141198018ca9769f8d96292a8928e6Jason Sams for (p.ar[0] = mtls->arrayStart; p.ar[0] < mtls->arrayEnd; p.ar[0]++) { 393709a0978ae141198018ca9769f8d96292a8928e6Jason Sams for (p.z = mtls->zStart; p.z < mtls->zEnd; p.z++) { 394709a0978ae141198018ca9769f8d96292a8928e6Jason Sams for (p.y = mtls->yStart; p.y < mtls->yEnd; p.y++) { 395709a0978ae141198018ca9769f8d96292a8928e6Jason Sams uint32_t offset = mtls->fep.dimY * mtls->fep.dimZ * p.ar[0] + 396709a0978ae141198018ca9769f8d96292a8928e6Jason Sams mtls->fep.dimY * p.z + p.y; 397709a0978ae141198018ca9769f8d96292a8928e6Jason Sams p.out = mtls->fep.ptrOut + (mtls->fep.yStrideOut * offset) + 398709a0978ae141198018ca9769f8d96292a8928e6Jason Sams (mtls->fep.eStrideOut * mtls->xStart); 399709a0978ae141198018ca9769f8d96292a8928e6Jason Sams p.in = mtls->fep.ptrIn + (mtls->fep.yStrideIn * offset) + 400709a0978ae141198018ca9769f8d96292a8928e6Jason Sams (mtls->fep.eStrideIn * mtls->xStart); 401709a0978ae141198018ca9769f8d96292a8928e6Jason Sams fn(&p, mtls->xStart, mtls->xEnd, mtls->fep.eStrideIn, mtls->fep.eStrideOut); 402709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 403709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 404709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 405709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 406709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 407709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 408709a0978ae141198018ca9769f8d96292a8928e6Jason SamsRsdCpuScriptImpl * RsdCpuReferenceImpl::setTLS(RsdCpuScriptImpl *sc) { 409709a0978ae141198018ca9769f8d96292a8928e6Jason Sams //ALOGE("setTls %p", sc); 410709a0978ae141198018ca9769f8d96292a8928e6Jason Sams ScriptTLSStruct * tls = (ScriptTLSStruct *)pthread_getspecific(gThreadTLSKey); 411709a0978ae141198018ca9769f8d96292a8928e6Jason Sams rsAssert(tls); 412709a0978ae141198018ca9769f8d96292a8928e6Jason Sams RsdCpuScriptImpl *old = tls->mImpl; 413709a0978ae141198018ca9769f8d96292a8928e6Jason Sams tls->mImpl = sc; 414709a0978ae141198018ca9769f8d96292a8928e6Jason Sams tls->mContext = mRSC; 415709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if (sc) { 416709a0978ae141198018ca9769f8d96292a8928e6Jason Sams tls->mScript = sc->getScript(); 417709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } else { 418709a0978ae141198018ca9769f8d96292a8928e6Jason Sams tls->mScript = NULL; 419709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 420709a0978ae141198018ca9769f8d96292a8928e6Jason Sams return old; 421709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 422709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 423709a0978ae141198018ca9769f8d96292a8928e6Jason Samsconst RsdCpuReference::CpuSymbol * RsdCpuReferenceImpl::symLookup(const char *name) { 424709a0978ae141198018ca9769f8d96292a8928e6Jason Sams return mSymLookupFn(mRSC, name); 425709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 426709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 427709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 428709a0978ae141198018ca9769f8d96292a8928e6Jason SamsRsdCpuReference::CpuScript * RsdCpuReferenceImpl::createScript(const ScriptC *s, 429709a0978ae141198018ca9769f8d96292a8928e6Jason Sams char const *resName, char const *cacheDir, 430709a0978ae141198018ca9769f8d96292a8928e6Jason Sams uint8_t const *bitcode, size_t bitcodeSize, 431709a0978ae141198018ca9769f8d96292a8928e6Jason Sams uint32_t flags) { 432709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 433709a0978ae141198018ca9769f8d96292a8928e6Jason Sams RsdCpuScriptImpl *i = new RsdCpuScriptImpl(this, s); 434709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if (!i->init(resName, cacheDir, bitcode, bitcodeSize, flags)) { 435709a0978ae141198018ca9769f8d96292a8928e6Jason Sams delete i; 436709a0978ae141198018ca9769f8d96292a8928e6Jason Sams return NULL; 437709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 438709a0978ae141198018ca9769f8d96292a8928e6Jason Sams return i; 439709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 440709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 4417c4b888f2147edf99690b6af75470774ff31c43bJason Samsextern RsdCpuScriptImpl * rsdIntrinsic_3DLUT(RsdCpuReferenceImpl *ctx, 4427c4b888f2147edf99690b6af75470774ff31c43bJason Sams const Script *s, const Element *e); 443c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Samsextern RsdCpuScriptImpl * rsdIntrinsic_Convolve3x3(RsdCpuReferenceImpl *ctx, 444c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams const Script *s, const Element *e); 445c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Samsextern RsdCpuScriptImpl * rsdIntrinsic_ColorMatrix(RsdCpuReferenceImpl *ctx, 446c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams const Script *s, const Element *e); 447c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Samsextern RsdCpuScriptImpl * rsdIntrinsic_LUT(RsdCpuReferenceImpl *ctx, 448c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams const Script *s, const Element *e); 449c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Samsextern RsdCpuScriptImpl * rsdIntrinsic_Convolve5x5(RsdCpuReferenceImpl *ctx, 450c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams const Script *s, const Element *e); 451c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Samsextern RsdCpuScriptImpl * rsdIntrinsic_Blur(RsdCpuReferenceImpl *ctx, 452c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams const Script *s, const Element *e); 453c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Samsextern RsdCpuScriptImpl * rsdIntrinsic_YuvToRGB(RsdCpuReferenceImpl *ctx, 454c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams const Script *s, const Element *e); 455c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Samsextern RsdCpuScriptImpl * rsdIntrinsic_Blend(RsdCpuReferenceImpl *ctx, 456c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams const Script *s, const Element *e); 457709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 458709a0978ae141198018ca9769f8d96292a8928e6Jason SamsRsdCpuReference::CpuScript * RsdCpuReferenceImpl::createIntrinsic(const Script *s, 459709a0978ae141198018ca9769f8d96292a8928e6Jason Sams RsScriptIntrinsicID iid, Element *e) { 460709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 461709a0978ae141198018ca9769f8d96292a8928e6Jason Sams RsdCpuScriptImpl *i = NULL; 462709a0978ae141198018ca9769f8d96292a8928e6Jason Sams switch (iid) { 4637c4b888f2147edf99690b6af75470774ff31c43bJason Sams case RS_SCRIPT_INTRINSIC_ID_3DLUT: 4647c4b888f2147edf99690b6af75470774ff31c43bJason Sams i = rsdIntrinsic_3DLUT(this, s, e); 4657c4b888f2147edf99690b6af75470774ff31c43bJason Sams break; 466709a0978ae141198018ca9769f8d96292a8928e6Jason Sams case RS_SCRIPT_INTRINSIC_ID_CONVOLVE_3x3: 467c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams i = rsdIntrinsic_Convolve3x3(this, s, e); 468709a0978ae141198018ca9769f8d96292a8928e6Jason Sams break; 469709a0978ae141198018ca9769f8d96292a8928e6Jason Sams case RS_SCRIPT_INTRINSIC_ID_COLOR_MATRIX: 470c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams i = rsdIntrinsic_ColorMatrix(this, s, e); 471709a0978ae141198018ca9769f8d96292a8928e6Jason Sams break; 472709a0978ae141198018ca9769f8d96292a8928e6Jason Sams case RS_SCRIPT_INTRINSIC_ID_LUT: 473c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams i = rsdIntrinsic_LUT(this, s, e); 474709a0978ae141198018ca9769f8d96292a8928e6Jason Sams break; 475709a0978ae141198018ca9769f8d96292a8928e6Jason Sams case RS_SCRIPT_INTRINSIC_ID_CONVOLVE_5x5: 476c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams i = rsdIntrinsic_Convolve5x5(this, s, e); 477709a0978ae141198018ca9769f8d96292a8928e6Jason Sams break; 478709a0978ae141198018ca9769f8d96292a8928e6Jason Sams case RS_SCRIPT_INTRINSIC_ID_BLUR: 479c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams i = rsdIntrinsic_Blur(this, s, e); 480709a0978ae141198018ca9769f8d96292a8928e6Jason Sams break; 481709a0978ae141198018ca9769f8d96292a8928e6Jason Sams case RS_SCRIPT_INTRINSIC_ID_YUV_TO_RGB: 482c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams i = rsdIntrinsic_YuvToRGB(this, s, e); 483709a0978ae141198018ca9769f8d96292a8928e6Jason Sams break; 484709a0978ae141198018ca9769f8d96292a8928e6Jason Sams case RS_SCRIPT_INTRINSIC_ID_BLEND: 485c905efd76fdcc1b8846b229bf7d991d185a7b4b7Jason Sams i = rsdIntrinsic_Blend(this, s, e); 486709a0978ae141198018ca9769f8d96292a8928e6Jason Sams break; 487709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 488709a0978ae141198018ca9769f8d96292a8928e6Jason Sams default: 489709a0978ae141198018ca9769f8d96292a8928e6Jason Sams rsAssert(0); 490709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 491709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 492709a0978ae141198018ca9769f8d96292a8928e6Jason Sams return i; 493709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 494709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 495709a0978ae141198018ca9769f8d96292a8928e6Jason SamsRsdCpuReference::CpuScriptGroup * RsdCpuReferenceImpl::createScriptGroup(const ScriptGroup *sg) { 496709a0978ae141198018ca9769f8d96292a8928e6Jason Sams CpuScriptGroupImpl *sgi = new CpuScriptGroupImpl(this, sg); 497709a0978ae141198018ca9769f8d96292a8928e6Jason Sams if (!sgi->init()) { 498709a0978ae141198018ca9769f8d96292a8928e6Jason Sams delete sgi; 499709a0978ae141198018ca9769f8d96292a8928e6Jason Sams return NULL; 500709a0978ae141198018ca9769f8d96292a8928e6Jason Sams } 501709a0978ae141198018ca9769f8d96292a8928e6Jason Sams return sgi; 502709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 503709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 504709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 505