basic capture functionality

scripts/ui/CaptureUI.cs

@@ -0,0 +1,121 @@
+using Godot;
+
+public partial class CaptureUI : CanvasLayer
+{
+	[Export] public float BaseTimeLimit = 3.0f;
+	[Export] public float BaseRequired = 14.0f; // “energy” required
+	[Export] public float EnergyPerPress = 1.0f;
+
+	private Creature? _target;
+	private SaveManager? _save;
+
+	private float _timeLeft;
+	private float _required;
+	private float _energy;
+
+	private OptionButton _ballOption = null!;
+	private ProgressBar _bar = null!;
+	private Label _timer = null!;
+	private Label _title = null!;
+
+	public override void _Ready()
+	{
+		_ballOption = GetNode<OptionButton>("Panel/VBoxContainer/SoulballOption");
+		_bar = GetNode<ProgressBar>("Panel/VBoxContainer/MashBar");
+		_timer = GetNode<Label>("Panel/VBoxContainer/TimerLabel");
+		_title = GetNode<Label>("Panel/VBoxContainer/CreatureNameLabel");
+
+		Visible = false;
+
+		// v0 balls
+		_ballOption.Clear();
+		_ballOption.AddItem("Basic Soulball (easy)", 0);
+		_ballOption.AddItem("Heavy Soulball (hard)", 1);
+
+		_save = GetNodeOrNull<SaveManager>("/root/Main/SaveManager"); // or export a path if you prefer
+	}
+
+	public void StartCapture(Creature target)
+	{
+		_target = target;
+		_energy = 0;
+
+		// Ball selection affects difficulty (v0)
+		int ballId = _ballOption.GetSelectedId();
+		float difficultyMul = ballId switch
+		{
+			0 => 1.0f,
+			1 => 1.4f,
+			_ => 1.0f
+		};
+
+		_required = BaseRequired * difficultyMul;
+		_timeLeft = BaseTimeLimit;
+
+		_title.Text = $"Capture: {target.CreatureId}";
+		_bar.MinValue = 0;
+		_bar.MaxValue = _required;
+		_bar.Value = 0;
+
+		Visible = true;
+
+		// Optional: pause world while capturing
+		GetTree().Paused = true;
+		ProcessMode = ProcessModeEnum.WhenPaused;
+	}
+
+	public override void _Process(double delta)
+	{
+		if (!Visible || _target == null) return;
+
+		// Count presses
+		if (Input.IsActionJustPressed("capture_mash"))
+		{
+			_energy += EnergyPerPress;
+			_bar.Value = _energy;
+		}
+
+		_timeLeft -= (float)delta;
+		_timer.Text = $"Time: {_timeLeft:0.00}s";
+
+		if (_energy >= _required)
+		{
+			OnCaptureSuccess();
+		}
+		else if (_timeLeft <= 0)
+		{
+			OnCaptureFail();
+		}
+	}
+
+	private void EndCapture()
+	{
+		Visible = false;
+		GetTree().Paused = false;
+		ProcessMode = ProcessModeEnum.Inherit;
+		_target = null;
+	}
+
+	private void OnCaptureSuccess()
+	{
+		if (_target == null) return;
+
+		GD.Print($"Captured {_target.CreatureId}");
+
+		// v0: store captured creature IDs
+		if (_save != null)
+		{
+			_save.State.CapturedCreatures.Add(_target.CreatureId);
+			_save.Save();
+		}
+
+		_target.Despawn();
+		EndCapture();
+	}
+
+	private void OnCaptureFail()
+	{
+		GD.Print("Capture failed");
+		EndCapture();
+	}
+}