Difficult-Rocket/mods/dr_game/menu.py

#  -------------------------------
#  Difficult Rocket
#  Copyright © 2020-2023 by shenjackyuanjie 3695888@qq.com
#  All rights reserved
#  -------------------------------

from typing import Optional, Tuple
from pyglet.graphics import Batch, Group

from Difficult_Rocket.client import ClientWindow
from Difficult_Rocket.api.screen import BaseScreen
from Difficult_Rocket.main import Game
from Difficult_Rocket.gui.widget.button import PressTextButton, MinecraftWikiButtonTheme, ButtonThemeOptions, BaseButtonTheme

# from . import DR_mod_runtime


class Menu(BaseScreen):
    """
    DR game 的 菜单
    """

    name = "DR_game_menu"

    def __init__(self, main_window: ClientWindow):
        super().__init__(main_window)
        self.main_batch = Batch()
        self.main_group = Group(parent=main_window.main_group, order=1)

        # 占位, 高二看看能不能咕出来点啥 (20230911)
        # 欸呀, 正好是 911 纪念日哦
        # 好, 高二 第一学期 期末都考完了, 我过来做测试了 (20240119)

        self.wiki_button1 = PressTextButton(
            x=200,
            y=300,
            width=150,
            height=30,
            text="wiki button1",
            batch=self.main_batch,
            group=self.main_group,
            draw_theme=MinecraftWikiButtonTheme,
            dict_theme={"pop_out": True},
        )
        self.wiki_button2 = PressTextButton(
            x=400,
            y=300,
            width=150,
            height=30,
            text="wiki button2",
            batch=self.main_batch,
            group=self.main_group,
            draw_theme=MinecraftWikiButtonTheme,
            dict_theme={"pop_out": False},
        )
        self.wiki_button3 = PressTextButton(
            x=200,
            y=250,
            width=150,
            height=30,
            text="wiki button3",
            batch=self.main_batch,
            group=self.main_group,
            draw_theme=MinecraftWikiButtonTheme,
            dict_theme={"pop_out": True, "drag_list": True},
        )
        self.wiki_button4 = PressTextButton(
            x=400,
            y=250,
            width=150,
            height=30,
            text="wiki button4",
            batch=self.main_batch,
            group=self.main_group,
            draw_theme=MinecraftWikiButtonTheme,
            dict_theme={"pop_out": False, "drag_list": True},
        )
        self.button3 = PressTextButton(
            x=200,
            y=200,
            width=150,
            height=30,
            text="imgui button",
            batch=self.main_batch,
            group=self.main_group,
        )
        self.enter_ship_editor_button = PressEnterShipEditorButton(
            window=main_window,
            x=100,
            y=100,
            width=150,
            height=30,
            text="进入编辑器",
            batch=self.main_batch,
            group=self.main_group,
        )
        main_window.push_handlers(self.wiki_button1)
        main_window.push_handlers(self.wiki_button2)
        main_window.push_handlers(self.wiki_button3)
        main_window.push_handlers(self.wiki_button4)
        main_window.push_handlers(self.button3)
        main_window.push_handlers(self.enter_ship_editor_button)

    def on_draw(self, dt: float, window: ClientWindow):
        self.main_batch.draw()

class PressEnterShipEditorButton(PressTextButton):
    def __init__(
        self,
        window: ClientWindow,
        x: int,
        y: int,
        width: int,
        height: int,
        text: str,
        batch: Optional[Batch] = None,
        group: Optional[Group] = None,
        theme: Optional[ButtonThemeOptions] = None,
        draw_theme: Optional[type(BaseButtonTheme)] = None,
        dict_theme: Optional[dict] = None,):
        super().__init__(x,y,width,height,text,batch,group,theme,draw_theme,dict_theme)
        self.window=window
        
    #from lib_not_dr import loggers
    def on_mouse_release(self, x, y, buttons, modifiers):
        if self.pressed and (x, y) in self:
            if self.draw_theme:
                self.draw_theme.on_disable(self)
            else:
                self.back_rec.color = self.touched_color
            self.pressed = False
                    
            from .sr1_ship import SR1ShipRender
            self.window.remove_sub_screen("DR_game_menu")
            self.window.add_sub_screen("SR1_ship", SR1ShipRender)
            #logger.info("added SR1_ship screen", tag="dr_game")