Multiantag Gamemode (#37783)
Co-authored-by: slarticodefast <161409025+slarticodefast@users.noreply.github.com> Co-authored-by: Southbridge <7013162+southbridge-fur@users.noreply.github.com> Co-authored-by: ScarKy0 <106310278+ScarKy0@users.noreply.github.com>
This commit is contained in:
51
Content.Shared/EntityTable/Conditions/HasBudgetCondition.cs
Normal file
51
Content.Shared/EntityTable/Conditions/HasBudgetCondition.cs
Normal file
@@ -0,0 +1,51 @@
|
||||
using Content.Shared.EntityTable.EntitySelectors;
|
||||
using Content.Shared.GameTicking.Rules;
|
||||
using Robust.Shared.Prototypes;
|
||||
|
||||
namespace Content.Shared.EntityTable.Conditions;
|
||||
|
||||
/// <summary>
|
||||
/// Condition that only succeeds if a table supplies a sufficient "cost" to a given
|
||||
/// </summary>
|
||||
public sealed partial class HasBudgetCondition : EntityTableCondition
|
||||
{
|
||||
public const string BudgetContextKey = "Budget";
|
||||
|
||||
/// <summary>
|
||||
/// Used for determining the cost for the budget.
|
||||
/// If null, attempts to fetch the cost from the attached selector.
|
||||
/// </summary>
|
||||
[DataField]
|
||||
public int? CostOverride;
|
||||
|
||||
protected override bool EvaluateImplementation(EntityTableSelector root,
|
||||
IEntityManager entMan,
|
||||
IPrototypeManager proto,
|
||||
EntityTableContext ctx)
|
||||
{
|
||||
if (!ctx.TryGetData<float>(BudgetContextKey, out var budget))
|
||||
return false;
|
||||
|
||||
int cost;
|
||||
if (CostOverride != null)
|
||||
{
|
||||
cost = CostOverride.Value;
|
||||
}
|
||||
else
|
||||
{
|
||||
if (root is not EntSelector entSelector)
|
||||
return false;
|
||||
|
||||
if (!proto.Index(entSelector.Id).TryGetComponent(out DynamicRuleCostComponent? costComponent, entMan.ComponentFactory))
|
||||
{
|
||||
var log = Logger.GetSawmill("HasBudgetCondition");
|
||||
log.Error($"Rule {entSelector.Id} does not have a DynamicRuleCostComponent.");
|
||||
return false;
|
||||
}
|
||||
|
||||
cost = costComponent.Cost;
|
||||
}
|
||||
|
||||
return budget >= cost;
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user