1#define NO_RS_FUNCS 1
2
3#include "stdio.h"
4#include "RenderScript.h"
5#include <vector>
6
7struct Element;
8
9struct ElementField {
10    const char *name;
11    Element *e;
12    ElementField(const char *n, Element *_e) {
13        name = n;
14        e = _e;
15    }
16    ElementField() {
17        name = NULL;
18        e = NULL;
19    }
20};
21
22struct Element {
23    ElementField *fields;
24    size_t fieldCount;
25    const char *name;
26    bool generated;
27
28    RsDataType compType;
29    uint32_t compVectorSize;
30
31    Element() {
32        fields = NULL;
33        fieldCount = 0;
34        name = NULL;
35        generated = false;
36        compType = RS_TYPE_ELEMENT;
37        compVectorSize = 0;
38    }
39
40    Element(uint32_t _fieldCount, const char *_name) {
41        fields = new ElementField[_fieldCount];
42        fieldCount = _fieldCount;
43        name = _name;
44        generated = false;
45        compType = RS_TYPE_ELEMENT;
46        compVectorSize = 0;
47    }
48
49    Element(RsDataType t, uint32_t s) {
50        fields = NULL;
51        fieldCount = 0;
52        name = NULL;
53        generated = false;
54        compType = t;
55        compVectorSize = s;
56    }
57
58};
59
60
61static void genHeader(FILE *f, const char *packageName)
62{
63    fprintf(f, "package %s;\n", packageName);
64    fprintf(f, "\n");
65    fprintf(f, "import android.renderscript.*;\n");
66    fprintf(f, "\n");
67    fprintf(f, "\n");
68}
69
70static const char * RSTypeToJava(RsDataType dt)
71{
72    switch(dt) {
73    //case RS_TYPE_FLOAT_16:         return "float";
74    case RS_TYPE_FLOAT_32:         return "float";
75    //case RS_TYPE_FLOAT_64:         return "double";
76
77    case RS_TYPE_SIGNED_8:         return "byte";
78    case RS_TYPE_SIGNED_16:        return "short";
79    case RS_TYPE_SIGNED_32:        return "int";
80    //case RS_TYPE_SIGNED_64:        return "long";
81
82    case RS_TYPE_UNSIGNED_8:       return "short";
83    case RS_TYPE_UNSIGNED_16:      return "int";
84    case RS_TYPE_UNSIGNED_32:      return "long";
85    //case RS_TYPE_UNSIGNED_64:      return NULL;
86
87    //case RS_TYPE_ELEMENT:          return "android.renderscript.Element";
88    //case RS_TYPE_TYPE:             return "android.renderscript.Type";
89    //case RS_TYPE_ALLOCATION:       return "android.renderscript.Allocation";
90    //case RS_TYPE_SAMPLER:          return "android.renderscript.Sampler";
91    //case RS_TYPE_SCRIPT:           return "android.renderscript.Script";
92    //case RS_TYPE_MESH:             return "android.renderscript.Mesh";
93    //case RS_TYPE_PROGRAM_FRAGMENT: return "android.renderscript.ProgramFragment";
94    //case RS_TYPE_PROGRAM_VERTEX:   return "android.renderscript.ProgramVertex";
95    //case RS_TYPE_PROGRAM_RASTER:   return "android.renderscript.ProgramRaster";
96    //case RS_TYPE_PROGRAM_STORE:    return "android.renderscript.ProgramStore";
97    default: return NULL;
98    }
99    return NULL;
100}
101
102static const char * RSTypeToString(RsDataType dt)
103{
104    switch(dt) {
105    case RS_TYPE_FLOAT_16:         return "F16";
106    case RS_TYPE_FLOAT_32:         return "F32";
107    case RS_TYPE_FLOAT_64:         return "F64";
108
109    case RS_TYPE_SIGNED_8:         return "I8";
110    case RS_TYPE_SIGNED_16:        return "I16";
111    case RS_TYPE_SIGNED_32:        return "I32";
112    case RS_TYPE_SIGNED_64:        return "I64";
113
114    case RS_TYPE_UNSIGNED_8:       return "U8";
115    case RS_TYPE_UNSIGNED_16:      return "U16";
116    case RS_TYPE_UNSIGNED_32:      return "U32";
117    case RS_TYPE_UNSIGNED_64:      return "U64";
118
119    //case RS_TYPE_ELEMENT:          return "android.renderscript.Element";
120    //case RS_TYPE_TYPE:             return "android.renderscript.Type";
121    //case RS_TYPE_ALLOCATION:       return "android.renderscript.Allocation";
122    //case RS_TYPE_SAMPLER:          return "android.renderscript.Sampler";
123    //case RS_TYPE_SCRIPT:           return "android.renderscript.Script";
124    //case RS_TYPE_MESH:             return "android.renderscript.Mesh";
125    //case RS_TYPE_PROGRAM_FRAGMENT: return "android.renderscript.ProgramFragment";
126    //case RS_TYPE_PROGRAM_VERTEX:   return "android.renderscript.ProgramVertex";
127    //case RS_TYPE_PROGRAM_RASTER:   return "android.renderscript.ProgramRaster";
128    //case RS_TYPE_PROGRAM_STORE:    return "android.renderscript.ProgramStore";
129    default: return NULL;
130    }
131    return NULL;
132}
133
134bool rsGenerateElementClass(const Element *e, const char *packageName, FILE *f)
135{
136    genHeader(f, packageName);
137
138    fprintf(f, "class Element_%s {\n", e->name);
139
140    for (size_t ct=0; ct < e->fieldCount; ct++) {
141        const char *ts = RSTypeToJava(e->fields[ct].e->compType);
142        if (ts == NULL) {
143            return false;
144        }
145        fprintf(f, "    public %s %s;\n", ts, e->fields[ct].name);
146    }
147
148    fprintf(f, "\n");
149    fprintf(f, "    static Element getElement(RenderScript rs) {\n");
150    fprintf(f, "        Element.Builder eb = new Element.Builder(rs);\n");
151    for (size_t ct=0; ct < e->fieldCount; ct++) {
152        const char *ts = RSTypeToString(e->fields[ct].e->compType);
153        fprintf(f, "         eb.add(Element.USER_%s(rs), \"%s\");\n", ts, e->fields[ct].name);
154    }
155    fprintf(f, "        return eb.create();\n");
156    fprintf(f, "    }\n");
157
158    fprintf(f, "    static Allocation createAllocation(RenderScript rs) {\n");
159    fprintf(f, "        Element e = getElement(rs);\n");
160    fprintf(f, "        Allocation a = Allocation.createSized(rs, e, 1);\n");
161    fprintf(f, "        return a;\n");
162    fprintf(f, "    }\n");
163
164
165    fprintf(f, "    void copyToAllocation(Allocation a) {\n");
166    fprintf(f, "        mIOBuffer.reset();\n");
167    for (size_t ct=0; ct < e->fieldCount; ct++) {
168        const char *ts = RSTypeToString(e->fields[ct].e->compType);
169        fprintf(f, "         mIOBuffer.add%s(%s);\n", ts, e->fields[ct].name);
170    }
171    fprintf(f, "        a.data(mIOBuffer.getData());\n");
172    fprintf(f, "    }\n");
173
174
175
176    fprintf(f, "    private FieldPacker mIOBuffer[];\n");
177    fprintf(f, "    public Element_%s() {\n", e->name);
178    fprintf(f, "        mIOBuffer = new FieldPacker(%i);\n", 100/*element->getSizeBytes()*/);
179    fprintf(f, "    }\n");
180
181
182    fprintf(f, "}\n");
183
184    return true;
185}
186
187bool rsGenerateElementClassFile(Element *e, const char *packageName)
188{
189    char buf[1024];
190    sprintf(buf, "Element_%s.java", e->name);
191    printf("Creating file %s \n", buf);
192    FILE *f = fopen(buf, "w");
193    bool ret = rsGenerateElementClass(e, packageName, f);
194    fclose(f);
195    return ret;
196}
197
198
199
200
201/*
202bool rsGenerateScriptClass(const ScriptC *script, const char *packageName, FILE *f)
203{
204    genHeader(f, packageName);
205
206    fprintf(f, "class ScriptC_%s {\n", script->getName());
207
208
209
210        ObjectBaseRef<const Type> mTypes[MAX_SCRIPT_BANKS];
211    String8 mSlotNames[MAX_SCRIPT_BANKS];
212    bool mSlotWritable[MAX_SCRIPT_BANKS];
213
214
215}
216*/
217
218
219
220int main(int argc, const char *argv)
221{
222    Element *u8 = new Element(RS_TYPE_UNSIGNED_8, 1);
223    Element *i32 = new Element(RS_TYPE_SIGNED_32, 1);
224    Element *f32 = new Element(RS_TYPE_FLOAT_32, 1);
225
226    Element *e_Pixel = new Element(4, "Pixel");
227    e_Pixel->fields[0].e = u8;
228    e_Pixel->fields[0].name = "a";
229    e_Pixel->fields[1].e = u8;
230    e_Pixel->fields[1].name = "b";
231    e_Pixel->fields[2].e = u8;
232    e_Pixel->fields[2].name = "g";
233    e_Pixel->fields[3].e = u8;
234    e_Pixel->fields[3].name = "r";
235
236    Element *e_Params = new Element(5, "Params");
237    e_Params->fields[0].e = i32;
238    e_Params->fields[0].name = "inHeight";
239    e_Params->fields[1].e = i32;
240    e_Params->fields[1].name = "inWidth";
241    e_Params->fields[2].e = i32;
242    e_Params->fields[2].name = "outHeight";
243    e_Params->fields[3].e = i32;
244    e_Params->fields[3].name = "outWidth";
245    e_Params->fields[4].e = f32;
246    e_Params->fields[4].name = "threshold";
247
248
249    printf("1\n");
250    rsGenerateElementClassFile(e_Pixel, "android");
251    rsGenerateElementClassFile(e_Params, "android");
252
253}
254
255