test handle

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

from collections import namedtuple
from glob import glob
from importlib import import_module
from os.path import basename
from inspect import isclass
from panda3d.core import Texture, TextNode
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 ya2.p3d.gfx import P3dGfxMgr
from pmachines.items.item import ItemStrategy, FixedStrategy, StillStrategy
from pmachines.items.box import HitStrategy
from pmachines.items.domino import DownStrategy, UpStrategy


class Inspector(DirectObject):

    def __init__(self, item, all_items):
        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, 1.04
        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()
        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(self.__item.json['strategy_args'])
        t, pos_entry = self.__add_row(_('position'), f'{round(p.x, 3)}, {round(p.z, 3)}', self.on_edit_position)
        t, rot_entry = self.__add_row(_('roll'), f'{round(r, 3)}', self.on_edit_roll)
        t, scale_entry = self.__add_row(_('scale'), f'{round(s, 3)}', self.on_edit_scale)
        t, mass_entry = self.__add_row(_('mass'), f'{round(m, 3)}', self.on_edit_mass)
        t, restitution_entry = self.__add_row(_('restitution'), f'{round(restitution, 3)}', self.on_edit_restitution)
        t, friction_entry = self.__add_row(_('friction'), f'{round(f, 3)}', self.on_edit_friction)
        t, id_entry = self.__add_row(_('id'), _id, self.on_edit_id)
        t, strategy_entry = self.__add_row_option(_('strategy'), _strategy, self.on_edit_strategy)
        t, strategy_args_entry = self.__add_row(_('strategy_args'), _strategy_args, self.on_edit_strategy_args)
        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)
        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 __add_row_option(self, label, text, callback):
        item_modules = glob('pmachines/items/*.py')
        item_modules = [basename(i)[:-3] for i in item_modules]
        new_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:
                    new_items = list(set(new_items + [member.__name__]))
        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 = DirectOptionMenu(
            scale=self._common['scale'],
            initialitem=text,
            pos=(.30, 1, self.__z),
            items=new_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'))
        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):
        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 = P3dGfxMgr.pos2d_p2d(np)
        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):
        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.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')