TFHCRemapNode.cs 3.4 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374
  1. // Amplify Shader Editor - Visual Shader Editing Tool
  2. // Copyright (c) Amplify Creations, Lda <info@amplify.pt>
  3. //
  4. // Custom Node Remap
  5. // Donated by The Four Headed Cat - @fourheadedcat
  6. using UnityEngine;
  7. using System;
  8. namespace AmplifyShaderEditor
  9. {
  10. [Serializable]
  11. [NodeAttributes( "Remap", "Math Operators", "Remap value from old min - max range to new min - max range", null, KeyCode.None, true, false, null, null, "The Four Headed Cat - @fourheadedcat" )]
  12. public sealed class TFHCRemapNode : DynamicTypeNode
  13. {
  14. protected override void CommonInit( int uniqueId )
  15. {
  16. base.CommonInit( uniqueId );
  17. m_inputPorts[ 0 ].Name = Constants.EmptyPortValue;
  18. m_inputPorts[ 1 ].Name = "Min Old";
  19. AddInputPort( WirePortDataType.FLOAT, false, "Max Old" );
  20. m_inputPorts[ 2 ].FloatInternalData = 1;
  21. m_inputPorts[ 2 ].Vector2InternalData = Vector2.one;
  22. m_inputPorts[ 2 ].Vector3InternalData = Vector3.one;
  23. m_inputPorts[ 2 ].Vector4InternalData = Vector4.one;
  24. m_inputPorts[ 2 ].ColorInternalData = Color.white;
  25. AddInputPort( WirePortDataType.FLOAT, false, "Min New" );
  26. AddInputPort( WirePortDataType.FLOAT, false, "Max New" );
  27. m_inputPorts[ 4 ].FloatInternalData = 1;
  28. m_inputPorts[ 4 ].Vector2InternalData = Vector2.one;
  29. m_inputPorts[ 4 ].Vector3InternalData = Vector3.one;
  30. m_inputPorts[ 4 ].Vector4InternalData = Vector4.one;
  31. m_inputPorts[ 4 ].ColorInternalData = Color.white;
  32. m_textLabelWidth = 100;
  33. m_useInternalPortData = true;
  34. m_previewShaderGUID = "72dd1cbea889fa047b929d5191e360c0";
  35. }
  36. public override void OnInputPortConnected( int portId, int otherNodeId, int otherPortId, bool activateNode = true )
  37. {
  38. base.OnInputPortConnected( portId, otherNodeId, otherPortId, activateNode );
  39. UpdateConnections();
  40. }
  41. public override void OnConnectedOutputNodeChanges( int outputPortId, int otherNodeId, int otherPortId, string name, WirePortDataType type )
  42. {
  43. base.OnConnectedOutputNodeChanges( outputPortId, otherNodeId, otherPortId, name, type );
  44. UpdateConnections();
  45. }
  46. void UpdateConnections()
  47. {
  48. m_inputPorts[ 0 ].MatchPortToConnection();
  49. m_inputPorts[ 1 ].ChangeType( m_inputPorts[ 0 ].DataType, false );
  50. m_inputPorts[ 2 ].ChangeType( m_inputPorts[ 0 ].DataType, false );
  51. m_inputPorts[ 3 ].ChangeType( m_inputPorts[ 0 ].DataType, false );
  52. m_inputPorts[ 4 ].ChangeType( m_inputPorts[ 0 ].DataType, false );
  53. m_outputPorts[ 0 ].ChangeType( m_inputPorts[ 0 ].DataType, false );
  54. }
  55. public override string GenerateShaderForOutput( int outputId, ref MasterNodeDataCollector dataCollector, bool ignoreLocalvar )
  56. {
  57. string value = m_inputPorts[ 0 ].GeneratePortInstructions( ref dataCollector );
  58. string oldMin = m_inputPorts[ 1 ].GenerateShaderForOutput( ref dataCollector, m_inputPorts[ 0 ].DataType, ignoreLocalvar, true );
  59. string oldMax = m_inputPorts[ 2 ].GenerateShaderForOutput( ref dataCollector, m_inputPorts[ 0 ].DataType, ignoreLocalvar, true );
  60. string newMin = m_inputPorts[ 3 ].GenerateShaderForOutput( ref dataCollector, m_inputPorts[ 0 ].DataType, ignoreLocalvar, true );
  61. string newMax = m_inputPorts[ 4 ].GenerateShaderForOutput( ref dataCollector, m_inputPorts[ 0 ].DataType, ignoreLocalvar, true );
  62. string strout = "(" + newMin + " + (" + value + " - " + oldMin + ") * (" + newMax + " - " + newMin + ") / (" + oldMax + " - " + oldMin + "))";
  63. return CreateOutputLocalVariable( 0, strout, ref dataCollector );
  64. }
  65. }
  66. }