fix: mine_resource 근본 수정 — 광석 타일로 먼저 걸어간 뒤 채굴

근본 원인: mining_state는 2.7타일 이내만 채굴 가능 이전: 반경 50에서 찾고 제자리에서 채굴 시도 → 손이 안 닿아 실패 이후: 1. 반경 80에서 가장 가까운 광석 찾기 2. 그 광석 타일 좌표로 move (걸어가기) 3. 도착 후 mining_state로 채굴 4. 타일 고갈/접근불가 시 다음 가까운 타일로 자동 이동 5. 최대 10라운드(이동+채굴) 반복
2026-03-25 20:51:43 +09:00
parent d27aff2824
commit 1c5c0e0a5a
1 changed files with 103 additions and 149 deletions
--- a/action_executor.py
+++ b/action_executor.py
@@ -1,16 +1,14 @@
 """
 action_executor.py — 순수 AI 플레이 버전

-핵심 변경:
- mine_resource: 접근 불가 시 광맥 주변 4방향으로 이동 후 재시도
- 검색 반경 20→50, 후보 5→10개
- 한 위치에서 실패 → 10칸씩 이동하며 접근 가능 지점 탐색
+핵심 수정:
+- mine_resource: 광석 타일로 먼저 WALK → 도착 후 mining_state
+- 손 채굴 거리 2.7타일 이내에서만 채굴 가능
+- 가장 가까운 광석으로 걸어가서 채굴 → 고갈 시 다음 타일로 이동
 """
 import time
-import json
 from factorio_rcon import FactorioRCON

-
 P = """local p = game.players[1]
 if not p then rcon.print("NO_PLAYER") return end
 if not p.character then rcon.print("NO_CHARACTER") return end
@@ -22,7 +20,7 @@ class ActionExecutor:
        self.rcon = rcon

    def execute(self, action: dict) -> tuple[bool, str]:
-        act    = action.get("action", "")
+        act = action.get("action", "")
        params = action.get("params", {})
        handlers = {
            "move": self.move, "explore": self.explore,
@@ -37,8 +35,8 @@ class ActionExecutor:
        try:
            result = handler(**params)
            if isinstance(result, tuple) and isinstance(result[1], str):
-                if "NO_PLAYER" in result[1]: return False, "서버에 접속한 플레이어가 없습니다."
-                if "NO_CHARACTER" in result[1]: return False, "플레이어 캐릭터가 없습니다."
+                if "NO_PLAYER" in result[1]: return False, "플레이어 없음"
+                if "NO_CHARACTER" in result[1]: return False, "캐릭터 없음"
            return result
        except TypeError as e: return False, f"파라미터 오류: {e}"
        except Exception as e: return False, f"실행 오류: {e}"
@@ -82,9 +80,9 @@ if ok then rcon.print(data) else rcon.print("ERROR") end
                    last_pos = pos_str
                    if stuck_count >= 3:
                        self.rcon.lua(P + "p.walking_state = {walking = false, direction = defines.direction.north}")
-                        return False, f"탐색 중 장애물에 막힘 위치({pos_str}) — 다른 방향 시도"
+                        return False, f"장애물에 막힘 위치({pos_str}) — 다른 방향 시도"
        self.rcon.lua(P + "p.walking_state = {walking = false, direction = defines.direction.north}")
-        return False, f"{direction} 방향 탐색 자원 미발견 — 다른 방향 시도"
+        return False, f"{direction} 방향 자원 미발견 — 다른 방향 시도"

    # ── 이동 ─────────────────────────────────────────────────────────
    def move(self, x: int, y: int) -> tuple[bool, str]:
@@ -123,120 +121,90 @@ rcon.print("WALK:" .. string.format("%.1f", dist))
            except: pass
            if stuck_count > 30:
                self.rcon.lua(P + "p.walking_state = {walking = false, direction = defines.direction.north}")
-                return False, f"({x},{y}) 장애물에 막힘 (남은 거리: {last_dist:.0f})"
+                return False, f"({x},{y}) 장애물 (거리: {last_dist:.0f})"
            time.sleep(0.1)
        self.rcon.lua(P + "p.walking_state = {walking = false, direction = defines.direction.north}")
-        return False, f"({x},{y}) 시간 초과 (남은 거리: {last_dist:.0f})"
+        return False, f"({x},{y}) 시간초과 (거리: {last_dist:.0f})"

-    # ── 한 지점에서 채굴 시도 (내부 헬퍼) ────────────────────────────
-    def _try_mine_at_position(self, ore: str, target_count: int, before_count: int) -> int:
-        """현재 위치에서 반경 50 내 광석 10개를 거리순으로 시도.
-        채굴 성공한 개수 반환. 0이면 접근 불가."""
-        find_result = self.rcon.lua(P + f"""
-local surface = p.surface
-local pos = p.position
-local resources = surface.find_entities_filtered{{position = pos, radius = 50, name = "{ore}"}}
-if #resources == 0 then rcon.print("NOT_FOUND") return end
-table.sort(resources, function(a, b)
-    local da = (a.position.x - pos.x)^2 + (a.position.y - pos.y)^2
-    local db = (b.position.x - pos.x)^2 + (b.position.y - pos.y)^2
-    return da < db
-end)
-local parts = {{}}
-for i = 1, math.min(10, #resources) do
-    local r = resources[i]
-    parts[#parts+1] = string.format("%.1f,%.1f", r.position.x, r.position.y)
+    # ── 채굴 (광석으로 걸어간 뒤 채굴) ──────────────────────────────
+    def mine_resource(self, ore: str = "iron-ore", count: int = 10) -> tuple[bool, str]:
+        """1. 가까운 광석 찾기 → 2. 그 위치로 걸어가기 → 3. 채굴.
+        한 타일 고갈 시 다음 가까운 타일로 자동 이동."""
+        before_count = self._get_item_count(ore)
+        target_count = before_count + count
+        total_mined = 0
+
+        for round_num in range(10):  # 최대 10번 이동+채굴 시도
+            # 1. 가장 가까운 광석 위치 찾기
+            find_result = self.rcon.lua(P + f"""
+local res = p.surface.find_entities_filtered{{position = p.position, radius = 80, name = "{ore}"}}
+if #res == 0 then rcon.print("NOT_FOUND") return end
+-- 거리순 정렬해서 가장 가까운 1개
+local closest = res[1]
+local min_dist = 999999
+for _, e in ipairs(res) do
+    local d = (e.position.x - p.position.x)^2 + (e.position.y - p.position.y)^2
+    if d < min_dist then min_dist = d closest = e end
 end
-rcon.print("FOUND:" .. table.concat(parts, ";"))
+rcon.print(string.format("%.1f,%.1f", closest.position.x, closest.position.y))
 """)
-        if not find_result or find_result == "NOT_FOUND":
-            return 0
+            if not find_result or find_result in ("NO_PLAYER", "NO_CHARACTER", "NOT_FOUND"):
+                if total_mined > 0:
+                    return True, f"{ore} {total_mined}개 채굴 (주변 광석 소진)"
+                return False, f"반경 80 내 {ore} 없음 — explore로 다른 광맥 찾기"

-        coords_str = find_result.replace("FOUND:", "")
-        coord_list = []
-        for c in coords_str.split(";"):
-            parts = c.split(",")
-            if len(parts) == 2:
-                coord_list.append((float(parts[0]), float(parts[1])))
+            try:
+                parts = find_result.split(",")
+                ox, oy = float(parts[0]), float(parts[1])
+            except:
+                return False, f"광석 좌표 파싱 실패: {find_result}"

-        for attempt, (rx, ry) in enumerate(coord_list):
+            # 2. 광석 위치로 걸어가기
+            print(f"    [채굴] 광석({ox:.0f},{oy:.0f})으로 이동 중...")
+            ok, msg = self.move(int(ox), int(oy))
+            if not ok:
+                print(f"    [채굴] 이동 실패: {msg} — 다음 시도")
+                # 이동 실패해도 현재 위치에서 채굴 시도
+                pass
+
+            # 3. 현재 위치에서 반경 3 이내 광석 채굴 (손 채굴 거리)
            stall_count = 0
-            last_item_count = self._get_item_count(ore)
+            last_item = self._get_item_count(ore)

-            for tick in range(150):  # 타일당 최대 15초
-                self.rcon.lua(P + f"p.mining_state = {{mining = true, position = {{{rx}, {ry}}}}}")
+            for tick in range(300):  # 최대 30초
+                self.rcon.lua(P + f"p.mining_state = {{mining = true, position = {{{ox}, {oy}}}}}")
                time.sleep(0.1)

-                if tick % 10 == 9:
+                if tick % 8 == 7:  # 매 0.8초마다 체크
                    current = self._get_item_count(ore)
                    if current >= target_count:
                        self.rcon.lua(P + "p.mining_state = {mining = false}")
-                        return current - before_count
+                        total_mined = current - before_count
+                        return True, f"{ore} {total_mined}개 채굴 완료"

-                    if current > last_item_count:
-                        # 채굴 진행 중! 계속
+                    if current > last_item:
                        stall_count = 0
-                        last_item_count = current
+                        last_item = current
                    else:
                        stall_count += 1

+                    # 2번 연속 진행 없음 → 이 타일 안 닿음, 다음으로
                    if stall_count >= 2:
-                        # 이 타일 접근 불가 → 다음 타일
+                        print(f"    [채굴] 타일({ox:.0f},{oy:.0f}) 채굴 불가/고갈 — 다음 타일 이동")
                        break

            self.rcon.lua(P + "p.mining_state = {mining = false}")
+            total_mined = self._get_item_count(ore) - before_count

-        return self._get_item_count(ore) - before_count
+            if total_mined >= count:
+                return True, f"{ore} {total_mined}개 채굴 완료"

-    # ── 채굴 (접근 불가 시 광맥 주변 이동 후 재시도) ──────────────────
-    def mine_resource(self, ore: str = "iron-ore", count: int = 10) -> tuple[bool, str]:
-        """현재 위치에서 시도 → 실패 시 광맥 주변 4방향으로 이동하며 접근점 탐색"""
-        before_count = self._get_item_count(ore)
-        target_count = before_count + count
+        # 10라운드 후
+        if total_mined > 0:
+            return True, f"{ore} {total_mined}개 채굴 (목표 {count}개 중 일부)"
+        return False, f"{ore} 채굴 실패 — 접근 가능한 광석 없음"

-        # 1차: 현재 위치에서 시도
-        print(f"    [채굴] 현재 위치에서 {ore} 채굴 시도...")
-        mined = self._try_mine_at_position(ore, target_count, before_count)
-        if mined >= count:
-            return True, f"{ore} {mined}개 채굴 완료"
-        if mined > 0:
-            return True, f"{ore} {mined}개 채굴 (일부 — 접근 가능 타일에서 최대한)"
-
-        # 2차: 실패 시 광맥 주변 4방향으로 이동하며 재시도
-        # 현재 위치 기준 동서남북 10칸씩 이동
-        offsets = [(10, 0), (-10, 0), (0, 10), (0, -10), (15, 10), (-15, 10), (10, -15), (-10, -15)]
-        cur_pos = self.rcon.lua(P + 'rcon.print(string.format("%.0f,%.0f", p.position.x, p.position.y))')
-        try:
-            cx, cy = [float(v) for v in cur_pos.split(",")]
-        except:
-            cx, cy = 0, 0
-
-        for i, (dx, dy) in enumerate(offsets):
-            tx, ty = int(cx + dx), int(cy + dy)
-            print(f"    [채굴] 위치({tx},{ty})로 이동하여 재시도 ({i+1}/{len(offsets)})...")
-
-            ok, msg = self.move(tx, ty)
-            if not ok:
-                print(f"    [채굴] 이동 실패: {msg}")
-                continue
-
-            before_now = self._get_item_count(ore)
-            target_now = before_count + count  # 원래 목표 기준
-            mined = self._try_mine_at_position(ore, target_now, before_count)
-
-            total = self._get_item_count(ore) - before_count
-            if total >= count:
-                return True, f"{ore} {total}개 채굴 완료 (위치 이동 {i+1}회)"
-            if total > 0:
-                print(f"    [채굴] {total}개 채굴 중... 계속 시도")
-
-        # 최종 결과
-        total = self._get_item_count(ore) - before_count
-        if total > 0:
-            return True, f"{ore} {total}개 채굴 (목표 {count}개 중 일부 — 광맥 주변 탐색 완료)"
-        return False, f"{ore} 채굴 실패 — 광맥 주변 8방향 이동 후에도 접근 불가. explore로 다른 광맥 찾기"
-
-    # ── 제작 ─────────────────────────────────────────────────────────
+    # ── 제작/배치/삽입/레시피/연구/대기 ──────────────────────────────
    def craft_item(self, item: str, count: int = 1) -> tuple[bool, str]:
        result = self.rcon.lua(P + f"""
 local recipe = p.force.recipes["{item}"]
@@ -255,7 +223,6 @@ if crafted > 0 then rcon.print("CRAFTING:" .. crafted) else rcon.print("NO_INGRE
        elif result == "NO_INGREDIENTS": return False, f"{item} 재료 부족"
        return False, f"제작 실패: {result}"

-    # ── 건물 배치 ────────────────────────────────────────────────────
    def place_entity(self, name: str, x: int, y: int, direction: str = "north") -> tuple[bool, str]:
        dir_map = {"north": "defines.direction.north", "south": "defines.direction.south",
                   "east": "defines.direction.east", "west": "defines.direction.west"}
@@ -263,28 +230,23 @@ if crafted > 0 then rcon.print("CRAFTING:" .. crafted) else rcon.print("NO_INGRE
        result = self.rcon.lua(P + f"""
 local inv = p.get_main_inventory()
 local have = inv.get_item_count("{name}")
-if have < 1 then rcon.print("NO_ITEM:" .. have) return end
+if have < 1 then rcon.print("NO_ITEM") return end
 local dist = math.sqrt(({x} - p.position.x)^2 + ({y} - p.position.y)^2)
 if dist > p.build_distance + 2 then rcon.print("TOO_FAR:" .. string.format("%.1f", dist)) return end
-local existing = p.surface.find_entity("{name}", {{{x}, {y}}})
-if existing then rcon.print("ALREADY_EXISTS") return end
 p.cursor_stack.set_stack({{name="{name}", count=1}})
 local built = p.build_from_cursor{{position = {{{x}, {y}}}, direction = {lua_dir}}}
-if built then p.cursor_stack.clear() rcon.print("OK")
-else p.cursor_stack.clear() rcon.print("BLOCKED") end
+p.cursor_stack.clear()
+if built then rcon.print("OK") else rcon.print("BLOCKED") end
 """)
        if not result or result in ("NO_PLAYER", "NO_CHARACTER"): return False, result or "플레이어 없음"
-        if result == "OK": return True, f"{name} 배치 완료 ({x},{y})"
-        elif result.startswith("NO_ITEM"): return False, f"인벤토리에 {name} 없음"
-        elif result.startswith("TOO_FAR"): return False, f"({x},{y})가 너무 멀음 — move 먼저"
-        elif result == "ALREADY_EXISTS": return True, f"{name}이 이미 ({x},{y})에 있음"
-        elif result == "BLOCKED": return False, f"({x},{y}) 배치 불가"
+        if result == "OK": return True, f"{name} 배치 ({x},{y})"
+        elif result == "NO_ITEM": return False, f"인벤토리에 {name} 없음"
+        elif result.startswith("TOO_FAR"): return False, f"너무 멀음 — move 먼저"
+        elif result == "BLOCKED": return False, f"배치 불가"
        return False, f"배치 실패: {result}"

-    # ── 벨트/삽입/레시피/연구/대기 (변경 없음) ───────────────────────
    def place_belt_line(self, from_x: int, from_y: int, to_x: int, to_y: int) -> tuple[bool, str]:
-        placed, failed = 0, 0
-        positions = []
+        placed, failed, positions = 0, 0, []
        if from_x != to_x:
            step = 1 if from_x < to_x else -1
            d = "east" if step == 1 else "west"
@@ -300,8 +262,8 @@ else p.cursor_stack.clear() rcon.print("BLOCKED") end
            ok, _ = self.place_entity("transport-belt", bx, by, d)
            if ok: placed += 1
            else: failed += 1
-        if placed == 0: return False, f"벨트 설치 실패"
-        return True, f"벨트 {placed}개 설치" + (f", {failed}개 실패" if failed else "")
+        if placed == 0: return False, "벨트 설치 실패"
+        return True, f"벨트 {placed}개" + (f", {failed}개 실패" if failed else "")

    def insert_to_entity(self, x: int, y: int, item: str = "coal", count: int = 50) -> tuple[bool, str]:
        result = self.rcon.lua(P + f"""
@@ -314,46 +276,40 @@ local actual = math.min(have, {count})
 local entities = p.surface.find_entities_filtered{{position = {{{x},{y}}}, radius = 2}}
 local inserted = false
 for _, e in ipairs(entities) do
-    if e.valid then
-        if e.burner then
-            local fi = e.burner.inventory
-            if fi then
-                local removed = inv.remove({{name="{item}", count=actual}})
-                if removed > 0 then fi.insert({{name="{item}", count=removed}}) inserted = true break end
-            end
+    if e.valid and e.burner then
+        local fi = e.burner.inventory
+        if fi then
+            local removed = inv.remove({{name="{item}", count=actual}})
+            if removed > 0 then fi.insert({{name="{item}", count=removed}}) inserted = true break end
        end
-        if not inserted then
-            for i = 1, 10 do
-                local ti = e.get_inventory(i)
-                if ti and ti.can_insert({{name="{item}", count=1}}) then
-                    local removed = inv.remove({{name="{item}", count=actual}})
-                    if removed > 0 then ti.insert({{name="{item}", count=removed}}) inserted = true end
-                    break
-                end
-            end
-        end
-        if inserted then break end
    end
+    if not inserted and e.valid then
+        for i = 1, 10 do
+            local ti = e.get_inventory(i)
+            if ti and ti.can_insert({{name="{item}", count=1}}) then
+                local removed = inv.remove({{name="{item}", count=actual}})
+                if removed > 0 then ti.insert({{name="{item}", count=removed}}) inserted = true end
+                break
+            end
+        end
+    end
+    if inserted then break end
 end
 rcon.print(inserted and "OK" or "NOT_FOUND")
 """)
-        if not result or result in ("NO_PLAYER", "NO_CHARACTER"): return False, result or "플레이어 없음"
-        if result == "OK": return True, f"({x},{y})에 {item} 삽입 완료"
-        elif result == "TOO_FAR": return False, f"({x},{y})가 너무 멀음"
-        elif result == "NO_ITEM": return False, f"인벤토리에 {item} 없음"
-        return False, f"({x},{y})에 건물 없음"
+        if not result or result in ("NO_PLAYER", "NO_CHARACTER"): return False, "플레이어 없음"
+        if result == "OK": return True, f"({x},{y})에 {item} 삽입"
+        elif result == "TOO_FAR": return False, "너무 멀음"
+        elif result == "NO_ITEM": return False, f"{item} 없음"
+        return False, "건물 없음"

    def set_recipe(self, x: int, y: int, recipe: str) -> tuple[bool, str]:
        result = self.rcon.lua(P + f"""
-local dist = math.sqrt(({x} - p.position.x)^2 + ({y} - p.position.y)^2)
-if dist > p.build_distance + 2 then rcon.print("TOO_FAR") return end
 local e = p.surface.find_entities_filtered{{position = {{{x},{y}}}, radius = 2, type = "assembling-machine"}}[1]
 if e then e.set_recipe("{recipe}") rcon.print("OK") else rcon.print("NOT_FOUND") end
 """)
-        if not result or result in ("NO_PLAYER", "NO_CHARACTER"): return False, result or "플레이어 없음"
-        if result == "OK": return True, f"({x},{y}) 레시피: {recipe}"
-        elif result == "TOO_FAR": return False, f"너무 멀음"
-        return False, f"조립기 없음"
+        if result == "OK": return True, f"레시피: {recipe}"
+        return False, "조립기 없음"

    def start_research(self, tech: str = "automation") -> tuple[bool, str]:
        result = self.rcon.lua(P + f"""
@@ -363,15 +319,13 @@ if not t then rcon.print("NO_TECH")
 elseif t.researched then rcon.print("ALREADY_DONE")
 else force.research_queue_enabled = true force.add_research("{tech}") rcon.print("OK") end
 """)
-        if not result or result in ("NO_PLAYER", "NO_CHARACTER"): return False, result or "플레이어 없음"
        if result == "OK": return True, f"{tech} 연구 시작"
-        elif result == "ALREADY_DONE": return True, f"{tech} 이미 완료"
-        elif result == "NO_TECH": return False, f"{tech} 기술 없음"
-        return False, result
+        elif result == "ALREADY_DONE": return True, f"{tech} 완료"
+        return False, result or "실패"

    def wait(self, seconds: int = 3) -> tuple[bool, str]:
        time.sleep(min(seconds, 30))
-        return True, f"{seconds}초 대기 완료"
+        return True, f"{seconds}초 대기"

    def _get_item_count(self, item: str) -> int:
        result = self.rcon.lua(P + f'rcon.print(tostring(p.get_main_inventory().get_item_count("{item}")))')