image.go

package codeview

import (
	"github.com/llgcode/draw2d/draw2dimg"
	"image"
)

type CodeView struct {
	text  Text
	title string
	scale float64
}

func NewCodeView() *CodeView {
	return &CodeView{
		scale: 1,
		text:  Text{},
		title: projectName,
	}
}

func (cv *CodeView) SetText(text Text) {
	cv.text = text
}

func (cv *CodeView) SetTitle(title string) {
	max := 32 - len(projectName)

	if len(title) > max {
		title = title[:max-3] + "..."
	}

	if projectName != "" {
		title = title + " - " + projectName
	}

	cv.title = title
}

func (cv *CodeView) SetScale(scale float64) {
	cv.scale = scale
}

func (cv *CodeView) Render() (image.Image, error) {
	return _render(cv.text, cv.title, cv.scale)
}

func (cv *CodeView) RenderToPng(path string) error {
	img, err := _render(cv.text, cv.title, cv.scale)
	if err != nil {
		return err
	}

	return draw2dimg.SaveToPngFile(path, img)
}