252 lines
7.2 KiB
Python
252 lines
7.2 KiB
Python
#
|
|
# Gramps - a GTK+/GNOME based genealogy program
|
|
#
|
|
# Copyright (C) 2000 Donald N. Allingham
|
|
#
|
|
# 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
|
|
#
|
|
|
|
"Generate files/Relationship graph"
|
|
|
|
from RelLib import *
|
|
import os
|
|
import re
|
|
import sort
|
|
import string
|
|
import utils
|
|
import intl
|
|
|
|
_ = intl.gettext
|
|
|
|
from gtk import *
|
|
from gnome.ui import *
|
|
from libglade import *
|
|
|
|
active_person = None
|
|
db = None
|
|
topDialog = None
|
|
glade_file = None
|
|
ind_list = []
|
|
|
|
#------------------------------------------------------------------------
|
|
#
|
|
#
|
|
#
|
|
#------------------------------------------------------------------------
|
|
def ancestor_filter(database,person,list):
|
|
|
|
if person == None:
|
|
return
|
|
if person not in list:
|
|
list.append(person)
|
|
family = person.getMainFamily()
|
|
if family != None:
|
|
ancestor_filter(database,family.getFather(),list)
|
|
ancestor_filter(database,family.getMother(),list)
|
|
|
|
#------------------------------------------------------------------------
|
|
#
|
|
#
|
|
#
|
|
#------------------------------------------------------------------------
|
|
def descendant_filter(database,person,list):
|
|
|
|
if person == None:
|
|
return
|
|
if person not in list:
|
|
list.append(person)
|
|
|
|
for family in person.getFamilyList():
|
|
for child in family.getChildList():
|
|
descendant_filter(database,child,list)
|
|
|
|
#------------------------------------------------------------------------
|
|
#
|
|
#
|
|
#
|
|
#------------------------------------------------------------------------
|
|
def an_des_filter(database,person,list):
|
|
|
|
descendant_filter(database,person,list)
|
|
ancestor_filter(database,person,list)
|
|
|
|
#------------------------------------------------------------------------
|
|
#
|
|
#
|
|
#
|
|
#------------------------------------------------------------------------
|
|
def entire_db_filter(database,person,list):
|
|
|
|
for entry in database.getPersonMap().values():
|
|
list.append(entry)
|
|
|
|
#------------------------------------------------------------------------
|
|
#
|
|
#
|
|
#
|
|
#------------------------------------------------------------------------
|
|
|
|
filter_map = {
|
|
"Ancestors" : ancestor_filter,
|
|
"Descendants" : descendant_filter,
|
|
"Ancestors and Descendants" : an_des_filter,
|
|
"Entire Database" : entire_db_filter
|
|
}
|
|
|
|
#------------------------------------------------------------------------
|
|
#
|
|
#
|
|
#
|
|
#------------------------------------------------------------------------
|
|
def report(database,person):
|
|
global active_person
|
|
global topDialog
|
|
global glade_file
|
|
global db
|
|
|
|
active_person = person
|
|
db = database
|
|
|
|
base = os.path.dirname(__file__)
|
|
glade_file = base + os.sep + "graphviz.glade"
|
|
dic = {
|
|
"destroy_passed_object" : utils.destroy_passed_object,
|
|
"on_ok_clicked" : on_ok_clicked,
|
|
}
|
|
|
|
topDialog = GladeXML(glade_file,"top")
|
|
topDialog.signal_autoconnect(dic)
|
|
|
|
top = topDialog.get_widget("top")
|
|
topDialog.get_widget("targetDirectory").set_default_path(os.getcwd())
|
|
filterName = topDialog.get_widget("filterName")
|
|
popdown_strings = filter_map.keys()
|
|
popdown_strings.sort()
|
|
filterName.set_popdown_strings(popdown_strings)
|
|
|
|
name = person.getPrimaryName().getName()
|
|
topDialog.get_widget("personName").set_text(name)
|
|
|
|
top.show()
|
|
|
|
#------------------------------------------------------------------------
|
|
#
|
|
#
|
|
#
|
|
#------------------------------------------------------------------------
|
|
def on_ok_clicked(obj):
|
|
global active_person
|
|
global filter_map
|
|
global db
|
|
global glade_file
|
|
global ind_list
|
|
|
|
filterName = topDialog.get_widget("filter").get_text()
|
|
file_name = topDialog.get_widget("filename").get_text()
|
|
if file_name == "":
|
|
return
|
|
|
|
paper = topDialog.get_widget("paper")
|
|
multi = topDialog.get_widget("multi").get_active()
|
|
scaled = topDialog.get_widget("scaled").get_active()
|
|
portrait = topDialog.get_widget("portrait").get_active()
|
|
|
|
width = paper.get_width()/72.0
|
|
height = paper.get_height()/72.0
|
|
|
|
file = open(file_name,"w")
|
|
|
|
filter = filter_map[filterName]
|
|
ind_list = []
|
|
|
|
filter(db,active_person,ind_list)
|
|
|
|
file.write("digraph g {\n")
|
|
file.write("bgcolor=white;\n")
|
|
file.write("rankdir=LR;\n")
|
|
file.write("center=1;\n")
|
|
|
|
if scaled == 1:
|
|
file.write("size=\"%3.1fin,%3.1fin\";\n" % (width-0.5,height-0.5))
|
|
file.write("ratio=compress;\n")
|
|
else:
|
|
file.write("ratio=auto;\n")
|
|
|
|
if multi == 1:
|
|
file.write("page=\"%3.1f,%3.1f\";\n" % (width,height))
|
|
|
|
if portrait == 1:
|
|
file.write("orientation=portrait;\n")
|
|
else:
|
|
file.write("orientation=landscape;\n")
|
|
|
|
if len(ind_list) > 1:
|
|
dump_index(ind_list,file)
|
|
dump_person(ind_list,file)
|
|
|
|
file.write("}\n")
|
|
file.close()
|
|
utils.destroy_passed_object(obj)
|
|
|
|
#########################################################################
|
|
#
|
|
# Write a web page for a specific person represented by the key
|
|
#
|
|
#########################################################################
|
|
|
|
def dump_person(person_list,file):
|
|
for person in person_list:
|
|
family = person.getMainFamily()
|
|
if family == None:
|
|
continue
|
|
father = family.getFather()
|
|
if father and father in person_list:
|
|
file.write('p' + str(person.getId()) + ' -> p')
|
|
file.write(str(father.getId()) + ';\n')
|
|
mother = family.getMother()
|
|
if mother and mother in person_list:
|
|
file.write('p' + str(person.getId()) + ' -> p')
|
|
file.write(str(mother.getId()) + ';\n')
|
|
|
|
#########################################################################
|
|
#
|
|
# Write a web page for a specific person represented by the key
|
|
#
|
|
#########################################################################
|
|
|
|
def dump_index(person_list,file):
|
|
|
|
for person in person_list:
|
|
name = person.getPrimaryName().getName()
|
|
id = person.getId()
|
|
# file.write('p' + str(id) + ' [shape=box, fontsize=11, ')
|
|
file.write('p' + str(id) + ' [shape=box, ')
|
|
file.write('fontname="Arial", label="' + name + '"];\n')
|
|
|
|
#------------------------------------------------------------------------
|
|
#
|
|
#
|
|
#
|
|
#------------------------------------------------------------------------
|
|
def get_description():
|
|
return _("Generates relationship graphs, currently only in GraphViz format.") + \
|
|
" " + \
|
|
_("GraphViz (dot) can transform the graph into postscript, jpeg, png, vrml, svg, and many other formats.") + \
|
|
" " + \
|
|
_("For more information or to get a copy of GraphViz, goto http://www.graphviz.org")
|
|
|
|
def get_name():
|
|
return _("Generate files/Relationship graph")
|