如何使用ByteBuffer从C ++返回到Java

时间:2012-07-23 18:26:06

标签: java c++ swig

How can I make Swig correctly wrap a char* buffer that is modified in C as a Java Something-or-other?中,这几乎是一个重复的问题 如果我使用bytebuffer而不是Stringbuffer,那么typemap会有什么变化?

1 个答案:

答案 0 :(得分:0)

我已经汇总了一个示例,说明如何使用以下头文件/函数作为测试:

#include <stdio.h>

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

我的解决方案是修改this answer,使代理获取ByteBuffer并将其转换为byte[],以便我们传递给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 typemap中,分配一个临时byte[],然后使用get填充它。实际上有一个array()函数,如果您使用的ByteBuffer支持,则应该使用,即typemap应该只是:

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

如果您的实现支持它(该方法是可选的,可能会抛出UnsuportedOperationException)。

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

%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以使用它:

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