Correct bridge / tree / mine color

2010-07-22 19:49:27 +12:00
parent 5f36933837
commit f7212ef757
9 changed files with 71 additions and 3 deletions
--- a/OpenRA.Game/Traits/Unit.cs
+++ b/OpenRA.Game/Traits/Unit.cs
@@ -10,6 +10,7 @@

 using System.Collections.Generic;
 using System.Drawing;
+using System.Linq;

 namespace OpenRA.Traits
 {
@@ -43,6 +44,10 @@ namespace OpenRA.Traits
 		
 		public Color RadarSignatureColor(Actor self)
 		{
+			var mod = self.traits.WithInterface<IRadarSignatureModifier>().FirstOrDefault();
+			if (mod != null)
+				return mod.RadarColorOverride(self);
+			
 			return self.Owner.Color;
 		}
 	}