visualization.go

package main

import (
	"image/color"
	"math"

	"github.com/hajimehoshi/ebiten/v2"
	"github.com/hajimehoshi/ebiten/v2/vector"
)

const (
	screenWidth  = 2400
	screenHeight = 1800
)

type Game struct {
	bodies     []Body
	forces     []Vector
	root       Node
	trailLayer *ebiten.Image
}

func NewGame() *Game {
	return &Game{
		bodies: bodies,
		forces: make([]Vector, len(bodies)),
		root:   Construct(bodies),
	}
}

func (g *Game) Update() error {
	BarnesHutUpdateForcesAndBodies(g.forces, g.bodies, g.root, dt)
	g.forces = make([]Vector, len(bodies))
	g.root = Construct(bodies)
	return nil
}

func (g *Game) Draw(screen *ebiten.Image) {
	for _, b := range g.bodies {
		vector.FillCircle(screen, float32(b.Position.X+screenWidth/2), float32(b.Position.Y+screenHeight/2), float32(math.Pow(b.Mass, .2)), color.White, false)
	}
}

func (g *Game) Layout(outsideWidth, outsideHeight int) (int, int) {
	return screenWidth, screenHeight
}