Files
AceField-New-Horizon/Scripts/System/PlacementManager.cs
2025-08-27 17:37:31 +08:00

112 lines
3.7 KiB
C#

using AceFieldNewHorizon.Scripts.Tiles;
using Godot;
namespace AceFieldNewHorizon.Scripts.System;
public partial class PlacementManager : Node2D
{
[Export] public GridManager Grid { get; set; }
[Export] public BuildingRegistry Registry { get; set; }
private string _currentBuildingId = "miner";
private Vector2I _hoveredCell;
private BaseTile _ghostBuilding;
public void SetCurrentBuilding(string buildingId)
{
_currentBuildingId = buildingId;
// Replace ghost immediately
if (_ghostBuilding == null) return;
_ghostBuilding.QueueFree();
_ghostBuilding = null;
}
// Add this field to PlacementManager
private float _currentRotation = 0f;
// Add this method to handle rotation
private void RotateGhost(bool reverse = false)
{
if (_ghostBuilding == null) return;
if (reverse)
_currentRotation = (_currentRotation - 90f) % 360f;
else
_currentRotation = (_currentRotation + 90f) % 360f;
_ghostBuilding.RotationDegrees = _currentRotation;
}
public override void _Process(double delta)
{
// Snap mouse to grid
var mousePos = GetGlobalMousePosition();
_hoveredCell = GridUtils.WorldToGrid(mousePos);
if (Input.IsActionJustPressed("rotate_tile_reverse"))
RotateGhost(reverse: true);
else if (Input.IsActionJustPressed("rotate_tile"))
RotateGhost();
if (_ghostBuilding == null)
{
var scene = Registry.GetBuilding(_currentBuildingId)?.Scene;
if (scene == null) return;
_ghostBuilding = (BaseTile)scene.Instantiate();
_ghostBuilding.SetGhostMode(true);
_ghostBuilding.RotationDegrees = _currentRotation;
AddChild(_ghostBuilding);
}
var placementPos = GridUtils.GridToWorld(_hoveredCell);
var spaceState = GetWorld2D().DirectSpaceState;
var centerPos = placementPos + new Vector2(GridUtils.TileSize / 2f, GridUtils.TileSize / 2f);
var query = new PhysicsPointQueryParameters2D
{
Position = centerPos,
CollideWithBodies = true,
CollideWithAreas = true,
CollisionMask = uint.MaxValue // check against all layers/masks
// Exclude = new Godot.Collections.Array<Rid>() // optional, see below
};
var collision = spaceState.IntersectPoint(query, 8);
var canPlace = Grid.IsCellFree(_hoveredCell) && collision.Count == 0;
_ghostBuilding.Position = placementPos;
_ghostBuilding.SetGhostMode(canPlace);
// Left click to place
if (Input.IsActionPressed("build_tile") && canPlace)
{
var scene = Registry.GetBuilding(_currentBuildingId)?.Scene;
if (scene == null) return;
_ghostBuilding.FinalizePlacement();
_ghostBuilding.RotationDegrees = _currentRotation;
var buildingData = Registry.GetBuilding(_currentBuildingId);
Grid.OccupyCell(_hoveredCell, _ghostBuilding);
if (buildingData is { BuildTime: > 0f })
_ghostBuilding.StartConstruction(buildingData.BuildTime);
_ghostBuilding = (BaseTile)scene.Instantiate();
_ghostBuilding.SetGhostMode(true);
_ghostBuilding.RotationDegrees = _currentRotation;
AddChild(_ghostBuilding);
}
if (Input.IsActionPressed("destroy_tile") && !Grid.IsCellFree(_hoveredCell))
{
// Right click to destroy
var building = Grid.GetBuildingAtCell(_hoveredCell);
if (building == null) return;
building.QueueFree();
Grid.FreeCell(_hoveredCell);
}
}
}