You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
62 lines
1.4 KiB
62 lines
1.4 KiB
package images
|
|
|
|
import (
|
|
"bytes"
|
|
"image"
|
|
"image/gif"
|
|
"image/jpeg"
|
|
"image/png"
|
|
|
|
"github.com/chrislusf/seaweedfs/weed/glog"
|
|
"github.com/disintegration/imaging"
|
|
)
|
|
|
|
func Resized(ext string, data []byte, width, height int) (resized []byte, w int, h int) {
|
|
if width == 0 && height == 0 {
|
|
return data, 0, 0
|
|
}
|
|
srcImage, _, err := image.Decode(bytes.NewReader(data))
|
|
|
|
dx := srcImage.Bounds().Dx()
|
|
dy := srcImage.Bounds().Dy()
|
|
|
|
if height == 0 {
|
|
height = width * dy / dx
|
|
} else if width == 0 {
|
|
width = height * dx / dy
|
|
} else {
|
|
if width/height > dx/dy { //定高
|
|
width = height * dx / dy
|
|
} else if width/height < dx/dy { //定宽
|
|
height = width * dy / dx
|
|
}
|
|
}
|
|
|
|
if err == nil {
|
|
bounds := srcImage.Bounds()
|
|
var dstImage *image.NRGBA
|
|
if bounds.Dx() > width && width != 0 || bounds.Dy() > height && height != 0 {
|
|
if width == height && bounds.Dx() != bounds.Dy() {
|
|
dstImage = imaging.Thumbnail(srcImage, width, height, imaging.Lanczos)
|
|
w, h = width, height
|
|
} else {
|
|
dstImage = imaging.Resize(srcImage, width, height, imaging.Lanczos)
|
|
}
|
|
} else {
|
|
return data, bounds.Dx(), bounds.Dy()
|
|
}
|
|
var buf bytes.Buffer
|
|
switch ext {
|
|
case ".png":
|
|
png.Encode(&buf, dstImage)
|
|
case ".jpg", ".jpeg":
|
|
jpeg.Encode(&buf, dstImage, nil)
|
|
case ".gif":
|
|
gif.Encode(&buf, dstImage, nil)
|
|
}
|
|
return buf.Bytes(), dstImage.Bounds().Dx(), dstImage.Bounds().Dy()
|
|
} else {
|
|
glog.Error(err)
|
|
}
|
|
return data, 0, 0
|
|
}
|