gramps/src/ImgManip.py

165 lines
4.8 KiB
Python
Raw Normal View History

2002-10-20 19:55:16 +05:30
#
# Gramps - a GTK+/GNOME based genealogy program
#
2003-05-15 08:49:43 +05:30
# Copyright (C) 2000-2003 Donald N. Allingham
2002-10-20 19:55:16 +05:30
#
# 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 const
import signal
import md5
import gtk
import gobject
2002-10-20 19:55:16 +05:30
import GrampsMime
2005-12-06 12:08:09 +05:30
import GrampsKeys
import Utils
2002-10-20 19:55:16 +05:30
class ImgManip:
def __init__(self,source):
self.src = source
def size(self):
2005-12-06 12:08:09 +05:30
try:
img = gtk.gdk.pixbuf_new_from_file(self.src)
except gobject.GError:
return (0,0)
return (img.get_width(),img.get_height())
def fmt_thumbnail(self,dest,width,height,cnv):
w = int(width)
h = int(height)
cmd = "%s -geometry %dx%d '%s' '%s:%s'" % (const.convert,w,h,self.src,cnv,dest)
os.system(cmd)
def fmt_convert(self,dest,cnv):
cmd = "%s '%s' '%s:%s'" % (const.convert,self.src,cnv,dest)
os.system(cmd)
2002-10-20 19:55:16 +05:30
def fmt_data(self,cnv):
import popen2
cmd = "%s '%s' '%s:-'" % (const.convert,self.src,cnv)
r,w = popen2.popen2(cmd)
buf = r.read()
r.close()
w.close()
return buf
def fmt_scale_data(self,x,y,cnv):
import popen2
cmd = "%s -geometry %dx%d '%s' '%s:-'" % (const.convert,x,y,self.src,cnv)
signal.signal (signal.SIGCHLD, signal.SIG_DFL)
r,w = popen2.popen2(cmd)
buf = r.read()
r.close()
w.close()
return buf
2002-10-20 19:55:16 +05:30
def jpg_thumbnail(self,dest,width,height):
self.fmt_thumbnail(dest,width,height,"jpeg")
def png_thumbnail(self,dest,width,height):
self.fmt_thumbnail(dest,width,height,"png")
def jpg_convert(self,dest):
self.fmt_convert(dest,"jpeg")
def png_convert(self,dest):
self.fmt_convert(dest,"png")
def eps_convert(self,dest):
self.fmt_convert(dest,"eps")
def jpg_data(self):
return self.fmt_data("jpeg")
def png_data(self):
return self.fmt_data("png")
def jpg_scale_data(self,x,y):
return self.fmt_scale_data(x,y,"jpeg")
def png_scale_data(self,x,y):
return self.fmt_scale_data(x,y,"png")
def _build_thumb_path(path):
base = os.path.expanduser('~/.gramps/thumb')
m = md5.md5(path)
2005-12-06 12:08:09 +05:30
return os.path.join(base,m.hexdigest()+'.png')
def run_thumbnailer(mtype, frm, to, size=const.thumbScale):
sublist = {
'%s' : "%dx%d" % (int(size),int(size)),
'%u' : frm,
'%o' : to,
}
base = '/desktop/gnome/thumbnailers/%s' % mtype.replace('/','@')
cmd = GrampsKeys.client.get_string(base + '/command')
enable = GrampsKeys.client.get_bool(base + '/enable')
if cmd and enable:
cmdlist = map(lambda x: sublist.get(x,x),cmd.split())
if os.fork() == 0:
os.execvp(cmdlist[0],cmdlist)
os.wait()
return True
return False
def set_thumbnail_image(path,mtype=None):
if mtype and not mtype.startswith('image/'):
run_thumbnailer(mtype,path,_build_thumb_path(path))
else:
try:
pixbuf = gtk.gdk.pixbuf_new_from_file(path)
w = pixbuf.get_width()
h = pixbuf.get_height()
scale = const.thumbScale / (float(max(w,h)))
pw = int(w*scale)
ph = int(h*scale)
pixbuf = pixbuf.scale_simple(pw,ph,gtk.gdk.INTERP_BILINEAR)
pixbuf.save(_build_thumb_path(path),"png")
except:
pass
def get_thumbnail_image(path,mtype=None):
filename = _build_thumb_path(path)
2005-12-06 12:08:09 +05:30
try:
2005-12-06 12:08:09 +05:30
path = Utils.find_file( path)
if not os.path.isfile(filename):
set_thumbnail_image(path,mtype)
elif os.path.getmtime(path) > os.path.getmtime(filename):
set_thumbnail_image(path,mtype)
return gtk.gdk.pixbuf_new_from_file(filename)
2005-12-06 12:08:09 +05:30
except (gobject.GError, OSError):
if mtype:
return GrampsMime.find_mime_type_pixbuf(mtype)
2005-12-06 12:08:09 +05:30
else:
return gtk.gdk.pixbuf_new_from_file(os.path.join(const.dataDir,"document.png"))
2005-12-06 12:08:09 +05:30
def get_thumbnail_path(path,mtype=None):
filename = _build_thumb_path(path)
if not os.path.isfile(filename):
2005-12-06 12:08:09 +05:30
set_thumbnail_image(path,mtype)
return filename