diff --git a/OutlawEditor/src/main/java/org/badvision/outlaweditor/MapEditor.java b/OutlawEditor/src/main/java/org/badvision/outlaweditor/MapEditor.java index 414569d9..2ec36f68 100644 --- a/OutlawEditor/src/main/java/org/badvision/outlaweditor/MapEditor.java +++ b/OutlawEditor/src/main/java/org/badvision/outlaweditor/MapEditor.java @@ -253,7 +253,7 @@ public class MapEditor extends Editor implements EventH idx = (idx + 1) % scripts.size(); gc.beginPath(); gc.moveTo(xx,yy); - gc.setStroke(currentMap.getScriptColor(scripts.get(idx))); + currentMap.getScriptColor(scripts.get(idx)).ifPresent(gc::setStroke); xx += dashLength; gc.lineTo(xx, yy); gc.setEffect(new DropShadow(2, Color.BLACK)); @@ -263,7 +263,7 @@ public class MapEditor extends Editor implements EventH idx = (idx + 1) % scripts.size(); gc.beginPath(); gc.moveTo(xx,yy); - gc.setStroke(currentMap.getScriptColor(scripts.get(idx))); + currentMap.getScriptColor(scripts.get(idx)).ifPresent(gc::setStroke); yy += dashLength; gc.lineTo(xx, yy); gc.setEffect(new DropShadow(2, Color.BLACK)); @@ -273,7 +273,7 @@ public class MapEditor extends Editor implements EventH idx = (idx + 1) % scripts.size(); gc.beginPath(); gc.moveTo(xx,yy); - gc.setStroke(currentMap.getScriptColor(scripts.get(idx))); + currentMap.getScriptColor(scripts.get(idx)).ifPresent(gc::setStroke); xx -= dashLength; gc.lineTo(xx, yy); gc.setEffect(new DropShadow(2, Color.BLACK)); @@ -283,7 +283,7 @@ public class MapEditor extends Editor implements EventH idx = (idx + 1) % scripts.size(); gc.beginPath(); gc.moveTo(xx,yy); - gc.setStroke(currentMap.getScriptColor(scripts.get(idx))); + currentMap.getScriptColor(scripts.get(idx)).ifPresent(gc::setStroke); yy -= dashLength; gc.lineTo(xx, yy); gc.setEffect(new DropShadow(2, Color.BLACK)); diff --git a/OutlawEditor/src/main/java/org/badvision/outlaweditor/data/TileMap.java b/OutlawEditor/src/main/java/org/badvision/outlaweditor/data/TileMap.java index 12c3c39d..97b2779e 100644 --- a/OutlawEditor/src/main/java/org/badvision/outlaweditor/data/TileMap.java +++ b/OutlawEditor/src/main/java/org/badvision/outlaweditor/data/TileMap.java @@ -6,6 +6,7 @@ import java.util.Collections; import java.util.HashMap; import java.util.HashSet; import java.util.List; +import java.util.Optional; import java.util.Set; import javafx.scene.image.WritableImage; import javafx.scene.paint.Color; @@ -46,8 +47,8 @@ public class TileMap extends ArrayList> implements Serializable private final java.util.Map> locationScripts = new HashMap<>(); private final java.util.Map scriptColors = new HashMap<>(); - public Color getScriptColor(Script s) { - return scriptColors.get(s); + public Optional getScriptColor(Script s) { + return Optional.ofNullable(scriptColors.get(s)); } public List