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