rsd_cpu.h revision 10adb0c2029f112b5738228617d5645f6ecea0c5
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, 6214ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Wala const Allocation *ain, 6314ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Wala Allocation *aout, 6414ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Wala const RsScriptCall *sc) = 0; 6514ce007a633b10e3b9a3fae29d8f53a7e8c9b59fMatt Wala 666c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross virtual void invokeReduceNew(uint32_t slot, 676c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross const Allocation ** ains, uint32_t inLen, 686c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross Allocation *aout, 696c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross const RsScriptCall *sc) = 0; 706c1876bbef1b2c89975dce91230a168bd2d2ce4cDavid Gross 71709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void invokeInit() = 0; 72709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void invokeFreeChildren() = 0; 73709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 74709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void setGlobalVar(uint32_t slot, const void *data, size_t dataLength) = 0; 759c64239ebbfa4170190ede812e69150035e008e0Tim Murray virtual void getGlobalVar(uint32_t slot, void *data, size_t dataLength) = 0; 76709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void setGlobalVarWithElemDims(uint32_t slot, const void *data, size_t dataLength, 77ac8d146a41f18afad5314ac8af440d6aedbe20bfStephen Hines const Element *e, const uint32_t *dims, size_t dimLength) = 0; 78709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void setGlobalBind(uint32_t slot, Allocation *data) = 0; 79709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void setGlobalObj(uint32_t slot, ObjectBase *obj) = 0; 80709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 81709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual Allocation * getAllocationForPointer(const void *ptr) const = 0; 828409d6414dd4a42aa59779fcfe9fce18648cb135Stephen Hines 838409d6414dd4a42aa59779fcfe9fce18648cb135Stephen Hines // Returns number of global variables in this Script (may be 0 if 848409d6414dd4a42aa59779fcfe9fce18648cb135Stephen Hines // compiler is not configured to emit this information). 858409d6414dd4a42aa59779fcfe9fce18648cb135Stephen Hines virtual int getGlobalEntries() const = 0; 868409d6414dd4a42aa59779fcfe9fce18648cb135Stephen Hines // Returns the name of the global variable at index i. 878409d6414dd4a42aa59779fcfe9fce18648cb135Stephen Hines virtual const char * getGlobalName(int i) const = 0; 888409d6414dd4a42aa59779fcfe9fce18648cb135Stephen Hines // Returns the CPU address of the global variable at index i. 898409d6414dd4a42aa59779fcfe9fce18648cb135Stephen Hines virtual const void * getGlobalAddress(int i) const = 0; 908409d6414dd4a42aa59779fcfe9fce18648cb135Stephen Hines // Returns the size (in bytes) of the global variable at index i. 918409d6414dd4a42aa59779fcfe9fce18648cb135Stephen Hines virtual size_t getGlobalSize(int i) const = 0; 925aa018cc36e589b07674957714d27ae3d1fa1c4eStephen Hines // Returns the properties of the global variable at index i. 935aa018cc36e589b07674957714d27ae3d1fa1c4eStephen Hines virtual uint32_t getGlobalProperties(int i) const = 0; 948409d6414dd4a42aa59779fcfe9fce18648cb135Stephen Hines 95709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual ~CpuScript() {} 96709a0978ae141198018ca9769f8d96292a8928e6Jason Sams }; 97709a0978ae141198018ca9769f8d96292a8928e6Jason Sams typedef CpuScript * (* script_lookup_t)(Context *, const Script *s); 98709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 991ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni class CpuScriptGroupBase { 1001ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni public: 1011ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni virtual void execute() = 0; 1021ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni virtual ~CpuScriptGroupBase() {} 1031ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni }; 1041ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni 1051ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni class CpuScriptGroup : public CpuScriptGroupBase { 106709a0978ae141198018ca9769f8d96292a8928e6Jason Sams public: 107709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void setInput(const ScriptKernelID *kid, Allocation *) = 0; 108709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void setOutput(const ScriptKernelID *kid, Allocation *) = 0; 109c060f1435e7b9405f3be8974417fa6f410f03753Stephen Hines ~CpuScriptGroup() override {}; 110709a0978ae141198018ca9769f8d96292a8928e6Jason Sams }; 111709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 1121ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni class CpuScriptGroup2 : public CpuScriptGroupBase { 1131ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni public: 114c060f1435e7b9405f3be8974417fa6f410f03753Stephen Hines ~CpuScriptGroup2() override {} 1151ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni }; 1161ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni 117709a0978ae141198018ca9769f8d96292a8928e6Jason Sams static Context * getTlsContext(); 118709a0978ae141198018ca9769f8d96292a8928e6Jason Sams static const Script * getTlsScript(); 119f218bf115af4ae4fd79adbb8842608b308a4cf07Stephen Hines static pthread_key_t getThreadTLSKey(); 120709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 121709a0978ae141198018ca9769f8d96292a8928e6Jason Sams static RsdCpuReference * create(Context *c, uint32_t version_major, 122cadfac411e6690e39de36c4f9e94deb9b7d2d08eJason Sams uint32_t version_minor, sym_lookup_t lfn, script_lookup_t slfn 123b043df0676fef226336deb3a00ead2f31e02343fDavid Gross , RSSelectRTCallback pSelectRTCallback = nullptr, 12444bef6fba6244292b751387f3d6c31cca96c28adChris Wailes const char *pBccPluginName = nullptr 125cadfac411e6690e39de36c4f9e94deb9b7d2d08eJason Sams ); 126709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual ~RsdCpuReference(); 127709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual void setPriority(int32_t priority) = 0; 128709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 129709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual CpuScript * createScript(const ScriptC *s, char const *resName, char const *cacheDir, 130709a0978ae141198018ca9769f8d96292a8928e6Jason Sams uint8_t const *bitcode, size_t bitcodeSize, 131709a0978ae141198018ca9769f8d96292a8928e6Jason Sams uint32_t flags) = 0; 132709a0978ae141198018ca9769f8d96292a8928e6Jason Sams virtual CpuScript * createIntrinsic(const Script *s, RsScriptIntrinsicID iid, Element *e) = 0; 1331ffd86b448d78366190c540f98f8b6d641cdb6cfYang Ni virtual void* createScriptGroup(const ScriptGroupBase *sg) = 0; 13410adb0c2029f112b5738228617d5645f6ecea0c5David Gross virtual bool getInKernel() = 0; // Is a parallel kernel execution underway? 135f218bf115af4ae4fd79adbb8842608b308a4cf07Stephen Hines 1365cb19fd40ad3225005f65b6960c875746c2136b9Pirama Arumuga Nainar // Set to true if we should embed global variable information in the code. 1375cb19fd40ad3225005f65b6960c875746c2136b9Pirama Arumuga Nainar virtual void setEmbedGlobalInfo(bool v) = 0; 1385cb19fd40ad3225005f65b6960c875746c2136b9Pirama Arumuga Nainar 1395cb19fd40ad3225005f65b6960c875746c2136b9Pirama Arumuga Nainar // Returns true if we should embed global variable information in the code. 1405cb19fd40ad3225005f65b6960c875746c2136b9Pirama Arumuga Nainar virtual bool getEmbedGlobalInfo() const = 0; 1415cb19fd40ad3225005f65b6960c875746c2136b9Pirama Arumuga Nainar 1425cb19fd40ad3225005f65b6960c875746c2136b9Pirama Arumuga Nainar // Set to true if we should skip constant (immutable) global variables when 1435cb19fd40ad3225005f65b6960c875746c2136b9Pirama Arumuga Nainar // potentially embedding information about globals. 1445cb19fd40ad3225005f65b6960c875746c2136b9Pirama Arumuga Nainar virtual void setEmbedGlobalInfoSkipConstant(bool v) = 0; 1455cb19fd40ad3225005f65b6960c875746c2136b9Pirama Arumuga Nainar 1465cb19fd40ad3225005f65b6960c875746c2136b9Pirama Arumuga Nainar // Returns true if we should skip constant (immutable) global variables when 1475cb19fd40ad3225005f65b6960c875746c2136b9Pirama Arumuga Nainar // potentially embedding information about globals. 1485cb19fd40ad3225005f65b6960c875746c2136b9Pirama Arumuga Nainar virtual bool getEmbedGlobalInfoSkipConstant() const = 0; 149709a0978ae141198018ca9769f8d96292a8928e6Jason Sams}; 150709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 151709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 152709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 153709a0978ae141198018ca9769f8d96292a8928e6Jason Sams} 154709a0978ae141198018ca9769f8d96292a8928e6Jason Sams 155709a0978ae141198018ca9769f8d96292a8928e6Jason Sams#endif 156