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

from copy import deepcopy
from json import dumps
import hashlib
from panda3d.core import Texture, TextNode
from direct.gui.OnscreenImage import OnscreenImage
from direct.gui.DirectGui import DirectButton, DirectFrame, DirectEntry, YesNoDialog
from direct.gui.DirectGuiGlobals import FLAT, NORMAL
from direct.gui.OnscreenText import OnscreenText
from direct.showbase.DirectObject import DirectObject
from pmachines.editor.scene_list import SceneList
from pmachines.editor.inspector import Inspector


class SceneEditor(DirectObject):

    def __init__(self, json, json_name):
        super().__init__()
        self.__json = json
        self.__json_name = json_name
        self.__inspector = None
        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, .9, 30, .36
        self._frm = DirectFrame(frameColor=(.4, .4, .4, .06),
                                frameSize=(0, w, 0, h),
                                parent=base.a2dBottomCenter,
                                pos=(-w/2, 0, 0))
        OnscreenText(
            _('Name'), 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.__name_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(
            _('Description'), 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)
        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 - .2),
            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'))
        messenger.send('editor-start')
        self.accept('editor-item-click-l', self.__on_item_click_l)
        self.accept('editor-inspector-destroy', self.__on_inspector_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 __actually_close(self, arg):
        if arg:
            self._frm.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()
        with open('assets/scenes/%s.json' % self.__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_l(self, item):
        if self.__inspector and self.__inspector.item == item: return
        if self.__inspector:
            self.__inspector.destroy()
        self.__inspector = Inspector(item)

    def __on_inspector_destroy(self):
        self.__inspector = None
Commit	Line	Data
d3a2e50a FC	1	from copy import deepcopy
	2	from json import dumps
	3	import hashlib
	4	from panda3d.core import Texture, TextNode
	5	from direct.gui.OnscreenImage import OnscreenImage
	6	from direct.gui.DirectGui import DirectButton, DirectFrame, DirectEntry, YesNoDialog
	7	from direct.gui.DirectGuiGlobals import FLAT, NORMAL
	8	from direct.gui.OnscreenText import OnscreenText
ee65fee0	9	from direct.showbase.DirectObject import DirectObject
012cf969	10	from pmachines.editor.scene_list import SceneList
ee65fee0	11	from pmachines.editor.inspector import Inspector
d3a2e50a FC	12
d3a2e50a FC	13
ee65fee0	14	class SceneEditor(DirectObject):
d3a2e50a FC	15
d3a2e50a FC	16	def __init__(self, json, json_name):
ee65fee0	17	super().__init__()
d3a2e50a FC	18	self.__json = json
d3a2e50a FC	19	self.__json_name = json_name
ee65fee0	20	self.__inspector = None
d3a2e50a FC	21	self._font = base.loader.load_font(
	22	'assets/fonts/Hanken-Book.ttf')
	23	self._font.clear()
	24	self._font.set_pixels_per_unit(60)
	25	self._font.set_minfilter(Texture.FTLinearMipmapLinear)
	26	self._font.set_outline((0, 0, 0, 1), .8, .2)
	27	self._common = {
	28	'scale': .046,
	29	'text_font': self._font,
	30	'text_fg': (.9, .9, .9, 1),
	31	'relief': FLAT,
	32	'frameColor': (.4, .4, .4, .14),
	33	'rolloverSound': loader.load_sfx(
	34	'assets/audio/sfx/rollover.ogg'),
	35	'clickSound': loader.load_sfx(
	36	'assets/audio/sfx/click.ogg')}
	37	w, h, tw, l = 1.8, .9, 30, .36
	38	self._frm = DirectFrame(frameColor=(.4, .4, .4, .06),
	39	frameSize=(0, w, 0, h),
	40	parent=base.a2dBottomCenter,
	41	pos=(-w/2, 0, 0))
	42	OnscreenText(
	43	_('Name'), pos=(l - .03, h - .1), parent=self._frm,
	44	font=self._common['text_font'],
	45	scale=self._common['scale'],
	46	fg=self._common['text_fg'],
	47	align=TextNode.A_right)
	48	self.__name_entry = DirectEntry(
	49	scale=self._common['scale'],
	50	pos=(l, 1, h - .1),
	51	entryFont=self._font,
	52	width=tw,
	53	frameColor=self._common['frameColor'],
	54	initialText=json['name'],
	55	parent=self._frm,
	56	text_fg=self._common['text_fg'])
	57	OnscreenText(
	58	_('Description'), pos=(l - .03, h - .2), parent=self._frm,
	59	font=self._common['text_font'],
	60	scale=self._common['scale'],
	61	fg=self._common['text_fg'],
	62	align=TextNode.A_right)
	63	def add_line_break(txt, entry):
	64	curpos = entry.node().getCursorPosition()
	65	entry.set(txt[:curpos]+ "\n" + txt[curpos:])
	66	entry.node().setCursorPosition(curpos+1)
	67	entry['focus']=1
	68	self.__instructions_entry = DirectEntry(
	69	scale=self._common['scale'],
	70	pos=(l, 1, h - .2),
	71	entryFont=self._font,
	72	width=tw,
	73	numLines=12,
	74	cursorKeys=True,
	75	frameColor=self._common['frameColor'],
	76	initialText=json['instructions'].replace('\\n', '\n'),
	77	parent=self._frm,
	78	text_fg=self._common['text_fg'])
	79	self.__instructions_entry['command'] = add_line_break
	80	self.__instructions_entry['extraArgs'] = [self.__instructions_entry]
	81	def load_images_btn(path, col):
	82	colors = {
	83	'gray': [
	84	(.6, .6, .6, 1), # ready
85	(1, 1, 1, 1), # press
86	(.8, .8, .8, 1), # rollover
87	(.4, .4, .4, .4)],
88	'green': [
89	(.1, .68, .1, 1),
90	(.1, 1, .1, 1),
91	(.1, .84, .1, 1),
92	(.4, .1, .1, .4)]}[col]
93	return [self.__load_img_btn(path, col) for col in colors]
94	fcols = (.4, .4, .4, .14), (.3, .3, .3, .05)
012cf969	95	DirectButton(
d3a2e50a FC	96	image=load_images_btn('exitRight', 'gray'), scale=.05,
	97	pos=(.06, 1, .06),
	98	parent=self._frm, command=self.__on_close, state=NORMAL, relief=FLAT,
	99	frameColor=fcols[0],
	100	rolloverSound=loader.load_sfx('assets/audio/sfx/rollover.ogg'),
	101	clickSound=loader.load_sfx('assets/audio/sfx/click.ogg'))
012cf969	102	DirectButton(
d3a2e50a FC	103	image=load_images_btn('save', 'gray'), scale=.05,
	104	pos=(.06, 1, .18),
	105	parent=self._frm, command=self.__on_save, state=NORMAL, relief=FLAT,
	106	frameColor=fcols[0],
	107	rolloverSound=loader.load_sfx('assets/audio/sfx/rollover.ogg'),
	108	clickSound=loader.load_sfx('assets/audio/sfx/click.ogg'))
012cf969 FC	109	DirectButton(
	110	image=load_images_btn('menuList', 'gray'), scale=.05,
	111	pos=(.06, 1, .30),
	112	parent=self._frm, command=self.__on_scene_list, state=NORMAL, relief=FLAT,
	113	frameColor=fcols[0],
	114	rolloverSound=loader.load_sfx('assets/audio/sfx/rollover.ogg'),
	115	clickSound=loader.load_sfx('assets/audio/sfx/click.ogg'))
ee65fee0 FC	116	messenger.send('editor-start')
	117	self.accept('editor-item-click-l', self.__on_item_click_l)
	118	self.accept('editor-inspector-destroy', self.__on_inspector_destroy)
d3a2e50a FC	119
	120	def __on_close(self):
	121	self.__json['name'] = self.__name_entry.get()
	122	self.__json['instructions'] = self.__instructions_entry.get()
	123	if self.__compute_hash() == self.__json['version']:
	124	self._frm.destroy()
	125	else:
	126	self.__dialog = YesNoDialog(dialogName='Unsaved changes',
	127	text=_('You have unsaved changes. Really quit?'),
	128	command=self.__actually_close)
012cf969 FC	129	self.__dialog['frameColor'] = (.4, .4, .4, .14)
	130	self.__dialog['relief'] = FLAT
	131	self.__dialog.component('text0')['fg'] = (.9, .9, .9, 1)
	132	self.__dialog.component('text0')['font'] = self._font
	133	for b in self.__dialog.buttonList:
	134	b['frameColor'] = (.4, .4, .4, .14)
	135	b.component('text0')['fg'] = (.9, .9, .9, 1)
	136	b.component('text0')['font'] = self._font
	137	b.component('text1')['fg'] = (.9, .1, .1, 1)
	138	b.component('text1')['font'] = self._font
	139	b.component('text2')['fg'] = (.9, .9, .1, 1)
	140	b.component('text2')['font'] = self._font
d3a2e50a FC	141
	142	def __actually_close(self, arg):
	143	if arg:
	144	self._frm.destroy()
ee65fee0	145	messenger.send('editor-stop')
d3a2e50a FC	146	self.__dialog.cleanup()
	147
	148	def __on_save(self):
	149	self.__json['name'] = self.__name_entry.get()
	150	self.__json['instructions'] = self.__instructions_entry.get()
	151	self.__json['version'] = self.__compute_hash()
	152	with open('assets/scenes/%s.json' % self.__json_name, 'w') as f:
	153	f.write(dumps(self.__json, indent=2, sort_keys=True))
	154
012cf969 FC	155	def __on_scene_list(self):
	156	SceneList()
	157
d3a2e50a FC	158	def __load_img_btn(self, path, col):
	159	img = OnscreenImage('assets/images/buttons/%s.dds' % path)
	160	img.set_transparency(True)
	161	img.set_color(col)
	162	img.detach_node()
	163	return img
	164
	165	def __compute_hash(self):
	166	new_dict = deepcopy(self.__json)
	167	del new_dict['version']
	168	new_dict_str = str(dumps(new_dict, indent=2, sort_keys=True))
	169	h = hashlib.new('sha256')
	170	h.update(new_dict_str.encode())
	171	return h.hexdigest()[:12]
ee65fee0 FC	172
	173	def __on_item_click_l(self, item):
	174	if self.__inspector and self.__inspector.item == item: return
	175	if self.__inspector:
	176	self.__inspector.destroy()
	177	self.__inspector = Inspector(item)
	178
	179	def __on_inspector_destroy(self):
	180	self.__inspector = None