438 lines
12 KiB
Go
438 lines
12 KiB
Go
package system
|
|
|
|
import (
|
|
"math/rand"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/hajimehoshi/ebiten/v2/audio"
|
|
|
|
"code.rocketnine.space/tslocum/citylimits/asset"
|
|
|
|
"code.rocketnine.space/tslocum/citylimits/component"
|
|
"code.rocketnine.space/tslocum/citylimits/world"
|
|
"code.rocketnine.space/tslocum/gohan"
|
|
"github.com/hajimehoshi/ebiten/v2"
|
|
"github.com/hajimehoshi/ebiten/v2/inpututil"
|
|
)
|
|
|
|
type playerMoveSystem struct {
|
|
player gohan.Entity
|
|
movement *MovementSystem
|
|
lastWalkDirL bool
|
|
|
|
rewindTicks int
|
|
nextRewindTick int
|
|
|
|
scrollDragX, scrollDragY int
|
|
scrollCamStartX, scrollCamStartY float64
|
|
}
|
|
|
|
func NewPlayerMoveSystem(player gohan.Entity, m *MovementSystem) *playerMoveSystem {
|
|
return &playerMoveSystem{
|
|
player: player,
|
|
movement: m,
|
|
scrollDragX: -1,
|
|
scrollDragY: -1,
|
|
}
|
|
}
|
|
|
|
func (_ *playerMoveSystem) Needs() []gohan.ComponentID {
|
|
return []gohan.ComponentID{
|
|
component.PositionComponentID,
|
|
component.VelocityComponentID,
|
|
component.WeaponComponentID,
|
|
}
|
|
}
|
|
|
|
func (_ *playerMoveSystem) Uses() []gohan.ComponentID {
|
|
return nil
|
|
}
|
|
|
|
func (s *playerMoveSystem) buildStructure(structureType int, tileX int, tileY int, playSound bool) (*world.Structure, error) {
|
|
structure, err := world.BuildStructure(world.World.HoverStructure, false, tileX, tileY)
|
|
if err == nil {
|
|
if world.IsPowerPlant(world.World.HoverStructure) {
|
|
plant := &world.PowerPlant{
|
|
Type: world.World.HoverStructure,
|
|
X: tileX,
|
|
Y: tileY,
|
|
}
|
|
world.World.PowerPlants = append(world.World.PowerPlants, plant)
|
|
}
|
|
|
|
if world.IsZone(structureType) {
|
|
zone := &world.Zone{
|
|
Type: world.World.HoverStructure,
|
|
X: tileX,
|
|
Y: tileY,
|
|
}
|
|
world.World.Zones = append(world.World.Zones, zone)
|
|
}
|
|
|
|
if world.World.HoverStructure != world.StructureBulldozer && playSound {
|
|
sounds := []*audio.Player{
|
|
asset.SoundPop2,
|
|
asset.SoundPop3,
|
|
}
|
|
sound := sounds[rand.Intn(len(sounds))]
|
|
sound.Rewind()
|
|
sound.Play()
|
|
}
|
|
|
|
cost := world.StructureCosts[structureType]
|
|
world.World.Funds -= cost
|
|
|
|
world.World.HUDUpdated = true
|
|
} else {
|
|
dX := tileX - world.World.LastBuildX
|
|
if dX < 0 {
|
|
dX *= -1
|
|
}
|
|
dY := tileY - world.World.LastBuildY
|
|
if dY < 0 {
|
|
dY *= -1
|
|
}
|
|
if (dX > 1 || dY > 1) && err != world.ErrNothingToBulldoze {
|
|
errMessage := err.Error()
|
|
if len(errMessage) > 0 {
|
|
errMessage = strings.ToUpper(errMessage[0:1]) + errMessage[1:]
|
|
}
|
|
world.ShowMessage(errMessage, 3)
|
|
}
|
|
}
|
|
return structure, err
|
|
}
|
|
|
|
func (s *playerMoveSystem) Update(ctx *gohan.Context) error {
|
|
if ebiten.IsKeyPressed(ebiten.KeyEscape) && !world.World.DisableEsc {
|
|
os.Exit(0)
|
|
return nil
|
|
}
|
|
|
|
if ebiten.IsKeyPressed(ebiten.KeyControl) && inpututil.IsKeyJustPressed(ebiten.KeyV) {
|
|
v := 1
|
|
if ebiten.IsKeyPressed(ebiten.KeyShift) {
|
|
v = 2
|
|
}
|
|
if world.World.Debug == v {
|
|
world.World.Debug = 0
|
|
} else {
|
|
world.World.Debug = v
|
|
}
|
|
return nil
|
|
}
|
|
if ebiten.IsKeyPressed(ebiten.KeyControl) && inpututil.IsKeyJustPressed(ebiten.KeyN) {
|
|
world.World.NoClip = !world.World.NoClip
|
|
return nil
|
|
}
|
|
|
|
if !world.World.GameStarted {
|
|
if ebiten.IsKeyPressed(ebiten.KeyEnter) || ebiten.IsMouseButtonPressed(ebiten.MouseButtonLeft) {
|
|
world.StartGame()
|
|
}
|
|
return nil
|
|
}
|
|
|
|
if inpututil.IsKeyJustPressed(ebiten.KeyM) {
|
|
world.World.MuteMusic = !world.World.MuteMusic
|
|
if world.World.MuteMusic {
|
|
asset.SoundMusic.Pause()
|
|
} else {
|
|
asset.SoundMusic.Play()
|
|
}
|
|
}
|
|
|
|
if world.World.GameOver {
|
|
if inpututil.IsKeyJustPressed(ebiten.KeyEnter) {
|
|
world.World.ResetGame = true
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// Update target zoom level.
|
|
var scrollY float64
|
|
if ebiten.IsKeyPressed(ebiten.KeyC) || ebiten.IsKeyPressed(ebiten.KeyPageDown) {
|
|
scrollY = -0.25
|
|
} else if ebiten.IsKeyPressed(ebiten.KeyE) || ebiten.IsKeyPressed(ebiten.KeyPageUp) {
|
|
scrollY = .25
|
|
} else {
|
|
_, scrollY = ebiten.Wheel()
|
|
if scrollY < -1 {
|
|
scrollY = -1
|
|
} else if scrollY > 1 {
|
|
scrollY = 1
|
|
}
|
|
}
|
|
world.World.CamScaleTarget += scrollY * (world.World.CamScaleTarget / 7)
|
|
if world.World.CamScaleTarget < world.CameraMinZoom {
|
|
world.World.CamScaleTarget = world.CameraMinZoom
|
|
} else if world.World.CamScaleTarget > world.CameraMaxZoom {
|
|
world.World.CamScaleTarget = world.CameraMaxZoom
|
|
}
|
|
|
|
// Smooth zoom transition.
|
|
div := 10.0
|
|
if world.World.CamScaleTarget > world.World.CamScale {
|
|
world.World.CamScale += (world.World.CamScaleTarget - world.World.CamScale) / div
|
|
} else if world.World.CamScaleTarget < world.World.CamScale {
|
|
world.World.CamScale -= (world.World.CamScale - world.World.CamScaleTarget) / div
|
|
}
|
|
|
|
pressLeft := ebiten.IsKeyPressed(ebiten.KeyLeft) || ebiten.IsKeyPressed(ebiten.KeyA)
|
|
pressRight := ebiten.IsKeyPressed(ebiten.KeyRight) || ebiten.IsKeyPressed(ebiten.KeyD)
|
|
pressUp := ebiten.IsKeyPressed(ebiten.KeyUp) || ebiten.IsKeyPressed(ebiten.KeyW)
|
|
pressDown := ebiten.IsKeyPressed(ebiten.KeyDown) || ebiten.IsKeyPressed(ebiten.KeyS)
|
|
|
|
const camSpeed = 10
|
|
if (pressLeft && !pressRight) ||
|
|
(pressRight && !pressLeft) {
|
|
if pressLeft {
|
|
world.World.CamX -= camSpeed
|
|
} else {
|
|
world.World.CamX += camSpeed
|
|
}
|
|
}
|
|
|
|
if (pressUp && !pressDown) ||
|
|
(pressDown && !pressUp) {
|
|
if pressUp {
|
|
world.World.CamY -= camSpeed
|
|
} else {
|
|
world.World.CamY += camSpeed
|
|
}
|
|
}
|
|
|
|
const scrollEdgeSize = 1
|
|
x, y := ebiten.CursorPosition()
|
|
if !world.World.GotCursorPosition {
|
|
if x != 0 || y != 0 {
|
|
world.World.GotCursorPosition = true
|
|
} else {
|
|
return nil
|
|
}
|
|
}
|
|
if ebiten.IsMouseButtonPressed(ebiten.MouseButtonMiddle) {
|
|
if s.scrollDragX == -1 && s.scrollDragY == -1 {
|
|
// TODO Disabled due to possible ebiten bug.
|
|
//ebiten.SetCursorMode(ebiten.CursorModeCaptured)
|
|
|
|
s.scrollDragX, s.scrollDragY = x, y
|
|
s.scrollCamStartX, s.scrollCamStartY = world.World.CamX, world.World.CamY
|
|
} else {
|
|
dx, dy := float64(x-s.scrollDragX)/world.World.CamScale, float64(y-s.scrollDragY)/world.World.CamScale
|
|
world.World.CamX, world.World.CamY = s.scrollCamStartX-dx, s.scrollCamStartY-dy
|
|
}
|
|
} else {
|
|
if s.scrollDragX != -1 && s.scrollDragY != -1 {
|
|
s.scrollDragX, s.scrollDragY = -1, -1
|
|
//ebiten.SetCursorMode(ebiten.CursorModeVisible)
|
|
} else if x >= -2 && y >= -2 && x < world.World.ScreenW+2 && y < world.World.ScreenH+2 {
|
|
// Pan via screen edge.
|
|
if x <= scrollEdgeSize {
|
|
world.World.CamX -= camSpeed
|
|
} else if x >= world.World.ScreenW-scrollEdgeSize-1 {
|
|
world.World.CamX += camSpeed
|
|
}
|
|
if y <= scrollEdgeSize {
|
|
world.World.CamY -= camSpeed
|
|
} else if y >= world.World.ScreenH-scrollEdgeSize-1 {
|
|
world.World.CamY += camSpeed
|
|
}
|
|
}
|
|
}
|
|
// Clamp viewport.
|
|
minCam := -256.0 * world.TileSize / 2
|
|
maxCam := 256.0 * world.TileSize / 2
|
|
if world.World.CamX < minCam {
|
|
world.World.CamX = minCam
|
|
} else if world.World.CamX > maxCam {
|
|
world.World.CamX = maxCam
|
|
}
|
|
if world.World.CamY < 0 {
|
|
world.World.CamY = 0
|
|
} else if world.World.CamY > maxCam {
|
|
world.World.CamY = maxCam
|
|
}
|
|
|
|
if x < world.SidebarWidth {
|
|
world.World.Level.ClearHoverSprites()
|
|
|
|
world.World.HoverX, world.World.HoverY = 0, 0
|
|
if inpututil.IsMouseButtonJustPressed(ebiten.MouseButtonLeft) {
|
|
button := world.HUDButtonAt(x, y)
|
|
if button != nil {
|
|
if button.StructureType != 0 {
|
|
if button.StructureType == world.StructureToggleTransparentStructures {
|
|
world.World.TransparentStructures = !world.World.TransparentStructures
|
|
world.World.HUDUpdated = true
|
|
|
|
if world.World.TransparentStructures {
|
|
world.ShowMessage("Enabled transparency", 3)
|
|
} else {
|
|
world.ShowMessage("Disabled transparency", 3)
|
|
}
|
|
} else {
|
|
if world.World.HoverStructure == button.StructureType {
|
|
world.SetHoverStructure(0) // Deselect.
|
|
} else {
|
|
world.SetHoverStructure(button.StructureType)
|
|
}
|
|
}
|
|
asset.SoundSelect.Rewind()
|
|
asset.SoundSelect.Play()
|
|
}
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
if x >= world.World.ScreenW-helpW && y >= world.World.ScreenH-helpH {
|
|
if inpututil.IsMouseButtonJustPressed(ebiten.MouseButtonLeft) {
|
|
const (
|
|
helpPrev = iota
|
|
helpClose
|
|
helpNext
|
|
)
|
|
|
|
helpButton := world.HelpButtonAt(x-(world.World.ScreenW-helpW), y-(world.World.ScreenH-helpH))
|
|
var updated bool
|
|
switch helpButton {
|
|
case helpPrev:
|
|
if world.World.HelpPage > 0 {
|
|
world.World.HelpPage--
|
|
updated = true
|
|
}
|
|
case helpClose:
|
|
world.World.HelpPage = -1
|
|
updated = true
|
|
case helpNext:
|
|
if world.World.HelpPage < len(world.HelpText)-1 {
|
|
world.World.HelpPage++
|
|
updated = true
|
|
}
|
|
}
|
|
if updated {
|
|
world.World.HelpUpdated = true
|
|
world.World.HUDUpdated = true
|
|
|
|
asset.SoundSelect.Rewind()
|
|
asset.SoundSelect.Play()
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
if world.World.HoverStructure != 0 {
|
|
roadTiles := func(fromX, fromY, toX, toY int) [][2]int {
|
|
var tiles [][2]int
|
|
fx, fy := float64(fromX), float64(fromY)
|
|
tx, ty := float64(toX), float64(toY)
|
|
dx, dy := tx-fx, ty-fy
|
|
for dx < -1 || dx > 1 || dy < -1 || dy > 1 {
|
|
dx /= 2
|
|
dy /= 2
|
|
}
|
|
tiles = append(tiles, [2]int{fromX, fromY})
|
|
for fx != tx || fy != ty {
|
|
fx, fy = fx+dx, fy+dy
|
|
tiles = append(tiles, [2]int{int(fx), int(fy)})
|
|
}
|
|
return tiles
|
|
}
|
|
|
|
tileX, tileY := world.ScreenToCartesian(x, y)
|
|
if tileX >= 0 && tileY >= 0 && tileX < 256 && tileY < 256 {
|
|
|
|
multiUseStructure := world.World.HoverStructure == world.StructureBulldozer || world.World.HoverStructure == world.StructureRoad || world.IsZone(world.World.HoverStructure)
|
|
dragStarted := world.World.BuildDragX != -1 || world.World.BuildDragY != -1
|
|
if inpututil.IsMouseButtonJustPressed(ebiten.MouseButtonLeft) || (multiUseStructure && ebiten.IsMouseButtonPressed(ebiten.MouseButtonLeft)) || dragStarted {
|
|
if !dragStarted {
|
|
world.World.BuildDragX, world.World.BuildDragY = int(tileX), int(tileY)
|
|
|
|
if world.World.HoverStructure == world.StructureBulldozer {
|
|
asset.SoundBulldoze.Play()
|
|
}
|
|
}
|
|
|
|
if world.World.HoverStructure == world.StructureRoad {
|
|
tiles := roadTiles(world.World.BuildDragX, world.World.BuildDragY, int(tileX), int(tileY))
|
|
|
|
if dragStarted && !ebiten.IsMouseButtonPressed(ebiten.MouseButtonLeft) {
|
|
// TODO build all tiles
|
|
world.World.Level.ClearHoverSprites()
|
|
var cost int
|
|
var builtRoad bool
|
|
for _, tile := range tiles {
|
|
_, err := s.buildStructure(world.World.HoverStructure, tile[0], tile[1], !builtRoad)
|
|
if err == nil {
|
|
cost += world.StructureCosts[world.World.HoverStructure]
|
|
builtRoad = true
|
|
}
|
|
}
|
|
if cost > 0 {
|
|
world.ShowBuildCost(world.World.HoverStructure, cost)
|
|
}
|
|
|
|
world.World.BuildDragX, world.World.BuildDragY = -1, -1
|
|
dragStarted = false
|
|
} else {
|
|
// TODO draw hover sprites
|
|
// TODO move below into shared func
|
|
world.World.Level.ClearHoverSprites()
|
|
for _, tile := range tiles {
|
|
world.BuildStructure(world.World.HoverStructure, true, tile[0], tile[1])
|
|
}
|
|
}
|
|
return nil
|
|
} else if dragStarted && !ebiten.IsMouseButtonPressed(ebiten.MouseButtonLeft) {
|
|
world.World.BuildDragX, world.World.BuildDragY = -1, -1
|
|
asset.SoundBulldoze.Pause()
|
|
}
|
|
|
|
cost := world.StructureCosts[world.World.HoverStructure]
|
|
if world.World.Funds < cost {
|
|
world.ShowMessage("Insufficient funds", 3)
|
|
} else {
|
|
world.World.Level.ClearHoverSprites()
|
|
|
|
// TODO draw hovers and build all roads in a line from drag start
|
|
structure, err := s.buildStructure(world.World.HoverStructure, int(tileX), int(tileY), true)
|
|
if err == nil {
|
|
tileX, tileY = float64(structure.X), float64(structure.Y)
|
|
world.ShowBuildCost(world.World.HoverStructure, cost)
|
|
}
|
|
|
|
world.BuildStructure(world.World.HoverStructure, true, int(tileX), int(tileY))
|
|
}
|
|
} else {
|
|
if world.World.LastBuildX != -1 || world.World.LastBuildY != -1 {
|
|
world.World.LastBuildX, world.World.LastBuildY = -1, -1
|
|
}
|
|
|
|
world.World.Level.ClearHoverSprites()
|
|
|
|
world.BuildStructure(world.World.HoverStructure, true, int(tileX), int(tileY))
|
|
}
|
|
world.World.HoverX, world.World.HoverY = int(tileX), int(tileY)
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (s *playerMoveSystem) Draw(_ *gohan.Context, _ *ebiten.Image) error {
|
|
return gohan.ErrSystemWithoutDraw
|
|
}
|
|
|
|
func deltaXY(x1, y1, x2, y2 float64) (dx float64, dy float64) {
|
|
dx, dy = x1-x2, y1-y2
|
|
if dx < 0 {
|
|
dx *= -1
|
|
}
|
|
if dy < 0 {
|
|
dy *= -1
|
|
}
|
|
return dx, dy
|
|
}
|