simplify CountShipDefenses code

alaingilbert · Dec 6, 2020 · c803f61 · c803f61
1 parent cc6dac6
commit c803f61
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/defence.go b/defence.go
@@ -30,10 +30,9 @@ func (d DefensesInfos) HasMissilesDefense() bool {
 // CountShipDefenses returns the count of defenses which can attack ships i.e., excluding AntiBallisticMissiles
 func (d DefensesInfos) CountShipDefenses() (out int64) {
 	for _, defense := range Defenses {
-		if defense == InterplanetaryMissiles || defense == AntiBallisticMissiles {
-			continue
+		if defense != InterplanetaryMissiles && defense != AntiBallisticMissiles {
+			out += d.ByID(defense.GetID())
 		}
-		out += d.ByID(defense.GetID())
 	}
 	return
 }