Witness: apworld support (#1885)

worlds/witness/static_logic.py

@@ -1,6 +1,6 @@
 import os
 
-from .utils import define_new_region, parse_lambda, lazy
+from .utils import define_new_region, parse_lambda, lazy, get_logic_file, get_items
 
 
 class StaticWitnessLogicObj:
@@ -8,16 +8,12 @@ class StaticWitnessLogicObj:
         """
         Reads the logic file and does the initial population of data structures
         """
-        path = os.path.join(os.path.dirname(__file__), file_path)
+        lines = get_logic_file(file_path)
 
-        with open(path, "r", encoding="utf-8") as file:
         current_region = dict()
-
         counter = 0
 
-            for line in file.readlines():
-                line = line.strip()
-
+        for line in lines:
             if line == "":
                 continue
 
@@ -166,13 +162,10 @@ class StaticWitnessLogic:
         Parses currently defined items from WitnessItems.txt
         """
 
-        path = os.path.join(os.path.dirname(__file__), "WitnessItems.txt")
-        with open(path, "r", encoding="utf-8") as file:
+        lines = get_items()
         current_set = self.ALL_SYMBOL_ITEMS
 
-            for line in file.readlines():
-                line = line.strip()
-
+        for line in lines:
             if line == "Progression:":
                 current_set = self.ALL_SYMBOL_ITEMS
                 continue

worlds/witness/utils.py

@@ -3,7 +3,7 @@ from Utils import cache_argsless
 from itertools import accumulate
 from typing import *
 from fractions import Fraction
-from collections import Counter
+from pkgutil import get_data
 
 
 def make_warning_string(any_j: bool, any_u: bool, any_d: bool, all_j: bool, all_u: bool, all_d: bool) -> str:
@@ -142,10 +142,8 @@ class lazy(object):
 
 
 def get_adjustment_file(adjustment_file):
-    path = os.path.join(os.path.dirname(__file__), adjustment_file)
-
-    with open(path) as f:
-        return [line.strip() for line in f.readlines()]
+    data = get_data(__name__, adjustment_file).decode('utf-8')
+    return [line.strip() for line in data.split("\n")]
 
 
 @cache_argsless
@@ -226,3 +224,33 @@ def get_ep_no_mountain():
 @cache_argsless
 def get_ep_no_videos():
     return get_adjustment_file("settings/EP_Shuffle/EP_Videos.txt")
+
+
+@cache_argsless
+def get_sigma_normal_logic():
+    return get_adjustment_file("WitnessLogic.txt")
+
+
+@cache_argsless
+def get_sigma_expert_logic():
+    return get_adjustment_file("WitnessLogicExpert.txt")
+
+
+@cache_argsless
+def get_vanilla_logic():
+    return get_adjustment_file("WitnessLogicVanilla.txt")
+
+
+@cache_argsless
+def get_items():
+    return get_adjustment_file("WitnessItems.txt")
+
+
+def get_logic_file(filepath: str):
+    if filepath == "WitnessLogic.txt":
+        return get_sigma_normal_logic()
+    if filepath == "WitnessLogicExpert.txt":
+        return get_sigma_expert_logic()
+    if filepath == "WitnessLogicVanilla.txt":
+        return get_vanilla_logic
+    return get_adjustment_file(filepath)