From 8eadc54ac293bdc639241718f97d23d5983edeff Mon Sep 17 00:00:00 2001 From: Don Allingham Date: Sat, 8 Feb 2003 23:45:21 +0000 Subject: [PATCH] Timeline report svn: r1291 --- gramps2/src/plugins/TimeLine.py | 453 ++++++++++++++++++++++++++++++++ 1 file changed, 453 insertions(+) create mode 100644 gramps2/src/plugins/TimeLine.py diff --git a/gramps2/src/plugins/TimeLine.py b/gramps2/src/plugins/TimeLine.py new file mode 100644 index 000000000..f6dc3d95c --- /dev/null +++ b/gramps2/src/plugins/TimeLine.py @@ -0,0 +1,453 @@ +# +# Gramps - a GTK+/GNOME based genealogy program +# +# Copyright (C) 2003 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 +# + +""" +TimeLine report +""" + +#------------------------------------------------------------------------ +# +# python modules +# +#------------------------------------------------------------------------ +import os + +#------------------------------------------------------------------------ +# +# GNOME/gtk +# +#------------------------------------------------------------------------ +import gtk + +#------------------------------------------------------------------------ +# +# GRAMPS modules +# +#------------------------------------------------------------------------ +import Utils +import Report +import TextDoc +import DrawDoc +import GenericFilter +import Errors +import Date +import FontScale +import sort +from QuestionDialog import ErrorDialog + +from intl import gettext as _ + +#------------------------------------------------------------------------ +# +# TimeLine +# +#------------------------------------------------------------------------ +class TimeLine: + + def __init__(self,database,person,output,document,filter,title,sort_func): + """ + Creates the TimeLine object that produces the report. This class + is used by the TimeLineDialog class. The arguments are: + + database - the GRAMPS database + person - currently selected person + output - name of the output file + document - DrawDoc instance for the output file. Any class derived + from DrawDoc may be used. + filter - filtering function selected by the TimeLineDialog + class. + """ + self.d = document + self.filter = filter + self.db = database + self.person = person + self.output = output + self.title = title + self.sort_func = sort_func + + def setup(self): + """ + Define the graphics styles used by the report. Paragraph definitions + have already been defined in the document. The styles used are: + + grid - 0.5pt wide line dashed line. Used for the lines that make up + the grid. + line - 0.5pt wide line. Used for the line connecting two endpoints + and for the birth marker. + solid - 0.5pt line with a black fill color. Used for the date of + death marker. + text - Contains the Name paragraph style used for the individual's + name + title - Contains the Title paragraph style used for the title of + the document + label - Contains the Label paragraph style used for the year label's + in the document. + """ + g = DrawDoc.GraphicsStyle() + g.set_line_width(0.5) + g.set_color((0,0,0)) + self.d.add_draw_style("line",g) + + g = DrawDoc.GraphicsStyle() + g.set_line_width(0.5) + g.set_color((0,0,0)) + g.set_fill_color((0,0,0)) + self.d.add_draw_style("solid",g) + + g = DrawDoc.GraphicsStyle() + g.set_line_width(0.5) + g.set_color((0,0,0)) + g.set_fill_color((255,255,255)) + self.d.add_draw_style("open",g) + + g = DrawDoc.GraphicsStyle() + g.set_line_width(0.5) + g.set_line_style(DrawDoc.DASHED) + g.set_color((0,0,0)) + self.d.add_draw_style("grid",g) + + g = DrawDoc.GraphicsStyle() + g.set_paragraph_style("Name") + g.set_color((255,255,255)) + g.set_fill_color((255,255,255)) + g.set_line_width(0) + self.d.add_draw_style("text",g) + + g = DrawDoc.GraphicsStyle() + g.set_paragraph_style("Title") + g.set_color((255,255,255)) + g.set_fill_color((255,255,255)) + g.set_line_width(0) + self.d.add_draw_style("title",g) + + g = DrawDoc.GraphicsStyle() + g.set_paragraph_style("Label") + g.set_color((255,255,255)) + g.set_fill_color((255,255,255)) + g.set_line_width(0) + self.d.add_draw_style("label",g) + + def write_report(self): + + (low,high) = self.find_year_range() + st_size = self.name_size() + + font = self.d.style_list['Name'].get_font() + + incr = pt2cm(font.get_size()) + pad = incr*.75 + + x1,x2,y1,y2 = (0,0,0,0) + + start = st_size+0.5 + stop = self.d.get_usable_width()-0.5 + size = (stop-start) + self.header = 2.0 + + self.d.open(self.output) + self.d.start_page() + self.build_grid(low,high,start,stop) + + index = 1 + current = 1; + + length = len(self.plist) + + self.plist.sort(self.sort_func) + + for p in self.plist: + b = p.getBirth().getDateObj().getYear() + d = p.getDeath().getDateObj().getYear() + + n = p.getPrimaryName().getName() + self.d.draw_text('text',n,incr+pad,self.header + (incr+pad)*index) + + y1 = self.header + (pad+incr)*index + y2 = self.header + ((pad+incr)*index)+incr + y3 = (y1+y2)/2.0 + w = 0.05 + + if b != Date.UNDEF: + start_offset = ((float(b-low)/float(high-low)) * (size)) + x1 = start+start_offset + path = [(x1,y1),(x1+w,y3),(x1,y2),(x1-w,y3)] + self.d.draw_path('line',path) + + if d != Date.UNDEF: + start_offset = ((float(d-low)/float(high-low)) * (size)) + x1 = start+start_offset + path = [(x1,y1),(x1+w,y3),(x1,y2),(x1-w,y3)] + self.d.draw_path('solid',path) + + if b != Date.UNDEF and d != Date.UNDEF: + start_offset = ((float(b-low)/float(high-low)) * size) + w + stop_offset = ((float(d-low)/float(high-low)) * size) - w + + x1 = start+start_offset + x2 = start+stop_offset + self.d.draw_line('open',x1,y3,x2,y3) + + if (y2 + incr) >= self.d.get_usable_height(): + if current != length: + self.d.end_page() + self.d.start_page() + self.build_grid(low,high,start,stop) + index = 1 + x1,x2,y1,y2 = (0,0,0,0) + else: + index += 1; + current += 1 + + self.d.end_page() + self.d.close() + + def build_grid(self,year_low,year_high,start_pos,stop_pos): + """ + Draws the grid outline for the chart. Sets the document label, + draws the vertical lines, and adds the year labels. Arguments + are: + + year_low - lowest year on the chart + year_high - highest year on the chart + start_pos - x position of the lowest leftmost grid line + stop_pos - x position of the rightmost grid line + """ + width = self.d.get_usable_width() + + title_font = self.d.style_list['Title'].get_font() + normal_font = self.d.style_list['Name'].get_font() + label_font = self.d.style_list['Name'].get_font() + + tstr_width = pt2cm(FontScale.string_width(title_font,self.title)) + + title_x = (width - tstr_width )/2.0 + title_y = 0 + self.d.draw_text('title',self.title,title_x,title_y) + + label_y = self.header - (pt2cm(normal_font.get_size())*1.2) + top_y = self.header + bottom_y = self.d.get_usable_height() + + incr = (year_high - year_low)/5 + delta = (stop_pos - start_pos)/ 5 + + for val in range(0,6): + year_str = str(year_low + (incr*val)) + year_width = pt2cm(FontScale.string_width(label_font,year_str))/2.0 + + xpos = start_pos+(val*delta) + label_xpos = start_pos+(val*delta) - year_width + + self.d.draw_text('label', year_str, label_xpos, label_y) + self.d.draw_line('grid', xpos, top_y, xpos, bottom_y) + + def find_year_range(self): + low = 999999 + high = -999999 + + self.plist = self.filter.apply(self.db,self.db.getPersonMap().values()) + + for p in self.plist: + b = p.getBirth().getDateObj().getYear() + d = p.getDeath().getDateObj().getYear() + + if b != Date.UNDEF: + low = min(low,b) + high = max(high,b) + + if d != Date.UNDEF: + low = min(low,b) + high = max(high,b) + + low = (low/10)*10 + high = ((high+9)/10)*10 + + return (low,high) + + def name_size(self): + self.plist = self.filter.apply(self.db,self.db.getPersonMap().values()) + + style_name = self.d.draw_styles['text'].get_paragraph_style() + font = self.d.style_list[style_name].get_font() + + size = 0 + for p in self.plist: + n = p.getPrimaryName().getName() + size = max(FontScale.string_width(font,n),size) + return pt2cm(size) + + +#------------------------------------------------------------------------ +# +# TimeLineDialog +# +#------------------------------------------------------------------------ +class TimeLineDialog(Report.DrawReportDialog): + + def __init__(self,database,person): + Report.DrawReportDialog.__init__(self,database,person) + + def get_title(self): + """The window title for this dialog""" + return "%s - %s - GRAMPS" % (_("Timeline"), + _("Graphical Reports")) + + def get_target_browser_title(self): + """The title of the window created when the 'browse' button is + clicked in the 'Save As' frame.""" + return _("TimeLine File") + + def get_report_generations(self): + """No generation options.""" + return (0, 0) + + def add_user_options(self): + """ + Override the base class add_user_options task to add a menu that allows + the user to select the sort method. + """ + + self.sort_style = gtk.OptionMenu() + self.sort_menu = gtk.Menu() + + for item in [(_("Birth Date"),sort.by_birthdate),(_("Name"),sort.by_last_name)]: + menuitem = gtk.MenuItem(item[0]) + menuitem.set_data('sort',item[1]) + menuitem.show() + self.sort_menu.append(menuitem) + + self.sort_style.set_menu(self.sort_menu) + self.add_option(_('Sort by'),self.sort_style) + + self.title_box = gtk.Entry() + self.title_box.show() + self.add_option(_('Title'),self.title_box) + + def get_report_filters(self): + """Set up the list of possible content filters.""" + + name = self.person.getPrimaryName().getName() + + all = GenericFilter.GenericFilter() + all.set_name(_("Entire Database")) + all.add_rule(GenericFilter.Everyone([])) + + des = GenericFilter.GenericFilter() + des.set_name(_("Descendants of %s") % name) + des.add_rule(GenericFilter.IsDescendantOf([self.person.getId()])) + + ans = GenericFilter.GenericFilter() + ans.set_name(_("Ancestors of %s") % name) + ans.add_rule(GenericFilter.IsAncestorOf([self.person.getId()])) + + com = GenericFilter.GenericFilter() + com.set_name(_("People with common ancestor with %s") % name) + com.add_rule(GenericFilter.HasCommonAncestorWith([self.person.getId()])) + + return [all,des,ans,com] + + def make_default_style(self): + """Make the default output style for the Ancestor Chart report.""" + f = TextDoc.FontStyle() + f.set_size(10) + f.set_type_face(TextDoc.FONT_SANS_SERIF) + p = TextDoc.ParagraphStyle() + p.set_font(f) + self.default_style.add_style("Name",p) + + f = TextDoc.FontStyle() + f.set_size(8) + f.set_type_face(TextDoc.FONT_SANS_SERIF) + p = TextDoc.ParagraphStyle() + p.set_font(f) + self.default_style.add_style("Label",p) + + f = TextDoc.FontStyle() + f.set_size(14) + f.set_type_face(TextDoc.FONT_SANS_SERIF) + p = TextDoc.ParagraphStyle() + p.set_font(f) + self.default_style.add_style("Title",p) + + def make_report(self): + + title = self.title_box.get_text() + sort_func = self.sort_menu.get_active().get_data('sort') + + try: + MyReport = TimeLine(self.db, self.person, self.target_path, + self.doc, self.filter, title, sort_func) + MyReport.setup() + MyReport.write_report() + except Errors.ReportError, msg: + ErrorDialog(str(msg)) + except: + import DisplayTrace + DisplayTrace.DisplayTrace() + +#------------------------------------------------------------------------ +# +# point to centimeter convertion +# +#------------------------------------------------------------------------ +def pt2cm(val): + return (float(val)/28.3465) + +#------------------------------------------------------------------------ +# +# Register the TimeLine report with the plugin system. The register_report +# task of the Plugins module takes the following arguments. +# +# task - function that starts the task +# name - Name of the report +# status - alpha/beta/production +# category - Category entry in the menu system. +# author_name - Name of the author +# author_email - Author's email address +# description - function that returns the description of the report +# +#------------------------------------------------------------------------ +from Plugins import register_report + +def report(database,person): + """ + report - task starts the report. The plugin system requires that the + task be in the format of task that takes a database and a person as + its arguments. + """ + TimeLineDialog(database,person) + +def get_description(): + """ + get_description - returns a descriptive name for the report. The plugin + system uses this to provide a description in the report selector. + """ + return _("Generates a timeline graph.") + +register_report( + task=report, + name=_("TimeLine Graph"), + status=(_("Beta")), + category=_("Graphical Reports"), + author_name="Donald N. Allingham", + author_email="dallingham@users.sourceforge.net", + description=get_description() + ) +