From 4c2cbc55586482e130f7c3f0971756107093b082 Mon Sep 17 00:00:00 2001
From: Sheikah45 <Sheikah450@gmail.com>
Date: Wed, 28 Apr 2021 09:02:00 -0400
Subject: [PATCH] Fix unexplored map crash with units

---
 .../neroxis/map/generator/prop/EnemyCivPropGenerator.java       | 2 ++
 .../neroxis/map/generator/prop/NeutralCivPropGenerator.java     | 2 ++
 .../com/faforever/neroxis/map/generator/MapGeneratorTest.java   | 2 +-
 3 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/main/java/com/faforever/neroxis/map/generator/prop/EnemyCivPropGenerator.java b/src/main/java/com/faforever/neroxis/map/generator/prop/EnemyCivPropGenerator.java
index 936f3e5f3..4fb5817d0 100644
--- a/src/main/java/com/faforever/neroxis/map/generator/prop/EnemyCivPropGenerator.java
+++ b/src/main/java/com/faforever/neroxis/map/generator/prop/EnemyCivPropGenerator.java
@@ -36,6 +36,8 @@ protected void setupCivilianPipeline() {
         if (!map.isUnexplored()) {
             baseMask.randomize(.005f).setSize(mapSize + 1);
             baseMask.multiply(passableLand.copy().subtract(unbuildable).deflate(24)).fillCenter(32, false).fillEdge(32, false);
+        } else {
+            baseMask.setSize(mapSize + 1);
         }
     }
 
diff --git a/src/main/java/com/faforever/neroxis/map/generator/prop/NeutralCivPropGenerator.java b/src/main/java/com/faforever/neroxis/map/generator/prop/NeutralCivPropGenerator.java
index f9cc22765..d5d20bc01 100644
--- a/src/main/java/com/faforever/neroxis/map/generator/prop/NeutralCivPropGenerator.java
+++ b/src/main/java/com/faforever/neroxis/map/generator/prop/NeutralCivPropGenerator.java
@@ -37,6 +37,8 @@ protected void setupCivilianPipeline() {
         if (!map.isUnexplored()) {
             civReclaimMask.randomize(.005f).setSize(mapSize + 1);
             civReclaimMask.multiply(passableLand.copy().subtract(unbuildable).deflate(24)).fillCenter(32, false).fillEdge(64, false);
+        } else {
+            civReclaimMask.setSize(mapSize + 1);
         }
     }
 
diff --git a/src/test/java/com/faforever/neroxis/map/generator/MapGeneratorTest.java b/src/test/java/com/faforever/neroxis/map/generator/MapGeneratorTest.java
index 90a46b1ec..bdde7abf2 100644
--- a/src/test/java/com/faforever/neroxis/map/generator/MapGeneratorTest.java
+++ b/src/test/java/com/faforever/neroxis/map/generator/MapGeneratorTest.java
@@ -373,7 +373,7 @@ public void TestEqualityBiomeSpecified() throws Exception {
 
     @Test
     public void TestUnexploredNoUnits() throws Exception {
-        for (int i = 0; i < 5; ++i) {
+        for (int i = 0; i < 10; ++i) {
             Pipeline.reset();
             instance = new MapGenerator();
             instance.interpretArguments(new String[]{"--unexplored", "--map-size", "256"});