Yang Ni | 1ffd86b | 2015-01-07 09:16:40 -0800 | [diff] [blame] | 1 | #ifndef CPU_REF_CPUSCRIPTGROUP2IMPL_H_ |
| 2 | #define CPU_REF_CPUSCRIPTGROUP2IMPL_H_ |
| 3 | |
Yang Ni | 1ffd86b | 2015-01-07 09:16:40 -0800 | [diff] [blame] | 4 | #include "rsd_cpu.h" |
Yang Ni | ff2bb54 | 2015-02-02 14:33:47 -0800 | [diff] [blame] | 5 | #include "rsList.h" |
Yang Ni | 1ffd86b | 2015-01-07 09:16:40 -0800 | [diff] [blame] | 6 | |
David Gross | b0abb14 | 2015-03-12 15:23:03 -0700 | [diff] [blame] | 7 | struct RsExpandKernelDriverInfo; |
David Gross | dced5c9 | 2015-03-11 16:12:42 -0700 | [diff] [blame] | 8 | |
Yang Ni | 1ffd86b | 2015-01-07 09:16:40 -0800 | [diff] [blame] | 9 | namespace android { |
| 10 | namespace renderscript { |
| 11 | |
| 12 | class Closure; |
| 13 | class RsdCpuScriptImpl; |
| 14 | class RsdCpuReferenceImpl; |
Yang Ni | da0f069 | 2015-01-12 13:03:40 -0800 | [diff] [blame] | 15 | class ScriptExecutable; |
Yang Ni | 1ffd86b | 2015-01-07 09:16:40 -0800 | [diff] [blame] | 16 | class ScriptGroup2; |
| 17 | |
David Gross | b0abb14 | 2015-03-12 15:23:03 -0700 | [diff] [blame] | 18 | typedef void (*ExpandFuncTy)(const RsExpandKernelDriverInfo*, uint32_t, uint32_t, |
Yang Ni | 1ffd86b | 2015-01-07 09:16:40 -0800 | [diff] [blame] | 19 | uint32_t); |
Yang Ni | eb9aa67 | 2015-01-27 14:32:25 -0800 | [diff] [blame] | 20 | typedef void (*InvokeFuncTy)(const void*, uint32_t); |
Yang Ni | 1ffd86b | 2015-01-07 09:16:40 -0800 | [diff] [blame] | 21 | |
| 22 | class CPUClosure { |
Yang Ni | eb9aa67 | 2015-01-27 14:32:25 -0800 | [diff] [blame] | 23 | public: |
Yang Ni | 062c287 | 2015-02-20 15:20:00 -0800 | [diff] [blame] | 24 | CPUClosure(const Closure* closure, RsdCpuScriptImpl* si, ExpandFuncTy func) : |
| 25 | mClosure(closure), mSi(si), mFunc(func) {} |
Yang Ni | 1ffd86b | 2015-01-07 09:16:40 -0800 | [diff] [blame] | 26 | |
Yang Ni | eb9aa67 | 2015-01-27 14:32:25 -0800 | [diff] [blame] | 27 | CPUClosure(const Closure* closure, RsdCpuScriptImpl* si) : |
Yang Ni | 062c287 | 2015-02-20 15:20:00 -0800 | [diff] [blame] | 28 | mClosure(closure), mSi(si), mFunc(nullptr) {} |
Yang Ni | eb9aa67 | 2015-01-27 14:32:25 -0800 | [diff] [blame] | 29 | |
| 30 | // It's important to do forwarding here than inheritance for unbound value |
| 31 | // binding to work. |
| 32 | const Closure* mClosure; |
| 33 | RsdCpuScriptImpl* mSi; |
| 34 | const ExpandFuncTy mFunc; |
Yang Ni | 1ffd86b | 2015-01-07 09:16:40 -0800 | [diff] [blame] | 35 | }; |
| 36 | |
Yang Ni | da0f069 | 2015-01-12 13:03:40 -0800 | [diff] [blame] | 37 | class CpuScriptGroup2Impl; |
| 38 | |
| 39 | class Batch { |
Yang Ni | eb9aa67 | 2015-01-27 14:32:25 -0800 | [diff] [blame] | 40 | public: |
Yang Ni | 062c287 | 2015-02-20 15:20:00 -0800 | [diff] [blame] | 41 | Batch(CpuScriptGroup2Impl* group, const char* name); |
Yang Ni | eb9aa67 | 2015-01-27 14:32:25 -0800 | [diff] [blame] | 42 | ~Batch(); |
Yang Ni | da0f069 | 2015-01-12 13:03:40 -0800 | [diff] [blame] | 43 | |
Yang Ni | eb9aa67 | 2015-01-27 14:32:25 -0800 | [diff] [blame] | 44 | // Returns true if closure depends on any closure in this batch for a global |
| 45 | // variable |
| 46 | bool conflict(CPUClosure* closure) const; |
Yang Ni | da0f069 | 2015-01-12 13:03:40 -0800 | [diff] [blame] | 47 | |
Yang Ni | 062c287 | 2015-02-20 15:20:00 -0800 | [diff] [blame] | 48 | void resolveFuncPtr(void* sharedObj); |
Yang Ni | eb9aa67 | 2015-01-27 14:32:25 -0800 | [diff] [blame] | 49 | void setGlobalsForBatch(); |
| 50 | void run(); |
Yang Ni | da0f069 | 2015-01-12 13:03:40 -0800 | [diff] [blame] | 51 | |
Yang Ni | 062c287 | 2015-02-20 15:20:00 -0800 | [diff] [blame] | 52 | size_t size() const { return mClosures.size(); } |
| 53 | |
Yang Ni | eb9aa67 | 2015-01-27 14:32:25 -0800 | [diff] [blame] | 54 | CpuScriptGroup2Impl* mGroup; |
Yang Ni | ff2bb54 | 2015-02-02 14:33:47 -0800 | [diff] [blame] | 55 | List<CPUClosure*> mClosures; |
Yang Ni | 062c287 | 2015-02-20 15:20:00 -0800 | [diff] [blame] | 56 | char* mName; |
| 57 | void* mFunc; |
Yang Ni | da0f069 | 2015-01-12 13:03:40 -0800 | [diff] [blame] | 58 | }; |
| 59 | |
Yang Ni | 1ffd86b | 2015-01-07 09:16:40 -0800 | [diff] [blame] | 60 | class CpuScriptGroup2Impl : public RsdCpuReference::CpuScriptGroup2 { |
Yang Ni | eb9aa67 | 2015-01-27 14:32:25 -0800 | [diff] [blame] | 61 | public: |
| 62 | CpuScriptGroup2Impl(RsdCpuReferenceImpl *cpuRefImpl, const ScriptGroupBase* group); |
Stephen Hines | c060f14 | 2015-05-13 19:26:09 -0700 | [diff] [blame] | 63 | ~CpuScriptGroup2Impl() override; |
Yang Ni | 1ffd86b | 2015-01-07 09:16:40 -0800 | [diff] [blame] | 64 | |
Yang Ni | eb9aa67 | 2015-01-27 14:32:25 -0800 | [diff] [blame] | 65 | bool init(); |
Stephen Hines | c060f14 | 2015-05-13 19:26:09 -0700 | [diff] [blame] | 66 | void execute() override; |
Yang Ni | 1ffd86b | 2015-01-07 09:16:40 -0800 | [diff] [blame] | 67 | |
Yang Ni | eb9aa67 | 2015-01-27 14:32:25 -0800 | [diff] [blame] | 68 | RsdCpuReferenceImpl* getCpuRefImpl() const { return mCpuRefImpl; } |
Yang Ni | 062c287 | 2015-02-20 15:20:00 -0800 | [diff] [blame] | 69 | ScriptExecutable* getExecutable() const { return mExecutable; } |
| 70 | |
| 71 | void compile(const char* cacheDir); |
Yang Ni | 1ffd86b | 2015-01-07 09:16:40 -0800 | [diff] [blame] | 72 | |
Yang Ni | eb9aa67 | 2015-01-27 14:32:25 -0800 | [diff] [blame] | 73 | private: |
| 74 | RsdCpuReferenceImpl* mCpuRefImpl; |
| 75 | const ScriptGroup2* mGroup; |
Yang Ni | ff2bb54 | 2015-02-02 14:33:47 -0800 | [diff] [blame] | 76 | List<Batch*> mBatches; |
Yang Ni | 062c287 | 2015-02-20 15:20:00 -0800 | [diff] [blame] | 77 | ScriptExecutable* mExecutable; |
| 78 | void* mScriptObj; |
Yang Ni | 1ffd86b | 2015-01-07 09:16:40 -0800 | [diff] [blame] | 79 | }; |
| 80 | |
| 81 | } // namespace renderscript |
| 82 | } // namespace android |
| 83 | |
| 84 | #endif // CPU_REF_CPUSCRIPTGROUP2IMPL_H_ |