sklad/label/label.go

109 lines
2.8 KiB
Go

package label
import (
"image"
"image/color"
"image/draw"
"strings"
"janouch.name/sklad/bdf"
"janouch.name/sklad/imgutil"
"github.com/boombuler/barcode"
"github.com/boombuler/barcode/qr"
)
// TODO: Rename to GenQRLabelForHeight.
func GenLabelForHeight(font *bdf.Font,
text string, height, scale int) image.Image {
// Create a scaled bitmap of the text label.
textRect, _ := font.BoundString(text)
textImg := image.NewRGBA(textRect)
draw.Draw(textImg, textRect, image.White, image.ZP, draw.Src)
font.DrawString(textImg, image.ZP, color.Black, text)
scaledTextImg := imgutil.Scale{Image: textImg, Scale: scale}
scaledTextRect := scaledTextImg.Bounds()
remains := height - scaledTextRect.Dy() - 20
width := scaledTextRect.Dx()
if remains > width {
width = remains
}
// Create a scaled bitmap of the QR code.
qrImg, _ := qr.Encode(text, qr.H, qr.Auto)
qrImg, _ = barcode.Scale(qrImg, remains, remains)
qrRect := qrImg.Bounds()
// Combine.
combinedRect := image.Rect(0, 0, width, height)
combinedImg := image.NewRGBA(combinedRect)
draw.Draw(combinedImg, combinedRect, image.White, image.ZP, draw.Src)
draw.Draw(combinedImg,
combinedRect.Add(image.Point{X: (width - qrRect.Dx()) / 2, Y: 0}),
qrImg, image.ZP, draw.Src)
target := image.Rect(
(width-scaledTextRect.Dx())/2, qrRect.Dy()+20,
combinedRect.Max.X, combinedRect.Max.Y)
draw.Draw(combinedImg, target, &scaledTextImg, scaledTextRect.Min, draw.Src)
return combinedImg
}
func max(a, b int) int {
if a > b {
return a
}
return b
}
func GenLabelForWidth(font *bdf.Font,
text string, width, scale int) image.Image {
var lines []string
for _, line := range strings.Split(text, "\n") {
lines = append(lines, strings.TrimSuffix(line, "\r"))
}
// Respect font ascent and descent so that there are gaps between lines.
rects := make([]image.Rectangle, len(lines))
jumps := make([]int, len(lines))
for i, line := range lines {
r, _ := font.BoundString(line)
rects[i] = r
if i > 0 {
deficitD := font.Descent - rects[i-1].Max.Y
jumps[i] += max(0, deficitD)
deficitA := font.Ascent - (-r.Min.Y)
jumps[i] += max(0, deficitA)
}
}
height := 0
for i := range lines {
height += jumps[i] + rects[i].Dy()
}
imgRect := image.Rect(0, 0, width, height*scale)
img := image.NewRGBA(imgRect)
draw.Draw(img, imgRect, image.White, image.ZP, draw.Src)
y := 0
for i, line := range lines {
textImg := image.NewRGBA(rects[i])
draw.Draw(textImg, rects[i], image.White, image.ZP, draw.Src)
font.DrawString(textImg, image.ZP, color.Black, line)
scaledImg := imgutil.Scale{Image: textImg, Scale: scale}
scaledRect := scaledImg.Bounds()
y += jumps[i]
target := image.Rect(0, y*scale, imgRect.Max.X, imgRect.Max.Y)
draw.Draw(img, target, &scaledImg, scaledRect.Min, draw.Src)
y += rects[i].Dy()
}
return img
}