removed uml diagrams

[pmachines.git] / ya2 / lib / p3d / gui.py

from inspect import getmro
from panda3d.core import TextNode, Texture
from direct.gui.DirectGuiGlobals import FLAT, ENTER, EXIT, DISABLED, NORMAL, \
    B1PRESS
from direct.showbase.DirectObject import DirectObject
from direct.gui.DirectButton import DirectButton
from direct.gui.DirectCheckButton import DirectCheckButton
from direct.gui.DirectOptionMenu import DirectOptionMenu
from direct.gui.OnscreenImage import OnscreenImage
from direct.gui.DirectSlider import DirectSlider
from direct.gui.DirectEntry import DirectEntry, ENTRY_FOCUS_STATE
from direct.gui.DirectLabel import DirectLabel
from direct.gui.DirectFrame import DirectFrame
from direct.gui.OnscreenText import OnscreenText
from direct.gui.DirectScrolledFrame import DirectScrolledFrame
from ya2.observer import Subject
# from ya2.lib.ivals import Seq, Wait, PosIval, Func


class CommonBase:

    def set_widget(self):
        from ya2.lib.gui import Frame, Slider, Btn, Label, OptionMenu, \
            CheckBtn, Entry, Img, Text, ScrolledFrame
        from ya2.lib.p3d.widget import FrameMixin, SliderMixin, BtnMixin, \
            OptionMenuMixin, CheckBtnMixin, EntryMixin, ImgMixin, \
            ScrolledFrameMixin
        self.__class__ = self.__class__  # for pylint
        libwdg2wdg = {
            FrameMixin: [Frame],
            ScrolledFrameMixin: [ScrolledFrame],
            SliderMixin: [Slider],
            BtnMixin: [Btn, Label],
            OptionMenuMixin: [OptionMenu],
            CheckBtnMixin: [CheckBtn],
            EntryMixin: [Entry],
            ImgMixin: [Img, Text]}
        for libwdg, wdgcls in libwdg2wdg.items():
            if any(cls in getmro(self.__class__) for cls in wdgcls):
                par_cls = libwdg
        clsname = self.__class__.__name__ + 'Widget'
        self.__class__ = type(clsname, (self.__class__, par_cls), {})
        self.init(self)
        if not hasattr(self, 'bind'): return
        bind_args = [(ENTER, self.on_wdg_enter), (EXIT, self.on_wdg_exit)]
        list(map(lambda args: self.bind(*args), bind_args))

    def set_enter_transition(self):
        start_pos = self.get_pos()
        pos = self.pos - (3.6, 0)
        self.set_pos((pos.x, 1, pos.y))
        Seq(
            Wait(abs(pos.y - 1) / 4),
            PosIval(self.get_np(), .5, start_pos)
        ).start()

    def set_exit_transition(self, destroy):
        start_pos = self.get_pos()
        end_pos = (self.pos.x + 3.6, 1, self.pos.y)
        seq = Seq(
            Wait(abs(self.pos.y - 1) / 4),
            PosIval(self.get_np(), .5, end_pos),
            Func(self.destroy if destroy else self.hide))
        if not destroy: seq += Func(self.set_pos, start_pos)
        seq.start()

    def translate(self):
        if hasattr(self, 'bind_transl'): self.wdg['text'] = self.bind_transl


class P3dImg(CommonBase):

    def __init__(self, filepath, pos=(0, 0), scale=1.0, background=False,
                 foreground=False, parent=None):
        self.img = OnscreenImage(
            filepath, pos=(pos[0], 1, pos[1]), scale=scale, parent=parent)
        if background: self.img.set_bin('background', 10)
        alpha_formats = [12]  # panda3d.core.texture.Frgba
        if self.img.get_texture().get_format() in alpha_formats:
            self.img.set_transparency(True)
        if foreground: self.img.set_bin('gui-popup', 50)

    def reparent_to(self, node): return self.img.reparent_to(node)
    def show(self): return self.img.show()
    def hide(self): return self.img.hide()
    def set_shader(self, shader): return self.img.set_shader(shader)
    def set_shader_input(self, name, val):
        return self.img.set_shader_input(name, val)
    def set_texture(self, texturestage, texture):
        return self.img.set_texture(texturestage, texture)

    def set_exit_transition(self, destroy):
        start_pos = self.get_pos()
        end_pos = (self.pos.x + 3.6, 1, self.pos.y)
        seq = Seq(
            Wait(abs(self.pos.y - 1) / 4),
            PosIval(self.get_np(), .5, end_pos),
            Func(self.destroy if destroy else self.hide))
        if not destroy: seq += Func(self.set_pos, (start_pos[0], start_pos[2]))
        seq.start()

    def set_pos(self, pos): return self.img.set_pos(pos[0], 1, pos[1])

    def get_pos(self, pos=None): return self.img.get_pos(*[pos] if pos else [])

    @property
    def parent(self): return self.img.get_parent()

    @property
    def hidden(self): return self.img.is_hidden()

    def set_transparent(self): return self.img.set_transparency(True)

    def destroy(self): self.img = self.img.destroy()


# class P3dBase(CommonBase):

#     def __init__(self, tra_src=None, tra_tra=None):
#         # self.text_src_tra = None  # it breaks the gui
#         if tra_src and tra_tra: self.bind_tra(tra_src, tra_tra)

#     def set_pos(self, pos): return self.wdg.set_pos(pos)
#     def show(self): return self.wdg.show()
#     def hide(self): return self.wdg.hide()

#     def bind_tra(self, text_src, text_transl):
#         # text_transl is not used, anyway we need it since we have this kind of
#         # use: self.bind_transl('example str', _('example str'))
#         # this allows to change translations on the fly keeping the source
#         # text for remapping it later
#         # TODO: try reverse mapping? i.e. retrieve the src string from the
#         # translated one
#         self.text_src_tra = text_src
#         self.text_tra_tra = text_transl
#         tra = lambda self: _(self.text_tra_tra)
#         self.__class__.bind_transl = property(tra)
#         self['text'] = self.bind_transl

#     def get_pos(self, pos=None):
#         return self.wdg.get_pos(*[pos] if pos else [])

#     def __setitem__(self, key, value): self.wdg[key] = value

#     def __getitem__(self, key): return self.wdg[key]

#     def get_np(self): return self.wdg

#     @property
#     def hidden(self): return self.wdg.is_hidden()

#     def destroy(self): self.wdg.destroy()


# class P3dAbs(P3dBase):

#     def get_value(self): return self.wdg.getValue()
#     def initialiseoptions(self): return self.wdg.initialiseoptions()
#     def set_z(self, val): return self.wdg.set_z(val)
#     def set_shader(self, shader): return self.wdg.set_shader(shader)
#     def set_shader_input(self, name, val):
#         return self.wdg.set_shader_input(name, val)
#     def set_transparency(self, val): return self.wdg.set_transparency(val)
#     def bind(self, evt, mth): return self.wdg.bind(evt, mth)

#     def attachNewNode(self, gui_itm, sort_order):
#         # it won't work if we name it attach_node. hopefully this will be
#         # possible when we'll use decorators in place of mixins
#         return self.wdg.attachNewNode(gui_itm, sort_order)

#     @property
#     def is_enabled(self): return self.wdg['state'] != DISABLED


# class P3dBtn(P3dAbs):

#     def __init__(
#             self, text='', parent=None, pos=(0, 0), scale=(1, 1),
#             cmd=None, frame_size=(-1, 1, -1, 1), click_snd=None,
#             text_fg=(1, 1, 1, 1), frame_col=(1, 1, 1, 1), text_font=None,
#             over_snd=None, extra_args=None, frame_texture=None, img=None,
#             tra_src=None, tra_tra=None, text_scale=1.0):
#         str2par = {'bottomcenter': base.a2dBottomCenter}
#         parent = str2par.get(parent, parent)
#         extra_args = extra_args or []
#         self.wdg = DirectButton(
#             text=text, parent=parent, pos=(pos[0], 1, pos[1]),
#             scale=(scale[0], 1, scale[1]), command=cmd,
#             frameSize=frame_size, clickSound=click_snd, text_fg=text_fg,
#             frameColor=frame_col, text_font=text_font, rolloverSound=over_snd,
#             extraArgs=extra_args, frameTexture=frame_texture, image=img,
#             text_scale=text_scale)
#         P3dAbs.__init__(self, tra_src, tra_tra)
#         self['relief'] = FLAT
#         args = [(ENTER, self._on_enter), (EXIT, self._on_exit)]
#         list(map(lambda args: self.bind(*args), args))

#     def _on_enter(self, pos): pass  # pos comes from mouse

#     def _on_exit(self, pos): pass  # pos comes from mouse

#     # we add these with the mixins
#     # def enable(self): self['state'] = NORMAL

#     # def disable(self): self['state'] = DISABLED


# class P3dSlider(P3dAbs):

#     def __init__(
#             self, parent=None, pos=(0, 0), scale=1, val=0,
#             frame_col=(1, 1, 1, 1), thumb_frame_col=(1, 1, 1, 1),
#             cmd=None, range_=(0, 1), tra_src=None, tra_tra=None):
#         self.wdg = DirectSlider(
#             parent=parent, pos=(pos[0], 1, pos[1]), scale=scale, value=val,
#             frameColor=frame_col, thumb_frameColor=thumb_frame_col,
#             command=cmd, range=range_)
#         P3dAbs.__init__(self, tra_src, tra_tra)


# class P3dCheckBtn(P3dAbs):

#     def __init__(
#             self, pos=(0, 0), text='', indicator_val=False,
#             indicator_frame_col=(1, 1, 1, 1), frame_col=(1, 1, 1, 1),
#             scale=(1, 1, 1), click_snd=None, over_snd=None,
#             text_fg=(1, 1, 1, 1), text_font=None, cmd=None, tra_src=None,
#             tra_tra=None):
#         self.wdg = DirectCheckButton(
#             pos=(pos[0], 1, pos[1]), text=text, indicatorValue=indicator_val,
#             indicator_frameColor=indicator_frame_col,
#             frameColor=frame_col, scale=scale, clickSound=click_snd,
#             rolloverSound=over_snd, text_fg=text_fg, text_font=text_font,
#             command=cmd)
#         P3dAbs.__init__(self, tra_src, tra_tra)


# class P3dOptionMenu(P3dAbs):

#     def __init__(
#             self, text='', items=None, pos=(0, 0), scale=(1, 1, 1),
#             initialitem='', cmd=None, frame_size=(-1, 1, -1, 1),
#             click_snd=None, over_snd=None, text_may_change=False,
#             text_fg=(1, 1, 1, 1), item_frame_col=(1, 1, 1, 1),
#             frame_col=(1, 1, 1, 1), highlight_col=(1, 1, 1, 1),
#             text_scale=.05, popup_marker_col=(1, 1, 1, 1),
#             item_relief=None, item_text_font=None, text_font=None,
#             tra_src=None, tra_tra=None):
#         items = items or []
#         self.wdg = DirectOptionMenu(
#             text=text, items=items, pos=(pos[0], 1, pos[1]), scale=scale,
#             initialitem=initialitem, command=cmd, frameSize=frame_size,
#             clickSound=click_snd, rolloverSound=over_snd,
#             textMayChange=text_may_change, text_fg=text_fg,
#             item_frameColor=item_frame_col, frameColor=frame_col,
#             highlightColor=highlight_col, text_scale=text_scale,
#             popupMarker_frameColor=popup_marker_col,
#             item_relief=item_relief, item_text_font=item_text_font,
#             text_font=text_font)
#         P3dAbs.__init__(self, tra_src, tra_tra)

#     def set(self, idx, f_cmd=1): return self.wdg.set(idx, f_cmd)

#     @property
#     def curr_val(self): return self.wdg.get()

#     @property
#     def curr_idx(self): return self.wdg.selectedIndex


# class P3dEntry(P3dAbs, DirectObject, Subject):

#     def __init__(
#             self, scale=.05, pos=(0, 0), entry_font=None, width=12,
#             frame_col=(1, 1, 1, 1), initial_text='', obscured=False,
#             cmd=None, focus_in_cmd=None, focus_in_args=None,
#             focus_out_cmd=None, focus_out_args=None, parent=None,
#             tra_src=None, tra_tra=None, text_fg=(1, 1, 1, 1), on_tab=None,
#             on_click=None):
#         self.__focused = False
#         self.__focus_in_cmd = focus_in_cmd
#         self.__focus_out_cmd = focus_out_cmd
#         DirectObject.__init__(self)
#         Subject.__init__(self)
#         focus_in_args = focus_in_args or []
#         focus_out_args = focus_out_args or []
#         self.wdg = DirectEntry(
#             scale=scale, pos=(pos[0], 1, pos[1]), entryFont=entry_font,
#             width=width, frameColor=frame_col, initialText=initial_text,
#             obscured=obscured, command=cmd, focusInCommand=self._focus_in_cmd,
#             focusInExtraArgs=focus_in_args,
#             focusOutCommand=self._focus_out_cmd,
#             focusOutExtraArgs=focus_out_args, parent=parent,
#             text_fg=text_fg)
#         P3dAbs.__init__(self, tra_src, tra_tra)
#         if on_tab:
#             self.on_tab_cb = on_tab
#             self.accept('tab-up', self.on_tab)
#         if on_click: self.wdg.bind(B1PRESS, on_click)

#     def set(self, txt): return self.wdg.set(txt)

#     def _focus_in_cmd(self, *args):
#         self.__focused = True
#         if self.__focus_in_cmd: self.__focus_in_cmd(*args)
#         self.notify('on_entry_enter')

#     def _focus_out_cmd(self, *args):
#         self.__focused = False
#         if self.__focus_out_cmd: self.__focus_out_cmd(*args)
#         self.notify('on_entry_exit')

#     def on_tab(self):
#         if self.wdg['focus'] == ENTRY_FOCUS_STATE: self.on_tab_cb()

#     @property
#     def focused(self): return self.__focused

#     @property
#     def text(self): return self.wdg.get()

#     def enter_text(self, txt):
#         return self.wdg.enterText(txt)

#     def enable(self): self['state'] = NORMAL

#     def disable(self): self['state'] = DISABLED

#     def destroy(self):
#         self.ignore('tab-up')
#         self.on_tab_cb = None
#         Subject.destroy(self)
#         P3dAbs.destroy(self)


# class P3dLabel(P3dAbs):

#     def __init__(
#             self, text='', pos=(0, 0), parent=None, text_wordwrap=12,
#             text_align=None, text_fg=(1, 1, 1, 1), text_font=None, scale=.05,
#             frame_col=(1, 1, 1, 1), tra_src=None, tra_tra=None, hpr=(0, 0, 0)):
#         self.wdg = DirectLabel(
#             text=text, pos=(pos[0], 1, pos[1]), parent=parent,
#             text_wordwrap=text_wordwrap, text_align=text_align,
#             text_fg=text_fg, text_font=text_font, scale=scale,
#             frameColor=frame_col, hpr=hpr)
#         P3dAbs.__init__(self, tra_src, tra_tra)

#     def set_bin(self, bin_name, priority): return self.wdg.set_bin(bin_name, priority)

#     def set_x(self, x): return self.wdg.set_x(x)

#     def set_alpha_scale(self, alpha): return self.wdg.set_alpha_scale(alpha)


# class P3dTxt(P3dBase):

#     def __init__(
#             self, txt='', pos=(0, 0), scale=.05, wordwrap=12, parent=None,
#             fg=(1, 1, 1, 1), font=None, align=None, tra_src=None,
#             tra_tra=None):
#         str2par = {'bottomleft': base.a2dBottomLeft,
#                    'bottomright': base.a2dBottomRight,
#                    'leftcenter': base.a2dLeftCenter}
#         str2al = {'left': TextNode.A_left, 'right': TextNode.A_right,
#                   'center': TextNode.A_center}
#         if parent and parent in str2par: parent = str2par[parent]
#         if align: align = str2al[align]
#         self.wdg = OnscreenText(
#             text=txt, pos=pos, scale=scale, wordwrap=wordwrap,
#             parent=parent, fg=fg, font=font, align=align)
#         P3dBase.__init__(self, tra_src, tra_tra)

#     def set_r(self, r): return self.wdg.set_r(r)


# class P3dFrame(P3dAbs):

#     def __init__(self, frame_size=(-1, 1, -1, 1), frame_col=(1, 1, 1, 1),
#                  pos=(0, 0), parent=None, texture_coord=False):
#         P3dAbs.__init__(self)
#         self.wdg = DirectFrame(frameSize=frame_size, frameColor=frame_col,
#                                pos=(pos[0], 1, pos[1]), parent=parent)
#         if texture_coord: self.wdg['frameTexture'] = Texture()


# class P3dScrolledFrame(P3dAbs):

#     def __init__(
#             self, frame_sz=(-1, 1, -1, 1), canvas_sz=(0, 1, 0, 1),
#             scrollbar_width=.05, frame_col=(1, 1, 1, 1),
#             pos=(0, 0), parent='topleft'):
#         P3dAbs.__init__(self)
#         par2p3d = {'topleft': base.a2dTopLeft}
#         if parent and parent in par2p3d: parent = par2p3d[parent]
#         self.wdg = DirectScrolledFrame(
#             frameSize=frame_sz,
#             canvasSize=canvas_sz,
#             scrollBarWidth=scrollbar_width,
#             frameColor=frame_col,
#             pos=(pos[0], 1, pos[1]),
#             parent=parent)

#     @property
#     def canvas(self): return self.wdg.getCanvas()