tools/util/image/text.go

package image

import (
	"fmt"
	"gopkg.in/gographics/imagick.v3/imagick"
	"icloudapp.cn/tools/util"
	"strings"
)

type Text struct {
	//字体内容
	Text string
	//字体
	Font string
	//字体大小
	FontSize float64
	//字体格式
	FontFamily string
	//文字颜色
	Color string
	//文字区域背景颜色
	Background string
	//文字背景色
	UnderColor string
	//下划线
	UnderLine bool
	//删除线
	Strike bool
	//斜体
	Italic bool
	//粗体
	Bold bool
	//字体间隙
	WordSpace int
	//是否自动换行
	Wrap bool
	//对齐方式 left center right
	Align string
	//坐标
	Coordinate *Point
	//文字区域大小
	ViewSize *Size

	Angle int64
}

// NewText text对象
func NewText(font string, fontSize float64, fontFamily string) *Text {
	t := &Text{
		Font:       font,
		FontSize:   fontSize,
		FontFamily: fontFamily,
	}
	t.Clean()
	return t
}

func (t *Text) SetText(text string) {
	t.Text = text
}

func (t *Text) SetFont(font string) {
	t.Font = font
}

func (t *Text) SetFontSize(fontSize float64) {
	t.FontSize = fontSize
}

func (t *Text) SetFontFamily(fontFamily string) {
	t.FontFamily = fontFamily
}

func (t *Text) SetColor(color string) {
	t.Color = color
}

func (t *Text) SetBackground(background string) {
	t.Background = background
}

func (t *Text) SetUnderColor(underColor string) {
	t.UnderColor = underColor
}

func (t *Text) SetUnderLine(underLine bool) {
	t.UnderLine = underLine
}

func (t *Text) SetStrike(strike bool) {
	t.Strike = strike
}

func (t *Text) SetBius(bius string) {
	t.SetItalic(false)
	t.SetBold(false)
	t.SetUnderLine(false)
	t.SetStrike(false)
	for _, v := range bius {
		if string(v) == "i" {
			t.SetItalic(true)
		}
		if string(v) == "b" {
			t.SetBold(true)
		}
		if string(v) == "u" {
			t.SetUnderLine(true)
		}
		if string(v) == "s" {
			t.SetStrike(true)
		}
	}

}

func (t *Text) SetItalic(italic bool) {
	t.Italic = italic
}

func (t *Text) SetBold(bold bool) {
	t.Bold = bold
}

func (t *Text) SetWordSpace(wordSpace int) {
	t.WordSpace = wordSpace
}

func (t *Text) SetWrap(wrap bool) {
	t.Wrap = wrap
}

func (t *Text) SetAlign(align string) {
	t.Align = align
}

func (t *Text) SetPoint(x, y float64) {
	t.Coordinate = NewPoint(x, y)
}

func (t *Text) SetSize(w, h uint) {
	t.ViewSize = NewSize(w, h)
}

func (t *Text) SetAngle(angle int64) {
	t.Angle = angle
}

func (t *Text) Create() (*imagick.MagickWand, error) {
	txtDraw := imagick.NewDrawingWand()
	if t.Font != "" {
		if err := txtDraw.SetFont(fmt.Sprintf("upload/fonts/%s", t.Font)); err != nil {
			return nil, util.NewError("txtDraw.SetFont", err.Error())
		}
	}

	if t.FontSize == 0 {
		t.FontSize = 12
	}

	txtDraw.SetFontSize(t.FontSize)

	if t.FontFamily != "" {
		if err := txtDraw.SetFontFamily(t.FontFamily); err != nil {
			return nil, util.NewError("txtDraw.SetFont", err.Error())
		}
	}
	if t.Color != "" {
		tColor := Pixel(t.Color)
		//tColor.SetOpacity(0)
		txtDraw.SetFillColor(tColor)
	}

	if t.Italic || t.Bold || t.UnderLine || t.Strike {
		txtDraw.SetStrokeColor(Pixel(t.Color))
	}

	if t.Italic {
		txtDraw.SkewX(-12)
	}

	txtDraw.SetFontStyle(imagick.STYLE_NORMAL)
	if t.Bold {
		//txtDraw.SetStrokeWidth(1)
		txtDraw.SetFontStyle(imagick.STYLE_OBLIQUE)
	}

	if t.UnderLine {
		txtDraw.SetTextDecoration(imagick.DECORATION_UNDERLINE)
	}

	if t.Background != "" {
		tb := Pixel(t.Background)
		//tb.SetOpacity(0)
		txtDraw.SetTextUnderColor(tb)
	}
	txtDraw.SetTextEncoding("UTF-8")

	if t.Wrap {
		txtArr := t.WapText(txtDraw, t.Text, t.Wrap)
		fmt.Println("txtArr", txtArr)
	}

	//透明图片
	transPNG := util.TransparentPNG()
	subMw := imagick.NewMagickWand()

	if err := subMw.ReadImageBlob(transPNG); err != nil {
		return nil, util.NewError("subMw.ReadImageBlob", err.Error())
	}

	if err := subMw.SetImageFormat("PNG"); err != nil {
		return nil, util.NewError("subMw.SetImageFormat", err.Error())
	}
	if err := subMw.ResizeImage(t.ViewSize.Width, t.ViewSize.Height, imagick.FILTER_LANCZOS); err != nil {
		return nil, util.NewError("subMw.ResizeImage", err.Error())
	}

	metrics := subMw.QueryFontMetrics(txtDraw, t.Text)
	alignX := 0.0
	if t.Align == "center" {
		alignX = float64(int(((float64(t.ViewSize.Width)-metrics.TextWidth)/2)*100)) / 100
	}

	if t.Align == "right" {
		alignX = float64(t.ViewSize.Width) - metrics.TextWidth
	}

	if t.Strike {
		txtDraw.SetStrokeDashOffset(metrics.TextHeight)
		txtDraw.SetTextDecoration(imagick.DECORATION_LINE_THROUGH)
	}

	txtDraw.Annotation(alignX, metrics.Ascender, t.Text)
	if t.Background != "" {
		if err := subMw.SetImageBackgroundColor(Pixel(t.Background)); err != nil {
			return nil, util.NewError("subMw.SetImageBackgroundColor", err.Error())
		}
	}
	if err := subMw.DrawImage(txtDraw); err != nil {
		return nil, util.NewError("subMw.DrawImage", err.Error())
	}

	parentMw := imagick.NewMagickWand()

	if err := parentMw.ReadImageBlob(transPNG); err != nil {
		return nil, util.NewError("parentMw.ReadImageBlob", err.Error())
	}
	if err := parentMw.ResizeImage(t.ViewSize.Width, t.ViewSize.Height, imagick.FILTER_LANCZOS); err != nil {
		return nil, util.NewError("subMw.ResizeImage", err.Error())
	}
	if err := parentMw.SetSize(t.ViewSize.Width, t.ViewSize.Height); err != nil {
		return nil, util.NewError("subMw.SetSize", err.Error())
	}

	if err := parentMw.CompositeImage(subMw, imagick.COMPOSITE_OP_OVER, true, 0, 0); err != nil {
		return nil, util.NewError("parentMw.CompositeImage ", err.Error())
	}
	pixBackground := Pixel("#ffffffff")
	if err := parentMw.RotateImage(pixBackground, float64(t.Angle)); err != nil {
		return nil, util.NewError("subMw.RotateImage", err.Error())
	}
	//由于图片旋转，图片的长宽和坐标位置发生变化，所以需要重置坐标位置
	if t.Angle > 0 {
		width := parentMw.GetImageWidth()
		height := parentMw.GetImageHeight()
		y := t.Coordinate.Y - float64(height)/3 + float64(t.ViewSize.Height)/2
		x := t.Coordinate.X - float64(width)/2 + float64(t.ViewSize.Width)/2
		t.Coordinate.X = x
		t.Coordinate.Y = y
		t.ViewSize.Width = width
		t.ViewSize.Height = height
	}

	return parentMw, nil
}

func (t *Text) WapText(dw *imagick.DrawingWand, text string, autoWrap bool) []string {
	queryMw := imagick.NewMagickWand()
	metrics := queryMw.QueryFontMetrics(dw, t.Text)
	fmt.Println("metrics", metrics)
	var lines []string
	if uint(metrics.TextWidth) <= t.ViewSize.Width {
		lines = append(lines, text)
	}
	if autoWrap {
		txtRune := []rune(text)
		var tmpRune []string
		for i, v := range txtRune {
			tmpRune = append(tmpRune, string(v))
			metrics = queryMw.QueryFontMetrics(dw, strings.Join(tmpRune, ""))
			if uint(metrics.TextWidth) > t.ViewSize.Width {
				fmt.Println("tmpRune", tmpRune)
				lines = append(lines, strings.Join(tmpRune[0:i-1], ""))
				tmpRune = []string{}
			}
		}
		if len(tmpRune) > 0 {
			lines = append(lines, strings.Join(tmpRune, ""))
		}
	}
	return lines
}

func (t *Text) Clean() {
	t.SetColor("#ffffff00")
	t.SetBackground("#ffffffff")
	t.SetSize(0, 0)
	t.SetPoint(0, 0)
	t.SetAlign("left")
	t.SetBius("")
	t.SetAngle(0)
}