Added FunctionNode.

Author: KN4CK3R

CodeGenerator/CSharpCodeGenerator.cs

@@ -5,6 +5,7 @@
 using System.Text;
 using ReClassNET.Logger;
 using ReClassNET.Nodes;
+using ReClassNET.Util;
 
 namespace ReClassNET.CodeGenerator
 {
@@ -84,7 +85,7 @@ private IEnumerable<MemberDefinition> YieldMemberDefinitions(IEnumerable<BaseNod
 			Contract.Ensures(Contract.Result<IEnumerable<MemberDefinition>>() != null);
 			Contract.Ensures(Contract.ForAll(Contract.Result<IEnumerable<MemberDefinition>>(), d => d != null));
 
-			foreach (var member in members.Where(n => !(n is BaseHexNode)))
+			foreach (var member in members.WhereNot(n => n is BaseHexNode))
 			{
 				if (member is BitFieldNode)
 				{

CodeGenerator/CppCodeGenerator.cs

@@ -5,6 +5,7 @@
 using System.Text;
 using ReClassNET.Logger;
 using ReClassNET.Nodes;
+using ReClassNET.Util;
 
 namespace ReClassNET.CodeGenerator
 {
@@ -48,7 +49,7 @@ public string GenerateCode(IEnumerable<ClassNode> classes, ILogger logger)
 			sb.AppendLine(
 				string.Join(
 					"\n\n",
-					OrderByInheritance(classes).Select(c =>
+					OrderByInheritance(classes.Where(c => c.Nodes.None(n => n is FunctionNode))).Select(c =>
 					{
 						var csb = new StringBuilder();
 						csb.Append($"class {c.Name}");
@@ -74,7 +75,7 @@ public string GenerateCode(IEnumerable<ClassNode> classes, ILogger logger)
 						csb.AppendLine(
 							string.Join(
 								"\n",
-								YieldMemberDefinitions(c.Nodes.Skip(skipFirstMember ? 1 : 0), logger)
+								YieldMemberDefinitions(c.Nodes.Skip(skipFirstMember ? 1 : 0).WhereNot(n => n is FunctionNode), logger)
 									.Select(m => MemberDefinitionToString(m))
 									.Select(s => "\t" + s)
 							)
@@ -92,6 +93,18 @@ public string GenerateCode(IEnumerable<ClassNode> classes, ILogger logger)
 							);
 						}
 
+						var functions = classes.SelectMany(c2 => c2.Nodes).OfType<FunctionNode>().Where(f => f.BelongsToClass == c);
+						if (functions.Any())
+						{
+							csb.AppendLine();
+							csb.AppendLine(
+								string.Join(
+									"\n",
+									functions.Select(f => $"\t{f.Signature} {{ }}")
+								)
+							);
+						}
+
 						csb.Append($"}}; //Size: 0x{c.MemorySize:X04}");
 						return csb.ToString();
 					})
@@ -145,7 +158,7 @@ private IEnumerable<MemberDefinition> YieldMemberDefinitions(IEnumerable<BaseNod
 			int fill = 0;
 			int fillStart = 0;
 
-			foreach (var member in members.Where(m => !(m is VTableNode)))
+			foreach (var member in members.WhereNot(m => m is VTableNode))
 			{
 				if (member is BaseHexNode)
 				{

DataExchange/ReClassFile.cs

@@ -381,7 +381,8 @@ public override int Draw(ViewInfo view, int x, int y)
 			typeof(UTF8TextPtrNode),
 			typeof(UTF16TextPtrNode),
 			typeof(BitFieldNode),
-			typeof(UInt64Node)
+			typeof(UInt64Node),
+			typeof(FunctionNode)
 		};
 
 		#endregion

DataExchange/ReClassNetFile.Read.cs

@@ -171,6 +171,17 @@ private IEnumerable<BaseNode> ReadNodeElements(IEnumerable<XElement> elements, C
 					TryGetAttributeValue(element, XmlBitsAttribute, out bits, logger);
 					bitFieldNode.Bits = bits;
 				}
+				var functionNode = node as FunctionNode;
+				if (functionNode != null)
+				{
+					functionNode.Signature = element.Attribute(XmlSignatureAttribute)?.Value ?? string.Empty;
+
+					var reference = NodeUuid.FromBase64String(element.Attribute(XmlReferenceAttribute)?.Value, false);
+					if (project.ContainsClass(reference))
+					{
+						functionNode.BelongsToClass = project.GetClassByUuid(reference);
+					}
+				}
 
 				yield return node;
 			}

DataExchange/ReClassNetFile.Write.cs

@@ -120,6 +120,13 @@ private IEnumerable<XElement> CreateNodeElements(IEnumerable<BaseNode> nodes, IL
 				{
 					element.SetAttributeValue(XmlBitsAttribute, bitFieldNode.Bits);
 				}
+				var functionNode = node as FunctionNode;
+				if (functionNode != null)
+				{
+					var uuid = functionNode.BelongsToClass == null ? NodeUuid.Zero : functionNode.BelongsToClass.Uuid;
+					element.SetAttributeValue(XmlReferenceAttribute, uuid.ToBase64String());
+					element.SetAttributeValue(XmlSignatureAttribute, functionNode.Signature);
+				}
 
 				yield return element;
 			}

DataExchange/ReClassNetFile.cs

@@ -37,6 +37,7 @@ public partial class ReClassNetFile : IReClassImport, IReClassExport
 		public const string XmlCountAttribute = "count";
 		public const string XmlBitsAttribute = "bits";
 		public const string XmlLengthAttribute = "length";
+		public const string XmlSignatureAttribute = "signature";
 
 		private ReClassNetProject project;
 
@@ -49,13 +50,15 @@ public ReClassNetFile(ReClassNetProject project)
 
 		private static Dictionary<string, Type> BuildInStringToTypeMap = new Type[]
 		{
+			typeof(BoolNode),
 			typeof(BitFieldNode),
 			typeof(ClassInstanceArrayNode),
 			typeof(ClassInstanceNode),
 			typeof(ClassPtrArrayNode),
 			typeof(ClassPtrNode),
 			typeof(DoubleNode),
 			typeof(FloatNode),
+			typeof(FunctionNode),
 			typeof(FunctionPtrNode),
 			typeof(Hex8Node),
 			typeof(Hex16Node),