godot/editor/groups_editor.cpp

184 lines
6.1 KiB
C++
Raw Normal View History

2014-02-10 01:10:30 +00:00
/*************************************************************************/
/* groups_editor.cpp */
/*************************************************************************/
/* This file is part of: */
/* GODOT ENGINE */
/* http://www.godotengine.org */
/*************************************************************************/
/* Copyright (c) 2007-2017 Juan Linietsky, Ariel Manzur. */
/* Copyright (c) 2014-2017 Godot Engine contributors (cf. AUTHORS.md) */
2014-02-10 01:10:30 +00:00
/* */
/* 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 "groups_editor.h"
#include "editor_node.h"
2015-11-26 17:59:04 +00:00
#include "scene/gui/box_container.h"
2014-02-10 01:10:30 +00:00
#include "scene/gui/label.h"
#include "scene/resources/packed_scene.h"
void GroupsEditor::_add_group(const String &p_group) {
2014-02-10 01:10:30 +00:00
2015-11-26 17:59:04 +00:00
if (!node)
return;
2014-02-10 01:10:30 +00:00
2015-11-26 17:59:04 +00:00
String name = group_name->get_text();
if (name.strip_edges() == "")
2015-11-26 17:59:04 +00:00
return;
2014-02-10 01:10:30 +00:00
2015-11-26 17:59:04 +00:00
if (node->is_in_group(name))
2014-02-10 01:10:30 +00:00
return;
undo_redo->create_action(TTR("Add to Group"));
2015-11-26 17:59:04 +00:00
undo_redo->add_do_method(node, "add_to_group", name, true);
undo_redo->add_do_method(this, "update_tree");
undo_redo->add_undo_method(node, "remove_from_group", name);
undo_redo->add_undo_method(this, "update_tree");
undo_redo->add_do_method(EditorNode::get_singleton()->get_scene_tree_dock()->get_tree_editor(), "update_tree"); //to force redraw of scene tree
undo_redo->add_undo_method(EditorNode::get_singleton()->get_scene_tree_dock()->get_tree_editor(), "update_tree"); //to force redraw of scene tree
2014-02-10 01:10:30 +00:00
undo_redo->commit_action();
group_name->clear();
2014-02-10 01:10:30 +00:00
}
2015-11-26 17:59:04 +00:00
void GroupsEditor::_remove_group(Object *p_item, int p_column, int p_id) {
2014-02-10 01:10:30 +00:00
if (!node)
return;
2015-11-26 17:59:04 +00:00
TreeItem *ti = p_item->cast_to<TreeItem>();
if (!ti)
2014-02-10 01:10:30 +00:00
return;
2015-11-26 17:59:04 +00:00
String name = ti->get_text(0);
undo_redo->create_action(TTR("Remove from Group"));
2015-11-26 17:59:04 +00:00
undo_redo->add_do_method(node, "remove_from_group", name);
undo_redo->add_do_method(this, "update_tree");
undo_redo->add_undo_method(node, "add_to_group", name, true);
undo_redo->add_undo_method(this, "update_tree");
undo_redo->add_do_method(EditorNode::get_singleton()->get_scene_tree_dock()->get_tree_editor(), "update_tree"); //to force redraw of scene tree
undo_redo->add_undo_method(EditorNode::get_singleton()->get_scene_tree_dock()->get_tree_editor(), "update_tree"); //to force redraw of scene tree
2015-11-26 17:59:04 +00:00
undo_redo->commit_action();
2014-02-10 01:10:30 +00:00
}
2015-11-26 17:59:04 +00:00
struct _GroupInfoComparator {
bool operator()(const Node::GroupInfo &p_a, const Node::GroupInfo &p_b) const {
2015-11-26 17:59:04 +00:00
return p_a.name.operator String() < p_b.name.operator String();
}
};
2014-02-10 01:10:30 +00:00
void GroupsEditor::update_tree() {
tree->clear();
2015-11-26 17:59:04 +00:00
2014-02-10 01:10:30 +00:00
if (!node)
return;
2015-11-26 17:59:04 +00:00
List<Node::GroupInfo> groups;
2014-02-10 01:10:30 +00:00
node->get_groups(&groups);
2015-11-26 17:59:04 +00:00
groups.sort_custom<_GroupInfoComparator>();
TreeItem *root = tree->create_item();
2015-11-26 17:59:04 +00:00
for (List<GroupInfo>::Element *E = groups.front(); E; E = E->next()) {
2015-11-26 17:59:04 +00:00
Node::GroupInfo gi = E->get();
if (!gi.persistent)
2014-02-10 01:10:30 +00:00
continue;
2015-11-26 17:59:04 +00:00
Node *n = node;
bool can_be_deleted = true;
while (n) {
Ref<SceneState> ss = (n == EditorNode::get_singleton()->get_edited_scene()) ? n->get_scene_inherited_state() : n->get_scene_instance_state();
if (ss.is_valid()) {
int path = ss->find_node_by_path(n->get_path_to(node));
if (path != -1) {
if (ss->is_node_in_group(path, gi.name)) {
can_be_deleted = false;
}
}
}
n = n->get_owner();
}
TreeItem *item = tree->create_item(root);
2015-11-26 17:59:04 +00:00
item->set_text(0, gi.name);
if (can_be_deleted) {
item->add_button(0, get_icon("Remove", "EditorIcons"), 0);
} else {
item->set_selectable(0, false);
}
2014-02-10 01:10:30 +00:00
}
}
void GroupsEditor::set_current(Node *p_node) {
2015-11-26 17:59:04 +00:00
node = p_node;
2014-02-10 01:10:30 +00:00
update_tree();
}
void GroupsEditor::_bind_methods() {
2015-11-26 17:59:04 +00:00
ClassDB::bind_method("_add_group", &GroupsEditor::_add_group);
ClassDB::bind_method("_remove_group", &GroupsEditor::_remove_group);
ClassDB::bind_method("update_tree", &GroupsEditor::update_tree);
2014-02-10 01:10:30 +00:00
}
GroupsEditor::GroupsEditor() {
node = NULL;
2015-11-26 17:59:04 +00:00
VBoxContainer *vbc = this;
2015-11-26 17:59:04 +00:00
HBoxContainer *hbc = memnew(HBoxContainer);
vbc->add_child(hbc);
2015-11-26 17:59:04 +00:00
group_name = memnew(LineEdit);
2015-11-26 17:59:04 +00:00
group_name->set_h_size_flags(SIZE_EXPAND_FILL);
hbc->add_child(group_name);
group_name->connect("text_entered", this, "_add_group");
2015-11-26 17:59:04 +00:00
add = memnew(Button);
add->set_text(TTR("Add"));
2015-11-26 17:59:04 +00:00
hbc->add_child(add);
add->connect("pressed", this, "_add_group", varray(String()));
2014-02-10 01:10:30 +00:00
tree = memnew(Tree);
2015-11-26 17:59:04 +00:00
tree->set_hide_root(true);
tree->set_v_size_flags(SIZE_EXPAND_FILL);
vbc->add_child(tree);
tree->connect("button_pressed", this, "_remove_group");
add_constant_override("separation", 3 * EDSCALE);
2014-02-10 01:10:30 +00:00
}
GroupsEditor::~GroupsEditor() {
2014-02-10 01:10:30 +00:00
}