Containment Field Rework (#9312)

Content.Server/Singularity/Components/ContainmentFieldGeneratorComponent.cs

@@ -1,33 +1,107 @@
-using System.Diagnostics.CodeAnalysis;
-using System.Linq;
 using Content.Shared.Physics;
 using Content.Shared.Singularity.Components;
-using Robust.Server.GameObjects;
-using Robust.Shared.Physics;
+using Content.Shared.Tag;
+using Robust.Shared.Prototypes;
+using Robust.Shared.Serialization.TypeSerializers.Implementations.Custom.Prototype;
 
-namespace Content.Server.Singularity.Components
+namespace Content.Server.Singularity.Components;
+[RegisterComponent]
+[ComponentReference(typeof(SharedContainmentFieldGeneratorComponent))]
+public sealed class ContainmentFieldGeneratorComponent : SharedContainmentFieldGeneratorComponent
 {
-    [RegisterComponent]
-    [ComponentReference(typeof(SharedContainmentFieldGeneratorComponent))]
-    public sealed class ContainmentFieldGeneratorComponent : SharedContainmentFieldGeneratorComponent
+    private int _powerBuffer;
+
+    /// <summary>
+    /// Store power with a cap. Decrease over time if not being powered from source.
+    /// </summary>
+    [ViewVariables]
+    [DataField("powerBuffer")]
+    public int PowerBuffer
     {
-        private int _powerBuffer;
-
-        [ViewVariables]
-        public int PowerBuffer
-        {
-            get => _powerBuffer;
-            set => _powerBuffer = Math.Clamp(value, 0, 6);
-        }
-
-        public Tuple<Direction, ContainmentFieldConnection>? Connection1;
-        public Tuple<Direction, ContainmentFieldConnection>? Connection2;
-
-        [ViewVariables]
-        public bool Enabled;
-
-        [ViewVariables]
-        public bool IsConnected;
-
+        get => _powerBuffer;
+        set => _powerBuffer = Math.Clamp(value, 0, 25); //have this decrease over time if not hit by a bolt
     }
+
+    /// <summary>
+    /// The minimum the field generator needs to start generating a connection
+    /// </summary>
+    [ViewVariables(VVAccess.ReadWrite)]
+    [DataField("powerMinimum")]
+    public int PowerMinimum = 6;
+
+    /// <summary>
+    /// How much power should this field generator receive from a collision
+    /// </summary>
+    [ViewVariables(VVAccess.ReadWrite)]
+    [DataField("power")]
+    public int PowerReceived = 3;
+
+    /// <summary>
+    /// How much power should this field generator lose if not powered?
+    /// </summary>
+    [ViewVariables(VVAccess.ReadWrite)]
+    [DataField("powerLoss")]
+    public int PowerLoss = 2;
+
+    /// <summary>
+    /// Used to check if it's received power recently.
+    /// </summary>
+    [ViewVariables]
+    [DataField("accumulator")]
+    public float Accumulator;
+
+    /// <summary>
+    /// How many seconds should the generators wait before losing power?
+    /// </summary>
+    [ViewVariables]
+    [DataField("threshold")]
+    public float Threshold = 10f;
+
+    /// <summary>
+    /// How many tiles should this field check before giving up?
+    /// </summary>
+    [ViewVariables]
+    [DataField("maxLength")]
+    public float MaxLength = 8F;
+
+    /// <summary>
+    /// What collision should power this generator?
+    /// It really shouldn't be anything but an emitter bolt but it's here for fun.
+    /// </summary>
+    [ViewVariables(VVAccess.ReadWrite)]
+    [DataField("idTag", customTypeSerializer: typeof(PrototypeIdSerializer<TagPrototype>))]
+    public string IDTag = "EmitterBolt";
+
+    /// <summary>
+    /// Is the generator toggled on?
+    /// </summary>
+    [ViewVariables]
+    public bool Enabled;
+
+    /// <summary>
+    /// Is this generator connected to fields?
+    /// </summary>
+    [ViewVariables(VVAccess.ReadWrite)]
+    public bool IsConnected;
+
+    /// <summary>
+    /// The masks the raycast should not go through
+    /// </summary>
+    [ViewVariables]
+    [DataField("collisionMask")]
+    public int CollisionMask = (int) (CollisionGroup.MobMask | CollisionGroup.Impassable | CollisionGroup.MachineMask);
+
+    /// <summary>
+    /// A collection of connections that the generator has based on direction.
+    /// Stores a list of fields connected between generators in this direction.
+    /// </summary>
+    [ViewVariables]
+    public Dictionary<Direction, (ContainmentFieldGeneratorComponent, List<EntityUid>)> Connections = new();
+
+    /// <summary>
+    /// What fields should this spawn?
+    /// </summary>
+    [ViewVariables(VVAccess.ReadWrite)]
+    [DataField("createdField", customTypeSerializer: typeof(PrototypeIdSerializer<EntityPrototype>))]
+    public string CreatedField = "ContainmentField";
 }