111 lines
3.1 KiB
Python
111 lines
3.1 KiB
Python
import os
|
|
from typing import Any, List, Tuple
|
|
from kivy.lang import Observable
|
|
import gettext
|
|
from kivy.uix.label import Label
|
|
from kivy.uix.textinput import TextInput
|
|
from kivy.core.text import Label as CoreLabel
|
|
|
|
try:
|
|
import importlib.resources as pkg_resources
|
|
except:
|
|
import importlib_resources as pkg_resources
|
|
|
|
OUTPUT_ERROR = -1
|
|
OUTPUT_KATAGO_STDERR = -0.5
|
|
OUTPUT_INFO = 0
|
|
OUTPUT_DEBUG = 1
|
|
OUTPUT_EXTRA_DEBUG = 2
|
|
|
|
MODE_PLAY, MODE_ANALYZE = "play", "analyze"
|
|
|
|
|
|
def var_to_grid(array_var: List[Any], size: Tuple[int, int]) -> List[List[Any]]:
|
|
"""convert ownership/policy to grid format such that grid[y][x] is for move with coords x,y"""
|
|
ix = 0
|
|
grid = [[]] * size[1]
|
|
for y in range(size[1] - 1, -1, -1):
|
|
grid[y] = array_var[ix : ix + size[0]]
|
|
ix += size[0]
|
|
return grid
|
|
|
|
|
|
def evaluation_class(points_lost: float, eval_thresholds: List[float]):
|
|
i = 0
|
|
while i < len(eval_thresholds) - 1 and points_lost < eval_thresholds[i]:
|
|
i += 1
|
|
return i
|
|
|
|
|
|
def find_package_resource(path):
|
|
if path.startswith("katrain"):
|
|
parts = path.replace("\\", "/").split("/")
|
|
try:
|
|
with pkg_resources.path(".".join(parts[:-1]), parts[-1]) as path_obj:
|
|
return str(path_obj) # this will clean up if egg etc, but these don't work anyway
|
|
except (ModuleNotFoundError, FileNotFoundError) as e:
|
|
print(f"File {path} not found, installation possibly broken")
|
|
return f"FILENOTFOUND::{path}"
|
|
else:
|
|
return path # absolute path
|
|
|
|
|
|
class Lang(Observable):
|
|
observers = []
|
|
lang = None
|
|
|
|
def __init__(self, defaultlang):
|
|
super(Lang, self).__init__()
|
|
self.ugettext = None
|
|
self.lang = defaultlang
|
|
self.switch_lang(self.lang)
|
|
|
|
def _(self, text):
|
|
str = self.ugettext(text)
|
|
return str
|
|
|
|
def fbind(self, name, func, *args, **kwargs):
|
|
if name == "_":
|
|
self.observers.append((func, args, kwargs))
|
|
else:
|
|
return super(Lang, self).fbind(name, func, *args, **kwargs)
|
|
|
|
def funbind(self, name, func, *args, **kwargs):
|
|
if name == "_":
|
|
key = (func, args, kwargs)
|
|
print("funbind", key in self.observers)
|
|
if key in self.observers:
|
|
self.observers.remove(key)
|
|
else:
|
|
return super(Lang, self).funbind(name, func, *args, **kwargs)
|
|
|
|
def switch_lang(self, lang):
|
|
# get the right locales directory, and instantiate a gettext
|
|
i18n_dir, _ = os.path.split(find_package_resource("katrain/i18n/__init__.py"))
|
|
locale_dir = os.path.join(i18n_dir, "locales")
|
|
locales = gettext.translation("katrain", locale_dir, languages=[lang])
|
|
self.ugettext = locales.gettext
|
|
|
|
# update all the kv rules attached to this text
|
|
for func, args, kwargs in self.observers:
|
|
try:
|
|
func(args[0], None, None)
|
|
except ReferenceError:
|
|
pass # proxy no longer exists
|
|
|
|
|
|
LANGUAGE = "nl"
|
|
i18n = Lang(LANGUAGE)
|
|
|
|
|
|
class I18NLabel(Label):
|
|
pass
|
|
|
|
|
|
class I18NCoreLabel(CoreLabel):
|
|
pass
|
|
|
|
|
|
class I18NTextInput(TextInput):
|
|
pass
|