# HG changeset patch
# User IBBoard <dev@ibboard.co.uk>
# Date 1261083295 0
# Node ID f4d3e64bdb1831e58a41d22360388416745ab90a
# Parent  6871c9788ce13fa05d43e73e3f893fa2e26fb231
Fixes #221: Changing unit size doesn't update text of equipment items
 * Simple fix - on unit size change, re-set all equipment item values in the table

diff -r 6871c9788ce1 -r f4d3e64bdb18 FrmUnit.cs
--- a/FrmUnit.cs	Sun Dec 13 20:15:26 2009 +0000
+++ b/FrmUnit.cs	Thu Dec 17 20:54:55 2009 +0000
@@ -460,6 +460,11 @@
 			if (obj is Unit && obj.Equals(unit))
 			{
 				unitSize.Value = newValue;
+
+				foreach (UnitEquipmentChoice choice in equipmentChoices.Values)
+				{
+					SetEquipmentListValue(choice);
+				}
 			}
 		}
 
@@ -490,17 +495,22 @@
 				}
 				else
 				{
-					int idx = equipmentList.Items.IndexOf(equipChoice);
+					SetEquipmentListValue(equipChoice);
+				}
+			}
+		}
 
-					if (idx > -1)
-					{
-						equipmentList.Items[idx] = equipChoice;
-					}
-					else
-					{
-						equipmentList.Items.Add(equipChoice);
-					}
-				}
+		private void SetEquipmentListValue(UnitEquipmentChoice equipChoice)
+		{
+			int idx = equipmentList.Items.IndexOf(equipChoice);
+
+			if (idx > -1)
+			{
+				equipmentList.Items[idx] = equipChoice;
+			}
+			else
+			{
+				equipmentList.Items.Add(equipChoice);
 			}
 		}