draw2d/path_storage.go

// Copyright 2010 The draw2d Authors. All rights reserved.
// created: 21/11/2010 by Laurent Le Goff

package draw2d

import (
	"fmt"
	"math"
)

type PathCmd int

const (
	MoveTo PathCmd = iota
	LineTo
	QuadCurveTo
	CubicCurveTo
	ArcTo
	Close
)

type Path struct {
	commands []PathCmd
	vertices []float64
	x, y     float64
}

func NewPathStorage() (p *Path) {
	p = new(Path)
	p.commands = make([]PathCmd, 0, 256)
	p.vertices = make([]float64, 0, 256)
	return
}

func (p *Path) Clear() {
	p.commands = p.commands[0:0]
	p.vertices = p.vertices[0:0]
	return
}

func (p *Path) appendToPath(cmd PathCmd, vertices ...float64) {
	p.commands = append(p.commands, cmd)
	p.vertices = append(p.vertices, vertices...)
}

func (src *Path) Copy() (dest *Path) {
	dest = new(Path)
	dest.commands = make([]PathCmd, len(src.commands))
	copy(dest.commands, src.commands)
	dest.vertices = make([]float64, len(src.vertices))
	copy(dest.vertices, src.vertices)
	return dest
}

func (p *Path) LastPoint() (x, y float64) {
	return p.x, p.y
}

func (p *Path) IsEmpty() bool {
	return len(p.commands) == 0
}

func (p *Path) Close() {
	p.appendToPath(Close)
}

func (p *Path) MoveTo(x, y float64) {
	p.appendToPath(MoveTo, x, y)

	p.x = x
	p.y = y
}

func (p *Path) RMoveTo(dx, dy float64) {
	x, y := p.LastPoint()
	p.MoveTo(x+dx, y+dy)
}

func (p *Path) LineTo(x, y float64) {
	if len(p.commands) == 0 { //special case when no move has been done
		p.MoveTo(0, 0)
	}
	p.appendToPath(LineTo, x, y)
	p.x = x
	p.y = y
}

func (p *Path) RLineTo(dx, dy float64) {
	x, y := p.LastPoint()
	p.LineTo(x+dx, y+dy)
}

func (p *Path) QuadCurveTo(cx, cy, x, y float64) {
	if len(p.commands) == 0 { //special case when no move has been done
		p.MoveTo(0, 0)
	}
	p.appendToPath(QuadCurveTo, cx, cy, x, y)
	p.x = x
	p.y = y
}

func (p *Path) RQuadCurveTo(dcx, dcy, dx, dy float64) {
	x, y := p.LastPoint()
	p.QuadCurveTo(x+dcx, y+dcy, x+dx, y+dy)
}

func (p *Path) CubicCurveTo(cx1, cy1, cx2, cy2, x, y float64) {
	if len(p.commands) == 0 { //special case when no move has been done
		p.MoveTo(0, 0)
	}
	p.appendToPath(CubicCurveTo, cx1, cy1, cx2, cy2, x, y)
	p.x = x
	p.y = y
}

func (p *Path) RCubicCurveTo(dcx1, dcy1, dcx2, dcy2, dx, dy float64) {
	x, y := p.LastPoint()
	p.CubicCurveTo(x+dcx1, y+dcy1, x+dcx2, y+dcy2, x+dx, y+dy)
}

func (p *Path) ArcTo(cx, cy, rx, ry, startAngle, angle float64) {
	endAngle := startAngle + angle
	clockWise := true
	if angle < 0 {
		clockWise = false
	}
	// normalize
	if clockWise {
		for endAngle < startAngle {
			endAngle += math.Pi * 2.0
		}
	} else {
		for startAngle < endAngle {
			startAngle += math.Pi * 2.0
		}
	}
	startX := cx + math.Cos(startAngle)*rx
	startY := cy + math.Sin(startAngle)*ry
	if len(p.commands) > 0 {
		p.LineTo(startX, startY)
	} else {
		p.MoveTo(startX, startY)
	}
	p.appendToPath(ArcTo, cx, cy, rx, ry, startAngle, angle)
	p.x = cx + math.Cos(endAngle)*rx
	p.y = cy + math.Sin(endAngle)*ry
}

func (p *Path) RArcTo(dcx, dcy, rx, ry, startAngle, angle float64) {
	x, y := p.LastPoint()
	p.ArcTo(x+dcx, y+dcy, rx, ry, startAngle, angle)
}

func (p *Path) String() string {
	s := ""
	j := 0
	for _, cmd := range p.commands {
		switch cmd {
		case MoveTo:
			s += fmt.Sprintf("MoveTo: %f, %f\n", p.vertices[j], p.vertices[j+1])
			j = j + 2
		case LineTo:
			s += fmt.Sprintf("LineTo: %f, %f\n", p.vertices[j], p.vertices[j+1])
			j = j + 2
		case QuadCurveTo:
			s += fmt.Sprintf("QuadCurveTo: %f, %f, %f, %f\n", p.vertices[j], p.vertices[j+1], p.vertices[j+2], p.vertices[j+3])
			j = j + 4
		case CubicCurveTo:
			s += fmt.Sprintf("CubicCurveTo: %f, %f, %f, %f, %f, %f\n", p.vertices[j], p.vertices[j+1], p.vertices[j+2], p.vertices[j+3], p.vertices[j+4], p.vertices[j+5])
			j = j + 6
		case ArcTo:
			s += fmt.Sprintf("ArcTo: %f, %f, %f, %f, %f, %f\n", p.vertices[j], p.vertices[j+1], p.vertices[j+2], p.vertices[j+3], p.vertices[j+4], p.vertices[j+5])
			j = j + 6
		case Close:
			s += "Close\n"
		}
	}
	return s
}