Home | History | Annotate | Download | only in cpu_ref 
      1 /*
      2  * Copyright (C) 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_CPU_H
     18 #define RSD_CPU_H
     19 
     20 #include "rsAllocation.h"
     21 
     22 typedef const char* (*RSSelectRTCallback) (const char*, size_t);
     23 
     24 namespace android {
     25 namespace renderscript {
     26 
     27 class ScriptC;
     28 class Script;
     29 class ScriptGroupBase;
     30 class ScriptKernelID;
     31 
     32 class RsdCpuReference {
     33 public:
     34     struct CpuSymbol {
     35         const char * name;
     36         void * fnPtr;
     37         bool threadable;
     38     };
     39 
     40     typedef const CpuSymbol * (* sym_lookup_t)(Context *, const char *name);
     41 
     42     struct CpuTls {
     43         Context *rsc;
     44         const ScriptC * sc;
     45     };
     46 
     47     class CpuScript {
     48     public:
     49         virtual void populateScript(Script *) = 0;
     50         virtual void invokeFunction(uint32_t slot, const void *params, size_t paramLength) = 0;
     51         virtual int invokeRoot() = 0;
     52 
     53         virtual void invokeForEach(uint32_t slot,
     54                                    const Allocation ** ains,
     55                                    uint32_t inLen,
     56                                    Allocation * aout,
     57                                    const void * usr,
     58                                    uint32_t usrLen,
     59                                    const RsScriptCall *sc) = 0;
     60 
     61         virtual void invokeReduce(uint32_t slot,
     62                                   const Allocation ** ains, uint32_t inLen,
     63                                   Allocation *aout,
     64                                   const RsScriptCall *sc) = 0;
     65 
     66         virtual void invokeInit() = 0;
     67         virtual void invokeFreeChildren() = 0;
     68 
     69         virtual void setGlobalVar(uint32_t slot, const void *data, size_t dataLength) = 0;
     70         virtual void getGlobalVar(uint32_t slot, void *data, size_t dataLength) = 0;
     71         virtual void setGlobalVarWithElemDims(uint32_t slot, const void *data, size_t dataLength,
     72                                       const Element *e, const uint32_t *dims, size_t dimLength) = 0;
     73         virtual void setGlobalBind(uint32_t slot, Allocation *data) = 0;
     74         virtual void setGlobalObj(uint32_t slot, ObjectBase *obj) = 0;
     75 
     76         virtual Allocation * getAllocationForPointer(const void *ptr) const = 0;
     77 
     78         // Returns number of global variables in this Script (may be 0 if
     79         // compiler is not configured to emit this information).
     80         virtual int getGlobalEntries() const = 0;
     81         // Returns the name of the global variable at index i.
     82         virtual const char * getGlobalName(int i) const = 0;
     83         // Returns the CPU address of the global variable at index i.
     84         virtual const void * getGlobalAddress(int i) const = 0;
     85         // Returns the size (in bytes) of the global variable at index i.
     86         virtual size_t getGlobalSize(int i) const = 0;
     87         // Returns the properties of the global variable at index i.
     88         virtual uint32_t getGlobalProperties(int i) const = 0;
     89 
     90         virtual ~CpuScript() {}
     91     };
     92     typedef CpuScript * (* script_lookup_t)(Context *, const Script *s);
     93 
     94     class CpuScriptGroupBase {
     95      public:
     96       virtual void execute() = 0;
     97       virtual ~CpuScriptGroupBase() {}
     98     };
     99 
    100     class CpuScriptGroup : public CpuScriptGroupBase {
    101     public:
    102         virtual void setInput(const ScriptKernelID *kid, Allocation *) = 0;
    103         virtual void setOutput(const ScriptKernelID *kid, Allocation *) = 0;
    104         ~CpuScriptGroup() override {};
    105     };
    106 
    107     class CpuScriptGroup2 : public CpuScriptGroupBase {
    108      public:
    109       ~CpuScriptGroup2() override {}
    110     };
    111 
    112     static Context * getTlsContext();
    113     static const Script * getTlsScript();
    114     static pthread_key_t getThreadTLSKey();
    115 
    116     static RsdCpuReference * create(Context *c, uint32_t version_major,
    117                                     uint32_t version_minor, sym_lookup_t lfn, script_lookup_t slfn
    118                                     , RSSelectRTCallback pSelectRTCallback = nullptr,
    119                                     const char *pBccPluginName = nullptr
    120                                     );
    121     virtual ~RsdCpuReference();
    122     virtual void setPriority(int32_t priority) = 0;
    123 
    124     virtual CpuScript * createScript(const ScriptC *s, char const *resName, char const *cacheDir,
    125                                      uint8_t const *bitcode, size_t bitcodeSize,
    126                                      uint32_t flags) = 0;
    127     virtual CpuScript * createIntrinsic(const Script *s, RsScriptIntrinsicID iid, Element *e) = 0;
    128     virtual void* createScriptGroup(const ScriptGroupBase *sg) = 0;
    129     virtual bool getInKernel() = 0;  // Is a parallel kernel execution underway?
    130 
    131     // Set to true if we should embed global variable information in the code.
    132     virtual void setEmbedGlobalInfo(bool v) = 0;
    133 
    134     // Returns true if we should embed global variable information in the code.
    135     virtual bool getEmbedGlobalInfo() const = 0;
    136 
    137     // Set to true if we should skip constant (immutable) global variables when
    138     // potentially embedding information about globals.
    139     virtual void setEmbedGlobalInfoSkipConstant(bool v) = 0;
    140 
    141     // Returns true if we should skip constant (immutable) global variables when
    142     // potentially embedding information about globals.
    143     virtual bool getEmbedGlobalInfoSkipConstant() const = 0;
    144 };
    145 
    146 
    147 }
    148 }
    149 
    150 #endif
    151