UltimateZoneBuilder/Source/Core/GZBuilder/Tools/UDMFTools.cs

using System;
using System.Collections.Generic;
using CodeImp.DoomBuilder.Map;
using CodeImp.DoomBuilder.Types;

namespace CodeImp.DoomBuilder.GZBuilder.Tools
{
	public static class UDMFTools
	{
		//float
		public static void SetFloat(UniFields fields, string key, float value) 
		{
			SetFloat(fields, key, value, 0f);
		}

		public static void SetFloat(UniFields fields, string key, float value, float defaultValue) 
		{
			if(fields == null) return;

			if(value != defaultValue) 
			{
				if(!fields.ContainsKey(key))
					fields.Add(key, new UniValue(UniversalType.Float, value));
				else
					fields[key].Value = value;
			} 
			else if(fields.ContainsKey(key)) //don't save default value
			{ 
				fields.Remove(key);
			}
		}

		public static float GetFloat(UniFields fields, string key) 
		{
			return GetFloat(fields, key, 0f);
		}

		public static float GetFloat(UniFields fields, string key, float defaultValue) 
		{
			if(fields == null) return defaultValue;
			return fields.GetValue(key, defaultValue);
		}

		//int
		public static void SetInteger(UniFields fields, string key, int value) 
		{
			SetInteger(fields, key, value, 0);
		}

		public static void SetInteger(UniFields fields, string key, int value, int defaultValue) 
		{
			if(fields == null) return;

			if(value != defaultValue) 
			{
				if(!fields.ContainsKey(key))
					fields.Add(key, new UniValue(UniversalType.Integer, value));
				else
					fields[key].Value = value;
			} 
			else if(fields.ContainsKey(key)) //don't save default value
			{ 
				fields.Remove(key);
			}
		}

		public static int GetInteger(UniFields fields, string key) 
		{
			return GetInteger(fields, key, 0);
		}

		public static int GetInteger(UniFields fields, string key, int defaultValue) 
		{
			if(fields == null) return defaultValue;
			return fields.GetValue(key, defaultValue);
		}

		public static void SetString(UniFields fields, string key, string value, string defaultValue) 
		{
			if(fields == null) return;

			if(value != defaultValue) 
			{
				if(!fields.ContainsKey(key))
					fields.Add(key, new UniValue(UniversalType.String, value));
				else
					fields[key].Value = value;
			} 
			else if(fields.ContainsKey(key)) //don't save default value
			{
				fields.Remove(key);
			}
		}

		public static void ClearFields(UniFields fields, string[] keys) 
		{
			if(fields == null) return;

			foreach(string key in keys)
			{
				if(fields.ContainsKey(key)) fields.Remove(key);
			}
		}

		public static void ClearField(UniFields fields, string key) 
		{
			if(fields == null || !fields.ContainsKey(key)) return;
			fields.Remove(key);
		}

		public static bool FieldsMatch(UniFields fields1, UniFields fields2) 
		{
			if (fields1.Keys.Count != fields2.Keys.Count) return false;
			foreach(KeyValuePair<string, UniValue> group in fields1) 
			{
				if (!fields2.ContainsKey(group.Key)) return false;
				if (fields2[group.Key].Type != fields1[group.Key].Type) return false;
				
				if (fields1[group.Key].Value is int) 
				{
					if ((int)fields1[group.Key].Value != (int)fields2[group.Key].Value) return false;
				}
				else if (fields1[group.Key].Value is float) 
				{
					if((float)fields1[group.Key].Value != (float)fields2[group.Key].Value) return false;
				}
				else if(fields1[group.Key].Value is bool) 
				{
					if((bool)fields1[group.Key].Value != (bool)fields2[group.Key].Value) return false;
				}
				else if (fields1[group.Key].Value is string) 
				{
					if ((string)fields1[group.Key].Value != (string)fields2[group.Key].Value) return false;
				} 
				else 
				{
					throw new Exception("Got unknown Custom Field type to compare: " + fields1[group.Key].Value.GetType());
				}
			}

			return true;
		}
	}
}