30 lines
843 B
C#
30 lines
843 B
C#
using Content.Shared.EntityTable.ValueSelector;
|
|
using Robust.Shared.Prototypes;
|
|
|
|
namespace Content.Shared.EntityTable.EntitySelectors;
|
|
|
|
/// <summary>
|
|
/// Gets the spawn for the entity prototype specified at whatever count specified.
|
|
/// </summary>
|
|
public sealed partial class EntSelector : EntityTableSelector
|
|
{
|
|
public const string IdDataFieldTag = "id";
|
|
|
|
[DataField(IdDataFieldTag, required: true)]
|
|
public EntProtoId Id;
|
|
|
|
[DataField]
|
|
public NumberSelector Amount = new ConstantNumberSelector(1);
|
|
|
|
protected override IEnumerable<EntProtoId> GetSpawnsImplementation(System.Random rand,
|
|
IEntityManager entMan,
|
|
IPrototypeManager proto)
|
|
{
|
|
var num = (int) Math.Round(Amount.Get(rand, entMan, proto));
|
|
for (var i = 0; i < num; i++)
|
|
{
|
|
yield return Id;
|
|
}
|
|
}
|
|
}
|