Merge pull request #1349 from MegaMek/infantry_refresh

Conventional infantry fixes
MegaMek · Dec 26, 2023 · 7adc1d0 · 7adc1d0
2 parents 37aa5ff + 9d360f3
commit 7adc1d0
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/megameklab/src/megameklab/ui/infantry/CIStructureTab.java b/megameklab/src/megameklab/ui/infantry/CIStructureTab.java
@@ -56,6 +56,7 @@ public class CIStructureTab extends ITab implements InfantryBuildListener {
     public static final int T_MOUNT            = 4;
     public static final int T_AUGMENTATION     = 5;
 
+
     private BasicInfoView panBasicInfo;
     private CIPlatoonTypeView panPlatoonType;
     private CIWeaponView panWeapons;
@@ -424,6 +425,8 @@ public void platoonSizeChanged(int numSquads, int squadSize) {
         getInfantry().setSquadSize(squadSize);
         getInfantry().autoSetInternal();
         panPlatoonType.setFromEntity(getInfantry());
+        refresh.refreshStatus();
+        refresh.refreshPreview();
     }
 
     @Override
@@ -440,10 +443,11 @@ public void specializationsChanged() {
     public void numSecondaryChanged(final int count) {
         if (getInfantry().getSecondaryWeapon() == null) {
             getInfantry().setSecondaryWeaponsPerSquad(0);
-        } else if (count == 0) {
-            UnitUtil.replaceMainWeapon(getInfantry(), null, true);
-            getInfantry().setSpecializations(getInfantry().getSpecializations() & ~Infantry.TAG_TROOPS);
         } else {
+            if (count == 0) {
+                UnitUtil.replaceMainWeapon(getInfantry(), null, true);
+                getInfantry().setSpecializations(getInfantry().getSpecializations() & ~Infantry.TAG_TROOPS);
+            }
             getInfantry().setSecondaryWeaponsPerSquad(count);
         }
         refresh.refreshStatus();