generator: verified

pkg/generator/map.go

@@ -31,7 +31,7 @@ type PlanetarySystem struct {
 }
 
 func NewMap(width, height, players uint32) (*Map, error) {
-	p, err := plotter.NewPlotter(width, height, 1.0)
+	p, err := plotter.NewPlotter(width, height, defaultFactor)
 	if err != nil {
 		return nil, fmt.Errorf("NewPlotter: %s", err)
 	}
@@ -43,6 +43,18 @@ func NewMap(width, height, players uint32) (*Map, error) {
 	}, nil
 }
 
+func (m *Map) CreatePlanets(num int, deadZoneRadius float32, size, resources func() float32) error {
+	for range num {
+		coord, err := m.NewCoordinate(deadZoneRadius)
+		if err != nil {
+			return err
+		}
+		planet := NewPlanet(coord, size(), resources())
+		m.AddPlanet(planet)
+	}
+	return nil
+}
+
 func (m *Map) AddPlanet(planet Planet) {
 	m.FreePlanets = append(m.FreePlanets, planet)
 }