godot/tools/editor/plugins/path_editor_plugin.cpp
2016-06-15 15:26:52 -05:00

596 lines
16 KiB
C++

/*************************************************************************/
/* path_editor_plugin.cpp */
/*************************************************************************/
/* This file is part of: */
/* GODOT ENGINE */
/* http://www.godotengine.org */
/*************************************************************************/
/* Copyright (c) 2007-2016 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 "path_editor_plugin.h"
#include "spatial_editor_plugin.h"
#include "scene/resources/curve.h"
#include "os/keyboard.h"
String PathSpatialGizmo::get_handle_name(int p_idx) const {
Ref<Curve3D> c = path->get_curve();
if (c.is_null())
return "";
if (p_idx<c->get_point_count()) {
return TTR("Curve Point #")+itos(p_idx);
}
p_idx=p_idx-c->get_point_count()+1;
int idx=p_idx/2;
int t=p_idx%2;
String n = TTR("Curve Point #")+itos(idx);
if (t==0)
n+=" In";
else
n+=" Out";
return n;
}
Variant PathSpatialGizmo::get_handle_value(int p_idx) const{
Ref<Curve3D> c = path->get_curve();
if (c.is_null())
return Variant();
if (p_idx<c->get_point_count()) {
original=c->get_point_pos(p_idx);
return original;
}
p_idx=p_idx-c->get_point_count()+1;
int idx=p_idx/2;
int t=p_idx%2;
Vector3 ofs;
if (t==0)
ofs=c->get_point_in(idx);
else
ofs= c->get_point_out(idx);
original=ofs+c->get_point_pos(idx);
return ofs;
}
void PathSpatialGizmo::set_handle(int p_idx,Camera *p_camera, const Point2& p_point){
Ref<Curve3D> c = path->get_curve();
if (c.is_null())
return;
Transform gt = path->get_global_transform();
Transform gi = gt.affine_inverse();
Vector3 ray_from = p_camera->project_ray_origin(p_point);
Vector3 ray_dir = p_camera->project_ray_normal(p_point);
if (p_idx<c->get_point_count()) {
Plane p(gt.xform(original),p_camera->get_transform().basis.get_axis(2));
Vector3 inters;
if (p.intersects_ray(ray_from,ray_dir,&inters)) {
Vector3 local = gi.xform(inters);
c->set_point_pos(p_idx,local);
}
return;
}
p_idx=p_idx-c->get_point_count()+1;
int idx=p_idx/2;
int t=p_idx%2;
Vector3 base = c->get_point_pos(idx);
Plane p(gt.xform(original),p_camera->get_transform().basis.get_axis(2));
Vector3 inters;
if (p.intersects_ray(ray_from,ray_dir,&inters)) {
Vector3 local = gi.xform(inters)-base;
if (t==0) {
c->set_point_in(idx,local);
} else {
c->set_point_out(idx,local);
}
}
}
void PathSpatialGizmo::commit_handle(int p_idx,const Variant& p_restore,bool p_cancel){
Ref<Curve3D> c = path->get_curve();
if (c.is_null())
return;
UndoRedo *ur = SpatialEditor::get_singleton()->get_undo_redo();
if (p_idx<c->get_point_count()) {
if (p_cancel) {
c->set_point_pos(p_idx,p_restore);
return;
}
ur->create_action(TTR("Set Curve Point Pos"));
ur->add_do_method(c.ptr(),"set_point_pos",p_idx,c->get_point_pos(p_idx));
ur->add_undo_method(c.ptr(),"set_point_pos",p_idx,p_restore);
ur->commit_action();
return;
}
p_idx=p_idx-c->get_point_count()+1;
int idx=p_idx/2;
int t=p_idx%2;
Vector3 ofs;
if (p_cancel) {
return;
}
if (t==0) {
if (p_cancel) {
c->set_point_in(p_idx,p_restore);
return;
}
ur->create_action(TTR("Set Curve In Pos"));
ur->add_do_method(c.ptr(),"set_point_in",idx,c->get_point_in(idx));
ur->add_undo_method(c.ptr(),"set_point_in",idx,p_restore);
ur->commit_action();
} else {
if (p_cancel) {
c->set_point_out(idx,p_restore);
return;
}
ur->create_action(TTR("Set Curve Out Pos"));
ur->add_do_method(c.ptr(),"set_point_out",idx,c->get_point_out(idx));
ur->add_undo_method(c.ptr(),"set_point_out",idx,p_restore);
ur->commit_action();
}
}
void PathSpatialGizmo::redraw(){
clear();
Ref<Curve3D> c = path->get_curve();
if (c.is_null())
return;
Vector3Array v3a=c->tesselate();
//Vector3Array v3a=c->get_baked_points();
int v3s = v3a.size();
if (v3s==0)
return;
Vector<Vector3> v3p;
Vector3Array::Read r = v3a.read();
for(int i=0;i<v3s-1;i++) {
v3p.push_back(r[i]);
v3p.push_back(r[i+1]);
//v3p.push_back(r[i]);
//v3p.push_back(r[i]+Vector3(0,0.2,0));
}
add_lines(v3p,PathEditorPlugin::singleton->path_material);
add_collision_segments(v3p);
if (PathEditorPlugin::singleton->get_edited_path()==path) {
v3p.clear();
Vector<Vector3> handles;
Vector<Vector3> sec_handles;
for(int i=0;i<c->get_point_count();i++) {
Vector3 p = c->get_point_pos(i);
handles.push_back(p);
if (i>0) {
v3p.push_back(p);
v3p.push_back(p+c->get_point_in(i));
sec_handles.push_back(p+c->get_point_in(i));
}
if (i<c->get_point_count()-1) {
v3p.push_back(p);
v3p.push_back(p+c->get_point_out(i));
sec_handles.push_back(p+c->get_point_out(i));
}
}
add_lines(v3p,PathEditorPlugin::singleton->path_thin_material);
add_handles(handles);
add_handles(sec_handles,false,true);
}
}
PathSpatialGizmo::PathSpatialGizmo(Path* p_path){
path=p_path;
set_spatial_node(p_path);
}
Ref<SpatialEditorGizmo> PathEditorPlugin::create_spatial_gizmo(Spatial* p_spatial) {
if (p_spatial->cast_to<Path>()) {
return memnew( PathSpatialGizmo(p_spatial->cast_to<Path>()));
}
return Ref<SpatialEditorGizmo>();
}
bool PathEditorPlugin::forward_spatial_input_event(Camera* p_camera,const InputEvent& p_event) {
if (!path)
return false;
Ref<Curve3D> c=path->get_curve();
if (c.is_null())
return false;
Transform gt = path->get_global_transform();
Transform it = gt.affine_inverse();
static const int click_dist = 10; //should make global
if (p_event.type==InputEvent::MOUSE_BUTTON) {
const InputEventMouseButton &mb=p_event.mouse_button;
Point2 mbpos(mb.x,mb.y);
if (mb.pressed && mb.button_index==BUTTON_LEFT && (curve_create->is_pressed() || (curve_edit->is_pressed() && mb.mod.control))) {
//click into curve, break it down
Vector3Array v3a = c->tesselate();
int idx=0;
int rc=v3a.size();
int closest_seg=-1;
Vector3 closest_seg_point;
float closest_d=1e20;
if (rc>=2) {
Vector3Array::Read r = v3a.read();
if (p_camera->unproject_position(gt.xform(c->get_point_pos(0))).distance_to(mbpos)<click_dist)
return false; //nope, existing
for(int i=0;i<c->get_point_count()-1;i++) {
//find the offset and point index of the place to break up
int j=idx;
if (p_camera->unproject_position(gt.xform(c->get_point_pos(i+1))).distance_to(mbpos)<click_dist)
return false; //nope, existing
while(j<rc && c->get_point_pos(i+1)!=r[j]) {
Vector3 from =r[j];
Vector3 to =r[j+1];
real_t cdist = from.distance_to(to);
from=gt.xform(from);
to=gt.xform(to);
if (cdist>0) {
Vector2 s[2];
s[0] = p_camera->unproject_position(from);
s[1] = p_camera->unproject_position(to);
Vector2 inters = Geometry::get_closest_point_to_segment_2d(mbpos,s);
float d = inters.distance_to(mbpos);
if (d<10 && d<closest_d) {
closest_d=d;
closest_seg=i;
Vector3 ray_from=p_camera->project_ray_origin(mbpos);
Vector3 ray_dir=p_camera->project_ray_normal(mbpos);
Vector3 ra,rb;
Geometry::get_closest_points_between_segments(ray_from,ray_from+ray_dir*4096,from,to,ra,rb);
closest_seg_point=it.xform(rb);
}
}
j++;
}
if (idx==j)
idx++; //force next
else
idx=j; //swap
if (j==rc)
break;
}
}
UndoRedo *ur = editor->get_undo_redo();
if (closest_seg!=-1) {
//subdivide
ur->create_action(TTR("Split Path"));
ur->add_do_method(c.ptr(),"add_point",closest_seg_point,Vector3(),Vector3(),closest_seg+1);
ur->add_undo_method(c.ptr(),"remove_point",closest_seg+1);
ur->commit_action();;
return true;
} else {
Vector3 org;
if (c->get_point_count()==0)
org=path->get_transform().get_origin();
else
org=gt.xform(c->get_point_pos(c->get_point_count()));
Plane p(org,p_camera->get_transform().basis.get_axis(2));
Vector3 ray_from=p_camera->project_ray_origin(mbpos);
Vector3 ray_dir=p_camera->project_ray_normal(mbpos);
Vector3 inters;
if (p.intersects_ray(ray_from,ray_dir,&inters)) {
ur->create_action(TTR("Add Point to Curve"));
ur->add_do_method(c.ptr(),"add_point",it.xform(inters),Vector3(),Vector3(),-1);
ur->add_undo_method(c.ptr(),"remove_point",c->get_point_count());
ur->commit_action();;
return true;
}
//add new at pos
}
} else if (mb.pressed && ((mb.button_index==BUTTON_LEFT && curve_del->is_pressed()) || (mb.button_index==BUTTON_RIGHT && curve_edit->is_pressed()))) {
int erase_idx=-1;
for(int i=0;i<c->get_point_count();i++) {
//find the offset and point index of the place to break up
if (p_camera->unproject_position(gt.xform(c->get_point_pos(i))).distance_to(mbpos)<click_dist) {
erase_idx=i;
break;
}
}
if (erase_idx!=-1) {
UndoRedo *ur = editor->get_undo_redo();
ur->create_action(TTR("Remove Path Point"));
ur->add_do_method(c.ptr(),"remove_point",erase_idx);
ur->add_undo_method(c.ptr(),"add_point",c->get_point_pos(erase_idx),c->get_point_in(erase_idx),c->get_point_out(erase_idx),erase_idx);
ur->commit_action();
return true;
}
}
}
return false;
}
void PathEditorPlugin::edit(Object *p_object) {
if (p_object) {
path=p_object->cast_to<Path>();
if (path) {
if (path->get_curve().is_valid()) {
path->get_curve()->emit_signal("changed");
}
}
} else {
Path *pre=path;
path=NULL;
if (pre) {
pre->get_curve()->emit_signal("changed");
}
}
// collision_polygon_editor->edit(p_object->cast_to<Node>());
}
bool PathEditorPlugin::handles(Object *p_object) const {
return p_object->is_type("Path");
}
void PathEditorPlugin::make_visible(bool p_visible) {
if (p_visible) {
curve_create->show();
curve_edit->show();
curve_del->show();
curve_close->show();
sep->show();
} else {
curve_create->hide();
curve_edit->hide();
curve_del->hide();
curve_close->hide();
sep->hide();
{
Path *pre=path;
path=NULL;
if (pre && pre->get_curve().is_valid()) {
pre->get_curve()->emit_signal("changed");
}
}
}
}
void PathEditorPlugin::_mode_changed(int p_idx) {
curve_create->set_pressed(p_idx==0);
curve_edit->set_pressed(p_idx==1);
curve_del->set_pressed(p_idx==2);
}
void PathEditorPlugin::_close_curve() {
Ref<Curve3D> c = path->get_curve();
if (c.is_null())
return ;
if (c->get_point_count()<2)
return;
c->add_point(c->get_point_pos(0),c->get_point_in(0),c->get_point_out(0));
}
void PathEditorPlugin::_notification(int p_what) {
if (p_what==NOTIFICATION_ENTER_TREE) {
curve_create->connect("pressed",this,"_mode_changed",make_binds(0));
curve_edit->connect("pressed",this,"_mode_changed",make_binds(1));
curve_del->connect("pressed",this,"_mode_changed",make_binds(2));
curve_close->connect("pressed",this,"_close_curve");
}
}
void PathEditorPlugin::_bind_methods() {
ObjectTypeDB::bind_method(_MD("_mode_changed"),&PathEditorPlugin::_mode_changed);
ObjectTypeDB::bind_method(_MD("_close_curve"),&PathEditorPlugin::_close_curve);
}
PathEditorPlugin* PathEditorPlugin::singleton=NULL;
PathEditorPlugin::PathEditorPlugin(EditorNode *p_node) {
path=NULL;
editor=p_node;
singleton=this;
path_material = Ref<FixedMaterial>( memnew( FixedMaterial ));
path_material->set_parameter( FixedMaterial::PARAM_DIFFUSE,Color(0.5,0.5,1.0,0.8) );
path_material->set_fixed_flag(FixedMaterial::FLAG_USE_ALPHA, true);
path_material->set_line_width(3);
path_material->set_flag(Material::FLAG_DOUBLE_SIDED,true);
path_material->set_flag(Material::FLAG_UNSHADED,true);
path_thin_material = Ref<FixedMaterial>( memnew( FixedMaterial ));
path_thin_material->set_parameter( FixedMaterial::PARAM_DIFFUSE,Color(0.5,0.5,1.0,0.4) );
path_thin_material->set_fixed_flag(FixedMaterial::FLAG_USE_ALPHA, true);
path_thin_material->set_line_width(1);
path_thin_material->set_flag(Material::FLAG_DOUBLE_SIDED,true);
path_thin_material->set_flag(Material::FLAG_UNSHADED,true);
// SpatialEditor::get_singleton()->add_gizmo_plugin(this);
sep = memnew( VSeparator);
sep->hide();
SpatialEditor::get_singleton()->add_control_to_menu_panel(sep);
curve_edit = memnew( ToolButton );
curve_edit->set_icon(EditorNode::get_singleton()->get_gui_base()->get_icon("CurveEdit","EditorIcons"));
curve_edit->set_toggle_mode(true);
curve_edit->hide();
curve_edit->set_focus_mode(Control::FOCUS_NONE);
curve_edit->set_tooltip(TTR("Select Points")+"\n"+TTR("Shift+Drag: Select Control Points")+"\n"+keycode_get_string(KEY_MASK_CMD)+TTR("Click: Add Point")+"\n"+TTR("Right Click: Delete Point"));
SpatialEditor::get_singleton()->add_control_to_menu_panel(curve_edit);
curve_create = memnew( ToolButton );
curve_create->set_icon(EditorNode::get_singleton()->get_gui_base()->get_icon("CurveCreate","EditorIcons"));
curve_create->set_toggle_mode(true);
curve_create->hide();
curve_create->set_focus_mode(Control::FOCUS_NONE);
curve_create->set_tooltip(TTR("Add Point (in empty space)")+"\n"+TTR("Split Segment (in curve)"));
SpatialEditor::get_singleton()->add_control_to_menu_panel(curve_create);
curve_del = memnew( ToolButton );
curve_del->set_icon(EditorNode::get_singleton()->get_gui_base()->get_icon("CurveDelete","EditorIcons"));
curve_del->set_toggle_mode(true);
curve_del->hide();
curve_del->set_focus_mode(Control::FOCUS_NONE);
curve_del->set_tooltip(TTR("Delete Point"));
SpatialEditor::get_singleton()->add_control_to_menu_panel(curve_del);
curve_close = memnew( ToolButton );
curve_close->set_icon(EditorNode::get_singleton()->get_gui_base()->get_icon("CurveClose","EditorIcons"));
curve_close->hide();
curve_close->set_focus_mode(Control::FOCUS_NONE);
curve_close->set_tooltip(TTR("Close Curve"));
SpatialEditor::get_singleton()->add_control_to_menu_panel(curve_close);
curve_edit->set_pressed(true);
/*
collision_polygon_editor = memnew( PathEditor(p_node) );
editor->get_viewport()->add_child(collision_polygon_editor);
collision_polygon_editor->set_margin(MARGIN_LEFT,200);
collision_polygon_editor->set_margin(MARGIN_RIGHT,230);
collision_polygon_editor->set_margin(MARGIN_TOP,0);
collision_polygon_editor->set_margin(MARGIN_BOTTOM,10);
collision_polygon_editor->hide();
*/
}
PathEditorPlugin::~PathEditorPlugin()
{
}