mirror of https://github.com/mindoc-org/mindoc.git
37 lines
698 B
Go
37 lines
698 B
Go
package graphics
|
|
|
|
import (
|
|
"image"
|
|
"image/gif"
|
|
"image/jpeg"
|
|
"image/png"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
)
|
|
|
|
// 将图片保存到指定的路径
|
|
func SaveImage(p string, src image.Image) error {
|
|
|
|
os.MkdirAll(filepath.Dir(p), 0666)
|
|
|
|
f, err := os.OpenFile(p, os.O_SYNC|os.O_RDWR|os.O_CREATE, 0666)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer f.Close()
|
|
ext := filepath.Ext(p)
|
|
|
|
if strings.EqualFold(ext, ".jpg") || strings.EqualFold(ext, ".jpeg") {
|
|
|
|
err = jpeg.Encode(f, src, &jpeg.Options{Quality: 80})
|
|
|
|
} else if strings.EqualFold(ext, ".png") {
|
|
err = png.Encode(f, src)
|
|
} else if strings.EqualFold(ext, ".gif") {
|
|
err = gif.Encode(f, src, &gif.Options{NumColors: 256})
|
|
}
|
|
return err
|
|
}
|