#region Copyright & License Information
/*
 * Copyright 2007-2010 The OpenRA Developers (see AUTHORS)
 * This file is part of OpenRA, which is free software. It is made 
 * available to you under the terms of the GNU General Public License
 * as published by the Free Software Foundation. For more information,
 * see LICENSE.
 */
#endregion

using System;
using System.Collections.Generic;
using System.Linq;
using OpenRA.Support;
using OpenRA.Traits;

namespace OpenRA
{
	public static class Exts
	{
		public static bool HasModifier(this Modifiers k, Modifiers mod)
		{
			return (k & mod) == mod;
		}

		public static IEnumerable<T> SymmetricDifference<T>(this IEnumerable<T> xs, IEnumerable<T> ys)
		{
			// this is probably a shockingly-slow way to do this, but it's concise.
			return xs.Except(ys).Concat(ys.Except(xs));
		}

		public static float Product(this IEnumerable<float> xs)
		{
			return xs.Aggregate(1f, (a, x) => a * x);
		}

		public static int GetMaxHP(this Actor self)
		{
			var oai = self.Info.Traits.GetOrDefault<OwnedActorInfo>();
			if (oai == null) return 0;
			return oai.HP;
		}

		public static V GetOrAdd<K, V>( this Dictionary<K, V> d, K k )
			where V : new()
		{
			return d.GetOrAdd( k, _ => new V() );
		}

		public static V GetOrAdd<K, V>( this Dictionary<K, V> d, K k, Func<K, V> createFn )
		{
			V ret;
			if( !d.TryGetValue( k, out ret ) )
				d.Add( k, ret = createFn( k ) );
			return ret;
		}

		public static T Random<T>(this IEnumerable<T> ts, Thirdparty.Random r)
		{
			var xs = ts.ToArray();
			return xs[r.Next(xs.Length)];
		}

		public static void DoTimed<T>( this IEnumerable<T> e, Action<T> a, string text, double time )
		{
			var sw = new Stopwatch();

			e.Do( x =>
			{
				var t = sw.ElapsedTime();
				a( x );
				var dt = sw.ElapsedTime() - t;
				if( dt > time )
					Log.Write("perf", text, x, dt * 1000);
			} );
		}
	}
}