소스 검색

add fire line

olinox14 3 년 전
부모
커밋
ded644530c
1개의 변경된 파일6개의 추가작업 그리고 3개의 파일을 삭제
  1. 6 3
      cultist_war/main.py

+ 6 - 3
cultist_war/main.py

@@ -290,9 +290,12 @@ class Grid(BaseClass):
             result.reverse()
         return result
 
-    @classmethod
-    def fire_dist(cls, from_, to_):
-        return len(cls.line(from_, to_))
+    def fire_line(self, from_, to_):
+        line = self.line(from_, to_)
+        return line if all(self.can_see_trough(c) for c in line) else []
+
+    def fire_dist(self, from_, to_):
+        return len(self.fire_line(from_, to_))
 
 
 # Create grid