asgardius/virtualx-engine
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions
virtualx-engine/modules/gridmap/grid_map_editor_plugin.cpp
Juan Linietsky 59154cccf9 -Changed Godot exit to be clean. 
-Added more debug information on memory cleanliness on exit (if run with -v)
-Fixed several memory leaks, fixes #1731, fixes #755
2015-04-20 19:38:02 -03:00

1475 lines
41 KiB
C++
Raw Blame History

/*************************************************************************/
/* grid_map_editor_plugin.cpp */
/*************************************************************************/
/* This file is part of: */
/* GODOT ENGINE */
/* http://www.godotengine.org */
/*************************************************************************/
/* Copyright (c) 2007-2015 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 "grid_map_editor_plugin.h"
#include "tools/editor/plugins/spatial_editor_plugin.h"
#include "scene/3d/camera.h"
#include "tools/editor/editor_settings.h"
#include "os/keyboard.h"
#include "geometry.h"
void GridMapEditor::_node_removed(Node *p_node) {
if(p_node==node) {
node=NULL;
hide();
theme_pallete->hide();
}
}
void GridMapEditor::_configure() {
if(!node)
return;
update_grid();
}
void GridMapEditor::_menu_option(int p_option) {
switch(p_option) {
case MENU_OPTION_CONFIGURE: {
} break;
case MENU_OPTION_LOCK_VIEW: {
int index=options->get_popup()->get_item_index(MENU_OPTION_LOCK_VIEW);
lock_view=!options->get_popup()->is_item_checked(index);
options->get_popup()->set_item_checked(index,lock_view);
} break;
case MENU_OPTION_CLIP_DISABLED:
case MENU_OPTION_CLIP_ABOVE:
case MENU_OPTION_CLIP_BELOW: {
clip_mode=ClipMode(p_option-MENU_OPTION_CLIP_DISABLED);
for(int i=0;i<3;i++) {
int index=options->get_popup()->get_item_index(MENU_OPTION_CLIP_DISABLED+i);
options->get_popup()->set_item_checked(index,i==clip_mode);
}
_update_clip();
} break;
case MENU_OPTION_X_AXIS:
case MENU_OPTION_Y_AXIS:
case MENU_OPTION_Z_AXIS: {
int new_axis = p_option-MENU_OPTION_X_AXIS;
for(int i=0;i<3;i++) {
int idx=options->get_popup()->get_item_index(MENU_OPTION_X_AXIS+i);
options->get_popup()->set_item_checked(idx,i==new_axis);
}
edit_axis=Vector3::Axis(new_axis);
update_grid();
_update_clip();
} break;
case MENU_OPTION_CURSOR_ROTATE_Y: {
Matrix3 r;
if (input_action==INPUT_DUPLICATE) {
r.set_orthogonal_index(selection.duplicate_rot);
r.rotate(Vector3(0,1,0),Math_PI/2.0);
selection.duplicate_rot=r.get_orthogonal_index();
_update_duplicate_indicator();
break;
}
r.set_orthogonal_index(cursor_rot);
r.rotate(Vector3(0,1,0),Math_PI/2.0);
cursor_rot=r.get_orthogonal_index();
_update_cursor_transform();
} break;
case MENU_OPTION_CURSOR_ROTATE_X: {
Matrix3 r;
if (input_action==INPUT_DUPLICATE) {
r.set_orthogonal_index(selection.duplicate_rot);
r.rotate(Vector3(1,0,0),Math_PI/2.0);
selection.duplicate_rot=r.get_orthogonal_index();
_update_duplicate_indicator();
break;
}
r.set_orthogonal_index(cursor_rot);
r.rotate(Vector3(1,0,0),Math_PI/2.0);
cursor_rot=r.get_orthogonal_index();
_update_cursor_transform();
} break;
case MENU_OPTION_CURSOR_ROTATE_Z: {
Matrix3 r;
if (input_action==INPUT_DUPLICATE) {
r.set_orthogonal_index(selection.duplicate_rot);
r.rotate(Vector3(0,0,1),Math_PI/2.0);
selection.duplicate_rot=r.get_orthogonal_index();
_update_duplicate_indicator();
break;
}
r.set_orthogonal_index(cursor_rot);
r.rotate(Vector3(0,0,1),Math_PI/2.0);
cursor_rot=r.get_orthogonal_index();
_update_cursor_transform();
} break;
case MENU_OPTION_CURSOR_BACK_ROTATE_Y: {
Matrix3 r;
r.set_orthogonal_index(cursor_rot);
r.rotate(Vector3(0,1,0),-Math_PI/2.0);
cursor_rot=r.get_orthogonal_index();
_update_cursor_transform();
} break;
case MENU_OPTION_CURSOR_BACK_ROTATE_X: {
Matrix3 r;
r.set_orthogonal_index(cursor_rot);
r.rotate(Vector3(1,0,0),-Math_PI/2.0);
cursor_rot=r.get_orthogonal_index();
_update_cursor_transform();
} break;
case MENU_OPTION_CURSOR_BACK_ROTATE_Z: {
Matrix3 r;
r.set_orthogonal_index(cursor_rot);
r.rotate(Vector3(0,0,1),-Math_PI/2.0);
cursor_rot=r.get_orthogonal_index();
_update_cursor_transform();
} break;
case MENU_OPTION_CURSOR_CLEAR_ROTATION: {
if (input_action==INPUT_DUPLICATE) {
selection.duplicate_rot=0;
_update_duplicate_indicator();
break;
}
cursor_rot=0;
_update_cursor_transform();
} break;
case MENU_OPTION_DUPLICATE_SELECTS: {
int idx = options->get_popup()->get_item_index(MENU_OPTION_DUPLICATE_SELECTS);
options->get_popup()->set_item_checked( idx, !options->get_popup()->is_item_checked( idx ) );
} break;
case MENU_OPTION_SELECTION_MAKE_AREA:
case MENU_OPTION_SELECTION_MAKE_EXTERIOR_CONNECTOR: {
if (!selection.active)
break;
int area = node->get_unused_area_id();
Error err = node->create_area(area,AABB(selection.begin,selection.end-selection.begin+Vector3(1,1,1)));
if (err!=OK) {
}
if (p_option==MENU_OPTION_SELECTION_MAKE_EXTERIOR_CONNECTOR) {
node->area_set_exterior_portal(area,true);
}
_update_areas_display();
update_areas();
} break;
case MENU_OPTION_REMOVE_AREA: {
if (selected_area<1)
return;
node->erase_area(selected_area);
_update_areas_display();
update_areas();
} break;
case MENU_OPTION_SELECTION_CLEAR: {
if (!selection.active)
return;
_delete_selection();
} break;
}
}
void GridMapEditor::_update_cursor_transform() {
cursor_transform=Transform();
cursor_transform.origin=cursor_origin;
cursor_transform.basis.set_orthogonal_index(cursor_rot);
cursor_transform = node->get_transform() * cursor_transform;
if (cursor_instance.is_valid()) {
VisualServer::get_singleton()->instance_set_transform(cursor_instance,cursor_transform);
VisualServer::get_singleton()->instance_geometry_set_flag(cursor_instance,VS::INSTANCE_FLAG_VISIBLE,cursor_visible);
}
}
void GridMapEditor::_update_selection_transform() {
if (!selection.active) {
Transform xf;
xf.basis.set_zero();
VisualServer::get_singleton()->instance_set_transform(selection_instance,xf);
return;
}
Transform xf;
xf.scale(Vector3(1,1,1)*(Vector3(1,1,1)+(selection.end-selection.begin))*node->get_cell_size());
xf.origin=selection.begin*node->get_cell_size();
VisualServer::get_singleton()->instance_set_transform(selection_instance,node->get_global_transform() * xf);
}
void GridMapEditor::_validate_selection() {
if (!selection.active)
return;
selection.begin=selection.click;
selection.end=selection.current;
if (selection.begin.x>selection.end.x)
SWAP(selection.begin.x,selection.end.x);
if (selection.begin.y>selection.end.y)
SWAP(selection.begin.y,selection.end.y);
if (selection.begin.z>selection.end.z)
SWAP(selection.begin.z,selection.end.z);
_update_selection_transform();
}
bool GridMapEditor::do_input_action(Camera* p_camera,const Point2& p_point,bool p_click) {
if (!spatial_editor)
return false;
if (selected_pallete<0 && input_action!=INPUT_COPY && input_action!=INPUT_SELECT && input_action!=INPUT_DUPLICATE)
return false;
Ref<MeshLibrary> theme = node->get_theme();
if (theme.is_null())
return false;
if (input_action!=INPUT_COPY && input_action!=INPUT_SELECT && input_action!=INPUT_DUPLICATE && !theme->has_item(selected_pallete))
return false;
Camera *camera = p_camera;
Vector3 from = camera->project_ray_origin(p_point);
Vector3 normal = camera->project_ray_normal(p_point);
Transform local_xform = node->get_global_transform().affine_inverse();
Vector<Plane> planes=camera->get_frustum();
from=local_xform.xform(from);
normal=local_xform.basis.xform(normal).normalized();
Plane p;
p.normal[edit_axis]=1.0;
p.d=edit_floor[edit_axis]*node->get_cell_size();
Vector3 inters;
if (!p.intersects_segment(from,from+normal*500,&inters))
return false;
//make sure the intersection is inside the frustum planes, to avoid
//painting on invisible regions
for(int i=0;i<planes.size();i++) {
Plane fp = local_xform.xform(planes[i]);
if (fp.is_point_over(inters))
return false;
}
int cell[3];
float cell_size[3]={node->get_cell_size(),node->get_cell_size(),node->get_cell_size()};
last_mouseover=Vector3(-1,-1,-1);
for(int i=0;i<3;i++) {
if (i==edit_axis)
cell[i]=edit_floor[i];
else {
cell[i]=inters[i]/node->get_cell_size();
if (inters[i]<0)
cell[i]-=1; //compensate negative
grid_ofs[i]=cell[i]*cell_size[i];
}
/*if (cell[i]<0 || cell[i]>=grid_size[i]) {
cursor_visible=false;
_update_cursor_transform();
return false;
}*/
}
last_mouseover=Vector3(cell[0],cell[1],cell[2]);
VS::get_singleton()->instance_set_transform(grid_instance[edit_axis],Transform(Matrix3(),grid_ofs));
if (cursor_instance.is_valid()) {
cursor_origin=(Vector3(cell[0],cell[1],cell[2])+Vector3(0.5*node->get_center_x(),0.5*node->get_center_y(),0.5*node->get_center_z()))*node->get_cell_size();
cursor_visible=true;
_update_cursor_transform();
}
if (input_action==INPUT_DUPLICATE) {
selection.current=Vector3(cell[0],cell[1],cell[2]);
_update_duplicate_indicator();
} else if (input_action==INPUT_SELECT) {
selection.current=Vector3(cell[0],cell[1],cell[2]);
if (p_click)
selection.click=selection.current;
selection.active=true;
_validate_selection();
return true;
} else if (input_action==INPUT_COPY) {
int item=node->get_cell_item(cell[0],cell[1],cell[2]);
if (item>=0) {
selected_pallete=item;
update_pallete();
}
return true;
} if (input_action==INPUT_PAINT) {
SetItem si;
si.pos=Vector3(cell[0],cell[1],cell[2]);
si.new_value=selected_pallete;
si.new_orientation=cursor_rot;
si.old_value=node->get_cell_item(cell[0],cell[1],cell[2]);
si.old_orientation=node->get_cell_item_orientation(cell[0],cell[1],cell[2]);
set_items.push_back(si);
node->set_cell_item(cell[0],cell[1],cell[2],selected_pallete,cursor_rot);
return true;
} else if (input_action==INPUT_ERASE) {
SetItem si;
si.pos=Vector3(cell[0],cell[1],cell[2]);
si.new_value=-1;
si.new_orientation=0;
si.old_value=node->get_cell_item(cell[0],cell[1],cell[2]);
si.old_orientation=node->get_cell_item_orientation(cell[0],cell[1],cell[2]);
set_items.push_back(si);
node->set_cell_item(cell[0],cell[1],cell[2],-1);
return true;
}
return false;
}
void GridMapEditor::_delete_selection() {
if (!selection.active)
return;
undo_redo->create_action("GridMap Delete Selection");
for(int i=selection.begin.x;i<=selection.end.x;i++) {
for(int j=selection.begin.y;j<=selection.end.y;j++) {
for(int k=selection.begin.z;k<=selection.end.z;k++) {
undo_redo->add_do_method(node,"set_cell_item",i,j,k,GridMap::INVALID_CELL_ITEM);
undo_redo->add_undo_method(node,"set_cell_item",i,j,k,node->get_cell_item(i,j,k),node->get_cell_item_orientation(i,j,k));
}
}
}
undo_redo->commit_action();
selection.active=false;
_validate_selection();
}
void GridMapEditor::_update_duplicate_indicator() {
if (!selection.active || input_action!=INPUT_DUPLICATE) {
Transform xf;
xf.basis.set_zero();
VisualServer::get_singleton()->instance_set_transform(duplicate_instance,xf);
return;
}
Transform xf;
xf.scale(Vector3(1,1,1)*(Vector3(1,1,1)+(selection.end-selection.begin))*node->get_cell_size());
xf.origin=(selection.begin+(selection.current-selection.click))*node->get_cell_size();
Matrix3 rot;
rot.set_orthogonal_index(selection.duplicate_rot);
xf.basis = rot * xf.basis;
VisualServer::get_singleton()->instance_set_transform(duplicate_instance,node->get_global_transform() * xf);
}
struct __Item { Vector3 pos; int rot; int item ; };
void GridMapEditor::_duplicate_paste() {
if (!selection.active)
return;
int idx = options->get_popup()->get_item_index(MENU_OPTION_DUPLICATE_SELECTS);
bool reselect = options->get_popup()->is_item_checked( idx );
List< __Item > items;
Matrix3 rot;
rot.set_orthogonal_index(selection.duplicate_rot);
for(int i=selection.begin.x;i<=selection.end.x;i++) {
for(int j=selection.begin.y;j<=selection.end.y;j++) {
for(int k=selection.begin.z;k<=selection.end.z;k++) {
int itm = node->get_cell_item(i,j,k);
if (itm==GridMap::INVALID_CELL_ITEM)
continue;
int orientation = node->get_cell_item_orientation(i,j,k);
__Item item;
Vector3 rel=Vector3(i,j,k)-selection.begin;
rel = rot.xform(rel);
Matrix3 orm;
orm.set_orthogonal_index(orientation);
orm = rot * orm;
item.pos=selection.begin+rel;
item.item=itm;
item.rot=orm.get_orthogonal_index();
items.push_back(item);
}
}
}
Vector3 ofs=selection.current-selection.click;
if (items.size()) {
undo_redo->create_action("GridMap Duplicate Selection");
for(List< __Item >::Element *E=items.front();E;E=E->next()) {
__Item &it=E->get();
Vector3 pos = it.pos+ofs;
undo_redo->add_do_method(node,"set_cell_item",pos.x,pos.y,pos.z,it.item,it.rot);
undo_redo->add_undo_method(node,"set_cell_item",pos.x,pos.y,pos.z,node->get_cell_item(pos.x,pos.y,pos.z),node->get_cell_item_orientation(pos.x,pos.y,pos.z));
}
undo_redo->commit_action();
}
if (reselect) {
selection.begin+=ofs;
selection.end+=ofs;
selection.click=selection.begin;
selection.current=selection.end;
_validate_selection();
}
}
bool GridMapEditor::forward_spatial_input_event(Camera* p_camera,const InputEvent& p_event) {
if (edit_mode->get_selected()==0) { // regular click
switch (p_event.type) {
case InputEvent::KEY: {
if (p_event.key.pressed && p_event.key.scancode==KEY_D && p_event.key.mod.shift && selection.active && input_action==INPUT_NONE) {
if (last_mouseover==Vector3(-1,-1,-1)) //nono mouseovering anythin
return false;
input_action=INPUT_DUPLICATE;
selection.click=last_mouseover;
selection.current=last_mouseover;
selection.duplicate_rot=0;
_update_duplicate_indicator();
}
if (p_event.key.pressed && p_event.key.scancode==KEY_DELETE && selection.active) {
_delete_selection();
return true;
}
} break;
case InputEvent::MOUSE_BUTTON: {
if (p_event.mouse_button.button_index==BUTTON_WHEEL_UP && (p_event.mouse_button.mod.command || p_event.mouse_button.mod.shift)) {
if (p_event.mouse_button.pressed)
floor->set_val( floor->get_val() +1);
return true; //eaten
} else if (p_event.mouse_button.button_index==BUTTON_WHEEL_DOWN && (p_event.mouse_button.mod.command || p_event.mouse_button.mod.shift)) {
if (p_event.mouse_button.pressed)
floor->set_val( floor->get_val() -1);
return true;
}
if (p_event.mouse_button.pressed) {
if (p_event.mouse_button.button_index==BUTTON_LEFT) {
if (input_action==INPUT_DUPLICATE) {
//paste
_duplicate_paste();
input_action=INPUT_NONE;
_update_duplicate_indicator();
} else if (p_event.mouse_button.mod.shift) {
input_action=INPUT_SELECT;
} else if (p_event.mouse_button.mod.command)
input_action=INPUT_COPY;
else {
input_action=INPUT_PAINT;
set_items.clear();
}
} else if (p_event.mouse_button.button_index==BUTTON_RIGHT)
if (input_action==INPUT_DUPLICATE) {
input_action=INPUT_NONE;
_update_duplicate_indicator();
} else {
input_action=INPUT_ERASE;
set_items.clear();
}
else
return false;
return do_input_action(p_camera,Point2(p_event.mouse_button.x,p_event.mouse_button.y),true);
} else {
if (
(p_event.mouse_button.button_index==BUTTON_RIGHT && input_action==INPUT_ERASE) ||
(p_event.mouse_button.button_index==BUTTON_LEFT && input_action==INPUT_PAINT) ) {
if (set_items.size()) {
undo_redo->create_action("GridMap Paint");
for(List<SetItem>::Element *E=set_items.front();E;E=E->next()) {
const SetItem &si=E->get();
undo_redo->add_do_method(node,"set_cell_item",si.pos.x,si.pos.y,si.pos.z,si.new_value,si.new_orientation);
}
for(List<SetItem>::Element *E=set_items.back();E;E=E->prev()) {
const SetItem &si=E->get();
undo_redo->add_undo_method(node,"set_cell_item",si.pos.x,si.pos.y,si.pos.z,si.old_value,si.old_orientation);
}
undo_redo->commit_action();
}
set_items.clear();
input_action=INPUT_NONE;
return true;
}
if (p_event.mouse_button.button_index==BUTTON_LEFT && input_action!=INPUT_NONE) {
set_items.clear();
input_action=INPUT_NONE;
return true;
}
if (p_event.mouse_button.button_index==BUTTON_RIGHT && (input_action==INPUT_ERASE || input_action==INPUT_DUPLICATE)) {
input_action=INPUT_NONE;
return true;
}
}
} break;
case InputEvent::MOUSE_MOTION: {
return do_input_action(p_camera,Point2(p_event.mouse_motion.x,p_event.mouse_motion.y),false);
} break;
}
} else if (edit_mode->get_selected()==1) {
//area mode, select an area
switch (p_event.type) {
case InputEvent::MOUSE_BUTTON: {
if (p_event.mouse_button.button_index==BUTTON_LEFT && p_event.mouse_button.pressed) {
Point2 point = Point2(p_event.mouse_motion.x,p_event.mouse_motion.y);
Camera *camera = p_camera;
Vector3 from = camera->project_ray_origin(point);
Vector3 normal = camera->project_ray_normal(point);
Transform local_xform = node->get_global_transform().affine_inverse();
from=local_xform.xform(from);
normal=local_xform.basis.xform(normal).normalized();
List<int> areas;
node->get_area_list(&areas);
float min_d=1e10;
int min_area=-1;
for(List<int>::Element *E=areas.front();E;E=E->next()) {
int area = E->get();
AABB aabb = node->area_get_bounds(area);
aabb.pos*=node->get_cell_size();
aabb.size*=node->get_cell_size();
Vector3 rclip,rnormal;
if (!aabb.intersects_segment(from,from+normal*10000,&rclip,&rnormal))
continue;
float d = normal.dot(rclip);
if (d<min_d) {
min_d=d;
min_area=area;
}
}
selected_area=min_area;
update_areas();
}
} break;
}
}
return false;
}
struct _CGMEItemSort {
String name;
int id;
_FORCE_INLINE_ bool operator<(const _CGMEItemSort& r_it) const { return name < r_it.name; }
};
void GridMapEditor::update_pallete() {
theme_pallete->clear();
Ref<MeshLibrary> theme = node->get_theme();
if (theme.is_null()) {
last_theme=NULL;
return;
}
Vector<int> ids;
ids = theme->get_item_list();
TreeItem *root = theme_pallete->create_item(NULL);
theme_pallete->set_hide_root(true);
TreeItem *selected=NULL;
List<_CGMEItemSort> il;
for(int i=0;i<ids.size();i++) {
_CGMEItemSort is;
is.id=ids[i];
is.name=theme->get_item_name(ids[i]);
il.push_back(is);
}
il.sort();
int col=0;
TreeItem *ti=NULL;
int selected_col=0;
for(List<_CGMEItemSort>::Element *E=il.front();E;E=E->next()) {
int id = E->get().id;
if (col==0) {
ti = theme_pallete->create_item(root);
}
String name=theme->get_item_name(id);
Ref<Texture> preview = theme->get_item_preview(id);
if (!preview.is_null()) {
ti->set_cell_mode(col,TreeItem::CELL_MODE_ICON);
ti->set_icon(col,preview);
ti->set_tooltip(col,name);
} else {
ti->set_text(col,name);
}
ti->set_metadata(col,id);
if (selected_pallete==id) {
selected=ti;
selected_col=col;
}
col++;
if (col==theme_pallete->get_columns())
col=0;
}
if (selected)
selected->select(selected_col);
last_theme=theme.operator->();
}
void GridMapEditor::_area_renamed() {
TreeItem * it = area_list->get_selected();
int area = it->get_metadata(0);
if (area<1)
return;
node->area_set_name(area,it->get_text(0));
}
void GridMapEditor::_area_selected() {
TreeItem * it = area_list->get_selected();
int area = it->get_metadata(0);
if (area<1)
return;
selected_area=area;
}
void GridMapEditor::update_areas() {
area_list->clear();
List<int> areas;
node->get_area_list(&areas);
TreeItem *root = area_list->create_item(NULL);
area_list->set_hide_root(true);
TreeItem *selected=NULL;
for (List<int>::Element *E=areas.front();E;E=E->next()) {
int area = E->get();
TreeItem *ti = area_list->create_item(root);
String name=node->area_get_name(area);
ti->set_metadata(0,area);
ti->set_text(0,name);
ti->set_editable(0,true);
if (area==selected_area)
selected=ti;
}
if (selected)
selected->select(0);
}
void GridMapEditor::edit(GridMap *p_gridmap) {
node=p_gridmap;
VS *vs = VS::get_singleton();
last_mouseover=Vector3(-1,-1,-1);
spatial_editor = editor->get_editor_plugin_screen()->cast_to<SpatialEditorPlugin>();
if (!node) {
set_process(false);
for(int i=0;i<3;i++) {
VisualServer::get_singleton()->instance_geometry_set_flag(grid_instance[i],VS::INSTANCE_FLAG_VISIBLE,false);
}
_clear_areas();
return;
}
update_pallete();
update_areas();
set_process(true);
Vector3 edited_floor = p_gridmap->get_meta("_editor_floor_");
clip_mode=p_gridmap->has_meta("_editor_clip_")?ClipMode(p_gridmap->get_meta("_editor_clip_").operator int()):CLIP_DISABLED;
for(int i=0;i<3;i++) {
if (vs->mesh_get_surface_count(grid[i])>0)
vs->mesh_remove_surface(grid[i],0);
edit_floor[i]=edited_floor[i];
}
{
//update grids
indicator_mat = VisualServer::get_singleton()->fixed_material_create();
VisualServer::get_singleton()->material_set_flag( indicator_mat, VisualServer::MATERIAL_FLAG_UNSHADED, true );
VisualServer::get_singleton()->material_set_flag( indicator_mat, VisualServer::MATERIAL_FLAG_ONTOP, false );
VisualServer::get_singleton()->fixed_material_set_param(indicator_mat,VisualServer::FIXED_MATERIAL_PARAM_DIFFUSE,Color(0.8,0.5,0.1));
VisualServer::get_singleton()->fixed_material_set_flag( indicator_mat, VisualServer::FIXED_MATERIAL_FLAG_USE_ALPHA, true );
VisualServer::get_singleton()->fixed_material_set_flag( indicator_mat, VisualServer::FIXED_MATERIAL_FLAG_USE_COLOR_ARRAY, true );
Vector<Vector3> grid_points[3];
Vector<Color> grid_colors[3];
float cell_size[3]={p_gridmap->get_cell_size(),p_gridmap->get_cell_size(),p_gridmap->get_cell_size()};
for(int i=0;i<3;i++) {
Vector3 axis;
axis[i]=1;
Vector3 axis_n1;
axis_n1[(i+1)%3]=cell_size[(i+1)%3];;
Vector3 axis_n2;
axis_n2[(i+2)%3]=cell_size[(i+2)%3];
for(int j=-GRID_CURSOR_SIZE;j<=GRID_CURSOR_SIZE;j++) {
for(int k=-GRID_CURSOR_SIZE;k<=GRID_CURSOR_SIZE;k++) {
Vector3 p = axis_n1*j + axis_n2 *k;
float trans = Math::pow(MAX(0,1.0-(Vector2(j,k).length()/GRID_CURSOR_SIZE)),2);
Vector3 pj = axis_n1*(j+1) + axis_n2 *k;
float transj = Math::pow(MAX(0,1.0-(Vector2(j+1,k).length()/GRID_CURSOR_SIZE)),2);
Vector3 pk = axis_n1*j + axis_n2 *(k+1);
float transk = Math::pow(MAX(0,1.0-(Vector2(j,k+1).length()/GRID_CURSOR_SIZE)),2);
grid_points[i].push_back(p);
grid_points[i].push_back(pk);
grid_colors[i].push_back(Color(1,1,1,trans));
grid_colors[i].push_back(Color(1,1,1,transk));
grid_points[i].push_back(p);
grid_points[i].push_back(pj);
grid_colors[i].push_back(Color(1,1,1,trans));
grid_colors[i].push_back(Color(1,1,1,transj));
}
}
Array d;
d.resize(VS::ARRAY_MAX);
d[VS::ARRAY_VERTEX]=grid_points[i];
d[VS::ARRAY_COLOR]=grid_colors[i];
VisualServer::get_singleton()->mesh_add_surface(grid[i],VisualServer::PRIMITIVE_LINES,d);
VisualServer::get_singleton()->mesh_surface_set_material(grid[i],0,indicator_mat);
}
}
update_grid();
_update_clip();
_update_areas_display();
}
void GridMapEditor::_update_clip() {
node->set_meta("_editor_clip_",clip_mode);
if (clip_mode==CLIP_DISABLED)
node->set_clip(false);
else
node->set_clip(true,clip_mode==CLIP_ABOVE,edit_floor[edit_axis],edit_axis);
}
void GridMapEditor::update_grid() {
grid_xform.origin.x-=1; //force update in hackish way.. what do i care
VS *vs = VS::get_singleton();
grid_ofs[edit_axis]=edit_floor[edit_axis]*node->get_cell_size();
edit_grid_xform.origin=grid_ofs;
edit_grid_xform.basis=Matrix3();
for(int i=0;i<3;i++) {
VisualServer::get_singleton()->instance_geometry_set_flag(grid_instance[i],VS::INSTANCE_FLAG_VISIBLE,i==edit_axis);
}
updating=true;
floor->set_val(edit_floor[edit_axis]);
updating=false;
}
void GridMapEditor::_notification(int p_what) {
if (p_what==NOTIFICATION_ENTER_TREE) {
theme_pallete->connect("cell_selected", this,"_item_selected_cbk");
edit_mode->connect("item_selected", this,"_edit_mode_changed");
area_list->connect("item_edited", this,"_area_renamed");
area_list->connect("item_selected", this,"_area_selected");
for(int i=0;i<3;i++) {
grid[i]=VS::get_singleton()->mesh_create();
grid_instance[i]=VS::get_singleton()->instance_create2(grid[i],get_tree()->get_root()->get_world()->get_scenario());
}
selection_instance = VisualServer::get_singleton()->instance_create2(selection_mesh,get_tree()->get_root()->get_world()->get_scenario());
duplicate_instance = VisualServer::get_singleton()->instance_create2(duplicate_mesh,get_tree()->get_root()->get_world()->get_scenario());
_update_selection_transform();
_update_duplicate_indicator();
} else if (p_what==NOTIFICATION_EXIT_TREE) {
for(int i=0;i<3;i++) {
VS::get_singleton()->free(grid_instance[i]);
VS::get_singleton()->free(grid[i]);
grid_instance[i]=RID();
grid[i]=RID();
}
VisualServer::get_singleton()->free(selection_instance);
VisualServer::get_singleton()->free(duplicate_instance);
selection_instance=RID();
duplicate_instance=RID();
} else if (p_what==NOTIFICATION_PROCESS) {
Transform xf = node->get_global_transform();
if (xf!=grid_xform) {
for(int i=0;i<3;i++) {
VS::get_singleton()->instance_set_transform(grid_instance[i],xf * edit_grid_xform);
}
grid_xform=xf;
}
Ref<MeshLibrary> cgmt = node->get_theme();
if (cgmt.operator->()!=last_theme)
update_pallete();
if (lock_view) {
EditorNode*editor = get_tree()->get_root()->get_child(0)->cast_to<EditorNode>();
Plane p;
p.normal[edit_axis]=1.0;
p.d=edit_floor[edit_axis]*node->get_cell_size();
p = node->get_transform().xform(p); // plane to snap
SpatialEditorPlugin *sep = editor->get_editor_plugin_screen()->cast_to<SpatialEditorPlugin>();
if (sep)
sep->snap_cursor_to_plane(p);
//editor->get_editor_plugin_screen()->call("snap_cursor_to_plane",p);
}
}
}
void GridMapEditor::_update_cursor_instance() {
if (cursor_instance.is_valid())
VisualServer::get_singleton()->free(cursor_instance);
cursor_instance=RID();
if (selected_pallete>=0) {
if (node && !node->get_theme().is_null()) {
Ref<Mesh> mesh = node->get_theme()->get_item_mesh(selected_pallete);
if (!mesh.is_null() && mesh->get_rid().is_valid()) {
cursor_instance=VisualServer::get_singleton()->instance_create2(mesh->get_rid(),get_tree()->get_root()->get_world()->get_scenario());
VisualServer::get_singleton()->instance_set_transform(cursor_instance,cursor_transform);
}
}
}
}
void GridMapEditor::_item_selected_cbk() {
TreeItem *it = theme_pallete->get_selected();
if (it) {
selected_pallete=it->get_metadata(theme_pallete->get_selected_column());
} else {
selected_pallete=-1;
}
_update_cursor_instance();
}
void GridMapEditor::_clear_areas() {
for(int i=0;i<areas.size();i++) {
VisualServer::get_singleton()->free(areas[i].instance);
VisualServer::get_singleton()->free(areas[i].mesh);
}
areas.clear();
}
void GridMapEditor::_update_areas_display() {
_clear_areas();
List<int> areas;
node->get_area_list(&areas);
Transform global_xf = node->get_global_transform();
for(List<int>::Element *E=areas.front();E;E=E->next()) {
int area = E->get();
Color color;
if (node->area_is_exterior_portal(area))
color=Color(1,1,1,0.2);
else
color.set_hsv(Math::fmod(area*0.37,1),Math::fmod(area*0.75,1),1.0,0.2);
RID material = VisualServer::get_singleton()->fixed_material_create();
VisualServer::get_singleton()->fixed_material_set_param( material, VS::FIXED_MATERIAL_PARAM_DIFFUSE,color );
VisualServer::get_singleton()->fixed_material_set_param( material, VS::FIXED_MATERIAL_PARAM_EMISSION,0.5 );
VisualServer::get_singleton()->fixed_material_set_flag( material, VisualServer::FIXED_MATERIAL_FLAG_USE_ALPHA, true );
RID mesh = VisualServer::get_singleton()->mesh_create();
DVector<Plane> planes;
for(int i=0;i<3;i++) {
Vector3 axis;
axis[i]=1.0;
planes.push_back(Plane(axis,1));
planes.push_back(Plane(-axis,0));
}
VisualServer::get_singleton()->mesh_add_surface_from_planes(mesh,planes);
VisualServer::get_singleton()->mesh_surface_set_material(mesh,0,material,true);
AreaDisplay ad;
ad.mesh=mesh;
ad.instance = VisualServer::get_singleton()->instance_create2(mesh,node->get_world()->get_scenario());
Transform xform;
AABB aabb = node->area_get_bounds(area);
xform.origin=aabb.pos * node->get_cell_size();
xform.basis.scale(aabb.size * node->get_cell_size());
VisualServer::get_singleton()->instance_set_transform(ad.instance,global_xf * xform);
this->areas.push_back(ad);
}
}
void GridMapEditor::_edit_mode_changed(int p_what) {
if (p_what==0) {
theme_pallete->show();
area_list->hide();
} else {
theme_pallete->hide();
area_list->show();
}
}
void GridMapEditor::_floor_changed(float p_value) {
if (updating)
return;
edit_floor[edit_axis]=p_value;
node->set_meta("_editor_floor_",Vector3(edit_floor[0],edit_floor[1],edit_floor[2]));
update_grid();
_update_clip();
}
void GridMapEditor::_bind_methods() {
ObjectTypeDB::bind_method("_menu_option",&GridMapEditor::_menu_option);
ObjectTypeDB::bind_method("_configure",&GridMapEditor::_configure);
ObjectTypeDB::bind_method("_item_selected_cbk",&GridMapEditor::_item_selected_cbk);
ObjectTypeDB::bind_method("_edit_mode_changed",&GridMapEditor::_edit_mode_changed);
ObjectTypeDB::bind_method("_area_renamed",&GridMapEditor::_area_renamed);
ObjectTypeDB::bind_method("_area_selected",&GridMapEditor::_area_selected);
ObjectTypeDB::bind_method("_floor_changed",&GridMapEditor::_floor_changed);
}
GridMapEditor::GridMapEditor(EditorNode *p_editor) {
input_action=INPUT_NONE;
editor=p_editor;
undo_redo=p_editor->get_undo_redo();
int mw = EDITOR_DEF("grid_map/palette_min_width",230);
Control *ec = memnew( Control);
ec->set_custom_minimum_size(Size2(mw,0));
add_child(ec);
spatial_editor_hb = memnew( HBoxContainer );
SpatialEditor::get_singleton()->add_control_to_menu_panel(spatial_editor_hb);
options = memnew( MenuButton );
spatial_editor_hb->add_child(options);
spatial_editor_hb->hide();
options->set_text("Grid");
options->get_popup()->add_check_item("Snap View",MENU_OPTION_LOCK_VIEW);
options->get_popup()->add_separator();
options->get_popup()->add_item("Prev Level ("+keycode_get_string(KEY_MASK_CMD)+"Down Wheel)",MENU_OPTION_PREV_LEVEL);
options->get_popup()->add_item("Next Level ("+keycode_get_string(KEY_MASK_CMD)+"Up Wheel)",MENU_OPTION_NEXT_LEVEL);
options->get_popup()->add_separator();
options->get_popup()->add_check_item("Clip Disabled",MENU_OPTION_CLIP_DISABLED);
options->get_popup()->set_item_checked( options->get_popup()->get_item_index(MENU_OPTION_CLIP_DISABLED), true );
options->get_popup()->add_check_item("Clip Above",MENU_OPTION_CLIP_ABOVE);
options->get_popup()->add_check_item("Clip Below",MENU_OPTION_CLIP_BELOW);
options->get_popup()->add_separator();
options->get_popup()->add_check_item("Edit X Axis",MENU_OPTION_X_AXIS,KEY_Z);
options->get_popup()->add_check_item("Edit Y Axis",MENU_OPTION_Y_AXIS,KEY_X);
options->get_popup()->add_check_item("Edit Z Axis",MENU_OPTION_Z_AXIS,KEY_C);
options->get_popup()->set_item_checked( options->get_popup()->get_item_index(MENU_OPTION_Y_AXIS), true );
options->get_popup()->add_separator();
options->get_popup()->add_item("Cursor Rotate X",MENU_OPTION_CURSOR_ROTATE_X,KEY_A);
options->get_popup()->add_item("Cursor Rotate Y",MENU_OPTION_CURSOR_ROTATE_Y,KEY_S);
options->get_popup()->add_item("Cursor Rotate Z",MENU_OPTION_CURSOR_ROTATE_Z,KEY_D);
options->get_popup()->add_item("Cursor Back Rotate X",MENU_OPTION_CURSOR_ROTATE_X,KEY_MASK_SHIFT+KEY_A);
options->get_popup()->add_item("Cursor Back Rotate Y",MENU_OPTION_CURSOR_ROTATE_Y,KEY_MASK_SHIFT+KEY_S);
options->get_popup()->add_item("Cursor Back Rotate Z",MENU_OPTION_CURSOR_ROTATE_Z,KEY_MASK_SHIFT+KEY_D);
options->get_popup()->add_item("Cursor Clear Rotation",MENU_OPTION_CURSOR_CLEAR_ROTATION,KEY_W);
options->get_popup()->add_separator();
options->get_popup()->add_check_item("Duplicate Selects",MENU_OPTION_DUPLICATE_SELECTS);
options->get_popup()->add_separator();
options->get_popup()->add_item("Create Area",MENU_OPTION_SELECTION_MAKE_AREA,KEY_CONTROL+KEY_C);
options->get_popup()->add_item("Create Exterior Connector",MENU_OPTION_SELECTION_MAKE_EXTERIOR_CONNECTOR);
options->get_popup()->add_item("Erase Area",MENU_OPTION_REMOVE_AREA);
options->get_popup()->add_separator();
options->get_popup()->add_item("Selection -> Clear",MENU_OPTION_SELECTION_CLEAR);
//options->get_popup()->add_separator();
//options->get_popup()->add_item("Configure",MENU_OPTION_CONFIGURE);
clip_mode=CLIP_DISABLED;
options->get_popup()->connect("item_pressed", this,"_menu_option");
edit_mode = memnew(OptionButton);
edit_mode->set_area_as_parent_rect();
edit_mode->set_anchor_and_margin(MARGIN_BOTTOM,ANCHOR_BEGIN,24);;
edit_mode->set_anchor_and_margin(MARGIN_RIGHT,ANCHOR_END,14);;
edit_mode->add_item("Tiles");
edit_mode->add_item("Areas");
add_child(edit_mode);
selected_area=-1;
theme_pallete = memnew( Tree );
theme_pallete->set_columns(3);
add_child(theme_pallete);
theme_pallete->set_v_size_flags(SIZE_EXPAND_FILL);
area_list = memnew( Tree );
add_child(area_list);
area_list->set_v_size_flags(SIZE_EXPAND_FILL);
area_list->hide();
spatial_editor_hb->add_child(memnew(VSeparator));
Label *fl = memnew(Label);
fl->set_text(" Floor: ");
spatial_editor_hb->add_child(fl);
floor = memnew( SpinBox );
floor->set_min(-32767);
floor->set_max(32767);
floor->set_step(1);
floor->get_line_edit()->add_constant_override("minimum_spaces",16);
spatial_editor_hb->add_child(floor);
floor->connect("value_changed",this,"_floor_changed");
edit_axis=Vector3::AXIS_Y;
edit_floor[0]=-1;
edit_floor[1]=-1;
edit_floor[2]=-1;
cursor_visible=false;
selected_pallete=-1;
lock_view=false;
cursor_rot=0;
last_mouseover=Vector3(-1,-1,-1);
selection_mesh = VisualServer::get_singleton()->mesh_create();
duplicate_mesh = VisualServer::get_singleton()->mesh_create();
{
//selection mesh create
DVector<Vector3> lines;
DVector<Vector3> triangles;
for (int i=0;i<6;i++) {
Vector3 face_points[4];
for (int j=0;j<4;j++) {
float v[3];
v[0]=1.0;
v[1]=1-2*((j>>1)&1);
v[2]=v[1]*(1-2*(j&1));
for (int k=0;k<3;k++) {
if (i<3)
face_points[j][(i+k)%3]=v[k]*(i>=3?-1:1);
else
face_points[3-j][(i+k)%3]=v[k]*(i>=3?-1:1);
}
}
triangles.push_back(face_points[0]*0.5+Vector3(0.5,0.5,0.5));
triangles.push_back(face_points[1]*0.5+Vector3(0.5,0.5,0.5));
triangles.push_back(face_points[2]*0.5+Vector3(0.5,0.5,0.5));
triangles.push_back(face_points[2]*0.5+Vector3(0.5,0.5,0.5));
triangles.push_back(face_points[3]*0.5+Vector3(0.5,0.5,0.5));
triangles.push_back(face_points[0]*0.5+Vector3(0.5,0.5,0.5));
}
for(int i=0;i<12;i++) {
AABB base(Vector3(0,0,0),Vector3(1,1,1));
Vector3 a,b;
base.get_edge(i,a,b);
lines.push_back(a);
lines.push_back(b);
}
Array d;
d.resize(VS::ARRAY_MAX);
inner_mat = VisualServer::get_singleton()->fixed_material_create();
VisualServer::get_singleton()->fixed_material_set_param(inner_mat,VS::FIXED_MATERIAL_PARAM_DIFFUSE,Color(0.7,0.7,1.0,0.3));
VisualServer::get_singleton()->material_set_flag(inner_mat,VS::MATERIAL_FLAG_ONTOP,true);
VisualServer::get_singleton()->material_set_flag(inner_mat,VS::MATERIAL_FLAG_UNSHADED,true);
VisualServer::get_singleton()->fixed_material_set_flag( inner_mat, VisualServer::FIXED_MATERIAL_FLAG_USE_ALPHA, true );
d[VS::ARRAY_VERTEX]=triangles;
VisualServer::get_singleton()->mesh_add_surface(selection_mesh,VS::PRIMITIVE_TRIANGLES,d);
VisualServer::get_singleton()->mesh_surface_set_material(selection_mesh,0,inner_mat);
outer_mat = VisualServer::get_singleton()->fixed_material_create();
VisualServer::get_singleton()->fixed_material_set_param(outer_mat,VS::FIXED_MATERIAL_PARAM_DIFFUSE,Color(0.7,0.7,1.0,0.8));
VisualServer::get_singleton()->material_set_line_width(outer_mat,3.0);
VisualServer::get_singleton()->material_set_flag(outer_mat,VS::MATERIAL_FLAG_ONTOP,true);
VisualServer::get_singleton()->material_set_flag(outer_mat,VS::MATERIAL_FLAG_UNSHADED,true);
VisualServer::get_singleton()->fixed_material_set_flag( outer_mat, VisualServer::FIXED_MATERIAL_FLAG_USE_ALPHA, true );
d[VS::ARRAY_VERTEX]=lines;
VisualServer::get_singleton()->mesh_add_surface(selection_mesh,VS::PRIMITIVE_LINES,d);
VisualServer::get_singleton()->mesh_surface_set_material(selection_mesh,1,outer_mat);
inner_mat_dup = VisualServer::get_singleton()->fixed_material_create();
VisualServer::get_singleton()->fixed_material_set_param(inner_mat_dup,VS::FIXED_MATERIAL_PARAM_DIFFUSE,Color(1.0,0.7,0.7,0.3));
VisualServer::get_singleton()->material_set_flag(inner_mat_dup,VS::MATERIAL_FLAG_ONTOP,true);
VisualServer::get_singleton()->material_set_flag(inner_mat_dup,VS::MATERIAL_FLAG_UNSHADED,true);
VisualServer::get_singleton()->fixed_material_set_flag( inner_mat_dup, VisualServer::FIXED_MATERIAL_FLAG_USE_ALPHA, true );
d[VS::ARRAY_VERTEX]=triangles;
VisualServer::get_singleton()->mesh_add_surface(duplicate_mesh,VS::PRIMITIVE_TRIANGLES,d);
VisualServer::get_singleton()->mesh_surface_set_material(duplicate_mesh,0,inner_mat_dup);
outer_mat_dup = VisualServer::get_singleton()->fixed_material_create();
VisualServer::get_singleton()->fixed_material_set_param(outer_mat_dup,VS::FIXED_MATERIAL_PARAM_DIFFUSE,Color(1.0,0.7,0.7,0.8));
VisualServer::get_singleton()->material_set_line_width(outer_mat_dup,3.0);
VisualServer::get_singleton()->material_set_flag(outer_mat_dup,VS::MATERIAL_FLAG_ONTOP,true);
VisualServer::get_singleton()->material_set_flag(outer_mat_dup,VS::MATERIAL_FLAG_UNSHADED,true);
VisualServer::get_singleton()->fixed_material_set_flag( outer_mat_dup, VisualServer::FIXED_MATERIAL_FLAG_USE_ALPHA, true );
d[VS::ARRAY_VERTEX]=lines;
VisualServer::get_singleton()->mesh_add_surface(duplicate_mesh,VS::PRIMITIVE_LINES,d);
VisualServer::get_singleton()->mesh_surface_set_material(duplicate_mesh,1,outer_mat_dup);
}
selection.active=false;
updating=false;
}
GridMapEditor::~GridMapEditor() {
for(int i=0;i<3;i++) {
if (grid[i].is_valid())
VisualServer::get_singleton()->free(grid[i]);
if (grid_instance[i].is_valid())
VisualServer::get_singleton()->free(grid_instance[i]);
if (cursor_instance)
VisualServer::get_singleton()->free(cursor_instance);
}
VisualServer::get_singleton()->free(inner_mat);
VisualServer::get_singleton()->free(outer_mat);
VisualServer::get_singleton()->free(inner_mat_dup);
VisualServer::get_singleton()->free(outer_mat_dup);
VisualServer::get_singleton()->free(selection_mesh);
if (selection_instance.is_valid())
VisualServer::get_singleton()->free(selection_instance);
VisualServer::get_singleton()->free(duplicate_mesh);
if (duplicate_instance.is_valid())
VisualServer::get_singleton()->free(duplicate_instance);
_clear_areas();
}
void GridMapEditorPlugin::edit(Object *p_object) {
gridmap_editor->edit(p_object?p_object->cast_to<GridMap>():NULL);
}
bool GridMapEditorPlugin::handles(Object *p_object) const {
return p_object->is_type("GridMap");
}
void GridMapEditorPlugin::make_visible(bool p_visible) {
if (p_visible) {
gridmap_editor->show();
gridmap_editor->spatial_editor_hb->show();
gridmap_editor->set_process(true);
} else {
gridmap_editor->spatial_editor_hb->hide();
gridmap_editor->hide();
gridmap_editor->edit(NULL);
gridmap_editor->set_process(false);
}
}
GridMapEditorPlugin::GridMapEditorPlugin(EditorNode *p_node) {
editor=p_node;
gridmap_editor = memnew( GridMapEditor(editor) );
SpatialEditor::get_singleton()->get_palette_split()->add_child(gridmap_editor);
SpatialEditor::get_singleton()->get_palette_split()->move_child(gridmap_editor,0);
gridmap_editor->hide();
}
GridMapEditorPlugin::~GridMapEditorPlugin()
{
}
Reference in a new issue View git blame Copy permalink