65 lines
1.8 KiB
C#
65 lines
1.8 KiB
C#
using Content.Shared.AME;
|
|
using Robust.Server.GameObjects;
|
|
|
|
namespace Content.Server.AME.Components
|
|
{
|
|
[RegisterComponent]
|
|
public sealed class AMEShieldComponent : SharedAMEShieldComponent
|
|
{
|
|
|
|
private bool _isCore = false;
|
|
|
|
[ViewVariables]
|
|
public int CoreIntegrity = 100;
|
|
|
|
private AppearanceComponent? _appearance;
|
|
private PointLightComponent? _pointLight;
|
|
|
|
protected override void Initialize()
|
|
{
|
|
base.Initialize();
|
|
var entMan = IoCManager.Resolve<IEntityManager>();
|
|
entMan.TryGetComponent(Owner, out _appearance);
|
|
entMan.TryGetComponent(Owner, out _pointLight);
|
|
}
|
|
|
|
public void SetCore()
|
|
{
|
|
if(_isCore) { return; }
|
|
_isCore = true;
|
|
_appearance?.SetData(AMEShieldVisuals.Core, "isCore");
|
|
}
|
|
|
|
public void UnsetCore()
|
|
{
|
|
_isCore = false;
|
|
_appearance?.SetData(AMEShieldVisuals.Core, "isNotCore");
|
|
UpdateCoreVisuals(0, false);
|
|
}
|
|
|
|
public void UpdateCoreVisuals(int injectionStrength, bool injecting)
|
|
{
|
|
if (!injecting)
|
|
{
|
|
_appearance?.SetData(AMEShieldVisuals.CoreState, "off");
|
|
if (_pointLight != null) { _pointLight.Enabled = false; }
|
|
return;
|
|
}
|
|
|
|
if (_pointLight != null)
|
|
{
|
|
_pointLight.Radius = Math.Clamp(injectionStrength, 1, 12);
|
|
_pointLight.Enabled = true;
|
|
}
|
|
|
|
if (injectionStrength > 2)
|
|
{
|
|
_appearance?.SetData(AMEShieldVisuals.CoreState, "strong");
|
|
return;
|
|
}
|
|
|
|
_appearance?.SetData(AMEShieldVisuals.CoreState, "weak");
|
|
}
|
|
}
|
|
}
|