Small craft unit type added GROUND_SUPPORT, ASSAULT, INFANTRY_CARRIER…

…, BA_CARRIER roles
MegaMek · May 29, 2024 · d1ef66d · d1ef66d
1 parent 0ff1491
commit d1ef66d
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/megamek/src/megamek/client/ratgenerator/MissionRole.java b/megamek/src/megamek/client/ratgenerator/MissionRole.java
@@ -197,8 +197,13 @@ public boolean fitsUnitType(int unitType) {
             // Roles for conventional and aerospace fighters
             case BOMBER:
             case INTERCEPTOR:
+                return unitType == UnitType.CONV_FIGHTER ||
+                        unitType == UnitType.AEROSPACEFIGHTER;
+
             case GROUND_SUPPORT:
-                return unitType == UnitType.CONV_FIGHTER || unitType == UnitType.AEROSPACEFIGHTER;
+                return unitType == UnitType.CONV_FIGHTER ||
+                        unitType == UnitType.AEROSPACEFIGHTER ||
+                        unitType == UnitType.SMALL_CRAFT;
 
             case ESCORT:
                 return  unitType == UnitType.CONV_FIGHTER ||
@@ -207,9 +212,12 @@ public boolean fitsUnitType(int unitType) {
 
             // Roles for DropShips
             case ASSAULT:
-            case VEE_CARRIER:
             case INFANTRY_CARRIER:
             case BA_CARRIER:
+                return unitType == UnitType.DROPSHIP ||
+                        unitType == UnitType.SMALL_CRAFT;
+
+            case VEE_CARRIER:
             case MECH_CARRIER:
             case TUG:
             case POCKET_WARSHIP: