1 /*************************************************************************/
2 /* collision_shape_2d_editor_plugin.cpp */
3 /*************************************************************************/
4 /* This file is part of: */
5 /* GODOT ENGINE */
6 /* https://godotengine.org */
7 /*************************************************************************/
8 /* Copyright (c) 2007-2020 Juan Linietsky, Ariel Manzur. */
9 /* Copyright (c) 2014-2020 Godot Engine contributors (cf. AUTHORS.md). */
10 /* */
11 /* Permission is hereby granted, free of charge, to any person obtaining */
12 /* a copy of this software and associated documentation files (the */
13 /* "Software"), to deal in the Software without restriction, including */
14 /* without limitation the rights to use, copy, modify, merge, publish, */
15 /* distribute, sublicense, and/or sell copies of the Software, and to */
16 /* permit persons to whom the Software is furnished to do so, subject to */
17 /* the following conditions: */
18 /* */
19 /* The above copyright notice and this permission notice shall be */
20 /* included in all copies or substantial portions of the Software. */
21 /* */
22 /* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, */
23 /* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF */
24 /* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.*/
25 /* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY */
26 /* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, */
27 /* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE */
28 /* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. */
29 /*************************************************************************/
30
31 #include "collision_shape_2d_editor_plugin.h"
32
33 #include "canvas_item_editor_plugin.h"
34 #include "scene/resources/capsule_shape_2d.h"
35 #include "scene/resources/circle_shape_2d.h"
36 #include "scene/resources/concave_polygon_shape_2d.h"
37 #include "scene/resources/convex_polygon_shape_2d.h"
38 #include "scene/resources/line_shape_2d.h"
39 #include "scene/resources/rectangle_shape_2d.h"
40 #include "scene/resources/segment_shape_2d.h"
41
_node_removed(Node * p_node)42 void CollisionShape2DEditor::_node_removed(Node *p_node) {
43
44 if (p_node == node) {
45 node = NULL;
46 }
47 }
48
get_handle_value(int idx) const49 Variant CollisionShape2DEditor::get_handle_value(int idx) const {
50
51 switch (shape_type) {
52 case CAPSULE_SHAPE: {
53 Ref<CapsuleShape2D> capsule = node->get_shape();
54
55 if (idx == 0) {
56 return capsule->get_radius();
57 } else if (idx == 1) {
58 return capsule->get_height();
59 }
60
61 } break;
62
63 case CIRCLE_SHAPE: {
64 Ref<CircleShape2D> circle = node->get_shape();
65
66 if (idx == 0) {
67 return circle->get_radius();
68 }
69
70 } break;
71
72 case CONCAVE_POLYGON_SHAPE: {
73
74 } break;
75
76 case CONVEX_POLYGON_SHAPE: {
77
78 } break;
79
80 case LINE_SHAPE: {
81 Ref<LineShape2D> line = node->get_shape();
82
83 if (idx == 0) {
84 return line->get_d();
85 } else {
86 return line->get_normal();
87 }
88
89 } break;
90
91 case RAY_SHAPE: {
92 Ref<RayShape2D> ray = node->get_shape();
93
94 if (idx == 0) {
95 return ray->get_length();
96 }
97
98 } break;
99
100 case RECTANGLE_SHAPE: {
101 Ref<RectangleShape2D> rect = node->get_shape();
102
103 if (idx < 3) {
104 return rect->get_extents().abs();
105 }
106
107 } break;
108
109 case SEGMENT_SHAPE: {
110 Ref<SegmentShape2D> seg = node->get_shape();
111
112 if (idx == 0) {
113 return seg->get_a();
114 } else if (idx == 1) {
115 return seg->get_b();
116 }
117
118 } break;
119 }
120
121 return Variant();
122 }
123
set_handle(int idx,Point2 & p_point)124 void CollisionShape2DEditor::set_handle(int idx, Point2 &p_point) {
125
126 switch (shape_type) {
127 case CAPSULE_SHAPE: {
128 if (idx < 2) {
129 Ref<CapsuleShape2D> capsule = node->get_shape();
130
131 real_t parameter = Math::abs(p_point[idx]);
132
133 if (idx == 0) {
134 capsule->set_radius(parameter);
135 } else if (idx == 1) {
136 capsule->set_height(parameter * 2 - capsule->get_radius() * 2);
137 }
138
139 canvas_item_editor->update_viewport();
140 }
141
142 } break;
143
144 case CIRCLE_SHAPE: {
145 Ref<CircleShape2D> circle = node->get_shape();
146 circle->set_radius(p_point.length());
147
148 canvas_item_editor->update_viewport();
149
150 } break;
151
152 case CONCAVE_POLYGON_SHAPE: {
153
154 } break;
155
156 case CONVEX_POLYGON_SHAPE: {
157
158 } break;
159
160 case LINE_SHAPE: {
161 if (idx < 2) {
162 Ref<LineShape2D> line = node->get_shape();
163
164 if (idx == 0) {
165 line->set_d(p_point.length());
166 } else {
167 line->set_normal(p_point.normalized());
168 }
169
170 canvas_item_editor->update_viewport();
171 }
172
173 } break;
174
175 case RAY_SHAPE: {
176 Ref<RayShape2D> ray = node->get_shape();
177
178 ray->set_length(Math::abs(p_point.y));
179
180 canvas_item_editor->update_viewport();
181
182 } break;
183
184 case RECTANGLE_SHAPE: {
185 if (idx < 3) {
186 Ref<RectangleShape2D> rect = node->get_shape();
187
188 Vector2 extents = rect->get_extents();
189 if (idx == 2) {
190 extents = p_point;
191 } else {
192 extents[idx] = p_point[idx];
193 }
194 rect->set_extents(extents.abs());
195
196 canvas_item_editor->update_viewport();
197 }
198
199 } break;
200
201 case SEGMENT_SHAPE: {
202 if (edit_handle < 2) {
203 Ref<SegmentShape2D> seg = node->get_shape();
204
205 if (idx == 0) {
206 seg->set_a(p_point);
207 } else if (idx == 1) {
208 seg->set_b(p_point);
209 }
210
211 canvas_item_editor->update_viewport();
212 }
213
214 } break;
215 }
216 node->get_shape()->_change_notify();
217 }
218
commit_handle(int idx,Variant & p_org)219 void CollisionShape2DEditor::commit_handle(int idx, Variant &p_org) {
220
221 undo_redo->create_action(TTR("Set Handle"));
222
223 switch (shape_type) {
224 case CAPSULE_SHAPE: {
225 Ref<CapsuleShape2D> capsule = node->get_shape();
226
227 if (idx == 0) {
228 undo_redo->add_do_method(capsule.ptr(), "set_radius", capsule->get_radius());
229 undo_redo->add_do_method(canvas_item_editor, "update_viewport");
230 undo_redo->add_undo_method(capsule.ptr(), "set_radius", p_org);
231 undo_redo->add_do_method(canvas_item_editor, "update_viewport");
232 } else if (idx == 1) {
233 undo_redo->add_do_method(capsule.ptr(), "set_height", capsule->get_height());
234 undo_redo->add_do_method(canvas_item_editor, "update_viewport");
235 undo_redo->add_undo_method(capsule.ptr(), "set_height", p_org);
236 undo_redo->add_undo_method(canvas_item_editor, "update_viewport");
237 }
238
239 } break;
240
241 case CIRCLE_SHAPE: {
242 Ref<CircleShape2D> circle = node->get_shape();
243
244 undo_redo->add_do_method(circle.ptr(), "set_radius", circle->get_radius());
245 undo_redo->add_do_method(canvas_item_editor, "update_viewport");
246 undo_redo->add_undo_method(circle.ptr(), "set_radius", p_org);
247 undo_redo->add_undo_method(canvas_item_editor, "update_viewport");
248
249 } break;
250
251 case CONCAVE_POLYGON_SHAPE: {
252 // Cannot be edited directly, use CollisionPolygon2D instead.
253 } break;
254
255 case CONVEX_POLYGON_SHAPE: {
256 // Cannot be edited directly, use CollisionPolygon2D instead.
257 } break;
258
259 case LINE_SHAPE: {
260 Ref<LineShape2D> line = node->get_shape();
261
262 if (idx == 0) {
263 undo_redo->add_do_method(line.ptr(), "set_d", line->get_d());
264 undo_redo->add_do_method(canvas_item_editor, "update_viewport");
265 undo_redo->add_undo_method(line.ptr(), "set_d", p_org);
266 undo_redo->add_undo_method(canvas_item_editor, "update_viewport");
267 } else {
268 undo_redo->add_do_method(line.ptr(), "set_normal", line->get_normal());
269 undo_redo->add_do_method(canvas_item_editor, "update_viewport");
270 undo_redo->add_undo_method(line.ptr(), "set_normal", p_org);
271 undo_redo->add_undo_method(canvas_item_editor, "update_viewport");
272 }
273
274 } break;
275
276 case RAY_SHAPE: {
277 Ref<RayShape2D> ray = node->get_shape();
278
279 undo_redo->add_do_method(ray.ptr(), "set_length", ray->get_length());
280 undo_redo->add_do_method(canvas_item_editor, "update_viewport");
281 undo_redo->add_undo_method(ray.ptr(), "set_length", p_org);
282 undo_redo->add_undo_method(canvas_item_editor, "update_viewport");
283
284 } break;
285
286 case RECTANGLE_SHAPE: {
287 Ref<RectangleShape2D> rect = node->get_shape();
288
289 undo_redo->add_do_method(rect.ptr(), "set_extents", rect->get_extents());
290 undo_redo->add_do_method(canvas_item_editor, "update_viewport");
291 undo_redo->add_undo_method(rect.ptr(), "set_extents", p_org);
292 undo_redo->add_undo_method(canvas_item_editor, "update_viewport");
293
294 } break;
295
296 case SEGMENT_SHAPE: {
297 Ref<SegmentShape2D> seg = node->get_shape();
298 if (idx == 0) {
299 undo_redo->add_do_method(seg.ptr(), "set_a", seg->get_a());
300 undo_redo->add_do_method(canvas_item_editor, "update_viewport");
301 undo_redo->add_undo_method(seg.ptr(), "set_a", p_org);
302 undo_redo->add_undo_method(canvas_item_editor, "update_viewport");
303 } else if (idx == 1) {
304 undo_redo->add_do_method(seg.ptr(), "set_b", seg->get_b());
305 undo_redo->add_do_method(canvas_item_editor, "update_viewport");
306 undo_redo->add_undo_method(seg.ptr(), "set_b", p_org);
307 undo_redo->add_undo_method(canvas_item_editor, "update_viewport");
308 }
309
310 } break;
311 }
312
313 undo_redo->commit_action();
314 }
315
forward_canvas_gui_input(const Ref<InputEvent> & p_event)316 bool CollisionShape2DEditor::forward_canvas_gui_input(const Ref<InputEvent> &p_event) {
317
318 if (!node) {
319 return false;
320 }
321
322 if (!node->get_shape().is_valid()) {
323 return false;
324 }
325
326 if (shape_type == -1) {
327 return false;
328 }
329
330 Ref<InputEventMouseButton> mb = p_event;
331 Transform2D xform = canvas_item_editor->get_canvas_transform() * node->get_global_transform();
332
333 if (mb.is_valid()) {
334
335 Vector2 gpoint = mb->get_position();
336
337 if (mb->get_button_index() == BUTTON_LEFT) {
338 if (mb->is_pressed()) {
339 for (int i = 0; i < handles.size(); i++) {
340 if (xform.xform(handles[i]).distance_to(gpoint) < 8) {
341 edit_handle = i;
342
343 break;
344 }
345 }
346
347 if (edit_handle == -1) {
348 pressed = false;
349
350 return false;
351 }
352
353 original = get_handle_value(edit_handle);
354 pressed = true;
355
356 return true;
357
358 } else {
359 if (pressed) {
360 commit_handle(edit_handle, original);
361
362 edit_handle = -1;
363 pressed = false;
364
365 return true;
366 }
367 }
368 }
369
370 return false;
371 }
372
373 Ref<InputEventMouseMotion> mm = p_event;
374
375 if (mm.is_valid()) {
376
377 if (edit_handle == -1 || !pressed) {
378 return false;
379 }
380
381 Vector2 cpoint = canvas_item_editor->snap_point(canvas_item_editor->get_canvas_transform().affine_inverse().xform(mm->get_position()));
382 cpoint = node->get_global_transform().affine_inverse().xform(cpoint);
383
384 set_handle(edit_handle, cpoint);
385
386 return true;
387 }
388
389 return false;
390 }
391
_get_current_shape_type()392 void CollisionShape2DEditor::_get_current_shape_type() {
393
394 if (!node) {
395 return;
396 }
397
398 Ref<Shape2D> s = node->get_shape();
399
400 if (!s.is_valid()) {
401 return;
402 }
403
404 if (Object::cast_to<CapsuleShape2D>(*s)) {
405 shape_type = CAPSULE_SHAPE;
406 } else if (Object::cast_to<CircleShape2D>(*s)) {
407 shape_type = CIRCLE_SHAPE;
408 } else if (Object::cast_to<ConcavePolygonShape2D>(*s)) {
409 shape_type = CONCAVE_POLYGON_SHAPE;
410 } else if (Object::cast_to<ConvexPolygonShape2D>(*s)) {
411 shape_type = CONVEX_POLYGON_SHAPE;
412 } else if (Object::cast_to<LineShape2D>(*s)) {
413 shape_type = LINE_SHAPE;
414 } else if (Object::cast_to<RayShape2D>(*s)) {
415 shape_type = RAY_SHAPE;
416 } else if (Object::cast_to<RectangleShape2D>(*s)) {
417 shape_type = RECTANGLE_SHAPE;
418 } else if (Object::cast_to<SegmentShape2D>(*s)) {
419 shape_type = SEGMENT_SHAPE;
420 } else {
421 shape_type = -1;
422 }
423
424 canvas_item_editor->update_viewport();
425 }
426
forward_canvas_draw_over_viewport(Control * p_overlay)427 void CollisionShape2DEditor::forward_canvas_draw_over_viewport(Control *p_overlay) {
428
429 if (!node) {
430 return;
431 }
432
433 if (!node->get_shape().is_valid()) {
434 return;
435 }
436
437 _get_current_shape_type();
438
439 if (shape_type == -1) {
440 return;
441 }
442
443 Transform2D gt = canvas_item_editor->get_canvas_transform() * node->get_global_transform();
444
445 Ref<Texture> h = get_icon("EditorHandle", "EditorIcons");
446 Vector2 size = h->get_size() * 0.5;
447
448 handles.clear();
449
450 switch (shape_type) {
451 case CAPSULE_SHAPE: {
452 Ref<CapsuleShape2D> shape = node->get_shape();
453
454 handles.resize(2);
455 float radius = shape->get_radius();
456 float height = shape->get_height() / 2;
457
458 handles.write[0] = Point2(radius, -height);
459 handles.write[1] = Point2(0, -(height + radius));
460
461 p_overlay->draw_texture(h, gt.xform(handles[0]) - size);
462 p_overlay->draw_texture(h, gt.xform(handles[1]) - size);
463
464 } break;
465
466 case CIRCLE_SHAPE: {
467 Ref<CircleShape2D> shape = node->get_shape();
468
469 handles.resize(1);
470 handles.write[0] = Point2(shape->get_radius(), 0);
471
472 p_overlay->draw_texture(h, gt.xform(handles[0]) - size);
473
474 } break;
475
476 case CONCAVE_POLYGON_SHAPE: {
477
478 } break;
479
480 case CONVEX_POLYGON_SHAPE: {
481
482 } break;
483
484 case LINE_SHAPE: {
485 Ref<LineShape2D> shape = node->get_shape();
486
487 handles.resize(2);
488 handles.write[0] = shape->get_normal() * shape->get_d();
489 handles.write[1] = shape->get_normal() * (shape->get_d() + 30.0);
490
491 p_overlay->draw_texture(h, gt.xform(handles[0]) - size);
492 p_overlay->draw_texture(h, gt.xform(handles[1]) - size);
493
494 } break;
495
496 case RAY_SHAPE: {
497 Ref<RayShape2D> shape = node->get_shape();
498
499 handles.resize(1);
500 handles.write[0] = Point2(0, shape->get_length());
501
502 p_overlay->draw_texture(h, gt.xform(handles[0]) - size);
503
504 } break;
505
506 case RECTANGLE_SHAPE: {
507 Ref<RectangleShape2D> shape = node->get_shape();
508
509 handles.resize(3);
510 Vector2 ext = shape->get_extents();
511 handles.write[0] = Point2(ext.x, 0);
512 handles.write[1] = Point2(0, -ext.y);
513 handles.write[2] = Point2(ext.x, -ext.y);
514
515 p_overlay->draw_texture(h, gt.xform(handles[0]) - size);
516 p_overlay->draw_texture(h, gt.xform(handles[1]) - size);
517 p_overlay->draw_texture(h, gt.xform(handles[2]) - size);
518
519 } break;
520
521 case SEGMENT_SHAPE: {
522 Ref<SegmentShape2D> shape = node->get_shape();
523
524 handles.resize(2);
525 handles.write[0] = shape->get_a();
526 handles.write[1] = shape->get_b();
527
528 p_overlay->draw_texture(h, gt.xform(handles[0]) - size);
529 p_overlay->draw_texture(h, gt.xform(handles[1]) - size);
530
531 } break;
532 }
533 }
534
_notification(int p_what)535 void CollisionShape2DEditor::_notification(int p_what) {
536
537 switch (p_what) {
538
539 case NOTIFICATION_ENTER_TREE: {
540 get_tree()->connect("node_removed", this, "_node_removed");
541 } break;
542
543 case NOTIFICATION_EXIT_TREE: {
544 get_tree()->disconnect("node_removed", this, "_node_removed");
545 } break;
546 }
547 }
548
edit(Node * p_node)549 void CollisionShape2DEditor::edit(Node *p_node) {
550
551 if (!canvas_item_editor) {
552 canvas_item_editor = CanvasItemEditor::get_singleton();
553 }
554
555 if (p_node) {
556 node = Object::cast_to<CollisionShape2D>(p_node);
557
558 _get_current_shape_type();
559
560 } else {
561 edit_handle = -1;
562 shape_type = -1;
563
564 node = NULL;
565 }
566
567 canvas_item_editor->update_viewport();
568 }
569
_bind_methods()570 void CollisionShape2DEditor::_bind_methods() {
571
572 ClassDB::bind_method("_get_current_shape_type", &CollisionShape2DEditor::_get_current_shape_type);
573 ClassDB::bind_method(D_METHOD("_node_removed"), &CollisionShape2DEditor::_node_removed);
574 }
575
CollisionShape2DEditor(EditorNode * p_editor)576 CollisionShape2DEditor::CollisionShape2DEditor(EditorNode *p_editor) {
577
578 node = NULL;
579 canvas_item_editor = NULL;
580 editor = p_editor;
581
582 undo_redo = p_editor->get_undo_redo();
583
584 edit_handle = -1;
585 pressed = false;
586 }
587
edit(Object * p_obj)588 void CollisionShape2DEditorPlugin::edit(Object *p_obj) {
589
590 collision_shape_2d_editor->edit(Object::cast_to<Node>(p_obj));
591 }
592
handles(Object * p_obj) const593 bool CollisionShape2DEditorPlugin::handles(Object *p_obj) const {
594
595 return p_obj->is_class("CollisionShape2D");
596 }
597
make_visible(bool visible)598 void CollisionShape2DEditorPlugin::make_visible(bool visible) {
599
600 if (!visible) {
601 edit(NULL);
602 }
603 }
604
CollisionShape2DEditorPlugin(EditorNode * p_editor)605 CollisionShape2DEditorPlugin::CollisionShape2DEditorPlugin(EditorNode *p_editor) {
606
607 editor = p_editor;
608
609 collision_shape_2d_editor = memnew(CollisionShape2DEditor(p_editor));
610 p_editor->get_gui_base()->add_child(collision_shape_2d_editor);
611 }
612
~CollisionShape2DEditorPlugin()613 CollisionShape2DEditorPlugin::~CollisionShape2DEditorPlugin() {
614 }
615