fix selected ammo retrieval

MegaMek · Oct 19, 2024 · f3afac0 · f3afac0
1 parent 795f2c5
commit f3afac0
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 15 deletions.
diff --git a/megamek/src/megamek/client/ui/swing/ClientGUI.java b/megamek/src/megamek/client/ui/swing/ClientGUI.java
@@ -3023,7 +3023,7 @@ public Optional<WeaponMounted> getDisplayedWeapon() {
     }
 
     public Optional<AmmoMounted> getDisplayedAmmo() {
-        return Optional.ofNullable(unitDisplay.wPan.getSelectedAmmo());
+        return unitDisplay.wPan.getSelectedAmmo();
     }
 
     @Override

diff --git a/megamek/src/megamek/client/ui/swing/unitDisplay/WeaponPanel.java b/megamek/src/megamek/client/ui/swing/unitDisplay/WeaponPanel.java
@@ -30,11 +30,7 @@
 import java.awt.event.ActionListener;
 import java.awt.event.KeyListener;
 import java.awt.event.MouseEvent;
-import java.util.ArrayList;
-import java.util.Comparator;
-import java.util.EnumSet;
-import java.util.List;
-import java.util.Objects;
+import java.util.*;
 
 import javax.swing.*;
 import javax.swing.event.ListSelectionEvent;
@@ -1348,16 +1344,16 @@ public WeaponMounted getSelectedWeapon() {
     }
 
     /**
-     *
-     * @return the AmmoMounted currently selected by the ammo selector combo box,
-     *         not the linked ammo per se.
+     * @return the AmmoMounted currently selected by the ammo selector combo box, if any.
+     * The returned AmmoMounted may or may not be the ammo that is linked to the weapon.
      */
-    public AmmoMounted getSelectedAmmo() {
+    public Optional<AmmoMounted> getSelectedAmmo() {
         int selected = m_chAmmo.getSelectedIndex();
-        if (selected == -1 || vAmmo == null) {
-            return null;
+        if ((selected == -1) || (vAmmo == null) || (selected >= vAmmo.size())) {
+            return Optional.empty();
+        } else {
+            return Optional.of(vAmmo.get(selected));
         }
-        return vAmmo.get(m_chAmmo.getSelectedIndex());
     }
 
     /**
@@ -1948,8 +1944,7 @@ private void displaySelected() {
 
         // Update the range display to account for the selected ammo, or the loaded ammo
         // if none is selected
-        int curDisplayed = m_chAmmo.getSelectedIndex();
-        AmmoMounted mAmmo = (curDisplayed != -1 && vAmmo != null) ? vAmmo.get(curDisplayed) : mounted.getLinkedAmmo();
+        AmmoMounted mAmmo = getSelectedAmmo().orElse(mounted.getLinkedAmmo());
         if (mAmmo != null) {
             updateRangeDisplayForAmmo(mAmmo);
         }
@@ -2073,6 +2068,7 @@ private void displaySelected() {
 
                     vAmmo.add(mountedAmmo);
                     m_chAmmo.addItem(formatAmmo(mountedAmmo));
+                    int curDisplayed = m_chAmmo.getSelectedIndex();
                     if (curDisplayed != -1) {
                         nCur = curDisplayed;
                     } else if ((mounted.getLinked() != null) &&