tommy/tbd-station-14
2
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Make smoking cool again. (#6046)

Browse Source
This commit is contained in:
Leon Friedrich
2022-01-07 01:45:02 +13:00
committed by GitHub
parent 6af1bfabf3
commit 4921ea6850
3 changed files with 38 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

56
Content.Client/Smoking/BurnStateVisualizer.cs
View File

@@ -1,16 +1,18 @@
using Content.Client.Clothing;
using Content.Shared.Smoking;
using JetBrains.Annotations;
using Robust.Client.GameObjects;
using Robust.Shared.GameObjects;
using Robust.Shared.IoC;
using Robust.Shared.Serialization;
using Robust.Shared.Serialization.Manager.Attributes;
namespace Content.Client.Smoking
{
[UsedImplicitly]
public class BurnStateVisualizer : AppearanceVisualizer
public sealed class BurnStateVisualizer : AppearanceVisualizer, ISerializationHooks
{
[Dependency] private readonly IEntityManager _entMan = default!;
[DataField("burntIcon")]
private string _burntIcon = "burnt-icon";
[DataField("litIcon")]
@@ -18,49 +20,29 @@ namespace Content.Client.Smoking
[DataField("unlitIcon")]
private string _unlitIcon = "icon";
[DataField("burntPrefix")]
private string _burntPrefix = "unlit";
[DataField("litPrefix")]
private string _litPrefix = "lit";
[DataField("unlitPrefix")]
private string _unlitPrefix = "unlit";
void ISerializationHooks.AfterDeserialization()
{
IoCManager.InjectDependencies(this);
}
public override void OnChangeData(AppearanceComponent component)
{
base.OnChangeData(component);
if (component.TryGetData<SmokableState>(SmokingVisuals.Smoking, out var smoking))
{
SetState(component, smoking);
}
}
if (!_entMan.TryGetComponent(component.Owner, out SpriteComponent? sprite))
return;
private void SetState(AppearanceComponent component, SmokableState burnState)
{
var entities = IoCManager.Resolve<IEntityManager>();
var clothing = entities.GetComponentOrNull<ClothingComponent>(component.Owner);
if (!component.TryGetData<SmokableState>(SmokingVisuals.Smoking, out var burnState))
return;
if (entities.TryGetComponent(component.Owner, out ISpriteComponent sprite))
var state = burnState switch
{
switch (burnState)
{
case SmokableState.Lit:
if (clothing != null)
clothing.EquippedPrefix = _litPrefix;
sprite.LayerSetState(0, _litIcon);
break;
case SmokableState.Burnt:
if (clothing != null)
clothing.EquippedPrefix = _burntPrefix;
sprite.LayerSetState(0, _burntIcon);
break;
case SmokableState.Unlit:
if (clothing != null)
clothing.EquippedPrefix = _unlitPrefix;
sprite.LayerSetState(0, _unlitIcon);
break;
}
}
SmokableState.Lit => _litIcon,
SmokableState.Burnt => _burntIcon,
_ => _unlitIcon
};
sprite.LayerSetState(0, state);
}
}
}