# HG changeset patch
# User IBBoard <dev@ibboard.co.uk>
# Date 1255291033 0
# Node ID e3488dd61a217bd9a80c835d4d63612816c5b6de
# Parent  431cc3ace26cad17e4c181dd7ec788097ce73ac8
Fixes #199: armytree unit point costs don't always show after update
 * Extract common node text setting method and use on unit name update

diff -r 431cc3ace26c -r e3488dd61a21 FrmArmyTree.cs
--- a/FrmArmyTree.cs	Tue Oct 06 13:39:50 2009 +0000
+++ b/FrmArmyTree.cs	Sun Oct 11 19:57:13 2009 +0000
@@ -241,7 +241,17 @@
 		{
 			if (obj is ICostedWarFoundryObject)
 			{
-				nodes[obj.ID].Text = CreateNodeName((ICostedWarFoundryObject)obj);
+				SetNodeName(obj);
+			}
+		}
+
+		private void SetNodeName(WarFoundryObject obj)
+		{
+			TreeNode node = nodes[obj.ID];
+
+			if (node != null)
+			{
+				node.Text = CreateNodeName((ICostedWarFoundryObject)obj);
 			}
 		}
 
@@ -393,15 +403,9 @@
 
 		private void UpdateUnitName(WarFoundryObject obj, string oldValue, string newValue)
 		{
-			if (obj is Unit)
+			if (obj is ICostedWarFoundryObject)
 			{
-				Unit unit = (Unit)obj;
-				TreeNode node = nodes[unit.ID];
-				
-				if (node!=null)
-				{
-					node.Text = unit.Name;
-				}
+				SetNodeName(obj);
 			}
 		}