#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.Linq;
using OpenRA.Mods.RA.Activities;
using OpenRA.Traits;
using System.Collections.Generic;
using System;
using System.Drawing;

namespace OpenRA.Mods.RA
{
	class MinelayerInfo : TraitInfo<Minelayer>
	{
		[ActorReference]
		public readonly string Mine = "minv";
		public readonly float MinefieldDepth = 1.5f;
		[ActorReference]
		public readonly string[] RearmBuildings = { "fix" };
	}

	class Minelayer : IIssueOrder, IResolveOrder, IProvideCursor
	{
		/* [Sync] when sync can cope with arrays! */ public int2[] minefield = null;
		[Sync] int2 minefieldStart;

		public Order IssueOrder(Actor self, int2 xy, MouseInput mi, Actor underCursor)
		{
			if (mi.Button == MouseButton.Right && underCursor == null && mi.Modifiers.HasModifier(Modifiers.Ctrl))
				return new Order("BeginMinefield", self, xy);

			return null;
		}

		public string CursorForOrderString(string s, Actor a, int2 location)
		{
			return (s == "BeginMinefield") ? "ability" : null;
		}
		
		public void ResolveOrder(Actor self, Order order)
		{
			if (order.OrderString == "BeginMinefield")
			{
				minefieldStart = order.TargetLocation;
				if (self.Owner == self.World.LocalPlayer)
					Game.controller.orderGenerator = new MinefieldOrderGenerator(self);
			}

			if (order.OrderString == "PlaceMinefield")
			{
				if (self.Owner == self.World.LocalPlayer)
					Game.controller.CancelInputMode();

				var movement = self.traits.Get<IMove>();

				minefield = GetMinefieldCells(minefieldStart, order.TargetLocation,
					self.Info.Traits.Get<MinelayerInfo>().MinefieldDepth)
					.Where(p => movement.CanEnterCell(p)).ToArray();

				/* todo: start the mnly actually laying mines there */
				self.CancelActivity();
				self.QueueActivity(new LayMines());
			}
		}

		static IEnumerable<int2> GetMinefieldCells(int2 start, int2 end, float depth)
		{
			var mins = int2.Min(start, end);
			var maxs = int2.Max(start, end);

			/* todo: proper endcaps, if anyone cares (which won't happen unless depth is large) */

			var p = end - start;
			var q = new float2(p.Y, -p.X);
			q = (start != end) ? (1 / q.Length) * q : new float2(1, 0);
			var c = -float2.Dot(q, start);

			/* return all points such that |ax + by + c| < depth */

			for (var i = mins.X; i <= maxs.X; i++)
				for (var j = mins.Y; j <= maxs.Y; j++)
					if (Math.Abs(q.X * i + q.Y * j + c) < depth)
						yield return new int2(i, j);
		}

		class MinefieldOrderGenerator : IOrderGenerator
		{
			Actor minelayer;

			public MinefieldOrderGenerator(Actor self) { minelayer = self; }

			public IEnumerable<Order> Order(World world, int2 xy, MouseInput mi)
			{
				if (mi.Button == MouseButton.Left)
				{
					Game.controller.CancelInputMode();
					yield break;
				}

				var underCursor = world.FindUnitsAtMouse(mi.Location)
					.Where(a => a.Info.Traits.Contains<SelectableInfo>())
					.OrderByDescending(a => a.Info.Traits.Get<SelectableInfo>().Priority)
					.FirstOrDefault();

				if (mi.Button == MouseButton.Right && underCursor == null)
					yield return new Order("PlaceMinefield", minelayer, xy);
			}

			public void Tick(World world)
			{
				if (minelayer.IsDead || !minelayer.IsInWorld)
					Game.controller.CancelInputMode();
			}

			int2 lastMousePos;
			public void Render(World world)
			{
				var ml = minelayer.traits.Get<Minelayer>();
				var movement = minelayer.traits.Get<IMove>();
				var minefield = GetMinefieldCells(ml.minefieldStart, lastMousePos, minelayer.Info.Traits.Get<MinelayerInfo>().MinefieldDepth)
					.Where(p => movement.CanEnterCell(p)).ToArray();

				Game.world.WorldRenderer.DrawLocus(Color.Cyan, minefield);
			}

			public string GetCursor(World world, int2 xy, MouseInput mi) { lastMousePos = xy; return "ability"; }	/* todo */
		}
	}
}