bauiv1lib.teamnamescolors

  1# Released under the MIT License. See LICENSE for details.
  2#
  3"""Provides a window to customize team names and colors."""
  4
  5from __future__ import annotations
  6
  7from typing import TYPE_CHECKING, cast, override
  8
  9from bauiv1lib.popup import PopupWindow
 10from bauiv1lib.colorpicker import ColorPicker
 11import bauiv1 as bui
 12
 13if TYPE_CHECKING:
 14    from typing import Sequence
 15
 16
 17class TeamNamesColorsWindow(PopupWindow):
 18    """A popup window for customizing team names and colors."""
 19
 20    def __init__(self, scale_origin: tuple[float, float]):
 21        from bascenev1 import DEFAULT_TEAM_COLORS, DEFAULT_TEAM_NAMES
 22
 23        self._width = 500
 24        self._height = 330
 25        self._transitioning_out = False
 26        self._max_name_length = 16
 27
 28        # Creates our _root_widget.
 29        assert bui.app.classic is not None
 30        uiscale = bui.app.ui_v1.uiscale
 31        scale = (
 32            1.69
 33            if uiscale is bui.UIScale.SMALL
 34            else 1.1 if uiscale is bui.UIScale.MEDIUM else 0.85
 35        )
 36        super().__init__(
 37            position=scale_origin, size=(self._width, self._height), scale=scale
 38        )
 39
 40        appconfig = bui.app.config
 41        self._names = list(
 42            appconfig.get('Custom Team Names', DEFAULT_TEAM_NAMES)
 43        )
 44
 45        # We need to flatten the translation since it will be an
 46        # editable string.
 47        self._names = [
 48            bui.Lstr(translate=('teamNames', n)).evaluate() for n in self._names
 49        ]
 50        self._colors = list(
 51            appconfig.get('Custom Team Colors', DEFAULT_TEAM_COLORS)
 52        )
 53
 54        self._color_buttons: list[bui.Widget] = []
 55        self._color_text_fields: list[bui.Widget] = []
 56
 57        resetbtn = bui.buttonwidget(
 58            parent=self.root_widget,
 59            label=bui.Lstr(resource='settingsWindowAdvanced.resetText'),
 60            autoselect=True,
 61            scale=0.7,
 62            on_activate_call=self._reset,
 63            size=(120, 50),
 64            position=(self._width * 0.5 - 60 * 0.7, self._height - 60),
 65        )
 66
 67        for i in range(2):
 68            self._color_buttons.append(
 69                bui.buttonwidget(
 70                    parent=self.root_widget,
 71                    autoselect=True,
 72                    position=(50, 0 + 195 - 90 * i),
 73                    on_activate_call=bui.Call(self._color_click, i),
 74                    size=(70, 70),
 75                    color=self._colors[i],
 76                    label='',
 77                    button_type='square',
 78                )
 79            )
 80            self._color_text_fields.append(
 81                bui.textwidget(
 82                    parent=self.root_widget,
 83                    position=(135, 0 + 201 - 90 * i),
 84                    size=(280, 46),
 85                    text=self._names[i],
 86                    h_align='left',
 87                    v_align='center',
 88                    max_chars=self._max_name_length,
 89                    color=self._colors[i],
 90                    description=bui.Lstr(resource='nameText'),
 91                    editable=True,
 92                    padding=4,
 93                )
 94            )
 95        bui.widget(
 96            edit=self._color_text_fields[0],
 97            down_widget=self._color_text_fields[1],
 98        )
 99        bui.widget(
100            edit=self._color_text_fields[1],
101            up_widget=self._color_text_fields[0],
102        )
103        bui.widget(edit=self._color_text_fields[0], up_widget=resetbtn)
104
105        cancelbtn = bui.buttonwidget(
106            parent=self.root_widget,
107            label=bui.Lstr(resource='cancelText'),
108            autoselect=True,
109            on_activate_call=self._on_cancel_press,
110            size=(150, 50),
111            position=(self._width * 0.5 - 200, 20),
112        )
113        okbtn = bui.buttonwidget(
114            parent=self.root_widget,
115            label=bui.Lstr(resource='okText'),
116            autoselect=True,
117            on_activate_call=self._ok,
118            size=(150, 50),
119            position=(self._width * 0.5 + 50, 20),
120        )
121        bui.containerwidget(
122            edit=self.root_widget, selected_child=self._color_buttons[0]
123        )
124        bui.widget(edit=okbtn, left_widget=cancelbtn)
125        self._update()
126
127    def _color_click(self, i: int) -> None:
128        ColorPicker(
129            parent=self.root_widget,
130            position=self._color_buttons[i].get_screen_space_center(),
131            offset=(270.0, 0),
132            initial_color=self._colors[i],
133            delegate=self,
134            tag=i,
135        )
136
137    def color_picker_closing(self, picker: ColorPicker) -> None:
138        """Called when the color picker is closing."""
139
140    def color_picker_selected_color(
141        self, picker: ColorPicker, color: Sequence[float]
142    ) -> None:
143        """Called when a color is selected in the color picker."""
144        self._colors[picker.get_tag()] = color
145        self._update()
146
147    def _reset(self) -> None:
148        from bascenev1 import DEFAULT_TEAM_NAMES, DEFAULT_TEAM_COLORS
149
150        for i in range(2):
151            self._colors[i] = DEFAULT_TEAM_COLORS[i]
152            name = bui.Lstr(
153                translate=('teamNames', DEFAULT_TEAM_NAMES[i])
154            ).evaluate()
155            if len(name) > self._max_name_length:
156                print('GOT DEFAULT TEAM NAME LONGER THAN MAX LENGTH')
157            bui.textwidget(edit=self._color_text_fields[i], text=name)
158        self._update()
159
160    def _update(self) -> None:
161        for i in range(2):
162            bui.buttonwidget(edit=self._color_buttons[i], color=self._colors[i])
163            bui.textwidget(
164                edit=self._color_text_fields[i], color=self._colors[i]
165            )
166
167    def _ok(self) -> None:
168        from bascenev1 import DEFAULT_TEAM_COLORS, DEFAULT_TEAM_NAMES
169
170        cfg = bui.app.config
171
172        # First, determine whether the values here are defaults, in which case
173        # we can clear any values from prefs.  Currently if the string matches
174        # either the default raw value or its translation we consider it
175        # default. (the fact that team names get translated makes this
176        # situation a bit sloppy)
177        new_names: list[str] = []
178        is_default = True
179        for i in range(2):
180            name = cast(str, bui.textwidget(query=self._color_text_fields[i]))
181            if not name:
182                bui.screenmessage(
183                    bui.Lstr(resource='nameNotEmptyText'), color=(1, 0, 0)
184                )
185                bui.getsound('error').play()
186                return
187            new_names.append(name)
188
189        for i in range(2):
190            if self._colors[i] != DEFAULT_TEAM_COLORS[i]:
191                is_default = False
192            default_team_name = DEFAULT_TEAM_NAMES[i]
193            default_team_name_translated = bui.Lstr(
194                translate=('teamNames', default_team_name)
195            ).evaluate()
196            if (
197                new_names[i] != default_team_name
198                and new_names[i] != default_team_name_translated
199            ):
200                is_default = False
201
202        if is_default:
203            for key in ('Custom Team Names', 'Custom Team Colors'):
204                if key in cfg:
205                    del cfg[key]
206        else:
207            cfg['Custom Team Names'] = list(new_names)
208            cfg['Custom Team Colors'] = list(self._colors)
209
210        cfg.commit()
211        self._transition_out()
212
213    def _transition_out(self, transition: str = 'out_scale') -> None:
214        if not self._transitioning_out:
215            self._transitioning_out = True
216            bui.containerwidget(edit=self.root_widget, transition=transition)
217
218    @override
219    def on_popup_cancel(self) -> None:
220        bui.getsound('swish').play()
221        self._transition_out()
222
223    def _on_cancel_press(self) -> None:
224        self._transition_out()