rsScriptC.cpp revision ace3e01f07252ee0fa47fcf1ac48864d8220b90e
1326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams/* 2326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams * Copyright (C) 2009 The Android Open Source Project 3326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams * 4326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams * Licensed under the Apache License, Version 2.0 (the "License"); 5326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams * you may not use this file except in compliance with the License. 6326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams * You may obtain a copy of the License at 7326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams * 8326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams * http://www.apache.org/licenses/LICENSE-2.0 9326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams * 10326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams * Unless required by applicable law or agreed to in writing, software 11326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams * distributed under the License is distributed on an "AS IS" BASIS, 12326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams * See the License for the specific language governing permissions and 14326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams * limitations under the License. 15326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams */ 16326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 17326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams#include "rsContext.h" 18326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams#include "rsScriptC.h" 19326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams#include "rsMatrix.h" 20be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams#include "../../../external/llvm/libbcc/include/bcc/bcc.h" 219c4e4ca9ff75a7fe18544c83fcf782e46c9b6ac2Joe Onorato#include "utils/Timers.h" 221ef8b80bc49f415b93912fe7335f852953bde4a4Jack Palevich 231aa5a4eb81b8b88aeb5d2b6f4c47356fd0a62923Jason Sams#include <GLES/gl.h> 241aa5a4eb81b8b88aeb5d2b6f4c47356fd0a62923Jason Sams#include <GLES/glext.h> 251aa5a4eb81b8b88aeb5d2b6f4c47356fd0a62923Jason Sams 26326e0ddf89e8df2837752fbfd7a014814b32082cJason Samsusing namespace android; 27326e0ddf89e8df2837752fbfd7a014814b32082cJason Samsusing namespace android::renderscript; 28326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 29e57691037aea219562ac686429b4b98202aab7bcJason Sams#define GET_TLS() Context::ScriptTLSStruct * tls = \ 30e57691037aea219562ac686429b4b98202aab7bcJason Sams (Context::ScriptTLSStruct *)pthread_getspecific(Context::gThreadTLSKey); \ 31e57691037aea219562ac686429b4b98202aab7bcJason Sams Context * rsc = tls->mContext; \ 32e57691037aea219562ac686429b4b98202aab7bcJason Sams ScriptC * sc = (ScriptC *) tls->mScript 33e57691037aea219562ac686429b4b98202aab7bcJason Sams 34326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 35e514b45de8561fbc6ef6770845102ca10b0a69d7Jason SamsScriptC::ScriptC(Context *rsc) : Script(rsc) 36326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams{ 37f2649a961db2995e9e24a6c98f8a419f1496c1b7Jason Sams mAllocFile = __FILE__; 38f2649a961db2995e9e24a6c98f8a419f1496c1b7Jason Sams mAllocLine = __LINE__; 39be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams mBccScript = NULL; 40efb8de1ef851c9c2a042ad06f64e33bb8b366041Jason Sams memset(&mProgram, 0, sizeof(mProgram)); 41326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams} 42326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 43326e0ddf89e8df2837752fbfd7a014814b32082cJason SamsScriptC::~ScriptC() 44326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams{ 45be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams if (mBccScript) { 46be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams bccDeleteScript(mBccScript); 471ef8b80bc49f415b93912fe7335f852953bde4a4Jack Palevich } 48e402ed33486730f1d06f080cdfc48132bf612b3aJason Sams free(mEnviroment.mScriptText); 49e402ed33486730f1d06f080cdfc48132bf612b3aJason Sams mEnviroment.mScriptText = NULL; 50326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams} 51326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 52c61346b91434307c5003029017b54ce9c49112beJason Samsvoid ScriptC::setupScript(Context *rsc) 53ada7f272890d8791bc518c95989ad7d13050834dJason Sams{ 54c61346b91434307c5003029017b54ce9c49112beJason Sams setupGLState(rsc); 55c61346b91434307c5003029017b54ce9c49112beJason Sams mEnviroment.mStartTimeMillis 56c61346b91434307c5003029017b54ce9c49112beJason Sams = nanoseconds_to_milliseconds(systemTime(SYSTEM_TIME_MONOTONIC)); 57c61346b91434307c5003029017b54ce9c49112beJason Sams 58be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams for (uint32_t ct=0; ct < mEnviroment.mFieldCount; ct++) { 59be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams if (!mSlots[ct].get()) 60be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams continue; 61be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams void *ptr = mSlots[ct]->getPtr(); 62be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams void **dest = ((void ***)mEnviroment.mFieldAddress)[ct]; 63be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams //LOGE("setupScript %i %p = %p %p %i", ct, dest, ptr, mSlots[ct]->getType(), mSlots[ct]->getType()->getDimX()); 64be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams 65be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams //const uint32_t *p32 = (const uint32_t *)ptr; 66be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams //for (uint32_t ct2=0; ct2 < mSlots[ct]->getType()->getDimX(); ct2++) { 67be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams //LOGE(" %i = 0x%08x ", ct2, p32[ct2]); 68be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams //} 69be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams 70be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams if (dest) { 71be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams *dest = ptr; 72be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams } else { 73be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams LOGE("ScriptC::setupScript, NULL var binding address."); 74ada7f272890d8791bc518c95989ad7d13050834dJason Sams } 75ada7f272890d8791bc518c95989ad7d13050834dJason Sams } 76ada7f272890d8791bc518c95989ad7d13050834dJason Sams} 77ada7f272890d8791bc518c95989ad7d13050834dJason Sams 78ce92d4baf7a5bce097228fdd4498601764cd4014Jason Samsconst Allocation *ScriptC::ptrToAllocation(const void *ptr) const 79ce92d4baf7a5bce097228fdd4498601764cd4014Jason Sams{ 80ce92d4baf7a5bce097228fdd4498601764cd4014Jason Sams if (!ptr) { 81ce92d4baf7a5bce097228fdd4498601764cd4014Jason Sams return NULL; 82ce92d4baf7a5bce097228fdd4498601764cd4014Jason Sams } 83ce92d4baf7a5bce097228fdd4498601764cd4014Jason Sams for (uint32_t ct=0; ct < mEnviroment.mFieldCount; ct++) { 84ce92d4baf7a5bce097228fdd4498601764cd4014Jason Sams if (!mSlots[ct].get()) 85ce92d4baf7a5bce097228fdd4498601764cd4014Jason Sams continue; 86ce92d4baf7a5bce097228fdd4498601764cd4014Jason Sams if (mSlots[ct]->getPtr() == ptr) { 87ce92d4baf7a5bce097228fdd4498601764cd4014Jason Sams return mSlots[ct].get(); 88ce92d4baf7a5bce097228fdd4498601764cd4014Jason Sams } 89ce92d4baf7a5bce097228fdd4498601764cd4014Jason Sams } 90ce92d4baf7a5bce097228fdd4498601764cd4014Jason Sams LOGE("ScriptC::ptrToAllocation, failed to find %p", ptr); 91ce92d4baf7a5bce097228fdd4498601764cd4014Jason Sams return NULL; 92ce92d4baf7a5bce097228fdd4498601764cd4014Jason Sams} 93ce92d4baf7a5bce097228fdd4498601764cd4014Jason Sams 94c61346b91434307c5003029017b54ce9c49112beJason SamsScript * ScriptC::setTLS(Script *sc) 9522fa371bf64b8aae786acfe5d711af21afb13a93Jason Sams{ 9622fa371bf64b8aae786acfe5d711af21afb13a93Jason Sams Context::ScriptTLSStruct * tls = (Context::ScriptTLSStruct *) 9722fa371bf64b8aae786acfe5d711af21afb13a93Jason Sams pthread_getspecific(Context::gThreadTLSKey); 9822fa371bf64b8aae786acfe5d711af21afb13a93Jason Sams rsAssert(tls); 99c61346b91434307c5003029017b54ce9c49112beJason Sams Script *old = tls->mScript; 100c61346b91434307c5003029017b54ce9c49112beJason Sams tls->mScript = sc; 101c61346b91434307c5003029017b54ce9c49112beJason Sams return old; 10222fa371bf64b8aae786acfe5d711af21afb13a93Jason Sams} 10322fa371bf64b8aae786acfe5d711af21afb13a93Jason Sams 104326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 105c61346b91434307c5003029017b54ce9c49112beJason Samsvoid ScriptC::setupGLState(Context *rsc) 106326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams{ 107a0a1b6fbece2eb8d72d788422ab3e5f58d5a9216Jason Sams if (mEnviroment.mFragmentStore.get()) { 108a0a1b6fbece2eb8d72d788422ab3e5f58d5a9216Jason Sams rsc->setFragmentStore(mEnviroment.mFragmentStore.get()); 109a0a1b6fbece2eb8d72d788422ab3e5f58d5a9216Jason Sams } 110a0a1b6fbece2eb8d72d788422ab3e5f58d5a9216Jason Sams if (mEnviroment.mFragment.get()) { 111a0a1b6fbece2eb8d72d788422ab3e5f58d5a9216Jason Sams rsc->setFragment(mEnviroment.mFragment.get()); 112a0a1b6fbece2eb8d72d788422ab3e5f58d5a9216Jason Sams } 1138ce125be69531dbf3a7e856d5e59d1b8e2789db0Jason Sams if (mEnviroment.mVertex.get()) { 1148ce125be69531dbf3a7e856d5e59d1b8e2789db0Jason Sams rsc->setVertex(mEnviroment.mVertex.get()); 1158ce125be69531dbf3a7e856d5e59d1b8e2789db0Jason Sams } 116b681c8af69ef6938c80dd5cc5fc920db94ee9f43Jason Sams if (mEnviroment.mRaster.get()) { 117b681c8af69ef6938c80dd5cc5fc920db94ee9f43Jason Sams rsc->setRaster(mEnviroment.mRaster.get()); 118b681c8af69ef6938c80dd5cc5fc920db94ee9f43Jason Sams } 119c61346b91434307c5003029017b54ce9c49112beJason Sams} 120a0a1b6fbece2eb8d72d788422ab3e5f58d5a9216Jason Sams 121c61346b91434307c5003029017b54ce9c49112beJason Samsuint32_t ScriptC::run(Context *rsc) 122c61346b91434307c5003029017b54ce9c49112beJason Sams{ 123c61346b91434307c5003029017b54ce9c49112beJason Sams if (mProgram.mRoot == NULL) { 124c61346b91434307c5003029017b54ce9c49112beJason Sams rsc->setError(RS_ERROR_BAD_SCRIPT, "Attempted to run bad script"); 125c61346b91434307c5003029017b54ce9c49112beJason Sams return 0; 1269c4e4ca9ff75a7fe18544c83fcf782e46c9b6ac2Joe Onorato } 127c61346b91434307c5003029017b54ce9c49112beJason Sams 128c61346b91434307c5003029017b54ce9c49112beJason Sams setupScript(rsc); 1291d54f10f3c23e0d7ec57e52ec3b0701a2a5ed24eJason Sams 1302dca84dd6c07992f78ad050177975f16486dd77eJason Sams uint32_t ret = 0; 131c61346b91434307c5003029017b54ce9c49112beJason Sams Script * oldTLS = setTLS(this); 132be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams //LOGE("ScriptC::run %p", mProgram.mRoot); 133be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams ret = mProgram.mRoot(); 134c61346b91434307c5003029017b54ce9c49112beJason Sams setTLS(oldTLS); 135be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams //LOGE("ScriptC::run ret %i", ret); 136e45ac6e91864f2a6617b6a2c0aa87cdb62cf58fdJason Sams return ret; 137326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams} 138326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 139c61346b91434307c5003029017b54ce9c49112beJason Sams 140ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Samsvoid ScriptC::runForEach(Context *rsc, 141ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams const Allocation * ain, 142ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams Allocation * aout, 143ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams const void * usr, 144ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams const RsScriptCall *sc) 145c61346b91434307c5003029017b54ce9c49112beJason Sams{ 146c61346b91434307c5003029017b54ce9c49112beJason Sams uint32_t dimX = ain->getType()->getDimX(); 147ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams uint32_t dimY = ain->getType()->getDimY(); 148ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams uint32_t dimZ = ain->getType()->getDimZ(); 149ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams uint32_t dimA = 0;//ain->getType()->getDimArray(); 150ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams 151ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams uint32_t xStart = 0; 152ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams uint32_t xEnd = 0; 153ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams uint32_t yStart = 0; 154ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams uint32_t yEnd = 0; 155ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams uint32_t zStart = 0; 156ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams uint32_t zEnd = 0; 157ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams uint32_t arrayStart = 0; 158ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams uint32_t arrayEnd = 0; 159ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams 160ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams if (!sc || (sc->xEnd == 0)) { 161ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams xStart = 0; 162ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams xEnd = ain->getType()->getDimX(); 163ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams } else { 164ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams rsAssert(xStart < dimX); 165ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams rsAssert(xEnd <= dimX); 166ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams rsAssert(sc->xStart < sc->xEnd); 167ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams xStart = rsMin(dimX, sc->xStart); 168ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams xEnd = rsMin(dimX, sc->xEnd); 169ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams if (xStart >= xEnd) return; 170ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams } 171c61346b91434307c5003029017b54ce9c49112beJason Sams 172ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams if (!sc || (sc->yEnd == 0)) { 173ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams yStart = 0; 174ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams yEnd = ain->getType()->getDimY(); 175ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams } else { 176ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams rsAssert(yStart < dimY); 177ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams rsAssert(yEnd <= dimY); 178ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams rsAssert(sc->yStart < sc->yEnd); 179ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams yStart = rsMin(dimY, sc->yStart); 180ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams yEnd = rsMin(dimY, sc->yEnd); 181ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams if (yStart >= yEnd) return; 182ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams } 183ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams 184ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams xEnd = rsMax((uint32_t)1, xEnd); 185ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams yEnd = rsMax((uint32_t)1, yEnd); 186ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams zEnd = rsMax((uint32_t)1, zEnd); 187ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams arrayEnd = rsMax((uint32_t)1, arrayEnd); 188ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams 189ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams rsAssert(ain->getType()->getDimZ() == 0); 190c61346b91434307c5003029017b54ce9c49112beJason Sams 191c61346b91434307c5003029017b54ce9c49112beJason Sams setupScript(rsc); 192c61346b91434307c5003029017b54ce9c49112beJason Sams Script * oldTLS = setTLS(this); 193c61346b91434307c5003029017b54ce9c49112beJason Sams 194ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams typedef int (*rs_t)(const void *, void *, const void *, uint32_t, uint32_t, uint32_t, uint32_t); 195ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams 196c61346b91434307c5003029017b54ce9c49112beJason Sams const uint8_t *ptrIn = (const uint8_t *)ain->getPtr(); 197ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams uint32_t eStrideIn = ain->getType()->getElementSizeBytes(); 198c61346b91434307c5003029017b54ce9c49112beJason Sams 199c61346b91434307c5003029017b54ce9c49112beJason Sams uint8_t *ptrOut = NULL; 200ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams uint32_t eStrideOut = 0; 201c61346b91434307c5003029017b54ce9c49112beJason Sams if (aout) { 202c61346b91434307c5003029017b54ce9c49112beJason Sams ptrOut = (uint8_t *)aout->getPtr(); 203ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams eStrideOut = aout->getType()->getElementSizeBytes(); 204c61346b91434307c5003029017b54ce9c49112beJason Sams } 205c61346b91434307c5003029017b54ce9c49112beJason Sams 206ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams for (uint32_t ar = arrayStart; ar < arrayEnd; ar++) { 207ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams for (uint32_t z = zStart; z < zEnd; z++) { 208ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams for (uint32_t y = yStart; y < yEnd; y++) { 209ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams uint32_t offset = dimX * dimY * dimZ * ar + 210ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams dimX * dimY * z + 211ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams dimX * y; 212ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams uint8_t *xPtrOut = ptrOut + (eStrideOut * offset); 213ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams const uint8_t *xPtrIn = ptrIn + (eStrideIn * offset); 214ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams 215ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams for (uint32_t x = xStart; x < xEnd; x++) { 216ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams ((rs_t)mProgram.mRoot) (xPtrIn, xPtrOut, usr, x, y, z, ar); 217ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams xPtrIn += eStrideIn; 218ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams xPtrOut += eStrideOut; 219ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams } 220ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams } 221ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams } 222ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams 223c61346b91434307c5003029017b54ce9c49112beJason Sams } 224c61346b91434307c5003029017b54ce9c49112beJason Sams 225c61346b91434307c5003029017b54ce9c49112beJason Sams setTLS(oldTLS); 226c61346b91434307c5003029017b54ce9c49112beJason Sams} 227c61346b91434307c5003029017b54ce9c49112beJason Sams 22822fa371bf64b8aae786acfe5d711af21afb13a93Jason Samsvoid ScriptC::Invoke(Context *rsc, uint32_t slot, const void *data, uint32_t len) 22922fa371bf64b8aae786acfe5d711af21afb13a93Jason Sams{ 23022fa371bf64b8aae786acfe5d711af21afb13a93Jason Sams //LOGE("rsi_ScriptInvoke %i", slot); 23122fa371bf64b8aae786acfe5d711af21afb13a93Jason Sams if ((slot >= mEnviroment.mInvokeFunctionCount) || 23222fa371bf64b8aae786acfe5d711af21afb13a93Jason Sams (mEnviroment.mInvokeFunctions[slot] == NULL)) { 23322fa371bf64b8aae786acfe5d711af21afb13a93Jason Sams rsc->setError(RS_ERROR_BAD_SCRIPT, "Calling invoke on bad script"); 23422fa371bf64b8aae786acfe5d711af21afb13a93Jason Sams return; 23522fa371bf64b8aae786acfe5d711af21afb13a93Jason Sams } 236c61346b91434307c5003029017b54ce9c49112beJason Sams setupScript(rsc); 237c61346b91434307c5003029017b54ce9c49112beJason Sams Script * oldTLS = setTLS(this); 23822fa371bf64b8aae786acfe5d711af21afb13a93Jason Sams 2392a63bf6c293d89c8e3725cfb7ee2add3dd3f0246Jason Sams ((void (*)(const void *, uint32_t)) 2402a63bf6c293d89c8e3725cfb7ee2add3dd3f0246Jason Sams mEnviroment.mInvokeFunctions[slot])(data, len); 2412a63bf6c293d89c8e3725cfb7ee2add3dd3f0246Jason Sams 242c61346b91434307c5003029017b54ce9c49112beJason Sams setTLS(oldTLS); 24322fa371bf64b8aae786acfe5d711af21afb13a93Jason Sams} 24422fa371bf64b8aae786acfe5d711af21afb13a93Jason Sams 245326e0ddf89e8df2837752fbfd7a014814b32082cJason SamsScriptCState::ScriptCState() 246326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams{ 2478c6bc6930d4996b897de8e5113da4a4efae2310cJason Sams mScript = NULL; 248326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams clear(); 249326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams} 250326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 251326e0ddf89e8df2837752fbfd7a014814b32082cJason SamsScriptCState::~ScriptCState() 252326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams{ 2538c6bc6930d4996b897de8e5113da4a4efae2310cJason Sams delete mScript; 2548c6bc6930d4996b897de8e5113da4a4efae2310cJason Sams mScript = NULL; 255326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams} 256326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 257326e0ddf89e8df2837752fbfd7a014814b32082cJason Samsvoid ScriptCState::clear() 258326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams{ 2598b2c065dfc16c148e2829a19e83d2269b9bcd4ccJason Sams for (uint32_t ct=0; ct < MAX_SCRIPT_BANKS; ct++) { 2608b2c065dfc16c148e2829a19e83d2269b9bcd4ccJason Sams mConstantBufferTypes[ct].clear(); 26190b36a88cf0cc549b296ac15a249ea7786c6de9eJason Sams mSlotWritable[ct] = false; 2628b2c065dfc16c148e2829a19e83d2269b9bcd4ccJason Sams } 263efb8de1ef851c9c2a042ad06f64e33bb8b366041Jason Sams 2648c6bc6930d4996b897de8e5113da4a4efae2310cJason Sams delete mScript; 265e514b45de8561fbc6ef6770845102ca10b0a69d7Jason Sams mScript = new ScriptC(NULL); 2661f52633fefd1862451e6a30209d590680d02beecJason Sams} 2671f52633fefd1862451e6a30209d590680d02beecJason Sams 268be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Samsstatic BCCvoid* symbolLookup(BCCvoid* pContext, const BCCchar* name) 26929df66f82aeef7fa7e2cf00edbf00d43c822b05aJason Sams{ 270aeb094b520d8ea49b74129927578f18f758c873eJason Sams const ScriptCState::SymbolTable_t *sym; 271aeb094b520d8ea49b74129927578f18f758c873eJason Sams sym = ScriptCState::lookupSymbol(name); 272aeb094b520d8ea49b74129927578f18f758c873eJason Sams if (sym) { 273aeb094b520d8ea49b74129927578f18f758c873eJason Sams return sym->mPtr; 274aeb094b520d8ea49b74129927578f18f758c873eJason Sams } 275aeb094b520d8ea49b74129927578f18f758c873eJason Sams sym = ScriptCState::lookupSymbolCL(name); 276aeb094b520d8ea49b74129927578f18f758c873eJason Sams if (sym) { 277aeb094b520d8ea49b74129927578f18f758c873eJason Sams return sym->mPtr; 278aeb094b520d8ea49b74129927578f18f758c873eJason Sams } 279aeb094b520d8ea49b74129927578f18f758c873eJason Sams sym = ScriptCState::lookupSymbolGL(name); 28029df66f82aeef7fa7e2cf00edbf00d43c822b05aJason Sams if (sym) { 28129df66f82aeef7fa7e2cf00edbf00d43c822b05aJason Sams return sym->mPtr; 28229df66f82aeef7fa7e2cf00edbf00d43c822b05aJason Sams } 28329df66f82aeef7fa7e2cf00edbf00d43c822b05aJason Sams LOGE("ScriptC sym lookup failed for %s", name); 28429df66f82aeef7fa7e2cf00edbf00d43c822b05aJason Sams return NULL; 28529df66f82aeef7fa7e2cf00edbf00d43c822b05aJason Sams} 286a4a54e42fc710a62b47cbcb9d64c34a190429d9eJason Sams 2878c6bc6930d4996b897de8e5113da4a4efae2310cJason Samsvoid ScriptCState::runCompiler(Context *rsc, ScriptC *s) 2881f52633fefd1862451e6a30209d590680d02beecJason Sams{ 289ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams LOGV("ScriptCState::runCompiler "); 290be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams 291be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams s->mBccScript = bccCreateScript(); 292be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams bccScriptBitcode(s->mBccScript, s->mEnviroment.mScriptText, s->mEnviroment.mScriptTextLength); 293be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams bccRegisterSymbolCallback(s->mBccScript, symbolLookup, NULL); 294be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams bccCompileScript(s->mBccScript); 295be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams bccGetScriptLabel(s->mBccScript, "root", (BCCvoid**) &s->mProgram.mRoot); 296be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams bccGetScriptLabel(s->mBccScript, "init", (BCCvoid**) &s->mProgram.mInit); 297ace3e01f07252ee0fa47fcf1ac48864d8220b90eJason Sams LOGV("root %p, init %p", s->mProgram.mRoot, s->mProgram.mInit); 298f1685045e3423f2e342c3d74c206a25d72eba89dJason Sams 2998c6bc6930d4996b897de8e5113da4a4efae2310cJason Sams if (s->mProgram.mInit) { 3008c6bc6930d4996b897de8e5113da4a4efae2310cJason Sams s->mProgram.mInit(); 3011d54f10f3c23e0d7ec57e52ec3b0701a2a5ed24eJason Sams } 3021d54f10f3c23e0d7ec57e52ec3b0701a2a5ed24eJason Sams 3038c88090e8cccab103eebe2ff569e116e9f5fb208Jason Sams bccGetExportFuncs(s->mBccScript, (BCCsizei*) &s->mEnviroment.mInvokeFunctionCount, 0, NULL); 3048c88090e8cccab103eebe2ff569e116e9f5fb208Jason Sams if(s->mEnviroment.mInvokeFunctionCount <= 0) 3058c88090e8cccab103eebe2ff569e116e9f5fb208Jason Sams s->mEnviroment.mInvokeFunctions = NULL; 3068c88090e8cccab103eebe2ff569e116e9f5fb208Jason Sams else { 3078c88090e8cccab103eebe2ff569e116e9f5fb208Jason Sams s->mEnviroment.mInvokeFunctions = (Script::InvokeFunc_t*) calloc(s->mEnviroment.mInvokeFunctionCount, sizeof(Script::InvokeFunc_t)); 3088c88090e8cccab103eebe2ff569e116e9f5fb208Jason Sams bccGetExportFuncs(s->mBccScript, NULL, s->mEnviroment.mInvokeFunctionCount, (BCCvoid **) s->mEnviroment.mInvokeFunctions); 3098c88090e8cccab103eebe2ff569e116e9f5fb208Jason Sams } 3108c88090e8cccab103eebe2ff569e116e9f5fb208Jason Sams 311be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams s->mEnviroment.mFieldAddress = (void **)calloc(100, sizeof(void *)); 312be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams bccGetExportVars(s->mBccScript, (BCCsizei *)&s->mEnviroment.mFieldCount, 313be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams 100, s->mEnviroment.mFieldAddress); 3141d54f10f3c23e0d7ec57e52ec3b0701a2a5ed24eJason Sams 3158c6bc6930d4996b897de8e5113da4a4efae2310cJason Sams s->mEnviroment.mFragment.set(rsc->getDefaultProgramFragment()); 3168c6bc6930d4996b897de8e5113da4a4efae2310cJason Sams s->mEnviroment.mVertex.set(rsc->getDefaultProgramVertex()); 317ccc010bb7c0f89e162bf60033968a20be90a903aJason Sams s->mEnviroment.mFragmentStore.set(rsc->getDefaultProgramStore()); 318b681c8af69ef6938c80dd5cc5fc920db94ee9f43Jason Sams s->mEnviroment.mRaster.set(rsc->getDefaultProgramRaster()); 319a4a54e42fc710a62b47cbcb9d64c34a190429d9eJason Sams 320be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams if (s->mProgram.mRoot) { 3211030893d9b99b72468034da13df025bda479bb97Jason Sams const static int pragmaMax = 16; 322be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams BCCsizei pragmaCount; 323be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams BCCchar * str[pragmaMax]; 324be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams bccGetPragmas(s->mBccScript, &pragmaCount, pragmaMax, &str[0]); 3251030893d9b99b72468034da13df025bda479bb97Jason Sams 3261030893d9b99b72468034da13df025bda479bb97Jason Sams for (int ct=0; ct < pragmaCount; ct+=2) { 327aeb094b520d8ea49b74129927578f18f758c873eJason Sams //LOGE("pragme %s %s", str[ct], str[ct+1]); 3281030893d9b99b72468034da13df025bda479bb97Jason Sams if (!strcmp(str[ct], "version")) { 3291030893d9b99b72468034da13df025bda479bb97Jason Sams continue; 3301030893d9b99b72468034da13df025bda479bb97Jason Sams } 3311030893d9b99b72468034da13df025bda479bb97Jason Sams 3321030893d9b99b72468034da13df025bda479bb97Jason Sams if (!strcmp(str[ct], "stateVertex")) { 3338ce125be69531dbf3a7e856d5e59d1b8e2789db0Jason Sams if (!strcmp(str[ct+1], "default")) { 3348ce125be69531dbf3a7e856d5e59d1b8e2789db0Jason Sams continue; 3358ce125be69531dbf3a7e856d5e59d1b8e2789db0Jason Sams } 3368ce125be69531dbf3a7e856d5e59d1b8e2789db0Jason Sams if (!strcmp(str[ct+1], "parent")) { 3378c6bc6930d4996b897de8e5113da4a4efae2310cJason Sams s->mEnviroment.mVertex.clear(); 3388ce125be69531dbf3a7e856d5e59d1b8e2789db0Jason Sams continue; 3398ce125be69531dbf3a7e856d5e59d1b8e2789db0Jason Sams } 3401030893d9b99b72468034da13df025bda479bb97Jason Sams LOGE("Unreconized value %s passed to stateVertex", str[ct+1]); 3411030893d9b99b72468034da13df025bda479bb97Jason Sams } 3421030893d9b99b72468034da13df025bda479bb97Jason Sams 3431030893d9b99b72468034da13df025bda479bb97Jason Sams if (!strcmp(str[ct], "stateRaster")) { 344b681c8af69ef6938c80dd5cc5fc920db94ee9f43Jason Sams if (!strcmp(str[ct+1], "default")) { 345b681c8af69ef6938c80dd5cc5fc920db94ee9f43Jason Sams continue; 346b681c8af69ef6938c80dd5cc5fc920db94ee9f43Jason Sams } 347b681c8af69ef6938c80dd5cc5fc920db94ee9f43Jason Sams if (!strcmp(str[ct+1], "parent")) { 348b681c8af69ef6938c80dd5cc5fc920db94ee9f43Jason Sams s->mEnviroment.mRaster.clear(); 349b681c8af69ef6938c80dd5cc5fc920db94ee9f43Jason Sams continue; 350b681c8af69ef6938c80dd5cc5fc920db94ee9f43Jason Sams } 3511030893d9b99b72468034da13df025bda479bb97Jason Sams LOGE("Unreconized value %s passed to stateRaster", str[ct+1]); 3521030893d9b99b72468034da13df025bda479bb97Jason Sams } 3531030893d9b99b72468034da13df025bda479bb97Jason Sams 3541030893d9b99b72468034da13df025bda479bb97Jason Sams if (!strcmp(str[ct], "stateFragment")) { 3558ce125be69531dbf3a7e856d5e59d1b8e2789db0Jason Sams if (!strcmp(str[ct+1], "default")) { 3568ce125be69531dbf3a7e856d5e59d1b8e2789db0Jason Sams continue; 3578ce125be69531dbf3a7e856d5e59d1b8e2789db0Jason Sams } 3588ce125be69531dbf3a7e856d5e59d1b8e2789db0Jason Sams if (!strcmp(str[ct+1], "parent")) { 3598c6bc6930d4996b897de8e5113da4a4efae2310cJason Sams s->mEnviroment.mFragment.clear(); 3608ce125be69531dbf3a7e856d5e59d1b8e2789db0Jason Sams continue; 3618ce125be69531dbf3a7e856d5e59d1b8e2789db0Jason Sams } 3621030893d9b99b72468034da13df025bda479bb97Jason Sams LOGE("Unreconized value %s passed to stateFragment", str[ct+1]); 3631030893d9b99b72468034da13df025bda479bb97Jason Sams } 3641030893d9b99b72468034da13df025bda479bb97Jason Sams 365b681c8af69ef6938c80dd5cc5fc920db94ee9f43Jason Sams if (!strcmp(str[ct], "stateStore")) { 3668ce125be69531dbf3a7e856d5e59d1b8e2789db0Jason Sams if (!strcmp(str[ct+1], "default")) { 3678ce125be69531dbf3a7e856d5e59d1b8e2789db0Jason Sams continue; 3688ce125be69531dbf3a7e856d5e59d1b8e2789db0Jason Sams } 3698ce125be69531dbf3a7e856d5e59d1b8e2789db0Jason Sams if (!strcmp(str[ct+1], "parent")) { 3708c6bc6930d4996b897de8e5113da4a4efae2310cJason Sams s->mEnviroment.mFragmentStore.clear(); 3718ce125be69531dbf3a7e856d5e59d1b8e2789db0Jason Sams continue; 3728ce125be69531dbf3a7e856d5e59d1b8e2789db0Jason Sams } 373b681c8af69ef6938c80dd5cc5fc920db94ee9f43Jason Sams LOGE("Unreconized value %s passed to stateStore", str[ct+1]); 3741030893d9b99b72468034da13df025bda479bb97Jason Sams } 3751030893d9b99b72468034da13df025bda479bb97Jason Sams 3761030893d9b99b72468034da13df025bda479bb97Jason Sams } 3771030893d9b99b72468034da13df025bda479bb97Jason Sams 378d34b725ac1752f5a234c7643a0b65245591ea1ccJason Sams 3791030893d9b99b72468034da13df025bda479bb97Jason Sams } else { 3801030893d9b99b72468034da13df025bda479bb97Jason Sams // Deal with an error. 3811030893d9b99b72468034da13df025bda479bb97Jason Sams } 38257b79ceb1126e3797fa42367b97dd7bcfcda1ed9Joe Onorato} 3831030893d9b99b72468034da13df025bda479bb97Jason Sams 3848b2c065dfc16c148e2829a19e83d2269b9bcd4ccJason Sams 38557b79ceb1126e3797fa42367b97dd7bcfcda1ed9Joe Onorato 386326e0ddf89e8df2837752fbfd7a014814b32082cJason Samsnamespace android { 387326e0ddf89e8df2837752fbfd7a014814b32082cJason Samsnamespace renderscript { 388326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 389326e0ddf89e8df2837752fbfd7a014814b32082cJason Samsvoid rsi_ScriptCBegin(Context * rsc) 390326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams{ 391326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams ScriptCState *ss = &rsc->mScriptC; 392326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams ss->clear(); 393326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams} 394326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 395efb8de1ef851c9c2a042ad06f64e33bb8b366041Jason Samsvoid rsi_ScriptCSetScript(Context * rsc, void *vp) 396326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams{ 3978c6bc6930d4996b897de8e5113da4a4efae2310cJason Sams rsAssert(0); 3988c6bc6930d4996b897de8e5113da4a4efae2310cJason Sams //ScriptCState *ss = &rsc->mScriptC; 3998c6bc6930d4996b897de8e5113da4a4efae2310cJason Sams //ss->mProgram.mScript = reinterpret_cast<ScriptC::RunScript_t>(vp); 400326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams} 401326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 4021f52633fefd1862451e6a30209d590680d02beecJason Samsvoid rsi_ScriptCSetText(Context *rsc, const char *text, uint32_t len) 4031f52633fefd1862451e6a30209d590680d02beecJason Sams{ 4041f52633fefd1862451e6a30209d590680d02beecJason Sams ScriptCState *ss = &rsc->mScriptC; 405e402ed33486730f1d06f080cdfc48132bf612b3aJason Sams 406e402ed33486730f1d06f080cdfc48132bf612b3aJason Sams char *t = (char *)malloc(len + 1); 407e402ed33486730f1d06f080cdfc48132bf612b3aJason Sams memcpy(t, text, len); 408e402ed33486730f1d06f080cdfc48132bf612b3aJason Sams t[len] = 0; 409e402ed33486730f1d06f080cdfc48132bf612b3aJason Sams ss->mScript->mEnviroment.mScriptText = t; 4108c6bc6930d4996b897de8e5113da4a4efae2310cJason Sams ss->mScript->mEnviroment.mScriptTextLength = len; 4111f52633fefd1862451e6a30209d590680d02beecJason Sams} 4121f52633fefd1862451e6a30209d590680d02beecJason Sams 4131f52633fefd1862451e6a30209d590680d02beecJason Sams 414326e0ddf89e8df2837752fbfd7a014814b32082cJason SamsRsScript rsi_ScriptCCreate(Context * rsc) 415326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams{ 416326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams ScriptCState *ss = &rsc->mScriptC; 417326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 4188c6bc6930d4996b897de8e5113da4a4efae2310cJason Sams ScriptC *s = ss->mScript; 4198c6bc6930d4996b897de8e5113da4a4efae2310cJason Sams ss->mScript = NULL; 4201f52633fefd1862451e6a30209d590680d02beecJason Sams 4218c6bc6930d4996b897de8e5113da4a4efae2310cJason Sams ss->runCompiler(rsc, s); 4229397e30ce5fe3f6af9212a93b490836b04fdfffaJason Sams s->incUserRef(); 423e514b45de8561fbc6ef6770845102ca10b0a69d7Jason Sams s->setContext(rsc); 424fa51719e6c1bd17b3f347477b4827face9ab647eJason Sams for (int ct=0; ct < MAX_SCRIPT_BANKS; ct++) { 425fa51719e6c1bd17b3f347477b4827face9ab647eJason Sams s->mTypes[ct].set(ss->mConstantBufferTypes[ct].get()); 42690b36a88cf0cc549b296ac15a249ea7786c6de9eJason Sams s->mSlotWritable[ct] = ss->mSlotWritable[ct]; 427fa51719e6c1bd17b3f347477b4827face9ab647eJason Sams } 4281030893d9b99b72468034da13df025bda479bb97Jason Sams 429fa51719e6c1bd17b3f347477b4827face9ab647eJason Sams ss->clear(); 430326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams return s; 431326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams} 432326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 433326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams} 434326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams} 435326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 436326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 437