add --shp and --png commands to util

OpenRA.Utility/Command.cs

@@ -15,6 +15,11 @@ using System.Windows.Forms;
 using ICSharpCode.SharpZipLib;
 using ICSharpCode.SharpZipLib.Zip;
 using OpenRA.GameRules;
+using OpenRA.FileFormats.Graphics;
+using System.Drawing;
+using System.Drawing.Imaging;
+using System.Runtime.InteropServices;
+using OpenRA.FileFormats;
 
 namespace OpenRA.Utility
 {
@@ -145,5 +150,67 @@ namespace OpenRA.Utility
         public static void AuthenticateAndExtractZip(string[] args) { AuthenticateAndExecute("--extract-zip-inner", args); }
 		public static void AuthenticateAndInstallRAPackages(string[] args) { AuthenticateAndExecute( "--install-ra-packages-inner", args ); }
 		public static void AuthenticateAndInstallCncPackages(string[] args) { AuthenticateAndExecute( "--install-cnc-packages-inner", args ); }
+
+		public static void ConvertPngToShp(string[] args)
+		{
+			var src = args[1];
+			var dest = Path.ChangeExtension(src, ".shp");
+			var width = int.Parse(args[2]);
+
+			var srcImage = PngLoader.Load(src);
+
+			if (srcImage.Width % width != 0)
+				throw new InvalidOperationException("Bogus width; not a whole number of frames");
+
+			using (var destStream = File.Create(dest))
+				ShpWriter.Write(destStream, width, srcImage.Height,
+					srcImage.ToFrames(width));
+		}
+
+		static IEnumerable<byte[]> ToFrames(this Bitmap bitmap, int width)
+		{
+			for (var x = 0; x < bitmap.Width; x += width)
+			{
+				var data = bitmap.LockBits(new Rectangle(x, 0, width, bitmap.Height), ImageLockMode.ReadOnly,
+					PixelFormat.Format8bppIndexed);
+
+				var bytes = new byte[width * bitmap.Height];
+				for (var i = 0; i < bitmap.Height; i++)
+					Marshal.Copy(new IntPtr(data.Scan0.ToInt64() + i * data.Stride),
+						bytes, i * width, width);
+
+				bitmap.UnlockBits(data);
+
+				yield return bytes;
+			}
+		}
+
+		public static void ConvertShpToPng(string[] args)
+		{
+			var src = args[1];
+			var dest = Path.ChangeExtension(src, ".png");
+
+			var srcImage = ShpReader.Load(src);
+
+			using (var bitmap = new Bitmap(srcImage.ImageCount * srcImage.Width, srcImage.Height, PixelFormat.Format8bppIndexed))
+			{
+				var x = 0;
+
+				foreach (var frame in srcImage)
+				{
+					var data = bitmap.LockBits(new Rectangle(x, 0, srcImage.Width, srcImage.Height), ImageLockMode.WriteOnly,
+						PixelFormat.Format8bppIndexed);
+
+					for (var i = 0; i < bitmap.Height; i++)
+						Marshal.Copy(frame.Image, i * srcImage.Width,
+							new IntPtr(data.Scan0.ToInt64() + i * data.Stride), srcImage.Width);
+
+					Marshal.Copy(frame.Image, 0, data.Scan0, frame.Image.Length);
+					x += srcImage.Width;
+				}
+
+				bitmap.Save(dest);
+			}
+		}
 	}
 }