Auditions: random cast generation

Content.Server/_ES/Auditions/AuditionsSystem.cs

@@ -0,0 +1,158 @@
+using System.Diagnostics;
+using System.Linq;
+using Content.Server.Administration;
+using Content.Shared._ES.Auditions;
+using Content.Shared.Administration;
+using Robust.Shared.Random;
+using Robust.Shared.Toolshed;
+
+namespace Content.Server._ES.Auditions;
+
+/// <summary>
+/// This handles the server-side of auditioning!
+/// </summary>
+public sealed class AuditionsSystem : SharedAuditionsSystem
+{
+    [Dependency] private readonly IRobustRandom _random = default!;
+
+    /// <summary>
+    /// Hires a cast, and integrates relationships between all of the characters.
+    /// </summary>
+    public void GenerateCast(
+        int captainCount = 26,
+        int crewCount = 10,
+        ProducerComponent? producer = null
+    )
+    {
+        if (!TryGetProducer(ref producer))
+            throw new Exception("Could not get ProducerComponent!");
+
+        var preEvt = new PreCastGenerateEvent(producer);
+        RaiseLocalEvent(ref preEvt);
+
+        var captains = GenerateEmptySocialGroup();
+        captains.Comp.RelativeContext = producer.CaptainContext;
+
+        var newCharacters = new List<EntityUid>();
+
+        for (var i = 0; i < captainCount; i++)
+        {
+            var newCrew = GenerateRandomCrew(crewCount);
+            captains.Comp.Members.Add(newCrew.Comp.Members[0]);
+            newCharacters.AddRange(newCrew.Comp.Members);
+        }
+
+        var psgEvt = new PostShipGenerateEvent(producer);
+        RaiseLocalEvent(ref psgEvt);
+
+        foreach (var group in producer.SocialGroups)
+        {
+            var comp = EnsureComp<SocialGroupComponent>(group);
+            if (comp.Integrated)
+                continue;
+
+            var ent = (group, comp);
+
+            var pre = new SocialGroupPreIntegrationEvent(ent);
+            RaiseLocalEvent(ref pre);
+
+            IntegrateRelationshipGroup(comp.RelativeContext, comp.Members);
+            comp.Integrated = true;
+
+            var post = new SocialGroupPostIntegrationEvent(ent);
+            RaiseLocalEvent(ref post);
+        }
+
+        IntegrateRelationshipGroup(producer.IntercrewContext, newCharacters);
+        producer.Characters.AddRange(newCharacters);
+
+        var postEvt = new PostCastGenerateEvent(producer);
+        RaiseLocalEvent(ref postEvt);
+    }
+
+    public void GenerateCast(
+        int captainCount = 26,
+        int minimumCrew = 5,
+        int maximumCrew = 12,
+        ProducerComponent? producer = null
+    )
+    {
+        GenerateCast(captainCount, _random.Next(minimumCrew, maximumCrew), producer);
+    }
+}
+
+[ToolshedCommand, AdminCommand(AdminFlags.Round)]
+public sealed class CastCommand : ToolshedCommand
+{
+    private AuditionsSystem? _auditions;
+
+    [CommandImplementation("generate")]
+    public IEnumerable<string> Generate(int captainCount = 26, int crewSize = 10)
+    {
+        _auditions ??= GetSys<AuditionsSystem>();
+
+        var stopwatch = new Stopwatch();
+        stopwatch.Start();
+
+        _auditions.GenerateCast(captainCount, crewSize, null);
+
+        yield return $"Generated cast in {stopwatch.Elapsed.TotalMilliseconds} ms.";
+    }
+
+    [CommandImplementation("view")]
+    public IEnumerable<string> View([PipedArgument] EntityUid castMember)
+    {
+        _auditions ??= GetSys<AuditionsSystem>();
+        if (!EntityManager.TryGetComponent<CharacterComponent>(castMember, out var character))
+        {
+            yield return "Invalid cast member object (did not have CharacterComponent)!";
+        }
+        else
+        {
+            yield return $"{character.Name}, {character.Age} years old ({character.DateOfBirth.ToShortDateString()})\nBackground: {character.Background}\nRelationships\n";
+            Dictionary<string, List<string>> relationships = new();
+            foreach (var relationship in character.Relationships)
+            {
+                if (relationships.ContainsKey(relationship.Value))
+                    relationships[relationship.Value].Add(relationship.Key);
+                else
+                    relationships[relationship.Value] = [relationship.Key];
+            }
+
+            foreach (var relationship in relationships)
+            {
+                yield return $"{relationship.Key} ({relationship.Value.Count}): {string.Join(", ", relationship.Value.ToArray())}";
+            }
+        }
+    }
+}
+
+/// <summary>
+/// Fires prior to this social group's relationships being integrated.
+/// </summary>
+[ByRefEvent]
+public readonly record struct SocialGroupPreIntegrationEvent(Entity<SocialGroupComponent> Group);
+
+/// <summary>
+/// Fires after this social group's relationships have been integrated.
+/// </summary>
+[ByRefEvent]
+public readonly record struct SocialGroupPostIntegrationEvent(Entity<SocialGroupComponent> Group);
+
+/// <summary>
+/// Fires prior to any generation events (captain group, crew groups, etc).
+/// </summary>
+[ByRefEvent]
+public readonly record struct PreCastGenerateEvent(ProducerComponent Producer);
+
+/// <summary>
+/// Fires after the primary generation events (captain group, crew group, etc) but before integration of relationships.
+/// </summary>
+[ByRefEvent]
+public readonly record struct PostShipGenerateEvent(ProducerComponent Producer);
+
+/// <summary>
+/// Fires after all relationships have been integrated.
+/// </summary>
+[ByRefEvent]
+public readonly record struct PostCastGenerateEvent(ProducerComponent Producer);

Content.Shared/_ES/Auditions/BackgroundPrototype.cs

@@ -0,0 +1,26 @@
+using Robust.Shared.Prototypes;
+
+namespace Content.Shared._ES.Auditions;
+
+/// <summary>
+/// This is a prototype for marking backgrounds
+/// </summary>
+[Prototype]
+public sealed partial class BackgroundPrototype : IPrototype
+{
+    /// <inheritdoc/>
+    [IdDataField]
+    public string ID { get; } = default!;
+
+    /// <summary>
+    /// Name of this relationship.
+    /// </summary>
+    [DataField, ViewVariables]
+    public LocId Name;
+
+    /// <summary>
+    /// Description of this relationship.
+    /// </summary>
+    [DataField, ViewVariables]
+    public LocId Description;
+}

Content.Shared/_ES/Auditions/CharacterComponent.cs

@@ -0,0 +1,38 @@
+using Content.Shared.Humanoid;
+using Robust.Shared.Enums;
+using Robust.Shared.GameStates;
+using Robust.Shared.Prototypes;
+using Robust.Shared.Serialization;
+
+namespace Content.Shared._ES.Auditions;
+
+/// <summary>
+/// This is used for marking the character of components.
+/// </summary>
+[RegisterComponent, NetworkedComponent, AutoGenerateComponentState]
+public sealed partial class CharacterComponent : Component
+{
+    [DataField, AutoNetworkedField]
+    public string Name = "Rain Miskovitch";
+
+    [DataField, AutoNetworkedField]
+    public int Age = 25;
+
+    [DataField, AutoNetworkedField]
+    public Gender Gender = Gender.Neuter;
+
+    [DataField, AutoNetworkedField]
+    public HumanoidCharacterAppearance Appearance = default!;
+
+    [DataField, AutoNetworkedField]
+    public DateTime DateOfBirth = new(2025, 1, 1);
+
+    [DataField, AutoNetworkedField]
+    public Dictionary<string, ProtoId<RelationshipPrototype>> Relationships = new ();
+
+    [DataField, AutoNetworkedField]
+    public ProtoId<BackgroundPrototype> Background = "Colonist";
+
+    [DataField, AutoNetworkedField]
+    public List<EntityUid> Memories = new ();
+}

Content.Shared/_ES/Auditions/ProducerComponent.cs

@@ -0,0 +1,84 @@
+using Content.Shared.Random;
+using Robust.Shared.Prototypes;
+using Robust.Shared.Serialization;
+
+namespace Content.Shared._ES.Auditions;
+
+/// <summary>
+/// This is the cast component placed onto the producer entity.
+/// </summary>
+[RegisterComponent]
+public sealed partial class ProducerComponent : Component
+{
+    /// <summary>
+    /// All the characters in the cast.
+    /// </summary>
+    [DataField]
+    public List<EntityUid> Characters = new ();
+
+    /// <summary>
+    /// List of all active social groups.
+    /// </summary>
+    [DataField]
+    public List<EntityUid> SocialGroups = new ();
+
+    [DataField]
+    public RelationshipContext CrewContext = new ("RelationshipPoolCrew", 0.75f);
+
+    [DataField]
+    public RelationshipContext CaptainContext = new ("RelationshipPoolCaptains", 0.5f);
+
+    [DataField]
+    public RelationshipContext IntercrewContext = new ("RelationshipPoolIntercrew", 0.1f);
+}
+
+/// <summary>
+/// Configuration for integrating relationships.
+/// </summary>
+[Serializable, NetSerializable, DataDefinition]
+public partial struct RelationshipContext
+{
+    /// <summary>
+    /// List of possible relationships.
+    /// </summary>
+    [DataField]
+    public ProtoId<WeightedRandomPrototype> PoolPrototype;
+
+    /// <summary>
+    /// How likely is a relationship to spark in this context?
+    /// </summary>
+    [DataField]
+    public float RelationshipProbability = 0.5f;
+
+    /// <summary>
+    /// How likely is a relationship to be mutual (both sides have the same relationship)?
+    /// </summary>
+    [DataField]
+    public float UnificationProbability = 1f;
+
+    /// <summary>
+    /// If the relationship isnt mutual, what are other possible relationships to give? If null, no relationship is assigned.
+    /// </summary>
+    [DataField]
+    public ProtoId<WeightedRandomPrototype>? SeperatePoolPrototype;
+
+    public RelationshipContext()
+    {
+    }
+
+    public RelationshipContext(string prototype, float probability)
+    {
+        RelationshipProbability = probability;
+        PoolPrototype = prototype;
+        UnificationProbability = 1f;
+        SeperatePoolPrototype = null;
+    }
+
+    public RelationshipContext(string prototype, float probability, float unificationProbability, string? seperatePrototype)
+    {
+        RelationshipProbability = probability;
+        PoolPrototype = prototype;
+        UnificationProbability = unificationProbability;
+        SeperatePoolPrototype = seperatePrototype;
+    }
+}

Content.Shared/_ES/Auditions/RelationshipPrototype.cs

@@ -0,0 +1,32 @@
+using Robust.Shared.Prototypes;
+
+namespace Content.Shared._ES.Auditions;
+
+/// <summary>
+/// This is a prototype for marking relationships
+/// </summary>
+[Prototype]
+public sealed partial class RelationshipPrototype : IPrototype
+{
+    /// <inheritdoc/>
+    [IdDataField]
+    public string ID { get; } = default!;
+
+    /// <summary>
+    /// Name of this relationship.
+    /// </summary>
+    [DataField, ViewVariables]
+    public LocId Name;
+
+    /// <summary>
+    /// Color of this relationship.
+    /// </summary>
+    [DataField, ViewVariables]
+    public Color Color;
+
+    /// <summary>
+    /// Whether or not this relationship must be mutual. Family members, ex-lovers, etc, all fall under this.
+    /// </summary>
+    [DataField, ViewVariables]
+    public bool ForceMutual;
+}