package away3d.materials.utils
{
	import away3d.arcane;
	import away3d.core.base.ISubGeometry;
	import away3d.entities.Mesh;
	
	import flash.display.BitmapData;
	import flash.display.Graphics;
	import flash.display.Sprite;
	import flash.display.TriangleCulling;
	
	use namespace arcane;
	
	/**
	 * WireframeMapGenerator is a utility class to generate a wireframe texture for uniquely mapped meshes.
	 */
	public class WireframeMapGenerator
	{
		/**
		 * Create a wireframe map with a texture fill.
		 * @param mesh The Mesh object for which to create the wireframe texture.
		 * @param bitmapData The BitmapData to use as the fill texture.
		 * @param lineColor The wireframe's line colour.
		 * @param lineThickness The wireframe's line thickness.
		 */
		public static function generateTexturedMap(mesh:Mesh, bitmapData:BitmapData, lineColor:uint = 0xffffff, lineThickness:Number = 2):BitmapData
		{
			bitmapData = bitmapData.clone();
			
			for (var i:uint = 0; i < mesh.subMeshes.length; ++i)
				drawLines(lineColor, lineThickness, bitmapData, mesh.subMeshes[i].subGeometry);
			
			return bitmapData;
		}
		
		/**
		 * Create a wireframe map with a solid colour fill.
		 * @param mesh The Mesh object for which to create the wireframe texture.
		 * @param lineColor The wireframe's line colour.
		 * @param lineThickness The wireframe's line thickness.
		 * @param fillColor The colour of the wireframe fill.
		 * @param fillAlpha The alpha of the wireframe fill.
		 * @param width The texture's width.
		 * @param height The texture's height.
		 * @return A BitmapData containing the texture underneath the wireframe.
		 */
		public static function generateSolidMap(mesh:Mesh, lineColor:uint = 0xffffff, lineThickness:Number = 2, fillColor:uint = 0, fillAlpha:Number = 0, width:uint = 512, height:uint = 512):BitmapData
		{
			var bitmapData:BitmapData;
			
			if (fillAlpha > 1)
				fillAlpha = 1;
			else if (fillAlpha < 0)
				fillAlpha = 0;
			
			bitmapData = new BitmapData(width, height, fillAlpha == 1? false : true, (fillAlpha << 24) | (fillColor & 0xffffff));
			
			for (var i:uint = 0; i < mesh.subMeshes.length; ++i)
				drawLines(lineColor, lineThickness, bitmapData, mesh.subMeshes[i].subGeometry);
			
			return bitmapData;
		}
		
		/**
		 * Draws the actual lines.
		 */
		private static function drawLines(lineColor:uint, lineThickness:Number, bitmapData:BitmapData, subGeom:ISubGeometry):void
		{
			var sprite:Sprite = new Sprite();
			var g:Graphics = sprite.graphics;
			var uvs:Vector.<Number> = subGeom.UVData;
			var i:uint, len:uint = uvs.length;
			var w:Number = bitmapData.width, h:Number = bitmapData.height;
			var texSpaceUV:Vector.<Number> = new Vector.<Number>(len, true);
			var indices:Vector.<uint> = subGeom.indexData;
			var indexClone:Vector.<int>;
			
			do {
				texSpaceUV[i] = uvs[i]*w;
				++i;
				texSpaceUV[i] = uvs[i]*h;
			} while (++i < len);
			
			len = indices.length;
			indexClone = new Vector.<int>(len, true);
			i = 0;
			// awesome, just to convert from uint to int vector -_-
			do
				indexClone[i] = indices[i];
			while (++i < len);
			
			g.lineStyle(lineThickness, lineColor);
			g.drawTriangles(texSpaceUV, indexClone, null, TriangleCulling.NONE);
			bitmapData.draw(sprite);
			g.clear();
		}
	}
}