editor: start items

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

from copy import deepcopy
from json import dumps
from importlib import import_module
from inspect import isclass
from glob import glob
from os.path import basename
from logging import info
import hashlib
from panda3d.core import Texture, TextNode
from direct.gui.OnscreenImage import OnscreenImage
from direct.gui.DirectGui import DirectButton, DirectFrame, DirectEntry, \
    YesNoDialog, DirectOptionMenu
from direct.gui.DirectGuiGlobals import FLAT, NORMAL
from direct.gui.OnscreenText import OnscreenText
from direct.showbase.DirectObject import DirectObject
from pmachines.items.item import Item
from pmachines.editor.scene_list import SceneList
from pmachines.editor.inspector import Inspector
from pmachines.editor.start_items import StartItems


class SceneEditor(DirectObject):

    def __init__(self, json, json_name, context, add_item, items):
        super().__init__()
        self.__items = items
        self.__json = json
        if not json_name:
            self.__json = json = {
                'name': '',
                'instructions': '',
                'version': '',
                'items': [],
                'start_items': [],
                'test_items': {
                    'pixel_space': [],
                    'world_space': []}}
        self.__inspector = None
        self.__context = context
        self.__add_item = add_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, tw, l = 1.8, 1, 30, .36
        self._frm = DirectFrame(frameColor=(.4, .4, .4, .06),
                                frameSize=(0, w, 0, h),
                                parent=base.a2dBottomCenter,
                                pos=(-w/2, 0, 0))
        OnscreenText(
            _('Filename'), pos=(l - .03, h - .1), parent=self._frm,
            font=self._common['text_font'],
            scale=self._common['scale'],
            fg=self._common['text_fg'],
            align=TextNode.A_right)
        self.__filenamename_entry = DirectEntry(
            scale=self._common['scale'],
            pos=(l, 1, h - .1),
            entryFont=self._font,
            width=tw,
            frameColor=self._common['frameColor'],
            initialText=json_name,
            parent=self._frm,
            text_fg=self._common['text_fg'])
        OnscreenText(
            _('Name'), pos=(l - .03, h - .2), parent=self._frm,
            font=self._common['text_font'],
            scale=self._common['scale'],
            fg=self._common['text_fg'],
            align=TextNode.A_right)
        self.__name_entry = DirectEntry(
            scale=self._common['scale'],
            pos=(l, 1, h - .2),
            entryFont=self._font,
            width=tw,
            frameColor=self._common['frameColor'],
            initialText=json['name'],
            parent=self._frm,
            text_fg=self._common['text_fg'])
        OnscreenText(
            _('Description'), pos=(l - .03, h - .3), parent=self._frm,
            font=self._common['text_font'],
            scale=self._common['scale'],
            fg=self._common['text_fg'],
            align=TextNode.A_right)
        def add_line_break(txt, entry):
            curpos = entry.node().getCursorPosition()
            entry.set(txt[:curpos]+ "\n" + txt[curpos:])
            entry.node().setCursorPosition(curpos+1)
            entry['focus']=1
        self.__instructions_entry = DirectEntry(
            scale=self._common['scale'],
            pos=(l, 1, h - .3),
            entryFont=self._font,
            width=tw,
            numLines=12,
            cursorKeys=True,
            frameColor=self._common['frameColor'],
            initialText=json['instructions'].replace('\\n', '\n'),
            parent=self._frm,
            text_fg=self._common['text_fg'])
        self.__instructions_entry['command'] = add_line_break
        self.__instructions_entry['extraArgs'] = [self.__instructions_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, .06),
            parent=self._frm, command=self.__on_close, 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'))
        DirectButton(
            image=load_images_btn('save', 'gray'), scale=.05,
            pos=(.06, 1, .18),
            parent=self._frm, command=self.__on_save, 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'))
        DirectButton(
            image=load_images_btn('menuList', 'gray'), scale=.05,
            pos=(.06, 1, .30),
            parent=self._frm, command=self.__on_scene_list, 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'))
        item_modules = glob('pmachines/items/*.py')
        item_modules = [basename(i)[:-3] for i in item_modules]
        self.__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, Item) and \
                        member != Item:
                    self.__new_items[member.__name__] = member
        OnscreenText(
            _('new item'), pos=(.02, .46), parent=self._frm,
            font=self._common['text_font'],
            scale=self._common['scale'],
            fg=self._common['text_fg'],
            align=TextNode.A_left)
        DirectOptionMenu(
            scale=.05,
            text=_('new item'), pos=(.02, 1, .4), items=list(self.__new_items.keys()),
            parent=self._frm, command=self.__on_new_item, state=NORMAL,
            relief=FLAT, item_relief=FLAT,
            frameColor=fcols[0], item_frameColor=fcols[0],
            popupMenu_frameColor=fcols[0],
            popupMarker_frameColor=fcols[0],
            text_font=self._font, text_fg=(.9, .9, .9, 1),
            highlightColor=(.9, .9, .9, .9),
            item_text_font=self._font, item_text_fg=(.9, .9, .9, 1),
            rolloverSound=loader.load_sfx('assets/audio/sfx/rollover.ogg'),
            clickSound=loader.load_sfx('assets/audio/sfx/click.ogg'))
        DirectButton(
            image=load_images_btn('start_items', 'gray'), scale=.05,
            pos=(.06, 1, .58),
            parent=self._frm, command=self.__on_start_items, 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'))
        DirectButton(
            image=load_images_btn('plus', 'gray'), scale=.05,
            pos=(.06, 1, .7),
            parent=self._frm, command=self.__on_new_scene, 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'))
        messenger.send('editor-start')
        self.accept('editor-item-click', self.__on_item_click)
        self.accept('editor-inspector-destroy', self.__on_inspector_destroy)
        self.accept('editor-start-items-save', self.__on_start_items_save)
        self.accept('editor-start-items-destroy', self.__on_start_items_destroy)

    def __on_close(self):
        self.__json['name'] = self.__name_entry.get()
        self.__json['instructions'] = self.__instructions_entry.get()
        if self.__compute_hash() == self.__json['version']:
            self._frm.destroy()
        else:
            self.__dialog = YesNoDialog(dialogName='Unsaved changes',
                                        text=_('You have unsaved changes. Really quit?'),
                                        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 __on_new_item(self, item):
        info(f'new {item}')
        item_json = {}
        _item = self.__new_items[item](
            self.__context.world,
            self.__context.plane_node,
            self.__context.cb_inst,
            self.__context.curr_bottom,
            self.__context.repos,
            item_json)
        _item._Item__editing = True
        self.__add_item(_item)
        item_json['class'] = _item.__class__.__name__
        item_json['position'] = list(_item._np.get_pos())
        self.__json['items'] += [item_json]

    def __actually_close(self, arg):
        if arg:
            self.destroy()
            messenger.send('editor-stop')
        self.__dialog.cleanup()

    def __on_save(self):
        self.__json['name'] = self.__name_entry.get()
        self.__json['instructions'] = self.__instructions_entry.get()
        self.__json['version'] = self.__compute_hash()
        json_name = self.__filenamename_entry.get()
        with open('assets/scenes/%s.json' % json_name, 'w') as f:
            f.write(dumps(self.__json, indent=2, sort_keys=True))

    def __on_scene_list(self):
        SceneList()

    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 __compute_hash(self):
        new_dict = deepcopy(self.__json)
        del new_dict['version']
        new_dict_str = str(dumps(new_dict, indent=2, sort_keys=True))
        h = hashlib.new('sha256')
        h.update(new_dict_str.encode())
        return h.hexdigest()[:12]

    def __on_item_click(self, item):
        if self.__inspector and self.__inspector.item == item: return
        if self.__inspector:
            self.__inspector.destroy()
        self.__inspector = Inspector(item, self.__items)

    def __on_inspector_destroy(self):
        self.__inspector = None

    def __on_new_scene(self):
        self.destroy()
        messenger.send('editor-stop')
        messenger.send('new_scene')

    def __on_start_items(self):
        self.__start_items = StartItems(self.__json['start_items'])

    def __on_start_items_save(self, start_items):
        self.__json['start_items'] = start_items
        self.__on_save()

    def __on_start_items_destroy(self):
        self.__start_items = None

    def destroy(self):
        self._frm.destroy()
        if self.__inspector:
            self.__inspector.destroy()
        self.ignore('editor-item-click')
        self.ignore('editor-inspector-destroy')
        self.ignore('editor-start-items-save')
        self.ignore('editor-start-items-destroy')