/*
 * ImageToolbox is an image editor for android
 * Copyright (c) 2026 T8RIN (Malik Mukhametzyanov)
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 * You should have received a copy of the Apache License
 * along with this program.  If not, see <http://www.apache.org/licenses/LICENSE-2.0>.
 */

package com.t8rin.imagetoolbox.core.data.image.utils.compressor

import android.graphics.Bitmap
import com.radzivon.bartoshyk.avif.coder.AvKind
import com.radzivon.bartoshyk.avif.coder.AvSpeed
import com.radzivon.bartoshyk.avif.coder.Coder
import com.radzivon.bartoshyk.avif.coder.PreciseMode
import com.t8rin.imagetoolbox.core.data.image.utils.ImageCompressorBackend
import com.t8rin.imagetoolbox.core.domain.image.model.AvifChromaSubsampling
import com.t8rin.imagetoolbox.core.domain.image.model.Quality
import com.radzivon.bartoshyk.avif.coder.AvifChromaSubsampling as BackendChromaSubsampling

internal data class AvifBackend(
    private val isLossless: Boolean,
    private val isAv1: Boolean
) : ImageCompressorBackend {

    override suspend fun compress(
        image: Bitmap,
        quality: Quality
    ): ByteArray {
        val avifQuality = quality as? Quality.Avif ?: Quality.Avif()

        return Coder().encodeAvif(
            bitmap = image,
            quality = avifQuality.qualityValue,
            preciseMode = if (isLossless) {
                PreciseMode.LOSSLESS
            } else {
                PreciseMode.LOSSY
            },
            avifChromaSubsampling = avifQuality.chromaSubsampling.toBackend(),
            avKind = if (isAv1) {
                AvKind.AV1
            } else {
                AvKind.AV2
            },
            speed = AvSpeed.entries.firstOrNull {
                it.ordinal == (3 - avifQuality.effort)
            } ?: AvSpeed.FAST
        )
    }

}

private fun AvifChromaSubsampling.toBackend(): BackendChromaSubsampling = when (this) {
    AvifChromaSubsampling.Auto -> BackendChromaSubsampling.AUTO
    AvifChromaSubsampling.Yuv420 -> BackendChromaSubsampling.YUV420
    AvifChromaSubsampling.Yuv422 -> BackendChromaSubsampling.YUV422
    AvifChromaSubsampling.Yuv444 -> BackendChromaSubsampling.YUV444
    AvifChromaSubsampling.Yuv400 -> BackendChromaSubsampling.YUV400
    AvifChromaSubsampling.Lossless -> BackendChromaSubsampling.LOSELESS
}