1 /* 2 * Copyright (C) 2009-2012 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17 #include "rsContext.h" 18 #include <time.h> 19 20 using namespace android; 21 using namespace android::renderscript; 22 23 Script::Script(Context *rsc) : ObjectBase(rsc) { 24 memset(&mEnviroment, 0, sizeof(mEnviroment)); 25 memset(&mHal, 0, sizeof(mHal)); 26 27 mSlots = NULL; 28 mTypes = NULL; 29 mInitialized = false; 30 mHasObjectSlots = false; 31 } 32 33 Script::~Script() { 34 if (mSlots) { 35 delete [] mSlots; 36 mSlots = NULL; 37 } 38 if (mTypes) { 39 delete [] mTypes; 40 mTypes = NULL; 41 } 42 } 43 44 void Script::setSlot(uint32_t slot, Allocation *a) { 45 //ALOGE("setSlot %i %p", slot, a); 46 if (slot >= mHal.info.exportedVariableCount) { 47 ALOGE("Script::setSlot unable to set allocation, invalid slot index"); 48 return; 49 } 50 51 mSlots[slot].set(a); 52 mHasObjectSlots = true; 53 mRSC->mHal.funcs.script.setGlobalBind(mRSC, this, slot, a); 54 } 55 56 void Script::setVar(uint32_t slot, const void *val, size_t len) { 57 //ALOGE("setVar %i %p %i", slot, val, len); 58 if (slot >= mHal.info.exportedVariableCount) { 59 ALOGE("Script::setVar unable to set allocation, invalid slot index"); 60 return; 61 } 62 mRSC->mHal.funcs.script.setGlobalVar(mRSC, this, slot, (void *)val, len); 63 } 64 65 void Script::getVar(uint32_t slot, const void *val, size_t len) { 66 //ALOGE("getVar %i %p %i", slot, val, len); 67 if (slot >= mHal.info.exportedVariableCount) { 68 ALOGE("Script::getVar unable to set allocation, invalid slot index"); 69 return; 70 } 71 mRSC->mHal.funcs.script.getGlobalVar(mRSC, this, slot, (void *)val, len); 72 } 73 74 void Script::setVar(uint32_t slot, const void *val, size_t len, Element *e, 75 const size_t *dims, size_t dimLen) { 76 if (slot >= mHal.info.exportedVariableCount) { 77 ALOGE("Script::setVar unable to set allocation, invalid slot index"); 78 return; 79 } 80 mRSC->mHal.funcs.script.setGlobalVarWithElemDims(mRSC, this, slot, 81 (void *)val, len, e, dims, dimLen); 82 } 83 84 void Script::setVarObj(uint32_t slot, ObjectBase *val) { 85 //ALOGE("setVarObj %i %p", slot, val); 86 if (slot >= mHal.info.exportedVariableCount) { 87 ALOGE("Script::setVarObj unable to set allocation, invalid slot index"); 88 return; 89 } 90 mHasObjectSlots = true; 91 //ALOGE("setvarobj %i %p", slot, val); 92 mRSC->mHal.funcs.script.setGlobalObj(mRSC, this, slot, val); 93 } 94 95 bool Script::freeChildren() { 96 incSysRef(); 97 mRSC->mHal.funcs.script.invokeFreeChildren(mRSC, this); 98 return decSysRef(); 99 } 100 101 ScriptKernelID::ScriptKernelID(Context *rsc, Script *s, int slot, int sig) 102 : ObjectBase(rsc) { 103 104 mScript = s; 105 mSlot = slot; 106 mHasKernelInput = (sig & 1) != 0; 107 mHasKernelOutput = (sig & 2) != 0; 108 } 109 110 ScriptKernelID::~ScriptKernelID() { 111 112 } 113 114 void ScriptKernelID::serialize(Context *rsc, OStream *stream) const { 115 116 } 117 118 RsA3DClassID ScriptKernelID::getClassId() const { 119 return RS_A3D_CLASS_ID_SCRIPT_KERNEL_ID; 120 } 121 122 ScriptFieldID::ScriptFieldID(Context *rsc, Script *s, int slot) : ObjectBase(rsc) { 123 mScript = s; 124 mSlot = slot; 125 } 126 127 ScriptFieldID::~ScriptFieldID() { 128 129 } 130 131 void ScriptFieldID::serialize(Context *rsc, OStream *stream) const { 132 133 } 134 135 RsA3DClassID ScriptFieldID::getClassId() const { 136 return RS_A3D_CLASS_ID_SCRIPT_FIELD_ID; 137 } 138 139 140 namespace android { 141 namespace renderscript { 142 143 RsScriptKernelID rsi_ScriptKernelIDCreate(Context *rsc, RsScript vs, int slot, int sig) { 144 ScriptKernelID *kid = new ScriptKernelID(rsc, (Script *)vs, slot, sig); 145 kid->incUserRef(); 146 return kid; 147 } 148 149 RsScriptFieldID rsi_ScriptFieldIDCreate(Context *rsc, RsScript vs, int slot) { 150 ScriptFieldID *fid = new ScriptFieldID(rsc, (Script *)vs, slot); 151 fid->incUserRef(); 152 return fid; 153 } 154 155 void rsi_ScriptBindAllocation(Context * rsc, RsScript vs, RsAllocation va, uint32_t slot) { 156 Script *s = static_cast<Script *>(vs); 157 Allocation *a = static_cast<Allocation *>(va); 158 s->setSlot(slot, a); 159 } 160 161 void rsi_ScriptSetTimeZone(Context * rsc, RsScript vs, const char * timeZone, size_t length) { 162 // We unfortunately need to make a new copy of the string, since it is 163 // not NULL-terminated. We then use setenv(), which properly handles 164 // freeing/duplicating the actual string for the environment. 165 char *tz = (char *) malloc(length + 1); 166 if (!tz) { 167 ALOGE("Couldn't allocate memory for timezone buffer"); 168 return; 169 } 170 strncpy(tz, timeZone, length); 171 tz[length] = '\0'; 172 if (setenv("TZ", tz, 1) == 0) { 173 tzset(); 174 } else { 175 ALOGE("Error setting timezone"); 176 } 177 free(tz); 178 } 179 180 void rsi_ScriptForEach(Context *rsc, RsScript vs, uint32_t slot, 181 RsAllocation vain, RsAllocation vaout, 182 const void *params, size_t paramLen, 183 const RsScriptCall *sc, size_t scLen) { 184 Script *s = static_cast<Script *>(vs); 185 // The rs.spec generated code does not handle the absence of an actual 186 // input for sc. Instead, it retains an existing pointer value (the prior 187 // field in the packed data object). This can cause confusion because 188 // drivers might now inspect bogus sc data. 189 if (scLen == 0) { 190 sc = NULL; 191 } 192 s->runForEach(rsc, slot, 193 static_cast<const Allocation *>(vain), static_cast<Allocation *>(vaout), 194 params, paramLen, sc); 195 196 } 197 198 void rsi_ScriptInvoke(Context *rsc, RsScript vs, uint32_t slot) { 199 Script *s = static_cast<Script *>(vs); 200 s->Invoke(rsc, slot, NULL, 0); 201 } 202 203 204 void rsi_ScriptInvokeData(Context *rsc, RsScript vs, uint32_t slot, void *data) { 205 Script *s = static_cast<Script *>(vs); 206 s->Invoke(rsc, slot, NULL, 0); 207 } 208 209 void rsi_ScriptInvokeV(Context *rsc, RsScript vs, uint32_t slot, const void *data, size_t len) { 210 Script *s = static_cast<Script *>(vs); 211 s->Invoke(rsc, slot, data, len); 212 } 213 214 void rsi_ScriptSetVarI(Context *rsc, RsScript vs, uint32_t slot, int value) { 215 Script *s = static_cast<Script *>(vs); 216 s->setVar(slot, &value, sizeof(value)); 217 } 218 219 void rsi_ScriptSetVarObj(Context *rsc, RsScript vs, uint32_t slot, RsObjectBase value) { 220 Script *s = static_cast<Script *>(vs); 221 ObjectBase *o = static_cast<ObjectBase *>(value); 222 s->setVarObj(slot, o); 223 } 224 225 void rsi_ScriptSetVarJ(Context *rsc, RsScript vs, uint32_t slot, int64_t value) { 226 Script *s = static_cast<Script *>(vs); 227 s->setVar(slot, &value, sizeof(value)); 228 } 229 230 void rsi_ScriptSetVarF(Context *rsc, RsScript vs, uint32_t slot, float value) { 231 Script *s = static_cast<Script *>(vs); 232 s->setVar(slot, &value, sizeof(value)); 233 } 234 235 void rsi_ScriptSetVarD(Context *rsc, RsScript vs, uint32_t slot, double value) { 236 Script *s = static_cast<Script *>(vs); 237 s->setVar(slot, &value, sizeof(value)); 238 } 239 240 void rsi_ScriptSetVarV(Context *rsc, RsScript vs, uint32_t slot, const void *data, size_t len) { 241 Script *s = static_cast<Script *>(vs); 242 s->setVar(slot, data, len); 243 } 244 245 void rsi_ScriptGetVarV(Context *rsc, RsScript vs, uint32_t slot, void *data, size_t len) { 246 Script *s = static_cast<Script *>(vs); 247 s->getVar(slot, data, len); 248 } 249 250 void rsi_ScriptSetVarVE(Context *rsc, RsScript vs, uint32_t slot, 251 const void *data, size_t len, RsElement ve, 252 const size_t *dims, size_t dimLen) { 253 Script *s = static_cast<Script *>(vs); 254 Element *e = static_cast<Element *>(ve); 255 s->setVar(slot, data, len, e, dims, dimLen); 256 } 257 258 } 259 } 260 261
