bauiv1lib.playoptions

  1# Released under the MIT License. See LICENSE for details.
  2#
  3"""Provides a window for configuring play options."""
  4
  5from __future__ import annotations
  6
  7import logging
  8from typing import TYPE_CHECKING, override
  9
 10import bascenev1 as bs
 11import bauiv1 as bui
 12
 13from bauiv1lib.popup import PopupWindow
 14
 15if TYPE_CHECKING:
 16    from typing import Any
 17
 18    from bauiv1lib.play import PlaylistSelectContext
 19
 20
 21class PlayOptionsWindow(PopupWindow):
 22    """A popup window for configuring play options."""
 23
 24    def __init__(
 25        self,
 26        sessiontype: type[bs.Session],
 27        playlist: str,
 28        scale_origin: tuple[float, float],
 29        delegate: Any = None,
 30        playlist_select_context: PlaylistSelectContext | None = None,
 31    ):
 32        # FIXME: Tidy this up.
 33        # pylint: disable=too-many-branches
 34        # pylint: disable=too-many-statements
 35        # pylint: disable=too-many-locals
 36        from bascenev1 import filter_playlist, get_map_class
 37        from bauiv1lib.playlist import PlaylistTypeVars
 38        from bauiv1lib.config import ConfigNumberEdit
 39
 40        self._r = 'gameListWindow'
 41        self._delegate = delegate
 42        self._pvars = PlaylistTypeVars(sessiontype)
 43        self._transitioning_out = False
 44
 45        self._playlist_select_context = playlist_select_context
 46
 47        self._do_randomize_val = bui.app.config.get(
 48            self._pvars.config_name + ' Playlist Randomize', 0
 49        )
 50
 51        self._sessiontype = sessiontype
 52        self._playlist = playlist
 53
 54        self._width = 500.0
 55        self._height = 370.0 - 50.0
 56
 57        # In teams games, show the custom names/colors button.
 58        if self._sessiontype is bs.DualTeamSession:
 59            self._height += 50.0
 60
 61        self._row_height = 45.0
 62
 63        # Grab our maps to display.
 64        mesh_opaque = bui.getmesh('level_select_button_opaque')
 65        mesh_transparent = bui.getmesh('level_select_button_transparent')
 66        mask_tex = bui.gettexture('mapPreviewMask')
 67
 68        # Poke into this playlist and see if we can display some of its
 69        # maps.
 70        map_textures = []
 71        map_texture_entries = []
 72        rows = 0
 73        columns = 0
 74        game_count = 0
 75        scl = 0.35
 76        c_width_total = 0.0
 77        try:
 78            max_columns = 5
 79            name = playlist
 80            if name == '__default__':
 81                plst = self._pvars.get_default_list_call()
 82            else:
 83                try:
 84                    plst = bui.app.config[
 85                        self._pvars.config_name + ' Playlists'
 86                    ][name]
 87                except Exception:
 88                    print(
 89                        'ERROR INFO: self._config_name is:',
 90                        self._pvars.config_name,
 91                    )
 92                    print(
 93                        'ERROR INFO: playlist names are:',
 94                        list(
 95                            bui.app.config[
 96                                self._pvars.config_name + ' Playlists'
 97                            ].keys()
 98                        ),
 99                    )
100                    raise
101            plst = filter_playlist(
102                plst,
103                self._sessiontype,
104                remove_unowned=False,
105                mark_unowned=True,
106                name=name,
107            )
108            game_count = len(plst)
109            for entry in plst:
110                mapname = entry['settings']['map']
111                maptype: type[bs.Map] | None
112                try:
113                    maptype = get_map_class(mapname)
114                except bui.NotFoundError:
115                    maptype = None
116                if maptype is not None:
117                    tex_name = maptype.get_preview_texture_name()
118                    if tex_name is not None:
119                        map_textures.append(tex_name)
120                        map_texture_entries.append(entry)
121            rows = (max(0, len(map_textures) - 1) // max_columns) + 1
122            columns = min(max_columns, len(map_textures))
123
124            if len(map_textures) == 1:
125                scl = 1.1
126            elif len(map_textures) == 2:
127                scl = 0.7
128            elif len(map_textures) == 3:
129                scl = 0.55
130            else:
131                scl = 0.35
132            self._row_height = 128.0 * scl
133            c_width_total = scl * 250.0 * columns
134            if map_textures:
135                self._height += self._row_height * rows
136
137        except Exception:
138            logging.exception('Error listing playlist maps.')
139
140        show_shuffle_check_box = game_count > 1
141
142        if show_shuffle_check_box:
143            self._height += 40
144
145        uiscale = bui.app.ui_v1.uiscale
146        scale = (
147            1.69
148            if uiscale is bui.UIScale.SMALL
149            else 1.1 if uiscale is bui.UIScale.MEDIUM else 0.85
150        )
151        # Creates our _root_widget.
152        super().__init__(
153            position=scale_origin, size=(self._width, self._height), scale=scale
154        )
155
156        playlist_name: str | bui.Lstr = (
157            self._pvars.default_list_name
158            if playlist == '__default__'
159            else playlist
160        )
161        self._title_text = bui.textwidget(
162            parent=self.root_widget,
163            position=(self._width * 0.5, self._height - 89 + 51),
164            size=(0, 0),
165            text=playlist_name,
166            scale=1.4,
167            color=(1, 1, 1),
168            maxwidth=self._width * 0.7,
169            h_align='center',
170            v_align='center',
171        )
172
173        self._cancel_button = bui.buttonwidget(
174            parent=self.root_widget,
175            position=(25, self._height - 53),
176            size=(50, 50),
177            scale=0.7,
178            label='',
179            color=(0.42, 0.73, 0.2),
180            on_activate_call=self._on_cancel_press,
181            autoselect=True,
182            icon=bui.gettexture('crossOut'),
183            iconscale=1.2,
184        )
185
186        h_offs_img = self._width * 0.5 - c_width_total * 0.5
187        v_offs_img = self._height - 118 - scl * 125.0 + 50
188        bottom_row_buttons = []
189        self._have_at_least_one_owned = False
190
191        for row in range(rows):
192            for col in range(columns):
193                tex_index = row * columns + col
194                if tex_index < len(map_textures):
195                    tex_name = map_textures[tex_index]
196                    h = h_offs_img + scl * 250 * col
197                    v = v_offs_img - self._row_height * row
198                    entry = map_texture_entries[tex_index]
199                    owned = not (
200                        ('is_unowned_map' in entry and entry['is_unowned_map'])
201                        or (
202                            'is_unowned_game' in entry
203                            and entry['is_unowned_game']
204                        )
205                    )
206
207                    if owned:
208                        self._have_at_least_one_owned = True
209
210                    try:
211                        desc = bui.getclass(
212                            entry['type'], subclassof=bs.GameActivity
213                        ).get_settings_display_string(entry)
214                        if not owned:
215                            desc = bui.Lstr(
216                                value='${DESC}\n${UNLOCK}',
217                                subs=[
218                                    ('${DESC}', desc),
219                                    (
220                                        '${UNLOCK}',
221                                        bui.Lstr(
222                                            resource='unlockThisInTheStoreText'
223                                        ),
224                                    ),
225                                ],
226                            )
227                        desc_color = (0, 1, 0) if owned else (1, 0, 0)
228                    except Exception:
229                        desc = bui.Lstr(value='(invalid)')
230                        desc_color = (1, 0, 0)
231
232                    btn = bui.buttonwidget(
233                        parent=self.root_widget,
234                        size=(scl * 240.0, scl * 120.0),
235                        position=(h, v),
236                        texture=bui.gettexture(tex_name if owned else 'empty'),
237                        mesh_opaque=mesh_opaque if owned else None,
238                        on_activate_call=bui.Call(
239                            bui.screenmessage, desc, desc_color
240                        ),
241                        label='',
242                        color=(1, 1, 1),
243                        autoselect=True,
244                        extra_touch_border_scale=0.0,
245                        mesh_transparent=mesh_transparent if owned else None,
246                        mask_texture=mask_tex if owned else None,
247                    )
248                    if row == 0 and col == 0:
249                        bui.widget(edit=self._cancel_button, down_widget=btn)
250                    if row == rows - 1:
251                        bottom_row_buttons.append(btn)
252                    if not owned:
253                        # Ewww; buttons don't currently have alpha so in this
254                        # case we draw an image over our button with an empty
255                        # texture on it.
256                        bui.imagewidget(
257                            parent=self.root_widget,
258                            size=(scl * 260.0, scl * 130.0),
259                            position=(h - 10.0 * scl, v - 4.0 * scl),
260                            draw_controller=btn,
261                            color=(1, 1, 1),
262                            texture=bui.gettexture(tex_name),
263                            mesh_opaque=mesh_opaque,
264                            opacity=0.25,
265                            mesh_transparent=mesh_transparent,
266                            mask_texture=mask_tex,
267                        )
268
269                        bui.imagewidget(
270                            parent=self.root_widget,
271                            size=(scl * 100, scl * 100),
272                            draw_controller=btn,
273                            position=(h + scl * 70, v + scl * 10),
274                            texture=bui.gettexture('lock'),
275                        )
276
277        y_offs = 50 if show_shuffle_check_box else 0
278
279        # Series Length
280        y_offs2 = 40 if self._sessiontype is bs.DualTeamSession else 0
281        self._series_length_numedit = ConfigNumberEdit(
282            parent=self.root_widget,
283            position=(100, 200 + y_offs + y_offs2),
284            configkey=(
285                'FFA' if self._sessiontype is bs.FreeForAllSession else 'Teams'
286            )
287            + ' Series Length',
288            displayname=bui.Lstr(
289                resource=self._r
290                + (
291                    '.pointsToWinText'
292                    if self._sessiontype is bs.FreeForAllSession
293                    else '.seriesLengthText'
294                )
295            ),
296            minval=1.0,
297            maxval=100.0 if self._sessiontype is bs.FreeForAllSession else 99.0,
298            increment=1.0 if self._sessiontype is bs.FreeForAllSession else 2.0,
299            fallback_value=(
300                24 if self._sessiontype is bs.FreeForAllSession else 7
301            ),
302            f=0,
303        )
304
305        # Team names/colors.
306        self._custom_colors_names_button: bui.Widget | None
307        if self._sessiontype is bs.DualTeamSession:
308            self._custom_colors_names_button = bui.buttonwidget(
309                parent=self.root_widget,
310                position=(100, 195 + y_offs),
311                size=(290, 35),
312                on_activate_call=bui.WeakCall(self._custom_colors_names_press),
313                autoselect=True,
314                textcolor=(0.8, 0.8, 0.8),
315                label=bui.Lstr(resource='teamNamesColorText'),
316            )
317            assert bui.app.classic is not None
318            if not bui.app.classic.accounts.have_pro():
319                bui.imagewidget(
320                    parent=self.root_widget,
321                    size=(30, 30),
322                    position=(95, 202 + y_offs),
323                    texture=bui.gettexture('lock'),
324                    draw_controller=self._custom_colors_names_button,
325                )
326        else:
327            self._custom_colors_names_button = None
328
329        # Shuffle.
330        def _cb_callback(val: bool) -> None:
331            self._do_randomize_val = val
332            cfg = bui.app.config
333            cfg[self._pvars.config_name + ' Playlist Randomize'] = (
334                self._do_randomize_val
335            )
336            cfg.commit()
337
338        if show_shuffle_check_box:
339            self._shuffle_check_box = bui.checkboxwidget(
340                parent=self.root_widget,
341                position=(110, 200),
342                scale=1.0,
343                size=(250, 30),
344                autoselect=True,
345                text=bui.Lstr(resource=f'{self._r}.shuffleGameOrderText'),
346                maxwidth=300,
347                textcolor=(0.8, 0.8, 0.8),
348                value=self._do_randomize_val,
349                on_value_change_call=_cb_callback,
350            )
351
352        # Show tutorial.
353        show_tutorial = bool(bui.app.config.get('Show Tutorial', True))
354
355        def _cb_callback_2(val: bool) -> None:
356            cfg = bui.app.config
357            cfg['Show Tutorial'] = val
358            cfg.commit()
359
360        self._show_tutorial_check_box = bui.checkboxwidget(
361            parent=self.root_widget,
362            position=(110, 151),
363            scale=1.0,
364            size=(250, 30),
365            autoselect=True,
366            text=bui.Lstr(resource=f'{self._r}.showTutorialText'),
367            maxwidth=300,
368            textcolor=(0.8, 0.8, 0.8),
369            value=show_tutorial,
370            on_value_change_call=_cb_callback_2,
371        )
372
373        # Grumble: current autoselect doesn't do a very good job
374        # with checkboxes.
375        if self._custom_colors_names_button is not None:
376            for btn in bottom_row_buttons:
377                bui.widget(
378                    edit=btn, down_widget=self._custom_colors_names_button
379                )
380            if show_shuffle_check_box:
381                bui.widget(
382                    edit=self._custom_colors_names_button,
383                    down_widget=self._shuffle_check_box,
384                )
385                bui.widget(
386                    edit=self._shuffle_check_box,
387                    up_widget=self._custom_colors_names_button,
388                )
389            else:
390                bui.widget(
391                    edit=self._custom_colors_names_button,
392                    down_widget=self._show_tutorial_check_box,
393                )
394                bui.widget(
395                    edit=self._show_tutorial_check_box,
396                    up_widget=self._custom_colors_names_button,
397                )
398
399        self._ok_button = bui.buttonwidget(
400            parent=self.root_widget,
401            position=(70, 44),
402            size=(200, 45),
403            scale=1.8,
404            text_res_scale=1.5,
405            on_activate_call=self._on_ok_press,
406            autoselect=True,
407            label=bui.Lstr(
408                resource=(
409                    'okText'
410                    if self._playlist_select_context is not None
411                    else 'playText'
412                )
413            ),
414        )
415
416        bui.widget(
417            edit=self._ok_button, up_widget=self._show_tutorial_check_box
418        )
419
420        bui.containerwidget(
421            edit=self.root_widget,
422            start_button=self._ok_button,
423            cancel_button=self._cancel_button,
424            selected_child=self._ok_button,
425        )
426
427        # Update now and once per second.
428        self._update_timer = bui.AppTimer(
429            1.0, bui.WeakCall(self._update), repeat=True
430        )
431        self._update()
432
433    def _custom_colors_names_press(self) -> None:
434        from bauiv1lib.account import show_sign_in_prompt
435        from bauiv1lib.teamnamescolors import TeamNamesColorsWindow
436        from bauiv1lib.purchase import PurchaseWindow
437
438        plus = bui.app.plus
439        assert plus is not None
440
441        assert bui.app.classic is not None
442        if not bui.app.classic.accounts.have_pro():
443            if plus.get_v1_account_state() != 'signed_in':
444                show_sign_in_prompt()
445            else:
446                PurchaseWindow(items=['pro'])
447            self._transition_out()
448            return
449        assert self._custom_colors_names_button
450        TeamNamesColorsWindow(
451            scale_origin=(
452                self._custom_colors_names_button.get_screen_space_center()
453            )
454        )
455
456    def _does_target_playlist_exist(self) -> bool:
457        if self._playlist == '__default__':
458            return True
459        return self._playlist in bui.app.config.get(
460            self._pvars.config_name + ' Playlists', {}
461        )
462
463    def _update(self) -> None:
464        # All we do here is make sure our targeted playlist still exists,
465        # and close ourself if not.
466        if not self._does_target_playlist_exist():
467            self._transition_out()
468
469    def _transition_out(self, transition: str = 'out_scale') -> None:
470        if not self._transitioning_out:
471            self._transitioning_out = True
472            bui.containerwidget(edit=self.root_widget, transition=transition)
473
474    @override
475    def on_popup_cancel(self) -> None:
476        bui.getsound('swish').play()
477        self._transition_out()
478
479    def _on_cancel_press(self) -> None:
480        self._transition_out()
481
482    def _on_ok_press(self) -> None:
483        # no-op if our underlying widget is dead or on its way out.
484        if not self.root_widget or self.root_widget.transitioning_out:
485            return
486
487        # Disallow if our playlist has disappeared.
488        if not self._does_target_playlist_exist():
489            return
490
491        # Disallow if we have no unlocked games.
492        if not self._have_at_least_one_owned:
493            bui.getsound('error').play()
494            bui.screenmessage(
495                bui.Lstr(resource='playlistNoValidGamesErrorText'),
496                color=(1, 0, 0),
497            )
498            return
499
500        cfg = bui.app.config
501        cfg[self._pvars.config_name + ' Playlist Selection'] = self._playlist
502
503        # Head back to the gather window in playlist-select mode or
504        # start the game in regular mode.
505        if self._playlist_select_context is not None:
506            # from bauiv1lib.gather import GatherWindow
507
508            if self._sessiontype is bs.FreeForAllSession:
509                typename = 'ffa'
510            elif self._sessiontype is bs.DualTeamSession:
511                typename = 'teams'
512            else:
513                raise RuntimeError('Only teams and ffa currently supported')
514            cfg['Private Party Host Session Type'] = typename
515            bui.getsound('gunCocking').play()
516
517            self._transition_out(transition='out_left')
518            if self._delegate is not None:
519                self._delegate.on_play_options_window_run_game()
520        else:
521            bui.fade_screen(False, endcall=self._run_selected_playlist)
522            bui.lock_all_input()
523            self._transition_out(transition='out_left')
524            if self._delegate is not None:
525                self._delegate.on_play_options_window_run_game()
526
527        cfg.commit()
528
529    def _run_selected_playlist(self) -> None:
530        bui.unlock_all_input()
531        try:
532            bs.new_host_session(self._sessiontype)
533        except Exception:
534            from bascenev1lib import mainmenu
535
536            logging.exception('Error running session %s.', self._sessiontype)
537
538            # Drop back into a main menu session.
539            bs.new_host_session(mainmenu.MainMenuSession)