Ilia Denisov
83 lines
1.8 KiB
Go
83 lines
1.8 KiB
Go
package generator
|
|
|
|
import (
|
|
"fmt"
|
|
"math"
|
|
"math/rand"
|
|
|
|
"github.com/iliadenisov/galaxy/pkg/generator/plotter"
|
|
)
|
|
|
|
type Map struct {
|
|
Width uint32
|
|
Height uint32
|
|
HomePlanets []PlanetarySystem
|
|
FreePlanets []Planet
|
|
plotter plotter.Plotter
|
|
}
|
|
|
|
type Coordinate struct {
|
|
X, Y float64
|
|
}
|
|
|
|
func NewMap(width, height, players uint32) (*Map, error) {
|
|
p, err := plotter.NewPlotter(width, height, defaultFactor)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("NewPlotter: %s", err)
|
|
}
|
|
return &Map{
|
|
Width: width,
|
|
Height: height,
|
|
HomePlanets: make([]PlanetarySystem, players),
|
|
plotter: p,
|
|
}, nil
|
|
}
|
|
|
|
func (m *Map) CreatePlanets(pc PlanetClass, num int, deadZoneRadius float64, size, resources func() float64) error {
|
|
for range num {
|
|
coord, err := m.NewCoordinate(deadZoneRadius)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
planet := NewPlanet(pc, coord, size(), resources())
|
|
m.AddPlanet(planet)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (m *Map) AddPlanet(planet Planet) {
|
|
m.FreePlanets = append(m.FreePlanets, planet)
|
|
}
|
|
|
|
func (m Map) NewCoordinate(deadZoneRaduis float64) (Coordinate, error) {
|
|
if x, y, err := m.plotter.RandomFreePoint(deadZoneRaduis); err != nil {
|
|
return Coordinate{}, fmt.Errorf("NewCoordinate: RandomFreePoint: %s", err)
|
|
} else {
|
|
return Coordinate{X: x, Y: y}, nil
|
|
}
|
|
}
|
|
|
|
func (m Map) ShortDistance(from, to Coordinate) float64 {
|
|
dx := math.Abs(to.X - from.X)
|
|
dy := math.Abs(to.Y - from.Y)
|
|
if dx > float64(m.Width/2) {
|
|
dx = float64(m.Width) - dx
|
|
}
|
|
if dy > float64(m.Height/2) {
|
|
dy = float64(m.Height) - dy
|
|
}
|
|
return math.Sqrt(math.Pow(dx, 2) + math.Pow(dy, 2))
|
|
}
|
|
|
|
// RandI returns a random float64 value between min and max
|
|
func RandI(min, max float64) float64 {
|
|
return min + rand.Float64()*(max-min)
|
|
}
|
|
|
|
// RandIFn is a wrapper for the [RandI] func
|
|
func RandIFn(min, max float64) func() float64 {
|
|
return func() float64 {
|
|
return RandI(min, max)
|
|
}
|
|
}
|