214 lines
6.0 KiB
Go
214 lines
6.0 KiB
Go
package arn
|
|
|
|
import (
|
|
"bytes"
|
|
"image"
|
|
"path"
|
|
"time"
|
|
|
|
"github.com/akyoto/imageserver"
|
|
)
|
|
|
|
const (
|
|
// AnimeImageLargeWidth is the minimum width in pixels of a large anime image.
|
|
AnimeImageLargeWidth = 250
|
|
|
|
// AnimeImageLargeHeight is the minimum height in pixels of a large anime image.
|
|
AnimeImageLargeHeight = 350
|
|
|
|
// AnimeImageMediumWidth is the minimum width in pixels of a medium anime image.
|
|
AnimeImageMediumWidth = 142
|
|
|
|
// AnimeImageMediumHeight is the minimum height in pixels of a medium anime image.
|
|
AnimeImageMediumHeight = 200
|
|
|
|
// AnimeImageSmallWidth is the minimum width in pixels of a small anime image.
|
|
AnimeImageSmallWidth = 55
|
|
|
|
// AnimeImageSmallHeight is the minimum height in pixels of a small anime image.
|
|
AnimeImageSmallHeight = 78
|
|
|
|
// AnimeImageWebPQuality is the WebP quality of anime images.
|
|
AnimeImageWebPQuality = 70
|
|
|
|
// AnimeImageJPEGQuality is the JPEG quality of anime images.
|
|
AnimeImageJPEGQuality = 70
|
|
|
|
// AnimeImageQualityBonusLowDPI ...
|
|
AnimeImageQualityBonusLowDPI = 10
|
|
|
|
// AnimeImageQualityBonusLarge ...
|
|
AnimeImageQualityBonusLarge = 5
|
|
|
|
// AnimeImageQualityBonusMedium ...
|
|
AnimeImageQualityBonusMedium = 10
|
|
|
|
// AnimeImageQualityBonusSmall ...
|
|
AnimeImageQualityBonusSmall = 10
|
|
)
|
|
|
|
// Define the anime image outputs
|
|
var animeImageOutputs = []imageserver.Output{
|
|
// Original at full size
|
|
&imageserver.OriginalFile{
|
|
Directory: path.Join(Root, "images/anime/original/"),
|
|
Width: 0,
|
|
Height: 0,
|
|
Quality: 0,
|
|
},
|
|
|
|
// JPEG - Large
|
|
&imageserver.JPEGFile{
|
|
Directory: path.Join(Root, "images/anime/large/"),
|
|
Width: AnimeImageLargeWidth,
|
|
Height: AnimeImageLargeHeight,
|
|
Quality: AnimeImageJPEGQuality + AnimeImageQualityBonusLowDPI + AnimeImageQualityBonusLarge,
|
|
},
|
|
|
|
// JPEG - Medium
|
|
&imageserver.JPEGFile{
|
|
Directory: path.Join(Root, "images/anime/medium/"),
|
|
Width: AnimeImageMediumWidth,
|
|
Height: AnimeImageMediumHeight,
|
|
Quality: AnimeImageJPEGQuality + AnimeImageQualityBonusLowDPI + AnimeImageQualityBonusMedium,
|
|
},
|
|
|
|
// JPEG - Small
|
|
&imageserver.JPEGFile{
|
|
Directory: path.Join(Root, "images/anime/small/"),
|
|
Width: AnimeImageSmallWidth,
|
|
Height: AnimeImageSmallHeight,
|
|
Quality: AnimeImageJPEGQuality + AnimeImageQualityBonusLowDPI + AnimeImageQualityBonusSmall,
|
|
},
|
|
|
|
// WebP - Large
|
|
&imageserver.WebPFile{
|
|
Directory: path.Join(Root, "images/anime/large/"),
|
|
Width: AnimeImageLargeWidth,
|
|
Height: AnimeImageLargeHeight,
|
|
Quality: AnimeImageWebPQuality + AnimeImageQualityBonusLowDPI + AnimeImageQualityBonusLarge,
|
|
},
|
|
|
|
// WebP - Medium
|
|
&imageserver.WebPFile{
|
|
Directory: path.Join(Root, "images/anime/medium/"),
|
|
Width: AnimeImageMediumWidth,
|
|
Height: AnimeImageMediumHeight,
|
|
Quality: AnimeImageWebPQuality + AnimeImageQualityBonusLowDPI + AnimeImageQualityBonusMedium,
|
|
},
|
|
|
|
// WebP - Small
|
|
&imageserver.WebPFile{
|
|
Directory: path.Join(Root, "images/anime/small/"),
|
|
Width: AnimeImageSmallWidth,
|
|
Height: AnimeImageSmallHeight,
|
|
Quality: AnimeImageWebPQuality + AnimeImageQualityBonusLowDPI + AnimeImageQualityBonusSmall,
|
|
},
|
|
}
|
|
|
|
// Define the high DPI anime image outputs
|
|
var animeImageOutputsHighDPI = []imageserver.Output{
|
|
// JPEG - Large
|
|
&imageserver.JPEGFile{
|
|
Directory: path.Join(Root, "images/anime/large/"),
|
|
Width: AnimeImageLargeWidth * 2,
|
|
Height: AnimeImageLargeHeight * 2,
|
|
Quality: AnimeImageJPEGQuality + AnimeImageQualityBonusLarge,
|
|
},
|
|
|
|
// JPEG - Medium
|
|
&imageserver.JPEGFile{
|
|
Directory: path.Join(Root, "images/anime/medium/"),
|
|
Width: AnimeImageMediumWidth * 2,
|
|
Height: AnimeImageMediumHeight * 2,
|
|
Quality: AnimeImageJPEGQuality + AnimeImageQualityBonusMedium,
|
|
},
|
|
|
|
// JPEG - Small
|
|
&imageserver.JPEGFile{
|
|
Directory: path.Join(Root, "images/anime/small/"),
|
|
Width: AnimeImageSmallWidth * 2,
|
|
Height: AnimeImageSmallHeight * 2,
|
|
Quality: AnimeImageJPEGQuality + AnimeImageQualityBonusSmall,
|
|
},
|
|
|
|
// WebP - Large
|
|
&imageserver.WebPFile{
|
|
Directory: path.Join(Root, "images/anime/large/"),
|
|
Width: AnimeImageLargeWidth * 2,
|
|
Height: AnimeImageLargeHeight * 2,
|
|
Quality: AnimeImageWebPQuality + AnimeImageQualityBonusLarge,
|
|
},
|
|
|
|
// WebP - Medium
|
|
&imageserver.WebPFile{
|
|
Directory: path.Join(Root, "images/anime/medium/"),
|
|
Width: AnimeImageMediumWidth * 2,
|
|
Height: AnimeImageMediumHeight * 2,
|
|
Quality: AnimeImageWebPQuality + AnimeImageQualityBonusMedium,
|
|
},
|
|
|
|
// WebP - Small
|
|
&imageserver.WebPFile{
|
|
Directory: path.Join(Root, "images/anime/small/"),
|
|
Width: AnimeImageSmallWidth * 2,
|
|
Height: AnimeImageSmallHeight * 2,
|
|
Quality: AnimeImageWebPQuality + AnimeImageQualityBonusSmall,
|
|
},
|
|
}
|
|
|
|
// AnimeImage ...
|
|
type AnimeImage struct {
|
|
Extension string `json:"extension"`
|
|
Width int `json:"width"`
|
|
Height int `json:"height"`
|
|
AverageColor HSLColor `json:"averageColor"`
|
|
LastModified int64 `json:"lastModified"`
|
|
}
|
|
|
|
// SetImageBytes accepts a byte buffer that represents an image file and updates the anime image.
|
|
func (anime *Anime) SetImageBytes(data []byte) error {
|
|
// Decode
|
|
img, format, err := image.Decode(bytes.NewReader(data))
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return anime.SetImage(&imageserver.MetaImage{
|
|
Image: img,
|
|
Format: format,
|
|
Data: data,
|
|
})
|
|
}
|
|
|
|
// SetImage sets the anime image to the given MetaImage.
|
|
func (anime *Anime) SetImage(metaImage *imageserver.MetaImage) error {
|
|
var lastError error
|
|
|
|
// Save the different image formats and sizes in low DPI
|
|
for _, output := range animeImageOutputs {
|
|
err := output.Save(metaImage, anime.ID)
|
|
|
|
if err != nil {
|
|
lastError = err
|
|
}
|
|
}
|
|
|
|
// Save the different image formats and sizes in high DPI
|
|
for _, output := range animeImageOutputsHighDPI {
|
|
err := output.Save(metaImage, anime.ID+"@2")
|
|
|
|
if err != nil {
|
|
lastError = err
|
|
}
|
|
}
|
|
|
|
anime.Image.Extension = metaImage.Extension()
|
|
anime.Image.Width = metaImage.Image.Bounds().Dx()
|
|
anime.Image.Height = metaImage.Image.Bounds().Dy()
|
|
anime.Image.AverageColor = GetAverageColor(metaImage.Image)
|
|
anime.Image.LastModified = time.Now().Unix()
|
|
return lastError
|
|
}
|