#!/usr/bin/env python ''' Copyright (C) 2006 Aaron Spike, aaron@ekips.org Copyright (C) 2010-2012 Nicolas Dufour, nicoduf@yahoo.fr (Windows support and various fixes) 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 library import os import shutil from subprocess import Popen, PIPE import sys import tempfile import datetime # local library import inkex inkex.localize() # Define extension exceptions class GimpXCFError(Exception): pass class GimpXCFExpectedIOError(GimpXCFError): pass class GimpXCFGimpNotInstalled(GimpXCFError): def __init__(self): inkex.errormsg(_('Gimp must be installed and set in your path variable.')) class GimpXCFScriptFuError(GimpXCFError): def __init__(self): inkex.errormsg(_('An error occurred while processing the XCF file.')) class MyEffect(inkex.Effect): def __init__(self): inkex.Effect.__init__(self) self.OptionParser.add_option("--tab", action="store", type="string", dest="tab") self.OptionParser.add_option("-d", "--guides", action="store", type="inkbool", dest="saveGuides", default=False, help="Save the Guides with the .XCF") self.OptionParser.add_option("-r", "--grid", action="store", type="inkbool", dest="saveGrid", default=False, help="Save the Grid with the .XCF") self.OptionParser.add_option("-b", "--background", action="store", type="inkbool", dest="layerBackground", default=False, help="Add background color to each layer") self.OptionParser.add_option("-i", "--dpi", action="store", type="string", dest="resolution", default="90", help="File resolution") self.OptionParser.add_option("-g", "--gimp", action="store", type="string", dest="gimp", default="gimp", help="GIMP path") self.OptionParser.add_option("-c", "--clrtmp", action="store", type="inkbool", dest="clrtmp", default=False, help="Clear TempDir") self.OptionParser.add_option("-t", "--tmpdesktop", action="store", type="inkbool", dest="tmpdesktop", default=False, help="TempDir is Desktop") def output(self): pass def clear_tmp(self, clrtmp): if clrtmp: shutil.rmtree(self.tmp_dir) def effect(self): svg_file = self.args[-1] ttmp_orig = self.document.getroot() docname = ttmp_orig.get(inkex.addNS('docname',u'sodipodi')) if docname is None: docname = self.args[-1] pageHeight = int(self.unittouu(self.xpathSingle('/svg:svg/@height').split('.')[0])) pageWidth = int(self.unittouu(self.xpathSingle('/svg:svg/@width').split('.')[0])) gimp_path = self.options.gimp clrtmp = self.options.clrtmp tempdir_is_desktop = self.options.tmpdesktop # Create os temp dir (to store exported pngs and Gimp log file) if os.name == 'nt' and tempdir_is_desktop: desktop_path = os.getenv("HOMEDRIVE") + os.getenv("HOMEPATH") + "\\Desktop" d = datetime.datetime.today() self.tmp_dir = os.path.join(desktop_path, "tmp%s" % d.strftime("%Y%m%d_%H%M%S")) os.mkdir(self.tmp_dir) else: self.tmp_dir = tempfile.mkdtemp() # Guides hGuides = [] vGuides = [] if self.options.saveGuides: # Grab all guide tags in the namedview tag guideXpath = "sodipodi:namedview/sodipodi:guide" for guideNode in self.document.xpath(guideXpath, namespaces=inkex.NSS): ori = guideNode.get('orientation') if ori == '0,1': # This is a horizontal guide pos = int(guideNode.get('position').split(',')[1].split('.')[0]) # GIMP doesn't like guides that are outside of the image if pos > 0 and pos < pageHeight: # The origin is at the top in GIMP land hGuides.append(str(pageHeight - pos)) elif ori == '1,0': # This is a vertical guide pos = int(guideNode.get('position').split(',')[0].split('.')[0]) # GIMP doesn't like guides that are outside of the image if pos > 0 and pos < pageWidth: vGuides.append(str(pos)) hGList = ' '.join(hGuides) vGList = ' '.join(vGuides) # Grid gridSpacingFunc = '' gridOriginFunc = '' # GIMP only allows one rectangular grid gridXpath = "sodipodi:namedview/inkscape:grid[@type='xygrid' and (not(@units) or @units='px')]" if (self.options.saveGrid and self.document.xpath(gridXpath, namespaces=inkex.NSS)): gridNode = self.xpathSingle(gridXpath) if gridNode != None: # These attributes could be nonexistant spacingX = gridNode.get('spacingx') if spacingX == None: spacingX = '1 ' spacingY = gridNode.get('spacingy') if spacingY == None: spacingY = '1 ' originX = gridNode.get('originx') if originX == None: originX = '0 ' originY = gridNode.get('originy') if originY == None: originY = '0 ' gridSpacingFunc = '(gimp-image-grid-set-spacing img %s %s)' % (spacingX[:-2], spacingY[:-2]) gridOriginFunc = '(gimp-image-grid-set-offset img %s %s)'% (originX[:-2], originY[:-2]) # Layers area = '--export-area-page' opacity = '--export-background-opacity=' resolution = '--export-dpi=' + self.options.resolution if self.options.layerBackground: opacity += "1" else: opacity += "0" pngs = [] names = [] self.valid = 0 path = "/svg:svg/*[name()='g' or @style][@id]" for node in self.document.xpath(path, namespaces=inkex.NSS): if len(node) > 0: # Get rid of empty layers self.valid = 1 id = node.get('id') if node.get("{" + inkex.NSS["inkscape"] + "}label"): name = node.get("{" + inkex.NSS["inkscape"] + "}label") else: name = id filename = os.path.join(self.tmp_dir, "%s.png" % id) command = "inkscape -i \"%s\" -j %s %s -e \"%s\" %s %s" % (id, area, opacity, filename, svg_file, resolution) p = Popen(command, shell=True, stdin=PIPE, stdout=PIPE, stderr=PIPE) return_code = p.wait() f = p.stdout err = p.stderr stdin = p.stdin f.read() f.close() err.close() stdin.close() if os.name == 'nt': filename = filename.replace("\\", "/") pngs.append(filename) names.append(name.encode('utf-8')) if (self.valid == 0): self.clear_tmp(clrtmp) inkex.errormsg(_('This extension requires at least one non empty layer.')) else: filelist = '"%s"' % '" "'.join(pngs) namelist = '"%s"' % '" "'.join(names) if not docname.endswith('.xcf'): docname = "%s.xcf" % docname xcf = os.path.join(self.tmp_dir, docname) if os.name == 'nt': xcf = xcf.replace("\\", "/") script_fu = """ (tracing 1) (define (png-to-layer img png_filename layer_name) (let* ( (png (car (file-png-load RUN-NONINTERACTIVE png_filename png_filename))) (png_layer (car (gimp-image-get-active-layer png))) (xcf_layer (car (gimp-layer-new-from-drawable png_layer img))) ) (gimp-image-add-layer img xcf_layer -1) (gimp-drawable-set-name xcf_layer layer_name) ) ) (let* ( (img (car (gimp-image-new 200 200 RGB))) ) (gimp-image-set-resolution img %s %s) (gimp-image-undo-disable img) (for-each (lambda (names) (png-to-layer img (car names) (cdr names)) ) (map cons '(%s) '(%s)) ) (gimp-image-resize-to-layers img) (for-each (lambda (hGuide) (gimp-image-add-hguide img hGuide) ) '(%s) ) (for-each (lambda (vGuide) (gimp-image-add-vguide img vGuide) ) '(%s) ) %s %s (gimp-image-undo-enable img) (gimp-file-save RUN-NONINTERACTIVE img (car (gimp-image-get-active-layer img)) "%s" "%s")) (gimp-quit 0) """ % (self.options.resolution, self.options.resolution, filelist, namelist, hGList, vGList, gridSpacingFunc, gridOriginFunc, xcf, xcf) junk = os.path.join(self.tmp_dir, 'junk_from_gimp.txt') logfile = os.path.join(self.tmp_dir, 'log.txt') scmfile = os.path.join(self.tmp_dir, '%s.scm' % docname) f = open(scmfile, 'w') f.write(script_fu.encode('utf-8')) f.close() if os.name == "nt": opts = '-i -f -d -s' scm = scmfile.replace("\\", "/") command = '%s %s -b "(load \\"%s\\")"' % (gimp_path, opts, scm) # command += ' > ' + junk + ' 2>&1' f = open(logfile, "w") if os.path.exists(gimp_path): f.write("found %s\n" % gimp_path) else: f.write("Error : not found %s\n" % gimp_path) f.write("docname: %s\n" % docname) f.write(command + "\n") f.close() p = Popen(command, shell=True) return_code = p.wait() else: command = 'gimp -i -f -d -s --batch-interpreter plug-in-script-fu-eval -b - > %s 2>&1' % junk p = Popen(command, shell=True, stdin=PIPE, stdout=PIPE, stderr=PIPE) f = p.stdin out = p.stdout err = p.stderr f.write(script_fu.encode('utf-8')) return_code = p.wait() if p.returncode != 0: self.clear_tmp(clrtmp) raise GimpXCFGimpNotInstalled f.close() err.close() out.close() # Uncomment these lines to see the output from gimp err = open(junk, 'r') inkex.debug(err.read()) err.close() try: x = open(xcf, 'rb') except: self.clear_tmp(clrtmp) raise GimpXCFScriptFuError if os.name == 'nt': try: import msvcrt msvcrt.setmode(1, os.O_BINARY) except: pass try: sys.stdout.write(x.read()) finally: x.close() self.clear_tmp(clrtmp) if __name__ == '__main__': e = MyEffect() e.affect()