diff options
Diffstat (limited to 'editor/plugins/cube_grid_theme_editor_plugin.cpp')
-rw-r--r-- | editor/plugins/cube_grid_theme_editor_plugin.cpp | 357 |
1 files changed, 357 insertions, 0 deletions
diff --git a/editor/plugins/cube_grid_theme_editor_plugin.cpp b/editor/plugins/cube_grid_theme_editor_plugin.cpp new file mode 100644 index 0000000000..b8abd2e9cc --- /dev/null +++ b/editor/plugins/cube_grid_theme_editor_plugin.cpp @@ -0,0 +1,357 @@ +/*************************************************************************/ +/* cube_grid_theme_editor_plugin.cpp */ +/*************************************************************************/ +/* This file is part of: */ +/* GODOT ENGINE */ +/* http://www.godotengine.org */ +/*************************************************************************/ +/* Copyright (c) 2007-2017 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 "cube_grid_theme_editor_plugin.h" + +#if 0 +#include "scene/3d/mesh_instance.h" +#include "scene/3d/physics_body.h" +#include "scene/main/viewport.h" +#include "scene/resources/packed_scene.h" +#include "editor/editor_node.h" +#include "main/main.h" +#include "editor/editor_settings.h" +#include "scene/3d/navigation_mesh.h" + +void MeshLibraryEditor::edit(const Ref<MeshLibrary>& p_theme) { + + theme=p_theme; + if (theme.is_valid()) + menu->get_popup()->set_item_disabled(menu->get_popup()->get_item_index(MENU_OPTION_UPDATE_FROM_SCENE),!theme->has_meta("_editor_source_scene")); + +} + +void MeshLibraryEditor::_menu_confirm() { + + + switch(option) { + + case MENU_OPTION_REMOVE_ITEM: { + + theme->remove_item(to_erase); + } break; + case MENU_OPTION_UPDATE_FROM_SCENE: { + String existing = theme->get_meta("_editor_source_scene"); + ERR_FAIL_COND(existing==""); + _import_scene_cbk(existing); + + } break; + default: {}; + } + +} + + +void MeshLibraryEditor::_import_scene(Node *p_scene, Ref<MeshLibrary> p_library, bool p_merge) { + + if (!p_merge) + p_library->clear(); + + + for(int i=0;i<p_scene->get_child_count();i++) { + + + Node *child = p_scene->get_child(i); + + if (!child->cast_to<MeshInstance>()) { + if (child->get_child_count()>0) { + child=child->get_child(0); + if (!child->cast_to<MeshInstance>()) { + continue; + } + + } else + continue; + + + } + + MeshInstance *mi = child->cast_to<MeshInstance>(); + Ref<Mesh> mesh=mi->get_mesh(); + if (mesh.is_null()) + continue; + + int id = p_library->find_item_name(mi->get_name()); + if (id<0) { + + id=p_library->get_last_unused_item_id(); + p_library->create_item(id); + p_library->set_item_name(id,mi->get_name()); + } + + + p_library->set_item_mesh(id,mesh); + + Ref<Shape> collision; + + + for(int j=0;j<mi->get_child_count();j++) { +#if 1 + Node *child2 = mi->get_child(j); + if (!child2->cast_to<StaticBody>()) + continue; + StaticBody *sb = child2->cast_to<StaticBody>(); + if (sb->get_shape_count()==0) + continue; + collision=sb->get_shape(0); + if (!collision.is_null()) + break; +#endif + } + + if (!collision.is_null()) { + + p_library->set_item_shape(id,collision); + } + Ref<NavigationMesh> navmesh; + for(int j=0;j<mi->get_child_count();j++) { + Node *child2 = mi->get_child(j); + if (!child2->cast_to<NavigationMeshInstance>()) + continue; + NavigationMeshInstance *sb = child2->cast_to<NavigationMeshInstance>(); + navmesh=sb->get_navigation_mesh(); + if (!navmesh.is_null()) + break; + } + if(!navmesh.is_null()){ + p_library->set_item_navmesh(id, navmesh); + } + } + + //generate previews! + + if (1) { + Vector<int> ids = p_library->get_item_list(); + RID vp = VS::get_singleton()->viewport_create(); + VS::ViewportRect vr; + vr.x=0; + vr.y=0; + vr.width=EditorSettings::get_singleton()->get("editors/grid_map/preview_size"); + vr.height=EditorSettings::get_singleton()->get("editors/grid_map/preview_size"); + VS::get_singleton()->viewport_set_rect(vp,vr); + VS::get_singleton()->viewport_set_as_render_target(vp,true); + VS::get_singleton()->viewport_set_render_target_update_mode(vp,VS::RENDER_TARGET_UPDATE_ALWAYS); + RID scen = VS::get_singleton()->scenario_create(); + VS::get_singleton()->viewport_set_scenario(vp,scen); + RID cam = VS::get_singleton()->camera_create(); + VS::get_singleton()->camera_set_transform(cam, Transform() ); + VS::get_singleton()->viewport_attach_camera(vp,cam); + RID light = VS::get_singleton()->light_create(VS::LIGHT_DIRECTIONAL); + RID lightinst = VS::get_singleton()->instance_create2(light,scen); + VS::get_singleton()->camera_set_orthogonal(cam,1.0,0.01,1000.0); + + + EditorProgress ep("mlib",TTR("Creating Mesh Library"),ids.size()); + + for(int i=0;i<ids.size();i++) { + + int id=ids[i]; + Ref<Mesh> mesh = p_library->get_item_mesh(id); + if (!mesh.is_valid()) + continue; + AABB aabb= mesh->get_aabb(); + print_line("aabb: "+aabb); + Vector3 ofs = aabb.pos + aabb.size*0.5; + aabb.pos-=ofs; + Transform xform; + xform.basis=Matrix3().rotated(Vector3(0,1,0),-Math_PI*0.25); + xform.basis = Matrix3().rotated(Vector3(1,0,0),Math_PI*0.25)*xform.basis; + AABB rot_aabb = xform.xform(aabb); + print_line("rot_aabb: "+rot_aabb); + float m = MAX(rot_aabb.size.x,rot_aabb.size.y)*0.5; + if (m==0) + continue; + m=1.0/m; + m*=0.5; + print_line("scale: "+rtos(m)); + xform.basis.scale(Vector3(m,m,m)); + xform.origin=-xform.basis.xform(ofs); //-ofs*m; + xform.origin.z-=rot_aabb.size.z*2; + RID inst = VS::get_singleton()->instance_create2(mesh->get_rid(),scen); + VS::get_singleton()->instance_set_transform(inst,xform); + ep.step(TTR("Thumbnail.."),i); + VS::get_singleton()->viewport_queue_screen_capture(vp); + Main::iteration(); + Image img = VS::get_singleton()->viewport_get_screen_capture(vp); + ERR_CONTINUE(img.empty()); + Ref<ImageTexture> it( memnew( ImageTexture )); + it->create_from_image(img); + p_library->set_item_preview(id,it); + + //print_line("loaded image, size: "+rtos(m)+" dist: "+rtos(dist)+" empty?"+itos(img.empty())+" w: "+itos(it->get_width())+" h: "+itos(it->get_height())); + VS::get_singleton()->free(inst); + } + + VS::get_singleton()->free(lightinst); + VS::get_singleton()->free(light); + VS::get_singleton()->free(vp); + VS::get_singleton()->free(cam); + VS::get_singleton()->free(scen); + } + + +} + + +void MeshLibraryEditor::_import_scene_cbk(const String& p_str) { + + + print_line("Impot Callback!"); + + Ref<PackedScene> ps = ResourceLoader::load(p_str,"PackedScene"); + ERR_FAIL_COND(ps.is_null()); + Node *scene = ps->instance(); + + _import_scene(scene,theme,option==MENU_OPTION_UPDATE_FROM_SCENE); + + memdelete(scene); + theme->set_meta("_editor_source_scene",p_str); + menu->get_popup()->set_item_disabled(menu->get_popup()->get_item_index(MENU_OPTION_UPDATE_FROM_SCENE),false); + +} + +Error MeshLibraryEditor::update_library_file(Node *p_base_scene, Ref<MeshLibrary> ml,bool p_merge) { + + _import_scene(p_base_scene,ml,p_merge); + return OK; +} + + +void MeshLibraryEditor::_menu_cbk(int p_option) { + + option=p_option; + switch(p_option) { + + case MENU_OPTION_ADD_ITEM: { + + theme->create_item(theme->get_last_unused_item_id()); + } break; + case MENU_OPTION_REMOVE_ITEM: { + + String p = editor->get_property_editor()->get_selected_path(); + if (p.begins_with("/MeshLibrary/item") && p.get_slice_count("/")>=3) { + + to_erase = p.get_slice("/",3).to_int(); + cd->set_text(vformat(TTR("Remove item %d?"),to_erase)); + cd->popup_centered(Size2(300,60)); + } + } break; + case MENU_OPTION_IMPORT_FROM_SCENE: { + + file->popup_centered_ratio(); + } break; + case MENU_OPTION_UPDATE_FROM_SCENE: { + + cd->set_text("Update from existing scene?:\n"+String(theme->get_meta("_editor_source_scene"))); + cd->popup_centered(Size2(500,60)); + } break; + } +} + + +void MeshLibraryEditor::_bind_methods() { + + ClassDB::bind_method("_menu_cbk",&MeshLibraryEditor::_menu_cbk); + ClassDB::bind_method("_menu_confirm",&MeshLibraryEditor::_menu_confirm); + ClassDB::bind_method("_import_scene_cbk",&MeshLibraryEditor::_import_scene_cbk); +} + +MeshLibraryEditor::MeshLibraryEditor(EditorNode *p_editor) { + + file = memnew( EditorFileDialog ); + file->set_mode(EditorFileDialog::MODE_OPEN_FILE); + //not for now? + List<String> extensions; + ResourceLoader::get_recognized_extensions_for_type("PackedScene",&extensions); + file->clear_filters(); + file->set_title(TTR("Import Scene")); + for(int i=0;i<extensions.size();i++) { + + file->add_filter("*."+extensions[i]+" ; "+extensions[i].to_upper()); + } + add_child(file); + file->connect("file_selected",this,"_import_scene_cbk"); + + Panel *panel = memnew( Panel ); + panel->set_area_as_parent_rect(); + add_child(panel); + MenuButton * options = memnew( MenuButton ); + panel->add_child(options); + options->set_pos(Point2(1,1)); + options->set_text("Theme"); + options->get_popup()->add_item(TTR("Add Item"),MENU_OPTION_ADD_ITEM); + options->get_popup()->add_item(TTR("Remove Selected Item"),MENU_OPTION_REMOVE_ITEM); + options->get_popup()->add_separator(); + options->get_popup()->add_item(TTR("Import from Scene"),MENU_OPTION_IMPORT_FROM_SCENE); + options->get_popup()->add_item(TTR("Update from Scene"),MENU_OPTION_UPDATE_FROM_SCENE); + options->get_popup()->set_item_disabled(options->get_popup()->get_item_index(MENU_OPTION_UPDATE_FROM_SCENE),true); + options->get_popup()->connect("id_pressed", this,"_menu_cbk"); + menu=options; + editor=p_editor; + cd = memnew(ConfirmationDialog); + add_child(cd); + cd->get_ok()->connect("pressed", this,"_menu_confirm"); + +} + +void MeshLibraryEditorPlugin::edit(Object *p_node) { + + if (p_node && p_node->cast_to<MeshLibrary>()) { + theme_editor->edit( p_node->cast_to<MeshLibrary>() ); + theme_editor->show(); + } else + theme_editor->hide(); +} + +bool MeshLibraryEditorPlugin::handles(Object *p_node) const{ + + return p_node->is_type("MeshLibrary"); +} + +void MeshLibraryEditorPlugin::make_visible(bool p_visible){ + + if (p_visible) + theme_editor->show(); + else + theme_editor->hide(); +} + +MeshLibraryEditorPlugin::MeshLibraryEditorPlugin(EditorNode *p_node) { + + EDITOR_DEF("editors/grid_map/preview_size",64); + theme_editor = memnew( MeshLibraryEditor(p_node) ); + + p_node->get_viewport()->add_child(theme_editor); + theme_editor->set_area_as_parent_rect(); + theme_editor->set_anchor( MARGIN_RIGHT, Control::ANCHOR_END ); + theme_editor->set_anchor( MARGIN_BOTTOM, Control::ANCHOR_BEGIN ); + theme_editor->set_end( Point2(0,22) ); + theme_editor->hide(); + +} +#endif |