137 lines
2.7 KiB
Go
137 lines
2.7 KiB
Go
package aoc
|
|
|
|
import (
|
|
"fmt"
|
|
"math"
|
|
"strings"
|
|
)
|
|
|
|
type Coordinate struct {
|
|
X, Y int
|
|
}
|
|
|
|
func NewCoordinate(x, y int) *Coordinate {
|
|
return &Coordinate{x, y}
|
|
}
|
|
|
|
func (c Coordinate) Relative(t Coordinate) Coordinate {
|
|
return Coordinate{X: c.X + t.X, Y: c.Y + t.Y}
|
|
}
|
|
|
|
func (c *Coordinate) MoveSouth() { c.Y++ }
|
|
func (c *Coordinate) MoveNorth() { c.Y-- }
|
|
func (c *Coordinate) MoveEast() { c.X++ }
|
|
func (c *Coordinate) MoveWest() { c.X-- }
|
|
func (c *Coordinate) MoveNE() {
|
|
c.X++
|
|
c.Y--
|
|
}
|
|
func (c *Coordinate) MoveSE() {
|
|
c.X++
|
|
c.Y++
|
|
}
|
|
func (c *Coordinate) MoveSW() {
|
|
c.X--
|
|
c.Y++
|
|
}
|
|
func (c *Coordinate) MoveNW() {
|
|
c.X--
|
|
c.Y--
|
|
}
|
|
|
|
func (c Coordinate) North() Coordinate {
|
|
return Coordinate{X: c.X, Y: c.Y - 1}
|
|
}
|
|
func (c Coordinate) East() Coordinate {
|
|
return Coordinate{X: c.X + 1, Y: c.Y}
|
|
}
|
|
func (c Coordinate) South() Coordinate {
|
|
return Coordinate{X: c.X, Y: c.Y + 1}
|
|
}
|
|
func (c Coordinate) West() Coordinate {
|
|
return Coordinate{X: c.X - 1, Y: c.Y}
|
|
}
|
|
func (c *Coordinate) NW() Coordinate {
|
|
return Coordinate{X: c.X - 1, Y: c.Y - 1}
|
|
}
|
|
func (c *Coordinate) NE() Coordinate {
|
|
return Coordinate{X: c.X + 1, Y: c.Y - 1}
|
|
}
|
|
func (c *Coordinate) SW() Coordinate {
|
|
return Coordinate{X: c.X - 1, Y: c.Y + 1}
|
|
}
|
|
func (c *Coordinate) SE() Coordinate {
|
|
return Coordinate{X: c.X + 1, Y: c.Y + 1}
|
|
}
|
|
func (c *Coordinate) GetOrthNeighbors() []Coordinate {
|
|
return []Coordinate{c.North(), c.East(), c.South(), c.West()}
|
|
}
|
|
|
|
func (c *Coordinate) GetNorthCoord() *Coordinate {
|
|
return &Coordinate{
|
|
X: c.X,
|
|
Y: c.Y - 1,
|
|
}
|
|
}
|
|
func (c *Coordinate) GetEastCoord() *Coordinate {
|
|
return &Coordinate{
|
|
X: c.X + 1,
|
|
Y: c.Y,
|
|
}
|
|
}
|
|
func (c *Coordinate) GetSouthCoord() *Coordinate {
|
|
return &Coordinate{
|
|
X: c.X,
|
|
Y: c.Y + 1,
|
|
}
|
|
}
|
|
func (c *Coordinate) GetWestCoord() *Coordinate {
|
|
return &Coordinate{
|
|
X: c.X - 1,
|
|
Y: c.Y,
|
|
}
|
|
}
|
|
|
|
func CoordinateFromString(str string) *Coordinate {
|
|
c := Coordinate{}
|
|
r := strings.NewReader(str)
|
|
_, err := fmt.Fscanf(r, "[%d, %d]", &c.X, &c.Y)
|
|
if err != nil {
|
|
_, err = fmt.Fscanf(r, "%d,%d", &c.X, &c.Y)
|
|
}
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return &c
|
|
}
|
|
|
|
func (c Coordinate) Angle(t Coordinate) float64 {
|
|
ret := math.Atan2(float64(t.X-c.X), float64(c.Y-t.Y)) * 180 / math.Pi
|
|
if ret < 0 {
|
|
ret = ret + 360
|
|
}
|
|
return ret
|
|
}
|
|
|
|
func (c Coordinate) String() string {
|
|
return fmt.Sprintf("[%d, %d]", c.X, c.Y)
|
|
}
|
|
|
|
func (c Coordinate) Distance(t Coordinate) int {
|
|
return ManhattanDistance(c.X, c.Y, t.X, t.Y)
|
|
}
|
|
|
|
func (c Coordinate) Equals(c2 Coordinate) bool {
|
|
return c.X == c2.X && c.Y == c2.Y
|
|
}
|
|
func (c Coordinate) Adjacent(c2 Coordinate) bool {
|
|
return c2.Equals(c.North()) ||
|
|
c2.Equals(c.NE()) ||
|
|
c2.Equals(c.East()) ||
|
|
c2.Equals(c.SE()) ||
|
|
c2.Equals(c.South()) ||
|
|
c2.Equals(c.SW()) ||
|
|
c2.Equals(c.West()) ||
|
|
c2.Equals(c.NW())
|
|
}
|