Кастомизация СИИ

Content.Client/ADT/UserInterface/Controls/ScaledAnimatedTextureRect.cs

@@ -0,0 +1,74 @@
+using System.Numerics;
+using Robust.Client.Graphics;
+using Robust.Client.UserInterface;
+using Robust.Client.UserInterface.Controls;
+using Robust.Client.Utility;
+using Robust.Shared.Graphics.RSI;
+using Robust.Shared.Timing;
+using Robust.Shared.Utility;
+
+namespace Content.Client.UserInterface.Controls
+{
+    /// <summary>
+    ///     Copy-paste of engine <see cref="AnimatedTextureRect"/> that can be scaled.
+    /// </summary>
+    public sealed class ScaledAnimatedTextureRect : Control
+    {
+        private IRsiStateLike? _state;
+        private int _curFrame;
+        private float _curFrameTime;
+
+        /// <summary>
+        ///     Internal TextureRect used to do actual drawing of the texture.
+        ///     You can use this property to change shaders or styling or such.
+        /// </summary>
+        public TextureRect DisplayRect { get; }
+
+        public RsiDirection RsiDirection { get; } = RsiDirection.South;
+
+        public ScaledAnimatedTextureRect()
+        {
+            IoCManager.InjectDependencies(this);
+
+            DisplayRect = new TextureRect()
+            {
+                TextureScale = new(1, 1)
+            };
+            AddChild(DisplayRect);
+        }
+
+        public void SetFromSpriteSpecifier(SpriteSpecifier specifier, Vector2 scale)
+        {
+            _curFrame = 0;
+            _state = specifier.RsiStateLike();
+            _curFrameTime = _state.GetDelay(0);
+            DisplayRect.Texture = _state.GetFrame(RsiDirection, 0);
+            DisplayRect.TextureScale = scale;
+        }
+
+        public void SetScale(Vector2 scale)
+        {
+            DisplayRect.TextureScale = scale;
+        }
+
+        protected override void FrameUpdate(FrameEventArgs args)
+        {
+            if (!VisibleInTree || _state == null || !_state.IsAnimated)
+                return;
+
+            var oldFrame = _curFrame;
+
+            _curFrameTime -= args.DeltaSeconds;
+            while (_curFrameTime < _state.GetDelay(_curFrame))
+            {
+                _curFrame = (_curFrame + 1) % _state.AnimationFrameCount;
+                _curFrameTime += _state.GetDelay(_curFrame);
+            }
+
+            if (_curFrame != oldFrame)
+            {
+                DisplayRect.Texture = _state.GetFrame(RsiDirection, _curFrame);
+            }
+        }
+    }
+}

Content.Client/Lobby/LobbyUIController.cs

@@ -15,7 +15,9 @@
 using Content.Shared.Preferences;
 using Content.Shared.Preferences.Loadouts;
 using Content.Shared.Roles;
+using Content.Shared.Silicons.StationAi;
 using Content.Shared.Traits;
+using Robust.Client.GameObjects;
 using Robust.Client.Player;
 using Robust.Client.ResourceManagement;
 using Robust.Client.State;
@@ -40,6 +42,8 @@ public sealed partial class LobbyUIController : UIController, IOnStateEntered<Lo
     [Dependency] private readonly IStateManager _stateManager = default!;
     [Dependency] private readonly JobRequirementsManager _requirements = default!;
     [Dependency] private readonly MarkingManager _markings = default!;
+    [Dependency] private readonly IDynamicTypeFactory _factory = default!;
+    [Dependency] private readonly IEntityManager _entMan = default!;
     [UISystemDependency] private readonly HumanoidAppearanceSystem _humanoid = default!;
     [UISystemDependency] private readonly ClientInventorySystem _inventory = default!;
     [UISystemDependency] private readonly StationSpawningSystem _spawn = default!;
@@ -281,7 +285,8 @@ private void OpenSavePanel()
             _prototypeManager,
             _resourceCache,
             _requirements,
-            _markings);
+            _markings,
+            _factory);  // ADT SAI Custom
 
         _profileEditor.OnOpenGuidebook += _guide.OpenHelp;
 
@@ -375,6 +380,8 @@ public void GiveDummyLoadout(EntityUid uid, RoleLoadout? roleLoadout)
                 _spawn.EquipStartingGear(uid, loadoutProto);
             }
         }
+
+        _spawn.ApplyLoadoutExtras(uid, roleLoadout);    // ADT SAI Custom
     }
 
     /// <summary>
@@ -473,6 +480,17 @@ public EntityUid LoadProfileEntity(HumanoidCharacterProfile? humanoid, JobProtot
         {
             // Special type like borg or AI, do not spawn a human just spawn the entity.
             dummyEnt = EntityManager.SpawnEntity(previewEntity, MapCoordinates.Nullspace);
+
+            // ADT SAI Custom start
+            // Applying loadout extras to dummy
+            if (_prototypeManager.HasIndex<RoleLoadoutPrototype>(LoadoutSystem.GetJobPrototype(job?.ID)))
+            {
+                var loadout = humanoid?.GetLoadoutOrDefault(LoadoutSystem.GetJobPrototype(job?.ID), _playerManager.LocalSession, humanoid.Species, EntityManager, _prototypeManager);
+                if (loadout != null)
+                    _spawn.ApplyLoadoutExtras(dummyEnt, loadout);
+            }
+            // ADT SAI Custom end
+
             return dummyEnt;
         }
         else if (humanoid is not null)

Content.Client/Lobby/UI/HumanoidProfileEditor.xaml.cs

@@ -4,13 +4,15 @@
 using Content.Client.ADT.Lobby.UI;
 using Content.Client.Corvax.Sponsors;
 using Content.Client.Guidebook;
+using System.Reflection;
 using Content.Client.Humanoid;
 using Content.Client.Lobby.UI.Loadouts;
 using Content.Client.Lobby.UI.Roles;
 using Content.Client.Message;
 using Content.Client.Players.PlayTimeTracking;
 using Content.Client.Sprite;
 using Content.Client.Stylesheets;
+using Content.Client.UserInterface.Controls;
 using Content.Client.UserInterface.Systems.Guidebook;
 using Content.Shared.ADT.CCVar;
 using Content.Client.Corvax.Sponsors;
@@ -22,6 +24,7 @@
 using Content.Shared.Humanoid;
 using Content.Shared.Humanoid.Markings;
 using Content.Shared.Humanoid.Prototypes;
+using Content.Shared.Mobs;
 using Content.Shared.Preferences;
 using Content.Shared.Preferences.Loadouts;
 using Content.Shared.Roles;
@@ -37,6 +40,7 @@
 using Robust.Shared.ContentPack;
 using Robust.Shared.Enums;
 using Robust.Shared.Prototypes;
+using Robust.Shared.Toolshed.Errors;
 using Robust.Shared.Utility;
 using Direction = Robust.Shared.Maths.Direction;
 
@@ -56,13 +60,14 @@ public sealed partial class HumanoidProfileEditor : BoxContainer
         private readonly MarkingManager _markingManager;
         private readonly JobRequirementsManager _requirements;
         private readonly LobbyUIController _controller;
+        private readonly IDynamicTypeFactory _factory;  // ADT SAI Custom
         [Dependency] private readonly DocumentParsingManager _parsingMan = default!;
 
         private FlavorText.FlavorText? _flavorText;
         private TextEdit? _flavorTextEdit;
 
         // One at a time.
-        private LoadoutWindow? _loadoutWindow;
+        private BaseLoadoutWindow? _loadoutWindow;  // ADT SAI Custom
 
         private bool _exporting;
         private bool _imaging;
@@ -125,7 +130,8 @@ public HumanoidProfileEditor(
             IPrototypeManager prototypeManager,
             IResourceManager resManager,
             JobRequirementsManager requirements,
-            MarkingManager markings)
+            MarkingManager markings,
+            IDynamicTypeFactory factory)    // ADT SAI Custom
         {
             RobustXamlLoader.Load(this);
             _sawmill = logManager.GetSawmill("profile.editor");
@@ -138,6 +144,7 @@ public HumanoidProfileEditor(
             _preferencesManager = preferencesManager;
             _resManager = resManager;
             _requirements = requirements;
+            _factory = factory; // ADT SAI Custom
             _controller = UserInterfaceManager.GetUIController<LobbyUIController>();
 
             ImportButton.OnPressed += args =>
@@ -1103,6 +1110,11 @@ public void RefreshJobs()
                     var collection = IoCManager.Instance!;
                     var protoManager = collection.Resolve<IPrototypeManager>();
 
+                    // ADT SAI Custom start
+                    if (job.LoadoutButtonText != null)
+                        loadoutWindowBtn.Text = Loc.GetString(job.LoadoutButtonText);
+                    // ADT SAI Custom end
+
                     // If no loadout found then disabled button
                     if (!protoManager.TryIndex<RoleLoadoutPrototype>(LoadoutSystem.GetJobPrototype(job.ID), out var roleLoadoutProto))
                     {
@@ -1118,14 +1130,18 @@ public void RefreshJobs()
                             // Clone so we don't modify the underlying loadout.
                             Profile?.Loadouts.TryGetValue(LoadoutSystem.GetJobPrototype(job.ID), out loadout);
                             loadout = loadout?.Clone();
-
                             if (loadout == null)
                             {
                                 loadout = new RoleLoadout(roleLoadoutProto.ID);
                                 loadout.SetDefault(Profile, _playerManager.LocalSession, _prototypeManager);
                             }
 
-                            OpenLoadout(job, loadout, roleLoadoutProto);
+                            // ADT SAI Custom start
+                            if (job.LoadoutOverride != null)
+                                OpenLoadoutOverride(job, loadout, job.LoadoutOverride);
+                            else
+                                OpenLoadout(job, loadout, roleLoadoutProto);
+                            // ADT SAI Custom end
                         };
                     }
 
@@ -1156,29 +1172,34 @@ private void OpenLoadout(JobPrototype? jobProto, RoleLoadout roleLoadout, RoleLo
                 Title = jobProto?.ID + "-loadout",
             };
 
+            // ADT SAI Custom start
+            if (_loadoutWindow is not LoadoutWindow window)
+                return;
+            // ADT SAI Custom end
+
             // Refresh the buttons etc.
-            _loadoutWindow.RefreshLoadouts(roleLoadout, session, collection);
+            window.RefreshLoadouts(roleLoadout, session, collection);   // ADT SAI Custom tweaked
             _loadoutWindow.OpenCenteredLeft();
 
-            _loadoutWindow.OnNameChanged += name =>
+            window.OnNameChanged += name => // ADT SAI Custom tweaked
             {
                 roleLoadout.EntityName = name;
                 Profile = Profile.WithLoadout(roleLoadout);
                 SetDirty();
             };
 
-            _loadoutWindow.OnLoadoutPressed += (loadoutGroup, loadoutProto) =>
+            window.OnLoadoutPressed += (loadoutGroup, loadoutProto) =>  // ADT SAI Custom tweaked
             {
                 roleLoadout.AddLoadout(loadoutGroup, loadoutProto, _prototypeManager);
-                _loadoutWindow.RefreshLoadouts(roleLoadout, session, collection);
+                window.RefreshLoadouts(roleLoadout, session, collection);   // ADT SAI Custom tweaked
                 Profile = Profile?.WithLoadout(roleLoadout);
                 ReloadPreview();
             };
 
-            _loadoutWindow.OnLoadoutUnpressed += (loadoutGroup, loadoutProto) =>
+            window.OnLoadoutUnpressed += (loadoutGroup, loadoutProto) =>    // ADT SAI Custom tweaked
             {
                 roleLoadout.RemoveLoadout(loadoutGroup, loadoutProto, _prototypeManager);
-                _loadoutWindow.RefreshLoadouts(roleLoadout, session, collection);
+                window.RefreshLoadouts(roleLoadout, session, collection);   // ADT SAI Custom tweaked
                 Profile = Profile?.WithLoadout(roleLoadout);
                 ReloadPreview();
             };
@@ -1198,6 +1219,48 @@ private void OpenLoadout(JobPrototype? jobProto, RoleLoadout roleLoadout, RoleLo
             UpdateJobPriorities();
         }
 
+        // ADT SAI Custom start
+        private void OpenLoadoutOverride(JobPrototype? jobProto, RoleLoadout roleLoadout, string windowName)
+        {
+            _loadoutWindow?.Dispose();
+            _loadoutWindow = null;
+
+            _loadoutWindow = _factory.CreateInstance(Type.GetType($"Content.Client.Lobby.UI.Loadouts.{windowName}")!) as BaseLoadoutWindow;
+            if (_loadoutWindow is not ILoadoutOverride loadoutWindow)
+                return;
+
+            loadoutWindow.Refresh(Profile, roleLoadout, _prototypeManager);
+            loadoutWindow.OnValueChanged += item =>
+            {
+                if (roleLoadout.ExtraData.ContainsKey(item.Key))
+                    roleLoadout.ExtraData[item.Key] = item.Value;
+                else
+                    roleLoadout.ExtraData.Add(item.Key, item.Value);
+
+                Profile = Profile?.WithLoadout(roleLoadout);
+                loadoutWindow.Refresh(Profile, roleLoadout, _prototypeManager);
+                ReloadPreview();
+                SetDirty();
+            };
+
+            _loadoutWindow.OpenCenteredLeft();
+
+            JobOverride = jobProto;
+            ReloadPreview();
+
+            _loadoutWindow.OnClose += () =>
+            {
+                JobOverride = null;
+                ReloadPreview();
+            };
+
+            if (Profile is null)
+                return;
+
+            UpdateJobPriorities();
+        }
+        // ADT SAI Custom end
+
         private void OnFlavorTextChange(string content)
         {
             if (Profile is null)

Content.Client/Lobby/UI/Loadouts/BaseLoadoutWindow.cs

@@ -0,0 +1,7 @@
+using Content.Client.UserInterface.Controls;
+
+namespace Content.Client.Lobby.UI.Loadouts;
+
+// ADT File
+[Virtual]
+public partial class BaseLoadoutWindow : FancyWindow;

Content.Client/Lobby/UI/Loadouts/ILoadoutOverride.cs

@@ -0,0 +1,14 @@
+using Content.Shared.Preferences;
+using Content.Shared.Preferences.Loadouts;
+using Robust.Shared.Prototypes;
+
+namespace Content.Client.Lobby.UI.Loadouts;
+
+// ADT File
+public interface ILoadoutOverride
+{
+    public Action<KeyValuePair<string, string>>? OnValueChanged { get; set; }
+    HumanoidCharacterProfile? Profile { get; set; }
+
+    void Refresh(HumanoidCharacterProfile? profile, RoleLoadout loadout, IPrototypeManager protoMan);
+}

Content.Client/Lobby/UI/Loadouts/LoadoutWindow.xaml

@@ -1,6 +1,6 @@
-<controls:FancyWindow xmlns="https://spacestation14.io"
+<loadout:BaseLoadoutWindow xmlns="https://spacestation14.io"
          xmlns:x="http://schemas.microsoft.com/winfx/2006/xaml"
-         xmlns:controls="clr-namespace:Content.Client.UserInterface.Controls"
+         xmlns:loadout="clr-namespace:Content.Client.Lobby.UI.Loadouts"
          xmlns:graphics="clr-namespace:Robust.Client.Graphics;assembly=Robust.Client"
          SetSize="800 800"
          MinSize="800 128">
@@ -19,4 +19,4 @@
                               HorizontalExpand="True">
         </VerticalTabContainer>
     </BoxContainer>
-</controls:FancyWindow>
+</loadout:BaseLoadoutWindow>

Content.Client/Lobby/UI/Loadouts/LoadoutWindow.xaml.cs

@@ -14,7 +14,7 @@
 namespace Content.Client.Lobby.UI.Loadouts;
 
 [GenerateTypedNameReferences]
-public sealed partial class LoadoutWindow : FancyWindow
+public sealed partial class LoadoutWindow : BaseLoadoutWindow   // ADT SAI Custom tweaked
 {
     public event Action<string>? OnNameChanged;
     public event Action<ProtoId<LoadoutGroupPrototype>, ProtoId<LoadoutPrototype>>? OnLoadoutPressed;