[pmachines.git] / pmachines / editor / inspector.py

from collections import namedtuple
from logging import info
from panda3d.core import Texture, TextNode, LPoint3f
from direct.gui.OnscreenImage import OnscreenImage
from direct.gui.DirectGui import DirectButton, DirectFrame, DirectEntry, DirectOptionMenu, OkDialog
from direct.gui.DirectGuiGlobals import FLAT, NORMAL
from direct.gui.OnscreenText import OnscreenText
from direct.showbase.DirectObject import DirectObject
from pmachines.items.item import FixedStrategy, StillStrategy
from pmachines.items.box import HitStrategy
from pmachines.items.domino import DownStrategy, UpStrategy
from pmachines.editor.augmented_frame import AugmentedDirectFrame
from ya2.utils.gui.base_page import DirectOptionMenuTestable
from ya2.utils.gfx import DirectGuiMixin


class Inspector(DirectObject):

    def __init__(self, item, all_items, pos_mgr, strategy_items):
        super().__init__()
        self.__item = item
        self.__all_items = all_items
        self.__pos_mgr = pos_mgr
        self._font = base.loader.load_font(
            'assets/fonts/Hanken-Book.ttf')
        self._font.clear()
        self._font.set_pixels_per_unit(60)
        self._font.set_minfilter(Texture.FTLinearMipmapLinear)
        self._font.set_outline((0, 0, 0, 1), .8, .2)
        self._common = {
            'scale': .046,
            'text_font': self._font,
            'text_fg': (.9, .9, .9, 1),
            'relief': FLAT,
            'frameColor': (.4, .4, .4, .14),
            'rolloverSound': loader.load_sfx(
                'assets/audio/sfx/rollover.ogg'),
            'clickSound': loader.load_sfx(
                'assets/audio/sfx/click.ogg')}
        tooltip_args = self._common['text_font'], self._common['scale'], self._common['text_fg']
        w, h = .8, 1.04
        self._frm = AugmentedDirectFrame(frameColor=(.4, .4, .4, .06),
                                         frameSize=(0, w, -h, 0),
                                         parent=base.a2dTopRight,
                                         pos=(-w, 0, 0),
                                         delta_drag=LPoint3f(-w, 0, -h),
                                         collapse_pos=(w - .06, 1, -h + .06),
                                         pos_mgr=pos_mgr,
                                         frame_name='inspector')
        self.__z = -.08
        p = self.__item._np.get_pos()
        r = self.__item._np.get_r()
        s = self.__item._np.get_scale()[0]
        m = self.__item._mass
        restitution = self.__item._restitution
        f = self.__item._friction
        _id = ''
        if 'id' in self.__item.json:
            _id = self.__item.json['id']
        _strategy = ''
        if 'strategy' in self.__item.json:
            _strategy = self.__item.json['strategy']
        _strategy_args = ''
        if 'strategy_args' in self.__item.json:
            _strategy_args = ' '.join(map(str, self.__item.json['strategy_args']))
        t, pos_entry = self.__add_row('position', _('position'), f'{round(p.x, 3)} {round(p.z, 3)}', self.on_edit_position, _('position (e.g. 0.1 2.3 4.5)'))
        t, rot_entry = self.__add_row('roll', _('roll'), f'{round(r, 3)}', self.on_edit_roll, _('roll (e.g. 90)'))
        t, scale_entry = self.__add_row('scale', _('scale'), f'{round(s, 3)}', self.on_edit_scale, _('scale (e.g. 1.2)'))
        t, mass_entry = self.__add_row('mass', _('mass'), f'{round(m, 3)}', self.on_edit_mass, _('mass (default 1; 0 if fixed)'))
        t, restitution_entry = self.__add_row('restitution', _('restitution'), f'{round(restitution, 3)}', self.on_edit_restitution, _('restitution (default 0.5)'))
        t, friction_entry = self.__add_row('friction', _('friction'), f'{round(f, 3)}', self.on_edit_friction, _('friction (default 0.5)'))
        t, id_entry = self.__add_row('id', _('id'), _id, self.on_edit_id, _('id of the item (for the strategies)'))
        # item_modules = glob('pmachines/items/*.py')
        # item_modules = [basename(i)[:-3] for i in item_modules]
        # strategy_items = ['']
        # for item_module in item_modules:
        #     mod_name = 'pmachines.items.' + item_module
        #     for member in import_module(mod_name).__dict__.values():
        #         if isclass(member) and issubclass(member, ItemStrategy) and \
        #                 member != ItemStrategy:
        #             strategy_items = list(set(strategy_items + [member.__name__]))
        strategy_names = [s.__name__ for s in strategy_items]
        t, strategy_entry = self.__add_row_option(_('strategy'), _strategy, strategy_names, self.on_edit_strategy, _('the strategy of the item'))

        def strategy_set(comps):
            strategy_labels = [f'inspector_strategy_{i.lower()}' for i in strategy_names]
            for i in strategy_labels:
                if i in self.__pos_mgr:
                    del self.__pos_mgr[i]
            for l, b in zip(strategy_labels, comps):
                b.__class__ = type('DirectFrameMixed', (DirectFrame, DirectGuiMixin), {})
                p = b.pos_pixel()
                self.__pos_mgr[l] = (p[0] + 5, p[1])
        strategy_entry._show_cb = strategy_set
        p = strategy_entry.pos_pixel()
        self.__pos_mgr['editor_inspector_strategy'] = (p[0] + 5, p[1])

        t, strategy_args_entry = self.__add_row('strategy_args', _('strategy_args'), _strategy_args, self.on_edit_strategy_args, _('the arguments of the strategy'))
        fields = ['position', 'roll', 'scale', 'mass', 'restitution', 'friction', 'id', 'strategy', 'strategy_args']
        Entries = namedtuple('Entries', fields)
        self.__entries = Entries(pos_entry, rot_entry, scale_entry, mass_entry, restitution_entry, friction_entry, id_entry, strategy_entry, strategy_args_entry)
        def load_images_btn(path, col):
            colors = {
                'gray': [
                    (.6, .6, .6, 1),  # ready
                    (1, 1, 1, 1), # press
                    (.8, .8, .8, 1), # rollover
                    (.4, .4, .4, .4)],
                'green': [
                    (.1, .68, .1, 1),
                    (.1, 1, .1, 1),
                    (.1, .84, .1, 1),
                    (.4, .1, .1, .4)]}[col]
            return [self.__load_img_btn(path, col) for col in colors]
        fcols = (.4, .4, .4, .14), (.3, .3, .3, .05)
        b = DirectButton(
            image=load_images_btn('exitRight', 'gray'), scale=.05,
            pos=(.06, 1, -h + .06),
            parent=self._frm, command=self.destroy, state=NORMAL, relief=FLAT,
            frameColor=fcols[0],
            rolloverSound=loader.load_sfx('assets/audio/sfx/rollover.ogg'),
            clickSound=loader.load_sfx('assets/audio/sfx/click.ogg'))
        b.__class__ = type('DirectButtonMixed', (DirectButton, DirectGuiMixin), {})
        pos_mgr['editor_inspector_close'] = b.pos_pixel()
        b.set_tooltip(_('Close'), *tooltip_args)
        self.accept('item-rototranslated', self.__on_item_rototranslated)
        b = DirectButton(
            image=load_images_btn('trashcan', 'gray'), scale=.05,
            pos=(.18, 1, -h + .06),
            parent=self._frm, command=self.__delete_item, state=NORMAL, relief=FLAT,
            frameColor=fcols[0],
            rolloverSound=loader.load_sfx('assets/audio/sfx/rollover.ogg'),
            clickSound=loader.load_sfx('assets/audio/sfx/click.ogg'))
        b.__class__ = type('DirectButtonMixed', (DirectButton, DirectGuiMixin), {})
        pos_mgr['editor_inspector_delete'] = b.pos_pixel()
        b.set_tooltip(_('Delete the item'), *tooltip_args)

    def __add_row(self, id_, label, text, callback, tooltip):
        tw = 10
        tooltip_args = self._common['text_font'], self._common['scale'], self._common['text_fg']
        t = OnscreenText(
            label,
            pos=(.03, self.__z), parent=self._frm,
            font=self._common['text_font'],
            scale=self._common['scale'],
            fg=self._common['text_fg'],
            wordwrap=20, align=TextNode.ALeft)
        e = DirectEntry(
            scale=self._common['scale'],
            pos=(.30, 1, self.__z),
            entryFont=self._font,
            width=tw,
            cursorKeys=True,
            frameColor=self._common['frameColor'],
            initialText=text,
            parent=self._frm,
            text_fg=self._common['text_fg'],
            command=callback)
        e.__class__ = type('DirectEntryMixed', (DirectEntry, DirectGuiMixin), {})
        e.set_tooltip(tooltip, *tooltip_args)
        self.__pos_mgr[f'editor_inspector_{id_}'] = e.pos_pixel()
        self.__z -= .1
        return t, e

    def __add_row_option(self, label, text, items, callback, tooltip):
        tooltip_args = self._common['text_font'], self._common['scale'], self._common['text_fg']
        t = OnscreenText(
            label,
            pos=(.03, self.__z), parent=self._frm,
            font=self._common['text_font'],
            scale=self._common['scale'],
            fg=self._common['text_fg'],
            wordwrap=20, align=TextNode.ALeft)
        e = DirectOptionMenuTestable(
            scale=self._common['scale'],
            initialitem=text,
            pos=(.30, 1, self.__z),
            items=items,
            parent=self._frm,
            command=callback,
            state=NORMAL,
            relief=FLAT,
            item_relief=FLAT,
            frameColor=self._common['frameColor'],
            item_frameColor=self._common['frameColor'],
            popupMenu_frameColor=self._common['frameColor'],
            popupMarker_frameColor=self._common['frameColor'],
            text_font=self._font,
            text_fg=self._common['text_fg'],
            highlightColor=(.9, .9, .9, .9),
            item_text_font=self._font,
            item_text_fg=self._common['text_fg'],
            rolloverSound=loader.load_sfx('assets/audio/sfx/rollover.ogg'),
            clickSound=loader.load_sfx('assets/audio/sfx/click.ogg'))
        e.__class__ = type('DirectOptionMenuMixed', (DirectOptionMenu, DirectGuiMixin), {})
        e.set_tooltip(tooltip, *tooltip_args)
        self.__z -= .1
        return t, e

    def __load_img_btn(self, path, col):
        img = OnscreenImage('assets/images/buttons/%s.dds' % path)
        img.set_transparency(True)
        img.set_color(col)
        img.detach_node()
        return img

    def __on_item_rototranslated(self, np):
        pos = np.get_pos()
        r = np.get_r()
        self.__entries.position.set('%s %s' % (str(round(pos.x, 3)), str(round(pos.z, 3))))
        self.__entries.roll.set('%s' % str(round(r, 3)))
        self.__item.json['position'] = list(pos)
        self.__item.json['roll'] = round(r, 3)

    def __delete_item(self):
        messenger.send('editor-inspector-delete', [self.__item])
        self.destroy()

    @property
    def item(self):
        return self.__item

    def on_edit_position(self, txt):
        x, z = map(float, txt.split())
        self.__item.position = [x, 0, z]

    def on_edit_roll(self, txt):
        self.__item.roll = float(txt)

    def on_edit_scale(self, txt):
        self.__item.scale = float(txt)

    def on_edit_mass(self, txt):
        self.__item.mass = float(txt)

    def on_edit_restitution(self, txt):
        self.__item.restitution = float(txt)

    def on_edit_friction(self, txt):
        self.__item.friction = float(txt)

    def on_edit_id(self, txt):
        self.__item.id = txt

    def on_edit_strategy(self, txt):
        if not txt:
            self.__entries.strategy_args.set('')
            return
        name2class = {
            'StillStrategy': StillStrategy,
            'UpStrategy': UpStrategy,
            'HitStrategy': HitStrategy,
            'DownStrategy': DownStrategy,
            'FixedStrategy': FixedStrategy}
        class_ = name2class[txt]
        args = []
        error = False
        if txt == 'StillStrategy':
            args += [self.__item._np]
        if txt in ['UpStrategy', 'DownStrategy']:
            args += [self.__item._np]
            try:
                args += [float(self.__entries.strategy_args.get())]
            except ValueError:
                error = True
        if txt == 'HitStrategy':
            for item in self.__all_items:
                if item.id == self.__entries.strategy_args.get():
                    args += [item]
            args += [self.__item.node]
            args += [self.__item._world]
        if not error:
            self.__item.strategy = class_(*args)
            self.__item.strategy_json = txt
        else:
            self.__show_error_popup()

    def on_edit_strategy_args(self, txt):
        self.__item.strategy_args_json = txt

    def __show_error_popup(self):
            self.__dialog = OkDialog(dialogName='Strategy args errors',
                                        text=_('There are errors in the strategy args.'),
                                        command=self.__actually_close)
            self.__dialog['frameColor'] = (.4, .4, .4, .14)
            self.__dialog['relief'] = FLAT
            self.__dialog.component('text0')['fg'] = (.9, .9, .9, 1)
            self.__dialog.component('text0')['font'] = self._font
            for b in self.__dialog.buttonList:
                b['frameColor'] = (.4, .4, .4, .14)
                b.component('text0')['fg'] = (.9, .9, .9, 1)
                b.component('text0')['font'] = self._font
                b.component('text1')['fg'] = (.9, .1, .1, 1)
                b.component('text1')['font'] = self._font
                b.component('text2')['fg'] = (.9, .9, .1, 1)
                b.component('text2')['font'] = self._font

    def __actually_close(self, arg):
        self.__entries.strategy.set('')
        self.__entries.strategy_args.set('')
        self.__dialog.cleanup()

    def destroy(self):
        self._frm.destroy()
        self.ignore('item-rototranslated')
        messenger.send('editor-inspector-destroy')


class PixelSpaceInspector(DirectObject):

    def __init__(self, item, all_items):
        info('PixelSpaceInspector')
        super().__init__()
        self.__item = item
        self.__all_items = all_items
        self._font = base.loader.load_font(
            'assets/fonts/Hanken-Book.ttf')
        self._font.clear()
        self._font.set_pixels_per_unit(60)
        self._font.set_minfilter(Texture.FTLinearMipmapLinear)
        self._font.set_outline((0, 0, 0, 1), .8, .2)
        self._common = {
            'scale': .046,
            'text_font': self._font,
            'text_fg': (.9, .9, .9, 1),
            'relief': FLAT,
            'frameColor': (.4, .4, .4, .14),
            'rolloverSound': loader.load_sfx(
                'assets/audio/sfx/rollover.ogg'),
            'clickSound': loader.load_sfx(
                'assets/audio/sfx/click.ogg')}
        w, h = .8, .36
        self._frm = DirectFrame(frameColor=(.4, .4, .4, .06),
                                frameSize=(0, w, -h, 0),
                                parent=base.a2dTopRight,
                                pos=(-w, 0, 0))
        self.__z = -.08
        p = self.__item._np.get_pos()
        _id = ''
        if 'id' in self.__item.json:
            _id = self.__item.json['id']
        t, pos_entry = self.__add_row(_('position'), f'{round(p.x, 3)}, {round(p.z, 3)}', self.on_edit_position)
        t, id_entry = self.__add_row(_('id'), _id, self.on_edit_id)
        fields = ['position', 'id']
        Entries = namedtuple('Entries', fields)
        self.__entries = Entries(pos_entry, id_entry)
        def load_images_btn(path, col):
            colors = {
                'gray': [
                    (.6, .6, .6, 1),  # ready
                    (1, 1, 1, 1), # press
                    (.8, .8, .8, 1), # rollover
                    (.4, .4, .4, .4)],
                'green': [
                    (.1, .68, .1, 1),
                    (.1, 1, .1, 1),
                    (.1, .84, .1, 1),
                    (.4, .1, .1, .4)]}[col]
            return [self.__load_img_btn(path, col) for col in colors]
        fcols = (.4, .4, .4, .14), (.3, .3, .3, .05)
        DirectButton(
            image=load_images_btn('exitRight', 'gray'), scale=.05,
            pos=(.06, 1, -h + .06),
            parent=self._frm, command=self.destroy, state=NORMAL, relief=FLAT,
            frameColor=fcols[0],
            rolloverSound=loader.load_sfx('assets/audio/sfx/rollover.ogg'),
            clickSound=loader.load_sfx('assets/audio/sfx/click.ogg'))
        self.accept('item-rototranslated', self.__on_item_rototranslated)
        DirectButton(
            image=load_images_btn('trashcan', 'gray'), scale=.05,
            pos=(.18, 1, -h + .06),
            parent=self._frm, command=self.__delete_item, state=NORMAL, relief=FLAT,
            frameColor=fcols[0],
            rolloverSound=loader.load_sfx('assets/audio/sfx/rollover.ogg'),
            clickSound=loader.load_sfx('assets/audio/sfx/click.ogg'))

    def __add_row(self, label, text, callback):
        tw = 10
        t = OnscreenText(
            label,
            pos=(.03, self.__z), parent=self._frm,
            font=self._common['text_font'],
            scale=self._common['scale'],
            fg=self._common['text_fg'],
            wordwrap=20, align=TextNode.ALeft)
        e = DirectEntry(
            scale=self._common['scale'],
            pos=(.30, 1, self.__z),
            entryFont=self._font,
            width=tw,
            cursorKeys=True,
            frameColor=self._common['frameColor'],
            initialText=text,
            parent=self._frm,
            text_fg=self._common['text_fg'],
            command=callback)
        self.__z -= .1
        return t, e

    def __load_img_btn(self, path, col):
        img = OnscreenImage('assets/images/buttons/%s.dds' % path)
        img.set_transparency(True)
        img.set_color(col)
        img.detach_node()
        return img

    def __on_item_rototranslated(self, np):
        pos = np.pos2d_pixel()
        self.__entries.position.set('%s %s' % (str(round(pos[0], 3)), str(round(pos[1], 3))))
        self.__item.json['position'] = list(pos)

    def __delete_item(self):
        messenger.send('editor-inspector-delete', [self.__item])
        self.destroy()

    @property
    def item(self):
        return self.__item

    def on_edit_position(self, txt):
        x, z = map(float, txt.split())
        self.__item.position = [x, 0, z]

    def on_edit_id(self, txt):
        self.__item.id = txt

    def __actually_close(self, arg):
        self.__entries.strategy.set('')
        self.__entries.strategy_args.set('')
        self.__dialog.cleanup()

    def destroy(self):
        self._frm.destroy()
        self.ignore('item-rototranslated')
        messenger.send('editor-inspector-destroy')


class WorldSpaceInspector(DirectObject):

    def __init__(self, item, all_items, pos_mgr):
        info('WorldSpaceInspector')
        super().__init__()
        self.__item = item
        self.__all_items = all_items
        self.__pos_mgr = pos_mgr
        self._font = base.loader.load_font(
            'assets/fonts/Hanken-Book.ttf')
        self._font.clear()
        self._font.set_pixels_per_unit(60)
        self._font.set_minfilter(Texture.FTLinearMipmapLinear)
        self._font.set_outline((0, 0, 0, 1), .8, .2)
        self._common = {
            'scale': .046,
            'text_font': self._font,
            'text_fg': (.9, .9, .9, 1),
            'relief': FLAT,
            'frameColor': (.4, .4, .4, .14),
            'rolloverSound': loader.load_sfx(
                'assets/audio/sfx/rollover.ogg'),
            'clickSound': loader.load_sfx(
                'assets/audio/sfx/click.ogg')}
        tooltip_args = self._common['text_font'], self._common['scale'], self._common['text_fg']
        w, h = .8, .36
        self._frm = DirectFrame(frameColor=(.4, .4, .4, .06),
                                frameSize=(0, w, -h, 0),
                                parent=base.a2dTopRight,
                                pos=(-w, 0, 0))
        self.__z = -.08
        p = self.__item._np.get_pos()
        _id = ''
        if 'id' in self.__item.json:
            _id = self.__item.json['id']
        t, pos_entry = self.__add_row('position', _('position'), f'{round(p.x, 3)} {round(p.z, 3)}', self.on_edit_position, _('position (e.g. 0.1 2.3 4.5)'))
        t, id_entry = self.__add_row('id', _('id'), _id, self.on_edit_id, _('id of the item (for the strategies)'))
        fields = ['position', 'id']
        Entries = namedtuple('Entries', fields)
        self.__entries = Entries(pos_entry, id_entry)
        def load_images_btn(path, col):
            colors = {
                'gray': [
                    (.6, .6, .6, 1),  # ready
                    (1, 1, 1, 1), # press
                    (.8, .8, .8, 1), # rollover
                    (.4, .4, .4, .4)],
                'green': [
                    (.1, .68, .1, 1),
                    (.1, 1, .1, 1),
                    (.1, .84, .1, 1),
                    (.4, .1, .1, .4)]}[col]
            return [self.__load_img_btn(path, col) for col in colors]
        fcols = (.4, .4, .4, .14), (.3, .3, .3, .05)
        b = DirectButton(
            image=load_images_btn('exitRight', 'gray'), scale=.05,
            pos=(.06, 1, -h + .06),
            parent=self._frm, command=self.destroy, state=NORMAL, relief=FLAT,
            frameColor=fcols[0],
            rolloverSound=loader.load_sfx('assets/audio/sfx/rollover.ogg'),
            clickSound=loader.load_sfx('assets/audio/sfx/click.ogg'))
        b.__class__ = type('DirectButtonMixed', (DirectButton, DirectGuiMixin), {})
        pos_mgr['editor_inspector_test_close'] = b.pos_pixel()
        b.set_tooltip(_('Close'), *tooltip_args)
        self.accept('item-rototranslated', self.__on_item_rototranslated)
        b = DirectButton(
            image=load_images_btn('trashcan', 'gray'), scale=.05,
            pos=(.18, 1, -h + .06),
            parent=self._frm, command=self.__delete_item, state=NORMAL, relief=FLAT,
            frameColor=fcols[0],
            rolloverSound=loader.load_sfx('assets/audio/sfx/rollover.ogg'),
            clickSound=loader.load_sfx('assets/audio/sfx/click.ogg'))
        b.__class__ = type('DirectButtonMixed', (DirectButton, DirectGuiMixin), {})
        pos_mgr['editor_inspector_test_delete'] = b.pos_pixel()
        b.set_tooltip(_('Delete the item'), *tooltip_args)

    def __add_row(self, id_, label, text, callback, tooltip):
        tw = 10
        tooltip_args = self._common['text_font'], self._common['scale'], self._common['text_fg']
        t = OnscreenText(
            label,
            pos=(.03, self.__z), parent=self._frm,
            font=self._common['text_font'],
            scale=self._common['scale'],
            fg=self._common['text_fg'],
            wordwrap=20, align=TextNode.ALeft)
        e = DirectEntry(
            scale=self._common['scale'],
            pos=(.30, 1, self.__z),
            entryFont=self._font,
            width=tw,
            cursorKeys=True,
            frameColor=self._common['frameColor'],
            initialText=text,
            parent=self._frm,
            text_fg=self._common['text_fg'],
            command=callback)
        e.__class__ = type('DirectEntryMixed', (DirectEntry, DirectGuiMixin), {})
        e.set_tooltip(tooltip, *tooltip_args)
        self.__pos_mgr[f'editor_inspector_test_{id_}'] = e.pos_pixel()
        self.__z -= .1
        return t, e

    def __load_img_btn(self, path, col):
        img = OnscreenImage('assets/images/buttons/%s.dds' % path)
        img.set_transparency(True)
        img.set_color(col)
        img.detach_node()
        return img

    def __on_item_rototranslated(self, np):
        pos = np.get_pos()
        self.__entries.position.set('%s %s' % (str(round(pos.x, 3)), str(round(pos.z, 3))))
        self.__item.json['position'] = list(pos)

    def __delete_item(self):
        messenger.send('editor-inspector-delete', [self.__item])
        self.destroy()

    @property
    def item(self):
        return self.__item

    def on_edit_position(self, txt):
        x, z = map(float, txt.split())
        self.__item.position = [x, 0, z]

    def on_edit_id(self, txt):
        self.__item.id = txt

    def __actually_close(self, arg):
        self.__entries.strategy.set('')
        self.__entries.strategy_args.set('')
        self.__dialog.cleanup()

    def destroy(self):
        self._frm.destroy()
        self.ignore('item-rototranslated')
        messenger.send('editor-inspector-destroy')