initial commit, 4.5 stable

editor/scene/2d/polygon_2d_editor_plugin.h

@@ -0,0 +1,197 @@
+/**************************************************************************/
+/*  polygon_2d_editor_plugin.h                                            */
+/**************************************************************************/
+/*                         This file is part of:                          */
+/*                             GODOT ENGINE                               */
+/*                        https://godotengine.org                         */
+/**************************************************************************/
+/* Copyright (c) 2014-present Godot Engine contributors (see AUTHORS.md). */
+/* Copyright (c) 2007-2014 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.                 */
+/**************************************************************************/
+
+#pragma once
+
+#include "editor/scene/2d/abstract_polygon_2d_editor.h"
+#include "scene/2d/polygon_2d.h"
+
+class AcceptDialog;
+class ButtonGroup;
+class EditorZoomWidget;
+class HScrollBar;
+class HSlider;
+class Label;
+class MenuButton;
+class Panel;
+class ScrollContainer;
+class SpinBox;
+class TextureRect;
+class ViewPanner;
+class VScrollBar;
+
+class Polygon2DEditor : public AbstractPolygon2DEditor {
+	GDCLASS(Polygon2DEditor, AbstractPolygon2DEditor);
+
+	enum {
+		MENU_POLYGON_TO_UV,
+		MENU_UV_TO_POLYGON,
+		MENU_UV_CLEAR,
+		MENU_GRID_SETTINGS,
+	};
+
+	enum Mode {
+		MODE_POINTS,
+		MODE_POLYGONS,
+		MODE_UV,
+		MODE_BONES,
+		MODE_MAX
+	};
+
+	enum Action {
+		ACTION_CREATE,
+		ACTION_CREATE_INTERNAL,
+		ACTION_REMOVE_INTERNAL,
+		ACTION_EDIT_POINT,
+		ACTION_MOVE,
+		ACTION_ROTATE,
+		ACTION_SCALE,
+		ACTION_ADD_POLYGON,
+		ACTION_REMOVE_POLYGON,
+		ACTION_PAINT_WEIGHT,
+		ACTION_CLEAR_WEIGHT,
+		ACTION_MAX
+	};
+
+	Polygon2D *node = nullptr;
+	Polygon2D *previous_node = nullptr;
+
+	Button *dock_button = nullptr;
+	VBoxContainer *polygon_edit = nullptr;
+	Mode current_mode = MODE_MAX; // Uninitialized.
+	Button *mode_buttons[MODE_MAX];
+	Action selected_action = ACTION_CREATE;
+	Button *action_buttons[ACTION_MAX];
+	Button *b_snap_enable = nullptr;
+	Button *b_snap_grid = nullptr;
+	MenuButton *edit_menu = nullptr;
+
+	Control *canvas = nullptr;
+	Panel *canvas_background = nullptr;
+	Polygon2D *preview_polygon = nullptr;
+	EditorZoomWidget *zoom_widget = nullptr;
+	HScrollBar *hscroll = nullptr;
+	VScrollBar *vscroll = nullptr;
+	bool center_view_on_draw = false;
+
+	Ref<ViewPanner> panner;
+	void _pan_callback(Vector2 p_scroll_vec, Ref<InputEvent> p_event);
+	void _zoom_callback(float p_zoom_factor, Vector2 p_origin, Ref<InputEvent> p_event);
+	Vector2 draw_offset;
+	real_t draw_zoom = 1.0;
+
+	VBoxContainer *bone_scroll_main_vb = nullptr;
+	ScrollContainer *bone_scroll = nullptr;
+	VBoxContainer *bone_scroll_vb = nullptr;
+	Button *sync_bones = nullptr;
+	HSlider *bone_paint_strength = nullptr;
+	SpinBox *bone_paint_radius = nullptr;
+	Label *bone_paint_radius_label = nullptr;
+	bool bone_painting = false;
+	int bone_painting_bone = 0;
+	Vector<float> prev_weights;
+	Vector2 bone_paint_pos;
+	AcceptDialog *grid_settings = nullptr;
+
+	void _sync_bones();
+	void _update_bone_list();
+
+	Vector<Vector2> editing_points;
+	Vector<Vector2> previous_uv;
+	Vector<Vector2> previous_polygon;
+	Vector<Color> previous_colors;
+	int previous_internal_vertices = 0;
+	Array previous_bones;
+	Array previous_polygons;
+
+	Vector2 create_to;
+	int point_drag_index = -1;
+	bool is_dragging = false;
+	bool is_creating = false;
+	Vector<int> polygon_create;
+	Action current_action = ACTION_CREATE;
+	Vector2 drag_from;
+
+	AcceptDialog *error = nullptr;
+
+	bool use_snap = false;
+	bool snap_show_grid = false;
+	Vector2 snap_offset;
+	Vector2 snap_step;
+
+	void _edit_menu_option(int p_option);
+
+	void _cancel_editing();
+	void _update_polygon_editing_state();
+	void _update_available_modes();
+
+	void _center_view();
+	void _update_zoom_and_pan(bool p_zoom_at_center);
+	void _canvas_input(const Ref<InputEvent> &p_input);
+	void _center_view_on_draw(bool p_enabled = true);
+	void _canvas_draw();
+	void _set_action(int p_mode);
+
+	void _set_use_snap(bool p_use);
+	void _set_show_grid(bool p_show);
+	void _set_snap_off_x(real_t p_val);
+	void _set_snap_off_y(real_t p_val);
+	void _set_snap_step_x(real_t p_val);
+	void _set_snap_step_y(real_t p_val);
+
+	void _select_mode(int p_mode);
+	void _bone_paint_selected(int p_index);
+
+	int _get_polygon_count() const override;
+
+protected:
+	virtual Node2D *_get_node() const override;
+	virtual void _set_node(Node *p_polygon) override;
+
+	virtual Vector2 _get_offset(int p_idx) const override;
+
+	virtual bool _has_uv() const override { return true; }
+	virtual void _commit_action() override;
+
+	void _notification(int p_what);
+	static void _bind_methods();
+
+	Vector2 snap_point(Vector2 p_target) const;
+
+public:
+	Polygon2DEditor();
+};
+
+class Polygon2DEditorPlugin : public AbstractPolygon2DEditorPlugin {
+	GDCLASS(Polygon2DEditorPlugin, AbstractPolygon2DEditorPlugin);
+
+public:
+	Polygon2DEditorPlugin();
+};