#define NO_RS_FUNCS 1 #include "stdio.h" #include "RenderScript.h" #include struct Element; struct ElementField { // An Element Field is a combination of an Element with a name assigned. const char *name; Element *e; ElementField(const char *n, Element *_e) { name = n; e = _e; } ElementField() { name = NULL; e = NULL; } }; struct Element { // An Element can take one of two forms. // 1: Basic. It contains a single basic type and vector size. // 2: Complex. It contains a list of fields with names. Each field // will in turn be another element. ElementField *fields; size_t fieldCount; // If field count is 0, the element is a Basic type. const char *name; bool generated; // The basic data type from RenderScript.h RsDataType compType; // The vector size of the data type for float2, float3, .... // Allowed sizes are 2,3,4,8,16 uint32_t compVectorSize; Element() { fields = NULL; fieldCount = 0; name = NULL; generated = false; compType = RS_TYPE_ELEMENT; compVectorSize = 0; } Element(uint32_t _fieldCount, const char *_name) { fields = new ElementField[_fieldCount]; fieldCount = _fieldCount; name = _name; generated = false; compType = RS_TYPE_ELEMENT; compVectorSize = 0; } Element(RsDataType t, uint32_t s) { fields = NULL; fieldCount = 0; name = NULL; generated = false; compType = t; compVectorSize = s; } }; static void genHeader(FILE *f, const char *packageName) { fprintf(f, "package %s;\n", packageName); fprintf(f, "\n"); fprintf(f, "import android.renderscript.*;\n"); fprintf(f, "\n"); fprintf(f, "\n"); } static const char * RSTypeToJava(RsDataType dt) { switch(dt) { //case RS_TYPE_FLOAT_16: return "float"; case RS_TYPE_FLOAT_32: return "float"; //case RS_TYPE_FLOAT_64: return "double"; case RS_TYPE_SIGNED_8: return "byte"; case RS_TYPE_SIGNED_16: return "short"; case RS_TYPE_SIGNED_32: return "int"; //case RS_TYPE_SIGNED_64: return "long"; case RS_TYPE_UNSIGNED_8: return "short"; case RS_TYPE_UNSIGNED_16: return "int"; case RS_TYPE_UNSIGNED_32: return "long"; //case RS_TYPE_UNSIGNED_64: return NULL; //case RS_TYPE_ELEMENT: return "android.renderscript.Element"; //case RS_TYPE_TYPE: return "android.renderscript.Type"; //case RS_TYPE_ALLOCATION: return "android.renderscript.Allocation"; //case RS_TYPE_SAMPLER: return "android.renderscript.Sampler"; //case RS_TYPE_SCRIPT: return "android.renderscript.Script"; //case RS_TYPE_MESH: return "android.renderscript.Mesh"; //case RS_TYPE_PROGRAM_FRAGMENT: return "android.renderscript.ProgramFragment"; //case RS_TYPE_PROGRAM_VERTEX: return "android.renderscript.ProgramVertex"; //case RS_TYPE_PROGRAM_RASTER: return "android.renderscript.ProgramRaster"; //case RS_TYPE_PROGRAM_STORE: return "android.renderscript.ProgramStore"; default: return NULL; } return NULL; } static const char * RSTypeToString(RsDataType dt) { switch(dt) { case RS_TYPE_FLOAT_16: return "F16"; case RS_TYPE_FLOAT_32: return "F32"; case RS_TYPE_FLOAT_64: return "F64"; case RS_TYPE_SIGNED_8: return "I8"; case RS_TYPE_SIGNED_16: return "I16"; case RS_TYPE_SIGNED_32: return "I32"; case RS_TYPE_SIGNED_64: return "I64"; case RS_TYPE_UNSIGNED_8: return "U8"; case RS_TYPE_UNSIGNED_16: return "U16"; case RS_TYPE_UNSIGNED_32: return "U32"; case RS_TYPE_UNSIGNED_64: return "U64"; //case RS_TYPE_ELEMENT: return "android.renderscript.Element"; //case RS_TYPE_TYPE: return "android.renderscript.Type"; //case RS_TYPE_ALLOCATION: return "android.renderscript.Allocation"; //case RS_TYPE_SAMPLER: return "android.renderscript.Sampler"; //case RS_TYPE_SCRIPT: return "android.renderscript.Script"; //case RS_TYPE_MESH: return "android.renderscript.Mesh"; //case RS_TYPE_PROGRAM_FRAGMENT: return "android.renderscript.ProgramFragment"; //case RS_TYPE_PROGRAM_VERTEX: return "android.renderscript.ProgramVertex"; //case RS_TYPE_PROGRAM_RASTER: return "android.renderscript.ProgramRaster"; //case RS_TYPE_PROGRAM_STORE: return "android.renderscript.ProgramStore"; default: return NULL; } return NULL; } bool rsGenerateElementClass(const Element *e, const char *packageName, FILE *f) { genHeader(f, packageName); fprintf(f, "class Element_%s {\n", e->name); for (size_t ct=0; ct < e->fieldCount; ct++) { const char *ts = RSTypeToJava(e->fields[ct].e->compType); if (ts == NULL) { return false; } fprintf(f, " public %s %s;\n", ts, e->fields[ct].name); } fprintf(f, "\n"); fprintf(f, " static Element getElement(RenderScript rs) {\n"); fprintf(f, " Element.Builder eb = new Element.Builder(rs);\n"); for (size_t ct=0; ct < e->fieldCount; ct++) { const char *ts = RSTypeToString(e->fields[ct].e->compType); fprintf(f, " eb.add(Element.USER_%s(rs), \"%s\");\n", ts, e->fields[ct].name); } fprintf(f, " return eb.create();\n"); fprintf(f, " }\n"); fprintf(f, " static Allocation createAllocation(RenderScript rs) {\n"); fprintf(f, " Element e = getElement(rs);\n"); fprintf(f, " Allocation a = Allocation.createSized(rs, e, 1);\n"); fprintf(f, " return a;\n"); fprintf(f, " }\n"); fprintf(f, " void copyToAllocation(Allocation a) {\n"); fprintf(f, " mIOBuffer.reset();\n"); for (size_t ct=0; ct < e->fieldCount; ct++) { const char *ts = RSTypeToString(e->fields[ct].e->compType); fprintf(f, " mIOBuffer.add%s(%s);\n", ts, e->fields[ct].name); } fprintf(f, " a.data(mIOBuffer.getData());\n"); fprintf(f, " }\n"); fprintf(f, " private FieldPacker mIOBuffer[];\n"); fprintf(f, " public Element_%s() {\n", e->name); fprintf(f, " mIOBuffer = new FieldPacker(%i);\n", 100/*element->getSizeBytes()*/); fprintf(f, " }\n"); fprintf(f, "}\n"); return true; } bool rsGenerateElementClassFile(Element *e, const char *packageName) { char buf[1024]; sprintf(buf, "Element_%s.java", e->name); printf("Creating file %s \n", buf); FILE *f = fopen(buf, "w"); bool ret = rsGenerateElementClass(e, packageName, f); fclose(f); return ret; } /* bool rsGenerateScriptClass(const ScriptC *script, const char *packageName, FILE *f) { genHeader(f, packageName); fprintf(f, "class ScriptC_%s {\n", script->getName()); ObjectBaseRef mTypes[MAX_SCRIPT_BANKS]; String8 mSlotNames[MAX_SCRIPT_BANKS]; bool mSlotWritable[MAX_SCRIPT_BANKS]; } */ int main(int argc, const char *argv) { Element *u8 = new Element(RS_TYPE_UNSIGNED_8, 1); Element *i32 = new Element(RS_TYPE_SIGNED_32, 1); Element *f32 = new Element(RS_TYPE_FLOAT_32, 1); Element *e_Pixel = new Element(4, "Pixel"); e_Pixel->fields[0].e = u8; e_Pixel->fields[0].name = "a"; e_Pixel->fields[1].e = u8; e_Pixel->fields[1].name = "b"; e_Pixel->fields[2].e = u8; e_Pixel->fields[2].name = "g"; e_Pixel->fields[3].e = u8; e_Pixel->fields[3].name = "r"; Element *e_Params = new Element(5, "Params"); e_Params->fields[0].e = i32; e_Params->fields[0].name = "inHeight"; e_Params->fields[1].e = i32; e_Params->fields[1].name = "inWidth"; e_Params->fields[2].e = i32; e_Params->fields[2].name = "outHeight"; e_Params->fields[3].e = i32; e_Params->fields[3].name = "outWidth"; e_Params->fields[4].e = f32; e_Params->fields[4].name = "threshold"; printf("1\n"); rsGenerateElementClassFile(e_Pixel, "android"); rsGenerateElementClassFile(e_Params, "android"); }