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

from collections import namedtuple
from glob import glob
from logging import info
from importlib import import_module
from os.path import basename
from inspect import isclass
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 ItemStrategy, 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')