Nuke flash (fixes #16)

OpenRA.Mods.RA/NukePaletteEffect.cs

@@ -0,0 +1,62 @@
+#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.Drawing;
+using OpenRA.Traits;
+using OpenRA.Graphics;
+
+namespace OpenRA.Mods.RA
+{
+	class NukePaletteEffectInfo : TraitInfo<NukePaletteEffect> { }
+
+	public class NukePaletteEffect : IPaletteModifier, ITick
+	{
+		const int nukeEffectLength = 20;
+		int remainingFrames;
+
+		public void Enable()
+		{
+			remainingFrames = nukeEffectLength;
+		}
+
+		public void Tick(Actor self)
+		{
+			if (remainingFrames > 0)
+				remainingFrames--;
+		}
+
+		public void AdjustPalette(Bitmap b)
+		{
+			if (remainingFrames == 0)
+				return;
+			
+			var frac = (float)remainingFrames / nukeEffectLength;
+
+			// TODO: Fix me to only affect "world" palettes
+			for( var y = 0; y < b.Height; y++ )
+				for (var x = 0; x < 256; x++)
+				{
+					var orig = b.GetPixel(x, y);
+					var white = Color.FromArgb(orig.A, 255, 255, 255);
+					b.SetPixel(x, y, OpenRA.Graphics.Util.Lerp(frac, orig, white));
+				}
+		}
+	}
+}