92 lines
2.8 KiB
C#
92 lines
2.8 KiB
C#
using Robust.Shared.GameStates;
|
|
using Robust.Shared.Serialization;
|
|
|
|
namespace Content.Shared.Gravity;
|
|
|
|
public abstract partial class SharedGravitySystem
|
|
{
|
|
protected const float GravityKick = 100.0f;
|
|
protected const float ShakeCooldown = 0.2f;
|
|
|
|
private void InitializeShake()
|
|
{
|
|
SubscribeLocalEvent<GravityShakeComponent, EntityUnpausedEvent>(OnShakeUnpaused);
|
|
SubscribeLocalEvent<GravityShakeComponent, ComponentGetState>(OnShakeGetState);
|
|
SubscribeLocalEvent<GravityShakeComponent, ComponentHandleState>(OnShakeHandleState);
|
|
}
|
|
|
|
private void OnShakeUnpaused(EntityUid uid, GravityShakeComponent component, ref EntityUnpausedEvent args)
|
|
{
|
|
component.NextShake += args.PausedTime;
|
|
}
|
|
|
|
private void UpdateShake()
|
|
{
|
|
var curTime = Timing.CurTime;
|
|
var gravityQuery = GetEntityQuery<GravityComponent>();
|
|
var query = EntityQueryEnumerator<GravityShakeComponent>();
|
|
|
|
while (query.MoveNext(out var uid, out var comp))
|
|
{
|
|
if (comp.NextShake <= curTime)
|
|
{
|
|
if (comp.ShakeTimes == 0 || !gravityQuery.TryGetComponent(uid, out var gravity))
|
|
{
|
|
RemCompDeferred<GravityShakeComponent>(uid);
|
|
continue;
|
|
}
|
|
|
|
ShakeGrid(uid, gravity);
|
|
comp.ShakeTimes--;
|
|
comp.NextShake += TimeSpan.FromSeconds(ShakeCooldown);
|
|
Dirty(comp);
|
|
}
|
|
}
|
|
}
|
|
|
|
public void StartGridShake(EntityUid uid, GravityComponent? gravity = null)
|
|
{
|
|
if (Terminating(uid))
|
|
return;
|
|
|
|
if (!Resolve(uid, ref gravity, false))
|
|
return;
|
|
|
|
if (!TryComp<GravityShakeComponent>(uid, out var shake))
|
|
{
|
|
shake = AddComp<GravityShakeComponent>(uid);
|
|
shake.NextShake = Timing.CurTime;
|
|
}
|
|
|
|
shake.ShakeTimes = 10;
|
|
Dirty(shake);
|
|
}
|
|
|
|
protected virtual void ShakeGrid(EntityUid uid, GravityComponent? comp = null) {}
|
|
|
|
private void OnShakeHandleState(EntityUid uid, GravityShakeComponent component, ref ComponentHandleState args)
|
|
{
|
|
if (args.Current is not GravityShakeComponentState state)
|
|
return;
|
|
|
|
component.ShakeTimes = state.ShakeTimes;
|
|
component.NextShake = state.NextShake;
|
|
}
|
|
|
|
private void OnShakeGetState(EntityUid uid, GravityShakeComponent component, ref ComponentGetState args)
|
|
{
|
|
args.State = new GravityShakeComponentState()
|
|
{
|
|
ShakeTimes = component.ShakeTimes,
|
|
NextShake = component.NextShake,
|
|
};
|
|
}
|
|
|
|
[Serializable, NetSerializable]
|
|
protected sealed class GravityShakeComponentState : ComponentState
|
|
{
|
|
public int ShakeTimes;
|
|
public TimeSpan NextShake;
|
|
}
|
|
}
|