# HG changeset patch
# User IBBoard <dev@ibboard.co.uk>
# Date 1346509755 -3600
# Node ID dc13cef2bc2bd31208b2b2205ff30991e6dcd4b5
# Parent  2193406301501cf0e10b5af22f8e1a067cb2a5a8
Re #420: Saved army does not save "contained" structure
 * Make sure GTK# UI loads nesting correctly

diff -r 219340630150 -r dc13cef2bc2b FrmMainWindow.cs
--- a/FrmMainWindow.cs	Wed Aug 08 20:53:54 2012 +0100
+++ b/FrmMainWindow.cs	Sat Sep 01 15:29:15 2012 +0100
@@ -337,7 +337,6 @@
 			TreeIter targetIter = TreeIter.Zero;
 			TreeIter iter;
 			model.GetIterFirst(out iter);
-			bool found = false;
 
 			do
 			{
@@ -649,14 +648,26 @@
 					logger.DebugFormat("Append category {0}", cat.Name);
 					iter = store.AppendValues(cat);
 
-					foreach (WFObjects.Unit unit in cat.GetUnits())
+					foreach (WFObjects.Unit unit in cat.GetMainUnits())
 					{
-						store.AppendValues(iter, unit);
+						AddUnitToTree(store, iter, unit);
 					}
 				}
 
 				logger.Debug("Finished loading tree categories");
 			}
+
+			treeUnits.ExpandAll();
+		}
+
+		private void AddUnitToTree(TreeStore store, TreeIter parentIter, WFObjects.Unit unit)
+		{
+			TreeIter unitIter = store.AppendValues(parentIter, unit);
+			
+			foreach (WFObjects.Unit childUnit in unit.ContainedUnits)
+			{
+				AddUnitToTree(store, unitIter, childUnit);
+			}
 		}
 
 		private void DisableCategoryButtons()