button solver, some tweaks

2 files changed, 19 insertions, 0 deletions

diff --git a/src/main/java/kr/syeyoung/dungeonsguide/dungeon/doorfinder/CatacombDataProvider.java b/src/main/java/kr/syeyoung/dungeonsguide/dungeon/doorfinder/CatacombDataProvider.java
index d39418bd..0c874458 100755
--- a/src/main/java/kr/syeyoung/dungeonsguide/dungeon/doorfinder/CatacombDataProvider.java
+++ b/src/main/java/kr/syeyoung/dungeonsguide/dungeon/doorfinder/CatacombDataProvider.java
@@ -92,4 +92,21 @@ public class CatacombDataProvider implements DungeonSpecificDataProvider {
         }
         return null;
     }
+
+    @Override
+    public boolean isTrapSpawn(String dungeonName) {
+        String floor = dungeonName.substring(14).trim();
+        if (floor.equals("F3")) {
+            return true;
+        } else if (floor.equals("F4")) {
+            return true;
+        } else if (floor.equals("F5")) {
+            return true;
+        } else if (floor.equals("F6")) {
+            return true;
+        } else if (floor.equals("F7")) {
+            return true;
+        }
+        return false;
+    }
 }
diff --git a/src/main/java/kr/syeyoung/dungeonsguide/dungeon/doorfinder/DungeonSpecificDataProvider.java b/src/main/java/kr/syeyoung/dungeonsguide/dungeon/doorfinder/DungeonSpecificDataProvider.java
index 3103adf0..4cc0e677 100755
--- a/src/main/java/kr/syeyoung/dungeonsguide/dungeon/doorfinder/DungeonSpecificDataProvider.java
+++ b/src/main/java/kr/syeyoung/dungeonsguide/dungeon/doorfinder/DungeonSpecificDataProvider.java
@@ -12,4 +12,6 @@ public interface DungeonSpecificDataProvider {
     Vector2d findDoorOffset(World w, String dungeonName);
 
     BossfightProcessor createBossfightProcessor(World w, String dungeonName);
+
+    boolean isTrapSpawn(String dungeonName);
 }
\ No newline at end of file