88 lines
3.8 KiB
C#
88 lines
3.8 KiB
C#
using Content.Server.Chemistry.EntitySystems;
|
|
using Content.Shared.Body.Prototypes;
|
|
using Content.Shared.Chemistry.Reagent;
|
|
using Content.Shared.FixedPoint;
|
|
using JetBrains.Annotations;
|
|
using Robust.Shared.Prototypes;
|
|
using Robust.Shared.Serialization.TypeSerializers.Implementations.Custom.Prototype;
|
|
|
|
namespace Content.Server.Chemistry.ReagentEffects
|
|
{
|
|
[UsedImplicitly]
|
|
public sealed partial class AdjustReagent : ReagentEffect
|
|
{
|
|
/// <summary>
|
|
/// The reagent ID to remove. Only one of this and <see cref="Group"/> should be active.
|
|
/// </summary>
|
|
[DataField("reagent", customTypeSerializer:typeof(PrototypeIdSerializer<ReagentPrototype>))]
|
|
public string? Reagent = null;
|
|
// TODO use ReagentId
|
|
|
|
/// <summary>
|
|
/// The metabolism group to remove, if the reagent satisfies any.
|
|
/// Only one of this and <see cref="Reagent"/> should be active.
|
|
/// </summary>
|
|
[DataField("group", customTypeSerializer:typeof(PrototypeIdSerializer<MetabolismGroupPrototype>))]
|
|
public string? Group = null;
|
|
|
|
[DataField("amount", required: true)]
|
|
public FixedPoint2 Amount = default!;
|
|
|
|
public override void Effect(ReagentEffectArgs args)
|
|
{
|
|
if (args.Source != null)
|
|
{
|
|
var solutionSys = args.EntityManager.EntitySysManager.GetEntitySystem<SolutionContainerSystem>();
|
|
var amount = Amount;
|
|
|
|
amount *= args.Scale;
|
|
|
|
if (Reagent != null)
|
|
{
|
|
if (amount < 0 && args.Source.ContainsPrototype(Reagent))
|
|
solutionSys.RemoveReagent(args.SolutionEntity, args.Source, Reagent, -amount);
|
|
if (amount > 0)
|
|
solutionSys.TryAddReagent(args.SolutionEntity, args.Source, Reagent, amount, out _);
|
|
}
|
|
else if (Group != null)
|
|
{
|
|
var prototypeMan = IoCManager.Resolve<IPrototypeManager>();
|
|
foreach (var quant in args.Source.Contents.ToArray())
|
|
{
|
|
var proto = prototypeMan.Index<ReagentPrototype>(quant.Reagent.Prototype);
|
|
if (proto.Metabolisms != null && proto.Metabolisms.ContainsKey(Group))
|
|
{
|
|
if (amount < 0)
|
|
solutionSys.RemoveReagent(args.SolutionEntity, args.Source, quant.Reagent, amount);
|
|
if (amount > 0)
|
|
solutionSys.TryAddReagent(args.SolutionEntity, args.Source, quant.Reagent, amount, out _);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
protected override string? ReagentEffectGuidebookText(IPrototypeManager prototype, IEntitySystemManager entSys)
|
|
{
|
|
if (Reagent is not null && prototype.TryIndex(Reagent, out ReagentPrototype? reagentProto))
|
|
{
|
|
return Loc.GetString("reagent-effect-guidebook-adjust-reagent-reagent",
|
|
("chance", Probability),
|
|
("deltasign", MathF.Sign(Amount.Float())),
|
|
("reagent", reagentProto.LocalizedName),
|
|
("amount", MathF.Abs(Amount.Float())));
|
|
}
|
|
else if (Group is not null && prototype.TryIndex(Group, out MetabolismGroupPrototype? groupProto))
|
|
{
|
|
return Loc.GetString("reagent-effect-guidebook-adjust-reagent-group",
|
|
("chance", Probability),
|
|
("deltasign", MathF.Sign(Amount.Float())),
|
|
("group", groupProto.ID),
|
|
("amount", MathF.Abs(Amount.Float())));
|
|
}
|
|
|
|
throw new NotImplementedException();
|
|
}
|
|
}
|
|
}
|