2007-12-20 11:56:38 +05:30
|
|
|
#
|
|
|
|
# Gramps - a GTK+/GNOME based genealogy program
|
|
|
|
#
|
2008-01-18 11:09:50 +05:30
|
|
|
# Copyright (C) 2007-2008 Brian G. Matherly
|
2007-12-20 11:56:38 +05:30
|
|
|
#
|
|
|
|
# Adapted from GraphViz.py (now deprecated)
|
|
|
|
# Copyright (C) 2000-2007 Donald N. Allingham
|
2008-05-12 00:37:30 +05:30
|
|
|
# Copyright (C) 2005-2006 Eero Tamminen
|
2007-12-20 11:56:38 +05:30
|
|
|
# Copyright (C) 2007 Johan Gonqvist <johan.gronqvist@gmail.com>
|
|
|
|
# Contributions by Lorenzo Cappelletti <lorenzo.cappelletti@email.it>
|
2008-05-31 14:12:38 +05:30
|
|
|
# Copyright (C) 2008 Stephane Charette <stephanecharette@gmail.com>
|
2009-01-25 20:34:22 +05:30
|
|
|
# Copyright (C) 2009 Gary Burton
|
2007-12-20 11:56:38 +05:30
|
|
|
#
|
|
|
|
# This program is free software; you can redistribute it and/or modify
|
|
|
|
# it under the terms of the GNU General Public License as published by
|
|
|
|
# the Free Software Foundation; either version 2 of the License, or
|
|
|
|
# (at your option) any later version.
|
|
|
|
#
|
|
|
|
# This program is distributed in the hope that it will be useful,
|
|
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
|
|
# GNU General Public License for more details.
|
|
|
|
#
|
|
|
|
# You should have received a copy of the GNU General Public License
|
|
|
|
# along with this program; if not, write to the Free Software
|
|
|
|
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
|
|
#
|
|
|
|
|
2008-01-23 11:11:46 +05:30
|
|
|
# $Id$
|
2007-12-20 11:56:38 +05:30
|
|
|
|
|
|
|
"""
|
|
|
|
Create a relationship graph using Graphviz
|
|
|
|
"""
|
|
|
|
|
2008-01-18 11:09:50 +05:30
|
|
|
#------------------------------------------------------------------------
|
|
|
|
#
|
|
|
|
# python modules
|
|
|
|
#
|
|
|
|
#------------------------------------------------------------------------
|
|
|
|
from TransUtils import sgettext as _
|
|
|
|
|
2007-12-20 11:56:38 +05:30
|
|
|
#------------------------------------------------------------------------
|
|
|
|
#
|
|
|
|
# GRAMPS modules
|
|
|
|
#
|
|
|
|
#------------------------------------------------------------------------
|
2008-10-02 09:32:10 +05:30
|
|
|
from gen.plug import PluginManager
|
2008-09-27 19:26:17 +05:30
|
|
|
from gen.plug.menu import BooleanOption, EnumeratedListOption, FilterOption, \
|
2009-01-26 03:04:48 +05:30
|
|
|
PersonOption, ColorOption
|
2008-10-02 09:32:10 +05:30
|
|
|
from ReportBase import Report, ReportUtils, MenuReportOptions, CATEGORY_GRAPHVIZ
|
2007-12-20 11:56:38 +05:30
|
|
|
from BasicUtils import name_displayer
|
|
|
|
import DateHandler
|
|
|
|
import gen.lib
|
|
|
|
import Utils
|
|
|
|
import ThumbNails
|
|
|
|
|
|
|
|
#------------------------------------------------------------------------
|
|
|
|
#
|
|
|
|
# Constant options items
|
|
|
|
#
|
|
|
|
#------------------------------------------------------------------------
|
2008-09-09 12:33:26 +05:30
|
|
|
_COLORS = [ { 'name' : _("B&W outline"), 'value' : 'outlined' },
|
|
|
|
{ 'name' : _("Colored outline"), 'value' : 'colored' },
|
|
|
|
{ 'name' : _("Color fill"), 'value' : 'filled' }]
|
2008-09-08 23:30:38 +05:30
|
|
|
|
2008-09-09 12:33:26 +05:30
|
|
|
_ARROWS = [ { 'name' : _("Descendants <- Ancestors"), 'value' : 'd' },
|
|
|
|
{ 'name' : _("Descendants -> Ancestors"), 'value' : 'a' },
|
|
|
|
{ 'name' : _("Descendants <-> Ancestors"), 'value' : 'da' },
|
|
|
|
{ 'name' : _("Descendants - Ancestors"), 'value' : '' }]
|
2007-12-20 11:56:38 +05:30
|
|
|
|
|
|
|
#------------------------------------------------------------------------
|
|
|
|
#
|
|
|
|
# Report class
|
|
|
|
#
|
|
|
|
#------------------------------------------------------------------------
|
|
|
|
class RelGraphReport(Report):
|
|
|
|
|
2008-02-20 10:22:10 +05:30
|
|
|
def __init__(self, database, options_class):
|
2007-12-20 11:56:38 +05:30
|
|
|
"""
|
2008-02-24 19:25:55 +05:30
|
|
|
Create ComprehensiveAncestorsReport object that produces the report.
|
2007-12-20 11:56:38 +05:30
|
|
|
|
|
|
|
The arguments are:
|
|
|
|
|
|
|
|
database - the GRAMPS database instance
|
|
|
|
person - currently selected person
|
|
|
|
options_class - instance of the Options class for this report
|
|
|
|
|
|
|
|
This report needs the following parameters (class variables)
|
|
|
|
that come in the options class.
|
|
|
|
|
|
|
|
filter - Filter to be applied to the people of the database.
|
|
|
|
The option class carries its number, and the function
|
|
|
|
returning the list of filters.
|
|
|
|
arrow - Arrow styles for heads and tails.
|
|
|
|
showfamily - Whether to show family nodes.
|
|
|
|
incid - Whether to include IDs.
|
|
|
|
incdate - Whether to include dates.
|
|
|
|
justyears - Use years only.
|
2008-02-01 18:34:35 +05:30
|
|
|
use_place - Whether to replace missing dates with place
|
2007-12-20 11:56:38 +05:30
|
|
|
url - Whether to include URLs.
|
|
|
|
inclimg - Include images or not
|
|
|
|
imgpos - Image position, above/beside name
|
2009-01-25 20:34:22 +05:30
|
|
|
color - Whether to use outline, colored outline or filled color
|
|
|
|
in graph
|
2009-01-26 03:04:48 +05:30
|
|
|
color_males - Colour to apply to males
|
|
|
|
color_females - Colour to apply to females
|
|
|
|
color_unknown - Colour to apply to unknown genders
|
|
|
|
color_families - Colour to apply to families
|
|
|
|
dashed - Whether to use dashed lines for non-birth relationships
|
2008-05-31 14:12:38 +05:30
|
|
|
use_roundedcorners - Whether to use rounded corners for females
|
2007-12-20 11:56:38 +05:30
|
|
|
"""
|
2008-02-20 10:22:10 +05:30
|
|
|
Report.__init__(self, database, options_class)
|
2007-12-20 11:56:38 +05:30
|
|
|
|
|
|
|
self.database = database
|
|
|
|
|
2008-01-23 11:11:46 +05:30
|
|
|
menu = options_class.menu
|
|
|
|
self.includeid = menu.get_option_by_name('incid').get_value()
|
|
|
|
self.includedates = menu.get_option_by_name('incdate').get_value()
|
|
|
|
self.includeurl = menu.get_option_by_name('url').get_value()
|
|
|
|
self.includeimg = menu.get_option_by_name('includeImages').get_value()
|
|
|
|
self.imgpos = menu.get_option_by_name('imageOnTheSide').get_value()
|
2009-01-25 20:34:22 +05:30
|
|
|
self.use_roundedcorners = \
|
|
|
|
menu.get_option_by_name('useroundedcorners').get_value()
|
2008-01-23 11:11:46 +05:30
|
|
|
self.adoptionsdashed = menu.get_option_by_name('dashed').get_value()
|
|
|
|
self.show_families = menu.get_option_by_name('showfamily').get_value()
|
|
|
|
self.just_years = menu.get_option_by_name('justyears').get_value()
|
2008-02-01 18:34:35 +05:30
|
|
|
self.use_place = menu.get_option_by_name('use_place').get_value()
|
2009-05-30 17:42:58 +05:30
|
|
|
self.use_subgraphs = menu.get_option_by_name('usesubgraphs').get_value()
|
2009-01-26 03:04:48 +05:30
|
|
|
|
2008-09-09 12:33:26 +05:30
|
|
|
self.colorize = menu.get_option_by_name('color').get_value()
|
2009-01-26 03:04:48 +05:30
|
|
|
color_males = menu.get_option_by_name('colormales').get_value()
|
|
|
|
color_females = menu.get_option_by_name('colorfemales').get_value()
|
|
|
|
color_unknown = menu.get_option_by_name('colorunknown').get_value()
|
|
|
|
color_families = menu.get_option_by_name('colorfamilies').get_value()
|
|
|
|
self.colors = {
|
|
|
|
'male': color_males,
|
|
|
|
'female': color_females,
|
|
|
|
'unknown': color_unknown,
|
|
|
|
'family': color_families
|
|
|
|
}
|
|
|
|
|
2008-09-09 12:33:26 +05:30
|
|
|
arrow_str = menu.get_option_by_name('arrow').get_value()
|
|
|
|
if arrow_str.find('a') + 1:
|
2007-12-20 11:56:38 +05:30
|
|
|
self.arrowheadstyle = 'normal'
|
2008-09-09 12:33:26 +05:30
|
|
|
else:
|
|
|
|
self.arrowheadstyle = 'none'
|
|
|
|
if arrow_str.find('d') + 1:
|
2007-12-20 11:56:38 +05:30
|
|
|
self.arrowtailstyle = 'normal'
|
2008-09-09 12:33:26 +05:30
|
|
|
else:
|
|
|
|
self.arrowtailstyle = 'none'
|
2007-12-20 11:56:38 +05:30
|
|
|
filter_option = options_class.menu.get_option_by_name('filter')
|
2008-01-23 11:11:46 +05:30
|
|
|
self._filter = filter_option.get_filter()
|
2007-12-20 11:56:38 +05:30
|
|
|
|
|
|
|
def write_report(self):
|
2008-01-23 11:11:46 +05:30
|
|
|
self.person_handles = self._filter.apply(self.database,
|
2007-12-20 11:56:38 +05:30
|
|
|
self.database.get_person_handles(sort_handles=False))
|
|
|
|
|
|
|
|
if len(self.person_handles) > 1:
|
|
|
|
self.add_persons_and_families()
|
|
|
|
self.add_child_links_to_families()
|
|
|
|
|
|
|
|
def add_child_links_to_families(self):
|
2009-01-25 20:34:22 +05:30
|
|
|
"returns string of GraphViz edges linking parents to families or \
|
|
|
|
children"
|
2007-12-20 11:56:38 +05:30
|
|
|
person_dict = {}
|
|
|
|
# Hash people in a dictionary for faster inclusion checking
|
|
|
|
for person_handle in self.person_handles:
|
|
|
|
person_dict[person_handle] = 1
|
|
|
|
for person_handle in self.person_handles:
|
|
|
|
person = self.database.get_person_from_handle(person_handle)
|
|
|
|
p_id = person.get_gramps_id()
|
|
|
|
for fam_handle in person.get_parent_family_handle_list():
|
|
|
|
family = self.database.get_family_from_handle(fam_handle)
|
|
|
|
father_handle = family.get_father_handle()
|
|
|
|
mother_handle = family.get_mother_handle()
|
|
|
|
for child_ref in family.get_child_ref_list():
|
|
|
|
if child_ref.ref == person_handle:
|
|
|
|
frel = child_ref.frel
|
|
|
|
mrel = child_ref.mrel
|
|
|
|
break
|
|
|
|
if (self.show_families and
|
|
|
|
((father_handle and father_handle in person_dict) or
|
|
|
|
(mother_handle and mother_handle in person_dict))):
|
|
|
|
# Link to the family node if either parent is in graph
|
2009-01-25 20:34:22 +05:30
|
|
|
self.add_family_link(p_id, family, frel, mrel)
|
2007-12-20 11:56:38 +05:30
|
|
|
else:
|
|
|
|
# Link to the parents' nodes directly, if they are in graph
|
|
|
|
if father_handle and father_handle in person_dict:
|
2009-01-25 20:34:22 +05:30
|
|
|
self.add_parent_link(p_id, father_handle, frel)
|
2007-12-20 11:56:38 +05:30
|
|
|
if mother_handle and mother_handle in person_dict:
|
2009-01-25 20:34:22 +05:30
|
|
|
self.add_parent_link(p_id, mother_handle, mrel)
|
2007-12-20 11:56:38 +05:30
|
|
|
|
|
|
|
def add_family_link(self, p_id, family, frel, mrel):
|
|
|
|
"Links the child to a family"
|
|
|
|
style = 'solid'
|
|
|
|
adopted = ((int(frel) != gen.lib.ChildRefType.BIRTH) or
|
|
|
|
(int(mrel) != gen.lib.ChildRefType.BIRTH))
|
2009-02-03 01:13:21 +05:30
|
|
|
# If birth relation to father is NONE, mening there is no father and
|
|
|
|
# if birth relation to mother is BIRTH then solid line
|
|
|
|
if ((int(frel) == gen.lib.ChildRefType.NONE) and
|
|
|
|
(int(mrel) == gen.lib.ChildRefType.BIRTH)):
|
|
|
|
adopted = False
|
2007-12-20 11:56:38 +05:30
|
|
|
if adopted and self.adoptionsdashed:
|
|
|
|
style = 'dotted'
|
|
|
|
self.doc.add_link( p_id, family.get_gramps_id(), style,
|
|
|
|
self.arrowheadstyle, self.arrowtailstyle )
|
|
|
|
|
|
|
|
def add_parent_link(self, p_id, parent_handle, rel):
|
|
|
|
"Links the child to a parent"
|
|
|
|
style = 'solid'
|
|
|
|
if (int(rel) != gen.lib.ChildRefType.BIRTH) and self.adoptionsdashed:
|
|
|
|
style = 'dotted'
|
|
|
|
parent = self.database.get_person_from_handle(parent_handle)
|
|
|
|
self.doc.add_link( p_id, parent.get_gramps_id(), style,
|
|
|
|
self.arrowheadstyle, self.arrowtailstyle )
|
|
|
|
|
|
|
|
def add_persons_and_families(self):
|
|
|
|
"adds nodes for persons and their families"
|
|
|
|
# variable to communicate with get_person_label
|
|
|
|
self.bUseHtmlOutput = False
|
|
|
|
|
|
|
|
# The list of families for which we have output the node,
|
|
|
|
# so we don't do it twice
|
|
|
|
families_done = {}
|
|
|
|
for person_handle in self.person_handles:
|
|
|
|
# determine per person if we use HTML style label
|
|
|
|
if self.includeimg:
|
|
|
|
self.bUseHtmlOutput = True
|
|
|
|
person = self.database.get_person_from_handle(person_handle)
|
|
|
|
p_id = person.get_gramps_id()
|
|
|
|
# Output the person's node
|
|
|
|
label = self.get_person_label(person)
|
2009-01-25 20:34:22 +05:30
|
|
|
(shape, style, color, fill) = self.get_gender_style(person)
|
2007-12-20 11:56:38 +05:30
|
|
|
url = ""
|
|
|
|
if self.includeurl:
|
|
|
|
h = person_handle
|
|
|
|
dirpath = "ppl/%s/%s" % (h[0], h[1])
|
|
|
|
dirpath = dirpath.lower()
|
2008-02-24 19:25:55 +05:30
|
|
|
url = "%s/%s.html" % (dirpath, h)
|
2007-12-20 11:56:38 +05:30
|
|
|
|
2009-01-25 20:34:22 +05:30
|
|
|
self.doc.add_node(p_id, label, shape, color, style, fill, url)
|
2007-12-20 11:56:38 +05:30
|
|
|
|
|
|
|
# Output families where person is a parent
|
|
|
|
if self.show_families:
|
|
|
|
family_list = person.get_family_handle_list()
|
|
|
|
for fam_handle in family_list:
|
2009-05-30 17:42:58 +05:30
|
|
|
family = self.database.get_family_from_handle(fam_handle)
|
|
|
|
fam_id = family.get_gramps_id()
|
2007-12-20 11:56:38 +05:30
|
|
|
if fam_handle not in families_done:
|
|
|
|
families_done[fam_handle] = 1
|
2009-05-30 17:42:58 +05:30
|
|
|
self.__add_family(fam_handle)
|
|
|
|
# If subgraphs are not chosen then each parent is linked
|
|
|
|
# separately to the family. This gives Graphviz greater
|
|
|
|
# control over the layout of the whole graph but
|
|
|
|
# may leave spouses not positioned together.
|
|
|
|
if not self.use_subgraphs:
|
|
|
|
self.doc.add_link(fam_id, p_id, "",
|
|
|
|
self.arrowheadstyle,
|
|
|
|
self.arrowtailstyle)
|
2009-04-27 02:17:42 +05:30
|
|
|
|
2009-05-30 17:42:58 +05:30
|
|
|
def __add_family(self, fam_handle):
|
|
|
|
"""Add a node for a family and optionally link the spouses to it"""
|
|
|
|
fam = self.database.get_family_from_handle(fam_handle)
|
|
|
|
fam_id = fam.get_gramps_id()
|
|
|
|
|
|
|
|
label = ""
|
|
|
|
for event_ref in fam.get_event_ref_list():
|
|
|
|
event = self.database.get_event_from_handle(event_ref.ref)
|
|
|
|
if event.type == gen.lib.EventType.MARRIAGE:
|
|
|
|
label = self.get_event_string(event)
|
|
|
|
break
|
|
|
|
if self.includeid:
|
|
|
|
label = "%s (%s)" % (label, fam_id)
|
|
|
|
color = ""
|
|
|
|
fill = ""
|
|
|
|
style = "solid"
|
|
|
|
if self.colorize == 'colored':
|
|
|
|
color = self.colors['family']
|
|
|
|
elif self.colorize == 'filled':
|
|
|
|
fill = self.colors['family']
|
|
|
|
style = "filled"
|
|
|
|
self.doc.add_node(fam_id, label, "ellipse", color, style, fill)
|
|
|
|
|
|
|
|
# If subgraphs are used then we add both spouses here and Graphviz
|
|
|
|
# will attempt to position both spouses closely together.
|
|
|
|
# TODO: A person who is a parent in more than one family may only be
|
|
|
|
# positioned next to one of their spouses. The code currently
|
|
|
|
# does not take into account multiple spouses.
|
|
|
|
if self.use_subgraphs:
|
|
|
|
self.doc.start_subgraph(fam_id)
|
|
|
|
f_handle = fam.get_father_handle()
|
|
|
|
m_handle = fam.get_mother_handle()
|
|
|
|
if f_handle:
|
|
|
|
father = self.database.get_person_from_handle(f_handle)
|
|
|
|
self.doc.add_link(fam_id,
|
|
|
|
father.get_gramps_id(), "",
|
|
|
|
self.arrowheadstyle,
|
|
|
|
self.arrowtailstyle)
|
|
|
|
if m_handle:
|
|
|
|
mother = self.database.get_person_from_handle(m_handle)
|
|
|
|
self.doc.add_link(fam_id,
|
|
|
|
mother.get_gramps_id(), "",
|
|
|
|
self.arrowheadstyle,
|
|
|
|
self.arrowtailstyle)
|
|
|
|
self.doc.end_subgraph()
|
2007-12-20 11:56:38 +05:30
|
|
|
|
|
|
|
def get_gender_style(self, person):
|
|
|
|
"return gender specific person style"
|
|
|
|
gender = person.get_gender()
|
|
|
|
shape = "box"
|
2008-05-31 14:12:38 +05:30
|
|
|
style = "solid"
|
2007-12-20 11:56:38 +05:30
|
|
|
color = ""
|
|
|
|
fill = ""
|
2008-05-31 14:12:38 +05:30
|
|
|
|
|
|
|
if gender == person.FEMALE and self.use_roundedcorners:
|
2009-01-25 20:34:22 +05:30
|
|
|
style = "rounded"
|
2008-05-31 14:12:38 +05:30
|
|
|
elif gender == person.UNKNOWN:
|
2009-01-25 20:34:22 +05:30
|
|
|
shape = "hexagon"
|
2008-05-31 14:12:38 +05:30
|
|
|
|
2007-12-20 11:56:38 +05:30
|
|
|
if self.colorize == 'colored':
|
|
|
|
if gender == person.MALE:
|
|
|
|
color = self.colors['male']
|
|
|
|
elif gender == person.FEMALE:
|
|
|
|
color = self.colors['female']
|
|
|
|
else:
|
|
|
|
color = self.colors['unknown']
|
|
|
|
elif self.colorize == 'filled':
|
2008-05-31 14:12:38 +05:30
|
|
|
style += ",filled"
|
2007-12-20 11:56:38 +05:30
|
|
|
if gender == person.MALE:
|
|
|
|
fill = self.colors['male']
|
|
|
|
elif gender == person.FEMALE:
|
|
|
|
fill = self.colors['female']
|
|
|
|
else:
|
|
|
|
fill = self.colors['unknown']
|
2009-01-25 20:34:22 +05:30
|
|
|
return(shape, style, color, fill)
|
2007-12-20 11:56:38 +05:30
|
|
|
|
|
|
|
def get_person_label(self, person):
|
|
|
|
"return person label string"
|
|
|
|
# see if we have an image to use for this person
|
|
|
|
imagePath = None
|
|
|
|
if self.bUseHtmlOutput:
|
|
|
|
mediaList = person.get_media_list()
|
|
|
|
if len(mediaList) > 0:
|
|
|
|
mediaHandle = mediaList[0].get_reference_handle()
|
|
|
|
media = self.database.get_object_from_handle(mediaHandle)
|
|
|
|
mediaMimeType = media.get_mime_type()
|
|
|
|
if mediaMimeType[0:5] == "image":
|
2008-02-18 19:36:41 +05:30
|
|
|
imagePath = ThumbNails.get_thumbnail_path(
|
|
|
|
Utils.media_path_full(self.database,
|
|
|
|
media.get_path()))
|
2009-01-25 20:34:22 +05:30
|
|
|
# test if thumbnail actually exists in thumbs
|
|
|
|
# (import of data means media files might not be present
|
2007-12-20 11:56:38 +05:30
|
|
|
imagePath = Utils.find_file(imagePath)
|
|
|
|
|
|
|
|
label = u""
|
|
|
|
lineDelimiter = '\\n'
|
|
|
|
|
2009-01-25 20:34:22 +05:30
|
|
|
# If we have an image, then start an HTML table; remember to close
|
|
|
|
# the table afterwards!
|
2007-12-28 15:38:19 +05:30
|
|
|
#
|
2009-01-25 20:34:22 +05:30
|
|
|
# This isn't a free-form HTML format here...just a few keywords that
|
|
|
|
# happen to be
|
|
|
|
# simillar to keywords commonly seen in HTML. For additional
|
|
|
|
# information on what
|
2007-12-28 15:38:19 +05:30
|
|
|
# is allowed, see:
|
|
|
|
#
|
|
|
|
# http://www.graphviz.org/info/shapes.html#html
|
|
|
|
#
|
2007-12-20 11:56:38 +05:30
|
|
|
if self.bUseHtmlOutput and imagePath:
|
|
|
|
lineDelimiter = '<BR/>'
|
|
|
|
label += '<TABLE BORDER="0" CELLSPACING="2" CELLPADDING="0" CELLBORDER="0"><TR><TD></TD><TD><IMG SRC="%s"/></TD><TD></TD>' % imagePath
|
|
|
|
if self.imgpos == 0:
|
|
|
|
#trick it into not stretching the image
|
|
|
|
label += '</TR><TR><TD COLSPAN="3">'
|
|
|
|
else :
|
|
|
|
label += '<TD>'
|
|
|
|
else :
|
|
|
|
#no need for html label with this person
|
|
|
|
self.bUseHtmlOutput = False
|
|
|
|
|
|
|
|
# at the very least, the label must have the person's name
|
|
|
|
nm = name_displayer.display_name(person.get_primary_name())
|
|
|
|
if self.bUseHtmlOutput :
|
|
|
|
# avoid < and > in the name, as this is html text
|
2009-01-25 20:34:22 +05:30
|
|
|
label += nm.replace('<', '<').replace('>', '>')
|
2007-12-20 11:56:38 +05:30
|
|
|
else :
|
|
|
|
label += nm
|
|
|
|
p_id = person.get_gramps_id()
|
|
|
|
if self.includeid:
|
|
|
|
label += " (%s)" % p_id
|
|
|
|
if self.includedates:
|
|
|
|
birth, death = self.get_date_strings(person)
|
2009-01-25 20:34:22 +05:30
|
|
|
label = label + '%s(%s - %s)' % (lineDelimiter, birth, death)
|
2007-12-20 11:56:38 +05:30
|
|
|
|
|
|
|
# see if we have a table that needs to be terminated
|
|
|
|
if self.bUseHtmlOutput:
|
|
|
|
label += '</TD></TR></TABLE>'
|
|
|
|
return label
|
|
|
|
else :
|
|
|
|
# non html label is enclosed by "" so excape other "
|
|
|
|
return label.replace('"', '\\\"')
|
|
|
|
|
|
|
|
def get_date_strings(self, person):
|
|
|
|
"returns tuple of birth/christening and death/burying date strings"
|
2009-01-25 20:34:22 +05:30
|
|
|
birth_event = ReportUtils.get_birth_or_fallback(self.database, person)
|
|
|
|
if birth_event:
|
2007-12-20 11:56:38 +05:30
|
|
|
birth = self.get_event_string(birth_event)
|
|
|
|
else:
|
2009-01-25 20:34:22 +05:30
|
|
|
birth = ""
|
|
|
|
|
|
|
|
death_event = ReportUtils.get_death_or_fallback(self.database, person)
|
|
|
|
if death_event:
|
2007-12-20 11:56:38 +05:30
|
|
|
death = self.get_event_string(death_event)
|
|
|
|
else:
|
2009-01-25 20:34:22 +05:30
|
|
|
death = ""
|
|
|
|
|
2007-12-20 11:56:38 +05:30
|
|
|
return (birth, death)
|
|
|
|
|
|
|
|
def get_event_string(self, event):
|
|
|
|
"""
|
|
|
|
return string for for an event label.
|
|
|
|
|
|
|
|
Based on the data availability and preferences, we select one
|
|
|
|
of the following for a given event:
|
|
|
|
year only
|
|
|
|
complete date
|
|
|
|
place name
|
|
|
|
empty string
|
|
|
|
"""
|
|
|
|
if event:
|
|
|
|
if event.get_date_object().get_year_valid():
|
|
|
|
if self.just_years:
|
|
|
|
return '%i' % event.get_date_object().get_year()
|
|
|
|
else:
|
|
|
|
return DateHandler.get_date(event)
|
2008-02-01 18:34:35 +05:30
|
|
|
elif self.use_place:
|
2007-12-20 11:56:38 +05:30
|
|
|
place_handle = event.get_place_handle()
|
|
|
|
place = self.database.get_place_from_handle(place_handle)
|
|
|
|
if place and place.get_title():
|
|
|
|
return place.get_title()
|
|
|
|
return ''
|
|
|
|
|
|
|
|
|
|
|
|
#------------------------------------------------------------------------
|
|
|
|
#
|
|
|
|
# Options class
|
|
|
|
#
|
|
|
|
#------------------------------------------------------------------------
|
|
|
|
class RelGraphOptions(MenuReportOptions):
|
|
|
|
"""
|
|
|
|
Defines options and provides handling interface.
|
|
|
|
"""
|
2008-01-24 18:20:33 +05:30
|
|
|
def __init__(self, name, dbase):
|
2008-01-18 11:09:50 +05:30
|
|
|
self.__pid = None
|
|
|
|
self.__filter = None
|
|
|
|
self.__include_images = None
|
|
|
|
self.__image_on_side = None
|
2008-01-24 18:20:33 +05:30
|
|
|
self.__db = dbase
|
|
|
|
MenuReportOptions.__init__(self, name, dbase)
|
2007-12-20 11:56:38 +05:30
|
|
|
|
2008-01-24 18:20:33 +05:30
|
|
|
def add_menu_options(self, menu):
|
2007-12-20 11:56:38 +05:30
|
|
|
################################
|
|
|
|
category_name = _("Report Options")
|
|
|
|
################################
|
2008-01-18 11:09:50 +05:30
|
|
|
|
2008-01-23 11:11:46 +05:30
|
|
|
self.__filter = FilterOption(_("Filter"), 0)
|
|
|
|
self.__filter.set_help(
|
|
|
|
_("Determines what people are included in the graph"))
|
|
|
|
menu.add_option(category_name, "filter", self.__filter)
|
|
|
|
self.__filter.connect('value-changed', self.__filter_changed)
|
|
|
|
|
2008-01-18 11:09:50 +05:30
|
|
|
self.__pid = PersonOption(_("Filter Person"))
|
|
|
|
self.__pid.set_help(_("The center person for the filter"))
|
|
|
|
menu.add_option(category_name, "pid", self.__pid)
|
|
|
|
self.__pid.connect('value-changed', self.__update_filters)
|
2007-12-20 11:56:38 +05:30
|
|
|
|
2008-01-18 11:09:50 +05:30
|
|
|
self.__update_filters()
|
2007-12-20 11:56:38 +05:30
|
|
|
|
2009-01-26 02:09:02 +05:30
|
|
|
self.incdate = BooleanOption(
|
2007-12-20 11:56:38 +05:30
|
|
|
_("Include Birth, Marriage and Death dates"), True)
|
2009-01-26 02:09:02 +05:30
|
|
|
self.incdate.set_help(_("Include the dates that the individual was "
|
|
|
|
"born, got married and/or died in the graph labels."))
|
|
|
|
menu.add_option(category_name, "incdate", self.incdate)
|
|
|
|
self.incdate.connect('value-changed', self.__include_dates_changed)
|
2007-12-20 11:56:38 +05:30
|
|
|
|
2009-01-26 02:09:02 +05:30
|
|
|
self.justyears = BooleanOption(_("Limit dates to years only"), False)
|
|
|
|
self.justyears.set_help(_("Prints just dates' year, neither "
|
|
|
|
"month or day nor date approximation "
|
|
|
|
"or interval are shown."))
|
|
|
|
menu.add_option(category_name, "justyears", self.justyears)
|
2007-12-20 11:56:38 +05:30
|
|
|
|
2008-02-01 18:34:35 +05:30
|
|
|
use_place = BooleanOption(_("Use place when no date"), True)
|
|
|
|
use_place.set_help(_("When no birth, marriage, or death date is "
|
|
|
|
"available, the correspondent place field "
|
|
|
|
"will be used."))
|
|
|
|
menu.add_option(category_name, "use_place", use_place)
|
2007-12-20 11:56:38 +05:30
|
|
|
|
|
|
|
url = BooleanOption(_("Include URLs"), False)
|
|
|
|
url.set_help(_("Include a URL in each graph node so "
|
|
|
|
"that PDF and imagemap files can be "
|
|
|
|
"generated that contain active links "
|
|
|
|
"to the files generated by the 'Narrated "
|
|
|
|
"Web Site' report."))
|
2008-01-18 11:09:50 +05:30
|
|
|
menu.add_option(category_name, "url", url)
|
2007-12-20 11:56:38 +05:30
|
|
|
|
|
|
|
incid = BooleanOption(_("Include IDs"), False)
|
|
|
|
incid.set_help(_("Include individual and family IDs."))
|
2008-01-18 11:09:50 +05:30
|
|
|
menu.add_option(category_name, "incid", incid)
|
2007-12-20 11:56:38 +05:30
|
|
|
|
2008-01-18 11:09:50 +05:30
|
|
|
self.__include_images = BooleanOption(
|
2007-12-20 11:56:38 +05:30
|
|
|
_('Include thumbnail images of people'), False)
|
2008-01-18 11:09:50 +05:30
|
|
|
self.__include_images.set_help(
|
|
|
|
_("Whether to include thumbnails of people."))
|
|
|
|
menu.add_option(category_name, "includeImages", self.__include_images)
|
|
|
|
self.__include_images.connect('value-changed', self.__image_changed)
|
2007-12-20 11:56:38 +05:30
|
|
|
|
2008-01-18 11:09:50 +05:30
|
|
|
self.__image_on_side = EnumeratedListOption(_("Thumbnail Location"), 0)
|
|
|
|
self.__image_on_side.add_item(0, _('Above the name'))
|
|
|
|
self.__image_on_side.add_item(1, _('Beside the name'))
|
|
|
|
self.__image_on_side.set_help(
|
|
|
|
_("Where the thumbnail image should appear "
|
|
|
|
"relative to the name"))
|
|
|
|
menu.add_option(category_name, "imageOnTheSide", self.__image_on_side)
|
2007-12-20 11:56:38 +05:30
|
|
|
|
|
|
|
################################
|
|
|
|
category_name = _("Graph Style")
|
|
|
|
################################
|
|
|
|
|
2008-09-09 12:33:26 +05:30
|
|
|
color = EnumeratedListOption(_("Graph coloring"), 'filled')
|
2007-12-20 11:56:38 +05:30
|
|
|
for i in range( 0, len(_COLORS) ):
|
|
|
|
color.add_item(_COLORS[i]["value"], _COLORS[i]["name"])
|
|
|
|
color.set_help(_("Males will be shown with blue, females "
|
|
|
|
"with red. If the sex of an individual "
|
|
|
|
"is unknown it will be shown with gray."))
|
2008-01-18 11:09:50 +05:30
|
|
|
menu.add_option(category_name, "color", color)
|
2009-01-26 03:04:48 +05:30
|
|
|
|
|
|
|
color_males = ColorOption(_('Males'), '#e0e0ff')
|
|
|
|
color_males.set_help(_('The colour to use to display men.'))
|
|
|
|
menu.add_option(category_name, 'colormales', color_males)
|
|
|
|
|
|
|
|
color_females = ColorOption(_('Females'), '#ffe0e0')
|
|
|
|
color_females.set_help(_('The colour to use to display women.'))
|
|
|
|
menu.add_option(category_name, 'colorfemales', color_females)
|
|
|
|
|
|
|
|
color_unknown = ColorOption(_('Unknown'), '#e0e0e0')
|
|
|
|
color_unknown.set_help(_('The colour to use when the gender is ' \
|
|
|
|
'unknown.'))
|
|
|
|
menu.add_option(category_name, 'colorunknown', color_unknown)
|
|
|
|
|
|
|
|
color_family = ColorOption(_('Families'), '#ffffe0')
|
|
|
|
color_family.set_help(_('The colour to use to display families.'))
|
|
|
|
menu.add_option(category_name, 'colorfamilies', color_family)
|
2007-12-20 11:56:38 +05:30
|
|
|
|
2008-09-09 12:33:26 +05:30
|
|
|
arrow = EnumeratedListOption(_("Arrowhead direction"), 'd')
|
2007-12-20 11:56:38 +05:30
|
|
|
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."))
|
2008-01-18 11:09:50 +05:30
|
|
|
menu.add_option(category_name, "arrow", arrow)
|
2008-05-31 14:12:38 +05:30
|
|
|
|
|
|
|
roundedcorners = BooleanOption( # see bug report #2180
|
|
|
|
_("Use rounded corners"), False)
|
|
|
|
roundedcorners.set_help(
|
|
|
|
_("Use rounded corners to differentiate "
|
|
|
|
"between women and men."))
|
|
|
|
menu.add_option(category_name, "useroundedcorners", roundedcorners)
|
2007-12-20 11:56:38 +05:30
|
|
|
|
|
|
|
dashed = BooleanOption(
|
|
|
|
_("Indicate non-birth relationships with dotted lines"), True)
|
|
|
|
dashed.set_help(_("Non-birth relationships will show up "
|
|
|
|
"as dotted lines in the graph."))
|
2008-01-18 11:09:50 +05:30
|
|
|
menu.add_option(category_name, "dashed", dashed)
|
2007-12-20 11:56:38 +05:30
|
|
|
|
|
|
|
showfamily = BooleanOption(_("Show family nodes"), True)
|
|
|
|
showfamily.set_help(_("Families will show up as ellipses, linked "
|
|
|
|
"to parents and children."))
|
2008-01-18 11:09:50 +05:30
|
|
|
menu.add_option(category_name, "showfamily", showfamily)
|
|
|
|
|
|
|
|
def __update_filters(self):
|
|
|
|
"""
|
|
|
|
Update the filter list based on the selected person
|
|
|
|
"""
|
|
|
|
gid = self.__pid.get_value()
|
2008-01-24 18:20:33 +05:30
|
|
|
person = self.__db.get_person_from_gramps_id(gid)
|
2008-01-18 11:09:50 +05:30
|
|
|
filter_list = ReportUtils.get_person_filters(person, False)
|
|
|
|
self.__filter.set_filters(filter_list)
|
|
|
|
|
2009-01-26 02:09:02 +05:30
|
|
|
def __include_dates_changed(self):
|
|
|
|
"""
|
|
|
|
Enable/disable menu items if dates are required
|
|
|
|
"""
|
|
|
|
if self.incdate.get_value():
|
|
|
|
self.justyears.set_available(True)
|
|
|
|
else:
|
|
|
|
self.justyears.set_available(False)
|
|
|
|
|
2008-01-18 11:09:50 +05:30
|
|
|
def __filter_changed(self):
|
|
|
|
"""
|
|
|
|
Handle filter change. If the filter is not specific to a person,
|
|
|
|
disable the person option
|
|
|
|
"""
|
|
|
|
filter_value = self.__filter.get_value()
|
|
|
|
if filter_value in [1, 2, 3, 4]:
|
|
|
|
# Filters 1, 2, 3 and 4 rely on the center person
|
|
|
|
self.__pid.set_available(True)
|
|
|
|
else:
|
|
|
|
# The rest don't
|
|
|
|
self.__pid.set_available(False)
|
2008-01-12 17:07:21 +05:30
|
|
|
|
2008-01-18 11:09:50 +05:30
|
|
|
def __image_changed(self):
|
|
|
|
"""
|
|
|
|
Handle thumbnail change. If the image is not to be included, make the
|
|
|
|
image location option unavailable.
|
|
|
|
"""
|
|
|
|
self.__image_on_side.set_available(self.__include_images.get_value())
|
2008-01-12 17:07:21 +05:30
|
|
|
|
2007-12-20 11:56:38 +05:30
|
|
|
#------------------------------------------------------------------------
|
|
|
|
#
|
|
|
|
#
|
|
|
|
#
|
|
|
|
#------------------------------------------------------------------------
|
2008-05-19 00:54:28 +05:30
|
|
|
pmgr = PluginManager.get_instance()
|
|
|
|
pmgr.register_report(
|
2008-10-02 09:32:10 +05:30
|
|
|
name = 'rel_graph',
|
|
|
|
category = CATEGORY_GRAPHVIZ,
|
|
|
|
report_class = RelGraphReport,
|
|
|
|
options_class = RelGraphOptions,
|
|
|
|
modes = PluginManager.REPORT_MODE_GUI | \
|
|
|
|
PluginManager.REPORT_MODE_CLI,
|
2008-03-02 04:17:48 +05:30
|
|
|
translated_name = _("Relationship Graph"),
|
2008-10-02 09:32:10 +05:30
|
|
|
status = _("Stable"),
|
|
|
|
description = _("Produces relationship graphs using Graphviz"),
|
|
|
|
author_name = "Brian G. Matherly",
|
|
|
|
author_email = "brian@gramps-project.org"
|
2007-12-28 15:38:19 +05:30
|
|
|
)
|
|
|
|
|
2008-01-06 01:40:26 +05:30
|
|
|
|
2008-06-02 06:29:14 +05:30
|
|
|
|