1 files changed, 139 insertions, 10 deletions

diff --git a/modules/visual_script/visual_script_editor.cpp b/modules/visual_script/visual_script_editor.cpp
index 09ca4a3e4..3f3b2f322 100644
--- a/modules/visual_script/visual_script_editor.cpp
+++ b/modules/visual_script/visual_script_editor.cpp
@@ -302,9 +302,45 @@ public:
 };
 
 static Color _color_from_type(Variant::Type p_type) {
-
 	Color color;
-	color.set_hsv(p_type/float(Variant::VARIANT_MAX),0.7,0.7);
+	switch(p_type) {
+		case Variant::NIL: color = Color::html("69ecbd"); break;
+
+		case Variant::BOOL: color = Color::html("8da6f0"); break;
+		case Variant::INT: color = Color::html("7dc6ef"); break;
+		case Variant::REAL: color = Color::html("61daf4"); break;
+		case Variant::STRING: color = Color::html("6ba7ec"); break;
+
+		case Variant::VECTOR2: color = Color::html("bd91f1"); break;
+		case Variant::RECT2: color = Color::html("f191a5"); break;
+		case Variant::VECTOR3: color = Color::html("d67dee"); break;
+		case Variant::MATRIX32: color = Color::html("c4ec69"); break;
+		case Variant::PLANE: color = Color::html("f77070"); break;
+		case Variant::QUAT: color = Color::html("ec69a3"); break;
+		case Variant::_AABB: color = Color::html("ee7991"); break;
+		case Variant::MATRIX3: color = Color::html("e3ec69"); break;
+		case Variant::TRANSFORM: color = Color::html("ecd669"); break;
+
+		case Variant::COLOR: color = Color::html("9dff70"); break;
+		case Variant::IMAGE: color = Color::html("93f1b9"); break;
+		case Variant::NODE_PATH: color = Color::html("6993ec"); break;
+		case Variant::_RID: color = Color::html("69ec9a"); break;
+		case Variant::OBJECT: color = Color::html("79f3e8"); break;
+		case Variant::INPUT_EVENT: color = Color::html("adf18f"); break;
+		case Variant::DICTIONARY: color = Color::html("77edb1"); break;
+
+		case Variant::ARRAY: color = Color::html("e0e0e0"); break;
+		case Variant::RAW_ARRAY: color = Color::html("aaf4c8"); break;
+		case Variant::INT_ARRAY: color = Color::html("afdcf5"); break;
+		case Variant::REAL_ARRAY: color = Color::html("97e7f8"); break;
+		case Variant::STRING_ARRAY: color = Color::html("9dc4f2"); break;
+		case Variant::VECTOR2_ARRAY: color = Color::html("d1b3f5"); break;
+		case Variant::VECTOR3_ARRAY: color = Color::html("df9bf2"); break;
+		case Variant::COLOR_ARRAY: color = Color::html("e9ff97"); break;
+
+		default:
+			color.set_hsv(p_type/float(Variant::VARIANT_MAX),0.7,0.7);
+	}
 	return color;
 }
 
@@ -390,7 +426,7 @@ void VisualScriptEditor::_update_graph(int p_only_id) {
 		Control::get_icon("MiniVector2","EditorIcons"),
 		Control::get_icon("MiniRect2","EditorIcons"),
 		Control::get_icon("MiniVector3","EditorIcons"),
-		Control::get_icon("MiniMatrix2","EditorIcons"),
+		Control::get_icon("MiniMatrix32","EditorIcons"),
 		Control::get_icon("MiniPlane","EditorIcons"),
 		Control::get_icon("MiniQuat","EditorIcons"),
 		Control::get_icon("MiniAabb","EditorIcons"),
@@ -1426,6 +1462,8 @@ bool VisualScriptEditor::can_drop_data_fw(const Point2& p_point,const Variant& p
 					String(d["type"])=="visual_script_variable_drag" ||
 					String(d["type"])=="visual_script_signal_drag" ||
 					String(d["type"])=="obj_property" ||
+					String(d["type"])=="resource" ||
+					String(d["type"])=="files" ||
 					String(d["type"])=="nodes"
 				) ) {
 
@@ -1586,13 +1624,14 @@ void VisualScriptEditor::drop_data_fw(const Point2& p_point,const Variant& p_dat
 			Ref<VisualScriptFunctionCall> vnode;
 			vnode.instance();
 			vnode->set_call_mode(VisualScriptFunctionCall::CALL_MODE_SELF);
-			vnode->set_base_type(script->get_instance_base_type());
-			vnode->set_function(d["function"]);
 
 			int new_id = script->get_available_id();
 
 			undo_redo->create_action(TTR("Add Node"));
 			undo_redo->add_do_method(script.ptr(),"add_node",edited_func,new_id,vnode,ofs);
+			undo_redo->add_do_method(vnode.ptr(),"set_base_type",script->get_instance_base_type());
+			undo_redo->add_do_method(vnode.ptr(),"set_function",d["function"]);
+
 			undo_redo->add_undo_method(script.ptr(),"remove_node",edited_func,new_id);
 			undo_redo->add_do_method(this,"_update_graph");
 			undo_redo->add_undo_method(this,"_update_graph");
@@ -1634,8 +1673,90 @@ void VisualScriptEditor::drop_data_fw(const Point2& p_point,const Variant& p_dat
 			if (node) {
 				graph->set_selected(node);
 				_node_selected(node);
+			}			
+		}
+
+		if (d.has("type") && String(d["type"])=="resource") {
+
+			Vector2 ofs = graph->get_scroll_ofs() + p_point;
+			if (graph->is_using_snap()) {
+				int snap = graph->get_snap();
+				ofs = ofs.snapped(Vector2(snap,snap));
+			}
+
+			ofs/=EDSCALE;
+
+			Ref<VisualScriptPreload> prnode;
+			prnode.instance();
+			prnode->set_preload(d["resource"]);
+
+			int new_id = script->get_available_id();
+
+			undo_redo->create_action(TTR("Add Preload Node"));
+			undo_redo->add_do_method(script.ptr(),"add_node",edited_func,new_id,prnode,ofs);
+			undo_redo->add_undo_method(script.ptr(),"remove_node",edited_func,new_id);
+			undo_redo->add_do_method(this,"_update_graph");
+			undo_redo->add_undo_method(this,"_update_graph");
+			undo_redo->commit_action();
+
+			Node* node = graph->get_node(itos(new_id));
+			if (node) {
+				graph->set_selected(node);
+				_node_selected(node);
 			}
 		}
+
+		if (d.has("type") && String(d["type"])=="files") {
+
+			Vector2 ofs = graph->get_scroll_ofs() + p_point;
+			if (graph->is_using_snap()) {
+				int snap = graph->get_snap();
+				ofs = ofs.snapped(Vector2(snap,snap));
+			}
+
+			ofs/=EDSCALE;
+
+			Array files = d["files"];
+
+			List<int> new_ids;
+			int new_id = script->get_available_id();
+
+			if (files.size()) {
+				undo_redo->create_action(TTR("Add Preload Node"));
+
+				for(int i=0;i<files.size();i++) {
+
+					Ref<Resource> res = ResourceLoader::load(files[i]);
+					if (!res.is_valid())
+						continue;
+
+					Ref<VisualScriptPreload> prnode;
+					prnode.instance();
+					prnode->set_preload(res);
+
+					undo_redo->add_do_method(script.ptr(),"add_node",edited_func,new_id,prnode,ofs);
+					undo_redo->add_undo_method(script.ptr(),"remove_node",edited_func,new_id);
+					new_ids.push_back(new_id);
+					new_id++;
+					ofs+=Vector2(20,20)*EDSCALE;
+				}
+
+
+				undo_redo->add_do_method(this,"_update_graph");
+				undo_redo->add_undo_method(this,"_update_graph");
+				undo_redo->commit_action();
+			}
+
+			for(List<int>::Element *E=new_ids.front();E;E=E->next()) {
+
+				Node* node = graph->get_node(itos(E->get()));
+				if (node) {
+					graph->set_selected(node);
+					_node_selected(node);
+				}
+			}
+		}
+
 		if (d.has("type") && String(d["type"])=="nodes") {
 
 			Node* sn = _find_script_node(get_tree()->get_edited_scene_root(),get_tree()->get_edited_scene_root(),script);
@@ -1808,16 +1929,24 @@ void VisualScriptEditor::drop_data_fw(const Point2& p_point,const Variant& p_dat
 
 					Ref<VisualScriptPropertySet> pset;
 					pset.instance();
-					pset->set_call_mode(VisualScriptPropertySet::CALL_MODE_NODE_PATH);
-					pset->set_base_path(sn->get_path_to(node));
+					if (sn==node) {
+						pset->set_call_mode(VisualScriptPropertySet::CALL_MODE_SELF);
+					} else {
+						pset->set_call_mode(VisualScriptPropertySet::CALL_MODE_NODE_PATH);
+						pset->set_base_path(sn->get_path_to(node));
+					}
 
 					vnode=pset;
 				} else {
 
 					Ref<VisualScriptPropertyGet> pget;
 					pget.instance();
-					pget->set_call_mode(VisualScriptPropertyGet::CALL_MODE_NODE_PATH);
-					pget->set_base_path(sn->get_path_to(node));
+					if (sn==node) {
+						pget->set_call_mode(VisualScriptPropertyGet::CALL_MODE_SELF);
+					} else {
+						pget->set_call_mode(VisualScriptPropertyGet::CALL_MODE_NODE_PATH);
+						pget->set_base_path(sn->get_path_to(node));
+					}
 					vnode=pget;
 
 				}
@@ -2771,7 +2900,7 @@ VisualScriptEditor::VisualScriptEditor() {
 	left_vb->set_v_size_flags(SIZE_EXPAND_FILL);
 	left_vb->set_custom_minimum_size(Size2(180,1)*EDSCALE);
 
-	base_type_select = memnew( Button );	
+	base_type_select = memnew( Button );
 	left_vb->add_margin_child(TTR("Base Type:"),base_type_select);
 	base_type_select->connect("pressed",this,"_change_base_type");