--- /dev/null
+using Robust.Shared.Audio;
+using Robust.Shared.Serialization.TypeSerializers.Implementations.Custom;
+
+namespace Content.Server.Pinpointer;
+
+/// <summary>
+/// This is used for an item that beeps based on
+/// proximity to a specified component.
+/// </summary>
+[RegisterComponent, Access(typeof(ProximityBeeperSystem))]
+public sealed class ProximityBeeperComponent : Component
+{
+ /// <summary>
+ /// Whether or not it's on.
+ /// </summary>
+ [DataField("enabled")]
+ public bool Enabled;
+
+ /// <summary>
+ /// The target component that is being searched for
+ /// </summary>
+ [DataField("component", required: true), ViewVariables(VVAccess.ReadWrite)]
+ public string Component = default!;
+
+ /// <summary>
+ /// The farthest distance a target can be for the beep to occur
+ /// </summary>
+ [DataField("maximumDistance"), ViewVariables(VVAccess.ReadWrite)]
+ public float MaximumDistance = 10f;
+
+ /// <summary>
+ /// The maximum interval between beeps.
+ /// </summary>
+ [DataField("maxBeepInterval"), ViewVariables(VVAccess.ReadWrite)]
+ public TimeSpan MaxBeepInterval = TimeSpan.FromSeconds(1.5f);
+
+ /// <summary>
+ /// The minimum interval between beeps.
+ /// </summary>
+ [DataField("minBeepInterval"), ViewVariables(VVAccess.ReadWrite)]
+ public TimeSpan MinBeepInterval = TimeSpan.FromSeconds(0.25f);
+
+ /// <summary>
+ /// When the next beep will occur
+ /// </summary>
+ [DataField("nextBeepTime", customTypeSerializer: typeof(TimeOffsetSerializer)), ViewVariables(VVAccess.ReadWrite)]
+ public TimeSpan NextBeepTime;
+
+ /// <summary>
+ /// The sound played when the locator beeps.
+ /// </summary>
+ [DataField("beepSound")]
+ public SoundSpecifier? BeepSound;
+}
--- /dev/null
+using Content.Server.PowerCell;
+using Content.Shared.Interaction.Events;
+using Content.Shared.Pinpointer;
+using Robust.Server.GameObjects;
+using Robust.Shared.Timing;
+
+namespace Content.Server.Pinpointer;
+
+/// <summary>
+/// This handles logic and interaction relating to <see cref="ProximityBeeperComponent"/>
+/// </summary>
+public sealed class ProximityBeeperSystem : EntitySystem
+{
+ [Dependency] private readonly IGameTiming _timing = default!;
+ [Dependency] private readonly AppearanceSystem _appearance = default!;
+ [Dependency] private readonly AudioSystem _audio = default!;
+ [Dependency] private readonly EntityLookupSystem _entityLookup = default!;
+ [Dependency] private readonly PowerCellSystem _powerCell = default!;
+ [Dependency] private readonly TransformSystem _transform = default!;
+
+ /// <inheritdoc/>
+ public override void Initialize()
+ {
+ SubscribeLocalEvent<ProximityBeeperComponent, UseInHandEvent>(OnUseInHand);
+ SubscribeLocalEvent<ProximityBeeperComponent, ComponentInit>(OnInit);
+ SubscribeLocalEvent<ProximityBeeperComponent, EntityUnpausedEvent>(OnUnpaused);
+ SubscribeLocalEvent<ProximityBeeperComponent, PowerCellSlotEmptyEvent>(OnPowerCellSlotEmpty);
+ }
+ private void OnUseInHand(EntityUid uid, ProximityBeeperComponent component, UseInHandEvent args)
+ {
+ if (args.Handled)
+ return;
+
+ args.Handled = TryToggle(uid, component, args.User);
+ }
+
+ private void OnInit(EntityUid uid, ProximityBeeperComponent component, ComponentInit args)
+ {
+ if (component.NextBeepTime < _timing.CurTime)
+ component.NextBeepTime = _timing.CurTime;
+ }
+
+ private void OnUnpaused(EntityUid uid, ProximityBeeperComponent component, ref EntityUnpausedEvent args)
+ {
+ component.NextBeepTime += args.PausedTime;
+ }
+
+ private void OnPowerCellSlotEmpty(EntityUid uid, ProximityBeeperComponent component, ref PowerCellSlotEmptyEvent args)
+ {
+ if (component.Enabled)
+ TryDisable(uid, component);
+ }
+
+ /// <summary>
+ /// Beeps the proximitybeeper as well as sets the time for the next beep
+ /// based on proximity to entities with the target component.
+ /// </summary>
+ public void UpdateBeep(EntityUid uid, ProximityBeeperComponent? component = null, bool playBeep = true)
+ {
+ if (!Resolve(uid, ref component))
+ return;
+
+ if (!component.Enabled)
+ {
+ component.NextBeepTime += component.MinBeepInterval;
+ return;
+ }
+
+ var xformQuery = GetEntityQuery<TransformComponent>();
+ var xform = xformQuery.GetComponent(uid);
+ var comp = EntityManager.ComponentFactory.GetRegistration(component.Component).Type;
+ float? closestDistance = null;
+ foreach (var targetXform in _entityLookup.GetComponentsInRange<TransformComponent>(xform.MapPosition, component.MaximumDistance))
+ {
+ // forgive me father, for i have sinned.
+ var ent = targetXform.Owner;
+
+ if (!HasComp(ent, comp))
+ continue;
+
+ var dist = (_transform.GetWorldPosition(xform, xformQuery) - _transform.GetWorldPosition(targetXform, xformQuery)).Length;
+ if (dist >= (closestDistance ?? float.MaxValue))
+ continue;
+ closestDistance = dist;
+ }
+
+ if (closestDistance is not { } distance)
+ return;
+
+ if (playBeep)
+ _audio.PlayPvs(component.BeepSound, uid);
+
+ var scalingFactor = distance / component.MaximumDistance;
+ var interval = (component.MaxBeepInterval - component.MinBeepInterval) * scalingFactor + component.MinBeepInterval;
+ component.NextBeepTime += interval;
+ }
+
+ /// <summary>
+ /// Enables the proximity beeper
+ /// </summary>
+ public bool TryEnable(EntityUid uid, ProximityBeeperComponent? component = null, EntityUid? user = null)
+ {
+ if (!Resolve(uid, ref component))
+ return false;
+
+ TryComp<PowerCellDrawComponent>(uid, out var draw);
+
+ if (!_powerCell.HasActivatableCharge(uid, battery: draw, user: user))
+ return false;
+
+ component.Enabled = true;
+ _appearance.SetData(uid, ProximityBeeperVisuals.Enabled, true);
+ component.NextBeepTime = _timing.CurTime;
+ UpdateBeep(uid, component, false);
+ if (draw != null)
+ draw.Enabled = true;
+ return true;
+ }
+
+ /// <summary>
+ /// Disables the proximity beeper
+ /// </summary>
+ public bool TryDisable(EntityUid uid, ProximityBeeperComponent? component = null)
+ {
+ if (!Resolve(uid, ref component))
+ return false;
+
+ if (!component.Enabled)
+ return false;
+
+ component.Enabled = false;
+ _appearance.SetData(uid, ProximityBeeperVisuals.Enabled, false);
+ if (TryComp<PowerCellDrawComponent>(uid, out var draw))
+ draw.Enabled = true;
+ UpdateBeep(uid, component);
+ return true;
+ }
+
+ /// <summary>
+ /// toggles the proximity beeper
+ /// </summary>
+ public bool TryToggle(EntityUid uid, ProximityBeeperComponent? component = null, EntityUid? user = null)
+ {
+ if (!Resolve(uid, ref component))
+ return false;
+
+ return component.Enabled
+ ? TryDisable(uid, component)
+ : TryEnable(uid, component, user);
+ }
+
+ public override void Update(float frameTime)
+ {
+ base.Update(frameTime);
+
+ var query = EntityQueryEnumerator<ProximityBeeperComponent>();
+ while (query.MoveNext(out var uid, out var beeper))
+ {
+ if (!beeper.Enabled)
+ continue;
+
+ if (_timing.CurTime < beeper.NextBeepTime)
+ continue;
+ UpdateBeep(uid, beeper);
+ }
+ }
+}
projects / space-station-14.git / commitdiff
