rsScriptC.cpp revision 466bdc4008841f095ff8d1f13bfc36bcf11c3dd9 
1/*
2 * Copyright (C) 2009 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 "rsScriptC.h"
19#include "utils/Timers.h"
20#include "utils/StopWatch.h"
21
22#ifndef ANDROID_RS_SERIALIZE
23#include <bcinfo/BitcodeTranslator.h>
24#endif
25
26using namespace android;
27using namespace android::renderscript;
28
29#define GET_TLS()  Context::ScriptTLSStruct * tls = \
30    (Context::ScriptTLSStruct *)pthread_getspecific(Context::gThreadTLSKey); \
31    Context * rsc = tls->mContext; \
32    ScriptC * sc = (ScriptC *) tls->mScript
33
34ScriptC::ScriptC(Context *rsc) : Script(rsc) {
35#ifndef ANDROID_RS_SERIALIZE
36    BT = NULL;
37#endif
38}
39
40ScriptC::~ScriptC() {
41#ifndef ANDROID_RS_SERIALIZE
42    if (BT) {
43        delete BT;
44        BT = NULL;
45    }
46#endif
47    mRSC->mHal.funcs.script.invokeFreeChildren(mRSC, this);
48    mRSC->mHal.funcs.script.destroy(mRSC, this);
49}
50
51void ScriptC::setupScript(Context *rsc) {
52    mEnviroment.mStartTimeMillis
53                = nanoseconds_to_milliseconds(systemTime(SYSTEM_TIME_MONOTONIC));
54
55    for (uint32_t ct=0; ct < mHal.info.exportedVariableCount; ct++) {
56        if (mSlots[ct].get() && !mTypes[ct].get()) {
57            mTypes[ct].set(mSlots[ct]->getType());
58        }
59
60        if (!mTypes[ct].get())
61            continue;
62        void *ptr = NULL;
63        if (mSlots[ct].get()) {
64            ptr = mSlots[ct]->getPtr();
65        }
66
67        rsc->mHal.funcs.script.setGlobalBind(rsc, this, ct, ptr);
68    }
69}
70
71const Allocation *ScriptC::ptrToAllocation(const void *ptr) const {
72    //LOGE("ptr to alloc %p", ptr);
73    if (!ptr) {
74        return NULL;
75    }
76    for (uint32_t ct=0; ct < mHal.info.exportedVariableCount; ct++) {
77        if (!mSlots[ct].get())
78            continue;
79        if (mSlots[ct]->getPtr() == ptr) {
80            return mSlots[ct].get();
81        }
82    }
83    LOGE("ScriptC::ptrToAllocation, failed to find %p", ptr);
84    return NULL;
85}
86
87void ScriptC::setupGLState(Context *rsc) {
88    if (mEnviroment.mFragmentStore.get()) {
89        rsc->setProgramStore(mEnviroment.mFragmentStore.get());
90    }
91    if (mEnviroment.mFragment.get()) {
92        rsc->setProgramFragment(mEnviroment.mFragment.get());
93    }
94    if (mEnviroment.mVertex.get()) {
95        rsc->setProgramVertex(mEnviroment.mVertex.get());
96    }
97    if (mEnviroment.mRaster.get()) {
98        rsc->setProgramRaster(mEnviroment.mRaster.get());
99    }
100}
101
102uint32_t ScriptC::run(Context *rsc) {
103    if (mHal.info.root == NULL) {
104        rsc->setError(RS_ERROR_BAD_SCRIPT, "Attempted to run bad script");
105        return 0;
106    }
107
108    setupGLState(rsc);
109    setupScript(rsc);
110
111    uint32_t ret = 0;
112
113    if (rsc->props.mLogScripts) {
114        LOGV("%p ScriptC::run invoking root,  ptr %p", rsc, mHal.info.root);
115    }
116
117    ret = rsc->mHal.funcs.script.invokeRoot(rsc, this);
118
119    if (rsc->props.mLogScripts) {
120        LOGV("%p ScriptC::run invoking complete, ret=%i", rsc, ret);
121    }
122
123    return ret;
124}
125
126
127void ScriptC::runForEach(Context *rsc,
128                         const Allocation * ain,
129                         Allocation * aout,
130                         const void * usr,
131                         size_t usrBytes,
132                         const RsScriptCall *sc) {
133
134    Context::PushState ps(rsc);
135
136    setupGLState(rsc);
137    setupScript(rsc);
138    rsc->mHal.funcs.script.invokeForEach(rsc, this, 0, ain, aout, usr, usrBytes, sc);
139}
140
141void ScriptC::Invoke(Context *rsc, uint32_t slot, const void *data, size_t len) {
142    if (slot >= mHal.info.exportedFunctionCount) {
143        rsc->setError(RS_ERROR_BAD_SCRIPT, "Calling invoke on bad script");
144        return;
145    }
146    setupScript(rsc);
147
148    if (rsc->props.mLogScripts) {
149        LOGV("%p ScriptC::Invoke invoking slot %i,  ptr %p", rsc, slot, this);
150    }
151    rsc->mHal.funcs.script.invokeFunction(rsc, this, slot, data, len);
152}
153
154ScriptCState::ScriptCState() {
155}
156
157ScriptCState::~ScriptCState() {
158}
159
160/*
161static void* symbolLookup(void* pContext, char const* name) {
162    const ScriptCState::SymbolTable_t *sym;
163    ScriptC *s = (ScriptC *)pContext;
164    if (!strcmp(name, "__isThreadable")) {
165      return (void*) s->mHal.info.isThreadable;
166    } else if (!strcmp(name, "__clearThreadable")) {
167      s->mHal.info.isThreadable = false;
168      return NULL;
169    }
170    sym = ScriptCState::lookupSymbol(name);
171    if (!sym) {
172        sym = ScriptCState::lookupSymbolCL(name);
173    }
174    if (!sym) {
175        sym = ScriptCState::lookupSymbolGL(name);
176    }
177    if (sym) {
178        s->mHal.info.isThreadable &= sym->threadable;
179        return sym->mPtr;
180    }
181    LOGE("ScriptC sym lookup failed for %s", name);
182    return NULL;
183}
184*/
185
186#if 0
187extern const char rs_runtime_lib_bc[];
188extern unsigned rs_runtime_lib_bc_size;
189#endif
190
191bool ScriptC::runCompiler(Context *rsc,
192                          const char *resName,
193                          const char *cacheDir,
194                          const uint8_t *bitcode,
195                          size_t bitcodeLen) {
196
197    //LOGE("runCompiler %p %p %p %p %p %i", rsc, this, resName, cacheDir, bitcode, bitcodeLen);
198#ifndef ANDROID_RS_SERIALIZE
199    uint32_t sdkVersion = rsc->getTargetSdkVersion();
200    if (BT) {
201        delete BT;
202    }
203    BT = new bcinfo::BitcodeTranslator((const char *)bitcode, bitcodeLen,
204                                       sdkVersion);
205    if (!BT->translate()) {
206        LOGE("Failed to translate bitcode from version: %u", sdkVersion);
207        delete BT;
208        BT = NULL;
209        return false;
210    }
211    bitcode = (const uint8_t *) BT->getTranslatedBitcode();
212    bitcodeLen = BT->getTranslatedBitcodeSize();
213#endif
214
215    rsc->mHal.funcs.script.init(rsc, this, resName, cacheDir, bitcode, bitcodeLen, 0);
216
217    mEnviroment.mFragment.set(rsc->getDefaultProgramFragment());
218    mEnviroment.mVertex.set(rsc->getDefaultProgramVertex());
219    mEnviroment.mFragmentStore.set(rsc->getDefaultProgramStore());
220    mEnviroment.mRaster.set(rsc->getDefaultProgramRaster());
221
222    rsc->mHal.funcs.script.invokeInit(rsc, this);
223
224    for (size_t i=0; i < mHal.info.exportedPragmaCount; ++i) {
225        const char * key = mHal.info.exportedPragmaKeyList[i];
226        const char * value = mHal.info.exportedPragmaValueList[i];
227        //LOGE("pragma %s %s", keys[i], values[i]);
228        if (!strcmp(key, "version")) {
229            if (!strcmp(value, "1")) {
230                continue;
231            }
232            LOGE("Invalid version pragma value: %s\n", value);
233            return false;
234        }
235
236        if (!strcmp(key, "stateVertex")) {
237            if (!strcmp(value, "default")) {
238                continue;
239            }
240            if (!strcmp(value, "parent")) {
241                mEnviroment.mVertex.clear();
242                continue;
243            }
244            LOGE("Unrecognized value %s passed to stateVertex", value);
245            return false;
246        }
247
248        if (!strcmp(key, "stateRaster")) {
249            if (!strcmp(value, "default")) {
250                continue;
251            }
252            if (!strcmp(value, "parent")) {
253                mEnviroment.mRaster.clear();
254                continue;
255            }
256            LOGE("Unrecognized value %s passed to stateRaster", value);
257            return false;
258        }
259
260        if (!strcmp(key, "stateFragment")) {
261            if (!strcmp(value, "default")) {
262                continue;
263            }
264            if (!strcmp(value, "parent")) {
265                mEnviroment.mFragment.clear();
266                continue;
267            }
268            LOGE("Unrecognized value %s passed to stateFragment", value);
269            return false;
270        }
271
272        if (!strcmp(key, "stateStore")) {
273            if (!strcmp(value, "default")) {
274                continue;
275            }
276            if (!strcmp(value, "parent")) {
277                mEnviroment.mFragmentStore.clear();
278                continue;
279            }
280            LOGE("Unrecognized value %s passed to stateStore", value);
281            return false;
282        }
283    }
284
285    mSlots = new ObjectBaseRef<Allocation>[mHal.info.exportedVariableCount];
286    mTypes = new ObjectBaseRef<const Type>[mHal.info.exportedVariableCount];
287
288    return true;
289}
290
291namespace android {
292namespace renderscript {
293
294RsScript rsi_ScriptCCreate(Context *rsc,
295                           const char *resName, size_t resName_length,
296                           const char *cacheDir, size_t cacheDir_length,
297                           const char *text, size_t text_length)
298{
299    ScriptC *s = new ScriptC(rsc);
300
301    if (!s->runCompiler(rsc, resName, cacheDir, (uint8_t *)text, text_length)) {
302        // Error during compile, destroy s and return null.
303        delete s;
304        return NULL;
305    }
306
307    s->incUserRef();
308    return s;
309}
310
311}
312}
313