mirror of
https://github.com/seaweedfs/seaweedfs.git
synced 2024-01-19 02:48:24 +00:00
37 lines
820 B
Go
37 lines
820 B
Go
|
package images
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"github.com/disintegration/imaging"
|
||
|
|
"image"
|
||
|
|
"image/gif"
|
||
|
|
"image/jpeg"
|
||
|
|
"image/png"
|
||
|
|
)
|
||
|
|
|
||
|
|
func Resized(ext string, data []byte, width, height int) (resized []byte) {
|
||
|
|
if width == 0 && height == 0 {
|
||
|
|
return data
|
||
|
|
}
|
||
|
|
if srcImage, _, err := image.Decode(bytes.NewReader(data)); err == nil {
|
||
|
|
bounds := srcImage.Bounds()
|
||
|
|
var dstImage *image.NRGBA
|
||
|
|
if width == height && bounds.Dx() != bounds.Dy() {
|
||
|
|
dstImage = imaging.Thumbnail(srcImage, width, height, imaging.Lanczos)
|
||
|
|
} else {
|
||
|
|
dstImage = imaging.Resize(srcImage, width, height, imaging.Lanczos)
|
||
|
|
}
|
||
|
|
var buf bytes.Buffer
|
||
|
|
switch ext {
|
||
|
|
case ".png":
|
||
|
|
png.Encode(&buf, dstImage)
|
||
|
|
case ".jpg":
|
||
|
|
jpeg.Encode(&buf, dstImage, nil)
|
||
|
|
case ".gif":
|
||
|
|
gif.Encode(&buf, dstImage, nil)
|
||
|
|
}
|
||
|
|
return buf.Bytes()
|
||
|
|
}
|
||
|
|
return data
|
||
|
|
}
|