# # 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 Timeline 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() )