using System;
using System.Collections.Generic;
using System.Text;
using System.Windows.Forms;
using System.Drawing;
using BluntDirectX.Direct3D;
using OpenRa.FileFormats;
using System.IO;
using System.Runtime.InteropServices;

namespace OpenRa.Game
{
	class MainWindow : Form
	{
		readonly Renderer renderer;

		readonly Map map;
		readonly TileSet tileSet;
		
		Palette pal, playerPal;
		Package TileMix;
		string TileSuffix;

		Dictionary<TileReference, SheetRectangle<Sheet>> tileMapping =
			new Dictionary<TileReference, SheetRectangle<Sheet>>();

		FvfVertexBuffer<Vertex> vertexBuffer;

		Dictionary<Sheet, IndexBuffer> drawBatches = new Dictionary<Sheet, IndexBuffer>();

		World world;
		TreeCache treeCache;

		void LoadTextures()
		{
			List<Sheet> sheets = new List<Sheet>();

			Size pageSize = new Size(1024, 512);

			Provider<Sheet> sheetProvider = delegate
			{
				Sheet t = new Sheet(pageSize, renderer.Device);
				sheets.Add(t);
				return t;
			};

			TileSheetBuilder<Sheet> builder = new TileSheetBuilder<Sheet>(pageSize, sheetProvider);

			for (int i = 0; i < map.Width; i++)
				for (int j = 0; j < map.Height; j++)
				{
					TileReference tileRef = map.MapTiles[i + map.XOffset, j + map.YOffset];

					if (!tileMapping.ContainsKey(tileRef))
					{
						SheetRectangle<Sheet> rect = builder.AddImage(new Size(24, 24));
						Util.CopyIntoChannel(rect.sheet.bitmap, TextureChannel.Red,
							tileSet.tiles[tileRef.tile].TileBitmapBytes[tileRef.image], rect);

						tileMapping.Add(tileRef, rect);
					}
				}

			world = new World(renderer.Device);
			treeCache = new TreeCache(renderer.Device, map, TileMix);

			foreach (TreeReference treeReference in map.Trees)
				world.Add(new Tree(treeReference, treeCache, map));

			UnitSheetBuilder.Initialize(renderer.Device);
			UnitSheetBuilder.AddUnit("mcv");
			UnitSheetBuilder.AddUnit("1tnk");
			UnitSheetBuilder.AddUnit("2tnk");
			UnitSheetBuilder.AddUnit("3tnk");

			world.Add(new Mcv(new PointF(24 * 5, 24 * 5), 3));
			world.Add(new Mcv(new PointF(24 * 7, 24 * 5), 2));
			world.Add(new Mcv(new PointF(24 * 9, 24 * 5), 1));
		}

		void LoadVertexBuffer()
		{
			Dictionary<Sheet, List<ushort>> indexMap = new Dictionary<Sheet, List<ushort>>();
			List<Vertex> vertices = new List<Vertex>();

			for (int j = 0; j < map.Height; j++)
				for (int i = 0; i < map.Width; i++)
				{
					SheetRectangle<Sheet> tile = tileMapping[map.MapTiles[i + map.XOffset, j + map.YOffset]];

					List<ushort> indexList;
					if (!indexMap.TryGetValue(tile.sheet, out indexList))
						indexMap.Add(tile.sheet, indexList = new List<ushort>());

					Util.CreateQuad(vertices, indexList, new PointF(24 * i, 24 * j), tile, 0);
				}

			vertexBuffer = new FvfVertexBuffer<Vertex>(renderer.Device, vertices.Count, Vertex.Format);
			vertexBuffer.SetData(vertices.ToArray());

			foreach (KeyValuePair<Sheet, List<ushort>> p in indexMap)
			{
				IndexBuffer indexBuffer = new IndexBuffer(renderer.Device, p.Value.Count);
				indexBuffer.SetData(p.Value.ToArray());
				drawBatches.Add(p.Key, indexBuffer);
			}
		}

		static Size GetResolution(Settings settings)
		{
			Size desktopResolution = Screen.PrimaryScreen.Bounds.Size;

			return new Size(settings.GetValue("width", desktopResolution.Width),
				settings.GetValue("height", desktopResolution.Height));
		}

		public MainWindow( Settings settings )
		{
			FormBorderStyle = FormBorderStyle.None;
			renderer = new Renderer(this, GetResolution(settings), false);
			Visible = true;

			string mapName = settings.GetValue("map", "scm12ea.ini");

			IniFile mapFile = new IniFile(File.OpenRead("../../../" + mapName));
			map = new Map(mapFile);

			Text = string.Format("OpenRA - {0} - {1}", map.Title, mapName);

			tileSet = LoadTileSet(map);

			LoadTextures();
			LoadVertexBuffer();
		}

		internal void Run()
		{
			while (Created && Visible)
			{
				Frame();
				Application.DoEvents();
			}
		}

		PointF scrollPos = new PointF(1, 5);
		PointF oldPos;
		int x1,y1;

		protected override void OnMouseDown(MouseEventArgs e)
		{
			base.OnMouseDown(e);

			x1 = e.X;
			y1 = e.Y;
			oldPos = scrollPos;
		}

		protected override void OnMouseMove(MouseEventArgs e)
		{
			base.OnMouseMove(e);

			if (e.Button != 0)
			{
				scrollPos = oldPos;
				scrollPos.X += x1 - e.X;
				scrollPos.Y += y1 - e.Y;
			}
		}

		void Frame()
		{
			PointF r1 = new PointF(2.0f / ClientSize.Width, -2.0f / ClientSize.Height);
			PointF r2 = new PointF(-1, 1);

			renderer.BeginFrame(r1, r2, scrollPos);

			foreach (KeyValuePair<Sheet, IndexBuffer> batch in drawBatches)
				DrawTerrainBatch(batch);

			world.Draw(renderer);

			renderer.EndFrame();
		}

		void DrawTerrainBatch(KeyValuePair<Sheet, IndexBuffer> batch)
		{
			int indicesPerRow = map.Width * 6;
			int verticesPerRow = map.Width * 4;

			int visibleRows = (int)(ClientSize.Height / 24.0f + 2);

			int firstRow = (int)(scrollPos.Y / 24.0f);
			int lastRow = firstRow + visibleRows;

			if (lastRow < 0 || firstRow > map.Height)
				return;

			if (firstRow < 0) firstRow = 0;
			if (lastRow > map.Height) lastRow = map.Height;

			renderer.DrawWithShader(ShaderQuality.Low, delegate
			{
				renderer.DrawBatch(vertexBuffer, batch.Value, 
					new Range<int>(verticesPerRow * firstRow, verticesPerRow * lastRow), 
					new Range<int>(indicesPerRow * firstRow, indicesPerRow * lastRow), 
					batch.Key.Texture);
			});
		}

		TileSet LoadTileSet(Map currentMap)
		{
			string theaterName = currentMap.Theater;
			if (theaterName.Length > 8)
				theaterName = theaterName.Substring(0, 8);

			pal = new Palette(File.OpenRead("../../../" + theaterName + ".pal"));
			TileMix = new Package("../../../" + theaterName + ".mix");
			TileSuffix = "." + theaterName.Substring(0, 3);

			playerPal = new Palette(pal, new PaletteRemap(File.OpenRead("../../../red.rem")));

			HardwarePalette hardwarePalette = new HardwarePalette(renderer.Device);
			hardwarePalette.AddPalette(pal);

			foreach (string remap in new string[] { "blue", "red", "orange", "teal", "salmon", "green", "gray" })
				hardwarePalette.AddPalette(new Palette(pal, new PaletteRemap(
					File.OpenRead("../../../" + remap + ".rem"))));

			renderer.SetPalette(hardwarePalette);

			return new TileSet(TileMix, TileSuffix, pal);
		}
	}
}