rsScript.cpp revision d2432b9691869b5b40a5b49c682c40d917ea9dcb
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" 18d2432b9691869b5b40a5b49c682c40d917ea9dcbStephen Hines#include <time.h> 19326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 20326e0ddf89e8df2837752fbfd7a014814b32082cJason Samsusing namespace android; 21326e0ddf89e8df2837752fbfd7a014814b32082cJason Samsusing namespace android::renderscript; 22326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 23afb743aca56c18beb7ab924e75cb6e070ef3e55aAlex SakhartchoukScript::Script(Context *rsc) : ObjectBase(rsc) { 24928b734b2fbba0800567c8b7e686bcf5f45483ccJason Sams memset(&mEnviroment, 0, sizeof(mEnviroment)); 25bad807405b2b9764372af1ad24bcfd4fb1f33d8eJason Sams memset(&mHal, 0, sizeof(mHal)); 26700ba38f022208686523ab4280c4fc9f102aa273Alex Sakhartchouk 27700ba38f022208686523ab4280c4fc9f102aa273Alex Sakhartchouk mSlots = NULL; 28700ba38f022208686523ab4280c4fc9f102aa273Alex Sakhartchouk mTypes = NULL; 29326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams} 30326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 31afb743aca56c18beb7ab924e75cb6e070ef3e55aAlex SakhartchoukScript::~Script() { 32afb743aca56c18beb7ab924e75cb6e070ef3e55aAlex Sakhartchouk if (mSlots) { 33700ba38f022208686523ab4280c4fc9f102aa273Alex Sakhartchouk delete [] mSlots; 34700ba38f022208686523ab4280c4fc9f102aa273Alex Sakhartchouk mSlots = NULL; 35700ba38f022208686523ab4280c4fc9f102aa273Alex Sakhartchouk } 36afb743aca56c18beb7ab924e75cb6e070ef3e55aAlex Sakhartchouk if (mTypes) { 37700ba38f022208686523ab4280c4fc9f102aa273Alex Sakhartchouk delete [] mTypes; 38700ba38f022208686523ab4280c4fc9f102aa273Alex Sakhartchouk mTypes = NULL; 39700ba38f022208686523ab4280c4fc9f102aa273Alex Sakhartchouk } 40700ba38f022208686523ab4280c4fc9f102aa273Alex Sakhartchouk} 41700ba38f022208686523ab4280c4fc9f102aa273Alex Sakhartchouk 42700ba38f022208686523ab4280c4fc9f102aa273Alex Sakhartchoukvoid Script::setSlot(uint32_t slot, Allocation *a) { 43bad807405b2b9764372af1ad24bcfd4fb1f33d8eJason Sams //LOGE("setSlot %i %p", slot, a); 44bad807405b2b9764372af1ad24bcfd4fb1f33d8eJason Sams if (slot >= mHal.info.exportedVariableCount) { 45700ba38f022208686523ab4280c4fc9f102aa273Alex Sakhartchouk LOGE("Script::setSlot unable to set allocation, invalid slot index"); 46700ba38f022208686523ab4280c4fc9f102aa273Alex Sakhartchouk return; 47700ba38f022208686523ab4280c4fc9f102aa273Alex Sakhartchouk } 48700ba38f022208686523ab4280c4fc9f102aa273Alex Sakhartchouk 49700ba38f022208686523ab4280c4fc9f102aa273Alex Sakhartchouk mSlots[slot].set(a); 50bad807405b2b9764372af1ad24bcfd4fb1f33d8eJason Sams if (a != NULL) { 51bad807405b2b9764372af1ad24bcfd4fb1f33d8eJason Sams mRSC->mHal.funcs.script.setGlobalBind(mRSC, this, slot, a->getPtr()); 52bad807405b2b9764372af1ad24bcfd4fb1f33d8eJason Sams } else { 53bad807405b2b9764372af1ad24bcfd4fb1f33d8eJason Sams mRSC->mHal.funcs.script.setGlobalBind(mRSC, this, slot, NULL); 54bad807405b2b9764372af1ad24bcfd4fb1f33d8eJason Sams } 55326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams} 56326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 57b81a0eb8180791e4eaab1253b59fa8bd562b046bAlex Sakhartchoukvoid Script::setVar(uint32_t slot, const void *val, size_t len) { 58bad807405b2b9764372af1ad24bcfd4fb1f33d8eJason Sams //LOGE("setVar %i %p %i", slot, val, len); 59bad807405b2b9764372af1ad24bcfd4fb1f33d8eJason Sams if (slot >= mHal.info.exportedVariableCount) { 60bad807405b2b9764372af1ad24bcfd4fb1f33d8eJason Sams LOGE("Script::setVar unable to set allocation, invalid slot index"); 61bad807405b2b9764372af1ad24bcfd4fb1f33d8eJason Sams return; 62be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams } 63bad807405b2b9764372af1ad24bcfd4fb1f33d8eJason Sams mRSC->mHal.funcs.script.setGlobalVar(mRSC, this, slot, (void *)val, len); 64be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams} 65be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams 66a5eb6e17c08426deeedf989d6be280d89662ee82Jason Samsvoid Script::setVarObj(uint32_t slot, ObjectBase *val) { 67bad807405b2b9764372af1ad24bcfd4fb1f33d8eJason Sams //LOGE("setVarObj %i %p", slot, val); 68bad807405b2b9764372af1ad24bcfd4fb1f33d8eJason Sams if (slot >= mHal.info.exportedVariableCount) { 69bad807405b2b9764372af1ad24bcfd4fb1f33d8eJason Sams LOGE("Script::setVarObj unable to set allocation, invalid slot index"); 70bad807405b2b9764372af1ad24bcfd4fb1f33d8eJason Sams return; 71a5eb6e17c08426deeedf989d6be280d89662ee82Jason Sams } 72bad807405b2b9764372af1ad24bcfd4fb1f33d8eJason Sams //LOGE("setvarobj %i %p", slot, val); 73bad807405b2b9764372af1ad24bcfd4fb1f33d8eJason Sams mRSC->mHal.funcs.script.setGlobalObj(mRSC, this, slot, val); 74a5eb6e17c08426deeedf989d6be280d89662ee82Jason Sams} 75a5eb6e17c08426deeedf989d6be280d89662ee82Jason Sams 764ee16ffbd9d1d72e1757c9b26715597fdc044117Stephen Hinesbool Script::freeChildren() { 774ee16ffbd9d1d72e1757c9b26715597fdc044117Stephen Hines incSysRef(); 784ee16ffbd9d1d72e1757c9b26715597fdc044117Stephen Hines mRSC->mHal.funcs.script.invokeFreeChildren(mRSC, this); 794ee16ffbd9d1d72e1757c9b26715597fdc044117Stephen Hines return decSysRef(); 804ee16ffbd9d1d72e1757c9b26715597fdc044117Stephen Hines} 814ee16ffbd9d1d72e1757c9b26715597fdc044117Stephen Hines 82326e0ddf89e8df2837752fbfd7a014814b32082cJason Samsnamespace android { 83326e0ddf89e8df2837752fbfd7a014814b32082cJason Samsnamespace renderscript { 84326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 85afb743aca56c18beb7ab924e75cb6e070ef3e55aAlex Sakhartchoukvoid rsi_ScriptBindAllocation(Context * rsc, RsScript vs, RsAllocation va, uint32_t slot) { 86326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams Script *s = static_cast<Script *>(vs); 87be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams Allocation *a = static_cast<Allocation *>(va); 88700ba38f022208686523ab4280c4fc9f102aa273Alex Sakhartchouk s->setSlot(slot, a); 89be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams //LOGE("rsi_ScriptBindAllocation %i %p %p", slot, a, a->getPtr()); 90326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams} 91326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 92b81a0eb8180791e4eaab1253b59fa8bd562b046bAlex Sakhartchoukvoid rsi_ScriptSetTimeZone(Context * rsc, RsScript vs, const char * timeZone, size_t length) { 93d2432b9691869b5b40a5b49c682c40d917ea9dcbStephen Hines // We unfortunately need to make a new copy of the string, since it is 94d2432b9691869b5b40a5b49c682c40d917ea9dcbStephen Hines // not NULL-terminated. We then use setenv(), which properly handles 95d2432b9691869b5b40a5b49c682c40d917ea9dcbStephen Hines // freeing/duplicating the actual string for the environment. 96d2432b9691869b5b40a5b49c682c40d917ea9dcbStephen Hines char *tz = (char *) malloc(length + 1); 97d2432b9691869b5b40a5b49c682c40d917ea9dcbStephen Hines if (!tz) { 98d2432b9691869b5b40a5b49c682c40d917ea9dcbStephen Hines LOGE("Couldn't allocate memory for timezone buffer"); 99d2432b9691869b5b40a5b49c682c40d917ea9dcbStephen Hines return; 100d2432b9691869b5b40a5b49c682c40d917ea9dcbStephen Hines } 101d2432b9691869b5b40a5b49c682c40d917ea9dcbStephen Hines strncpy(tz, timeZone, length); 102d2432b9691869b5b40a5b49c682c40d917ea9dcbStephen Hines tz[length] = '\0'; 103d2432b9691869b5b40a5b49c682c40d917ea9dcbStephen Hines if (setenv("TZ", tz, 1) == 0) { 104d2432b9691869b5b40a5b49c682c40d917ea9dcbStephen Hines tzset(); 105d2432b9691869b5b40a5b49c682c40d917ea9dcbStephen Hines } else { 106d2432b9691869b5b40a5b49c682c40d917ea9dcbStephen Hines LOGE("Error setting timezone"); 107d2432b9691869b5b40a5b49c682c40d917ea9dcbStephen Hines } 108d2432b9691869b5b40a5b49c682c40d917ea9dcbStephen Hines free(tz); 109d34b725ac1752f5a234c7643a0b65245591ea1ccJason Sams} 110d34b725ac1752f5a234c7643a0b65245591ea1ccJason Sams 1115fb1aeb6374f814d72117803682bc141cc6b0895Jason Samsvoid rsi_ScriptForEach(Context *rsc, RsScript vs, uint32_t slot, 1125fb1aeb6374f814d72117803682bc141cc6b0895Jason Sams RsAllocation vain, RsAllocation vaout, 113b81a0eb8180791e4eaab1253b59fa8bd562b046bAlex Sakhartchouk const void *params, size_t paramLen) { 1145fb1aeb6374f814d72117803682bc141cc6b0895Jason Sams Script *s = static_cast<Script *>(vs); 1155fb1aeb6374f814d72117803682bc141cc6b0895Jason Sams s->runForEach(rsc, 1165fb1aeb6374f814d72117803682bc141cc6b0895Jason Sams static_cast<const Allocation *>(vain), static_cast<Allocation *>(vaout), 1175fb1aeb6374f814d72117803682bc141cc6b0895Jason Sams params, paramLen); 1185fb1aeb6374f814d72117803682bc141cc6b0895Jason Sams 1195fb1aeb6374f814d72117803682bc141cc6b0895Jason Sams} 1205fb1aeb6374f814d72117803682bc141cc6b0895Jason Sams 121afb743aca56c18beb7ab924e75cb6e070ef3e55aAlex Sakhartchoukvoid rsi_ScriptInvoke(Context *rsc, RsScript vs, uint32_t slot) { 1228c6bc6930d4996b897de8e5113da4a4efae2310cJason Sams Script *s = static_cast<Script *>(vs); 12322fa371bf64b8aae786acfe5d711af21afb13a93Jason Sams s->Invoke(rsc, slot, NULL, 0); 1248c6bc6930d4996b897de8e5113da4a4efae2310cJason Sams} 1258c6bc6930d4996b897de8e5113da4a4efae2310cJason Sams 1268c6bc6930d4996b897de8e5113da4a4efae2310cJason Sams 127afb743aca56c18beb7ab924e75cb6e070ef3e55aAlex Sakhartchoukvoid rsi_ScriptInvokeData(Context *rsc, RsScript vs, uint32_t slot, void *data) { 128be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams Script *s = static_cast<Script *>(vs); 12922fa371bf64b8aae786acfe5d711af21afb13a93Jason Sams s->Invoke(rsc, slot, NULL, 0); 130be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams} 131be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams 132b81a0eb8180791e4eaab1253b59fa8bd562b046bAlex Sakhartchoukvoid rsi_ScriptInvokeV(Context *rsc, RsScript vs, uint32_t slot, const void *data, size_t len) { 133be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams Script *s = static_cast<Script *>(vs); 13422fa371bf64b8aae786acfe5d711af21afb13a93Jason Sams s->Invoke(rsc, slot, data, len); 135be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams} 136be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams 137afb743aca56c18beb7ab924e75cb6e070ef3e55aAlex Sakhartchoukvoid rsi_ScriptSetVarI(Context *rsc, RsScript vs, uint32_t slot, int value) { 138be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams Script *s = static_cast<Script *>(vs); 139be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams s->setVar(slot, &value, sizeof(value)); 140be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams} 141be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams 142a5eb6e17c08426deeedf989d6be280d89662ee82Jason Samsvoid rsi_ScriptSetVarObj(Context *rsc, RsScript vs, uint32_t slot, RsObjectBase value) { 143a5eb6e17c08426deeedf989d6be280d89662ee82Jason Sams Script *s = static_cast<Script *>(vs); 144a5eb6e17c08426deeedf989d6be280d89662ee82Jason Sams ObjectBase *o = static_cast<ObjectBase *>(value); 145a5eb6e17c08426deeedf989d6be280d89662ee82Jason Sams s->setVarObj(slot, o); 146a5eb6e17c08426deeedf989d6be280d89662ee82Jason Sams} 147a5eb6e17c08426deeedf989d6be280d89662ee82Jason Sams 148afb743aca56c18beb7ab924e75cb6e070ef3e55aAlex Sakhartchoukvoid rsi_ScriptSetVarJ(Context *rsc, RsScript vs, uint32_t slot, long long value) { 1490977c94a4b83dd8ef6d3808160710ebbaa1b1768Stephen Hines Script *s = static_cast<Script *>(vs); 1500977c94a4b83dd8ef6d3808160710ebbaa1b1768Stephen Hines s->setVar(slot, &value, sizeof(value)); 1510977c94a4b83dd8ef6d3808160710ebbaa1b1768Stephen Hines} 1520977c94a4b83dd8ef6d3808160710ebbaa1b1768Stephen Hines 153afb743aca56c18beb7ab924e75cb6e070ef3e55aAlex Sakhartchoukvoid rsi_ScriptSetVarF(Context *rsc, RsScript vs, uint32_t slot, float value) { 154be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams Script *s = static_cast<Script *>(vs); 155be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams s->setVar(slot, &value, sizeof(value)); 156be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams} 157be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams 158afb743aca56c18beb7ab924e75cb6e070ef3e55aAlex Sakhartchoukvoid rsi_ScriptSetVarD(Context *rsc, RsScript vs, uint32_t slot, double value) { 1596d0a0740530d57e2a5759df994352600946ff47dStephen Hines Script *s = static_cast<Script *>(vs); 1606d0a0740530d57e2a5759df994352600946ff47dStephen Hines s->setVar(slot, &value, sizeof(value)); 1616d0a0740530d57e2a5759df994352600946ff47dStephen Hines} 1626d0a0740530d57e2a5759df994352600946ff47dStephen Hines 163b81a0eb8180791e4eaab1253b59fa8bd562b046bAlex Sakhartchoukvoid rsi_ScriptSetVarV(Context *rsc, RsScript vs, uint32_t slot, const void *data, size_t len) { 164be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams Script *s = static_cast<Script *>(vs); 165be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams s->setVar(slot, data, len); 166fa51719e6c1bd17b3f347477b4827face9ab647eJason Sams} 167fa51719e6c1bd17b3f347477b4827face9ab647eJason Sams 168326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams} 169326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams} 170326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 171