|
|
|
|
@ -14,7 +14,8 @@ import (
|
|
|
|
|
model "github.com/cloudreve/Cloudreve/v3/models"
|
|
|
|
|
"github.com/cloudreve/Cloudreve/v3/pkg/util"
|
|
|
|
|
|
|
|
|
|
"github.com/nfnt/resize"
|
|
|
|
|
//"github.com/nfnt/resize"
|
|
|
|
|
"golang.org/x/image/draw"
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
// Thumb 缩略图
|
|
|
|
|
@ -58,7 +59,8 @@ func NewThumbFromFile(file io.Reader, name string) (*Thumb, error) {
|
|
|
|
|
|
|
|
|
|
// GetThumb 生成给定最大尺寸的缩略图
|
|
|
|
|
func (image *Thumb) GetThumb(width, height uint) {
|
|
|
|
|
image.src = resize.Thumbnail(width, height, image.src, resize.Lanczos3)
|
|
|
|
|
//image.src = resize.Thumbnail(width, height, image.src, resize.Lanczos3)
|
|
|
|
|
image.src = Thumbnail(width, height, image.src)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// GetSize 获取图像尺寸
|
|
|
|
|
@ -81,6 +83,48 @@ func (image *Thumb) Save(path string) (err error) {
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Thumbnail will downscale provided image to max width and height preserving
|
|
|
|
|
// original aspect ratio and using the interpolation function interp.
|
|
|
|
|
// It will return original image, without processing it, if original sizes
|
|
|
|
|
// are already smaller than provided constraints.
|
|
|
|
|
func Thumbnail(maxWidth, maxHeight uint, img image.Image) image.Image {
|
|
|
|
|
origBounds := img.Bounds()
|
|
|
|
|
origWidth := uint(origBounds.Dx())
|
|
|
|
|
origHeight := uint(origBounds.Dy())
|
|
|
|
|
newWidth, newHeight := origWidth, origHeight
|
|
|
|
|
|
|
|
|
|
// Return original image if it have same or smaller size as constraints
|
|
|
|
|
if maxWidth >= origWidth && maxHeight >= origHeight {
|
|
|
|
|
return img
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Preserve aspect ratio
|
|
|
|
|
if origWidth > maxWidth {
|
|
|
|
|
newHeight = uint(origHeight * maxWidth / origWidth)
|
|
|
|
|
if newHeight < 1 {
|
|
|
|
|
newHeight = 1
|
|
|
|
|
}
|
|
|
|
|
newWidth = maxWidth
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if newHeight > maxHeight {
|
|
|
|
|
newWidth = uint(newWidth * maxHeight / newHeight)
|
|
|
|
|
if newWidth < 1 {
|
|
|
|
|
newWidth = 1
|
|
|
|
|
}
|
|
|
|
|
newHeight = maxHeight
|
|
|
|
|
}
|
|
|
|
|
return Resize(newWidth, newHeight, img)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func Resize(newWidth, newHeight uint, img image.Image) image.Image {
|
|
|
|
|
// Set the expected size that you want:
|
|
|
|
|
dst := image.NewRGBA(image.Rect(0, 0, int(newWidth), int(newHeight)))
|
|
|
|
|
// Resize:
|
|
|
|
|
draw.BiLinear.Scale(dst, dst.Rect, img, img.Bounds(), draw.Src, nil)
|
|
|
|
|
return dst
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// CreateAvatar 创建头像
|
|
|
|
|
func (image *Thumb) CreateAvatar(uid uint) error {
|
|
|
|
|
// 读取头像相关设定
|
|
|
|
|
@ -92,7 +136,8 @@ func (image *Thumb) CreateAvatar(uid uint) error {
|
|
|
|
|
// 生成头像缩略图
|
|
|
|
|
src := image.src
|
|
|
|
|
for k, size := range []int{s, m, l} {
|
|
|
|
|
image.src = resize.Resize(uint(size), uint(size), src, resize.Lanczos3)
|
|
|
|
|
//image.src = resize.Resize(uint(size), uint(size), src, resize.Lanczos3)
|
|
|
|
|
image.src = Resize(uint(size), uint(size), src)
|
|
|
|
|
err := image.Save(filepath.Join(savePath, fmt.Sprintf("avatar_%d_%d.png", uid, k)))
|
|
|
|
|
if err != nil {
|
|
|
|
|
return err
|
|
|
|
|
|
kikoqiu
4 years ago