def __init__(self, module: 'DungeonGraphicsModule', item_id: int):
self.module = module
self.item_id = item_id
self.builder = None
self.dma: Dma = module.get_dma(item_id)
self.dpl: Dpl = module.get_dpl(item_id)
self.dpla: Dpla = module.get_dpla(item_id)
self.dpc: Dpc = module.get_dpc(item_id)
self.dpci: Dpci = module.get_dpci(item_id)
self.rules = [
[0, 0, 0],
[0, 0, 0],
[0, 0, 0]
]
self.pal_ani_durations = 0
self.current_icon_view_renderers: List[DungeonChunkCellDrawer] = []
self.chunks_surfaces: Iterable[Iterable[List[cairo.Surface]]] = []
self._init_chunk_imgs()
self.dtef = None
self.menu_controller = BgMenuController(self)
def __init__(self, module: 'DungeonGraphicsModule', item_id: int):
self.module = module
self.item_id = item_id
self.builder: Gtk.Builder = None
self.dbg: Dbg = module.get_bg_dbg(item_id)
self.dpl: Dpl = module.get_bg_dpl(item_id)
self.dpla: Dpla = module.get_bg_dpla(item_id)
self.dpc: Dpc = module.get_bg_dpc(item_id)
self.dpci: Dpci = module.get_bg_dpci(item_id)
# Cairo surfaces for each tile in each layer for each frame
# chunks_surfaces[chunk_idx][palette_animation_frame]
self.chunks_surfaces: List[List[List[cairo.Surface]]] = []
self.drawer: Optional[Drawer] = None
self.current_icon_view_renderer: Optional[DrawerCellRenderer] = None
self.bg_draw: Gtk.DrawingArea = None
self.bg_draw_event_box: Gtk.EventBox = None
self.scale_factor: float = 1.0
self.bg_draw_is_clicked = False
self._init_chunk_imgs()
self.menu_controller = BgMenuController(self) # type: ignore
class DungeonBgController(AbstractController):
def __init__(self, module: 'DungeonGraphicsModule', item_id: int):
self.module = module
self.item_id = item_id
self.builder: Gtk.Builder = None
self.dbg: Dbg = module.get_bg_dbg(item_id)
self.dpl: Dpl = module.get_bg_dpl(item_id)
self.dpla: Dpla = module.get_bg_dpla(item_id)
self.dpc: Dpc = module.get_bg_dpc(item_id)
self.dpci: Dpci = module.get_bg_dpci(item_id)
# Cairo surfaces for each tile in each layer for each frame
# chunks_surfaces[chunk_idx][palette_animation_frame]
self.chunks_surfaces: List[List[List[cairo.Surface]]] = []
self.drawer: Optional[Drawer] = None
self.current_icon_view_renderer: Optional[DrawerCellRenderer] = None
self.bg_draw: Gtk.DrawingArea = None
self.bg_draw_event_box: Gtk.EventBox = None
self.scale_factor: float = 1.0
self.bg_draw_is_clicked = False
self._init_chunk_imgs()
self.menu_controller = BgMenuController(self) # type: ignore
def get_view(self) -> Gtk.Widget:
self.builder = self._get_builder(__file__, 'dungeon_bg.glade')
self._init_drawer()
self._init_main_area()
self.builder.connect_signals(self)
return self.builder.get_object('editor_map_bg')
def on_bg_draw_click(self, box, button: Gdk.EventButton):
correct_mouse_x = int(button.x / self.scale_factor)
correct_mouse_y = int(button.y / self.scale_factor)
if button.button == 1:
self.bg_draw_is_clicked = True
snap_x = correct_mouse_x - correct_mouse_x % (DBG_TILING_DIM * DPCI_TILE_DIM)
snap_y = correct_mouse_y - correct_mouse_y % (DBG_TILING_DIM * DPCI_TILE_DIM)
self._set_chunk_at_pos(snap_x, snap_y)
def on_bg_draw_release(self, box, button: Gdk.EventButton):
if button.button == 1:
self.bg_draw_is_clicked = False
def on_bg_draw_mouse_move(self, box, motion: Gdk.EventMotion):
correct_mouse_x = int(motion.x / self.scale_factor)
correct_mouse_y = int(motion.y / self.scale_factor)
if self.drawer:
snap_x = correct_mouse_x - correct_mouse_x % (DBG_TILING_DIM * DPCI_TILE_DIM)
snap_y = correct_mouse_y - correct_mouse_y % (DBG_TILING_DIM * DPCI_TILE_DIM)
self.drawer.set_mouse_position(snap_x, snap_y)
if self.bg_draw_is_clicked:
self._set_chunk_at_pos(snap_x, snap_y)
def _set_chunk_at_pos(self, mouse_x, mouse_y):
if self.drawer:
chunk_x = int(mouse_x / (DBG_TILING_DIM * DPCI_TILE_DIM))
chunk_y = int(mouse_y / (DBG_TILING_DIM * DPCI_TILE_DIM))
if 0 <= chunk_x < DBG_WIDTH_AND_HEIGHT and 0 <= chunk_y < DBG_WIDTH_AND_HEIGHT:
chunk_mapping_idx = int(chunk_y * DBG_WIDTH_AND_HEIGHT + chunk_x)
# Set chunk at current position
self.mark_as_modified()
self.dbg.mappings[chunk_mapping_idx] = self.drawer.get_selected_chunk_id()
def on_current_icon_view_selection_changed(self, icon_view: Gtk.IconView):
model, treeiter = icon_view.get_model(), icon_view.get_selected_items()
if model is not None and treeiter is not None and treeiter != []:
chunk_id = model[treeiter][0]
if self.drawer:
self.drawer.set_selected_chunk(chunk_id)
def on_tb_zoom_out_clicked(self, w):
self.scale_factor /= 2
self._update_scales()
def on_tb_zoom_in_clicked(self, w):
self.scale_factor *= 2
self._update_scales()
def on_tb_chunk_grid_toggled(self, w):
if self.drawer:
self.drawer.set_draw_chunk_grid(w.get_active())
def on_tb_tile_grid_toggled(self, w):
if self.drawer:
self.drawer.set_draw_tile_grid(w.get_active())
def on_tb_bg_color_toggled(self, w):
if self.drawer:
self.drawer.set_pink_bg(w.get_active())
if self.current_icon_view_renderer:
self.current_icon_view_renderer.set_pink_bg(w.get_active())
def on_men_map_export_activate(self, *args):
self.menu_controller.on_men_map_export_activate()
def on_men_map_import_activate(self, *args):
self.menu_controller.on_men_map_import_activate()
def on_men_chunks_layer1_export_activate(self, *args):
self.menu_controller.on_men_chunks_layer1_export_activate()
def on_men_chunks_layer1_import_activate(self, *args):
self.menu_controller.on_men_chunks_layer1_import_activate()
def on_men_chunks_layer1_edit_activate(self, *args):
self.menu_controller.on_men_chunks_layer1_edit_activate()
def on_men_tiles_layer1_export_activate(self, *args):
self.menu_controller.on_men_tiles_layer1_export_activate()
def on_men_tiles_layer1_import_activate(self, *args):
self.menu_controller.on_men_tiles_layer1_import_activate()
def on_men_palettes_edit_activate(self, *args):
self.menu_controller.on_men_palettes_edit_activate()
def on_men_palettes_ani_settings_activate(self, *args):
self.menu_controller.on_men_palettes_ani_settings_activate()
def on_men_palettes_ani_edit_11_activate(self, *args):
self.menu_controller.edit_palette_ani(0)
def on_men_palettes_ani_edit_12_activate(self, *args):
self.menu_controller.edit_palette_ani(1)
def on_format_details_entire_clicked(self, *args):
md = SkyTempleMessageDialog(MainController.window(),
Gtk.DialogFlags.DESTROY_WITH_PARENT, Gtk.MessageType.INFO,
Gtk.ButtonsType.OK, INFO_IMEXPORT_ENTIRE)
md.set_position(Gtk.WindowPosition.CENTER)
md.run()
md.destroy()
def on_format_details_chunks_clicked(self, *args):
md = SkyTempleMessageDialog(MainController.window(),
Gtk.DialogFlags.DESTROY_WITH_PARENT, Gtk.MessageType.INFO,
Gtk.ButtonsType.OK, INFO_IMEXPORT_CHUNK)
md.set_position(Gtk.WindowPosition.CENTER)
md.run()
md.destroy()
def on_format_details_tiles_clicked(self, *args):
md = SkyTempleMessageDialog(MainController.window(),
Gtk.DialogFlags.DESTROY_WITH_PARENT, Gtk.MessageType.INFO,
Gtk.ButtonsType.OK, INFO_IMEXPORT_TILES)
md.set_position(Gtk.WindowPosition.CENTER)
md.run()
md.destroy()
def on_converter_tool_clicked(self, *args):
MainController.show_tilequant_dialog(DPL_MAX_PAL, DPL_PAL_LEN)
def on_men_tools_tilequant_activate(self, *args):
MainController.show_tilequant_dialog(DPL_MAX_PAL, DPL_PAL_LEN)
def _init_chunk_imgs(self):
"""(Re)-draw the chunk images"""
self.chunks_surfaces = []
# For each chunk...
for chunk_idx in range(0, len(self.dpc.chunks)):
# For each frame of palette animation... ( applicable for this chunk )
pal_ani_frames: List[List[cairo.Surface]] = []
self.chunks_surfaces.append(pal_ani_frames)
chunk_data = self.dpc.chunks[chunk_idx]
chunk_image = self.dpc.single_chunk_to_pil(chunk_idx, self.dpci, self.dpl.palettes)
has_pal_ani = any(chunk.pal_idx >= 10 and self.dpla.has_for_palette(chunk.pal_idx - 10) for chunk in chunk_data)
if not has_pal_ani:
len_pal_ani = 1
else:
ani_pal_lengths = [self.dpla.get_frame_count_for_palette(x) for x in (0, 1) if self.dpla.has_for_palette(x)]
if len(ani_pal_lengths) < 2:
len_pal_ani = ani_pal_lengths[0]
else:
len_pal_ani = lcm(*ani_pal_lengths)
for pal_ani in range(0, len_pal_ani):
# We don't have animated tiles, so ani_frames just has one entry.
ani_frames: List[cairo.Surface] = []
pal_ani_frames.append(ani_frames)
# Switch out the palette with that from the palette animation
if has_pal_ani:
pal_for_frame = itertools.chain.from_iterable(self.dpla.apply_palette_animations(self.dpl.palettes, pal_ani))
chunk_image.putpalette(pal_for_frame)
ani_frames.append(pil_to_cairo_surface(chunk_image.convert('RGBA')))
# TODO: No DPLA animations at different speeds supported at the moment
ani_pal11 = 9999
ani_pal12 = 9999
if self.dpla.has_for_palette(0):
ani_pal11 = self.dpla.get_duration_for_palette(0)
if self.dpla.has_for_palette(1):
ani_pal12 = self.dpla.get_duration_for_palette(1)
self.pal_ani_durations = min(ani_pal11, ani_pal12)
def _init_drawer(self):
"""(Re)-initialize the main drawing area"""
bg_draw_sw: Gtk.ScrolledWindow = self.builder.get_object('bg_draw_sw')
for child in bg_draw_sw.get_children():
bg_draw_sw.remove(child)
if self.bg_draw_event_box:
for child in self.bg_draw_event_box.get_children():
self.bg_draw_event_box.remove(child)
self.bg_draw_event_box.destroy()
if self.bg_draw:
self.bg_draw.destroy()
self.bg_draw_event_box = Gtk.EventBox()
self.bg_draw_event_box.add_events(Gdk.EventMask.POINTER_MOTION_MASK)
self.bg_draw_event_box.connect("button-press-event", self.on_bg_draw_click)
self.bg_draw_event_box.connect("button-release-event", self.on_bg_draw_release)
self.bg_draw_event_box.connect("motion-notify-event", self.on_bg_draw_mouse_move)
self.bg_draw = Gtk.DrawingArea.new()
self.bg_draw_event_box.add(self.bg_draw)
bg_draw_sw.add(self.bg_draw_event_box)
self.bg_draw.set_size_request(
DBG_WIDTH_AND_HEIGHT * DBG_TILING_DIM * DPCI_TILE_DIM,
DBG_WIDTH_AND_HEIGHT * DBG_TILING_DIM * DPCI_TILE_DIM
)
self.drawer = Drawer(self.bg_draw, self.dbg, self.pal_ani_durations, self.chunks_surfaces)
self.drawer.start()
def _init_chunks_icon_view(self):
"""Fill the icon view"""
self._deinit_chunks_icon_view()
icon_view: Gtk.IconView = self.builder.get_object(f'bg_chunks_view')
icon_view.set_selection_mode(Gtk.SelectionMode.BROWSE)
self.current_icon_view_renderer = DrawerCellRenderer(icon_view, self.pal_ani_durations, self.chunks_surfaces)
store = Gtk.ListStore(int)
icon_view.set_model(store)
icon_view.pack_start(self.current_icon_view_renderer, True)
icon_view.add_attribute(self.current_icon_view_renderer, 'chunkidx', 0)
icon_view.connect('selection-changed', self.on_current_icon_view_selection_changed)
for idx in range(0, len(self.chunks_surfaces)):
store.append([idx])
icon_view.select_path(store.get_path(store.get_iter_first()))
self.current_icon_view_renderer.start()
self.current_icon_view_renderer.set_pink_bg(self.builder.get_object(f'tb_bg_color').get_active())
def _deinit_chunks_icon_view(self):
"""Remove the icon view for the specified layer"""
icon_view: Gtk.IconView = self.builder.get_object(f'bg_chunks_view')
icon_view.clear()
icon_view.set_model(None)
def mark_as_modified(self):
self.module.mark_as_modified(self.item_id, True)
def _init_main_area(self):
self._init_chunks_icon_view()
self._update_scales()
def _update_scales(self):
"""Update drawers+DrawingArea and iconview+Renderer scales"""
self.bg_draw.set_size_request(
DBG_WIDTH_AND_HEIGHT * DBG_TILING_DIM * DPCI_TILE_DIM * self.scale_factor,
DBG_WIDTH_AND_HEIGHT * DBG_TILING_DIM * DPCI_TILE_DIM * self.scale_factor
)
if self.drawer:
self.drawer.set_scale(self.scale_factor)
if self.current_icon_view_renderer:
self.current_icon_view_renderer.set_scale(self.scale_factor)
# update size
self.current_icon_view_renderer.set_fixed_size(
int(DPCI_TILE_DIM * 3 * self.scale_factor), int(DPCI_TILE_DIM * 3 * self.scale_factor)
)
icon_view: Gtk.IconView = self.builder.get_object(f'bg_chunks_view')
icon_view.queue_resize()
def reload_all(self):
"""Reload all image related things"""
if self.current_icon_view_renderer:
self.current_icon_view_renderer.stop()
self._init_chunk_imgs()
assert self.drawer is not None
self.drawer.reset(self.dbg, self.pal_ani_durations, self.chunks_surfaces)
self._init_main_area()
class TilesetController(AbstractController):
_last_open_tab_id = 0
def __init__(self, module: 'DungeonGraphicsModule', item_id: int):
self.module = module
self.item_id = item_id
self.builder = None
self.dma: Dma = module.get_dma(item_id)
self.dpl: Dpl = module.get_dpl(item_id)
self.dpla: Dpla = module.get_dpla(item_id)
self.dpc: Dpc = module.get_dpc(item_id)
self.dpci: Dpci = module.get_dpci(item_id)
self.rules = [
[0, 0, 0],
[0, 0, 0],
[0, 0, 0]
]
self.pal_ani_durations = 0
self.current_icon_view_renderers: List[DungeonChunkCellDrawer] = []
self.chunks_surfaces: Iterable[Iterable[List[cairo.Surface]]] = []
self._init_chunk_imgs()
self.dtef = None
self.menu_controller = BgMenuController(self)
def get_view(self) -> Gtk.Widget:
self.builder = self._get_builder(__file__, 'tileset.glade')
self._init_rules()
self._init_rule_icon_views()
self._init_chunk_picker_icon_view()
self._init_secondary_terrain()
self.builder.connect_signals(self)
editor = self.builder.get_object('editor')
root = self.builder.get_object('editor_root')
root.set_current_page(self.__class__._last_open_tab_id)
self.on_editor_root_switch_page(None, None, self.__class__._last_open_tab_id)
self.builder.get_object('label_tileset_name').set_text(f(_('Dungeon Tileset {self.item_id} Rules')))
self.builder.get_object('label_tileset_name2').set_text(f(_('Dungeon Tileset {self.item_id}')))
return editor
def on_editor_root_switch_page(self, w, p, pnum, *args):
self.__class__._last_open_tab_id = pnum
if pnum == 1:
for renderer in self.current_icon_view_renderers:
renderer.start()
else:
self._load_dtef_rendering()
for renderer in self.current_icon_view_renderers:
renderer.stop()
# SIMPLE MODE
def on_btn_import_clicked(self, *args):
md = SkyTempleMessageDialog(MainController.window(),
Gtk.DialogFlags.DESTROY_WITH_PARENT, Gtk.MessageType.INFO,
Gtk.ButtonsType.OK, _("To import select the XML file in the DTEF tileset package. If it "
"is still zipped, unzip it first."),
title="SkyTemple")
md.run()
md.destroy()
dialog = Gtk.FileChooserNative.new(
_("Import dungeon tileset..."),
MainController.window(),
Gtk.FileChooserAction.OPEN,
None, None
)
filter = Gtk.FileFilter()
filter.set_name(_("DTEF XML document (*.dtef.xml)"))
filter.add_pattern("*.dtef.xml")
dialog.add_filter(filter)
add_dialog_xml_filter(dialog)
response = dialog.run()
fn = dialog.get_filename()
dialog.destroy()
if response == Gtk.ResponseType.ACCEPT:
try:
dirname = os.path.dirname(fn)
fn_xml = fn
fn_var0 = os.path.join(dirname, VAR0_FN)
fn_var1 = os.path.join(dirname, VAR1_FN)
fn_var2 = os.path.join(dirname, VAR2_FN)
dtef_importer = ExplorersDtefImporter(self.dma, self.dpc, self.dpci, self.dpl, self.dpla)
dtef_importer.do_import(
dirname, fn_xml, fn_var0, fn_var1, fn_var2
)
md = SkyTempleMessageDialog(MainController.window(),
Gtk.DialogFlags.DESTROY_WITH_PARENT, Gtk.MessageType.INFO,
Gtk.ButtonsType.OK, _("The tileset was successfully imported."),
title=_("Success!"), is_success=True)
md.run()
md.destroy()
self.mark_as_modified()
MainController.reload_view()
except BaseException as e:
display_error(
sys.exc_info(),
str(e),
_("Error importing the tileset.")
)
def on_btn_export_clicked(self, *args):
dialog = Gtk.FileChooserNative.new(
_("Export dungeon tileset..."),
MainController.window(),
Gtk.FileChooserAction.SELECT_FOLDER,
None, None
)
response = dialog.run()
fn = dialog.get_filename()
dialog.destroy()
if response == Gtk.ResponseType.ACCEPT:
try:
# Write XML
with open(os.path.join(fn, 'tileset.dtef.xml'), 'w') as f:
f.write(prettify(self.dtef.get_xml()))
# Write Tiles
var0, var1, var2, rest = self.dtef.get_tiles()
var0fn, var1fn, var2fn, restfn = self.dtef.get_filenames()
var0.save(os.path.join(fn, var0fn))
var1.save(os.path.join(fn, var1fn))
var2.save(os.path.join(fn, var2fn))
rest.save(os.path.join(fn, restfn))
shutil.copy(get_template_file(), os.path.join(fn, 'template.png'))
md = SkyTempleMessageDialog(MainController.window(),
Gtk.DialogFlags.DESTROY_WITH_PARENT, Gtk.MessageType.INFO,
Gtk.ButtonsType.OK, _("The tileset was successfully exported."),
title=_("Success!"), is_success=True)
md.run()
md.destroy()
except BaseException as e:
display_error(
sys.exc_info(),
str(e),
_("Error exporting the tileset.")
)
def on_btn_help_clicked(self, *args):
webbrowser.open_new_tab(URL_HELP)
# END SIMPLE MODE
def on_men_chunks_edit_activate(self, *args):
self.menu_controller.on_men_chunks_layer1_edit_activate()
def on_men_chunks_export_activate(self, *args):
self.menu_controller.on_men_chunks_layer1_export_activate()
def on_men_chunks_import_activate(self, *args):
self.menu_controller.on_men_chunks_layer1_import_activate()
def on_men_tiles_export_activate(self, *args):
self.menu_controller.on_men_tiles_layer1_export_activate()
def on_men_tiles_import_activate(self, *args):
self.menu_controller.on_men_tiles_layer1_import_activate()
def on_men_palettes_edit_activate(self, *args):
self.menu_controller.on_men_palettes_edit_activate()
def on_men_palettes_ani_settings_activate(self, *args):
self.menu_controller.on_men_palettes_ani_settings_activate()
def on_men_palettes_ani_edit_11_activate(self, *args):
self.menu_controller.edit_palette_ani(0)
def on_men_palettes_ani_edit_12_activate(self, *args):
self.menu_controller.edit_palette_ani(1)
def on_men_tools_tilequant_activate(self, *args):
MainController.show_tilequant_dialog(12, 16)
def on_rules_active_type_changed(self, widget, *area):
self.update_chunks_from_current_rules()
def on_rules_a0_toggled(self, widget: Gtk.CheckButton, *area):
self._rules_pos_toggle(0, widget.get_active())
def on_rules_a1_toggled(self, widget: Gtk.CheckButton, *area):
self._rules_pos_toggle(1, widget.get_active())
def on_rules_a2_toggled(self, widget: Gtk.CheckButton, *area):
self._rules_pos_toggle(2, widget.get_active())
def on_rules_a3_toggled(self, widget: Gtk.CheckButton, *area):
self._rules_pos_toggle(3, widget.get_active())
def on_rules_a4_toggled(self, widget: Gtk.CheckButton, *area):
self._rules_pos_toggle(4, widget.get_active())
def on_rules_a5_toggled(self, widget: Gtk.CheckButton, *area):
self._rules_pos_toggle(5, widget.get_active())
def on_rules_a6_toggled(self, widget: Gtk.CheckButton, *area):
self._rules_pos_toggle(6, widget.get_active())
def on_rules_a7_toggled(self, widget: Gtk.CheckButton, *area):
self._rules_pos_toggle(7, widget.get_active())
def on_rules_a8_toggled(self, widget: Gtk.CheckButton, *area):
self._rules_pos_toggle(8, widget.get_active())
def on_rules_chunk_picker_selection_changed(self, icon_view: Gtk.IconView):
pass
def on_secondary_terrain_type_changed(self, w: Gtk.ComboBox):
idx = w.get_active()
static = self.module.project.get_rom_module().get_static_data()
secondary_terrains = HardcodedDungeons.get_secondary_terrains(
self.module.project.get_binary(BinaryName.ARM9), static
)
secondary_terrains[self.item_id] = SecondaryTerrainTableEntry(idx)
def update(arm9):
HardcodedDungeons.set_secondary_terrains(secondary_terrains, arm9, static)
self.module.project.modify_binary(BinaryName.ARM9, update)
self.mark_as_modified()
def on_rules_main_1_selection_changed(self, icon_view):
model, treeiter = icon_view.get_model(), icon_view.get_selected_items()
if model is not None and treeiter is not None and treeiter != []:
self._rule_updated(model[treeiter][0], 0)
def on_rules_main_2_selection_changed(self, icon_view):
model, treeiter = icon_view.get_model(), icon_view.get_selected_items()
if model is not None and treeiter is not None and treeiter != []:
self._rule_updated(model[treeiter][0], 1)
def on_rules_main_3_selection_changed(self, icon_view):
model, treeiter = icon_view.get_model(), icon_view.get_selected_items()
if model is not None and treeiter is not None and treeiter != []:
self._rule_updated(model[treeiter][0], 2)
def on_rules_extra_selection_changed(self, icon_view_extra):
model, treeiter = icon_view_extra.get_model(), icon_view_extra.get_selected_items()
if model is not None and treeiter is not None and treeiter != []:
i = model[treeiter][0]
icon_view_picker = self.builder.get_object('rules_chunk_picker')
model, treeiter = icon_view_picker.get_model(), icon_view_picker.get_selected_items()
if model is not None and treeiter is not None and treeiter != []:
edited_value = model[treeiter][1]
extra_type = DmaExtraType.FLOOR1
if i > 15:
extra_type = DmaExtraType.WALL_OR_VOID
if i > 31:
extra_type = DmaExtraType.FLOOR2
self.dma.set_extra(extra_type, i % 16, edited_value)
self.update_chunks_extra()
self.mark_as_modified()
def mark_as_modified(self):
self.module.mark_as_modified(self.item_id, False)
def reload_all(self):
"""Reload all image related things"""
for renderer in self.current_icon_view_renderers:
renderer.stop()
self._init_chunk_imgs()
for renderer in self.current_icon_view_renderers:
renderer.reset(self.pal_ani_durations, self.chunks_surfaces)
self._init_rule_icon_views()
self._init_chunk_picker_icon_view()
def _init_chunk_imgs(self):
"""(Re)-draw the chunk images"""
self.chunks_surfaces = []
# For each chunk...
for chunk_idx in range(0, len(self.dpc.chunks)):
# For each frame of palette animation... ( applicable for this chunk )
pal_ani_frames = []
self.chunks_surfaces.append(pal_ani_frames)
chunk_data = self.dpc.chunks[chunk_idx]
chunk_image = self.dpc.single_chunk_to_pil(chunk_idx, self.dpci, self.dpl.palettes)
has_pal_ani = any(chunk.pal_idx >= 10 and self.dpla.has_for_palette(chunk.pal_idx - 10) for chunk in chunk_data)
if not has_pal_ani:
len_pal_ani = 1
else:
ani_pal_lengths = [self.dpla.get_frame_count_for_palette(x) for x in (0, 1) if self.dpla.has_for_palette(x)]
if len(ani_pal_lengths) < 2:
len_pal_ani = ani_pal_lengths[0]
else:
len_pal_ani = lcm(*ani_pal_lengths)
for pal_ani in range(0, len_pal_ani):
# We don't have animated tiles, so ani_frames just has one entry.
ani_frames = []
pal_ani_frames.append(ani_frames)
# Switch out the palette with that from the palette animation
if has_pal_ani:
pal_for_frame = itertools.chain.from_iterable(self.dpla.apply_palette_animations(self.dpl.palettes, pal_ani))
chunk_image.putpalette(pal_for_frame)
ani_frames.append(pil_to_cairo_surface(chunk_image.convert('RGBA')))
# TODO: No DPLA animations at different speeds supported at the moment
ani_pal11 = 9999
ani_pal12 = 9999
if self.dpla.has_for_palette(0):
ani_pal11 = self.dpla.get_duration_for_palette(0)
if self.dpla.has_for_palette(1):
ani_pal12 = self.dpla.get_duration_for_palette(1)
self.pal_ani_durations = min(ani_pal11, ani_pal12)
def _init_rule_icon_views(self):
"""Fill the icon views for the three variations and the extra rules."""
def init_main_rules_store(store):
for idx in range(0, 9):
store.append([idx, 0])
def init_extra_rules_store(store):
for extra_type in (DmaExtraType.FLOOR1, DmaExtraType.WALL_OR_VOID, DmaExtraType.FLOOR2):
for idx, val in enumerate(self.dma.get_extra(extra_type)):
store.append([idx + 16 * extra_type.value, val])
for i, v_icon_view_name in enumerate(("rules_main_1", "rules_main_2", "rules_main_3")):
self._init_an_icon_view(v_icon_view_name, init_main_rules_store, False)
self._init_an_icon_view("rules_extra", init_extra_rules_store, False)
self.update_chunks_from_current_rules()
def _init_chunk_picker_icon_view(self):
"""Fill the icon view for the chunk picker"""
def init_store(store):
for idx in range(0, len(self.dpc.chunks)):
store.append([idx, idx])
self._init_an_icon_view("rules_chunk_picker", init_store, True, True)
def _init_an_icon_view(self, name: str, init_store: Callable[[Gtk.ListStore], None], selection_draw_solid, select_first=False):
icon_view: Gtk.IconView = self.builder.get_object(name)
if icon_view.get_model() == None:
# id, val
store = Gtk.ListStore(int, int)
icon_view.set_model(store)
renderer = DungeonChunkCellDrawer(icon_view, self.pal_ani_durations, self.chunks_surfaces, selection_draw_solid)
self.current_icon_view_renderers.append(renderer)
icon_view.pack_start(renderer, True)
icon_view.add_attribute(renderer, 'chunkidx', 1)
else:
store = icon_view.get_model()
store.clear()
self.renderer = None
for child in icon_view.get_cells():
if isinstance(child, DungeonChunkCellDrawer):
self.renderer = child
break
init_store(store)
if select_first:
icon_view.select_path(store.get_path(store.get_iter_first()))
def _rules_pos_toggle(self, i, state):
y = i % 3
x = math.floor(i / 3)
self.rules[x][y] = state
self.update_chunks_from_current_rules()
def _init_rules(self):
def ia(i):
return self.builder.get_object(f'rules_a{i}').get_active()
self.rules = list(chunks([ia(i) for i in range(0, 9)], 3))
def update_chunks_from_current_rules(self):
solid_type = self._get_current_solid_type()
all_chunk_mapping_vars = []
for y, row in enumerate(self.rules):
for x, solid in enumerate(row):
chunk_type = solid_type if solid else DmaType.FLOOR
solid_neighbors = Dma.get_tile_neighbors(self.rules, x, y, bool(solid))
all_chunk_mapping_vars.append(self.dma.get(chunk_type, solid_neighbors))
for i, v_icon_view_name in enumerate(("rules_main_1", "rules_main_2", "rules_main_3")):
icon_view_model: Gtk.ListStore = self.builder.get_object(v_icon_view_name).get_model()
icon_view_model.clear()
for j, idxs in enumerate(all_chunk_mapping_vars):
icon_view_model.append([j, idxs[i]])
def update_chunks_extra(self):
store = self.builder.get_object('rules_extra').get_model()
store.clear()
for extra_type in (DmaExtraType.FLOOR1, DmaExtraType.WALL_OR_VOID, DmaExtraType.FLOOR2):
for idx, val in enumerate(self.dma.get_extra(extra_type)):
store.append([idx + 16 * extra_type.value, val])
def _get_current_solid_type(self):
combo_box: Gtk.ComboBoxText = self.builder.get_object("rules_active_type")
return DmaType.WALL if combo_box.get_active_text() == 'Wall' else DmaType.WATER
def _rule_updated(self, i, variation):
icon_view = self.builder.get_object('rules_chunk_picker')
model, treeiter = icon_view.get_model(), icon_view.get_selected_items()
if model is not None and treeiter is not None and treeiter != []:
x = i % 3
y = math.floor(i / 3)
edited_value = model[treeiter][1]
solid = bool(self.rules[x][y])
self.dma.set(
self._get_current_solid_type() if solid else DmaType.FLOOR,
Dma.get_tile_neighbors(self.rules, x, y, solid),
variation, edited_value
)
self.update_chunks_from_current_rules()
self.mark_as_modified()
def _load_dtef_rendering(self):
self.dtef = ExplorersDtef(self.dma, self.dpc, self.dpci, self.dpl, self.dpla)
box: Gtk.Box = self.builder.get_object('dungeon_image_placeholder')
for child in box:
box.remove(child)
image: Gtk.Image = Gtk.Image.new_from_surface(pil_to_cairo_surface(self.dtef.get_tiles()[0].convert('RGBA')))
box.pack_start(image, True, True, 0)
box.show_all()
def _init_secondary_terrain(self):
w: Gtk.ComboBox = self.builder.get_object('secondary_terrain_type')
s: Gtk.ListStore = w.get_model()
for v in SecondaryTerrainTableEntry:
s.append([v.value, v.name.capitalize()])
secondary_terrain = HardcodedDungeons.get_secondary_terrains(
self.module.project.get_binary(BinaryName.ARM9), self.module.project.get_rom_module().get_static_data()
)[self.item_id]
w.set_active(secondary_terrain.value)