abstract_polygon_2d_editor.cpp 24 KB

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