-using System.Linq;
-using Content.Server.GameTicking.Rules.Components;
+using System.Linq;
+using Content.Shared.Dataset;
using Content.Server.Ghost.Roles.Components;
using Content.Server.StationEvents.Components;
using Content.Shared.GameTicking.Components;
+using Content.Shared.Random.Helpers;
+using Robust.Shared.Prototypes;
+using Robust.Shared.Random;
namespace Content.Server.StationEvents.Events;
public sealed class RandomSentienceRule : StationEventSystem<RandomSentienceRuleComponent>
{
+ [Dependency] private readonly IPrototypeManager _prototype = default!;
+ [Dependency] private readonly IRobustRandom _random = default!;
protected override void Started(EntityUid uid, RandomSentienceRuleComponent component, GameRuleComponent gameRule, GameRuleStartedEvent args)
{
- HashSet<EntityUid> stationsToNotify = new();
+ if (!TryGetRandomStation(out var station))
+ return;
var targetList = new List<Entity<SentienceTargetComponent>>();
- var query = EntityQueryEnumerator<SentienceTargetComponent>();
- while (query.MoveNext(out var targetUid, out var target))
+ var query = EntityQueryEnumerator<SentienceTargetComponent, TransformComponent>();
+ while (query.MoveNext(out var targetUid, out var target, out var xform))
{
+ if (StationSystem.GetOwningStation(targetUid, xform) != station)
+ continue;
+
targetList.Add((targetUid, target));
}
- RobustRandom.Shuffle(targetList);
+ var toMakeSentient = _random.Next(component.MinSentiences, component.MaxSentiences);
- var toMakeSentient = RobustRandom.Next(2, 5);
var groups = new HashSet<string>();
- foreach (var target in targetList)
+ for (var i = 0; i < toMakeSentient && targetList.Count > 0; i++)
{
- if (toMakeSentient-- == 0)
- break;
+ // weighted random to pick a sentience target
+ var totalWeight = targetList.Sum(x => x.Comp.Weight);
+ // This initial target should never be picked.
+ // It's just so that target doesn't need to be nullable and as a safety fallback for id floating point errors ever mess up the comparison in the foreach.
+ var target = targetList[0];
+ var chosenWeight = _random.NextFloat(totalWeight);
+ var currentWeight = 0.0;
+ foreach (var potentialTarget in targetList)
+ {
+ currentWeight += potentialTarget.Comp.Weight;
+ if (currentWeight > chosenWeight)
+ {
+ target = potentialTarget;
+ break;
+ }
+ }
+ targetList.Remove(target);
RemComp<SentienceTargetComponent>(target);
var ghostRole = EnsureComp<GhostRoleComponent>(target);
var kind2 = groupList.Count > 1 ? groupList[1] : "???";
var kind3 = groupList.Count > 2 ? groupList[2] : "???";
- foreach (var target in targetList)
- {
- var station = StationSystem.GetOwningStation(target);
- if(station == null)
- continue;
- stationsToNotify.Add((EntityUid) station);
- }
- foreach (var station in stationsToNotify)
- {
- ChatSystem.DispatchStationAnnouncement(
- station,
- Loc.GetString("station-event-random-sentience-announcement",
- ("kind1", kind1), ("kind2", kind2), ("kind3", kind3), ("amount", groupList.Count),
- ("data", Loc.GetString($"random-sentience-event-data-{RobustRandom.Next(1, 6)}")),
- ("strength", Loc.GetString($"random-sentience-event-strength-{RobustRandom.Next(1, 8)}"))),
- playDefaultSound: false,
- colorOverride: Color.Gold
- );
- }
+ ChatSystem.DispatchStationAnnouncement(
+ station.Value,
+ Loc.GetString("station-event-random-sentience-announcement",
+ ("kind1", kind1), ("kind2", kind2), ("kind3", kind3), ("amount", groupList.Count),
+ ("data", _random.Pick(_prototype.Index<LocalizedDatasetPrototype>("RandomSentienceEventData"))),
+ ("strength", _random.Pick(_prototype.Index<LocalizedDatasetPrototype>("RandomSentienceEventStrength")))
+ ),
+ playDefaultSound: false,
+ colorOverride: Color.Gold
+ );
}
}
projects / space-station-14.git / commitdiff