Screen fade on chonotank teleport (needs tweaking to replicate real-ra)

OpenRa.Game/Traits/ChronoshiftDeploy.cs

@@ -1,19 +1,40 @@
 using System.Collections.Generic;
 using System.Linq;
 using OpenRa.Game.Orders;
+using System.Drawing;
 
 namespace OpenRa.Game.Traits
 {
-    class ChronoshiftDeploy : IOrder, ISpeedModifier, ITick, IPips
+	class ChronoshiftDeploy : IOrder, ISpeedModifier, ITick, IPips, IPaletteModifier
     {
         public ChronoshiftDeploy(Actor self) { }
+        // Recharge logic
         int remainingChargeTime = 0; // How long until we can chronoshift again?
         int chargeTime = (int)(Rules.Aftermath.ChronoTankDuration * 60 * 25); // How long between shifts?
 		
+		// Screen fade logic
+		int animationTick = 0;
+		int animationLength = 10;
+		bool animationStarted = false;
         public void Tick(Actor self)
         {
             if (remainingChargeTime > 0)
                 remainingChargeTime--;
+
+			if (animationStarted)
+			{
+				if (animationTick < animationLength)
+					animationTick++;
+				else
+					animationStarted = false;
+			}
+			if (!animationStarted)
+			{
+				if (animationTick > 0)
+					animationTick--;
+				//else
+				//	animationForwards = true;
+			}
         }
 
         public Order IssueOrder(Actor self, int2 xy, MouseInput mi, Actor underCursor)
@@ -42,6 +63,7 @@ namespace OpenRa.Game.Traits
            		self.QueueActivity(new Activities.Teleport(order.TargetLocation));
                 Sound.Play("chrotnk1.aud");
                 remainingChargeTime = chargeTime;
+                animationStarted = true;
             }
         }
         
@@ -78,5 +100,51 @@ namespace OpenRa.Game.Traits
 				}
 			}
 		}
+
+		public void AdjustPalette(Bitmap bmp)
+		{
+			if (!animationStarted && animationTick == 0)
+				return;
+
+			// saturation modifier
+			var f = 1 - (animationTick * 1.0f / animationLength);
+
+			using (var bitmapCopy = new Bitmap(bmp))
+				for (int j = 0; j < 8; j++)
+					for (int i = 0; i < bmp.Width; i++)
+					{
+						var h = bitmapCopy.GetPixel(i, j).GetHue(); // 0-360
+						var s = f * bitmapCopy.GetPixel(i, j).GetSaturation(); // 0-1.0
+						var l = bitmapCopy.GetPixel(i, j).GetBrightness(); // 0-1.0
+						var alpha = bitmapCopy.GetPixel(i, j).A;
+
+						// Convert from HSL to RGB
+						// Refactor me!
+						var q = (l < 0.5f) ? l * (1 + s) : l + s - (l * s);
+						var p = 2 * l - q;
+						var hk = h / 360.0f;
+
+						double[] trgb = { hk + 1 / 3.0f,
+										  hk,
+										  hk - 1/3.0f };
+						double[] rgb = { 0, 0, 0 };
+
+						for (int k = 0; k < 3; k++)
+						{
+							// mod doesn't seem to work right... do it manually
+							while (trgb[k] < 0) trgb[k] += 1.0f;
+							while (trgb[k] > 1) trgb[k] -= 1.0f;
+						}
+
+						for (int k = 0; k < 3; k++)
+						{
+							if (trgb[k] < 1 / 6.0f) { rgb[k] = (p + ((q - p) * 6 * trgb[k])); }
+							else if (trgb[k] >= 1 / 6.0f && trgb[k] < 0.5) { rgb[k] = q; }
+							else if (trgb[k] >= 0.5f && trgb[k] < 2.0f / 3) { rgb[k] = (p + ((q - p) * 6 * (2.0f / 3 - trgb[k]))); }
+							else { rgb[k] = p; }
+						}
+						bmp.SetPixel(i, j, Color.FromArgb(alpha, (int)(rgb[0] * 255), (int)(rgb[1] * 255), (int)(rgb[2] * 255)));
+					}
+		}
     }
 }