rove/pkg/vector/vector.go

package vector

import (
	"math"

	"github.com/mdiluz/rove/pkg/maths"
)

// Vector desribes a 3D vector
type Vector struct {
	X int `json:"x"`
	Y int `json:"y"`
}

// Add adds one vector to another
func (v *Vector) Add(v2 Vector) {
	v.X += v2.X
	v.Y += v2.Y
}

// Added calculates a new vector
func (v Vector) Added(v2 Vector) Vector {
	v.Add(v2)
	return v
}

// Negated returns a negated vector
func (v Vector) Negated() Vector {
	return Vector{-v.X, -v.Y}
}

// Length returns the length of the vector
func (v Vector) Length() float64 {
	return math.Sqrt(float64(v.X*v.X + v.Y*v.Y))
}

// Distance returns the distance between two vectors
func (v Vector) Distance(v2 Vector) float64 {
	// Negate the two vectors and calciate the length
	return v.Added(v2.Negated()).Length()
}

// Multiplied returns the vector multiplied by an int
func (v Vector) Multiplied(val int) Vector {
	return Vector{v.X * val, v.Y * val}
}

// Divided returns the vector divided by an int
func (v Vector) Divided(val int) Vector {
	return Vector{v.X / val, v.Y / val}
}

// DividedFloor returns the vector divided but floors the value regardless
func (v Vector) DividedFloor(val int) Vector {
	x := float64(v.X) / float64(val)

	if x < 0 {
		x = math.Floor(x)
	} else {
		x = math.Floor(x)
	}
	y := float64(v.Y) / float64(val)
	if y < 0 {
		y = math.Floor(y)
	} else {
		y = math.Floor(y)
	}

	return Vector{X: int(x), Y: int(y)}
}

// Abs returns an absolute version of the vector
func (v Vector) Abs() Vector {
	return Vector{maths.Abs(v.X), maths.Abs(v.Y)}
}

// Min returns the minimum values in both vectors
func Min(v1 Vector, v2 Vector) Vector {
	return Vector{maths.Min(v1.X, v2.X), maths.Min(v1.Y, v2.Y)}
}

// Max returns the max values in both vectors
func Max(v1 Vector, v2 Vector) Vector {
	return Vector{maths.Max(v1.X, v2.X), maths.Max(v1.Y, v2.Y)}
}