Traits for roundstart species size

Content.Server/Cloning/CloningSystem.cs

@@ -33,6 +33,7 @@
 using Robust.Shared.Physics.Components;
 using Robust.Shared.Prototypes;
 using Robust.Shared.Random;
+using Robust.Shared.Serialization.Manager;
 
 namespace Content.Server.Cloning
 {
@@ -61,6 +62,9 @@ public sealed class CloningSystem : EntitySystem
         [Dependency] private readonly MetaDataSystem _metaSystem = default!;
         [Dependency] private readonly SharedJobSystem _jobs = default!;
 
+        // corvax-next
+        [Dependency] private readonly ISerializationManager _serialization = default!;
+
         public readonly Dictionary<MindComponent, EntityUid> ClonesWaitingForMind = new();
         public const float EasyModeCloningCost = 0.7f;
 
@@ -213,6 +217,17 @@ public bool TryCloning(EntityUid uid, EntityUid bodyToClone, Entity<MindComponen
             var mob = Spawn(speciesPrototype.Prototype, _transformSystem.GetMapCoordinates(uid));
             _humanoidSystem.CloneAppearance(bodyToClone, mob);
 
+            // corvax-next
+            foreach (var comp in EntityManager.GetComponents(bodyToClone))
+            {
+                if (comp is ITransferredByCloning)
+                {
+                    var copy = _serialization.CreateCopy(comp, notNullableOverride: true);
+                    copy.Owner = mob;
+                    EntityManager.AddComponent(mob, copy, overwrite: true);
+                }
+            }
+
             var ev = new CloningEvent(bodyToClone, mob);
             RaiseLocalEvent(bodyToClone, ref ev);
 

Content.Server/Traits/Assorted/SizeAttributeComponent.cs

@@ -0,0 +1,14 @@
+using Content.Shared.Cloning;
+
+namespace Content.Server.Traits.Assorted
+{
+    [RegisterComponent]
+    public sealed partial class SizeAttributeComponent : Component, ITransferredByCloning
+    {
+        [DataField("short")]
+        public bool Short = false;
+
+        [DataField("tall")]
+        public bool Tall = false;
+    }
+}

Content.Server/Traits/Assorted/SizeAttributeSystem.cs

@@ -0,0 +1,67 @@
+using Robust.Server.GameObjects;
+using Robust.Shared.Physics;
+using Robust.Shared.Physics.Collision.Shapes;
+using Robust.Shared.Physics.Systems;
+using System.Numerics;
+
+namespace Content.Server.Traits.Assorted
+{
+    public sealed class SizeAttributeSystem : EntitySystem
+    {
+        [Dependency] private readonly IEntityManager _entityManager = default!;
+        [Dependency] private readonly SharedPhysicsSystem _physics = default!;
+        [Dependency] private readonly AppearanceSystem _appearance = default!;
+        public override void Initialize()
+        {
+            base.Initialize();
+            SubscribeLocalEvent<SizeAttributeComponent, ComponentInit>(OnComponentInit);
+        }
+
+        private void OnComponentInit(EntityUid uid, SizeAttributeComponent component, ComponentInit args)
+        {
+            if (!TryComp<SizeAttributeWhitelistComponent>(uid, out var whitelist))
+                return;
+
+            if (whitelist.Tall && component.Tall)
+            {
+                Scale(uid, component, whitelist.TallScale);
+            }
+            else if (whitelist.Short && component.Short)
+            {
+                Scale(uid, component, whitelist.ShortScale);
+            }
+        }
+
+        private void Scale(EntityUid uid, SizeAttributeComponent component, float scale)
+        {
+            if (scale <= 0f)
+                return;
+
+            _entityManager.EnsureComponent<ScaleVisualsComponent>(uid);
+
+            var appearanceComponent = _entityManager.EnsureComponent<AppearanceComponent>(uid);
+            if (!_appearance.TryGetData<Vector2>(uid, ScaleVisuals.Scale, out var oldScale, appearanceComponent))
+                oldScale = Vector2.One;
+
+            _appearance.SetData(uid, ScaleVisuals.Scale, oldScale * scale, appearanceComponent);
+
+            if (_entityManager.TryGetComponent(uid, out FixturesComponent? manager))
+            {
+                foreach (var (id, fixture) in manager.Fixtures)
+                {
+                    if (!fixture.Hard || fixture.Density <= 1f)
+                        continue; // This will skip the flammable fixture and any other fixture that is not supposed to contribute to mass
+
+                    switch (fixture.Shape)
+                    {
+                        case PhysShapeCircle circle:
+                            _physics.SetPositionRadius(uid, id, fixture, circle, circle.Position * scale, circle.Radius * scale, manager);
+                            break;
+                        default:
+                            throw new NotImplementedException();
+                    }
+                }
+            }
+        }
+    }
+}

Content.Server/Traits/Assorted/SizeAttributeWhitelistComponent.cs

@@ -0,0 +1,19 @@
+namespace Content.Server.Traits.Assorted
+{
+    [RegisterComponent]
+    public sealed partial class SizeAttributeWhitelistComponent : Component
+    {
+        [DataField("short")]
+        public bool Short = false;
+
+        [DataField("shortscale")]
+        public float ShortScale = 0f;
+
+        [DataField("tall")]
+        public bool Tall = false;
+
+        [DataField("tallscale")]
+        public float TallScale = 0f;
+
+    }
+}

Content.Shared/Cloning/ITransferredByCloning.cs

@@ -0,0 +1,9 @@
+namespace Content.Shared.Cloning
+{
+    /// <summary>
+    ///   Indicates that this Component should be transferred to the new entity when the entity is cloned (for example, using a cloner)
+    /// </summary>
+    public interface ITransferredByCloning
+    {
+    }
+}

Resources/Prototypes/Entities/Mobs/Species/base.yml

@@ -211,6 +211,11 @@
     - CanPilot
     - FootstepSound
     - DoorBumpOpener
+  - type: SizeAttributeWhitelist # corvax-next
+    tall: true
+    tallscale: 1.1
+    short: true
+    shortscale: 0.90
 
 - type: entity
   save: false

Resources/Prototypes/Entities/Mobs/Species/dwarf.yml

@@ -63,6 +63,9 @@
           32:
             sprite: Mobs/Species/Human/displacement.rsi
             state: jumpsuit-female
+  - type: SizeAttributeWhitelist # corvax-next
+    tall: false
+    short: false
 
 - type: entity
   parent: BaseSpeciesDummy

Resources/Prototypes/Traits/categories.yml

@@ -10,3 +10,8 @@
 - type: traitCategory
   id: Quirks
   name: trait-category-quirks
+
+- type: traitCategory # corvax-next
+  id: SizeTraits
+  name: Рост
+  maxTraitPoints: 1

Resources/Prototypes/Traits/sizeattribute.yml

@@ -0,0 +1,31 @@
+- type: trait # corvax-next
+  id: Tall
+  name: Высокий
+  description: У вас проблемы с гигантизмом.
+  category: SizeTraits
+  cost: 1
+  whitelist:
+    components:
+      - SizeAttributeWhitelist
+  blacklist:
+    components:
+      - SizeAttribute
+  components:
+    - type: SizeAttribute
+      tall: true
+
+- type: trait
+  id: Short
+  name: Низкий
+  description: Вы коротковаты ростом.
+  category: SizeTraits
+  cost: 1
+  whitelist:
+    components:
+      - SizeAttributeWhitelist
+  blacklist:
+    components:
+      - SizeAttribute
+  components:
+    - type: SizeAttribute
+      short: true