62 lines
1.8 KiB
C#
62 lines
1.8 KiB
C#
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);
|
|
}
|
|
|
|
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) {}
|
|
}
|