private static void generateStringDeallocations(
PrintWriter writer, Collection<ParameterDeclaration> params) {
for (ParameterDeclaration param : params) {
if (Utils.getJavaType(param.getType()).equals(String.class)
&& param.getAnnotation(Result.class) == null)
writer.println("\tfree(" + param.getSimpleName() + BUFFER_ADDRESS_POSTFIX + ");");
else if (param.getAnnotation(StringList.class) != null) // Free the string array mem
writer.println("\tfree(" + param.getSimpleName() + STRING_LIST_POSTFIX + ");");
}
}
private static void generateParameter(PrintWriter writer, ParameterDeclaration param, Mode mode) {
writer.print(", ");
if (mode == Mode.BUFFEROBJECT && param.getAnnotation(BufferObject.class) != null) {
writer.print("jlong " + param.getSimpleName() + Utils.BUFFER_OBJECT_PARAMETER_POSTFIX);
} else if (param.getAnnotation(GLpointer.class) != null) {
writer.print("jlong " + param.getSimpleName());
} else {
JNITypeTranslator translator = new JNITypeTranslator();
param.getType().accept(translator);
writer.print(translator.getSignature() + " " + param.getSimpleName());
if (Utils.getNIOBufferType(param.getType()) != null)
writer.print(", jint " + param.getSimpleName() + BUFFER_POSITION_POSTFIX);
}
}
private static void generateStringListInits(
PrintWriter writer, Collection<ParameterDeclaration> params) {
for (ParameterDeclaration param : params) {
StringList stringList_annotation = param.getAnnotation(StringList.class);
if (stringList_annotation != null) {
String lengths = stringList_annotation.lengths();
// Init vars
writer.println("\t_str_i = 0;");
writer.println(
"\t_str_address = (GLchar *)" + param.getSimpleName() + BUFFER_ADDRESS_POSTFIX + ";");
// Fill string array with the string pointers
writer.println("\twhile ( _str_i < " + stringList_annotation.value() + " ) {");
if (lengths.length() == 0) {
writer.println(
"\t\t" + param.getSimpleName() + STRING_LIST_POSTFIX + "[_str_i++] = _str_address;");
writer.println("\t\t_str_address += strlen(_str_address) + 1;");
} else {
writer.println(
"\t\t" + param.getSimpleName() + STRING_LIST_POSTFIX + "[_str_i] = _str_address;");
writer.println("\t\t_str_address += " + lengths + BUFFER_ADDRESS_POSTFIX + "[_str_i++];");
}
writer.println("\t}");
}
}
}
private static void generateCallParameters(
PrintWriter writer, TypeMap type_map, Collection<ParameterDeclaration> params) {
if (params.size() > 0) {
Iterator<ParameterDeclaration> it = params.iterator();
generateCallParameter(writer, type_map, it.next());
while (it.hasNext()) {
writer.print(", ");
generateCallParameter(writer, type_map, it.next());
}
}
}
private static void generateCallParameter(
PrintWriter writer, TypeMap type_map, ParameterDeclaration param) {
boolean is_indirect = param.getAnnotation(Indirect.class) != null;
if (is_indirect || param.getAnnotation(StringList.class) != null) {
writer.print("(");
NativeTypeTranslator translator = new NativeTypeTranslator(type_map, param);
param.getType().accept(translator);
writer.print(translator.getSignature());
writer.print("*)");
}
if (param.getAnnotation(GLpointer.class) != null)
writer.print("(" + param.getAnnotation(GLpointer.class).value() + ")(intptr_t)");
if (param.getAnnotation(Result.class) != null || is_indirect) writer.print("&");
if (param.getAnnotation(Result.class) != null) {
writer.print(Utils.RESULT_VAR_NAME);
} else {
writer.print(param.getSimpleName());
if (param.getAnnotation(StringList.class) != null) writer.print(STRING_LIST_POSTFIX);
else if (Utils.isAddressableType(param.getType())) writer.print(BUFFER_ADDRESS_POSTFIX);
}
}
public void process() {
for (TypeDeclaration typeDecl : env.getSpecifiedTypeDeclarations()) {
ExtractInterface annot = typeDecl.getAnnotation(ExtractInterface.class);
if (annot == null) break;
for (MethodDeclaration m : typeDecl.getMethods())
if (m.getModifiers().contains(Modifier.PUBLIC)
&& !(m.getModifiers().contains(Modifier.STATIC))) interfaceMethods.add(m);
if (interfaceMethods.size() > 0) {
try {
PrintWriter writer = env.getFiler().createSourceFile(annot.value());
writer.println("package " + typeDecl.getPackage().getQualifiedName() + ";");
writer.println("public interface " + annot.value() + " {");
for (MethodDeclaration m : interfaceMethods) {
writer.print(" public ");
writer.print(m.getReturnType() + " ");
writer.print(m.getSimpleName() + " (");
int i = 0;
for (ParameterDeclaration parm : m.getParameters()) {
writer.print(parm.getType() + " " + parm.getSimpleName());
if (++i < m.getParameters().size()) writer.print(", ");
}
writer.println(");");
}
writer.println("}");
writer.close();
} catch (IOException ioe) {
throw new RuntimeException(ioe);
}
}
}
}
private static boolean generateBufferParameterAddress(
TypeMap type_map,
PrintWriter writer,
MethodDeclaration method,
ParameterDeclaration param,
Mode mode,
boolean loopDeclared) {
NativeTypeTranslator translator = new NativeTypeTranslator(type_map, param);
param.getType().accept(translator);
writer.print("\t" + translator.getSignature() + param.getSimpleName());
writer.print(BUFFER_ADDRESS_POSTFIX + " = ((");
writer.print(translator.getSignature());
Check check_annotation = param.getAnnotation(Check.class);
writer.print(")");
if (mode == Mode.BUFFEROBJECT && param.getAnnotation(BufferObject.class) != null) {
writer.print(
"offsetToPointer("
+ param.getSimpleName()
+ Utils.BUFFER_OBJECT_PARAMETER_POSTFIX
+ "))");
} else {
Class java_type = Utils.getJavaType(param.getType());
if (Buffer.class.isAssignableFrom(java_type)
|| java_type.equals(CharSequence.class)
|| java_type.equals(CharSequence[].class)) {
boolean explicitly_byte_sized =
java_type.equals(Buffer.class)
|| translator.getAnnotationType().equals(type_map.getVoidType());
if (explicitly_byte_sized) writer.print("(((char *)");
if (method.getAnnotation(GenerateAutos.class) != null
|| (check_annotation != null && check_annotation.canBeNull())) {
writer.print("safeGetBufferAddress(env, " + param.getSimpleName());
} else {
writer.print("(*env)->GetDirectBufferAddress(env, " + param.getSimpleName());
}
writer.print("))");
writer.print(" + " + param.getSimpleName() + BUFFER_POSITION_POSTFIX);
if (explicitly_byte_sized) writer.print("))");
} else if (java_type.equals(String.class)) {
writer.print("GetStringNativeChars(env, " + param.getSimpleName() + "))");
} else throw new RuntimeException("Illegal type " + java_type);
}
writer.println(";");
if (param.getAnnotation(StringList.class) != null) {
if (Utils.getJavaType(param.getType()) != CharSequence[].class
&& (param.getAnnotation(GLchar.class) == null
|| param.getAnnotation(NullTerminated.class) == null
|| param.getAnnotation(NullTerminated.class).value().length() == 0))
throw new RuntimeException(
"StringList annotation can only be applied on null-terminated GLchar buffers.");
if ("_str".equals(param.getSimpleName()))
throw new RuntimeException(
"The name '_str' is not valid for arguments annotated with StringList");
// Declare loop counters and allocate string array
if (!loopDeclared) {
writer.println("\tunsigned int _str_i;");
writer.println("\tGLchar *_str_address;");
loopDeclared = true;
}
writer.println(
"\tGLchar **"
+ param.getSimpleName()
+ STRING_LIST_POSTFIX
+ " = (GLchar **) malloc("
+ param.getAnnotation(StringList.class).value()
+ "*sizeof(GLchar*));");
}
return loopDeclared;
}
private static void generateMethodStub(
AnnotationProcessorEnvironment env,
TypeMap type_map,
PrintWriter writer,
String interface_name,
MethodDeclaration method,
Mode mode,
boolean generate_error_checks,
boolean context_specific) {
if (!context_specific && method.getAnnotation(Alternate.class) == null) writer.print("static ");
else writer.print("JNIEXPORT ");
TypeMirror result_type = Utils.getMethodReturnType(method);
if (method.getAnnotation(GLpointer.class) != null) {
writer.print("jlong");
} else {
JNITypeTranslator translator = new JNITypeTranslator();
result_type.accept(translator);
writer.print(translator.getSignature());
}
writer.print(" JNICALL ");
writer.print(
Utils.getQualifiedNativeMethodName(
interface_name, method, generate_error_checks, context_specific));
if (mode == Mode.BUFFEROBJECT) writer.print(Utils.BUFFER_OBJECT_METHOD_POSTFIX);
writer.print("(JNIEnv *env, jclass clazz");
generateParameters(writer, method.getParameters(), mode);
if (Utils.getNIOBufferType(result_type) != null) {
CachedResult cached_result_annotation = method.getAnnotation(CachedResult.class);
if (cached_result_annotation == null || !cached_result_annotation.isRange())
writer.print(", jlong " + Utils.RESULT_SIZE_NAME);
if (cached_result_annotation != null) writer.print(", jobject " + Utils.CACHED_BUFFER_NAME);
}
if (context_specific) {
writer.print(", jlong " + Utils.FUNCTION_POINTER_VAR_NAME);
}
writer.println(") {");
generateBufferParameterAddresses(type_map, writer, method, mode);
Alternate alt_annotation = method.getAnnotation(Alternate.class);
if (context_specific) {
String typedef_name = Utils.getTypedefName(method);
writer.print(
"\t"
+ typedef_name
+ " "
+ (alt_annotation == null ? method.getSimpleName() : alt_annotation.value()));
writer.print(" = (" + typedef_name + ")((intptr_t)");
writer.println(Utils.FUNCTION_POINTER_VAR_NAME + ");");
}
generateStringListInits(writer, method.getParameters());
writer.print("\t");
if (!result_type.equals(env.getTypeUtils().getVoidType())) {
Declaration return_declaration;
ParameterDeclaration result_param = Utils.getResultParameter(method);
if (result_param != null) return_declaration = result_param;
else return_declaration = method;
NativeTypeTranslator native_translator =
new NativeTypeTranslator(type_map, return_declaration);
result_type.accept(native_translator);
writer.print(native_translator.getSignature() + " " + Utils.RESULT_VAR_NAME);
if (result_param != null) {
writer.println(";");
writer.print("\t");
} else writer.print(" = ");
}
writer.print((alt_annotation == null ? method.getSimpleName() : alt_annotation.value()) + "(");
generateCallParameters(writer, type_map, method.getParameters());
writer.print(")");
writer.println(";");
generateStringDeallocations(writer, method.getParameters());
if (!result_type.equals(env.getTypeUtils().getVoidType())) {
writer.print("\treturn ");
Class java_result_type = Utils.getJavaType(result_type);
if (Buffer.class.isAssignableFrom(java_result_type)) {
if (method.getAnnotation(CachedResult.class) != null)
writer.print("safeNewBufferCached(env, ");
else writer.print("safeNewBuffer(env, ");
} else if (String.class.equals(java_result_type)) {
writer.print("NewStringNativeUnsigned(env, ");
} else if (method.getAnnotation(GLpointer.class) != null) {
writer.print("(intptr_t)");
}
writer.print(Utils.RESULT_VAR_NAME);
if (Buffer.class.isAssignableFrom(java_result_type)) {
writer.print(", ");
if (method.getAnnotation(CachedResult.class) != null
&& method.getAnnotation(CachedResult.class).isRange())
Utils.printExtraCallArguments(
writer, method, method.getAnnotation(AutoResultSize.class).value());
else Utils.printExtraCallArguments(writer, method, Utils.RESULT_SIZE_NAME);
}
if (Buffer.class.isAssignableFrom(java_result_type) || String.class.equals(java_result_type))
writer.print(")");
writer.println(";");
}
writer.println("}");
writer.println();
}