inspector

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

from collections import namedtuple
from panda3d.core import Texture, TextNode
from direct.gui.OnscreenImage import OnscreenImage
from direct.gui.DirectGui import DirectButton, DirectFrame, DirectEntry
from direct.gui.DirectGuiGlobals import FLAT, NORMAL
from direct.gui.OnscreenText import OnscreenText
from direct.showbase.DirectObject import DirectObject


class Inspector(DirectObject):

    def __init__(self, item):
        super().__init__()
        self.__item = item
        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, .76
        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
        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)
        fields = ['position', 'roll', 'scale', 'mass', 'restitution', 'friction']
        Entries = namedtuple('Entries', fields)
        self.__entries = Entries(pos_entry, rot_entry, scale_entry, mass_entry, restitution_entry, friction_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)

    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()
        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)))

    @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 destroy(self):
        self._frm.destroy()
        self.ignore('item-rototranslated')
        messenger.send('editor-inspector-destroy')