mathe/Library/PackageCache/com.unity.shadergraph@14.0.8/Editor/Drawing/Views/GraphEditorView.cs

1445 lines
57 KiB
C#
Raw Normal View History

2024-09-20 20:30:10 +02:00
using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using UnityEngine;
using UnityEditor.Graphing;
using UnityEditor.Graphing.Util;
using UnityEditor.ShaderGraph.Drawing.Inspector;
using Object = UnityEngine.Object;
using UnityEditor.Experimental.GraphView;
using UnityEditor.ShaderGraph.Drawing.Colors;
using UnityEngine.UIElements;
using Edge = UnityEditor.Experimental.GraphView.Edge;
using UnityEditor.VersionControl;
using UnityEditor.Searcher;
using Unity.Profiling;
namespace UnityEditor.ShaderGraph.Drawing
{
[Serializable]
class FloatingWindowsLayout
{
public WindowDockingLayout previewLayout = new WindowDockingLayout
{
dockingTop = false,
dockingLeft = false,
verticalOffset = 8,
horizontalOffset = 8
};
}
[Serializable]
class UserViewSettings
{
public bool isBlackboardVisible = true;
public bool isPreviewVisible = true;
public bool isInspectorVisible = true;
public string colorProvider = NoColors.Title;
}
class GraphEditorView : VisualElement, IDisposable
{
MaterialGraphView m_GraphView;
MasterPreviewView m_MasterPreviewView;
InspectorView m_InspectorView;
GraphData m_Graph;
PreviewManager m_PreviewManager;
MessageManager m_MessageManager;
SearchWindowProvider m_SearchWindowProvider;
EdgeConnectorListener m_EdgeConnectorListener;
VisualElement m_HoveredContextView;
BlackboardController m_BlackboardController;
internal BlackboardController blackboardController
{
get => m_BlackboardController;
set
{
if (value != null)
m_BlackboardController = value;
}
}
ColorManager m_ColorManager;
EditorWindow m_EditorWindow;
const string k_UserViewSettings = "UnityEditor.ShaderGraph.ToggleSettings";
UserViewSettings m_UserViewSettings;
internal UserViewSettings viewSettings { get => m_UserViewSettings; }
const string k_FloatingWindowsLayoutKey = "UnityEditor.ShaderGraph.FloatingWindowsLayout2";
FloatingWindowsLayout m_FloatingWindowsLayout = new FloatingWindowsLayout();
public Action saveRequested { get; set; }
public Action saveAsRequested { get; set; }
public Func<bool> isCheckedOut { get; set; }
public Action checkOut { get; set; }
public Action convertToSubgraphRequested
{
get { return m_GraphView.onConvertToSubgraphClick; }
set { m_GraphView.onConvertToSubgraphClick = value; }
}
public Action showInProjectRequested { get; set; }
public MaterialGraphView graphView
{
get { return m_GraphView; }
}
internal PreviewManager previewManager
{
get { return m_PreviewManager; }
set { m_PreviewManager = value; }
}
public string assetName
{
get => m_AssetName;
set
{
m_AssetName = value;
// Also update blackboard title
m_BlackboardController.UpdateBlackboardTitle(m_AssetName);
}
}
public ColorManager colorManager
{
get => m_ColorManager;
}
private static readonly ProfilerMarker AddGroupsMarker = new ProfilerMarker("AddGroups");
private static readonly ProfilerMarker AddStickyNotesMarker = new ProfilerMarker("AddStickyNotes");
public GraphEditorView(EditorWindow editorWindow, GraphData graph, MessageManager messageManager, string graphName)
{
m_GraphViewGroupTitleChanged = OnGroupTitleChanged;
m_GraphViewElementsAddedToGroup = OnElementsAddedToGroup;
m_GraphViewElementsRemovedFromGroup = OnElementsRemovedFromGroup;
m_EditorWindow = editorWindow;
m_Graph = graph;
m_AssetName = graphName;
m_MessageManager = messageManager;
previewManager = new PreviewManager(graph, messageManager);
previewManager.RenderPreviews(m_EditorWindow, false);
styleSheets.Add(Resources.Load<StyleSheet>("Styles/GraphEditorView"));
var serializedSettings = EditorUserSettings.GetConfigValue(k_UserViewSettings);
m_UserViewSettings = JsonUtility.FromJson<UserViewSettings>(serializedSettings) ?? new UserViewSettings();
m_ColorManager = new ColorManager(m_UserViewSettings.colorProvider);
var colorProviders = m_ColorManager.providerNames.ToArray();
var toolbar = new IMGUIContainer(() =>
{
GUILayout.BeginHorizontal(EditorStyles.toolbar);
if (GUILayout.Button("Save Asset", EditorStyles.toolbarButton))
{
if (saveRequested != null)
saveRequested();
}
GUILayout.Space(6);
if (GUILayout.Button("Save As...", EditorStyles.toolbarButton))
{
saveAsRequested();
}
GUILayout.Space(6);
if (GUILayout.Button("Show In Project", EditorStyles.toolbarButton))
{
if (showInProjectRequested != null)
showInProjectRequested();
}
if (isCheckedOut != null)
{
if (!isCheckedOut() && Provider.enabled && Provider.isActive)
{
if (GUILayout.Button("Check Out", EditorStyles.toolbarButton))
{
if (checkOut != null)
checkOut();
}
}
else
{
EditorGUI.BeginDisabledGroup(true);
GUILayout.Button("Check Out", EditorStyles.toolbarButton);
EditorGUI.EndDisabledGroup();
}
}
GUILayout.FlexibleSpace();
EditorGUI.BeginChangeCheck();
GUILayout.Label("Color Mode");
var newColorIndex = EditorGUILayout.Popup(m_ColorManager.activeIndex, colorProviders, GUILayout.Width(100f));
GUILayout.Space(4);
m_UserViewSettings.isBlackboardVisible = GUILayout.Toggle(m_UserViewSettings.isBlackboardVisible, "Blackboard", EditorStyles.toolbarButton);
GUILayout.Space(6);
m_UserViewSettings.isInspectorVisible = GUILayout.Toggle(m_UserViewSettings.isInspectorVisible, "Graph Inspector", EditorStyles.toolbarButton);
GUILayout.Space(6);
m_UserViewSettings.isPreviewVisible = GUILayout.Toggle(m_UserViewSettings.isPreviewVisible, "Main Preview", EditorStyles.toolbarButton);
if (EditorGUI.EndChangeCheck())
{
UserViewSettingsChangeCheck(newColorIndex);
}
GUILayout.EndHorizontal();
});
Add(toolbar);
var content = new VisualElement { name = "content" };
{
m_GraphView = new MaterialGraphView(graph, () => m_PreviewManager.UpdateMasterPreview(ModificationScope.Topological))
{ name = "GraphView", viewDataKey = "MaterialGraphView" };
m_GraphView.SetupZoom(0.05f, 8);
m_GraphView.AddManipulator(new ContentDragger());
m_GraphView.AddManipulator(new SelectionDragger());
m_GraphView.AddManipulator(new RectangleSelector());
m_GraphView.AddManipulator(new ClickSelector());
m_GraphView.RegisterCallback<KeyDownEvent>(OnKeyDown);
// Bugfix 1312222. Running 'ResetSelectedBlockNodes' on all mouse up interactions will break selection
// after changing tabs. This was originally added to fix a bug with middle-mouse clicking while dragging a block node.
m_GraphView.RegisterCallback<MouseUpEvent>(evt => { if (evt.button == (int)MouseButton.MiddleMouse) m_GraphView.ResetSelectedBlockNodes(); });
// This takes care of when a property is dragged from BB and then the drag is ended by the Escape key, hides the scroll boundary regions and drag indicator if so
m_GraphView.RegisterCallback<DragExitedEvent>(evt =>
{
blackboardController.blackboard.OnDragExitedEvent(evt);
blackboardController.blackboard.hideDragIndicatorAction?.Invoke();
});
RegisterGraphViewCallbacks();
content.Add(m_GraphView);
string serializedWindowLayout = EditorUserSettings.GetConfigValue(k_FloatingWindowsLayoutKey);
if (!string.IsNullOrEmpty(serializedWindowLayout))
{
m_FloatingWindowsLayout = JsonUtility.FromJson<FloatingWindowsLayout>(serializedWindowLayout);
}
CreateMasterPreview();
CreateInspector();
CreateBlackboard();
UpdateSubWindowsVisibility();
m_GraphView.graphViewChanged = GraphViewChanged;
RegisterCallback<GeometryChangedEvent>(ApplySerializedWindowLayouts);
if (m_Graph.isSubGraph)
{
m_GraphView.AddToClassList("subgraph");
}
}
m_SearchWindowProvider = new SearcherProvider();
m_SearchWindowProvider.Initialize(editorWindow, m_Graph, m_GraphView);
m_GraphView.nodeCreationRequest = NodeCreationRequest;
//regenerate entries when graph view is refocused, to propogate subgraph changes
m_GraphView.RegisterCallback<FocusInEvent>(evt => { m_SearchWindowProvider.regenerateEntries = true; });
m_EdgeConnectorListener = new EdgeConnectorListener(m_Graph, m_SearchWindowProvider, editorWindow);
if (!m_Graph.isSubGraph)
{
AddContexts();
}
using (AddGroupsMarker.Auto())
{
foreach (var graphGroup in graph.groups)
AddGroup(graphGroup);
}
using (AddStickyNotesMarker.Auto())
{
foreach (var stickyNote in graph.stickyNotes)
AddStickyNote(stickyNote);
}
AddNodes(graph.GetNodes<AbstractMaterialNode>());
AddBlocks(graph.GetNodes<BlockNode>());
AddEdges(graph.edges);
Add(content);
// Active block lists need to be initialized on window start up
// Do this here to as we cant do this inside GraphData
// This is due to targets not being deserialized yet
var context = new TargetSetupContext();
foreach (var target in m_Graph.activeTargets)
{
target.Setup(ref context);
}
var activeBlocks = m_Graph.GetActiveBlocksForAllActiveTargets();
m_Graph.UpdateActiveBlocks(activeBlocks);
// Graph settings need to be initialized after the target setup
m_InspectorView.InitializeGraphSettings();
}
private void CreateBlackboard()
{
var blackboardViewModel = new BlackboardViewModel() { parentView = graphView, model = m_Graph, title = assetName };
m_BlackboardController = new BlackboardController(m_Graph, blackboardViewModel, m_Graph.owner.graphDataStore);
}
void AddContexts()
{
ContextView AddContext(string name, ContextData contextData, Direction portDirection)
{
//need to eventually remove this reference to editor window in context views
var contextView = new ContextView(name, contextData, m_EditorWindow);
// GraphView marks ContextViews' stacks, but not the actual root elements, as insertable. We want the
// contextual searcher menu to come up when *any* part of the ContextView is hovered. As a workaround,
// we keep track of the hovered ContextView and offer it if no targets are found.
contextView.RegisterCallback((MouseOverEvent _) => m_HoveredContextView = contextView);
contextView.RegisterCallback((MouseOutEvent _) =>
{
if (m_HoveredContextView == contextView) m_HoveredContextView = null;
});
contextView.SetPosition(new Rect(contextData.position, Vector2.zero));
contextView.AddPort(portDirection);
m_GraphView.AddElement(contextView);
return contextView;
}
// Add Contexts
// As Contexts are hardcoded and contain a single port we can just give the direction
var vertexContext = AddContext("Vertex", m_Graph.vertexContext, Direction.Output);
var fragmentContext = AddContext("Fragment", m_Graph.fragmentContext, Direction.Input);
// Connect Contexts
// Vertical Edges have no representation in Model
// Therefore just draw it and dont allow interaction
var contextEdge = new Edge()
{
output = vertexContext.port,
input = fragmentContext.port,
pickingMode = PickingMode.Ignore,
};
m_GraphView.AddElement(contextEdge);
// Update the Context list on MaterialGraphView
m_GraphView.UpdateContextList();
}
internal void UserViewSettingsChangeCheck(int newColorIndex)
{
if (newColorIndex != m_ColorManager.activeIndex)
{
m_ColorManager.SetActiveProvider(newColorIndex, m_GraphView.Query<MaterialNodeView>().ToList());
m_UserViewSettings.colorProvider = m_ColorManager.activeProviderName;
}
var serializedViewSettings = JsonUtility.ToJson(m_UserViewSettings);
EditorUserSettings.SetConfigValue(k_UserViewSettings, serializedViewSettings);
UpdateSubWindowsVisibility();
}
void NodeCreationRequest(NodeCreationContext c)
{
if (EditorWindow.focusedWindow == m_EditorWindow) //only display the search window when current graph view is focused
{
m_SearchWindowProvider.connectedPort = null;
m_SearchWindowProvider.target = c.target ?? m_HoveredContextView;
var displayPosition = (c.screenMousePosition - m_EditorWindow.position.position);
SearcherWindow.Show(m_EditorWindow, (m_SearchWindowProvider as SearcherProvider).LoadSearchWindow(),
item => (m_SearchWindowProvider as SearcherProvider).OnSearcherSelectEntry(item, c.screenMousePosition - m_EditorWindow.position.position),
displayPosition, null, new SearcherWindow.Alignment(SearcherWindow.Alignment.Vertical.Center, SearcherWindow.Alignment.Horizontal.Left));
}
}
// Master Preview, Inspector and Blackboard all need to keep their layouts when hidden in order to restore user preferences.
// Because of their differences we do this is different ways, for now.
void UpdateSubWindowsVisibility()
{
// Blackboard needs to be effectively removed when hidden to avoid bugs.
if (m_UserViewSettings.isBlackboardVisible)
blackboardController.blackboard.ShowWindow();
else
blackboardController.blackboard.HideWindow();
// Same for the inspector
if (m_UserViewSettings.isInspectorVisible)
m_InspectorView.ShowWindow();
else
m_InspectorView.HideWindow();
m_MasterPreviewView.visible = m_UserViewSettings.isPreviewVisible;
}
Action<Group, string> m_GraphViewGroupTitleChanged;
Action<Group, IEnumerable<GraphElement>> m_GraphViewElementsAddedToGroup;
Action<Group, IEnumerable<GraphElement>> m_GraphViewElementsRemovedFromGroup;
void RegisterGraphViewCallbacks()
{
m_GraphView.groupTitleChanged = m_GraphViewGroupTitleChanged;
m_GraphView.elementsAddedToGroup = m_GraphViewElementsAddedToGroup;
m_GraphView.elementsRemovedFromGroup = m_GraphViewElementsRemovedFromGroup;
}
void UnregisterGraphViewCallbacks()
{
m_GraphView.groupTitleChanged = null;
m_GraphView.elementsAddedToGroup = null;
m_GraphView.elementsRemovedFromGroup = null;
}
void CreateMasterPreview()
{
m_MasterPreviewView = new MasterPreviewView(previewManager, m_Graph) { name = "masterPreview" };
var masterPreviewViewDraggable = new WindowDraggable(null, this);
m_MasterPreviewView.AddManipulator(masterPreviewViewDraggable);
m_GraphView.Add(m_MasterPreviewView);
masterPreviewViewDraggable.OnDragFinished += UpdateSerializedWindowLayout;
m_MasterPreviewView.previewResizeBorderFrame.OnResizeFinished += UpdateSerializedWindowLayout;
}
void CreateInspector()
{
var inspectorViewModel = new InspectorViewModel() { parentView = this.graphView };
m_InspectorView = new InspectorView(inspectorViewModel);
graphView.OnSelectionChange += m_InspectorView.TriggerInspectorUpdate;
// Undo/redo actions that only affect selection don't trigger the above callback for some reason, so we also have to do this
Undo.undoRedoPerformed += (() => { m_InspectorView?.TriggerInspectorUpdate(graphView?.selection); });
}
void OnKeyDown(KeyDownEvent evt)
{
if (evt.keyCode == KeyCode.F1)
{
var selection = m_GraphView.selection.OfType<IShaderNodeView>();
if (selection.Count() == 1)
{
var nodeView = selection.First();
if (nodeView.node.documentationURL != null)
{
System.Diagnostics.Process.Start(nodeView.node.documentationURL);
}
}
}
if (evt.actionKey && evt.keyCode == KeyCode.G)
{
if (m_GraphView.selection.OfType<GraphElement>().Any())
{
m_GraphView.GroupSelection();
}
}
if (evt.actionKey && evt.keyCode == KeyCode.U)
{
if (m_GraphView.selection.OfType<GraphElement>().Any())
{
m_GraphView.RemoveFromGroupNode();
}
}
}
GraphViewChange GraphViewChanged(GraphViewChange graphViewChange)
{
if (graphViewChange.edgesToCreate != null)
{
foreach (var edge in graphViewChange.edgesToCreate)
{
var leftSlot = edge.output.GetSlot();
var rightSlot = edge.input.GetSlot();
if (leftSlot != null && rightSlot != null)
{
m_Graph.owner.RegisterCompleteObjectUndo("Connect Edge");
m_Graph.Connect(leftSlot.slotReference, rightSlot.slotReference);
}
}
graphViewChange.edgesToCreate.Clear();
}
if (graphViewChange.movedElements != null)
{
m_Graph.owner.RegisterCompleteObjectUndo("Move Elements");
List<GraphElement> nodesInsideGroup = new List<GraphElement>();
foreach (var element in graphViewChange.movedElements)
{
var groupNode = element as ShaderGroup;
if (groupNode == null)
continue;
foreach (GraphElement graphElement in groupNode.containedElements)
{
nodesInsideGroup.Add(graphElement);
}
SetGroupPosition(groupNode);
}
if (nodesInsideGroup.Any())
graphViewChange.movedElements.AddRange(nodesInsideGroup);
foreach (var element in graphViewChange.movedElements)
{
if (element.userData is AbstractMaterialNode node)
{
var drawState = node.drawState;
drawState.position = element.parent.ChangeCoordinatesTo(m_GraphView.contentViewContainer, element.GetPosition());
node.drawState = drawState;
// BlockNode moved outside a Context
// This isnt allowed but there is no way to disallow it on the GraphView
if (node is BlockNode blockNode &&
element.GetFirstAncestorOfType<ContextView>() == null)
{
var context = graphView.GetContext(blockNode.contextData);
// isDragging ensures we arent calling this when moving
// the BlockNode into the GraphView during dragging
if (context.isDragging)
continue;
// Remove from GraphView and add back to Context
m_GraphView.RemoveElement(element);
context.InsertBlock(element as MaterialNodeView);
}
}
if (element is StickyNote stickyNote)
{
SetStickyNotePosition(stickyNote);
}
if (element is ContextView contextView)
{
var rect = element.parent.ChangeCoordinatesTo(m_GraphView.contentViewContainer, element.GetPosition());
contextView.contextData.position = rect.position;
}
}
}
var nodesToUpdate = m_NodeViewHashSet;
nodesToUpdate.Clear();
if (graphViewChange.elementsToRemove != null)
{
m_Graph.owner.RegisterCompleteObjectUndo("Remove Elements");
m_Graph.RemoveElements(graphViewChange.elementsToRemove.OfType<IShaderNodeView>().Select(v => v.node).ToArray(),
graphViewChange.elementsToRemove.OfType<Edge>().Select(e => (IEdge)e.userData).ToArray(),
graphViewChange.elementsToRemove.OfType<ShaderGroup>().Select(g => g.userData).ToArray(),
graphViewChange.elementsToRemove.OfType<StickyNote>().Select(n => n.userData).ToArray());
foreach (var edge in graphViewChange.elementsToRemove.OfType<Edge>())
{
if (edge.input != null)
{
if (edge.input.node is IShaderNodeView materialNodeView)
nodesToUpdate.Add(materialNodeView);
}
if (edge.output != null)
{
if (edge.output.node is IShaderNodeView materialNodeView)
nodesToUpdate.Add(materialNodeView);
}
}
}
foreach (var node in nodesToUpdate)
{
node.OnModified(ModificationScope.Topological);
}
UpdateEdgeColors(nodesToUpdate);
return graphViewChange;
}
void SetGroupPosition(ShaderGroup groupNode)
{
var pos = groupNode.GetPosition();
groupNode.userData.position = new Vector2(pos.x, pos.y);
}
void SetStickyNotePosition(StickyNote stickyNote)
{
var pos = stickyNote.GetPosition();
stickyNote.userData.position = new Rect(pos);
}
void OnGroupTitleChanged(Group graphGroup, string title)
{
var groupData = graphGroup.userData as GroupData;
if (groupData != null)
{
groupData.title = graphGroup.title;
}
}
void OnElementsAddedToGroup(Group graphGroup, IEnumerable<GraphElement> elements)
{
if (graphGroup.userData is GroupData groupData)
{
var anyChanged = false;
foreach (var element in elements)
{
if (element.userData is IGroupItem groupItem && groupItem.group != groupData)
{
anyChanged = true;
break;
}
}
if (!anyChanged)
return;
m_Graph.owner.RegisterCompleteObjectUndo(groupData.title);
foreach (var element in elements)
{
if (element.userData is IGroupItem groupItem)
{
m_Graph.SetGroup(groupItem, groupData);
}
}
}
}
void OnElementsRemovedFromGroup(Group graphGroup, IEnumerable<GraphElement> elements)
{
if (graphGroup.userData is GroupData groupData)
{
var anyChanged = false;
foreach (var element in elements)
{
if (element.userData is IGroupItem groupItem && groupItem.group == groupData)
{
anyChanged = true;
break;
}
}
if (!anyChanged)
return;
m_Graph.owner.RegisterCompleteObjectUndo("Ungroup Node(s)");
foreach (var element in elements)
{
if (element.userData is IGroupItem groupItem)
{
m_Graph.SetGroup(groupItem, null);
SetGroupPosition((ShaderGroup)graphGroup); //, (GraphElement)nodeView);
}
}
}
}
void OnNodeChanged(AbstractMaterialNode inNode, ModificationScope scope)
{
if (m_GraphView == null)
return;
var dependentNodes = new List<AbstractMaterialNode>();
if (!inNode.owner.graphIsConcretizing)
NodeUtils.CollectNodesNodeFeedsInto(dependentNodes, inNode);
else dependentNodes.Add(inNode);
foreach (var node in dependentNodes)
{
var nodeView = m_GraphView.GetNodeByGuid(node.objectId) as IShaderNodeView;
if (nodeView != null)
nodeView.OnModified(scope);
}
}
HashSet<IShaderNodeView> m_NodeViewHashSet = new HashSet<IShaderNodeView>();
HashSet<ShaderGroup> m_GroupHashSet = new HashSet<ShaderGroup>();
public void HandleGraphChanges(bool wasUndoRedoPerformed)
{
UnregisterGraphViewCallbacks();
previewManager.HandleGraphChanges();
m_InspectorView.HandleGraphChanges();
if (m_Graph.addedEdges.Any() || m_Graph.removedEdges.Any())
{
var nodeList = m_GraphView.Query<MaterialNodeView>().ToList();
m_ColorManager.SetNodesDirty(nodeList);
m_ColorManager.UpdateNodeViews(nodeList);
}
previewManager.RenderPreviews(m_EditorWindow);
m_GraphView.wasUndoRedoPerformed = wasUndoRedoPerformed;
if (wasUndoRedoPerformed || m_InspectorView.doesInspectorNeedUpdate)
m_InspectorView.Update();
if (wasUndoRedoPerformed)
m_GraphView.RestorePersistentSelectionAfterUndoRedo();
m_GroupHashSet.Clear();
HandleRemovedNodes();
foreach (var noteData in m_Graph.removedNotes)
{
var note = m_GraphView.graphElements.ToList().OfType<StickyNote>().First(n => n.userData == noteData);
m_GraphView.RemoveElement(note);
}
foreach (GroupData groupData in m_Graph.removedGroups)
{
var group = m_GraphView.graphElements.ToList().OfType<ShaderGroup>().First(g => g.userData == groupData);
m_GraphView.RemoveElement(group);
}
foreach (var groupData in m_Graph.addedGroups)
{
AddGroup(groupData);
}
foreach (var stickyNote in m_Graph.addedStickyNotes)
{
AddStickyNote(stickyNote);
}
foreach (var node in m_Graph.addedNodes)
{
AddNode(node);
}
foreach (var groupChange in m_Graph.parentGroupChanges)
{
GraphElement graphElement = null;
if (groupChange.groupItem is AbstractMaterialNode node)
{
graphElement = m_GraphView.GetNodeByGuid(node.objectId);
}
else if (groupChange.groupItem is StickyNoteData stickyNote)
{
graphElement = m_GraphView.GetElementByGuid(stickyNote.objectId);
}
else
{
throw new InvalidOperationException("Unknown group item type.");
}
if (graphElement != null)
{
var groupView = graphElement.GetContainingScope() as ShaderGroup;
if (groupView?.userData != groupChange.newGroup)
{
groupView?.RemoveElement(graphElement);
if (groupChange.newGroup != null)
{
var newGroupView = m_GraphView.graphElements.ToList()
.OfType<ShaderGroup>()
.First(x => x.userData == groupChange.newGroup);
newGroupView.AddElement(graphElement);
}
}
}
}
foreach (var groupData in m_Graph.pastedGroups)
{
var group = m_GraphView.graphElements.ToList().OfType<ShaderGroup>().ToList().First(g => g.userData == groupData);
m_GraphView.AddToSelection(group);
}
foreach (var stickyNoteData in m_Graph.pastedStickyNotes)
{
var stickyNote = m_GraphView.graphElements.ToList().OfType<StickyNote>().First(s => s.userData == stickyNoteData);
m_GraphView.AddToSelection(stickyNote);
}
foreach (var node in m_Graph.pastedNodes)
{
var nodeView = m_GraphView.nodes.ToList().OfType<IShaderNodeView>()
.FirstOrDefault(p => p.node != null && p.node == node);
m_GraphView.AddToSelection((Node)nodeView);
}
foreach (var shaderGroup in m_GroupHashSet)
{
SetGroupPosition(shaderGroup);
}
var nodesToUpdate = m_NodeViewHashSet;
nodesToUpdate.Clear();
foreach (var edge in m_Graph.removedEdges)
{
var edgeView = m_GraphView.graphElements.ToList().OfType<Edge>()
.FirstOrDefault(p => p.userData is IEdge && Equals((IEdge)p.userData, edge));
if (edgeView != null)
{
var nodeView = (IShaderNodeView)edgeView.input.node;
if (nodeView?.node != null)
{
nodesToUpdate.Add(nodeView);
}
edgeView.output.Disconnect(edgeView);
edgeView.input.Disconnect(edgeView);
edgeView.output = null;
edgeView.input = null;
m_GraphView.RemoveElement(edgeView);
}
}
foreach (var edge in m_Graph.addedEdges)
{
var edgeView = AddEdge(edge);
if (edgeView != null)
nodesToUpdate.Add((IShaderNodeView)edgeView.input.node);
}
foreach (var node in nodesToUpdate)
{
node.OnModified(ModificationScope.Topological);
}
UpdateEdgeColors(nodesToUpdate);
if (m_Graph.movedContexts)
{
foreach (var context in m_GraphView.contexts)
{
context.SetPosition(new Rect(context.contextData.position, Vector2.zero));
}
}
// Checking if any new Group Nodes just got added
if (m_Graph.mostRecentlyCreatedGroup != null)
{
var groups = m_GraphView.graphElements.ToList().OfType<ShaderGroup>();
foreach (ShaderGroup shaderGroup in groups)
{
if (shaderGroup.userData == m_Graph.mostRecentlyCreatedGroup)
{
shaderGroup.FocusTitleTextField();
break;
}
}
}
// If we auto-remove blocks and something has happened to trigger a check (don't re-check constantly)
if (m_Graph.checkAutoAddRemoveBlocks && ShaderGraphPreferences.autoAddRemoveBlocks)
{
var activeBlocks = m_Graph.GetActiveBlocksForAllActiveTargets();
m_Graph.AddRemoveBlocksFromActiveList(activeBlocks);
m_Graph.checkAutoAddRemoveBlocks = false;
// We have to re-check any nodes views that need to be removed since we already handled this above. After leaving this function the states on m_Graph will be cleared so we'll lose track of removed blocks.
HandleRemovedNodes();
}
UpdateBadges();
RegisterGraphViewCallbacks();
}
void HandleRemovedNodes()
{
foreach (var node in m_Graph.removedNodes)
{
node.UnregisterCallback(OnNodeChanged);
var nodeView = m_GraphView.nodes.ToList().OfType<IShaderNodeView>()
.FirstOrDefault(p => p.node != null && p.node == node);
// When deleting a node make sure to clear any input observers
switch (node)
{
case PropertyNode propertyNode:
propertyNode.property.RemoveObserver(propertyNode);
propertyNode.property.RemoveObserver(nodeView as IShaderInputObserver);
break;
case KeywordNode keywordNode:
keywordNode.keyword.RemoveObserver(keywordNode);
break;
case DropdownNode dropdownNode:
dropdownNode.dropdown.RemoveObserver(dropdownNode);
break;
}
if (nodeView != null)
{
nodeView.Dispose();
if (node is BlockNode blockNode)
{
var context = m_GraphView.GetContext(blockNode.contextData);
// blocknode may be floating and not actually in the stacknode's visual hierarchy.
if (context.Contains(nodeView as Node))
{
context.RemoveElement(nodeView as Node);
}
else
{
m_GraphView.RemoveElement((Node)nodeView);
}
}
else
{
m_GraphView.RemoveElement((Node)nodeView);
}
if (node.group != null)
{
var shaderGroup = m_GraphView.graphElements.ToList().OfType<ShaderGroup>().First(g => g.userData == node.group);
m_GroupHashSet.Add(shaderGroup);
}
}
}
}
void UpdateBadges()
{
if (!m_MessageManager.nodeMessagesChanged)
return;
foreach (var messageData in m_MessageManager.GetNodeMessages())
{
var node = m_Graph.GetNodeFromId(messageData.Key);
if (node == null || !(m_GraphView.GetNodeByGuid(node.objectId) is IShaderNodeView nodeView))
continue;
if (messageData.Value.Count == 0)
{
nodeView.ClearMessage();
}
else
{
var foundMessage = messageData.Value.First();
string messageString;
if (foundMessage.line > 0)
messageString = foundMessage.message + " at line " + foundMessage.line;
else
messageString = foundMessage.message;
nodeView.AttachMessage(messageString, foundMessage.severity);
}
}
}
List<GraphElement> m_GraphElementsTemp = new List<GraphElement>();
void AddNode(AbstractMaterialNode node, bool usePrebuiltVisualGroupMap = false)
{
var materialNode = node;
Node nodeView;
if (node is PropertyNode propertyNode)
{
var tokenNode = new PropertyNodeView(propertyNode, m_EdgeConnectorListener);
m_GraphView.AddElement(tokenNode);
nodeView = tokenNode;
// Register node model and node view as observer of property
propertyNode.property.AddObserver(propertyNode);
propertyNode.property.AddObserver(tokenNode);
}
else if (node is BlockNode blockNode)
{
var blockNodeView = new MaterialNodeView { userData = blockNode };
blockNodeView.Initialize(blockNode, m_PreviewManager, m_EdgeConnectorListener, graphView);
blockNodeView.MarkDirtyRepaint();
nodeView = blockNodeView;
var context = m_GraphView.GetContext(blockNode.contextData);
context.InsertBlock(blockNodeView);
}
else if (node is RedirectNodeData redirectNodeData)
{
var redirectNodeView = new RedirectNodeView { userData = redirectNodeData };
m_GraphView.AddElement(redirectNodeView);
redirectNodeView.ConnectToData(materialNode, m_EdgeConnectorListener);
nodeView = redirectNodeView;
}
else
{
var materialNodeView = new MaterialNodeView { userData = materialNode };
// For keywords and dropdowns, we only register the node model itself as an observer,
// the material node view redraws completely on changes so it doesn't need to be an observer
switch (node)
{
case KeywordNode keywordNode:
keywordNode.keyword.AddObserver(keywordNode);
break;
case DropdownNode dropdownNode:
dropdownNode.dropdown.AddObserver(dropdownNode);
break;
}
m_GraphView.AddElement(materialNodeView);
materialNodeView.Initialize(materialNode, m_PreviewManager, m_EdgeConnectorListener, graphView);
m_ColorManager.UpdateNodeView(materialNodeView);
nodeView = materialNodeView;
}
node.RegisterCallback(OnNodeChanged);
nodeView.MarkDirtyRepaint();
if (m_SearchWindowProvider.nodeNeedsRepositioning &&
m_SearchWindowProvider.targetSlotReference.node == node)
{
m_SearchWindowProvider.nodeNeedsRepositioning = false;
if (nodeView is IShaderNodeView shaderView &&
shaderView.FindPort(m_SearchWindowProvider.targetSlotReference, out var port))
{
port.RegisterCallback<GeometryChangedEvent>(RepositionNode);
return;
}
}
if (materialNode.group != null)
{
if (usePrebuiltVisualGroupMap)
{
// cheaper way to add the node to groups it is in
ShaderGroup groupView;
visualGroupMap.TryGetValue(materialNode.group, out groupView);
if (groupView != null)
groupView.AddElement(nodeView);
}
else
{
// This should also work for sticky notes
m_GraphElementsTemp.Clear();
m_GraphView.graphElements.ToList(m_GraphElementsTemp);
foreach (var element in m_GraphElementsTemp)
{
if (element is ShaderGroup groupView && groupView.userData == materialNode.group)
{
groupView.AddElement(nodeView);
}
}
}
}
}
private static Dictionary<GroupData, ShaderGroup> visualGroupMap = new Dictionary<GroupData, ShaderGroup>();
private static void AddToVisualGroupMap(GraphElement e)
{
if (e is ShaderGroup sg)
{
visualGroupMap.Add(sg.userData, sg);
}
}
private static Action<GraphElement> AddToVisualGroupMapAction = AddToVisualGroupMap;
void BuildVisualGroupMap()
{
visualGroupMap.Clear();
m_GraphView.graphElements.ForEach(AddToVisualGroupMapAction);
}
private static readonly ProfilerMarker AddNodesMarker = new ProfilerMarker("AddNodes");
void AddNodes(IEnumerable<AbstractMaterialNode> nodes)
{
using (AddNodesMarker.Auto())
{
BuildVisualGroupMap();
foreach (var node in nodes)
{
// Skip BlockNodes as we need to order them
if (node is BlockNode)
continue;
AddNode(node, true);
}
visualGroupMap.Clear();
}
}
private static readonly ProfilerMarker AddBlocksMarker = new ProfilerMarker("AddBlocks");
void AddBlocks(IEnumerable<BlockNode> blocks)
{
using (AddBlocksMarker.Auto())
{
// As they can be reordered, we cannot be sure BlockNodes are deserialized in the same order as their stack position
// To handle this we reorder the BlockNodes here to avoid having to reorder them on the fly as they are added
foreach (var node in blocks.OrderBy(s => s.index))
{
AddNode(node);
}
}
}
void AddGroup(GroupData groupData)
{
ShaderGroup graphGroup = new ShaderGroup();
graphGroup.userData = groupData;
graphGroup.title = groupData.title;
graphGroup.SetPosition(new Rect(graphGroup.userData.position, Vector2.zero));
m_GraphView.AddElement(graphGroup);
}
void AddStickyNote(StickyNoteData stickyNoteData)
{
var stickyNote = new StickyNote(stickyNoteData.position, m_Graph);
stickyNote.userData = stickyNoteData;
stickyNote.viewDataKey = stickyNoteData.objectId;
stickyNote.title = stickyNoteData.title;
stickyNote.contents = stickyNoteData.content;
stickyNote.textSize = (StickyNote.TextSize)stickyNoteData.textSize;
stickyNote.theme = (StickyNote.Theme)stickyNoteData.theme;
stickyNote.userData.group = stickyNoteData.group;
stickyNote.SetPosition(new Rect(stickyNote.userData.position));
m_GraphView.AddElement(stickyNote);
// Add Sticky Note to group
m_GraphElementsTemp.Clear();
m_GraphView.graphElements.ToList(m_GraphElementsTemp);
if (stickyNoteData.group != null)
{
foreach (var element in m_GraphElementsTemp)
{
if (element is ShaderGroup groupView && groupView.userData == stickyNoteData.group)
{
groupView.AddElement(stickyNote);
}
}
}
}
static void RepositionNode(GeometryChangedEvent evt)
{
var port = evt.target as ShaderPort;
if (port == null)
return;
port.UnregisterCallback<GeometryChangedEvent>(RepositionNode);
var nodeView = port.node as IShaderNodeView;
if (nodeView == null)
return;
var offset = nodeView.gvNode.mainContainer.WorldToLocal(port.GetGlobalCenter() + new Vector3(3f, 3f, 0f));
var position = nodeView.gvNode.GetPosition();
position.position -= offset;
nodeView.gvNode.SetPosition(position);
var drawState = nodeView.node.drawState;
drawState.position = position;
nodeView.node.drawState = drawState;
nodeView.gvNode.MarkDirtyRepaint();
port.MarkDirtyRepaint();
}
private static Dictionary<AbstractMaterialNode, IShaderNodeView> visualNodeMap = new Dictionary<AbstractMaterialNode, IShaderNodeView>();
private static void AddToVisualNodeMap(Node n)
{
IShaderNodeView snv = n as IShaderNodeView;
if (snv != null)
visualNodeMap.Add(snv.node, snv);
}
private static Action<Node> AddToVisualNodeMapAction = AddToVisualNodeMap;
void BuildVisualNodeMap()
{
visualNodeMap.Clear();
m_GraphView.nodes.ForEach(AddToVisualNodeMapAction);
}
private static readonly ProfilerMarker AddEdgesMarker = new ProfilerMarker("AddEdges");
void AddEdges(IEnumerable<IEdge> edges)
{
using (AddEdgesMarker.Auto())
{
// fast way
BuildVisualNodeMap();
foreach (IEdge edge in edges)
{
AddEdge(edge, true, false);
}
// apply the port update on every node
foreach (IShaderNodeView nodeView in visualNodeMap.Values)
{
nodeView.gvNode.RefreshPorts();
nodeView.UpdatePortInputTypes();
}
// cleanup temp data
visualNodeMap.Clear();
}
}
Edge AddEdge(IEdge edge, bool useVisualNodeMap = false, bool updateNodePorts = true)
{
var sourceNode = edge.outputSlot.node;
if (sourceNode == null)
{
Debug.LogWarning("Source node is null");
return null;
}
var sourceSlot = sourceNode.FindOutputSlot<MaterialSlot>(edge.outputSlot.slotId);
var targetNode = edge.inputSlot.node;
if (targetNode == null)
{
Debug.LogWarning("Target node is null");
return null;
}
var targetSlot = targetNode.FindInputSlot<MaterialSlot>(edge.inputSlot.slotId);
IShaderNodeView sourceNodeView;
if (useVisualNodeMap)
visualNodeMap.TryGetValue(sourceNode, out sourceNodeView);
else
sourceNodeView = m_GraphView.nodes.ToList().OfType<IShaderNodeView>().FirstOrDefault(x => x.node == sourceNode);
if (sourceNodeView != null)
{
sourceNodeView.FindPort(sourceSlot.slotReference, out var sourceAnchor);
IShaderNodeView targetNodeView;
if (useVisualNodeMap)
visualNodeMap.TryGetValue(targetNode, out targetNodeView);
else
targetNodeView = m_GraphView.nodes.ToList().OfType<IShaderNodeView>().First(x => x.node == targetNode);
targetNodeView.FindPort(targetSlot.slotReference, out var targetAnchor);
var edgeView = new Edge
{
userData = edge,
output = sourceAnchor,
input = targetAnchor
};
edgeView.RegisterCallback<MouseDownEvent>(OnMouseDown);
edgeView.output.Connect(edgeView);
edgeView.input.Connect(edgeView);
m_GraphView.AddElement(edgeView);
if (updateNodePorts)
{
sourceNodeView.gvNode.RefreshPorts();
targetNodeView.gvNode.RefreshPorts();
sourceNodeView.UpdatePortInputTypes();
targetNodeView.UpdatePortInputTypes();
}
return edgeView;
}
return null;
}
void OnMouseDown(MouseDownEvent evt)
{
if (evt.button == (int)MouseButton.LeftMouse && evt.clickCount == 2)
{
if (evt.target is Edge edgeTarget)
{
Vector2 pos = evt.mousePosition;
m_GraphView.CreateRedirectNode(pos, edgeTarget);
}
}
}
Stack<Node> m_NodeStack = new Stack<Node>();
string m_AssetName;
void UpdateEdgeColors(HashSet<IShaderNodeView> nodeViews)
{
var nodeStack = m_NodeStack;
nodeStack.Clear();
foreach (var nodeView in nodeViews)
nodeStack.Push((Node)nodeView);
PooledList<Edge> edgesToUpdate = PooledList<Edge>.Get();
while (nodeStack.Any())
{
var nodeView = nodeStack.Pop();
if (nodeView is IShaderNodeView shaderNodeView)
{
shaderNodeView.UpdatePortInputTypes();
}
foreach (var anchorView in nodeView.outputContainer.Children().OfType<Port>())
{
foreach (var edgeView in anchorView.connections)
{
//update edges based on the active state of any modified nodes
if (edgeView.input.node is MaterialNodeView inputNode && edgeView.output.node is MaterialNodeView outputNode)
{
//force redraw on update to prevent visual lag in the graph
//Now has to be delayed a frame because setting port styles wont update colors till next frame
edgesToUpdate.Add(edgeView);
}
//update edges based on dynamic vector length of any modified nodes
var targetSlot = edgeView.input.GetSlot();
if (targetSlot.valueType == SlotValueType.DynamicVector || targetSlot.valueType == SlotValueType.DynamicMatrix || targetSlot.valueType == SlotValueType.Dynamic)
{
var connectedNodeView = edgeView.input.node;
if (connectedNodeView != null && !nodeViews.Contains((IShaderNodeView)connectedNodeView))
{
nodeStack.Push(connectedNodeView);
nodeViews.Add((IShaderNodeView)connectedNodeView);
}
}
}
}
foreach (var anchorView in nodeView.inputContainer.Query<Port>().ToList())
{
var targetSlot = anchorView.GetSlot();
if (targetSlot.valueType != SlotValueType.DynamicVector)
continue;
foreach (var edgeView in anchorView.connections)
{
//update edges based on the active state of any modified nodes
if (edgeView.input.node is MaterialNodeView inputNode && edgeView.output.node is MaterialNodeView outputNode)
{
//force redraw on update to prevent visual lag in the graph
//Now has to be delayed a frame because setting port styles wont update colors till next frame
edgesToUpdate.Add(edgeView);
}
//update edge color for upstream dynamic vector types
var connectedNodeView = edgeView.output.node;
if (connectedNodeView != null && !nodeViews.Contains((IShaderNodeView)connectedNodeView))
{
nodeStack.Push(connectedNodeView);
nodeViews.Add((IShaderNodeView)connectedNodeView);
}
}
}
}
schedule.Execute(() =>
{
foreach (Edge e in edgesToUpdate)
{
e.UpdateEdgeControl();
}
edgesToUpdate.Dispose();
}).StartingIn(0);
}
void ApplySerializedWindowLayouts(GeometryChangedEvent evt)
{
UnregisterCallback<GeometryChangedEvent>(ApplySerializedWindowLayouts);
ApplyMasterPreviewLayout();
m_BlackboardController.blackboard.DeserializeLayout();
m_InspectorView.DeserializeLayout();
}
void ApplyMasterPreviewLayout()
{
// If a preview size was loaded in from saved user settings use that
if (m_FloatingWindowsLayout.previewLayout.size.x > 0f && m_FloatingWindowsLayout.previewLayout.size.y > 0f)
{
previewManager.ResizeMasterPreview(m_FloatingWindowsLayout.previewLayout.size);
}
else // Use default specified in the stylesheet for master preview
{
m_FloatingWindowsLayout.previewLayout.size = m_MasterPreviewView.layout.size;
}
m_FloatingWindowsLayout.previewLayout.ApplyPosition(m_MasterPreviewView);
m_MasterPreviewView.style.width = m_FloatingWindowsLayout.previewLayout.size.x;
m_MasterPreviewView.style.height = m_FloatingWindowsLayout.previewLayout.size.y;
m_MasterPreviewView.RegisterCallback<GeometryChangedEvent>(SerializeMasterPreviewLayout);
}
void SerializeMasterPreviewLayout(GeometryChangedEvent evt)
{
UpdateSerializedWindowLayout();
}
void UpdateSerializedWindowLayout()
{
m_FloatingWindowsLayout.previewLayout.CalculateDockingCornerAndOffset(m_MasterPreviewView.layout, m_GraphView.layout);
m_FloatingWindowsLayout.previewLayout.ClampToParentWindow();
blackboardController.blackboard.ClampToParentLayout(m_GraphView.layout);
m_InspectorView.ClampToParentLayout(m_GraphView.layout);
if (m_MasterPreviewView.visible)
{
m_FloatingWindowsLayout.previewLayout.size = m_MasterPreviewView.layout.size;
}
string serializedWindowLayout = JsonUtility.ToJson(m_FloatingWindowsLayout);
EditorUserSettings.SetConfigValue(k_FloatingWindowsLayoutKey, serializedWindowLayout);
}
public void Dispose()
{
if (m_GraphView != null)
{
saveRequested = null;
saveAsRequested = null;
convertToSubgraphRequested = null;
showInProjectRequested = null;
isCheckedOut = null;
checkOut = null;
foreach (var materialNodeView in m_GraphView.Query<MaterialNodeView>().ToList())
materialNodeView.Dispose();
foreach (var propertyNodeView in m_GraphView.Query<PropertyNodeView>().ToList())
propertyNodeView.Dispose();
foreach (var redirectNodeView in m_GraphView.Query<RedirectNodeView>().ToList())
redirectNodeView.Dispose();
foreach (var contextView in m_GraphView.Query<ContextView>().ToList())
contextView.Dispose();
foreach (var edge in m_GraphView.Query<Edge>().ToList())
{
edge.output = null;
edge.input = null;
}
m_GraphView.nodeCreationRequest = null;
m_GraphView = null;
}
m_BlackboardController?.Dispose();
m_BlackboardController = null;
m_InspectorView?.Dispose();
m_InspectorView = null;
if (previewManager != null)
{
previewManager.Dispose();
previewManager = null;
}
// Unload any static resources here
Resources.UnloadAsset(ShaderPort.styleSheet);
if (m_SearchWindowProvider != null)
{
m_SearchWindowProvider.Dispose();
m_SearchWindowProvider = null;
}
}
}
}