adventofcode/helpers/coordinate.go

113 lines
2.3 KiB
Go
Raw Normal View History

2019-12-10 15:17:50 +00:00
package aoc
import (
"fmt"
"math"
"strings"
)
type Coordinate struct {
X, Y int
}
2019-12-16 23:31:11 +00:00
func NewCoordinate(x, y int) *Coordinate {
return &Coordinate{x, y}
}
2020-12-20 21:57:10 +00:00
func (c Coordinate) Relative(t Coordinate) Coordinate {
return Coordinate{X: c.X + t.X, Y: c.Y + t.Y}
}
func (c Coordinate) North() Coordinate {
2020-11-04 17:56:23 +00:00
return Coordinate{X: c.X, Y: c.Y - 1}
}
func (c Coordinate) East() Coordinate {
2020-11-04 17:56:23 +00:00
return Coordinate{X: c.X + 1, Y: c.Y}
}
func (c Coordinate) South() Coordinate {
2020-11-04 17:56:23 +00:00
return Coordinate{X: c.X, Y: c.Y + 1}
}
func (c Coordinate) West() Coordinate {
2020-11-04 17:56:23 +00:00
return Coordinate{X: c.X - 1, Y: c.Y}
}
2020-12-11 14:11:30 +00:00
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}
}
2022-12-13 16:08:27 +00:00
func (c *Coordinate) GetOrthNeighbors() []Coordinate {
return []Coordinate{c.North(), c.East(), c.South(), c.West()}
}
2020-11-04 17:56:23 +00:00
2019-12-16 23:31:11 +00:00
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,
}
}
2019-12-10 15:17:50 +00:00
func CoordinateFromString(str string) *Coordinate {
c := Coordinate{}
r := strings.NewReader(str)
2019-12-18 22:44:05 +00:00
_, err := fmt.Fscanf(r, "[%d, %d]", &c.X, &c.Y)
2019-12-10 15:17:50 +00:00
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 AbsInt(c.X-t.X) + AbsInt(c.Y-t.Y)
}
2019-12-16 23:31:11 +00:00
func (c Coordinate) Equals(c2 Coordinate) bool {
return c.X == c2.X && c.Y == c2.Y
}
2022-12-09 15:03:12 +00:00
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())
}