1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
|
// Amplify Shader Editor - Visual Shader Editing Tool
// Copyright (c) Amplify Creations, Lda <info@amplify.pt>
//https://www.shadertoy.com/view/ldX3D4
using UnityEngine;
using UnityEditor;
using System;
namespace AmplifyShaderEditor
{
[Serializable]
[NodeAttributes( "Posterize", "Image Effects", "Converts a continuous gradation of tones to multiple regions of fewer tones" )]
public sealed class PosterizeNode : ParentNode
{
private const string PosterizationPowerStr = "Power";
[SerializeField]
private int m_posterizationPower = 1;
protected override void CommonInit( int uniqueId )
{
base.CommonInit( uniqueId );
AddInputPort( WirePortDataType.COLOR, false, "RGBA", -1, MasterNodePortCategory.Fragment, 1 );
AddInputPort( WirePortDataType.INT, false, "Power", -1, MasterNodePortCategory.Fragment, 0 );
m_inputPorts[ 1 ].AutoDrawInternalData = true;
AddOutputPort( WirePortDataType.COLOR, Constants.EmptyPortValue );
m_textLabelWidth = 60;
m_autoWrapProperties = true;
m_previewShaderGUID = "ecb3048ef0eec1645bad1d72a98d8279";
}
public override void DrawProperties()
{
base.DrawProperties();
if( !m_inputPorts[ 1 ].IsConnected )
{
EditorGUILayout.BeginVertical();
{
EditorGUI.BeginChangeCheck();
m_posterizationPower = EditorGUILayoutIntSlider( PosterizationPowerStr, m_posterizationPower, 1, 256 );
if( EditorGUI.EndChangeCheck() )
{
GetInputPortByUniqueId( 0 ).IntInternalData = m_posterizationPower;
}
}
EditorGUILayout.EndVertical();
}
else
{
EditorGUILayout.Space();
}
}
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 posterizationPower = "1";
if( m_inputPorts[ 1 ].IsConnected )
{
posterizationPower = m_inputPorts[ 1 ].GeneratePortInstructions( ref dataCollector );
}
else
{
posterizationPower = m_posterizationPower.ToString();
}
string colorTarget = m_inputPorts[ 0 ].GeneratePortInstructions( ref dataCollector );
string divVar = "div" + OutputId;
dataCollector.AddLocalVariable( UniqueId, "float " + divVar + "=256.0/float(" + posterizationPower + ");" );
string result = "( floor( " + colorTarget + " * " + divVar + " ) / " + divVar + " )";
RegisterLocalVariable( 0, result, ref dataCollector, "posterize" + OutputId );
return m_outputPorts[ 0 ].LocalValue( dataCollector.PortCategory );
}
public override void RefreshExternalReferences()
{
base.RefreshExternalReferences();
m_inputPorts[ 0 ].ChangeType( WirePortDataType.COLOR, false );
m_inputPorts[ 1 ].ChangeType( WirePortDataType.INT, false );
}
public override void WriteToString( ref string nodeInfo, ref string connectionsInfo )
{
base.WriteToString( ref nodeInfo, ref connectionsInfo );
IOUtils.AddFieldValueToString( ref nodeInfo, m_posterizationPower );
}
public override void ReadFromString( ref string[] nodeParams )
{
base.ReadFromString( ref nodeParams );
m_posterizationPower = Convert.ToInt32( GetCurrentParam( ref nodeParams ) );
}
}
}
|
