rsCpuScript.h revision e195a3f57ace3b66d313a6ee88c6e93d5c9d87f4 
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#ifndef RSD_BCC_H
18#define RSD_BCC_H
19
20#include <rs_hal.h>
21#include <rsRuntime.h>
22
23#include "rsCpuCore.h"
24
25namespace bcc {
26    class BCCContext;
27    class RSCompilerDriver;
28    class RSExecutable;
29}
30
31namespace android {
32namespace renderscript {
33
34
35
36class RsdCpuScriptImpl : public RsdCpuReferenceImpl::CpuScript {
37public:
38    typedef void (*outer_foreach_t)(
39        const RsForEachStubParamStruct *,
40        uint32_t x1, uint32_t x2,
41        uint32_t instep, uint32_t outstep);
42#ifdef RS_COMPATIBILITY_LIB
43    typedef void (* InvokeFunc_t)(void);
44    typedef void (* ForEachFunc_t)(void);
45    typedef int (* RootFunc_t)(void);
46    typedef void (*WorkerCallback_t)(void *usr, uint32_t idx);
47#endif
48
49    bool init(char const *resName, char const *cacheDir,
50              uint8_t const *bitcode, size_t bitcodeSize, uint32_t flags);
51    virtual void populateScript(Script *);
52
53    virtual void invokeFunction(uint32_t slot, const void *params, size_t paramLength);
54    virtual int invokeRoot();
55    virtual void preLaunch(uint32_t slot, const Allocation * ain,
56                           Allocation * aout, const void * usr,
57                           uint32_t usrLen, const RsScriptCall *sc);
58    virtual void postLaunch(uint32_t slot, const Allocation * ain,
59                            Allocation * aout, const void * usr,
60                            uint32_t usrLen, const RsScriptCall *sc);
61    virtual void invokeForEach(uint32_t slot,
62                       const Allocation * ain,
63                       Allocation * aout,
64                       const void * usr,
65                       uint32_t usrLen,
66                       const RsScriptCall *sc);
67    virtual void invokeInit();
68    virtual void invokeFreeChildren();
69
70    virtual void setGlobalVar(uint32_t slot, const void *data, size_t dataLength);
71    virtual void getGlobalVar(uint32_t slot, void *data, size_t dataLength);
72    virtual void setGlobalVarWithElemDims(uint32_t slot, const void *data, size_t dataLength,
73                                  const Element *e, const size_t *dims, size_t dimLength);
74    virtual void setGlobalBind(uint32_t slot, Allocation *data);
75    virtual void setGlobalObj(uint32_t slot, ObjectBase *data);
76
77
78    virtual ~RsdCpuScriptImpl();
79    RsdCpuScriptImpl(RsdCpuReferenceImpl *ctx, const Script *s);
80
81    const Script * getScript() {return mScript;}
82
83    void forEachMtlsSetup(const Allocation * ain, Allocation * aout,
84                          const void * usr, uint32_t usrLen,
85                          const RsScriptCall *sc, MTLaunchStruct *mtls);
86    virtual void forEachKernelSetup(uint32_t slot, MTLaunchStruct *mtls);
87
88
89    const RsdCpuReference::CpuSymbol * lookupSymbolMath(const char *sym);
90    static void * lookupRuntimeStub(void* pContext, char const* name);
91
92    virtual Allocation * getAllocationForPointer(const void *ptr) const;
93
94#ifndef FAKE_ARM64_BUILD
95#ifndef RS_COMPATIBILITY_LIB
96    virtual  void * getRSExecutable() { return mExecutable; }
97#endif
98#else
99    virtual void* getRSExecutable() { return NULL; }
100#endif
101
102protected:
103    RsdCpuReferenceImpl *mCtx;
104    const Script *mScript;
105
106#ifndef FAKE_ARM64_BUILD
107#ifndef RS_COMPATIBILITY_LIB
108    int (*mRoot)();
109    int (*mRootExpand)();
110    void (*mInit)();
111    void (*mFreeChildren)();
112
113    bcc::BCCContext *mCompilerContext;
114    bcc::RSCompilerDriver *mCompilerDriver;
115    bcc::RSExecutable *mExecutable;
116#else
117    void *mScriptSO;
118    RootFunc_t mRoot;
119    RootFunc_t mRootExpand;
120    InvokeFunc_t mInit;
121    InvokeFunc_t mFreeChildren;
122    InvokeFunc_t *mInvokeFunctions;
123    ForEachFunc_t *mForEachFunctions;
124
125    void **mFieldAddress;
126    bool *mFieldIsObject;
127    uint32_t *mForEachSignatures;
128
129    // for populate script
130    //int mVersionMajor;
131    //int mVersionMinor;
132    size_t mExportedVariableCount;
133    size_t mExportedFunctionCount;
134#endif
135#endif
136
137    Allocation **mBoundAllocs;
138    void * mIntrinsicData;
139    bool mIsThreadable;
140
141};
142
143
144Allocation * rsdScriptGetAllocationForPointer(
145                        const Context *dc,
146                        const Script *script,
147                        const void *);
148
149
150
151}
152}
153
154#endif
155