/// <summary>
/// Writes the header. Contains a warning that it is
/// script controlled and declares the enum name.
/// </summary>
private static void WriteHeader(StreamWriter writer, GenerateEnum_Data theData)
{
writer.WriteLine("//SCRIPT GENERATED, do not modify manually!");
writer.WriteLine("public enum " + theData.enumType);
writer.WriteLine("{");
}
/// <summary>
/// Using theData.enumType, finds the existing compiled enum values.
/// Then loads theData.enumValues with the found existing values.
/// If no enum type is found, exits without changing theData.enumValues.
/// </summary>
public static void LoadExisting(GenerateEnum_Data theData)
{
theData.enumValues.Clear();
// Get Type by string
System.Type theType = EditorHelper.GetEnumType(theData.enumType);
if (theType == null)
{
return;
}
List <string> values = GenerateEnum.FindValuesByType(theData.enumType);
for (int i = 0; i < values.Count; i++)
{
// Turn enum value string into int value
int enumID = System.Convert.ToInt32(System.Enum.Parse(theType, values[i]));
// Add to given data structure, theData.enumValues
theData.enumValues.Add(new GenerateEnum_Data.SingleEnum_Data(values[i], enumID));
}
}
/// <summary>
/// Given theData.enumValues, writes or rewrites an enum script.
/// Script is located at theData.enumPath.
/// After generating the script it recompiles (by calling AssetDatabse.Refresh()).
/// </summary>
public static void Generate(GenerateEnum_Data theData, bool forceIDtoInOrder = false)
{
if (Application.isPlaying)
{
return;
}
string fullPath = theData.enumPath + theData.enumType + ".cs";
using (StreamWriter writer = new StreamWriter(fullPath))
{
WriteHeader(writer, theData);
List <int> usedIDs = new List <int> ();
for (int i = 0; i < theData.enumValues.Count; i++)
{
int id = forceIDtoInOrder ? i : theData.enumValues[i].enumID;
string name = theData.enumValues[i].enumName;
// Check for duplicate IDs
if (usedIDs.Contains(id))
{
Debug.LogError("Enum generation has duplicate ID!" +
"\r\nSkipping enum name/id: " + theData.enumValues[i].enumName + " / " + id);
continue;
}
WriteEnumValue(writer, name, id);
usedIDs.Add(id);
}
WriteFooter(writer, theData);
}
Recompile();
}
/// <summary>
/// Closes the enum.
/// </summary>
private static void WriteFooter(StreamWriter writer, GenerateEnum_Data theData)
{
writer.WriteLine("}");
}