Nuke flash (fixes #16)

OpenRA.Mods.Aftermath/ChronoshiftDeploy.cs

@@ -1,4 +1,4 @@
-#region Copyright & License Information
+#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.
@@ -77,7 +77,7 @@ namespace OpenRA.Mods.Aftermath
 				chargeTick = 25 * self.Info.Traits.Get<ChronoshiftDeployInfo>().ChargeTime;
 
 				foreach (var a in self.World.Queries.WithTrait<ChronoshiftPaletteEffect>())
-					a.Trait.DoChronoshift();
+					a.Trait.Enable();
 			}
 		}
 

OpenRA.Mods.RA/ChronoshiftPaletteEffect.cs

@@ -31,7 +31,7 @@ namespace OpenRA.Mods.RA
 		const int chronoEffectLength = 20;
 		int remainingFrames;
 
-		public void DoChronoshift()
+		public void Enable()
 		{
 			remainingFrames = chronoEffectLength;
 		}

OpenRA.Mods.RA/ChronoshiftPower.cs

@@ -183,9 +183,10 @@ namespace OpenRA.Mods.RA
 			{
 				Sound.Play("chrono2.aud", self.CenterLocation);
 				Sound.Play("chrono2.aud", targetActor.CenterLocation);
+				
 				// Trigger screen desaturate effect
 				foreach (var a in self.World.Queries.WithTrait<ChronoshiftPaletteEffect>())
-					a.Trait.DoChronoshift();
+					a.Trait.Enable();
 
 				self.traits.Get<RenderBuilding>().PlayCustomAnim(self, "active");
 			}

OpenRA.Mods.RA/Effects/NukeLaunch.cs

@@ -80,7 +80,13 @@ namespace OpenRA.Mods.RA.Effects
 			{
 				altitude -= 10;
 				if (altitude <= 0)
+				{
+					// Trigger screen desaturate effect
+					foreach (var a in Game.world.Queries.WithTrait<NukePaletteEffect>())
+						a.Trait.Enable();
+					
 					Explode(world);
+				}
 			}
 		}
 

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));
+				}
+		}
+	}
+}

OpenRA.Mods.RA/OpenRA.Mods.RA.csproj

@@ -1,4 +1,4 @@
-<?xml version="1.0" encoding="utf-8"?>
+<?xml version="1.0" encoding="utf-8"?>
 <Project ToolsVersion="3.5" DefaultTargets="Build" xmlns="http://schemas.microsoft.com/developer/msbuild/2003">
   <PropertyGroup>
     <Configuration Condition=" '$(Configuration)' == '' ">Debug</Configuration>
@@ -177,6 +177,7 @@
     <Compile Include="WaterPaletteRotation.cs" />
     <Compile Include="WithMuzzleFlash.cs" />
     <Compile Include="WithShadow.cs" />
+    <Compile Include="NukePaletteEffect.cs" />
   </ItemGroup>
   <ItemGroup>
     <ProjectReference Include="..\OpenRA.FileFormats\OpenRA.FileFormats.csproj">

mods/ra/system.yaml

@@ -88,6 +88,7 @@ World:
 	ScreenShaker:
 	WaterPaletteRotation:
 	ChronoshiftPaletteEffect:
+	NukePaletteEffect:
 	LightPaletteRotator:
 	BuildingInfluence:
 	UnitInfluence: