OpenRA/OpenRA.Game/Traits/Production.cs

#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.Collections.Generic;
using System.Linq;
using OpenRA.GameRules;
using System.Drawing;

namespace OpenRA.Traits
{
	public class ProductionInfo : TraitInfo<Production>
	{
		public readonly int[] SpawnOffset = null;
		public readonly int[] ProductionOffset = null;
		public readonly int[] ExitOffset = null;
		public readonly bool EnablePrimary = true;
		public readonly string[] Produces = { };
	}

	public class Production : IIssueOrder, IResolveOrder, ITags, IOrderCursor
	{	
		public virtual int2? CreationLocation( Actor self, ActorInfo producee )
		{
			var pos = Util.CellContaining(self.CenterLocation);
			var pi = self.Info.Traits.Get<ProductionInfo>();
			if (pi.ProductionOffset != null)
				pos += pi.ProductionOffset.AsInt2();
			return pos;
		}

		public virtual int2? ExitLocation(Actor self, ActorInfo producee)
		{
			var pos = Util.CellContaining(self.CenterLocation);
			var pi = self.Info.Traits.Get<ProductionInfo>();
			if (pi.ExitOffset != null)
				pos += pi.ExitOffset.AsInt2();
			return pos;
		}
		
		public virtual int CreationFacing( Actor self, Actor newUnit )
		{
			return newUnit.traits.Get<IMove>().InitialFacing;
		}

		public virtual bool Produce( Actor self, ActorInfo producee )
		{
			var location = CreationLocation( self, producee );
			if( location == null || self.World.WorldActor.traits.Get<UnitInfluence>().GetUnitsAt( location.Value ).Any() )
				return false;

			var newUnit = self.World.CreateActor( producee.Name, location.Value, self.Owner );
			newUnit.traits.Get<Unit>().Facing = CreationFacing( self, newUnit ); ;

			var pi = self.Info.Traits.Get<ProductionInfo>();
			var rp = self.traits.GetOrDefault<RallyPoint>();
			if (rp != null || pi.ExitOffset != null)
			{
				var mobile = newUnit.traits.GetOrDefault<Mobile>();
				if (mobile != null)
				{
					int2? target = null;
					if (pi.ExitOffset != null)
					{
						target = ExitLocation(self, producee).Value;
						newUnit.QueueActivity(new Activities.Move(target.Value, 1));
					}

					if (rp != null)
					{
						target = rp.rallyPoint;
						newUnit.QueueActivity(new Activities.Move(target.Value, 1));
					}
					
					if (target != null && newUnit.Owner == self.World.LocalPlayer)
					{
						self.World.AddFrameEndTask(w =>
						{
							var line = newUnit.traits.GetOrDefault<DrawLineToTarget>();
							if (line != null)
								line.SetTargetSilently(newUnit, Target.FromCell(target.Value), Color.Green);
						});
					}
				}
			}

			if (pi != null && pi.SpawnOffset != null)
				newUnit.CenterLocation = self.CenterLocation + pi.SpawnOffset.AsInt2();

			foreach (var t in self.traits.WithInterface<INotifyProduction>())
				t.UnitProduced(self, newUnit);

			Log.Write("debug", "{0} #{1} produced by {2} #{3}", newUnit.Info.Name, newUnit.ActorID, self.Info.Name, self.ActorID);

			return true;
		}

		// "primary building" crap - perhaps this should be split?

		bool isPrimary = false;
		public bool IsPrimary { get { return isPrimary; } }

		public IEnumerable<TagType> GetTags()
		{
			yield return (isPrimary) ? TagType.Primary : TagType.None;
		}

		public Order IssueOrder(Actor self, int2 xy, MouseInput mi, Actor underCursor)
		{			
			if (mi.Button == MouseButton.Right && underCursor == self && self.Info.Traits.Get<ProductionInfo>().EnablePrimary)
				return new Order("PrimaryProducer", self);
			return null;
		}
		
		public string CursorForOrder(Actor self, Order order)
		{
			return (order.OrderString == "PrimaryProducer") ? "deploy" : null;
		}
		
		public void ResolveOrder(Actor self, Order order)
		{
			if (order.OrderString == "PrimaryProducer")
				SetPrimaryProducer(self, !isPrimary);
		}
		
		public void SetPrimaryProducer(Actor self, bool state)
		{
			if (state == false)
			{
				isPrimary = false;
				return;
			}
			
			// Cancel existing primaries
			foreach (var p in self.Info.Traits.Get<ProductionInfo>().Produces)
			{
				foreach (var b in self.World.Queries.OwnedBy[self.Owner]
					.WithTrait<Production>()
					.Where(x => x.Trait.IsPrimary
						&& (x.Actor.Info.Traits.Get<ProductionInfo>().Produces.Contains(p))))
				{
					b.Trait.SetPrimaryProducer(b.Actor, false);
				}
			}
			isPrimary = true;
			
			var eva = self.World.WorldActor.Info.Traits.Get<EvaAlertsInfo>();
			Sound.PlayToPlayer(self.Owner,eva.PrimaryBuildingSelected);
		}
	}
}