You've already forked FateViewer
Add project files.
This commit is contained in:
@@ -0,0 +1,352 @@
|
||||
// Amplify Shader Editor - Visual Shader Editing Tool
|
||||
// Copyright (c) Amplify Creations, Lda <info@amplify.pt>
|
||||
|
||||
using UnityEngine;
|
||||
using UnityEditor;
|
||||
using System;
|
||||
using System.Collections.Generic;
|
||||
|
||||
namespace AmplifyShaderEditor
|
||||
{
|
||||
[Serializable]
|
||||
[NodeAttributes( "Register Local Var", "Miscellaneous", "Forces a local variable to be written with the given name. Can then be fetched at any place with a <b>Get Local Var</b> node.", null, KeyCode.R )]
|
||||
public sealed class RegisterLocalVarNode : ParentNode
|
||||
{
|
||||
private const double MaxEditingTimestamp = 1;
|
||||
|
||||
private const string LocalDefaultNameStr = "myVarName";
|
||||
private const string LocalVarNameStr = "Var Name";
|
||||
private const string OrderIndexStr = "Order Index";
|
||||
private const string AutoOrderIndexStr = "Auto Order";
|
||||
private const string ReferencesStr = "References";
|
||||
|
||||
private const string GetLocalVarLabel = "( {0} ) Get Local Var";
|
||||
private string m_oldName = string.Empty;
|
||||
private bool m_reRegisterName = false;
|
||||
private int m_autoOrderIndex = int.MaxValue;
|
||||
private bool m_forceUpdate = true;
|
||||
private bool m_refSelect = false;
|
||||
|
||||
private bool m_referencesVisible = false;
|
||||
|
||||
[SerializeField]
|
||||
private string m_variableName = LocalDefaultNameStr;
|
||||
|
||||
[SerializeField]
|
||||
private int m_orderIndex = -1;
|
||||
|
||||
[SerializeField]
|
||||
private bool m_autoIndexActive = true;
|
||||
|
||||
[SerializeField]
|
||||
private List<GetLocalVarNode> m_registeredGetLocalVars = new List<GetLocalVarNode>();
|
||||
|
||||
[NonSerialized]
|
||||
private double m_editingTimestamp;
|
||||
|
||||
[NonSerialized]
|
||||
private bool m_editingTimestampFlag;
|
||||
|
||||
protected override void CommonInit( int uniqueId )
|
||||
{
|
||||
base.CommonInit( uniqueId );
|
||||
AddInputPort( WirePortDataType.FLOAT, false, Constants.EmptyPortValue );
|
||||
AddOutputPort( WirePortDataType.FLOAT, Constants.EmptyPortValue );
|
||||
m_inputPorts[ 0 ].SetFreeForAll();
|
||||
m_textLabelWidth = 85;
|
||||
m_customPrecision = true;
|
||||
|
||||
if( m_containerGraph != null )
|
||||
m_variableName += m_containerGraph.LocalVarNodes.NodesList.Count;
|
||||
|
||||
m_oldName = m_variableName;
|
||||
UpdateTitle();
|
||||
m_previewShaderGUID = "5aaa1d3ea9e1fa64781647e035a82334";
|
||||
}
|
||||
|
||||
protected override void OnUniqueIDAssigned()
|
||||
{
|
||||
base.OnUniqueIDAssigned();
|
||||
m_containerGraph.LocalVarNodes.AddNode( this );
|
||||
}
|
||||
|
||||
public override void OnInputPortConnected( int portId, int otherNodeId, int otherPortId, bool activateNode = true )
|
||||
{
|
||||
base.OnInputPortConnected( portId, otherNodeId, otherPortId, activateNode );
|
||||
m_inputPorts[ 0 ].MatchPortToConnection();
|
||||
m_outputPorts[ 0 ].ChangeType( m_inputPorts[ 0 ].DataType, false );
|
||||
}
|
||||
|
||||
public override void OnConnectedOutputNodeChanges( int outputPortId, int otherNodeId, int otherPortId, string name, WirePortDataType type )
|
||||
{
|
||||
base.OnConnectedOutputNodeChanges( outputPortId, otherNodeId, otherPortId, name, type );
|
||||
m_inputPorts[ 0 ].MatchPortToConnection();
|
||||
m_outputPorts[ 0 ].ChangeType( m_inputPorts[ 0 ].DataType, false );
|
||||
|
||||
for( int i = 0 ; i < m_registeredGetLocalVars.Count ; i++ )
|
||||
{
|
||||
m_registeredGetLocalVars[ i ].UpdateLocalVar();
|
||||
}
|
||||
}
|
||||
|
||||
void UpdateTitle()
|
||||
{
|
||||
SetAdditonalTitleText( string.Format( Constants.SubTitleVarNameFormatStr, m_variableName ) );
|
||||
}
|
||||
|
||||
public override void OnNodeLogicUpdate( DrawInfo drawInfo )
|
||||
{
|
||||
base.OnNodeLogicUpdate( drawInfo );
|
||||
if( m_editingTimestampFlag && ( EditorApplication.timeSinceStartup - m_editingTimestamp ) > MaxEditingTimestamp )
|
||||
{
|
||||
m_editingTimestampFlag = false;
|
||||
CheckAndChangeName();
|
||||
}
|
||||
}
|
||||
|
||||
void DrawMainProperties()
|
||||
{
|
||||
EditorGUI.BeginChangeCheck();
|
||||
m_variableName = EditorGUILayoutTextField( LocalVarNameStr, m_variableName );
|
||||
if( EditorGUI.EndChangeCheck() )
|
||||
{
|
||||
m_editingTimestampFlag = true;
|
||||
m_editingTimestamp = EditorApplication.timeSinceStartup;
|
||||
//CheckAndChangeName();
|
||||
}
|
||||
|
||||
DrawPrecisionProperty();
|
||||
}
|
||||
|
||||
public override void AfterDuplication()
|
||||
{
|
||||
base.AfterDuplication();
|
||||
CheckAndChangeName();
|
||||
}
|
||||
|
||||
void CheckAndChangeName()
|
||||
{
|
||||
m_variableName = UIUtils.RemoveInvalidCharacters( m_variableName );
|
||||
if( string.IsNullOrEmpty( m_variableName ) )
|
||||
{
|
||||
m_variableName = LocalDefaultNameStr + OutputId;
|
||||
}
|
||||
bool isNumericName = UIUtils.IsNumericName( m_variableName );
|
||||
bool isLocalVarNameAvailable = m_containerGraph.ParentWindow.DuplicatePrevBufferInstance.IsLocalvariableNameAvailable( m_variableName );
|
||||
if( !isNumericName && isLocalVarNameAvailable )
|
||||
{
|
||||
m_containerGraph.ParentWindow.DuplicatePrevBufferInstance.ReleaseLocalVariableName( UniqueId, m_oldName );
|
||||
m_containerGraph.ParentWindow.DuplicatePrevBufferInstance.RegisterLocalVariableName( UniqueId, m_variableName );
|
||||
m_oldName = m_variableName;
|
||||
m_containerGraph.LocalVarNodes.UpdateDataOnNode( UniqueId, m_variableName );
|
||||
UpdateTitle();
|
||||
m_forceUpdate = true;
|
||||
}
|
||||
else
|
||||
{
|
||||
if( isNumericName )
|
||||
{
|
||||
UIUtils.ShowMessage( UniqueId, string.Format( "Local variable name '{0}' cannot start or be numerical values. Reverting back to '{1}'.", m_variableName, m_oldName ), MessageSeverity.Warning );
|
||||
}
|
||||
else if( !isLocalVarNameAvailable )
|
||||
{
|
||||
UIUtils.ShowMessage( UniqueId, string.Format( "Local variable name '{0}' already being used. Reverting back to '{1}'.", m_variableName, m_oldName ), MessageSeverity.Warning );
|
||||
}
|
||||
|
||||
m_variableName = m_oldName;
|
||||
m_containerGraph.LocalVarNodes.UpdateDataOnNode( UniqueId, m_variableName );
|
||||
}
|
||||
}
|
||||
|
||||
void DrawReferences()
|
||||
{
|
||||
int count = m_registeredGetLocalVars.Count;
|
||||
if( m_registeredGetLocalVars.Count > 0 )
|
||||
{
|
||||
for( int i = 0; i < count; i++ )
|
||||
{
|
||||
EditorGUILayout.BeginHorizontal();
|
||||
EditorGUILayout.LabelField( string.Format( GetLocalVarLabel, m_registeredGetLocalVars[ i ].UniqueId ) );
|
||||
if( GUILayout.Button( "\u25BA", "minibutton", GUILayout.Width( 17 ) ) )
|
||||
{
|
||||
m_containerGraph.ParentWindow.FocusOnNode( m_registeredGetLocalVars[ i ], 0, false );
|
||||
}
|
||||
EditorGUILayout.EndHorizontal();
|
||||
}
|
||||
|
||||
if( GUILayout.Button( "Back" ) )
|
||||
{
|
||||
m_containerGraph.ParentWindow.FocusOnNode( this, 0, false );
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
EditorGUILayout.HelpBox( "This node is not being referenced by any Get Local Var.", MessageType.Info );
|
||||
}
|
||||
}
|
||||
|
||||
public override void DrawProperties()
|
||||
{
|
||||
NodeUtils.DrawPropertyGroup( ref m_propertiesFoldout, Constants.ParameterLabelStr, DrawMainProperties );
|
||||
NodeUtils.DrawPropertyGroup( ref m_referencesVisible, ReferencesStr, DrawReferences );
|
||||
//EditorGUILayout.LabelField(ConnStatus.ToString()+" "+m_activeConnections);
|
||||
}
|
||||
|
||||
public override void OnEnable()
|
||||
{
|
||||
base.OnEnable();
|
||||
m_reRegisterName = true;
|
||||
}
|
||||
|
||||
public void CheckReferenceSelection()
|
||||
{
|
||||
m_refSelect = false;
|
||||
int count = m_registeredGetLocalVars.Count;
|
||||
for( int i = 0; i < count; i++ )
|
||||
{
|
||||
if( m_registeredGetLocalVars[ i ].Selected )
|
||||
m_refSelect = true;
|
||||
}
|
||||
}
|
||||
|
||||
public override void OnNodeRepaint( DrawInfo drawInfo )
|
||||
{
|
||||
base.OnNodeRepaint( drawInfo );
|
||||
if( m_isVisible && m_refSelect && !m_selected )
|
||||
{
|
||||
GUI.color = Constants.SpecialRegisterLocalVarSelectionColor;
|
||||
GUI.Label( m_globalPosition, string.Empty, UIUtils.GetCustomStyle( CustomStyle.NodeWindowOn ) );
|
||||
GUI.color = m_colorBuffer;
|
||||
}
|
||||
}
|
||||
|
||||
public override void Draw( DrawInfo drawInfo )
|
||||
{
|
||||
|
||||
base.Draw( drawInfo );
|
||||
if( m_reRegisterName )
|
||||
{
|
||||
m_reRegisterName = false;
|
||||
m_containerGraph.ParentWindow.DuplicatePrevBufferInstance.RegisterLocalVariableName( UniqueId, m_variableName );
|
||||
}
|
||||
|
||||
if( m_forceUpdate )
|
||||
{
|
||||
m_forceUpdate = false;
|
||||
UpdateTitle();
|
||||
}
|
||||
}
|
||||
|
||||
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 result = m_inputPorts[ 0 ].GeneratePortInstructions( ref dataCollector );
|
||||
if( m_inputPorts[ 0 ].DataType == WirePortDataType.OBJECT )
|
||||
m_outputPorts[ 0 ].SetLocalValue( result, dataCollector.PortCategory );
|
||||
else
|
||||
RegisterLocalVariable( 0, result, ref dataCollector, m_variableName + OutputId );
|
||||
return m_outputPorts[ 0 ].LocalValue( dataCollector.PortCategory );
|
||||
}
|
||||
|
||||
public override void ReadFromString( ref string[] nodeParams )
|
||||
{
|
||||
base.ReadFromString( ref nodeParams );
|
||||
m_variableName = GetCurrentParam( ref nodeParams );
|
||||
m_oldName = m_variableName;
|
||||
if( UIUtils.CurrentShaderVersion() > 14 )
|
||||
m_orderIndex = Convert.ToInt32( GetCurrentParam( ref nodeParams ) );
|
||||
|
||||
if( UIUtils.CurrentShaderVersion() > 3106 )
|
||||
{
|
||||
m_autoIndexActive = Convert.ToBoolean( GetCurrentParam( ref nodeParams ) );
|
||||
}
|
||||
else
|
||||
{
|
||||
m_autoIndexActive = false;
|
||||
}
|
||||
if( !m_isNodeBeingCopied )
|
||||
{
|
||||
m_containerGraph.LocalVarNodes.UpdateDataOnNode( UniqueId, m_variableName );
|
||||
m_containerGraph.ParentWindow.DuplicatePrevBufferInstance.ReleaseLocalVariableName( UniqueId, m_oldName );
|
||||
m_containerGraph.ParentWindow.DuplicatePrevBufferInstance.RegisterLocalVariableName( UniqueId, m_variableName );
|
||||
}
|
||||
m_forceUpdate = true;
|
||||
}
|
||||
|
||||
public override void WriteToString( ref string nodeInfo, ref string connectionsInfo )
|
||||
{
|
||||
base.WriteToString( ref nodeInfo, ref connectionsInfo );
|
||||
IOUtils.AddFieldValueToString( ref nodeInfo, m_variableName );
|
||||
IOUtils.AddFieldValueToString( ref nodeInfo, m_orderIndex );
|
||||
IOUtils.AddFieldValueToString( ref nodeInfo, m_autoIndexActive );
|
||||
}
|
||||
|
||||
public override void PropagateNodeData( NodeData nodeData, ref MasterNodeDataCollector dataCollector )
|
||||
{
|
||||
if( m_autoOrderIndex < nodeData.OrderIndex )
|
||||
{
|
||||
nodeData.OrderIndex = m_autoOrderIndex - 1;
|
||||
}
|
||||
else
|
||||
{
|
||||
m_autoOrderIndex = nodeData.OrderIndex;
|
||||
nodeData.OrderIndex -= 1;
|
||||
}
|
||||
|
||||
base.PropagateNodeData( nodeData, ref dataCollector );
|
||||
}
|
||||
|
||||
public override void ResetNodeData()
|
||||
{
|
||||
base.ResetNodeData();
|
||||
m_autoOrderIndex = int.MaxValue;
|
||||
}
|
||||
|
||||
public void RegisterGetLocalVar( GetLocalVarNode node )
|
||||
{
|
||||
if( !m_registeredGetLocalVars.Contains( node ) )
|
||||
{
|
||||
m_registeredGetLocalVars.Add( node );
|
||||
CheckReferenceSelection();
|
||||
}
|
||||
}
|
||||
|
||||
public void UnregisterGetLocalVar( GetLocalVarNode node )
|
||||
{
|
||||
if( m_registeredGetLocalVars.Contains( node ) )
|
||||
{
|
||||
m_registeredGetLocalVars.Remove( node );
|
||||
CheckReferenceSelection();
|
||||
}
|
||||
}
|
||||
|
||||
public override void Destroy()
|
||||
{
|
||||
base.Destroy();
|
||||
m_containerGraph.LocalVarNodes.RemoveNode( this );
|
||||
m_containerGraph.ParentWindow.DuplicatePrevBufferInstance.ReleaseLocalVariableName( UniqueId, m_variableName );
|
||||
|
||||
int count = m_registeredGetLocalVars.Count;
|
||||
for( int i = 0; i < count; i++ )
|
||||
{
|
||||
//GetLocalVarNode node = m_containerGraph.GetNode( m_registeredGetLocalVars[ i ] ) as GetLocalVarNode;
|
||||
if( m_registeredGetLocalVars[ i ] != null )
|
||||
m_registeredGetLocalVars[ i ].ResetReference();
|
||||
}
|
||||
m_registeredGetLocalVars.Clear();
|
||||
m_registeredGetLocalVars = null;
|
||||
|
||||
m_containerGraph.LocalVarNodes.RemoveNode( this );
|
||||
}
|
||||
|
||||
public override void ActivateNode( int signalGenNodeId, int signalGenPortId, Type signalGenNodeType )
|
||||
{
|
||||
base.ActivateNode( signalGenNodeId, signalGenPortId, signalGenNodeType );
|
||||
}
|
||||
public override string DataToArray { get { return m_variableName; } }
|
||||
public List<GetLocalVarNode> NodeReferences { get { return m_registeredGetLocalVars; } }
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user