Передача указателя на указатель на float из Java через JNA в динамическую библиотеку C

Catboost предлагает динамическая библиотека C, который теоретически можно использовать на любом языке программирования.

Пытаюсь вызвать через Java с помощью JNA.

У меня проблема с функцией CalcModelPrediction, определенной в заголовочный файл следующим образом:

EXPORT bool CalcModelPrediction(
    ModelCalcerHandle* calcer,
    size_t docCount,
    const float** floatFeatures, size_t floatFeaturesSize,
    const char*** catFeatures, size_t catFeaturesSize,
    double* result, size_t resultSize);

В Java я определил функцию интерфейса следующим образом:

public interface CatboostModel extends Library {
        public Pointer ModelCalcerCreate();
        public String GetErrorString();
        public boolean LoadFullModelFromFile(Pointer calcer, String filename);
        public boolean CalcModelPrediction(Pointer calcer, int docCount,
                PointerByReference floatFeatures, int floatFeaturesSize,
                PointerByReference catFeatures, int catFeaturesSize,
                Pointer result, int resultSize);
        public int GetFloatFeaturesCount(Pointer calcer);
        public int GetCatFeaturesCount(Pointer calcer);
    }

а потом я называю это так:

CatboostModel catboost;
Pointer modelHandle;

catboost = Native.loadLibrary("catboostmodel", CatboostModel.class);
            modelHandle = catboost.ModelCalcerCreate();
if (!catboost.LoadFullModelFromFile(modelHandle, "catboost_test.model"))
{
    throw new RuntimeException("Cannot load Catboost model.");
}

final PointerByReference ppFloatFeatures = new PointerByReference();
final PointerByReference ppCatFeatures = new PointerByReference();
final Pointer pResult = new Memory(Native.getNativeSize(Double.TYPE));

float[] floatFeatures = {0.5f, 0.8f, 0.3f, 0.3f, 0.1f, 0.5f, 0.4f, 0.8f, 0.3f, 0.3f} ;
String[] catFeatures = {"1", "2", "3", "4"};
int catFeaturesLength = 0;
for (String s : catFeatures)
{
    catFeaturesLength += s.length() + 1;
}

try
{
    final Pointer pFloatFeatures = new Memory(floatFeatures.length * Native.getNativeSize(Float.TYPE));
    for (int dloop=0; dloop<floatFeatures.length; dloop++) {
        pFloatFeatures.setFloat(dloop * Native.getNativeSize(Float.TYPE), floatFeatures[dloop]);
    }
    ppFloatFeatures.setValue(pFloatFeatures);

    final Pointer pCatFeatures = new Memory(catFeaturesLength * Native.getNativeSize(Character.TYPE));
    long offset = 0;
    for (final String s : catFeatures) {
        pCatFeatures.setString(offset, s);
        pCatFeatures.setMemory(offset + s.length(), 1, (byte)(0));
        offset += s.length() + 1;
    }
    ppCatFeatures.setValue(pCatFeatures);

}
catch (Exception e)
{
    throw new RuntimeException("Couldn't initialize parameters for catboost");
}

try
{
    if (!catboost.CalcModelPrediction(
                modelHandle,
                1,
                ppFloatFeatures, 10,
                ppCatFeatures, 4,
                pResult, 1
                ))
    {
        throw new RuntimeException("No prediction made: " + catboost.GetErrorString());
    }
    else
    {
        double[] result = pResult.getDoubleArray(0, 1);
        log.info("Catboost prediction: " + String.valueOf(result[0]));
        Assert.assertFalse("ERROR: Result empty", result.length == 0);
    }
}
catch (Exception e)
{
    throw new RuntimeException("Prediction failed: " + e);
}

Я пытался передать Pointer, PointerByReference и Pointer[] функции CalcModelPrediction вместо float **floatFeatures и char ***catFeatures, но ничего не помогло. Я всегда получаю ошибку сегментации, предположительно, когда функция CalcModelPrediction пытается получить элементы floatFeatures и catFeatures, вызывая floatFeatures[0][0] и catFeatures[0][0].

Итак, вопрос в том, как правильно передать многомерный массив из Java через JNA в C, где он может рассматриваться как указатель на указатель на значение?

Интересно то, что функция CalcModelPredictionFlat, которая принимает только float **floatFeatures, а затем просто вызывает *floatFeatures, отлично работает при передаче PointerByReference.

ОБНОВЛЕНИЕ - 5.5.2018

Часть 1

После попытки отладить segfault, слегка изменив исходные файлы Catboost .cpp и .h и перекомпилировав библиотеку libcatboost.so, я обнаружил, что segfault возник из-за того, что я сопоставил size_t на C с int на Java. После исправления моя функция интерфейса в Java выглядит так:

public interface CatboostModel extends Library {
        public boolean LoadFullModelFromFile(Pointer calcer, String filename);
        public boolean CalcModelPrediction(Pointer calcer, size_t docCount,
                Pointer[] floatFeatures, size_t floatFeaturesSize,
                String[] catFeatures, size_t catFeaturesSize,
                Pointer result, size_t resultSize);
    }

Где класс size_t определяется следующим образом:

public static class size_t extends IntegerType {                                           
    public size_t() { this(0); }                                                           
    public size_t(long value) { super(Native.SIZE_T_SIZE, value); }                       
}

Часть 2 Изучая код Catboost, я заметил, что доступ к **floatFeatures осуществляется по строкам, например floatFeatures[i], а к ***catFeatures - по строкам и столбцам, например catFetures[i][catFeatureIdx].

После замены floatFeatures в Java на массив Pointer мой код начал работать с моделью, обученной без категориальных функций, т.е.длина catFeatures равна нулю.

Однако этот трюк не работал с catFeatures, доступ к которым осуществляется с помощью оператора двойного индекса [i][catFeatureidx]. Итак, на данный момент я изменил исходный код Catboost, чтобы он принимал char **catFeatures - массив строк. В функции интерфейса Java я установил String[] catFeatures. Теперь я могу делать прогнозы для одного элемента за раз, что не идеально.