|
|
- // Amplify Shader Editor - Visual Shader Editing Tool
- // Copyright (c) Amplify Creations, Lda <info@amplify.pt>
-
- //https://www.shadertoy.com/view/XdS3RW
- //http://www.deepskycolors.com/archivo/2010/04/21/formulas-for-Photoshop-blending-modes.html
- //http://www.pegtop.net/delphi/articles/blendmodes/softlight.htm
-
- using UnityEngine;
- using UnityEditor;
- using System;
-
- namespace AmplifyShaderEditor
- {
- public enum BlendOps
- {
- ColorBurn,
- ColorDodge,
- Darken,
- Divide,
- Difference,
- Exclusion,
- SoftLight,
- HardLight,
- HardMix,
- Lighten,
- LinearBurn,
- LinearDodge,
- LinearLight,
- Multiply,
- Overlay,
- PinLight,
- Subtract,
- Screen,
- VividLight
- }
- [Serializable]
- [NodeAttributes( "Blend Operations", "Image Effects", "Common layer blending modes" )]
- public class BlendOpsNode : ParentNode
- {
- //private const string ASEHardLightCall = "ASEHardLight({0},{1})";
- //private const string ASEHardLightFunc =
- //"inline float ASEHardLight( float srcLocalVar, float dstLocalVar ){" +
- //" return ( ( srcLocalVar > 0.5 ) ? ( 1.0 - ( 1.0 - 2.0 * ( srcLocalVar - 0.5 ) ) * ( 1.0 - dstLocalVar ) ) : ( 2.0 * srcLocalVar * dstLocalVar ) ); }";
-
- //private const string ASELinearLightCall = "ASELinearLight({0},{1})";
- //private const string ASELinearLightFunc =
- //"inline float ASELinearLight( float srcLocalVar, float dstLocalVar ){" +
- //" return ( ( srcLocalVar > 0.5 ) ? ( dstLocalVar + 2.0 * srcLocalVar - 1.0 ) : ( dstLocalVar + 2.0 * ( srcLocalVar - 0.5 ) ) ); }";
-
- //private const string ASEOverlayCall = "ASEOverlay({0},{1})";
- //private const string ASEOverlayFunc =
- //"inline float ASEOverlay( float srcLocalVar, float dstLocalVar ){" +
- //" return ( ( dstLocalVar > 0.5 ) ? ( 1.0 - ( 1.0 - 2.0 * ( dstLocalVar - 0.5 ) ) * ( 1.0 - srcLocalVar ) ) : ( 2.0 * dstLocalVar * srcLocalVar ) ); }";
- ////" return (dstLocalVar < 0.5) ? 2.0 * srcLocalVar * dstLocalVar : 1.0 - 2.0 * (1.0 - srcLocalVar) * (1.0 - dstLocalVar); }";
-
- //private const string ASEPinLightCall = "ASEPinLight({0},{1})";
- //private const string ASEPinLightFunc =
- //"inline float ASEPinLight( float srcLocalVar, float dstLocalVar ){" +
- //" return ( ( srcLocalVar > 0.5 ) ? max( dstLocalVar , 2.0 * ( srcLocalVar - 0.5 ) ) : min( dstLocalVar , 2.0 * srcLocalVar ) ); }";
-
- //private const string ASEVividLightCall = "ASEVividLight({0},{1})";
- //private const string ASEVividLightFunc = "inline float ASEVividLight( float srcLocalVar, float dstLocalVar ){" +
- //" return ( ( srcLocalVar > 0.5 ) ? ( dstLocalVar / ( ( 1.0 - srcLocalVar ) * 2.0 ) ) : ( 1.0 - ( ( ( 1.0 - dstLocalVar ) * 0.5 ) / srcLocalVar ) ) ); }";
-
- private const string ASEDarkerColorCall = "ASEDarkerColor{}({0},{1})";
- private const string ASEDarkerColorFunc = "inline float ASEDarkerColor{0}( float srcLocalVar, float dstLocalVar ){" +
- " return ({1} < {2}) ? s : d; }";
-
- private const string ASELighterColorCall = "ASELighterColor{}({0},{1})";
- private const string ASELighterColorFunc = "inline float ASELighterColor{0}( float srcLocalVar, float dstLocalVar ){" +
- " return ({1} > {2}) ? s : d; }";
-
- private const string BlendOpsModeStr = "Blend Op";
- private const string SaturateResultStr = "Saturate";
-
- [SerializeField]
- private BlendOps m_currentBlendOp = BlendOps.ColorBurn;
-
- [SerializeField]
- private WirePortDataType m_mainDataType = WirePortDataType.COLOR;
-
- [SerializeField]
- private bool m_saturate = true;
-
- private UpperLeftWidgetHelper m_upperLeftWidget = new UpperLeftWidgetHelper();
-
- protected override void CommonInit( int uniqueId )
- {
- base.CommonInit( uniqueId );
- AddInputPort( WirePortDataType.COLOR, false, "Source" );
- AddInputPort( WirePortDataType.COLOR, false, "Destiny" );
- AddOutputPort( WirePortDataType.COLOR, Constants.EmptyPortValue );
- m_inputPorts[ 0 ].AddPortForbiddenTypes( WirePortDataType.FLOAT3x3,
- WirePortDataType.FLOAT4x4,
- WirePortDataType.SAMPLER1D,
- WirePortDataType.SAMPLER2D,
- WirePortDataType.SAMPLER3D,
- WirePortDataType.SAMPLERCUBE );
- m_inputPorts[ 1 ].AddPortForbiddenTypes( WirePortDataType.FLOAT3x3,
- WirePortDataType.FLOAT4x4,
- WirePortDataType.SAMPLER1D,
- WirePortDataType.SAMPLER2D,
- WirePortDataType.SAMPLER3D,
- WirePortDataType.SAMPLERCUBE );
- m_textLabelWidth = 75;
- m_autoWrapProperties = true;
- m_hasLeftDropdown = true;
- SetAdditonalTitleText( string.Format( Constants.SubTitleTypeFormatStr, m_currentBlendOp ) );
- m_previewShaderGUID = "6d6b3518705b3ba49acdc6e18e480257";
- }
-
- public override void SetPreviewInputs()
- {
- base.SetPreviewInputs();
-
- m_previewMaterialPassId = (int)m_currentBlendOp;
- PreviewMaterial.SetInt( "_Sat", m_saturate ? 1 : 0 );
- }
-
- public override void AfterCommonInit()
- {
- base.AfterCommonInit();
-
- if( PaddingTitleLeft == 0 )
- {
- PaddingTitleLeft = Constants.PropertyPickerWidth + Constants.IconsLeftRightMargin;
- if( PaddingTitleRight == 0 )
- PaddingTitleRight = Constants.PropertyPickerWidth + Constants.IconsLeftRightMargin;
- }
- }
-
- public override void OnInputPortConnected( int portId, int otherNodeId, int otherPortId, bool activateNode = true )
- {
- base.OnInputPortConnected( portId, otherNodeId, otherPortId, activateNode );
- UpdateConnection( portId );
- }
-
- public override void OnConnectedOutputNodeChanges( int inputPortId, int otherNodeId, int otherPortId, string name, WirePortDataType type )
- {
- base.OnConnectedOutputNodeChanges( inputPortId, otherNodeId, otherPortId, name, type );
- UpdateConnection( inputPortId );
- }
-
- public override void OnInputPortDisconnected( int portId )
- {
- base.OnInputPortDisconnected( portId );
- UpdateDisconnection( portId );
- }
-
- void UpdateConnection( int portId )
- {
- m_inputPorts[ portId ].MatchPortToConnection();
- int otherPortId = ( portId + 1 ) % 2;
- if( m_inputPorts[ otherPortId ].IsConnected )
- {
- m_mainDataType = UIUtils.GetPriority( m_inputPorts[ 0 ].DataType ) > UIUtils.GetPriority( m_inputPorts[ 1 ].DataType ) ? m_inputPorts[ 0 ].DataType : m_inputPorts[ 1 ].DataType;
- }
- else
- {
- m_mainDataType = m_inputPorts[ portId ].DataType;
- m_inputPorts[ otherPortId ].ChangeType( m_mainDataType, false );
- }
- m_outputPorts[ 0 ].ChangeType( m_mainDataType, false );
- }
-
- void UpdateDisconnection( int portId )
- {
- int otherPortId = ( portId + 1 ) % 2;
- if( m_inputPorts[ otherPortId ].IsConnected )
- {
- m_mainDataType = m_inputPorts[ otherPortId ].DataType;
- m_inputPorts[ portId ].ChangeType( m_mainDataType, false );
- m_outputPorts[ 0 ].ChangeType( m_mainDataType, false );
- }
- }
-
- public override void DrawProperties()
- {
- base.DrawProperties();
- EditorGUI.BeginChangeCheck();
- m_currentBlendOp = (BlendOps)EditorGUILayoutEnumPopup( BlendOpsModeStr, m_currentBlendOp );
- if( EditorGUI.EndChangeCheck() )
- {
- SetAdditonalTitleText( string.Format( Constants.SubTitleTypeFormatStr, m_currentBlendOp ) );
- }
- m_saturate = EditorGUILayoutToggle( SaturateResultStr, m_saturate );
- }
-
- public override void Draw( DrawInfo drawInfo )
- {
- base.Draw( drawInfo );
- m_upperLeftWidget.DrawWidget<BlendOps>( ref m_currentBlendOp, this, OnWidgetUpdate );
- }
-
- private readonly Action<ParentNode> OnWidgetUpdate = ( x ) =>
- {
- x.SetAdditonalTitleText( string.Format( Constants.SubTitleTypeFormatStr, ( x as BlendOpsNode ).m_currentBlendOp ) );
- };
-
- private string CreateMultiChannel( ref MasterNodeDataCollector dataCollector, string function, string srcLocalVar, string dstLocalVar, string varName )
- {
- switch( m_outputPorts[ 0 ].DataType )
- {
- default:
- {
- return string.Format( function, srcLocalVar, dstLocalVar );
- }
- case WirePortDataType.FLOAT2:
- {
- string xChannelName = varName + OutputId + "X";
- string xChannelValue = string.Format( function, srcLocalVar + ".x", dstLocalVar + ".x" );
- dataCollector.AddLocalVariable( UniqueId, m_currentPrecisionType, WirePortDataType.FLOAT, xChannelName, xChannelValue );
-
- string yChannelName = varName + OutputId + "Y";
- string yChannelValue = string.Format( function, srcLocalVar + ".y", dstLocalVar + ".y" );
- dataCollector.AddLocalVariable( UniqueId, m_currentPrecisionType, WirePortDataType.FLOAT, yChannelName, yChannelValue );
-
- return string.Format( "float2({0},{1})", xChannelName, yChannelName );
- }
- case WirePortDataType.FLOAT3:
- {
- string xChannelName = varName + OutputId + "X";
- string xChannelValue = string.Format( function, srcLocalVar + ".x", dstLocalVar + ".x" );
- dataCollector.AddLocalVariable( UniqueId, m_currentPrecisionType, WirePortDataType.FLOAT, xChannelName, xChannelValue );
-
- string yChannelName = varName + OutputId + "Y";
- string yChannelValue = string.Format( function, srcLocalVar + ".y", dstLocalVar + ".y" );
- dataCollector.AddLocalVariable( UniqueId, m_currentPrecisionType, WirePortDataType.FLOAT, yChannelName, yChannelValue );
-
- string zChannelName = varName + OutputId + "Z";
- string zChannelValue = string.Format( function, srcLocalVar + ".z", dstLocalVar + ".z" );
- dataCollector.AddLocalVariable( UniqueId, m_currentPrecisionType, WirePortDataType.FLOAT, zChannelName, zChannelValue );
-
- return string.Format( "float3({0},{1},{2})", xChannelName, yChannelName, zChannelName );
- }
- case WirePortDataType.FLOAT4:
- case WirePortDataType.COLOR:
- {
- string xChannelName = varName + OutputId + "X";
- string xChannelValue = string.Format( function, srcLocalVar + ".x", dstLocalVar + ".x" );
- dataCollector.AddLocalVariable( UniqueId, m_currentPrecisionType, WirePortDataType.FLOAT, xChannelName, xChannelValue );
-
- string yChannelName = varName + OutputId + "Y";
- string yChannelValue = string.Format( function, srcLocalVar + ".y", dstLocalVar + ".y" );
- dataCollector.AddLocalVariable( UniqueId, m_currentPrecisionType, WirePortDataType.FLOAT, yChannelName, yChannelValue );
-
- string zChannelName = varName + OutputId + "Z";
- string zChannelValue = string.Format( function, srcLocalVar + ".z", dstLocalVar + ".z" );
- dataCollector.AddLocalVariable( UniqueId, m_currentPrecisionType, WirePortDataType.FLOAT, zChannelName, zChannelValue );
-
- string wChannelName = varName + OutputId + "W";
- string wChannelValue = string.Format( function, srcLocalVar + ".w", dstLocalVar + ".w" );
- dataCollector.AddLocalVariable( UniqueId, m_currentPrecisionType, WirePortDataType.FLOAT, wChannelName, wChannelValue );
-
- return string.Format( "float4({0},{1},{2},{3})", xChannelName, yChannelName, zChannelName, wChannelName );
- }
- }
- }
-
- public override string GenerateShaderForOutput( int outputId, ref MasterNodeDataCollector dataCollector, bool ignoreLocalvar )
- {
- if( m_outputPorts[ 0 ].IsLocalValue( dataCollector.PortCategory ) )
- return m_outputPorts[ 0 ].LocalValue( dataCollector.PortCategory );
-
- string src = m_inputPorts[ 0 ].GenerateShaderForOutput( ref dataCollector, m_mainDataType, false, true );
- string dst = m_inputPorts[ 1 ].GenerateShaderForOutput( ref dataCollector, m_mainDataType, false, true );
-
- string srcLocalVar = "blendOpSrc" + OutputId;
- string dstLocalVar = "blendOpDest" + OutputId;
- dataCollector.AddLocalVariable( UniqueId, UIUtils.FinalPrecisionWirePortToCgType( m_currentPrecisionType, m_mainDataType ) + " " + srcLocalVar, src + ";" );
- dataCollector.AddLocalVariable( UniqueId, UIUtils.FinalPrecisionWirePortToCgType( m_currentPrecisionType, m_mainDataType ) + " " + dstLocalVar, dst + ";" );
-
- int currIndent = UIUtils.ShaderIndentLevel;
- if( dataCollector.MasterNodeCategory == AvailableShaderTypes.Template )
- {
- UIUtils.ShaderIndentLevel = 0;
- }
- else
- {
- UIUtils.ShaderIndentLevel = 1;
- UIUtils.ShaderIndentLevel++;
- }
-
- string result = string.Empty;
- switch( m_currentBlendOp )
- {
- case BlendOps.ColorBurn:
- {
- result = "( 1.0 - ( ( 1.0 - " + dstLocalVar + ") / " + srcLocalVar + ") )";
- }
- break;
- case BlendOps.ColorDodge:
- {
- result = "( " + dstLocalVar + "/ ( 1.0 - " + srcLocalVar + " ) )";
- }
- break;
- case BlendOps.Darken:
- {
- result = "min( " + srcLocalVar + " , " + dstLocalVar + " )";
- }
- break;
- case BlendOps.Divide:
- {
- result = "( " + dstLocalVar + " / " + srcLocalVar + " )";
- }
- break;
- case BlendOps.Difference:
- {
- result = "abs( " + srcLocalVar + " - " + dstLocalVar + " )";
- }
- break;
- case BlendOps.Exclusion:
- {
- result = "( 0.5 - 2.0 * ( " + srcLocalVar + " - 0.5 ) * ( " + dstLocalVar + " - 0.5 ) )";
- }
- break;
- case BlendOps.SoftLight:
- {
- result = string.Format( "2.0f*{0}*{1} + {0}*{0}*(1.0f - 2.0f*{1})", dstLocalVar, srcLocalVar );
- }
- break;
- case BlendOps.HardLight:
- {
- result = " (( " + srcLocalVar + " > 0.5 ) ? ( 1.0 - ( 1.0 - 2.0 * ( " + srcLocalVar + " - 0.5 ) ) * ( 1.0 - " + dstLocalVar + " ) ) : ( 2.0 * " + srcLocalVar + " * " + dstLocalVar + " ) )";
- //dataCollector.AddFunction( ASEHardLightCall, UIUtils.ShaderIndentTabs + ASEHardLightFunc );
- //result = CreateMultiChannel( ref dataCollector, ASEHardLightCall, srcLocalVar, dstLocalVar, "hardLightBlend" );
- }
- break;
- case BlendOps.HardMix:
- {
- result = " round( 0.5 * ( " + srcLocalVar + " + " + dstLocalVar + " ) )";
- }
- break;
- case BlendOps.Lighten:
- {
- result = " max( " + srcLocalVar + ", " + dstLocalVar + " )";
- }
- break;
- case BlendOps.LinearBurn:
- {
- result = "( " + srcLocalVar + " + " + dstLocalVar + " - 1.0 )";
- }
- break;
- case BlendOps.LinearDodge:
- {
- result = "( " + srcLocalVar + " + " + dstLocalVar + " )";
- }
- break;
- case BlendOps.LinearLight:
- {
- result = "(( " + srcLocalVar + " > 0.5 )? ( " + dstLocalVar + " + 2.0 * " + srcLocalVar + " - 1.0 ) : ( " + dstLocalVar + " + 2.0 * ( " + srcLocalVar + " - 0.5 ) ) )";
- //dataCollector.AddFunction( ASELinearLightCall, UIUtils.ShaderIndentTabs + ASELinearLightFunc );
- //result = CreateMultiChannel( ref dataCollector, ASELinearLightCall, srcLocalVar, dstLocalVar, "linearLightBlend" );
- }
- break;
- case BlendOps.Multiply:
- {
- result = "( " + srcLocalVar + " * " + dstLocalVar + " )";
- }
- break;
- case BlendOps.Overlay:
- {
- result = "(( " + dstLocalVar + " > 0.5 ) ? ( 1.0 - ( 1.0 - 2.0 * ( " + dstLocalVar + " - 0.5 ) ) * ( 1.0 - " + srcLocalVar + " ) ) : ( 2.0 * " + dstLocalVar + " * " + srcLocalVar + " ) )";
- //dataCollector.AddFunction( ASEOverlayCall, UIUtils.ShaderIndentTabs + ASEOverlayFunc );
- //result = CreateMultiChannel( ref dataCollector, ASEOverlayCall, srcLocalVar, dstLocalVar, "overlayBlend" );
- }
- break;
- case BlendOps.PinLight:
- {
- result = "(( " + srcLocalVar + " > 0.5 ) ? max( " + dstLocalVar + ", 2.0 * ( " + srcLocalVar + " - 0.5 ) ) : min( " + dstLocalVar + ", 2.0 * " + srcLocalVar + " ) )";
- //dataCollector.AddFunction( ASEPinLightCall, UIUtils.ShaderIndentTabs + ASEPinLightFunc );
- //result = CreateMultiChannel( ref dataCollector, ASEPinLightCall, srcLocalVar, dstLocalVar, "pinLightBlend" );
- }
- break;
- case BlendOps.Subtract:
- {
- result = "( " + dstLocalVar + " - " + srcLocalVar + " )";
- }
- break;
- case BlendOps.Screen:
- {
- result = "( 1.0 - ( 1.0 - " + srcLocalVar + " ) * ( 1.0 - " + dstLocalVar + " ) )";
- }
- break;
- case BlendOps.VividLight:
- {
- result = "(( " + srcLocalVar + " > 0.5 ) ? ( " + dstLocalVar + " / ( ( 1.0 - " + srcLocalVar + " ) * 2.0 ) ) : ( 1.0 - ( ( ( 1.0 - " + dstLocalVar + " ) * 0.5 ) / " + srcLocalVar + " ) ) )";
- //dataCollector.AddFunction( ASEVividLightCall, UIUtils.ShaderIndentTabs + ASEVividLightFunc );
- //result = CreateMultiChannel( ref dataCollector, ASEVividLightCall, srcLocalVar, dstLocalVar, "vividLightBlend" );
- }
- break;
- }
-
- UIUtils.ShaderIndentLevel = currIndent;
-
- if( m_saturate )
- result = "( saturate( " + result + " ))";
-
- return CreateOutputLocalVariable( 0, result, ref dataCollector );
- }
-
- public override void WriteToString( ref string nodeInfo, ref string connectionsInfo )
- {
- base.WriteToString( ref nodeInfo, ref connectionsInfo );
- IOUtils.AddFieldValueToString( ref nodeInfo, m_currentBlendOp );
- IOUtils.AddFieldValueToString( ref nodeInfo, m_saturate );
- }
-
- public override void ReadFromString( ref string[] nodeParams )
- {
- base.ReadFromString( ref nodeParams );
- m_currentBlendOp = (BlendOps)Enum.Parse( typeof( BlendOps ), GetCurrentParam( ref nodeParams ) );
- m_saturate = Convert.ToBoolean( GetCurrentParam( ref nodeParams ) );
- SetAdditonalTitleText( string.Format( Constants.SubTitleTypeFormatStr, m_currentBlendOp ) );
- }
-
- public override void Destroy()
- {
- base.Destroy();
- m_upperLeftWidget = null;
- }
- }
- }
|