Android Renderscript-在Renderscript中旋转YUV数据

Question

基于我在Camera2 api Imageformat.yuv_420_888旋转图像结果上的讨论，我想知道如何调整通过rsGetElementAt_uchar方法完成的查找，以便将YUV数据旋转90度。 我也有一个类似Google提供的HdrViewfinder的项目。 问题在于输出是横向的，因为用作目标表面的输出表面已连接到yuv分配，而yuv分配并不关心设备是横向还是纵向模式。 但是我想调整代码，使其处于纵向模式。 因此，我采用了自定义的YUVToRGBA渲染脚本，但不知道要更改哪些内容以旋转输出。 有人可以帮我将以下自定义YUVtoRGBA脚本调整90度，因为我想在纵向模式下使用输出：

// Needed directive for RS to work
#pragma version(1)

// The java_package_name directive needs to use your Activity's package path
#pragma rs java_package_name(net.hydex11.cameracaptureexample)

rs_allocation inputAllocation;

int wIn, hIn;
int numTotalPixels;

// Function to invoke before applying conversion
void setInputImageSize(int _w, int _h)
{
    wIn = _w;
    hIn = _h;
    numTotalPixels = wIn * hIn;
}

// Kernel that converts a YUV element to a RGBA one
uchar4 __attribute__((kernel)) convert(uint32_t x, uint32_t y)
{

    // YUV 4:2:0 planar image, with 8 bit Y samples, followed by
    // interleaved V/U plane with 8bit 2x2 subsampled chroma samples
    int baseIdx = x + y * wIn;
    int baseUYIndex = numTotalPixels + (y >> 1) * wIn + (x & 0xfffffe);

    uchar _y = rsGetElementAt_uchar(inputAllocation, baseIdx);
    uchar _u = rsGetElementAt_uchar(inputAllocation, baseUYIndex);
    uchar _v = rsGetElementAt_uchar(inputAllocation, baseUYIndex + 1);
    _y = _y < 16 ? 16 : _y;

    short Y = ((short)_y) - 16;
    short U = ((short)_u) - 128;
    short V = ((short)_v) - 128;

    uchar4 out;
    out.r = (uchar) clamp((float)(
        (Y * 298 + V * 409 + 128) >> 8), 0.f, 255.f);   
    out.g = (uchar) clamp((float)(
        (Y * 298 - U * 100 - V * 208 + 128) >> 8), 0.f, 255.f); 
    out.b = (uchar) clamp((float)(
        (Y * 298 + U * 516 + 128) >> 8), 0.f, 255.f); // 
    out.a = 255;

    return out;
}

我在https://bitbucket.org/cmaster11/rsbookexamples/src/tip/CameraCaptureExample/app/src/main/rs/customYUVToRGBAConverter.fs中找到了自定义脚本。

在这里，有人放置了Java代码来旋转YUV数据。 但我想在Renderscript中执行此操作，因为这样做速度更快。

任何帮助都会很棒。

最好的祝福，

Answer 1

我假设您希望输出像在转换脚本中一样是RGBA。 您应该能够使用此答案中所使用的方法； 也就是说，只需简单地将x和y坐标修改为转换内核的第一步即可：

//Rotate 90 deg clockwise during the conversion
uchar4 __attribute__((kernel)) convert(uint32_t inX, uint32_t inY)
{
    uint32_t x = wIn - 1 - inY;
    uint32_t y = inX;

    //...rest of the function

注意对参数名称的更改。

假设您已经正确设置了输出尺寸（请参阅链接的答案）。 270度旋转可以类似的方式完成。