resimmed offmed

2025-10-06 01:55:42 -04:00
parent 847154b320
commit 173f24590f
84 changed files with 1210 additions and 1004 deletions
--- a/Content.Client/_Offbrand/Overlays/HeartrateOverlay.cs
+++ b/Content.Client/_Offbrand/Overlays/HeartrateOverlay.cs
@@ -0,0 +1,101 @@
+using System.Numerics;
+using Content.Client.StatusIcon;
+using Content.Shared._Offbrand.Wounds;
+using Content.Shared.StatusIcon.Components;
+using Content.Shared.StatusIcon;
+using Robust.Client.GameObjects;
+using Robust.Client.Graphics;
+using Robust.Shared.Enums;
+using Robust.Shared.Prototypes;
+using Robust.Shared.Timing;
+using Robust.Shared.Utility;
+
+namespace Content.Client._Offbrand.Overlays;
+
+public sealed class HeartrateOverlay : Overlay
+{
+    [Dependency] private readonly IEntityManager _entityManager = default!;
+    [Dependency] private readonly IGameTiming _timing = default!;
+    [Dependency] private readonly IPrototypeManager _prototype = default!;
+
+    private readonly HeartSystem _heart;
+    private readonly SharedTransformSystem _transform;
+    private readonly SpriteSystem _sprite;
+    private readonly StatusIconSystem _statusIcon;
+
+    public override OverlaySpace Space => OverlaySpace.WorldSpaceBelowFOV;
+
+    public ProtoId<HealthIconPrototype>? StatusIcon;
+
+    private static readonly SpriteSpecifier HudStopped = new SpriteSpecifier.Rsi(new("/Textures/_Offbrand/heart_rate_hud.rsi"), "hud_stopped");
+    private static readonly SpriteSpecifier HudGood = new SpriteSpecifier.Rsi(new("/Textures/_Offbrand/heart_rate_hud.rsi"), "hud_normal");
+    private static readonly SpriteSpecifier HudOkay = new SpriteSpecifier.Rsi(new("/Textures/_Offbrand/heart_rate_hud.rsi"), "hud_okay");
+    private static readonly SpriteSpecifier HudPoor = new SpriteSpecifier.Rsi(new("/Textures/_Offbrand/heart_rate_hud.rsi"), "hud_poor");
+    private static readonly SpriteSpecifier HudBad = new SpriteSpecifier.Rsi(new("/Textures/_Offbrand/heart_rate_hud.rsi"), "hud_bad");
+    private static readonly SpriteSpecifier HudDanger = new SpriteSpecifier.Rsi(new("/Textures/_Offbrand/heart_rate_hud.rsi"), "hud_danger");
+    private static readonly IReadOnlyList<SpriteSpecifier> Severities = new List<SpriteSpecifier>() { HudGood, HudOkay, HudPoor, HudBad, HudDanger };
+
+    public HeartrateOverlay()
+    {
+        IoCManager.InjectDependencies(this);
+
+        _transform = _entityManager.System<SharedTransformSystem>();
+        _sprite = _entityManager.System<SpriteSystem>();
+        _statusIcon = _entityManager.System<StatusIconSystem>();
+        _heart = _entityManager.System<HeartSystem>();
+    }
+
+    private SpriteSpecifier GetIcon(Entity<HeartrateComponent> ent)
+    {
+        if (!ent.Comp.Running)
+            return HudStopped;
+
+        var max = 4;
+        var severity = Math.Min((int)Math.Round(max * _heart.Strain(ent)), max);
+        return Severities[severity];
+    }
+
+    protected override void Draw(in OverlayDrawArgs args)
+    {
+        var handle = args.WorldHandle;
+        var rotation = args.Viewport.Eye?.Rotation ?? Angle.Zero;
+
+        const float scale = 1f;
+        var scaleMatrix = Matrix3Helpers.CreateScale(new Vector2(scale, scale));
+        var rotationMatrix = Matrix3Helpers.CreateRotation(-rotation);
+
+        _prototype.TryIndex(StatusIcon, out var statusIcon);
+
+        var query = _entityManager.AllEntityQueryEnumerator<MetaDataComponent, TransformComponent, HeartrateComponent, SpriteComponent>();
+        while (query.MoveNext(out var uid,
+            out var metadata,
+            out var xform,
+            out var heartrate,
+            out var sprite))
+        {
+            if (statusIcon != null && !_statusIcon.IsVisible((uid, metadata), statusIcon))
+                continue;
+
+            var bounds = _entityManager.GetComponentOrNull<StatusIconComponent>(uid)?.Bounds ?? _sprite.GetLocalBounds((uid, sprite));
+            var worldPos = _transform.GetWorldPosition(xform);
+
+            if (!bounds.Translated(worldPos).Intersects(args.WorldAABB))
+                continue;
+
+            var worldPosition = _transform.GetWorldPosition(xform);
+            var worldMatrix = Matrix3Helpers.CreateTranslation(worldPosition);
+
+            var scaledWorld = Matrix3x2.Multiply(scaleMatrix, worldMatrix);
+            var matty = Matrix3x2.Multiply(rotationMatrix, scaledWorld);
+
+            handle.SetTransform(matty);
+
+            var curTime = _timing.RealTime;
+            var texture = _sprite.GetFrame(GetIcon((uid, heartrate)), curTime);
+
+            handle.DrawTexture(texture, new Vector2(-8f, 8f) / EyeManager.PixelsPerMeter);
+        }
+
+        handle.SetTransform(Matrix3x2.Identity);
+    }
+}