diff options
author | chai <chaifix@163.com> | 2020-10-23 13:08:43 +0800 |
---|---|---|
committer | chai <chaifix@163.com> | 2020-10-23 13:08:43 +0800 |
commit | b82da95b5181ac8bbae38efb13e950d5e88a4caa (patch) | |
tree | 48a6f3269276484bbc7cfc95f0651f40a2176aa1 /Assets/AmplifyShaderEditor/Plugins/Editor/Menu/GraphContextMenu.cs | |
parent | 917e9e0b320775634dc2e710f7deac74fd0822f0 (diff) |
*移动amplify shader editor到third party目录
Diffstat (limited to 'Assets/AmplifyShaderEditor/Plugins/Editor/Menu/GraphContextMenu.cs')
-rw-r--r-- | Assets/AmplifyShaderEditor/Plugins/Editor/Menu/GraphContextMenu.cs | 375 |
1 files changed, 0 insertions, 375 deletions
diff --git a/Assets/AmplifyShaderEditor/Plugins/Editor/Menu/GraphContextMenu.cs b/Assets/AmplifyShaderEditor/Plugins/Editor/Menu/GraphContextMenu.cs deleted file mode 100644 index adcbfe4e..00000000 --- a/Assets/AmplifyShaderEditor/Plugins/Editor/Menu/GraphContextMenu.cs +++ /dev/null @@ -1,375 +0,0 @@ -// Amplify Shader Editor - Visual Shader Editing Tool -// Copyright (c) Amplify Creations, Lda <info@amplify.pt> - -using UnityEngine; -using UnityEditor; -using System; -using System.Text; -using System.Linq; -using System.Collections.Generic; -using System.Reflection; - -namespace AmplifyShaderEditor -{ - public class ShortcutKeyData - { - public bool IsPressed; - public System.Type NodeType; - public string Name; - public ShortcutKeyData( System.Type type, string name ) - { - NodeType = type; - Name = name; - IsPressed = false; - } - } - - public class GraphContextMenu - { - private List<ContextMenuItem> m_items; - private List<ContextMenuItem> m_itemFunctions; - private Dictionary<System.Type, NodeAttributes> m_itemsDict; - private Dictionary<System.Type, NodeAttributes> m_deprecatedItemsDict; - private Dictionary<System.Type, System.Type> m_castTypes; - private Dictionary<KeyCode, ShortcutKeyData> m_shortcutTypes; - - private KeyCode m_lastKeyPressed; - private ParentGraph m_currentGraph; - private bool m_correctlyLoaded = false; - - public GraphContextMenu( ParentGraph currentGraph ) - { - m_currentGraph = currentGraph; - m_correctlyLoaded = RefreshNodes( currentGraph ); - } - - private Type[] GetTypesInNamespace( Assembly assembly, string nameSpace ) - { - return assembly.GetTypes().Where( t => String.Equals( t.Namespace, nameSpace, StringComparison.Ordinal ) ).ToArray(); - } - - public bool RefreshNodes( ParentGraph currentGraph ) - { - if( m_items != null ) - { - m_items.Clear(); - m_items = null; - } - - if( m_itemFunctions != null ) - { - m_itemFunctions.Clear(); - m_itemFunctions = null; - } - - m_items = new List<ContextMenuItem>(); - m_itemFunctions = new List<ContextMenuItem>(); - - if( m_itemsDict != null ) - m_itemsDict.Clear(); - - m_itemsDict = new Dictionary<System.Type, NodeAttributes>(); - - if( m_deprecatedItemsDict != null ) - m_deprecatedItemsDict.Clear(); - - m_deprecatedItemsDict = new Dictionary<System.Type, NodeAttributes>(); - - if( m_castTypes != null ) - m_castTypes.Clear(); - - m_castTypes = new Dictionary<System.Type, System.Type>(); - - if( m_shortcutTypes != null ) - m_shortcutTypes.Clear(); - - m_shortcutTypes = new Dictionary<KeyCode, ShortcutKeyData>(); - - m_lastKeyPressed = KeyCode.None; - - // Fetch all available nodes by their attributes - try - { - //IEnumerable<System.Type> availableTypes = AppDomain.CurrentDomain.GetAssemblies().ToList().SelectMany( type => type.GetTypes() ); - Type[] availableTypes = GetTypesInNamespace( Assembly.GetExecutingAssembly(), "AmplifyShaderEditor" ); - foreach( System.Type type in availableTypes ) - { - foreach( NodeAttributes attribute in Attribute.GetCustomAttributes( type ).OfType<NodeAttributes>() ) - { - if( attribute.Available && !attribute.Deprecated ) - { - //if ( !UIUtils.CurrentWindow.IsShaderFunctionWindow && attribute.AvailableInFunctionsOnly ) - // continue; - - if( !UIUtils.HasColorCategory( attribute.Category ) ) - { - if( !String.IsNullOrEmpty( attribute.CustomCategoryColor ) ) - { - try - { - Color color = new Color(); - ColorUtility.TryParseHtmlString( attribute.CustomCategoryColor, out color ); - UIUtils.AddColorCategory( attribute.Category, color ); - } - catch( Exception e ) - { - Debug.LogException( e ); - UIUtils.AddColorCategory( attribute.Category, Constants.DefaultCategoryColor ); - } - } - //else - //{ - // UIUtils.AddColorCategory( attribute.Category, Constants.DefaultCategoryColor ); - //} - } - - if( attribute.CastType != null && attribute.CastType.Length > 0 && type != null ) - { - for( int i = 0; i < attribute.CastType.Length; i++ ) - { - m_castTypes.Add( attribute.CastType[ i ], type ); - } - } - - if( attribute.ShortcutKey != KeyCode.None && type != null ) - m_shortcutTypes.Add( attribute.ShortcutKey, new ShortcutKeyData( type, attribute.Name ) ); - - ContextMenuItem newItem = new ContextMenuItem( attribute, type, attribute.Name, attribute.Tags, attribute.Category, attribute.Description, null, attribute.ShortcutKey ); - if( UIUtils.GetNodeAvailabilityInBitArray( attribute.NodeAvailabilityFlags, NodeAvailability.SurfaceShader ) ) - m_items.Add( newItem ); - else if( UIUtils.GetNodeAvailabilityInBitArray( attribute.NodeAvailabilityFlags, currentGraph.ParentWindow.CurrentNodeAvailability ) ) - m_items.Add( newItem ); - else if( UIUtils.GetNodeAvailabilityInBitArray( attribute.NodeAvailabilityFlags, currentGraph.CurrentCanvasMode ) ) - m_items.Add( newItem ); - - m_itemsDict.Add( type, attribute ); - m_itemFunctions.Add( newItem ); - } - else - { - m_deprecatedItemsDict.Add( type, attribute ); - } - } - } - } - catch( ReflectionTypeLoadException exception ) - { - Debug.LogException( exception ); - return false; - } - - string[] guids = AssetDatabase.FindAssets( "t:AmplifyShaderFunction" ); - List<AmplifyShaderFunction> allFunctions = new List<AmplifyShaderFunction>(); - - for( int i = 0; i < guids.Length; i++ ) - { - allFunctions.Add( AssetDatabase.LoadAssetAtPath<AmplifyShaderFunction>( AssetDatabase.GUIDToAssetPath( guids[ i ] ) ) ); - } - - int functionCount = allFunctions.Count; - if( functionCount > 0 ) - { - m_castTypes.Add( typeof( AmplifyShaderFunction ), typeof( FunctionNode ) ); - } - - for( int i = 0; i < functionCount; i++ ) - { - if( !allFunctions[ i ].Hidden ) - { - NodeAttributes attribute = new NodeAttributes( allFunctions[ i ].FunctionName, allFunctions[ i ].CustomNodeCategory, allFunctions[ i ].Description, KeyCode.None, true, 0, int.MaxValue, typeof( AmplifyShaderFunction ) ); - System.Type type = typeof( FunctionNode ); - - ContextMenuItem newItem = new ContextMenuItem( attribute, type, AddSpacesToSentence( attribute.Name ), attribute.Tags, attribute.Category, attribute.Description, allFunctions[ i ], attribute.ShortcutKey ); - m_items.Add( newItem ); - m_itemFunctions.Add( newItem ); - } - } - - //Sort out the final list by name - m_items.Sort( ( x, y ) => x.Category.CompareTo( y.Category ) ); - m_itemFunctions.Sort( ( x, y ) => x.Category.CompareTo( y.Category ) ); - return true; - } - - public void Destroy() - { - for( int i = 0; i < m_items.Count; i++ ) - { - m_items[ i ].Destroy(); - } - - for( int i = 0; i < m_itemFunctions.Count; i++ ) - { - if( m_itemFunctions[ i ] != null ) - m_itemFunctions[ i ].Destroy(); - } - - m_items.Clear(); - m_items = null; - - m_itemFunctions.Clear(); - m_itemFunctions = null; - - m_itemsDict.Clear(); - m_itemsDict = null; - - m_deprecatedItemsDict.Clear(); - m_deprecatedItemsDict = null; - - m_castTypes.Clear(); - m_castTypes = null; - - m_shortcutTypes.Clear(); - m_shortcutTypes = null; - - } - - public static string AddSpacesToSentence( string text ) - { - if( string.IsNullOrEmpty( text ) ) - return string.Empty; - - bool lastIsUpper = char.IsUpper( text, 0 ); - bool lastIsLetter = char.IsLetter( text, 0 ); - StringBuilder title = new StringBuilder(); - title.Append( text[ 0 ] ); - for( int i = 1; i < text.Length; i++ ) - { - bool currIsUpper = char.IsUpper( text, i ); - bool currIsLetter = char.IsLetter( text, i ); - if( currIsUpper && !lastIsUpper && lastIsLetter ) - { - title.Append( " " ); - } - - // if current is a number and previous is a letter we space it (ie: Rotation2D = Rotation 2D) - if( lastIsLetter && char.IsNumber( text, i ) ) - { - title.Append( " " ); - } - - // if previous is upper, current is upper and the next two following are lower then we space it (ie: UVDistortion = UV Distortion) - if( i < text.Length - 1 ) - { - bool nextIsLower = char.IsLower( text, i + 1 ) && char.IsLetter( text, i + 1 ); - bool lastIsLower = i < text.Length - 2 ? char.IsLower( text, i + 2 ) && char.IsLetter( text, i + 2 ) : false; - if( lastIsUpper && currIsUpper && currIsLetter && nextIsLower && lastIsLower ) - { - title.Append( " " ); - } - } - lastIsUpper = currIsUpper; - lastIsLetter = currIsLetter; - title.Append( text[ i ] ); - } - return title.ToString(); - } - - public NodeAttributes GetNodeAttributesForType( System.Type type ) - { - if( type == null ) - { - Debug.LogError( "Invalid type detected" ); - return null; - } - - if( m_itemsDict.ContainsKey( type ) ) - return m_itemsDict[ type ]; - return null; - } - - public NodeAttributes GetDeprecatedNodeAttributesForType( System.Type type ) - { - if( m_deprecatedItemsDict.ContainsKey( type ) ) - return m_deprecatedItemsDict[ type ]; - return null; - } - - public void UpdateKeyPress( KeyCode key ) - { - if( key == KeyCode.None ) - return; - - m_lastKeyPressed = key; - if( m_shortcutTypes.ContainsKey( key ) ) - { - m_shortcutTypes[ key ].IsPressed = true; - } - } - - public void UpdateKeyReleased( KeyCode key ) - { - if( key == KeyCode.None ) - return; - - if( m_shortcutTypes.ContainsKey( key ) ) - { - m_shortcutTypes[ key ].IsPressed = false; - } - } - - public void ResetShortcutKeyStates() - { - foreach( KeyValuePair<KeyCode, ShortcutKeyData> kvp in m_shortcutTypes ) - { - kvp.Value.IsPressed = false; - } - } - - public ParentNode CreateNodeFromCastType( System.Type type ) - { - if( m_castTypes.ContainsKey( type ) ) - { - ParentNode newNode = (ParentNode)ScriptableObject.CreateInstance( m_castTypes[ type ] ); - return newNode; - } - return null; - } - - - public ParentNode CreateNodeFromShortcutKey() - { - if( m_lastKeyPressed == KeyCode.None ) - return null; - - if( m_shortcutTypes.ContainsKey( m_lastKeyPressed ) && m_shortcutTypes[ m_lastKeyPressed ].IsPressed ) - { - ParentNode newNode = (ParentNode)ScriptableObject.CreateInstance( m_shortcutTypes[ m_lastKeyPressed ].NodeType ); - return newNode; - } - return null; - } - - public bool CheckShortcutKey() - { - if( m_lastKeyPressed == KeyCode.None ) - return false; - - if( m_shortcutTypes.ContainsKey( m_lastKeyPressed ) && m_shortcutTypes[ m_lastKeyPressed ].IsPressed ) - { - return true; - } - return false; - } - - public List<ContextMenuItem> MenuItems - { - get - { - if( m_currentGraph.ParentWindow.IsShaderFunctionWindow ) - return m_itemFunctions; - else - return m_items; - } - } - - public List<ContextMenuItem> ItemFunctions { get { return m_itemFunctions; } } - public KeyCode LastKeyPressed - { - get { return m_lastKeyPressed; } - } - - public Dictionary<KeyCode, ShortcutKeyData> NodeShortcuts { get { return m_shortcutTypes; } } - public bool CorrectlyLoaded { get { return m_correctlyLoaded; } } - } -} |