746dddc067
* Map is unnecessary and inefficient in almost every case. * Replaced by the new HashMap. * Renamed Map to RBMap and Set to RBSet for cases that still make sense (order matters) but use is discouraged. There were very few cases where replacing by HashMap was undesired because keeping the key order was intended. I tried to keep those (as RBMap) as much as possible, but might have missed some. Review appreciated!
328 lines
14 KiB
C++
328 lines
14 KiB
C++
/*************************************************************************/
|
|
/* atlas_merging_dialog.cpp */
|
|
/*************************************************************************/
|
|
/* This file is part of: */
|
|
/* GODOT ENGINE */
|
|
/* https://godotengine.org */
|
|
/*************************************************************************/
|
|
/* Copyright (c) 2007-2022 Juan Linietsky, Ariel Manzur. */
|
|
/* Copyright (c) 2014-2022 Godot Engine contributors (cf. AUTHORS.md). */
|
|
/* */
|
|
/* Permission is hereby granted, free of charge, to any person obtaining */
|
|
/* a copy of this software and associated documentation files (the */
|
|
/* "Software"), to deal in the Software without restriction, including */
|
|
/* without limitation the rights to use, copy, modify, merge, publish, */
|
|
/* distribute, sublicense, and/or sell copies of the Software, and to */
|
|
/* permit persons to whom the Software is furnished to do so, subject to */
|
|
/* the following conditions: */
|
|
/* */
|
|
/* The above copyright notice and this permission notice shall be */
|
|
/* included in all copies or substantial portions of the Software. */
|
|
/* */
|
|
/* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, */
|
|
/* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF */
|
|
/* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.*/
|
|
/* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY */
|
|
/* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, */
|
|
/* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE */
|
|
/* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. */
|
|
/*************************************************************************/
|
|
|
|
#include "atlas_merging_dialog.h"
|
|
|
|
#include "editor/editor_file_dialog.h"
|
|
#include "editor/editor_node.h"
|
|
#include "editor/editor_scale.h"
|
|
|
|
#include "scene/gui/control.h"
|
|
#include "scene/gui/split_container.h"
|
|
|
|
void AtlasMergingDialog::_property_changed(const StringName &p_property, const Variant &p_value, const String &p_field, bool p_changing) {
|
|
_set(p_property, p_value);
|
|
}
|
|
|
|
void AtlasMergingDialog::_generate_merged(Vector<Ref<TileSetAtlasSource>> p_atlas_sources, int p_max_columns) {
|
|
merged.instantiate();
|
|
merged_mapping.clear();
|
|
|
|
if (p_atlas_sources.size() >= 2) {
|
|
Ref<Image> output_image;
|
|
output_image.instantiate();
|
|
output_image->create(1, 1, false, Image::FORMAT_RGBA8);
|
|
|
|
// Compute the new texture region size.
|
|
Vector2i new_texture_region_size;
|
|
for (int source_index = 0; source_index < p_atlas_sources.size(); source_index++) {
|
|
Ref<TileSetAtlasSource> atlas_source = p_atlas_sources[source_index];
|
|
new_texture_region_size = new_texture_region_size.max(atlas_source->get_texture_region_size());
|
|
}
|
|
|
|
// Generate the merged TileSetAtlasSource.
|
|
Vector2i atlas_offset;
|
|
int line_height = 0;
|
|
for (int source_index = 0; source_index < p_atlas_sources.size(); source_index++) {
|
|
Ref<TileSetAtlasSource> atlas_source = p_atlas_sources[source_index];
|
|
merged_mapping.push_back(HashMap<Vector2i, Vector2i>());
|
|
|
|
// Layout the tiles.
|
|
Vector2i atlas_size;
|
|
|
|
for (int tile_index = 0; tile_index < atlas_source->get_tiles_count(); tile_index++) {
|
|
Vector2i tile_id = atlas_source->get_tile_id(tile_index);
|
|
atlas_size = atlas_size.max(tile_id + atlas_source->get_tile_size_in_atlas(tile_id));
|
|
|
|
Rect2i new_tile_rect_in_altas = Rect2i(atlas_offset + tile_id, atlas_source->get_tile_size_in_atlas(tile_id));
|
|
|
|
// Create tiles and alternatives, then copy their properties.
|
|
for (int alternative_index = 0; alternative_index < atlas_source->get_alternative_tiles_count(tile_id); alternative_index++) {
|
|
int alternative_id = atlas_source->get_alternative_tile_id(tile_id, alternative_index);
|
|
if (alternative_id == 0) {
|
|
merged->create_tile(new_tile_rect_in_altas.position, new_tile_rect_in_altas.size);
|
|
} else {
|
|
merged->create_alternative_tile(new_tile_rect_in_altas.position, alternative_index);
|
|
}
|
|
|
|
// Copy the properties.
|
|
TileData *original_tile_data = atlas_source->get_tile_data(tile_id, alternative_id);
|
|
List<PropertyInfo> properties;
|
|
original_tile_data->get_property_list(&properties);
|
|
for (List<PropertyInfo>::Element *E = properties.front(); E; E = E->next()) {
|
|
const StringName &property_name = E->get().name;
|
|
merged->set(property_name, original_tile_data->get(property_name));
|
|
}
|
|
|
|
// Add to the mapping.
|
|
merged_mapping[source_index][tile_id] = new_tile_rect_in_altas.position;
|
|
}
|
|
|
|
// Copy the texture.
|
|
for (int frame = 0; frame < atlas_source->get_tile_animation_frames_count(tile_id); frame++) {
|
|
Rect2i src_rect = atlas_source->get_tile_texture_region(tile_id, frame);
|
|
Rect2 dst_rect_wide = Rect2i(new_tile_rect_in_altas.position * new_texture_region_size, new_tile_rect_in_altas.size * new_texture_region_size);
|
|
if (dst_rect_wide.get_end().x > output_image->get_width() || dst_rect_wide.get_end().y > output_image->get_height()) {
|
|
output_image->crop(MAX(dst_rect_wide.get_end().x, output_image->get_width()), MAX(dst_rect_wide.get_end().y, output_image->get_height()));
|
|
}
|
|
output_image->blit_rect(atlas_source->get_texture()->get_image(), src_rect, dst_rect_wide.get_center() - src_rect.size / 2);
|
|
}
|
|
}
|
|
|
|
// Compute the atlas offset.
|
|
line_height = MAX(atlas_size.y, line_height);
|
|
atlas_offset.x += atlas_size.x;
|
|
if (atlas_offset.x >= p_max_columns) {
|
|
atlas_offset.x = 0;
|
|
atlas_offset.y += line_height;
|
|
line_height = 0;
|
|
}
|
|
}
|
|
|
|
Ref<ImageTexture> output_image_texture;
|
|
output_image_texture.instantiate();
|
|
output_image_texture->create_from_image(output_image);
|
|
|
|
merged->set_name(p_atlas_sources[0]->get_name());
|
|
merged->set_texture(output_image_texture);
|
|
merged->set_texture_region_size(new_texture_region_size);
|
|
}
|
|
}
|
|
|
|
void AtlasMergingDialog::_update_texture() {
|
|
Vector<int> selected = atlas_merging_atlases_list->get_selected_items();
|
|
if (selected.size() >= 2) {
|
|
Vector<Ref<TileSetAtlasSource>> to_merge;
|
|
for (int i = 0; i < selected.size(); i++) {
|
|
int source_id = atlas_merging_atlases_list->get_item_metadata(selected[i]);
|
|
to_merge.push_back(tile_set->get_source(source_id));
|
|
}
|
|
_generate_merged(to_merge, next_line_after_column);
|
|
preview->set_texture(merged->get_texture());
|
|
preview->show();
|
|
select_2_atlases_label->hide();
|
|
get_ok_button()->set_disabled(false);
|
|
merge_button->set_disabled(false);
|
|
} else {
|
|
_generate_merged(Vector<Ref<TileSetAtlasSource>>(), next_line_after_column);
|
|
preview->set_texture(Ref<Texture2D>());
|
|
preview->hide();
|
|
select_2_atlases_label->show();
|
|
get_ok_button()->set_disabled(true);
|
|
merge_button->set_disabled(true);
|
|
}
|
|
}
|
|
|
|
void AtlasMergingDialog::_merge_confirmed(String p_path) {
|
|
ERR_FAIL_COND(!merged.is_valid());
|
|
|
|
Ref<ImageTexture> output_image_texture = merged->get_texture();
|
|
output_image_texture->get_image()->save_png(p_path);
|
|
|
|
Ref<Texture2D> new_texture_resource = ResourceLoader::load(p_path, "Texture2D");
|
|
merged->set_texture(new_texture_resource);
|
|
|
|
undo_redo->create_action(TTR("Merge TileSetAtlasSource"));
|
|
int next_id = tile_set->get_next_source_id();
|
|
undo_redo->add_do_method(*tile_set, "add_source", merged, next_id);
|
|
undo_redo->add_undo_method(*tile_set, "remove_source", next_id);
|
|
|
|
if (delete_original_atlases) {
|
|
// Delete originals if needed.
|
|
Vector<int> selected = atlas_merging_atlases_list->get_selected_items();
|
|
for (int i = 0; i < selected.size(); i++) {
|
|
int source_id = atlas_merging_atlases_list->get_item_metadata(selected[i]);
|
|
Ref<TileSetAtlasSource> tas = tile_set->get_source(source_id);
|
|
undo_redo->add_do_method(*tile_set, "remove_source", source_id);
|
|
undo_redo->add_undo_method(*tile_set, "add_source", tas, source_id);
|
|
|
|
// Add the tile proxies.
|
|
for (int tile_index = 0; tile_index < tas->get_tiles_count(); tile_index++) {
|
|
Vector2i tile_id = tas->get_tile_id(tile_index);
|
|
undo_redo->add_do_method(*tile_set, "set_coords_level_tile_proxy", source_id, tile_id, next_id, merged_mapping[i][tile_id]);
|
|
if (tile_set->has_coords_level_tile_proxy(source_id, tile_id)) {
|
|
Array a = tile_set->get_coords_level_tile_proxy(source_id, tile_id);
|
|
undo_redo->add_undo_method(*tile_set, "set_coords_level_tile_proxy", a[0], a[1]);
|
|
} else {
|
|
undo_redo->add_undo_method(*tile_set, "remove_coords_level_tile_proxy", source_id, tile_id);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
undo_redo->commit_action();
|
|
commited_actions_count++;
|
|
|
|
hide();
|
|
}
|
|
|
|
void AtlasMergingDialog::ok_pressed() {
|
|
delete_original_atlases = false;
|
|
editor_file_dialog->popup_file_dialog();
|
|
}
|
|
|
|
void AtlasMergingDialog::cancel_pressed() {
|
|
for (int i = 0; i < commited_actions_count; i++) {
|
|
undo_redo->undo();
|
|
}
|
|
commited_actions_count = 0;
|
|
}
|
|
|
|
void AtlasMergingDialog::custom_action(const String &p_action) {
|
|
if (p_action == "merge") {
|
|
delete_original_atlases = true;
|
|
editor_file_dialog->popup_file_dialog();
|
|
}
|
|
}
|
|
|
|
bool AtlasMergingDialog::_set(const StringName &p_name, const Variant &p_value) {
|
|
if (p_name == "next_line_after_column" && p_value.get_type() == Variant::INT) {
|
|
next_line_after_column = p_value;
|
|
_update_texture();
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
bool AtlasMergingDialog::_get(const StringName &p_name, Variant &r_ret) const {
|
|
if (p_name == "next_line_after_column") {
|
|
r_ret = next_line_after_column;
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
void AtlasMergingDialog::update_tile_set(Ref<TileSet> p_tile_set) {
|
|
ERR_FAIL_COND(!p_tile_set.is_valid());
|
|
tile_set = p_tile_set;
|
|
|
|
atlas_merging_atlases_list->clear();
|
|
for (int i = 0; i < p_tile_set->get_source_count(); i++) {
|
|
int source_id = p_tile_set->get_source_id(i);
|
|
Ref<TileSetAtlasSource> atlas_source = p_tile_set->get_source(source_id);
|
|
if (atlas_source.is_valid()) {
|
|
Ref<Texture2D> texture = atlas_source->get_texture();
|
|
if (texture.is_valid()) {
|
|
String item_text = vformat("%s (id:%d)", texture->get_path().get_file(), source_id);
|
|
atlas_merging_atlases_list->add_item(item_text, texture);
|
|
atlas_merging_atlases_list->set_item_metadata(-1, source_id);
|
|
}
|
|
}
|
|
}
|
|
|
|
get_ok_button()->set_disabled(true);
|
|
merge_button->set_disabled(true);
|
|
|
|
commited_actions_count = 0;
|
|
}
|
|
|
|
AtlasMergingDialog::AtlasMergingDialog() {
|
|
undo_redo = EditorNode::get_singleton()->get_undo_redo();
|
|
|
|
// Atlas merging window.
|
|
set_title(TTR("Atlas Merging"));
|
|
set_hide_on_ok(false);
|
|
|
|
// Ok buttons
|
|
get_ok_button()->set_text(TTR("Merge (Keep original Atlases)"));
|
|
get_ok_button()->set_disabled(true);
|
|
merge_button = add_button(TTR("Merge"), true, "merge");
|
|
merge_button->set_disabled(true);
|
|
|
|
HSplitContainer *atlas_merging_h_split_container = memnew(HSplitContainer);
|
|
atlas_merging_h_split_container->set_h_size_flags(Control::SIZE_EXPAND_FILL);
|
|
atlas_merging_h_split_container->set_v_size_flags(Control::SIZE_EXPAND_FILL);
|
|
add_child(atlas_merging_h_split_container);
|
|
|
|
// Atlas sources item list.
|
|
atlas_merging_atlases_list = memnew(ItemList);
|
|
atlas_merging_atlases_list->set_fixed_icon_size(Size2i(60, 60) * EDSCALE);
|
|
atlas_merging_atlases_list->set_h_size_flags(Control::SIZE_EXPAND_FILL);
|
|
atlas_merging_atlases_list->set_v_size_flags(Control::SIZE_EXPAND_FILL);
|
|
atlas_merging_atlases_list->set_texture_filter(CanvasItem::TEXTURE_FILTER_NEAREST);
|
|
atlas_merging_atlases_list->set_custom_minimum_size(Size2(100, 200));
|
|
atlas_merging_atlases_list->set_select_mode(ItemList::SELECT_MULTI);
|
|
atlas_merging_atlases_list->connect("multi_selected", callable_mp(this, &AtlasMergingDialog::_update_texture).unbind(2));
|
|
atlas_merging_h_split_container->add_child(atlas_merging_atlases_list);
|
|
|
|
VBoxContainer *atlas_merging_right_panel = memnew(VBoxContainer);
|
|
atlas_merging_right_panel->set_h_size_flags(Control::SIZE_EXPAND_FILL);
|
|
atlas_merging_h_split_container->add_child(atlas_merging_right_panel);
|
|
|
|
// Settings.
|
|
Label *settings_label = memnew(Label);
|
|
settings_label->set_text(TTR("Settings:"));
|
|
atlas_merging_right_panel->add_child(settings_label);
|
|
|
|
columns_editor_property = memnew(EditorPropertyInteger);
|
|
columns_editor_property->set_label(TTR("Next Line After Column"));
|
|
columns_editor_property->set_object_and_property(this, "next_line_after_column");
|
|
columns_editor_property->update_property();
|
|
columns_editor_property->connect("property_changed", callable_mp(this, &AtlasMergingDialog::_property_changed));
|
|
atlas_merging_right_panel->add_child(columns_editor_property);
|
|
|
|
// Preview.
|
|
Label *preview_label = memnew(Label);
|
|
preview_label->set_text(TTR("Preview:"));
|
|
atlas_merging_right_panel->add_child(preview_label);
|
|
|
|
preview = memnew(TextureRect);
|
|
preview->set_h_size_flags(Control::SIZE_EXPAND_FILL);
|
|
preview->set_v_size_flags(Control::SIZE_EXPAND_FILL);
|
|
preview->set_ignore_texture_size(true);
|
|
preview->hide();
|
|
preview->set_stretch_mode(TextureRect::STRETCH_KEEP_ASPECT_CENTERED);
|
|
atlas_merging_right_panel->add_child(preview);
|
|
|
|
select_2_atlases_label = memnew(Label);
|
|
select_2_atlases_label->set_h_size_flags(Control::SIZE_EXPAND_FILL);
|
|
select_2_atlases_label->set_v_size_flags(Control::SIZE_EXPAND_FILL);
|
|
select_2_atlases_label->set_horizontal_alignment(HORIZONTAL_ALIGNMENT_CENTER);
|
|
select_2_atlases_label->set_vertical_alignment(VERTICAL_ALIGNMENT_CENTER);
|
|
select_2_atlases_label->set_text(TTR("Please select two atlases or more."));
|
|
atlas_merging_right_panel->add_child(select_2_atlases_label);
|
|
|
|
// The file dialog to choose the texture path.
|
|
editor_file_dialog = memnew(EditorFileDialog);
|
|
editor_file_dialog->set_file_mode(EditorFileDialog::FILE_MODE_SAVE_FILE);
|
|
editor_file_dialog->add_filter("*.png");
|
|
editor_file_dialog->connect("file_selected", callable_mp(this, &AtlasMergingDialog::_merge_confirmed));
|
|
add_child(editor_file_dialog);
|
|
}
|