68ce53ae17
* Use new Subs.CVar helper Removes manual config OnValueChanged calls, removes need to remember to manually unsubscribe. This both reduces boilerplate and fixes many issues where subscriptions weren't removed on entity system shutdown. * Fix a bunch of warnings * More warning fixes * Use new DateTime serializer to get rid of ISerializationHooks in changelog code. * Get rid of some more ISerializationHooks for enums * And a little more * Apply suggestions from code review Co-authored-by: 0x6273 <0x40@keemail.me> --------- Co-authored-by: 0x6273 <0x40@keemail.me>
72 lines
2.4 KiB
C#
72 lines
2.4 KiB
C#
using Robust.Shared.Physics.Events;
|
|
using Robust.Shared.Physics.Systems;
|
|
|
|
namespace Content.Shared.Placeable;
|
|
|
|
/// <summary>
|
|
/// Tracks placed entities
|
|
/// Subscribe to <see cref="ItemPlacedEvent"/> or <see cref="ItemRemovedEvent"/> to do things when items or placed or removed.
|
|
/// </summary>
|
|
public sealed class ItemPlacerSystem : EntitySystem
|
|
{
|
|
[Dependency] private readonly CollisionWakeSystem _wake = default!;
|
|
[Dependency] private readonly PlaceableSurfaceSystem _placeableSurface = default!;
|
|
|
|
public override void Initialize()
|
|
{
|
|
base.Initialize();
|
|
|
|
SubscribeLocalEvent<ItemPlacerComponent, StartCollideEvent>(OnStartCollide);
|
|
SubscribeLocalEvent<ItemPlacerComponent, EndCollideEvent>(OnEndCollide);
|
|
}
|
|
|
|
private void OnStartCollide(EntityUid uid, ItemPlacerComponent comp, ref StartCollideEvent args)
|
|
{
|
|
if (comp.Whitelist != null && !comp.Whitelist.IsValid(args.OtherEntity))
|
|
return;
|
|
|
|
if (TryComp<CollisionWakeComponent>(args.OtherEntity, out var wakeComp))
|
|
_wake.SetEnabled(args.OtherEntity, false, wakeComp);
|
|
|
|
var count = comp.PlacedEntities.Count;
|
|
if (comp.MaxEntities == 0 || count < comp.MaxEntities)
|
|
{
|
|
comp.PlacedEntities.Add(args.OtherEntity);
|
|
|
|
var ev = new ItemPlacedEvent(args.OtherEntity);
|
|
RaiseLocalEvent(uid, ref ev);
|
|
}
|
|
|
|
if (comp.MaxEntities > 0 && count >= (comp.MaxEntities - 1))
|
|
{
|
|
// Don't let any more items be placed if it's reached its limit.
|
|
_placeableSurface.SetPlaceable(uid, false);
|
|
}
|
|
}
|
|
|
|
private void OnEndCollide(EntityUid uid, ItemPlacerComponent comp, ref EndCollideEvent args)
|
|
{
|
|
if (TryComp<CollisionWakeComponent>(args.OtherEntity, out var wakeComp))
|
|
_wake.SetEnabled(args.OtherEntity, true, wakeComp);
|
|
|
|
comp.PlacedEntities.Remove(args.OtherEntity);
|
|
|
|
var ev = new ItemRemovedEvent(args.OtherEntity);
|
|
RaiseLocalEvent(uid, ref ev);
|
|
|
|
_placeableSurface.SetPlaceable(uid, true);
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// Raised on the <see cref="ItemPlacer"/> when an item is placed and it is under the item limit.
|
|
/// </summary>
|
|
[ByRefEvent]
|
|
public readonly record struct ItemPlacedEvent(EntityUid OtherEntity);
|
|
|
|
/// <summary>
|
|
/// Raised on the <see cref="ItemPlacer"/> when an item is removed from it.
|
|
/// </summary>
|
|
[ByRefEvent]
|
|
public readonly record struct ItemRemovedEvent(EntityUid OtherEntity);
|