Skip to content

Commit 3223415

Browse files
author
bytekeeper
committed
#13 - Some more Set -> List; also using EnumSet for enums.
1 parent b67b7f0 commit 3223415

File tree

4 files changed

+12
-17
lines changed

4 files changed

+12
-17
lines changed

src/main/java/bwapi/TechType.java

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,7 @@
11
package bwapi;
22

33
import java.util.Arrays;
4-
import java.util.Set;
5-
import java.util.stream.Collectors;
4+
import java.util.List;
65

76
import static bwapi.Order.*;
87

@@ -223,8 +222,8 @@ public boolean targetsPosition() {
223222
return (techTypeFlags[id] & TARG_POS) != 0;
224223
}
225224

226-
public Set<UnitType> whatsUses() {
227-
return Arrays.stream(techWhatUses[id]).collect(Collectors.toSet());
225+
public List<UnitType> whatsUses() {
226+
return Arrays.asList(techWhatUses[id]);
228227
}
229228

230229
public Order getOrder() {

src/main/java/bwapi/Unit.java

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -13,10 +13,10 @@
1313
import static bwapi.UnitType.*;
1414

1515
public class Unit {
16-
private static Set<Order> gatheringGasOrders = new HashSet<>(Arrays.asList(
17-
Harvest1, Harvest2, MoveToGas, WaitForGas, HarvestGas, ReturnGas, ResetCollision));
18-
private static Set<Order> gatheringMineralOrders = new HashSet<>(Arrays.asList(
19-
Harvest1, Harvest2, MoveToMinerals, WaitForMinerals, MiningMinerals, ReturnMinerals, ResetCollision));
16+
private static Set<Order> gatheringGasOrders = EnumSet.of(
17+
Harvest1, Harvest2, MoveToGas, WaitForGas, HarvestGas, ReturnGas, ResetCollision);
18+
private static Set<Order> gatheringMineralOrders = EnumSet.of(
19+
Harvest1, Harvest2, MoveToMinerals, WaitForMinerals, MiningMinerals, ReturnMinerals, ResetCollision);
2020
private final UnitData unitData;
2121
private final Game game;
2222
// static

src/main/java/bwapi/UnitCommand.java

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,16 +1,13 @@
11
package bwapi;
22

33

4-
import java.util.Arrays;
5-
import java.util.HashSet;
4+
import java.util.EnumSet;
65
import java.util.Set;
76

87
import static bwapi.UnitCommandType.*;
98

109
public class UnitCommand {
11-
private static Set<UnitCommandType> posComs = new HashSet<>(
12-
Arrays.asList(Build, Land, Place_COP)
13-
);
10+
private static Set<UnitCommandType> posComs = EnumSet.of(Build, Land, Place_COP);
1411
final UnitCommandType type;
1512
final Unit target;
1613
final int x;

src/main/java/bwapi/UpgradeType.java

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,7 @@
11
package bwapi;
22

33
import java.util.Arrays;
4-
import java.util.Set;
5-
import java.util.stream.Collectors;
4+
import java.util.List;
65

76
public enum UpgradeType {
87
Terran_Infantry_Armor(0),
@@ -252,8 +251,8 @@ public UnitType whatUpgrades() {
252251
return whatUpgrades[id];
253252
}
254253

255-
public Set<UnitType> whatUses() {
256-
return Arrays.stream(upgradeWhatUses[id]).collect(Collectors.toSet());
254+
public List<UnitType> whatUses() {
255+
return Arrays.asList(upgradeWhatUses[id]);
257256
}
258257

259258
public int maxRepeats() {

0 commit comments

Comments
 (0)