java - 如何使用从 C++ 到 Java 的 java.nio.ByteBuffer 返回

Question

这几乎是如何让 Swig 正确包装在 C 中作为 Java 或其他东西修改的 char* 缓冲区中的问题的重复？...
如果我要使用字节缓冲区而不是 Stringbuffer，那么 typemap 会有什么变化？

Answer 1

我已经整理了一个示例，说明如何使用以下头文件/函数作为测试来执行此操作：

#include <stdio.h>

static void foo(char *buf, int len) {
  while(len--)
    putchar(*buf++);
}

我的解决方案是修改这个答案，使代理接受 aByteBuffer并将其转换为 abyte[]以供我们传递给 JNI 代码，然后将其转换为我们的指针 + 长度组合。

%module test

%{
#include "test.h"
%}

%typemap(jtype) (char *buf, int len) "byte[]"
%typemap(jstype) (char *buf, int len) "java.nio.ByteBuffer"
%typemap(jni) (char *buf, int len) "jbyteArray"
%typemap(javain,pre="    byte[] temp$javainput = new byte[$javainput.capacity()];"
                    "    $javainput.get(temp$javainput);")
        (char *buf, int len) "temp$javainput"

%typemap(in,numinputs=1) (char *buf, int len) {
  $1 = JCALL2(GetByteArrayElements, jenv, $input, NULL);
  $2 = JCALL1(GetArrayLength, jenv, $input);
}

%typemap(freearg) (const signed char *arr, size_t sz) {
  // Or use  0 instead of ABORT to keep changes if it was a copy
  JCALL3(ReleaseByteArrayElements, jenv, $input, $1, JNI_ABORT); 
}

%include "test.h"

这里的新位在 javain 类型映射中，分配一个临时的byte[]然后使用get来填充它。实际上有一个array()功能，如果ByteBuffer您使用的支持您应该使用它，即类型映射应该只是：

%typemap(javain) (char *buf, int len) "$javainput.array()"

如果您的实现支持它（该方法是可选的并且可能会抛出UnsuportedOperationException）。

实际上，这可以通过 SWIG 2.0 进一步简化，从前面引用的问题开始，因为我们期望类型始终是byte我们可以使用 SWIG 2.0 中的内置 int 类型映射来简化我们的界面，现在变成：

%module test

%{
#include "test.h"
%}

%apply (char *STRING, size_t LENGTH) { (char *buf, int len) }
%typemap(javain) (char *buf, int len) "$javainput.array()"
%typemap(jstype) (char *buf, int len) "java.nio.ByteBuffer"

%include "test.h"

我使用以下 Java 测试了所有三个版本：

public class run {
  public static void main(String[] argv) {
    System.loadLibrary("test");
    byte value[] = "hello world\n".getBytes();
    java.nio.ByteBuffer buf = java.nio.ByteBuffer.wrap(value);
    test.foo(buf);
  }
}

为了安全起见，array()您可能想要做的是在带有编译指示的函数中添加 try/catch：

%pragma(java) modulecode = %{
  private static byte[] buf2bytearr(java.nio.ByteBuffer buf) {
    try {
      return buf.array();
    }
    catch (UnsupportedOperationException e) {
      byte arr[] = new byte[buf.capacity()];
      buf.get(arr);
      return arr;
    }
  }
%}

然后修改类型映射以使用它：

%typemap(javain) (char *buf, int len) "buf2bytearr($javainput)"