gtsocial-umbx

resize.go (14148B)

---

 package imaging
 
 import (
 	"image"
 	"math"
 )
 
 type indexWeight struct {
 	index  int
 	weight float64
 }
 
 func precomputeWeights(dstSize, srcSize int, filter ResampleFilter) [][]indexWeight {
 	du := float64(srcSize) / float64(dstSize)
 	scale := du
 	if scale < 1.0 {
 		scale = 1.0
 	}
 	ru := math.Ceil(scale * filter.Support)
 
 	out := make([][]indexWeight, dstSize)
 	tmp := make([]indexWeight, 0, dstSize*int(ru+2)*2)
 
 	for v := 0; v < dstSize; v++ {
 		fu := (float64(v)+0.5)*du - 0.5
 
 		begin := int(math.Ceil(fu - ru))
 		if begin < 0 {
 			begin = 0
 		}
 		end := int(math.Floor(fu + ru))
 		if end > srcSize-1 {
 			end = srcSize - 1
 		}
 
 		var sum float64
 		for u := begin; u <= end; u++ {
 			w := filter.Kernel((float64(u) - fu) / scale)
 			if w != 0 {
 				sum += w
 				tmp = append(tmp, indexWeight{index: u, weight: w})
 			}
 		}
 		if sum != 0 {
 			for i := range tmp {
 				tmp[i].weight /= sum
 			}
 		}
 
 		out[v] = tmp
 		tmp = tmp[len(tmp):]
 	}
 
 	return out
 }
 
 // Resize resizes the image to the specified width and height using the specified resampling
 // filter and returns the transformed image. If one of width or height is 0, the image aspect
 // ratio is preserved.
 //
 // Example:
 //
 //	dstImage := imaging.Resize(srcImage, 800, 600, imaging.Lanczos)
 //
 func Resize(img image.Image, width, height int, filter ResampleFilter) *image.NRGBA {
 	dstW, dstH := width, height
 	if dstW < 0 || dstH < 0 {
 		return &image.NRGBA{}
 	}
 	if dstW == 0 && dstH == 0 {
 		return &image.NRGBA{}
 	}
 
 	srcW := img.Bounds().Dx()
 	srcH := img.Bounds().Dy()
 	if srcW <= 0 || srcH <= 0 {
 		return &image.NRGBA{}
 	}
 
 	// If new width or height is 0 then preserve aspect ratio, minimum 1px.
 	if dstW == 0 {
 		tmpW := float64(dstH) * float64(srcW) / float64(srcH)
 		dstW = int(math.Max(1.0, math.Floor(tmpW+0.5)))
 	}
 	if dstH == 0 {
 		tmpH := float64(dstW) * float64(srcH) / float64(srcW)
 		dstH = int(math.Max(1.0, math.Floor(tmpH+0.5)))
 	}
 
 	if filter.Support <= 0 {
 		// Nearest-neighbor special case.
 		return resizeNearest(img, dstW, dstH)
 	}
 
 	if srcW != dstW && srcH != dstH {
 		return resizeVertical(resizeHorizontal(img, dstW, filter), dstH, filter)
 	}
 	if srcW != dstW {
 		return resizeHorizontal(img, dstW, filter)
 	}
 	if srcH != dstH {
 		return resizeVertical(img, dstH, filter)
 	}
 	return Clone(img)
 }
 
 func resizeHorizontal(img image.Image, width int, filter ResampleFilter) *image.NRGBA {
 	src := newScanner(img)
 	dst := image.NewNRGBA(image.Rect(0, 0, width, src.h))
 	weights := precomputeWeights(width, src.w, filter)
 	parallel(0, src.h, func(ys <-chan int) {
 		scanLine := make([]uint8, src.w*4)
 		for y := range ys {
 			src.scan(0, y, src.w, y+1, scanLine)
 			j0 := y * dst.Stride
 			for x := range weights {
 				var r, g, b, a float64
 				for _, w := range weights[x] {
 					i := w.index * 4
 					s := scanLine[i : i+4 : i+4]
 					aw := float64(s[3]) * w.weight
 					r += float64(s[0]) * aw
 					g += float64(s[1]) * aw
 					b += float64(s[2]) * aw
 					a += aw
 				}
 				if a != 0 {
 					aInv := 1 / a
 					j := j0 + x*4
 					d := dst.Pix[j : j+4 : j+4]
 					d[0] = clamp(r * aInv)
 					d[1] = clamp(g * aInv)
 					d[2] = clamp(b * aInv)
 					d[3] = clamp(a)
 				}
 			}
 		}
 	})
 	return dst
 }
 
 func resizeVertical(img image.Image, height int, filter ResampleFilter) *image.NRGBA {
 	src := newScanner(img)
 	dst := image.NewNRGBA(image.Rect(0, 0, src.w, height))
 	weights := precomputeWeights(height, src.h, filter)
 	parallel(0, src.w, func(xs <-chan int) {
 		scanLine := make([]uint8, src.h*4)
 		for x := range xs {
 			src.scan(x, 0, x+1, src.h, scanLine)
 			for y := range weights {
 				var r, g, b, a float64
 				for _, w := range weights[y] {
 					i := w.index * 4
 					s := scanLine[i : i+4 : i+4]
 					aw := float64(s[3]) * w.weight
 					r += float64(s[0]) * aw
 					g += float64(s[1]) * aw
 					b += float64(s[2]) * aw
 					a += aw
 				}
 				if a != 0 {
 					aInv := 1 / a
 					j := y*dst.Stride + x*4
 					d := dst.Pix[j : j+4 : j+4]
 					d[0] = clamp(r * aInv)
 					d[1] = clamp(g * aInv)
 					d[2] = clamp(b * aInv)
 					d[3] = clamp(a)
 				}
 			}
 		}
 	})
 	return dst
 }
 
 // resizeNearest is a fast nearest-neighbor resize, no filtering.
 func resizeNearest(img image.Image, width, height int) *image.NRGBA {
 	dst := image.NewNRGBA(image.Rect(0, 0, width, height))
 	dx := float64(img.Bounds().Dx()) / float64(width)
 	dy := float64(img.Bounds().Dy()) / float64(height)
 
 	if dx > 1 && dy > 1 {
 		src := newScanner(img)
 		parallel(0, height, func(ys <-chan int) {
 			for y := range ys {
 				srcY := int((float64(y) + 0.5) * dy)
 				dstOff := y * dst.Stride
 				for x := 0; x < width; x++ {
 					srcX := int((float64(x) + 0.5) * dx)
 					src.scan(srcX, srcY, srcX+1, srcY+1, dst.Pix[dstOff:dstOff+4])
 					dstOff += 4
 				}
 			}
 		})
 	} else {
 		src := toNRGBA(img)
 		parallel(0, height, func(ys <-chan int) {
 			for y := range ys {
 				srcY := int((float64(y) + 0.5) * dy)
 				srcOff0 := srcY * src.Stride
 				dstOff := y * dst.Stride
 				for x := 0; x < width; x++ {
 					srcX := int((float64(x) + 0.5) * dx)
 					srcOff := srcOff0 + srcX*4
 					copy(dst.Pix[dstOff:dstOff+4], src.Pix[srcOff:srcOff+4])
 					dstOff += 4
 				}
 			}
 		})
 	}
 
 	return dst
 }
 
 // Fit scales down the image using the specified resample filter to fit the specified
 // maximum width and height and returns the transformed image.
 //
 // Example:
 //
 //	dstImage := imaging.Fit(srcImage, 800, 600, imaging.Lanczos)
 //
 func Fit(img image.Image, width, height int, filter ResampleFilter) *image.NRGBA {
 	maxW, maxH := width, height
 
 	if maxW <= 0 || maxH <= 0 {
 		return &image.NRGBA{}
 	}
 
 	srcBounds := img.Bounds()
 	srcW := srcBounds.Dx()
 	srcH := srcBounds.Dy()
 
 	if srcW <= 0 || srcH <= 0 {
 		return &image.NRGBA{}
 	}
 
 	if srcW <= maxW && srcH <= maxH {
 		return Clone(img)
 	}
 
 	srcAspectRatio := float64(srcW) / float64(srcH)
 	maxAspectRatio := float64(maxW) / float64(maxH)
 
 	var newW, newH int
 	if srcAspectRatio > maxAspectRatio {
 		newW = maxW
 		newH = int(float64(newW) / srcAspectRatio)
 	} else {
 		newH = maxH
 		newW = int(float64(newH) * srcAspectRatio)
 	}
 
 	return Resize(img, newW, newH, filter)
 }
 
 // Fill creates an image with the specified dimensions and fills it with the scaled source image.
 // To achieve the correct aspect ratio without stretching, the source image will be cropped.
 //
 // Example:
 //
 //	dstImage := imaging.Fill(srcImage, 800, 600, imaging.Center, imaging.Lanczos)
 //
 func Fill(img image.Image, width, height int, anchor Anchor, filter ResampleFilter) *image.NRGBA {
 	dstW, dstH := width, height
 
 	if dstW <= 0 || dstH <= 0 {
 		return &image.NRGBA{}
 	}
 
 	srcBounds := img.Bounds()
 	srcW := srcBounds.Dx()
 	srcH := srcBounds.Dy()
 
 	if srcW <= 0 || srcH <= 0 {
 		return &image.NRGBA{}
 	}
 
 	if srcW == dstW && srcH == dstH {
 		return Clone(img)
 	}
 
 	if srcW >= 100 && srcH >= 100 {
 		return cropAndResize(img, dstW, dstH, anchor, filter)
 	}
 	return resizeAndCrop(img, dstW, dstH, anchor, filter)
 }
 
 // cropAndResize crops the image to the smallest possible size that has the required aspect ratio using
 // the given anchor point, then scales it to the specified dimensions and returns the transformed image.
 //
 // This is generally faster than resizing first, but may result in inaccuracies when used on small source images.
 func cropAndResize(img image.Image, width, height int, anchor Anchor, filter ResampleFilter) *image.NRGBA {
 	dstW, dstH := width, height
 
 	srcBounds := img.Bounds()
 	srcW := srcBounds.Dx()
 	srcH := srcBounds.Dy()
 	srcAspectRatio := float64(srcW) / float64(srcH)
 	dstAspectRatio := float64(dstW) / float64(dstH)
 
 	var tmp *image.NRGBA
 	if srcAspectRatio < dstAspectRatio {
 		cropH := float64(srcW) * float64(dstH) / float64(dstW)
 		tmp = CropAnchor(img, srcW, int(math.Max(1, cropH)+0.5), anchor)
 	} else {
 		cropW := float64(srcH) * float64(dstW) / float64(dstH)
 		tmp = CropAnchor(img, int(math.Max(1, cropW)+0.5), srcH, anchor)
 	}
 
 	return Resize(tmp, dstW, dstH, filter)
 }
 
 // resizeAndCrop resizes the image to the smallest possible size that will cover the specified dimensions,
 // crops the resized image to the specified dimensions using the given anchor point and returns
 // the transformed image.
 func resizeAndCrop(img image.Image, width, height int, anchor Anchor, filter ResampleFilter) *image.NRGBA {
 	dstW, dstH := width, height
 
 	srcBounds := img.Bounds()
 	srcW := srcBounds.Dx()
 	srcH := srcBounds.Dy()
 	srcAspectRatio := float64(srcW) / float64(srcH)
 	dstAspectRatio := float64(dstW) / float64(dstH)
 
 	var tmp *image.NRGBA
 	if srcAspectRatio < dstAspectRatio {
 		tmp = Resize(img, dstW, 0, filter)
 	} else {
 		tmp = Resize(img, 0, dstH, filter)
 	}
 
 	return CropAnchor(tmp, dstW, dstH, anchor)
 }
 
 // Thumbnail scales the image up or down using the specified resample filter, crops it
 // to the specified width and hight and returns the transformed image.
 //
 // Example:
 //
 //	dstImage := imaging.Thumbnail(srcImage, 100, 100, imaging.Lanczos)
 //
 func Thumbnail(img image.Image, width, height int, filter ResampleFilter) *image.NRGBA {
 	return Fill(img, width, height, Center, filter)
 }
 
 // ResampleFilter specifies a resampling filter to be used for image resizing.
 //
 //	General filter recommendations:
 //
 //	- Lanczos
 //		A high-quality resampling filter for photographic images yielding sharp results.
 //
 //	- CatmullRom
 //		A sharp cubic filter that is faster than Lanczos filter while providing similar results.
 //
 //	- MitchellNetravali
 //		A cubic filter that produces smoother results with less ringing artifacts than CatmullRom.
 //
 //	- Linear
 //		Bilinear resampling filter, produces a smooth output. Faster than cubic filters.
 //
 //	- Box
 //		Simple and fast averaging filter appropriate for downscaling.
 //		When upscaling it's similar to NearestNeighbor.
 //
 //	- NearestNeighbor
 //		Fastest resampling filter, no antialiasing.
 //
 type ResampleFilter struct {
 	Support float64
 	Kernel  func(float64) float64
 }
 
 // NearestNeighbor is a nearest-neighbor filter (no anti-aliasing).
 var NearestNeighbor ResampleFilter
 
 // Box filter (averaging pixels).
 var Box ResampleFilter
 
 // Linear filter.
 var Linear ResampleFilter
 
 // Hermite cubic spline filter (BC-spline; B=0; C=0).
 var Hermite ResampleFilter
 
 // MitchellNetravali is Mitchell-Netravali cubic filter (BC-spline; B=1/3; C=1/3).
 var MitchellNetravali ResampleFilter
 
 // CatmullRom is a Catmull-Rom - sharp cubic filter (BC-spline; B=0; C=0.5).
 var CatmullRom ResampleFilter
 
 // BSpline is a smooth cubic filter (BC-spline; B=1; C=0).
 var BSpline ResampleFilter
 
 // Gaussian is a Gaussian blurring filter.
 var Gaussian ResampleFilter
 
 // Bartlett is a Bartlett-windowed sinc filter (3 lobes).
 var Bartlett ResampleFilter
 
 // Lanczos filter (3 lobes).
 var Lanczos ResampleFilter
 
 // Hann is a Hann-windowed sinc filter (3 lobes).
 var Hann ResampleFilter
 
 // Hamming is a Hamming-windowed sinc filter (3 lobes).
 var Hamming ResampleFilter
 
 // Blackman is a Blackman-windowed sinc filter (3 lobes).
 var Blackman ResampleFilter
 
 // Welch is a Welch-windowed sinc filter (parabolic window, 3 lobes).
 var Welch ResampleFilter
 
 // Cosine is a Cosine-windowed sinc filter (3 lobes).
 var Cosine ResampleFilter
 
 func bcspline(x, b, c float64) float64 {
 	var y float64
 	x = math.Abs(x)
 	if x < 1.0 {
 		y = ((12-9*b-6*c)*x*x*x + (-18+12*b+6*c)*x*x + (6 - 2*b)) / 6
 	} else if x < 2.0 {
 		y = ((-b-6*c)*x*x*x + (6*b+30*c)*x*x + (-12*b-48*c)*x + (8*b + 24*c)) / 6
 	}
 	return y
 }
 
 func sinc(x float64) float64 {
 	if x == 0 {
 		return 1
 	}
 	return math.Sin(math.Pi*x) / (math.Pi * x)
 }
 
 func init() {
 	NearestNeighbor = ResampleFilter{
 		Support: 0.0, // special case - not applying the filter
 	}
 
 	Box = ResampleFilter{
 		Support: 0.5,
 		Kernel: func(x float64) float64 {
 			x = math.Abs(x)
 			if x <= 0.5 {
 				return 1.0
 			}
 			return 0
 		},
 	}
 
 	Linear = ResampleFilter{
 		Support: 1.0,
 		Kernel: func(x float64) float64 {
 			x = math.Abs(x)
 			if x < 1.0 {
 				return 1.0 - x
 			}
 			return 0
 		},
 	}
 
 	Hermite = ResampleFilter{
 		Support: 1.0,
 		Kernel: func(x float64) float64 {
 			x = math.Abs(x)
 			if x < 1.0 {
 				return bcspline(x, 0.0, 0.0)
 			}
 			return 0
 		},
 	}
 
 	MitchellNetravali = ResampleFilter{
 		Support: 2.0,
 		Kernel: func(x float64) float64 {
 			x = math.Abs(x)
 			if x < 2.0 {
 				return bcspline(x, 1.0/3.0, 1.0/3.0)
 			}
 			return 0
 		},
 	}
 
 	CatmullRom = ResampleFilter{
 		Support: 2.0,
 		Kernel: func(x float64) float64 {
 			x = math.Abs(x)
 			if x < 2.0 {
 				return bcspline(x, 0.0, 0.5)
 			}
 			return 0
 		},
 	}
 
 	BSpline = ResampleFilter{
 		Support: 2.0,
 		Kernel: func(x float64) float64 {
 			x = math.Abs(x)
 			if x < 2.0 {
 				return bcspline(x, 1.0, 0.0)
 			}
 			return 0
 		},
 	}
 
 	Gaussian = ResampleFilter{
 		Support: 2.0,
 		Kernel: func(x float64) float64 {
 			x = math.Abs(x)
 			if x < 2.0 {
 				return math.Exp(-2 * x * x)
 			}
 			return 0
 		},
 	}
 
 	Bartlett = ResampleFilter{
 		Support: 3.0,
 		Kernel: func(x float64) float64 {
 			x = math.Abs(x)
 			if x < 3.0 {
 				return sinc(x) * (3.0 - x) / 3.0
 			}
 			return 0
 		},
 	}
 
 	Lanczos = ResampleFilter{
 		Support: 3.0,
 		Kernel: func(x float64) float64 {
 			x = math.Abs(x)
 			if x < 3.0 {
 				return sinc(x) * sinc(x/3.0)
 			}
 			return 0
 		},
 	}
 
 	Hann = ResampleFilter{
 		Support: 3.0,
 		Kernel: func(x float64) float64 {
 			x = math.Abs(x)
 			if x < 3.0 {
 				return sinc(x) * (0.5 + 0.5*math.Cos(math.Pi*x/3.0))
 			}
 			return 0
 		},
 	}
 
 	Hamming = ResampleFilter{
 		Support: 3.0,
 		Kernel: func(x float64) float64 {
 			x = math.Abs(x)
 			if x < 3.0 {
 				return sinc(x) * (0.54 + 0.46*math.Cos(math.Pi*x/3.0))
 			}
 			return 0
 		},
 	}
 
 	Blackman = ResampleFilter{
 		Support: 3.0,
 		Kernel: func(x float64) float64 {
 			x = math.Abs(x)
 			if x < 3.0 {
 				return sinc(x) * (0.42 - 0.5*math.Cos(math.Pi*x/3.0+math.Pi) + 0.08*math.Cos(2.0*math.Pi*x/3.0))
 			}
 			return 0
 		},
 	}
 
 	Welch = ResampleFilter{
 		Support: 3.0,
 		Kernel: func(x float64) float64 {
 			x = math.Abs(x)
 			if x < 3.0 {
 				return sinc(x) * (1.0 - (x * x / 9.0))
 			}
 			return 0
 		},
 	}
 
 	Cosine = ResampleFilter{
 		Support: 3.0,
 		Kernel: func(x float64) float64 {
 			x = math.Abs(x)
 			if x < 3.0 {
 				return sinc(x) * math.Cos((math.Pi/2.0)*(x/3.0))
 			}
 			return 0
 		},
 	}
 }