From bfd2a9629fbe2a99593171296300bb04c68d2104 Mon Sep 17 00:00:00 2001
From: Nick Hall <nick__hall@hotmail.com>
Date: Wed, 16 Dec 2009 22:56:55 +0000
Subject: [PATCH] Improved export for hierarchical views

svn: r13830
---
 src/gui/views/listview.py                  | 35 ++++++++++++++--------
 src/gui/views/treemodels/peoplemodel.py    |  6 ++++
 src/gui/views/treemodels/placetreemodel.py |  6 ++++
 3 files changed, 35 insertions(+), 12 deletions(-)

diff --git a/src/gui/views/listview.py b/src/gui/views/listview.py
index 0734860f4..b3afef0e4 100644
--- a/src/gui/views/listview.py
+++ b/src/gui/views/listview.py
@@ -803,8 +803,16 @@ class ListView(NavigationView):
         ofile.open(name)
         ofile.start_page()
         ofile.start_row()
+
         # Headings
-        map(ofile.write_cell, column_names)
+        if self.model.get_flags() & gtk.TREE_MODEL_LIST_ONLY:
+            headings = column_names
+        else:
+            levels = self.model.get_tree_levels()
+            headings = levels + column_names[1:]
+            data_cols = data_cols[1:]
+
+        map(ofile.write_cell, headings)
         ofile.end_row()
 
         if self.model.get_flags() & gtk.TREE_MODEL_LIST_ONLY:
@@ -816,25 +824,28 @@ class ListView(NavigationView):
                 ofile.end_row()
         else:
             # Tree model
-            node = self.model.get_iter_first()
-            self.write_node(node, 0, ofile, data_cols)
+            node = self.model.on_get_iter((0,))
+            self.write_node(node, len(levels), [], ofile, data_cols)
         
         ofile.end_page()
         ofile.close()
         
-    def write_node(self, node, level, ofile, data_cols):
+    def write_node(self, node, depth, level, ofile, data_cols):
         if node is None:
             return
         while node is not None:
-            ofile.start_row()
-            map(ofile.write_cell, ('',)*level)
-            for index in data_cols:
-                ofile.write_cell(self.model.get_value(node, index))
-            ofile.end_row()
+            new_level = level + [self.model.on_get_value(node, 0)]
+            if self.model.get_handle(node):
+                ofile.start_row()
+                padded_level = new_level + [''] * (depth - len(new_level))
+                map(ofile.write_cell, padded_level)
+                for index in data_cols:
+                    ofile.write_cell(self.model.on_get_value(node, index))
+                ofile.end_row()
 
-            first_child = self.model.iter_children(node)
-            self.write_node(first_child, level + 1, ofile, data_cols)
-            node = self.model.iter_next(node)
+            first_child = self.model.on_iter_children(node)
+            self.write_node(first_child, depth, new_level, ofile, data_cols)
+            node = self.model.on_iter_next(node)
 
     ####################################################################
     # Template functions
diff --git a/src/gui/views/treemodels/peoplemodel.py b/src/gui/views/treemodels/peoplemodel.py
index 17db432fe..d53a96b3f 100644
--- a/src/gui/views/treemodels/peoplemodel.py
+++ b/src/gui/views/treemodels/peoplemodel.py
@@ -159,6 +159,12 @@ class PeopleModel(TreeBaseModel):
         """ Return the number of columns in the model """
         return len(self.fmap)+1
 
+    def get_tree_levels(self):
+        """
+        Return the headings of the levels in the hierarchy.
+        """
+        return ['Group As', 'Name']
+
     def add_row(self, handle, data):
         """
         Add nodes to the node map for a single person.
diff --git a/src/gui/views/treemodels/placetreemodel.py b/src/gui/views/treemodels/placetreemodel.py
index 7a3894645..b2f7bbcec 100644
--- a/src/gui/views/treemodels/placetreemodel.py
+++ b/src/gui/views/treemodels/placetreemodel.py
@@ -73,6 +73,12 @@ class PlaceTreeModel(PlaceBaseModel, TreeBaseModel):
                                 nrgroups = 3,
                                 group_can_have_handle = True)
 
+    def get_tree_levels(self):
+        """
+        Return the headings of the levels in the hierarchy.
+        """
+        return ['Country', 'State', 'County', 'Place Name']
+
     def add_row(self, handle, data):
         """
         Add nodes to the node map for a single place.