440 lines
16 KiB
Python
440 lines
16 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
|
||
|
|
#
|
||
|
|
|
||
|
|
"Utilities/Relationship calculator"
|
||
|
|
|
||
|
|
import os
|
||
|
|
|
||
|
|
from gtk import *
|
||
|
|
from gnome.ui import *
|
||
|
|
import gtk.glade
|
||
|
|
|
||
|
|
import RelLib
|
||
|
|
import sort
|
||
|
|
import intl
|
||
|
|
import Utils
|
||
|
|
|
||
|
|
_ = intl.gettext
|
||
|
|
|
||
|
|
#-------------------------------------------------------------------------
|
||
|
|
#
|
||
|
|
#
|
||
|
|
#
|
||
|
|
#-------------------------------------------------------------------------
|
||
|
|
def filter(person,index,list,map):
|
||
|
|
if person == None:
|
||
|
|
return
|
||
|
|
list.append(person)
|
||
|
|
map[person] = index
|
||
|
|
|
||
|
|
family = person.getMainParents()
|
||
|
|
if family != None:
|
||
|
|
filter(family.getFather(),index+1,list,map)
|
||
|
|
filter(family.getMother(),index+1,list,map)
|
||
|
|
|
||
|
|
#-------------------------------------------------------------------------
|
||
|
|
#
|
||
|
|
#
|
||
|
|
#
|
||
|
|
#-------------------------------------------------------------------------
|
||
|
|
def get_cousin(f,s,level,removed):
|
||
|
|
if level == 1:
|
||
|
|
if removed == 0:
|
||
|
|
return _("%(p1)s is the first cousin of %(p2)s.") % {
|
||
|
|
'p1' : f, 'p2' : s}
|
||
|
|
elif removed == 1:
|
||
|
|
return _("%(p1)s is the first cousin once removed of %(p2)s.") % {
|
||
|
|
'p1' : f, 'p2' : s}
|
||
|
|
elif removed == 2:
|
||
|
|
return _("%(p1)s is the first cousin twice removed of %(p2)s.") % {
|
||
|
|
'p1' : f,'p2' : s}
|
||
|
|
else:
|
||
|
|
return _("%(p1)s is the first cousin %(removed)d times removed of %(p2)s.") % {
|
||
|
|
'p1' :f, 'p2' :s, 'removed' : removed}
|
||
|
|
elif level == 2:
|
||
|
|
if removed == 0:
|
||
|
|
return _("%(p1)s is the second cousin of %(p2)s.") % {
|
||
|
|
'p1' : f, 'p2' : s }
|
||
|
|
elif removed == 1:
|
||
|
|
return _("%(p1)s is the second cousin once removed of %(p2)s.") % {
|
||
|
|
'p1' : f, 'p2' : s }
|
||
|
|
elif removed == 2:
|
||
|
|
return _("%(p1)s is the second cousin twice removed of %(p2)s.") % {
|
||
|
|
'p1' : f, 'p2' : s }
|
||
|
|
else:
|
||
|
|
return _("%(p1)s is the second cousin %(removed)d times removed of %(p2)s.") % {
|
||
|
|
'p1' : f, 'p2' : s, 'removed' : removed }
|
||
|
|
elif level == 3:
|
||
|
|
if removed == 0:
|
||
|
|
return _("%(p1)s is the third cousin of %(p2)s.") % {
|
||
|
|
'p1' : f, 'p2' : s }
|
||
|
|
elif removed == 1:
|
||
|
|
return _("%(p1)s is the third cousin once removed of %(p2)s.") % {
|
||
|
|
'p1' : f, 'p2' : s }
|
||
|
|
elif removed == 2:
|
||
|
|
return _("%(p1)s is the third cousin twice removed of %(p2)s.") % {
|
||
|
|
'p1' : f, 'p2' : s }
|
||
|
|
else:
|
||
|
|
return _("%(p1)s is the third cousin %(removed)d times removed of %(p2)s.") % {
|
||
|
|
'p1' : f, 'p2' : s, 'removed' : removed }
|
||
|
|
else:
|
||
|
|
if removed == 0:
|
||
|
|
return _("%(p1)s is the %(level)dth cousin of %(p2)s.") % {
|
||
|
|
'p1': f, 'level': level, 'p2' : s }
|
||
|
|
elif removed == 1:
|
||
|
|
return _("%(p1)s is the %(level)dth cousin once removed of %(p2)s.") % {
|
||
|
|
'p1': f, 'level': level, 'p2' : s }
|
||
|
|
elif removed == 2:
|
||
|
|
return _("%(p1)s is the %(level)dth cousin twice removed of %(p2)s.") % {
|
||
|
|
'p1': f, 'level': level, 'p2' : s }
|
||
|
|
else:
|
||
|
|
return _("%(p1)s is the %(level)dth cousin %(removed)d times removed of %(p2)s.") % {
|
||
|
|
'p1': f, 'level': level, 'removed' : removed, 'p2' : s }
|
||
|
|
|
||
|
|
def get_father(f,s,level):
|
||
|
|
if level == 1:
|
||
|
|
return _("%(p1)s is the father of %(p2)s.") % {
|
||
|
|
'p1': s, 'p2': f }
|
||
|
|
elif level == 2:
|
||
|
|
return _("%(p1)s is the grandfather of %(p2)s.") % {
|
||
|
|
'p1': s, 'p2': f }
|
||
|
|
elif level == 3:
|
||
|
|
return _("%(p1)s is the great grandfather of %(p2)s.") % {
|
||
|
|
'p1': s, 'p2': f }
|
||
|
|
elif level == 4:
|
||
|
|
return _("%(p1)s is the second great grandfather of %(p2)s.") % {
|
||
|
|
'p1': s, 'p2': f }
|
||
|
|
elif level == 5:
|
||
|
|
return _("%(p1)s is the third great grandfather of %(p2)s.") % {
|
||
|
|
'p1': s, 'p2': f }
|
||
|
|
else:
|
||
|
|
return _("%(p1)s is the %(level)dth great grandfather of %(p2)s.") % {
|
||
|
|
'p1': s, 'level' : level-2, 'p2': f }
|
||
|
|
|
||
|
|
def get_son(f,s,level):
|
||
|
|
if level == 1:
|
||
|
|
return _("%(p1)s is the son of %(p2)s.") % {
|
||
|
|
'p1': s, 'p2': f }
|
||
|
|
elif level == 2:
|
||
|
|
return _("%(p1)s is the grandson of %(p2)s.") % {
|
||
|
|
'p1': s, 'p2': f }
|
||
|
|
elif level == 3:
|
||
|
|
return _("%(p1)s is the great grandson of %(p2)s.") % {
|
||
|
|
'p1': s, 'p2': f }
|
||
|
|
elif level == 4:
|
||
|
|
return _("%(p1)s is the second great grandson of %(p2)s.") % {
|
||
|
|
'p1': s, 'p2': f }
|
||
|
|
elif level == 5:
|
||
|
|
return _("%(p1)s is the third great grandson of %(p2)s.") % {
|
||
|
|
'p1': s, 'p2': f }
|
||
|
|
else:
|
||
|
|
return _("%(p1)s is the %(level)dth great grandson of %(p2)s.") % {
|
||
|
|
'p1': s, 'level' : level-2, 'p2': f }
|
||
|
|
|
||
|
|
def get_mother(f,s,level):
|
||
|
|
if level == 1:
|
||
|
|
return _("%(p1)s is the mother of %(p2)s.") % {
|
||
|
|
'p1': s, 'p2': f }
|
||
|
|
elif level == 2:
|
||
|
|
return _("%(p1)s is the grandmother of %(p2)s.") % {
|
||
|
|
'p1': s, 'p2': f }
|
||
|
|
elif level == 3:
|
||
|
|
return _("%(p1)s is the great grandmother of %(p2)s.") % {
|
||
|
|
'p1': s, 'p2': f }
|
||
|
|
elif level == 4:
|
||
|
|
return _("%(p1)s is the second great grandmother of %(p2)s.") % {
|
||
|
|
'p1': s, 'p2': f }
|
||
|
|
elif level == 5:
|
||
|
|
return _("%(p1)s is the third great grandmother of %(p2)s.") % {
|
||
|
|
'p1': s, 'p2': f }
|
||
|
|
else:
|
||
|
|
return _("%(p1)s is the %(level)dth great grandmother of %(p2)s.") % {
|
||
|
|
'p1': s, 'level' : level-2, 'p2': f }
|
||
|
|
|
||
|
|
def get_daughter(f,s,level):
|
||
|
|
if level == 1:
|
||
|
|
return _("%(p1)s is the daughter of %(p2)s.") % {
|
||
|
|
'p1': s, 'p2': f }
|
||
|
|
elif level == 2:
|
||
|
|
return _("%(p1)s is the granddaughter of %(p2)s.") % {
|
||
|
|
'p1': s, 'p2': f }
|
||
|
|
elif level == 3:
|
||
|
|
return _("%(p1)s is the great granddaughter of %(p2)s.") % {
|
||
|
|
'p1': s, 'p2': f }
|
||
|
|
elif level == 4:
|
||
|
|
return _("%(p1)s is the second great granddaughter of %(p2)s.") % {
|
||
|
|
'p1': s, 'p2': f }
|
||
|
|
elif level == 5:
|
||
|
|
return _("%(p1)s is the third great granddaughter of %(p2)s.") % {
|
||
|
|
'p1': s, 'p2': f }
|
||
|
|
else:
|
||
|
|
return _("%(p1)s is the %(level)dth great granddaughter of %(p2)s.") % {
|
||
|
|
'p1': s, 'level' : level-2, 'p2': f }
|
||
|
|
|
||
|
|
def get_aunt(f,s,level):
|
||
|
|
if level == 1:
|
||
|
|
return _("%(p1)s is the sister of %(p2)s.") % {
|
||
|
|
'p1': s, 'p2': f }
|
||
|
|
elif level == 2:
|
||
|
|
return _("%(p1)s is the aunt of %(p2)s.") % {
|
||
|
|
'p1': s, 'p2': f }
|
||
|
|
elif level == 3:
|
||
|
|
return _("%(p1)s is the grandaunt of %(p2)s.") % {
|
||
|
|
'p1': s, 'p2': f }
|
||
|
|
elif level == 4:
|
||
|
|
return _("%(p1)s is the great grandaunt of %(p2)s.") % {
|
||
|
|
'p1': s, 'p2': f }
|
||
|
|
elif level == 5:
|
||
|
|
return _("%(p1)s is the second great grandaunt of %(p2)s.") % {
|
||
|
|
'p1': s, 'p2': f }
|
||
|
|
elif level == 6:
|
||
|
|
return _("%(p1)s is the third great grandaunt of %(p2)s.") % {
|
||
|
|
'p1': s, 'p2': f }
|
||
|
|
else:
|
||
|
|
return _("%(p1)s is the %(level)dth great grandaunt of %(p2)s.") % {
|
||
|
|
'p1': s, 'level' : level-3, 'p2': f }
|
||
|
|
|
||
|
|
def get_uncle(f,s,level):
|
||
|
|
if level == 1:
|
||
|
|
return _("%(p1)s is the brother of %(p2)s.") % {
|
||
|
|
'p1': s, 'p2': f }
|
||
|
|
elif level == 2:
|
||
|
|
return _("%(p1)s is the uncle of %(p2)s.") % {
|
||
|
|
'p1': s, 'p2': f }
|
||
|
|
elif level == 3:
|
||
|
|
return _("%(p1)s is the granduncle of %(p2)s.") % {
|
||
|
|
'p1': s, 'p2': f }
|
||
|
|
elif level == 4:
|
||
|
|
return _("%(p1)s is the great granduncle of %(p2)s.") % {
|
||
|
|
'p1': s, 'p2': f }
|
||
|
|
elif level == 5:
|
||
|
|
return _("%(p1)s is the second great granduncle of %(p2)s.") % {
|
||
|
|
'p1': s, 'p2': f }
|
||
|
|
elif level == 6:
|
||
|
|
return _("%(p1)s is the third great granduncle of %(p2)s.") % {
|
||
|
|
'p1': s, 'p2': f }
|
||
|
|
else:
|
||
|
|
return _("%(p1)s is the %(level)dth great granduncle of %(p2)s.") % {
|
||
|
|
'p1': s, 'level' : level-3, 'p2': f }
|
||
|
|
|
||
|
|
def get_nephew(f,s,level):
|
||
|
|
if level == 1:
|
||
|
|
return _("%(p1)s is the nephew of %(p2)s.") % {
|
||
|
|
'p1': s, 'p2': f }
|
||
|
|
elif level == 2:
|
||
|
|
return _("%(p1)s is the grandnephew of %(p2)s.") % {
|
||
|
|
'p1': s, 'p2': f }
|
||
|
|
elif level == 3:
|
||
|
|
return _("%(p1)s is the great grandnephew of %(p2)s.") % {
|
||
|
|
'p1': s, 'p2': f }
|
||
|
|
elif level == 4:
|
||
|
|
return _("%(p1)s is the second great grandnephew of %(p2)s.") % {
|
||
|
|
'p1': s, 'p2': f }
|
||
|
|
elif level == 5:
|
||
|
|
return _("%(p1)s is the third great grandnephew of %(p2)s.") % {
|
||
|
|
'p1': s, 'p2': f }
|
||
|
|
else:
|
||
|
|
return _("%(p1)s is the %(level)dth great grandnephew of %(p2)s.") % {
|
||
|
|
'p1': s, 'level' : level-2, 'p2': f }
|
||
|
|
|
||
|
|
def get_niece(f,s,level):
|
||
|
|
if level == 1:
|
||
|
|
return _("%(p1)s is the niece of %(p2)s.") % {
|
||
|
|
'p1': s, 'p2': f }
|
||
|
|
elif level == 2:
|
||
|
|
return _("%(p1)s is the grandniece of %(p2)s.") % {
|
||
|
|
'p1': s, 'p2': f }
|
||
|
|
elif level == 3:
|
||
|
|
return _("%(p1)s is the great grandniece of %(p2)s.") % {
|
||
|
|
'p1': s, 'p2': f }
|
||
|
|
elif level == 4:
|
||
|
|
return _("%(p1)s is the second great grandniece of %(p2)s.") % {
|
||
|
|
'p1': s, 'p2': f }
|
||
|
|
elif level == 5:
|
||
|
|
return _("%(p1)s is the third great grandniece of %(p2)s.") % {
|
||
|
|
'p1': s, 'p2': f }
|
||
|
|
else:
|
||
|
|
return _("%(p1)s is the %(level)dth great grandniece of %(p2)s.") % {
|
||
|
|
'p1': s, 'level' : level-2, 'p2': f }
|
||
|
|
|
||
|
|
#-------------------------------------------------------------------------
|
||
|
|
#
|
||
|
|
#
|
||
|
|
#
|
||
|
|
#-------------------------------------------------------------------------
|
||
|
|
def runTool(database,person,callback):
|
||
|
|
RelCalc(database,person)
|
||
|
|
|
||
|
|
#-------------------------------------------------------------------------
|
||
|
|
#
|
||
|
|
#
|
||
|
|
#
|
||
|
|
#-------------------------------------------------------------------------
|
||
|
|
class RelCalc:
|
||
|
|
"""
|
||
|
|
Relationship calculator class.
|
||
|
|
"""
|
||
|
|
|
||
|
|
def __init__(self,database,person):
|
||
|
|
self.person = person
|
||
|
|
self.db = database
|
||
|
|
|
||
|
|
base = os.path.dirname(__file__)
|
||
|
|
glade_file = base + os.sep + "relcalc.glade"
|
||
|
|
self.glade = gtk.glade.XML(glade_file,"relcalc")
|
||
|
|
|
||
|
|
name = self.person.getPrimaryName().getRegularName()
|
||
|
|
|
||
|
|
self.glade.get_widget("name").set_text(_("Relationship to %s") % name)
|
||
|
|
self.people = self.glade.get_widget("peopleList")
|
||
|
|
|
||
|
|
name_list = self.db.getPersonMap().values()
|
||
|
|
name_list.sort(sort.by_last_name)
|
||
|
|
index = 0
|
||
|
|
self.people.freeze()
|
||
|
|
for p in name_list:
|
||
|
|
name = p.getPrimaryName().getName()
|
||
|
|
birthday = p.getBirth().getDate()
|
||
|
|
id = p.getId()
|
||
|
|
self.people.append([name,id,birthday])
|
||
|
|
self.people.set_row_data(index,p)
|
||
|
|
index = index + 1
|
||
|
|
self.people.thaw()
|
||
|
|
|
||
|
|
self.glade.signal_autoconnect({
|
||
|
|
"on_close_clicked" : Utils.destroy_passed_object,
|
||
|
|
"on_apply_clicked" : self.on_apply_clicked
|
||
|
|
})
|
||
|
|
|
||
|
|
def on_apply_clicked(self,obj):
|
||
|
|
firstMap = {}
|
||
|
|
firstList = []
|
||
|
|
secondMap = {}
|
||
|
|
secondList = []
|
||
|
|
common = []
|
||
|
|
rank = 9999999
|
||
|
|
|
||
|
|
if len(self.people.selection) == 0:
|
||
|
|
return
|
||
|
|
|
||
|
|
other_person = self.people.get_row_data(self.people.selection[0])
|
||
|
|
filter(self.person,0,firstList,firstMap)
|
||
|
|
filter(other_person,0,secondList,secondMap)
|
||
|
|
|
||
|
|
for person in firstList:
|
||
|
|
if person in secondList:
|
||
|
|
new_rank = firstMap[person]
|
||
|
|
if new_rank < rank:
|
||
|
|
rank = new_rank
|
||
|
|
common = [ person ]
|
||
|
|
elif new_rank == rank:
|
||
|
|
common.append(person)
|
||
|
|
|
||
|
|
firstRel = -1
|
||
|
|
secondRel = -1
|
||
|
|
|
||
|
|
firstName = self.person.getPrimaryName().getRegularName()
|
||
|
|
secondName = other_person.getPrimaryName().getRegularName()
|
||
|
|
|
||
|
|
length = len(common)
|
||
|
|
|
||
|
|
if length == 1:
|
||
|
|
person = common[0]
|
||
|
|
secondRel = firstMap[person]
|
||
|
|
firstRel = secondMap[person]
|
||
|
|
name = person.getPrimaryName().getRegularName()
|
||
|
|
commontext = " " + _("Their common ancestor is %s.") % name
|
||
|
|
elif length == 2:
|
||
|
|
p1 = common[0]
|
||
|
|
p2 = common[1]
|
||
|
|
secondRel = firstMap[p1]
|
||
|
|
firstRel = secondMap[p1]
|
||
|
|
commontext = " " + _("Their common ancestors are %s and %s.") % \
|
||
|
|
(p1.getPrimaryName().getRegularName(),\
|
||
|
|
p2.getPrimaryName().getRegularName())
|
||
|
|
elif length > 2:
|
||
|
|
index = 0
|
||
|
|
commontext = " " + _("Their common ancestors are : ")
|
||
|
|
for person in common:
|
||
|
|
secondRel = firstMap[person]
|
||
|
|
firstRel = secondMap[person]
|
||
|
|
if index != 0:
|
||
|
|
commontext = commontext + ", "
|
||
|
|
commontext = commontext + person.getPrimaryName().getRegularName()
|
||
|
|
index = index + 1
|
||
|
|
commontext = commontext + "."
|
||
|
|
else:
|
||
|
|
commontext = ""
|
||
|
|
|
||
|
|
if firstRel == -1:
|
||
|
|
msg = _("There is no relationship between %s and %s.")
|
||
|
|
text = msg % (firstName,secondName)
|
||
|
|
elif firstRel == 0:
|
||
|
|
if secondRel == 0:
|
||
|
|
text = _("%s and %s are the same person.") % (firstName,secondName)
|
||
|
|
elif other_person.getGender() == RelLib.Person.male:
|
||
|
|
text = get_father(firstName,secondName,secondRel)
|
||
|
|
else:
|
||
|
|
text = get_mother(firstName,secondName,secondRel)
|
||
|
|
elif secondRel == 0:
|
||
|
|
if other_person.getGender() == RelLib.Person.male:
|
||
|
|
text = get_son(firstName,secondName,firstRel)
|
||
|
|
else:
|
||
|
|
text = get_daughter(firstName,secondName,firstRel)
|
||
|
|
elif firstRel == 1:
|
||
|
|
if other_person.getGender() == RelLib.Person.male:
|
||
|
|
text = get_uncle(firstName,secondName,secondRel)
|
||
|
|
else:
|
||
|
|
text = get_aunt(firstName,secondName,secondRel)
|
||
|
|
elif secondRel == 1:
|
||
|
|
if other_person.getGender() == RelLib.Person.male:
|
||
|
|
text = get_nephew(firstName,secondName,firstRel-1)
|
||
|
|
else:
|
||
|
|
text = get_niece(firstName,secondName,firstRel-1)
|
||
|
|
else:
|
||
|
|
if secondRel > firstRel:
|
||
|
|
text = get_cousin(firstName,secondName,firstRel-1,secondRel-firstRel)
|
||
|
|
else:
|
||
|
|
text = get_cousin(firstName,secondName,secondRel-1,firstRel-secondRel)
|
||
|
|
|
||
|
|
text1 = self.glade.get_widget("text1")
|
||
|
|
text1.set_point(0)
|
||
|
|
length = text1.get_length()
|
||
|
|
text1.forward_delete(length)
|
||
|
|
if firstRel == 0 or secondRel == 0:
|
||
|
|
text1.insert_defaults(text)
|
||
|
|
else:
|
||
|
|
text1.insert_defaults(text + commontext)
|
||
|
|
text1.set_word_wrap(1)
|
||
|
|
|
||
|
|
#-------------------------------------------------------------------------
|
||
|
|
#
|
||
|
|
#
|
||
|
|
#
|
||
|
|
#-------------------------------------------------------------------------
|
||
|
|
from Plugins import register_tool
|
||
|
|
|
||
|
|
register_tool(
|
||
|
|
runTool,
|
||
|
|
_("Relationship calculator"),
|
||
|
|
category=_("Utilities"),
|
||
|
|
description=_("Calculates the relationship between two people")
|
||
|
|
)
|