annotate mamba/habitats/editor.py @ 573:c01e1e329b47

Add a sentence to help. Tweak help display
author Neil Muller <drnlmuller@gmail.com>
date Wed, 21 Nov 2012 13:38:01 +0200
parents 84616c7db2e7
children cd51099911ca
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
42
8521c142cd43 Add habitat for editor and reshuffle when options are checked to before initializing the display window.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
1 """Habitat for editing levels."""
8521c142cd43 Add habitat for editor and reshuffle when options are checked to before initializing the display window.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
2
52
d09f63429b80 Add screen size fiddling to editor habitat
Neil Muller <drnlmuller@gmail.com>
parents: 42
diff changeset
3 import pygame.display
551
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
4 from pygame.locals import (SWSURFACE, KEYDOWN, K_1, K_2, MOUSEBUTTONDOWN,
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
5 K_LEFT, K_RIGHT, K_DOWN, K_UP, K_SPACE)
353
62a2a659c69b Call sprite editor dialog
Neil Muller <drnlmuller@gmail.com>
parents: 343
diff changeset
6 import sys
62a2a659c69b Call sprite editor dialog
Neil Muller <drnlmuller@gmail.com>
parents: 343
diff changeset
7 import traceback
489
e522db37e5c5 Refactor level saving to allow saving to a passed in file (e.g. StringIO object).
Simon Cross <hodgestar@gmail.com>
parents: 485
diff changeset
8 from StringIO import StringIO
490
be1e22e3f17c Finish hooking up level uploading.
Simon Cross <hodgestar@gmail.com>
parents: 489
diff changeset
9 import urllib
be1e22e3f17c Finish hooking up level uploading.
Simon Cross <hodgestar@gmail.com>
parents: 489
diff changeset
10 import urllib2
52
d09f63429b80 Add screen size fiddling to editor habitat
Neil Muller <drnlmuller@gmail.com>
parents: 42
diff changeset
11
551
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
12 from mamba.engine import (Habitat, NewHabitatEvent, SnakeDiedEvent,
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
13 LevelCompletedEvent)
508
6c61d5862310 Massive speical sprite editor rewrite
Neil Muller <drnlmuller@gmail.com>
parents: 494
diff changeset
14 from mamba.sprites import find_special_sprites
572
84616c7db2e7 ensure save dir exists before trying to save
Neil Muller <drnlmuller@gmail.com>
parents: 571
diff changeset
15 from mamba.gamestate import ensure_save_path_exists
120
6b1cdbdd34ca Rename LevelWidget
Neil Muller <drnlmuller@gmail.com>
parents: 113
diff changeset
16 from mamba.widgets.level import EditLevelWidget
174
893fb9f8f468 Hook up basic saving
Neil Muller <drnlmuller@gmail.com>
parents: 172
diff changeset
17 from mamba.widgets.text import TextWidget, TextButton
94
aba653f8383e Start adding editor toolbar
Neil Muller <drnlmuller@gmail.com>
parents: 64
diff changeset
18 from mamba.widgets.imagebutton import ImageButtonWidget
276
9c57025781e7 Warn on enter filename results
Neil Muller <drnlmuller@gmail.com>
parents: 275
diff changeset
19 from mamba.widgets.messagebox import MessageBox
229
b0ef7eecc67a Make filename editable
Neil Muller <drnlmuller@gmail.com>
parents: 180
diff changeset
20 from mamba.widgets.entrybox import EntryBox
272
a061dd62127c Hook up load and new buttons
Neil Muller <drnlmuller@gmail.com>
parents: 270
diff changeset
21 from mamba.widgets.listbox import ListBox
265
74ae1645df6e Move hackery into list widget
Neil Muller <drnlmuller@gmail.com>
parents: 240
diff changeset
22 from mamba.widgets.toollist import ToolListWidget
353
62a2a659c69b Call sprite editor dialog
Neil Muller <drnlmuller@gmail.com>
parents: 343
diff changeset
23 from mamba.widgets.editsprite import EditSpriteBox
454
8cfa757dd7d3 Epic level editor reworking.
Jeremy Thurgood <firxen@gmail.com>
parents: 423
diff changeset
24 from mamba.widgets.editlevel import EditLevelBox
278
f10d06fc64b2 Add some robustness against broken levels
Neil Muller <drnlmuller@gmail.com>
parents: 277
diff changeset
25 from mamba.level import Level, Tileset, TILE_MAP, THING_MAP, InvalidMapError
475
4986c22fe5fd Re-set window icon when the editor re-creates windows.
Simon Cross <hodgestar@gmail.com>
parents: 454
diff changeset
26 from mamba.data import (check_level_exists, get_level_list, load_file,
508
6c61d5862310 Massive speical sprite editor rewrite
Neil Muller <drnlmuller@gmail.com>
parents: 494
diff changeset
27 load_image, load_tile_image)
270
3c95ba7408f1 Add protected level name list. Add load & new buttons
Neil Muller <drnlmuller@gmail.com>
parents: 267
diff changeset
28 from mamba.constants import (SCREEN, EDIT_SCREEN, NAME, ESCAPE_KEYS,
551
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
29 RESERVED_NAMES, WINDOW_ICON, LEVEL_SERVER, UP, DOWN, LEFT, RIGHT)
42
8521c142cd43 Add habitat for editor and reshuffle when options are checked to before initializing the display window.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
30
270
3c95ba7408f1 Add protected level name list. Add load & new buttons
Neil Muller <drnlmuller@gmail.com>
parents: 267
diff changeset
31 MAX_TOOLS = 6
481
68898de788cc Add upload button to editor.
Simon Cross <hodgestar@gmail.com>
parents: 479
diff changeset
32 MODE_HEIGHT = 370
68898de788cc Add upload button to editor.
Simon Cross <hodgestar@gmail.com>
parents: 479
diff changeset
33 LOAD_SAVE_HEIGHT = 500
557
fda360905a1e Move help text out into a seperate file
Neil Muller <drnlmuller@gmail.com>
parents: 555
diff changeset
34 HELP = 'editor_help.txt'
544
13d02d1f5ae0 Add help to editor.
Simon Cross <hodgestar@gmail.com>
parents: 540
diff changeset
35
42
8521c142cd43 Add habitat for editor and reshuffle when options are checked to before initializing the display window.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
36
8521c142cd43 Add habitat for editor and reshuffle when options are checked to before initializing the display window.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
37 class EditorHabitat(Habitat):
314
fba74abad7b7 Pass Level() objects to LevelHabitat and LevelEditorHabitat, in preparation for generating thumbnails from levels in levelmenu
Stefano Rivera <stefano@rivera.za.net>
parents: 284
diff changeset
38 def __init__(self, level):
94
aba653f8383e Start adding editor toolbar
Neil Muller <drnlmuller@gmail.com>
parents: 64
diff changeset
39 super(EditorHabitat, self).__init__(EDIT_SCREEN)
479
3f37e011c3b6 Neater buttons
Neil Muller <drnlmuller@gmail.com>
parents: 475
diff changeset
40 self.toolbar = {}
314
fba74abad7b7 Pass Level() objects to LevelHabitat and LevelEditorHabitat, in preparation for generating thumbnails from levels in levelmenu
Stefano Rivera <stefano@rivera.za.net>
parents: 284
diff changeset
41 self.level = level
337
170d4a43c00e Don't edit the level when interacting with dialogs
Neil Muller <drnlmuller@gmail.com>
parents: 325
diff changeset
42 self.container.paused = False
172
dabf13abb3fd Add basic drawing to level editor
Neil Muller <drnlmuller@gmail.com>
parents: 170
diff changeset
43 self.edit_widget = EditLevelWidget(self.level)
dabf13abb3fd Add basic drawing to level editor
Neil Muller <drnlmuller@gmail.com>
parents: 170
diff changeset
44 self.container.add(self.edit_widget)
64
fbb5cc655b47 More screen size fiddling hackery
Neil Muller <drnlmuller@gmail.com>
parents: 52
diff changeset
45 self.container.add_callback(KEYDOWN, self.keydown_event)
343
5bf35267ddc0 Hook up sprite deletion. Framework work for sprite editing & adding
Neil Muller <drnlmuller@gmail.com>
parents: 337
diff changeset
46 self.container.add_callback(MOUSEBUTTONDOWN, self.mouse_event)
551
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
47 self.edit_widget.add_callback(SnakeDiedEvent, self.halt_test)
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
48 self.edit_widget.add_callback(LevelCompletedEvent, self.halt_test)
284
96b4ad9b4d30 Start work on sprite editor. Tweak layout issues
Neil Muller <drnlmuller@gmail.com>
parents: 279
diff changeset
49 self.mode = 'Tiles'
96b4ad9b4d30 Start work on sprite editor. Tweak layout issues
Neil Muller <drnlmuller@gmail.com>
parents: 279
diff changeset
50 self.sprite_mode = 'Add'
508
6c61d5862310 Massive speical sprite editor rewrite
Neil Muller <drnlmuller@gmail.com>
parents: 494
diff changeset
51 self.sprite_cls = None
551
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
52 # Map for test mode lookups
555
c014f5023cd2 Test run replays
Neil Muller <drnlmuller@gmail.com>
parents: 553
diff changeset
53 self.replays = {}
551
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
54 self.action_map = {
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
55 K_UP: UP,
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
56 K_DOWN: DOWN,
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
57 K_LEFT: LEFT,
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
58 K_RIGHT: RIGHT,
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
59 K_SPACE: None,
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
60 }
557
fda360905a1e Move help text out into a seperate file
Neil Muller <drnlmuller@gmail.com>
parents: 555
diff changeset
61 helpfile = load_file(HELP)
fda360905a1e Move help text out into a seperate file
Neil Muller <drnlmuller@gmail.com>
parents: 555
diff changeset
62 self.help_msg = ''.join(helpfile.readlines())
52
d09f63429b80 Add screen size fiddling to editor habitat
Neil Muller <drnlmuller@gmail.com>
parents: 42
diff changeset
63
d09f63429b80 Add screen size fiddling to editor habitat
Neil Muller <drnlmuller@gmail.com>
parents: 42
diff changeset
64 def on_enter(self):
d09f63429b80 Add screen size fiddling to editor habitat
Neil Muller <drnlmuller@gmail.com>
parents: 42
diff changeset
65 # We need to juggle the display to the correct size
d09f63429b80 Add screen size fiddling to editor habitat
Neil Muller <drnlmuller@gmail.com>
parents: 42
diff changeset
66 # This is a horrible hack
d09f63429b80 Add screen size fiddling to editor habitat
Neil Muller <drnlmuller@gmail.com>
parents: 42
diff changeset
67 pygame.display.quit()
d09f63429b80 Add screen size fiddling to editor habitat
Neil Muller <drnlmuller@gmail.com>
parents: 42
diff changeset
68 pygame.display.init()
d09f63429b80 Add screen size fiddling to editor habitat
Neil Muller <drnlmuller@gmail.com>
parents: 42
diff changeset
69 pygame.display.set_mode(EDIT_SCREEN, SWSURFACE)
475
4986c22fe5fd Re-set window icon when the editor re-creates windows.
Simon Cross <hodgestar@gmail.com>
parents: 454
diff changeset
70 pygame.display.set_icon(load_image(WINDOW_ICON))
52
d09f63429b80 Add screen size fiddling to editor habitat
Neil Muller <drnlmuller@gmail.com>
parents: 42
diff changeset
71 pygame.display.set_caption('%s Level editor' % NAME)
64
fbb5cc655b47 More screen size fiddling hackery
Neil Muller <drnlmuller@gmail.com>
parents: 52
diff changeset
72 super(EditorHabitat, self).on_enter()
94
aba653f8383e Start adding editor toolbar
Neil Muller <drnlmuller@gmail.com>
parents: 64
diff changeset
73 self.setup_toolbar()
52
d09f63429b80 Add screen size fiddling to editor habitat
Neil Muller <drnlmuller@gmail.com>
parents: 42
diff changeset
74
d09f63429b80 Add screen size fiddling to editor habitat
Neil Muller <drnlmuller@gmail.com>
parents: 42
diff changeset
75 def on_exit(self):
d09f63429b80 Add screen size fiddling to editor habitat
Neil Muller <drnlmuller@gmail.com>
parents: 42
diff changeset
76 # We need to juggle the display to the correct size
d09f63429b80 Add screen size fiddling to editor habitat
Neil Muller <drnlmuller@gmail.com>
parents: 42
diff changeset
77 # This is a horrible hack
64
fbb5cc655b47 More screen size fiddling hackery
Neil Muller <drnlmuller@gmail.com>
parents: 52
diff changeset
78 super(EditorHabitat, self).on_exit()
52
d09f63429b80 Add screen size fiddling to editor habitat
Neil Muller <drnlmuller@gmail.com>
parents: 42
diff changeset
79 pygame.display.quit()
d09f63429b80 Add screen size fiddling to editor habitat
Neil Muller <drnlmuller@gmail.com>
parents: 42
diff changeset
80 pygame.display.init()
d09f63429b80 Add screen size fiddling to editor habitat
Neil Muller <drnlmuller@gmail.com>
parents: 42
diff changeset
81 pygame.display.set_mode(SCREEN, SWSURFACE)
475
4986c22fe5fd Re-set window icon when the editor re-creates windows.
Simon Cross <hodgestar@gmail.com>
parents: 454
diff changeset
82 pygame.display.set_icon(load_image(WINDOW_ICON))
52
d09f63429b80 Add screen size fiddling to editor habitat
Neil Muller <drnlmuller@gmail.com>
parents: 42
diff changeset
83 pygame.display.set_caption(NAME)
64
fbb5cc655b47 More screen size fiddling hackery
Neil Muller <drnlmuller@gmail.com>
parents: 52
diff changeset
84
fbb5cc655b47 More screen size fiddling hackery
Neil Muller <drnlmuller@gmail.com>
parents: 52
diff changeset
85 def keydown_event(self, ev, widget):
fbb5cc655b47 More screen size fiddling hackery
Neil Muller <drnlmuller@gmail.com>
parents: 52
diff changeset
86 if ev.key in ESCAPE_KEYS:
fbb5cc655b47 More screen size fiddling hackery
Neil Muller <drnlmuller@gmail.com>
parents: 52
diff changeset
87 from mamba.habitats.mainmenu import MainMenu
fbb5cc655b47 More screen size fiddling hackery
Neil Muller <drnlmuller@gmail.com>
parents: 52
diff changeset
88 NewHabitatEvent.post(MainMenu())
337
170d4a43c00e Don't edit the level when interacting with dialogs
Neil Muller <drnlmuller@gmail.com>
parents: 325
diff changeset
89 return True
267
1e8dca95c48a Hook up hot-keys in editor for jerith
Neil Muller <drnlmuller@gmail.com>
parents: 265
diff changeset
90 elif ev.key == K_1:
1e8dca95c48a Hook up hot-keys in editor for jerith
Neil Muller <drnlmuller@gmail.com>
parents: 265
diff changeset
91 # Activate floor button
1e8dca95c48a Hook up hot-keys in editor for jerith
Neil Muller <drnlmuller@gmail.com>
parents: 265
diff changeset
92 self.floor_button.forced_click()
337
170d4a43c00e Don't edit the level when interacting with dialogs
Neil Muller <drnlmuller@gmail.com>
parents: 325
diff changeset
93 return True
551
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
94 elif self.mode == 'Test' and ev.key in self.action_map:
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
95 self.edit_widget.apply_action(self.action_map[ev.key])
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
96 return True
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
97
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
98 def halt_test(self, ev, widget):
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
99 if SnakeDiedEvent.matches(ev):
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
100 text = 'Snake died: %s' % ev.reason
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
101 else:
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
102 text = 'Level completed'
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
103 if self.edit_widget.snake_alive:
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
104 self.edit_widget.kill_snake()
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
105 message = MessageBox((300, 200), text,
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
106 post_callback=self.test_restart)
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
107 self.display_dialog(message)
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
108 return True
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
109
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
110 def test_restart(self):
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
111 self.unpause()
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
112 self.edit_widget.restart()
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
113 return True
94
aba653f8383e Start adding editor toolbar
Neil Muller <drnlmuller@gmail.com>
parents: 64
diff changeset
114
aba653f8383e Start adding editor toolbar
Neil Muller <drnlmuller@gmail.com>
parents: 64
diff changeset
115 def setup_toolbar(self):
aba653f8383e Start adding editor toolbar
Neil Muller <drnlmuller@gmail.com>
parents: 64
diff changeset
116 """Draw the editor toolbar"""
236
09247e08aeda Refactor edit calling code. Tweak toolbar layout
Neil Muller <drnlmuller@gmail.com>
parents: 229
diff changeset
117 button_height = 5
94
aba653f8383e Start adding editor toolbar
Neil Muller <drnlmuller@gmail.com>
parents: 64
diff changeset
118 button_left = 820
236
09247e08aeda Refactor edit calling code. Tweak toolbar layout
Neil Muller <drnlmuller@gmail.com>
parents: 229
diff changeset
119 button_padding = 2
180
d9ee0e50a78f Teach editor about THING_MAP
Neil Muller <drnlmuller@gmail.com>
parents: 175
diff changeset
120
454
8cfa757dd7d3 Epic level editor reworking.
Jeremy Thurgood <firxen@gmail.com>
parents: 423
diff changeset
121 level_name = TextWidget((button_left, button_height),
8cfa757dd7d3 Epic level editor reworking.
Jeremy Thurgood <firxen@gmail.com>
parents: 423
diff changeset
122 'Level: %s' % self.level.level_name)
8cfa757dd7d3 Epic level editor reworking.
Jeremy Thurgood <firxen@gmail.com>
parents: 423
diff changeset
123 self.container.add(level_name)
8cfa757dd7d3 Epic level editor reworking.
Jeremy Thurgood <firxen@gmail.com>
parents: 423
diff changeset
124 button_height += level_name.rect.height + button_padding
8cfa757dd7d3 Epic level editor reworking.
Jeremy Thurgood <firxen@gmail.com>
parents: 423
diff changeset
125
8cfa757dd7d3 Epic level editor reworking.
Jeremy Thurgood <firxen@gmail.com>
parents: 423
diff changeset
126 edit_level = TextButton((button_left, button_height),
8cfa757dd7d3 Epic level editor reworking.
Jeremy Thurgood <firxen@gmail.com>
parents: 423
diff changeset
127 'Edit Level Metadata')
8cfa757dd7d3 Epic level editor reworking.
Jeremy Thurgood <firxen@gmail.com>
parents: 423
diff changeset
128 edit_level.add_callback('clicked', self.edit_level_data)
8cfa757dd7d3 Epic level editor reworking.
Jeremy Thurgood <firxen@gmail.com>
parents: 423
diff changeset
129 self.container.add(edit_level)
8cfa757dd7d3 Epic level editor reworking.
Jeremy Thurgood <firxen@gmail.com>
parents: 423
diff changeset
130 button_height += edit_level.rect.height + button_padding
94
aba653f8383e Start adding editor toolbar
Neil Muller <drnlmuller@gmail.com>
parents: 64
diff changeset
131
172
dabf13abb3fd Add basic drawing to level editor
Neil Muller <drnlmuller@gmail.com>
parents: 170
diff changeset
132 # TODO: Add Image widget for the current tool
284
96b4ad9b4d30 Start work on sprite editor. Tweak layout issues
Neil Muller <drnlmuller@gmail.com>
parents: 279
diff changeset
133 if self.mode != 'Sprites':
96b4ad9b4d30 Start work on sprite editor. Tweak layout issues
Neil Muller <drnlmuller@gmail.com>
parents: 279
diff changeset
134 self.current_tool = TextWidget((button_left, button_height),
96b4ad9b4d30 Start work on sprite editor. Tweak layout issues
Neil Muller <drnlmuller@gmail.com>
parents: 279
diff changeset
135 'Tool: Floor', color='white')
96b4ad9b4d30 Start work on sprite editor. Tweak layout issues
Neil Muller <drnlmuller@gmail.com>
parents: 279
diff changeset
136 else:
96b4ad9b4d30 Start work on sprite editor. Tweak layout issues
Neil Muller <drnlmuller@gmail.com>
parents: 279
diff changeset
137 self.current_tool = TextWidget((button_left, button_height),
96b4ad9b4d30 Start work on sprite editor. Tweak layout issues
Neil Muller <drnlmuller@gmail.com>
parents: 279
diff changeset
138 '%s Sprite' % self.sprite_mode, color='white')
172
dabf13abb3fd Add basic drawing to level editor
Neil Muller <drnlmuller@gmail.com>
parents: 170
diff changeset
139 self.container.add(self.current_tool)
dabf13abb3fd Add basic drawing to level editor
Neil Muller <drnlmuller@gmail.com>
parents: 170
diff changeset
140 button_height += self.current_tool.surface.get_height()
dabf13abb3fd Add basic drawing to level editor
Neil Muller <drnlmuller@gmail.com>
parents: 170
diff changeset
141 button_height += button_padding
284
96b4ad9b4d30 Start work on sprite editor. Tweak layout issues
Neil Muller <drnlmuller@gmail.com>
parents: 279
diff changeset
142 if self.mode != 'Sprites':
96b4ad9b4d30 Start work on sprite editor. Tweak layout issues
Neil Muller <drnlmuller@gmail.com>
parents: 279
diff changeset
143 self.floor_button = ImageButtonWidget(
96b4ad9b4d30 Start work on sprite editor. Tweak layout issues
Neil Muller <drnlmuller@gmail.com>
parents: 279
diff changeset
144 (button_left, button_height), self.level.tileset.floor,
96b4ad9b4d30 Start work on sprite editor. Tweak layout issues
Neil Muller <drnlmuller@gmail.com>
parents: 279
diff changeset
145 'Floor', color='white')
96b4ad9b4d30 Start work on sprite editor. Tweak layout issues
Neil Muller <drnlmuller@gmail.com>
parents: 279
diff changeset
146 self.container.add(self.floor_button)
96b4ad9b4d30 Start work on sprite editor. Tweak layout issues
Neil Muller <drnlmuller@gmail.com>
parents: 279
diff changeset
147 self.floor_button.add_callback('clicked', self.change_tool,
96b4ad9b4d30 Start work on sprite editor. Tweak layout issues
Neil Muller <drnlmuller@gmail.com>
parents: 279
diff changeset
148 '.', 'Floor')
96b4ad9b4d30 Start work on sprite editor. Tweak layout issues
Neil Muller <drnlmuller@gmail.com>
parents: 279
diff changeset
149 self.edit_widget.set_tool('.')
96b4ad9b4d30 Start work on sprite editor. Tweak layout issues
Neil Muller <drnlmuller@gmail.com>
parents: 279
diff changeset
150 button_height += (self.floor_button.surface.get_height()
96b4ad9b4d30 Start work on sprite editor. Tweak layout issues
Neil Muller <drnlmuller@gmail.com>
parents: 279
diff changeset
151 + button_padding)
479
3f37e011c3b6 Neater buttons
Neil Muller <drnlmuller@gmail.com>
parents: 475
diff changeset
152 if self.mode in self.toolbar:
3f37e011c3b6 Neater buttons
Neil Muller <drnlmuller@gmail.com>
parents: 475
diff changeset
153 # FIXME: This needs to be recreated on tileset changes
3f37e011c3b6 Neater buttons
Neil Muller <drnlmuller@gmail.com>
parents: 475
diff changeset
154 self.tool_widget = self.toolbar[self.mode]
284
96b4ad9b4d30 Start work on sprite editor. Tweak layout issues
Neil Muller <drnlmuller@gmail.com>
parents: 279
diff changeset
155 else:
479
3f37e011c3b6 Neater buttons
Neil Muller <drnlmuller@gmail.com>
parents: 475
diff changeset
156 if self.mode == 'Tiles':
3f37e011c3b6 Neater buttons
Neil Muller <drnlmuller@gmail.com>
parents: 475
diff changeset
157 tile_map = TILE_MAP
3f37e011c3b6 Neater buttons
Neil Muller <drnlmuller@gmail.com>
parents: 475
diff changeset
158 elif self.mode == 'Things':
3f37e011c3b6 Neater buttons
Neil Muller <drnlmuller@gmail.com>
parents: 475
diff changeset
159 tile_map = THING_MAP
170
77f1dae64019 Use name in editor
Neil Muller <drnlmuller@gmail.com>
parents: 120
diff changeset
160 else:
479
3f37e011c3b6 Neater buttons
Neil Muller <drnlmuller@gmail.com>
parents: 475
diff changeset
161 tile_map = []
3f37e011c3b6 Neater buttons
Neil Muller <drnlmuller@gmail.com>
parents: 475
diff changeset
162 tool_list = []
3f37e011c3b6 Neater buttons
Neil Muller <drnlmuller@gmail.com>
parents: 475
diff changeset
163 for tile_char in sorted(tile_map):
3f37e011c3b6 Neater buttons
Neil Muller <drnlmuller@gmail.com>
parents: 475
diff changeset
164 try:
3f37e011c3b6 Neater buttons
Neil Muller <drnlmuller@gmail.com>
parents: 475
diff changeset
165 tile = self.level.tileset[tile_char]
3f37e011c3b6 Neater buttons
Neil Muller <drnlmuller@gmail.com>
parents: 475
diff changeset
166 except pygame.error:
3f37e011c3b6 Neater buttons
Neil Muller <drnlmuller@gmail.com>
parents: 475
diff changeset
167 # Ignore stuff we can't load for now
3f37e011c3b6 Neater buttons
Neil Muller <drnlmuller@gmail.com>
parents: 475
diff changeset
168 continue
3f37e011c3b6 Neater buttons
Neil Muller <drnlmuller@gmail.com>
parents: 475
diff changeset
169 if tile is None:
3f37e011c3b6 Neater buttons
Neil Muller <drnlmuller@gmail.com>
parents: 475
diff changeset
170 continue
3f37e011c3b6 Neater buttons
Neil Muller <drnlmuller@gmail.com>
parents: 475
diff changeset
171 if tile.name:
3f37e011c3b6 Neater buttons
Neil Muller <drnlmuller@gmail.com>
parents: 475
diff changeset
172 text = tile.name
3f37e011c3b6 Neater buttons
Neil Muller <drnlmuller@gmail.com>
parents: 475
diff changeset
173 else:
3f37e011c3b6 Neater buttons
Neil Muller <drnlmuller@gmail.com>
parents: 475
diff changeset
174 text = 'Tile'
3f37e011c3b6 Neater buttons
Neil Muller <drnlmuller@gmail.com>
parents: 475
diff changeset
175 tile_button = ImageButtonWidget((0, 0), tile.image, text,
3f37e011c3b6 Neater buttons
Neil Muller <drnlmuller@gmail.com>
parents: 475
diff changeset
176 color='white')
3f37e011c3b6 Neater buttons
Neil Muller <drnlmuller@gmail.com>
parents: 475
diff changeset
177 tile_button.add_callback('clicked', self.change_tool,
3f37e011c3b6 Neater buttons
Neil Muller <drnlmuller@gmail.com>
parents: 475
diff changeset
178 tile_char, text)
284
96b4ad9b4d30 Start work on sprite editor. Tweak layout issues
Neil Muller <drnlmuller@gmail.com>
parents: 279
diff changeset
179 tool_list.append(tile_button)
479
3f37e011c3b6 Neater buttons
Neil Muller <drnlmuller@gmail.com>
parents: 475
diff changeset
180 if self.mode == "Sprites":
508
6c61d5862310 Massive speical sprite editor rewrite
Neil Muller <drnlmuller@gmail.com>
parents: 494
diff changeset
181 for cls_name, sprite_cls in find_special_sprites():
6c61d5862310 Massive speical sprite editor rewrite
Neil Muller <drnlmuller@gmail.com>
parents: 494
diff changeset
182 image = load_tile_image(sprite_cls.image_name,
6c61d5862310 Massive speical sprite editor rewrite
Neil Muller <drnlmuller@gmail.com>
parents: 494
diff changeset
183 self.level.tileset.name)
6c61d5862310 Massive speical sprite editor rewrite
Neil Muller <drnlmuller@gmail.com>
parents: 494
diff changeset
184 name = sprite_cls.name
6c61d5862310 Massive speical sprite editor rewrite
Neil Muller <drnlmuller@gmail.com>
parents: 494
diff changeset
185 tile_button = ImageButtonWidget((0, 0), image, name,
6c61d5862310 Massive speical sprite editor rewrite
Neil Muller <drnlmuller@gmail.com>
parents: 494
diff changeset
186 color='white')
6c61d5862310 Massive speical sprite editor rewrite
Neil Muller <drnlmuller@gmail.com>
parents: 494
diff changeset
187 tile_button.add_callback('clicked', self.sprite_tool,
6c61d5862310 Massive speical sprite editor rewrite
Neil Muller <drnlmuller@gmail.com>
parents: 494
diff changeset
188 'Add', cls_name, sprite_cls)
6c61d5862310 Massive speical sprite editor rewrite
Neil Muller <drnlmuller@gmail.com>
parents: 494
diff changeset
189 tool_list.append(tile_button)
6c61d5862310 Massive speical sprite editor rewrite
Neil Muller <drnlmuller@gmail.com>
parents: 494
diff changeset
190 for name in ['Edit', 'Delete']:
479
3f37e011c3b6 Neater buttons
Neil Muller <drnlmuller@gmail.com>
parents: 475
diff changeset
191 tile_button = TextButton((0, 0), '%s Sprite' % name)
3f37e011c3b6 Neater buttons
Neil Muller <drnlmuller@gmail.com>
parents: 475
diff changeset
192 tile_button.add_callback('clicked', self.sprite_tool,
508
6c61d5862310 Massive speical sprite editor rewrite
Neil Muller <drnlmuller@gmail.com>
parents: 494
diff changeset
193 name, None, None)
479
3f37e011c3b6 Neater buttons
Neil Muller <drnlmuller@gmail.com>
parents: 475
diff changeset
194 tool_list.append(tile_button)
555
c014f5023cd2 Test run replays
Neil Muller <drnlmuller@gmail.com>
parents: 553
diff changeset
195 elif self.mode == 'Test':
c014f5023cd2 Test run replays
Neil Muller <drnlmuller@gmail.com>
parents: 553
diff changeset
196 for replay in [1, 2]:
c014f5023cd2 Test run replays
Neil Muller <drnlmuller@gmail.com>
parents: 553
diff changeset
197 store_button = TextButton((0, 0),
c014f5023cd2 Test run replays
Neil Muller <drnlmuller@gmail.com>
parents: 553
diff changeset
198 'Store run as %s' % replay)
c014f5023cd2 Test run replays
Neil Muller <drnlmuller@gmail.com>
parents: 553
diff changeset
199 store_button.add_callback('clicked', self.store_replay,
c014f5023cd2 Test run replays
Neil Muller <drnlmuller@gmail.com>
parents: 553
diff changeset
200 replay)
c014f5023cd2 Test run replays
Neil Muller <drnlmuller@gmail.com>
parents: 553
diff changeset
201 tool_list.append(store_button)
c014f5023cd2 Test run replays
Neil Muller <drnlmuller@gmail.com>
parents: 553
diff changeset
202 replay_button = TextButton((0, 0),
c014f5023cd2 Test run replays
Neil Muller <drnlmuller@gmail.com>
parents: 553
diff changeset
203 'Replay run %s' % replay)
c014f5023cd2 Test run replays
Neil Muller <drnlmuller@gmail.com>
parents: 553
diff changeset
204 replay_button.add_callback('clicked', self.do_replay,
c014f5023cd2 Test run replays
Neil Muller <drnlmuller@gmail.com>
parents: 553
diff changeset
205 replay)
c014f5023cd2 Test run replays
Neil Muller <drnlmuller@gmail.com>
parents: 553
diff changeset
206 tool_list.append(replay_button)
c014f5023cd2 Test run replays
Neil Muller <drnlmuller@gmail.com>
parents: 553
diff changeset
207 last_button = TextButton((0, 0),
c014f5023cd2 Test run replays
Neil Muller <drnlmuller@gmail.com>
parents: 553
diff changeset
208 'Replay last run')
c014f5023cd2 Test run replays
Neil Muller <drnlmuller@gmail.com>
parents: 553
diff changeset
209 last_button.add_callback('clicked', self.do_replay, None)
c014f5023cd2 Test run replays
Neil Muller <drnlmuller@gmail.com>
parents: 553
diff changeset
210 tool_list.append(last_button)
479
3f37e011c3b6 Neater buttons
Neil Muller <drnlmuller@gmail.com>
parents: 475
diff changeset
211 self.tool_widget = ToolListWidget((button_left, button_height),
3f37e011c3b6 Neater buttons
Neil Muller <drnlmuller@gmail.com>
parents: 475
diff changeset
212 tool_list, MAX_TOOLS, start_key=K_2)
3f37e011c3b6 Neater buttons
Neil Muller <drnlmuller@gmail.com>
parents: 475
diff changeset
213 self.toolbar[self.mode] = self.tool_widget
267
1e8dca95c48a Hook up hot-keys in editor for jerith
Neil Muller <drnlmuller@gmail.com>
parents: 265
diff changeset
214 self.container.add(self.tool_widget)
481
68898de788cc Add upload button to editor.
Simon Cross <hodgestar@gmail.com>
parents: 479
diff changeset
215
284
96b4ad9b4d30 Start work on sprite editor. Tweak layout issues
Neil Muller <drnlmuller@gmail.com>
parents: 279
diff changeset
216 button_height = self.container.rect.top + MODE_HEIGHT
481
68898de788cc Add upload button to editor.
Simon Cross <hodgestar@gmail.com>
parents: 479
diff changeset
217 tile_button = TextButton((button_left, button_height),
68898de788cc Add upload button to editor.
Simon Cross <hodgestar@gmail.com>
parents: 479
diff changeset
218 'Tiles')
68898de788cc Add upload button to editor.
Simon Cross <hodgestar@gmail.com>
parents: 479
diff changeset
219 tile_button.add_callback('clicked', self.change_toolbar, 'Tiles')
68898de788cc Add upload button to editor.
Simon Cross <hodgestar@gmail.com>
parents: 479
diff changeset
220 thing_button = TextButton((button_left + tile_button.rect.width +
68898de788cc Add upload button to editor.
Simon Cross <hodgestar@gmail.com>
parents: 479
diff changeset
221 button_padding, button_height),
68898de788cc Add upload button to editor.
Simon Cross <hodgestar@gmail.com>
parents: 479
diff changeset
222 'Things')
68898de788cc Add upload button to editor.
Simon Cross <hodgestar@gmail.com>
parents: 479
diff changeset
223 thing_button.add_callback('clicked', self.change_toolbar, 'Things')
68898de788cc Add upload button to editor.
Simon Cross <hodgestar@gmail.com>
parents: 479
diff changeset
224 sprite_button = TextButton((button_left,
540
efe1b61aad08 Start of help and an exit button.
Simon Cross <hodgestar@gmail.com>
parents: 533
diff changeset
225 thing_button.rect.bottom +
481
68898de788cc Add upload button to editor.
Simon Cross <hodgestar@gmail.com>
parents: 479
diff changeset
226 button_padding),
68898de788cc Add upload button to editor.
Simon Cross <hodgestar@gmail.com>
parents: 479
diff changeset
227 'Sprites')
68898de788cc Add upload button to editor.
Simon Cross <hodgestar@gmail.com>
parents: 479
diff changeset
228 sprite_button.add_callback('clicked', self.change_toolbar,
68898de788cc Add upload button to editor.
Simon Cross <hodgestar@gmail.com>
parents: 479
diff changeset
229 'Sprites')
551
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
230 test_button = TextButton((sprite_button.rect.right + button_padding,
540
efe1b61aad08 Start of help and an exit button.
Simon Cross <hodgestar@gmail.com>
parents: 533
diff changeset
231 thing_button.rect.bottom + button_padding),
551
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
232 'Test')
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
233 test_button.add_callback('clicked', self.change_toolbar, 'Test')
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
234 help_button = TextButton((button_left,
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
235 test_button.rect.bottom + button_padding),
540
efe1b61aad08 Start of help and an exit button.
Simon Cross <hodgestar@gmail.com>
parents: 533
diff changeset
236 'Help')
efe1b61aad08 Start of help and an exit button.
Simon Cross <hodgestar@gmail.com>
parents: 533
diff changeset
237 help_button.add_callback('clicked', self.show_help)
efe1b61aad08 Start of help and an exit button.
Simon Cross <hodgestar@gmail.com>
parents: 533
diff changeset
238
481
68898de788cc Add upload button to editor.
Simon Cross <hodgestar@gmail.com>
parents: 479
diff changeset
239 if self.mode == "Tiles":
68898de788cc Add upload button to editor.
Simon Cross <hodgestar@gmail.com>
parents: 479
diff changeset
240 tile_button.disable()
68898de788cc Add upload button to editor.
Simon Cross <hodgestar@gmail.com>
parents: 479
diff changeset
241 elif self.mode == "Things":
68898de788cc Add upload button to editor.
Simon Cross <hodgestar@gmail.com>
parents: 479
diff changeset
242 thing_button.disable()
68898de788cc Add upload button to editor.
Simon Cross <hodgestar@gmail.com>
parents: 479
diff changeset
243 elif self.mode == "Sprites":
68898de788cc Add upload button to editor.
Simon Cross <hodgestar@gmail.com>
parents: 479
diff changeset
244 sprite_button.disable()
551
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
245 elif self.mode == 'Test':
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
246 test_button.disable()
481
68898de788cc Add upload button to editor.
Simon Cross <hodgestar@gmail.com>
parents: 479
diff changeset
247 self.container.add(tile_button)
68898de788cc Add upload button to editor.
Simon Cross <hodgestar@gmail.com>
parents: 479
diff changeset
248 self.container.add(thing_button)
68898de788cc Add upload button to editor.
Simon Cross <hodgestar@gmail.com>
parents: 479
diff changeset
249 self.container.add(sprite_button)
551
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
250 self.container.add(test_button)
540
efe1b61aad08 Start of help and an exit button.
Simon Cross <hodgestar@gmail.com>
parents: 533
diff changeset
251 self.container.add(help_button)
481
68898de788cc Add upload button to editor.
Simon Cross <hodgestar@gmail.com>
parents: 479
diff changeset
252
68898de788cc Add upload button to editor.
Simon Cross <hodgestar@gmail.com>
parents: 479
diff changeset
253 button_height = LOAD_SAVE_HEIGHT
68898de788cc Add upload button to editor.
Simon Cross <hodgestar@gmail.com>
parents: 479
diff changeset
254
68898de788cc Add upload button to editor.
Simon Cross <hodgestar@gmail.com>
parents: 479
diff changeset
255 upload = TextButton((button_left, button_height), "Upload Level")
68898de788cc Add upload button to editor.
Simon Cross <hodgestar@gmail.com>
parents: 479
diff changeset
256 upload.add_callback('clicked', self.upload)
68898de788cc Add upload button to editor.
Simon Cross <hodgestar@gmail.com>
parents: 479
diff changeset
257 self.container.add(upload)
68898de788cc Add upload button to editor.
Simon Cross <hodgestar@gmail.com>
parents: 479
diff changeset
258
540
efe1b61aad08 Start of help and an exit button.
Simon Cross <hodgestar@gmail.com>
parents: 533
diff changeset
259 mainmenu = TextButton((upload.rect.right + button_padding,
efe1b61aad08 Start of help and an exit button.
Simon Cross <hodgestar@gmail.com>
parents: 533
diff changeset
260 button_height), 'Exit')
efe1b61aad08 Start of help and an exit button.
Simon Cross <hodgestar@gmail.com>
parents: 533
diff changeset
261 mainmenu.add_callback('clicked', self.go_mainmenu)
efe1b61aad08 Start of help and an exit button.
Simon Cross <hodgestar@gmail.com>
parents: 533
diff changeset
262 self.container.add(mainmenu)
efe1b61aad08 Start of help and an exit button.
Simon Cross <hodgestar@gmail.com>
parents: 533
diff changeset
263
481
68898de788cc Add upload button to editor.
Simon Cross <hodgestar@gmail.com>
parents: 479
diff changeset
264 button_height = upload.rect.bottom + button_padding
180
d9ee0e50a78f Teach editor about THING_MAP
Neil Muller <drnlmuller@gmail.com>
parents: 175
diff changeset
265
270
3c95ba7408f1 Add protected level name list. Add load & new buttons
Neil Muller <drnlmuller@gmail.com>
parents: 267
diff changeset
266 new = TextButton((button_left, button_height), "New")
3c95ba7408f1 Add protected level name list. Add load & new buttons
Neil Muller <drnlmuller@gmail.com>
parents: 267
diff changeset
267 new.add_callback('clicked', self.new)
3c95ba7408f1 Add protected level name list. Add load & new buttons
Neil Muller <drnlmuller@gmail.com>
parents: 267
diff changeset
268 self.container.add(new)
481
68898de788cc Add upload button to editor.
Simon Cross <hodgestar@gmail.com>
parents: 479
diff changeset
269
68898de788cc Add upload button to editor.
Simon Cross <hodgestar@gmail.com>
parents: 479
diff changeset
270 load = TextButton((new.rect.right + button_padding, button_height),
68898de788cc Add upload button to editor.
Simon Cross <hodgestar@gmail.com>
parents: 479
diff changeset
271 "Load")
270
3c95ba7408f1 Add protected level name list. Add load & new buttons
Neil Muller <drnlmuller@gmail.com>
parents: 267
diff changeset
272 load.add_callback('clicked', self.load)
3c95ba7408f1 Add protected level name list. Add load & new buttons
Neil Muller <drnlmuller@gmail.com>
parents: 267
diff changeset
273 self.container.add(load)
3c95ba7408f1 Add protected level name list. Add load & new buttons
Neil Muller <drnlmuller@gmail.com>
parents: 267
diff changeset
274
481
68898de788cc Add upload button to editor.
Simon Cross <hodgestar@gmail.com>
parents: 479
diff changeset
275 save = TextButton((load.rect.right + button_padding, button_height),
68898de788cc Add upload button to editor.
Simon Cross <hodgestar@gmail.com>
parents: 479
diff changeset
276 "Save")
270
3c95ba7408f1 Add protected level name list. Add load & new buttons
Neil Muller <drnlmuller@gmail.com>
parents: 267
diff changeset
277 save.add_callback('clicked', self.save)
3c95ba7408f1 Add protected level name list. Add load & new buttons
Neil Muller <drnlmuller@gmail.com>
parents: 267
diff changeset
278 self.container.add(save)
172
dabf13abb3fd Add basic drawing to level editor
Neil Muller <drnlmuller@gmail.com>
parents: 170
diff changeset
279
551
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
280 if self.mode == 'Test':
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
281 save.disable()
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
282 load.disable()
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
283 new.disable()
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
284
172
dabf13abb3fd Add basic drawing to level editor
Neil Muller <drnlmuller@gmail.com>
parents: 170
diff changeset
285 def change_tool(self, ev, widget, new_tool, text):
dabf13abb3fd Add basic drawing to level editor
Neil Muller <drnlmuller@gmail.com>
parents: 170
diff changeset
286 self.edit_widget.set_tool(new_tool)
dabf13abb3fd Add basic drawing to level editor
Neil Muller <drnlmuller@gmail.com>
parents: 170
diff changeset
287 self.current_tool.text = 'Tool: %s' % text
dabf13abb3fd Add basic drawing to level editor
Neil Muller <drnlmuller@gmail.com>
parents: 170
diff changeset
288 self.current_tool.prepare()
337
170d4a43c00e Don't edit the level when interacting with dialogs
Neil Muller <drnlmuller@gmail.com>
parents: 325
diff changeset
289 return True
174
893fb9f8f468 Hook up basic saving
Neil Muller <drnlmuller@gmail.com>
parents: 172
diff changeset
290
540
efe1b61aad08 Start of help and an exit button.
Simon Cross <hodgestar@gmail.com>
parents: 533
diff changeset
291 def show_help(self, ev, widget):
573
c01e1e329b47 Add a sentence to help. Tweak help display
Neil Muller <drnlmuller@gmail.com>
parents: 572
diff changeset
292 message = MessageBox((5, 5), self.help_msg, color="black",
544
13d02d1f5ae0 Add help to editor.
Simon Cross <hodgestar@gmail.com>
parents: 540
diff changeset
293 fontsize=12)
13d02d1f5ae0 Add help to editor.
Simon Cross <hodgestar@gmail.com>
parents: 540
diff changeset
294 self.display_dialog(message)
540
efe1b61aad08 Start of help and an exit button.
Simon Cross <hodgestar@gmail.com>
parents: 533
diff changeset
295 return True
efe1b61aad08 Start of help and an exit button.
Simon Cross <hodgestar@gmail.com>
parents: 533
diff changeset
296
efe1b61aad08 Start of help and an exit button.
Simon Cross <hodgestar@gmail.com>
parents: 533
diff changeset
297 def go_mainmenu(self, ev, widget):
efe1b61aad08 Start of help and an exit button.
Simon Cross <hodgestar@gmail.com>
parents: 533
diff changeset
298 from mamba.habitats.mainmenu import MainMenu
efe1b61aad08 Start of help and an exit button.
Simon Cross <hodgestar@gmail.com>
parents: 533
diff changeset
299 NewHabitatEvent.post(MainMenu())
efe1b61aad08 Start of help and an exit button.
Simon Cross <hodgestar@gmail.com>
parents: 533
diff changeset
300 return True
efe1b61aad08 Start of help and an exit button.
Simon Cross <hodgestar@gmail.com>
parents: 533
diff changeset
301
485
e4e9e607e6f9 Factor out level checking messages for re-use in upload.
Simon Cross <hodgestar@gmail.com>
parents: 481
diff changeset
302 def check_level(self):
278
f10d06fc64b2 Add some robustness against broken levels
Neil Muller <drnlmuller@gmail.com>
parents: 277
diff changeset
303 message = None
277
82aa91664e27 Warn on save failures
Neil Muller <drnlmuller@gmail.com>
parents: 276
diff changeset
304 if not self.level.level_name:
278
f10d06fc64b2 Add some robustness against broken levels
Neil Muller <drnlmuller@gmail.com>
parents: 277
diff changeset
305 message = MessageBox((300, 300), 'Please enter a name')
277
82aa91664e27 Warn on save failures
Neil Muller <drnlmuller@gmail.com>
parents: 276
diff changeset
306 elif self.level.level_name in RESERVED_NAMES:
278
f10d06fc64b2 Add some robustness against broken levels
Neil Muller <drnlmuller@gmail.com>
parents: 277
diff changeset
307 message = MessageBox((300, 300), 'Reserved level name')
402
001c3797a63b Editor now uses templates and the user level directory.
Jeremy Thurgood <firxen@gmail.com>
parents: 396
diff changeset
308 elif '/' in self.level.level_name:
001c3797a63b Editor now uses templates and the user level directory.
Jeremy Thurgood <firxen@gmail.com>
parents: 396
diff changeset
309 message = MessageBox((300, 300), 'Illegal level name')
485
e4e9e607e6f9 Factor out level checking messages for re-use in upload.
Simon Cross <hodgestar@gmail.com>
parents: 481
diff changeset
310
e4e9e607e6f9 Factor out level checking messages for re-use in upload.
Simon Cross <hodgestar@gmail.com>
parents: 481
diff changeset
311 if message is None:
e4e9e607e6f9 Factor out level checking messages for re-use in upload.
Simon Cross <hodgestar@gmail.com>
parents: 481
diff changeset
312 try:
e4e9e607e6f9 Factor out level checking messages for re-use in upload.
Simon Cross <hodgestar@gmail.com>
parents: 481
diff changeset
313 self.level.validate_level()
e4e9e607e6f9 Factor out level checking messages for re-use in upload.
Simon Cross <hodgestar@gmail.com>
parents: 481
diff changeset
314 except InvalidMapError, error:
e4e9e607e6f9 Factor out level checking messages for re-use in upload.
Simon Cross <hodgestar@gmail.com>
parents: 481
diff changeset
315 message = MessageBox((300, 300), "Map isn't valid\n%s" % error)
e4e9e607e6f9 Factor out level checking messages for re-use in upload.
Simon Cross <hodgestar@gmail.com>
parents: 481
diff changeset
316
e4e9e607e6f9 Factor out level checking messages for re-use in upload.
Simon Cross <hodgestar@gmail.com>
parents: 481
diff changeset
317 return message
e4e9e607e6f9 Factor out level checking messages for re-use in upload.
Simon Cross <hodgestar@gmail.com>
parents: 481
diff changeset
318
e4e9e607e6f9 Factor out level checking messages for re-use in upload.
Simon Cross <hodgestar@gmail.com>
parents: 481
diff changeset
319 def save(self, ev, widget):
e4e9e607e6f9 Factor out level checking messages for re-use in upload.
Simon Cross <hodgestar@gmail.com>
parents: 481
diff changeset
320 message = self.check_level()
278
f10d06fc64b2 Add some robustness against broken levels
Neil Muller <drnlmuller@gmail.com>
parents: 277
diff changeset
321 if message:
404
38f95fbc3710 Better dialog handling.
Jeremy Thurgood <firxen@gmail.com>
parents: 402
diff changeset
322 self.display_dialog(message)
544
13d02d1f5ae0 Add help to editor.
Simon Cross <hodgestar@gmail.com>
parents: 540
diff changeset
323 return True
571
cb8cc5b4d6be Don't crash if saving a level fails
Neil Muller <drnlmuller@gmail.com>
parents: 565
diff changeset
324 try:
572
84616c7db2e7 ensure save dir exists before trying to save
Neil Muller <drnlmuller@gmail.com>
parents: 571
diff changeset
325 ensure_save_path_exists('user_levels')
571
cb8cc5b4d6be Don't crash if saving a level fails
Neil Muller <drnlmuller@gmail.com>
parents: 565
diff changeset
326 self.level.save_level('user_levels', is_user_dir=True)
cb8cc5b4d6be Don't crash if saving a level fails
Neil Muller <drnlmuller@gmail.com>
parents: 565
diff changeset
327 self.refresh_display()
cb8cc5b4d6be Don't crash if saving a level fails
Neil Muller <drnlmuller@gmail.com>
parents: 565
diff changeset
328 message = MessageBox((300, 300),
cb8cc5b4d6be Don't crash if saving a level fails
Neil Muller <drnlmuller@gmail.com>
parents: 565
diff changeset
329 'Success!\nYou have saved a user level')
cb8cc5b4d6be Don't crash if saving a level fails
Neil Muller <drnlmuller@gmail.com>
parents: 565
diff changeset
330 except Exception, e:
cb8cc5b4d6be Don't crash if saving a level fails
Neil Muller <drnlmuller@gmail.com>
parents: 565
diff changeset
331 # Catch-all if things go wrong
cb8cc5b4d6be Don't crash if saving a level fails
Neil Muller <drnlmuller@gmail.com>
parents: 565
diff changeset
332 message = MessageBox((300, 300),
cb8cc5b4d6be Don't crash if saving a level fails
Neil Muller <drnlmuller@gmail.com>
parents: 565
diff changeset
333 "Disaster! Level not saved:\n%s" % e)
521
8a3884201d00 Save success dialog
Neil Muller <drnlmuller@gmail.com>
parents: 512
diff changeset
334 self.display_dialog(message)
337
170d4a43c00e Don't edit the level when interacting with dialogs
Neil Muller <drnlmuller@gmail.com>
parents: 325
diff changeset
335 return True
180
d9ee0e50a78f Teach editor about THING_MAP
Neil Muller <drnlmuller@gmail.com>
parents: 175
diff changeset
336
270
3c95ba7408f1 Add protected level name list. Add load & new buttons
Neil Muller <drnlmuller@gmail.com>
parents: 267
diff changeset
337 def new(self, ev, widget):
402
001c3797a63b Editor now uses templates and the user level directory.
Jeremy Thurgood <firxen@gmail.com>
parents: 396
diff changeset
338 return self.load(ev, widget, 'levels', subdir='templates')
270
3c95ba7408f1 Add protected level name list. Add load & new buttons
Neil Muller <drnlmuller@gmail.com>
parents: 267
diff changeset
339
481
68898de788cc Add upload button to editor.
Simon Cross <hodgestar@gmail.com>
parents: 479
diff changeset
340 def upload(self, ev, widget):
485
e4e9e607e6f9 Factor out level checking messages for re-use in upload.
Simon Cross <hodgestar@gmail.com>
parents: 481
diff changeset
341 message = self.check_level()
e4e9e607e6f9 Factor out level checking messages for re-use in upload.
Simon Cross <hodgestar@gmail.com>
parents: 481
diff changeset
342 if message:
e4e9e607e6f9 Factor out level checking messages for re-use in upload.
Simon Cross <hodgestar@gmail.com>
parents: 481
diff changeset
343 self.display_dialog(message)
490
be1e22e3f17c Finish hooking up level uploading.
Simon Cross <hodgestar@gmail.com>
parents: 489
diff changeset
344 return True
489
e522db37e5c5 Refactor level saving to allow saving to a passed in file (e.g. StringIO object).
Simon Cross <hodgestar@gmail.com>
parents: 485
diff changeset
345 save_file = StringIO()
e522db37e5c5 Refactor level saving to allow saving to a passed in file (e.g. StringIO object).
Simon Cross <hodgestar@gmail.com>
parents: 485
diff changeset
346 self.level.save_level(save_file=save_file)
490
be1e22e3f17c Finish hooking up level uploading.
Simon Cross <hodgestar@gmail.com>
parents: 489
diff changeset
347
494
72976997bdd9 Use LEVEL_SERVER from constants.
Simon Cross <hodgestar@gmail.com>
parents: 490
diff changeset
348 url = "%s/save/%s" % (LEVEL_SERVER, self.level.level_name)
490
be1e22e3f17c Finish hooking up level uploading.
Simon Cross <hodgestar@gmail.com>
parents: 489
diff changeset
349 args = urllib.urlencode([('data', save_file.getvalue())])
be1e22e3f17c Finish hooking up level uploading.
Simon Cross <hodgestar@gmail.com>
parents: 489
diff changeset
350 try:
be1e22e3f17c Finish hooking up level uploading.
Simon Cross <hodgestar@gmail.com>
parents: 489
diff changeset
351 result = urllib2.urlopen(url, args)
be1e22e3f17c Finish hooking up level uploading.
Simon Cross <hodgestar@gmail.com>
parents: 489
diff changeset
352 mtxt = result.read()
be1e22e3f17c Finish hooking up level uploading.
Simon Cross <hodgestar@gmail.com>
parents: 489
diff changeset
353 except:
be1e22e3f17c Finish hooking up level uploading.
Simon Cross <hodgestar@gmail.com>
parents: 489
diff changeset
354 mtxt = "Failed to upload level. :("
be1e22e3f17c Finish hooking up level uploading.
Simon Cross <hodgestar@gmail.com>
parents: 489
diff changeset
355 else:
533
cb9ee0ba01ce Explain curation in upload dialog box.
Simon Cross <hodgestar@gmail.com>
parents: 521
diff changeset
356 mtxt = "Success! %s\n%s" % (mtxt,
cb9ee0ba01ce Explain curation in upload dialog box.
Simon Cross <hodgestar@gmail.com>
parents: 521
diff changeset
357 "Your level is now awaiting curation.")
490
be1e22e3f17c Finish hooking up level uploading.
Simon Cross <hodgestar@gmail.com>
parents: 489
diff changeset
358
be1e22e3f17c Finish hooking up level uploading.
Simon Cross <hodgestar@gmail.com>
parents: 489
diff changeset
359 message = MessageBox((300, 300), mtxt)
be1e22e3f17c Finish hooking up level uploading.
Simon Cross <hodgestar@gmail.com>
parents: 489
diff changeset
360 self.display_dialog(message)
485
e4e9e607e6f9 Factor out level checking messages for re-use in upload.
Simon Cross <hodgestar@gmail.com>
parents: 481
diff changeset
361 return True
481
68898de788cc Add upload button to editor.
Simon Cross <hodgestar@gmail.com>
parents: 479
diff changeset
362
402
001c3797a63b Editor now uses templates and the user level directory.
Jeremy Thurgood <firxen@gmail.com>
parents: 396
diff changeset
363 def load(self, ev, widget, level_dir=None, is_user_dir=False, subdir=''):
001c3797a63b Editor now uses templates and the user level directory.
Jeremy Thurgood <firxen@gmail.com>
parents: 396
diff changeset
364 if level_dir is None:
001c3797a63b Editor now uses templates and the user level directory.
Jeremy Thurgood <firxen@gmail.com>
parents: 396
diff changeset
365 level_dir = 'user_levels'
001c3797a63b Editor now uses templates and the user level directory.
Jeremy Thurgood <firxen@gmail.com>
parents: 396
diff changeset
366 is_user_dir = True
337
170d4a43c00e Don't edit the level when interacting with dialogs
Neil Muller <drnlmuller@gmail.com>
parents: 325
diff changeset
367 self.container.paused = True
402
001c3797a63b Editor now uses templates and the user level directory.
Jeremy Thurgood <firxen@gmail.com>
parents: 396
diff changeset
368 levels = get_level_list('/'.join([level_dir, subdir]), is_user_dir)
272
a061dd62127c Hook up load and new buttons
Neil Muller <drnlmuller@gmail.com>
parents: 270
diff changeset
369 load_list = []
a061dd62127c Hook up load and new buttons
Neil Muller <drnlmuller@gmail.com>
parents: 270
diff changeset
370 for level_name in levels:
a061dd62127c Hook up load and new buttons
Neil Muller <drnlmuller@gmail.com>
parents: 270
diff changeset
371 if level_name in RESERVED_NAMES:
a061dd62127c Hook up load and new buttons
Neil Muller <drnlmuller@gmail.com>
parents: 270
diff changeset
372 continue
402
001c3797a63b Editor now uses templates and the user level directory.
Jeremy Thurgood <firxen@gmail.com>
parents: 396
diff changeset
373 if subdir:
001c3797a63b Editor now uses templates and the user level directory.
Jeremy Thurgood <firxen@gmail.com>
parents: 396
diff changeset
374 level_name = '/'.join([subdir, level_name])
272
a061dd62127c Hook up load and new buttons
Neil Muller <drnlmuller@gmail.com>
parents: 270
diff changeset
375 load_button = TextButton((0, 0), level_name)
402
001c3797a63b Editor now uses templates and the user level directory.
Jeremy Thurgood <firxen@gmail.com>
parents: 396
diff changeset
376 load_button.add_callback(
001c3797a63b Editor now uses templates and the user level directory.
Jeremy Thurgood <firxen@gmail.com>
parents: 396
diff changeset
377 'clicked', self.load_level, level_name, level_dir, is_user_dir)
272
a061dd62127c Hook up load and new buttons
Neil Muller <drnlmuller@gmail.com>
parents: 270
diff changeset
378 load_list.append(load_button)
404
38f95fbc3710 Better dialog handling.
Jeremy Thurgood <firxen@gmail.com>
parents: 402
diff changeset
379 self.display_dialog(ListBox((200, 200), 'Select Level', load_list, 6))
337
170d4a43c00e Don't edit the level when interacting with dialogs
Neil Muller <drnlmuller@gmail.com>
parents: 325
diff changeset
380 return True
272
a061dd62127c Hook up load and new buttons
Neil Muller <drnlmuller@gmail.com>
parents: 270
diff changeset
381
402
001c3797a63b Editor now uses templates and the user level directory.
Jeremy Thurgood <firxen@gmail.com>
parents: 396
diff changeset
382 def load_level(self, ev, widget, level_name, level_dir, is_user_dir):
278
f10d06fc64b2 Add some robustness against broken levels
Neil Muller <drnlmuller@gmail.com>
parents: 277
diff changeset
383 try:
402
001c3797a63b Editor now uses templates and the user level directory.
Jeremy Thurgood <firxen@gmail.com>
parents: 396
diff changeset
384 source = load_file("%s/%s.txt" % (level_dir, level_name),
001c3797a63b Editor now uses templates and the user level directory.
Jeremy Thurgood <firxen@gmail.com>
parents: 396
diff changeset
385 is_user_dir=is_user_dir)
001c3797a63b Editor now uses templates and the user level directory.
Jeremy Thurgood <firxen@gmail.com>
parents: 396
diff changeset
386 new_level = Level(level_name, 'user', source.read())
278
f10d06fc64b2 Add some robustness against broken levels
Neil Muller <drnlmuller@gmail.com>
parents: 277
diff changeset
387 except (IOError, InvalidMapError, pygame.error), error:
f10d06fc64b2 Add some robustness against broken levels
Neil Muller <drnlmuller@gmail.com>
parents: 277
diff changeset
388 message = MessageBox((300, 300),
f10d06fc64b2 Add some robustness against broken levels
Neil Muller <drnlmuller@gmail.com>
parents: 277
diff changeset
389 'Loading Level Failed: %s' % error, color='red')
404
38f95fbc3710 Better dialog handling.
Jeremy Thurgood <firxen@gmail.com>
parents: 402
diff changeset
390 self.display_dialog(message)
278
f10d06fc64b2 Add some robustness against broken levels
Neil Muller <drnlmuller@gmail.com>
parents: 277
diff changeset
391 return False
272
a061dd62127c Hook up load and new buttons
Neil Muller <drnlmuller@gmail.com>
parents: 270
diff changeset
392 self.container.remove(self.edit_widget)
278
f10d06fc64b2 Add some robustness against broken levels
Neil Muller <drnlmuller@gmail.com>
parents: 277
diff changeset
393 self.level = new_level
272
a061dd62127c Hook up load and new buttons
Neil Muller <drnlmuller@gmail.com>
parents: 270
diff changeset
394 if level_name in RESERVED_NAMES:
a061dd62127c Hook up load and new buttons
Neil Muller <drnlmuller@gmail.com>
parents: 270
diff changeset
395 self.level.level_name = '' # Special case for new level
337
170d4a43c00e Don't edit the level when interacting with dialogs
Neil Muller <drnlmuller@gmail.com>
parents: 325
diff changeset
396 self.container.paused = False
272
a061dd62127c Hook up load and new buttons
Neil Muller <drnlmuller@gmail.com>
parents: 270
diff changeset
397 self.edit_widget = EditLevelWidget(self.level)
551
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
398 self.edit_widget.add_callback(SnakeDiedEvent, self.halt_test)
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
399 self.edit_widget.add_callback(LevelCompletedEvent, self.halt_test)
555
c014f5023cd2 Test run replays
Neil Muller <drnlmuller@gmail.com>
parents: 553
diff changeset
400 self.replays = {}
272
a061dd62127c Hook up load and new buttons
Neil Muller <drnlmuller@gmail.com>
parents: 270
diff changeset
401 self.container.add(self.edit_widget)
a061dd62127c Hook up load and new buttons
Neil Muller <drnlmuller@gmail.com>
parents: 270
diff changeset
402 self.clear_toolbar()
a061dd62127c Hook up load and new buttons
Neil Muller <drnlmuller@gmail.com>
parents: 270
diff changeset
403 self.setup_toolbar()
387
f86188a0f259 Tweak load list length
Neil Muller <drnlmuller@gmail.com>
parents: 363
diff changeset
404 return True
270
3c95ba7408f1 Add protected level name list. Add load & new buttons
Neil Muller <drnlmuller@gmail.com>
parents: 267
diff changeset
405
284
96b4ad9b4d30 Start work on sprite editor. Tweak layout issues
Neil Muller <drnlmuller@gmail.com>
parents: 279
diff changeset
406 def change_toolbar(self, ev, widget, new_mode):
553
a2648295b7cf Only stop test if we were just testing (otherwise the level gets cleared).
Simon Cross <hodgestar@gmail.com>
parents: 552
diff changeset
407 old_mode = self.mode
551
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
408 if new_mode == 'Test':
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
409 try:
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
410 self.level.validate_level()
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
411 except InvalidMapError, error:
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
412 # Fail to change mode on invalid maps
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
413 message = MessageBox((300, 300), "Map isn't valid\n%s" % error)
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
414 self.display_dialog(message)
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
415 return True
284
96b4ad9b4d30 Start work on sprite editor. Tweak layout issues
Neil Muller <drnlmuller@gmail.com>
parents: 279
diff changeset
416 self.mode = new_mode
560
a85993b00fd7 Unbreak thing placement
Neil Muller <drnlmuller@gmail.com>
parents: 558
diff changeset
417 self.edit_widget.tile_mode = (self.mode in ('Tiles', 'Things'))
180
d9ee0e50a78f Teach editor about THING_MAP
Neil Muller <drnlmuller@gmail.com>
parents: 175
diff changeset
418 self.clear_toolbar()
d9ee0e50a78f Teach editor about THING_MAP
Neil Muller <drnlmuller@gmail.com>
parents: 175
diff changeset
419 self.setup_toolbar()
551
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
420 if self.mode == 'Test':
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
421 self.edit_widget.start_test()
553
a2648295b7cf Only stop test if we were just testing (otherwise the level gets cleared).
Simon Cross <hodgestar@gmail.com>
parents: 552
diff changeset
422 elif old_mode == 'Test':
551
9d3ed4d05f55 Hack in test level mode to editor
Neil Muller <drnlmuller@gmail.com>
parents: 544
diff changeset
423 self.edit_widget.stop_test()
337
170d4a43c00e Don't edit the level when interacting with dialogs
Neil Muller <drnlmuller@gmail.com>
parents: 325
diff changeset
424 return True
180
d9ee0e50a78f Teach editor about THING_MAP
Neil Muller <drnlmuller@gmail.com>
parents: 175
diff changeset
425
d9ee0e50a78f Teach editor about THING_MAP
Neil Muller <drnlmuller@gmail.com>
parents: 175
diff changeset
426 def clear_toolbar(self):
d9ee0e50a78f Teach editor about THING_MAP
Neil Muller <drnlmuller@gmail.com>
parents: 175
diff changeset
427 """Remove every non-edit widget from the container"""
d9ee0e50a78f Teach editor about THING_MAP
Neil Muller <drnlmuller@gmail.com>
parents: 175
diff changeset
428 for widget in self.container.children[:]:
d9ee0e50a78f Teach editor about THING_MAP
Neil Muller <drnlmuller@gmail.com>
parents: 175
diff changeset
429 if widget is not self.edit_widget:
d9ee0e50a78f Teach editor about THING_MAP
Neil Muller <drnlmuller@gmail.com>
parents: 175
diff changeset
430 self.container.remove(widget)
229
b0ef7eecc67a Make filename editable
Neil Muller <drnlmuller@gmail.com>
parents: 180
diff changeset
431
236
09247e08aeda Refactor edit calling code. Tweak toolbar layout
Neil Muller <drnlmuller@gmail.com>
parents: 229
diff changeset
432 def do_edit(self, ev, widget, message, init_value, callback):
404
38f95fbc3710 Better dialog handling.
Jeremy Thurgood <firxen@gmail.com>
parents: 402
diff changeset
433 self.display_dialog(
38f95fbc3710 Better dialog handling.
Jeremy Thurgood <firxen@gmail.com>
parents: 402
diff changeset
434 EntryBox((200, 200), message, init_value, callback))
337
170d4a43c00e Don't edit the level when interacting with dialogs
Neil Muller <drnlmuller@gmail.com>
parents: 325
diff changeset
435 return True
229
b0ef7eecc67a Make filename editable
Neil Muller <drnlmuller@gmail.com>
parents: 180
diff changeset
436
512
0ccdcc970ae2 Improved unpausing in the level editor.
Jeremy Thurgood <firxen@gmail.com>
parents: 508
diff changeset
437 def unpause(self):
0ccdcc970ae2 Improved unpausing in the level editor.
Jeremy Thurgood <firxen@gmail.com>
parents: 508
diff changeset
438 self.container.paused = False
0ccdcc970ae2 Improved unpausing in the level editor.
Jeremy Thurgood <firxen@gmail.com>
parents: 508
diff changeset
439
565
fa673a8312ec Add author to editor
Neil Muller <drnlmuller@gmail.com>
parents: 560
diff changeset
440 def update_level_data(self, filename, name, author, tileset, track):
454
8cfa757dd7d3 Epic level editor reworking.
Jeremy Thurgood <firxen@gmail.com>
parents: 423
diff changeset
441
8cfa757dd7d3 Epic level editor reworking.
Jeremy Thurgood <firxen@gmail.com>
parents: 423
diff changeset
442 self.level.name = name
565
fa673a8312ec Add author to editor
Neil Muller <drnlmuller@gmail.com>
parents: 560
diff changeset
443 self.level.author = author
454
8cfa757dd7d3 Epic level editor reworking.
Jeremy Thurgood <firxen@gmail.com>
parents: 423
diff changeset
444 self.level.background_track = track
8cfa757dd7d3 Epic level editor reworking.
Jeremy Thurgood <firxen@gmail.com>
parents: 423
diff changeset
445 # err_ts = self.change_tileset(tileset)
8cfa757dd7d3 Epic level editor reworking.
Jeremy Thurgood <firxen@gmail.com>
parents: 423
diff changeset
446 err_fn = self.check_file(filename)
8cfa757dd7d3 Epic level editor reworking.
Jeremy Thurgood <firxen@gmail.com>
parents: 423
diff changeset
447
236
09247e08aeda Refactor edit calling code. Tweak toolbar layout
Neil Muller <drnlmuller@gmail.com>
parents: 229
diff changeset
448 self.clear_toolbar()
09247e08aeda Refactor edit calling code. Tweak toolbar layout
Neil Muller <drnlmuller@gmail.com>
parents: 229
diff changeset
449 self.setup_toolbar()
09247e08aeda Refactor edit calling code. Tweak toolbar layout
Neil Muller <drnlmuller@gmail.com>
parents: 229
diff changeset
450
454
8cfa757dd7d3 Epic level editor reworking.
Jeremy Thurgood <firxen@gmail.com>
parents: 423
diff changeset
451 if err_fn:
512
0ccdcc970ae2 Improved unpausing in the level editor.
Jeremy Thurgood <firxen@gmail.com>
parents: 508
diff changeset
452 err_fn.post_callback = self.unpause
454
8cfa757dd7d3 Epic level editor reworking.
Jeremy Thurgood <firxen@gmail.com>
parents: 423
diff changeset
453 self.display_dialog(err_fn)
512
0ccdcc970ae2 Improved unpausing in the level editor.
Jeremy Thurgood <firxen@gmail.com>
parents: 508
diff changeset
454 else:
0ccdcc970ae2 Improved unpausing in the level editor.
Jeremy Thurgood <firxen@gmail.com>
parents: 508
diff changeset
455 self.container.paused = False
325
00c36ff53de9 Teach editor about background tracks
Neil Muller <drnlmuller@gmail.com>
parents: 314
diff changeset
456
279
162bddbbc5d0 Protect against bad tilesets as well
Neil Muller <drnlmuller@gmail.com>
parents: 278
diff changeset
457 def refresh_display(self):
275
51bec5e6191e Rework selecting tilesets
Neil Muller <drnlmuller@gmail.com>
parents: 272
diff changeset
458 self.level.restart()
51bec5e6191e Rework selecting tilesets
Neil Muller <drnlmuller@gmail.com>
parents: 272
diff changeset
459 self.clear_toolbar()
51bec5e6191e Rework selecting tilesets
Neil Muller <drnlmuller@gmail.com>
parents: 272
diff changeset
460 self.setup_toolbar()
51bec5e6191e Rework selecting tilesets
Neil Muller <drnlmuller@gmail.com>
parents: 272
diff changeset
461 return True
236
09247e08aeda Refactor edit calling code. Tweak toolbar layout
Neil Muller <drnlmuller@gmail.com>
parents: 229
diff changeset
462
454
8cfa757dd7d3 Epic level editor reworking.
Jeremy Thurgood <firxen@gmail.com>
parents: 423
diff changeset
463 def change_tileset(self, new_tileset):
8cfa757dd7d3 Epic level editor reworking.
Jeremy Thurgood <firxen@gmail.com>
parents: 423
diff changeset
464 print "changing tileset"
8cfa757dd7d3 Epic level editor reworking.
Jeremy Thurgood <firxen@gmail.com>
parents: 423
diff changeset
465 self.level.update_tiles_ascii()
8cfa757dd7d3 Epic level editor reworking.
Jeremy Thurgood <firxen@gmail.com>
parents: 423
diff changeset
466 old_tileset = self.level.tileset
8cfa757dd7d3 Epic level editor reworking.
Jeremy Thurgood <firxen@gmail.com>
parents: 423
diff changeset
467 try:
8cfa757dd7d3 Epic level editor reworking.
Jeremy Thurgood <firxen@gmail.com>
parents: 423
diff changeset
468 self.level.tileset = Tileset(new_tileset)
8cfa757dd7d3 Epic level editor reworking.
Jeremy Thurgood <firxen@gmail.com>
parents: 423
diff changeset
469 self.level.restart()
8cfa757dd7d3 Epic level editor reworking.
Jeremy Thurgood <firxen@gmail.com>
parents: 423
diff changeset
470 return None
8cfa757dd7d3 Epic level editor reworking.
Jeremy Thurgood <firxen@gmail.com>
parents: 423
diff changeset
471 except pygame.error, error:
8cfa757dd7d3 Epic level editor reworking.
Jeremy Thurgood <firxen@gmail.com>
parents: 423
diff changeset
472 self.level.tileset = old_tileset
8cfa757dd7d3 Epic level editor reworking.
Jeremy Thurgood <firxen@gmail.com>
parents: 423
diff changeset
473 return MessageBox(
8cfa757dd7d3 Epic level editor reworking.
Jeremy Thurgood <firxen@gmail.com>
parents: 423
diff changeset
474 (300, 300), 'Unable to change tileset:: %s' % error,
8cfa757dd7d3 Epic level editor reworking.
Jeremy Thurgood <firxen@gmail.com>
parents: 423
diff changeset
475 color='red')
8cfa757dd7d3 Epic level editor reworking.
Jeremy Thurgood <firxen@gmail.com>
parents: 423
diff changeset
476
229
b0ef7eecc67a Make filename editable
Neil Muller <drnlmuller@gmail.com>
parents: 180
diff changeset
477 def check_file(self, new_name):
278
f10d06fc64b2 Add some robustness against broken levels
Neil Muller <drnlmuller@gmail.com>
parents: 277
diff changeset
478 message = None
229
b0ef7eecc67a Make filename editable
Neil Muller <drnlmuller@gmail.com>
parents: 180
diff changeset
479 if new_name == self.level.level_name:
454
8cfa757dd7d3 Epic level editor reworking.
Jeremy Thurgood <firxen@gmail.com>
parents: 423
diff changeset
480 return None # No-op change
276
9c57025781e7 Warn on enter filename results
Neil Muller <drnlmuller@gmail.com>
parents: 275
diff changeset
481 if not new_name:
278
f10d06fc64b2 Add some robustness against broken levels
Neil Muller <drnlmuller@gmail.com>
parents: 277
diff changeset
482 message = MessageBox((300, 300), 'Please enter a name')
277
82aa91664e27 Warn on save failures
Neil Muller <drnlmuller@gmail.com>
parents: 276
diff changeset
483 if new_name in RESERVED_NAMES:
82aa91664e27 Warn on save failures
Neil Muller <drnlmuller@gmail.com>
parents: 276
diff changeset
484 # This case is caught by the existance check, but the
82aa91664e27 Warn on save failures
Neil Muller <drnlmuller@gmail.com>
parents: 276
diff changeset
485 # importance of the reserved names means we use a different
82aa91664e27 Warn on save failures
Neil Muller <drnlmuller@gmail.com>
parents: 276
diff changeset
486 # message
278
f10d06fc64b2 Add some robustness against broken levels
Neil Muller <drnlmuller@gmail.com>
parents: 277
diff changeset
487 message = MessageBox((300, 300), 'Reserved level name')
402
001c3797a63b Editor now uses templates and the user level directory.
Jeremy Thurgood <firxen@gmail.com>
parents: 396
diff changeset
488 elif check_level_exists(new_name, 'user_levels', is_user_dir=True):
278
f10d06fc64b2 Add some robustness against broken levels
Neil Muller <drnlmuller@gmail.com>
parents: 277
diff changeset
489 message = MessageBox((300, 300), 'Name already in use')
f10d06fc64b2 Add some robustness against broken levels
Neil Muller <drnlmuller@gmail.com>
parents: 277
diff changeset
490 if message:
454
8cfa757dd7d3 Epic level editor reworking.
Jeremy Thurgood <firxen@gmail.com>
parents: 423
diff changeset
491 return message
229
b0ef7eecc67a Make filename editable
Neil Muller <drnlmuller@gmail.com>
parents: 180
diff changeset
492 self.level.level_name = new_name
454
8cfa757dd7d3 Epic level editor reworking.
Jeremy Thurgood <firxen@gmail.com>
parents: 423
diff changeset
493 return None
284
96b4ad9b4d30 Start work on sprite editor. Tweak layout issues
Neil Muller <drnlmuller@gmail.com>
parents: 279
diff changeset
494
508
6c61d5862310 Massive speical sprite editor rewrite
Neil Muller <drnlmuller@gmail.com>
parents: 494
diff changeset
495 def sprite_tool(self, ev, widget, sprite_mode, cls_name, sprite_cls):
284
96b4ad9b4d30 Start work on sprite editor. Tweak layout issues
Neil Muller <drnlmuller@gmail.com>
parents: 279
diff changeset
496 """Handle sprite stuff"""
96b4ad9b4d30 Start work on sprite editor. Tweak layout issues
Neil Muller <drnlmuller@gmail.com>
parents: 279
diff changeset
497 self.sprite_mode = sprite_mode
508
6c61d5862310 Massive speical sprite editor rewrite
Neil Muller <drnlmuller@gmail.com>
parents: 494
diff changeset
498 self.sprite_cls_name = cls_name
6c61d5862310 Massive speical sprite editor rewrite
Neil Muller <drnlmuller@gmail.com>
parents: 494
diff changeset
499 self.sprite_cls = sprite_cls
343
5bf35267ddc0 Hook up sprite deletion. Framework work for sprite editing & adding
Neil Muller <drnlmuller@gmail.com>
parents: 337
diff changeset
500 self.level.update_tiles_ascii() # commit any changes
284
96b4ad9b4d30 Start work on sprite editor. Tweak layout issues
Neil Muller <drnlmuller@gmail.com>
parents: 279
diff changeset
501 self.clear_toolbar()
96b4ad9b4d30 Start work on sprite editor. Tweak layout issues
Neil Muller <drnlmuller@gmail.com>
parents: 279
diff changeset
502 self.setup_toolbar()
337
170d4a43c00e Don't edit the level when interacting with dialogs
Neil Muller <drnlmuller@gmail.com>
parents: 325
diff changeset
503 return True
343
5bf35267ddc0 Hook up sprite deletion. Framework work for sprite editing & adding
Neil Muller <drnlmuller@gmail.com>
parents: 337
diff changeset
504
5bf35267ddc0 Hook up sprite deletion. Framework work for sprite editing & adding
Neil Muller <drnlmuller@gmail.com>
parents: 337
diff changeset
505 def mouse_event(self, ev, widget):
5bf35267ddc0 Hook up sprite deletion. Framework work for sprite editing & adding
Neil Muller <drnlmuller@gmail.com>
parents: 337
diff changeset
506 """Handle mouse clicks when we are in sprite mode"""
5bf35267ddc0 Hook up sprite deletion. Framework work for sprite editing & adding
Neil Muller <drnlmuller@gmail.com>
parents: 337
diff changeset
507 if self.mode != 'Sprites':
5bf35267ddc0 Hook up sprite deletion. Framework work for sprite editing & adding
Neil Muller <drnlmuller@gmail.com>
parents: 337
diff changeset
508 return False
5bf35267ddc0 Hook up sprite deletion. Framework work for sprite editing & adding
Neil Muller <drnlmuller@gmail.com>
parents: 337
diff changeset
509 if self.container.paused:
5bf35267ddc0 Hook up sprite deletion. Framework work for sprite editing & adding
Neil Muller <drnlmuller@gmail.com>
parents: 337
diff changeset
510 return False
5bf35267ddc0 Hook up sprite deletion. Framework work for sprite editing & adding
Neil Muller <drnlmuller@gmail.com>
parents: 337
diff changeset
511 tile_pos = self.edit_widget.convert_pos(ev.pos)
508
6c61d5862310 Massive speical sprite editor rewrite
Neil Muller <drnlmuller@gmail.com>
parents: 494
diff changeset
512 sprite, sprite_ascii = self.level.get_sprite_at(tile_pos)
343
5bf35267ddc0 Hook up sprite deletion. Framework work for sprite editing & adding
Neil Muller <drnlmuller@gmail.com>
parents: 337
diff changeset
513 if self.sprite_mode == 'Delete' and sprite:
508
6c61d5862310 Massive speical sprite editor rewrite
Neil Muller <drnlmuller@gmail.com>
parents: 494
diff changeset
514 self.level.remove_sprite(sprite_ascii)
353
62a2a659c69b Call sprite editor dialog
Neil Muller <drnlmuller@gmail.com>
parents: 343
diff changeset
515 self.level.restart()
343
5bf35267ddc0 Hook up sprite deletion. Framework work for sprite editing & adding
Neil Muller <drnlmuller@gmail.com>
parents: 337
diff changeset
516 elif self.sprite_mode == 'Edit' and sprite:
5bf35267ddc0 Hook up sprite deletion. Framework work for sprite editing & adding
Neil Muller <drnlmuller@gmail.com>
parents: 337
diff changeset
517 self.edit_sprite(tile_pos, sprite)
558
73c4985bf726 Only attempt to place a sprite if there is a sprite class selected.
Simon Cross <hodgestar@gmail.com>
parents: 557
diff changeset
518 elif (self.sprite_mode == 'Add' and sprite is None
73c4985bf726 Only attempt to place a sprite if there is a sprite class selected.
Simon Cross <hodgestar@gmail.com>
parents: 557
diff changeset
519 and self.sprite_cls is not None):
508
6c61d5862310 Massive speical sprite editor rewrite
Neil Muller <drnlmuller@gmail.com>
parents: 494
diff changeset
520 self.edit_sprite(tile_pos,
6c61d5862310 Massive speical sprite editor rewrite
Neil Muller <drnlmuller@gmail.com>
parents: 494
diff changeset
521 (self.sprite_cls_name, self.sprite_cls, None, []))
343
5bf35267ddc0 Hook up sprite deletion. Framework work for sprite editing & adding
Neil Muller <drnlmuller@gmail.com>
parents: 337
diff changeset
522 return True
5bf35267ddc0 Hook up sprite deletion. Framework work for sprite editing & adding
Neil Muller <drnlmuller@gmail.com>
parents: 337
diff changeset
523
508
6c61d5862310 Massive speical sprite editor rewrite
Neil Muller <drnlmuller@gmail.com>
parents: 494
diff changeset
524 def edit_sprite(self, tile_pos, sprite_info):
6c61d5862310 Massive speical sprite editor rewrite
Neil Muller <drnlmuller@gmail.com>
parents: 494
diff changeset
525 sprite_editor = EditSpriteBox((200, 100), tile_pos, sprite_info,
353
62a2a659c69b Call sprite editor dialog
Neil Muller <drnlmuller@gmail.com>
parents: 343
diff changeset
526 post_callback=self.commit_line)
404
38f95fbc3710 Better dialog handling.
Jeremy Thurgood <firxen@gmail.com>
parents: 402
diff changeset
527 self.display_dialog(sprite_editor)
353
62a2a659c69b Call sprite editor dialog
Neil Muller <drnlmuller@gmail.com>
parents: 343
diff changeset
528
454
8cfa757dd7d3 Epic level editor reworking.
Jeremy Thurgood <firxen@gmail.com>
parents: 423
diff changeset
529 def edit_level_data(self, ev, widget):
8cfa757dd7d3 Epic level editor reworking.
Jeremy Thurgood <firxen@gmail.com>
parents: 423
diff changeset
530 elb = EditLevelBox((200, 100), self.level, self.update_level_data)
8cfa757dd7d3 Epic level editor reworking.
Jeremy Thurgood <firxen@gmail.com>
parents: 423
diff changeset
531 self.display_dialog(elb)
8cfa757dd7d3 Epic level editor reworking.
Jeremy Thurgood <firxen@gmail.com>
parents: 423
diff changeset
532
353
62a2a659c69b Call sprite editor dialog
Neil Muller <drnlmuller@gmail.com>
parents: 343
diff changeset
533 def commit_line(self, sprite):
62a2a659c69b Call sprite editor dialog
Neil Muller <drnlmuller@gmail.com>
parents: 343
diff changeset
534 try:
62a2a659c69b Call sprite editor dialog
Neil Muller <drnlmuller@gmail.com>
parents: 343
diff changeset
535 self.level.validate_sprite(sprite)
62a2a659c69b Call sprite editor dialog
Neil Muller <drnlmuller@gmail.com>
parents: 343
diff changeset
536 except:
62a2a659c69b Call sprite editor dialog
Neil Muller <drnlmuller@gmail.com>
parents: 343
diff changeset
537 # We don't know what errors thwe constructor may show, so
62a2a659c69b Call sprite editor dialog
Neil Muller <drnlmuller@gmail.com>
parents: 343
diff changeset
538 # we catch everything
62a2a659c69b Call sprite editor dialog
Neil Muller <drnlmuller@gmail.com>
parents: 343
diff changeset
539 # We use sys.exc_info to get slight neater info
62a2a659c69b Call sprite editor dialog
Neil Muller <drnlmuller@gmail.com>
parents: 343
diff changeset
540 exc_type, info, _ = sys.exc_info()
62a2a659c69b Call sprite editor dialog
Neil Muller <drnlmuller@gmail.com>
parents: 343
diff changeset
541 info = traceback.format_exception_only(exc_type, info)[0]
62a2a659c69b Call sprite editor dialog
Neil Muller <drnlmuller@gmail.com>
parents: 343
diff changeset
542 message = MessageBox((300, 300),
62a2a659c69b Call sprite editor dialog
Neil Muller <drnlmuller@gmail.com>
parents: 343
diff changeset
543 'Validation failed:\n%s' % info)
404
38f95fbc3710 Better dialog handling.
Jeremy Thurgood <firxen@gmail.com>
parents: 402
diff changeset
544 self.display_dialog(message)
353
62a2a659c69b Call sprite editor dialog
Neil Muller <drnlmuller@gmail.com>
parents: 343
diff changeset
545 return False
62a2a659c69b Call sprite editor dialog
Neil Muller <drnlmuller@gmail.com>
parents: 343
diff changeset
546 # Validation successful, so add to level
62a2a659c69b Call sprite editor dialog
Neil Muller <drnlmuller@gmail.com>
parents: 343
diff changeset
547 if self.sprite_mode == 'Add':
62a2a659c69b Call sprite editor dialog
Neil Muller <drnlmuller@gmail.com>
parents: 343
diff changeset
548 self.level.add_sprite(sprite)
62a2a659c69b Call sprite editor dialog
Neil Muller <drnlmuller@gmail.com>
parents: 343
diff changeset
549 elif self.sprite_mode == 'Edit':
62a2a659c69b Call sprite editor dialog
Neil Muller <drnlmuller@gmail.com>
parents: 343
diff changeset
550 self.level.replace_sprite(sprite)
62a2a659c69b Call sprite editor dialog
Neil Muller <drnlmuller@gmail.com>
parents: 343
diff changeset
551 self.level.restart()
62a2a659c69b Call sprite editor dialog
Neil Muller <drnlmuller@gmail.com>
parents: 343
diff changeset
552 return True
555
c014f5023cd2 Test run replays
Neil Muller <drnlmuller@gmail.com>
parents: 553
diff changeset
553
c014f5023cd2 Test run replays
Neil Muller <drnlmuller@gmail.com>
parents: 553
diff changeset
554 def store_replay(self, ev, widget, number):
c014f5023cd2 Test run replays
Neil Muller <drnlmuller@gmail.com>
parents: 553
diff changeset
555 self.replays[number] = self.edit_widget.get_replay()
c014f5023cd2 Test run replays
Neil Muller <drnlmuller@gmail.com>
parents: 553
diff changeset
556
c014f5023cd2 Test run replays
Neil Muller <drnlmuller@gmail.com>
parents: 553
diff changeset
557 def do_replay(self, ev, widget, number):
c014f5023cd2 Test run replays
Neil Muller <drnlmuller@gmail.com>
parents: 553
diff changeset
558 if number:
c014f5023cd2 Test run replays
Neil Muller <drnlmuller@gmail.com>
parents: 553
diff changeset
559 # We do nothing if the user tries to replay something thay
c014f5023cd2 Test run replays
Neil Muller <drnlmuller@gmail.com>
parents: 553
diff changeset
560 # isn't there
c014f5023cd2 Test run replays
Neil Muller <drnlmuller@gmail.com>
parents: 553
diff changeset
561 if number in self.replays:
c014f5023cd2 Test run replays
Neil Muller <drnlmuller@gmail.com>
parents: 553
diff changeset
562 # Replay stored replay
c014f5023cd2 Test run replays
Neil Muller <drnlmuller@gmail.com>
parents: 553
diff changeset
563 self.edit_widget.replay(self.replays[number])
c014f5023cd2 Test run replays
Neil Muller <drnlmuller@gmail.com>
parents: 553
diff changeset
564 else:
c014f5023cd2 Test run replays
Neil Muller <drnlmuller@gmail.com>
parents: 553
diff changeset
565 # Replaying last run
c014f5023cd2 Test run replays
Neil Muller <drnlmuller@gmail.com>
parents: 553
diff changeset
566 self.edit_widget.replay()