barcode/aztec/azteccode.go

package aztec

import (
	"bytes"
	"image"
	"image/color"

	"git.bbr-dev.info/brajkovic/barcode"
	"git.bbr-dev.info/brajkovic/barcode/utils"
)

type aztecCode struct {
	*utils.BitList
	size    int
	content []byte
	color   barcode.ColorScheme
}

func newAztecCode(size int, color barcode.ColorScheme) *aztecCode {
	return &aztecCode{utils.NewBitList(size * size), size, nil, barcode.ColorScheme16}
}

func (c *aztecCode) Content() string {
	return string(c.content)
}

func (c *aztecCode) Metadata() barcode.Metadata {
	return barcode.Metadata{barcode.TypeAztec, 2}
}

func (c *aztecCode) ColorModel() color.Model {
	return c.color.Model
}

func (c *aztecCode) ColorScheme() barcode.ColorScheme {
	return c.color
}

func (c *aztecCode) Bounds() image.Rectangle {
	return image.Rect(0, 0, c.size, c.size)
}

func (c *aztecCode) At(x, y int) color.Color {
	if c.GetBit(x*c.size + y) {
		return c.color.Foreground
	}
	return c.color.Background
}

func (c *aztecCode) set(x, y int) {
	c.SetBit(x*c.size+y, true)
}

func (c *aztecCode) string() string {
	buf := new(bytes.Buffer)
	for y := 0; y < c.size; y++ {
		for x := 0; x < c.size; x++ {
			if c.GetBit(x*c.size + y) {
				buf.WriteString("X ")
			} else {
				buf.WriteString("  ")
			}
		}
		buf.WriteRune('\n')
	}
	return buf.String()
}