#!/usr/bin/env python #coding:utf-8 # Author: mozman --<mozman@gmx.at> # Purpose: filters module # Created: 03.11.2010 # Copyright (C) 2010, Manfred Moitzi # License: MIT License from svgwrite.base import BaseElement from svgwrite.mixins import XLink, Presentation from svgwrite.utils import strlist, is_string __all__ = ['Filter'] class _feDistantLight(BaseElement): elementname = 'feDistantLight' def __init__(self, azimuth=0, elevation=0, **extra): super(_feDistantLight, self).__init__(**extra) if azimuth != 0: self['azimuth'] = azimuth if elevation != 0: self['elevation'] = elevation class _fePointLight(BaseElement): elementname = 'fePointLight' def __init__(self, source=(0, 0, 0), **extra): super(_fePointLight, self).__init__(**extra) x, y, z = source if x != 0: self['x'] = x if y != 0: self['y'] = y if z != 0: self['z'] = z class _feSpotLight(_fePointLight): elementname = 'feSpotLight' def __init__(self, source=(0, 0, 0), target=(0, 0, 0), **extra): super(_feSpotLight, self).__init__(source, **extra) x, y, z = target if x != 0: self['pointsAtX'] = x if y != 0: self['pointsAtY'] = y if z != 0: self['pointsAtZ'] = z class _FilterPrimitive(BaseElement, Presentation): pass class _FilterNoInput(_FilterPrimitive): def __init__(self, start=None, size=None, **extra): super(_FilterNoInput, self).__init__(**extra) if start is not None: self['x'] = start[0] self['y'] = start[1] if size is not None: self['width'] = size[0] self['height'] = size[1] class _FilterRequireInput(_FilterNoInput): def __init__(self, in_='SourceGraphic', **extra): super(_FilterRequireInput, self).__init__(**extra) self['in'] = in_ class _feBlend(_FilterRequireInput): elementname = 'feBlend' class _feColorMatrix(_FilterRequireInput): elementname = 'feColorMatrix' class _feComponentTransfer(_FilterRequireInput): elementname = 'feComponentTransfer' def feFuncR(self, type_, **extra): return self.add(_feFuncR(type_, factory=self, **extra)) def feFuncG(self, type_, **extra): return self.add(_feFuncG(type_, factory=self, **extra)) def feFuncB(self, type_, **extra): return self.add(_feFuncB(type_, factory=self, **extra)) def feFuncA(self, type_, **extra): return self.add(_feFuncA(type_, factory=self, **extra)) class _feFuncR(_FilterPrimitive): elementname = 'feFuncR' def __init__(self, type_, **extra): super(_feFuncR, self).__init__(**extra) self['type'] = type_ class _feFuncG(_feFuncR): elementname = 'feFuncG' class _feFuncB(_feFuncR): elementname = 'feFuncB' class _feFuncA(_feFuncR): elementname = 'feFuncA' class _feComposite(_FilterRequireInput): elementname = 'feComposite' class _feConvolveMatrix(_FilterRequireInput): elementname = 'feConvolveMatrix' class _feDiffuseLighting(_FilterRequireInput): elementname = 'feDiffuseLighting' def feDistantLight(self, azimuth=0, elevation=0, **extra): return self.add(_feDistantLight(azimuth, elevation, **extra)) def fePointLight(self, source=(0, 0, 0), **extra): return self.add(_fePointLight(source, **extra)) def feSpotLight(self, source=(0, 0, 0), target=(0, 0, 0), **extra): return self.add(_feSpotLight(source, target, **extra)) class _feDisplacementMap(_FilterRequireInput): elementname = 'feDisplacementMap' class _feFlood(_FilterNoInput): elementname = 'feFlood' class _feGaussianBlur(_FilterRequireInput): elementname = 'feGaussianBlur' class _feImage(_FilterNoInput, XLink): elementname = 'feImage' def __init__(self, href, start=None, size=None, **extra): super(_feImage, self).__init__(start, size, **extra) self.set_href(href) class _feMergeNode(_FilterPrimitive): elementname = 'feMergeNode' class _feMerge(_FilterNoInput): elementname = 'feMerge' def __init__(self, layernames, **extra): super(_feMerge, self).__init__(**extra) self.feMergeNode(layernames) def feMergeNode(self, layernames): for layername in layernames: self.add(_feMergeNode(in_=layername, factory=self)) class _feMorphology(_FilterRequireInput): elementname = 'feMorphology' class _feOffset(_FilterRequireInput): elementname = 'feOffset' class _feSpecularLighting(_feDiffuseLighting): elementname = 'feSpecularLighting' class _feTile(_FilterRequireInput): elementname = 'feTile' class _feTurbulence(_FilterNoInput): elementname = 'feTurbulence' filter_factory = { 'feBlend': _feBlend, 'feColorMatrix': _feColorMatrix, 'feComponentTransfer': _feComponentTransfer, 'feComposite': _feComposite, 'feConvolveMatrix': _feConvolveMatrix, 'feDiffuseLighting': _feDiffuseLighting, 'feDisplacementMap': _feDisplacementMap, 'feFlood': _feFlood, 'feGaussianBlur': _feGaussianBlur, 'feImage': _feImage, 'feMerge': _feMerge, 'feMorphology': _feMorphology, 'feOffset': _feOffset, 'feSpecularLighting': _feSpecularLighting, 'feTile': _feTile, 'feTurbulence': _feTurbulence, } class _FilterBuilder(object): def __init__(self, cls, parent): self.cls = cls # primitive filter class to build self.parent = parent # the parent Filter() object def __call__(self, *args, **kwargs): kwargs['factory'] = self.parent # to get the _paramters object obj = self.cls(*args, **kwargs) # create an object of type 'cls' self.parent.add(obj) # add primitive filter to parent Filter() return obj class Filter(BaseElement, XLink, Presentation): """ The filter element is a container element for filter primitives, and also a **factory** for filter primitives. """ elementname = 'filter' def __init__(self, start=None, size=None, resolution=None, inherit=None, **extra): """ :param 2-tuple start: defines the start point of the filter effects region (**x**, **y**) :param 2-tuple size: defines the size of the filter effects region (**width**, **height**) :param resolution: takes the form ``'x-pixels [y-pixels]'``, and indicates the width and height of the intermediate images in pixels. :param inherit: inherits properties from Filter `inherit` see: **xlink:href** """ super(Filter, self).__init__(**extra) if start is not None: self['x'] = start[0] self['y'] = start[1] if size is not None: self['width'] = size[0] self['height'] = size[1] if resolution is not None: if is_string(resolution): self['filterRes'] = resolution elif hasattr(resolution, '__iter__'): self['filterRes'] = strlist(resolution, ' ') else: self['filterRes'] = str(resolution) if inherit is not None: self.href = inherit self.update_id() def get_xml(self): self.update_id() return super(Filter, self).get_xml() def __getattr__(self, name): # create primitive filters by Filter.<filtername>(...) # and auto-add the new filter as subelement of Filter() if name in filter_factory: return _FilterBuilder(filter_factory[name], self) else: raise AttributeError("'%s' has no attribute '%s'" % (self.__class__.__name__, name))