rsd_cpu.h revision ae2ec3febedfc29376b9104413fb4042028f1265
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 221d476620399d54774e4fd386c1d23cc583d49522Stephen Hinestypedef const char* (*RSSelectRTCallback) (const char*, size_t); 23b7d9c80c98fc96aa7c638e3124be24f13a6436b2Stephen Hines 24709a0978ae141198018ca9769f8d96292a8928e6Jason Samsnamespace android { 25709a0978ae141198018ca9769f8d96292a8928e6Jason Samsnamespace renderscript { 26709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 27709a0978ae141198018ca9769f8d96292a8928e6Jason Samsclass ScriptC; 28709a0978ae141198018ca9769f8d96292a8928e6Jason Samsclass Script; 291ffd86b448d78366190c540f98f8b6d641cdb6cfYang Niclass ScriptGroupBase; 30709a0978ae141198018ca9769f8d96292a8928e6Jason Samsclass ScriptKernelID; 31709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 32709a0978ae141198018ca9769f8d96292a8928e6Jason Samsclass RsdCpuReference { 33709a0978ae141198018ca9769f8d96292a8928e6Jason Samspublic: 34709a0978ae141198018ca9769f8d96292a8928e6Jason Sams struct CpuSymbol { 35709a0978ae141198018ca9769f8d96292a8928e6Jason Sams const char * name; 36709a0978ae141198018ca9769f8d96292a8928e6Jason Sams void * fnPtr; 37709a0978ae141198018ca9769f8d96292a8928e6Jason Sams bool threadable; 38709a0978ae141198018ca9769f8d96292a8928e6Jason Sams }; 39709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 40709a0978ae141198018ca9769f8d96292a8928e6Jason Sams typedef const CpuSymbol * (* sym_lookup_t)(Context *, const char *name); 41709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 42709a0978ae141198018ca9769f8d96292a8928e6Jason Sams struct CpuTls { 43709a0978ae141198018ca9769f8d96292a8928e6Jason Sams Context *rsc; 44709a0978ae141198018ca9769f8d96292a8928e6Jason Sams const ScriptC * sc; 45709a0978ae141198018ca9769f8d96292a8928e6Jason Sams }; 46709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 47709a0978ae141198018ca9769f8d96292a8928e6Jason Sams class CpuScript { 48709a0978ae141198018ca9769f8d96292a8928e6Jason Sams public: 49709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void populateScript(Script *) = 0; 50709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void invokeFunction(uint32_t slot, const void *params, size_t paramLength) = 0; 51709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual int invokeRoot() = 0; 52f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes 53709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void invokeForEach(uint32_t slot, 54f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes const Allocation ** ains, 55f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes uint32_t inLen, 56f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes Allocation * aout, 57f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes const void * usr, 58f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes uint32_t usrLen, 59f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes const RsScriptCall *sc) = 0; 60f37121300217d3b39ab66dd9c8881bcbcad932dfChris Wailes 6114ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Wala virtual void invokeReduce(uint32_t slot, 62ae2ec3febedfc29376b9104413fb4042028f1265David Gross const Allocation ** ains, uint32_t inLen, 6314ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Wala Allocation *aout, 6414ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Wala const RsScriptCall *sc) = 0; 6514ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Wala 66709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void invokeInit() = 0; 67709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void invokeFreeChildren() = 0; 68709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 69709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void setGlobalVar(uint32_t slot, const void *data, size_t dataLength) = 0; 709c64239ebbfa4170190ede812e69150035e008e0Tim Murray virtual void getGlobalVar(uint32_t slot, void *data, size_t dataLength) = 0; 71709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void setGlobalVarWithElemDims(uint32_t slot, const void *data, size_t dataLength, 72ac8d146a41f18afad5314ac8af440d6aedbe20bfStephen Hines const Element *e, const uint32_t *dims, size_t dimLength) = 0; 73709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void setGlobalBind(uint32_t slot, Allocation *data) = 0; 74709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void setGlobalObj(uint32_t slot, ObjectBase *obj) = 0; 75709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 76709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual Allocation * getAllocationForPointer(const void *ptr) const = 0; 778409d6414dd4a42aa59779fcfe9fce18648cb135Stephen Hines 788409d6414dd4a42aa59779fcfe9fce18648cb135Stephen Hines // Returns number of global variables in this Script (may be 0 if 798409d6414dd4a42aa59779fcfe9fce18648cb135Stephen Hines // compiler is not configured to emit this information). 808409d6414dd4a42aa59779fcfe9fce18648cb135Stephen Hines virtual int getGlobalEntries() const = 0; 818409d6414dd4a42aa59779fcfe9fce18648cb135Stephen Hines // Returns the name of the global variable at index i. 828409d6414dd4a42aa59779fcfe9fce18648cb135Stephen Hines virtual const char * getGlobalName(int i) const = 0; 838409d6414dd4a42aa59779fcfe9fce18648cb135Stephen Hines // Returns the CPU address of the global variable at index i. 848409d6414dd4a42aa59779fcfe9fce18648cb135Stephen Hines virtual const void * getGlobalAddress(int i) const = 0; 858409d6414dd4a42aa59779fcfe9fce18648cb135Stephen Hines // Returns the size (in bytes) of the global variable at index i. 868409d6414dd4a42aa59779fcfe9fce18648cb135Stephen Hines virtual size_t getGlobalSize(int i) const = 0; 875aa018cc36e589b07674957714d27ae3d1fa1c4eStephen Hines // Returns the properties of the global variable at index i. 885aa018cc36e589b07674957714d27ae3d1fa1c4eStephen Hines virtual uint32_t getGlobalProperties(int i) const = 0; 898409d6414dd4a42aa59779fcfe9fce18648cb135Stephen Hines 90709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual ~CpuScript() {} 91709a0978ae141198018ca9769f8d96292a8928e6Jason Sams }; 92709a0978ae141198018ca9769f8d96292a8928e6Jason Sams typedef CpuScript * (* script_lookup_t)(Context *, const Script *s); 93709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 941ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni class CpuScriptGroupBase { 951ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni public: 961ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni virtual void execute() = 0; 971ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni virtual ~CpuScriptGroupBase() {} 981ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni }; 991ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni 1001ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni class CpuScriptGroup : public CpuScriptGroupBase { 101709a0978ae141198018ca9769f8d96292a8928e6Jason Sams public: 102709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void setInput(const ScriptKernelID *kid, Allocation *) = 0; 103709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void setOutput(const ScriptKernelID *kid, Allocation *) = 0; 104c060f1435e7b9405f3be8974417fa6f410f03753Stephen Hines ~CpuScriptGroup() override {}; 105709a0978ae141198018ca9769f8d96292a8928e6Jason Sams }; 106709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 1071ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni class CpuScriptGroup2 : public CpuScriptGroupBase { 1081ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni public: 109c060f1435e7b9405f3be8974417fa6f410f03753Stephen Hines ~CpuScriptGroup2() override {} 1101ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni }; 1111ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni 112709a0978ae141198018ca9769f8d96292a8928e6Jason Sams static Context * getTlsContext(); 113709a0978ae141198018ca9769f8d96292a8928e6Jason Sams static const Script * getTlsScript(); 114f218bf115af4ae4fd79adbb8842608b308a4cf07Stephen Hines static pthread_key_t getThreadTLSKey(); 115709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 116709a0978ae141198018ca9769f8d96292a8928e6Jason Sams static RsdCpuReference * create(Context *c, uint32_t version_major, 117cadfac411e6690e39de36c4f9e94deb9b7d2d08eJason Sams uint32_t version_minor, sym_lookup_t lfn, script_lookup_t slfn 118b043df0676fef226336deb3a00ead2f31e02343fDavid Gross , RSSelectRTCallback pSelectRTCallback = nullptr, 11944bef6fba6244292b751387f3d6c31cca96c28adChris Wailes const char *pBccPluginName = nullptr 120cadfac411e6690e39de36c4f9e94deb9b7d2d08eJason Sams ); 121709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual ~RsdCpuReference(); 122709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void setPriority(int32_t priority) = 0; 123709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 124709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual CpuScript * createScript(const ScriptC *s, char const *resName, char const *cacheDir, 125709a0978ae141198018ca9769f8d96292a8928e6Jason Sams uint8_t const *bitcode, size_t bitcodeSize, 126709a0978ae141198018ca9769f8d96292a8928e6Jason Sams uint32_t flags) = 0; 127709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual CpuScript * createIntrinsic(const Script *s, RsScriptIntrinsicID iid, Element *e) = 0; 1281ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni virtual void* createScriptGroup(const ScriptGroupBase *sg) = 0; 12910adb0c2029f112b5738228617d5645f6ecea0c5David Gross virtual bool getInKernel() = 0; // Is a parallel kernel execution underway? 130f218bf115af4ae4fd79adbb8842608b308a4cf07Stephen Hines 1315cb19fd40ad3225005f65b6960c875746c2136b9Pirama Arumuga Nainar // Set to true if we should embed global variable information in the code. 1325cb19fd40ad3225005f65b6960c875746c2136b9Pirama Arumuga Nainar virtual void setEmbedGlobalInfo(bool v) = 0; 1335cb19fd40ad3225005f65b6960c875746c2136b9Pirama Arumuga Nainar 1345cb19fd40ad3225005f65b6960c875746c2136b9Pirama Arumuga Nainar // Returns true if we should embed global variable information in the code. 1355cb19fd40ad3225005f65b6960c875746c2136b9Pirama Arumuga Nainar virtual bool getEmbedGlobalInfo() const = 0; 1365cb19fd40ad3225005f65b6960c875746c2136b9Pirama Arumuga Nainar 1375cb19fd40ad3225005f65b6960c875746c2136b9Pirama Arumuga Nainar // Set to true if we should skip constant (immutable) global variables when 1385cb19fd40ad3225005f65b6960c875746c2136b9Pirama Arumuga Nainar // potentially embedding information about globals. 1395cb19fd40ad3225005f65b6960c875746c2136b9Pirama Arumuga Nainar virtual void setEmbedGlobalInfoSkipConstant(bool v) = 0; 1405cb19fd40ad3225005f65b6960c875746c2136b9Pirama Arumuga Nainar 1415cb19fd40ad3225005f65b6960c875746c2136b9Pirama Arumuga Nainar // Returns true if we should skip constant (immutable) global variables when 1425cb19fd40ad3225005f65b6960c875746c2136b9Pirama Arumuga Nainar // potentially embedding information about globals. 1435cb19fd40ad3225005f65b6960c875746c2136b9Pirama Arumuga Nainar virtual bool getEmbedGlobalInfoSkipConstant() const = 0; 144709a0978ae141198018ca9769f8d96292a8928e6Jason Sams}; 145709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 146709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 147709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 148709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 149709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 150709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#endif 151