summaryrefslogtreecommitdiff
path: root/Assets/ThirdParty/AmplifyShaderEditor/Plugins/Editor/Templates/TemplateOptionsDefinesContainer.cs
blob: affbd406341112d162997ece556c76f74d11b541 (plain)
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
using UnityEngine;
using System;
using System.Collections.Generic;

namespace AmplifyShaderEditor
{
	[Serializable]
	public class TemplateOptionsDefinesContainer
	{
		[SerializeField]
		private List<PropertyDataCollector> m_definesList = new List<PropertyDataCollector>();

		[NonSerialized]
		private Dictionary<string, PropertyDataCollector> m_definesDict = new Dictionary<string, PropertyDataCollector>();

		void Refresh()
		{
			if( m_definesDict.Count != m_definesList.Count )
			{
				m_definesDict.Clear();
				for( int i = 0; i < m_definesList.Count; i++ )
				{
					m_definesDict.Add( m_definesList[ i ].PropertyName, m_definesList[ i ] );
				}
			}
		}

		public void RemoveTemporaries()
		{
			List<PropertyDataCollector> temporaries = m_definesList.FindAll( ( x ) => ( x.NodeId == 1 ) );
			for( int i = 0; i < temporaries.Count; i++ )
			{
				m_definesList.Remove( temporaries[ i ] );
				m_definesDict.Remove( temporaries[ i ].PropertyName );
			}
		}

		public void AddDefine( string define , bool temporary )
		{
			Refresh();
			if( !m_definesDict.ContainsKey( define ) )
			{
				int nodeId = temporary ? 1 : 0;
				PropertyDataCollector data = new PropertyDataCollector( nodeId, define );
				m_definesDict.Add( define, data );
				m_definesList.Add( data );
			}
		}

		public void RemoveDefine( string define )
		{
			Refresh();
			if( m_definesDict.ContainsKey( define ) )
			{
				m_definesList.Remove( m_definesDict[define] );
				m_definesDict.Remove( define );
			}
		}

		public void Destroy()
		{
			m_definesDict.Clear();
			m_definesDict = null;
			m_definesList.Clear();
			m_definesList = null;
		}
		public List<PropertyDataCollector> DefinesList { get { return m_definesList; } }
	}
}