切换主武器

Assets/Scripts/MainGame/Characters/Player/Inventory/PlayerEquipmentSubmodule.cs

@@ -13,11 +13,24 @@ namespace Cielonos.MainGame.Characters
         public PlayerEquipmentSubmodule(PlayerInventorySubcontroller owner) : base(owner)
         {
             preparedMainWeapons = new List<MainWeaponBase>();
+            foreach (MainWeaponBase mainWeapon in owner.backpack.mainWeapons)
+            {
+                ObtainMainWeapon(mainWeapon);
+            }
+            EquipMainWeapon(preparedMainWeapons[0]);
+        }
+        
+        public void ObtainMainWeapon(MainWeaponBase newWeapon)
+        {
+            if (!preparedMainWeapons.Contains(newWeapon))
+            {
+                preparedMainWeapons.Add(newWeapon);
+                newWeapon.Initialize();
+            }
         }
         
         public void EquipMainWeapon(MainWeaponBase newWeapon)
         {
-            newWeapon.Initialize();
             currentMainWeapon = newWeapon;
             currentMainWeapon.OnEquipped();
             currentMainWeapon.RegisterFullBodyFuncAnims();
@@ -27,8 +40,23 @@ namespace Cielonos.MainGame.Characters
 
         public void RemoveMainWeapon()
         {
+            Debug.Log("Unequipping main weapon: " + currentMainWeapon);
             currentMainWeapon.OnUnequipped();
             currentMainWeapon = null;
         }
+        
+        public void DiscardMainWeapon(MainWeaponBase weaponToDiscard)
+        {
+            if (preparedMainWeapons.Contains(weaponToDiscard))
+            {
+                if (currentMainWeapon == weaponToDiscard)
+                {
+                    RemoveMainWeapon();
+                }
+                
+                preparedMainWeapons.Remove(weaponToDiscard);
+                Object.Destroy(weaponToDiscard);
+            }
+        }
     }
 }