Поворот YUV420 android .media.Image Android - PullRequest
Новая
       39

Поворот YUV420 android .media.Image Android

1 голос
/ 03 апреля 2020

Я пытаюсь получить несколько изображений и поставить их через алгоритм. Формат, который я использую, YUV_420_8888, например: 

streamImageReader = ImageReader.newInstance(
  imageSize.width,
  imageSize.height,
  ImageFormat.YUV_420_888,
  30
)

. После этого мне нужно преобразовать экземпляр android.media.Image результата (изображение результата) в байтовый массив RGB. Я делаю это с помощью этой функции расширения, вызываемой с помощью image.planes: 

fun Array<Plane>.YUV_420_888_toRGB(
    context: Context,
    width: Int,
    height: Int
): ByteArray {
    // Get the three image planes
    val planes: Array<Plane> = this
    var buffer: ByteBuffer = planes[0].buffer
    buffer.rewind()

    val y = ByteArray(buffer.capacity())
    buffer.get(y)
    buffer = planes[1].buffer

    val u = ByteArray(buffer.capacity())
    buffer.get(u)
    buffer = planes[2].buffer

    val v = ByteArray(buffer.capacity())
    buffer.get(v)

    // get the relevant RowStrides and PixelStrides
    // (we know from documentation that PixelStride is 1 for y)
    val yRowStride = planes[0].rowStride
    val uvRowStride =
        planes[1].rowStride // we know from documentation that RowStride is the same for u and v.
    val uvPixelStride =
        planes[1].pixelStride // we know from documentation that PixelStride is the same for u and v.

    // rs creation just for demo. Create rs just once in onCreate and use it again.
    val rs = RenderScript.create(context)
    //RenderScript rs = MainActivity.rs;
    val mYuv420 = ScriptC_yuv420888(rs)
    // Y,U,V are defined as global allocations, the out-Allocation is the Bitmap.
    // Note also that uAlloc and vAlloc are 1-dimensional while yAlloc is 2-dimensional.
    val typeUcharY: Type.Builder = Type.Builder(rs, Element.U8(rs))
    typeUcharY.setX(yRowStride).setY(height)

    val yAlloc = Allocation.createTyped(rs, typeUcharY.create())
    yAlloc.copy1DRangeFrom(0, y.size, y)
    mYuv420.set_ypsIn(yAlloc)

    val typeUcharUV: Type.Builder = Type.Builder(rs, Element.U8(rs))
    // note that the size of the u's and v's are as follows:
    // (  (width/2)*PixelStride + padding  ) * (height/2)
    // = (RowStride) * (height/2)
    typeUcharUV.setX(u.size)

    val uAlloc = Allocation.createTyped(rs, typeUcharUV.create())
    uAlloc.copyFrom(u)
    mYuv420._uIn = uAlloc

    val vAlloc = Allocation.createTyped(rs, typeUcharUV.create())
    vAlloc.copyFrom(v)
    mYuv420._vIn = vAlloc
    // handover parameters
    mYuv420._picWidth = width.toLong()
    mYuv420._uvRowStride = uvRowStride.toLong()
    mYuv420._uvPixelStride = uvPixelStride.toLong()

    val outBitmap = Bitmap.createBitmap(width, height, Bitmap.Config.ARGB_8888)
    val outAlloc = Allocation.createFromBitmap(
        rs,
        outBitmap,
        Allocation.MipmapControl.MIPMAP_NONE,
        Allocation.USAGE_SCRIPT
    )

    val lo = LaunchOptions()
    lo.setX(
        0,
        width
    ) // by this we ignore the y’s padding zone, i.e. the right side of x between width and yRowStride

    lo.setY(0, height)
    mYuv420.forEach_doConvert(outAlloc, lo)
    outAlloc.copyTo(outBitmap)

    return outBitmap.toByteArray()
}

Проблема, с которой я столкнулся, заключается в том, что этот результирующий байтовый массив вращается. Я попытался исправить это с помощью этого предложения: Поворот байтового массива YUV на Android

НО , получившийся байтовый массив после алгоритма выше имеет вдвое меньше элементов как первый (некоторая потеря здесь).

Я думал, что самым простым решением для этого было бы вращение изображения непосредственно, а не байтового массива.

...