imgproxy/process.go

package main

/*
#cgo pkg-config: vips
#include "vips.h"
*/
import "C"

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  resizeType
	width   int
	height  int
	gravity gravityType
	enlarge bool
	format  imageType
}

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)
}

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 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(width), float64(height), float64(po.width), float64(po.height)

	wr := fow / fsw
	hr := foh / fsh

	if po.resize == FIT {
		return math.Min(wr, hr)
	}

	return math.Max(wr, hr)
}

func calcCrop(width, height int, po processingOptions) (left, top int) {
	left = (width - po.width + 1) / 2
	top = (height - po.height + 1) / 2

	if po.gravity == NORTH {
		top = 0
	}

	if po.gravity == EAST {
		left = width - po.width
	}

	if po.gravity == SOUTH {
		top = height - po.height
	}

	if po.gravity == WEST {
		left = 0
	}

	return
}

func processImage(data []byte, imgtype imageType, po processingOptions) ([]byte, error) {
	defer 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()))
}