rsdBcc.cpp revision fb6dfa66fe274ae6562c00084869f8fadf8d197e 
1/*
2 * Copyright (C) 2011-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 "rsdCore.h"
18
19#include <bcc/BCCContext.h>
20#include <bcc/Renderscript/RSCompilerDriver.h>
21#include <bcc/Renderscript/RSExecutable.h>
22#include <bcc/Renderscript/RSInfo.h>
23
24#include "rsdBcc.h"
25#include "rsdRuntime.h"
26#include "rsdAllocation.h"
27#include "rsdIntrinsics.h"
28
29#include "rsContext.h"
30#include "rsElement.h"
31#include "rsScriptC.h"
32
33#include "utils/Vector.h"
34#include "utils/Timers.h"
35#include "utils/StopWatch.h"
36
37using namespace android;
38using namespace android::renderscript;
39
40
41static Script * setTLS(Script *sc) {
42    ScriptTLSStruct * tls = (ScriptTLSStruct *)pthread_getspecific(rsdgThreadTLSKey);
43    rsAssert(tls);
44    Script *old = tls->mScript;
45    tls->mScript = sc;
46    return old;
47}
48
49
50bool rsdScriptInit(const Context *rsc,
51                     ScriptC *script,
52                     char const *resName,
53                     char const *cacheDir,
54                     uint8_t const *bitcode,
55                     size_t bitcodeSize,
56                     uint32_t flags) {
57    //ALOGE("rsdScriptCreate %p %p %p %p %i %i %p", rsc, resName, cacheDir, bitcode, bitcodeSize, flags, lookupFunc);
58    //ALOGE("rsdScriptInit %p %p", rsc, script);
59
60    pthread_mutex_lock(&rsdgInitMutex);
61
62    bcc::RSExecutable *exec;
63    const bcc::RSInfo *info;
64    DrvScript *drv = (DrvScript *)calloc(1, sizeof(DrvScript));
65    if (drv == NULL) {
66        goto error;
67    }
68    script->mHal.drv = drv;
69
70    drv->mCompilerContext = NULL;
71    drv->mCompilerDriver = NULL;
72    drv->mExecutable = NULL;
73
74    drv->mCompilerContext = new bcc::BCCContext();
75    if (drv->mCompilerContext == NULL) {
76        ALOGE("bcc: FAILS to create compiler context (out of memory)");
77        goto error;
78    }
79
80    drv->mCompilerDriver = new bcc::RSCompilerDriver();
81    if (drv->mCompilerDriver == NULL) {
82        ALOGE("bcc: FAILS to create compiler driver (out of memory)");
83        goto error;
84    }
85
86    script->mHal.info.isThreadable = true;
87
88    drv->mCompilerDriver->setRSRuntimeLookupFunction(rsdLookupRuntimeStub);
89    drv->mCompilerDriver->setRSRuntimeLookupContext(script);
90
91    exec = drv->mCompilerDriver->build(*drv->mCompilerContext,
92                                       cacheDir, resName,
93                                       (const char *)bitcode, bitcodeSize);
94
95    if (exec == NULL) {
96        ALOGE("bcc: FAILS to prepare executable for '%s'", resName);
97        goto error;
98    }
99
100    drv->mExecutable = exec;
101
102    exec->setThreadable(script->mHal.info.isThreadable);
103    if (!exec->syncInfo()) {
104        ALOGW("bcc: FAILS to synchronize the RS info file to the disk");
105    }
106
107    drv->mRoot = reinterpret_cast<int (*)()>(exec->getSymbolAddress("root"));
108    drv->mRootExpand =
109        reinterpret_cast<int (*)()>(exec->getSymbolAddress("root.expand"));
110    drv->mInit = reinterpret_cast<void (*)()>(exec->getSymbolAddress("init"));
111    drv->mFreeChildren =
112        reinterpret_cast<void (*)()>(exec->getSymbolAddress(".rs.dtor"));
113
114    info = &drv->mExecutable->getInfo();
115    // Copy info over to runtime
116    script->mHal.info.exportedFunctionCount = info->getExportFuncNames().size();
117    script->mHal.info.exportedVariableCount = info->getExportVarNames().size();
118    script->mHal.info.exportedPragmaCount = info->getPragmas().size();
119    script->mHal.info.exportedPragmaKeyList =
120        const_cast<const char**>(exec->getPragmaKeys().array());
121    script->mHal.info.exportedPragmaValueList =
122        const_cast<const char**>(exec->getPragmaValues().array());
123
124    if (drv->mRootExpand) {
125        script->mHal.info.root = drv->mRootExpand;
126    } else {
127        script->mHal.info.root = drv->mRoot;
128    }
129
130    if (script->mHal.info.exportedVariableCount) {
131        drv->mBoundAllocs = new Allocation *[script->mHal.info.exportedVariableCount];
132        memset(drv->mBoundAllocs, 0, sizeof(void *) * script->mHal.info.exportedVariableCount);
133    }
134
135    pthread_mutex_unlock(&rsdgInitMutex);
136    return true;
137
138error:
139
140    pthread_mutex_unlock(&rsdgInitMutex);
141    if (drv) {
142        delete drv->mCompilerContext;
143        delete drv->mCompilerDriver;
144        delete drv->mExecutable;
145        delete[] drv->mBoundAllocs;
146        free(drv);
147    }
148    script->mHal.drv = NULL;
149    return false;
150
151}
152
153bool rsdInitIntrinsic(const Context *rsc, Script *s, RsScriptIntrinsicID iid, Element *e) {
154    pthread_mutex_lock(&rsdgInitMutex);
155
156    DrvScript *drv = (DrvScript *)calloc(1, sizeof(DrvScript));
157    if (drv == NULL) {
158        goto error;
159    }
160    s->mHal.drv = drv;
161    drv->mIntrinsicID = iid;
162    drv->mIntrinsicData = rsdIntrinsic_Init(rsc, s, iid, &drv->mIntrinsicFuncs);
163    s->mHal.info.isThreadable = true;
164
165    pthread_mutex_unlock(&rsdgInitMutex);
166    return true;
167
168error:
169    pthread_mutex_unlock(&rsdgInitMutex);
170    return false;
171}
172
173typedef void (*rs_t)(const void *, void *, const void *, uint32_t, uint32_t, uint32_t, uint32_t);
174
175static void wc_xy(void *usr, uint32_t idx) {
176    MTLaunchStruct *mtls = (MTLaunchStruct *)usr;
177    RsForEachStubParamStruct p;
178    memcpy(&p, &mtls->fep, sizeof(p));
179    p.lid = idx;
180    RsdHal * dc = (RsdHal *)mtls->rsc->mHal.drv;
181    uint32_t sig = mtls->sig;
182
183#if defined(ARCH_ARM_RS_USE_CACHED_SCANLINE_WRITE)
184    unsigned char buf[1024 * 8];
185#endif
186
187    outer_foreach_t fn = (outer_foreach_t) mtls->kernel;
188    while (1) {
189        uint32_t slice = (uint32_t)android_atomic_inc(&mtls->mSliceNum);
190        uint32_t yStart = mtls->yStart + slice * mtls->mSliceSize;
191        uint32_t yEnd = yStart + mtls->mSliceSize;
192        yEnd = rsMin(yEnd, mtls->yEnd);
193        if (yEnd <= yStart) {
194            return;
195        }
196
197        //ALOGE("usr idx %i, x %i,%i  y %i,%i", idx, mtls->xStart, mtls->xEnd, yStart, yEnd);
198        //ALOGE("usr ptr in %p,  out %p", mtls->fep.ptrIn, mtls->fep.ptrOut);
199
200#if defined(ARCH_ARM_RS_USE_CACHED_SCANLINE_WRITE)
201        if (mtls->fep.yStrideOut < sizeof(buf)) {
202            p.out = buf;
203            for (p.y = yStart; p.y < yEnd; p.y++) {
204                p.in = mtls->fep.ptrIn + (mtls->fep.yStrideIn * p.y);
205                fn(&p, mtls->xStart, mtls->xEnd, mtls->fep.eStrideIn, mtls->fep.eStrideOut);
206                memcpy(mtls->fep.ptrOut + (mtls->fep.yStrideOut * p.y), buf, mtls->fep.yStrideOut);
207            }
208        } else
209#endif
210            {
211            for (p.y = yStart; p.y < yEnd; p.y++) {
212                p.out = mtls->fep.ptrOut + (mtls->fep.yStrideOut * p.y) +
213                        (mtls->fep.eStrideOut * mtls->xStart);
214                p.in = mtls->fep.ptrIn + (mtls->fep.yStrideIn * p.y) +
215                       (mtls->fep.eStrideIn * mtls->xStart);
216                fn(&p, mtls->xStart, mtls->xEnd, mtls->fep.eStrideIn, mtls->fep.eStrideOut);
217            }
218        }
219    }
220}
221
222static void wc_x(void *usr, uint32_t idx) {
223    MTLaunchStruct *mtls = (MTLaunchStruct *)usr;
224    RsForEachStubParamStruct p;
225    memcpy(&p, &mtls->fep, sizeof(p));
226    p.lid = idx;
227    RsdHal * dc = (RsdHal *)mtls->rsc->mHal.drv;
228    uint32_t sig = mtls->sig;
229
230    outer_foreach_t fn = (outer_foreach_t) mtls->kernel;
231    while (1) {
232        uint32_t slice = (uint32_t)android_atomic_inc(&mtls->mSliceNum);
233        uint32_t xStart = mtls->xStart + slice * mtls->mSliceSize;
234        uint32_t xEnd = xStart + mtls->mSliceSize;
235        xEnd = rsMin(xEnd, mtls->xEnd);
236        if (xEnd <= xStart) {
237            return;
238        }
239
240        //ALOGE("usr slice %i idx %i, x %i,%i", slice, idx, xStart, xEnd);
241        //ALOGE("usr ptr in %p,  out %p", mtls->fep.ptrIn, mtls->fep.ptrOut);
242
243        p.out = mtls->fep.ptrOut + (mtls->fep.eStrideOut * xStart);
244        p.in = mtls->fep.ptrIn + (mtls->fep.eStrideIn * xStart);
245        fn(&p, xStart, xEnd, mtls->fep.eStrideIn, mtls->fep.eStrideOut);
246    }
247}
248
249void rsdScriptInvokeForEachMtlsSetup(const Context *rsc,
250                                     const Allocation * ain,
251                                     Allocation * aout,
252                                     const void * usr,
253                                     uint32_t usrLen,
254                                     const RsScriptCall *sc,
255                                     MTLaunchStruct *mtls) {
256
257    memset(mtls, 0, sizeof(MTLaunchStruct));
258
259    if (ain) {
260        mtls->fep.dimX = ain->getType()->getDimX();
261        mtls->fep.dimY = ain->getType()->getDimY();
262        mtls->fep.dimZ = ain->getType()->getDimZ();
263        //mtls->dimArray = ain->getType()->getDimArray();
264    } else if (aout) {
265        mtls->fep.dimX = aout->getType()->getDimX();
266        mtls->fep.dimY = aout->getType()->getDimY();
267        mtls->fep.dimZ = aout->getType()->getDimZ();
268        //mtls->dimArray = aout->getType()->getDimArray();
269    } else {
270        rsc->setError(RS_ERROR_BAD_SCRIPT, "rsForEach called with null allocations");
271        return;
272    }
273
274    if (!sc || (sc->xEnd == 0)) {
275        mtls->xEnd = mtls->fep.dimX;
276    } else {
277        rsAssert(sc->xStart < mtls->fep.dimX);
278        rsAssert(sc->xEnd <= mtls->fep.dimX);
279        rsAssert(sc->xStart < sc->xEnd);
280        mtls->xStart = rsMin(mtls->fep.dimX, sc->xStart);
281        mtls->xEnd = rsMin(mtls->fep.dimX, sc->xEnd);
282        if (mtls->xStart >= mtls->xEnd) return;
283    }
284
285    if (!sc || (sc->yEnd == 0)) {
286        mtls->yEnd = mtls->fep.dimY;
287    } else {
288        rsAssert(sc->yStart < mtls->fep.dimY);
289        rsAssert(sc->yEnd <= mtls->fep.dimY);
290        rsAssert(sc->yStart < sc->yEnd);
291        mtls->yStart = rsMin(mtls->fep.dimY, sc->yStart);
292        mtls->yEnd = rsMin(mtls->fep.dimY, sc->yEnd);
293        if (mtls->yStart >= mtls->yEnd) return;
294    }
295
296    mtls->xEnd = rsMax((uint32_t)1, mtls->xEnd);
297    mtls->yEnd = rsMax((uint32_t)1, mtls->yEnd);
298    mtls->zEnd = rsMax((uint32_t)1, mtls->zEnd);
299    mtls->arrayEnd = rsMax((uint32_t)1, mtls->arrayEnd);
300
301    rsAssert(!ain || (ain->getType()->getDimZ() == 0));
302
303    Context *mrsc = (Context *)rsc;
304    mtls->rsc = mrsc;
305    mtls->ain = ain;
306    mtls->aout = aout;
307    mtls->fep.usr = usr;
308    mtls->fep.usrLen = usrLen;
309    mtls->mSliceSize = 10;
310    mtls->mSliceNum = 0;
311
312    mtls->fep.ptrIn = NULL;
313    mtls->fep.eStrideIn = 0;
314
315    if (ain) {
316        DrvAllocation *aindrv = (DrvAllocation *)ain->mHal.drv;
317        mtls->fep.ptrIn = (const uint8_t *)aindrv->lod[0].mallocPtr;
318        mtls->fep.eStrideIn = ain->getType()->getElementSizeBytes();
319        mtls->fep.yStrideIn = aindrv->lod[0].stride;
320    }
321
322    mtls->fep.ptrOut = NULL;
323    mtls->fep.eStrideOut = 0;
324    if (aout) {
325        DrvAllocation *aoutdrv = (DrvAllocation *)aout->mHal.drv;
326        mtls->fep.ptrOut = (uint8_t *)aoutdrv->lod[0].mallocPtr;
327        mtls->fep.eStrideOut = aout->getType()->getElementSizeBytes();
328        mtls->fep.yStrideOut = aoutdrv->lod[0].stride;
329    }
330}
331
332void rsdScriptLaunchThreads(const Context *rsc,
333                            Script *s,
334                            uint32_t slot,
335                            const Allocation * ain,
336                            Allocation * aout,
337                            const void * usr,
338                            uint32_t usrLen,
339                            const RsScriptCall *sc,
340                            MTLaunchStruct *mtls) {
341
342    Script * oldTLS = setTLS(s);
343    Context *mrsc = (Context *)rsc;
344    RsdHal * dc = (RsdHal *)mtls->rsc->mHal.drv;
345
346    if ((dc->mWorkers.mCount >= 1) && s->mHal.info.isThreadable && !dc->mInForEach) {
347        const size_t targetByteChunk = 16 * 1024;
348        dc->mInForEach = true;
349        if (mtls->fep.dimY > 1) {
350            uint32_t s1 = mtls->fep.dimY / ((dc->mWorkers.mCount + 1) * 4);
351            uint32_t s2 = 0;
352
353            // This chooses our slice size to rate limit atomic ops to
354            // one per 16k bytes of reads/writes.
355            if (mtls->fep.yStrideOut) {
356                s2 = targetByteChunk / mtls->fep.yStrideOut;
357            } else {
358                s2 = targetByteChunk / mtls->fep.yStrideIn;
359            }
360            mtls->mSliceSize = rsMin(s1, s2);
361
362            if(mtls->mSliceSize < 1) {
363                mtls->mSliceSize = 1;
364            }
365
366            rsdLaunchThreads(mrsc, wc_xy, mtls);
367        } else {
368            uint32_t s1 = mtls->fep.dimX / ((dc->mWorkers.mCount + 1) * 4);
369            uint32_t s2 = 0;
370
371            // This chooses our slice size to rate limit atomic ops to
372            // one per 16k bytes of reads/writes.
373            if (mtls->fep.eStrideOut) {
374                s2 = targetByteChunk / mtls->fep.eStrideOut;
375            } else {
376                s2 = targetByteChunk / mtls->fep.eStrideIn;
377            }
378            mtls->mSliceSize = rsMin(s1, s2);
379
380            if(mtls->mSliceSize < 1) {
381                mtls->mSliceSize = 1;
382            }
383
384            rsdLaunchThreads(mrsc, wc_x, mtls);
385        }
386        dc->mInForEach = false;
387
388        //ALOGE("launch 1");
389    } else {
390        RsForEachStubParamStruct p;
391        memcpy(&p, &mtls->fep, sizeof(p));
392        p.lid = 0;
393        uint32_t sig = mtls->sig;
394
395        //ALOGE("launch 3");
396        outer_foreach_t fn = (outer_foreach_t) mtls->kernel;
397        for (p.ar[0] = mtls->arrayStart; p.ar[0] < mtls->arrayEnd; p.ar[0]++) {
398            for (p.z = mtls->zStart; p.z < mtls->zEnd; p.z++) {
399                for (p.y = mtls->yStart; p.y < mtls->yEnd; p.y++) {
400                    uint32_t offset = mtls->fep.dimY * mtls->fep.dimZ * p.ar[0] +
401                                      mtls->fep.dimY * p.z + p.y;
402                    p.out = mtls->fep.ptrOut + (mtls->fep.yStrideOut * offset) +
403                            (mtls->fep.eStrideOut * mtls->xStart);
404                    p.in = mtls->fep.ptrIn + (mtls->fep.yStrideIn * offset) +
405                           (mtls->fep.eStrideIn * mtls->xStart);
406                    fn(&p, mtls->xStart, mtls->xEnd, mtls->fep.eStrideIn, mtls->fep.eStrideOut);
407                }
408            }
409        }
410    }
411
412    setTLS(oldTLS);
413}
414
415void rsdScriptInvokeForEach(const Context *rsc,
416                            Script *s,
417                            uint32_t slot,
418                            const Allocation * ain,
419                            Allocation * aout,
420                            const void * usr,
421                            uint32_t usrLen,
422                            const RsScriptCall *sc) {
423
424    RsdHal * dc = (RsdHal *)rsc->mHal.drv;
425
426    MTLaunchStruct mtls;
427    rsdScriptInvokeForEachMtlsSetup(rsc, ain, aout, usr, usrLen, sc, &mtls);
428    mtls.script = s;
429    mtls.fep.slot = slot;
430
431    DrvScript *drv = (DrvScript *)s->mHal.drv;
432    if (drv->mIntrinsicID) {
433        mtls.kernel = (void (*)())drv->mIntrinsicFuncs.root;
434        mtls.fep.usr = drv->mIntrinsicData;
435    } else {
436        rsAssert(slot < drv->mExecutable->getExportForeachFuncAddrs().size());
437        mtls.kernel = reinterpret_cast<ForEachFunc_t>(
438                          drv->mExecutable->getExportForeachFuncAddrs()[slot]);
439        rsAssert(mtls.kernel != NULL);
440        mtls.sig = drv->mExecutable->getInfo().getExportForeachFuncs()[slot].second;
441    }
442
443
444    rsdScriptLaunchThreads(rsc, s, slot, ain, aout, usr, usrLen, sc, &mtls);
445}
446
447
448int rsdScriptInvokeRoot(const Context *dc, Script *script) {
449    DrvScript *drv = (DrvScript *)script->mHal.drv;
450
451    Script * oldTLS = setTLS(script);
452    int ret = drv->mRoot();
453    setTLS(oldTLS);
454
455    return ret;
456}
457
458void rsdScriptInvokeInit(const Context *dc, Script *script) {
459    DrvScript *drv = (DrvScript *)script->mHal.drv;
460
461    if (drv->mInit) {
462        drv->mInit();
463    }
464}
465
466void rsdScriptInvokeFreeChildren(const Context *dc, Script *script) {
467    DrvScript *drv = (DrvScript *)script->mHal.drv;
468
469    if (drv->mFreeChildren) {
470        drv->mFreeChildren();
471    }
472}
473
474void rsdScriptInvokeFunction(const Context *dc, Script *script,
475                            uint32_t slot,
476                            const void *params,
477                            size_t paramLength) {
478    DrvScript *drv = (DrvScript *)script->mHal.drv;
479    //ALOGE("invoke %p %p %i %p %i", dc, script, slot, params, paramLength);
480
481    Script * oldTLS = setTLS(script);
482    reinterpret_cast<void (*)(const void *, uint32_t)>(
483        drv->mExecutable->getExportFuncAddrs()[slot])(params, paramLength);
484    setTLS(oldTLS);
485}
486
487void rsdScriptSetGlobalVar(const Context *dc, const Script *script,
488                           uint32_t slot, void *data, size_t dataLength) {
489    DrvScript *drv = (DrvScript *)script->mHal.drv;
490    //rsAssert(!script->mFieldIsObject[slot]);
491    //ALOGE("setGlobalVar %p %p %i %p %i", dc, script, slot, data, dataLength);
492
493    if (drv->mIntrinsicID) {
494        drv->mIntrinsicFuncs.setVar(dc, script, drv->mIntrinsicData, slot, data, dataLength);
495        return;
496    }
497
498    int32_t *destPtr = reinterpret_cast<int32_t *>(
499                          drv->mExecutable->getExportVarAddrs()[slot]);
500    if (!destPtr) {
501        //ALOGV("Calling setVar on slot = %i which is null", slot);
502        return;
503    }
504
505    memcpy(destPtr, data, dataLength);
506}
507
508void rsdScriptSetGlobalVarWithElemDims(
509        const android::renderscript::Context *dc,
510        const android::renderscript::Script *script,
511        uint32_t slot, void *data, size_t dataLength,
512        const android::renderscript::Element *elem,
513        const size_t *dims, size_t dimLength) {
514    DrvScript *drv = (DrvScript *)script->mHal.drv;
515
516    int32_t *destPtr = reinterpret_cast<int32_t *>(
517        drv->mExecutable->getExportVarAddrs()[slot]);
518    if (!destPtr) {
519        //ALOGV("Calling setVar on slot = %i which is null", slot);
520        return;
521    }
522
523    // We want to look at dimension in terms of integer components,
524    // but dimLength is given in terms of bytes.
525    dimLength /= sizeof(int);
526
527    // Only a single dimension is currently supported.
528    rsAssert(dimLength == 1);
529    if (dimLength == 1) {
530        // First do the increment loop.
531        size_t stride = elem->getSizeBytes();
532        char *cVal = reinterpret_cast<char *>(data);
533        for (size_t i = 0; i < dims[0]; i++) {
534            elem->incRefs(cVal);
535            cVal += stride;
536        }
537
538        // Decrement loop comes after (to prevent race conditions).
539        char *oldVal = reinterpret_cast<char *>(destPtr);
540        for (size_t i = 0; i < dims[0]; i++) {
541            elem->decRefs(oldVal);
542            oldVal += stride;
543        }
544    }
545
546    memcpy(destPtr, data, dataLength);
547}
548
549void rsdScriptSetGlobalBind(const Context *dc, const Script *script, uint32_t slot, Allocation *data) {
550    DrvScript *drv = (DrvScript *)script->mHal.drv;
551
552    //rsAssert(!script->mFieldIsObject[slot]);
553    //ALOGE("setGlobalBind %p %p %i %p", dc, script, slot, data);
554
555    rsAssert(!drv->mIntrinsicID);
556
557    int32_t *destPtr = reinterpret_cast<int32_t *>(
558                          drv->mExecutable->getExportVarAddrs()[slot]);
559    if (!destPtr) {
560        //ALOGV("Calling setVar on slot = %i which is null", slot);
561        return;
562    }
563
564    void *ptr = NULL;
565    drv->mBoundAllocs[slot] = data;
566    if(data) {
567        DrvAllocation *allocDrv = (DrvAllocation *)data->mHal.drv;
568        ptr = allocDrv->lod[0].mallocPtr;
569    }
570    memcpy(destPtr, &ptr, sizeof(void *));
571}
572
573void rsdScriptSetGlobalObj(const Context *dc, const Script *script, uint32_t slot, ObjectBase *data) {
574    DrvScript *drv = (DrvScript *)script->mHal.drv;
575    //rsAssert(script->mFieldIsObject[slot]);
576    //ALOGE("setGlobalObj %p %p %i %p", dc, script, slot, data);
577
578    if (drv->mIntrinsicID) {
579        drv->mIntrinsicFuncs.setVarObj(dc, script, drv->mIntrinsicData, slot,
580                                       static_cast<Allocation *>(data));
581        return;
582    }
583
584    int32_t *destPtr = reinterpret_cast<int32_t *>(
585                          drv->mExecutable->getExportVarAddrs()[slot]);
586    if (!destPtr) {
587        //ALOGV("Calling setVar on slot = %i which is null", slot);
588        return;
589    }
590
591    rsrSetObject(dc, script, (ObjectBase **)destPtr, data);
592}
593
594void rsdScriptDestroy(const Context *dc, Script *script) {
595    DrvScript *drv = (DrvScript *)script->mHal.drv;
596
597    if (drv == NULL) {
598        return;
599    }
600
601    if (drv->mExecutable) {
602        Vector<void *>::const_iterator var_addr_iter =
603            drv->mExecutable->getExportVarAddrs().begin();
604        Vector<void *>::const_iterator var_addr_end =
605            drv->mExecutable->getExportVarAddrs().end();
606
607        bcc::RSInfo::ObjectSlotListTy::const_iterator is_object_iter =
608            drv->mExecutable->getInfo().getObjectSlots().begin();
609        bcc::RSInfo::ObjectSlotListTy::const_iterator is_object_end =
610            drv->mExecutable->getInfo().getObjectSlots().end();
611
612        while ((var_addr_iter != var_addr_end) &&
613               (is_object_iter != is_object_end)) {
614            // The field address can be NULL if the script-side has optimized
615            // the corresponding global variable away.
616            ObjectBase **obj_addr =
617                reinterpret_cast<ObjectBase **>(*var_addr_iter);
618            if (*is_object_iter) {
619                if (*var_addr_iter != NULL) {
620                    rsrClearObject(dc, script, obj_addr);
621                }
622            }
623            var_addr_iter++;
624            is_object_iter++;
625        }
626    }
627
628    delete drv->mCompilerContext;
629    delete drv->mCompilerDriver;
630    delete drv->mExecutable;
631    delete[] drv->mBoundAllocs;
632    free(drv);
633    script->mHal.drv = NULL;
634}
635
636Allocation * rsdScriptGetAllocationForPointer(const android::renderscript::Context *dc,
637                                              const android::renderscript::Script *sc,
638                                              const void *ptr) {
639    DrvScript *drv = (DrvScript *)sc->mHal.drv;
640    if (!ptr) {
641        return NULL;
642    }
643
644    for (uint32_t ct=0; ct < sc->mHal.info.exportedVariableCount; ct++) {
645        Allocation *a = drv->mBoundAllocs[ct];
646        if (!a) continue;
647        DrvAllocation *adrv = (DrvAllocation *)a->mHal.drv;
648        if (adrv->lod[0].mallocPtr == ptr) {
649            return a;
650        }
651    }
652    ALOGE("rsGetAllocation, failed to find %p", ptr);
653    return NULL;
654}
655
656