ffi/java: use existing Set of callbacks per language and modules

[nit.git] / src / common_ffi / java.nit

# This file is part of NIT ( http://www.nitlanguage.org ).
#
# Copyright 2013-2014 Alexis Laferrière <alexis.laf@xymus.net>
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

# FFI support for the Java language
#
# TODO support callbacks to super and casts
module java

import extern_classes
import c
import c_compiler_options

redef class FFILanguageAssignationPhase
        var java_language: FFILanguage = new JavaLanguage(self)
end

class JavaLanguage
        super FFILanguage

        redef fun identify_language(n) do return n.is_java

        redef fun compile_module_block(block, ccu, nmodule)
        do
                nmodule.ensure_java_files
                var java_file = nmodule.java_file
                assert java_file != null

                java_file.header.add(block.code)
        end

        redef fun compile_extern_method(block, m, ccu, nmodule)
        do
                ffi_ccu = ccu
                nmodule.ensure_java_files
                var java_file = nmodule.java_file
                assert java_file != null

                var mmodule = nmodule.mmodule.as(not null)
                var mclass_type = m.parent.as(AClassdef).mclass.mclass_type
                var mmethodef = m.mpropdef
                var mproperty = m.mpropdef.mproperty

                # C function calling the Java method through JNI
                var fc = new ExternCFunction(m, mmodule)

                fc.exprs.add """
        jclass java_class;
        jmethodID java_meth_id;

        // retrieve the current JVM
        Sys sys = {{{mmodule.name}}}___Pointer_sys(NULL);
        JNIEnv *nit_ffi_jni_env = {{{mmodule.name}}}___Sys_jni_env(sys);

        // retrieve the implementation Java class
        java_class = {{{mmodule.name}}}___Sys_load_jclass(sys, "{{{mmodule.impl_java_class_name}}}");
        if (java_class == NULL) {
                fprintf(stderr, "Nit FFI with Java error: failed to load class.\\n");
                (*nit_ffi_jni_env)->ExceptionDescribe(nit_ffi_jni_env);
                exit(1);
        }

        // register callbacks (only once per Nit module)
        if (!nit_ffi_with_java_registered_natives) nit_ffi_with_java_register_natives(nit_ffi_jni_env, java_class);
"""

                # Retrieve the Java implementation function id
                var java_fun_name = mproperty.build_cname(mclass_type, mmodule, "___java_impl", long_signature)
                var jni_format = mproperty.build_jni_format(mclass_type, mmodule)
                fc.exprs.add """
        // retreive the implementation static function
        java_meth_id = (*nit_ffi_jni_env)->GetStaticMethodID(nit_ffi_jni_env, java_class, "{{{java_fun_name}}}", "{{{jni_format}}}");
        if (java_meth_id == NULL) {
                fprintf(stderr, "Nit FFI with Java error: Java implementation not found.\\n");
                (*nit_ffi_jni_env)->ExceptionDescribe(nit_ffi_jni_env);
                exit(1);
        }
"""

                # Call the C Java implementation method from C
                var signature = mmethodef.msignature
                assert signature != null

                var jni_signature_alt
                var return_type
                var c_return_type
                var params = new Array[String]
                params.add "nit_ffi_jni_env"
                params.add "java_class"
                params.add "java_meth_id"

                if mproperty.is_init then
                        jni_signature_alt = mclass_type.jni_signature_alt
                        return_type = mclass_type
                        c_return_type = mclass_type.cname
                else
                        params.add "recv"
                        if signature.return_mtype != null then
                                var ret_mtype = signature.return_mtype
                                ret_mtype = ret_mtype.resolve_for(mclass_type, mclass_type, mmodule, true)
                                return_type = signature.return_mtype
                                c_return_type = mclass_type.cname
                                jni_signature_alt = return_type.jni_signature_alt
                        else
                                jni_signature_alt = "Void"
                                return_type = null
                                c_return_type = null
                        end
                end

                for p in signature.mparameters do params.add(p.name)

                var cname = "(*nit_ffi_jni_env)->CallStatic{jni_signature_alt}Method"
                var ccall
                if return_type != null then
                        ccall = "{return_type.jni_type} jni_res = {cname}({params.join(", ")});"
                else ccall = "{cname}({params.join(", ")});"

                fc.exprs.add """
        // execute implementation code
        {{{ccall}}}
        if ((*nit_ffi_jni_env)->ExceptionCheck(nit_ffi_jni_env)) {
                fprintf(stderr, "Nit FFI with Java error: Exception after call.\\n");
                (*nit_ffi_jni_env)->ExceptionDescribe(nit_ffi_jni_env);
                exit(1);
        }
"""

                if return_type != null then
                        fc.exprs.add "\treturn {to_java_call_context.cast_from(return_type, "jni_res")};"
                end

                ccu.add_exported_function( fc )

                # Java implementation function in Java
                var java_csig = mproperty.build_csignature(mclass_type, mmodule, "___java_impl", long_signature, java_call_context)
                nmodule.java_file.class_content.add """
        public static {{{java_csig}}} {
                // from Nit FII at: {{{block.location}}}
                {{{block.code}}}
        }
"""
        end

        redef fun compile_extern_class(block, m, ccu, nmodule) do end

        redef fun get_ftype(block, m) do return new ForeignJavaType(block.code)

        redef fun compile_to_files(nmodule, compdir)
        do
                # Make sure we have a .java file
                nmodule.ensure_java_files

                # Needed compiler and linker options
                nmodule.insert_compiler_options

                # Enable linking C callbacks to java native methods
                nmodule.ensure_linking_callback_methods(ffi_ccu, nmodule.ffi_callbacks[self])

                # Java implementation code
                var java_file = nmodule.java_file
                assert java_file != null
                var extern_java_file = java_file.write_to_files(compdir)
                nmodule.ffi_files.add(extern_java_file)
        end

        var ffi_ccu: CCompilationUnit # HACK

        redef fun compile_callback(callback, nmodule, mainmodule, ccu)
        do
                ffi_ccu = ccu
                callback.compile_callback_to_java(nmodule, ccu)
        end
end

redef class AModule

        # Pure java class source file
        private var java_file: nullable JavaClassTemplate = null

        # Set up the templates of the Java implementation class
        private fun ensure_java_files
        do
                if java_file != null then return

                # Java implementation code
                java_file = new JavaClassTemplate(mmodule.impl_java_class_name)
        end

        # Compile C code to call JNI and link C callbacks implementations to Java extern methods
        private fun ensure_linking_callback_methods(ccu: CCompilationUnit, callbacks: Set[NitniCallback])
        do
                if callbacks.is_empty then
                        ccu.body_decl.add "static int nit_ffi_with_java_registered_natives = 1;\n"
                        return
                end

                ccu.body_decl.add "static int nit_ffi_with_java_registered_natives = 0;\n"

                var jni_methods = new Array[String]
                for cb in callbacks do
                        jni_methods.add_all(cb.jni_methods_declaration(mmodule.as(not null)))
                end

                var cf = new CFunction("static void nit_ffi_with_java_register_natives(JNIEnv* env, jclass jclazz)")
                cf.exprs.add """
        nit_ffi_with_java_registered_natives = 1;

        jint n_methods = {{{jni_methods.length}}};
        JNINativeMethod methods[] = {
                {{{jni_methods.join(",\n\t\t")}}}
        };
        jint res = (*env)->RegisterNatives(env, jclazz, methods, n_methods);
        if (res != JNI_OK) {
                fprintf(stderr, "RegisterNatives failed\\n");
                (*env)->ExceptionDescribe(env);
                exit(1);
        }
                """
                ccu.add_local_function cf
        end

        # Tell the C compiler where to find jni.h and how to link with libjvm
        private fun insert_compiler_options
        do
                mmodule.c_compiler_options = "{mmodule.c_compiler_options} -I $(JAVA_HOME)/include/"
                mmodule.c_linker_options = "{mmodule.c_linker_options} -L $(JNI_LIB_PATH) -ljvm"
        end
end

redef class MModule
        # Name of the generated Java class where to store all implementation methods of this module
        # as well as generated callbacks.
        private fun impl_java_class_name: String do return "NitFFIWithJava_{name}"
end

redef class AExternPropdef
        redef fun verify_nitni_callbacks(toolcontext)
        do
                super

                var block = n_extern_code_block
                if block != null and block.is_java then
                        insert_artificial_callbacks(toolcontext)
                end
        end

        # Insert additionnal explicit calls to get the current `JNIEnv`
        #
        # This forces declaration of callbacks to Nit. The callbacks will be available in Java
        # but will be used mainly by the FFI itself.
        #
        # The developper can aso customize the JNIEnv used by the FFI by redefing `Sys::jni_env`.
        private fun insert_artificial_callbacks(toolcontext: ToolContext)
        do
                var fcc = foreign_callbacks
                assert fcc != null

                var modelbuilder = toolcontext.modelbuilder
                var mmodule = mpropdef.mclassdef.mmodule

                # Pointer::sys
                var pointer_class = modelbuilder.try_get_mclass_by_name(self, mmodule, "Pointer")
                assert pointer_class != null
                var pointer_sys_meth = modelbuilder.try_get_mproperty_by_name2(self, mmodule, pointer_class.mclass_type, "sys")
                assert pointer_sys_meth != null and pointer_sys_meth isa MMethod

                var explicit_call = new MExplicitCall(pointer_class.mclass_type, pointer_sys_meth, mmodule)
                fcc.callbacks.add(explicit_call)
                explicit_call.fill_type_for(fcc, mmodule)

                # Sys::jni_env
                var sys_class = modelbuilder.try_get_mclass_by_name(self, mmodule, "Sys")
                assert sys_class != null
                var sys_jni_env_meth = modelbuilder.try_get_mproperty_by_name2(self, mmodule, sys_class.mclass_type, "jni_env")
                assert sys_jni_env_meth != null
                assert sys_jni_env_meth isa MMethod

                explicit_call = new MExplicitCall(sys_class.mclass_type, sys_jni_env_meth, mmodule)
                fcc.callbacks.add(explicit_call)
                explicit_call.fill_type_for(fcc, mmodule)

                # Sys::load_jclass
                var sys_jni_load_jclass_meth = modelbuilder.try_get_mproperty_by_name2(self, mmodule, sys_class.mclass_type, "load_jclass")
                assert sys_jni_load_jclass_meth != null
                assert sys_jni_load_jclass_meth isa MMethod

                explicit_call = new MExplicitCall(sys_class.mclass_type, sys_jni_load_jclass_meth, mmodule)
                fcc.callbacks.add(explicit_call)
                explicit_call.fill_type_for(fcc, mmodule)
        end
end

redef class AExternCodeBlock
        fun is_java : Bool do return language_name != null and
                language_name_lowered == "java"
end

# Java class source template
class JavaClassTemplate
        super Template

        var java_class_name: String
        init(name: String) do self.java_class_name = name

        var header = new Template
        var class_content = new Template

        fun write_to_files(compdir: String): ExternFile
        do
                var filename = "{java_class_name}.java"
                var filepath = "{compdir}/{filename}"

                write_to_file filepath

                return new JavaFile(filename)
        end

        redef fun rendering
        do
                add header
                add "\n"
                add "public class {java_class_name} \{\n"
                add class_content
                add "\}"
        end
end

# A generated Java source file, represent the corresponding Makefile rules
class JavaFile
        super ExternFile

        redef fun makefile_rule_name do return "{filename.basename(".java")}.class"
        redef fun makefile_rule_content do return "javac {filename} -d ."
        redef fun add_to_jar do return true
end

# Context in pure Java code
private class JavaCallContext
        super CallContext

        redef fun name_mtype(mtype) do return mtype.java_type
end

# Context in C, when call are from normal C to JNI
private class ToJavaCallContext
        super CallContext

        redef fun cast_to(mtype, name) do return "({mtype.jni_type})({name})"
        redef fun cast_from(mtype, name) do return "({mtype.cname})({name})"
        redef fun name_mtype(mtype) do return mtype.jni_type
end

# Context in C, when call are from JNI to normal C
private class FromJavaCallContext
        super CallContext

        redef fun cast_to(mtype, name) do return "({mtype.cname})({name})"
        redef fun cast_from(mtype, name) do return "({mtype.jni_type})({name})"
        redef fun name_mtype(mtype) do return mtype.jni_type
end

# Foreign type attach to Nit extern Java classes
class ForeignJavaType
        super ForeignType

        var java_type: String
        init (java_type: String) do self.java_type = java_type
end

redef class NitniCallback
        # Compile C and Java code to implement this callback
        fun compile_callback_to_java(nmodule: AModule, ccu: CCompilationUnit) do end

        # Returns the list of C functions to link with extern Java methods, as required
        # to enable this callback from Java code.
        # 
        # Return used by `AModule::ensure_linking_callback_methods`
        #
        # TODO we return an Array to support cast and other features like that
        fun jni_methods_declaration(from_module: MModule): Array[String] do return new Array[String]
end

redef class MExplicitCall
        redef fun compile_callback_to_java(nmodule, ccu)
        do
                var mproperty = mproperty
                assert mproperty isa MMethod
                var mmodule = nmodule.mmodule.as(not null)

                # In C, indirection implementing the Java extern methods
                var csignature = mproperty.build_c_implementation_signature(recv_mtype, mmodule, "___indirect", long_signature, from_java_call_context)
                var cf = new CFunction("JNIEXPORT {csignature}")
                cf.exprs.add "\t{mproperty.build_ccall(recv_mtype, mmodule, null, long_signature, from_java_call_context, null)}\n"
                ccu.add_local_function cf

                # In Java, declare the extern method as a private static local method
                var java_signature = mproperty.build_csignature(recv_mtype, mmodule, null, short_signature, java_call_context)
                nmodule.java_file.class_content.add "private native static {java_signature};\n"
        end

        redef fun jni_methods_declaration(from_mmodule)
        do
                var mproperty = mproperty
                assert mproperty isa MMethod

                var java_name = mproperty.build_cname(recv_mtype, from_mmodule, null, short_signature)
                var jni_format = mproperty.build_jni_format(recv_mtype, from_mmodule)
                var c_name = mproperty.build_cname(recv_mtype, from_mmodule, "___indirect", long_signature)

                return ["""{"{{{java_name}}}", "{{{jni_format}}}", {{{c_name}}}}"""]
        end
end

redef class MType

        # Type name in Java
        #
        # * Primitives common to both languages use their Java primitive type
        # * Nit extern Java classes are reprensented by their full Java type
        # * Other Nit objects are represented by `int` in Java. It holds the
        #       pointer to the underlying C structure.
        #       TODO create static Java types to store and hide the pointer
        private fun java_type: String do return "int"

        # JNI type name (in C)
        #
        # So this is a C type, usually defined in `jni.h`
        private fun jni_type: String do return "jint"

        # JNI short type name (for signatures)
        # 
        # Is used by `MMethod::build_jni_format` to pass a Java method signature
        # to the JNI function `GetStaticMetodId`.
        private fun jni_format: String do return "I"

        # Type name appearing within JNI function names.
        #
        # Used by `JavaLanguage::compile_extern_method` when calling JNI's `CallStatic*Method`.
        # This strategy is used by JNI to type the return of callbacks to Java.
        private fun jni_signature_alt: String do return "Int"
end

redef class MClassType
        redef fun java_type
        do
                var ftype = mclass.ftype
                if ftype isa ForeignJavaType then return ftype.java_type
                if mclass.name == "Bool" then return "boolean"
                if mclass.name == "Char" then return "char"
                if mclass.name == "Int" then return "int"
                if mclass.name == "Float" then return "double"
                return super
        end

        redef fun jni_type
        do
                var ftype = mclass.ftype
                if ftype isa ForeignJavaType then return "jobject"
                if mclass.name == "Bool" then return "jboolean"
                if mclass.name == "Char" then return "jchar"
                if mclass.name == "Int" then return "jint"
                if mclass.name == "Float" then return "jdouble"
                return super
        end

        redef fun jni_format
        do
                var ftype = mclass.ftype
                if ftype isa ForeignJavaType then return "L{ftype.java_type.replace('.', "/").replace(' ', "")};"
                if mclass.name == "Bool" then return "Z"
                if mclass.name == "Char" then return "C"
                if mclass.name == "Int" then return "I"
                if mclass.name == "Float" then return "D"
                return super
        end

        redef fun jni_signature_alt
        do
                var ftype = mclass.ftype
                if ftype isa ForeignJavaType then return "Object"
                if mclass.name == "Bool" then return "Bool"
                if mclass.name == "Char" then return "Char"
                if mclass.name == "Int" then return "Int"
                if mclass.name == "Float" then return "Double"
                return super
        end
end

redef class MMethod
        # Returns the JNI signature format of this Nit method
        # 
        # Example: a Nity signature `(Bool, Int, Float, JavaString)` is represented by
        # the JNI format `(ZIDLjava/lang/string;)V"
        private fun build_jni_format(recv_mtype: MClassType, from_mmodule: MModule): String
        do
                var mmethoddef = lookup_first_definition(from_mmodule, recv_mtype)
                var msignature = mmethoddef.msignature
                var format = new Array[String]

                format.add "("

                # receiver
                if not self.is_init then format.add recv_mtype.jni_format

                # parameters
                for p in msignature.mparameters do
                        var param_mtype = p.mtype.resolve_for(recv_mtype, recv_mtype, from_mmodule, true)
                        format.add param_mtype.jni_format
                end
                format.add ")"

                # return
                if self.is_init then
                        format.add recv_mtype.jni_format
                else
                        var return_mtype = msignature.return_mtype
                        if return_mtype != null then
                                format.add return_mtype.jni_format
                        else format.add "V"
                end

                return format.join("")
        end

        # Similar to `build_c_signature` but adapted to create the signature expected by JNI for C functions
        # implementing Java extern methods.
        #
        # Is used to generate FFI callbacks to Nit at `MExplicitCall::compile_callback_to_java`.
        private fun build_c_implementation_signature(recv_mtype: MClassType, from_mmodule: MModule,
                suffix: nullable String, length: SignatureLength, call_context: CallContext): String
        do
                var mmethoddef = lookup_first_definition(from_mmodule, recv_mtype)
                var signature = mmethoddef.msignature
                assert signature != null

                var creturn_type
                if self.is_init then
                        creturn_type = call_context.name_mtype(recv_mtype)
                else if signature.return_mtype != null then
                        var ret_mtype = signature.return_mtype
                        ret_mtype = ret_mtype.resolve_for(recv_mtype, recv_mtype, from_mmodule, true)
                        creturn_type = call_context.name_mtype(ret_mtype)
                else
                        creturn_type = "void"
                end

                var cname = build_cname(recv_mtype, from_mmodule, suffix, length)

                var cparams = new List[String]

                # This is different
                cparams.add "JNIEnv *env"
                cparams.add "jclass clazz"

                if not self.is_init then
                        cparams.add "{call_context.name_mtype(recv_mtype)} recv"
                end
                for p in signature.mparameters do
                        var param_mtype = p.mtype.resolve_for(recv_mtype, recv_mtype, from_mmodule, true)
                        cparams.add "{call_context.name_mtype(param_mtype)} {p.name}"
                end

                return "{creturn_type} {cname}( {cparams.join(", ")} )"
        end
end

private fun java_call_context: JavaCallContext do return new JavaCallContext
private fun to_java_call_context: ToJavaCallContext do return new ToJavaCallContext
private fun from_java_call_context: FromJavaCallContext do return new FromJavaCallContext