From a6276cd38f4fca2e87ddc447c06a8ae997725b3f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Agn=C3=A8s=20Charrel-Berthillier?= <agnes@charrel.net>
Date: Fri, 27 Jan 2017 13:54:35 +1100
Subject: [PATCH] 9914: Add [Arrowhead direction] option to Hourglass graph

---
 gramps/plugins/graph/gvhourglass.py | 38 +++++++++++++++++++++++++----
 1 file changed, 33 insertions(+), 5 deletions(-)

diff --git a/gramps/plugins/graph/gvhourglass.py b/gramps/plugins/graph/gvhourglass.py
index 3662898fd..6f46cc278 100644
--- a/gramps/plugins/graph/gvhourglass.py
+++ b/gramps/plugins/graph/gvhourglass.py
@@ -58,6 +58,11 @@ _COLORS = [{'name' : _("B&W outline"), 'value' : "outline"},
            {'name' : _("Colored outline"), 'value' : "colored"},
            {'name' : _("Color fill"), 'value' : "filled"}]
 
+_ARROWS = [ { 'name' : _("Center -> Others"),  'value' : 'o' },
+            { 'name' : _("Center <- Others"),  'value' : 'c' },
+            { 'name' : _("Center <-> Other"), 'value' : 'co' },
+            { 'name' : _("Center - Other"),   'value' : '' }]
+
 #------------------------------------------------------------------------
 #
 # HourGlassReport
@@ -109,6 +114,16 @@ class HourGlassReport(Report):
 
         self.includeid = menu.get_option_by_name('incid').get_value()
 
+        arrow_str = menu.get_option_by_name('arrow').get_value()
+        if 'o' in arrow_str:
+            self.arrowheadstyle = 'normal'
+        else:
+            self.arrowheadstyle = 'none'
+        if 'c' in arrow_str:
+            self.arrowtailstyle = 'normal'
+        else:
+            self.arrowtailstyle = 'none'
+
         stdoptions.run_name_format_option(self, menu)
 
     def write_report(self):
@@ -128,7 +143,9 @@ class HourGlassReport(Report):
         for family_handle in person.get_family_handle_list():
             family = self.__db.get_family_from_handle(family_handle)
             self.add_family(family)
-            self.doc.add_link(person.get_gramps_id(), family.get_gramps_id())
+            self.doc.add_link(person.get_gramps_id(), family.get_gramps_id(),
+                               head=self.arrowheadstyle,
+                               tail=self.arrowtailstyle)
             for child_ref in family.get_child_ref_list():
                 child_handle = child_ref.get_reference_handle()
                 if child_handle not in self.__used_people:
@@ -137,7 +154,9 @@ class HourGlassReport(Report):
                     child = self.__db.get_person_from_handle(child_handle)
                     self.add_person(child)
                     self.doc.add_link(family.get_gramps_id(),
-                                      child.get_gramps_id())
+                                      child.get_gramps_id(),
+                                      head=self.arrowheadstyle,
+                                      tail=self.arrowtailstyle)
                     self.traverse_down(child, gen+1)
 
     def traverse_up(self, person, gen):
@@ -152,7 +171,8 @@ class HourGlassReport(Report):
             family_id = family.get_gramps_id()
             self.add_family(family)
             self.doc.add_link(family_id, person.get_gramps_id(),
-                              head='none', tail='normal')
+                              head=self.arrowtailstyle,
+                              tail=self.arrowheadstyle )
 
             # create link from family to father
             father_handle = family.get_father_handle()
@@ -162,7 +182,8 @@ class HourGlassReport(Report):
                 father = self.__db.get_person_from_handle(father_handle)
                 self.add_person(father)
                 self.doc.add_link(father.get_gramps_id(), family_id,
-                                  head='none', tail='normal')
+                                  head=self.arrowtailstyle,
+                                  tail=self.arrowheadstyle )
                 # no need to go up if he is a father in another family
                 if father_handle not in self.__used_people:
                     self.__used_people.append(father_handle)
@@ -176,7 +197,8 @@ class HourGlassReport(Report):
                 mother = self.__db.get_person_from_handle(mother_handle)
                 self.add_person(mother)
                 self.doc.add_link(mother.get_gramps_id(), family_id,
-                                  head='none', tail='normal')
+                                  head=self.arrowtailstyle,
+                                  tail=self.arrowheadstyle)
                 # no need to go up if she is a mother in another family
                 if mother_handle not in self.__used_people:
                     self.__used_people.append(mother_handle)
@@ -347,6 +369,12 @@ class HourGlassOptions(MenuReportOptions):
         color_family.set_help(_('The color to use to display families.'))
         menu.add_option(category_name, 'colorfamilies', color_family)
 
+        arrow = EnumeratedListOption(_("Arrowhead direction"), 'o')
+        for i in range( 0, len(_ARROWS) ):
+            arrow.add_item(_ARROWS[i]["value"], _ARROWS[i]["name"])
+        arrow.set_help(_("Choose the direction that the arrows point."))
+        menu.add_option(category_name, "arrow", arrow)
+
         roundedcorners = BooleanOption(_("Use rounded corners"), False) # 2180
         roundedcorners.set_help(
             _("Use rounded corners to differentiate between women and men."))