|
|
- // Amplify Shader Editor - Visual Shader Editing Tool
- // Copyright (c) Amplify Creations, Lda <info@amplify.pt>
-
- using UnityEngine;
- using System;
- using UnityEditor;
- namespace AmplifyShaderEditor
- {
- [Serializable]
-
- public class WeightedAvgNode : ParentNode
- {
- protected string[] AmountsStr = { "Layer 1", "Layer 2", "Layer 3", "Layer 4" };
-
- [SerializeField]
- protected int m_minimumSize = 1;
-
- [SerializeField]
- protected WirePortDataType m_mainDataType = WirePortDataType.FLOAT;
-
- [SerializeField]
- protected string[] m_inputData;
- [SerializeField]
- protected int m_activeCount = 0;
-
- protected override void CommonInit( int uniqueId )
- {
- base.CommonInit( uniqueId );
- AddInputPort( WirePortDataType.FLOAT, false, "Weights" );
- AddInputPort( WirePortDataType.FLOAT, false, AmountsStr[ 0 ] );
- AddInputPort( WirePortDataType.FLOAT, false, AmountsStr[ 1 ] );
- AddInputPort( WirePortDataType.FLOAT, false, AmountsStr[ 2 ] );
- AddInputPort( WirePortDataType.FLOAT, false, AmountsStr[ 3 ] );
- AddOutputPort( WirePortDataType.FLOAT, Constants.EmptyPortValue );
-
- for( int i = 0; i < m_inputPorts.Count; i++ )
- {
- m_inputPorts[ i ].AddPortForbiddenTypes( WirePortDataType.FLOAT3x3,
- WirePortDataType.FLOAT4x4,
- WirePortDataType.SAMPLER1D,
- WirePortDataType.SAMPLER2D,
- WirePortDataType.SAMPLER3D,
- WirePortDataType.SAMPLERCUBE );
- }
- UpdateConnection( 0 );
- m_useInternalPortData = true;
- }
-
- 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 OnInputPortConnected( int portId, int otherNodeId, int otherPortId, bool activateNode = true )
- {
- base.OnInputPortConnected( portId, otherNodeId, otherPortId, activateNode );
- UpdateConnection( portId );
- }
-
- void UpdateInputPorts( int activePorts )
- {
- int idx = 1;
- for ( ; idx < m_minimumSize + activePorts; idx++ )
- {
- m_inputPorts[ idx ].Visible = true;
- }
-
- m_activeCount = idx - 1;
-
- for ( ; idx < m_inputPorts.Count; idx++ )
- {
- m_inputPorts[ idx ].Visible = false;
- }
- }
-
- protected void UpdateConnection( int portId )
- {
- if ( portId == 0 )
- {
- if( m_inputPorts[ portId ].IsConnected )
- m_inputPorts[ portId ].MatchPortToConnection();
-
- switch ( m_inputPorts[ 0 ].DataType )
- {
- case WirePortDataType.INT:
- case WirePortDataType.FLOAT:
- {
- UpdateInputPorts( 1 );
- m_previewMaterialPassId = 0;
- }
- break;
- case WirePortDataType.FLOAT2:
- {
- UpdateInputPorts( 2 );
- m_previewMaterialPassId = 1;
- }
- break;
- case WirePortDataType.FLOAT3:
- {
- UpdateInputPorts( 3 );
- m_previewMaterialPassId = 2;
- }
- break;
- case WirePortDataType.COLOR:
- case WirePortDataType.FLOAT4:
- {
- UpdateInputPorts( 4 );
- m_previewMaterialPassId = 3;
- }
- break;
- case WirePortDataType.OBJECT:
- case WirePortDataType.FLOAT3x3:
- case WirePortDataType.FLOAT4x4:
- {
- for ( int i = 1; i < m_inputPorts.Count; i++ )
- {
- m_inputPorts[ i ].Visible = false;
- }
- m_activeCount = 0;
- }
- break;
- }
- }
- //else
- //{
- // SetMainOutputType();
- //}
-
- SetMainOutputType();
- m_sizeIsDirty = true;
- }
-
- protected void SetMainOutputType()
- {
- m_mainDataType = WirePortDataType.OBJECT;
- int count = m_inputPorts.Count;
- for ( int i = 1; i < count; i++ )
- {
- if ( m_inputPorts[ i ].Visible )
- {
- WirePortDataType portType = m_inputPorts[ i ].IsConnected ? m_inputPorts[ i ].ConnectionType() : WirePortDataType.FLOAT;
- if ( m_mainDataType != portType &&
- UIUtils.GetPriority( portType ) > UIUtils.GetPriority( m_mainDataType ) )
- {
- m_mainDataType = portType;
- }
- }
- }
-
- for( int i = 1; i < count; i++ )
- {
- if( m_inputPorts[ i ].Visible )
- {
- m_inputPorts[ i ].ChangeType( m_mainDataType, false );
- }
- }
-
- m_outputPorts[ 0 ].ChangeType( m_mainDataType, false );
- }
-
- protected void GetInputData( ref MasterNodeDataCollector dataCollector, bool ignoreLocalvar )
- {
- m_inputData[ 0 ] = m_inputPorts[ 0 ].GeneratePortInstructions( ref dataCollector );
- for ( int i = 1; i < m_inputPorts.Count; i++ )
- {
- if ( m_inputPorts[ i ].Visible )
- {
- m_inputData[ i ] = m_inputPorts[ i ].GeneratePortInstructions( ref dataCollector );
- }
- }
- }
-
- public override void ReadInputDataFromString( ref string[] nodeParams )
- {
- base.ReadInputDataFromString( ref nodeParams );
- UpdateConnection( 0 );
- }
- }
- }
|