scan.go

// 2017: This is the anti-aliasing algorithym from the golang
// translation of FreeType. It has been adapted for use by the rasterx package
// which can use either this for the scanner component or a scanner based on
// the golang.org/x/image/vector package.
//__
// Copyright 2010 The Freetype-Go Authors. All rights reserved.
// Use of this source code is governed by your choice of either the
// FreeType License or the GNU General Public License version 2 (or
// any later version), both of which can be found in the LICENSE file.
//_
// Package provides an anti-aliasing 2-D rasterizer.
// taken fron the larger Freetype suite of font-related packages, but the
// raster package is not specific to font rasterization, and can be used
// standalone without any other Freetype package.
// Rasterization is done by the same area/coverage accumulation algorithm as
// the Freetype "smooth" module, and the Anti-Grain Geometry library. A
// description of the area/coverage algorithm is at
// http://projects.tuxee.net/cl-vectors/section-the-cl-aa-algorithm

package scanFT

import (
	"image"
	"image/color"
	"math"

	"image/draw"

	"github.com/srwiley/rasterx"
	"golang.org/x/image/math/fixed"
)

type (

	// cell is part of a linked list (for a given yi co-ordinate) of accumulated
	// area/coverage for the pixel at (xi, yi).
	cell struct {
		xi          int
		area, cover int
		next        int
	}
	// Rasterizer converts a path to a raster using the grainless algorithm.
	ScannerFT struct {
		// If false, the behavior is to use the even-odd winding fill
		// rule during Rasterize.
		UseNonZeroWinding bool
		// An offset (in pixels) to the painted spans.
		Dx, Dy int

		// additional clipping rectangle
		clip image.Rectangle

		// The width of the Rasterizer. The height is implicit in len(cellIndex).
		width int

		// The current pen position.
		a fixed.Point26_6
		// The current cell and its area/coverage being accumulated.
		xi, yi      int
		area, cover int

		// Saved cells.
		cell []cell
		// Linked list of cells, one per row.
		cellIndex []int
		// Buffers.
		cellBuf                [256]cell
		cellIndexBuf           [64]int
		spanBuf                [64]Span
		Pntr                   *RGBAColFuncPainter
		minX, minY, maxX, maxY fixed.Int26_6 // keep track of bounds
	}
)

func (s *ScannerFT) set(a fixed.Point26_6) {
	if s.maxX < a.X {
		s.maxX = a.X
	}
	if s.maxY < a.Y {
		s.maxY = a.Y
	}
	if s.minX > a.X {
		s.minX = a.X
	}
	if s.minY > a.Y {
		s.minY = a.Y
	}
}

func (s *ScannerFT) SetWinding(useNonZeroWinding bool) {
	s.UseNonZeroWinding = useNonZeroWinding
}

func (s *ScannerFT) SetColor(clr interface{}) {
	switch c := clr.(type) {
	case color.Color:
		s.Pntr.SetColor(c)
		s.Pntr.colorFunc = nil
	case rasterx.ColorFunc:
		s.Pntr.colorFunc = c
	}
}

// findCell returns the index in r.cell for the cell corresponding to
// (r.xi, r.yi). The cell is created if necessary.
func (s *ScannerFT) findCell() int {
	if s.yi < 0 || s.yi >= len(s.cellIndex) {
		return -1
	}
	xi := s.xi
	if xi < 0 {
		xi = -1
	} else if xi > s.width {
		xi = s.width
	}
	i, prev := s.cellIndex[s.yi], -1
	for i != -1 && s.cell[i].xi <= xi {
		if s.cell[i].xi == xi {
			return i
		}
		i, prev = s.cell[i].next, i
	}
	c := len(s.cell)
	if c == cap(s.cell) {
		buf := make([]cell, c, 4*c)
		copy(buf, s.cell)
		s.cell = buf[0 : c+1]
	} else {
		s.cell = s.cell[0 : c+1]
	}
	s.cell[c] = cell{xi, 0, 0, i}
	if prev == -1 {
		s.cellIndex[s.yi] = c
	} else {
		s.cell[prev].next = c
	}
	return c
}

// saveCell saves any accumulated r.area/r.cover for (r.xi, r.yi).
func (s *ScannerFT) saveCell() {
	if s.area != 0 || s.cover != 0 {
		i := s.findCell()
		if i != -1 {
			s.cell[i].area += s.area
			s.cell[i].cover += s.cover
		}
		s.area = 0
		s.cover = 0
	}
}

// setCell sets the (xi, yi) cell that r is accumulating area/coverage for.
func (s *ScannerFT) setCell(xi, yi int) {
	if s.xi != xi || s.yi != yi {
		s.saveCell()
		s.xi, s.yi = xi, yi
	}
}

// scan accumulates area/coverage for the yi'th scanline, going from
// x0 to x1 in the horizontal direction (in 26.6 fixed point co-ordinates)
// and from y0f to y1f fractional vertical units within that scanline.
func (s *ScannerFT) scan(yi int, x0, y0f, x1, y1f fixed.Int26_6) {
	// Break the 26.6 fixed point X co-ordinates into integral and fractional parts.
	x0i := int(x0) / 64
	x0f := x0 - fixed.Int26_6(64*x0i)
	x1i := int(x1) / 64
	x1f := x1 - fixed.Int26_6(64*x1i)

	// A perfectly horizontal scan.
	if y0f == y1f {
		s.setCell(x1i, yi)
		return
	}
	dx, dy := x1-x0, y1f-y0f
	// A single cell scan.
	if x0i == x1i {
		s.area += int((x0f + x1f) * dy)
		s.cover += int(dy)
		return
	}
	// There are at least two cells. Apart from the first and last cells,
	// all intermediate cells go through the full width of the cell,
	// or 64 units in 26.6 fixed point format.
	var (
		p, q, edge0, edge1 fixed.Int26_6
		xiDelta            int
	)
	if dx > 0 {
		p, q = (64-x0f)*dy, dx
		edge0, edge1, xiDelta = 0, 64, 1
	} else {
		p, q = x0f*dy, -dx
		edge0, edge1, xiDelta = 64, 0, -1
	}
	yDelta, yRem := p/q, p%q
	if yRem < 0 {
		yDelta--
		yRem += q
	}
	// Do the first cell.
	xi, y := x0i, y0f
	s.area += int((x0f + edge1) * yDelta)
	s.cover += int(yDelta)
	xi, y = xi+xiDelta, y+yDelta
	s.setCell(xi, yi)
	if xi != x1i {
		// Do all the intermediate cells.
		p = 64 * (y1f - y + yDelta)
		fullDelta, fullRem := p/q, p%q
		if fullRem < 0 {
			fullDelta--
			fullRem += q
		}
		yRem -= q
		for xi != x1i {
			yDelta = fullDelta
			yRem += fullRem
			if yRem >= 0 {
				yDelta++
				yRem -= q
			}
			s.area += int(64 * yDelta)
			s.cover += int(yDelta)
			xi, y = xi+xiDelta, y+yDelta
			s.setCell(xi, yi)
		}
	}
	// Do the last cell.
	yDelta = y1f - y
	s.area += int((edge0 + x1f) * yDelta)
	s.cover += int(yDelta)
}

// Start starts a new path at the given point.
func (s *ScannerFT) Start(a fixed.Point26_6) {
	s.set(a)
	s.setCell(int(a.X/64), int(a.Y/64))
	s.a = a
}

// Line adds a linear segment to the current curve.
func (s *ScannerFT) Line(b fixed.Point26_6) {
	s.set(b)
	x0, y0 := s.a.X, s.a.Y
	x1, y1 := b.X, b.Y
	dx, dy := x1-x0, y1-y0
	// Break the 26.6 fixed point Y co-ordinates into integral and fractional
	// parts.
	y0i := int(y0) / 64
	y0f := y0 - fixed.Int26_6(64*y0i)
	y1i := int(y1) / 64
	y1f := y1 - fixed.Int26_6(64*y1i)

	if y0i == y1i {
		// There is only one scanline.
		s.scan(y0i, x0, y0f, x1, y1f)

	} else if dx == 0 {
		// This is a vertical line segment. We avoid calling r.scan and instead
		// manipulate r.area and r.cover directly.
		var (
			edge0, edge1 fixed.Int26_6
			yiDelta      int
		)
		if dy > 0 {
			edge0, edge1, yiDelta = 0, 64, 1
		} else {
			edge0, edge1, yiDelta = 64, 0, -1
		}
		x0i, yi := int(x0)/64, y0i
		x0fTimes2 := (int(x0) - (64 * x0i)) * 2
		// Do the first pixel.
		dcover := int(edge1 - y0f)
		darea := int(x0fTimes2 * dcover)
		s.area += darea
		s.cover += dcover
		yi += yiDelta
		s.setCell(x0i, yi)
		// Do all the intermediate pixels.
		dcover = int(edge1 - edge0)
		darea = int(x0fTimes2 * dcover)
		for yi != y1i {
			s.area += darea
			s.cover += dcover
			yi += yiDelta
			s.setCell(x0i, yi)
		}
		// Do the last pixel.
		dcover = int(y1f - edge0)
		darea = int(x0fTimes2 * dcover)
		s.area += darea
		s.cover += dcover

	} else {
		// There are at least two scanlines. Apart from the first and last
		// scanlines, all intermediate scanlines go through the full height of
		// the row, or 64 units in 26.6 fixed point format.
		var (
			p, q, edge0, edge1 fixed.Int26_6
			yiDelta            int
		)
		if dy > 0 {
			p, q = (64-y0f)*dx, dy
			edge0, edge1, yiDelta = 0, 64, 1
		} else {
			p, q = y0f*dx, -dy
			edge0, edge1, yiDelta = 64, 0, -1
		}
		xDelta, xRem := p/q, p%q
		if xRem < 0 {
			xDelta--
			xRem += q
		}
		// Do the first scanline.
		x, yi := x0, y0i
		s.scan(yi, x, y0f, x+xDelta, edge1)
		x, yi = x+xDelta, yi+yiDelta
		s.setCell(int(x)/64, yi)
		if yi != y1i {
			// Do all the intermediate scanlines.
			p = 64 * dx
			fullDelta, fullRem := p/q, p%q
			if fullRem < 0 {
				fullDelta--
				fullRem += q
			}
			xRem -= q
			for yi != y1i {
				xDelta = fullDelta
				xRem += fullRem
				if xRem >= 0 {
					xDelta++
					xRem -= q
				}
				s.scan(yi, x, edge0, x+xDelta, edge1)
				x, yi = x+xDelta, yi+yiDelta
				s.setCell(int(x)/64, yi)
			}
		}
		// Do the last scanline.
		s.scan(yi, x, edge0, x1, y1f)
	}
	// The next lineTo starts from b.
	s.a = b
}

// areaToAlpha converts an area value to a uint32 alpha value. A completely
// filled pixel corresponds to an area of 64*64*2, and an alpha of 0xffff. The
// conversion of area values greater than this depends on the winding rule:
// even-odd or non-zero.
func (s *ScannerFT) areaToAlpha(area int) uint32 {
	// The C Freetype implementation (version 2.3.12) does "alpha := area>>1"
	// without the +1. Round-to-nearest gives a more symmetric result than
	// round-down. The C implementation also returns 8-bit alpha, not 16-bit
	// alpha.
	a := (area + 1) >> 1
	if a < 0 {
		a = -a
	}
	alpha := uint32(a)
	if s.UseNonZeroWinding {
		if alpha > 0x0fff {
			alpha = 0x0fff
		}
	} else {
		alpha &= 0x1fff
		if alpha > 0x1000 {
			alpha = 0x2000 - alpha
		} else if alpha == 0x1000 {
			alpha = 0x0fff
		}
	}
	// alpha is now in the range [0x0000, 0x0fff]. Convert that 12-bit alpha to
	// 16-bit alpha.
	return alpha<<4 | alpha>>8
}

// Draw converts r's accumulated curves into Spans for p. The Spans passed
// to the painter are non-overlapping, and sorted by Y and then X. They all have non-zero
// width (and 0 <= X0 < X1 <= r.width) and non-zero A, except for the final
// Span, which has Y, X0, X1 and A all equal to zero.
func (s *ScannerFT) Draw() {
	s.saveCell()
	t := 0
	for yi := 0; yi < len(s.cellIndex); yi++ {
		xi, cover := 0, 0
		for c := s.cellIndex[yi]; c != -1; c = s.cell[c].next {
			if cover != 0 && s.cell[c].xi > xi {
				alpha := s.areaToAlpha(cover * 64 * 2)
				if alpha != 0 {
					xi0, xi1 := xi, s.cell[c].xi
					if xi0 < 0 {
						xi0 = 0
					}
					if xi1 >= s.width {
						xi1 = s.width
					}
					if xi0 < xi1 {
						s.spanBuf[t] = Span{yi + s.Dy, xi0 + s.Dx, xi1 + s.Dx, alpha}
						t++
					}
				}
			}
			cover += s.cell[c].cover
			alpha := s.areaToAlpha(cover*64*2 - s.cell[c].area)
			xi = s.cell[c].xi + 1
			if alpha != 0 {
				xi0, xi1 := s.cell[c].xi, xi
				if xi0 < 0 {
					xi0 = 0
				}
				if xi1 >= s.width {
					xi1 = s.width
				}
				if xi0 < xi1 {
					s.spanBuf[t] = Span{yi + s.Dy, xi0 + s.Dx, xi1 + s.Dx, alpha}
					t++
				}
			}
			if t > len(s.spanBuf)-2 {
				s.Pntr.Paint(s.spanBuf[:t], false, s.clip)
				t = 0
			}
		}
	}
	s.Pntr.Paint(s.spanBuf[:t], true, s.clip)
}

func (s *ScannerFT) GetPathExtent() fixed.Rectangle26_6 {
	//fmt.Println("Get path extent")
	return fixed.Rectangle26_6{fixed.Point26_6{s.minX, s.minY}, fixed.Point26_6{s.maxX, s.maxY}}
}

// Clear cancels any previous accumulated scans
func (s *ScannerFT) Clear() {
	s.a = fixed.Point26_6{}
	s.xi = 0
	s.yi = 0
	s.area = 0
	s.cover = 0
	s.cell = s.cell[:0]
	for i := 0; i < len(s.cellIndex); i++ {
		s.cellIndex[i] = -1
	}
	const mxfi = fixed.Int26_6(math.MaxInt32)
	s.minX, s.minY, s.maxX, s.maxY = mxfi, mxfi, -mxfi, -mxfi
}

// SetBounds sets the maximum width and height of the rasterized image and
// calls Clear. The width and height are in pixels, not fixed.Int26_6 units.
func (s *ScannerFT) SetBounds(width, height int) {
	if width < 0 {
		width = 0
	}
	if height < 0 {
		height = 0
	}
	s.width = width
	s.cell = s.cellBuf[:0]
	if height > len(s.cellIndexBuf) {
		s.cellIndex = make([]int, height)
	} else {
		s.cellIndex = s.cellIndexBuf[:height]
	}
	s.width = width
	s.Clear()
}

// SetClip sets an optional clipping rectangle to restrict rendering only to
// that region -- if size is 0 then ignored (set to image.ZR to clear)
func (s *ScannerFT) SetClip(rect image.Rectangle) {
	s.clip = rect
}

// NewScannerFT creates a new Scanner with the given bounds.
func NewScannerFT(width, height int, p *RGBAPainter) *ScannerFT {
	s := new(ScannerFT)
	s.Pntr = &RGBAColFuncPainter{RGBAPainter: *p}
	s.SetBounds(width, height)
	s.UseNonZeroWinding = true
	return s
}

// RGBAColFuncPainter is a Painter that paints Spans onto a *image.RGBA,
// and uses a color function as a the color source, or the composed RGBA
// paint func for a solid color
type RGBAColFuncPainter struct {
	RGBAPainter
	//Op draw.Op
	// cr, cg, cb and ca are the 16-bit color to paint the spans.
	colorFunc rasterx.ColorFunc
}

// Paint satisfies the Painter interface.
func (r *RGBAColFuncPainter) Paint(ss []Span, done bool, clip image.Rectangle) {
	if r.colorFunc == nil {
		r.RGBAPainter.Paint(ss, done, clip)
		return
	}
	b := r.Image.Bounds()
	if clip.Size() != image.ZP {
		b = b.Intersect(clip)
	}
	for _, s := range ss {
		if s.Y < b.Min.Y {
			continue
		}
		if s.Y >= b.Max.Y {
			return
		}
		if s.X0 < b.Min.X {
			s.X0 = b.Min.X
		}
		if s.X1 > b.Max.X {
			s.X1 = b.Max.X
		}
		if s.X0 >= s.X1 {
			continue
		}
		// This code mimics drawGlyphOver in $GOROOT/src/image/draw/draw.go.
		ma := s.Alpha
		const m = 1<<16 - 1
		i0 := (s.Y-r.Image.Rect.Min.Y)*r.Image.Stride + (s.X0-r.Image.Rect.Min.X)*4
		i1 := i0 + (s.X1-s.X0)*4
		cx := s.X0
		if r.Op == draw.Over {
			for i := i0; i < i1; i += 4 {
				rcr, rcg, rcb, rca := r.colorFunc(cx, s.Y).RGBA()
				//fmt.Println("rgb x y ", rcr, rcg, rcg, rca, cx, s.Y)
				cx++
				dr := uint32(r.Image.Pix[i+0])
				dg := uint32(r.Image.Pix[i+1])
				db := uint32(r.Image.Pix[i+2])
				da := uint32(r.Image.Pix[i+3])
				a := (m - (rca * ma / m)) * 0x101
				r.Image.Pix[i+0] = uint8((dr*a + rcr*ma) / m >> 8)
				r.Image.Pix[i+1] = uint8((dg*a + rcg*ma) / m >> 8)
				r.Image.Pix[i+2] = uint8((db*a + rcb*ma) / m >> 8)
				r.Image.Pix[i+3] = uint8((da*a + rca*ma) / m >> 8)
			}
		} else {
			for i := i0; i < i1; i += 4 {
				c := r.colorFunc(cx, s.Y)
				cx++
				rcr, rcb, rcg, rca := c.RGBA()

				r.Image.Pix[i+0] = uint8(rcr * ma / m >> 8)
				r.Image.Pix[i+1] = uint8(rcg * ma / m >> 8)
				r.Image.Pix[i+2] = uint8(rcb * ma / m >> 8)
				r.Image.Pix[i+3] = uint8(rca * ma / m >> 8)
			}
		}
	}
}