如何从Renderscript内核中检索结构

时间:2016-07-22 19:15:56

标签: java android renderscript android-renderscript

我有问题。我想从renderscript内核中检索一个结构。我想要的是我会得到一个输入结构元素...我会修改它然后返回它修改。但是在反射层中没有这样的方法。我试图手动反序列化缓冲区中的数据,但我甚至无法将缓冲区复制到ByteBuffer,因为Allocation在一个类型的copyTo中有验证,所以我不知道我应该做什么...

1 个答案:

答案 0 :(得分:1)

RenderScript支持自定义元素。要创建一个,请在RS脚本中声明自定义typedef struct,如下所示:

typedef struct MyElement {

    int x;
    int y;
    bool simpleBool;

} MyElement_t;

在构建过程之后,将出现ScriptField_MyElement Java类,镜像RS结构。您将能够使用此类创建使用您自己的元素的自定义分配:

// Declares a new Allocation, based upon the custom struct Element
Element myElement = ScriptField_MyElement.createElement(mRS);
Allocation myElementsAllocation = Allocation.createSized(mRS, myElement, 5);

// Or

Allocation myElementsAllocation = ScriptField_MyElement.create1D(mRS, sizeX).getAllocation();

您可以在CustomElementExample sample project中找到此过程的示例。 此外,在SurfaceRenderExample sample project内,您可以看到如何使用自定义元素来建模数学结构(在这种情况下是一个粒子,随着加速度下降)。

RenderScript脚本内部:

  • 从分配中获取自定义元素:

    MyElement_t el = * (MyElement_t *) rsGetElementAt(aIn, index);
    
  • 要更改自定义元素成员:

    el.x = 10;
    
  • 在分配中设置自定义元素:

    rsSetElementAt(myAlloc, (void *)&el);
    

参考:RenderScript: parallel computing on Android, the easy way

修改

目前,没有直接的方法将自定义struct元素复制到Java端。

CustomStructElementCopyToJava sample project提供了一个流程示例。

示例

的简短说明

注意:以下过程实验,根本无法执行!如果您打算大量使用此过程,请使用Android NDK访问分配。 此外,在Android SDK的未来版本中,此代码可能会中断,因为它依赖于Java反射;一些通常隐藏的方法可以在Android SDK中没有任何通知的情况下进行更改。

让我们假设使用以下自定义struct元素:

typedef struct Point {
    int x;
    int y;
} Point_t;

在查看生成的结构代码时(可以在Android Studio中看到,通过按 CTRL + B ,同时关注Java端的ScriptField_Point元素),可以看到以下要素:

public static Element createElement(RenderScript rs) {
        Element.Builder eb = new Element.Builder(rs);
        eb.add(Element.I32(rs), "x");
        eb.add(Element.I32(rs), "y");
        return eb.create();
    }

您可以用hacky方式映射自定义结构的内容:

1)定义目标字节数组:

    byte destinationArray[] = new byte[allocationGrayPointOrdered.getBytesSize()];

2)使用Java反射来访问隐藏的Allocation.copyTo方法:

private static Method getCopyToWithoutValidationMethod(){
    // private void copyTo(Object array, Element.DataType dt, int arrayLen)
    Method allocationHiddenCopyToMethod = null;
    try {
        allocationHiddenCopyToMethod = Allocation.class.getDeclaredMethod("copyTo", Object.class, Element.DataType.class, int.class);
        allocationHiddenCopyToMethod.setAccessible(true);
    } catch (NoSuchMethodException e) {
        throw new RuntimeException("Could not find allocationHiddenCopyToMethod");
    }

    return allocationHiddenCopyToMethod;
}

3)执行复制:

// Gets reflected method
Method copyToWithoutValidationMethod = getCopyToWithoutValidationMethod();

// Tries to copy contents
try {
    copyToWithoutValidationMethod.invoke(allocationGrayPointOrdered, destinationArray, 
                                                Element.DataType.UNSIGNED_8, destinationArray.length);
} catch (IllegalAccessException e) {
    throw new RuntimeException(e);
} catch (InvocationTargetException e) {
    throw new RuntimeException(e);
}

4)一旦数组填充了源数据,就可以将其内容映射到人类可读的结构中。

// Defines the destination array
ScriptField_Point.Item mappedItems[][] = new ScriptField_Point.Item[sizeX][sizeY];

// Wraps array contents
ByteBuffer byteBuffer = ByteBuffer.wrap(destinationArray);
// Sets byte order to be Android-like
byteBuffer.order(ByteOrder.LITTLE_ENDIAN);

// Iterates on every column and row
for (int x = 0; x < sizeX; x++) {
    for (int y = 0; y < sizeY; y++) {

        // Allocates a new item
        ScriptField_Point.Item currentItem = new ScriptField_Point.Item();

        // Calculate the offset in the source array
        int currentOffset = (x + y * sizeX) * ScriptField_Point.Item.sizeof;

        // Gets data from the byte array
        currentItem.x = byteBuffer.getInt(currentOffset);
        currentItem.y = byteBuffer.getInt(currentOffset + 4);

        mappedItems[x][y] = currentItem;
    }
}

有关完整说明,请参阅book