bastd.ui.iconpicker

  1# Released under the MIT License. See LICENSE for details.
  2#
  3"""Provides a picker for icons."""
  4
  5from __future__ import annotations
  6
  7import math
  8from typing import TYPE_CHECKING
  9
 10import ba
 11import ba.internal
 12from bastd.ui import popup
 13
 14if TYPE_CHECKING:
 15    from typing import Any, Sequence
 16
 17
 18class IconPicker(popup.PopupWindow):
 19    """Picker for icons."""
 20
 21    def __init__(
 22        self,
 23        parent: ba.Widget,
 24        position: tuple[float, float] = (0.0, 0.0),
 25        delegate: Any = None,
 26        scale: float | None = None,
 27        offset: tuple[float, float] = (0.0, 0.0),
 28        tint_color: Sequence[float] = (1.0, 1.0, 1.0),
 29        tint2_color: Sequence[float] = (1.0, 1.0, 1.0),
 30        selected_icon: str | None = None,
 31    ):
 32        # pylint: disable=too-many-locals
 33        del parent  # unused here
 34        del tint_color  # unused_here
 35        del tint2_color  # unused here
 36        uiscale = ba.app.ui.uiscale
 37        if scale is None:
 38            scale = (
 39                1.85
 40                if uiscale is ba.UIScale.SMALL
 41                else 1.65
 42                if uiscale is ba.UIScale.MEDIUM
 43                else 1.23
 44            )
 45
 46        self._delegate = delegate
 47        self._transitioning_out = False
 48
 49        self._icons = [
 50            ba.charstr(ba.SpecialChar.LOGO)
 51        ] + ba.app.accounts_v1.get_purchased_icons()
 52        count = len(self._icons)
 53        columns = 4
 54        rows = int(math.ceil(float(count) / columns))
 55
 56        button_width = 50
 57        button_height = 50
 58        button_buffer_h = 10
 59        button_buffer_v = 5
 60
 61        self._width = 10 + columns * (button_width + 2 * button_buffer_h) * (
 62            1.0 / 0.95
 63        ) * (1.0 / 0.8)
 64        self._height = self._width * (
 65            0.8 if uiscale is ba.UIScale.SMALL else 1.06
 66        )
 67
 68        self._scroll_width = self._width * 0.8
 69        self._scroll_height = self._height * 0.8
 70        self._scroll_position = (
 71            (self._width - self._scroll_width) * 0.5,
 72            (self._height - self._scroll_height) * 0.5,
 73        )
 74
 75        # creates our _root_widget
 76        popup.PopupWindow.__init__(
 77            self,
 78            position=position,
 79            size=(self._width, self._height),
 80            scale=scale,
 81            bg_color=(0.5, 0.5, 0.5),
 82            offset=offset,
 83            focus_position=self._scroll_position,
 84            focus_size=(self._scroll_width, self._scroll_height),
 85        )
 86
 87        self._scrollwidget = ba.scrollwidget(
 88            parent=self.root_widget,
 89            size=(self._scroll_width, self._scroll_height),
 90            color=(0.55, 0.55, 0.55),
 91            highlight=False,
 92            position=self._scroll_position,
 93        )
 94        ba.containerwidget(edit=self._scrollwidget, claims_left_right=True)
 95
 96        self._sub_width = self._scroll_width * 0.95
 97        self._sub_height = (
 98            5 + rows * (button_height + 2 * button_buffer_v) + 100
 99        )
100        self._subcontainer = ba.containerwidget(
101            parent=self._scrollwidget,
102            size=(self._sub_width, self._sub_height),
103            background=False,
104        )
105        index = 0
106        for y in range(rows):
107            for x in range(columns):
108                pos = (
109                    x * (button_width + 2 * button_buffer_h) + button_buffer_h,
110                    self._sub_height
111                    - (y + 1) * (button_height + 2 * button_buffer_v)
112                    + 0,
113                )
114                btn = ba.buttonwidget(
115                    parent=self._subcontainer,
116                    button_type='square',
117                    size=(button_width, button_height),
118                    autoselect=True,
119                    text_scale=1.2,
120                    label='',
121                    color=(0.65, 0.65, 0.65),
122                    on_activate_call=ba.Call(
123                        self._select_icon, self._icons[index]
124                    ),
125                    position=pos,
126                )
127                ba.textwidget(
128                    parent=self._subcontainer,
129                    h_align='center',
130                    v_align='center',
131                    size=(0, 0),
132                    position=(pos[0] + 0.5 * button_width - 1, pos[1] + 15),
133                    draw_controller=btn,
134                    text=self._icons[index],
135                    scale=1.8,
136                )
137                ba.widget(edit=btn, show_buffer_top=60, show_buffer_bottom=60)
138                if self._icons[index] == selected_icon:
139                    ba.containerwidget(
140                        edit=self._subcontainer,
141                        selected_child=btn,
142                        visible_child=btn,
143                    )
144                index += 1
145
146                if index >= count:
147                    break
148            if index >= count:
149                break
150        self._get_more_icons_button = btn = ba.buttonwidget(
151            parent=self._subcontainer,
152            size=(self._sub_width * 0.8, 60),
153            position=(self._sub_width * 0.1, 30),
154            label=ba.Lstr(resource='editProfileWindow.getMoreIconsText'),
155            on_activate_call=self._on_store_press,
156            color=(0.6, 0.6, 0.6),
157            textcolor=(0.8, 0.8, 0.8),
158            autoselect=True,
159        )
160        ba.widget(edit=btn, show_buffer_top=30, show_buffer_bottom=30)
161
162    def _on_store_press(self) -> None:
163        from bastd.ui.account import show_sign_in_prompt
164        from bastd.ui.store.browser import StoreBrowserWindow
165
166        if ba.internal.get_v1_account_state() != 'signed_in':
167            show_sign_in_prompt()
168            return
169        self._transition_out()
170        StoreBrowserWindow(
171            modal=True,
172            show_tab=StoreBrowserWindow.TabID.ICONS,
173            origin_widget=self._get_more_icons_button,
174        )
175
176    def _select_icon(self, icon: str) -> None:
177        if self._delegate is not None:
178            self._delegate.on_icon_picker_pick(icon)
179        self._transition_out()
180
181    def _transition_out(self) -> None:
182        if not self._transitioning_out:
183            self._transitioning_out = True
184            ba.containerwidget(edit=self.root_widget, transition='out_scale')
185
186    def on_popup_cancel(self) -> None:
187        ba.playsound(ba.getsound('swish'))
188        self._transition_out()