react-native-vision-camera/package/android/src/main/java/com/mrousavy/camera/types/PixelFormat.kt

package com.mrousavy.camera.types

import android.graphics.ImageFormat
import android.util.Log
import com.mrousavy.camera.core.InvalidTypeScriptUnionError
import com.mrousavy.camera.core.PixelFormatNotSupportedError
import com.mrousavy.camera.utils.ImageFormatUtils

enum class PixelFormat(override val unionValue: String) : JSUnionValue {
  YUV("yuv"),
  RGB("rgb"),
  NATIVE("native"),
  UNKNOWN("unknown");

  fun toImageFormat(): Int =
    when (this) {
      YUV -> ImageFormat.YUV_420_888
      NATIVE -> ImageFormat.PRIVATE
      else -> throw PixelFormatNotSupportedError(this.unionValue)
    }

  companion object : JSUnionValue.Companion<PixelFormat> {
    private const val TAG = "PixelFormat"
    fun fromImageFormat(imageFormat: Int): PixelFormat =
      when (imageFormat) {
        ImageFormat.YUV_420_888 -> YUV
        ImageFormat.PRIVATE -> NATIVE
        else -> {
          Log.w(TAG, "Unknown PixelFormat! ${ImageFormatUtils.imageFormatToString(imageFormat)}")
          UNKNOWN
        }
      }

    override fun fromUnionValue(unionValue: String?): PixelFormat =
      when (unionValue) {
        "yuv" -> YUV
        "rgb" -> RGB
        "native" -> NATIVE
        "unknown" -> UNKNOWN
        else -> throw InvalidTypeScriptUnionError("pixelFormat", unionValue)
      }
  }
}