using System; using System.Collections.Generic; using System.Linq; using System.Reflection; using UnityEngine; using UnityEditor.Experimental.UIElements; using UnityEngine.Experimental.UIElements; using UnityEditor.Graphing; using UnityEditor.Graphing.Util; using UnityEngine.Experimental.UIElements.StyleSheets; using Object = UnityEngine.Object; #if UNITY_2018_3_OR_NEWER using ContextualMenu = UnityEngine.Experimental.UIElements.DropdownMenu; #endif namespace UnityEditor.ShaderGraph.Drawing.Inspector { public class MasterPreviewView : VisualElement { PreviewManager m_PreviewManager; AbstractMaterialGraph m_Graph; PreviewRenderData m_PreviewRenderHandle; Image m_PreviewTextureView; public Image previewTextureView { get { return m_PreviewTextureView; } } Vector2 m_PreviewScrollPosition; ObjectField m_PreviewMeshPicker; IMasterNode m_MasterNode; Mesh m_PreviousMesh; bool m_Expanded = true; public bool expanded { get { return m_Expanded; } } bool m_RecalculateLayout; Vector2 m_ExpandedPreviewSize; VisualElement m_CollapsePreviewContainer; VisualElement m_CollapsePreviewButton; ResizeBorderFrame m_PreviewResizeBorderFrame; public ResizeBorderFrame previewResizeBorderFrame { get { return m_PreviewResizeBorderFrame; } } VisualElement m_Preview; Label m_Title; public VisualElement preview { get { return m_Preview; } } List m_DoNotShowPrimitives = new List(new string[] {PrimitiveType.Plane.ToString()}); static Type s_ContextualMenuManipulator = AppDomain.CurrentDomain.GetAssemblies().SelectMany(x => x.GetTypesOrNothing()).FirstOrDefault(t => t.FullName == "UnityEngine.Experimental.UIElements.ContextualMenuManipulator"); static Type s_ObjectSelector = AppDomain.CurrentDomain.GetAssemblies().SelectMany(x => x.GetTypesOrNothing()).FirstOrDefault(t => t.FullName == "UnityEditor.ObjectSelector"); public string assetName { get { return m_Title.text; } set { m_Title.text = value; } } public MasterPreviewView(PreviewManager previewManager, AbstractMaterialGraph graph) { this.clippingOptions = ClippingOptions.ClipAndCacheContents; m_PreviewManager = previewManager; m_Graph = graph; AddStyleSheetPath("Styles/MasterPreviewView"); m_PreviewRenderHandle = previewManager.masterRenderData; var topContainer = new VisualElement() { name = "top" }; { m_Title = new Label() { name = "title" }; // Add preview collapse button on top of preview m_CollapsePreviewContainer = new VisualElement { name = "collapse-container" }; m_CollapsePreviewContainer.AddToClassList("collapse-container"); m_CollapsePreviewButton = new VisualElement { name = "icon" }; m_CollapsePreviewButton.AddToClassList("icon"); m_CollapsePreviewContainer.Add(m_CollapsePreviewButton); m_CollapsePreviewContainer.AddManipulator(new Clickable(() => { m_Graph.owner.RegisterCompleteObjectUndo("Collapse Preview"); m_Expanded ^= true; UpdateExpandedButtonState(); UpdatePreviewVisibility(); })); topContainer.Add(m_Title); topContainer.Add(m_CollapsePreviewContainer); } Add(topContainer); m_Preview = new VisualElement {name = "middle"}; { m_PreviewTextureView = CreatePreview(Texture2D.blackTexture); m_PreviewScrollPosition = new Vector2(0f, 0f); preview.Add(m_PreviewTextureView); preview.AddManipulator(new Scrollable(OnScroll)); } m_PreviewRenderHandle.onPreviewChanged += OnPreviewChanged; Add(preview); m_PreviewResizeBorderFrame = new ResizeBorderFrame(previewTextureView, this) { name = "resizeBorderFrame" }; m_PreviewResizeBorderFrame.maintainAspectRatio = true; Add(m_PreviewResizeBorderFrame); m_ExpandedPreviewSize = new Vector2(256f, 256f); m_RecalculateLayout = false; previewTextureView.RegisterCallback(OnGeometryChanged); } void UpdateExpandedButtonState() { m_CollapsePreviewButton.RemoveFromClassList(!m_Expanded ? "expanded" : "collapsed"); m_CollapsePreviewButton.AddToClassList(!m_Expanded ? "collapsed" : "expanded"); } void UpdatePreviewVisibility() { if (m_Expanded) { RemoveFromClassList("collapsed"); AddToClassList("expanded"); m_PreviewResizeBorderFrame.visible = true; style.positionLeft = StyleValue.Create(float.NaN); style.positionBottom = StyleValue.Create(float.NaN); style.positionRight = StyleValue.Create(parent.layout.width - layout.xMax); style.positionTop = StyleValue.Create(layout.yMin); previewTextureView.style.width = StyleValue.Create(m_ExpandedPreviewSize.x); previewTextureView.style.height = StyleValue.Create(m_ExpandedPreviewSize.y); } else { m_ExpandedPreviewSize = previewTextureView.layout.size; m_PreviewResizeBorderFrame.visible = false; style.positionLeft = StyleValue.Create(float.NaN); style.positionBottom = StyleValue.Create(float.NaN); style.positionRight = StyleValue.Create(parent.layout.width - layout.xMax); style.positionTop = StyleValue.Create(layout.yMin); previewTextureView.style.width = StyleValue.Create(0f); previewTextureView.style.height = StyleValue.Create(0f); RemoveFromClassList("expanded"); AddToClassList("collapsed"); } m_RecalculateLayout = true; } Image CreatePreview(Texture texture) { var image = new Image { name = "preview", image = StyleValue.Create(m_PreviewRenderHandle.texture ?? texture) }; image.AddManipulator(new Draggable(OnMouseDragPreviewMesh, true)); image.AddManipulator((IManipulator)Activator.CreateInstance(s_ContextualMenuManipulator, (Action)BuildContextualMenu)); return image; } void BuildContextualMenu(ContextualMenuPopulateEvent evt) { foreach (var primitiveTypeName in Enum.GetNames(typeof(PrimitiveType))) { if (m_DoNotShowPrimitives.Contains(primitiveTypeName)) continue; evt.menu.AppendAction(primitiveTypeName, e => ChangePrimitiveMesh(primitiveTypeName), ContextualMenu.MenuAction.AlwaysEnabled); } evt.menu.AppendAction("Custom Mesh", e => ChangeMeshCustom(), ContextualMenu.MenuAction.AlwaysEnabled); } IMasterNode masterNode { get { return m_PreviewRenderHandle.shaderData != null ? m_PreviewRenderHandle.shaderData.node as IMasterNode : null; } } void DirtyMasterNode(ModificationScope scope) { var amn = masterNode as AbstractMaterialNode; if (amn != null) amn.Dirty(scope); // If currently editing a subgraph, dirty the output node rather than master node. if (m_Graph is SubGraph) { var subgraph = m_Graph as SubGraph; if (subgraph != null && subgraph.outputNode != null) subgraph.outputNode.Dirty(scope); } } void OnPreviewChanged() { m_PreviewTextureView.image = StyleValue.Create(m_PreviewRenderHandle.texture ?? Texture2D.blackTexture); m_PreviewTextureView.MarkDirtyRepaint(); } void ChangePrimitiveMesh(string primitiveName) { Mesh changedPrimitiveMesh = Resources.GetBuiltinResource(typeof(Mesh), string.Format("{0}.fbx", primitiveName)) as Mesh; ChangeMesh(changedPrimitiveMesh); } void ChangeMesh(Mesh mesh) { Mesh changedMesh = mesh; DirtyMasterNode(ModificationScope.Node); if (m_Graph.previewData.serializedMesh.mesh != changedMesh) { m_Graph.previewData.rotation = Quaternion.identity; } m_Graph.previewData.serializedMesh.mesh = changedMesh; } private static EditorWindow Get() { PropertyInfo P = s_ObjectSelector.GetProperty("get", BindingFlags.Public | BindingFlags.Static); return P.GetValue(null, null) as EditorWindow; } void OnMeshChanged(Object obj) { var mesh = obj as Mesh; if (mesh == null) mesh = m_PreviousMesh; ChangeMesh(mesh); } void ChangeMeshCustom() { MethodInfo ShowMethod = s_ObjectSelector.GetMethod("Show", BindingFlags.NonPublic | BindingFlags.Instance | BindingFlags.DeclaredOnly, Type.DefaultBinder, new[] {typeof(Object), typeof(Type), typeof(SerializedProperty), typeof(bool), typeof(List), typeof(Action), typeof(Action)}, new ParameterModifier[7]); m_PreviousMesh = m_Graph.previewData.serializedMesh.mesh; ShowMethod.Invoke(Get(), new object[] { null, typeof(Mesh), null, false, null, (Action)OnMeshChanged, (Action)OnMeshChanged }); } void OnGeometryChanged(GeometryChangedEvent evt) { if (m_RecalculateLayout) { WindowDockingLayout dockingLayout = new WindowDockingLayout(); dockingLayout.CalculateDockingCornerAndOffset(layout, parent.layout); dockingLayout.ClampToParentWindow(); dockingLayout.ApplyPosition(this); m_RecalculateLayout = false; } if (!expanded) return; var currentWidth = m_PreviewRenderHandle.texture != null ? m_PreviewRenderHandle.texture.width : -1; var currentHeight = m_PreviewRenderHandle.texture != null ? m_PreviewRenderHandle.texture.height : -1; var targetWidth = Mathf.Max(1f, m_PreviewTextureView.contentRect.width); var targetHeight = Mathf.Max(1f, m_PreviewTextureView.contentRect.height); if (Mathf.Approximately(currentWidth, targetHeight) && Mathf.Approximately(currentHeight, targetWidth)) return; m_PreviewManager.ResizeMasterPreview(new Vector2(targetWidth, targetHeight)); } void OnScroll(float scrollValue) { float rescaleAmount = -scrollValue * .03f; m_Graph.previewData.scale = Mathf.Clamp(m_Graph.previewData.scale + rescaleAmount, 0.2f, 5f); DirtyMasterNode(ModificationScope.Node); } void OnMouseDragPreviewMesh(Vector2 deltaMouse) { Vector2 previewSize = m_PreviewTextureView.contentRect.size; m_PreviewScrollPosition -= deltaMouse * (Event.current.shift ? 3f : 1f) / Mathf.Min(previewSize.x, previewSize.y) * 140f; m_PreviewScrollPosition.y = Mathf.Clamp(m_PreviewScrollPosition.y, -90f, 90f); Quaternion previewRotation = Quaternion.Euler(m_PreviewScrollPosition.y, 0, 0) * Quaternion.Euler(0, m_PreviewScrollPosition.x, 0); m_Graph.previewData.rotation = previewRotation; DirtyMasterNode(ModificationScope.Node); } } }