322 lines
15 KiB
C#
322 lines
15 KiB
C#
using System.Linq;
|
|
using Content.Server.Chemistry.EntitySystems;
|
|
using Content.Server.Construction;
|
|
using Content.Server.Kitchen.Components;
|
|
using Content.Server.Power.Components;
|
|
using Content.Server.Power.EntitySystems;
|
|
using Content.Server.Stack;
|
|
using Content.Shared.Chemistry.Components;
|
|
using Content.Shared.Containers.ItemSlots;
|
|
using Content.Shared.FixedPoint;
|
|
using Content.Shared.Interaction;
|
|
using Content.Shared.Kitchen;
|
|
using Content.Shared.Popups;
|
|
using Content.Shared.Random.Helpers;
|
|
using Content.Shared.Stacks;
|
|
using JetBrains.Annotations;
|
|
using Robust.Server.GameObjects;
|
|
using Robust.Shared.Audio;
|
|
using Robust.Shared.Containers;
|
|
using Robust.Shared.Player;
|
|
using Robust.Shared.Timing;
|
|
|
|
namespace Content.Server.Kitchen.EntitySystems
|
|
{
|
|
[UsedImplicitly]
|
|
internal sealed class ReagentGrinderSystem : EntitySystem
|
|
{
|
|
[Dependency] private readonly IGameTiming _timing = default!;
|
|
[Dependency] private readonly SolutionContainerSystem _solutionsSystem = default!;
|
|
[Dependency] private readonly ItemSlotsSystem _itemSlotsSystem = default!;
|
|
[Dependency] private readonly SharedPopupSystem _popupSystem = default!;
|
|
[Dependency] private readonly UserInterfaceSystem _userInterfaceSystem = default!;
|
|
[Dependency] private readonly StackSystem _stackSystem = default!;
|
|
[Dependency] private readonly SharedAudioSystem _audioSystem = default!;
|
|
[Dependency] private readonly SharedAppearanceSystem _appearanceSystem = default!;
|
|
[Dependency] private readonly SharedContainerSystem _containerSystem = default!;
|
|
|
|
public override void Initialize()
|
|
{
|
|
base.Initialize();
|
|
|
|
SubscribeLocalEvent<ReagentGrinderComponent, ComponentStartup>((uid, _, _) => UpdateUiState(uid));
|
|
SubscribeLocalEvent((EntityUid uid, ReagentGrinderComponent _, ref PowerChangedEvent _) => UpdateUiState(uid));
|
|
SubscribeLocalEvent<ReagentGrinderComponent, InteractUsingEvent>(OnInteractUsing);
|
|
SubscribeLocalEvent<ReagentGrinderComponent, RefreshPartsEvent>(OnRefreshParts);
|
|
SubscribeLocalEvent<ReagentGrinderComponent, UpgradeExamineEvent>(OnUpgradeExamine);
|
|
|
|
SubscribeLocalEvent<ReagentGrinderComponent, EntInsertedIntoContainerMessage>(OnContainerModified);
|
|
SubscribeLocalEvent<ReagentGrinderComponent, EntRemovedFromContainerMessage>(OnContainerModified);
|
|
SubscribeLocalEvent<ReagentGrinderComponent, ContainerIsRemovingAttemptEvent>(OnEntRemoveAttempt);
|
|
|
|
SubscribeLocalEvent<ReagentGrinderComponent, ReagentGrinderStartMessage>(OnStartMessage);
|
|
SubscribeLocalEvent<ReagentGrinderComponent, ReagentGrinderEjectChamberAllMessage>(OnEjectChamberAllMessage);
|
|
SubscribeLocalEvent<ReagentGrinderComponent, ReagentGrinderEjectChamberContentMessage>(OnEjectChamberContentMessage);
|
|
}
|
|
|
|
public override void Update(float frameTime)
|
|
{
|
|
base.Update(frameTime);
|
|
|
|
foreach (var (active, reagentGrinder) in EntityQuery<ActiveReagentGrinderComponent, ReagentGrinderComponent>())
|
|
{
|
|
var uid = reagentGrinder.Owner;
|
|
|
|
if (active.EndTime > _timing.CurTime)
|
|
continue;
|
|
|
|
reagentGrinder.AudioStream?.Stop();
|
|
RemCompDeferred<ActiveReagentGrinderComponent>(uid);
|
|
|
|
var inputContainer = _containerSystem.EnsureContainer<Container>(uid, SharedReagentGrinder.InputContainerId);
|
|
var outputContainer = _itemSlotsSystem.GetItemOrNull(uid, SharedReagentGrinder.BeakerSlotId);
|
|
if (outputContainer is null || !_solutionsSystem.TryGetFitsInDispenser(outputContainer.Value, out var containerSolution))
|
|
continue;
|
|
|
|
foreach (var item in inputContainer.ContainedEntities.ToList())
|
|
{
|
|
var solution = active.Program switch
|
|
{
|
|
GrinderProgram.Grind => GetGrindSolution(item),
|
|
GrinderProgram.Juice => CompOrNull<ExtractableComponent>(item)?.JuiceSolution,
|
|
_ => null,
|
|
};
|
|
|
|
if (solution is null)
|
|
continue;
|
|
|
|
if (TryComp<StackComponent>(item, out var stack))
|
|
{
|
|
var totalVolume = solution.TotalVolume * stack.Count;
|
|
if (totalVolume <= 0)
|
|
continue;
|
|
|
|
// Maximum number of items we can process in the stack without going over AvailableVolume
|
|
// We add a small tolerance, because floats are inaccurate.
|
|
var fitsCount = (int) (stack.Count * FixedPoint2.Min(containerSolution.AvailableVolume / totalVolume + 0.01, 1));
|
|
if (fitsCount <= 0)
|
|
continue;
|
|
|
|
solution.ScaleSolution(fitsCount);
|
|
_stackSystem.SetCount(item, stack.Count - fitsCount); // Setting to 0 will QueueDel
|
|
}
|
|
else
|
|
{
|
|
if (solution.TotalVolume > containerSolution.AvailableVolume)
|
|
continue;
|
|
|
|
QueueDel(item);
|
|
}
|
|
|
|
_solutionsSystem.TryAddSolution(outputContainer.Value, containerSolution, solution);
|
|
}
|
|
|
|
_userInterfaceSystem.TrySendUiMessage(uid, ReagentGrinderUiKey.Key,
|
|
new ReagentGrinderWorkCompleteMessage());
|
|
|
|
UpdateUiState(uid);
|
|
}
|
|
}
|
|
|
|
private void OnEntRemoveAttempt(EntityUid uid, ReagentGrinderComponent reagentGrinder, ContainerIsRemovingAttemptEvent args)
|
|
{
|
|
if (HasComp<ActiveReagentGrinderComponent>(uid))
|
|
args.Cancel();
|
|
}
|
|
|
|
private void OnContainerModified(EntityUid uid, ReagentGrinderComponent reagentGrinder, ContainerModifiedMessage args)
|
|
{
|
|
UpdateUiState(uid);
|
|
|
|
var outputContainer = _itemSlotsSystem.GetItemOrNull(uid, SharedReagentGrinder.BeakerSlotId);
|
|
_appearanceSystem.SetData(uid, ReagentGrinderVisualState.BeakerAttached, outputContainer.HasValue);
|
|
}
|
|
|
|
private void OnInteractUsing(EntityUid uid, ReagentGrinderComponent reagentGrinder, InteractUsingEvent args)
|
|
{
|
|
var heldEnt = args.Used;
|
|
var inputContainer = _containerSystem.EnsureContainer<Container>(uid, SharedReagentGrinder.InputContainerId);
|
|
|
|
if (!HasComp<ExtractableComponent>(heldEnt))
|
|
{
|
|
if (!HasComp<FitsInDispenserComponent>(heldEnt))
|
|
{
|
|
// This is ugly but we can't use whitelistFailPopup because there are 2 containers with different whitelists.
|
|
_popupSystem.PopupEntity(Loc.GetString("reagent-grinder-component-cannot-put-entity-message"), uid, args.User);
|
|
}
|
|
|
|
// Entity did NOT pass the whitelist for grind/juice.
|
|
// Wouldn't want the clown grinding up the Captain's ID card now would you?
|
|
// Why am I asking you? You're biased.
|
|
return;
|
|
}
|
|
|
|
if (args.Handled)
|
|
return;
|
|
|
|
// Cap the chamber. Don't want someone putting in 500 entities and ejecting them all at once.
|
|
// Maybe I should have done that for the microwave too?
|
|
if (inputContainer.ContainedEntities.Count >= reagentGrinder.StorageMaxEntities)
|
|
return;
|
|
|
|
if (!inputContainer.Insert(heldEnt, EntityManager))
|
|
return;
|
|
|
|
args.Handled = true;
|
|
}
|
|
|
|
/// <remarks>
|
|
/// Gotta be efficient, you know? you're saving a whole extra second here and everything.
|
|
/// </remarks>
|
|
private void OnRefreshParts(EntityUid uid, ReagentGrinderComponent component, RefreshPartsEvent args)
|
|
{
|
|
var ratingWorkTime = args.PartRatings[component.MachinePartWorkTime];
|
|
var ratingStorage = args.PartRatings[component.MachinePartStorageMax];
|
|
|
|
component.WorkTimeMultiplier = MathF.Pow(component.PartRatingWorkTimerMulitplier, ratingWorkTime - 1);
|
|
component.StorageMaxEntities = component.BaseStorageMaxEntities + (int) (component.StoragePerPartRating * (ratingStorage - 1));
|
|
}
|
|
|
|
private void OnUpgradeExamine(EntityUid uid, ReagentGrinderComponent component, UpgradeExamineEvent args)
|
|
{
|
|
args.AddPercentageUpgrade("reagent-grinder-component-upgrade-work-time", component.WorkTimeMultiplier);
|
|
args.AddNumberUpgrade("reagent-grinder-component-upgrade-storage", component.StorageMaxEntities - component.BaseStorageMaxEntities);
|
|
}
|
|
|
|
private void UpdateUiState(EntityUid uid)
|
|
{
|
|
var inputContainer = _containerSystem.EnsureContainer<Container>(uid, SharedReagentGrinder.InputContainerId);
|
|
var outputContainer = _itemSlotsSystem.GetItemOrNull(uid, SharedReagentGrinder.BeakerSlotId);
|
|
Solution? containerSolution = null;
|
|
var isBusy = HasComp<ActiveReagentGrinderComponent>(uid);
|
|
var canJuice = false;
|
|
var canGrind = false;
|
|
|
|
if (outputContainer is not null
|
|
&& _solutionsSystem.TryGetFitsInDispenser(outputContainer.Value, out containerSolution)
|
|
&& inputContainer.ContainedEntities.Count > 0)
|
|
{
|
|
canGrind = inputContainer.ContainedEntities.All(CanGrind);
|
|
canJuice = inputContainer.ContainedEntities.All(CanJuice);
|
|
}
|
|
|
|
var state = new ReagentGrinderInterfaceState(
|
|
isBusy,
|
|
outputContainer.HasValue,
|
|
this.IsPowered(uid, EntityManager),
|
|
canJuice,
|
|
canGrind,
|
|
inputContainer.ContainedEntities.Select(item => item).ToArray(),
|
|
containerSolution?.Contents.ToArray()
|
|
);
|
|
_userInterfaceSystem.TrySetUiState(uid, ReagentGrinderUiKey.Key, state);
|
|
}
|
|
|
|
private void OnStartMessage(EntityUid uid, ReagentGrinderComponent reagentGrinder, ReagentGrinderStartMessage message)
|
|
{
|
|
if (!this.IsPowered(uid, EntityManager) || HasComp<ActiveReagentGrinderComponent>(uid))
|
|
return;
|
|
|
|
DoWork(uid, reagentGrinder, message.Program);
|
|
}
|
|
|
|
private void OnEjectChamberAllMessage(EntityUid uid, ReagentGrinderComponent reagentGrinder, ReagentGrinderEjectChamberAllMessage message)
|
|
{
|
|
var inputContainer = _containerSystem.EnsureContainer<Container>(uid, SharedReagentGrinder.InputContainerId);
|
|
|
|
if (HasComp<ActiveReagentGrinderComponent>(uid) || inputContainer.ContainedEntities.Count <= 0)
|
|
return;
|
|
|
|
ClickSound(uid, reagentGrinder);
|
|
foreach (var entity in inputContainer.ContainedEntities.ToList())
|
|
{
|
|
inputContainer.Remove(entity);
|
|
entity.RandomOffset(0.4f);
|
|
}
|
|
UpdateUiState(uid);
|
|
}
|
|
|
|
private void OnEjectChamberContentMessage(EntityUid uid, ReagentGrinderComponent reagentGrinder, ReagentGrinderEjectChamberContentMessage message)
|
|
{
|
|
if (HasComp<ActiveReagentGrinderComponent>(uid))
|
|
return;
|
|
|
|
var inputContainer = _containerSystem.EnsureContainer<Container>(uid, SharedReagentGrinder.InputContainerId);
|
|
|
|
if (inputContainer.Remove(message.EntityId))
|
|
{
|
|
message.EntityId.RandomOffset(0.4f);
|
|
ClickSound(uid, reagentGrinder);
|
|
UpdateUiState(uid);
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// The wzhzhzh of the grinder. Processes the contents of the grinder and puts the output in the beaker.
|
|
/// </summary>
|
|
/// <param name="uid">The grinder itself</param>
|
|
/// <param name="reagentGrinder"></param>
|
|
/// <param name="program">Which program, such as grind or juice</param>
|
|
private void DoWork(EntityUid uid, ReagentGrinderComponent reagentGrinder, GrinderProgram program)
|
|
{
|
|
var inputContainer = _containerSystem.EnsureContainer<Container>(uid, SharedReagentGrinder.InputContainerId);
|
|
var outputContainer = _itemSlotsSystem.GetItemOrNull(uid, SharedReagentGrinder.BeakerSlotId);
|
|
|
|
// Do we have anything to grind/juice and a container to put the reagents in?
|
|
if (inputContainer.ContainedEntities.Count <= 0 || !HasComp<FitsInDispenserComponent>(outputContainer))
|
|
return;
|
|
|
|
SoundSpecifier? sound;
|
|
switch (program)
|
|
{
|
|
case GrinderProgram.Grind when inputContainer.ContainedEntities.All(CanGrind):
|
|
sound = reagentGrinder.GrindSound;
|
|
break;
|
|
case GrinderProgram.Juice when inputContainer.ContainedEntities.All(CanJuice):
|
|
sound = reagentGrinder.JuiceSound;
|
|
break;
|
|
default:
|
|
return;
|
|
}
|
|
|
|
var active = AddComp<ActiveReagentGrinderComponent>(uid);
|
|
active.EndTime = _timing.CurTime + reagentGrinder.WorkTime * reagentGrinder.WorkTimeMultiplier;
|
|
active.Program = program;
|
|
|
|
reagentGrinder.AudioStream = _audioSystem.PlayPvs(sound, uid,
|
|
AudioParams.Default.WithPitchScale(1 / reagentGrinder.WorkTimeMultiplier)); //slightly higher pitched
|
|
_userInterfaceSystem.TrySendUiMessage(uid, ReagentGrinderUiKey.Key,
|
|
new ReagentGrinderWorkStartedMessage(program));
|
|
}
|
|
|
|
private void ClickSound(EntityUid uid, ReagentGrinderComponent reagentGrinder)
|
|
{
|
|
_audioSystem.PlayPvs(reagentGrinder.ClickSound, uid, AudioParams.Default.WithVolume(-2f));
|
|
}
|
|
|
|
private Solution? GetGrindSolution(EntityUid uid)
|
|
{
|
|
if (TryComp<ExtractableComponent>(uid, out var extractable)
|
|
&& extractable.GrindableSolution is not null
|
|
&& _solutionsSystem.TryGetSolution(uid, extractable.GrindableSolution, out var solution))
|
|
{
|
|
return solution;
|
|
}
|
|
else
|
|
return null;
|
|
}
|
|
|
|
private bool CanGrind(EntityUid uid)
|
|
{
|
|
var solutionName = CompOrNull<ExtractableComponent>(uid)?.GrindableSolution;
|
|
|
|
return solutionName is not null && _solutionsSystem.TryGetSolution(uid, solutionName, out _);
|
|
}
|
|
|
|
private bool CanJuice(EntityUid uid)
|
|
{
|
|
return CompOrNull<ExtractableComponent>(uid)?.JuiceSolution is not null;
|
|
}
|
|
}
|
|
}
|