parent
874012fe56
commit
4b69d2b8bf
@ -0,0 +1,194 @@
|
||||
package tools
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"image"
|
||||
"io/ioutil"
|
||||
"os"
|
||||
|
||||
"github.com/golang/freetype"
|
||||
|
||||
"github.com/nfnt/resize"
|
||||
|
||||
"image/color"
|
||||
"image/draw"
|
||||
_ "image/gif"
|
||||
_ "image/jpeg"
|
||||
"image/png"
|
||||
_ "image/png"
|
||||
|
||||
"b612.me/starainrt"
|
||||
)
|
||||
|
||||
func OpenImage(name string) (image.Image, error) {
|
||||
if !starainrt.Exists(name) {
|
||||
return nil, errors.New("File Not Exists")
|
||||
}
|
||||
fso, err := os.Open(name)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
img, _, err := image.Decode(fso)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return img, nil
|
||||
}
|
||||
|
||||
func MergePhoto(big, small image.Image, bigsize, smallsize uint, x, y int) image.Image {
|
||||
big = resize.Resize(bigsize, bigsize, big, resize.Lanczos3)
|
||||
small = resize.Resize(smallsize, smallsize, small, resize.Lanczos3)
|
||||
offset := image.Pt(x, y)
|
||||
b := big.Bounds()
|
||||
nimg := image.NewRGBA(b)
|
||||
draw.Draw(nimg, b, big, image.ZP, draw.Src)
|
||||
draw.Draw(nimg, small.Bounds(), small, offset, draw.Over)
|
||||
return nimg
|
||||
}
|
||||
|
||||
func SavePhoto(path string, img image.Image) error {
|
||||
imgf, err := os.Create(path)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer imgf.Close()
|
||||
return png.Encode(imgf, img)
|
||||
}
|
||||
|
||||
func SetAlpha(img image.Image, alpha uint8) image.Image {
|
||||
size := img.Bounds()
|
||||
nimg := image.NewRGBA(size)
|
||||
for x := 0; x < size.Dx(); x++ {
|
||||
for y := 0; y < size.Dy(); y++ {
|
||||
r, g, b, a := img.At(x, y).RGBA()
|
||||
r = r >> 8
|
||||
g = g >> 8
|
||||
b = b >> 8
|
||||
a = a >> 8
|
||||
nimg.Set(x, y, color.NRGBA{uint8(r), uint8(g), uint8(b), alpha})
|
||||
//nimg.Set(x, y, color.Alpha{alpha})
|
||||
//nimg.Set(x, y, img.At(x, y))
|
||||
}
|
||||
}
|
||||
return nimg
|
||||
}
|
||||
|
||||
func AddText(text string, img image.Image, x, y int, dpi, size float64, colors color.RGBA, ttf string) (image.Image, error) {
|
||||
if !starainrt.Exists(ttf) {
|
||||
return nil, errors.New("File Not Exists")
|
||||
}
|
||||
fontbyte, err := ioutil.ReadFile(ttf)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
font, err := freetype.ParseFont(fontbyte)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
nimg, ok := img.(draw.Image)
|
||||
if !ok {
|
||||
size := img.Bounds()
|
||||
nimg = image.NewRGBA(img.Bounds())
|
||||
for x := 0; x < size.Dx(); x++ {
|
||||
for y := 0; y < size.Dy(); y++ {
|
||||
r, g, b, a := img.At(x, y).RGBA()
|
||||
nimg.Set(x, y, color.NRGBA{uint8(r), uint8(g), uint8(b), uint8(a)})
|
||||
}
|
||||
}
|
||||
}
|
||||
f := freetype.NewContext()
|
||||
f.SetDPI(dpi)
|
||||
f.SetFontSize(size)
|
||||
f.SetFont(font)
|
||||
f.SetClip(nimg.Bounds())
|
||||
f.SetDst(nimg)
|
||||
f.SetSrc(image.NewUniform(colors))
|
||||
_, err = f.DrawString(text, freetype.Pt(x, y))
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return nimg, nil
|
||||
}
|
||||
|
||||
type TextImg struct {
|
||||
Text string
|
||||
X int
|
||||
Y int
|
||||
Dpi float64
|
||||
Size float64
|
||||
Color color.NRGBA
|
||||
}
|
||||
|
||||
type TextList struct {
|
||||
List []TextImg
|
||||
TTF []byte
|
||||
}
|
||||
|
||||
func AddListTests(list TextList, img image.Image) (image.Image, error) {
|
||||
font, err := freetype.ParseFont(list.TTF)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
nimg, ok := img.(draw.Image)
|
||||
if !ok {
|
||||
size := img.Bounds()
|
||||
nimg = image.NewRGBA(img.Bounds())
|
||||
for x := 0; x < size.Dx(); x++ {
|
||||
for y := 0; y < size.Dy(); y++ {
|
||||
r, g, b, a := img.At(x, y).RGBA()
|
||||
nimg.Set(x, y, color.NRGBA{uint8(r), uint8(g), uint8(b), uint8(a)})
|
||||
}
|
||||
}
|
||||
}
|
||||
for _, v := range list.List {
|
||||
f := freetype.NewContext()
|
||||
f.SetDPI(v.Dpi)
|
||||
f.SetFontSize(v.Size)
|
||||
f.SetFont(font)
|
||||
f.SetClip(nimg.Bounds())
|
||||
f.SetDst(nimg)
|
||||
f.SetSrc(image.NewUniform(v.Color))
|
||||
_, err = f.DrawString(v.Text, freetype.Pt(v.X, v.Y))
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
}
|
||||
return nimg, nil
|
||||
}
|
||||
|
||||
func AddTexts(text string, img image.Image, x, y int, dpi, size float64, colors color.NRGBA, ttf string) (image.Image, error) {
|
||||
if !starainrt.Exists(ttf) {
|
||||
return nil, errors.New("File Not Exists")
|
||||
}
|
||||
fontbyte, err := ioutil.ReadFile(ttf)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
font, err := freetype.ParseFont(fontbyte)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
nimg, ok := img.(draw.Image)
|
||||
if !ok {
|
||||
size := img.Bounds()
|
||||
nimg = image.NewRGBA(img.Bounds())
|
||||
for x := 0; x < size.Dx(); x++ {
|
||||
for y := 0; y < size.Dy(); y++ {
|
||||
r, g, b, a := img.At(x, y).RGBA()
|
||||
nimg.Set(x, y, color.NRGBA{uint8(r), uint8(g), uint8(b), uint8(a)})
|
||||
}
|
||||
}
|
||||
}
|
||||
f := freetype.NewContext()
|
||||
f.SetDPI(dpi)
|
||||
f.SetFontSize(size)
|
||||
f.SetFont(font)
|
||||
f.SetClip(nimg.Bounds())
|
||||
f.SetDst(nimg)
|
||||
f.SetSrc(image.NewUniform(colors))
|
||||
_, err = f.DrawString(text, freetype.Pt(x, y))
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return nimg, nil
|
||||
}
|
||||
@ -0,0 +1,83 @@
|
||||
package tools
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"image"
|
||||
|
||||
"b612.me/starlog"
|
||||
"github.com/spf13/cobra"
|
||||
)
|
||||
|
||||
func init() {
|
||||
imageCmd.AddCommand(imgMirrorCmd)
|
||||
Maincmd.AddCommand(imageCmd)
|
||||
}
|
||||
|
||||
var imageCmd = &cobra.Command{
|
||||
Use: "image",
|
||||
Short: "图像处理",
|
||||
Long: "简单的图像处理工具",
|
||||
Run: func(this *cobra.Command, args []string) {
|
||||
this.Help()
|
||||
},
|
||||
}
|
||||
|
||||
var imgMirrorCmd = &cobra.Command{
|
||||
Use: "mirror",
|
||||
Short: "图像镜像翻转",
|
||||
Long: "图像镜像翻转<水平>",
|
||||
Run: func(this *cobra.Command, args []string) {
|
||||
if len(args) == 0 {
|
||||
starlog.StdPrintln(0, starlog.RED, "请指定需要转换的图像！")
|
||||
return
|
||||
}
|
||||
for _, v := range args {
|
||||
img, err := OpenImage(v)
|
||||
if err != nil {
|
||||
starlog.StdPrintln(0, starlog.RED, err, v)
|
||||
continue
|
||||
}
|
||||
size := img.Bounds()
|
||||
nimg := image.NewRGBA(size)
|
||||
for x := 0; x < size.Dx(); x++ {
|
||||
for y := 0; y < size.Dy(); y++ {
|
||||
nimg.Set(size.Dx()-x, y, img.At(x, y))
|
||||
}
|
||||
}
|
||||
if err := SavePhoto(v, nimg); err != nil {
|
||||
starlog.StdPrintln(0, starlog.RED, err, v)
|
||||
continue
|
||||
} else {
|
||||
fmt.Println(v, "转换已完成！")
|
||||
}
|
||||
}
|
||||
fmt.Println("任务完成！")
|
||||
},
|
||||
}
|
||||
|
||||
var imgAlpha = &cobra.Command{
|
||||
Use: "alpha",
|
||||
Short: "设置透明度",
|
||||
Long: "设置alpha通道透明度",
|
||||
Run: func(this *cobra.Command, args []string) {
|
||||
if len(args) == 0 {
|
||||
starlog.StdPrintln(0, starlog.RED, "请指定需要转换的图像！")
|
||||
return
|
||||
}
|
||||
for _, v := range args {
|
||||
img, err := OpenImage(v)
|
||||
if err != nil {
|
||||
starlog.StdPrintln(0, starlog.RED, err, v)
|
||||
continue
|
||||
}
|
||||
img = SetAlpha(img, 4)
|
||||
if err := SavePhoto(v, img); err != nil {
|
||||
starlog.StdPrintln(0, starlog.RED, err, v)
|
||||
continue
|
||||
} else {
|
||||
fmt.Println(v, "转换已完成！")
|
||||
}
|
||||
}
|
||||
fmt.Println("任务完成！")
|
||||
},
|
||||
}
|
||||
Loading…
Reference in New Issue