Everything is now OpenRA, not OpenRa

OpenRA.Game/GameRules/ActorInfo.cs

@@ -0,0 +1,139 @@
+#region Copyright & License Information
+/*
+ * Copyright 2007,2009,2010 Chris Forbes, Robert Pepperell, Matthew Bowra-Dean, Paul Chote, Alli Witheford.
+ * This file is part of OpenRA.
+ * 
+ *  OpenRA is free software: you can redistribute it and/or modify
+ *  it under the terms of the GNU General Public License as published by
+ *  the Free Software Foundation, either version 3 of the License, or
+ *  (at your option) any later version.
+ * 
+ *  OpenRA is distributed in the hope that it will be useful,
+ *  but WITHOUT ANY WARRANTY; without even the implied warranty of
+ *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ *  GNU General Public License for more details.
+ * 
+ *  You should have received a copy of the GNU General Public License
+ *  along with OpenRA.  If not, see <http://www.gnu.org/licenses/>.
+ */
+#endregion
+
+using System;
+using System.Collections.Generic;
+using System.IO;
+using System.Linq;
+using System.Reflection;
+using OpenRA.FileFormats;
+using OpenRA.Traits;
+
+namespace OpenRA.GameRules
+{
+	public class ActorInfo
+	{
+		public readonly string Name;
+		public readonly string Category;
+		public readonly TypeDictionary Traits = new TypeDictionary();
+
+		public ActorInfo( string name, MiniYaml node, Dictionary<string, MiniYaml> allUnits )
+		{
+			var mergedNode = MergeWithParent( node, allUnits ).Nodes;
+
+			Name = name;
+			MiniYaml categoryNode;
+			if( mergedNode.TryGetValue( "Category", out categoryNode ) )
+				Category = categoryNode.Value;
+
+			foreach( var t in mergedNode )
+				if( t.Key != "Inherits" && t.Key != "Category" )
+					Traits.Add( LoadTraitInfo( t.Key, t.Value ) );
+		}
+
+		static MiniYaml GetParent( MiniYaml node, Dictionary<string, MiniYaml> allUnits )
+		{
+			MiniYaml inherits;
+			node.Nodes.TryGetValue( "Inherits", out inherits );
+			if( inherits == null || string.IsNullOrEmpty( inherits.Value ) )
+				return null;
+
+			MiniYaml parent;
+			allUnits.TryGetValue( inherits.Value, out parent );
+			if( parent == null )
+				return null;
+
+			return parent;
+		}
+
+		static MiniYaml MergeWithParent( MiniYaml node, Dictionary<string, MiniYaml> allUnits )
+		{
+			var parent = GetParent( node, allUnits );
+			if( parent != null )
+				return MiniYaml.Merge( node, MergeWithParent( parent, allUnits ) );
+			return node;
+		}
+
+		static Pair<Assembly, string>[] ModAssemblies;
+		public static void LoadModAssemblies(Manifest m)
+		{
+			var asms = new List<Pair<Assembly, string>>();
+
+			// all the core stuff is in this assembly
+			asms.Add(Pair.New(typeof(ITraitInfo).Assembly, typeof(ITraitInfo).Namespace));
+
+			// add the mods
+			foreach (var a in m.Assemblies)
+				asms.Add(Pair.New(Assembly.LoadFile(Path.GetFullPath(a)), Path.GetFileNameWithoutExtension(a)));
+			ModAssemblies = asms.ToArray();
+		}
+
+		static ITraitInfo LoadTraitInfo(string traitName, MiniYaml my)
+		{
+			if (traitName.Contains('@'))
+				traitName = traitName.Substring(0, traitName.IndexOf('@'));
+
+			foreach (var mod in ModAssemblies)
+			{
+				var fullTypeName = mod.Second + "." + traitName + "Info";
+				var info = (ITraitInfo)mod.First.CreateInstance(fullTypeName);
+				if (info == null) continue;
+				FieldLoader.Load(info, my);
+				return info;
+			}
+
+			throw new InvalidOperationException("Cannot locate trait: {0}".F(traitName));
+		}
+
+		public IEnumerable<ITraitInfo> TraitsInConstructOrder()
+		{
+			var ret = new List<ITraitInfo>();
+			var t = Traits.WithInterface<ITraitInfo>().ToList();
+			int index = 0;
+			while( t.Count != 0 )
+			{
+				if( index >= t.Count )
+					throw new InvalidOperationException( "Trait prerequisites not satisfied (or prerequisite loop)" );
+
+				var prereqs = PrerequisitesOf( t[ index ] );
+				if( prereqs.Count == 0 || prereqs.All( n => ret.Any( x => x.GetType().IsSubclassOf( n ) ) ) )
+				{
+					ret.Add( t[ index ] );
+					t.RemoveAt( index );
+					index = 0;
+				}
+				else
+					++index;
+			}
+
+			return ret;
+		}
+
+		static List<Type> PrerequisitesOf( ITraitInfo info )
+		{
+			return info
+				.GetType()
+				.GetInterfaces()
+				.Where( t => t.IsGenericType && t.GetGenericTypeDefinition() == typeof( ITraitPrerequisite<> ) )
+				.Select( t => t.GetGenericArguments()[ 0 ] )
+				.ToList();
+		}
+	}
+}