motdGOi/figlet/render.go

package figlet

import (
	//"github.com/fatih/color"
	"errors"
	"strings"
)

type RenderOptions struct {
	FontName string
}

func NewRenderOptions() *RenderOptions {
	opt := &RenderOptions{}
	opt.FontName = "default"
	return opt
}

type AsciiRender struct {
	fontMgr *fontManager
}

func NewAsciiRender() *AsciiRender {
	this := &AsciiRender{}

	this.fontMgr = newFontManager()
	return this
}

// walk through the path, load all the *.flf font file
func (this *AsciiRender) LoadFont(fontPath string) error {
	return this.fontMgr.loadFont(fontPath)
}

// render with default options
func (this *AsciiRender) Render(asciiStr string) (string, error) {
	return this.render(asciiStr, NewRenderOptions())
}

// render with options
func (this *AsciiRender) RenderOpts(asciiStr string, opts *RenderOptions) (string, error) {
	return this.render(asciiStr, opts)
}

func (this *AsciiRender) convertChar(font *font, char rune) ([]string, error) {

	if char < 0 || char > 127 {
		return nil, errors.New("Not Ascii")
	}

	height := font.height
	begintRow := (int(char) - 32) * height

	word := make([]string, height, height)

	for i := 0; i < height; i++ {
		row := font.fontSlice[begintRow+i]
		row = strings.Replace(row, "@", "", -1)
		row = strings.Replace(row, font.hardblank, " ", -1)
		word[i] = row
	}

	return word, nil
}

func (this *AsciiRender) render(asciiStr string, opt *RenderOptions) (string, error) {

	font, _ := this.fontMgr.getFont(opt.FontName)

	wordlist := make([][]string, 0)
	for _, char := range asciiStr {
		word, err := this.convertChar(font, char)
		if err != nil {
			return "", err
		}
		wordlist = append(wordlist, word)
	}

	result := ""

	for i := 0; i < font.height; i++ {
		for j := 0; j < len(wordlist); j++ {
			result += wordlist[j][i]
		}
		result += "\n"
	}
	return result, nil
}