bauiv1lib.playlist.mapselect

  1# Released under the MIT License. See LICENSE for details.
  2#
  3"""Provides UI for selecting maps in playlists."""
  4
  5from __future__ import annotations
  6
  7import math
  8from typing import TYPE_CHECKING
  9
 10import bauiv1 as bui
 11
 12if TYPE_CHECKING:
 13    from typing import Any, Callable
 14
 15    import bascenev1 as bs
 16
 17
 18class PlaylistMapSelectWindow(bui.Window):
 19    """Window to select a map."""
 20
 21    def __init__(
 22        self,
 23        gametype: type[bs.GameActivity],
 24        sessiontype: type[bs.Session],
 25        config: dict[str, Any],
 26        edit_info: dict[str, Any],
 27        completion_call: Callable[[dict[str, Any] | None], Any],
 28        transition: str = 'in_right',
 29    ):
 30        from bascenev1 import get_filtered_map_name
 31
 32        self._gametype = gametype
 33        self._sessiontype = sessiontype
 34        self._config = config
 35        self._completion_call = completion_call
 36        self._edit_info = edit_info
 37        self._maps: list[tuple[str, bui.Texture]] = []
 38        try:
 39            self._previous_map = get_filtered_map_name(
 40                config['settings']['map']
 41            )
 42        except Exception:
 43            self._previous_map = ''
 44
 45        assert bui.app.classic is not None
 46        uiscale = bui.app.ui_v1.uiscale
 47        width = 715 if uiscale is bui.UIScale.SMALL else 615
 48        x_inset = 50 if uiscale is bui.UIScale.SMALL else 0
 49        height = (
 50            400
 51            if uiscale is bui.UIScale.SMALL
 52            else 480
 53            if uiscale is bui.UIScale.MEDIUM
 54            else 600
 55        )
 56
 57        top_extra = 20 if uiscale is bui.UIScale.SMALL else 0
 58        super().__init__(
 59            root_widget=bui.containerwidget(
 60                size=(width, height + top_extra),
 61                transition=transition,
 62                scale=(
 63                    2.17
 64                    if uiscale is bui.UIScale.SMALL
 65                    else 1.3
 66                    if uiscale is bui.UIScale.MEDIUM
 67                    else 1.0
 68                ),
 69                stack_offset=(0, -27)
 70                if uiscale is bui.UIScale.SMALL
 71                else (0, 0),
 72            )
 73        )
 74
 75        self._cancel_button = btn = bui.buttonwidget(
 76            parent=self._root_widget,
 77            position=(38 + x_inset, height - 67),
 78            size=(140, 50),
 79            scale=0.9,
 80            text_scale=1.0,
 81            autoselect=True,
 82            label=bui.Lstr(resource='cancelText'),
 83            on_activate_call=self._cancel,
 84        )
 85
 86        bui.containerwidget(edit=self._root_widget, cancel_button=btn)
 87        bui.textwidget(
 88            parent=self._root_widget,
 89            position=(width * 0.5, height - 46),
 90            size=(0, 0),
 91            maxwidth=260,
 92            scale=1.1,
 93            text=bui.Lstr(
 94                resource='mapSelectTitleText',
 95                subs=[('${GAME}', self._gametype.get_display_string())],
 96            ),
 97            color=bui.app.ui_v1.title_color,
 98            h_align='center',
 99            v_align='center',
100        )
101        v = height - 70
102        self._scroll_width = width - (80 + 2 * x_inset)
103        self._scroll_height = height - 140
104
105        self._scrollwidget = bui.scrollwidget(
106            parent=self._root_widget,
107            position=(40 + x_inset, v - self._scroll_height),
108            size=(self._scroll_width, self._scroll_height),
109        )
110        bui.containerwidget(
111            edit=self._root_widget, selected_child=self._scrollwidget
112        )
113        bui.containerwidget(edit=self._scrollwidget, claims_left_right=True)
114
115        self._subcontainer: bui.Widget | None = None
116        self._refresh()
117
118    def _refresh(self, select_get_more_maps_button: bool = False) -> None:
119        # pylint: disable=too-many-statements
120        # pylint: disable=too-many-branches
121        # pylint: disable=too-many-locals
122        from bascenev1 import (
123            get_map_class,
124            get_map_display_string,
125        )
126
127        assert bui.app.classic is not None
128        store = bui.app.classic.store
129        # Kill old.
130        if self._subcontainer is not None:
131            self._subcontainer.delete()
132
133        mesh_opaque = bui.getmesh('level_select_button_opaque')
134        mesh_transparent = bui.getmesh('level_select_button_transparent')
135
136        self._maps = []
137        map_list = self._gametype.get_supported_maps(self._sessiontype)
138        map_list_sorted = list(map_list)
139        map_list_sorted.sort()
140        unowned_maps = store.get_unowned_maps()
141
142        for mapname in map_list_sorted:
143            # Disallow ones we don't own.
144            if mapname in unowned_maps:
145                continue
146            map_tex_name = get_map_class(mapname).get_preview_texture_name()
147            if map_tex_name is not None:
148                try:
149                    map_tex = bui.gettexture(map_tex_name)
150                    self._maps.append((mapname, map_tex))
151                except Exception:
152                    print(f'Invalid map preview texture: "{map_tex_name}".')
153            else:
154                print('Error: no map preview texture for map:', mapname)
155
156        count = len(self._maps)
157        columns = 2
158        rows = int(math.ceil(float(count) / columns))
159        button_width = 220
160        button_height = button_width * 0.5
161        button_buffer_h = 16
162        button_buffer_v = 19
163        self._sub_width = self._scroll_width * 0.95
164        self._sub_height = (
165            5 + rows * (button_height + 2 * button_buffer_v) + 100
166        )
167        self._subcontainer = bui.containerwidget(
168            parent=self._scrollwidget,
169            size=(self._sub_width, self._sub_height),
170            background=False,
171        )
172        index = 0
173        mask_texture = bui.gettexture('mapPreviewMask')
174        h_offs = 130 if len(self._maps) == 1 else 0
175        for y in range(rows):
176            for x in range(columns):
177                pos = (
178                    x * (button_width + 2 * button_buffer_h)
179                    + button_buffer_h
180                    + h_offs,
181                    self._sub_height
182                    - (y + 1) * (button_height + 2 * button_buffer_v)
183                    + 12,
184                )
185                btn = bui.buttonwidget(
186                    parent=self._subcontainer,
187                    button_type='square',
188                    size=(button_width, button_height),
189                    autoselect=True,
190                    texture=self._maps[index][1],
191                    mask_texture=mask_texture,
192                    mesh_opaque=mesh_opaque,
193                    mesh_transparent=mesh_transparent,
194                    label='',
195                    color=(1, 1, 1),
196                    on_activate_call=bui.Call(
197                        self._select_with_delay, self._maps[index][0]
198                    ),
199                    position=pos,
200                )
201                if x == 0:
202                    bui.widget(edit=btn, left_widget=self._cancel_button)
203                if y == 0:
204                    bui.widget(edit=btn, up_widget=self._cancel_button)
205                if x == columns - 1 and bui.app.ui_v1.use_toolbars:
206                    bui.widget(
207                        edit=btn,
208                        right_widget=bui.get_special_widget('party_button'),
209                    )
210
211                bui.widget(edit=btn, show_buffer_top=60, show_buffer_bottom=60)
212                if self._maps[index][0] == self._previous_map:
213                    bui.containerwidget(
214                        edit=self._subcontainer,
215                        selected_child=btn,
216                        visible_child=btn,
217                    )
218                name = get_map_display_string(self._maps[index][0])
219                bui.textwidget(
220                    parent=self._subcontainer,
221                    text=name,
222                    position=(pos[0] + button_width * 0.5, pos[1] - 12),
223                    size=(0, 0),
224                    scale=0.5,
225                    maxwidth=button_width,
226                    draw_controller=btn,
227                    h_align='center',
228                    v_align='center',
229                    color=(0.8, 0.8, 0.8, 0.8),
230                )
231                index += 1
232
233                if index >= count:
234                    break
235            if index >= count:
236                break
237        self._get_more_maps_button = btn = bui.buttonwidget(
238            parent=self._subcontainer,
239            size=(self._sub_width * 0.8, 60),
240            position=(self._sub_width * 0.1, 30),
241            label=bui.Lstr(resource='mapSelectGetMoreMapsText'),
242            on_activate_call=self._on_store_press,
243            color=(0.6, 0.53, 0.63),
244            textcolor=(0.75, 0.7, 0.8),
245            autoselect=True,
246        )
247        bui.widget(edit=btn, show_buffer_top=30, show_buffer_bottom=30)
248        if select_get_more_maps_button:
249            bui.containerwidget(
250                edit=self._subcontainer, selected_child=btn, visible_child=btn
251            )
252
253    def _on_store_press(self) -> None:
254        from bauiv1lib import account
255        from bauiv1lib.store.browser import StoreBrowserWindow
256
257        plus = bui.app.plus
258        assert plus is not None
259
260        if plus.get_v1_account_state() != 'signed_in':
261            account.show_sign_in_prompt()
262            return
263        StoreBrowserWindow(
264            modal=True,
265            show_tab=StoreBrowserWindow.TabID.MAPS,
266            on_close_call=self._on_store_close,
267            origin_widget=self._get_more_maps_button,
268        )
269
270    def _on_store_close(self) -> None:
271        self._refresh(select_get_more_maps_button=True)
272
273    def _select(self, map_name: str) -> None:
274        from bauiv1lib.playlist.editgame import PlaylistEditGameWindow
275
276        self._config['settings']['map'] = map_name
277        bui.containerwidget(edit=self._root_widget, transition='out_right')
278        assert bui.app.classic is not None
279        bui.app.ui_v1.set_main_menu_window(
280            PlaylistEditGameWindow(
281                self._gametype,
282                self._sessiontype,
283                self._config,
284                self._completion_call,
285                default_selection='map',
286                transition='in_left',
287                edit_info=self._edit_info,
288            ).get_root_widget()
289        )
290
291    def _select_with_delay(self, map_name: str) -> None:
292        bui.lock_all_input()
293        bui.apptimer(0.1, bui.unlock_all_input)
294        bui.apptimer(0.1, bui.WeakCall(self._select, map_name))
295
296    def _cancel(self) -> None:
297        from bauiv1lib.playlist.editgame import PlaylistEditGameWindow
298
299        bui.containerwidget(edit=self._root_widget, transition='out_right')
300        assert bui.app.classic is not None
301        bui.app.ui_v1.set_main_menu_window(
302            PlaylistEditGameWindow(
303                self._gametype,
304                self._sessiontype,
305                self._config,
306                self._completion_call,
307                default_selection='map',
308                transition='in_left',
309                edit_info=self._edit_info,
310            ).get_root_widget()
311        )