initial approximation
This commit is contained in:
Ilia Denisov
@@ -0,0 +1,148 @@
|
||||
package draw
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"math"
|
||||
)
|
||||
|
||||
const batchSize = 32
|
||||
|
||||
type Plane struct {
|
||||
Width uint
|
||||
Height uint
|
||||
pixels []uint32
|
||||
}
|
||||
|
||||
func (p *Plane) pixelsValue() []uint32 {
|
||||
return p.pixels
|
||||
}
|
||||
|
||||
func New(width uint, height uint) Plane {
|
||||
elements := int(math.Ceil(float64(width*height) / batchSize))
|
||||
return Plane{Width: width, Height: height, pixels: make([]uint32, elements)}
|
||||
}
|
||||
|
||||
func (p Plane) Mark(x, y int) {
|
||||
boundX := (p.Width + uint(x)) % p.Width
|
||||
boundY := (p.Height + uint(y)) % p.Height
|
||||
p.mark(boundX + boundY*p.Width)
|
||||
|
||||
// p.mark(x + y*p.Width)
|
||||
|
||||
// number := x + y*p.width
|
||||
// index := number / batchSize
|
||||
// bit := number % batchSize
|
||||
// p.pixels[index] |= (0b1 << bit)
|
||||
}
|
||||
|
||||
func (p Plane) mark(number uint) {
|
||||
// index := number / batchSize
|
||||
// bit := number % batchSize
|
||||
p.pixels[number/batchSize] |= (0b1 << (number % batchSize))
|
||||
}
|
||||
|
||||
func (p Plane) Marked(x, y uint) bool {
|
||||
return p.marked(x + y*p.Width)
|
||||
// number := x + y*p.width
|
||||
// index := number / batchSize
|
||||
// bit := number % batchSize
|
||||
// return p.pixels[index]&(0b1<<bit) > 0
|
||||
}
|
||||
|
||||
func (p Plane) marked(number uint) bool {
|
||||
// index := number / batchSize
|
||||
// bit := number % batchSize
|
||||
return p.pixels[number/batchSize]&(0b1<<(number%batchSize)) > 0
|
||||
}
|
||||
|
||||
func (p Plane) Circle(x, y int, r float64) {
|
||||
plotX := 0
|
||||
plotY := int(math.Ceil(r))
|
||||
delta := 3 - 2*plotY
|
||||
lastY := plotY
|
||||
for plotX <= plotY {
|
||||
p.octant(x, y, plotX, plotY)
|
||||
if plotY < lastY {
|
||||
for lineX := 0; lineX < plotX; lineX++ {
|
||||
p.octant(x, y, lineX, plotY)
|
||||
}
|
||||
lastY = plotY
|
||||
}
|
||||
if delta < 0 {
|
||||
delta += 4*plotX + 6
|
||||
} else {
|
||||
delta += 4*(plotX-plotY) + 10
|
||||
plotY -= 1
|
||||
}
|
||||
plotX += 1
|
||||
}
|
||||
for fillX := 0; fillX < plotX; fillX++ {
|
||||
for fillY := 0; fillY <= fillX; fillY++ {
|
||||
p.octant(x, y, fillX, fillY)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func (p Plane) CircleAdjacent(x, y int, r float64) {
|
||||
plotX := 0
|
||||
plotY := int(math.Ceil(r))
|
||||
delta := 1 - 2*plotY
|
||||
err := 0
|
||||
for plotX <= plotY {
|
||||
p.octant(x, y, plotX, plotY)
|
||||
err = 2*(delta+plotY) - 1
|
||||
if delta < 0 && err <= 0 {
|
||||
plotX += 1
|
||||
delta += 2*plotX + 1
|
||||
continue
|
||||
}
|
||||
if delta > 0 && err > 0 {
|
||||
plotY -= 1
|
||||
delta -= 2*plotY + 1
|
||||
continue
|
||||
}
|
||||
plotX += 1
|
||||
plotY -= 1
|
||||
delta += 2 * (plotX - plotY)
|
||||
}
|
||||
}
|
||||
|
||||
func (p Plane) octant(x, y int, plotX, plotY int) {
|
||||
p.Mark(x+plotX, y+plotY)
|
||||
p.Mark(x+plotX, y-plotY)
|
||||
p.Mark(x-plotX, y+plotY)
|
||||
p.Mark(x-plotX, y-plotY)
|
||||
p.Mark(x+plotY, y+plotX)
|
||||
p.Mark(x+plotY, y-plotX)
|
||||
p.Mark(x-plotY, y+plotX)
|
||||
p.Mark(x-plotY, y-plotX)
|
||||
}
|
||||
|
||||
func (p Plane) Clear() {
|
||||
for i := range p.pixels {
|
||||
p.pixels[i] &= 0
|
||||
}
|
||||
}
|
||||
|
||||
func (p Plane) String() string {
|
||||
px := map[bool]string{true: "x", false: "_"}
|
||||
var result string
|
||||
// for y := 0; y < int(p.height); y++ {
|
||||
// for x := 0; x < int(p.width); x++ {
|
||||
// result += px[p.Marked(uint(x), uint(y))] + " "
|
||||
// }
|
||||
// result += "\n"
|
||||
// }
|
||||
// result += "\n"
|
||||
cnt := 0
|
||||
for i := 0; i < len(p.pixels); i++ {
|
||||
for bit := 0; bit < batchSize && cnt < int(p.Width*p.Height); bit++ {
|
||||
result += fmt.Sprintf("%s ", px[p.pixels[i]&(0b1<<bit) > 0])
|
||||
cnt++
|
||||
if cnt%int(p.Width) == 0 {
|
||||
result += "\n"
|
||||
}
|
||||
}
|
||||
}
|
||||
return result
|
||||
}
|
||||
@@ -0,0 +1,82 @@
|
||||
package draw_test
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"testing"
|
||||
|
||||
"github.com/iliadenisov/galaxy/pkg/generator/draw"
|
||||
)
|
||||
|
||||
func Test_CreatePlot(t *testing.T) {
|
||||
var pane = draw.New(10, 10)
|
||||
if len(draw.PixelsHolder(&pane)) != 4 {
|
||||
t.Errorf("wrong Pane data size, expected: %d given: %d", 4, len(draw.PixelsHolder(&pane)))
|
||||
}
|
||||
|
||||
pane = draw.New(1, 1)
|
||||
if len(draw.PixelsHolder(&pane)) != 1 {
|
||||
t.Errorf("wrong Pane data size, expected: %d given: %d", 4, len(draw.PixelsHolder(&pane)))
|
||||
}
|
||||
|
||||
pane = draw.New(32, 32)
|
||||
if len(draw.PixelsHolder(&pane)) != 32 {
|
||||
t.Errorf("wrong Pane data size, expected: %d given: %d", 4, len(draw.PixelsHolder(&pane)))
|
||||
}
|
||||
}
|
||||
|
||||
func Test_Mark(t *testing.T) {
|
||||
var size int = 7
|
||||
var pane = draw.New(uint(size), uint(size))
|
||||
for i := 0; i < int(size); i++ {
|
||||
pane.Mark(0, i)
|
||||
pane.Mark(i, 0)
|
||||
pane.Mark(size-1, i)
|
||||
pane.Mark(i, size-1)
|
||||
pane.Mark(i, i)
|
||||
pane.Mark(size-1-i, i)
|
||||
}
|
||||
|
||||
pixelsHolder := draw.PixelsHolder(&pane)
|
||||
|
||||
if len(pixelsHolder) != 2 {
|
||||
t.Errorf("expected len: %d given: %d", 2, len(pixelsHolder))
|
||||
}
|
||||
|
||||
// if !pane.Marked(0, 0) {
|
||||
// t.Errorf("should be marked at x=%d y=%d", 0, 0)
|
||||
// }
|
||||
|
||||
fmt.Println(pane)
|
||||
}
|
||||
|
||||
func Test_Cleat(t *testing.T) {
|
||||
var pane = draw.New(10, 10)
|
||||
pane.Mark(5, 5)
|
||||
pane.Mark(0, 0)
|
||||
pane.Mark(9, 9)
|
||||
if !pane.Marked(0, 0) {
|
||||
t.Errorf("should be marked at x=%d y=%d", 0, 0)
|
||||
}
|
||||
if !pane.Marked(5, 5) {
|
||||
t.Errorf("should be marked at x=%d y=%d", 5, 5)
|
||||
}
|
||||
if !pane.Marked(9, 9) {
|
||||
t.Errorf("should be marked at x=%d y=%d", 9, 9)
|
||||
}
|
||||
pane.Clear()
|
||||
for _, holder := range draw.PixelsHolder(&pane) {
|
||||
if holder != 0 {
|
||||
t.Errorf("should be all bits clear")
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func Test_Circle(t *testing.T) {
|
||||
var size int = 40
|
||||
var pane1 = draw.New(uint(size), uint(size))
|
||||
pane1.Circle(size/2+5, size/2-5, float64(size/2)-3)
|
||||
fmt.Println(pane1)
|
||||
var pane2 = draw.New(uint(size), uint(size))
|
||||
pane2.CircleAdjacent(size/2+20, size/2, float64(size/2)-5)
|
||||
// fmt.Println(pane2)
|
||||
}
|
||||
@@ -0,0 +1,3 @@
|
||||
package draw
|
||||
|
||||
var PixelsHolder = (*Plane).pixelsValue
|
||||
@@ -0,0 +1,160 @@
|
||||
package generator
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"fmt"
|
||||
"math"
|
||||
"math/rand"
|
||||
|
||||
"github.com/iliadenisov/galaxy/pkg/generator/draw"
|
||||
)
|
||||
|
||||
type Map struct {
|
||||
Width uint
|
||||
Height uint
|
||||
HomePlanets []PlanetarySystem
|
||||
FreePlanets []Planet
|
||||
plotter Plotter
|
||||
}
|
||||
|
||||
type Coordinate struct {
|
||||
X, Y float64
|
||||
}
|
||||
|
||||
type Planet struct {
|
||||
Position Coordinate
|
||||
Size float64
|
||||
Resources float64 // Сырьё
|
||||
}
|
||||
|
||||
type PlanetarySystem struct {
|
||||
HW Planet
|
||||
DW []Planet
|
||||
}
|
||||
|
||||
type Plotter struct {
|
||||
factor float64
|
||||
sectors draw.Plane
|
||||
}
|
||||
|
||||
func Generate(param MapParameters) (result Map) {
|
||||
pl := func(c Coordinate, param UninhabitedPlanetParameters) Planet {
|
||||
return Planet{
|
||||
Position: c,
|
||||
Size: param.MinSize + rand.Float64()*(param.MaxSize-param.MinSize),
|
||||
Resources: float64(param.MinResource) + rand.Float64()*(param.MaxResource-param.MinResource)}
|
||||
}
|
||||
// mapSize := uint(math.Ceil(math.Sqrt(float64(param.Players)))) * param.HW_MinDistance
|
||||
var mapSize uint = 200
|
||||
|
||||
result = NewMap(mapSize, mapSize, param.Players)
|
||||
|
||||
totalPlanets := param.Players * 10
|
||||
freePlanets := totalPlanets - param.Players*(param.DW_Count+1)
|
||||
|
||||
fmt.Println("map:", mapSize, "players:", param.Players, "planets:", totalPlanets, "uninhabited:", freePlanets)
|
||||
|
||||
giantsNum := int(math.Ceil(float64(freePlanets) * param.GiantPlanets.Probability))
|
||||
fmt.Println("generating", giantsNum, "giant planets")
|
||||
for i := 0; i < giantsNum; i++ {
|
||||
coord := result.newPlanet(float64(param.GiantPlanets.MinDistanceHW))
|
||||
planet := pl(coord, param.GiantPlanets)
|
||||
result.addPlanet(planet)
|
||||
// result.FreePlanets = append(result.FreePlanets, planet)
|
||||
}
|
||||
|
||||
bigsNum := int(math.Ceil(float64(freePlanets) * param.BigPlanets.Probability))
|
||||
fmt.Println("generating", bigsNum, "big planets")
|
||||
for i := 0; i < bigsNum; i++ {
|
||||
coord := result.newPlanet(float64(param.BigPlanets.MinDistanceHW))
|
||||
planet := pl(coord, param.BigPlanets)
|
||||
result.addPlanet(planet)
|
||||
// result.FreePlanets = append(result.FreePlanets, planet)
|
||||
}
|
||||
|
||||
for player := 0; player < int(param.Players); player++ {
|
||||
fmt.Println("generating HW #", player)
|
||||
coord := result.newPlanet(float64(param.HW_MinDistance))
|
||||
planet := Planet{Position: coord, Size: float64(param.HW_Size), Resources: float64(param.HW_Resources)}
|
||||
// fmt.Println("HW: ", planet)
|
||||
result.HomePlanets[player] = PlanetarySystem{HW: planet}
|
||||
}
|
||||
|
||||
fmt.Println("free sectors left: ", len(result.plotter.freeSectors()))
|
||||
|
||||
result.plotter.sectors.Clear()
|
||||
|
||||
for _, hw := range result.HomePlanets {
|
||||
result.plotter.markNoGoZone(hw.HW.Position.X, hw.HW.Position.Y, 5)
|
||||
for _, dw := range hw.DW {
|
||||
result.plotter.markNoGoZone(dw.Position.X, dw.Position.Y, 5)
|
||||
}
|
||||
}
|
||||
|
||||
for _, planet := range result.FreePlanets {
|
||||
result.plotter.markNoGoZone(planet.Position.X, planet.Position.Y, 5)
|
||||
|
||||
}
|
||||
|
||||
fmt.Println("free sectors after reset: ", len(result.plotter.freeSectors()))
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
func (m Map) newPlanet(deadZoneRaduis float64) Coordinate {
|
||||
fs := m.plotter.freeSectors()
|
||||
fmt.Println("free sectors: ", len(fs))
|
||||
fsCount := len(fs)
|
||||
if fsCount == 0 {
|
||||
panic("no more space for planets")
|
||||
}
|
||||
next := rand.Intn(fsCount)
|
||||
x := fs[next][0]
|
||||
y := fs[next][1]
|
||||
fmt.Println("planet on plot: x =", x, "y =", y)
|
||||
planetX := float64(x)*m.plotter.factor + rand.Float64()*m.plotter.factor
|
||||
planetY := float64(y)*m.plotter.factor + rand.Float64()*m.plotter.factor
|
||||
m.plotter.markDeadZone(int(x), int(y), deadZoneRaduis)
|
||||
return Coordinate{X: planetX, Y: planetY}
|
||||
}
|
||||
|
||||
func (m *Map) addPlanet(planet Planet) {
|
||||
m.FreePlanets = append(m.FreePlanets, planet)
|
||||
}
|
||||
|
||||
func (p Plotter) freeSectors() (result [][]uint) {
|
||||
result = make([][]uint, 0)
|
||||
for x := uint(0); x < p.sectors.Width; x++ {
|
||||
for y := uint(0); y < p.sectors.Height; y++ {
|
||||
if !p.sectors.Marked(x, y) {
|
||||
result = append(result, []uint{x, y})
|
||||
}
|
||||
}
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func (p Plotter) markDeadZone(x, y int, radius float64) {
|
||||
p.sectors.Circle(x, y, radius)
|
||||
}
|
||||
|
||||
func (p Plotter) markNoGoZone(x, y float64, radius float64) { // TODO: test
|
||||
p.markDeadZone(int(x/p.factor), int(y/p.factor), radius)
|
||||
}
|
||||
|
||||
func NewMap(width, height, players uint) Map {
|
||||
var factor float64 = 1.0
|
||||
sectorsX := uint(float64(width) / factor)
|
||||
sectorsY := uint(float64(height) / factor)
|
||||
return Map{
|
||||
Width: width,
|
||||
Height: height,
|
||||
HomePlanets: make([]PlanetarySystem, players),
|
||||
plotter: Plotter{
|
||||
factor: factor,
|
||||
sectors: draw.New(sectorsX, sectorsY)}}
|
||||
}
|
||||
|
||||
func errfmt(message string) error {
|
||||
return errors.New("generator: " + message)
|
||||
}
|
||||
@@ -0,0 +1,11 @@
|
||||
package generator_test
|
||||
|
||||
import (
|
||||
"testing"
|
||||
|
||||
"github.com/iliadenisov/galaxy/pkg/generator"
|
||||
)
|
||||
|
||||
func Test_Generator(t *testing.T) {
|
||||
generator.Generate(generator.DefaultMapParameters())
|
||||
}
|
||||
@@ -0,0 +1,82 @@
|
||||
package generator
|
||||
|
||||
type MapParameters struct {
|
||||
Players uint
|
||||
HW_Size uint
|
||||
HW_Resources uint
|
||||
HW_MinDistance uint
|
||||
DW_Count uint
|
||||
DW_Size uint
|
||||
DW_Resources uint
|
||||
DW_MinDistance uint
|
||||
DW_MaxDistance uint
|
||||
|
||||
GiantPlanets UninhabitedPlanetParameters
|
||||
BigPlanets UninhabitedPlanetParameters
|
||||
NormalPlanets UninhabitedPlanetParameters
|
||||
RichPlanets UninhabitedPlanetParameters
|
||||
Asterioids UninhabitedPlanetParameters
|
||||
}
|
||||
|
||||
type UninhabitedPlanetParameters struct {
|
||||
MinDistanceHW uint
|
||||
MinSize float64
|
||||
MaxSize float64
|
||||
MinResource float64
|
||||
MaxResource float64
|
||||
Probability float64
|
||||
}
|
||||
|
||||
func DefaultMapParameters() MapParameters {
|
||||
return MapParameters{
|
||||
Players: 25,
|
||||
HW_Size: 1000,
|
||||
HW_Resources: 10,
|
||||
HW_MinDistance: 30,
|
||||
DW_Count: 2,
|
||||
DW_Size: 500,
|
||||
DW_Resources: 10,
|
||||
DW_MinDistance: 5,
|
||||
DW_MaxDistance: 15,
|
||||
GiantPlanets: UninhabitedPlanetParameters{
|
||||
MinDistanceHW: 20,
|
||||
MinSize: 1500,
|
||||
MaxSize: 2500,
|
||||
MinResource: 0,
|
||||
MaxResource: 3,
|
||||
Probability: 0.06,
|
||||
},
|
||||
BigPlanets: UninhabitedPlanetParameters{
|
||||
MinDistanceHW: 10,
|
||||
MinSize: 1000,
|
||||
MaxSize: 2000,
|
||||
MinResource: 1,
|
||||
MaxResource: 10,
|
||||
Probability: 0.18,
|
||||
},
|
||||
NormalPlanets: UninhabitedPlanetParameters{
|
||||
MinDistanceHW: 0,
|
||||
MinSize: 0.001,
|
||||
MaxSize: 1000,
|
||||
MinResource: 0,
|
||||
MaxResource: 10,
|
||||
Probability: 0.5,
|
||||
},
|
||||
RichPlanets: UninhabitedPlanetParameters{
|
||||
MinDistanceHW: 0,
|
||||
MinSize: 0.001,
|
||||
MaxSize: 500,
|
||||
MinResource: 5,
|
||||
MaxResource: 25,
|
||||
Probability: 0.18,
|
||||
},
|
||||
Asterioids: UninhabitedPlanetParameters{
|
||||
MinDistanceHW: 0,
|
||||
MinSize: 0.001,
|
||||
MaxSize: 10,
|
||||
MinResource: 0,
|
||||
MaxResource: 0,
|
||||
Probability: 0.08,
|
||||
},
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user