renderscript consider yuv type

gordinmitya · Aug 1, 2020 · 2ecaab9 · 2ecaab9
1 parent 5760011
commit 2ecaab9
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/app/src/main/java/ru/gordinmitya/yuv2buf_demo/RenderScriptConverter.kt b/app/src/main/java/ru/gordinmitya/yuv2buf_demo/RenderScriptConverter.kt
@@ -34,10 +34,14 @@ class RenderScriptConverter(context: Context) : ImageConverter {
  || input!!.type.yuv != converted.type.format
  || bytes.size != converted.buffer.capacity()
  ) {
+ val yuvFormat = when (converted.type) {
+ Yuv.Type.YUV_I420 -> ImageFormat.YUV_420_888
+ Yuv.Type.YUV_NV21 -> ImageFormat.NV21
+ }
  val yuvType: Type.Builder = Type.Builder(rs, Element.U8(rs))
  .setX(image.width)
  .setY(image.height)
- .setYuvFormat(ImageFormat.NV21)
+ .setYuvFormat(yuvFormat)
  input = Allocation.createTyped(rs, yuvType.create(), Allocation.USAGE_SCRIPT)
  bytes = ByteArray(converted.buffer.capacity())
  val rgbaType: Type.Builder = Type.Builder(rs, Element.RGBA_8888(rs))