abstract_polygon_2d_editor.cpp 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776
  1. /**************************************************************************/
  2. /* abstract_polygon_2d_editor.cpp */
  3. /**************************************************************************/
  4. /* This file is part of: */
  5. /* GODOT ENGINE */
  6. /* https://godotengine.org */
  7. /**************************************************************************/
  8. /* Copyright (c) 2014-present Godot Engine contributors (see AUTHORS.md). */
  9. /* Copyright (c) 2007-2014 Juan Linietsky, Ariel Manzur. */
  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. #include "abstract_polygon_2d_editor.h"
  31. #include "canvas_item_editor_plugin.h"
  32. #include "core/math/geometry_2d.h"
  33. #include "core/os/keyboard.h"
  34. #include "editor/editor_node.h"
  35. #include "editor/editor_scale.h"
  36. #include "editor/editor_settings.h"
  37. #include "editor/editor_string_names.h"
  38. #include "editor/editor_undo_redo_manager.h"
  39. #include "scene/gui/button.h"
  40. #include "scene/gui/dialogs.h"
  41. #include "scene/gui/separator.h"
  42. bool AbstractPolygon2DEditor::Vertex::operator==(const AbstractPolygon2DEditor::Vertex &p_vertex) const {
  43. return polygon == p_vertex.polygon && vertex == p_vertex.vertex;
  44. }
  45. bool AbstractPolygon2DEditor::Vertex::operator!=(const AbstractPolygon2DEditor::Vertex &p_vertex) const {
  46. return !(*this == p_vertex);
  47. }
  48. bool AbstractPolygon2DEditor::Vertex::valid() const {
  49. return vertex >= 0;
  50. }
  51. bool AbstractPolygon2DEditor::_is_empty() const {
  52. if (!_get_node()) {
  53. return true;
  54. }
  55. const int n = _get_polygon_count();
  56. for (int i = 0; i < n; i++) {
  57. Vector<Vector2> vertices = _get_polygon(i);
  58. if (vertices.size() != 0) {
  59. return false;
  60. }
  61. }
  62. return true;
  63. }
  64. bool AbstractPolygon2DEditor::_is_line() const {
  65. return false;
  66. }
  67. bool AbstractPolygon2DEditor::_has_uv() const {
  68. return false;
  69. }
  70. int AbstractPolygon2DEditor::_get_polygon_count() const {
  71. return 1;
  72. }
  73. Variant AbstractPolygon2DEditor::_get_polygon(int p_idx) const {
  74. return _get_node()->get("polygon");
  75. }
  76. void AbstractPolygon2DEditor::_set_polygon(int p_idx, const Variant &p_polygon) const {
  77. _get_node()->set("polygon", p_polygon);
  78. }
  79. void AbstractPolygon2DEditor::_action_set_polygon(int p_idx, const Variant &p_previous, const Variant &p_polygon) {
  80. Node2D *node = _get_node();
  81. EditorUndoRedoManager *undo_redo = EditorUndoRedoManager::get_singleton();
  82. undo_redo->add_do_method(node, "set_polygon", p_polygon);
  83. undo_redo->add_undo_method(node, "set_polygon", p_previous);
  84. }
  85. Vector2 AbstractPolygon2DEditor::_get_offset(int p_idx) const {
  86. return Vector2(0, 0);
  87. }
  88. void AbstractPolygon2DEditor::_commit_action() {
  89. EditorUndoRedoManager *undo_redo = EditorUndoRedoManager::get_singleton();
  90. undo_redo->add_do_method(canvas_item_editor, "update_viewport");
  91. undo_redo->add_undo_method(canvas_item_editor, "update_viewport");
  92. undo_redo->commit_action();
  93. }
  94. void AbstractPolygon2DEditor::_action_add_polygon(const Variant &p_polygon) {
  95. _action_set_polygon(0, p_polygon);
  96. }
  97. void AbstractPolygon2DEditor::_action_remove_polygon(int p_idx) {
  98. _action_set_polygon(p_idx, _get_polygon(p_idx), Vector<Vector2>());
  99. }
  100. void AbstractPolygon2DEditor::_action_set_polygon(int p_idx, const Variant &p_polygon) {
  101. _action_set_polygon(p_idx, _get_polygon(p_idx), p_polygon);
  102. }
  103. bool AbstractPolygon2DEditor::_has_resource() const {
  104. return true;
  105. }
  106. void AbstractPolygon2DEditor::_create_resource() {
  107. }
  108. void AbstractPolygon2DEditor::_menu_option(int p_option) {
  109. switch (p_option) {
  110. case MODE_CREATE: {
  111. mode = MODE_CREATE;
  112. button_create->set_pressed(true);
  113. button_edit->set_pressed(false);
  114. button_delete->set_pressed(false);
  115. } break;
  116. case MODE_EDIT: {
  117. _wip_close();
  118. mode = MODE_EDIT;
  119. button_create->set_pressed(false);
  120. button_edit->set_pressed(true);
  121. button_delete->set_pressed(false);
  122. } break;
  123. case MODE_DELETE: {
  124. _wip_close();
  125. mode = MODE_DELETE;
  126. button_create->set_pressed(false);
  127. button_edit->set_pressed(false);
  128. button_delete->set_pressed(true);
  129. } break;
  130. }
  131. }
  132. void AbstractPolygon2DEditor::_notification(int p_what) {
  133. switch (p_what) {
  134. case NOTIFICATION_ENTER_TREE:
  135. case NOTIFICATION_THEME_CHANGED: {
  136. button_create->set_icon(get_editor_theme_icon(SNAME("CurveCreate")));
  137. button_edit->set_icon(get_editor_theme_icon(SNAME("CurveEdit")));
  138. button_delete->set_icon(get_editor_theme_icon(SNAME("CurveDelete")));
  139. } break;
  140. case NOTIFICATION_READY: {
  141. disable_polygon_editing(false, String());
  142. button_edit->set_pressed(true);
  143. get_tree()->connect("node_removed", callable_mp(this, &AbstractPolygon2DEditor::_node_removed));
  144. create_resource->connect("confirmed", callable_mp(this, &AbstractPolygon2DEditor::_create_resource));
  145. } break;
  146. }
  147. }
  148. void AbstractPolygon2DEditor::_node_removed(Node *p_node) {
  149. if (p_node == _get_node()) {
  150. edit(nullptr);
  151. hide();
  152. canvas_item_editor->update_viewport();
  153. }
  154. }
  155. void AbstractPolygon2DEditor::_wip_changed() {
  156. if (wip_active && _is_line()) {
  157. _set_polygon(0, wip);
  158. }
  159. }
  160. void AbstractPolygon2DEditor::_wip_cancel() {
  161. wip.clear();
  162. wip_active = false;
  163. edited_point = PosVertex();
  164. hover_point = Vertex();
  165. selected_point = Vertex();
  166. canvas_item_editor->update_viewport();
  167. }
  168. void AbstractPolygon2DEditor::_wip_close() {
  169. if (!wip_active) {
  170. return;
  171. }
  172. if (_is_line()) {
  173. _set_polygon(0, wip);
  174. } else if (wip.size() >= (_is_line() ? 2 : 3)) {
  175. EditorUndoRedoManager *undo_redo = EditorUndoRedoManager::get_singleton();
  176. undo_redo->create_action(TTR("Create Polygon"));
  177. _action_add_polygon(wip);
  178. if (_has_uv()) {
  179. undo_redo->add_do_method(_get_node(), "set_uv", Vector<Vector2>());
  180. undo_redo->add_undo_method(_get_node(), "set_uv", _get_node()->get("uv"));
  181. }
  182. _commit_action();
  183. } else {
  184. return;
  185. }
  186. mode = MODE_EDIT;
  187. button_edit->set_pressed(true);
  188. button_create->set_pressed(false);
  189. button_delete->set_pressed(false);
  190. wip.clear();
  191. wip_active = false;
  192. edited_point = PosVertex();
  193. hover_point = Vertex();
  194. selected_point = Vertex();
  195. }
  196. void AbstractPolygon2DEditor::disable_polygon_editing(bool p_disable, String p_reason) {
  197. _polygon_editing_enabled = !p_disable;
  198. button_create->set_disabled(p_disable);
  199. button_edit->set_disabled(p_disable);
  200. button_delete->set_disabled(p_disable);
  201. if (p_disable) {
  202. button_create->set_tooltip_text(p_reason);
  203. button_edit->set_tooltip_text(p_reason);
  204. button_delete->set_tooltip_text(p_reason);
  205. } else {
  206. button_create->set_tooltip_text(TTR("Create points."));
  207. button_edit->set_tooltip_text(TTR("Edit points.\nLMB: Move Point\nRMB: Erase Point"));
  208. button_delete->set_tooltip_text(TTR("Erase points."));
  209. }
  210. }
  211. bool AbstractPolygon2DEditor::forward_gui_input(const Ref<InputEvent> &p_event) {
  212. if (!_get_node() || !_polygon_editing_enabled) {
  213. return false;
  214. }
  215. if (!_get_node()->is_visible_in_tree()) {
  216. return false;
  217. }
  218. EditorUndoRedoManager *undo_redo = EditorUndoRedoManager::get_singleton();
  219. Ref<InputEventMouseButton> mb = p_event;
  220. if (!_has_resource()) {
  221. if (mb.is_valid() && mb->get_button_index() == MouseButton::LEFT && mb->is_pressed()) {
  222. create_resource->set_text(String("No polygon resource on this node.\nCreate and assign one?"));
  223. create_resource->popup_centered();
  224. }
  225. return (mb.is_valid() && mb->get_button_index() == MouseButton::LEFT);
  226. }
  227. CanvasItemEditor::Tool tool = CanvasItemEditor::get_singleton()->get_current_tool();
  228. if (tool != CanvasItemEditor::TOOL_SELECT) {
  229. return false;
  230. }
  231. if (mb.is_valid()) {
  232. Transform2D xform = canvas_item_editor->get_canvas_transform() * _get_node()->get_global_transform();
  233. Vector2 gpoint = mb->get_position();
  234. Vector2 cpoint = _get_node()->to_local(canvas_item_editor->snap_point(canvas_item_editor->get_canvas_transform().affine_inverse().xform(mb->get_position())));
  235. if (mode == MODE_EDIT || (_is_line() && mode == MODE_CREATE)) {
  236. if (mb->get_button_index() == MouseButton::LEFT) {
  237. if (mb->is_pressed()) {
  238. if (mb->is_meta_pressed() || mb->is_ctrl_pressed() || mb->is_shift_pressed() || mb->is_alt_pressed()) {
  239. return false;
  240. }
  241. const PosVertex insert = closest_edge_point(gpoint);
  242. if (insert.valid()) {
  243. Vector<Vector2> vertices = _get_polygon(insert.polygon);
  244. if (vertices.size() < (_is_line() ? 2 : 3)) {
  245. vertices.push_back(cpoint);
  246. undo_redo->create_action(TTR("Edit Polygon"));
  247. selected_point = Vertex(insert.polygon, vertices.size());
  248. _action_set_polygon(insert.polygon, vertices);
  249. _commit_action();
  250. return true;
  251. } else {
  252. edited_point = PosVertex(insert.polygon, insert.vertex + 1, xform.affine_inverse().xform(insert.pos));
  253. vertices.insert(edited_point.vertex, edited_point.pos);
  254. pre_move_edit = vertices;
  255. selected_point = Vertex(edited_point.polygon, edited_point.vertex);
  256. edge_point = PosVertex();
  257. undo_redo->create_action(TTR("Insert Point"));
  258. _action_set_polygon(insert.polygon, vertices);
  259. _commit_action();
  260. return true;
  261. }
  262. } else {
  263. //look for points to move
  264. const PosVertex closest = closest_point(gpoint);
  265. if (closest.valid()) {
  266. pre_move_edit = _get_polygon(closest.polygon);
  267. edited_point = PosVertex(closest, xform.affine_inverse().xform(closest.pos));
  268. selected_point = closest;
  269. edge_point = PosVertex();
  270. canvas_item_editor->update_viewport();
  271. return true;
  272. } else {
  273. selected_point = Vertex();
  274. }
  275. }
  276. } else {
  277. if (edited_point.valid()) {
  278. //apply
  279. Vector<Vector2> vertices = _get_polygon(edited_point.polygon);
  280. ERR_FAIL_INDEX_V(edited_point.vertex, vertices.size(), false);
  281. vertices.write[edited_point.vertex] = edited_point.pos - _get_offset(edited_point.polygon);
  282. undo_redo->create_action(TTR("Edit Polygon"));
  283. _action_set_polygon(edited_point.polygon, pre_move_edit, vertices);
  284. _commit_action();
  285. edited_point = PosVertex();
  286. return true;
  287. }
  288. }
  289. } else if (mb->get_button_index() == MouseButton::RIGHT && mb->is_pressed() && !edited_point.valid()) {
  290. const PosVertex closest = closest_point(gpoint);
  291. if (closest.valid()) {
  292. remove_point(closest);
  293. return true;
  294. }
  295. }
  296. } else if (mode == MODE_DELETE) {
  297. if (mb->get_button_index() == MouseButton::LEFT && mb->is_pressed()) {
  298. const PosVertex closest = closest_point(gpoint);
  299. if (closest.valid()) {
  300. remove_point(closest);
  301. return true;
  302. }
  303. }
  304. }
  305. if (mode == MODE_CREATE) {
  306. if (mb->get_button_index() == MouseButton::LEFT && mb->is_pressed()) {
  307. if (_is_line()) {
  308. // for lines, we don't have a wip mode, and we can undo each single add point.
  309. Vector<Vector2> vertices = _get_polygon(0);
  310. vertices.push_back(cpoint);
  311. undo_redo->create_action(TTR("Insert Point"));
  312. _action_set_polygon(0, vertices);
  313. _commit_action();
  314. return true;
  315. } else if (!wip_active) {
  316. wip.clear();
  317. wip.push_back(cpoint);
  318. wip_active = true;
  319. _wip_changed();
  320. edited_point = PosVertex(-1, 1, cpoint);
  321. canvas_item_editor->update_viewport();
  322. hover_point = Vertex();
  323. selected_point = Vertex(0);
  324. edge_point = PosVertex();
  325. return true;
  326. } else {
  327. const real_t grab_threshold = EDITOR_GET("editors/polygon_editor/point_grab_radius");
  328. if (!_is_line() && wip.size() > 1 && xform.xform(wip[0]).distance_to(xform.xform(cpoint)) < grab_threshold) {
  329. //wip closed
  330. _wip_close();
  331. return true;
  332. } else {
  333. //add wip point
  334. wip.push_back(cpoint);
  335. _wip_changed();
  336. edited_point = PosVertex(-1, wip.size(), cpoint);
  337. selected_point = Vertex(wip.size() - 1);
  338. canvas_item_editor->update_viewport();
  339. return true;
  340. }
  341. }
  342. } else if (mb->get_button_index() == MouseButton::RIGHT && mb->is_pressed() && wip_active) {
  343. _wip_cancel();
  344. }
  345. }
  346. }
  347. Ref<InputEventMouseMotion> mm = p_event;
  348. if (mm.is_valid()) {
  349. Vector2 gpoint = mm->get_position();
  350. if (edited_point.valid() && (wip_active || mm->get_button_mask().has_flag(MouseButtonMask::LEFT))) {
  351. Vector2 cpoint = _get_node()->to_local(canvas_item_editor->snap_point(canvas_item_editor->get_canvas_transform().affine_inverse().xform(gpoint)));
  352. //Move the point in a single axis. Should only work when editing a polygon and while holding shift.
  353. if (mode == MODE_EDIT && mm->is_shift_pressed()) {
  354. Vector2 old_point = pre_move_edit.get(selected_point.vertex);
  355. if (ABS(cpoint.x - old_point.x) > ABS(cpoint.y - old_point.y)) {
  356. cpoint.y = old_point.y;
  357. } else {
  358. cpoint.x = old_point.x;
  359. }
  360. }
  361. edited_point = PosVertex(edited_point, cpoint);
  362. if (!wip_active) {
  363. Vector<Vector2> vertices = _get_polygon(edited_point.polygon);
  364. ERR_FAIL_INDEX_V(edited_point.vertex, vertices.size(), false);
  365. vertices.write[edited_point.vertex] = cpoint - _get_offset(edited_point.polygon);
  366. _set_polygon(edited_point.polygon, vertices);
  367. }
  368. canvas_item_editor->update_viewport();
  369. } else if (mode == MODE_EDIT || (_is_line() && mode == MODE_CREATE)) {
  370. const PosVertex onEdgeVertex = closest_edge_point(gpoint);
  371. if (onEdgeVertex.valid()) {
  372. hover_point = Vertex();
  373. edge_point = onEdgeVertex;
  374. canvas_item_editor->update_viewport();
  375. } else {
  376. if (edge_point.valid()) {
  377. edge_point = PosVertex();
  378. canvas_item_editor->update_viewport();
  379. }
  380. const PosVertex new_hover_point = closest_point(gpoint);
  381. if (hover_point != new_hover_point) {
  382. hover_point = new_hover_point;
  383. canvas_item_editor->update_viewport();
  384. }
  385. }
  386. }
  387. }
  388. Ref<InputEventKey> k = p_event;
  389. if (k.is_valid() && k->is_pressed()) {
  390. if (k->get_keycode() == Key::KEY_DELETE || k->get_keycode() == Key::BACKSPACE) {
  391. if (wip_active && selected_point.polygon == -1) {
  392. if (wip.size() > selected_point.vertex) {
  393. wip.remove_at(selected_point.vertex);
  394. _wip_changed();
  395. selected_point = wip.size() - 1;
  396. canvas_item_editor->update_viewport();
  397. return true;
  398. }
  399. } else {
  400. const Vertex active_point = get_active_point();
  401. if (active_point.valid()) {
  402. remove_point(active_point);
  403. return true;
  404. }
  405. }
  406. } else if (wip_active && k->get_keycode() == Key::ENTER) {
  407. _wip_close();
  408. } else if (wip_active && k->get_keycode() == Key::ESCAPE) {
  409. _wip_cancel();
  410. }
  411. }
  412. return false;
  413. }
  414. void AbstractPolygon2DEditor::forward_canvas_draw_over_viewport(Control *p_overlay) {
  415. if (!_get_node()) {
  416. return;
  417. }
  418. if (!_get_node()->is_visible_in_tree()) {
  419. return;
  420. }
  421. Transform2D xform = canvas_item_editor->get_canvas_transform() * _get_node()->get_global_transform();
  422. // All polygon points are sharp, so use the sharp handle icon
  423. const Ref<Texture2D> handle = get_editor_theme_icon(SNAME("EditorPathSharpHandle"));
  424. const Vertex active_point = get_active_point();
  425. const int n_polygons = _get_polygon_count();
  426. const bool is_closed = !_is_line();
  427. for (int j = -1; j < n_polygons; j++) {
  428. if (wip_active && wip_destructive && j != -1) {
  429. continue;
  430. }
  431. Vector<Vector2> points;
  432. Vector2 offset;
  433. if (wip_active && j == edited_point.polygon) {
  434. points = Variant(wip);
  435. offset = Vector2(0, 0);
  436. } else {
  437. if (j == -1) {
  438. continue;
  439. }
  440. points = _get_polygon(j);
  441. offset = _get_offset(j);
  442. }
  443. if (!wip_active && j == edited_point.polygon && EDITOR_GET("editors/polygon_editor/show_previous_outline")) {
  444. const Color col = Color(0.5, 0.5, 0.5); // FIXME polygon->get_outline_color();
  445. const int n = pre_move_edit.size();
  446. for (int i = 0; i < n - (is_closed ? 0 : 1); i++) {
  447. Vector2 p, p2;
  448. p = pre_move_edit[i] + offset;
  449. p2 = pre_move_edit[(i + 1) % n] + offset;
  450. Vector2 point = xform.xform(p);
  451. Vector2 next_point = xform.xform(p2);
  452. p_overlay->draw_line(point, next_point, col, Math::round(2 * EDSCALE));
  453. }
  454. }
  455. const int n_points = points.size();
  456. const Color col = Color(1, 0.3, 0.1, 0.8);
  457. for (int i = 0; i < n_points; i++) {
  458. const Vertex vertex(j, i);
  459. const Vector2 p = (vertex == edited_point) ? edited_point.pos : (points[i] + offset);
  460. const Vector2 point = xform.xform(p);
  461. if (is_closed || i < n_points - 1) {
  462. Vector2 p2;
  463. if (j == edited_point.polygon &&
  464. ((wip_active && i == n_points - 1) || (((i + 1) % n_points) == edited_point.vertex))) {
  465. p2 = edited_point.pos;
  466. } else {
  467. p2 = points[(i + 1) % n_points] + offset;
  468. }
  469. const Vector2 next_point = xform.xform(p2);
  470. p_overlay->draw_line(point, next_point, col, Math::round(2 * EDSCALE));
  471. }
  472. }
  473. for (int i = 0; i < n_points; i++) {
  474. const Vertex vertex(j, i);
  475. const Vector2 p = (vertex == edited_point) ? edited_point.pos : (points[i] + offset);
  476. const Vector2 point = xform.xform(p);
  477. const Color overlay_modulate = vertex == active_point ? Color(0.5, 1, 2) : Color(1, 1, 1);
  478. p_overlay->draw_texture(handle, point - handle->get_size() * 0.5, overlay_modulate);
  479. if (vertex == hover_point) {
  480. Ref<Font> font = get_theme_font(SNAME("bold"), EditorStringName(EditorFonts));
  481. int font_size = 1.3 * get_theme_font_size(SNAME("bold_size"), EditorStringName(EditorFonts));
  482. String num = String::num(vertex.vertex);
  483. Size2 num_size = font->get_string_size(num, HORIZONTAL_ALIGNMENT_LEFT, -1, font_size);
  484. const float outline_size = 4;
  485. Color font_color = get_theme_color(SNAME("font_color"), EditorStringName(Editor));
  486. Color outline_color = font_color.inverted();
  487. p_overlay->draw_string_outline(font, point - num_size * 0.5, num, HORIZONTAL_ALIGNMENT_LEFT, -1, font_size, outline_size, outline_color);
  488. p_overlay->draw_string(font, point - num_size * 0.5, num, HORIZONTAL_ALIGNMENT_LEFT, -1, font_size, font_color);
  489. }
  490. }
  491. }
  492. if (edge_point.valid()) {
  493. Ref<Texture2D> add_handle = get_editor_theme_icon(SNAME("EditorHandleAdd"));
  494. p_overlay->draw_texture(add_handle, edge_point.pos - add_handle->get_size() * 0.5);
  495. }
  496. }
  497. void AbstractPolygon2DEditor::edit(Node *p_polygon) {
  498. if (!canvas_item_editor) {
  499. canvas_item_editor = CanvasItemEditor::get_singleton();
  500. }
  501. if (p_polygon) {
  502. _set_node(p_polygon);
  503. // Enable the pencil tool if the polygon is empty.
  504. if (_is_empty()) {
  505. _menu_option(MODE_CREATE);
  506. } else {
  507. _menu_option(MODE_EDIT);
  508. }
  509. wip.clear();
  510. wip_active = false;
  511. edited_point = PosVertex();
  512. hover_point = Vertex();
  513. selected_point = Vertex();
  514. } else {
  515. _set_node(nullptr);
  516. }
  517. canvas_item_editor->update_viewport();
  518. }
  519. void AbstractPolygon2DEditor::_bind_methods() {
  520. }
  521. void AbstractPolygon2DEditor::remove_point(const Vertex &p_vertex) {
  522. EditorUndoRedoManager *undo_redo = EditorUndoRedoManager::get_singleton();
  523. Vector<Vector2> vertices = _get_polygon(p_vertex.polygon);
  524. if (vertices.size() > (_is_line() ? 2 : 3)) {
  525. vertices.remove_at(p_vertex.vertex);
  526. undo_redo->create_action(TTR("Edit Polygon (Remove Point)"));
  527. _action_set_polygon(p_vertex.polygon, vertices);
  528. _commit_action();
  529. } else {
  530. undo_redo->create_action(TTR("Remove Polygon And Point"));
  531. _action_remove_polygon(p_vertex.polygon);
  532. _commit_action();
  533. }
  534. if (_is_empty()) {
  535. _menu_option(MODE_CREATE);
  536. }
  537. hover_point = Vertex();
  538. if (selected_point == p_vertex) {
  539. selected_point = Vertex();
  540. }
  541. }
  542. AbstractPolygon2DEditor::Vertex AbstractPolygon2DEditor::get_active_point() const {
  543. return hover_point.valid() ? hover_point : selected_point;
  544. }
  545. AbstractPolygon2DEditor::PosVertex AbstractPolygon2DEditor::closest_point(const Vector2 &p_pos) const {
  546. const real_t grab_threshold = EDITOR_GET("editors/polygon_editor/point_grab_radius");
  547. const int n_polygons = _get_polygon_count();
  548. const Transform2D xform = canvas_item_editor->get_canvas_transform() * _get_node()->get_global_transform();
  549. PosVertex closest;
  550. real_t closest_dist = 1e10;
  551. for (int j = 0; j < n_polygons; j++) {
  552. Vector<Vector2> points = _get_polygon(j);
  553. const Vector2 offset = _get_offset(j);
  554. const int n_points = points.size();
  555. for (int i = 0; i < n_points; i++) {
  556. Vector2 cp = xform.xform(points[i] + offset);
  557. real_t d = cp.distance_to(p_pos);
  558. if (d < closest_dist && d < grab_threshold) {
  559. closest_dist = d;
  560. closest = PosVertex(j, i, cp);
  561. }
  562. }
  563. }
  564. return closest;
  565. }
  566. AbstractPolygon2DEditor::PosVertex AbstractPolygon2DEditor::closest_edge_point(const Vector2 &p_pos) const {
  567. const real_t grab_threshold = EDITOR_GET("editors/polygon_editor/point_grab_radius");
  568. const real_t eps = grab_threshold * 2;
  569. const real_t eps2 = eps * eps;
  570. const int n_polygons = _get_polygon_count();
  571. const Transform2D xform = canvas_item_editor->get_canvas_transform() * _get_node()->get_global_transform();
  572. PosVertex closest;
  573. real_t closest_dist = 1e10;
  574. for (int j = 0; j < n_polygons; j++) {
  575. Vector<Vector2> points = _get_polygon(j);
  576. const Vector2 offset = _get_offset(j);
  577. const int n_points = points.size();
  578. const int n_segments = n_points - (_is_line() ? 1 : 0);
  579. for (int i = 0; i < n_segments; i++) {
  580. Vector2 segment[2] = { xform.xform(points[i] + offset),
  581. xform.xform(points[(i + 1) % n_points] + offset) };
  582. Vector2 cp = Geometry2D::get_closest_point_to_segment(p_pos, segment);
  583. if (cp.distance_squared_to(segment[0]) < eps2 || cp.distance_squared_to(segment[1]) < eps2) {
  584. continue; //not valid to reuse point
  585. }
  586. real_t d = cp.distance_to(p_pos);
  587. if (d < closest_dist && d < grab_threshold) {
  588. closest_dist = d;
  589. closest = PosVertex(j, i, cp);
  590. }
  591. }
  592. }
  593. return closest;
  594. }
  595. AbstractPolygon2DEditor::AbstractPolygon2DEditor(bool p_wip_destructive) {
  596. edited_point = PosVertex();
  597. wip_destructive = p_wip_destructive;
  598. hover_point = Vertex();
  599. selected_point = Vertex();
  600. edge_point = PosVertex();
  601. button_create = memnew(Button);
  602. button_create->set_theme_type_variation("FlatButton");
  603. add_child(button_create);
  604. button_create->connect("pressed", callable_mp(this, &AbstractPolygon2DEditor::_menu_option).bind(MODE_CREATE));
  605. button_create->set_toggle_mode(true);
  606. button_edit = memnew(Button);
  607. button_edit->set_theme_type_variation("FlatButton");
  608. add_child(button_edit);
  609. button_edit->connect("pressed", callable_mp(this, &AbstractPolygon2DEditor::_menu_option).bind(MODE_EDIT));
  610. button_edit->set_toggle_mode(true);
  611. button_delete = memnew(Button);
  612. button_delete->set_theme_type_variation("FlatButton");
  613. add_child(button_delete);
  614. button_delete->connect("pressed", callable_mp(this, &AbstractPolygon2DEditor::_menu_option).bind(MODE_DELETE));
  615. button_delete->set_toggle_mode(true);
  616. create_resource = memnew(ConfirmationDialog);
  617. add_child(create_resource);
  618. create_resource->set_ok_button_text(TTR("Create"));
  619. }
  620. void AbstractPolygon2DEditorPlugin::edit(Object *p_object) {
  621. polygon_editor->edit(Object::cast_to<Node>(p_object));
  622. }
  623. bool AbstractPolygon2DEditorPlugin::handles(Object *p_object) const {
  624. return p_object->is_class(klass);
  625. }
  626. void AbstractPolygon2DEditorPlugin::make_visible(bool p_visible) {
  627. if (p_visible) {
  628. polygon_editor->show();
  629. } else {
  630. polygon_editor->hide();
  631. polygon_editor->edit(nullptr);
  632. }
  633. }
  634. AbstractPolygon2DEditorPlugin::AbstractPolygon2DEditorPlugin(AbstractPolygon2DEditor *p_polygon_editor, String p_class) :
  635. polygon_editor(p_polygon_editor),
  636. klass(p_class) {
  637. CanvasItemEditor::get_singleton()->add_control_to_menu_panel(polygon_editor);
  638. polygon_editor->hide();
  639. }
  640. AbstractPolygon2DEditorPlugin::~AbstractPolygon2DEditorPlugin() {
  641. }