/*************************************************************************/ /* shader_graph_editor_plugin.cpp */ /*************************************************************************/ /* 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. */ /*************************************************************************/ #include "shader_graph_editor_plugin.h" #include "scene/gui/check_box.h" #include "scene/gui/menu_button.h" #include "scene/gui/panel.h" #include "spatial_editor_plugin.h" #include "os/keyboard.h" #include "canvas_item_editor_plugin.h" void GraphColorRampEdit::_input_event(const InputEvent& p_event) { if (p_event.type==InputEvent::KEY && p_event.key.pressed && p_event.key.scancode==KEY_DELETE && grabbed!=-1) { points.remove(grabbed); grabbed=-1; update(); emit_signal("ramp_changed"); accept_event(); } if (p_event.type==InputEvent::MOUSE_BUTTON && p_event.mouse_button.button_index==1 && p_event.mouse_button.pressed) { update(); int x = p_event.mouse_button.x; int total_w = get_size().width-get_size().height-3; if (x>total_w+3) { if (grabbed==-1) return; Size2 ms = Size2(350, picker->get_combined_minimum_size().height+10); picker->set_color(points[grabbed].color); popup->set_pos(get_global_pos()-Size2(0,ms.height)); popup->set_size(ms); popup->popup(); return; } float ofs = CLAMP(x/float(total_w),0,1); grabbed=-1; grabbing=true; int pos=-1; for(int i=0;iis_connected("color_changed",this,"_color_changed")) { picker->connect("color_changed",this,"_color_changed"); } } if (p_what==NOTIFICATION_DRAW) { Point prev; prev.offset=0; prev.color=Color(0,0,0); int h = get_size().y; int total_w = get_size().width-get_size().height-3; for(int i=-1;i points; Vector colors; points.push_back(Vector2(prev.offset*total_w,h)); points.push_back(Vector2(prev.offset*total_w,0)); points.push_back(Vector2(next.offset*total_w,0)); points.push_back(Vector2(next.offset*total_w,h)); colors.push_back(prev.color); colors.push_back(prev.color); colors.push_back(next.color); colors.push_back(next.color); draw_primitive(points,colors,Vector()); prev=next; } for(int i=0;i& p_offsets,const Vector& p_colors) { ERR_FAIL_COND(p_offsets.size()!=p_colors.size()); points.clear(); for(int i=0;i GraphColorRampEdit::get_offsets() const{ Vector ret; for(int i=0;i GraphColorRampEdit::get_colors() const{ Vector ret; for(int i=0;iadd_child(picker); popup->set_child_rect(picker); add_child(popup); } //////////// void GraphCurveMapEdit::_input_event(const InputEvent& p_event) { if (p_event.type==InputEvent::KEY && p_event.key.pressed && p_event.key.scancode==KEY_DELETE && grabbed!=-1) { points.remove(grabbed); grabbed=-1; update(); emit_signal("curve_changed"); accept_event(); } if (p_event.type==InputEvent::MOUSE_BUTTON && p_event.mouse_button.button_index==1 && p_event.mouse_button.pressed) { update(); Point2 p = Vector2(p_event.mouse_button.x,p_event.mouse_button.y)/get_size(); p.y=1.0-p.y; grabbed=-1; grabbing=true; for(int i=0;icurve[cd->outline][lastx] = lasty; } else { cd->curve_ptr[cd->outline][lastx] = lasty; if(gb_debug) printf("bender_plot_curve xmax:%d ymax:%d\n", (int)xmax, (int)ymax); } */ /* loop over the curve */ for (i = 0; i < ntimes; i++) { /* increment the x values */ x += dx; dx += dx2; dx2 += dx3; /* increment the y values */ y += dy; dy += dy2; dy2 += dy3; newx = CLAMP ((Math::round (x)), 0, xmax); newy = CLAMP ((Math::round (y)), 0, ymax); /* if this point is different than the last one...then draw it */ if ((lastx != newx) || (lasty != newy)) { #if 0 if(fix255) { /* use fixed array size (for the curve graph) */ cd->curve[cd->outline][newx] = newy; } else { /* use dynamic allocated curve_ptr (for the real curve) */ cd->curve_ptr[cd->outline][newx] = newy; if(gb_debug) printf("outline: %d cX: %d cY: %d\n", (int)cd->outline, (int)newx, (int)newy); } #endif draw_line(Vector2(lastx,ymax-lasty),Vector2(newx,ymax-newy),Color(0.8,0.8,0.8,0.8),2.0); } lastx = newx; lasty = newy; } } void GraphCurveMapEdit::_notification(int p_what){ if (p_what==NOTIFICATION_DRAW) { draw_style_box(get_stylebox("bg","Tree"),Rect2(Point2(),get_size())); int w = get_size().x; int h = get_size().y; Vector2 prev=Vector2(0,0); Vector2 prev2=Vector2(0,0); for(int i=-1;i=points.size()) { next=Vector2(1,1); } else { next=Vector2(points[i+1].offset,points[i+1].height); } if (i+2>=points.size()) { next2=Vector2(1,1); } else { next2=Vector2(points[i+2].offset,points[i+2].height); } /*if (i==-1 && prev.offset==next.offset) { prev=next; continue; }*/ _plot_curve(prev2,prev,next,next2); prev2=prev; prev=next; } for(int i=0;i& p_points) { points.clear(); for(int i=0;i GraphCurveMapEdit::get_points() const { Vector ret; for(int i=0;iget_undo_redo(); ur->create_action(TTR("Change Scalar Constant"),UndoRedo::MERGE_ENDS); ur->add_do_method(graph.ptr(),"scalar_const_node_set_value",type,p_id,p_value); ur->add_undo_method(graph.ptr(),"scalar_const_node_set_value",type,p_id,graph->scalar_const_node_get_value(type,p_id)); ur->add_do_method(this,"_update_graph"); ur->add_undo_method(this,"_update_graph"); block_update=true; ur->commit_action(); block_update=false; } void ShaderGraphView::_vec_const_changed(double p_value, int p_id,Array p_arr){ Vector3 val; for(int i=0;iget_undo_redo(); ur->create_action(TTR("Change Vec Constant"),UndoRedo::MERGE_ENDS); ur->add_do_method(graph.ptr(),"vec_const_node_set_value",type,p_id,val); ur->add_undo_method(graph.ptr(),"vec_const_node_set_value",type,p_id,graph->vec_const_node_get_value(type,p_id)); ur->add_do_method(this,"_update_graph"); ur->add_undo_method(this,"_update_graph"); block_update=true; ur->commit_action(); block_update=false; } void ShaderGraphView::_rgb_const_changed(const Color& p_color, int p_id){ UndoRedo *ur=EditorNode::get_singleton()->get_undo_redo(); ur->create_action(TTR("Change RGB Constant"),UndoRedo::MERGE_ENDS); ur->add_do_method(graph.ptr(),"rgb_const_node_set_value",type,p_id,p_color); ur->add_undo_method(graph.ptr(),"rgb_const_node_set_value",type,p_id,graph->rgb_const_node_get_value(type,p_id)); ur->add_do_method(this,"_update_graph"); ur->add_undo_method(this,"_update_graph"); block_update=true; ur->commit_action(); block_update=false; } void ShaderGraphView::_scalar_op_changed(int p_op, int p_id){ UndoRedo *ur=EditorNode::get_singleton()->get_undo_redo(); ur->create_action(TTR("Change Scalar Operator")); ur->add_do_method(graph.ptr(),"scalar_op_node_set_op",type,p_id,p_op); ur->add_undo_method(graph.ptr(),"scalar_op_node_set_op",type,p_id,graph->scalar_op_node_get_op(type,p_id)); ur->add_do_method(this,"_update_graph"); ur->add_undo_method(this,"_update_graph"); block_update=true; ur->commit_action(); block_update=false; } void ShaderGraphView::_vec_op_changed(int p_op, int p_id){ UndoRedo *ur=EditorNode::get_singleton()->get_undo_redo(); ur->create_action(TTR("Change Vec Operator")); ur->add_do_method(graph.ptr(),"vec_op_node_set_op",type,p_id,p_op); ur->add_undo_method(graph.ptr(),"vec_op_node_set_op",type,p_id,graph->vec_op_node_get_op(type,p_id)); ur->add_do_method(this,"_update_graph"); ur->add_undo_method(this,"_update_graph"); block_update=true; ur->commit_action(); block_update=false; } void ShaderGraphView::_vec_scalar_op_changed(int p_op, int p_id){ UndoRedo *ur=EditorNode::get_singleton()->get_undo_redo(); ur->create_action(TTR("Change Vec Scalar Operator")); ur->add_do_method(graph.ptr(),"vec_scalar_op_node_set_op",type,p_id,p_op); ur->add_undo_method(graph.ptr(),"vec_scalar_op_node_set_op",type,p_id,graph->vec_scalar_op_node_get_op(type,p_id)); ur->add_do_method(this,"_update_graph"); ur->add_undo_method(this,"_update_graph"); block_update=true; ur->commit_action(); block_update=false; } void ShaderGraphView::_rgb_op_changed(int p_op, int p_id){ UndoRedo *ur=EditorNode::get_singleton()->get_undo_redo(); ur->create_action(TTR("Change RGB Operator")); ur->add_do_method(graph.ptr(),"rgb_op_node_set_op",type,p_id,p_op); ur->add_undo_method(graph.ptr(),"rgb_op_node_set_op",type,p_id,graph->rgb_op_node_get_op(type,p_id)); ur->add_do_method(this,"_update_graph"); ur->add_undo_method(this,"_update_graph"); block_update=true; ur->commit_action(); block_update=false; } void ShaderGraphView::_xform_inv_rev_changed(bool p_enabled, int p_id){ UndoRedo *ur=EditorNode::get_singleton()->get_undo_redo(); ur->create_action(TTR("Toggle Rot Only")); ur->add_do_method(graph.ptr(),"xform_vec_mult_node_set_no_translation",type,p_id,p_enabled); ur->add_undo_method(graph.ptr(),"xform_vec_mult_node_set_no_translation",type,p_id,graph->xform_vec_mult_node_get_no_translation(type,p_id)); ur->add_do_method(this,"_update_graph"); ur->add_undo_method(this,"_update_graph"); block_update=true; ur->commit_action(); block_update=false; } void ShaderGraphView::_scalar_func_changed(int p_func, int p_id){ UndoRedo *ur=EditorNode::get_singleton()->get_undo_redo(); ur->create_action(TTR("Change Scalar Function")); ur->add_do_method(graph.ptr(),"scalar_func_node_set_function",type,p_id,p_func); ur->add_undo_method(graph.ptr(),"scalar_func_node_set_function",type,p_id,graph->scalar_func_node_get_function(type,p_id)); ur->add_do_method(this,"_update_graph"); ur->add_undo_method(this,"_update_graph"); block_update=true; ur->commit_action(); block_update=false; } void ShaderGraphView::_vec_func_changed(int p_func, int p_id){ UndoRedo *ur=EditorNode::get_singleton()->get_undo_redo(); ur->create_action(TTR("Change Vec Function")); ur->add_do_method(graph.ptr(),"vec_func_node_set_function",type,p_id,p_func); ur->add_undo_method(graph.ptr(),"vec_func_node_set_function",type,p_id,graph->vec_func_node_get_function(type,p_id)); ur->add_do_method(this,"_update_graph"); ur->add_undo_method(this,"_update_graph"); block_update=true; ur->commit_action(); block_update=false; } void ShaderGraphView::_scalar_input_changed(double p_value,int p_id){ UndoRedo *ur=EditorNode::get_singleton()->get_undo_redo(); ur->create_action(TTR("Change Scalar Uniform"),UndoRedo::MERGE_ENDS); ur->add_do_method(graph.ptr(),"scalar_input_node_set_value",type,p_id,p_value); ur->add_undo_method(graph.ptr(),"scalar_input_node_set_value",type,p_id,graph->scalar_input_node_get_value(type,p_id)); ur->add_do_method(this,"_update_graph"); ur->add_undo_method(this,"_update_graph"); block_update=true; ur->commit_action(); block_update=false; } void ShaderGraphView::_vec_input_changed(double p_value, int p_id,Array p_arr){ Vector3 val; for(int i=0;iget_undo_redo(); ur->create_action(TTR("Change Vec Uniform"),UndoRedo::MERGE_ENDS); ur->add_do_method(graph.ptr(),"vec_input_node_set_value",type,p_id,val); ur->add_undo_method(graph.ptr(),"vec_input_node_set_value",type,p_id,graph->vec_input_node_get_value(type,p_id)); ur->add_do_method(this,"_update_graph"); ur->add_undo_method(this,"_update_graph"); block_update=true; ur->commit_action(); block_update=false; } void ShaderGraphView::_xform_input_changed(int p_id, Node *p_button){ ToolButton *tb = p_button->cast_to(); ped_popup->set_pos(tb->get_global_pos()+Vector2(0,tb->get_size().height)); ped_popup->set_size(tb->get_size()); edited_id=p_id; edited_def=-1; ped_popup->edit(NULL,"",Variant::TRANSFORM,graph->xform_input_node_get_value(type,p_id),PROPERTY_HINT_NONE,""); ped_popup->popup(); } void ShaderGraphView::_xform_const_changed(int p_id, Node *p_button){ ToolButton *tb = p_button->cast_to(); ped_popup->set_pos(tb->get_global_pos()+Vector2(0,tb->get_size().height)); ped_popup->set_size(tb->get_size()); edited_id=p_id; edited_def=-1; ped_popup->edit(NULL,"",Variant::TRANSFORM,graph->xform_const_node_get_value(type,p_id),PROPERTY_HINT_NONE,""); ped_popup->popup(); } void ShaderGraphView::_rgb_input_changed(const Color& p_color, int p_id){ UndoRedo *ur=EditorNode::get_singleton()->get_undo_redo(); ur->create_action(TTR("Change RGB Uniform"),UndoRedo::MERGE_ENDS); ur->add_do_method(graph.ptr(),"rgb_input_node_set_value",type,p_id,p_color); ur->add_undo_method(graph.ptr(),"rgb_input_node_set_value",type,p_id,graph->rgb_input_node_get_value(type,p_id)); ur->add_do_method(this,"_update_graph"); ur->add_undo_method(this,"_update_graph"); block_update=true; ur->commit_action(); block_update=false; } void ShaderGraphView::_tex_input_change(int p_id, Node *p_button){ } void ShaderGraphView::_cube_input_change(int p_id){ } void ShaderGraphView::_variant_edited() { if (edited_def != -1) { Variant v = ped_popup->get_variant(); Variant v2 = graph->default_get_value(type,edited_id,edited_def); if (v2.get_type() == Variant::NIL) switch (v.get_type()) { case Variant::VECTOR3: v2=Vector3(); break; case Variant::REAL: v2=0.0; break; case Variant::TRANSFORM: v2=Transform(); break; case Variant::COLOR: v2=Color(); break; default: {} } UndoRedo *ur=EditorNode::get_singleton()->get_undo_redo(); ur->create_action(TTR("Change Default Value")); ur->add_do_method(graph.ptr(),"default_set_value",type,edited_id,edited_def, v); ur->add_undo_method(graph.ptr(),"default_set_value",type,edited_id,edited_def, v2); ur->add_do_method(this,"_update_graph"); ur->add_undo_method(this,"_update_graph"); ur->commit_action(); return; } if (graph->node_get_type(type,edited_id)==ShaderGraph::NODE_XFORM_CONST) { UndoRedo *ur=EditorNode::get_singleton()->get_undo_redo(); ur->create_action(TTR("Change XForm Uniform")); ur->add_do_method(graph.ptr(),"xform_const_node_set_value",type,edited_id,ped_popup->get_variant()); ur->add_undo_method(graph.ptr(),"xform_const_node_set_value",type,edited_id,graph->xform_const_node_get_value(type,edited_id)); ur->add_do_method(this,"_update_graph"); ur->add_undo_method(this,"_update_graph"); ur->commit_action(); } if (graph->node_get_type(type,edited_id)==ShaderGraph::NODE_XFORM_INPUT) { UndoRedo *ur=EditorNode::get_singleton()->get_undo_redo(); ur->create_action(TTR("Change XForm Uniform")); ur->add_do_method(graph.ptr(),"xform_input_node_set_value",type,edited_id,ped_popup->get_variant()); ur->add_undo_method(graph.ptr(),"xform_input_node_set_value",type,edited_id,graph->xform_input_node_get_value(type,edited_id)); ur->add_do_method(this,"_update_graph"); ur->add_undo_method(this,"_update_graph"); ur->commit_action(); } if (graph->node_get_type(type,edited_id)==ShaderGraph::NODE_TEXTURE_INPUT) { UndoRedo *ur=EditorNode::get_singleton()->get_undo_redo(); ur->create_action(TTR("Change Texture Uniform")); ur->add_do_method(graph.ptr(),"texture_input_node_set_value",type,edited_id,ped_popup->get_variant()); ur->add_undo_method(graph.ptr(),"texture_input_node_set_value",type,edited_id,graph->texture_input_node_get_value(type,edited_id)); ur->add_do_method(this,"_update_graph"); ur->add_undo_method(this,"_update_graph"); ur->commit_action(); } if (graph->node_get_type(type,edited_id)==ShaderGraph::NODE_CUBEMAP_INPUT) { UndoRedo *ur=EditorNode::get_singleton()->get_undo_redo(); ur->create_action(TTR("Change Cubemap Uniform")); ur->add_do_method(graph.ptr(),"cubemap_input_node_set_value",type,edited_id,ped_popup->get_variant()); ur->add_undo_method(graph.ptr(),"cubemap_input_node_set_value",type,edited_id,graph->cubemap_input_node_get_value(type,edited_id)); ur->add_do_method(this,"_update_graph"); ur->add_undo_method(this,"_update_graph"); ur->commit_action(); } } void ShaderGraphView::_comment_edited(int p_id,Node* p_button) { UndoRedo *ur=EditorNode::get_singleton()->get_undo_redo(); TextEdit *te=p_button->cast_to(); ur->create_action(TTR("Change Comment"),UndoRedo::MERGE_ENDS); ur->add_do_method(graph.ptr(),"comment_node_set_text",type,p_id,te->get_text()); ur->add_undo_method(graph.ptr(),"comment_node_set_text",type,p_id,graph->comment_node_get_text(type,p_id)); ur->add_do_method(this,"_update_graph"); ur->add_undo_method(this,"_update_graph"); block_update=true; ur->commit_action(); block_update=false; } void ShaderGraphView::_color_ramp_changed(int p_id,Node* p_ramp) { GraphColorRampEdit *cr=p_ramp->cast_to(); UndoRedo *ur=EditorNode::get_singleton()->get_undo_redo(); Vector offsets=cr->get_offsets(); Vector colors=cr->get_colors(); DVector new_offsets; DVector new_colors; { new_offsets.resize(offsets.size()); new_colors.resize(colors.size()); DVector::Write ow=new_offsets.write(); DVector::Write cw=new_colors.write(); for(int i=0;i old_offsets=graph->color_ramp_node_get_offsets(type,p_id); DVector old_colors=graph->color_ramp_node_get_colors(type,p_id); if (old_offsets.size()!=new_offsets.size()) ur->create_action(TTR("Add/Remove to Color Ramp")); else ur->create_action(TTR("Modify Color Ramp"),UndoRedo::MERGE_ENDS); ur->add_do_method(graph.ptr(),"color_ramp_node_set_ramp",type,p_id,new_colors,new_offsets); ur->add_undo_method(graph.ptr(),"color_ramp_node_set_ramp",type,p_id,old_colors,old_offsets); ur->add_do_method(this,"_update_graph"); ur->add_undo_method(this,"_update_graph"); block_update=true; ur->commit_action(); block_update=false; } void ShaderGraphView::_curve_changed(int p_id,Node* p_curve) { GraphCurveMapEdit *cr=p_curve->cast_to(); UndoRedo *ur=EditorNode::get_singleton()->get_undo_redo(); Vector points=cr->get_points(); DVector new_points; { new_points.resize(points.size()); DVector::Write ow=new_points.write(); for(int i=0;i old_points=graph->curve_map_node_get_points(type,p_id); if (old_points.size()!=new_points.size()) ur->create_action(TTR("Add/Remove to Curve Map")); else ur->create_action(TTR("Modify Curve Map"),UndoRedo::MERGE_ENDS); ur->add_do_method(graph.ptr(),"curve_map_node_set_points",type,p_id,new_points); ur->add_undo_method(graph.ptr(),"curve_map_node_set_points",type,p_id,old_points); ur->add_do_method(this,"_update_graph"); ur->add_undo_method(this,"_update_graph"); block_update=true; ur->commit_action(); block_update=false; } void ShaderGraphView::_input_name_changed(const String& p_name, int p_id, Node *p_line_edit) { LineEdit *le=p_line_edit->cast_to(); ERR_FAIL_COND(!le); UndoRedo *ur=EditorNode::get_singleton()->get_undo_redo(); ur->create_action(TTR("Change Input Name")); ur->add_do_method(graph.ptr(),"input_node_set_name",type,p_id,p_name); ur->add_undo_method(graph.ptr(),"input_node_set_name",type,p_id,graph->input_node_get_name(type,p_id)); ur->add_do_method(this,"_update_graph"); ur->add_undo_method(this,"_update_graph"); block_update=true; ur->commit_action(); block_update=false; le->set_text(graph->input_node_get_name(type,p_id)); } void ShaderGraphView::_tex_edited(int p_id,Node* p_button) { ToolButton *tb = p_button->cast_to(); ped_popup->set_pos(tb->get_global_pos()+Vector2(0,tb->get_size().height)); ped_popup->set_size(tb->get_size()); edited_id=p_id; edited_def=-1; ped_popup->edit(NULL,"",Variant::OBJECT,graph->texture_input_node_get_value(type,p_id),PROPERTY_HINT_RESOURCE_TYPE,"Texture"); } void ShaderGraphView::_cube_edited(int p_id,Node* p_button) { ToolButton *tb = p_button->cast_to(); ped_popup->set_pos(tb->get_global_pos()+Vector2(0,tb->get_size().height)); ped_popup->set_size(tb->get_size()); edited_id=p_id; edited_def=-1; ped_popup->edit(NULL,"",Variant::OBJECT,graph->cubemap_input_node_get_value(type,p_id),PROPERTY_HINT_RESOURCE_TYPE,"CubeMap"); } //////////////view///////////// void ShaderGraphView::_connection_request(const String& p_from, int p_from_slot,const String& p_to,int p_to_slot) { UndoRedo *ur=EditorNode::get_singleton()->get_undo_redo(); int from_idx=-1; int to_idx=-1; for (Map::Element *E=node_map.front();E;E=E->next()) { if (p_from==E->get()->get_name()) from_idx=E->key(); if (p_to==E->get()->get_name()) to_idx=E->key(); } ERR_FAIL_COND(from_idx==-1); ERR_FAIL_COND(to_idx==-1); ur->create_action(TTR("Connect Graph Nodes")); List conns; graph->get_node_connections(type,&conns); //disconnect/reconnect dependencies ur->add_undo_method(graph.ptr(),"disconnect_node",type,from_idx,p_from_slot,to_idx,p_to_slot); for(List::Element *E=conns.front();E;E=E->next()) { if (E->get().dst_id==to_idx && E->get().dst_slot==p_to_slot) { ur->add_do_method(graph.ptr(),"disconnect_node",type,E->get().src_id,E->get().src_slot,E->get().dst_id,E->get().dst_slot); ur->add_undo_method(graph.ptr(),"connect_node",type,E->get().src_id,E->get().src_slot,E->get().dst_id,E->get().dst_slot); } } ur->add_do_method(graph.ptr(),"connect_node",type,from_idx,p_from_slot,to_idx,p_to_slot); ur->add_do_method(this,"_update_graph"); ur->add_undo_method(this,"_update_graph"); ur->commit_action(); } void ShaderGraphView::_disconnection_request(const String& p_from, int p_from_slot,const String& p_to,int p_to_slot) { UndoRedo *ur=EditorNode::get_singleton()->get_undo_redo(); int from_idx=-1; int to_idx=-1; for (Map::Element *E=node_map.front();E;E=E->next()) { if (p_from==E->get()->get_name()) from_idx=E->key(); if (p_to==E->get()->get_name()) to_idx=E->key(); } ERR_FAIL_COND(from_idx==-1); ERR_FAIL_COND(to_idx==-1); if (!graph->is_node_connected(type,from_idx,p_from_slot,to_idx,p_to_slot)) return; //nothing to disconnect ur->create_action(TTR("Disconnect Graph Nodes")); List conns; graph->get_node_connections(type,&conns); //disconnect/reconnect dependencies ur->add_do_method(graph.ptr(),"disconnect_node",type,from_idx,p_from_slot,to_idx,p_to_slot); ur->add_undo_method(graph.ptr(),"connect_node",type,from_idx,p_from_slot,to_idx,p_to_slot); ur->add_do_method(this,"_update_graph"); ur->add_undo_method(this,"_update_graph"); ur->commit_action(); } void ShaderGraphView::_node_removed(int p_id) { UndoRedo *ur=EditorNode::get_singleton()->get_undo_redo(); ur->create_action(TTR("Remove Shader Graph Node")); ur->add_do_method(graph.ptr(),"node_remove",type,p_id); ur->add_undo_method(graph.ptr(),"node_add",type,graph->node_get_type(type,p_id),p_id); ur->add_undo_method(graph.ptr(),"node_set_state",type,p_id,graph->node_get_state(type,p_id)); List conns; graph->get_node_connections(type,&conns); for(List::Element *E=conns.front();E;E=E->next()) { if (E->get().dst_id==p_id || E->get().src_id==p_id) { ur->add_undo_method(graph.ptr(),"connect_node",type,E->get().src_id,E->get().src_slot,E->get().dst_id,E->get().dst_slot); } } ur->add_do_method(this,"_update_graph"); ur->add_undo_method(this,"_update_graph"); ur->commit_action(); } void ShaderGraphView::_begin_node_move() { UndoRedo *ur=EditorNode::get_singleton()->get_undo_redo(); ur->create_action(TTR("Move Shader Graph Node")); } void ShaderGraphView::_node_moved(const Vector2& p_from, const Vector2& p_to,int p_id) { ERR_FAIL_COND(!node_map.has(p_id)); UndoRedo *ur=EditorNode::get_singleton()->get_undo_redo(); ur->add_do_method(this,"_move_node",p_id,p_to); ur->add_undo_method(this,"_move_node",p_id,p_from); } void ShaderGraphView::_end_node_move() { UndoRedo *ur=EditorNode::get_singleton()->get_undo_redo(); ur->commit_action(); } void ShaderGraphView::_move_node(int p_id,const Vector2& p_to) { ERR_FAIL_COND(!node_map.has(p_id)); node_map[p_id]->set_offset(p_to); graph->node_set_pos(type,p_id,p_to); } void ShaderGraphView::_duplicate_nodes_request() { Array s_id; for(Map::Element *E=node_map.front();E;E=E->next()) { ShaderGraph::NodeType t=graph->node_get_type(type, E->key()); if (t==ShaderGraph::NODE_OUTPUT || t==ShaderGraph::NODE_INPUT) continue; GraphNode *gn = E->get(); if (gn && gn->is_selected()) s_id.push_back(E->key()); } if (s_id.size()==0) return; UndoRedo *ur=EditorNode::get_singleton()->get_undo_redo(); ur->create_action(TTR("Duplicate Graph Node(s)")); ur->add_do_method(this,"_duplicate_nodes",s_id); List n_ids = graph->generate_ids(type, s_id.size()); for (List::Element *E=n_ids.front();E;E=E->next()) ur->add_undo_method(graph.ptr(),"node_remove",type,E->get()); ur->add_do_method(this,"_update_graph"); ur->add_undo_method(this,"_update_graph"); ur->commit_action(); } void ShaderGraphView::_duplicate_nodes(const Array &p_nodes) { List n = List(); for (int i=0; iduplicate_nodes(type, n); call_deferred("_update_graph"); } void ShaderGraphView::_delete_nodes_request() { List s_id=List(); for(Map::Element *E=node_map.front();E;E=E->next()) { ShaderGraph::NodeType t=graph->node_get_type(type, E->key()); if (t==ShaderGraph::NODE_OUTPUT) continue; GraphNode *gn = E->get(); if (gn && gn->is_selected()) s_id.push_back(E->key()); } if (s_id.size()==0) return; UndoRedo *ur=EditorNode::get_singleton()->get_undo_redo(); ur->create_action(TTR("Delete Shader Graph Node(s)")); for (List::Element *N=s_id.front();N;N=N->next()) { ur->add_do_method(graph.ptr(),"node_remove",type,N->get()); ur->add_undo_method(graph.ptr(),"node_add",type,graph->node_get_type(type,N->get()),N->get()); ur->add_undo_method(graph.ptr(),"node_set_state",type,N->get(),graph->node_get_state(type,N->get())); List conns; graph->get_node_connections(type,&conns); for(List::Element *E=conns.front();E;E=E->next()) { if (E->get().dst_id==N->get() || E->get().src_id==N->get()) { ur->add_undo_method(graph.ptr(),"connect_node",type,E->get().src_id,E->get().src_slot,E->get().dst_id,E->get().dst_slot); } } } ur->add_do_method(this,"_update_graph"); ur->add_undo_method(this,"_update_graph"); ur->commit_action(); } void ShaderGraphView::_default_changed(int p_id, Node *p_button, int p_param, int v_type, String p_hint) { ToolButton *tb = p_button->cast_to(); ped_popup->set_pos(tb->get_global_pos()+Vector2(0,tb->get_size().height)); ped_popup->set_size(tb->get_size()); edited_id=p_id; edited_def=p_param; Variant::Type vt = (Variant::Type)v_type; Variant v = graph->default_get_value(type,p_id,edited_def); int h=PROPERTY_HINT_NONE; if (v.get_type() == Variant::NIL) switch (vt) { case Variant::VECTOR3: v=Vector3(); break; case Variant::REAL: h=PROPERTY_HINT_RANGE; v=0.0; break; case Variant::TRANSFORM: v=Transform(); break; case Variant::COLOR: h=PROPERTY_HINT_COLOR_NO_ALPHA; v=Color(); break; default: {} } ped_popup->edit(NULL,"",vt,v,h,p_hint); ped_popup->popup(); } ToolButton *ShaderGraphView::make_label(String text, Variant::Type v_type) { ToolButton *l = memnew( ToolButton ); l->set_text(text); l->set_text_align(ToolButton::ALIGN_LEFT); l->add_style_override("hover", l->get_stylebox("normal", "ToolButton")); l->add_style_override("pressed", l->get_stylebox("normal", "ToolButton")); l->add_style_override("focus", l->get_stylebox("normal", "ToolButton")); switch (v_type) { case Variant::REAL: l->set_icon(ped_popup->get_icon("Real", "EditorIcons")); break; case Variant::VECTOR3: l->set_icon(ped_popup->get_icon("Vector", "EditorIcons")); break; case Variant::TRANSFORM: l->set_icon(ped_popup->get_icon("Matrix", "EditorIcons")); break; case Variant::COLOR: l->set_icon(ped_popup->get_icon("Color", "EditorIcons")); break; default: {} } return l; } ToolButton *ShaderGraphView::make_editor(String text,GraphNode* gn,int p_id,int param,Variant::Type v_type, String p_hint) { ToolButton *edit = memnew( ToolButton ); edit->set_text(text); edit->set_text_align(ToolButton::ALIGN_LEFT); edit->set_flat(false); edit->add_style_override("normal", gn->get_stylebox("defaultframe", "GraphNode")); edit->add_style_override("hover", gn->get_stylebox("defaultframe", "GraphNode")); edit->add_style_override("pressed", gn->get_stylebox("defaultframe", "GraphNode")); edit->add_style_override("focus", gn->get_stylebox("defaultfocus", "GraphNode")); edit->connect("pressed",this,"_default_changed",varray(p_id,edit,param,v_type,p_hint)); switch (v_type) { case Variant::REAL: edit->set_icon(ped_popup->get_icon("Real", "EditorIcons")); break; case Variant::VECTOR3: edit->set_icon(ped_popup->get_icon("Vector", "EditorIcons")); break; case Variant::TRANSFORM: edit->set_icon(ped_popup->get_icon("Matrix", "EditorIcons")); break; case Variant::COLOR: { Image icon_color = Image(15,15,false,Image::FORMAT_RGB); Color c = graph->default_get_value(type,p_id,param); for (int x=1;x<14;x++) for (int y=1;y<14;y++) icon_color.put_pixel(x,y,c); Ref t; t.instance(); t->create_from_image(icon_color); edit->set_icon(t); } break; default: {} } return edit; } void ShaderGraphView::_create_node(int p_id) { GraphNode *gn = memnew( GraphNode ); gn->set_show_close_button(true); Color typecol[4]={ Color(0.9,0.4,1), Color(0.8,1,0.2), Color(1,0.2,0.2), Color(0,1,1) }; const String hint_spin = "-65536,65535,0.001"; const String hint_slider = "0.0,1.0,0.01,slider"; switch(graph->node_get_type(type,p_id)) { case ShaderGraph::NODE_INPUT: { gn->set_title("Input"); List si; ShaderGraph::get_input_output_node_slot_info(graph->get_mode(),type,&si); int idx=0; for (List::Element *E=si.front();E;E=E->next()) { ShaderGraph::SlotInfo& s=E->get(); if (s.dir==ShaderGraph::SLOT_IN) { Label *l= memnew( Label ); l->set_text(s.name); l->set_align(Label::ALIGN_RIGHT); gn->add_child(l); gn->set_slot(idx,false,0,Color(),true,s.type,typecol[s.type]); idx++; } } } break; // all inputs (case Shader type dependent) case ShaderGraph::NODE_SCALAR_CONST: { gn->set_title("Scalar"); SpinBox *sb = memnew( SpinBox ); sb->set_min(-100000); sb->set_max(100000); sb->set_step(0.001); sb->set_val(graph->scalar_const_node_get_value(type,p_id)); sb->connect("value_changed",this,"_scalar_const_changed",varray(p_id)); gn->add_child(sb); gn->set_slot(0,false,0,Color(),true,ShaderGraph::SLOT_TYPE_SCALAR,typecol[ShaderGraph::SLOT_TYPE_SCALAR]); } break; //scalar constant case ShaderGraph::NODE_VEC_CONST: { gn->set_title("Vector"); Array v3p(true); for(int i=0;i<3;i++) { HBoxContainer *hbc = memnew( HBoxContainer ); Label *l = memnew( Label ); l->set_text(String::chr('X'+i)); hbc->add_child(l); SpinBox *sb = memnew( SpinBox ); sb->set_h_size_flags(Control::SIZE_EXPAND_FILL); sb->set_min(-100000); sb->set_max(100000); sb->set_step(0.001); sb->set_val(graph->vec_const_node_get_value(type,p_id)[i]); sb->connect("value_changed",this,"_vec_const_changed",varray(p_id,v3p)); v3p.push_back(sb); hbc->add_child(sb); gn->add_child(hbc); } gn->set_slot(0,false,0,Color(),true,ShaderGraph::SLOT_TYPE_VEC,typecol[ShaderGraph::SLOT_TYPE_VEC]); } break; //vec3 constant case ShaderGraph::NODE_RGB_CONST: { gn->set_title("Color"); ColorPickerButton *cpb = memnew( ColorPickerButton ); cpb->set_color(graph->rgb_const_node_get_value(type,p_id)); cpb->connect("color_changed",this,"_rgb_const_changed",varray(p_id)); gn->add_child(cpb); Label *l = memnew( Label ); l->set_text("RGB"); l->set_align(Label::ALIGN_RIGHT); gn->add_child(l); l = memnew( Label ); l->set_text("Alpha"); l->set_align(Label::ALIGN_RIGHT); gn->add_child(l); gn->set_slot(1,false,0,Color(),true,ShaderGraph::SLOT_TYPE_VEC,typecol[ShaderGraph::SLOT_TYPE_VEC]); gn->set_slot(2,false,0,Color(),true,ShaderGraph::SLOT_TYPE_SCALAR,typecol[ShaderGraph::SLOT_TYPE_SCALAR]); } break; //rgb constant (shows a color picker instead) case ShaderGraph::NODE_XFORM_CONST: { gn->set_title("XForm"); ToolButton *edit = memnew( ToolButton ); edit->set_text("edit.."); edit->connect("pressed",this,"_xform_const_changed",varray(p_id,edit)); gn->add_child(edit); gn->set_slot(0,false,0,Color(),true,ShaderGraph::SLOT_TYPE_XFORM,typecol[ShaderGraph::SLOT_TYPE_XFORM]); } break; // 4x4 matrix constant case ShaderGraph::NODE_TIME: { gn->set_title("Time"); Label *l = memnew( Label ); l->set_text("(s)"); l->set_align(Label::ALIGN_RIGHT); gn->add_child(l); gn->set_slot(0,false,0,Color(),true,ShaderGraph::SLOT_TYPE_SCALAR,typecol[ShaderGraph::SLOT_TYPE_SCALAR]); } break; // time in seconds case ShaderGraph::NODE_SCREEN_TEX: { gn->set_title("ScreenTex"); HBoxContainer *hbc = memnew( HBoxContainer ); hbc->add_constant_override("separation",0); if (!graph->is_slot_connected(type,p_id,0)) { Vector3 v = graph->default_get_value(type, p_id, 0); hbc->add_child(make_editor("UV: " + v,gn,p_id,0,Variant::VECTOR3)); } else { hbc->add_child(make_label("UV",Variant::VECTOR3)); } hbc->add_spacer(); hbc->add_child( memnew(Label("RGB"))); gn->add_child(hbc); gn->set_slot(0,true,ShaderGraph::SLOT_TYPE_VEC,typecol[ShaderGraph::SLOT_TYPE_VEC],true,ShaderGraph::SLOT_TYPE_VEC,typecol[ShaderGraph::SLOT_TYPE_VEC]); } break; // screen texture sampler (takes UV) (only usable in fragment case Shader) case ShaderGraph::NODE_SCALAR_OP: { gn->set_title("ScalarOp"); static const char* op_name[ShaderGraph::SCALAR_MAX_OP]={ ("Add"), ("Sub"), ("Mul"), ("Div"), ("Mod"), ("Pow"), ("Max"), ("Min"), ("Atan2") }; OptionButton *ob = memnew( OptionButton ); for(int i=0;iadd_item(op_name[i],i); } ob->select(graph->scalar_op_node_get_op(type,p_id)); ob->connect("item_selected",this,"_scalar_op_changed",varray(p_id)); gn->add_child(ob); HBoxContainer *hbc = memnew( HBoxContainer ); hbc->add_constant_override("separation",0); if (graph->is_slot_connected(type, p_id, 0)) { hbc->add_child(make_label("a",Variant::REAL)); } else { float v = graph->default_get_value(type,p_id,0); hbc->add_child(make_editor(String("a: ")+Variant(v),gn,p_id,0,Variant::REAL,hint_spin)); } hbc->add_spacer(); hbc->add_child( memnew(Label("out"))); gn->add_child(hbc); if (graph->is_slot_connected(type, p_id, 1)) { gn->add_child(make_label("b",Variant::REAL)); } else { float v = graph->default_get_value(type,p_id,1); gn->add_child(make_editor(String("b: ")+Variant(v),gn,p_id,1,Variant::REAL,hint_spin)); } gn->set_slot(1,true,ShaderGraph::SLOT_TYPE_SCALAR,typecol[ShaderGraph::SLOT_TYPE_SCALAR],true,ShaderGraph::SLOT_TYPE_SCALAR,typecol[ShaderGraph::SLOT_TYPE_SCALAR]); gn->set_slot(2,true,ShaderGraph::SLOT_TYPE_SCALAR,typecol[ShaderGraph::SLOT_TYPE_SCALAR],false,0,Color()); } break; // scalar vs scalar op (mul: { } break; add: { } break; div: { } break; etc) case ShaderGraph::NODE_VEC_OP: { gn->set_title("VecOp"); static const char* op_name[ShaderGraph::VEC_MAX_OP]={ ("Add"), ("Sub"), ("Mul"), ("Div"), ("Mod"), ("Pow"), ("Max"), ("Min"), ("Cross") }; OptionButton *ob = memnew( OptionButton ); for(int i=0;iadd_item(op_name[i],i); } ob->select(graph->vec_op_node_get_op(type,p_id)); ob->connect("item_selected",this,"_vec_op_changed",varray(p_id)); gn->add_child(ob); HBoxContainer *hbc = memnew( HBoxContainer ); hbc->add_constant_override("separation",0); if (graph->is_slot_connected(type, p_id, 0)) { hbc->add_child(make_label("a",Variant::VECTOR3)); } else { Vector3 v = graph->default_get_value(type,p_id,0); hbc->add_child(make_editor(String("a: ")+v,gn,p_id,0,Variant::VECTOR3)); } hbc->add_spacer(); hbc->add_child( memnew(Label("out"))); gn->add_child(hbc); if (graph->is_slot_connected(type, p_id, 1)) { gn->add_child(make_label("b",Variant::VECTOR3)); } else { Vector3 v = graph->default_get_value(type,p_id,1); gn->add_child(make_editor(String("b: ")+v,gn,p_id,1,Variant::VECTOR3)); } gn->set_slot(1,true,ShaderGraph::SLOT_TYPE_VEC,typecol[ShaderGraph::SLOT_TYPE_VEC],true,ShaderGraph::SLOT_TYPE_VEC,typecol[ShaderGraph::SLOT_TYPE_VEC]); gn->set_slot(2,true,ShaderGraph::SLOT_TYPE_VEC,typecol[ShaderGraph::SLOT_TYPE_VEC],false,0,Color()); } break; // vec3 vs vec3 op (mul: { } break;ad: { } break;div: { } break;crossprod: { } break;etc) case ShaderGraph::NODE_VEC_SCALAR_OP: { gn->set_title("VecScalarOp"); static const char* op_name[ShaderGraph::VEC_SCALAR_MAX_OP]={ ("Mul"), ("Div"), ("Pow"), }; OptionButton *ob = memnew( OptionButton ); for(int i=0;iadd_item(op_name[i],i); } ob->select(graph->vec_scalar_op_node_get_op(type,p_id)); ob->connect("item_selected",this,"_vec_scalar_op_changed",varray(p_id)); gn->add_child(ob); HBoxContainer *hbc = memnew( HBoxContainer ); hbc->add_constant_override("separation",0); if (graph->is_slot_connected(type, p_id, 0)) { hbc->add_child(make_label("a",Variant::VECTOR3)); } else { Vector3 v = graph->default_get_value(type,p_id,0); hbc->add_child(make_editor(String("a: ")+v,gn,p_id,0,Variant::VECTOR3)); } hbc->add_spacer(); hbc->add_child( memnew(Label("out"))); gn->add_child(hbc); if (graph->is_slot_connected(type, p_id, 1)) { gn->add_child(make_label("b",Variant::REAL)); } else { float v = graph->default_get_value(type,p_id,1); gn->add_child(make_editor(String("b: ")+Variant(v),gn,p_id,1,Variant::REAL,hint_spin)); } gn->set_slot(1,true,ShaderGraph::SLOT_TYPE_VEC,typecol[ShaderGraph::SLOT_TYPE_VEC],true,ShaderGraph::SLOT_TYPE_VEC,typecol[ShaderGraph::SLOT_TYPE_VEC]); gn->set_slot(2,true,ShaderGraph::SLOT_TYPE_SCALAR,typecol[ShaderGraph::SLOT_TYPE_SCALAR],false,0,Color()); } break; // vec3 vs scalar op (mul: { } break; add: { } break; div: { } break; etc) case ShaderGraph::NODE_RGB_OP: { gn->set_title("RGB Op"); static const char* op_name[ShaderGraph::RGB_MAX_OP]={ ("Screen"), ("Difference"), ("Darken"), ("Lighten"), ("Overlay"), ("Dodge"), ("Burn"), ("SoftLight"), ("HardLight") }; OptionButton *ob = memnew( OptionButton ); for(int i=0;iadd_item(op_name[i],i); } ob->select(graph->rgb_op_node_get_op(type,p_id)); ob->connect("item_selected",this,"_rgb_op_changed",varray(p_id)); gn->add_child(ob); HBoxContainer *hbc = memnew( HBoxContainer ); hbc->add_constant_override("separation",0); if (graph->is_slot_connected(type, p_id, 0)) { hbc->add_child(make_label("a",Variant::COLOR)); } else { hbc->add_child(make_editor(String("a: "),gn,p_id,0,Variant::COLOR)); } hbc->add_spacer(); hbc->add_child( memnew(Label("out"))); gn->add_child(hbc); if (graph->is_slot_connected(type, p_id, 1)) { gn->add_child(make_label("b",Variant::COLOR)); } else { gn->add_child(make_editor(String("b: "),gn,p_id,1,Variant::COLOR)); } gn->set_slot(1,true,ShaderGraph::SLOT_TYPE_VEC,typecol[ShaderGraph::SLOT_TYPE_VEC],true,ShaderGraph::SLOT_TYPE_VEC,typecol[ShaderGraph::SLOT_TYPE_VEC]); gn->set_slot(2,true,ShaderGraph::SLOT_TYPE_VEC,typecol[ShaderGraph::SLOT_TYPE_VEC],false,0,Color()); } break; // vec3 vs vec3 rgb op (with scalar amount): { } break; like brighten: { } break; darken: { } break; burn: { } break; dodge: { } break; multiply: { } break; etc. case ShaderGraph::NODE_XFORM_MULT: { gn->set_title("XFMult"); HBoxContainer *hbc = memnew( HBoxContainer ); if (graph->is_slot_connected(type, p_id, 0)) { hbc->add_child(make_label("a",Variant::TRANSFORM)); } else { hbc->add_child(make_editor(String("a: edit..."),gn,p_id,0,Variant::TRANSFORM)); } hbc->add_spacer(); hbc->add_child( memnew(Label("out"))); gn->add_child(hbc); if (graph->is_slot_connected(type, p_id, 1)) { gn->add_child(make_label("b",Variant::TRANSFORM)); } else { gn->add_child(make_editor(String("b: edit..."),gn,p_id,1,Variant::TRANSFORM)); } gn->set_slot(0,true,ShaderGraph::SLOT_TYPE_XFORM,typecol[ShaderGraph::SLOT_TYPE_XFORM],true,ShaderGraph::SLOT_TYPE_XFORM,typecol[ShaderGraph::SLOT_TYPE_XFORM]); gn->set_slot(1,true,ShaderGraph::SLOT_TYPE_XFORM,typecol[ShaderGraph::SLOT_TYPE_XFORM],false,0,Color()); } break; // mat4 x mat4 case ShaderGraph::NODE_XFORM_VEC_MULT: { gn->set_title("XFVecMult"); CheckBox *button = memnew (CheckBox("RotOnly")); button->set_pressed(graph->xform_vec_mult_node_get_no_translation(type,p_id)); button->connect("toggled",this,"_xform_inv_rev_changed",varray(p_id)); gn->add_child(button); HBoxContainer *hbc = memnew( HBoxContainer ); hbc->add_constant_override("separation",0); if (graph->is_slot_connected(type, p_id, 0)) { hbc->add_child(make_label("xf",Variant::TRANSFORM)); } else { hbc->add_child(make_editor(String("xf: edit..."),gn,p_id,0,Variant::TRANSFORM)); } hbc->add_spacer(); Label *l = memnew(Label("out")); l->set_align(Label::ALIGN_RIGHT); hbc->add_child( l); gn->add_child(hbc); if (graph->is_slot_connected(type, p_id, 1)) { gn->add_child(make_label("a",Variant::VECTOR3)); } else { Vector3 v = graph->default_get_value(type,p_id,1); gn->add_child(make_editor(String("a: ")+v,gn,p_id,1,Variant::VECTOR3)); } gn->set_slot(1,true,ShaderGraph::SLOT_TYPE_XFORM,typecol[ShaderGraph::SLOT_TYPE_XFORM],true,ShaderGraph::SLOT_TYPE_VEC,typecol[ShaderGraph::SLOT_TYPE_VEC]); gn->set_slot(2,true,ShaderGraph::SLOT_TYPE_VEC,typecol[ShaderGraph::SLOT_TYPE_VEC],false,0,Color()); } break; case ShaderGraph::NODE_XFORM_VEC_INV_MULT: { gn->set_title("XFVecInvMult"); CheckBox *button = memnew( CheckBox("RotOnly")); button->set_pressed(graph->xform_vec_mult_node_get_no_translation(type,p_id)); button->connect("toggled",this,"_xform_inv_rev_changed",varray(p_id)); gn->add_child(button); if (graph->is_slot_connected(type, p_id, 0)) { gn->add_child(make_label("a",Variant::VECTOR3)); } else { Vector3 v = graph->default_get_value(type,p_id,0); gn->add_child(make_editor(String("a: ")+v,gn,p_id,0,Variant::VECTOR3)); } HBoxContainer *hbc = memnew( HBoxContainer ); hbc->add_constant_override("separation",0); if (graph->is_slot_connected(type, p_id, 1)) { hbc->add_child(make_label("xf", Variant::TRANSFORM)); } else { hbc->add_child(make_editor(String("xf: edit..."),gn,p_id,1,Variant::TRANSFORM)); } hbc->add_spacer(); Label *l = memnew(Label("out")); l->set_align(Label::ALIGN_RIGHT); hbc->add_child( l); gn->add_child(hbc); gn->set_slot(1,true,ShaderGraph::SLOT_TYPE_VEC,typecol[ShaderGraph::SLOT_TYPE_VEC],false,0,Color()); gn->set_slot(2,true,ShaderGraph::SLOT_TYPE_XFORM,typecol[ShaderGraph::SLOT_TYPE_XFORM],true,ShaderGraph::SLOT_TYPE_VEC,typecol[ShaderGraph::SLOT_TYPE_VEC]); } break; // mat4 x vec3 inverse mult (with no-translation option) case ShaderGraph::NODE_SCALAR_FUNC: { gn->set_title("ScalarFunc"); static const char* func_name[ShaderGraph::SCALAR_MAX_FUNC]={ ("Sin"), ("Cos"), ("Tan"), ("ASin"), ("ACos"), ("ATan"), ("SinH"), ("CosH"), ("TanH"), ("Log"), ("Exp"), ("Sqrt"), ("Abs"), ("Sign"), ("Floor"), ("Round"), ("Ceil"), ("Frac"), ("Satr"), ("Neg") }; OptionButton *ob = memnew( OptionButton ); for(int i=0;iadd_item(func_name[i],i); } ob->select(graph->scalar_func_node_get_function(type,p_id)); ob->connect("item_selected",this,"_scalar_func_changed",varray(p_id)); gn->add_child(ob); HBoxContainer *hbc = memnew( HBoxContainer ); if (graph->is_slot_connected(type, p_id, 0)) { hbc->add_child(make_label("in", Variant::REAL)); } else { float v = graph->default_get_value(type,p_id,0); hbc->add_child(make_editor(String("in: ")+Variant(v),gn,p_id,0,Variant::REAL,hint_spin)); } hbc->add_spacer(); hbc->add_child( memnew(Label("out"))); gn->add_child(hbc); gn->set_slot(1,true,ShaderGraph::SLOT_TYPE_SCALAR,typecol[ShaderGraph::SLOT_TYPE_SCALAR],true,ShaderGraph::SLOT_TYPE_SCALAR,typecol[ShaderGraph::SLOT_TYPE_SCALAR]); } break; // scalar function (sin: { } break; cos: { } break; etc) case ShaderGraph::NODE_VEC_FUNC: { gn->set_title("VecFunc"); static const char* func_name[ShaderGraph::VEC_MAX_FUNC]={ ("Normalize"), ("Saturate"), ("Negate"), ("Reciprocal"), ("RGB to HSV"), ("HSV to RGB"), }; OptionButton *ob = memnew( OptionButton ); for(int i=0;iadd_item(func_name[i],i); } ob->select(graph->vec_func_node_get_function(type,p_id)); ob->connect("item_selected",this,"_vec_func_changed",varray(p_id)); gn->add_child(ob); HBoxContainer *hbc = memnew( HBoxContainer ); hbc->add_constant_override("separation",0); if (graph->is_slot_connected(type, p_id, 0)) { hbc->add_child(make_label("in", Variant::VECTOR3)); } else { Vector3 v = graph->default_get_value(type,p_id,0); hbc->add_child(make_editor(String("in: ")+v,gn,p_id,0,Variant::VECTOR3)); } hbc->add_spacer(); hbc->add_child( memnew(Label("out"))); gn->add_child(hbc); gn->set_slot(1,true,ShaderGraph::SLOT_TYPE_VEC,typecol[ShaderGraph::SLOT_TYPE_VEC],true,ShaderGraph::SLOT_TYPE_VEC,typecol[ShaderGraph::SLOT_TYPE_VEC]); } break; // vector function (normalize: { } break; negate: { } break; reciprocal: { } break; rgb2hsv: { } break; hsv2rgb: { } break; etc: { } break; etc) case ShaderGraph::NODE_VEC_LEN: { gn->set_title("VecLength"); HBoxContainer *hbc = memnew( HBoxContainer ); if (graph->is_slot_connected(type, p_id, 0)) { hbc->add_child(make_label("in", Variant::VECTOR3)); } else { Vector3 v = graph->default_get_value(type,p_id,0); hbc->add_child(make_editor(String("in: ")+v,gn,p_id,0,Variant::VECTOR3)); } hbc->add_spacer(); hbc->add_child( memnew(Label("len"))); gn->add_child(hbc); gn->set_slot(0,true,ShaderGraph::SLOT_TYPE_VEC,typecol[ShaderGraph::SLOT_TYPE_VEC],true,ShaderGraph::SLOT_TYPE_SCALAR,typecol[ShaderGraph::SLOT_TYPE_SCALAR]); } break; // vec3 length case ShaderGraph::NODE_DOT_PROD: { gn->set_title("DotProduct"); HBoxContainer *hbc = memnew( HBoxContainer ); hbc->add_constant_override("separation",0); if (graph->is_slot_connected(type, p_id, 0)) { hbc->add_child(make_label("a", Variant::VECTOR3)); } else { Vector3 v = graph->default_get_value(type,p_id,0); hbc->add_child(make_editor(String("a: ")+v,gn,p_id,0,Variant::VECTOR3)); } hbc->add_spacer(); hbc->add_child( memnew(Label("dp"))); gn->add_child(hbc); if (graph->is_slot_connected(type, p_id, 1)) { gn->add_child(make_label("b", Variant::VECTOR3)); } else { Vector3 v = graph->default_get_value(type,p_id,1); gn->add_child(make_editor(String("b: ")+v,gn,p_id,1,Variant::VECTOR3)); } gn->set_slot(0,true,ShaderGraph::SLOT_TYPE_VEC,typecol[ShaderGraph::SLOT_TYPE_VEC],true,ShaderGraph::SLOT_TYPE_SCALAR,typecol[ShaderGraph::SLOT_TYPE_SCALAR]); gn->set_slot(1,true,ShaderGraph::SLOT_TYPE_VEC,typecol[ShaderGraph::SLOT_TYPE_VEC],false,0,Color()); } break; // vec3 . vec3 (dot product -> scalar output) case ShaderGraph::NODE_VEC_TO_SCALAR: { gn->set_title("Vec2Scalar"); HBoxContainer *hbc = memnew( HBoxContainer ); hbc->add_constant_override("separation",0); if (graph->is_slot_connected(type, p_id, 0)) { hbc->add_child(make_label("vec", Variant::VECTOR3)); } else { Vector3 v = graph->default_get_value(type,p_id,0); hbc->add_child(make_editor(String("vec: ")+v,gn,p_id,0,Variant::VECTOR3)); } hbc->add_spacer(); Label *l=memnew(Label("x")); l->set_align(Label::ALIGN_RIGHT); hbc->add_child( l); gn->add_child(hbc); l=memnew(Label("y")); l->set_align(Label::ALIGN_RIGHT); gn->add_child( l ); l=memnew(Label("z")); l->set_align(Label::ALIGN_RIGHT); gn->add_child( l); gn->set_slot(0,true,ShaderGraph::SLOT_TYPE_VEC,typecol[ShaderGraph::SLOT_TYPE_VEC],true,ShaderGraph::SLOT_TYPE_SCALAR,typecol[ShaderGraph::SLOT_TYPE_SCALAR]); gn->set_slot(1,false,0,Color(),true,ShaderGraph::SLOT_TYPE_SCALAR,typecol[ShaderGraph::SLOT_TYPE_SCALAR]); gn->set_slot(2,false,0,Color(),true,ShaderGraph::SLOT_TYPE_SCALAR,typecol[ShaderGraph::SLOT_TYPE_SCALAR]); } break; // 1 vec3 input: { } break; 3 scalar outputs case ShaderGraph::NODE_SCALAR_TO_VEC: { gn->set_title("Scalar2Vec"); HBoxContainer *hbc = memnew( HBoxContainer ); if (graph->is_slot_connected(type, p_id, 0)) { hbc->add_child(make_label("x", Variant::REAL)); } else { float v = graph->default_get_value(type,p_id,0); hbc->add_child(make_editor(String("x: ")+Variant(v),gn,p_id,0,Variant::REAL)); } hbc->add_spacer(); hbc->add_child( memnew(Label("vec"))); gn->add_child(hbc); if (graph->is_slot_connected(type, p_id, 1)) { gn->add_child(make_label("y", Variant::REAL)); } else { float v = graph->default_get_value(type,p_id,1); gn->add_child(make_editor(String("y: ")+Variant(v),gn,p_id,1,Variant::REAL)); } if (graph->is_slot_connected(type, p_id, 2)) { gn->add_child(make_label("in", Variant::REAL)); } else { float v = graph->default_get_value(type,p_id,2); gn->add_child(make_editor(String("in: ")+Variant(v),gn,p_id,2,Variant::REAL)); } gn->set_slot(0,true,ShaderGraph::SLOT_TYPE_SCALAR,typecol[ShaderGraph::SLOT_TYPE_SCALAR],true,ShaderGraph::SLOT_TYPE_VEC,typecol[ShaderGraph::SLOT_TYPE_VEC]); gn->set_slot(1,true,ShaderGraph::SLOT_TYPE_SCALAR,typecol[ShaderGraph::SLOT_TYPE_SCALAR],false,0,Color()); gn->set_slot(2,true,ShaderGraph::SLOT_TYPE_SCALAR,typecol[ShaderGraph::SLOT_TYPE_SCALAR],false,0,Color()); } break; // 3 scalar input: { } break; 1 vec3 output case ShaderGraph::NODE_VEC_TO_XFORM: { gn->set_title("Vec2XForm"); HBoxContainer *hbc = memnew( HBoxContainer ); hbc->add_constant_override("separation",0); if (graph->is_slot_connected(type, p_id, 0)) { hbc->add_child(make_label("x", Variant::VECTOR3)); } else { Vector3 v = graph->default_get_value(type,p_id,0); hbc->add_child(make_editor(String("x: ")+v,gn,p_id,0,Variant::VECTOR3)); } hbc->add_spacer(); hbc->add_child( memnew(Label("xf"))); gn->add_child(hbc); if (graph->is_slot_connected(type, p_id, 1)) { gn->add_child(make_label("y", Variant::VECTOR3)); } else { Vector3 v = graph->default_get_value(type,p_id,1); gn->add_child(make_editor(String("y: ")+v,gn,p_id,1,Variant::VECTOR3)); } if (graph->is_slot_connected(type, p_id, 2)) { gn->add_child(make_label("z", Variant::VECTOR3)); } else { Vector3 v = graph->default_get_value(type,p_id,2); gn->add_child(make_editor(String("z: ")+v,gn,p_id,2,Variant::VECTOR3)); } if (graph->is_slot_connected(type, p_id, 3)) { gn->add_child(make_label("ofs", Variant::VECTOR3)); } else { Vector3 v = graph->default_get_value(type,p_id,3); gn->add_child(make_editor(String("ofs: ")+v,gn,p_id,3,Variant::VECTOR3)); } gn->set_slot(0,true,ShaderGraph::SLOT_TYPE_VEC,typecol[ShaderGraph::SLOT_TYPE_VEC],true,ShaderGraph::SLOT_TYPE_XFORM,typecol[ShaderGraph::SLOT_TYPE_XFORM]); gn->set_slot(1,true,ShaderGraph::SLOT_TYPE_VEC,typecol[ShaderGraph::SLOT_TYPE_VEC],false,0,Color()); gn->set_slot(2,true,ShaderGraph::SLOT_TYPE_VEC,typecol[ShaderGraph::SLOT_TYPE_VEC],false,0,Color()); gn->set_slot(3,true,ShaderGraph::SLOT_TYPE_VEC,typecol[ShaderGraph::SLOT_TYPE_VEC],false,0,Color()); } break; // 3 vec input: { } break; 1 xform output case ShaderGraph::NODE_XFORM_TO_VEC: { gn->set_title("XForm2Vec"); HBoxContainer *hbc = memnew( HBoxContainer ); hbc->add_constant_override("separation",0); if (graph->is_slot_connected(type, p_id, 0)) { hbc->add_child(make_label("fx", Variant::TRANSFORM)); } else { hbc->add_child(make_editor(String("fx: edit..."),gn,p_id,0,Variant::TRANSFORM)); } hbc->add_spacer(); Label *l=memnew(Label("x")); l->set_align(Label::ALIGN_RIGHT); hbc->add_child( l); gn->add_child(hbc); l=memnew(Label("y")); l->set_align(Label::ALIGN_RIGHT); gn->add_child( l ); l=memnew(Label("z")); l->set_align(Label::ALIGN_RIGHT); gn->add_child( l); l=memnew(Label("ofs")); l->set_align(Label::ALIGN_RIGHT); gn->add_child( l); gn->set_slot(0,true,ShaderGraph::SLOT_TYPE_XFORM,typecol[ShaderGraph::SLOT_TYPE_XFORM],true,ShaderGraph::SLOT_TYPE_VEC,typecol[ShaderGraph::SLOT_TYPE_VEC]); gn->set_slot(1,false,0,Color(),true,ShaderGraph::SLOT_TYPE_VEC,typecol[ShaderGraph::SLOT_TYPE_VEC]); gn->set_slot(2,false,0,Color(),true,ShaderGraph::SLOT_TYPE_VEC,typecol[ShaderGraph::SLOT_TYPE_VEC]); gn->set_slot(3,false,0,Color(),true,ShaderGraph::SLOT_TYPE_VEC,typecol[ShaderGraph::SLOT_TYPE_VEC]); } break; // 3 vec input: { } break; 1 xform output case ShaderGraph::NODE_SCALAR_INTERP: { gn->set_title("ScalarInterp"); HBoxContainer *hbc = memnew( HBoxContainer ); hbc->add_constant_override("separation",0); if (graph->is_slot_connected(type, p_id, 0)) { hbc->add_child(make_label("a", Variant::REAL)); } else { float v = graph->default_get_value(type,p_id,0); hbc->add_child(make_editor(String("a: ")+Variant(v),gn,p_id,0,Variant::REAL,hint_spin)); } hbc->add_spacer(); hbc->add_child( memnew(Label("interp"))); gn->add_child(hbc); if (graph->is_slot_connected(type, p_id, 1)) { gn->add_child(make_label("b", Variant::REAL)); } else { float v = graph->default_get_value(type,p_id,1); gn->add_child(make_editor(String("b: ")+Variant(v),gn,p_id,1,Variant::REAL,hint_spin)); } if (graph->is_slot_connected(type, p_id, 2)) { gn->add_child(make_label("c", Variant::REAL)); } else { float v = graph->default_get_value(type,p_id,2); gn->add_child(make_editor(String("c: ")+Variant(v),gn,p_id,2,Variant::REAL,hint_slider)); } gn->set_slot(0,true,ShaderGraph::SLOT_TYPE_SCALAR,typecol[ShaderGraph::SLOT_TYPE_SCALAR],true,ShaderGraph::SLOT_TYPE_SCALAR,typecol[ShaderGraph::SLOT_TYPE_SCALAR]); gn->set_slot(1,true,ShaderGraph::SLOT_TYPE_SCALAR,typecol[ShaderGraph::SLOT_TYPE_SCALAR],false,0,Color()); gn->set_slot(2,true,ShaderGraph::SLOT_TYPE_SCALAR,typecol[ShaderGraph::SLOT_TYPE_SCALAR],false,0,Color()); } break; // scalar interpolation (with optional curve) case ShaderGraph::NODE_VEC_INTERP: { gn->set_title("VecInterp"); HBoxContainer *hbc = memnew( HBoxContainer ); if (graph->is_slot_connected(type, p_id, 0)) { hbc->add_child(make_label("a", Variant::VECTOR3)); } else { Vector3 v = graph->default_get_value(type,p_id,0); hbc->add_child(make_editor(String("a: ")+v,gn,p_id,0,Variant::VECTOR3)); } hbc->add_spacer(); hbc->add_child( memnew(Label("interp"))); gn->add_child(hbc); if (graph->is_slot_connected(type, p_id, 1)) { gn->add_child(make_label("b", Variant::VECTOR3)); } else { Vector3 v = graph->default_get_value(type,p_id,1); gn->add_child(make_editor(String("b: ")+v,gn,p_id,1,Variant::VECTOR3)); } if (graph->is_slot_connected(type, p_id, 2)) { gn->add_child(make_label("c", Variant::REAL)); } else { float v = graph->default_get_value(type,p_id,2); gn->add_child(make_editor(String("c: ")+Variant(v),gn,p_id,2,Variant::REAL,hint_slider)); } gn->set_slot(0,true,ShaderGraph::SLOT_TYPE_VEC,typecol[ShaderGraph::SLOT_TYPE_VEC],true,ShaderGraph::SLOT_TYPE_VEC,typecol[ShaderGraph::SLOT_TYPE_VEC]); gn->set_slot(1,true,ShaderGraph::SLOT_TYPE_VEC,typecol[ShaderGraph::SLOT_TYPE_VEC],false,0,Color()); gn->set_slot(2,true,ShaderGraph::SLOT_TYPE_SCALAR,typecol[ShaderGraph::SLOT_TYPE_SCALAR],false,0,Color()); } break; // vec3 interpolation (with optional curve) case ShaderGraph::NODE_COLOR_RAMP: { gn->set_title("ColorRamp"); GraphColorRampEdit * ramp = memnew( GraphColorRampEdit ); DVector offsets = graph->color_ramp_node_get_offsets(type,p_id); DVector colors = graph->color_ramp_node_get_colors(type,p_id); int oc = offsets.size(); if (oc) { DVector::Read rofs = offsets.read(); DVector::Read rcol = colors.read(); Vector ofsv; Vector colorv; for(int i=0;iset_ramp(ofsv,colorv); } ramp->connect("ramp_changed",this,"_color_ramp_changed",varray(p_id,ramp)); ramp->set_custom_minimum_size(Size2(128,1)); gn->add_child(ramp); HBoxContainer *hbc = memnew( HBoxContainer ); hbc->add_constant_override("separation",0); if (graph->is_slot_connected(type, p_id, 0)) { hbc->add_child(make_label("c", Variant::REAL)); } else { float v = graph->default_get_value(type,p_id,0); hbc->add_child(make_editor(String("c: ")+Variant(v),gn,p_id,0,Variant::REAL,hint_slider)); } hbc->add_spacer(); Label *l=memnew(Label("rgb")); l->set_align(Label::ALIGN_RIGHT); hbc->add_child( l); gn->add_child(hbc); l=memnew(Label("alpha")); l->set_align(Label::ALIGN_RIGHT); gn->add_child( l); gn->set_slot(1,true,ShaderGraph::SLOT_TYPE_SCALAR,typecol[ShaderGraph::SLOT_TYPE_SCALAR],true,ShaderGraph::SLOT_TYPE_VEC,typecol[ShaderGraph::SLOT_TYPE_VEC]); gn->set_slot(2,false,ShaderGraph::SLOT_MAX,Color(),true,ShaderGraph::SLOT_TYPE_SCALAR,typecol[ShaderGraph::SLOT_TYPE_SCALAR]); } break; // scalar interpolation (with optional curve) case ShaderGraph::NODE_CURVE_MAP: { gn->set_title("CurveMap"); GraphCurveMapEdit * map = memnew( GraphCurveMapEdit ); DVector points = graph->curve_map_node_get_points(type,p_id); int oc = points.size(); if (oc) { DVector::Read rofs = points.read(); Vector ofsv; for(int i=0;iset_points(ofsv); } map->connect("curve_changed",this,"_curve_changed",varray(p_id,map)); //map->connect("map_changed",this,"_curve_map_changed",varray(p_id,map)); map->set_custom_minimum_size(Size2(128,64)); gn->add_child(map); HBoxContainer *hbc = memnew( HBoxContainer ); hbc->add_constant_override("separation",0); if (graph->is_slot_connected(type, p_id, 0)) { hbc->add_child(make_label("c", Variant::REAL)); } else { float v = graph->default_get_value(type,p_id,0); hbc->add_child(make_editor(String("c: ")+Variant(v),gn,p_id,0,Variant::REAL,hint_slider)); } hbc->add_spacer(); Label *l=memnew(Label("cmap")); l->set_align(Label::ALIGN_RIGHT); hbc->add_child( l); gn->add_child(hbc); gn->set_slot(1,true,ShaderGraph::SLOT_TYPE_SCALAR,typecol[ShaderGraph::SLOT_TYPE_SCALAR],true,ShaderGraph::SLOT_TYPE_SCALAR,typecol[ShaderGraph::SLOT_TYPE_SCALAR]); } break; // scalar interpolation (with optional curve) case ShaderGraph::NODE_SCALAR_INPUT: { gn->set_title("ScalarUniform"); LineEdit *le = memnew( LineEdit ); gn->add_child(le); le->set_text(graph->input_node_get_name(type,p_id)); le->connect("text_entered",this,"_input_name_changed",varray(p_id,le)); SpinBox *sb = memnew( SpinBox ); sb->set_min(-100000); sb->set_max(100000); sb->set_step(0.001); sb->set_val(graph->scalar_input_node_get_value(type,p_id)); sb->connect("value_changed",this,"_scalar_input_changed",varray(p_id)); gn->add_child(sb); gn->set_slot(1,false,0,Color(),true,ShaderGraph::SLOT_TYPE_SCALAR,typecol[ShaderGraph::SLOT_TYPE_SCALAR]); } break; // scalar uniform (assignable in material) case ShaderGraph::NODE_VEC_INPUT: { gn->set_title("VectorUniform"); LineEdit *le = memnew( LineEdit ); gn->add_child(le); le->set_text(graph->input_node_get_name(type,p_id)); le->connect("text_entered",this,"_input_name_changed",varray(p_id,le)); Array v3p(true); for(int i=0;i<3;i++) { HBoxContainer *hbc = memnew( HBoxContainer ); Label *l = memnew( Label ); l->set_text(String::chr('X'+i)); hbc->add_child(l); SpinBox *sb = memnew( SpinBox ); sb->set_h_size_flags(Control::SIZE_EXPAND_FILL); sb->set_min(-100000); sb->set_max(100000); sb->set_step(0.001); sb->set_val(graph->vec_input_node_get_value(type,p_id)[i]); sb->connect("value_changed",this,"_vec_input_changed",varray(p_id,v3p)); v3p.push_back(sb); hbc->add_child(sb); gn->add_child(hbc); } gn->set_slot(1,false,0,Color(),true,ShaderGraph::SLOT_TYPE_VEC,typecol[ShaderGraph::SLOT_TYPE_VEC]); } break; // vec3 uniform (assignable in material) case ShaderGraph::NODE_RGB_INPUT: { gn->set_title("ColorUniform"); LineEdit *le = memnew( LineEdit ); gn->add_child(le); le->set_text(graph->input_node_get_name(type,p_id)); le->connect("text_entered",this,"_input_name_changed",varray(p_id,le)); ColorPickerButton *cpb = memnew( ColorPickerButton ); cpb->set_color(graph->rgb_input_node_get_value(type,p_id)); cpb->connect("color_changed",this,"_rgb_input_changed",varray(p_id)); gn->add_child(cpb); Label *l = memnew( Label ); l->set_text("RGB"); l->set_align(Label::ALIGN_RIGHT); gn->add_child(l); l = memnew( Label ); l->set_text("Alpha"); l->set_align(Label::ALIGN_RIGHT); gn->add_child(l); gn->set_slot(2,false,0,Color(),true,ShaderGraph::SLOT_TYPE_VEC,typecol[ShaderGraph::SLOT_TYPE_VEC]); gn->set_slot(3,false,0,Color(),true,ShaderGraph::SLOT_TYPE_SCALAR,typecol[ShaderGraph::SLOT_TYPE_SCALAR]); } break; // color uniform (assignable in material) case ShaderGraph::NODE_XFORM_INPUT: { gn->set_title("XFUniform"); LineEdit *le = memnew( LineEdit ); gn->add_child(le); le->set_text(graph->input_node_get_name(type,p_id)); le->connect("text_entered",this,"_input_name_changed",varray(p_id,le)); ToolButton *edit = memnew( ToolButton ); edit->set_text("edit.."); edit->connect("pressed",this,"_xform_input_changed",varray(p_id,edit)); gn->add_child(edit); gn->set_slot(1,false,0,Color(),true,ShaderGraph::SLOT_TYPE_XFORM,typecol[ShaderGraph::SLOT_TYPE_XFORM]); } break; // mat4 uniform (assignable in material) case ShaderGraph::NODE_TEXTURE_INPUT: { gn->set_title("TexUniform"); LineEdit *le = memnew( LineEdit ); gn->add_child(le); le->set_text(graph->input_node_get_name(type,p_id)); le->connect("text_entered",this,"_input_name_changed",varray(p_id,le)); TextureFrame *tex = memnew( TextureFrame ); tex->set_expand(true); tex->set_custom_minimum_size(Size2(80,80)); tex->set_drag_forwarding(this); gn->add_child(tex); tex->set_ignore_mouse(false); tex->set_texture(graph->texture_input_node_get_value(type,p_id)); ToolButton *edit = memnew( ToolButton ); edit->set_text("edit.."); edit->connect("pressed",this,"_tex_edited",varray(p_id,edit)); gn->add_child(edit); HBoxContainer *hbc = memnew( HBoxContainer ); hbc->add_constant_override("separation",0); if (graph->is_slot_connected(type, p_id, 0)) { hbc->add_child(make_label("UV", Variant::VECTOR3)); } else { Vector3 v = graph->default_get_value(type,p_id,0); hbc->add_child(make_editor(String("UV: ")+v,gn,p_id,0,Variant::VECTOR3)); } hbc->add_spacer(); Label *l=memnew(Label("RGB")); l->set_align(Label::ALIGN_RIGHT); hbc->add_child(l); gn->add_child(hbc); l = memnew( Label ); l->set_text("Alpha"); l->set_align(Label::ALIGN_RIGHT); gn->add_child(l); gn->set_slot(3,true,ShaderGraph::SLOT_TYPE_VEC,typecol[ShaderGraph::SLOT_TYPE_VEC],true,ShaderGraph::SLOT_TYPE_VEC,typecol[ShaderGraph::SLOT_TYPE_VEC]); gn->set_slot(4,false,0,Color(),true,ShaderGraph::SLOT_TYPE_SCALAR,typecol[ShaderGraph::SLOT_TYPE_SCALAR]); } break; // texture input (assignable in material) case ShaderGraph::NODE_CUBEMAP_INPUT: { gn->set_title("TexUniform"); LineEdit *le = memnew( LineEdit ); gn->add_child(le); le->set_text(graph->input_node_get_name(type,p_id)); le->connect("text_entered",this,"_input_name_changed",varray(p_id,le)); ToolButton *edit = memnew( ToolButton ); edit->set_text("edit.."); edit->connect("pressed",this,"_cube_edited",varray(p_id,edit)); gn->add_child(edit); HBoxContainer *hbc = memnew( HBoxContainer ); hbc->add_constant_override("separation",0); if (graph->is_slot_connected(type, p_id, 0)) { hbc->add_child(make_label("UV", Variant::VECTOR3)); } else { Vector3 v = graph->default_get_value(type,p_id,0); hbc->add_child(make_editor(String("UV: ")+v,gn,p_id,0,Variant::VECTOR3)); } hbc->add_spacer(); Label *l=memnew(Label("RGB")); l->set_align(Label::ALIGN_RIGHT); hbc->add_child(l); gn->add_child(hbc); l = memnew( Label ); l->set_text("Alpha"); l->set_align(Label::ALIGN_RIGHT); gn->add_child(l); gn->set_slot(2,true,ShaderGraph::SLOT_TYPE_VEC,typecol[ShaderGraph::SLOT_TYPE_VEC],true,ShaderGraph::SLOT_TYPE_VEC,typecol[ShaderGraph::SLOT_TYPE_VEC]); gn->set_slot(3,false,0,Color(),true,ShaderGraph::SLOT_TYPE_SCALAR,typecol[ShaderGraph::SLOT_TYPE_SCALAR]); } break; // cubemap input (assignable in material) case ShaderGraph::NODE_DEFAULT_TEXTURE: { gn->set_title("CanvasItemTex"); HBoxContainer *hbc = memnew( HBoxContainer ); hbc->add_constant_override("separation",0); if (graph->is_slot_connected(type, p_id, 0)) { hbc->add_child(make_label("UV", Variant::VECTOR3)); } else { Vector3 v = graph->default_get_value(type,p_id,0); hbc->add_child(make_editor(String("UV: ")+v,gn,p_id,0,Variant::VECTOR3)); } hbc->add_spacer(); Label *l=memnew(Label("RGB")); l->set_align(Label::ALIGN_RIGHT); hbc->add_child(l); gn->add_child(hbc); l = memnew( Label ); l->set_text("Alpha"); l->set_align(Label::ALIGN_RIGHT); gn->add_child(l); gn->set_slot(0,true,ShaderGraph::SLOT_TYPE_VEC,typecol[ShaderGraph::SLOT_TYPE_VEC],true,ShaderGraph::SLOT_TYPE_VEC,typecol[ShaderGraph::SLOT_TYPE_VEC]); gn->set_slot(1,false,0,Color(),true,ShaderGraph::SLOT_TYPE_SCALAR,typecol[ShaderGraph::SLOT_TYPE_SCALAR]); } break; // screen texture sampler (takes UV) (only usable in fragment case Shader) case ShaderGraph::NODE_OUTPUT: { gn->set_title("Output"); gn->set_show_close_button(false); List si; ShaderGraph::get_input_output_node_slot_info(graph->get_mode(),type,&si); Array colors; colors.push_back("Color"); colors.push_back("LightColor"); colors.push_back("Light"); colors.push_back("Diffuse"); colors.push_back("Specular"); colors.push_back("Emmision"); Array reals; reals.push_back("Alpha"); reals.push_back("DiffuseAlpha"); reals.push_back("NormalMapDepth"); reals.push_back("SpecExp"); reals.push_back("Glow"); reals.push_back("ShadeParam"); reals.push_back("SpecularExp"); reals.push_back("LightAlpha"); reals.push_back("PointSize"); reals.push_back("Discard"); int idx=0; for (List::Element *E=si.front();E;E=E->next()) { ShaderGraph::SlotInfo& s=E->get(); if (s.dir==ShaderGraph::SLOT_OUT) { Variant::Type v; if (colors.find(s.name)>=0) v=Variant::COLOR; else if (reals.find(s.name)>=0) v=Variant::REAL; else v=Variant::VECTOR3; gn->add_child(make_label(s.name, v)); gn->set_slot(idx,true,s.type,typecol[s.type],false,0,Color()); idx++; } } } break; // output (case Shader type dependent) case ShaderGraph::NODE_COMMENT: { gn->set_title("Comment"); TextEdit *te = memnew(TextEdit); te->set_custom_minimum_size(Size2(100,100)); gn->add_child(te); te->set_text(graph->comment_node_get_text(type,p_id)); te->connect("text_changed",this,"_comment_edited",varray(p_id,te)); } break; // comment } gn->connect("dragged",this,"_node_moved",varray(p_id)); gn->connect("close_request",this,"_node_removed",varray(p_id),CONNECT_DEFERRED); graph_edit->add_child(gn); node_map[p_id]=gn; gn->set_offset(graph->node_get_pos(type,p_id)); } void ShaderGraphView::_update_graph() { if (block_update) return; for (Map::Element *E=node_map.front();E;E=E->next()) { memdelete(E->get()); } node_map.clear(); if (!graph.is_valid()) return; List nl; graph->get_node_list(type,&nl); for(List::Element *E=nl.front();E;E=E->next()) { _create_node(E->get()); } graph_edit->clear_connections(); List connections; graph->get_node_connections(type,&connections); for(List::Element *E=connections.front();E;E=E->next()) { ERR_CONTINUE(!node_map.has(E->get().src_id) || !node_map.has(E->get().dst_id)); graph_edit->connect_node(node_map[E->get().src_id]->get_name(),E->get().src_slot,node_map[E->get().dst_id]->get_name(),E->get().dst_slot); } } void ShaderGraphView::_sg_updated() { if (!graph.is_valid()) return; switch(graph->get_graph_error(type)) { case ShaderGraph::GRAPH_OK: status->set_text(""); break; case ShaderGraph::GRAPH_ERROR_CYCLIC: status->set_text(TTR("Error: Cyclic Connection Link")); break; case ShaderGraph::GRAPH_ERROR_MISSING_CONNECTIONS: status->set_text(TTR("Error: Missing Input Connections")); break; } } Variant ShaderGraphView::get_drag_data_fw(const Point2 &p_point, Control *p_from) { TextureFrame* frame = p_from->cast_to(); if (!frame) return Variant(); if (!frame->get_texture().is_valid()) return Variant(); RES res = frame->get_texture(); return EditorNode::get_singleton()->drag_resource(res,p_from); return Variant(); } bool ShaderGraphView::can_drop_data_fw(const Point2 &p_point, const Variant &p_data, Control *p_from) const { if (p_data.get_type() != Variant::DICTIONARY) return false; Dictionary d = p_data; if (d.has("type")){ if (d["type"] == "resource" && d.has("resource")) { Variant val = d["resource"]; if (val.get_type()==Variant::OBJECT) { RES res = val; if (res.is_valid() && res->cast_to()) return true; } } else if (d["type"] == "files" && d.has("files")) { Vector files = d["files"]; if (files.size() != 1) return false; return (ResourceLoader::get_resource_type(files[0]) == "ImageTexture"); } } return false; } void ShaderGraphView::drop_data_fw(const Point2 &p_point, const Variant &p_data, Control *p_from) { if (!can_drop_data_fw(p_point, p_data, p_from)) return; TextureFrame *frame = p_from->cast_to(); if (!frame) return; Dictionary d = p_data; Ref tex; if (d.has("type")) { if (d["type"] == "resource" && d.has("resource")){ Variant val = d["resource"]; if (val.get_type()==Variant::OBJECT) { RES res = val; if (res.is_valid()) tex = Ref(res->cast_to()); } } else if (d["type"] == "files" && d.has("files")) { Vector files = d["files"]; RES res = ResourceLoader::load(files[0]); if (res.is_valid()) tex = Ref(res->cast_to()); } } if (!tex.is_valid()) return; GraphNode *gn = frame->get_parent()->cast_to(); if (!gn) return; int id = -1; for(Map::Element *E = node_map.front();E;E=E->next()) if (E->get() == gn) { id = E->key(); break; } print_line(String::num(double(id))); if (id < 0) return; if (graph->node_get_type(type,id)==ShaderGraph::NODE_TEXTURE_INPUT) { UndoRedo *ur=EditorNode::get_singleton()->get_undo_redo(); ur->create_action(TTR("Change Texture Uniform")); ur->add_do_method(graph.ptr(),"texture_input_node_set_value",type,id,tex); ur->add_undo_method(graph.ptr(),"texture_input_node_set_value",type,id,graph->texture_input_node_get_value(type,id)); ur->add_do_method(this,"_update_graph"); ur->add_undo_method(this,"_update_graph"); ur->commit_action(); } } void ShaderGraphView::set_graph(Ref p_graph){ if (graph.is_valid()) { graph->disconnect("updated",this,"_sg_updated"); } graph=p_graph; if (graph.is_valid()) { graph->connect("updated",this,"_sg_updated"); } _update_graph(); _sg_updated(); } void ShaderGraphView::_notification(int p_what) { if (p_what==NOTIFICATION_ENTER_TREE) { ped_popup->connect("variant_changed",this,"_variant_edited"); } } void ShaderGraphView::add_node(int p_type, const Vector2 &location) { if (p_type==ShaderGraph::NODE_INPUT && graph->node_count(type, p_type)>0) return; List existing; graph->get_node_list(type,&existing); existing.sort(); int newid=1; for(List::Element *E=existing.front();E;E=E->next()) { if (!E->next() || (E->get()+1!=E->next()->get())){ newid=E->get()+1; break; } } Vector2 init_ofs = location; while(true) { bool valid=true; for(List::Element *E=existing.front();E;E=E->next()) { Vector2 pos = graph->node_get_pos(type,E->get()); if (init_ofs==pos) { init_ofs+=Vector2(20,20); valid=false; break; } } if (valid) break; } UndoRedo *ur=EditorNode::get_singleton()->get_undo_redo(); ur->create_action(TTR("Add Shader Graph Node")); ur->add_do_method(graph.ptr(),"node_add",type,p_type,newid); ur->add_do_method(graph.ptr(),"node_set_pos",type,newid,init_ofs); ur->add_undo_method(graph.ptr(),"node_remove",type,newid); ur->add_do_method(this,"_update_graph"); ur->add_undo_method(this,"_update_graph"); ur->commit_action(); } void ShaderGraphView::_bind_methods() { ObjectTypeDB::bind_method("_update_graph",&ShaderGraphView::_update_graph); ObjectTypeDB::bind_method("_begin_node_move", &ShaderGraphView::_begin_node_move); ObjectTypeDB::bind_method("_node_moved",&ShaderGraphView::_node_moved); ObjectTypeDB::bind_method("_end_node_move", &ShaderGraphView::_end_node_move); ObjectTypeDB::bind_method("_move_node",&ShaderGraphView::_move_node); ObjectTypeDB::bind_method("_node_removed",&ShaderGraphView::_node_removed); ObjectTypeDB::bind_method("_connection_request",&ShaderGraphView::_connection_request); ObjectTypeDB::bind_method("_disconnection_request",&ShaderGraphView::_disconnection_request); ObjectTypeDB::bind_method("_duplicate_nodes_request", &ShaderGraphView::_duplicate_nodes_request); ObjectTypeDB::bind_method("_duplicate_nodes", &ShaderGraphView::_duplicate_nodes); ObjectTypeDB::bind_method("_delete_nodes_request", &ShaderGraphView::_delete_nodes_request); ObjectTypeDB::bind_method("_default_changed",&ShaderGraphView::_default_changed); ObjectTypeDB::bind_method("_scalar_const_changed",&ShaderGraphView::_scalar_const_changed); ObjectTypeDB::bind_method("_vec_const_changed",&ShaderGraphView::_vec_const_changed); ObjectTypeDB::bind_method("_rgb_const_changed",&ShaderGraphView::_rgb_const_changed); ObjectTypeDB::bind_method("_xform_const_changed",&ShaderGraphView::_xform_const_changed); ObjectTypeDB::bind_method("_scalar_op_changed",&ShaderGraphView::_scalar_op_changed); ObjectTypeDB::bind_method("_vec_op_changed",&ShaderGraphView::_vec_op_changed); ObjectTypeDB::bind_method("_vec_scalar_op_changed",&ShaderGraphView::_vec_scalar_op_changed); ObjectTypeDB::bind_method("_rgb_op_changed",&ShaderGraphView::_rgb_op_changed); ObjectTypeDB::bind_method("_xform_inv_rev_changed",&ShaderGraphView::_xform_inv_rev_changed); ObjectTypeDB::bind_method("_scalar_func_changed",&ShaderGraphView::_scalar_func_changed); ObjectTypeDB::bind_method("_vec_func_changed",&ShaderGraphView::_vec_func_changed); ObjectTypeDB::bind_method("_scalar_input_changed",&ShaderGraphView::_scalar_input_changed); ObjectTypeDB::bind_method("_vec_input_changed",&ShaderGraphView::_vec_input_changed); ObjectTypeDB::bind_method("_xform_input_changed",&ShaderGraphView::_xform_input_changed); ObjectTypeDB::bind_method("_rgb_input_changed",&ShaderGraphView::_rgb_input_changed); ObjectTypeDB::bind_method("_tex_input_change",&ShaderGraphView::_tex_input_change); ObjectTypeDB::bind_method("_cube_input_change",&ShaderGraphView::_cube_input_change); ObjectTypeDB::bind_method("_input_name_changed",&ShaderGraphView::_input_name_changed); ObjectTypeDB::bind_method("_tex_edited",&ShaderGraphView::_tex_edited); ObjectTypeDB::bind_method("_variant_edited",&ShaderGraphView::_variant_edited); ObjectTypeDB::bind_method("_cube_edited",&ShaderGraphView::_cube_edited); ObjectTypeDB::bind_method("_comment_edited",&ShaderGraphView::_comment_edited); ObjectTypeDB::bind_method("_color_ramp_changed",&ShaderGraphView::_color_ramp_changed); ObjectTypeDB::bind_method("_curve_changed",&ShaderGraphView::_curve_changed); ObjectTypeDB::bind_method(_MD("get_drag_data_fw"), &ShaderGraphView::get_drag_data_fw); ObjectTypeDB::bind_method(_MD("can_drop_data_fw"), &ShaderGraphView::can_drop_data_fw); ObjectTypeDB::bind_method(_MD("drop_data_fw"), &ShaderGraphView::drop_data_fw); ObjectTypeDB::bind_method("_sg_updated",&ShaderGraphView::_sg_updated); } ShaderGraphView::ShaderGraphView(ShaderGraph::ShaderType p_type) { type=p_type; graph_edit = memnew( GraphEdit ); block_update=false; ped_popup = memnew( CustomPropertyEditor ); graph_edit->add_child(ped_popup); status = memnew( Label ); graph_edit->get_top_layer()->add_child(status); graph_edit->connect("_begin_node_move", this, "_begin_node_move"); graph_edit->connect("_end_node_move", this, "_end_node_move"); status->set_pos(Vector2(5,5)); status->add_color_override("font_color_shadow",Color(0,0,0)); status->add_color_override("font_color",Color(1,0.4,0.3)); status->add_constant_override("shadow_as_outline",1); status->add_constant_override("shadow_offset_x",2); status->add_constant_override("shadow_offset_y",2); status->set_text(""); } //////////////edit////////////// void ShaderGraphEditor::edit(Ref p_shader) { for(int i=0;iset_graph(p_shader); } } void ShaderGraphEditor::_add_node(int p_type) { ShaderGraph::ShaderType shader_type=ShaderGraph::ShaderType(tabs->get_current_tab()); graph_edits[shader_type]->add_node(p_type, next_location); } void ShaderGraphEditor::_popup_requested(const Vector2 &p_position) { Vector2 scroll_ofs=graph_edits[tabs->get_current_tab()]->get_graph_edit()->get_scroll_ofs(); next_location = get_local_mouse_pos() + scroll_ofs; popup->set_global_pos(p_position); popup->set_size( Size2( 200, 0) ); popup->popup(); popup->call_deferred("grab_click_focus"); popup->set_invalidate_click_until_motion(); } void ShaderGraphEditor::_notification(int p_what) { if (p_what==NOTIFICATION_ENTER_TREE) { for(int i=0;iadd_icon_item(get_icon(ic,"EditorIcons"),v,i); if (addsep) popup->add_separator(); } popup->connect("item_pressed",this,"_add_node"); } } void ShaderGraphEditor::_bind_methods() { ObjectTypeDB::bind_method("_add_node",&ShaderGraphEditor::_add_node); ObjectTypeDB::bind_method("_popup_requested",&ShaderGraphEditor::_popup_requested); } const char* ShaderGraphEditor::node_names[ShaderGraph::NODE_TYPE_MAX]={ ("GraphInput:Input"), // all inputs (shader type dependent) ("GraphScalar:Scalar Constant"), //scalar constant ("GraphVector:Vector Constant"), //vec3 constant ("GraphRgb:RGB Constant"), //rgb constant (shows a color picker instead) ("GraphXform:XForm Constant"), // 4x4 matrix constant ("GraphTime:Time:"), // time in seconds ("GraphTexscreen:Screen Sample"), // screen texture sampler (takes uv) (only usable in fragment shader) ("GraphScalarOp:Scalar Operator"), // scalar vs scalar op (mul", add", div", etc) ("GraphVecOp:Vector Operator"), // vec3 vs vec3 op (mul",ad",div",crossprod",etc) ("GraphVecScalarOp:Scalar+Vector Operator"), // vec3 vs scalar op (mul", add", div", etc) ("GraphRgbOp:RGB Operator:"), // vec3 vs vec3 rgb op (with scalar amount)", like brighten", darken", burn", dodge", multiply", etc. ("GraphXformMult:XForm Multiply"), // mat4 x mat4 ("GraphXformVecMult:XForm+Vector Multiply"), // mat4 x vec3 mult (with no-translation option) ("GraphXformVecImult:Form+Vector InvMultiply:"), // mat4 x vec3 inverse mult (with no-translation option) ("GraphXformScalarFunc:Scalar Function"), // scalar function (sin", cos", etc) ("GraphXformVecFunc:Vector Function"), // vector function (normalize", negate", reciprocal", rgb2hsv", hsv2rgb", etc", etc) ("GraphVecLength:Vector Length"), // vec3 length ("GraphVecDp:Dot Product:"), // vec3 . vec3 (dot product -> scalar output) ("GraphVecToScalars:Vector -> Scalars"), // 1 vec3 input", 3 scalar outputs ("GraphScalarsToVec:Scalars -> Vector"), // 3 scalar input", 1 vec3 output ("GraphXformToVecs:XForm -> Vectors"), // 3 vec input", 1 xform output ("GraphVecsToXform:Vectors -> XForm:"), // 3 vec input", 1 xform output ("GraphScalarInterp:Scalar Interpolate"), // scalar interpolation (with optional curve) ("GraphVecInterp:Vector Interpolate:"), // vec3 interpolation (with optional curve) ("GraphColorRamp:Color Ramp"), // vec3 interpolation (with optional curve) ("GraphCurveMap:Curve Remap:"), // vec3 interpolation (with optional curve) ("GraphScalarUniform:Scalar Uniform"), // scalar uniform (assignable in material) ("GraphVectorUniform:Vector Uniform"), // vec3 uniform (assignable in material) ("GraphRgbUniform:RGB Uniform"), // color uniform (assignable in material) ("GraphXformUniform:XForm Uniform"), // mat4 uniform (assignable in material) ("GraphTextureUniform:Texture Uniform"), // texture input (assignable in material) ("GraphCubeUniform:CubeMap Uniform:"), // cubemap input (assignable in material) ("GraphDefaultTexture:CanvasItem Texture:"), // cubemap input (assignable in material) ("Output"), // output (shader type dependent) ("GraphComment:Comment"), // comment }; ShaderGraphEditor::ShaderGraphEditor(bool p_2d) { _2d=p_2d; popup = memnew( PopupMenu ); add_child(popup); tabs = memnew(TabContainer); tabs->set_v_size_flags(SIZE_EXPAND_FILL); add_child(tabs); const char* sname[ShaderGraph::SHADER_TYPE_MAX]={ "Vertex", "Fragment", "Light" }; for(int i=0;iget_graph_edit()->set_name(sname[i]); tabs->add_child(graph_edits[i]->get_graph_edit()); graph_edits[i]->get_graph_edit()->connect("connection_request",graph_edits[i],"_connection_request"); graph_edits[i]->get_graph_edit()->connect("disconnection_request",graph_edits[i],"_disconnection_request"); graph_edits[i]->get_graph_edit()->connect("duplicate_nodes_request", graph_edits[i], "_duplicate_nodes_request"); graph_edits[i]->get_graph_edit()->connect("popup_request",this,"_popup_requested"); graph_edits[i]->get_graph_edit()->connect("delete_nodes_request",graph_edits[i],"_delete_nodes_request"); graph_edits[i]->get_graph_edit()->set_right_disconnects(true); } tabs->set_current_tab(1); set_custom_minimum_size(Size2(100,300)); } void ShaderGraphEditorPlugin::edit(Object *p_object) { shader_editor->edit(p_object->cast_to()); } bool ShaderGraphEditorPlugin::handles(Object *p_object) const { ShaderGraph *shader=p_object->cast_to(); if (!shader) return false; if (_2d) return shader->get_mode()==Shader::MODE_CANVAS_ITEM; else return shader->get_mode()==Shader::MODE_MATERIAL; } void ShaderGraphEditorPlugin::make_visible(bool p_visible) { if (p_visible) { shader_editor->show(); } else { shader_editor->hide(); } } ShaderGraphEditorPlugin::ShaderGraphEditorPlugin(EditorNode *p_node, bool p_2d) { _2d=p_2d; editor=p_node; shader_editor = memnew( ShaderGraphEditor(p_2d) ); shader_editor->hide(); if (p_2d) CanvasItemEditor::get_singleton()->get_bottom_split()->add_child(shader_editor); else SpatialEditor::get_singleton()->get_shader_split()->add_child(shader_editor); // editor->get_viewport()->add_child(shader_editor); // shader_editor->set_area_as_parent_rect(); // shader_editor->hide(); } ShaderGraphEditorPlugin::~ShaderGraphEditorPlugin() { }