rsd_cpu.h revision 1ffd86b448d78366190c540f98f8b6d641cdb6cf
1709a0978ae141198018ca9769f8d96292a8928e6Jason Sams/* 2709a0978ae141198018ca9769f8d96292a8928e6Jason Sams * Copyright (C) 2012 The Android Open Source Project 3709a0978ae141198018ca9769f8d96292a8928e6Jason Sams * 4709a0978ae141198018ca9769f8d96292a8928e6Jason Sams * Licensed under the Apache License, Version 2.0 (the "License"); 5709a0978ae141198018ca9769f8d96292a8928e6Jason Sams * you may not use this file except in compliance with the License. 6709a0978ae141198018ca9769f8d96292a8928e6Jason Sams * You may obtain a copy of the License at 7709a0978ae141198018ca9769f8d96292a8928e6Jason Sams * 8709a0978ae141198018ca9769f8d96292a8928e6Jason Sams * http://www.apache.org/licenses/LICENSE-2.0 9709a0978ae141198018ca9769f8d96292a8928e6Jason Sams * 10709a0978ae141198018ca9769f8d96292a8928e6Jason Sams * Unless required by applicable law or agreed to in writing, software 11709a0978ae141198018ca9769f8d96292a8928e6Jason Sams * distributed under the License is distributed on an "AS IS" BASIS, 12709a0978ae141198018ca9769f8d96292a8928e6Jason Sams * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13709a0978ae141198018ca9769f8d96292a8928e6Jason Sams * See the License for the specific language governing permissions and 14709a0978ae141198018ca9769f8d96292a8928e6Jason Sams * limitations under the License. 15709a0978ae141198018ca9769f8d96292a8928e6Jason Sams */ 16709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 17709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#ifndef RSD_CPU_H 18709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#define RSD_CPU_H 19709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 20709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#include "rsAllocation.h" 21709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 22110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams#ifndef RS_COMPATIBILITY_LIB 23f218bf115af4ae4fd79adbb8842608b308a4cf07Stephen Hinesnamespace llvm { 24f218bf115af4ae4fd79adbb8842608b308a4cf07Stephen Hines 25f218bf115af4ae4fd79adbb8842608b308a4cf07Stephen Hinesclass Module; 26f218bf115af4ae4fd79adbb8842608b308a4cf07Stephen Hines 27f218bf115af4ae4fd79adbb8842608b308a4cf07Stephen Hines} // end namespace llvm 28f218bf115af4ae4fd79adbb8842608b308a4cf07Stephen Hines 29f218bf115af4ae4fd79adbb8842608b308a4cf07Stephen Hinesnamespace bcc { 30f218bf115af4ae4fd79adbb8842608b308a4cf07Stephen Hines 31b7d9c80c98fc96aa7c638e3124be24f13a6436b2Stephen Hinesclass RSCompilerDriver; 32f218bf115af4ae4fd79adbb8842608b308a4cf07Stephen Hinesclass RSScript; 33b7d9c80c98fc96aa7c638e3124be24f13a6436b2Stephen Hinestypedef llvm::Module* (*RSLinkRuntimeCallback) 34b7d9c80c98fc96aa7c638e3124be24f13a6436b2Stephen Hines (bcc::RSScript *, llvm::Module *, llvm::Module *); 35f218bf115af4ae4fd79adbb8842608b308a4cf07Stephen Hines 36f218bf115af4ae4fd79adbb8842608b308a4cf07Stephen Hines} // end namespace bcc; 371d476620399d54774e4fd386c1d23cc583d49522Stephen Hines 381d476620399d54774e4fd386c1d23cc583d49522Stephen Hinestypedef const char* (*RSSelectRTCallback) (const char*, size_t); 39b7d9c80c98fc96aa7c638e3124be24f13a6436b2Stephen Hines 40b7d9c80c98fc96aa7c638e3124be24f13a6436b2Stephen Hinestypedef void (*RSSetupCompilerCallback) (bcc::RSCompilerDriver *); 41110f181b7966212a36ef18016f9b81c7322d0a2fJason Sams#endif 42709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 43709a0978ae141198018ca9769f8d96292a8928e6Jason Samsnamespace android { 44709a0978ae141198018ca9769f8d96292a8928e6Jason Samsnamespace renderscript { 45709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 46709a0978ae141198018ca9769f8d96292a8928e6Jason Samsclass ScriptC; 47709a0978ae141198018ca9769f8d96292a8928e6Jason Samsclass Script; 481ffd86b448d78366190c540f98f8b6d641cdb6cfYang Niclass ScriptGroupBase; 49709a0978ae141198018ca9769f8d96292a8928e6Jason Samsclass ScriptKernelID; 50709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 51709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 52709a0978ae141198018ca9769f8d96292a8928e6Jason Samsclass RsdCpuReference { 53709a0978ae141198018ca9769f8d96292a8928e6Jason Samspublic: 54709a0978ae141198018ca9769f8d96292a8928e6Jason Sams struct CpuSymbol { 55709a0978ae141198018ca9769f8d96292a8928e6Jason Sams const char * name; 56709a0978ae141198018ca9769f8d96292a8928e6Jason Sams void * fnPtr; 57709a0978ae141198018ca9769f8d96292a8928e6Jason Sams bool threadable; 58709a0978ae141198018ca9769f8d96292a8928e6Jason Sams }; 59709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 60709a0978ae141198018ca9769f8d96292a8928e6Jason Sams typedef const CpuSymbol * (* sym_lookup_t)(Context *, const char *name); 61709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 62709a0978ae141198018ca9769f8d96292a8928e6Jason Sams struct CpuTls { 63709a0978ae141198018ca9769f8d96292a8928e6Jason Sams Context *rsc; 64709a0978ae141198018ca9769f8d96292a8928e6Jason Sams const ScriptC * sc; 65709a0978ae141198018ca9769f8d96292a8928e6Jason Sams }; 66709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 67709a0978ae141198018ca9769f8d96292a8928e6Jason Sams class CpuScript { 68709a0978ae141198018ca9769f8d96292a8928e6Jason Sams public: 69709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void populateScript(Script *) = 0; 70709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void invokeFunction(uint32_t slot, const void *params, size_t paramLength) = 0; 71709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual int invokeRoot() = 0; 72f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes 73709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void invokeForEach(uint32_t slot, 74f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes const Allocation ** ains, 75f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes uint32_t inLen, 76f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes Allocation * aout, 77f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes const void * usr, 78f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes uint32_t usrLen, 79f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes const RsScriptCall *sc) = 0; 80f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes 81709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void invokeInit() = 0; 82709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void invokeFreeChildren() = 0; 83709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 84709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void setGlobalVar(uint32_t slot, const void *data, size_t dataLength) = 0; 859c64239ebbfa4170190ede812e69150035e008e0Tim Murray virtual void getGlobalVar(uint32_t slot, void *data, size_t dataLength) = 0; 86709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void setGlobalVarWithElemDims(uint32_t slot, const void *data, size_t dataLength, 87ac8d146a41f18afad5314ac8af440d6aedbe20bfStephen Hines const Element *e, const uint32_t *dims, size_t dimLength) = 0; 88709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void setGlobalBind(uint32_t slot, Allocation *data) = 0; 89709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void setGlobalObj(uint32_t slot, ObjectBase *obj) = 0; 90709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 91709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual Allocation * getAllocationForPointer(const void *ptr) const = 0; 92709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual ~CpuScript() {} 93f218bf115af4ae4fd79adbb8842608b308a4cf07Stephen Hines 94cadfac411e6690e39de36c4f9e94deb9b7d2d08eJason Sams#ifndef RS_COMPATIBILITY_LIB 95f218bf115af4ae4fd79adbb8842608b308a4cf07Stephen Hines virtual void * getRSExecutable() = 0; 96cadfac411e6690e39de36c4f9e94deb9b7d2d08eJason Sams#endif 97709a0978ae141198018ca9769f8d96292a8928e6Jason Sams }; 98709a0978ae141198018ca9769f8d96292a8928e6Jason Sams typedef CpuScript * (* script_lookup_t)(Context *, const Script *s); 99709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 1001ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni class CpuScriptGroupBase { 1011ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni public: 1021ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni virtual void execute() = 0; 1031ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni virtual ~CpuScriptGroupBase() {} 1041ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni }; 1051ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni 1061ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni class CpuScriptGroup : public CpuScriptGroupBase { 107709a0978ae141198018ca9769f8d96292a8928e6Jason Sams public: 108709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void setInput(const ScriptKernelID *kid, Allocation *) = 0; 109709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void setOutput(const ScriptKernelID *kid, Allocation *) = 0; 110709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void execute() = 0; 111709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual ~CpuScriptGroup() {}; 112709a0978ae141198018ca9769f8d96292a8928e6Jason Sams }; 113709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 1141ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni class CpuScriptGroup2 : public CpuScriptGroupBase { 1151ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni public: 1161ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni virtual void execute() = 0; 1171ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni virtual ~CpuScriptGroup2() {} 1181ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni }; 1191ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni 120709a0978ae141198018ca9769f8d96292a8928e6Jason Sams static Context * getTlsContext(); 121709a0978ae141198018ca9769f8d96292a8928e6Jason Sams static const Script * getTlsScript(); 122f218bf115af4ae4fd79adbb8842608b308a4cf07Stephen Hines static pthread_key_t getThreadTLSKey(); 123709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 124709a0978ae141198018ca9769f8d96292a8928e6Jason Sams static RsdCpuReference * create(Context *c, uint32_t version_major, 125cadfac411e6690e39de36c4f9e94deb9b7d2d08eJason Sams uint32_t version_minor, sym_lookup_t lfn, script_lookup_t slfn 126cadfac411e6690e39de36c4f9e94deb9b7d2d08eJason Sams#ifndef RS_COMPATIBILITY_LIB 12744bef6fba6244292b751387f3d6c31cca96c28adChris Wailes , bcc::RSLinkRuntimeCallback pLinkRuntimeCallback = nullptr, 12844bef6fba6244292b751387f3d6c31cca96c28adChris Wailes RSSelectRTCallback pSelectRTCallback = nullptr, 12944bef6fba6244292b751387f3d6c31cca96c28adChris Wailes const char *pBccPluginName = nullptr 130cadfac411e6690e39de36c4f9e94deb9b7d2d08eJason Sams#endif 131cadfac411e6690e39de36c4f9e94deb9b7d2d08eJason Sams ); 132709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual ~RsdCpuReference(); 133709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void setPriority(int32_t priority) = 0; 134709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 135709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual CpuScript * createScript(const ScriptC *s, char const *resName, char const *cacheDir, 136709a0978ae141198018ca9769f8d96292a8928e6Jason Sams uint8_t const *bitcode, size_t bitcodeSize, 137709a0978ae141198018ca9769f8d96292a8928e6Jason Sams uint32_t flags) = 0; 138709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual CpuScript * createIntrinsic(const Script *s, RsScriptIntrinsicID iid, Element *e) = 0; 1391ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni virtual void* createScriptGroup(const ScriptGroupBase *sg) = 0; 140f218bf115af4ae4fd79adbb8842608b308a4cf07Stephen Hines virtual bool getInForEach() = 0; 141f218bf115af4ae4fd79adbb8842608b308a4cf07Stephen Hines 142b7d9c80c98fc96aa7c638e3124be24f13a6436b2Stephen Hines#ifndef RS_COMPATIBILITY_LIB 143b7d9c80c98fc96aa7c638e3124be24f13a6436b2Stephen Hines virtual void setSetupCompilerCallback( 144b7d9c80c98fc96aa7c638e3124be24f13a6436b2Stephen Hines RSSetupCompilerCallback pSetupCompilerCallback) = 0; 145b7d9c80c98fc96aa7c638e3124be24f13a6436b2Stephen Hines virtual RSSetupCompilerCallback getSetupCompilerCallback() const = 0; 146b7d9c80c98fc96aa7c638e3124be24f13a6436b2Stephen Hines#endif 147709a0978ae141198018ca9769f8d96292a8928e6Jason Sams}; 148709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 149709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 150709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 151709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 152709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 153709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#endif 154