Generally implemented tracking of nearest valid relay even if out of range.
[AntennaRange.git] / RelayExtensions.cs
blob:a/RelayExtensions.cs -> blob:b/RelayExtensions.cs
--- a/RelayExtensions.cs
+++ b/RelayExtensions.cs
@@ -78,7 +78,7 @@
 			return relay.vessel.sqrDistanceTo(body);
 		}
 
-		public static double sqrDistanceTo(this AntennaRelay relayOne, AntennaRelay relayTwo)
+		public static double sqrDistanceTo(this AntennaRelay relayOne, IAntennaRelay relayTwo)
 		{
 			return relayOne.vessel.sqrDistanceTo(relayTwo.vessel);
 		}
@@ -89,7 +89,7 @@
 		/// <param name="vessel">This <see cref="Vessel"/></param>
 		public static IEnumerable<IAntennaRelay> GetAntennaRelays (this Vessel vessel)
 		{
-			return RelayDatabase.Instance[vessel].Values.ToList();
+			return RelayDatabase.Instance[vessel].Values.ToList().AsReadOnly();
 		}
 
 		/// <summary>
@@ -109,6 +109,59 @@
 
 			return false;
 		}
+
+		public static ConnectionStatus GetConnectionStatus(this Vessel vessel)
+		{
+			bool canTransmit = false;
+
+			foreach (IAntennaRelay relay in RelayDatabase.Instance[vessel].Values)
+			{
+				if (relay.CanTransmit())
+				{
+					canTransmit = true;
+					if (relay.transmitDistance <= relay.nominalTransmitDistance)
+					{
+						return ConnectionStatus.Optimal;
+					}
+				}
+			}
+
+			if (canTransmit)
+			{
+				return ConnectionStatus.Suboptimal;
+			}
+			else
+			{
+				return ConnectionStatus.None;
+			}
+		}
+
+		public static IAntennaRelay GetBestRelay(this Vessel vessel)
+		{
+			IAntennaRelay bestRelay = null;
+			double bestScore = double.PositiveInfinity;
+			double relayScore = double.NaN;
+
+			foreach (IAntennaRelay relay in vessel.GetAntennaRelays())
+			{
+				relayScore = relay.transmitDistance / relay.maxTransmitDistance;
+
+				if (relayScore < bestScore)
+				{
+					bestScore = relayScore;
+					bestRelay = relay;
+				}
+			}
+
+			return bestRelay;
+		}
+	}
+
+	public enum ConnectionStatus
+	{
+		None,
+		Suboptimal,
+		Optimal
 	}
 }