96 lines
2.6 KiB
Go
96 lines
2.6 KiB
Go
package nmslib
|
|
|
|
import (
|
|
"fmt"
|
|
"image/png"
|
|
"io/ioutil"
|
|
"os"
|
|
"strconv"
|
|
|
|
gim "github.com/ozankasikci/go-image-merge"
|
|
)
|
|
|
|
/*CreateBanner - Output PNG of Portal Glyphs
|
|
Requires 3 vars and returns 0 var, Outputs a PNG file
|
|
var portalex string
|
|
Portal Glyph hex string
|
|
var savename string
|
|
Output name of PNG file
|
|
var vopt int
|
|
1 enables vertical banner
|
|
0 enables horizontal banner
|
|
*/
|
|
func CreateBanner(portalhex string, savename string, vopt int) {
|
|
var err error
|
|
var GlyphHex [12]int64
|
|
var glyphImg [12]string
|
|
var vert int
|
|
var horz int
|
|
|
|
// Setup temp dir
|
|
tempdir, err := ioutil.TempDir("", "nmslib-")
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
defer os.RemoveAll(tempdir)
|
|
|
|
// verify len of portalhex
|
|
if len(portalhex) == 12 {
|
|
// get hex value from each digit in given string to an array of int64
|
|
for i := 0; i < len(portalhex); i++ {
|
|
GlyphHex[i], err = strconv.ParseInt(portalhex[i:int(i+1)], 16, 16)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
// assign image location of its glyph hex value to an array of strings
|
|
for j := 0; j < len(glyphImg); j++ {
|
|
glyphImg[j] = fmt.Sprintf("glyphs/GLYPH-%X.png", GlyphHex[j])
|
|
}
|
|
// pull images need from glyph.go and saved them to ./glyphs/
|
|
for k := 0; k < len(glyphImg); k++ {
|
|
data, err := Asset(glyphImg[k])
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
_, err2 := os.Stat(tempdir + "/glyphs/")
|
|
if os.IsNotExist(err2) {
|
|
errDir := os.MkdirAll(tempdir+"/glyphs", 0755)
|
|
if errDir != nil {
|
|
panic(errDir)
|
|
}
|
|
}
|
|
|
|
err = ioutil.WriteFile(tempdir+"/"+glyphImg[k], []byte(data), 0644)
|
|
}
|
|
}
|
|
// load images for processing using github.com/ozankasikci/go-image-merge
|
|
grids := []*gim.Grid{
|
|
{ImageFilePath: tempdir + "/" + glyphImg[0]}, {ImageFilePath: tempdir + "/" + glyphImg[1]},
|
|
{ImageFilePath: tempdir + "/" + glyphImg[2]}, {ImageFilePath: tempdir + "/" + glyphImg[3]},
|
|
{ImageFilePath: tempdir + "/" + glyphImg[4]}, {ImageFilePath: tempdir + "/" + glyphImg[5]},
|
|
{ImageFilePath: tempdir + "/" + glyphImg[6]}, {ImageFilePath: tempdir + "/" + glyphImg[7]},
|
|
{ImageFilePath: tempdir + "/" + glyphImg[8]}, {ImageFilePath: tempdir + "/" + glyphImg[9]},
|
|
{ImageFilePath: tempdir + "/" + glyphImg[10]}, {ImageFilePath: tempdir + "/" + glyphImg[11]},
|
|
}
|
|
|
|
if vopt == 1 { // set vertical rendering
|
|
vert, horz = 1, 12
|
|
} else { // set horizontal rendering (default)
|
|
vert, horz = 12, 1
|
|
}
|
|
|
|
// create new image from grids
|
|
rgba, err := gim.New(grids, vert, horz).Merge()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
// save the output to png
|
|
fmt.Printf("Saving %s to %s\n", portalhex, savename)
|
|
file, err := os.Create(savename)
|
|
err = png.Encode(file, rgba)
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
}
|
|
}
|