// Copyright ©2018 The Gonum Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

package vgpdf_test

import (
	"bytes"
	"fmt"
	"image/color"
	"image/png"
	"log"
	"os"
	"testing"

	"gonum.org/v1/plot"
	"gonum.org/v1/plot/cmpimg"
	"gonum.org/v1/plot/plotter"
	"gonum.org/v1/plot/vg"
	"gonum.org/v1/plot/vg/draw"
	"gonum.org/v1/plot/vg/vgpdf"
)

func TestEmbedFonts(t *testing.T) {
	for _, tc := range []struct {
		name  string
		embed bool
	}{
		{
			name:  "testdata/disable-embedded-fonts_golden.pdf",
			embed: false,
		},
		{
			name:  "testdata/enable-embedded-fonts_golden.pdf",
			embed: true,
		},
	} {
		t.Run(fmt.Sprintf("embed=%v", tc.embed), func(t *testing.T) {
			p := plot.New()

			pts := plotter.XYs{{X: 0, Y: 0}, {X: 0, Y: 1}, {X: 1, Y: 0}, {X: 1, Y: 1}}
			line, err := plotter.NewLine(pts)
			if err != nil {
				t.Fatalf("could not create line: %v", err)
			}
			p.Add(line)
			p.X.Label.Text = "X axis"
			p.Y.Label.Text = "Y axis"

			c := vgpdf.New(100, 100)

			// enable/disable embedding fonts
			c.EmbedFonts(tc.embed)
			p.Draw(draw.New(c))

			var buf bytes.Buffer
			_, err = c.WriteTo(&buf)
			if err != nil {
				t.Fatalf("could not write canvas: %v", err)
			}

			if *cmpimg.GenerateTestData {
				// Recreate Golden images and exit.
				err = os.WriteFile(tc.name, buf.Bytes(), 0o644)
				if err != nil {
					t.Fatal(err)
				}
				return
			}

			want, err := os.ReadFile(tc.name)
			if err != nil {
				t.Fatalf("failed to read golden plot: %v", err)
			}

			ok, err := cmpimg.Equal("pdf", buf.Bytes(), want)
			if err != nil {
				t.Fatalf("failed to run cmpimg test: %v", err)
			}

			if !ok {
				t.Fatalf("plot mismatch: %v", tc.name)
			}
		})
	}
}

func TestArc(t *testing.T) {
	pts := plotter.XYs{{X: 1, Y: 1}, {X: 2, Y: 2}}
	scat, err := plotter.NewScatter(pts)
	if err != nil {
		t.Fatal(err)
	}
	p := plot.New()
	p.Add(scat)

	c := vgpdf.New(100, 100)

	c.EmbedFonts(false)
	p.Draw(draw.New(c))

	if *cmpimg.GenerateTestData {
		// Recreate Golden images and exit.
		f, err := os.Create("testdata/arc_golden.pdf")
		if err != nil {
			t.Fatal(err)
		}
		defer f.Close()

		_, err = c.WriteTo(f)
		if err != nil {
			t.Fatalf("could not write canvas: %v", err)
		}
		return
	}

	f, err := os.Create("testdata/arc.pdf")
	if err != nil {
		t.Fatal(err)
	}
	defer f.Close()

	_, err = c.WriteTo(f)
	if err != nil {
		t.Fatalf("could not write canvas: %v", err)
	}

	err = f.Close()
	if err != nil {
		t.Fatal(err)
	}

	want, err := os.ReadFile("testdata/arc_golden.pdf")
	if err != nil {
		t.Fatalf("failed to read golden plot: %v", err)
	}

	got, err := os.ReadFile("testdata/arc.pdf")
	if err != nil {
		t.Fatalf("failed to read plot: %v", err)
	}

	ok, err := cmpimg.Equal("pdf", got, want)
	if err != nil {
		t.Fatalf("failed to run cmpimg test: %v", err)
	}

	if !ok {
		t.Fatalf("plot mismatch")
	}
}

func TestMultipage(t *testing.T) {
	cmpimg.CheckPlot(Example_multipage, t, "multipage.pdf")
}

func TestIssue540(t *testing.T) {
	p := plot.New()

	xys := plotter.XYs{
		plotter.XY{X: 0, Y: 0},
		plotter.XY{X: 1, Y: 1},
		plotter.XY{X: 2, Y: 2},
	}

	p.Title.Text = "My title"
	p.X.Tick.Label.Font.Size = 0 // hide X-axis labels
	p.Y.Tick.Label.Font.Size = 0 // hide Y-axis labels

	lines, points, err := plotter.NewLinePoints(xys)
	if err != nil {
		log.Fatal(err)
	}
	lines.Color = color.RGBA{B: 255, A: 255}

	p.Add(lines, points)
	p.Add(plotter.NewGrid())

	if *cmpimg.GenerateTestData {
		// Recreate Golden images and exit.
		err = p.Save(100, 100, "testdata/issue540_golden.pdf")
		if err != nil {
			t.Fatal(err)
		}
		return
	}

	err = p.Save(100, 100, "testdata/issue540.pdf")
	if err != nil {
		t.Fatal(err)
	}

	want, err := os.ReadFile("testdata/issue540_golden.pdf")
	if err != nil {
		t.Fatal(err)
	}

	got, err := os.ReadFile("testdata/issue540.pdf")
	if err != nil {
		t.Fatal(err)
	}

	ok, err := cmpimg.Equal("pdf", got, want)
	if err != nil {
		t.Fatal(err)
	}
	if !ok {
		t.Fatalf("images differ")
	}
}

func BenchmarkCanvas(b *testing.B) {
	p := plot.New()

	xys := plotter.XYs{
		plotter.XY{X: 0, Y: 0},
		plotter.XY{X: 1, Y: 1},
		plotter.XY{X: 2, Y: 2},
	}

	p.Title.Text = "My title"
	p.X.Tick.Label.Font.Size = 0 // hide X-axis labels
	p.Y.Tick.Label.Font.Size = 0 // hide Y-axis labels

	lines, points, err := plotter.NewLinePoints(xys)
	if err != nil {
		log.Fatal(err)
	}
	lines.Color = color.RGBA{B: 255, A: 255}

	p.Add(lines, points)
	p.Add(plotter.NewGrid())

	c := vgpdf.New(5*vg.Centimeter, 5*vg.Centimeter)
	d := draw.New(c)

	b.ResetTimer()
	for i := 0; i < b.N; i++ {
		p.Draw(d)
	}
}

func BenchmarkCanvasImage(b *testing.B) {
	c := vgpdf.New(5*vg.Centimeter, 5*vg.Centimeter)
	raw, err := os.ReadFile("../../plotter/testdata/gopher.png")
	if err != nil {
		b.Fatalf("could not read test image: %+v", err)
	}
	img, err := png.Decode(bytes.NewReader(raw))
	if err != nil {
		b.Fatalf("could not decode test image: %+v", err)
	}

	b.ResetTimer()
	for i := 0; i < b.N; i++ {
		c.DrawImage(vg.Rectangle{Max: vg.Point{X: 2, Y: 2}}, img)
	}
}