// Thanks to sh1r0, reference: https://github.com/sh1r0/caffe-android-demo

// TODO: add your package name

import java.nio.charset.StandardCharsets;

public class CaffeMobile {

    /* private class */
    private static byte[] stringToBytes(String s) {
        return s.getBytes(StandardCharsets.US_ASCII);
    }

    private native void setMeanWithMeanFile(String meanFile);

    private native void setMeanWithMeanValues(float[] meanValues);

    private native int[] predictImage(byte[] data, int width, int height, int k);

    private native float[] getConfidenceScore(byte[] data, int width, int height);

    private native float[][] extractFeatures(byte[] data, int width, int height, String blobNames);

    private native void native_setNumThreads(int numThreads);

    //private native void enableLog(boolean enabled);  // currently nonfunctional

    private native int native_loadModel(String modelPath, String weightsPath);  // required

    //private native void setScale(float scale);


    /* public class */

    /*
        * CPU2RK33992Cortex-A72
        * numThreads
        */
    public void setNumThreads(int numThreads) {
        native_setNumThreads(numThreads);
    }

    /*
        * 
        * modelPath
        * weightsPath
        * 0
        */
    public int loadModel(String modelPath, String weightsPath) {
        return native_loadModel(modelPath, weightsPath);
    }

    /*
        * 
        * imgPath
        * 
        */
    public float[] getConfidenceScore(String imgPath) {
        return getConfidenceScore(stringToBytes(imgPath), 0, 0);
    }

    /*
        * 
        * imgPath
        * kTop kid
        * Top kid
        */
    public int[] predictImage(String imgPath, int k) {
        return predictImage(stringToBytes(imgPath), 0, 0, k);
    }

    /*
        * 
        * imgPath
        * Topid
        */
    public int[] predictImage(String imgPath) {
        return predictImage(imgPath, 1);
    }

    /*
        * 
        * imgPath
        * blobNames
        */
    public float[][] extractFeatures(String imgPath, String blobNames) {
        return extractFeatures(stringToBytes(imgPath), 0, 0, blobNames);
    }

    /*
        * 
        * meanValues
        */
    public void setMean(float[] meanValues) {
        setMeanWithMeanValues(meanValues);
    }

    /*
        * 
        * meanFile
        */
    public void setMean(String meanFile) {
        setMeanWithMeanFile(meanFile);
    }
}