godot/tools/editor/scenes_dock.cpp
2014-02-23 17:12:36 +08:00

310 lines
8.4 KiB
C++

/*************************************************************************/
/* scenes_dock.cpp */
/*************************************************************************/
/* This file is part of: */
/* GODOT ENGINE */
/* http://www.godotengine.org */
/*************************************************************************/
/* Copyright (c) 2007-2014 Juan Linietsky, Ariel Manzur. */
/* */
/* 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 "scenes_dock.h"
#include "os/dir_access.h"
#include "os/file_access.h"
#include "globals.h"
#include "scene/io/scene_loader.h"
#include "io/resource_loader.h"
#include "os/os.h"
#include "editor_node.h"
bool ScenesDock::_create_tree(TreeItem *p_parent,EditorFileSystemDirectory *p_dir) {
TreeItem *item = tree->create_item(p_parent);
item->set_text(0,p_dir->get_name()+"/");
item->set_icon(0,get_icon("Folder","EditorIcons"));
item->set_custom_bg_color(0,get_color("prop_subsection","Editor"));
bool has_items=false;
for(int i=0;i<p_dir->get_subdir_count();i++) {
if (_create_tree(item,p_dir->get_subdir(i)))
has_items=true;
}
for (int i=0;i<p_dir->get_file_count();i++) {
bool isfave = favorites.has(p_dir->get_file_path(i));
if (button_favorite->is_pressed() && !isfave)
continue;
TreeItem *fitem = tree->create_item(item);
fitem->set_cell_mode(0,TreeItem::CELL_MODE_CHECK);
fitem->set_editable(0,true);
fitem->set_checked(0,isfave);
fitem->set_text(0,p_dir->get_file(i));
Ref<Texture> icon = get_icon( (has_icon(p_dir->get_file_type(i),"EditorIcons")?p_dir->get_file_type(i):String("Object")),"EditorIcons");
fitem->set_icon(0, icon );
fitem->set_metadata(0,p_dir->get_file_path(i));
//if (p_dir->files[i]->icon.is_valid()) {
// fitem->set_icon(0,p_dir->files[i]->icon);
// }
has_items=true;
}
if (!has_items) {
memdelete(item);
return false;
}
return true;
}
void ScenesDock::_update_tree() {
tree->clear();
updating_tree=true;
_create_tree(NULL,EditorFileSystem::get_singleton()->get_filesystem());
updating_tree=false;
}
void ScenesDock::_notification(int p_what) {
switch(p_what) {
case NOTIFICATION_ENTER_SCENE: {
EditorFileSystem::get_singleton()->connect("filesystem_changed",this,"_update_tree");
button_reload->set_icon( get_icon("Reload","EditorIcons"));
button_favorite->set_icon( get_icon("Favorites","EditorIcons"));
button_instance->set_icon( get_icon("Add","EditorIcons"));
button_open->set_icon( get_icon("Folder","EditorIcons"));
String path = Globals::get_singleton()->get_resource_path()+"/favorites.cfg";
FileAccess *f=FileAccess::open(path,FileAccess::READ);
if (f) {
String l = f->get_line();
while(l!="") {
favorites.insert(l);
l = f->get_line();
}
f->close();
memdelete(f);
}
_update_tree(); //maybe it finished already
} break;
case NOTIFICATION_EXIT_SCENE: {
} break;
case NOTIFICATION_PROCESS: {
} break;
}
}
void ScenesDock::_favorite_toggled() {
if (updating_tree)
return;
TreeItem *sel = tree->get_selected();
if (!sel)
return; //?
bool faved = sel->is_checked(0);
String path = sel->get_metadata(0);
if (faved)
favorites.insert(path);
else
favorites.erase(path);
timer->start();
}
void ScenesDock::_favorites_toggled(bool p_toggled) {
_update_tree();
}
String ScenesDock::get_selected_path() const {
TreeItem *sel = tree->get_selected();
if (!sel)
return "";
String path = sel->get_metadata(0);
return "res://"+path;
}
void ScenesDock::_instance_pressed() {
TreeItem *sel = tree->get_selected();
if (!sel)
return;
String path = sel->get_metadata(0);
emit_signal("instance",path);
}
void ScenesDock::_open_pressed(){
TreeItem *sel = tree->get_selected();
if (!sel)
return;
String path = sel->get_metadata(0);
if (ResourceLoader::get_resource_type(path)=="PackedScene") {
editor->open_request(path);
} else {
/*
RES res = ResourceLoader::load(path);
if (res.is_null()) {
return;
}*/
editor->load_resource(path);
}
// emit_signal("open",path);
}
void ScenesDock::_save_favorites() {
String path = Globals::get_singleton()->get_resource_path()+"/favorites.cfg";
FileAccess *f=FileAccess::open(path,FileAccess::WRITE);
ERR_FAIL_COND(!f);
for(Set<String>::Element *E=favorites.front();E;E=E->next() ) {
CharString utf8f = E->get().utf8();
f->store_buffer((const uint8_t*)utf8f.get_data(),utf8f.length());
f->store_8('\n');
}
f->close();
memdelete(f);
}
void ScenesDock::_rescan() {
EditorFileSystem::get_singleton()->scan();
}
void ScenesDock::_bind_methods() {
ObjectTypeDB::bind_method(_MD("_update_tree"),&ScenesDock::_update_tree);
ObjectTypeDB::bind_method(_MD("_rescan"),&ScenesDock::_rescan);
ObjectTypeDB::bind_method(_MD("_favorites_toggled"),&ScenesDock::_favorites_toggled);
ObjectTypeDB::bind_method(_MD("_favorite_toggled"),&ScenesDock::_favorite_toggled);
ObjectTypeDB::bind_method(_MD("_instance_pressed"),&ScenesDock::_instance_pressed);
ObjectTypeDB::bind_method(_MD("_open_pressed"),&ScenesDock::_open_pressed);
ObjectTypeDB::bind_method(_MD("_save_favorites"),&ScenesDock::_save_favorites);
ADD_SIGNAL(MethodInfo("instance"));
ADD_SIGNAL(MethodInfo("open"));
}
ScenesDock::ScenesDock(EditorNode *p_editor) {
editor=p_editor;
tree = memnew( Tree );
add_child(tree);
tree->set_area_as_parent_rect();
tree->set_anchor_and_margin(MARGIN_TOP,Control::ANCHOR_BEGIN,25);
tree->connect("item_edited",this,"_favorite_toggled");
button_reload = memnew( Button );
button_reload->set_pos(Point2(3,2));
button_reload->set_size(Point2(20,5));
button_reload->set_flat(true);
add_child(button_reload);
button_reload->connect("pressed",this,"_rescan");
button_favorite = memnew( Button );
button_favorite->set_pos(Point2(28,2));
button_favorite->set_size(Point2(20,5));
button_favorite->set_flat(true);
button_favorite->set_toggle_mode(true);
add_child(button_favorite);
button_favorite->connect("toggled",this,"_favorites_toggled");
button_instance = memnew( Button );
button_instance->set_anchor(MARGIN_LEFT,ANCHOR_END);
button_instance->set_anchor(MARGIN_RIGHT,ANCHOR_END);
button_instance->set_begin(Point2(3+20,2));
button_instance->set_end(Point2(2+15,5));
button_instance->set_flat(true);
add_child(button_instance);
button_instance->connect("pressed",this,"_instance_pressed");
button_open = memnew( Button );
button_open->set_anchor(MARGIN_LEFT,ANCHOR_END);
button_open->set_anchor(MARGIN_RIGHT,ANCHOR_END);
button_open->set_begin(Point2(3+45,2));
button_open->set_end(Point2(2+34,5));
button_open->set_flat(true);
add_child(button_open);
button_open->connect("pressed",this,"_open_pressed");
timer = memnew( Timer );
timer->set_one_shot(true);
timer->set_wait_time(2);
timer->connect("timeout",this,"_save_favorites");
add_child(timer);
updating_tree=false;
}
ScenesDock::~ScenesDock() {
}