mirror of
https://github.com/imgproxy/imgproxy.git
synced 2025-10-11 04:32:29 +02:00
Drop bimb dependency
This commit is contained in:
DarthSim
288
process.go
288
process.go
@@ -1,91 +1,269 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"math"
|
||||
/*
|
||||
#cgo pkg-config: vips
|
||||
#include "vips.h"
|
||||
*/
|
||||
import "C"
|
||||
|
||||
"github.com/h2non/bimg"
|
||||
import (
|
||||
"errors"
|
||||
"log"
|
||||
"math"
|
||||
"runtime"
|
||||
"unsafe"
|
||||
)
|
||||
|
||||
type imageType int
|
||||
|
||||
const (
|
||||
UNKNOWN imageType = iota
|
||||
JPEG
|
||||
PNG
|
||||
WEBP
|
||||
GIF
|
||||
)
|
||||
|
||||
var imageTypes = map[string]imageType{
|
||||
"jpeg": JPEG,
|
||||
"jpg": JPEG,
|
||||
"png": PNG,
|
||||
"webp": WEBP,
|
||||
"gif": GIF,
|
||||
}
|
||||
|
||||
type gravityType int
|
||||
|
||||
const (
|
||||
CENTER gravityType = iota
|
||||
NORTH
|
||||
EAST
|
||||
SOUTH
|
||||
WEST
|
||||
SMART
|
||||
)
|
||||
|
||||
var gravityTypes = map[string]gravityType{
|
||||
"ce": CENTER,
|
||||
"no": NORTH,
|
||||
"ea": EAST,
|
||||
"so": SOUTH,
|
||||
"we": WEST,
|
||||
"sm": SMART,
|
||||
}
|
||||
|
||||
type resizeType int
|
||||
|
||||
const (
|
||||
FIT resizeType = iota
|
||||
FILL
|
||||
CROP
|
||||
)
|
||||
|
||||
var resizeTypes = map[string]resizeType{
|
||||
"fit": FIT,
|
||||
"fill": FILL,
|
||||
"crop": CROP,
|
||||
}
|
||||
|
||||
type processingOptions struct {
|
||||
resize string
|
||||
resize resizeType
|
||||
width int
|
||||
height int
|
||||
gravity bimg.Gravity
|
||||
gravity gravityType
|
||||
enlarge bool
|
||||
format bimg.ImageType
|
||||
format imageType
|
||||
}
|
||||
|
||||
var imageTypes = map[string]bimg.ImageType{
|
||||
"jpeg": bimg.JPEG,
|
||||
"jpg": bimg.JPEG,
|
||||
"png": bimg.PNG,
|
||||
"webp": bimg.WEBP,
|
||||
func initVips() {
|
||||
runtime.LockOSThread()
|
||||
defer runtime.UnlockOSThread()
|
||||
|
||||
if err := C.vips_initialize(); err != 0 {
|
||||
C.vips_shutdown()
|
||||
log.Fatalln("unable to start vips!")
|
||||
}
|
||||
|
||||
C.vips_concurrency_set(1)
|
||||
C.vips_cache_set_max_mem(100 * 1024 * 1024) // 100Mb
|
||||
C.vips_cache_set_max(500)
|
||||
}
|
||||
|
||||
var gravityTypes = map[string]bimg.Gravity{
|
||||
"ce": bimg.GravityCentre,
|
||||
"no": bimg.GravityNorth,
|
||||
"ea": bimg.GravityEast,
|
||||
"so": bimg.GravitySouth,
|
||||
"we": bimg.GravityWest,
|
||||
"sm": bimg.GravitySmart,
|
||||
func vipsTypeSupportedLoad(imgtype imageType) bool {
|
||||
switch imgtype {
|
||||
case JPEG:
|
||||
return int(C.vips_type_find_load_go(C.JPEG)) != 0
|
||||
case PNG:
|
||||
return int(C.vips_type_find_load_go(C.PNG)) != 0
|
||||
case WEBP:
|
||||
return int(C.vips_type_find_load_go(C.WEBP)) != 0
|
||||
case GIF:
|
||||
return int(C.vips_type_find_load_go(C.GIF)) != 0
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
func vipsTypeSupportedSave(imgtype imageType) bool {
|
||||
switch imgtype {
|
||||
case JPEG:
|
||||
return int(C.vips_type_find_save_go(C.JPEG)) != 0
|
||||
case PNG:
|
||||
return int(C.vips_type_find_save_go(C.PNG)) != 0
|
||||
case WEBP:
|
||||
return int(C.vips_type_find_save_go(C.WEBP)) != 0
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
func round(f float64) int {
|
||||
return int(f + .5)
|
||||
}
|
||||
func calcSize(size bimg.ImageSize, po processingOptions) (int, int) {
|
||||
if (po.width == size.Width && po.height == size.Height) || (po.resize != "fill" && po.resize != "fit") {
|
||||
return po.width, po.height
|
||||
|
||||
func calcScale(width, height int, po processingOptions) float64 {
|
||||
if (po.width == width && po.height == height) || (po.resize != FILL && po.resize != FIT) {
|
||||
return 1
|
||||
}
|
||||
|
||||
fsw, fsh, fow, foh := float64(size.Width), float64(size.Height), float64(po.width), float64(po.height)
|
||||
fsw, fsh, fow, foh := float64(width), float64(height), float64(po.width), float64(po.height)
|
||||
|
||||
wr := fow / fsw
|
||||
hr := foh / fsh
|
||||
|
||||
var rate float64
|
||||
if po.resize == "fit" {
|
||||
rate = math.Min(wr, hr)
|
||||
} else {
|
||||
rate = math.Max(wr, hr)
|
||||
if po.resize == FIT {
|
||||
return math.Min(wr, hr)
|
||||
}
|
||||
|
||||
return round(math.Min(fsw*rate, fow)), round(math.Min(fsh*rate, foh))
|
||||
return math.Max(wr, hr)
|
||||
}
|
||||
|
||||
func processImage(p []byte, po processingOptions) ([]byte, error) {
|
||||
var err error
|
||||
func calcCrop(width, height int, po processingOptions) (left, top int) {
|
||||
left = (width - po.width + 1) / 2
|
||||
top = (height - po.height + 1) / 2
|
||||
|
||||
img := bimg.NewImage(p)
|
||||
|
||||
size, err := img.Size()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
if po.gravity == NORTH {
|
||||
top = 0
|
||||
}
|
||||
|
||||
// Default options
|
||||
opts := bimg.Options{
|
||||
Interpolator: bimg.Bicubic,
|
||||
Quality: conf.Quality,
|
||||
Gravity: po.gravity,
|
||||
Enlarge: po.enlarge,
|
||||
Type: po.format,
|
||||
if po.gravity == EAST {
|
||||
left = width - po.width
|
||||
}
|
||||
|
||||
opts.Width, opts.Height = calcSize(size, po)
|
||||
|
||||
switch po.resize {
|
||||
case "fit":
|
||||
opts.Embed = true
|
||||
case "fill":
|
||||
opts.Embed = true
|
||||
opts.Crop = true
|
||||
case "crop":
|
||||
opts.Crop = true
|
||||
default:
|
||||
opts.Force = true
|
||||
if po.gravity == SOUTH {
|
||||
top = height - po.height
|
||||
}
|
||||
|
||||
return img.Process(opts)
|
||||
if po.gravity == WEST {
|
||||
left = 0
|
||||
}
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
func processImage(data []byte, imgtype imageType, po processingOptions) ([]byte, error) {
|
||||
defer runtime.KeepAlive(data)
|
||||
|
||||
err := C.int(0)
|
||||
|
||||
var img, tmpImg *C.struct__VipsImage
|
||||
|
||||
// Cleanup after all
|
||||
defer func() {
|
||||
C.vips_thread_shutdown()
|
||||
C.vips_error_clear()
|
||||
}()
|
||||
|
||||
// Load the image
|
||||
switch imgtype {
|
||||
case JPEG:
|
||||
err = C.vips_jpegload_buffer_go(unsafe.Pointer(&data[0]), C.size_t(len(data)), &img)
|
||||
case PNG:
|
||||
err = C.vips_pngload_buffer_go(unsafe.Pointer(&data[0]), C.size_t(len(data)), &img)
|
||||
case GIF:
|
||||
err = C.vips_gifload_buffer_go(unsafe.Pointer(&data[0]), C.size_t(len(data)), &img)
|
||||
case WEBP:
|
||||
err = C.vips_webpload_buffer_go(unsafe.Pointer(&data[0]), C.size_t(len(data)), &img)
|
||||
}
|
||||
if err != 0 {
|
||||
return nil, vipsError()
|
||||
}
|
||||
|
||||
imgWidth := int(img.Xsize)
|
||||
imgHeight := int(img.Ysize)
|
||||
|
||||
// Ensure we won't crop out of bounds
|
||||
if !po.enlarge || po.resize == CROP {
|
||||
if imgWidth < po.width {
|
||||
po.width = imgWidth
|
||||
}
|
||||
|
||||
if imgHeight < po.height {
|
||||
po.height = imgHeight
|
||||
}
|
||||
}
|
||||
|
||||
if po.width != imgWidth || po.height != imgHeight {
|
||||
// Resize image for "fill" and "fit"
|
||||
if po.resize == FILL || po.resize == FIT {
|
||||
scale := calcScale(imgWidth, imgHeight, po)
|
||||
err = C.vips_resize_go(img, &tmpImg, C.double(scale))
|
||||
C.g_object_unref(C.gpointer(img))
|
||||
img = tmpImg
|
||||
if err != 0 {
|
||||
return nil, vipsError()
|
||||
}
|
||||
}
|
||||
// Crop image for "fill" and "crop"
|
||||
if po.resize == FILL || po.resize == CROP {
|
||||
if po.gravity == SMART && C.vips_support_smartcrop() == 1 {
|
||||
err = C.vips_smartcrop_go(img, &tmpImg, C.int(po.width), C.int(po.height))
|
||||
C.g_object_unref(C.gpointer(img))
|
||||
img = tmpImg
|
||||
if err != 0 {
|
||||
return nil, vipsError()
|
||||
}
|
||||
} else {
|
||||
left, top := calcCrop(int(img.Xsize), int(img.Ysize), po)
|
||||
err = C.vips_extract_area_go(img, &tmpImg, C.int(left), C.int(top), C.int(po.width), C.int(po.height))
|
||||
C.g_object_unref(C.gpointer(img))
|
||||
img = tmpImg
|
||||
if err != 0 {
|
||||
return nil, vipsError()
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Convert to sRGB colour space
|
||||
err = C.vips_colourspace_go(img, &tmpImg, C.VIPS_INTERPRETATION_sRGB)
|
||||
C.g_object_unref(C.gpointer(img))
|
||||
img = tmpImg
|
||||
if err != 0 {
|
||||
return nil, vipsError()
|
||||
}
|
||||
|
||||
// Finally, save
|
||||
imgsize := C.size_t(0)
|
||||
var ptr unsafe.Pointer
|
||||
switch po.format {
|
||||
case JPEG:
|
||||
err = C.vips_jpegsave_go(img, &ptr, &imgsize, 1, C.int(conf.Quality), 0)
|
||||
case PNG:
|
||||
err = C.vips_pngsave_go(img, &ptr, &imgsize)
|
||||
case WEBP:
|
||||
err = C.vips_webpsave_go(img, &ptr, &imgsize, 1, C.int(conf.Quality))
|
||||
}
|
||||
if err != 0 {
|
||||
return nil, vipsError()
|
||||
}
|
||||
|
||||
C.g_object_unref(C.gpointer(img))
|
||||
|
||||
buf := C.GoBytes(ptr, C.int(imgsize))
|
||||
C.g_free(C.gpointer(ptr))
|
||||
|
||||
return buf, nil
|
||||
}
|
||||
|
||||
func vipsError() error {
|
||||
return errors.New(C.GoString(C.vips_error_buffer()))
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user