/*************************************************************************/ /* spatial_editor_plugin.h */ /*************************************************************************/ /* This file is part of: */ /* GODOT ENGINE */ /* http://www.godotengine.org */ /*************************************************************************/ /* Copyright (c) 2007-2017 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 */ class Camera; class SpatialEditor; class SpatialEditorGizmos; class SpatialEditorGizmo : public SpatialGizmo { GDCLASS(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 &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 { GDCLASS( 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 &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, }; enum NavigationZoomStyle { NAVIGATION_ZOOM_VERTICAL, NAVIGATION_ZOOM_HORIZONTAL }; 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 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(); void focus_selection(); Viewport *get_viewport_node() { return viewport; } SpatialEditorViewport(SpatialEditor *p_spatial_editor,EditorNode *p_editor,int p_index); }; class SpatialEditorSelectedItem : public Object { GDCLASS(SpatialEditorSelectedItem,Object); public: Rect3 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 { GDCLASS(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 move_gizmo[3], rotate_gizmo[3]; Ref gizmo_color[3]; Ref gizmo_hl; int over_gizmo_handle; Ref selection_box; RID indicators; RID indicators_instance; RID cursor_mesh; RID cursor_instance; Ref indicator_mat; Ref cursor_material; /* struct Selected { AABB aabb; Transform original; // original location when moving Transform last_xform; // last transform Spatial *sp; RID poly_instance; }; Map 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; ViewportContainer *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 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 _gui_input(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_value(); } float get_zfar() const { return settings_zfar->get_value(); } float get_fov() const { return settings_fov->get_value(); } 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 get_move_gizmo(int idx) const { return move_gizmo[idx]; } Ref 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 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 { GDCLASS( 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