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