-
Notifications
You must be signed in to change notification settings - Fork 177
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Issue 3650: Moved Potential Transports in ATB thread to new class
- Loading branch information
1 parent
9c464cd
commit ecf29e4
Showing
2 changed files
with
124 additions
and
27 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,108 @@ | ||
| /** | ||
| * Copyright (c) 2025-2025 The MegaMek Team. All Rights Reserved. | ||
| * | ||
| * This file is part of MekHQ. | ||
| * | ||
| * MekHQ is free software: you can redistribute it and/or modify | ||
| * it under the terms of the GNU General Public License as published by | ||
| * the Free Software Foundation, either version 3 of the License, or | ||
| * (at your option) any later version. | ||
| * | ||
| * MekHQ is distributed in the hope that it will be useful, | ||
| * but WITHOUT ANY WARRANTY; without even the implied warranty of | ||
| * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | ||
| * GNU General Public License for more details. | ||
| * | ||
| * You should have received a copy of the GNU General Public License | ||
| * along with MekHQ. If not, see <http://www.gnu.org/licenses/>. | ||
| */ | ||
|
|
||
| package mekhq.utilities; | ||
|
|
||
| import mekhq.campaign.enums.CampaignTransportType; | ||
|
|
||
| import java.util.*; | ||
|
|
||
| /** | ||
| * Map to help hold what transports are transporting what units for a given campaign transport type | ||
| * @see CampaignTransportType | ||
| */ | ||
| public class PotentialTransportsMap { | ||
|
|
||
| private final HashMap<CampaignTransportType, Map<UUID, List<UUID>>> hashMap = new HashMap<CampaignTransportType, Map<UUID, List<UUID>>>(); | ||
|
|
||
| public PotentialTransportsMap(CampaignTransportType[] campaignTransportTypes) { | ||
| for (CampaignTransportType campaignTransportType : campaignTransportTypes) { | ||
| hashMap.put(campaignTransportType, new HashMap<>()); | ||
| } | ||
| } | ||
|
|
||
| /** | ||
| * For the provided campaign transport type, get the transports | ||
| * @param campaignTransportType type (enum) of campaign transport | ||
| * @return transports for the given campaign transport type | ||
| */ | ||
| public Set<UUID> getTransports(CampaignTransportType campaignTransportType) { | ||
| return hashMap.get(campaignTransportType).keySet(); | ||
| } | ||
|
|
||
| /** | ||
| * For the provided campaign transport type and transport id, get the transported units | ||
| * @param campaignTransportType type (enum) of campaign transport | ||
| * @param uuid transport id | ||
| * @return list of uuids of units on that transport | ||
| */ | ||
| public List<UUID> getTransportedUnits(CampaignTransportType campaignTransportType, UUID uuid) { | ||
| return hashMap.get(campaignTransportType).get(uuid); | ||
| } | ||
|
|
||
| /** | ||
| * For the provided campaign transport type, does the provided transport exist in the map? | ||
| * @param campaignTransportType type (enum) of campaign transport | ||
| * @param key transport id | ||
| * @return true if that transport exists, false if not | ||
| */ | ||
| public boolean containsTransportKey(CampaignTransportType campaignTransportType, UUID key) { | ||
| return hashMap.containsKey(campaignTransportType) && hashMap.get(campaignTransportType).containsKey(key); | ||
| } | ||
|
|
||
| /** | ||
| * For the provided campaign transport type, add a transport | ||
| * @param campaignTransportType type (enum) of campaign transport | ||
| * @param key transport id | ||
| */ | ||
| public void putNewTransport(CampaignTransportType campaignTransportType, UUID key) { | ||
| hashMap.get(campaignTransportType).put(key, new ArrayList<>()); | ||
| } | ||
|
|
||
| /** | ||
| * For the provided campaign transport type and transport, add a transported unit | ||
| * @param campaignTransportType type (enum) of campaign transport | ||
| * @param key transport unit id | ||
| * @param value transported unit id | ||
| */ | ||
| public void addTransportedUnit(CampaignTransportType campaignTransportType, UUID key, UUID value) { | ||
| hashMap.get(campaignTransportType).get(key).add(value); | ||
| } | ||
|
|
||
| /** | ||
| * Removes any transports that are empty | ||
| */ | ||
| public void removeEmptyTransports() { | ||
| if (hashMap.isEmpty()) { | ||
| return; | ||
| } | ||
| for (CampaignTransportType campaignTransportType : hashMap.keySet()) { | ||
| Set<UUID> emptyTransports = new HashSet<>(); | ||
| if (!(hashMap.get(campaignTransportType).isEmpty())) { | ||
| for (UUID transport : hashMap.get(campaignTransportType).keySet()) { | ||
| if (hashMap.get(campaignTransportType).get(transport).isEmpty()) | ||
| emptyTransports.add(transport); | ||
| } | ||
| } | ||
| for (UUID emptyTransport : emptyTransports) { | ||
| hashMap.get(campaignTransportType).remove(emptyTransport); | ||
| } | ||
| } | ||
| } | ||
| } |