|
|
package main
|
|
|
|
|
|
import (
|
|
|
"fmt"
|
|
|
"log"
|
|
|
"os"
|
|
|
"strings"
|
|
|
|
|
|
"github.com/chubin/vt10x"
|
|
|
"github.com/fogleman/gg"
|
|
|
)
|
|
|
|
|
|
func StringSliceToRuneSlice(s string) [][]rune {
|
|
|
strings := strings.Split(s, "\n")
|
|
|
result := make([][]rune, len(strings))
|
|
|
|
|
|
i := 0
|
|
|
for _, str := range strings {
|
|
|
if len(str) == 0 {
|
|
|
continue
|
|
|
}
|
|
|
result[i] = []rune(str)
|
|
|
i++
|
|
|
}
|
|
|
|
|
|
return result
|
|
|
}
|
|
|
|
|
|
func maxRowLength(rows [][]rune) int {
|
|
|
maxLen := 0
|
|
|
for _, row := range rows {
|
|
|
if len(row) > maxLen {
|
|
|
maxLen = len(row)
|
|
|
}
|
|
|
}
|
|
|
return maxLen
|
|
|
}
|
|
|
|
|
|
func GeneratePng() {
|
|
|
runes := StringSliceToRuneSlice(`
|
|
|
Weather report: Hochstadt an der Aisch, Germany
|
|
|
|
|
|
\ / Partly cloudy
|
|
|
_ /"".-. +5(2) °C
|
|
|
\_( ). ↗ 9 km/h
|
|
|
/(___(__) 10 km
|
|
|
0.0 mm
|
|
|
┌─────────────┐
|
|
|
┌───────────────────────┤ Sat 11 Nov ├───────────────────────┐
|
|
|
│ Noon └──────┬──────┘ Night │
|
|
|
├──────────────────────────────┼──────────────────────────────┤
|
|
|
│ _'/"".-. Patchy rain po…│ _'/"".-. Patchy rain po…│
|
|
|
│ ,\_( ). +6(3) °C │ ,\_( ). +5(2) °C │
|
|
|
│ /(___(__) → 22-29 km/h │ /(___(__) ↗ 14-20 km/h │
|
|
|
│ ‘ ‘ ‘ ‘ 10 km │ ‘ ‘ ‘ ‘ 10 km │
|
|
|
│ ‘ ‘ ‘ ‘ 0.1 mm | 86% │ ‘ ‘ ‘ ‘ 0.0 mm | 89% │
|
|
|
└──────────────────────────────┴──────────────────────────────┘
|
|
|
┌─────────────┐
|
|
|
┌───────────────────────┤ Sun 12 Nov ├───────────────────────┐
|
|
|
│ Noon └──────┬──────┘ Night │
|
|
|
├──────────────────────────────┼──────────────────────────────┤
|
|
|
│ \ / Partly cloudy │ .-. Light drizzle │
|
|
|
│ _ /"".-. +8(7) °C │ ( ). +5(2) °C │
|
|
|
│ \_( ). ↑ 7-8 km/h │ (___(__) ↑ 13-18 km/h │
|
|
|
│ /(___(__) 10 km │ ‘ ‘ ‘ ‘ 2 km │
|
|
|
│ 0.0 mm | 0% │ ‘ ‘ ‘ ‘ 0.3 mm | 76% │
|
|
|
└──────────────────────────────┴──────────────────────────────┘
|
|
|
`)
|
|
|
|
|
|
// Dimensions of each rune in pixels
|
|
|
runeWidth := 8
|
|
|
runeHeight := 14
|
|
|
|
|
|
// Compute the width and height of the final image
|
|
|
imageWidth := runeWidth * maxRowLength(runes)
|
|
|
imageHeight := runeHeight * len(runes)
|
|
|
|
|
|
// Create a new context with the computed dimensions
|
|
|
dc := gg.NewContext(imageWidth, imageHeight)
|
|
|
|
|
|
// fontPath := "/usr/share/fonts/truetype/dejavu/DejaVuSansMono.ttf"
|
|
|
// fontPath := "/usr/share/fonts/truetype/wqy/wqy-zenhei.ttc"
|
|
|
fontPath := "/usr/share/fonts/truetype/lexi/LexiGulim.ttf"
|
|
|
|
|
|
err := dc.LoadFontFace(fontPath, 13)
|
|
|
if err != nil {
|
|
|
log.Fatal(err)
|
|
|
}
|
|
|
|
|
|
// Loop through each rune in the array and draw it on the context
|
|
|
for i, row := range runes {
|
|
|
for j, char := range row {
|
|
|
// Compute the x and y coordinates for drawing the current rune
|
|
|
x := float64(j*runeWidth + runeWidth/2)
|
|
|
y := float64(i*runeHeight + runeHeight/2)
|
|
|
|
|
|
// Set the appropriate color for the current rune
|
|
|
if char == '#' {
|
|
|
dc.SetRGB(0, 0, 0) // Black
|
|
|
} else if char == '@' {
|
|
|
dc.SetRGB(1, 0, 0) // Red
|
|
|
} else {
|
|
|
dc.SetRGB(1, 1, 1) // White
|
|
|
}
|
|
|
|
|
|
character := string(char)
|
|
|
// if char == ' ' {
|
|
|
// character = fmt.Sprint(j % 10)
|
|
|
// }
|
|
|
dc.DrawRectangle(x, y, x+float64(runeWidth), y+float64(runeHeight))
|
|
|
dc.Fill()
|
|
|
|
|
|
// Draw a rectangle with the rune's dimensions and color
|
|
|
dc.DrawString(character, x, y) // Draw the character centered on the canvas
|
|
|
// dc.DrawStringAnchored(character, x, y, 0.5, 0.5) // Draw the character centered on the canvas
|
|
|
}
|
|
|
}
|
|
|
|
|
|
// Save the image to a PNG file
|
|
|
err = dc.SavePNG("output.png")
|
|
|
if err != nil {
|
|
|
fmt.Println("Error saving PNG:", err)
|
|
|
return
|
|
|
}
|
|
|
|
|
|
fmt.Println("PNG generated successfully")
|
|
|
}
|
|
|
|
|
|
func GeneratePngFromANSI(input []byte, outputFile string) error {
|
|
|
// Dimensions of each rune in pixels
|
|
|
runeWidth := 8
|
|
|
runeHeight := 14
|
|
|
fontSize := 13.0
|
|
|
// fontPath := "/usr/share/fonts/truetype/dejavu/DejaVuSansMono.ttf"
|
|
|
fontPath := "/usr/share/fonts/truetype/wqy/wqy-zenhei.ttc"
|
|
|
|
|
|
imageCols := 80
|
|
|
imageRows := 25
|
|
|
|
|
|
// Compute the width and height of the final image
|
|
|
imageWidth := runeWidth * imageCols
|
|
|
imageHeight := runeHeight * imageRows
|
|
|
|
|
|
// Create terminal and feed it with input.
|
|
|
term := vt10x.New(vt10x.WithSize(imageCols, imageRows))
|
|
|
_, err := term.Write([]byte("\033[20h"))
|
|
|
if err != nil {
|
|
|
return fmt.Errorf("virtual terminal write error: %w", err)
|
|
|
}
|
|
|
|
|
|
_, err = term.Write(input)
|
|
|
if err != nil {
|
|
|
return fmt.Errorf("virtual terminal write error: %w", err)
|
|
|
}
|
|
|
|
|
|
// Create a new context with the computed dimensions
|
|
|
dc := gg.NewContext(imageWidth, imageHeight)
|
|
|
|
|
|
err = dc.LoadFontFace(fontPath, fontSize) // Set font size to 96
|
|
|
if err != nil {
|
|
|
return fmt.Errorf("error loading font: %w", err)
|
|
|
}
|
|
|
|
|
|
// Loop through each rune in the array and draw it on the context
|
|
|
for i := 0; i < imageRows; i++ {
|
|
|
for j := 0; j < imageCols; j++ {
|
|
|
// Compute the x and y coordinates for drawing the current rune
|
|
|
x := float64(j * runeWidth)
|
|
|
y := float64(i * runeHeight)
|
|
|
|
|
|
cell := term.Cell(j, i)
|
|
|
character := string(cell.Char)
|
|
|
|
|
|
dc.DrawRectangle(x, y, float64(runeWidth), float64(runeHeight))
|
|
|
bg := colorANSItoRGB(cell.BG)
|
|
|
dc.SetRGB(bg[0], bg[1], bg[2])
|
|
|
dc.Fill()
|
|
|
|
|
|
fg := colorANSItoRGB(cell.FG)
|
|
|
dc.SetRGB(fg[0], fg[1], fg[2])
|
|
|
|
|
|
// Draw a rectangle with the rune's dimensions and color
|
|
|
dc.DrawString(character, x, y+float64(runeHeight)-3) // Draw the character centered on the canvas
|
|
|
// dc.DrawStringAnchored(character, x, y, 0.5, 0.5) // Draw the character centered on the canvas
|
|
|
}
|
|
|
}
|
|
|
|
|
|
// Save the image to a PNG file
|
|
|
err = dc.SavePNG(outputFile)
|
|
|
if err != nil {
|
|
|
return fmt.Errorf("error saving png: %w", err)
|
|
|
}
|
|
|
|
|
|
return nil
|
|
|
}
|
|
|
|
|
|
func colorANSItoRGB(colorANSI vt10x.Color) [3]float64 {
|
|
|
defaultBG := vt10x.Color(0)
|
|
|
defaultFG := vt10x.Color(8)
|
|
|
|
|
|
if colorANSI == vt10x.DefaultFG {
|
|
|
colorANSI = defaultFG
|
|
|
}
|
|
|
if colorANSI == vt10x.DefaultBG {
|
|
|
colorANSI = defaultBG
|
|
|
}
|
|
|
|
|
|
if colorANSI > 255 {
|
|
|
return [3]float64{127, 127, 127}
|
|
|
}
|
|
|
return ansiColorsDB[colorANSI]
|
|
|
}
|
|
|
|
|
|
func main() {
|
|
|
data, err := os.ReadFile("zh-text.txt")
|
|
|
if err != nil {
|
|
|
log.Fatalln(err)
|
|
|
}
|
|
|
|
|
|
err = GeneratePngFromANSI(data, "output.png")
|
|
|
if err != nil {
|
|
|
log.Fatalln(err)
|
|
|
}
|
|
|
}
|
