574 lines
14 KiB
C++
574 lines
14 KiB
C++
/*************************************************************************/
|
|
/* spatial_editor_plugin.h */
|
|
/*************************************************************************/
|
|
/* 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. */
|
|
/*************************************************************************/
|
|
#ifndef SPATIAL_EDITOR_PLUGIN_H
|
|
#define SPATIAL_EDITOR_PLUGIN_H
|
|
|
|
#include "tools/editor/editor_plugin.h"
|
|
#include "tools/editor/editor_node.h"
|
|
#include "scene/3d/visual_instance.h"
|
|
#include "scene/3d/immediate_geometry.h"
|
|
#include "scene/3d/light.h"
|
|
#include "scene/gui/panel_container.h"
|
|
/**
|
|
@author Juan Linietsky <reduzio@gmail.com>
|
|
*/
|
|
|
|
class Camera;
|
|
class SpatialEditor;
|
|
class SpatialEditorGizmos;
|
|
|
|
class SpatialEditorGizmo : public SpatialGizmo {
|
|
|
|
OBJ_TYPE(SpatialEditorGizmo,SpatialGizmo);
|
|
|
|
bool selected;
|
|
public:
|
|
|
|
void set_selected(bool p_selected) { selected=p_selected; }
|
|
bool is_selected() const { return selected; }
|
|
|
|
virtual String get_handle_name(int p_idx) const;
|
|
virtual Variant get_handle_value(int p_idx) const;
|
|
virtual void set_handle(int p_idx,Camera *p_camera, const Point2& p_point);
|
|
virtual void commit_handle(int p_idx,const Variant& p_restore,bool p_cancel=false);
|
|
|
|
virtual bool intersect_frustum(const Camera *p_camera,const Vector<Plane> &p_frustum);
|
|
virtual bool intersect_ray(const Camera *p_camera,const Point2& p_point, Vector3& r_pos, Vector3& r_normal,int *r_gizmo_handle=NULL,bool p_sec_first=false);
|
|
SpatialEditorGizmo();
|
|
};
|
|
|
|
|
|
class SpatialEditorViewport : public Control {
|
|
|
|
OBJ_TYPE( SpatialEditorViewport, Control );
|
|
friend class SpatialEditor;
|
|
enum {
|
|
|
|
VIEW_TOP,
|
|
VIEW_BOTTOM,
|
|
VIEW_LEFT,
|
|
VIEW_RIGHT,
|
|
VIEW_FRONT,
|
|
VIEW_REAR,
|
|
VIEW_CENTER_TO_ORIGIN,
|
|
VIEW_CENTER_TO_SELECTION,
|
|
VIEW_ALIGN_SELECTION_WITH_VIEW,
|
|
VIEW_PERSPECTIVE,
|
|
VIEW_ENVIRONMENT,
|
|
VIEW_ORTHOGONAL,
|
|
VIEW_AUDIO_LISTENER,
|
|
VIEW_GIZMOS,
|
|
};
|
|
public:
|
|
enum {
|
|
GIZMO_BASE_LAYER=27,
|
|
GIZMO_EDIT_LAYER=26,
|
|
GIZMO_GRID_LAYER=25
|
|
};
|
|
private:
|
|
int index;
|
|
String name;
|
|
void _menu_option(int p_option);
|
|
Size2 prev_size;
|
|
|
|
EditorNode *editor;
|
|
EditorSelection *editor_selection;
|
|
UndoRedo *undo_redo;
|
|
|
|
Button *preview_camera;
|
|
|
|
MenuButton *view_menu;
|
|
|
|
Control *surface;
|
|
Viewport *viewport;
|
|
Camera *camera;
|
|
bool transforming;
|
|
bool orthogonal;
|
|
float gizmo_scale;
|
|
|
|
struct _RayResult {
|
|
|
|
Spatial* item;
|
|
float depth;
|
|
int handle;
|
|
_FORCE_INLINE_ bool operator<(const _RayResult& p_rr) const { return depth<p_rr.depth; }
|
|
};
|
|
|
|
void _update_name();
|
|
void _compute_edit(const Point2& p_point);
|
|
void _clear_selected();
|
|
void _select_clicked(bool p_append,bool p_single);
|
|
void _select(Spatial *p_node, bool p_append,bool p_single);
|
|
ObjectID _select_ray(const Point2& p_pos, bool p_append,bool &r_includes_current,int *r_gizmo_handle=NULL,bool p_alt_select=false);
|
|
void _find_items_at_pos(const Point2& p_pos,bool &r_includes_current,Vector<_RayResult> &results,bool p_alt_select=false);
|
|
Vector3 _get_ray_pos(const Vector2& p_pos) const;
|
|
Vector3 _get_ray(const Vector2& p_pos);
|
|
Point2 _point_to_screen(const Vector3& p_point);
|
|
Transform _get_camera_transform() const;
|
|
int get_selected_count() const;
|
|
|
|
Vector3 _get_camera_pos() const;
|
|
Vector3 _get_camera_normal() const;
|
|
Vector3 _get_screen_to_space(const Vector3& p_vector3);
|
|
|
|
void _select_region();
|
|
bool _gizmo_select(const Vector2& p_screenpos,bool p_hilite_only=false);
|
|
|
|
float get_znear() const;
|
|
float get_zfar() const;
|
|
float get_fov() const;
|
|
|
|
ObjectID clicked;
|
|
Vector<_RayResult> selection_results;
|
|
bool clicked_includes_current;
|
|
bool clicked_wants_append;
|
|
|
|
PopupMenu *selection_menu;
|
|
|
|
enum NavigationScheme {
|
|
NAVIGATION_GODOT,
|
|
NAVIGATION_MAYA,
|
|
NAVIGATION_MODO,
|
|
};
|
|
NavigationScheme _get_navigation_schema(const String& p_property);
|
|
|
|
enum NavigationZoomStyle {
|
|
NAVIGATION_ZOOM_VERTICAL,
|
|
NAVIGATION_ZOOM_HORIZONTAL
|
|
};
|
|
NavigationZoomStyle _get_navigation_zoom_style(const String& p_property);
|
|
|
|
enum NavigationMode {
|
|
NAVIGATION_NONE,
|
|
NAVIGATION_PAN,
|
|
NAVIGATION_ZOOM,
|
|
NAVIGATION_ORBIT
|
|
};
|
|
enum TransformMode {
|
|
TRANSFORM_NONE,
|
|
TRANSFORM_ROTATE,
|
|
TRANSFORM_TRANSLATE,
|
|
TRANSFORM_SCALE
|
|
|
|
};
|
|
enum TransformPlane {
|
|
TRANSFORM_VIEW,
|
|
TRANSFORM_X_AXIS,
|
|
TRANSFORM_Y_AXIS,
|
|
TRANSFORM_Z_AXIS,
|
|
};
|
|
|
|
struct EditData {
|
|
TransformMode mode;
|
|
TransformPlane plane;
|
|
Transform original;
|
|
Vector3 click_ray;
|
|
Vector3 click_ray_pos;
|
|
Vector3 center;
|
|
Vector3 orig_gizmo_pos;
|
|
int edited_gizmo;
|
|
Point2 mouse_pos;
|
|
bool snap;
|
|
Ref<SpatialEditorGizmo> gizmo;
|
|
int gizmo_handle;
|
|
Variant gizmo_initial_value;
|
|
Vector3 gizmo_initial_pos;
|
|
} _edit;
|
|
|
|
struct Cursor {
|
|
|
|
Vector3 cursor_pos;
|
|
|
|
Vector3 pos;
|
|
float x_rot,y_rot,distance;
|
|
bool region_select;
|
|
Point2 region_begin,region_end;
|
|
|
|
Cursor() { x_rot=y_rot=0; distance=4; region_select=false; }
|
|
} cursor;
|
|
|
|
RID move_gizmo_instance[3], rotate_gizmo_instance[3];
|
|
|
|
|
|
String last_message;
|
|
String message;
|
|
float message_time;
|
|
|
|
void set_message(String p_message,float p_time=5);
|
|
|
|
//
|
|
void _update_camera();
|
|
void _draw();
|
|
|
|
void _smouseenter();
|
|
void _sinput(const InputEvent& p_ie);
|
|
SpatialEditor *spatial_editor;
|
|
|
|
Camera* previewing;
|
|
Camera* preview;
|
|
|
|
void _preview_exited_scene();
|
|
void _toggle_camera_preview(bool);
|
|
void _init_gizmo_instance(int p_idx);
|
|
void _finish_gizmo_instances();
|
|
void _selection_result_pressed(int);
|
|
void _selection_menu_hide();
|
|
void _list_select(InputEventMouseButton b);
|
|
|
|
|
|
protected:
|
|
|
|
void _notification(int p_what);
|
|
static void _bind_methods();
|
|
public:
|
|
|
|
void update_transform_gizmo_view();
|
|
|
|
void set_can_preview(Camera* p_preview);
|
|
void set_state(const Dictionary& p_state);
|
|
Dictionary get_state() const;
|
|
void reset();
|
|
Viewport *get_viewport_node() { return viewport; }
|
|
|
|
|
|
SpatialEditorViewport(SpatialEditor *p_spatial_editor,EditorNode *p_editor,int p_index);
|
|
};
|
|
|
|
|
|
|
|
class SpatialEditorSelectedItem : public Object {
|
|
|
|
OBJ_TYPE(SpatialEditorSelectedItem,Object);
|
|
|
|
public:
|
|
|
|
AABB aabb;
|
|
Transform original; // original location when moving
|
|
Transform last_xform; // last transform
|
|
Spatial *sp;
|
|
RID sbox_instance;
|
|
|
|
SpatialEditorSelectedItem() { sp=NULL; }
|
|
~SpatialEditorSelectedItem();
|
|
};
|
|
|
|
class SpatialEditor : public VBoxContainer {
|
|
|
|
OBJ_TYPE(SpatialEditor, VBoxContainer );
|
|
public:
|
|
|
|
enum ToolMode {
|
|
|
|
TOOL_MODE_SELECT,
|
|
TOOL_MODE_MOVE,
|
|
TOOL_MODE_ROTATE,
|
|
TOOL_MODE_SCALE,
|
|
TOOL_MODE_LIST_SELECT,
|
|
TOOL_MAX
|
|
|
|
};
|
|
|
|
|
|
private:
|
|
EditorNode *editor;
|
|
EditorSelection *editor_selection;
|
|
|
|
|
|
|
|
Control *viewport_base;
|
|
SpatialEditorViewport *viewports[4];
|
|
VSplitContainer *shader_split;
|
|
HSplitContainer *palette_split;
|
|
|
|
/////
|
|
|
|
ToolMode tool_mode;
|
|
bool orthogonal;
|
|
|
|
|
|
VisualServer::ScenarioDebugMode scenario_debug;
|
|
|
|
RID light;
|
|
RID light_instance;
|
|
Transform light_transform;
|
|
|
|
|
|
RID origin;
|
|
RID origin_instance;
|
|
RID grid[3];
|
|
RID grid_instance[3];
|
|
bool grid_visible[3]; //currently visible
|
|
float last_grid_snap;
|
|
bool grid_enable[3]; //should be always visible if true
|
|
bool grid_enabled;
|
|
|
|
Ref<Mesh> move_gizmo[3], rotate_gizmo[3];
|
|
Ref<FixedMaterial> gizmo_color[3];
|
|
Ref<FixedMaterial> gizmo_hl;
|
|
|
|
|
|
int over_gizmo_handle;
|
|
|
|
|
|
|
|
Ref<Mesh> selection_box;
|
|
RID indicators;
|
|
RID indicators_instance;
|
|
RID cursor_mesh;
|
|
RID cursor_instance;
|
|
RID indicator_mat;
|
|
RID cursor_material;
|
|
|
|
/*
|
|
struct Selected {
|
|
AABB aabb;
|
|
Transform original; // original location when moving
|
|
Transform last_xform; // last transform
|
|
Spatial *sp;
|
|
RID poly_instance;
|
|
};
|
|
|
|
Map<uint32_t,Selected> selected;
|
|
*/
|
|
struct Gizmo {
|
|
|
|
bool visible;
|
|
float scale;
|
|
Transform transform;
|
|
} gizmo;
|
|
|
|
|
|
|
|
|
|
enum MenuOption {
|
|
|
|
MENU_TOOL_SELECT,
|
|
MENU_TOOL_MOVE,
|
|
MENU_TOOL_ROTATE,
|
|
MENU_TOOL_SCALE,
|
|
MENU_TOOL_LIST_SELECT,
|
|
MENU_TRANSFORM_USE_SNAP,
|
|
MENU_TRANSFORM_CONFIGURE_SNAP,
|
|
MENU_TRANSFORM_LOCAL_COORDS,
|
|
MENU_TRANSFORM_DIALOG,
|
|
MENU_VIEW_USE_1_VIEWPORT,
|
|
MENU_VIEW_USE_2_VIEWPORTS,
|
|
MENU_VIEW_USE_2_VIEWPORTS_ALT,
|
|
MENU_VIEW_USE_3_VIEWPORTS,
|
|
MENU_VIEW_USE_3_VIEWPORTS_ALT,
|
|
MENU_VIEW_USE_4_VIEWPORTS,
|
|
MENU_VIEW_USE_DEFAULT_LIGHT,
|
|
MENU_VIEW_USE_DEFAULT_SRGB,
|
|
MENU_VIEW_DISPLAY_NORMAL,
|
|
MENU_VIEW_DISPLAY_WIREFRAME,
|
|
MENU_VIEW_DISPLAY_OVERDRAW,
|
|
MENU_VIEW_DISPLAY_SHADELESS,
|
|
MENU_VIEW_ORIGIN,
|
|
MENU_VIEW_GRID,
|
|
MENU_VIEW_CAMERA_SETTINGS,
|
|
|
|
};
|
|
|
|
|
|
Button *tool_button[TOOL_MAX];
|
|
Button *instance_button;
|
|
|
|
|
|
MenuButton* transform_menu;
|
|
MenuButton* view_menu;
|
|
|
|
ConfirmationDialog *snap_dialog;
|
|
ConfirmationDialog *xform_dialog;
|
|
ConfirmationDialog *settings_dialog;
|
|
|
|
bool snap_enabled;
|
|
LineEdit *snap_translate;
|
|
LineEdit *snap_rotate;
|
|
LineEdit *snap_scale;
|
|
PanelContainer* menu_panel;
|
|
|
|
LineEdit *xform_translate[3];
|
|
LineEdit *xform_rotate[3];
|
|
LineEdit *xform_scale[3];
|
|
OptionButton *xform_type;
|
|
|
|
VBoxContainer *settings_vbc;
|
|
SpinBox *settings_fov;
|
|
SpinBox *settings_znear;
|
|
SpinBox *settings_zfar;
|
|
DirectionalLight *settings_dlight;
|
|
ImmediateGeometry *settings_sphere;
|
|
Camera *settings_camera;
|
|
float settings_default_light_rot_x;
|
|
float settings_default_light_rot_y;
|
|
|
|
Control *settings_light_base;
|
|
Viewport *settings_light_vp;
|
|
ColorPickerButton *settings_ambient_color;
|
|
Image settings_light_dir_image;
|
|
|
|
|
|
void _xform_dialog_action();
|
|
void _menu_item_pressed(int p_option);
|
|
|
|
HBoxContainer *hbc_menu;
|
|
|
|
|
|
|
|
//
|
|
//
|
|
void _generate_selection_box();
|
|
UndoRedo *undo_redo;
|
|
|
|
void _instance_scene();
|
|
void _init_indicators();
|
|
void _finish_indicators();
|
|
|
|
void _toggle_maximize_view(Object* p_viewport);
|
|
|
|
Node *custom_camera;
|
|
|
|
Object *_get_editor_data(Object *p_what);
|
|
|
|
Ref<Environment> viewport_environment;
|
|
|
|
Spatial *selected;
|
|
|
|
void _request_gizmo(Object* p_obj);
|
|
|
|
static SpatialEditor *singleton;
|
|
|
|
void _node_removed(Node* p_node);
|
|
SpatialEditorGizmos *gizmos;
|
|
SpatialEditor();
|
|
|
|
void _update_ambient_light_color(const Color& p_color);
|
|
void _update_default_light_angle();
|
|
void _default_light_angle_input(const InputEvent& p_event);
|
|
|
|
protected:
|
|
|
|
|
|
|
|
|
|
void _notification(int p_what);
|
|
//void _input_event(InputEvent p_event);
|
|
void _unhandled_key_input(InputEvent p_event);
|
|
|
|
static void _bind_methods();
|
|
public:
|
|
|
|
|
|
static SpatialEditor *get_singleton() { return singleton; }
|
|
void snap_cursor_to_plane(const Plane& p_plane);
|
|
|
|
float get_znear() const { return settings_znear->get_val(); }
|
|
float get_zfar() const { return settings_zfar->get_val(); }
|
|
float get_fov() const { return settings_fov->get_val(); }
|
|
|
|
Transform get_gizmo_transform() const { return gizmo.transform; }
|
|
bool is_gizmo_visible() const { return gizmo.visible; }
|
|
|
|
ToolMode get_tool_mode() const { return tool_mode; }
|
|
bool is_snap_enabled() const { return snap_enabled; }
|
|
float get_translate_snap() const { return snap_translate->get_text().to_double(); }
|
|
float get_rotate_snap() const { return snap_rotate->get_text().to_double(); }
|
|
float get_scale_snap() const { return snap_scale->get_text().to_double(); }
|
|
|
|
Ref<Mesh> get_move_gizmo(int idx) const { return move_gizmo[idx]; }
|
|
Ref<Mesh> get_rotate_gizmo(int idx) const { return rotate_gizmo[idx]; }
|
|
|
|
void update_transform_gizmo();
|
|
|
|
void select_gizmo_hilight_axis(int p_axis);
|
|
void set_custom_camera(Node *p_camera) { custom_camera=p_camera; }
|
|
|
|
void set_undo_redo(UndoRedo *p_undo_redo) {undo_redo=p_undo_redo; }
|
|
Dictionary get_state() const;
|
|
void set_state(const Dictionary& p_state);
|
|
|
|
Ref<Environment> get_viewport_environment() { return viewport_environment; }
|
|
|
|
UndoRedo *get_undo_redo() { return undo_redo; }
|
|
|
|
void add_control_to_menu_panel(Control *p_control);
|
|
|
|
VSplitContainer *get_shader_split();
|
|
HSplitContainer *get_palette_split();
|
|
|
|
Spatial *get_selected() { return selected; }
|
|
|
|
int get_over_gizmo_handle() const { return over_gizmo_handle; }
|
|
void set_over_gizmo_handle(int idx) { over_gizmo_handle=idx; }
|
|
|
|
void set_can_preview(Camera* p_preview);
|
|
|
|
SpatialEditorViewport *get_editor_viewport(int p_idx) {
|
|
ERR_FAIL_INDEX_V(p_idx,4,NULL);
|
|
return viewports[p_idx];
|
|
}
|
|
|
|
Camera *get_camera() { return NULL; }
|
|
void edit(Spatial *p_spatial);
|
|
void clear();
|
|
SpatialEditor(EditorNode *p_editor);
|
|
~SpatialEditor();
|
|
};
|
|
|
|
class SpatialEditorPlugin : public EditorPlugin {
|
|
|
|
OBJ_TYPE( SpatialEditorPlugin, EditorPlugin );
|
|
|
|
SpatialEditor *spatial_editor;
|
|
EditorNode *editor;
|
|
protected:
|
|
static void _bind_methods();
|
|
public:
|
|
|
|
void snap_cursor_to_plane(const Plane& p_plane);
|
|
|
|
SpatialEditor *get_spatial_editor() { return spatial_editor; }
|
|
virtual String get_name() const { return "3D"; }
|
|
bool has_main_screen() const { return true; }
|
|
virtual void make_visible(bool p_visible);
|
|
virtual void edit(Object *p_object);
|
|
virtual bool handles(Object *p_object) const;
|
|
|
|
virtual Dictionary get_state() const;
|
|
virtual void set_state(const Dictionary& p_state);
|
|
virtual void clear() { spatial_editor->clear(); }
|
|
|
|
|
|
SpatialEditorPlugin(EditorNode *p_node);
|
|
~SpatialEditorPlugin();
|
|
|
|
};
|
|
|
|
#endif
|