Improved toolbar capabilities

Author: alelievr

Assets/com.alelievr.NodeGraphProcessor/Editor/Views/ToolbarView.cs

@@ -19,6 +19,9 @@ protected enum ElementType
 			Button,
 			Toggle,
 			DropDownButton,
+			Separator,
+			Custom,
+			FlexibleSpace,
 		}
 
 		protected class ToolbarButtonData
@@ -29,6 +32,8 @@ protected class ToolbarButtonData
 			public bool				visible = true;
 			public Action			buttonCallback;
 			public Action< bool >	toggleCallback;
+			public int				size;
+			public Action			customDrawFunction;
 		}
 
 		List< ToolbarButtonData >	leftButtonDatas = new List< ToolbarButtonData >();
@@ -66,6 +71,32 @@ protected ToolbarButtonData AddButton(GUIContent content, Action callback, bool
 			return data;
 		}
 
+		protected void AddSeparator(int sizeInPixels = 10, bool left = true)
+		{
+			var data = new ToolbarButtonData{
+				type = ElementType.Separator,
+				size = sizeInPixels,
+			};
+			((left) ? leftButtonDatas : rightButtonDatas).Add(data);
+		}
+
+		protected void AddCustom(Action imguiDrawFunction, bool left = true)
+		{
+			if (imguiDrawFunction == null)
+				throw new ArgumentException("imguiDrawFunction can't be null");
+
+			var data = new ToolbarButtonData{
+				type = ElementType.Custom,
+				customDrawFunction = imguiDrawFunction,
+			};
+			((left) ? leftButtonDatas : rightButtonDatas).Add(data);
+		}
+
+		protected void AddFlexibleSpace(bool left = true)
+		{
+			((left) ? leftButtonDatas : rightButtonDatas).Add(new ToolbarButtonData{ type = ElementType.FlexibleSpace });
+		}
+
 		protected ToolbarButtonData AddToggle(string name, bool defaultValue, Action< bool > callback, bool left = true)
 			=> AddToggle(new GUIContent(name), defaultValue, callback, left);
 
@@ -112,7 +143,7 @@ protected void RemoveButton(string name, bool left)
 		protected void HideButton(string name)
 		{
 			leftButtonDatas.Concat(rightButtonDatas).All(b => {
-				if (b.content.text == name)
+				if (b?.content?.text == name)
 					b.visible = false;
 				return true;
 			});
@@ -125,7 +156,7 @@ protected void HideButton(string name)
 		protected void ShowButton(string name)
 		{
 			leftButtonDatas.Concat(rightButtonDatas).All(b => {
-				if (b.content.text == name)
+				if (b?.content?.text == name)
 					b.visible = true;
 				return true;
 			});
@@ -174,6 +205,16 @@ void DrawImGUIButtonList(List< ToolbarButtonData > buttons)
 						if (EditorGUILayout.DropdownButton(button.content, FocusType.Passive, EditorStyles.toolbarDropDown))
 							button.buttonCallback();
 						break;
+					case ElementType.Separator:
+						EditorGUILayout.Separator();
+						EditorGUILayout.Space(button.size);
+						break;
+					case ElementType.Custom:
+						button.customDrawFunction();
+						break;
+					case ElementType.FlexibleSpace:
+						GUILayout.FlexibleSpace();
+						break;
 				}
 			}
 		}

CHANGELOG.md

@@ -8,6 +8,7 @@ and this project adheres to [Semantic Versioning](http://semver.org/spec/v2.0.0.
 
 ### Added
 - Renamable nodes
+- Added an API in the toolbar view to add separators and custom UI fields.
 
 ## [1.1.2]