FancyBot/src/main/kotlin/cn/rtast/fancybot/util/misc/GIF.kt

/*
 * Copyright © 2024 RTAkland
 * Author: RTAkland
 * Date: 2024/9/21
 */


package cn.rtast.fancybot.util.misc

import com.madgag.gif.fmsware.AnimatedGifEncoder
import com.madgag.gif.fmsware.GifDecoder
import java.awt.image.BufferedImage
import java.io.ByteArrayOutputStream

fun GifDecoder.makeGif(frames: List<BufferedImage>): ByteArray {
    val estimatedSize = frames.size * 50 * 1024
    val byteArrayOutputStream = ByteArrayOutputStream(estimatedSize)
    val encoder = AnimatedGifEncoder()
    encoder.start(byteArrayOutputStream)
    encoder.setRepeat(0)
    val delays = frames.indices.map { this.getDelay(this.frameCount - it - 1) }
    frames.indices.forEach { i ->
        encoder.setDelay(delays[i])
        encoder.addFrame(frames[i])
    }
    encoder.finish()
    return byteArrayOutputStream.toByteArray()
}