/*
 * 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.Coder
import com.radzivon.bartoshyk.avif.coder.HeifQualityArg
import com.radzivon.bartoshyk.avif.coder.PreciseMode
import com.t8rin.imagetoolbox.core.data.image.utils.ImageCompressorBackend
import com.t8rin.imagetoolbox.core.domain.image.model.HeicChromaSubsampling
import com.t8rin.imagetoolbox.core.domain.image.model.Quality
import com.radzivon.bartoshyk.avif.coder.HeicChromaSubsampling as BackendChromaSubsampling

internal data class HeicBackend(
    private val isLossless: Boolean
) : ImageCompressorBackend {

    override suspend fun compress(
        image: Bitmap,
        quality: Quality
    ): ByteArray {
        val heicQuality = quality as? Quality.Heic ?: Quality.Heic(
            qualityValue = quality.qualityValue
        )

        return Coder().encodeHeic(
            bitmap = image,
            quality = HeifQualityArg.Quality(heicQuality.qualityValue),
            preciseMode = if (isLossless) {
                PreciseMode.LOSSLESS
            } else {
                PreciseMode.LOSSY
            },
            chromaSubsampling = heicQuality.chromaSubsampling.toBackend()
        )
    }

}

private fun HeicChromaSubsampling.toBackend(): BackendChromaSubsampling = when (this) {
    HeicChromaSubsampling.Yuv420 -> BackendChromaSubsampling.YUV420
    HeicChromaSubsampling.Yuv422 -> BackendChromaSubsampling.YUV422
    HeicChromaSubsampling.Yuv444 -> BackendChromaSubsampling.YUV444
}