using System.Collections.Generic;
using OpenRa.GameRules;
using OpenRa.Graphics;
using OpenRa.Traits;

namespace OpenRa.Effects
{
	public class Bullet : IEffect
	{
		readonly Player Owner;
		readonly Actor FiredBy;
		readonly WeaponInfo Weapon;
		readonly ProjectileInfo Projectile;
		readonly WarheadInfo Warhead;
		readonly int2 Src;
		readonly int2 Dest;
		readonly int2 VisualDest;
		readonly int SrcAltitude;
		readonly int DestAltitude;

		int t = 0;
		Animation anim;

		const int BaseBulletSpeed = 100;		/* pixels / 40ms frame */

		public Bullet(string weapon, Player owner, Actor firedBy,
			int2 src, int2 dest, int srcAltitude, int destAltitude)
			: this(Rules.WeaponInfo[weapon], owner, firedBy, src, dest, srcAltitude, destAltitude) { }

		/* src, dest are *pixel* coords */
		public Bullet(WeaponInfo weapon, Player owner, Actor firedBy, 
			int2 src, int2 dest, int srcAltitude, int destAltitude)
		{
			Owner = owner;
			FiredBy = firedBy;
			Src = src;
			Dest = dest;
			SrcAltitude = srcAltitude;
			DestAltitude = destAltitude;
			VisualDest = Dest + new int2(
						Game.CosmeticRandom.Next(-10, 10),
						Game.CosmeticRandom.Next(-10, 10));
			Weapon = weapon;
			Projectile = Rules.ProjectileInfo[Weapon.Projectile];
			Warhead = Rules.WarheadInfo[Weapon.Warhead];

			if (Projectile.Image != null && Projectile.Image != "none")
			{
				anim = new Animation(Projectile.Image);
				if (Projectile.Rotates)
					Traits.Util.PlayFacing(anim, "idle", () => Traits.Util.GetFacing((dest - src).ToFloat2(), 0));
				else
					anim.PlayRepeating("idle");
			}
		}

		int TotalTime() { return (Dest - Src).Length * BaseBulletSpeed / Weapon.Speed; }

		public void Tick()
		{
			t += 40;

			if (t > TotalTime())		/* remove finished bullets */
			{
				Game.world.AddFrameEndTask(w => w.Remove(this));
				Combat.DoImpact(Dest, VisualDest - new int2( 0, DestAltitude ), 
					Weapon, Projectile, Warhead, FiredBy);
			}
		}

		const float height = .1f;

		public IEnumerable<Renderable> Render()
		{
			if (anim != null)
			{
				var at = (float)t / TotalTime();

				var altitude = float2.Lerp(SrcAltitude, DestAltitude, at);
				var pos = float2.Lerp( Src.ToFloat2(), VisualDest.ToFloat2(), at)
					- 0.5f * anim.Image.size - new float2( 0, altitude );

				if (Projectile.High || Projectile.Arcing)
				{
					if (Projectile.Shadow)
						yield return new Renderable(anim.Image, pos, PaletteType.Shadow);

					var highPos = pos - new float2(0, (VisualDest - Src).Length * height * 4 * at * (1 - at));

					yield return new Renderable(anim.Image, highPos, Owner.Palette);
				}
				else
					yield return new Renderable(anim.Image, pos, Projectile.UnderWater ? PaletteType.Shadow : Owner.Palette);
			}
		}
	}
}