Goで画像をリサイズ or トリミングする

package main

import (
	"os"
	"image"
	"image/jpeg"  // JPEGを読み書きする場合
	// "image/png"  // PNGを読み書きする場合
	// "image/gif"  // GIFを読み書きする場合

	"golang.org/x/image/draw"
)

// 画像を読み取るための関数。
// ファイルパスを指定すると、画像データを返してくれる。
func LoadImage(path string) (image.Image, error) {
	f, err := os.Open(path)
	if err != nil {
		return nil, err
	}
	defer f.Close()

	img, err := jpeg.Decode(f)
	if err != nil {
		return nil, err
	}

	return img, nil
}

// 画像を保存する関数。
// 保存先のパスと画像データを渡すと保存してくれる。
func SaveImage(path string, img image.Image) error {
	f, err := os.Create(path)
	if err != nil {
		return err
	}
	defer f.Close()

	err = jpeg.Encode(f, img, &jpeg.Options{
		Quality: 80, // JPEGのクオリティ設定。省略するとjpeg.DefaultQualityの値（75）が使われる。
	})
	return err
}

func ResizeImage(img image.Image, width, height int) image.Image {
	// 欲しいサイズの画像を新しく作る
	newImage := image.NewRGBA(image.Rect(0, 0, width, height))

	// サイズを変更しながら画像をコピーする
	draw.BiLinear.Scale(newImage, newImage.Bounds(), img, img.Bounds(), draw.Over, nil)

	return newImage
}

func ResizeImageKeepAspect(img image.Image, size int) image.Image {
	// 画像のサイズを取得する
	width := img.Bounds().Max.X
	height := img.Bounds().Max.Y

	// 結果となる画像のサイズを計算する
	if width > height {
		height = height * size / width
		width = size
	} else {
		width = width * size / height
		height = size
	}

	// 先ほどの関数を使って画像をリサイズする
	return ResizeImage(img, width, height)
}

func TrimImage(img image.Image, top, left, width, height int) image.Image {
	// 新しい画像を用意する
	newImage := image.NewRGBA(image.Rect(0, 0, width, height))

	// 左上(top, left)から右下(top+width, left+height)までの範囲を、新しい画像にコピーする
	draw.BiLinear.Scale(newImage, newImage.Bounds(), img, image.Rect(left, top, width, height), draw.Over, nil)

	return newImage
}

func SquareTrimImage(img image.Image, size int) image.Image {
	// 画像のサイズを取得する
	width := img.Bounds().Max.X
	height := img.Bounds().Max.Y

	// 短辺の長さを取得する
	shorter := width
	if height < shorter {
		shorter = height
	}

	// 左上の座標を計算する
	top := (height - shorter) / 2
	left := (width - shorter) / 2

	// 新しい画像を用意する
	newImage := image.NewRGBA(image.Rect(0, 0, size, size))

	// 画像の中心を切り抜きつつ、最終的なサイズ(size × size)になるようにリサイズする
	draw.BiLinear.Scale(newImage, newImage.Bounds(), img, image.Rect(left, top, width-left, height-top), draw.Over, nil)

	return newImage
}

package main

import (
	"os"
	"image"
	"image/jpeg"
	"bytes"

	"github.com/dsoprea/go-jpeg-image-structure/v2"
)

func main() {
	// 元画像のExif情報を取得しておく
	jmp := jpegstructure.NewJpegMediaParser()
	original, err := jmp.ParseFile("input.jpg")
	if err != nil {
		panic(err)
	}
	exif, err := original.(*jpegstructure.SegmentList).ConstructExifBuilder()
	if err != nil {
		panic(err)
	}

	// 画像を読み込んで加工する
	img, _ := LoadImage("input.jpg")
	img = EditImage(img)

	// 作った画像をメモリ上でJPEGにエンコードする
	var buf bytes.Buffer
	err = jpeg.Encode(&buf, img, nil)
	if err != nil {
		panic(err)
	}

	// エンコード結果をdsprea/go-jpeg-image-structureでパースしなおす
	output, err := jmp.ParseBytes(buf.Bytes())
	if err != nil {
		panic(err)
	}
	outputSL := output.(*jpegstructure.SegmentList)

	// 元画像から取得しておいたExif情報を新しい画像にセットする。
	outputSL.SetExif(exif)

	// 新しい画像をファイルに保存する
	f, err := os.Create("output.jpg")
	if err != nil {
		panic(err)
	}
	defer f.Close()
	err = outputSL.Write(f)
	if err != nil {
		panic(err)
	}
}