|
|
- // Amplify Shader Editor - Visual Shader Editing Tool
- // Copyright (c) Amplify Creations, Lda <info@amplify.pt>
-
- using UnityEngine;
- using UnityEditor;
- using System;
- using System.Reflection;
- using AmplifyShaderEditor;
-
-
- public static class MaterialPropertyHandlerEx
- {
- private static System.Type type = null;
- public static System.Type Type { get { return ( type == null ) ? type = System.Type.GetType( "UnityEditor.MaterialPropertyHandler, UnityEditor" ) : type; } }
- public static object GetHandler( Shader shader, string name )
- {
- return MaterialPropertyHandlerEx.Type.InvokeMember( "GetHandler", BindingFlags.Static | BindingFlags.NonPublic | BindingFlags.InvokeMethod, null, null, new object[] { shader, name } );
- }
-
- public static void OnGUI( object obj, ref Rect position, MaterialProperty prop, GUIContent label, MaterialEditor editor )
- {
- Type.InvokeMember( "OnGUI", BindingFlags.Instance | BindingFlags.Public | BindingFlags.InvokeMethod, null, obj, new object[] { position, prop, label, editor } );
- }
-
- public static float GetPropertyHeight( object obj, MaterialProperty prop, string label, MaterialEditor editor )
- {
- return (float)Type.InvokeMember( "GetPropertyHeight", BindingFlags.Instance | BindingFlags.Public | BindingFlags.InvokeMethod, null, obj, new object[] { prop, label, editor } );
- }
-
- public static object PropertyDrawer( object obj )
- {
- return Type.InvokeMember( "propertyDrawer", BindingFlags.Instance | BindingFlags.Public | BindingFlags.GetProperty, null, obj, new object[] {} );
- }
- }
-
- internal class ASEMaterialInspector : ShaderGUI
- {
- private const string CopyButtonStr = "Copy Values";
- private const string PasteButtonStr = "Paste Values";
- private const string PreviewModelPref = "ASEMI_PREVIEWMODEL";
-
- private static MaterialEditor m_instance = null;
- private static bool m_refreshOnUndo = false;
-
- private bool m_initialized = false;
- private double m_lastRenderedTime;
- private PreviewRenderUtility m_previewRenderUtility;
- private Mesh m_targetMesh;
- private Vector2 m_previewDir = new Vector2( 120f, -20f );
- private int m_selectedMesh = 0;
-
-
- // Reflection Fields
- private Type m_modelInspectorType = null;
- private MethodInfo m_renderMeshMethod = null;
- private Type m_previewGUIType = null;
- private MethodInfo m_dragMethod = null;
- private FieldInfo m_selectedField = null;
- private FieldInfo m_infoField = null;
-
- #if UNITY_2018_2_OR_NEWER
- public override void OnClosed( Material material )
- {
- base.OnClosed( material );
- CleanUp();
- }
- #endif
-
- void CleanUp()
- {
- if( m_previewRenderUtility != null )
- {
- m_previewRenderUtility.Cleanup();
- m_previewRenderUtility = null;
- }
- }
-
- void UndoRedoPerformed()
- {
- m_refreshOnUndo = true;
- }
-
- ~ASEMaterialInspector()
- {
- Undo.undoRedoPerformed -= UndoRedoPerformed;
- CleanUp();
- }
- public override void OnGUI( MaterialEditor materialEditor, MaterialProperty[] properties )
- {
- IOUtils.Init();
- Material mat = materialEditor.target as Material;
-
- if( mat == null )
- return;
-
- m_instance = materialEditor;
-
- if( !m_initialized )
- {
- Init();
- m_initialized = true;
- Undo.undoRedoPerformed += UndoRedoPerformed;
- }
-
- if( Event.current.type == EventType.Repaint &&
- mat.HasProperty( IOUtils.DefaultASEDirtyCheckId ) &&
- mat.GetInt( IOUtils.DefaultASEDirtyCheckId ) == 1 )
- {
- mat.SetInt( IOUtils.DefaultASEDirtyCheckId, 0 );
- UIUtils.ForceUpdateFromMaterial();
- //Event.current.Use();
- }
-
-
-
- if( materialEditor.isVisible )
- {
- GUILayout.BeginVertical();
- {
- GUILayout.Space( 3 );
- if( GUILayout.Button( "Open in Shader Editor" ) )
- {
- AmplifyShaderEditorWindow.LoadMaterialToASE( mat );
- }
-
- GUILayout.BeginHorizontal();
- {
- if( GUILayout.Button( CopyButtonStr ) )
- {
- System.Threading.Thread.CurrentThread.CurrentCulture = System.Globalization.CultureInfo.InvariantCulture;
-
- Shader shader = mat.shader;
- int propertyCount = UnityEditor.ShaderUtil.GetPropertyCount( shader );
- string allProperties = string.Empty;
- for( int i = 0; i < propertyCount; i++ )
- {
- UnityEditor.ShaderUtil.ShaderPropertyType type = UnityEditor.ShaderUtil.GetPropertyType( shader, i );
- string name = UnityEditor.ShaderUtil.GetPropertyName( shader, i );
- string valueStr = string.Empty;
- switch( type )
- {
- case UnityEditor.ShaderUtil.ShaderPropertyType.Color:
- {
- Color value = mat.GetColor( name );
- valueStr = value.r.ToString() + IOUtils.VECTOR_SEPARATOR +
- value.g.ToString() + IOUtils.VECTOR_SEPARATOR +
- value.b.ToString() + IOUtils.VECTOR_SEPARATOR +
- value.a.ToString();
- }
- break;
- case UnityEditor.ShaderUtil.ShaderPropertyType.Vector:
- {
- Vector4 value = mat.GetVector( name );
- valueStr = value.x.ToString() + IOUtils.VECTOR_SEPARATOR +
- value.y.ToString() + IOUtils.VECTOR_SEPARATOR +
- value.z.ToString() + IOUtils.VECTOR_SEPARATOR +
- value.w.ToString();
- }
- break;
- case UnityEditor.ShaderUtil.ShaderPropertyType.Float:
- {
- float value = mat.GetFloat( name );
- valueStr = value.ToString();
- }
- break;
- case UnityEditor.ShaderUtil.ShaderPropertyType.Range:
- {
- float value = mat.GetFloat( name );
- valueStr = value.ToString();
- }
- break;
- case UnityEditor.ShaderUtil.ShaderPropertyType.TexEnv:
- {
- Texture value = mat.GetTexture( name );
- valueStr = AssetDatabase.GetAssetPath( value );
- Vector2 offset = mat.GetTextureOffset( name );
- Vector2 scale = mat.GetTextureScale( name );
- valueStr += IOUtils.VECTOR_SEPARATOR + scale.x.ToString() +
- IOUtils.VECTOR_SEPARATOR + scale.y.ToString() +
- IOUtils.VECTOR_SEPARATOR + offset.x.ToString() +
- IOUtils.VECTOR_SEPARATOR + offset.y.ToString();
- }
- break;
- }
-
- allProperties += name + IOUtils.FIELD_SEPARATOR + type + IOUtils.FIELD_SEPARATOR + valueStr;
-
- if( i < ( propertyCount - 1 ) )
- {
- allProperties += IOUtils.LINE_TERMINATOR;
- }
- }
- EditorPrefs.SetString( IOUtils.MAT_CLIPBOARD_ID, allProperties );
- System.Threading.Thread.CurrentThread.CurrentCulture = System.Threading.Thread.CurrentThread.CurrentUICulture;
- }
-
- if( GUILayout.Button( PasteButtonStr ) )
- {
- System.Threading.Thread.CurrentThread.CurrentCulture = System.Globalization.CultureInfo.InvariantCulture;
- string propertiesStr = EditorPrefs.GetString( IOUtils.MAT_CLIPBOARD_ID, string.Empty );
- if( !string.IsNullOrEmpty( propertiesStr ) )
- {
- string[] propertyArr = propertiesStr.Split( IOUtils.LINE_TERMINATOR );
- bool validData = true;
- try
- {
- for( int i = 0; i < propertyArr.Length; i++ )
- {
- string[] valuesArr = propertyArr[ i ].Split( IOUtils.FIELD_SEPARATOR );
- if( valuesArr.Length != 3 )
- {
- Debug.LogWarning( "Material clipboard data is corrupted" );
- validData = false;
- break;
- }
- else if( mat.HasProperty( valuesArr[ 0 ] ) )
- {
- UnityEditor.ShaderUtil.ShaderPropertyType type = (UnityEditor.ShaderUtil.ShaderPropertyType)Enum.Parse( typeof( UnityEditor.ShaderUtil.ShaderPropertyType ), valuesArr[ 1 ] );
- switch( type )
- {
- case UnityEditor.ShaderUtil.ShaderPropertyType.Color:
- {
- string[] colorVals = valuesArr[ 2 ].Split( IOUtils.VECTOR_SEPARATOR );
- if( colorVals.Length != 4 )
- {
- Debug.LogWarning( "Material clipboard data is corrupted" );
- validData = false;
- break;
- }
- else
- {
- mat.SetColor( valuesArr[ 0 ], new Color( Convert.ToSingle( colorVals[ 0 ] ),
- Convert.ToSingle( colorVals[ 1 ] ),
- Convert.ToSingle( colorVals[ 2 ] ),
- Convert.ToSingle( colorVals[ 3 ] ) ) );
- }
- }
- break;
- case UnityEditor.ShaderUtil.ShaderPropertyType.Vector:
- {
- string[] vectorVals = valuesArr[ 2 ].Split( IOUtils.VECTOR_SEPARATOR );
- if( vectorVals.Length != 4 )
- {
- Debug.LogWarning( "Material clipboard data is corrupted" );
- validData = false;
- break;
- }
- else
- {
- mat.SetVector( valuesArr[ 0 ], new Vector4( Convert.ToSingle( vectorVals[ 0 ] ),
- Convert.ToSingle( vectorVals[ 1 ] ),
- Convert.ToSingle( vectorVals[ 2 ] ),
- Convert.ToSingle( vectorVals[ 3 ] ) ) );
- }
- }
- break;
- case UnityEditor.ShaderUtil.ShaderPropertyType.Float:
- {
- mat.SetFloat( valuesArr[ 0 ], Convert.ToSingle( valuesArr[ 2 ] ) );
- }
- break;
- case UnityEditor.ShaderUtil.ShaderPropertyType.Range:
- {
- mat.SetFloat( valuesArr[ 0 ], Convert.ToSingle( valuesArr[ 2 ] ) );
- }
- break;
- case UnityEditor.ShaderUtil.ShaderPropertyType.TexEnv:
- {
- string[] texVals = valuesArr[ 2 ].Split( IOUtils.VECTOR_SEPARATOR );
- if( texVals.Length != 5 )
- {
- Debug.LogWarning( "Material clipboard data is corrupted" );
- validData = false;
- break;
- }
- else
- {
- mat.SetTexture( valuesArr[ 0 ], AssetDatabase.LoadAssetAtPath<Texture>( texVals[ 0 ] ) );
- mat.SetTextureScale( valuesArr[ 0 ], new Vector2( Convert.ToSingle( texVals[ 1 ] ), Convert.ToSingle( texVals[ 2 ] ) ) );
- mat.SetTextureOffset( valuesArr[ 0 ], new Vector2( Convert.ToSingle( texVals[ 3 ] ), Convert.ToSingle( texVals[ 4 ] ) ) );
- }
- }
- break;
- }
- }
- }
- }
- catch( Exception e )
- {
- Debug.LogException( e );
- validData = false;
- }
-
-
- if( validData )
- {
- materialEditor.PropertiesChanged();
- UIUtils.CopyValuesFromMaterial( mat );
- }
- else
- {
- EditorPrefs.SetString( IOUtils.MAT_CLIPBOARD_ID, string.Empty );
- }
- }
- System.Threading.Thread.CurrentThread.CurrentCulture = System.Threading.Thread.CurrentThread.CurrentUICulture;
- }
- }
- GUILayout.EndHorizontal();
- GUILayout.Space( 5 );
- }
- GUILayout.EndVertical();
- }
- EditorGUI.BeginChangeCheck();
- //base.OnGUI( materialEditor, properties );
-
- // Draw custom properties instead of calling BASE to use single line texture properties
- materialEditor.SetDefaultGUIWidths();
-
- if( m_infoField == null )
- {
- m_infoField = typeof( MaterialEditor ).GetField( "m_InfoMessage", BindingFlags.Instance | BindingFlags.NonPublic );
- }
-
- string info = m_infoField.GetValue( materialEditor ) as string;
- if( !string.IsNullOrEmpty( info ) )
- {
- EditorGUILayout.HelpBox( info, MessageType.Info );
- }
- else
- {
- GUIUtility.GetControlID( "EditorTextField".GetHashCode(), FocusType.Passive, new Rect( 0f, 0f, 0f, 0f ) );
- }
-
- for( int i = 0; i < properties.Length; i++ )
- {
- if( ( properties[ i ].flags & ( MaterialProperty.PropFlags.HideInInspector | MaterialProperty.PropFlags.PerRendererData ) ) == MaterialProperty.PropFlags.None )
- {
- if( ( properties[ i ].flags & MaterialProperty.PropFlags.NoScaleOffset ) == MaterialProperty.PropFlags.NoScaleOffset )
- {
- object obj = MaterialPropertyHandlerEx.GetHandler( mat.shader, properties[ i ].name );
- if( obj != null )
- {
- float height = MaterialPropertyHandlerEx.GetPropertyHeight( obj, properties[ i ], properties[ i ].displayName, materialEditor );
- //Rect rect = (Rect)materialEditor.GetType().InvokeMember( "GetPropertyRect", BindingFlags.NonPublic | BindingFlags.Instance | BindingFlags.InvokeMethod, null, materialEditor, new object[] { properties[ i ], properties[ i ].displayName, true } );
- Rect rect = EditorGUILayout.GetControlRect( true, height, EditorStyles.layerMaskField );
- MaterialPropertyHandlerEx.OnGUI( obj, ref rect, properties[ i ], new GUIContent( properties[ i ].displayName ), materialEditor );
-
- if( MaterialPropertyHandlerEx.PropertyDrawer( obj ) != null )
- continue;
-
- rect = EditorGUILayout.GetControlRect( true, height, EditorStyles.layerMaskField );
- materialEditor.TexturePropertyMiniThumbnail( rect, properties[ i ], properties[ i ].displayName, string.Empty );
- }
- else
- {
- materialEditor.TexturePropertySingleLine( new GUIContent( properties[ i ].displayName ), properties[ i ] );
- }
- }
- else
- {
- float propertyHeight = materialEditor.GetPropertyHeight( properties[ i ], properties[ i ].displayName );
- Rect controlRect = EditorGUILayout.GetControlRect( true, propertyHeight, EditorStyles.layerMaskField, new GUILayoutOption[ 0 ] );
- materialEditor.ShaderProperty( controlRect, properties[ i ], properties[ i ].displayName );
- }
- }
- }
-
- EditorGUILayout.Space();
- materialEditor.RenderQueueField();
- #if UNITY_5_6_OR_NEWER
- materialEditor.EnableInstancingField();
- #endif
- #if UNITY_5_6_2 || UNITY_5_6_3 || UNITY_5_6_4 || UNITY_2017_1_OR_NEWER
- materialEditor.DoubleSidedGIField();
- #endif
- materialEditor.LightmapEmissionProperty();
- if( m_refreshOnUndo || EditorGUI.EndChangeCheck() )
- {
- m_refreshOnUndo = false;
-
- string isEmissive = mat.GetTag( "IsEmissive", false, "false" );
- if( isEmissive.Equals( "true" ) )
- {
- mat.globalIlluminationFlags &= (MaterialGlobalIlluminationFlags)3;
- }
- else
- {
- mat.globalIlluminationFlags |= MaterialGlobalIlluminationFlags.EmissiveIsBlack;
- }
-
- UIUtils.CopyValuesFromMaterial( mat );
- }
-
- if( materialEditor.RequiresConstantRepaint() && m_lastRenderedTime + 0.032999999821186066 < EditorApplication.timeSinceStartup )
- {
- this.m_lastRenderedTime = EditorApplication.timeSinceStartup;
- materialEditor.Repaint();
- }
- }
-
- private void Init()
- {
- string guid = EditorPrefs.GetString( PreviewModelPref, "" );
- if( !string.IsNullOrEmpty( guid ) )
- {
- m_targetMesh = AssetDatabase.LoadAssetAtPath<Mesh>( AssetDatabase.GUIDToAssetPath( guid ) );
- }
- }
-
- public override void OnMaterialPreviewSettingsGUI( MaterialEditor materialEditor )
- {
-
- base.OnMaterialPreviewSettingsGUI( materialEditor );
-
- if( UnityEditor.ShaderUtil.hardwareSupportsRectRenderTexture )
- {
- EditorGUI.BeginChangeCheck();
- m_targetMesh = (Mesh)EditorGUILayout.ObjectField( m_targetMesh, typeof( Mesh ), false, GUILayout.MaxWidth( 120 ) );
- if( EditorGUI.EndChangeCheck() )
- {
- if( m_targetMesh != null )
- {
- EditorPrefs.SetString( PreviewModelPref, AssetDatabase.AssetPathToGUID( AssetDatabase.GetAssetPath( m_targetMesh ) ) );
- }
- else
- {
- EditorPrefs.SetString( PreviewModelPref, "" );
- }
- }
-
- if( m_selectedField == null )
- {
- m_selectedField = typeof( MaterialEditor ).GetField( "m_SelectedMesh", BindingFlags.Instance | BindingFlags.NonPublic );
- }
-
- m_selectedMesh = (int)m_selectedField.GetValue( materialEditor );
-
- if( m_selectedMesh != 0 )
- {
- if( m_targetMesh != null )
- {
- m_targetMesh = null;
- EditorPrefs.SetString( PreviewModelPref, "" );
- }
- }
- }
- }
-
- public override void OnMaterialInteractivePreviewGUI( MaterialEditor materialEditor, Rect r, GUIStyle background )
- {
- if( Event.current.type == EventType.DragExited )
- {
- if( DragAndDrop.objectReferences.Length > 0 )
- {
- GameObject dropped = DragAndDrop.objectReferences[ 0 ] as GameObject;
- if( dropped != null )
- {
- m_targetMesh = AssetDatabase.LoadAssetAtPath<Mesh>( AssetDatabase.GetAssetPath( dropped ) );
- EditorPrefs.SetString( PreviewModelPref, AssetDatabase.AssetPathToGUID( AssetDatabase.GetAssetPath( m_targetMesh ) ) );
- }
- }
- }
-
- if( m_targetMesh == null )
- {
- base.OnMaterialInteractivePreviewGUI( materialEditor, r, background );
- return;
- }
-
- Material mat = materialEditor.target as Material;
-
- if( m_previewRenderUtility == null )
- {
- m_previewRenderUtility = new PreviewRenderUtility();
- #if UNITY_2017_1_OR_NEWER
- m_previewRenderUtility.cameraFieldOfView = 30f;
- #else
- m_previewRenderUtility.m_CameraFieldOfView = 30f;
- #endif
- }
-
- if( m_previewGUIType == null )
- {
- m_previewGUIType = Type.GetType( "PreviewGUI, UnityEditor, Version=0.0.0.0, Culture=neutral, PublicKeyToken=null" );
- m_dragMethod = m_previewGUIType.GetMethod( "Drag2D", BindingFlags.Static | BindingFlags.Public );
- }
-
- if( m_modelInspectorType == null )
- {
- m_modelInspectorType = Type.GetType( "UnityEditor.ModelInspector, UnityEditor, Version=0.0.0.0, Culture=neutral, PublicKeyToken=null" );
- m_renderMeshMethod = m_modelInspectorType.GetMethod( "RenderMeshPreview", BindingFlags.Static | BindingFlags.NonPublic );
- }
-
- m_previewDir = (Vector2)m_dragMethod.Invoke( m_previewGUIType, new object[] { m_previewDir, r } );
-
- if( Event.current.type == EventType.Repaint )
- {
- m_previewRenderUtility.BeginPreview( r, background );
- m_renderMeshMethod.Invoke( m_modelInspectorType, new object[] { m_targetMesh, m_previewRenderUtility, mat, null, m_previewDir, -1 } );
- m_previewRenderUtility.EndAndDrawPreview( r );
- }
- }
-
- public static MaterialEditor Instance { get { return m_instance; } set { m_instance = value; } }
- }
|