120 lines
2.8 KiB
Go
120 lines
2.8 KiB
Go
package logic
|
||
|
||
import (
|
||
"github.com/dubbogo/gost/log/logger"
|
||
"github.com/rwcarlsen/goexif/exif"
|
||
"golang.org/x/image/bmp"
|
||
"image"
|
||
"image/jpeg"
|
||
"image/png"
|
||
"os"
|
||
"path"
|
||
"strings"
|
||
)
|
||
|
||
// ReadOrientation 方向判断
|
||
func ReadOrientation(filename string) (direct int, err error) {
|
||
file, err := os.Open(filename)
|
||
defer file.Close()
|
||
if err != nil {
|
||
logger.Error("failed to open file, err: ", err)
|
||
return
|
||
}
|
||
x, err := exif.Decode(file)
|
||
if err != nil {
|
||
logger.Error("failed to decode file, err: ", err)
|
||
return
|
||
}
|
||
orientation, err := x.Get(exif.Orientation)
|
||
if err != nil {
|
||
logger.Error("failed to orientation file, err: ", err)
|
||
return
|
||
}
|
||
direct, err = orientation.Int(0)
|
||
if err != nil {
|
||
logger.Error("failed to convert type file, err: ", err)
|
||
return
|
||
}
|
||
return
|
||
}
|
||
|
||
// 旋转90度
|
||
func rotate90(m image.Image) image.Image {
|
||
rotate90 := image.NewRGBA(image.Rect(0, 0, m.Bounds().Dy(), m.Bounds().Dx()))
|
||
// 矩阵旋转
|
||
for x := m.Bounds().Min.Y; x < m.Bounds().Max.Y; x++ {
|
||
for y := m.Bounds().Max.X - 1; y >= m.Bounds().Min.X; y-- {
|
||
// 设置像素点
|
||
rotate90.Set(m.Bounds().Max.Y-x, y, m.At(y, x))
|
||
}
|
||
}
|
||
return rotate90
|
||
}
|
||
|
||
// 旋转180度
|
||
func rotate180(m image.Image) image.Image {
|
||
rotate180 := image.NewRGBA(image.Rect(0, 0, m.Bounds().Dx(), m.Bounds().Dy()))
|
||
// 矩阵旋转
|
||
for x := m.Bounds().Min.X; x < m.Bounds().Max.X; x++ {
|
||
for y := m.Bounds().Min.Y; y < m.Bounds().Max.Y; y++ {
|
||
// 设置像素点
|
||
rotate180.Set(m.Bounds().Max.X-x, m.Bounds().Max.Y-y, m.At(x, y))
|
||
}
|
||
}
|
||
return rotate180
|
||
}
|
||
|
||
// 旋转270度
|
||
func rotate270(m image.Image) image.Image {
|
||
rotate270 := image.NewRGBA(image.Rect(0, 0, m.Bounds().Dy(), m.Bounds().Dx()))
|
||
// 矩阵旋转
|
||
for x := m.Bounds().Min.Y; x < m.Bounds().Max.Y; x++ {
|
||
for y := m.Bounds().Max.X - 1; y >= m.Bounds().Min.X; y-- {
|
||
// 设置像素点
|
||
rotate270.Set(x, m.Bounds().Max.X-y, m.At(y, x))
|
||
}
|
||
}
|
||
return rotate270
|
||
}
|
||
|
||
func MakeThumbnail(imagePath, savePath string) error {
|
||
prefix := strings.ToLower(path.Ext("./2.jpg"))
|
||
ori, err := ReadOrientation(imagePath)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
file, _ := os.Open(imagePath)
|
||
defer file.Close()
|
||
img, _, err := image.Decode(file)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
//苹果手机拍照的图片，会有方向属性Orientation，
|
||
//经过Decode和Encode，编码处理后，方向属性会丢失，导致图片被旋转
|
||
switch ori {
|
||
case 6: //90度图片旋转
|
||
img = rotate90(img)
|
||
case 3:
|
||
img = rotate180(img)
|
||
case 8:
|
||
img = rotate270(img)
|
||
}
|
||
newImg, _ := os.Create(savePath)
|
||
defer newImg.Close()
|
||
switch prefix {
|
||
case ".jpg", ".jpeg":
|
||
err = jpeg.Encode(newImg, img, &jpeg.Options{Quality: 100})
|
||
case "png":
|
||
err = png.Encode(newImg, img)
|
||
case "bmp":
|
||
err = bmp.Encode(newImg, img)
|
||
default:
|
||
err = jpeg.Encode(newImg, img, &jpeg.Options{Quality: 100})
|
||
}
|
||
if err != nil {
|
||
logger.Error("Encode err", err)
|
||
return err
|
||
}
|
||
return nil
|
||
}
|