Files
gramps/src/plugins/RelCalc.py
Don Allingham faca534de4 Renamed the categories for reports
svn: r676
2002-01-02 14:27:03 +00:00

367 lines
13 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 *
from libglade import *
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.getMainFamily()
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 _("%s is the first cousin of %s.") %(f,s)
elif removed == 1:
return _("%s is the first cousin once removed of %s.") %(f,s)
elif removed == 2:
return _("%s is the first cousin twice removed of %s.") %(f,s)
else:
return _("%s is the first cousin %d times removed of %s.") % (f,s,removed)
elif level == 2:
if removed == 0:
return _("%s is the second cousin of %s.") %(f,s)
elif removed == 1:
return _("%s is the second cousin once removed of %s.") %(f,s)
elif removed == 2:
return _("%s is the second cousin twice removed of %s.") %(f,s)
else:
return _("%s is the second cousin %d times removed of %s.") % (f,s,removed)
elif level == 3:
if removed == 0:
return _("%s is the third cousin of %s.") %(f,s)
elif removed == 1:
return _("%s is the third cousin once removed of %s.") %(f,s)
elif removed == 2:
return _("%s is the third cousin twice removed of %s.") %(f,s)
else:
return _("%s is the third cousin %d times removed of %s.") % (f,s,removed)
else:
if removed == 0:
return _("%s is the %dth cousin of %s.") %(f,level,s)
elif removed == 1:
return _("%s is the %dth cousin once removed of %s.") %(f,level,s)
elif removed == 2:
return _("%s is the %dth cousin twice removed of %s.") %(f,level,s)
else:
return _("%s is the %dth cousin %d times removed of %s.") % (f,level,s,removed)
def get_father(f,s,level):
if level == 1:
return _("%s is the father of %s.") % (s,f)
elif level == 2:
return _("%s is the grandfather of %s.") % (s,f)
elif level == 3:
return _("%s is the great grandfather of %s.") % (s,f)
elif level == 4:
return _("%s is the second great grandfather of %s.") % (s,f)
elif level == 5:
return _("%s is the third great grandfather of %s.") % (s,f)
else:
return _("%s is the %dth great grandfather of %s.") % (s,level-2,f)
def get_son(f,s,level):
if level == 1:
return _("%s is the son of %s.") % (s,f)
elif level == 2:
return _("%s is the grandson of %s.") % (s,f)
elif level == 3:
return _("%s is the great grandson of %s.") % (s,f)
elif level == 4:
return _("%s is the second great grandson of %s.") % (s,f)
elif level == 5:
return _("%s is the third great grandson of %s.") % (s,f)
else:
return _("%s is the %dth great grandson of %s.") % (s,level-2,f)
def get_mother(f,s,level):
if level == 1:
return _("%s is the mother of %s.") % (s,f)
elif level == 2:
return _("%s is the grandmother of %s.") % (s,f)
elif level == 3:
return _("%s is the great grandmother of %s.") % (s,f)
elif level == 4:
return _("%s is the second great grandmother of %s.") % (s,f)
elif level == 5:
return _("%s is the third great grandmother of %s.") % (s,f)
else:
return _("%s is the %dth great grandmother of %s.") % (s,level-2,f)
def get_daughter(f,s,level):
if level == 1:
return _("%s is the daughter of %s.") % (s,f)
elif level == 2:
return _("%s is the granddaughter of %s.") % (s,f)
elif level == 3:
return _("%s is the great granddaughter of %s.") % (s,f)
elif level == 4:
return _("%s is the second great granddaughter of %s.") % (s,f)
elif level == 5:
return _("%s is the third great granddaughter of %s.") % (s,f)
else:
return _("%s is the %dth great granddaughter of %s.") % (s,level-2,f)
def get_aunt(f,s,level):
if level == 1:
return _("%s is the sister of %s.") % (s,f)
elif level == 2:
return _("%s is the aunt of %s.") % (s,f)
elif level == 3:
return _("%s is the grandaunt of %s.") % (s,f)
elif level == 4:
return _("%s is the great grandaunt of %s.") % (s,f)
elif level == 5:
return _("%s is the second great grandaunt of %s.") % (s,f)
elif level == 6:
return _("%s is the third great grandaunt of %s.") % (s,f)
else:
return _("%s is the %dth great grandaunt of %s.") % (s,level-3,f)
def get_uncle(f,s,level):
if level == 1:
return _("%s is the brother of %s.") % (s,f)
elif level == 2:
return _("%s is the uncle of %s.") % (s,f)
elif level == 3:
return _("%s is the granduncle of %s.") % (s,f)
elif level == 4:
return _("%s is the great granduncle of %s.") % (s,f)
elif level == 5:
return _("%s is the second great granduncle of %s.") % (s,f)
elif level == 6:
return _("%s is the third great granduncle of %s.") % (s,f)
else:
return _("%s is the %dth great granduncle of %s.") % (s,level-3,f)
def get_nephew(f,s,level):
if level == 1:
return _("%s is the nephew of %s.") % (s,f)
elif level == 2:
return _("%s is the grandnephew of %s.") % (s,f)
elif level == 3:
return _("%s is the great grandnephew of %s.") % (s,f)
elif level == 4:
return _("%s is the second great grandnephew of %s.") % (s,f)
elif level == 5:
return _("%s is the third great grandnephew of %s.") % (s,f)
else:
return _("%s is the %dth great grandnephew of %s.") % (s,level-2,f)
def get_niece(f,s,level):
if level == 1:
return _("%s is the niece of %s.") % (s,f)
elif level == 2:
return _("%s is the grandniece of %s.") % (s,f)
elif level == 3:
return _("%s is the great grandniece of %s.") % (s,f)
elif level == 4:
return _("%s is the second great grandniece of %s.") % (s,f)
elif level == 5:
return _("%s is the third great grandniece of %s.") % (s,f)
else:
return _("%s is the %dth great grandniece of %s.") % (s,level-2,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 = GladeXML(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 > 1:
index = 0
commontext = " Their common ancestors are "
for person in common:
secondRel = firstMap[person]
firstRel = secondMap[person]
if length == index + 1:
commontext = commontext + " and "
elif index != 0:
commontext = commontext + ", "
commontext = commontext + person.getPrimaryName().getRegularName()
index = index + 1
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")
)