Class MLModel

java_package/brainflow/src/main/java/brainflow/MLModel.java:11–245

import com.sun.jna.Native;

@SuppressWarnings ("deprecation")
public class MLModel
{
    private interface DllInterface extends Library
    {

        int set_log_level_ml_module (int log_level);

        int set_log_file_ml_module (String log_file);

        int log_message_ml_module (int log_level, String message);

        int prepare (String params);

        int release (String params);

        int predict (double[] data, int data_len, double[] output, int[] output_len, String params);

        int release_all ();

        int get_version_ml_module (byte[] version, int[] len, int max_len);
    }

    private static DllInterface instance;

    static
    {
        // SystemUtils doesnt have smth like IS_OS_ANDROID, need to check using
        // properties
        boolean is_os_android = "The Android Project".equals (System.getProperty ("java.specification.vendor"));

        String lib_name = "libMLModule.so";
        if (SystemUtils.IS_OS_WINDOWS)
        {
            lib_name = "MLModule.dll";
            String arch = System.getProperty ("os.arch");
            switch (arch) {
                case "x86":
                    JarHelper.unpack_from_jar ("onnxruntime_x86.dll");
                    break;
                case "x86_64":
                case "amd64":
                    JarHelper.unpack_from_jar ("onnxruntime_x64.dll");
                    break;
                case "arm":
                    JarHelper.unpack_from_jar ("onnxruntime_arm.dll");
                    break;
                case "arm64":
                    JarHelper.unpack_from_jar ("onnxruntime_arm64.dll");
                    break;
                default:
                    System.err.println("Unsupported Windows architecture: " + arch);
            }
        } else if (SystemUtils.IS_OS_MAC)
        {
            lib_name = "libMLModule.dylib";
            JarHelper.unpack_from_jar ("onnxruntime_x86.dll");
            JarHelper.unpack_from_jar ("onnxruntime_x86.dll");