116 lines
2.9 KiB
C#
116 lines
2.9 KiB
C#
using UnityEngine;
|
|
using UnityEngine.UI;
|
|
|
|
public class WeaponSwitcher : MonoBehaviour
|
|
{
|
|
public Weapon[] weapons;
|
|
public int currentIndex;
|
|
public Text weaponNameText;
|
|
public WeaponHUD hud;
|
|
|
|
private void Start()
|
|
{
|
|
if (weapons == null || weapons.Length == 0)
|
|
return;
|
|
|
|
for (int i = 0; i < weapons.Length; i++)
|
|
{
|
|
if (weapons[i] == null)
|
|
continue;
|
|
|
|
weapons[i].Initialize();
|
|
weapons[i].gameObject.SetActive(false);
|
|
}
|
|
|
|
currentIndex = Mathf.Clamp(currentIndex, 0, weapons.Length - 1);
|
|
SwitchTo(currentIndex);
|
|
}
|
|
|
|
private void Update()
|
|
{
|
|
if (weapons == null || weapons.Length == 0)
|
|
return;
|
|
|
|
if (Input.GetKeyDown(KeyCode.Alpha1))
|
|
SwitchTo(0);
|
|
else if (Input.GetKeyDown(KeyCode.Alpha2))
|
|
SwitchTo(1);
|
|
else if (Input.GetKeyDown(KeyCode.Alpha3))
|
|
SwitchTo(2);
|
|
|
|
float scroll = Input.GetAxis("Mouse ScrollWheel");
|
|
if (scroll > 0.01f)
|
|
NextWeapon();
|
|
else if (scroll < -0.01f)
|
|
PreviousWeapon();
|
|
}
|
|
|
|
public void SwitchTo(int index)
|
|
{
|
|
if (weapons == null || weapons.Length == 0 || index < 0 || index >= weapons.Length)
|
|
return;
|
|
|
|
// Reloading locks weapon switching to keep ammo state predictable.
|
|
Weapon currentWeapon = GetCurrentWeapon();
|
|
if (currentWeapon != null && currentWeapon.IsReloading())
|
|
{
|
|
if (hud != null)
|
|
hud.ShowMessage("Cannot switch while reloading.");
|
|
return;
|
|
}
|
|
|
|
for (int i = 0; i < weapons.Length; i++)
|
|
{
|
|
if (weapons[i] == null)
|
|
continue;
|
|
|
|
bool active = i == index;
|
|
if (!active && weapons[i].gameObject.activeSelf)
|
|
weapons[i].OnDeselected();
|
|
|
|
weapons[i].gameObject.SetActive(active);
|
|
}
|
|
|
|
currentIndex = index;
|
|
Weapon selected = GetCurrentWeapon();
|
|
if (selected != null)
|
|
{
|
|
selected.OnSelected();
|
|
|
|
if (weaponNameText != null && selected.data != null)
|
|
weaponNameText.text = "Weapon: " + selected.data.weaponName;
|
|
|
|
if (hud != null)
|
|
hud.SetWeapon(selected.data);
|
|
}
|
|
}
|
|
|
|
public void NextWeapon()
|
|
{
|
|
if (weapons == null || weapons.Length == 0)
|
|
return;
|
|
|
|
SwitchTo((currentIndex + 1) % weapons.Length);
|
|
}
|
|
|
|
public void PreviousWeapon()
|
|
{
|
|
if (weapons == null || weapons.Length == 0)
|
|
return;
|
|
|
|
int nextIndex = currentIndex - 1;
|
|
if (nextIndex < 0)
|
|
nextIndex = weapons.Length - 1;
|
|
|
|
SwitchTo(nextIndex);
|
|
}
|
|
|
|
public Weapon GetCurrentWeapon()
|
|
{
|
|
if (weapons == null || weapons.Length == 0 || currentIndex < 0 || currentIndex >= weapons.Length)
|
|
return null;
|
|
|
|
return weapons[currentIndex];
|
|
}
|
|
}
|