Cinka/OldThink
1
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

Health bar overlay refactor

Browse Source
This commit is contained in:
Aviu00
2024-01-24 14:42:36 +03:00
parent b0e55ed2a9
commit 8c75cde3e2
1 changed files with 35 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

69
Content.Client/White/EntityHealthBar/EntityHealthBarOverlay.cs
View File

@@ -7,7 +7,6 @@ using Content.Shared.FixedPoint;
using Robust.Client.GameObjects; using Robust.Client.GameObjects;
using Robust.Client.Graphics; using Robust.Client.Graphics;
using Robust.Shared.Enums; using Robust.Shared.Enums;
using Robust.Shared.Prototypes;
using Robust.Shared.Utility; using Robust.Shared.Utility;
using Robust.Shared.Timing; using Robust.Shared.Timing;
@@ -27,8 +26,8 @@ public sealed class EntityHealthBarOverlay : Overlay
public override OverlaySpace Space => OverlaySpace.WorldSpaceBelowFOV; public override OverlaySpace Space => OverlaySpace.WorldSpaceBelowFOV;
public List<string>? DamageContainers; public List<string>? DamageContainers;
// for icon frame change timer // for icon frame change timer
int iconFrame = 1; private int _iconFrame = 1;
double delayTime = 0.25; private const double DelayTime = 0.25;
public EntityHealthBarOverlay(IEntityManager entManager) public EntityHealthBarOverlay(IEntityManager entManager)
{ {
@@ -40,12 +39,12 @@ public sealed class EntityHealthBarOverlay : Overlay
var sprite = new SpriteSpecifier.Rsi(new ResPath("/Textures/Interface/Misc/health_status.rsi"), "background"); var sprite = new SpriteSpecifier.Rsi(new ResPath("/Textures/Interface/Misc/health_status.rsi"), "background");
_barTexture = _entManager.EntitySysManager.GetEntitySystem<SpriteSystem>().Frame0(sprite); _barTexture = _entManager.EntitySysManager.GetEntitySystem<SpriteSystem>().Frame0(sprite);
Timer.SpawnRepeating(TimeSpan.FromSeconds(delayTime), () => Timer.SpawnRepeating(TimeSpan.FromSeconds(DelayTime), () =>
{ {
if (iconFrame < 8) if (_iconFrame < 8)
iconFrame++; _iconFrame++;
else else
iconFrame = 1; _iconFrame = 1;
}, new System.Threading.CancellationToken()); }, new System.Threading.CancellationToken());
} }
@@ -56,15 +55,17 @@ public sealed class EntityHealthBarOverlay : Overlay
var spriteQuery = _entManager.GetEntityQuery<SpriteComponent>(); var spriteQuery = _entManager.GetEntityQuery<SpriteComponent>();
var xformQuery = _entManager.GetEntityQuery<TransformComponent>(); var xformQuery = _entManager.GetEntityQuery<TransformComponent>();
var _spriteSys = _entManager.EntitySysManager.GetEntitySystem<SpriteSystem>(); var spriteSys = _entManager.EntitySysManager.GetEntitySystem<SpriteSystem>();
const float scale = 1f; const float scale = 1f;
var scaleMatrix = Matrix3.CreateScale(new Vector2(scale, scale)); var scaleMatrix = Matrix3.CreateScale(new Vector2(scale, scale));
var rotationMatrix = Matrix3.CreateRotation(-rotation); var rotationMatrix = Matrix3.CreateRotation(-rotation);
foreach (var (mob, dmg, threasholds) in _entManager.EntityQuery<MobStateComponent, DamageableComponent, MobThresholdsComponent>(true)) var query = _entManager.EntityQueryEnumerator<MobStateComponent, DamageableComponent, MobThresholdsComponent>();
while (query.MoveNext(out var uid, out var mob, out var dmg, out var thresholds))
{ {
if (!xformQuery.TryGetComponent(mob.Owner, out var xform) || if (!xformQuery.TryGetComponent(uid, out var xform) ||
xform.MapID != args.MapId) xform.MapID != args.MapId)
{ {
continue; continue;
@@ -86,7 +87,7 @@ public sealed class EntityHealthBarOverlay : Overlay
float yOffset; float yOffset;
float xIconOffset; float xIconOffset;
float yIconOffset; float yIconOffset;
if (spriteQuery.TryGetComponent(mob.Owner, out var sprite)) if (spriteQuery.TryGetComponent(uid, out var sprite))
{ {
yOffset = sprite.Bounds.Height + 12f; yOffset = sprite.Bounds.Height + 12f;
yIconOffset = sprite.Bounds.Height + 7f; yIconOffset = sprite.Bounds.Height + 7f;
@@ -105,7 +106,7 @@ public sealed class EntityHealthBarOverlay : Overlay
yOffset / EyeManager.PixelsPerMeter); yOffset / EyeManager.PixelsPerMeter);
// Draw the underlying bar texture // Draw the underlying bar texture
if (sprite != null && !sprite.ContainerOccluded) if (sprite is {ContainerOccluded: false})
handle.DrawTexture(_barTexture, position); handle.DrawTexture(_barTexture, position);
else else
continue; continue;
@@ -113,34 +114,34 @@ public sealed class EntityHealthBarOverlay : Overlay
// Draw state icon // Draw state icon
if (dmg.DamageContainerID == "Biological") if (dmg.DamageContainerID == "Biological")
{ {
string current_state; string currentState;
if (_mobStateSystem.IsAlive(mob.Owner, mob)) if (_mobStateSystem.IsAlive(uid, mob))
{ {
current_state = "life_state"; currentState = "life_state";
} }
else else
{ {
if (_mobStateSystem.IsCritical(mob.Owner, mob) && if (_mobStateSystem.IsCritical(uid, mob) &&
_mobThresholdSystem.TryGetThresholdForState(mob.Owner, MobState.Critical, _mobThresholdSystem.TryGetThresholdForState(uid, MobState.Critical,
out var critThreshold)) out var _, thresholds))
current_state = "defib_state"; currentState = "defib_state";
else else
current_state = "dead_state"; currentState = "dead_state";
} }
var icon_sprite = new SpriteSpecifier.Rsi(new ResPath("/Textures/Interface/Misc/health_state.rsi"), var iconSprite = new SpriteSpecifier.Rsi(new ResPath("/Textures/Interface/Misc/health_state.rsi"),
current_state); currentState);
Texture _stateIcon = _spriteSys.RsiStateLike(icon_sprite) var stateIcon = spriteSys.RsiStateLike(iconSprite)
.GetFrame(0, GetIconFrame(_spriteSys.RsiStateLike(icon_sprite))); .GetFrame(0, GetIconFrame(spriteSys.RsiStateLike(iconSprite)));
var icon_position = new Vector2(xIconOffset / EyeManager.PixelsPerMeter, var iconPosition = new Vector2(xIconOffset / EyeManager.PixelsPerMeter,
yIconOffset / EyeManager.PixelsPerMeter); yIconOffset / EyeManager.PixelsPerMeter);
handle.DrawTexture(_stateIcon, icon_position); handle.DrawTexture(stateIcon, iconPosition);
} }
// we are all progressing towards death every day // we are all progressing towards death every day
(float ratio, bool inCrit) deathProgress = CalcProgress(mob.Owner, mob, dmg); (float ratio, bool inCrit) deathProgress = CalcProgress(uid, mob, dmg, thresholds);
var color = GetProgressColor(deathProgress.ratio, deathProgress.inCrit); var color = GetProgressColor(deathProgress.ratio, deathProgress.inCrit);
@@ -161,13 +162,13 @@ public sealed class EntityHealthBarOverlay : Overlay
private int GetIconFrame(IRsiStateLike sprite) private int GetIconFrame(IRsiStateLike sprite)
{ {
var _spriteSys = _entManager.EntitySysManager.GetEntitySystem<SpriteSystem>(); // var _spriteSys = _entManager.EntitySysManager.GetEntitySystem<SpriteSystem>();
if (sprite.AnimationFrameCount <= 1) if (sprite.AnimationFrameCount <= 1)
return 0; return 0;
var currentFrame = iconFrame; var currentFrame = _iconFrame;
var result = 0; int result;
while (true) while (true)
{ {
if (currentFrame > 0 && currentFrame > sprite.AnimationFrameCount) if (currentFrame > 0 && currentFrame > sprite.AnimationFrameCount)
@@ -186,11 +187,11 @@ public sealed class EntityHealthBarOverlay : Overlay
/// <summary> /// <summary>
/// Returns a ratio between 0 and 1, and whether the entity is in crit. /// Returns a ratio between 0 and 1, and whether the entity is in crit.
/// </summary> /// </summary>
private (float, bool) CalcProgress(EntityUid uid, MobStateComponent component, DamageableComponent dmg) private (float, bool) CalcProgress(EntityUid uid, MobStateComponent component, DamageableComponent dmg, MobThresholdsComponent thresholds)
{ {
if (_mobStateSystem.IsAlive(uid, component)) if (_mobStateSystem.IsAlive(uid, component))
{ {
if (!_mobThresholdSystem.TryGetThresholdForState(uid, MobState.Critical, out var threshold)) if (!_mobThresholdSystem.TryGetThresholdForState(uid, MobState.Critical, out var threshold, thresholds))
return (1, false); return (1, false);
var ratio = 1 - ((FixedPoint2) (dmg.TotalDamage / threshold)).Float(); var ratio = 1 - ((FixedPoint2) (dmg.TotalDamage / threshold)).Float();
@@ -199,8 +200,8 @@ public sealed class EntityHealthBarOverlay : Overlay
if (_mobStateSystem.IsCritical(uid, component)) if (_mobStateSystem.IsCritical(uid, component))
{ {
if (!_mobThresholdSystem.TryGetThresholdForState(uid, MobState.Critical, out var critThreshold) || if (!_mobThresholdSystem.TryGetThresholdForState(uid, MobState.Critical, out var critThreshold, thresholds) ||
!_mobThresholdSystem.TryGetThresholdForState(uid, MobState.Dead, out var deadThreshold)) !_mobThresholdSystem.TryGetThresholdForState(uid, MobState.Dead, out var deadThreshold, thresholds))
{ {
return (1, true); return (1, true);
} }