rsScript.cpp revision 2980f07d3dbbca727e8efe24ace7e7928a935648
1326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams/* 24419977d78018a9933c7f455fe001f644f2d638bStephen Hines * Copyright (C) 2009-2012 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; 2977020c5248ca0921d7d24fb60dec5105ba39026cJason Sams mInitialized = false; 30326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams} 31326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 32afb743aca56c18beb7ab924e75cb6e070ef3e55aAlex SakhartchoukScript::~Script() { 33afb743aca56c18beb7ab924e75cb6e070ef3e55aAlex Sakhartchouk if (mSlots) { 34700ba38f022208686523ab4280c4fc9f102aa273Alex Sakhartchouk delete [] mSlots; 35700ba38f022208686523ab4280c4fc9f102aa273Alex Sakhartchouk mSlots = NULL; 36700ba38f022208686523ab4280c4fc9f102aa273Alex Sakhartchouk } 37afb743aca56c18beb7ab924e75cb6e070ef3e55aAlex Sakhartchouk if (mTypes) { 38700ba38f022208686523ab4280c4fc9f102aa273Alex Sakhartchouk delete [] mTypes; 39700ba38f022208686523ab4280c4fc9f102aa273Alex Sakhartchouk mTypes = NULL; 40700ba38f022208686523ab4280c4fc9f102aa273Alex Sakhartchouk } 41700ba38f022208686523ab4280c4fc9f102aa273Alex Sakhartchouk} 42700ba38f022208686523ab4280c4fc9f102aa273Alex Sakhartchouk 43700ba38f022208686523ab4280c4fc9f102aa273Alex Sakhartchoukvoid Script::setSlot(uint32_t slot, Allocation *a) { 44af12ac6a08651464f8d823add667c706f993b587Steve Block //ALOGE("setSlot %i %p", slot, a); 45bad807405b2b9764372af1ad24bcfd4fb1f33d8eJason Sams if (slot >= mHal.info.exportedVariableCount) { 46af12ac6a08651464f8d823add667c706f993b587Steve Block ALOGE("Script::setSlot unable to set allocation, invalid slot index"); 47700ba38f022208686523ab4280c4fc9f102aa273Alex Sakhartchouk return; 48700ba38f022208686523ab4280c4fc9f102aa273Alex Sakhartchouk } 49700ba38f022208686523ab4280c4fc9f102aa273Alex Sakhartchouk 50700ba38f022208686523ab4280c4fc9f102aa273Alex Sakhartchouk mSlots[slot].set(a); 51bad807405b2b9764372af1ad24bcfd4fb1f33d8eJason Sams if (a != NULL) { 52bad807405b2b9764372af1ad24bcfd4fb1f33d8eJason Sams mRSC->mHal.funcs.script.setGlobalBind(mRSC, this, slot, a->getPtr()); 53bad807405b2b9764372af1ad24bcfd4fb1f33d8eJason Sams } else { 54bad807405b2b9764372af1ad24bcfd4fb1f33d8eJason Sams mRSC->mHal.funcs.script.setGlobalBind(mRSC, this, slot, NULL); 55bad807405b2b9764372af1ad24bcfd4fb1f33d8eJason Sams } 56326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams} 57326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 58b81a0eb8180791e4eaab1253b59fa8bd562b046bAlex Sakhartchoukvoid Script::setVar(uint32_t slot, const void *val, size_t len) { 59af12ac6a08651464f8d823add667c706f993b587Steve Block //ALOGE("setVar %i %p %i", slot, val, len); 60bad807405b2b9764372af1ad24bcfd4fb1f33d8eJason Sams if (slot >= mHal.info.exportedVariableCount) { 61af12ac6a08651464f8d823add667c706f993b587Steve Block ALOGE("Script::setVar unable to set allocation, invalid slot index"); 62bad807405b2b9764372af1ad24bcfd4fb1f33d8eJason Sams return; 63be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams } 64bad807405b2b9764372af1ad24bcfd4fb1f33d8eJason Sams mRSC->mHal.funcs.script.setGlobalVar(mRSC, this, slot, (void *)val, len); 65be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams} 66be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams 672980f07d3dbbca727e8efe24ace7e7928a935648Stephen Hinesvoid Script::setVar(uint32_t slot, const void *val, size_t len, Element *e, 682980f07d3dbbca727e8efe24ace7e7928a935648Stephen Hines const size_t *dims, size_t dimLen) { 692980f07d3dbbca727e8efe24ace7e7928a935648Stephen Hines if (slot >= mHal.info.exportedVariableCount) { 702980f07d3dbbca727e8efe24ace7e7928a935648Stephen Hines ALOGE("Script::setVar unable to set allocation, invalid slot index"); 712980f07d3dbbca727e8efe24ace7e7928a935648Stephen Hines return; 722980f07d3dbbca727e8efe24ace7e7928a935648Stephen Hines } 732980f07d3dbbca727e8efe24ace7e7928a935648Stephen Hines mRSC->mHal.funcs.script.setGlobalVarWithElemDims(mRSC, this, slot, 742980f07d3dbbca727e8efe24ace7e7928a935648Stephen Hines (void *)val, len, e, dims, dimLen); 752980f07d3dbbca727e8efe24ace7e7928a935648Stephen Hines} 762980f07d3dbbca727e8efe24ace7e7928a935648Stephen Hines 77a5eb6e17c08426deeedf989d6be280d89662ee82Jason Samsvoid Script::setVarObj(uint32_t slot, ObjectBase *val) { 78af12ac6a08651464f8d823add667c706f993b587Steve Block //ALOGE("setVarObj %i %p", slot, val); 79bad807405b2b9764372af1ad24bcfd4fb1f33d8eJason Sams if (slot >= mHal.info.exportedVariableCount) { 80af12ac6a08651464f8d823add667c706f993b587Steve Block ALOGE("Script::setVarObj unable to set allocation, invalid slot index"); 81bad807405b2b9764372af1ad24bcfd4fb1f33d8eJason Sams return; 82a5eb6e17c08426deeedf989d6be280d89662ee82Jason Sams } 83af12ac6a08651464f8d823add667c706f993b587Steve Block //ALOGE("setvarobj %i %p", slot, val); 84bad807405b2b9764372af1ad24bcfd4fb1f33d8eJason Sams mRSC->mHal.funcs.script.setGlobalObj(mRSC, this, slot, val); 85a5eb6e17c08426deeedf989d6be280d89662ee82Jason Sams} 86a5eb6e17c08426deeedf989d6be280d89662ee82Jason Sams 874ee16ffbd9d1d72e1757c9b26715597fdc044117Stephen Hinesbool Script::freeChildren() { 884ee16ffbd9d1d72e1757c9b26715597fdc044117Stephen Hines incSysRef(); 894ee16ffbd9d1d72e1757c9b26715597fdc044117Stephen Hines mRSC->mHal.funcs.script.invokeFreeChildren(mRSC, this); 904ee16ffbd9d1d72e1757c9b26715597fdc044117Stephen Hines return decSysRef(); 914ee16ffbd9d1d72e1757c9b26715597fdc044117Stephen Hines} 924ee16ffbd9d1d72e1757c9b26715597fdc044117Stephen Hines 93326e0ddf89e8df2837752fbfd7a014814b32082cJason Samsnamespace android { 94326e0ddf89e8df2837752fbfd7a014814b32082cJason Samsnamespace renderscript { 95326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 96afb743aca56c18beb7ab924e75cb6e070ef3e55aAlex Sakhartchoukvoid rsi_ScriptBindAllocation(Context * rsc, RsScript vs, RsAllocation va, uint32_t slot) { 97326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams Script *s = static_cast<Script *>(vs); 98be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams Allocation *a = static_cast<Allocation *>(va); 99700ba38f022208686523ab4280c4fc9f102aa273Alex Sakhartchouk s->setSlot(slot, a); 100af12ac6a08651464f8d823add667c706f993b587Steve Block //ALOGE("rsi_ScriptBindAllocation %i %p %p", slot, a, a->getPtr()); 101326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams} 102326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 103b81a0eb8180791e4eaab1253b59fa8bd562b046bAlex Sakhartchoukvoid rsi_ScriptSetTimeZone(Context * rsc, RsScript vs, const char * timeZone, size_t length) { 104d2432b9691869b5b40a5b49c682c40d917ea9dcbStephen Hines // We unfortunately need to make a new copy of the string, since it is 105d2432b9691869b5b40a5b49c682c40d917ea9dcbStephen Hines // not NULL-terminated. We then use setenv(), which properly handles 106d2432b9691869b5b40a5b49c682c40d917ea9dcbStephen Hines // freeing/duplicating the actual string for the environment. 107d2432b9691869b5b40a5b49c682c40d917ea9dcbStephen Hines char *tz = (char *) malloc(length + 1); 108d2432b9691869b5b40a5b49c682c40d917ea9dcbStephen Hines if (!tz) { 109af12ac6a08651464f8d823add667c706f993b587Steve Block ALOGE("Couldn't allocate memory for timezone buffer"); 110d2432b9691869b5b40a5b49c682c40d917ea9dcbStephen Hines return; 111d2432b9691869b5b40a5b49c682c40d917ea9dcbStephen Hines } 112d2432b9691869b5b40a5b49c682c40d917ea9dcbStephen Hines strncpy(tz, timeZone, length); 113d2432b9691869b5b40a5b49c682c40d917ea9dcbStephen Hines tz[length] = '\0'; 114d2432b9691869b5b40a5b49c682c40d917ea9dcbStephen Hines if (setenv("TZ", tz, 1) == 0) { 115d2432b9691869b5b40a5b49c682c40d917ea9dcbStephen Hines tzset(); 116d2432b9691869b5b40a5b49c682c40d917ea9dcbStephen Hines } else { 117af12ac6a08651464f8d823add667c706f993b587Steve Block ALOGE("Error setting timezone"); 118d2432b9691869b5b40a5b49c682c40d917ea9dcbStephen Hines } 119d2432b9691869b5b40a5b49c682c40d917ea9dcbStephen Hines free(tz); 120d34b725ac1752f5a234c7643a0b65245591ea1ccJason Sams} 121d34b725ac1752f5a234c7643a0b65245591ea1ccJason Sams 1225fb1aeb6374f814d72117803682bc141cc6b0895Jason Samsvoid rsi_ScriptForEach(Context *rsc, RsScript vs, uint32_t slot, 1235fb1aeb6374f814d72117803682bc141cc6b0895Jason Sams RsAllocation vain, RsAllocation vaout, 124b81a0eb8180791e4eaab1253b59fa8bd562b046bAlex Sakhartchouk const void *params, size_t paramLen) { 1255fb1aeb6374f814d72117803682bc141cc6b0895Jason Sams Script *s = static_cast<Script *>(vs); 1264419977d78018a9933c7f455fe001f644f2d638bStephen Hines s->runForEach(rsc, slot, 1275fb1aeb6374f814d72117803682bc141cc6b0895Jason Sams static_cast<const Allocation *>(vain), static_cast<Allocation *>(vaout), 1285fb1aeb6374f814d72117803682bc141cc6b0895Jason Sams params, paramLen); 1295fb1aeb6374f814d72117803682bc141cc6b0895Jason Sams 1305fb1aeb6374f814d72117803682bc141cc6b0895Jason Sams} 1315fb1aeb6374f814d72117803682bc141cc6b0895Jason Sams 132afb743aca56c18beb7ab924e75cb6e070ef3e55aAlex Sakhartchoukvoid rsi_ScriptInvoke(Context *rsc, RsScript vs, uint32_t slot) { 1338c6bc6930d4996b897de8e5113da4a4efae2310cJason Sams Script *s = static_cast<Script *>(vs); 13422fa371bf64b8aae786acfe5d711af21afb13a93Jason Sams s->Invoke(rsc, slot, NULL, 0); 1358c6bc6930d4996b897de8e5113da4a4efae2310cJason Sams} 1368c6bc6930d4996b897de8e5113da4a4efae2310cJason Sams 1378c6bc6930d4996b897de8e5113da4a4efae2310cJason Sams 138afb743aca56c18beb7ab924e75cb6e070ef3e55aAlex Sakhartchoukvoid rsi_ScriptInvokeData(Context *rsc, RsScript vs, uint32_t slot, void *data) { 139be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams Script *s = static_cast<Script *>(vs); 14022fa371bf64b8aae786acfe5d711af21afb13a93Jason Sams s->Invoke(rsc, slot, NULL, 0); 141be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams} 142be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams 143b81a0eb8180791e4eaab1253b59fa8bd562b046bAlex Sakhartchoukvoid rsi_ScriptInvokeV(Context *rsc, RsScript vs, uint32_t slot, const void *data, size_t len) { 144be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams Script *s = static_cast<Script *>(vs); 14522fa371bf64b8aae786acfe5d711af21afb13a93Jason Sams s->Invoke(rsc, slot, data, len); 146be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams} 147be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams 148afb743aca56c18beb7ab924e75cb6e070ef3e55aAlex Sakhartchoukvoid rsi_ScriptSetVarI(Context *rsc, RsScript vs, uint32_t slot, int value) { 149be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams Script *s = static_cast<Script *>(vs); 150be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams s->setVar(slot, &value, sizeof(value)); 151be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams} 152be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams 153a5eb6e17c08426deeedf989d6be280d89662ee82Jason Samsvoid rsi_ScriptSetVarObj(Context *rsc, RsScript vs, uint32_t slot, RsObjectBase value) { 154a5eb6e17c08426deeedf989d6be280d89662ee82Jason Sams Script *s = static_cast<Script *>(vs); 155a5eb6e17c08426deeedf989d6be280d89662ee82Jason Sams ObjectBase *o = static_cast<ObjectBase *>(value); 156a5eb6e17c08426deeedf989d6be280d89662ee82Jason Sams s->setVarObj(slot, o); 157a5eb6e17c08426deeedf989d6be280d89662ee82Jason Sams} 158a5eb6e17c08426deeedf989d6be280d89662ee82Jason Sams 159afb743aca56c18beb7ab924e75cb6e070ef3e55aAlex Sakhartchoukvoid rsi_ScriptSetVarJ(Context *rsc, RsScript vs, uint32_t slot, long long value) { 1600977c94a4b83dd8ef6d3808160710ebbaa1b1768Stephen Hines Script *s = static_cast<Script *>(vs); 1610977c94a4b83dd8ef6d3808160710ebbaa1b1768Stephen Hines s->setVar(slot, &value, sizeof(value)); 1620977c94a4b83dd8ef6d3808160710ebbaa1b1768Stephen Hines} 1630977c94a4b83dd8ef6d3808160710ebbaa1b1768Stephen Hines 164afb743aca56c18beb7ab924e75cb6e070ef3e55aAlex Sakhartchoukvoid rsi_ScriptSetVarF(Context *rsc, RsScript vs, uint32_t slot, float value) { 165be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams Script *s = static_cast<Script *>(vs); 166be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams s->setVar(slot, &value, sizeof(value)); 167be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams} 168be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams 169afb743aca56c18beb7ab924e75cb6e070ef3e55aAlex Sakhartchoukvoid rsi_ScriptSetVarD(Context *rsc, RsScript vs, uint32_t slot, double value) { 1706d0a0740530d57e2a5759df994352600946ff47dStephen Hines Script *s = static_cast<Script *>(vs); 1716d0a0740530d57e2a5759df994352600946ff47dStephen Hines s->setVar(slot, &value, sizeof(value)); 1726d0a0740530d57e2a5759df994352600946ff47dStephen Hines} 1736d0a0740530d57e2a5759df994352600946ff47dStephen Hines 174b81a0eb8180791e4eaab1253b59fa8bd562b046bAlex Sakhartchoukvoid rsi_ScriptSetVarV(Context *rsc, RsScript vs, uint32_t slot, const void *data, size_t len) { 175be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams Script *s = static_cast<Script *>(vs); 176be36bf3a76481737a7fa606a04144ceef80eb4f2Jason Sams s->setVar(slot, data, len); 177fa51719e6c1bd17b3f347477b4827face9ab647eJason Sams} 178fa51719e6c1bd17b3f347477b4827face9ab647eJason Sams 1792980f07d3dbbca727e8efe24ace7e7928a935648Stephen Hinesvoid rsi_ScriptSetVarVE(Context *rsc, RsScript vs, uint32_t slot, 1802980f07d3dbbca727e8efe24ace7e7928a935648Stephen Hines const void *data, size_t len, RsElement ve, 1812980f07d3dbbca727e8efe24ace7e7928a935648Stephen Hines const size_t *dims, size_t dimLen) { 1822980f07d3dbbca727e8efe24ace7e7928a935648Stephen Hines Script *s = static_cast<Script *>(vs); 1832980f07d3dbbca727e8efe24ace7e7928a935648Stephen Hines Element *e = static_cast<Element *>(ve); 1842980f07d3dbbca727e8efe24ace7e7928a935648Stephen Hines s->setVar(slot, data, len, e, dims, dimLen); 1852980f07d3dbbca727e8efe24ace7e7928a935648Stephen Hines} 1862980f07d3dbbca727e8efe24ace7e7928a935648Stephen Hines 187326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams} 188326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams} 189326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams 190