slang_rs_reflection.cpp revision 92b344a51c6c4934e96882bd401e4b13d6d03db8 
1#include "slang_rs_reflection.h"
2
3#include <utility>
4#include <cstdarg>
5#include <cctype>
6#include <sys/stat.h>
7
8#include "llvm/ADT/APFloat.h"
9
10#include "slang_rs_context.h"
11#include "slang_rs_export_var.h"
12#include "slang_rs_export_func.h"
13#include "slang_rs_reflect_utils.h"
14
15#define RS_SCRIPT_CLASS_NAME_PREFIX      "ScriptC_"
16#define RS_SCRIPT_CLASS_SUPER_CLASS_NAME "ScriptC"
17
18#define RS_TYPE_CLASS_NAME_PREFIX        "ScriptField_"
19#define RS_TYPE_CLASS_SUPER_CLASS_NAME   "android.renderscript.Script.FieldBase"
20
21#define RS_TYPE_ITEM_CLASS_NAME          "Item"
22
23#define RS_TYPE_ITEM_BUFFER_NAME         "mItemArray"
24#define RS_TYPE_ITEM_BUFFER_PACKER_NAME  "mIOBuffer"
25
26#define RS_EXPORT_VAR_INDEX_PREFIX       "mExportVarIdx_"
27#define RS_EXPORT_VAR_PREFIX             "mExportVar_"
28
29#define RS_EXPORT_FUNC_INDEX_PREFIX      "mExportFuncIdx_"
30
31#define RS_EXPORT_VAR_ALLOCATION_PREFIX  "mAlloction_"
32#define RS_EXPORT_VAR_DATA_STORAGE_PREFIX "mData_"
33
34using namespace slang;
35
36// Some utility function using internal in RSReflection
37static bool GetClassNameFromFileName(const std::string &FileName,
38                                     std::string &ClassName) {
39  ClassName.clear();
40
41  if (FileName.empty() || (FileName == "-"))
42    return true;
43
44  ClassName =
45      RSSlangReflectUtils::JavaClassNameFromRSFileName(FileName.c_str());
46
47  return true;
48}
49
50static const char *GetPrimitiveTypeName(const RSExportPrimitiveType *EPT) {
51  static const char *PrimitiveTypeJavaNameMap[] = {
52    "",         // RSExportPrimitiveType::DataTypeFloat16
53    "float",    // RSExportPrimitiveType::DataTypeFloat32
54    "double",   // RSExportPrimitiveType::DataTypeFloat64
55    "byte",     // RSExportPrimitiveType::DataTypeSigned8
56    "short",    // RSExportPrimitiveType::DataTypeSigned16
57    "int",      // RSExportPrimitiveType::DataTypeSigned32
58    "long",     // RSExportPrimitiveType::DataTypeSigned64
59    "short",    // RSExportPrimitiveType::DataTypeUnsigned8
60    "int",      // RSExportPrimitiveType::DataTypeUnsigned16
61    "long",     // RSExportPrimitiveType::DataTypeUnsigned32
62    "long",     // RSExportPrimitiveType::DataTypeUnsigned64
63    "boolean",  // RSExportPrimitiveType::DataTypeBoolean
64
65    "int",      // RSExportPrimitiveType::DataTypeUnsigned565
66    "int",      // RSExportPrimitiveType::DataTypeUnsigned5551
67    "int",      // RSExportPrimitiveType::DataTypeUnsigned4444
68
69    "",     // (Dummy) RSExportPrimitiveType::DataTypeRSMatrix2x2
70    "",     // (Dummy) RSExportPrimitiveType::DataTypeRSMatrix3x3
71    "",     // (Dummy) RSExportPrimitiveType::DataTypeRSMatrix4x4
72
73    "Element",      // RSExportPrimitiveType::DataTypeRSElement
74    "Type",         // RSExportPrimitiveType::DataTypeRSType
75    "Allocation",   // RSExportPrimitiveType::DataTypeRSAllocation
76    "Sampler",      // RSExportPrimitiveType::DataTypeRSSampler
77    "Script",       // RSExportPrimitiveType::DataTypeRSScript
78    "Mesh",         // RSExportPrimitiveType::DataTypeRSMesh
79    "ProgramFragment",  // RSExportPrimitiveType::DataTypeRSProgramFragment
80    "ProgramVertex",    // RSExportPrimitiveType::DataTypeRSProgramVertex
81    "ProgramRaster",    // RSExportPrimitiveType::DataTypeRSProgramRaster
82    "ProgramStore",     // RSExportPrimitiveType::DataTypeRSProgramStore
83    "Font",     // RSExportPrimitiveType::DataTypeRSFont
84  };
85  unsigned TypeId = EPT->getType();
86
87  if (TypeId < (sizeof(PrimitiveTypeJavaNameMap) / sizeof(const char*))) {
88    return PrimitiveTypeJavaNameMap[ EPT->getType() ];
89  }
90
91  assert(false && "GetPrimitiveTypeName : Unknown primitive data type");
92  return NULL;
93}
94
95static const char *GetVectorTypeName(const RSExportVectorType *EVT) {
96  static const char *VectorTypeJavaNameMap[][3] = {
97    /* 0 */ { "Byte2",  "Byte3",    "Byte4" },
98    /* 1 */ { "Short2", "Short3",   "Short4" },
99    /* 2 */ { "Int2",   "Int3",     "Int4" },
100    /* 3 */ { "Long2",  "Long3",    "Long4" },
101    /* 4 */ { "Float2", "Float3",   "Float4" },
102  };
103
104  const char **BaseElement = NULL;
105
106  switch (EVT->getType()) {
107    case RSExportPrimitiveType::DataTypeSigned8:
108    case RSExportPrimitiveType::DataTypeBoolean: {
109      BaseElement = VectorTypeJavaNameMap[0];
110      break;
111    }
112    case RSExportPrimitiveType::DataTypeSigned16:
113    case RSExportPrimitiveType::DataTypeUnsigned8: {
114      BaseElement = VectorTypeJavaNameMap[1];
115      break;
116    }
117    case RSExportPrimitiveType::DataTypeSigned32:
118    case RSExportPrimitiveType::DataTypeUnsigned16: {
119      BaseElement = VectorTypeJavaNameMap[2];
120      break;
121    }
122    case RSExportPrimitiveType::DataTypeSigned64:
123    case RSExportPrimitiveType::DataTypeUnsigned32: {
124      BaseElement = VectorTypeJavaNameMap[3];
125      break;
126    }
127    case RSExportPrimitiveType::DataTypeFloat32: {
128      BaseElement = VectorTypeJavaNameMap[4];
129      break;
130    }
131    default: {
132      assert(false && "RSReflection::genElementTypeName : Unsupported vector "
133                      "element data type");
134      break;
135    }
136  }
137
138  assert((EVT->getNumElement() > 1) &&
139         (EVT->getNumElement() <= 4) &&
140         "Number of element in vector type is invalid");
141
142  return BaseElement[EVT->getNumElement() - 2];
143}
144
145static const char *GetMatrixTypeName(const RSExportMatrixType *EMT) {
146  static const char *MatrixTypeJavaNameMap[] = {
147    /* 2x2 */ "Matrix2f",
148    /* 3x3 */ "Matrix3f",
149    /* 4x4 */ "Matrix4f",
150  };
151  unsigned Dim = EMT->getDim();
152
153  if ((Dim - 2) < (sizeof(MatrixTypeJavaNameMap) / sizeof(const char*)))
154    return MatrixTypeJavaNameMap[ EMT->getDim() - 2 ];
155
156  assert(false && "GetMatrixTypeName : Unsupported matrix dimension");
157  return NULL;
158}
159
160static const char *GetVectorAccessor(int Index) {
161  static const char *VectorAccessorMap[] = {
162    /* 0 */ "x",
163    /* 1 */ "y",
164    /* 2 */ "z",
165    /* 3 */ "w",
166  };
167
168  assert((Index >= 0) &&
169         (Index < (sizeof(VectorAccessorMap) / sizeof(const char*))) &&
170         "Out-of-bound index to access vector member");
171
172  return VectorAccessorMap[Index];
173}
174
175static const char *GetPackerAPIName(const RSExportPrimitiveType *EPT) {
176  static const char *PrimitiveTypePackerAPINameMap[] = {
177    "",         // RSExportPrimitiveType::DataTypeFloat16
178    "addF32",   // RSExportPrimitiveType::DataTypeFloat32
179    "addF64",   // RSExportPrimitiveType::DataTypeFloat64
180    "addI8",    // RSExportPrimitiveType::DataTypeSigned8
181    "addI16",   // RSExportPrimitiveType::DataTypeSigned16
182    "addI32",   // RSExportPrimitiveType::DataTypeSigned32
183    "addI64",   // RSExportPrimitiveType::DataTypeSigned64
184    "addU8",    // RSExportPrimitiveType::DataTypeUnsigned8
185    "addU16",   // RSExportPrimitiveType::DataTypeUnsigned16
186    "addU32",   // RSExportPrimitiveType::DataTypeUnsigned32
187    "addU64",   // RSExportPrimitiveType::DataTypeUnsigned64
188    "addBoolean",  // RSExportPrimitiveType::DataTypeBoolean
189
190    "addU16",   // RSExportPrimitiveType::DataTypeUnsigned565
191    "addU16",   // RSExportPrimitiveType::DataTypeUnsigned5551
192    "addU16",   // RSExportPrimitiveType::DataTypeUnsigned4444
193
194    "addObj",   // RSExportPrimitiveType::DataTypeRSMatrix2x2
195    "addObj",   // RSExportPrimitiveType::DataTypeRSMatrix3x3
196    "addObj",   // RSExportPrimitiveType::DataTypeRSMatrix4x4
197
198    "addObj",   // RSExportPrimitiveType::DataTypeRSElement
199    "addObj",   // RSExportPrimitiveType::DataTypeRSType
200    "addObj",   // RSExportPrimitiveType::DataTypeRSAllocation
201    "addObj",   // RSExportPrimitiveType::DataTypeRSSampler
202    "addObj",   // RSExportPrimitiveType::DataTypeRSScript
203    "addObj",   // RSExportPrimitiveType::DataTypeRSMesh
204    "addObj",   // RSExportPrimitiveType::DataTypeRSProgramFragment
205    "addObj",   // RSExportPrimitiveType::DataTypeRSProgramVertex
206    "addObj",   // RSExportPrimitiveType::DataTypeRSProgramRaster
207    "addObj",   // RSExportPrimitiveType::DataTypeRSProgramStore
208    "addObj",   // RSExportPrimitiveType::DataTypeRSFont
209  };
210  unsigned TypeId = EPT->getType();
211
212  if (TypeId < (sizeof(PrimitiveTypePackerAPINameMap) / sizeof(const char*)))
213    return PrimitiveTypePackerAPINameMap[ EPT->getType() ];
214
215  assert(false && "GetPackerAPIName : Unknown primitive data type");
216  return NULL;
217}
218
219static std::string GetTypeName(const RSExportType *ET) {
220  switch (ET->getClass()) {
221    case RSExportType::ExportClassPrimitive:
222    case RSExportType::ExportClassConstantArray: {
223      return GetPrimitiveTypeName(
224                static_cast<const RSExportPrimitiveType*>(ET));
225      break;
226    }
227    case RSExportType::ExportClassPointer: {
228      const RSExportType *PointeeType =
229          static_cast<const RSExportPointerType*>(ET)->getPointeeType();
230
231      if (PointeeType->getClass() != RSExportType::ExportClassRecord)
232        return "Allocation";
233      else
234        return RS_TYPE_CLASS_NAME_PREFIX + PointeeType->getName();
235      break;
236    }
237    case RSExportType::ExportClassVector: {
238      return GetVectorTypeName(static_cast<const RSExportVectorType*>(ET));
239      break;
240    }
241    case RSExportType::ExportClassMatrix: {
242      return GetMatrixTypeName(static_cast<const RSExportMatrixType*>(ET));
243      break;
244    }
245    case RSExportType::ExportClassRecord: {
246      return RS_TYPE_CLASS_NAME_PREFIX + ET->getName() +
247                 "."RS_TYPE_ITEM_CLASS_NAME;
248      break;
249    }
250    default: {
251      assert(false && "Unknown class of type");
252    }
253  }
254
255  return "";
256}
257
258static const char *GetBuiltinElementConstruct(const RSExportType *ET) {
259  if (ET->getClass() == RSExportType::ExportClassPrimitive ||
260      ET->getClass() == RSExportType::ExportClassConstantArray) {
261    const RSExportPrimitiveType *EPT =
262        static_cast<const RSExportPrimitiveType*>(ET);
263    if (EPT->getKind() == RSExportPrimitiveType::DataKindUser) {
264      static const char *PrimitiveBuiltinElementConstructMap[] = {
265        NULL,   // RSExportPrimitiveType::DataTypeFloat16
266        "F32",  // RSExportPrimitiveType::DataTypeFloat32
267        "F64",  // RSExportPrimitiveType::DataTypeFloat64
268        "I8",   // RSExportPrimitiveType::DataTypeSigned8
269        NULL,   // RSExportPrimitiveType::DataTypeSigned16
270        "I32",  // RSExportPrimitiveType::DataTypeSigned32
271        "I64",  // RSExportPrimitiveType::DataTypeSigned64
272        "U8",   // RSExportPrimitiveType::DataTypeUnsigned8
273        NULL,   // RSExportPrimitiveType::DataTypeUnsigned16
274        "U32",  // RSExportPrimitiveType::DataTypeUnsigned32
275        NULL,   // RSExportPrimitiveType::DataTypeUnsigned64
276        "BOOLEAN",  // RSExportPrimitiveType::DataTypeBoolean
277
278        NULL,   // RSExportPrimitiveType::DataTypeUnsigned565
279        NULL,   // RSExportPrimitiveType::DataTypeUnsigned5551
280        NULL,   // RSExportPrimitiveType::DataTypeUnsigned4444
281
282        NULL,   // (Dummy) RSExportPrimitiveType::DataTypeRSMatrix2x2
283        NULL,   // (Dummy) RSExportPrimitiveType::DataTypeRSMatrix3x3
284        NULL,   // (Dummy) RSExportPrimitiveType::DataTypeRSMatrix4x4
285
286        "ELEMENT",      // RSExportPrimitiveType::DataTypeRSElement
287        "TYPE",         // RSExportPrimitiveType::DataTypeRSType
288        "ALLOCATION",   // RSExportPrimitiveType::DataTypeRSAllocation
289        "SAMPLER",      // RSExportPrimitiveType::DataTypeRSSampler
290        "SCRIPT",       // RSExportPrimitiveType::DataTypeRSScript
291        "MESH",         // RSExportPrimitiveType::DataTypeRSMesh
292        "PROGRAM_FRAGMENT",  // RSExportPrimitiveType::DataTypeRSProgramFragment
293        "PROGRAM_VERTEX",    // RSExportPrimitiveType::DataTypeRSProgramVertex
294        "PROGRAM_RASTER",    // RSExportPrimitiveType::DataTypeRSProgramRaster
295        "PROGRAM_STORE",     // RSExportPrimitiveType::DataTypeRSProgramStore
296        "FONT",       // RSExportPrimitiveType::DataTypeRSFont
297      };
298      unsigned TypeId = EPT->getType();
299
300      if (TypeId <
301          (sizeof(PrimitiveBuiltinElementConstructMap) / sizeof(const char*)))
302        return PrimitiveBuiltinElementConstructMap[ EPT->getType() ];
303    } else if (EPT->getKind() == RSExportPrimitiveType::DataKindPixelA) {
304      if (EPT->getType() == RSExportPrimitiveType::DataTypeUnsigned8)
305        return "A_8";
306    } else if (EPT->getKind() == RSExportPrimitiveType::DataKindPixelRGB) {
307      if (EPT->getType() == RSExportPrimitiveType::DataTypeUnsigned565)
308        return "RGB_565";
309      else if (EPT->getType() == RSExportPrimitiveType::DataTypeUnsigned8)
310        return "RGB_888";
311    } else if (EPT->getKind() == RSExportPrimitiveType::DataKindPixelRGBA) {
312      if (EPT->getType() == RSExportPrimitiveType::DataTypeUnsigned5551)
313        return "RGBA_5551";
314      else if (EPT->getType() == RSExportPrimitiveType::DataTypeUnsigned4444)
315        return "RGBA_4444";
316      else if (EPT->getType() == RSExportPrimitiveType::DataTypeUnsigned8)
317        return "RGBA_8888";
318    }
319  } else if (ET->getClass() == RSExportType::ExportClassVector) {
320    const RSExportVectorType *EVT = static_cast<const RSExportVectorType*>(ET);
321    if (EVT->getKind() == RSExportPrimitiveType::DataKindUser) {
322      if (EVT->getType() == RSExportPrimitiveType::DataTypeFloat32) {
323        if (EVT->getNumElement() == 2)
324          return "F32_2";
325        else if (EVT->getNumElement() == 3)
326          return "F32_3";
327        else if (EVT->getNumElement() == 4)
328          return "F32_4";
329      } else if (EVT->getType() == RSExportPrimitiveType::DataTypeUnsigned8) {
330        if (EVT->getNumElement() == 4)
331          return "U8_4";
332      }
333    }
334  } else if (ET->getClass() == RSExportType::ExportClassMatrix) {
335    const RSExportMatrixType *EMT = static_cast<const RSExportMatrixType *>(ET);
336    switch (EMT->getDim()) {
337      case 2: {
338        return "MATRIX_2X2";
339        break;
340      }
341      case 3: {
342        return "MATRIX_3X3";
343        break;
344      }
345      case 4: {
346        return "MATRIX_4X4";
347        break;
348      }
349      default: {
350        assert(false && "Unsupported dimension of matrix");
351      }
352    }
353  } else if (ET->getClass() == RSExportType::ExportClassPointer) {
354    // Treat pointer type variable as unsigned int
355    // TODO(zonr): this is target dependent
356    return "USER_I32";
357  }
358
359  return NULL;
360}
361
362static const char *GetElementDataKindName(RSExportPrimitiveType::DataKind DK) {
363  static const char *ElementDataKindNameMap[] = {
364    "Element.DataKind.USER",        // RSExportPrimitiveType::DataKindUser
365    "Element.DataKind.PIXEL_L",     // RSExportPrimitiveType::DataKindPixelL
366    "Element.DataKind.PIXEL_A",     // RSExportPrimitiveType::DataKindPixelA
367    "Element.DataKind.PIXEL_LA",    // RSExportPrimitiveType::DataKindPixelLA
368    "Element.DataKind.PIXEL_RGB",   // RSExportPrimitiveType::DataKindPixelRGB
369    "Element.DataKind.PIXEL_RGBA",  // RSExportPrimitiveType::DataKindPixelRGBA
370  };
371
372  if (static_cast<unsigned>(DK) <
373      (sizeof(ElementDataKindNameMap) / sizeof(const char*)))
374    return ElementDataKindNameMap[ DK ];
375  else
376    return NULL;
377}
378
379static const char *GetElementDataTypeName(RSExportPrimitiveType::DataType DT) {
380  static const char *ElementDataTypeNameMap[] = {
381    NULL,                           // RSExportPrimitiveType::DataTypeFloat16
382    "Element.DataType.FLOAT_32",    // RSExportPrimitiveType::DataTypeFloat32
383    "Element.DataType.FLOAT_64",    // RSExportPrimitiveType::DataTypeFloat64
384    "Element.DataType.SIGNED_8",    // RSExportPrimitiveType::DataTypeSigned8
385    "Element.DataType.SIGNED_16",   // RSExportPrimitiveType::DataTypeSigned16
386    "Element.DataType.SIGNED_32",   // RSExportPrimitiveType::DataTypeSigned32
387    "Element.DataType.SIGNED_64",   // RSExportPrimitiveType::DataTypeSigned64
388    "Element.DataType.UNSIGNED_8",  // RSExportPrimitiveType::DataTypeUnsigned8
389    "Element.DataType.UNSIGNED_16", // RSExportPrimitiveType::DataTypeUnsigned16
390    "Element.DataType.UNSIGNED_32", // RSExportPrimitiveType::DataTypeUnsigned32
391    NULL,                           // RSExportPrimitiveType::DataTypeUnsigned64
392    "Element.DataType.BOOLEAN",     // RSExportPrimitiveType::DataTypeBoolean
393
394    // RSExportPrimitiveType::DataTypeUnsigned565
395    "Element.DataType.UNSIGNED_5_6_5",
396    // RSExportPrimitiveType::DataTypeUnsigned5551
397    "Element.DataType.UNSIGNED_5_5_5_1",
398    // RSExportPrimitiveType::DataTypeUnsigned4444
399    "Element.DataType.UNSIGNED_4_4_4_4",
400
401    // DataTypeRSMatrix* must have been resolved in GetBuiltinElementConstruct()
402    NULL, // (Dummy) RSExportPrimitiveType::DataTypeRSMatrix2x2
403    NULL, // (Dummy) RSExportPrimitiveType::DataTypeRSMatrix3x3
404    NULL, // (Dummy) RSExportPrimitiveType::DataTypeRSMatrix4x4
405
406    "Element.DataType.RS_ELEMENT",  // RSExportPrimitiveType::DataTypeRSElement
407    "Element.DataType.RS_TYPE",     // RSExportPrimitiveType::DataTypeRSType
408      // RSExportPrimitiveType::DataTypeRSAllocation
409    "Element.DataType.RS_ALLOCATION",
410      // RSExportPrimitiveType::DataTypeRSSampler
411    "Element.DataType.RS_SAMPLER",
412      // RSExportPrimitiveType::DataTypeRSScript
413    "Element.DataType.RS_SCRIPT",
414      // RSExportPrimitiveType::DataTypeRSMesh
415    "Element.DataType.RS_MESH",
416      // RSExportPrimitiveType::DataTypeRSProgramFragment
417    "Element.DataType.RS_PROGRAM_FRAGMENT",
418      // RSExportPrimitiveType::DataTypeRSProgramVertex
419    "Element.DataType.RS_PROGRAM_VERTEX",
420      // RSExportPrimitiveType::DataTypeRSProgramRaster
421    "Element.DataType.RS_PROGRAM_RASTER",
422      // RSExportPrimitiveType::DataTypeRSProgramStore
423    "Element.DataType.RS_PROGRAM_STORE",
424      // RSExportPrimitiveType::DataTypeRSFont
425    "Element.DataType.RS_FONT",
426  };
427
428  if (static_cast<unsigned>(DT) <
429      (sizeof(ElementDataTypeNameMap) / sizeof(const char*)))
430    return ElementDataTypeNameMap[ DT ];
431  else
432    return NULL;
433}
434
435bool RSReflection::openScriptFile(Context &C,
436                                  const std::string &ClassName,
437                                  std::string &ErrorMsg) {
438  if (!C.mUseStdout) {
439    C.mOF.clear();
440    std::string _path = RSSlangReflectUtils::ComputePackagedPath(
441        mRSContext->getReflectJavaPathName().c_str(),
442        C.getPackageName().c_str());
443
444    RSSlangReflectUtils::mkdir_p(_path.c_str());
445    C.mOF.open((_path + "/" + ClassName + ".java").c_str());
446    if (!C.mOF.good()) {
447      ErrorMsg = "failed to open file '" + _path + "/" + ClassName
448          + ".java' for write";
449
450      return false;
451    }
452  }
453  return true;
454}
455
456/********************** Methods to generate script class **********************/
457bool RSReflection::genScriptClass(Context &C,
458                                  const std::string &ClassName,
459                                  std::string &ErrorMsg) {
460  // Open the file
461  if (!openScriptFile(C, ClassName, ErrorMsg)) {
462    return false;
463  }
464
465  if (!C.startClass(Context::AM_Public,
466                    false,
467                    ClassName,
468                    RS_SCRIPT_CLASS_SUPER_CLASS_NAME,
469                    ErrorMsg))
470    return false;
471
472  genScriptClassConstructor(C);
473
474  // Reflect export variable
475  for (RSContext::const_export_var_iterator I = mRSContext->export_vars_begin(),
476           E = mRSContext->export_vars_end();
477       I != E;
478       I++)
479    genExportVariable(C, *I);
480
481  // Reflect export function
482  for (RSContext::const_export_func_iterator
483           I = mRSContext->export_funcs_begin(),
484           E = mRSContext->export_funcs_end();
485       I != E; I++)
486    genExportFunction(C, *I);
487
488  C.endClass();
489
490  return true;
491}
492
493void RSReflection::genScriptClassConstructor(Context &C) {
494  C.indent() << "// Constructor" << std::endl;
495  C.startFunction(Context::AM_Public,
496                  false,
497                  NULL,
498                  C.getClassName(),
499                  4,
500                  "RenderScript",
501                  "rs",
502                  "Resources",
503                  "resources",
504                  "int",
505                  "id",
506                  "boolean",
507                  "isRoot");
508  // Call constructor of super class
509  C.indent() << "super(rs, resources, id, isRoot);" << std::endl;
510
511  // If an exported variable has initial value, reflect it
512
513  for (RSContext::const_export_var_iterator I = mRSContext->export_vars_begin(),
514           E = mRSContext->export_vars_end();
515       I != E;
516       I++) {
517    const RSExportVar *EV = *I;
518    if (!EV->getInit().isUninit())
519      genInitExportVariable(C, EV->getType(), EV->getName(), EV->getInit());
520  }
521
522  C.endFunction();
523  return;
524}
525
526void RSReflection::genInitBoolExportVariable(Context &C,
527                                             const std::string &VarName,
528                                             const clang::APValue &Val) {
529  assert(!Val.isUninit() && "Not a valid initializer");
530
531  C.indent() << RS_EXPORT_VAR_PREFIX << VarName << " = ";
532  assert((Val.getKind() == clang::APValue::Int) &&
533         "Bool type has wrong initial APValue");
534
535  C.out() << ((Val.getInt().getSExtValue() == 0) ? "false" : "true")
536          << ";" << std::endl;
537
538  return;
539}
540
541void RSReflection::genInitPrimitiveExportVariable(Context &C,
542                                                  const std::string &VarName,
543                                                  const clang::APValue &Val) {
544  assert(!Val.isUninit() && "Not a valid initializer");
545
546  C.indent() << RS_EXPORT_VAR_PREFIX << VarName << " = ";
547  switch (Val.getKind()) {
548    case clang::APValue::Int: {
549      C.out() << Val.getInt().getSExtValue();
550      break;
551    }
552    case clang::APValue::Float: {
553      llvm::APFloat apf = Val.getFloat();
554      if (&apf.getSemantics() == &llvm::APFloat::IEEEsingle) {
555        C.out() << apf.convertToFloat() << "f";
556      } else {
557        C.out() << apf.convertToDouble();
558      }
559      break;
560    }
561
562    case clang::APValue::ComplexInt:
563    case clang::APValue::ComplexFloat:
564    case clang::APValue::LValue:
565    case clang::APValue::Vector: {
566      assert(false && "Primitive type cannot have such kind of initializer");
567      break;
568    }
569    default: {
570      assert(false && "Unknown kind of initializer");
571    }
572  }
573  C.out() << ";" << std::endl;
574
575  return;
576}
577
578void RSReflection::genInitExportVariable(Context &C,
579                                         const RSExportType *ET,
580                                         const std::string &VarName,
581                                         const clang::APValue &Val) {
582  assert(!Val.isUninit() && "Not a valid initializer");
583
584  switch (ET->getClass()) {
585    case RSExportType::ExportClassPrimitive:
586    case RSExportType::ExportClassConstantArray: {
587      const RSExportPrimitiveType *EPT =
588          static_cast<const RSExportPrimitiveType*>(ET);
589      if (EPT->getType() == RSExportPrimitiveType::DataTypeBoolean) {
590        genInitBoolExportVariable(C, VarName, Val);
591      } else {
592        genInitPrimitiveExportVariable(C, VarName, Val);
593      }
594      break;
595    }
596
597    case RSExportType::ExportClassPointer: {
598      if (!Val.isInt() || Val.getInt().getSExtValue() != 0)
599        std::cout << "Initializer which is non-NULL to pointer type variable "
600                     "will be ignored" << std::endl;
601      break;
602    }
603
604    case RSExportType::ExportClassVector: {
605      const RSExportVectorType *EVT =
606          static_cast<const RSExportVectorType*>(ET);
607      switch (Val.getKind()) {
608        case clang::APValue::Int:
609        case clang::APValue::Float: {
610          for (int i = 0; i < EVT->getNumElement(); i++) {
611            std::string Name =  VarName + "." + GetVectorAccessor(i);
612            genInitPrimitiveExportVariable(C, Name, Val);
613          }
614          break;
615        }
616
617        case clang::APValue::Vector: {
618          C.indent() << RS_EXPORT_VAR_PREFIX << VarName << " = new "
619                     << GetVectorTypeName(EVT) << "();" << std::endl;
620
621          unsigned NumElements =
622              std::min(static_cast<unsigned>(EVT->getNumElement()),
623                       Val.getVectorLength());
624          for (unsigned i = 0; i < NumElements; i++) {
625            const clang::APValue &ElementVal = Val.getVectorElt(i);
626            std::string Name = VarName + "." + GetVectorAccessor(i);
627            genInitPrimitiveExportVariable(C, Name, ElementVal);
628          }
629          break;
630        }
631
632        case clang::APValue::Uninitialized:
633        case clang::APValue::ComplexInt:
634        case clang::APValue::ComplexFloat:
635        case clang::APValue::LValue: {
636          assert(false && "Unexpected type of value of initializer.");
637        }
638      }
639      break;
640    }
641
642    // TODO(zonr): Resolving initializer of a record (and matrix) type variable
643    // is complex. It cannot obtain by just simply evaluating the initializer
644    // expression.
645    case RSExportType::ExportClassMatrix:
646    case RSExportType::ExportClassRecord: {
647#if 0
648      unsigned InitIndex = 0;
649      const RSExportRecordType *ERT =
650          static_cast<const RSExportRecordType*>(ET);
651
652      assert((Val.getKind() == clang::APValue::Vector) && "Unexpected type of "
653             "initializer for record type variable");
654
655      C.indent() << RS_EXPORT_VAR_PREFIX << VarName
656                 << " = new "RS_TYPE_CLASS_NAME_PREFIX << ERT->getName()
657                 <<  "."RS_TYPE_ITEM_CLASS_NAME"();" << std::endl;
658
659      for (RSExportRecordType::const_field_iterator I = ERT->fields_begin(),
660               E = ERT->fields_end();
661           I != E;
662           I++) {
663        const RSExportRecordType::Field *F = *I;
664        std::string FieldName = VarName + "." + F->getName();
665
666        if (InitIndex > Val.getVectorLength())
667          break;
668
669        genInitPrimitiveExportVariable(C,
670                                       FieldName,
671                                       Val.getVectorElt(InitIndex++));
672      }
673#endif
674      assert(false && "Unsupported initializer for record type variable "
675                      "currently");
676      break;
677    }
678
679    default: {
680      assert(false && "Unknown class of type");
681    }
682  }
683  return;
684}
685
686void RSReflection::genExportVariable(Context &C, const RSExportVar *EV) {
687  const RSExportType *ET = EV->getType();
688
689  C.indent() << "private final static int "RS_EXPORT_VAR_INDEX_PREFIX
690             << EV->getName() << " = " << C.getNextExportVarSlot() << ";"
691             << std::endl;
692
693  switch (ET->getClass()) {
694    case RSExportType::ExportClassPrimitive:
695    case RSExportType::ExportClassConstantArray: {
696      genPrimitiveTypeExportVariable(C, EV);
697      break;
698    }
699    case RSExportType::ExportClassPointer: {
700      genPointerTypeExportVariable(C, EV);
701      break;
702    }
703    case RSExportType::ExportClassVector: {
704      genVectorTypeExportVariable(C, EV);
705      break;
706    }
707    case RSExportType::ExportClassMatrix: {
708      genMatrixTypeExportVariable(C, EV);
709      break;
710    }
711    case RSExportType::ExportClassRecord: {
712      genRecordTypeExportVariable(C, EV);
713      break;
714    }
715    default: {
716      assert(false && "Unknown class of type");
717    }
718  }
719
720  return;
721}
722
723void RSReflection::genExportFunction(Context &C, const RSExportFunc *EF) {
724  C.indent() << "private final static int "RS_EXPORT_FUNC_INDEX_PREFIX
725             << EF->getName() << " = " << C.getNextExportFuncSlot() << ";"
726             << std::endl;
727
728  // invoke_*()
729  Context::ArgTy Args;
730
731  for (RSExportFunc::const_param_iterator I = EF->params_begin(),
732           E = EF->params_end();
733       I != E;
734       I++) {
735    const RSExportFunc::Parameter *P = *I;
736    Args.push_back(make_pair(GetTypeName(P->getType()), P->getName()));
737  }
738
739  C.startFunction(Context::AM_Public,
740                  false,
741                  "void",
742                  "invoke_" + EF->getName(),
743                  Args);
744
745  if (!EF->hasParam()) {
746    C.indent() << "invoke("RS_EXPORT_FUNC_INDEX_PREFIX << EF->getName() << ");"
747               << std::endl;
748  } else {
749    const RSExportRecordType *ERT = EF->getParamPacketType();
750    std::string FieldPackerName = EF->getName() + "_fp";
751
752    if (genCreateFieldPacker(C, ERT, FieldPackerName.c_str()))
753      genPackVarOfType(C, ERT, NULL, FieldPackerName.c_str());
754
755    C.indent() << "invoke("RS_EXPORT_FUNC_INDEX_PREFIX << EF->getName() << ", "
756               << FieldPackerName << ");" << std::endl;
757  }
758
759  C.endFunction();
760  return;
761}
762
763void RSReflection::genPrimitiveTypeExportVariable(Context &C,
764                                                  const RSExportVar *EV) {
765  assert((EV->getType()->getClass() == RSExportType::ExportClassPrimitive ||
766          EV->getType()->getClass() == RSExportType::ExportClassConstantArray)
767         && "Variable should be type of primitive here");
768
769  const RSExportPrimitiveType *EPT =
770      static_cast<const RSExportPrimitiveType*>(EV->getType());
771  const char *TypeName = GetPrimitiveTypeName(EPT);
772
773  C.indent() << "private " << TypeName << " "RS_EXPORT_VAR_PREFIX
774             << EV->getName() << ";" << std::endl;
775
776  // set_*()
777  if (!EV->isConst()) {
778    C.startFunction(Context::AM_Public,
779                    false,
780                    "void",
781                    "set_" + EV->getName(),
782                    1,
783                    TypeName,
784                    "v");
785    C.indent() << RS_EXPORT_VAR_PREFIX << EV->getName() << " = v;" << std::endl;
786
787    if (EPT->isRSObjectType())
788      C.indent() << "setVar("RS_EXPORT_VAR_INDEX_PREFIX << EV->getName()
789                 << ", (v == null) ? 0 : v.getID());" << std::endl;
790    else
791      C.indent() << "setVar("RS_EXPORT_VAR_INDEX_PREFIX << EV->getName()
792                 << ", v);" << std::endl;
793
794    C.endFunction();
795  }
796
797  genGetExportVariable(C, TypeName, EV->getName());
798
799  return;
800}
801
802void RSReflection::genPointerTypeExportVariable(Context &C,
803                                                const RSExportVar *EV) {
804  const RSExportType *ET = EV->getType();
805  const RSExportType *PointeeType;
806  std::string TypeName;
807
808  assert((ET->getClass() == RSExportType::ExportClassPointer) &&
809         "Variable should be type of pointer here");
810
811  PointeeType = static_cast<const RSExportPointerType*>(ET)->getPointeeType();
812  TypeName = GetTypeName(ET);
813
814  // bind_*()
815  C.indent() << "private " << TypeName << " "RS_EXPORT_VAR_PREFIX
816             << EV->getName() << ";" << std::endl;
817
818  C.startFunction(Context::AM_Public,
819                  false,
820                  "void",
821                  "bind_" + EV->getName(),
822                  1,
823                  TypeName.c_str(),
824                  "v");
825
826  C.indent() << RS_EXPORT_VAR_PREFIX << EV->getName() << " = v;" << std::endl;
827  C.indent() << "if (v == null) bindAllocation(null, "RS_EXPORT_VAR_INDEX_PREFIX
828             << EV->getName() << ");" << std::endl;
829
830  if (PointeeType->getClass() == RSExportType::ExportClassRecord)
831    C.indent() << "else bindAllocation(v.getAllocation(), "
832        RS_EXPORT_VAR_INDEX_PREFIX << EV->getName() << ");"
833               << std::endl;
834  else
835    C.indent() << "else bindAllocation(v, "RS_EXPORT_VAR_INDEX_PREFIX
836               << EV->getName() << ");" << std::endl;
837
838  C.endFunction();
839
840  genGetExportVariable(C, TypeName, EV->getName());
841
842  return;
843}
844
845void RSReflection::genVectorTypeExportVariable(Context &C,
846                                               const RSExportVar *EV) {
847  assert((EV->getType()->getClass() == RSExportType::ExportClassVector) &&
848         "Variable should be type of vector here");
849
850  const RSExportVectorType *EVT =
851      static_cast<const RSExportVectorType*>(EV->getType());
852  const char *TypeName = GetVectorTypeName(EVT);
853  const char *FieldPackerName = "fp";
854
855  C.indent() << "private " << TypeName << " "RS_EXPORT_VAR_PREFIX
856             << EV->getName() << ";" << std::endl;
857
858  // set_*()
859  if (!EV->isConst()) {
860    C.startFunction(Context::AM_Public,
861                    false,
862                    "void",
863                    "set_" + EV->getName(),
864                    1,
865                    TypeName,
866                    "v");
867    C.indent() << RS_EXPORT_VAR_PREFIX << EV->getName() << " = v;" << std::endl;
868
869    if (genCreateFieldPacker(C, EVT, FieldPackerName))
870      genPackVarOfType(C, EVT, "v", FieldPackerName);
871    C.indent() << "setVar("RS_EXPORT_VAR_INDEX_PREFIX << EV->getName() << ", "
872               << FieldPackerName << ");" << std::endl;
873
874    C.endFunction();
875  }
876
877  genGetExportVariable(C, TypeName, EV->getName());
878  return;
879}
880
881void RSReflection::genMatrixTypeExportVariable(Context &C,
882                                               const RSExportVar *EV) {
883  assert((EV->getType()->getClass() == RSExportType::ExportClassMatrix) &&
884         "Variable should be type of matrix here");
885
886  const RSExportMatrixType *EMT =
887      static_cast<const RSExportMatrixType*>(EV->getType());
888  const char *TypeName = GetMatrixTypeName(EMT);
889  const char *FieldPackerName = "fp";
890
891  C.indent() << "private " << TypeName << " "RS_EXPORT_VAR_PREFIX
892             << EV->getName() << ";" << std::endl;
893
894  // set_*()
895  if (!EV->isConst()) {
896    C.startFunction(Context::AM_Public,
897                    false,
898                    "void",
899                    "set_" + EV->getName(),
900                    1,
901                    TypeName, "v");
902    C.indent() << RS_EXPORT_VAR_PREFIX << EV->getName() << " = v;" << std::endl;
903
904    if (genCreateFieldPacker(C, EMT, FieldPackerName))
905      genPackVarOfType(C, EMT, "v", FieldPackerName);
906    C.indent() << "setVar("RS_EXPORT_VAR_INDEX_PREFIX << EV->getName() << ", "
907               << FieldPackerName << ");" << std::endl;
908
909    C.endFunction();
910  }
911
912  genGetExportVariable(C, TypeName, EV->getName());
913  return;
914}
915
916void RSReflection::genRecordTypeExportVariable(Context &C,
917                                               const RSExportVar *EV) {
918  assert((EV->getType()->getClass() == RSExportType::ExportClassRecord) &&
919         "Variable should be type of struct here");
920
921  const RSExportRecordType *ERT =
922      static_cast<const RSExportRecordType*>(EV->getType());
923  std::string TypeName =
924      RS_TYPE_CLASS_NAME_PREFIX + ERT->getName() + "."RS_TYPE_ITEM_CLASS_NAME;
925  const char *FieldPackerName = "fp";
926
927  C.indent() << "private " << TypeName << " "RS_EXPORT_VAR_PREFIX
928             << EV->getName() << ";" << std::endl;
929
930  // set_*()
931  if (!EV->isConst()) {
932    C.startFunction(Context::AM_Public,
933                    false,
934                    "void",
935                    "set_" + EV->getName(),
936                    1,
937                    TypeName.c_str(),
938                    "v");
939    C.indent() << RS_EXPORT_VAR_PREFIX << EV->getName() << " = v;" << std::endl;
940
941    if (genCreateFieldPacker(C, ERT, FieldPackerName))
942      genPackVarOfType(C, ERT, "v", FieldPackerName);
943    C.indent() << "setVar("RS_EXPORT_VAR_INDEX_PREFIX << EV->getName()
944               << ", " << FieldPackerName << ");" << std::endl;
945
946    C.endFunction();
947  }
948
949  genGetExportVariable(C, TypeName.c_str(), EV->getName());
950  return;
951}
952
953void RSReflection::genGetExportVariable(Context &C,
954                                        const std::string &TypeName,
955                                        const std::string &VarName) {
956  C.startFunction(Context::AM_Public,
957                  false,
958                  TypeName.c_str(),
959                  "get_" + VarName,
960                  0);
961
962  C.indent() << "return "RS_EXPORT_VAR_PREFIX << VarName << ";" << std::endl;
963
964  C.endFunction();
965  return;
966}
967
968/******************* Methods to generate script class /end *******************/
969
970bool RSReflection::genCreateFieldPacker(Context &C,
971                                        const RSExportType *ET,
972                                        const char *FieldPackerName) {
973  size_t AllocSize = RSExportType::GetTypeAllocSize(ET);
974  if (AllocSize > 0)
975    C.indent() << "FieldPacker " << FieldPackerName << " = new FieldPacker("
976               << AllocSize << ");" << std::endl;
977  else
978    return false;
979  return true;
980}
981
982void RSReflection::genPackVarOfType(Context &C,
983                                    const RSExportType *ET,
984                                    const char *VarName,
985                                    const char *FieldPackerName) {
986  switch (ET->getClass()) {
987    case RSExportType::ExportClassPrimitive:
988    case RSExportType::ExportClassVector: {
989      C.indent() << FieldPackerName << "."
990                 << GetPackerAPIName(
991                     static_cast<const RSExportPrimitiveType*>(ET))
992                 << "(" << VarName << ");" << std::endl;
993      break;
994    }
995    case RSExportType::ExportClassConstantArray: {
996      if (ET->getName().compare("addObj") == 0) {
997        C.indent() << FieldPackerName << "." << "addObj" << "("
998                   << VarName << ");" << std::endl;
999      } else {
1000        C.indent() << FieldPackerName << "."
1001                   << GetPackerAPIName(
1002                       static_cast<const RSExportPrimitiveType*>(ET))
1003                   << "(" << VarName << ");" << std::endl;
1004      }
1005      break;
1006    }
1007    case RSExportType::ExportClassPointer: {
1008      // Must reflect as type Allocation in Java
1009      const RSExportType *PointeeType =
1010          static_cast<const RSExportPointerType*>(ET)->getPointeeType();
1011
1012      if (PointeeType->getClass() != RSExportType::ExportClassRecord)
1013        C.indent() << FieldPackerName << ".addI32(" << VarName
1014                   << ".getPtr());" << std::endl;
1015      else
1016        C.indent() << FieldPackerName << ".addI32(" << VarName
1017                   << ".getAllocation().getPtr());" << std::endl;
1018      break;
1019    }
1020    case RSExportType::ExportClassMatrix: {
1021      C.indent() << FieldPackerName << ".addObj(" << VarName << ");"
1022                 << std::endl;
1023      break;
1024    }
1025    case RSExportType::ExportClassRecord: {
1026      const RSExportRecordType *ERT =
1027          static_cast<const RSExportRecordType*>(ET);
1028      // Relative pos from now on in field packer
1029      unsigned Pos = 0;
1030
1031      for (RSExportRecordType::const_field_iterator I = ERT->fields_begin(),
1032               E = ERT->fields_end();
1033           I != E;
1034           I++) {
1035        const RSExportRecordType::Field *F = *I;
1036        std::string FieldName;
1037        size_t FieldOffset = F->getOffsetInParent();
1038        size_t FieldStoreSize = RSExportType::GetTypeStoreSize(F->getType());
1039        size_t FieldAllocSize = RSExportType::GetTypeAllocSize(F->getType());
1040
1041        if (VarName != NULL)
1042          FieldName = VarName + ("." + F->getName());
1043        else
1044          FieldName = F->getName();
1045
1046        if (FieldOffset > Pos)
1047          C.indent() << FieldPackerName << ".skip("
1048                     << (FieldOffset - Pos) << ");" << std::endl;
1049
1050        genPackVarOfType(C, F->getType(), FieldName.c_str(), FieldPackerName);
1051
1052        // There is padding in the field type
1053        if (FieldAllocSize > FieldStoreSize)
1054            C.indent() << FieldPackerName << ".skip("
1055                       << (FieldAllocSize - FieldStoreSize)
1056                       << ");" << std::endl;
1057
1058          Pos = FieldOffset + FieldAllocSize;
1059      }
1060
1061      // There maybe some padding after the struct
1062      size_t Padding = RSExportType::GetTypeAllocSize(ERT) - Pos;
1063      if (Padding > 0)
1064        C.indent() << FieldPackerName << ".skip(" << Padding << ");"
1065                   << std::endl;
1066      break;
1067    }
1068    default: {
1069      assert(false && "Unknown class of type");
1070    }
1071  }
1072
1073  return;
1074}
1075
1076void RSReflection::genNewItemBufferIfNull(Context &C, const char *Index) {
1077  C.indent() << "if ("RS_TYPE_ITEM_BUFFER_NAME" == null) "
1078                  RS_TYPE_ITEM_BUFFER_NAME" = "
1079                    "new "RS_TYPE_ITEM_CLASS_NAME"[mType.getX() /* count */];"
1080             << std::endl;
1081  if (Index != NULL)
1082    C.indent() << "if ("RS_TYPE_ITEM_BUFFER_NAME"[" << Index << "] == null) "
1083                    RS_TYPE_ITEM_BUFFER_NAME"[" << Index << "] = "
1084                      "new "RS_TYPE_ITEM_CLASS_NAME"();" << std::endl;
1085  return;
1086}
1087
1088void RSReflection::genNewItemBufferPackerIfNull(Context &C) {
1089  C.indent() << "if ("RS_TYPE_ITEM_BUFFER_PACKER_NAME" == null) "
1090                  RS_TYPE_ITEM_BUFFER_PACKER_NAME" = "
1091                    "new FieldPacker("
1092                      RS_TYPE_ITEM_CLASS_NAME".sizeof * mType.getX()/* count */"
1093                    ");" << std::endl;
1094  return;
1095}
1096
1097/********************** Methods to generate type class  **********************/
1098bool RSReflection::genTypeClass(Context &C,
1099                                const RSExportRecordType *ERT,
1100                                std::string &ErrorMsg) {
1101  std::string ClassName = RS_TYPE_CLASS_NAME_PREFIX + ERT->getName();
1102
1103  // Open the file
1104  if (!openScriptFile(C, ClassName, ErrorMsg)) {
1105    return false;
1106  }
1107
1108  if (!C.startClass(Context::AM_Public,
1109                    false,
1110                    ClassName,
1111                    RS_TYPE_CLASS_SUPER_CLASS_NAME,
1112                    ErrorMsg))
1113    return false;
1114
1115  if (!genTypeItemClass(C, ERT, ErrorMsg))
1116    return false;
1117
1118  // Declare item buffer and item buffer packer
1119  C.indent() << "private "RS_TYPE_ITEM_CLASS_NAME" "RS_TYPE_ITEM_BUFFER_NAME"[]"
1120      ";" << std::endl;
1121  C.indent() << "private FieldPacker "RS_TYPE_ITEM_BUFFER_PACKER_NAME";"
1122             << std::endl;
1123
1124  genTypeClassConstructor(C, ERT);
1125  genTypeClassCopyToArray(C, ERT);
1126  genTypeClassItemSetter(C, ERT);
1127  genTypeClassItemGetter(C, ERT);
1128  genTypeClassComponentSetter(C, ERT);
1129  genTypeClassComponentGetter(C, ERT);
1130  genTypeClassCopyAll(C, ERT);
1131
1132  C.endClass();
1133
1134  C.resetFieldIndex();
1135  C.clearFieldIndexMap();
1136
1137  return true;
1138}
1139
1140bool RSReflection::genTypeItemClass(Context &C,
1141                                    const RSExportRecordType *ERT,
1142                                    std::string &ErrorMsg) {
1143  C.indent() << "static public class "RS_TYPE_ITEM_CLASS_NAME;
1144  C.startBlock();
1145
1146  C.indent() << "public static final int sizeof = "
1147             << RSExportType::GetTypeAllocSize(ERT) << ";" << std::endl;
1148
1149  // Member elements
1150  C.out() << std::endl;
1151  for (RSExportRecordType::const_field_iterator FI = ERT->fields_begin(),
1152           FE = ERT->fields_end();
1153       FI != FE;
1154       FI++) {
1155    C.indent() << GetTypeName((*FI)->getType()) << " " << (*FI)->getName()
1156               << ";" << std::endl;
1157  }
1158
1159  // Constructor
1160  C.out() << std::endl;
1161  C.indent() << RS_TYPE_ITEM_CLASS_NAME"()";
1162  C.startBlock();
1163
1164  for (RSExportRecordType::const_field_iterator FI = ERT->fields_begin(),
1165           FE = ERT->fields_end();
1166       FI != FE;
1167       FI++) {
1168    const RSExportRecordType::Field *F = *FI;
1169    if ((F->getType()->getClass() == RSExportType::ExportClassVector) ||
1170        (F->getType()->getClass() == RSExportType::ExportClassMatrix) ||
1171        (F->getType()->getClass() == RSExportType::ExportClassRecord) ||
1172        (F->getType()->getClass() == RSExportType::ExportClassConstantArray)) {
1173      C.indent() << F->getName() << " = new " << GetTypeName(F->getType())
1174                 << "();" << std::endl;
1175    }
1176  }
1177
1178  // end Constructor
1179  C.endBlock();
1180
1181  // end Item class
1182  C.endBlock();
1183
1184  return true;
1185}
1186
1187void RSReflection::genTypeClassConstructor(Context &C,
1188                                           const RSExportRecordType *ERT) {
1189  const char *RenderScriptVar = "rs";
1190
1191  C.startFunction(Context::AM_Public,
1192                  true,
1193                  "Element",
1194                  "createElement",
1195                  1,
1196                  "RenderScript",
1197                  RenderScriptVar);
1198  genBuildElement(C, ERT, RenderScriptVar);
1199  C.endFunction();
1200
1201  C.startFunction(Context::AM_Public,
1202                  false,
1203                  NULL,
1204                  C.getClassName(),
1205                  2,
1206                  "RenderScript",
1207                  RenderScriptVar,
1208                  "int",
1209                  "count");
1210
1211  C.indent() << RS_TYPE_ITEM_BUFFER_NAME" = null;" << std::endl;
1212  C.indent() << RS_TYPE_ITEM_BUFFER_PACKER_NAME" = null;" << std::endl;
1213  C.indent() << "mElement = createElement(" << RenderScriptVar << ");"
1214             << std::endl;
1215  // Call init() in super class
1216  C.indent() << "init(" << RenderScriptVar << ", count);" << std::endl;
1217  C.endFunction();
1218
1219  return;
1220}
1221
1222void RSReflection::genTypeClassCopyToArray(Context &C,
1223                                           const RSExportRecordType *ERT) {
1224  C.startFunction(Context::AM_Private,
1225                  false,
1226                  "void",
1227                  "copyToArray",
1228                  2,
1229                  RS_TYPE_ITEM_CLASS_NAME,
1230                  "i",
1231                  "int",
1232                  "index");
1233
1234  genNewItemBufferPackerIfNull(C);
1235  C.indent() << RS_TYPE_ITEM_BUFFER_PACKER_NAME
1236                ".reset(index * "RS_TYPE_ITEM_CLASS_NAME".sizeof);"
1237             << std::endl;
1238
1239  genPackVarOfType(C, ERT, "i", RS_TYPE_ITEM_BUFFER_PACKER_NAME);
1240
1241  C.endFunction();
1242  return;
1243}
1244
1245void RSReflection::genTypeClassItemSetter(Context &C,
1246                                          const RSExportRecordType *ERT) {
1247  C.startFunction(Context::AM_Public,
1248                  false,
1249                  "void",
1250                  "set",
1251                  3,
1252                  RS_TYPE_ITEM_CLASS_NAME,
1253                  "i",
1254                  "int",
1255                  "index",
1256                  "boolean",
1257                  "copyNow");
1258  genNewItemBufferIfNull(C, NULL);
1259  C.indent() << RS_TYPE_ITEM_BUFFER_NAME"[index] = i;" << std::endl;
1260
1261  C.indent() << "if (copyNow) ";
1262  C.startBlock();
1263
1264  C.indent() << "copyToArray(i, index);" << std::endl;
1265  C.indent() << "mAllocation.subData1D(index, 1, "
1266                  RS_TYPE_ITEM_BUFFER_PACKER_NAME".getData());" << std::endl;
1267
1268  // End of if (copyNow)
1269  C.endBlock();
1270
1271  C.endFunction();
1272  return;
1273}
1274
1275void RSReflection::genTypeClassItemGetter(Context &C,
1276                                          const RSExportRecordType *ERT) {
1277  C.startFunction(Context::AM_Public,
1278                  false,
1279                  RS_TYPE_ITEM_CLASS_NAME,
1280                  "get",
1281                  1,
1282                  "int",
1283                  "index");
1284  C.indent() << "if ("RS_TYPE_ITEM_BUFFER_NAME" == null) return null;"
1285             << std::endl;
1286  C.indent() << "return "RS_TYPE_ITEM_BUFFER_NAME"[index];" << std::endl;
1287  C.endFunction();
1288  return;
1289}
1290
1291void RSReflection::genTypeClassComponentSetter(Context &C,
1292                                               const RSExportRecordType *ERT) {
1293  for (RSExportRecordType::const_field_iterator FI = ERT->fields_begin(),
1294           FE = ERT->fields_end();
1295       FI != FE;
1296       FI++) {
1297    const RSExportRecordType::Field *F = *FI;
1298    size_t FieldOffset = F->getOffsetInParent();
1299    size_t FieldStoreSize = RSExportType::GetTypeStoreSize(F->getType());
1300    unsigned FieldIndex = C.getFieldIndex(F);
1301
1302    C.startFunction(Context::AM_Public,
1303                    false,
1304                    "void",
1305                    "set_" + F->getName(), 3,
1306                    "int",
1307                    "index",
1308                    GetTypeName(F->getType()).c_str(),
1309                    "v",
1310                    "boolean",
1311                    "copyNow");
1312    genNewItemBufferPackerIfNull(C);
1313    genNewItemBufferIfNull(C, "index");
1314    C.indent() << RS_TYPE_ITEM_BUFFER_NAME"[index]." << F->getName()
1315               << " = v;" << std::endl;
1316
1317    C.indent() << "if (copyNow) ";
1318    C.startBlock();
1319
1320    if (FieldOffset > 0)
1321      C.indent() << RS_TYPE_ITEM_BUFFER_PACKER_NAME
1322                    ".reset(index * "RS_TYPE_ITEM_CLASS_NAME".sizeof + "
1323                 << FieldOffset << ");" << std::endl;
1324    else
1325      C.indent() << RS_TYPE_ITEM_BUFFER_PACKER_NAME
1326                    ".reset(index * "RS_TYPE_ITEM_CLASS_NAME".sizeof);"
1327                 << std::endl;
1328    genPackVarOfType(C, F->getType(), "v", RS_TYPE_ITEM_BUFFER_PACKER_NAME);
1329
1330    C.indent() << "FieldPacker fp = new FieldPacker(" << FieldStoreSize << ");"
1331               << std::endl;
1332    genPackVarOfType(C, F->getType(), "v", "fp");
1333    C.indent() << "mAllocation.subElementData(index, " << FieldIndex << ", fp);"
1334               << std::endl;
1335
1336    // End of if (copyNow)
1337    C.endBlock();
1338
1339    C.endFunction();
1340  }
1341  return;
1342}
1343
1344void RSReflection::genTypeClassComponentGetter(Context &C,
1345                                               const RSExportRecordType *ERT) {
1346  for (RSExportRecordType::const_field_iterator FI = ERT->fields_begin(),
1347           FE = ERT->fields_end();
1348       FI != FE;
1349       FI++) {
1350    const RSExportRecordType::Field *F = *FI;
1351    C.startFunction(Context::AM_Public,
1352                    false,
1353                    GetTypeName(F->getType()).c_str(),
1354                    "get_" + F->getName(),
1355                    1,
1356                    "int",
1357                    "index");
1358    C.indent() << "return "RS_TYPE_ITEM_BUFFER_NAME"[index]." << F->getName()
1359               << ";" << std::endl;
1360    C.endFunction();
1361  }
1362  return;
1363}
1364
1365void RSReflection::genTypeClassCopyAll(Context &C,
1366                                       const RSExportRecordType *ERT) {
1367  C.startFunction(Context::AM_Public, false, "void", "copyAll", 0);
1368
1369  C.indent() << "for (int ct=0; ct < "RS_TYPE_ITEM_BUFFER_NAME".length; ct++) "
1370      "copyToArray("RS_TYPE_ITEM_BUFFER_NAME"[ct], ct);"
1371             << std::endl;
1372  C.indent() << "mAllocation.data("RS_TYPE_ITEM_BUFFER_PACKER_NAME".getData());"
1373             << std::endl;
1374
1375  C.endFunction();
1376  return;
1377}
1378
1379/******************** Methods to generate type class /end ********************/
1380
1381/********** Methods to create Element in Java of given record type ***********/
1382void RSReflection::genBuildElement(Context &C, const RSExportRecordType *ERT,
1383                                   const char *RenderScriptVar) {
1384  const char *ElementBuilderName = "eb";
1385
1386  // Create element builder
1387  //    C.startBlock(true);
1388
1389  C.indent() << "Element.Builder " << ElementBuilderName << " = "
1390      "new Element.Builder(" << RenderScriptVar << ");" << std::endl;
1391
1392  // eb.add(...)
1393  genAddElementToElementBuilder(C,
1394                                ERT,
1395                                "",
1396                                ElementBuilderName,
1397                                RenderScriptVar);
1398
1399  C.indent() << "return " << ElementBuilderName << ".create();" << std::endl;
1400
1401  //   C.endBlock();
1402  return;
1403}
1404
1405#define EB_ADD(x)                                                   \
1406  C.indent() << ElementBuilderName                                  \
1407             << ".add(Element." << x << ", \"" << VarName << "\");" \
1408             << std::endl;                                          \
1409  C.incFieldIndex()
1410
1411void RSReflection::genAddElementToElementBuilder(Context &C,
1412                                                 const RSExportType *ET,
1413                                                 const std::string &VarName,
1414                                                 const char *ElementBuilderName,
1415                                                 const char *RenderScriptVar) {
1416  const char *ElementConstruct = GetBuiltinElementConstruct(ET);
1417
1418  if (ElementConstruct != NULL) {
1419    EB_ADD(ElementConstruct << "(" << RenderScriptVar << ")");
1420  } else {
1421    if ((ET->getClass() == RSExportType::ExportClassPrimitive) ||
1422        (ET->getClass() == RSExportType::ExportClassVector)    ||
1423        (ET->getClass() == RSExportType::ExportClassConstantArray)) {
1424      const RSExportPrimitiveType *EPT =
1425          static_cast<const RSExportPrimitiveType*>(ET);
1426      const char *DataKindName = GetElementDataKindName(EPT->getKind());
1427      const char *DataTypeName = GetElementDataTypeName(EPT->getType());
1428      int Size = (ET->getClass() == RSExportType::ExportClassVector) ?
1429          static_cast<const RSExportVectorType*>(ET)->getNumElement() :
1430          1;
1431
1432      switch (EPT->getKind()) {
1433        case RSExportPrimitiveType::DataKindPixelL:
1434        case RSExportPrimitiveType::DataKindPixelA:
1435        case RSExportPrimitiveType::DataKindPixelLA:
1436        case RSExportPrimitiveType::DataKindPixelRGB:
1437        case RSExportPrimitiveType::DataKindPixelRGBA: {
1438          // Element.createPixel()
1439          EB_ADD("createPixel(" << RenderScriptVar << ", "
1440                                << DataTypeName << ", "
1441                                << DataKindName << ")");
1442          break;
1443        }
1444        case RSExportPrimitiveType::DataKindUser:
1445        default: {
1446          if (EPT->getClass() == RSExportType::ExportClassPrimitive ||
1447              EPT->getClass() == RSExportType::ExportClassConstantArray) {
1448            // Element.createUser()
1449            EB_ADD("createUser(" << RenderScriptVar << ", "
1450                                 << DataTypeName << ")");
1451          } else {
1452            assert((ET->getClass() == RSExportType::ExportClassVector) &&
1453                   "Unexpected type.");
1454            EB_ADD("createVector(" << RenderScriptVar << ", "
1455                                   << DataTypeName << ", "
1456                                   << Size << ")");
1457          }
1458          break;
1459        }
1460      }
1461#ifndef NDEBUG
1462    } else if (ET->getClass() == RSExportType::ExportClassPointer) {
1463      // Pointer type variable should be resolved in
1464      // GetBuiltinElementConstruct()
1465      assert(false && "??");
1466    } else if (ET->getClass() == RSExportType::ExportClassMatrix) {
1467      // Matrix type variable should be resolved
1468      // in GetBuiltinElementConstruct()
1469      assert(false && "??");
1470#endif
1471    } else if (ET->getClass() == RSExportType::ExportClassRecord) {
1472      // Simalar to case of RSExportType::ExportClassRecord in genPackVarOfType.
1473      //
1474      // TODO(zonr): Generalize these two function such that there's no
1475      //             duplicated codes.
1476      const RSExportRecordType *ERT =
1477          static_cast<const RSExportRecordType*>(ET);
1478      int Pos = 0;    // relative pos from now on
1479
1480      for (RSExportRecordType::const_field_iterator I = ERT->fields_begin(),
1481               E = ERT->fields_end();
1482           I != E;
1483           I++) {
1484        const RSExportRecordType::Field *F = *I;
1485        std::string FieldName;
1486        int FieldOffset = F->getOffsetInParent();
1487        int FieldStoreSize = RSExportType::GetTypeStoreSize(F->getType());
1488        int FieldAllocSize = RSExportType::GetTypeAllocSize(F->getType());
1489
1490        if (!VarName.empty())
1491          FieldName = VarName + "." + F->getName();
1492        else
1493          FieldName = F->getName();
1494
1495        // Alignment
1496        genAddPaddingToElementBuiler(C,
1497                                     (FieldOffset - Pos),
1498                                     ElementBuilderName,
1499                                     RenderScriptVar);
1500
1501        // eb.add(...)
1502        C.addFieldIndexMapping(F);
1503        genAddElementToElementBuilder(C,
1504                                      F->getType(),
1505                                      FieldName,
1506                                      ElementBuilderName,
1507                                      RenderScriptVar);
1508
1509        // There is padding within the field type
1510        genAddPaddingToElementBuiler(C,
1511                                     (FieldAllocSize - FieldStoreSize),
1512                                     ElementBuilderName,
1513                                     RenderScriptVar);
1514
1515        Pos = FieldOffset + FieldAllocSize;
1516      }
1517
1518      // There maybe some padding after the struct
1519      //unsigned char align = RSExportType::GetTypeAlignment(ERT);
1520      //size_t siz = RSExportType::GetTypeAllocSize(ERT);
1521      size_t siz1 = RSExportType::GetTypeStoreSize(ERT);
1522
1523      genAddPaddingToElementBuiler(C,
1524                                   siz1 - Pos,
1525                                   ElementBuilderName,
1526                                   RenderScriptVar);
1527    } else {
1528      assert(false && "Unknown class of type");
1529    }
1530  }
1531}
1532
1533void RSReflection::genAddPaddingToElementBuiler(Context &C,
1534                                                int PaddingSize,
1535                                                const char *ElementBuilderName,
1536                                                const char *RenderScriptVar) {
1537  while (PaddingSize > 0) {
1538    const std::string &VarName = C.createPaddingField();
1539    if (PaddingSize >= 4) {
1540      EB_ADD("U32(" << RenderScriptVar << ")");
1541      PaddingSize -= 4;
1542    } else if (PaddingSize >= 2) {
1543      EB_ADD("U16(" << RenderScriptVar << ")");
1544      PaddingSize -= 2;
1545    } else if (PaddingSize >= 1) {
1546      EB_ADD("U8(" << RenderScriptVar << ")");
1547      PaddingSize -= 1;
1548    }
1549  }
1550  return;
1551}
1552
1553#undef EB_ADD
1554/******** Methods to create Element in Java of given record type /end ********/
1555
1556bool RSReflection::reflect(const char *OutputPackageName,
1557                           const std::string &InputFileName,
1558                           const std::string &OutputBCFileName) {
1559  Context *C = NULL;
1560  std::string ResourceId = "";
1561
1562  if (!GetClassNameFromFileName(OutputBCFileName, ResourceId))
1563    return false;
1564
1565  if (ResourceId.empty())
1566    ResourceId = "<Resource ID>";
1567
1568  if ((OutputPackageName == NULL) ||
1569      (*OutputPackageName == '\0') ||
1570      strcmp(OutputPackageName, "-") == 0)
1571    C = new Context(InputFileName, "<Package Name>", ResourceId, true);
1572  else
1573    C = new Context(InputFileName, OutputPackageName, ResourceId, false);
1574
1575  if (C != NULL) {
1576    std::string ErrorMsg, ScriptClassName;
1577    // class ScriptC_<ScriptName>
1578    if (!GetClassNameFromFileName(InputFileName, ScriptClassName))
1579      return false;
1580
1581    if (ScriptClassName.empty())
1582      ScriptClassName = "<Input Script Name>";
1583
1584    ScriptClassName.insert(0, RS_SCRIPT_CLASS_NAME_PREFIX);
1585
1586    if (mRSContext->getLicenseNote() != NULL) {
1587      C->setLicenseNote(*(mRSContext->getLicenseNote()));
1588    }
1589
1590    if (!genScriptClass(*C, ScriptClassName, ErrorMsg)) {
1591      std::cerr << "Failed to generate class " << ScriptClassName << " ("
1592                << ErrorMsg << ")" << std::endl;
1593      return false;
1594    }
1595
1596    // class ScriptField_<TypeName>
1597    for (RSContext::const_export_type_iterator TI =
1598             mRSContext->export_types_begin(),
1599             TE = mRSContext->export_types_end();
1600         TI != TE;
1601         TI++) {
1602      const RSExportType *ET = TI->getValue();
1603
1604      if (ET->getClass() == RSExportType::ExportClassRecord) {
1605        const RSExportRecordType *ERT =
1606            static_cast<const RSExportRecordType*>(ET);
1607
1608        if (!ERT->isArtificial() && !genTypeClass(*C, ERT, ErrorMsg)) {
1609          std::cerr << "Failed to generate type class for struct '"
1610                    << ERT->getName() << "' (" << ErrorMsg << ")" << std::endl;
1611          return false;
1612        }
1613      }
1614    }
1615  }
1616
1617  return true;
1618}
1619
1620/************************** RSReflection::Context **************************/
1621const char *const RSReflection::Context::ApacheLicenseNote =
1622    "/*\n"
1623    " * Copyright (C) 2010 The Android Open Source Project\n"
1624    " *\n"
1625    " * Licensed under the Apache License, Version 2.0 (the \"License\");\n"
1626    " * you may not use this file except in compliance with the License.\n"
1627    " * You may obtain a copy of the License at\n"
1628    " *\n"
1629    " *      http://www.apache.org/licenses/LICENSE-2.0\n"
1630    " *\n"
1631    " * Unless required by applicable law or agreed to in writing, software\n"
1632    " * distributed under the License is distributed on an \"AS IS\" BASIS,\n"
1633    " * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or "
1634    "implied.\n"
1635    " * See the License for the specific language governing permissions and\n"
1636    " * limitations under the License.\n"
1637    " */\n"
1638    "\n";
1639
1640const char *const RSReflection::Context::Import[] = {
1641  // RenderScript java class
1642  "android.renderscript.*",
1643  // Import R
1644  "android.content.res.Resources",
1645  // Import for debugging
1646  "android.util.Log",
1647};
1648
1649const char *RSReflection::Context::AccessModifierStr(AccessModifier AM) {
1650  switch (AM) {
1651    case AM_Public: return "public"; break;
1652    case AM_Protected: return "protected"; break;
1653    case AM_Private: return "private"; break;
1654    default: return ""; break;
1655  }
1656}
1657
1658bool RSReflection::Context::startClass(AccessModifier AM,
1659                                       bool IsStatic,
1660                                       const std::string &ClassName,
1661                                       const char *SuperClassName,
1662                                       std::string &ErrorMsg) {
1663  if (mVerbose)
1664    std::cout << "Generating " << ClassName << ".java ..." << std::endl;
1665
1666  // License
1667  out() << mLicenseNote;
1668
1669  // Notice of generated file
1670  out() << "/*" << std::endl;
1671  out() << " * This file is auto-generated. DO NOT MODIFY!" << std::endl;
1672  out() << " * The source RenderScript file: " << mInputRSFile << std::endl;
1673  out() << " */" << std::endl;
1674
1675  // Package
1676  if (!mPackageName.empty())
1677    out() << "package " << mPackageName << ";" << std::endl;
1678  out() << std::endl;
1679
1680  // Imports
1681  for (unsigned i = 0;i < (sizeof(Import) / sizeof(const char*)); i++)
1682    out() << "import " << Import[i] << ";" << std::endl;
1683  out() << std::endl;
1684
1685  // All reflected classes should be annotated as hidden, so that they won't
1686  // be exposed in SDK.
1687  out() << "/**" << std::endl;
1688  out() << " * @hide" << std::endl;
1689  out() << " */" << std::endl;
1690
1691  out() << AccessModifierStr(AM) << ((IsStatic) ? " static" : "") << " class "
1692        << ClassName;
1693  if (SuperClassName != NULL)
1694    out() << " extends " << SuperClassName;
1695
1696  startBlock();
1697
1698  mClassName = ClassName;
1699
1700  return true;
1701}
1702
1703void RSReflection::Context::endClass() {
1704  endBlock();
1705  if (!mUseStdout)
1706    mOF.close();
1707  clear();
1708  return;
1709}
1710
1711void RSReflection::Context::startBlock(bool ShouldIndent) {
1712  if (ShouldIndent)
1713    indent() << "{" << std::endl;
1714  else
1715    out() << " {" << std::endl;
1716  incIndentLevel();
1717  return;
1718}
1719
1720void RSReflection::Context::endBlock() {
1721  decIndentLevel();
1722  indent() << "}" << std::endl << std::endl;
1723  return;
1724}
1725
1726void RSReflection::Context::startTypeClass(const std::string &ClassName) {
1727  indent() << "public static class " << ClassName;
1728  startBlock();
1729  return;
1730}
1731
1732void RSReflection::Context::endTypeClass() {
1733  endBlock();
1734  return;
1735}
1736
1737void RSReflection::Context::startFunction(AccessModifier AM,
1738                                          bool IsStatic,
1739                                          const char *ReturnType,
1740                                          const std::string &FunctionName,
1741                                          int Argc, ...) {
1742  ArgTy Args;
1743  va_list vl;
1744  va_start(vl, Argc);
1745
1746  for (int i = 0; i < Argc; i++) {
1747    const char *ArgType = va_arg(vl, const char*);
1748    const char *ArgName = va_arg(vl, const char*);
1749
1750    Args.push_back(std::make_pair(ArgType, ArgName));
1751  }
1752  va_end(vl);
1753
1754  startFunction(AM, IsStatic, ReturnType, FunctionName, Args);
1755
1756  return;
1757}
1758
1759void RSReflection::Context::startFunction(AccessModifier AM,
1760                                          bool IsStatic,
1761                                          const char *ReturnType,
1762                                          const std::string &FunctionName,
1763                                          const ArgTy &Args) {
1764  indent() << AccessModifierStr(AM) << ((IsStatic) ? " static " : " ")
1765           << ((ReturnType) ? ReturnType : "") << " " << FunctionName << "(";
1766
1767  bool FirstArg = true;
1768  for (ArgTy::const_iterator I = Args.begin(), E = Args.end();
1769       I != E;
1770       I++) {
1771    if (!FirstArg)
1772      out() << ", ";
1773    else
1774      FirstArg = false;
1775
1776    out() << I->first << " " << I->second;
1777  }
1778
1779  out() << ")";
1780  startBlock();
1781
1782  return;
1783}
1784
1785void RSReflection::Context::endFunction() {
1786  endBlock();
1787  return;
1788}
1789