barcode/palette_test.go at master · boombuler/barcode · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
package barcode_test

import (
	"bytes"
	"image/png"
	"testing"

	"github.com/boombuler/barcode"
	"github.com/boombuler/barcode/aztec"
	"github.com/boombuler/barcode/codabar"
	"github.com/boombuler/barcode/code128"
	"github.com/boombuler/barcode/code39"
	"github.com/boombuler/barcode/code93"
	"github.com/boombuler/barcode/datamatrix"
	"github.com/boombuler/barcode/ean"
	"github.com/boombuler/barcode/pdf417"
	"github.com/boombuler/barcode/qr"
	"github.com/boombuler/barcode/twooffive"
)

const (
	testWidth       = 900
	testHeight      = 300
	testPDF417Level = 3
)

type barcodeCase struct {
	name   string
	encode func(barcode.ColorScheme) (barcode.Barcode, error)
}

var barcodeCases = []barcodeCase{
	{
		name: "Aztec",
		encode: func(scheme barcode.ColorScheme) (barcode.Barcode, error) {
			return aztec.EncodeWithColor([]byte("1234567890"), 23, aztec.DEFAULT_LAYERS, scheme)
		},
	},
	{
		name: "Codabar",
		encode: func(scheme barcode.ColorScheme) (barcode.Barcode, error) {
			return codabar.EncodeWithColor("A12345B", scheme)
		},
	},
	{
		name: "Code128",
		encode: func(scheme barcode.ColorScheme) (barcode.Barcode, error) {
			return code128.EncodeWithColor("1234567890123456", scheme)
		},
	},
	{
		name: "Code39",
		encode: func(scheme barcode.ColorScheme) (barcode.Barcode, error) {
			return code39.EncodeWithColor("1234567890", false, false, scheme)
		},
	},
	{
		name: "Code93",
		encode: func(scheme barcode.ColorScheme) (barcode.Barcode, error) {
			return code93.EncodeWithColor("1234567890", false, false, scheme)
		},
	},
	{
		name: "DataMatrix",
		encode: func(scheme barcode.ColorScheme) (barcode.Barcode, error) {
			return datamatrix.EncodeWithColor("1234567890", scheme)
		},
	},
	{
		name: "EAN",
		encode: func(scheme barcode.ColorScheme) (barcode.Barcode, error) {
			return ean.EncodeWithColor("5901234123457", scheme)
		},
	},
	{
		name: "PDF417",
		encode: func(scheme barcode.ColorScheme) (barcode.Barcode, error) {
			return pdf417.EncodeWithColor("1234567890123456789012345678901234567890", testPDF417Level, scheme)
		},
	},
	{
		name: "QRCode",
		encode: func(scheme barcode.ColorScheme) (barcode.Barcode, error) {
			return qr.EncodeWithColor("1234567890", qr.M, qr.Auto, scheme)
		},
	},
	{
		name: "TwoOfFive",
		encode: func(scheme barcode.ColorScheme) (barcode.Barcode, error) {
			return twooffive.EncodeWithColor("1234567890", false, scheme)
		},
	},
}

func BenchmarkGenerateBarcode(b *testing.B) {
	variants := []struct {
		name   string
		scheme barcode.ColorScheme
	}{
		{name: "gray16", scheme: barcode.ColorScheme16},
		{name: "palette", scheme: barcode.ColorScheme1},
	}

	for _, tc := range barcodeCases {
		for _, variant := range variants {
			b.Run(tc.name+"/"+variant.name, func(b *testing.B) {
				b.ReportAllocs()
				pngSize := 0

				for i := 0; i < b.N; i++ {
					code, err := tc.encode(variant.scheme)
					if err != nil {
						b.Fatal(err)
					}

					scaled, err := barcode.Scale(code, testWidth, testHeight)
					if err != nil {
						b.Fatal(err)
					}

					var output bytes.Buffer
					if err := png.Encode(&output, scaled); err != nil {
						b.Fatal(err)
					}
					pngSize = output.Len()
				}

				b.ReportMetric(float64(pngSize), "png_B")
			})
		}
	}
}