1 #define NO_RS_FUNCS 1
2
3 #include "stdio.h"
4 #include "RenderScript.h"
5 #include <vector>
6
7 struct Element;
8
9 struct ElementField {
10 const char *name;
11 Element *e;
ElementFieldElementField12 ElementField(const char *n, Element *_e) {
13 name = n;
14 e = _e;
15 }
ElementFieldElementField16 ElementField() {
17 name = NULL;
18 e = NULL;
19 }
20 };
21
22 struct Element {
23 ElementField *fields;
24 size_t fieldCount;
25 const char *name;
26 bool generated;
27
28 RsDataType compType;
29 uint32_t compVectorSize;
30
ElementElement31 Element() {
32 fields = NULL;
33 fieldCount = 0;
34 name = NULL;
35 generated = false;
36 compType = RS_TYPE_ELEMENT;
37 compVectorSize = 0;
38 }
39
ElementElement40 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
ElementElement49 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
genHeader(FILE * f,const char * packageName)61 static 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
RSTypeToJava(RsDataType dt)70 static 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
RSTypeToString(RsDataType dt)102 static 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
rsGenerateElementClass(const Element * e,const char * packageName,FILE * f)134 bool 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
rsGenerateElementClassFile(Element * e,const char * packageName)187 bool 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 /*
202 bool 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
main(int argc,const char * argv)220 int 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