370 lines
11 KiB
Python
370 lines
11 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
|
|
#
|
|
|
|
#-------------------------------------------------------------------------
|
|
#
|
|
# Standard python libraries
|
|
#
|
|
#-------------------------------------------------------------------------
|
|
from math import sin, cos, pi
|
|
|
|
#-------------------------------------------------------------------------
|
|
#
|
|
# GRAMPS libraries
|
|
#
|
|
#-------------------------------------------------------------------------
|
|
import Plugins
|
|
import Errors
|
|
|
|
import TextDoc
|
|
import DrawDoc
|
|
from intl import gettext as _
|
|
|
|
#-------------------------------------------------------------------------
|
|
#
|
|
# ReportLab PDF detection, PIL detection
|
|
#
|
|
#-------------------------------------------------------------------------
|
|
try:
|
|
from reportlab.pdfgen import canvas
|
|
from reportlab.lib.units import cm
|
|
from reportlab.lib.colors import Color
|
|
except ImportError:
|
|
raise Errors.PluginError( _("The ReportLab modules are not installed"))
|
|
|
|
|
|
_H = 'Helvetica'
|
|
_HB = 'Helvetica-Bold'
|
|
_T = 'Times-Roman'
|
|
_TB = 'Times-Bold'
|
|
|
|
#-------------------------------------------------------------------------
|
|
#
|
|
# PdfDrawDoc
|
|
#
|
|
#-------------------------------------------------------------------------
|
|
class PdfDrawDoc(DrawDoc.DrawDoc):
|
|
"""
|
|
PDF drawing class. Uses the ReportLab libraries to build the PDF files
|
|
"""
|
|
|
|
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):
|
|
try:
|
|
self.f.save()
|
|
except IOError,msg:
|
|
raise Errors.ReportError(_("Could not create %s") % self.filename,msg)
|
|
except:
|
|
raise Errors.ReportError(_("Could not create %s") % self.filename)
|
|
|
|
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_wedge(self, style, centerx, centery, radius, start_angle,
|
|
end_angle, short_radius=0):
|
|
|
|
centerx += self.lmargin
|
|
centery += self.bmargin
|
|
|
|
def rnd(val):
|
|
return val*cm
|
|
|
|
while end_angle < start_angle:
|
|
end_angle += 360
|
|
|
|
stype = self.draw_styles[style]
|
|
if stype.get_line_style() == DrawDoc.SOLID:
|
|
self.f.setDash([],0)
|
|
else:
|
|
self.f.setDash([2,4],0)
|
|
|
|
degreestoradians = pi/180.0
|
|
radiansdelta = degreestoradians/2
|
|
sangle = start_angle*degreestoradians
|
|
eangle = end_angle*degreestoradians
|
|
while eangle<sangle:
|
|
eangle = eangle+2*pi
|
|
angle = sangle
|
|
|
|
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()
|
|
if short_radius == 0:
|
|
p.moveTo(rnd(centerx),rnd(centery))
|
|
else:
|
|
origx = rnd(centerx + cos(angle)*short_radius)
|
|
origy = rnd(centery + sin(angle)*short_radius)
|
|
p.moveTo(origx, origy)
|
|
|
|
while angle<eangle:
|
|
x = centerx + cos(angle)*radius
|
|
y = centery + sin(angle)*radius
|
|
p.lineTo(rnd(x),rnd(y))
|
|
angle = angle+radiansdelta
|
|
x = centerx + cos(eangle)*radius
|
|
y = centery + sin(eangle)*radius
|
|
p.lineTo(rnd(x),rnd(y))
|
|
|
|
if short_radius:
|
|
x = centerx + cos(eangle)*short_radius
|
|
y = centery + sin(eangle)*short_radius
|
|
p.lineTo(rnd(x),rnd(y))
|
|
|
|
angle = eangle
|
|
while angle>=sangle:
|
|
x = centerx + cos(angle)*short_radius
|
|
y = centery + sin(angle)*short_radius
|
|
p.lineTo(rnd(x),rnd(y))
|
|
angle = angle-radiansdelta
|
|
p.close()
|
|
self.f.drawPath(p,stroke=1,fill=1)
|
|
|
|
delta = (eangle - sangle)/2.0
|
|
rad = short_radius + (radius-short_radius)/2.0
|
|
|
|
return ( (centerx + cos(sangle+delta) * rad)-self.lmargin,
|
|
(centery + sin(sangle+delta) * rad)-self.tmargin)
|
|
|
|
|
|
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(box_style.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_fill_color()))
|
|
|
|
self.f.rect(x*cm,y*cm,w,h,fill=1)
|
|
|
|
if text != "":
|
|
lines = text.split('\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 pdf_set_font(self,font):
|
|
size = font.get_size()
|
|
if font.get_type_face() == TextDoc.FONT_SANS_SERIF:
|
|
if font.get_bold():
|
|
self.f.setFont(_HB,size)
|
|
else:
|
|
self.f.setFont(_H,size)
|
|
else:
|
|
if font.get_bold():
|
|
self.f.setFont(_TB,size)
|
|
else:
|
|
self.f.setFont(_T,size)
|
|
|
|
def rotate_text(self,style,text,x,y,angle):
|
|
|
|
x += self.lmargin
|
|
y += self.tmargin
|
|
|
|
stype = self.draw_styles[style]
|
|
pname = stype.get_paragraph_style()
|
|
p = self.style_list[pname]
|
|
font = p.get_font()
|
|
|
|
size = font.get_size()
|
|
|
|
self.f.saveState()
|
|
self.pdf_set_font(font)
|
|
|
|
self.f.translate(x*cm,y*cm)
|
|
self.f.rotate(angle)
|
|
self.f.setStrokeColor(make_color(font.get_color()))
|
|
self.f.setFillColor(make_color(font.get_color()))
|
|
|
|
val = len(text)
|
|
y = ((-font.get_size() * val)/2.0) + font.get_size()
|
|
|
|
for line in text:
|
|
self.f.drawCentredString(0,y,line.encode('iso-8859-1'))
|
|
y += font.get_size()
|
|
|
|
self.f.restoreState()
|
|
|
|
def center_text(self,style,text,x,y):
|
|
x += self.lmargin
|
|
y += self.tmargin
|
|
|
|
stype = self.draw_styles[style]
|
|
pname = stype.get_paragraph_style()
|
|
p = self.style_list[pname]
|
|
font = p.get_font()
|
|
|
|
self.f.saveState()
|
|
self.f.setStrokeColor(make_color(font.get_color()))
|
|
self.f.setFillColor(make_color(font.get_color()))
|
|
|
|
self.pdf_set_font(font)
|
|
|
|
self.f.drawCentredString(x*cm,y*cm,text.encode('iso-8859-1'))
|
|
self.f.restoreState()
|
|
|
|
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()))
|
|
self.f.setStrokeColor(make_color(font.get_color()))
|
|
self.pdf_set_font(font)
|
|
|
|
for text in lines:
|
|
self.f.drawCentredString(start_x,start_y,text.encode('iso-8859-1'))
|
|
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()
|
|
|
|
self.f.saveState()
|
|
self.f.setStrokeColor(make_color(font.get_color()))
|
|
|
|
self.pdf_set_font(font)
|
|
|
|
self.f.drawString(start_x,start_y,text.encode('iso-8859-1'))
|
|
self.f.restoreState()
|
|
|
|
|
|
def make_color(c):
|
|
return Color(float(c[0])/255.0, float(c[1])/255.0, float(c[2])/255.0)
|
|
|
|
|
|
#-------------------------------------------------------------------------
|
|
#
|
|
# Register the document class
|
|
#
|
|
#-------------------------------------------------------------------------
|
|
|
|
Plugins.register_draw_doc(_("PDF"),PdfDrawDoc,1,1,".pdf");
|
|
|