mirror of
https://github.com/space-syndicate/space-station-14.git
synced 2026-10-09 18:31:17 +03:00
e7897d76a4
* more-item-status * update 1 * update 2 * Update pistol.yml * update 3 * yml * oops * BatteryItemStatus * fix ChargeItemStatus * Update item-status.ftl * review update 1 * review fix 1 * Update BatteryInfoSystem.cs * Update TimerTriggerItemStatus * merge fix * merge fix * review update 2 * review update 3 * fix marge * StyleNano * review update 4 * review update 5 * BatteryItemStatus * BatteryItemStatus Client * code cleanup * merge * merge and GasTank fix * Merge * unnecessary components * review * oops * GasTank * wawa
79 lines
2.9 KiB
C#
79 lines
2.9 KiB
C#
using Content.Client.Items.UI;
|
|
using Content.Client.Message;
|
|
using Content.Client.Power.EntitySystems;
|
|
using Content.Client.Stylesheets;
|
|
using Content.Client.Weapons.Ranged.Components;
|
|
using Content.Shared.Item.ItemToggle.Components;
|
|
using Content.Shared.PowerCell;
|
|
using Content.Shared.PowerCell.Components;
|
|
using Content.Shared.Power.EntitySystems;
|
|
using Robust.Client.UserInterface.Controls;
|
|
|
|
namespace Content.Client.Power.UI;
|
|
|
|
/// <summary>
|
|
/// Displays battery charge information for entities with <see cref="BatteryItemStatusComponent"/>.
|
|
/// </summary>
|
|
/// <seealso cref="BatteryItemStatusSystem"/>
|
|
public sealed partial class BatteryStatusControl : PollingItemStatusControl<BatteryStatusControl.Data>
|
|
{
|
|
private readonly EntityUid _parent;
|
|
private readonly IEntityManager _entityManager;
|
|
private readonly SharedBatterySystem _battery;
|
|
private readonly PowerCellSystem _powerCell;
|
|
private readonly RichTextLabel _label;
|
|
|
|
public BatteryStatusControl(
|
|
EntityUid parent,
|
|
IEntityManager entityManager,
|
|
SharedBatterySystem battery,
|
|
PowerCellSystem powerCell)
|
|
{
|
|
_parent = parent;
|
|
_entityManager = entityManager;
|
|
_battery = battery;
|
|
_powerCell = powerCell;
|
|
_label = new RichTextLabel { StyleClasses = { StyleClass.ItemStatus } };
|
|
AddChild(_label);
|
|
}
|
|
|
|
protected override Data PollData()
|
|
{
|
|
// Do not add battery status to guns that already show an ammo counter.
|
|
if (_entityManager.HasComponent<AmmoCounterComponent>(_parent))
|
|
return default;
|
|
|
|
// Battery charge level.
|
|
int? chargePercent = null;
|
|
if (_powerCell.TryGetBatteryFromEntityOrSlot(_parent, out var battery))
|
|
chargePercent = (int)(_battery.GetChargeLevel(battery.Value.AsNullable()) * 100);
|
|
|
|
// On/off state.
|
|
bool? toggleState = null;
|
|
if (_entityManager.TryGetComponent<ItemToggleComponent>(_parent, out var toggle))
|
|
toggleState = toggle.Activated;
|
|
else if (_entityManager.TryGetComponent<PowerCellDrawComponent>(_parent, out var powerDraw))
|
|
toggleState = powerDraw.Enabled;
|
|
|
|
return new Data(chargePercent, toggleState);
|
|
}
|
|
|
|
protected override void Update(in Data data)
|
|
{
|
|
var markup = "";
|
|
if (data.ChargePercent.HasValue)
|
|
markup = Loc.GetString("battery-status-charge", ("percent", data.ChargePercent));
|
|
|
|
if (data.ToggleState.HasValue)
|
|
{
|
|
var stateValue = data.ToggleState.Value ? "on" : "off";
|
|
var stateColor = Loc.GetString("battery-status-switchable-state", ("state", stateValue));
|
|
var stateLine = Loc.GetString("battery-status-state", ("state", stateColor));
|
|
markup += "\n" + stateLine;
|
|
}
|
|
|
|
_label.SetMarkup(markup);
|
|
}
|
|
|
|
public readonly record struct Data(int? ChargePercent, bool? ToggleState);
|
|
}
|