// Copyright 2016 The Ebiten Authors
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// 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.

package ebiten_test

import (
	"bytes"
	"image"
	"image/color"
	"image/color/palette"
	"testing"

	"github.com/hajimehoshi/ebiten/v2"
)

func TestImageToBytes(t *testing.T) {
	pal := make(color.Palette, 256)
	for i := range pal {
		pal[i] = color.White
	}
	p := make([]color.Color, 255)
	for i := range p {
		if i == 64 {
			p[i] = color.White
		} else {
			p[i] = color.Transparent
		}
	}
	bigPalette := color.Palette(p)
	cases := []struct {
		Image  image.Image
		Premul bool
		Out    []uint8
	}{
		{
			Image: &image.Paletted{
				Pix:    []uint8{0, 1, 1, 2},
				Stride: 2,
				Rect:   image.Rect(0, 0, 2, 2),
				Palette: color.Palette([]color.Color{
					color.Transparent, color.White, color.RGBA{0x80, 0x80, 0x80, 0x80},
				}),
			},
			Premul: true,
			Out:    []uint8{0, 0, 0, 0, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0x80, 0x80, 0x80, 0x80},
		},
		{
			Image: &image.Paletted{
				Pix:    []uint8{0, 1, 1, 2},
				Stride: 2,
				Rect:   image.Rect(0, 0, 2, 2),
				Palette: color.Palette([]color.Color{
					color.Transparent, color.White, color.RGBA{0x80, 0x80, 0x80, 0x80},
				}),
			},
			Premul: false,
			Out:    []uint8{0, 0, 0, 0, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0x80},
		},
		{
			Image:  image.NewPaletted(image.Rect(0, 0, 240, 160), pal).SubImage(image.Rect(238, 158, 240, 160)),
			Premul: true,
			Out:    []uint8{0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff},
		},
		{
			Image:  image.NewPaletted(image.Rect(0, 0, 240, 160), pal).SubImage(image.Rect(238, 158, 240, 160)),
			Premul: false,
			Out:    []uint8{0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff},
		},
		{
			Image: &image.RGBA{
				Pix:    []uint8{0, 0, 0, 0, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0, 0, 0, 0},
				Stride: 8,
				Rect:   image.Rect(0, 0, 2, 2),
			},
			Premul: true,
			Out:    []uint8{0, 0, 0, 0, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0, 0, 0, 0},
		},
		{
			Image: &image.RGBA{
				Pix:    []uint8{0, 0, 0, 0, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0, 0, 0, 0},
				Stride: 8,
				Rect:   image.Rect(0, 0, 2, 2),
			},
			Premul: false,
			Out:    []uint8{0, 0, 0, 0, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0, 0, 0, 0},
		},
		{
			Image: &image.NRGBA{
				Pix:    []uint8{0, 0, 0, 0, 0xff, 0xff, 0xff, 0x80, 0x80, 0x80, 0x80, 0x80, 0, 0, 0, 0},
				Stride: 8,
				Rect:   image.Rect(0, 0, 2, 2),
			},
			Premul: true,
			Out:    []uint8{0, 0, 0, 0, 0x80, 0x80, 0x80, 0x80, 0x40, 0x40, 0x40, 0x80, 0, 0, 0, 0},
		},
		{
			Image: &image.NRGBA{
				Pix:    []uint8{0, 0, 0, 0, 0xff, 0xff, 0xff, 0x80, 0x80, 0x80, 0x80, 0x80, 0, 0, 0, 0},
				Stride: 8,
				Rect:   image.Rect(0, 0, 2, 2),
			},
			Premul: false,
			Out:    []uint8{0, 0, 0, 0, 0xff, 0xff, 0xff, 0x80, 0x80, 0x80, 0x80, 0x80, 0, 0, 0, 0},
		},
		{
			Image: &image.Paletted{
				Pix:     []uint8{0, 64, 0, 0},
				Stride:  2,
				Rect:    image.Rect(0, 0, 2, 2),
				Palette: bigPalette,
			},
			Premul: true,
			Out:    []uint8{0, 0, 0, 0, 0xff, 0xff, 0xff, 0xff, 0, 0, 0, 0, 0, 0, 0, 0},
		},
		{
			Image: &image.Paletted{
				Pix:     []uint8{0, 64, 0, 0},
				Stride:  2,
				Rect:    image.Rect(0, 0, 2, 2),
				Palette: bigPalette,
			},
			Premul: false,
			Out:    []uint8{0, 0, 0, 0, 0xff, 0xff, 0xff, 0xff, 0, 0, 0, 0, 0, 0, 0, 0},
		},
		{
			Image: (&image.Paletted{
				Pix:     []uint8{0, 64, 0, 0},
				Stride:  2,
				Rect:    image.Rect(0, 0, 2, 2),
				Palette: bigPalette,
			}).SubImage(image.Rect(1, 0, 2, 1)),
			Premul: true,
			Out:    []uint8{0xff, 0xff, 0xff, 0xff},
		},
		{
			Image: (&image.Paletted{
				Pix:     []uint8{0, 64, 0, 0},
				Stride:  2,
				Rect:    image.Rect(0, 0, 2, 2),
				Palette: bigPalette,
			}).SubImage(image.Rect(1, 0, 2, 1)),
			Premul: false,
			Out:    []uint8{0xff, 0xff, 0xff, 0xff},
		},
	}
	for i, c := range cases {
		got := ebiten.ImageToBytes(c.Image, c.Premul)
		want := c.Out
		if !bytes.Equal(got, want) {
			t.Errorf("Test %d: got: %v, want: %v", i, got, want)
		}
	}
}

func BenchmarkImageToBytesRGBA(b *testing.B) {
	img := image.NewRGBA(image.Rect(0, 0, 4096, 4096))
	b.ResetTimer()
	for range b.N {
		ebiten.ImageToBytes(img, true)
	}
}

func BenchmarkImageToBytesNRGBA(b *testing.B) {
	img := image.NewNRGBA(image.Rect(0, 0, 4096, 4096))
	b.ResetTimer()
	for range b.N {
		ebiten.ImageToBytes(img, true)
	}
}

func BenchmarkImageToBytesPaletted(b *testing.B) {
	img := image.NewPaletted(image.Rect(0, 0, 4096, 4096), palette.Plan9)
	b.ResetTimer()
	for range b.N {
		ebiten.ImageToBytes(img, true)
	}
}