gramps/gramps2/src/docgen/PdfDrawDoc.py
Don Allingham dae5e3548e Drawing improvements
svn: r1290
2003-02-08 23:44:44 +00:00

218 lines
6.5 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
#
import os
import string
import Plugins
import Errors
import TextDoc
import DrawDoc
from intl import gettext as _
try:
from reportlab.pdfgen import canvas
from reportlab.lib.units import cm
from reportlab.lib.colors import Color
except:
raise Errors.PluginError( _("The ReportLab modules are not installed"))
def make_color(color):
return Color(float(color[0])/255.0, float(color[1])/255.0,
float(color[2])/255.0)
class PdfDrawDoc(DrawDoc.DrawDoc):
def __init__(self,styles,type,orientation):
DrawDoc.DrawDoc.__init__(self,styles,type,orientation)
self.f = None
self.filename = None
self.level = 0
self.time = "0000-00-00T00:00:00"
self.page = 0
def open(self,filename):
if filename[-4:] != ".pdf":
self.filename = filename + ".pdf"
else:
self.filename = filename
self.f = canvas.Canvas(self.filename,(self.width*cm,self.height*cm),0)
if self.name:
self.f.setAuthor(self.name)
def close(self):
self.f.save()
def start_paragraph(self,style_name):
pass
def end_paragraph(self):
pass
def write_text(self,text):
pass
def start_page(self,orientation=None):
pass
def end_page(self):
self.f.showPage()
def draw_line(self,style,x1,y1,x2,y2):
x1 = x1 + self.lmargin
x2 = x2 + self.lmargin
y1 = y1 + self.tmargin
y2 = y2 + self.tmargin
stype = self.draw_styles[style]
if stype.get_line_style() == DrawDoc.SOLID:
self.f.setDash([],0)
else:
self.f.setDash([2,4],0)
self.f.setLineWidth(stype.get_line_width())
self.f.line(x1*cm,y1*cm,x2*cm,y2*cm)
def draw_bar(self,style,x1,y1,x2,y2):
x1 = x1 + self.lmargin
x2 = x2 + self.lmargin
y1 = y1 + self.tmargin
y2 = y2 + self.tmargin
stype = self.draw_styles[style]
if stype.get_line_style() == DrawDoc.SOLID:
self.f.setDash([],0)
else:
self.f.setDash([2,4],0)
self.f.setLineWidth(stype.get_line_width())
self.f.rect(x1*cm,y1*cm,(x2-x1)*cm,(y2-y1)*cm,fill=0,stroke=1)
def draw_path(self,style,path):
stype = self.draw_styles[style]
if stype.get_line_style() == DrawDoc.SOLID:
self.f.setDash([],0)
else:
self.f.setDash([2,4],0)
self.f.setLineWidth(stype.get_line_width())
color = stype.get_fill_color()
self.f.setFillColor((float(color[0])/255.0,float(color[1])/255.0,float(color[2])/255.0))
p = self.f.beginPath()
point = path[0]
p.moveTo((point[0]+self.lmargin)*cm,(point[1]+self.tmargin)*cm)
for point in path[1:]:
p.lineTo((point[0]+self.lmargin)*cm,(point[1]+self.tmargin)*cm)
p.close()
fill = stype.get_color()
if fill[0] == 0:
self.f.drawPath(p,stroke=1,fill=1)
else:
self.f.drawPath(p,stroke=1,fill=0)
def draw_box(self,style,text,x,y):
x = x + self.lmargin
y = y + self.tmargin
box_style = self.draw_styles[style]
para_name = box_style.get_paragraph_style()
p = self.style_list[para_name]
w = box_style.get_width()*cm
h = box_style.get_height()*cm
self.f.setLineWidth(stype.get_line_width())
if box_style.get_shadow():
self.f.setFillColorRGB(0.5,0.5,0.5)
self.f.rect((x+0.3)*cm,(y+0.3)*cm,w,h,fill=1,stroke=0)
font = p.get_font()
self.f.setStrokeColor(make_color(font.get_color()))
self.f.setFillColor(make_color(box_style.get_color()))
self.f.rect(x*cm,y*cm,w,h,fill=1)
if text != "":
lines = string.split(text,'\n')
self.center_print(lines,font,x*cm,y*cm,w,h)
def draw_text(self,style,text,x,y):
x = x + self.lmargin
y = y + self.tmargin
stype = self.draw_styles[style]
pname = stype.get_paragraph_style()
p = self.style_list[pname]
font = p.get_font()
self.f.setStrokeColor(make_color(font.get_color()))
self.left_print(text,font,x*cm,(y*cm)+font.get_size())
def center_print(self,lines,font,x,y,w,h):
l = len(lines)
size = font.get_size()
start_y = (y + h/2.0 + l/2.0 + l) - ((l*size) + ((l-1)*0.2))/2.0
start_x = (x + w/2.0)
self.f.saveState()
self.f.setFillColor(make_color(font.get_color()))
if font.get_type_face() == TextDoc.FONT_SANS_SERIF:
if font.get_bold():
self.f.setFont("Helvetica-Bold",font.get_size())
else:
self.f.setFont("Helvetica",font.get_size())
else:
if font.get_bold():
self.f.setFont("Times-Bold",font.get_size())
else:
self.f.setFont("Times-Roman",font.get_size())
for text in lines:
self.f.drawCentredString(start_x,start_y,text)
start_y = start_y + size*1.2
start_y = start_y + size*1.2
self.f.restoreState()
def left_print(self,text,font,x,y):
size = font.get_size()
start_y = y
start_x = x
self.f.saveState()
self.f.setFillColor(make_color(font.get_color()))
if font.get_type_face() == TextDoc.FONT_SANS_SERIF:
if font.get_bold():
self.f.setFont("Helvetica-Bold",font.get_size())
else:
self.f.setFont("Helvetica",font.get_size())
else:
if font.get_bold():
self.f.setFont("Times-Bold",font.get_size())
else:
self.f.setFont("Times-Roman",font.get_size())
self.f.drawString(start_x,start_y,text)
self.f.restoreState()
Plugins.register_draw_doc(_("PDF"),PdfDrawDoc);