2013-04-11 22:34:04 +08:00
#!/usr/bin/env python
2012-11-24 02:57:22 +08:00
2011-02-24 02:11:25 +08:00
# svgfig.py copyright (C) 2008 Jim Pivarski <jpivarski@gmail.com>
#
# 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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA
#
# Full licence is in the file COPYING and at http://www.gnu.org/copyleft/gpl.html
import re , codecs , os , platform , copy , itertools , math , cmath , random , sys , copy
_epsilon = 1e-5
if re . search ( " windows " , platform . system ( ) , re . I ) :
try :
import _winreg
_default_directory = _winreg . QueryValueEx ( _winreg . OpenKey ( _winreg . HKEY_CURRENT_USER ,
r " Software \ Microsoft \ Windows \ Current Version \ Explorer \ Shell Folders " ) , " Desktop " ) [ 0 ]
# tmpdir = _winreg.QueryValueEx(_winreg.OpenKey(_winreg.HKEY_CURRENT_USER, "Environment"), "TEMP")[0]
# if tmpdir[0:13] != "%USERPROFILE%":
# tmpdir = os.path.expanduser("~") + tmpdir[13:]
except :
_default_directory = os . path . expanduser ( " ~ " ) + os . sep + " Desktop "
_default_fileName = " tmp.svg "
_hacks = { }
_hacks [ " inkscape-text-vertical-shift " ] = False
def rgb ( r , g , b , maximum = 1. ) :
""" Create an SVG color string " #xxyyzz " from r, g, and b.
r , g , b = 0 is black and r , g , b = maximum is white .
"""
return " # %02x %02x %02x " % ( max ( 0 , min ( r * 255. / maximum , 255 ) ) ,
max ( 0 , min ( g * 255. / maximum , 255 ) ) ,
max ( 0 , min ( b * 255. / maximum , 255 ) ) )
def attr_preprocess ( attr ) :
for name in attr . keys ( ) :
name_colon = re . sub ( " __ " , " : " , name )
if name_colon != name :
attr [ name_colon ] = attr [ name ]
del attr [ name ]
name = name_colon
name_dash = re . sub ( " _ " , " - " , name )
if name_dash != name :
attr [ name_dash ] = attr [ name ]
del attr [ name ]
name = name_dash
return attr
class SVG :
""" A tree representation of an SVG image or image fragment.
SVG ( t , sub , sub , sub . . . , attribute = value )
t required SVG type name
sub optional list nested SVG elements or text / Unicode
attribute = value pairs optional keywords SVG attributes
In attribute names , " __ " becomes " : " and " _ " becomes " - " .
SVG in XML
< g id = " mygroup " fill = " blue " >
< rect x = " 1 " y = " 1 " width = " 2 " height = " 2 " / >
< rect x = " 3 " y = " 3 " width = " 2 " height = " 2 " / >
< / g >
SVG in Python
>> > svg = SVG ( " g " , SVG ( " rect " , x = 1 , y = 1 , width = 2 , height = 2 ) , \
. . . SVG ( " rect " , x = 3 , y = 3 , width = 2 , height = 2 ) , \
. . . id = " mygroup " , fill = " blue " )
Sub - elements and attributes may be accessed through tree - indexing :
>> > svg = SVG ( " text " , SVG ( " tspan " , " hello there " ) , stroke = " none " , fill = " black " )
>> > svg [ 0 ]
< tspan ( 1 sub ) / >
>> > svg [ 0 , 0 ]
' hello there '
>> > svg [ " fill " ]
' black '
Iteration is depth - first :
>> > svg = SVG ( " g " , SVG ( " g " , SVG ( " line " , x1 = 0 , y1 = 0 , x2 = 1 , y2 = 1 ) ) , \
. . . SVG ( " text " , SVG ( " tspan " , " hello again " ) ) )
. . .
>> > for ti , s in svg :
. . . print ti , repr ( s )
. . .
( 0 , ) < g ( 1 sub ) / >
( 0 , 0 ) < line x2 = 1 y1 = 0 x1 = 0 y2 = 1 / >
( 0 , 0 , ' x2 ' ) 1
( 0 , 0 , ' y1 ' ) 0
( 0 , 0 , ' x1 ' ) 0
( 0 , 0 , ' y2 ' ) 1
( 1 , ) < text ( 1 sub ) / >
( 1 , 0 ) < tspan ( 1 sub ) / >
( 1 , 0 , 0 ) ' hello again '
Use " print " to navigate :
>> > print svg
None < g ( 2 sub ) / >
[ 0 ] < g ( 1 sub ) / >
[ 0 , 0 ] < line x2 = 1 y1 = 0 x1 = 0 y2 = 1 / >
[ 1 ] < text ( 1 sub ) / >
[ 1 , 0 ] < tspan ( 1 sub ) / >
"""
def __init__ ( self , * t_sub , * * attr ) :
if len ( t_sub ) == 0 :
raise TypeError , " SVG element must have a t (SVG type) "
# first argument is t (SVG type)
self . t = t_sub [ 0 ]
# the rest are sub-elements
self . sub = list ( t_sub [ 1 : ] )
# keyword arguments are attributes
# need to preprocess to handle differences between SVG and Python syntax
self . attr = attr_preprocess ( attr )
def __getitem__ ( self , ti ) :
""" Index is a list that descends tree, returning a sub-element if
it ends with a number and an attribute if it ends with a string . """
obj = self
if isinstance ( ti , ( list , tuple ) ) :
for i in ti [ : - 1 ] :
obj = obj [ i ]
ti = ti [ - 1 ]
if isinstance ( ti , ( int , long , slice ) ) :
return obj . sub [ ti ]
else :
return obj . attr [ ti ]
def __setitem__ ( self , ti , value ) :
""" Index is a list that descends tree, returning a sub-element if
it ends with a number and an attribute if it ends with a string . """
obj = self
if isinstance ( ti , ( list , tuple ) ) :
for i in ti [ : - 1 ] :
obj = obj [ i ]
ti = ti [ - 1 ]
if isinstance ( ti , ( int , long , slice ) ) :
obj . sub [ ti ] = value
else :
obj . attr [ ti ] = value
def __delitem__ ( self , ti ) :
""" Index is a list that descends tree, returning a sub-element if
it ends with a number and an attribute if it ends with a string . """
obj = self
if isinstance ( ti , ( list , tuple ) ) :
for i in ti [ : - 1 ] :
obj = obj [ i ]
ti = ti [ - 1 ]
if isinstance ( ti , ( int , long , slice ) ) :
del obj . sub [ ti ]
else :
del obj . attr [ ti ]
def __contains__ ( self , value ) :
""" x in svg == True iff x is an attribute in svg. """
return value in self . attr
def __eq__ ( self , other ) :
""" x == y iff x represents the same SVG as y. """
if id ( self ) == id ( other ) :
return True
return ( isinstance ( other , SVG ) and
self . t == other . t and self . sub == other . sub and self . attr == other . attr )
def __ne__ ( self , other ) :
""" x != y iff x does not represent the same SVG as y. """
return not ( self == other )
def append ( self , x ) :
""" Appends x to the list of sub-elements (drawn last, overlaps
other primitives ) . """
self . sub . append ( x )
def prepend ( self , x ) :
""" Prepends x to the list of sub-elements (drawn first may be
overlapped by other primitives ) . """
self . sub [ 0 : 0 ] = [ x ]
def extend ( self , x ) :
""" Extends list of sub-elements by a list x. """
self . sub . extend ( x )
def clone ( self , shallow = False ) :
""" Deep copy of SVG tree. Set shallow=True for a shallow copy. """
if shallow :
return copy . copy ( self )
else :
return copy . deepcopy ( self )
### nested class
class SVGDepthIterator :
""" Manages SVG iteration. """
def __init__ ( self , svg , ti , depth_limit ) :
self . svg = svg
self . ti = ti
self . shown = False
self . depth_limit = depth_limit
def __iter__ ( self ) :
return self
def next ( self ) :
if not self . shown :
self . shown = True
if self . ti != ( ) :
return self . ti , self . svg
if not isinstance ( self . svg , SVG ) :
raise StopIteration
if self . depth_limit is not None and len ( self . ti ) > = self . depth_limit :
raise StopIteration
if " iterators " not in self . __dict__ :
self . iterators = [ ]
for i , s in enumerate ( self . svg . sub ) :
self . iterators . append ( self . __class__ ( s , self . ti + ( i , ) , self . depth_limit ) )
for k , s in self . svg . attr . items ( ) :
self . iterators . append ( self . __class__ ( s , self . ti + ( k , ) , self . depth_limit ) )
self . iterators = itertools . chain ( * self . iterators )
return self . iterators . next ( )
### end nested class
def depth_first ( self , depth_limit = None ) :
""" Returns a depth-first generator over the SVG. If depth_limit
is a number , stop recursion at that depth . """
return self . SVGDepthIterator ( self , ( ) , depth_limit )
def breadth_first ( self , depth_limit = None ) :
""" Not implemented yet. Any ideas on how to do it?
Returns a breadth - first generator over the SVG . If depth_limit
is a number , stop recursion at that depth . """
raise NotImplementedError , " Got an algorithm for breadth-first searching a tree without effectively copying the tree? "
def __iter__ ( self ) :
return self . depth_first ( )
def items ( self , sub = True , attr = True , text = True ) :
""" Get a recursively-generated list of tree-index, sub-element/attribute pairs.
If sub == False , do not show sub - elements .
If attr == False , do not show attributes .
If text == False , do not show text / Unicode sub - elements .
"""
output = [ ]
for ti , s in self :
show = False
if isinstance ( ti [ - 1 ] , ( int , long ) ) :
if isinstance ( s , basestring ) :
show = text
else :
show = sub
else :
show = attr
if show :
output . append ( ( ti , s ) )
return output
def keys ( self , sub = True , attr = True , text = True ) :
""" Get a recursively-generated list of tree-indexes.
If sub == False , do not show sub - elements .
If attr == False , do not show attributes .
If text == False , do not show text / Unicode sub - elements .
"""
return [ ti for ti , s in self . items ( sub , attr , text ) ]
def values ( self , sub = True , attr = True , text = True ) :
""" Get a recursively-generated list of sub-elements and attributes.
If sub == False , do not show sub - elements .
If attr == False , do not show attributes .
If text == False , do not show text / Unicode sub - elements .
"""
return [ s for ti , s in self . items ( sub , attr , text ) ]
def __repr__ ( self ) :
return self . xml ( depth_limit = 0 )
def __str__ ( self ) :
""" Print (actually, return a string of) the tree in a form useful for browsing. """
return self . tree ( sub = True , attr = False , text = False )
def tree ( self , depth_limit = None , sub = True , attr = True , text = True , tree_width = 20 , obj_width = 80 ) :
""" Print (actually, return a string of) the tree in a form useful for browsing.
If depth_limit == a number , stop recursion at that depth .
If sub == False , do not show sub - elements .
If attr == False , do not show attributes .
If text == False , do not show text / Unicode sub - elements .
tree_width is the number of characters reserved for printing tree indexes .
obj_width is the number of characters reserved for printing sub - elements / attributes .
"""
output = [ ]
line = " %s %s " % ( ( " %% - %d s " % tree_width ) % repr ( None ) ,
( " %% - %d s " % obj_width ) % ( repr ( self ) ) [ 0 : obj_width ] )
output . append ( line )
for ti , s in self . depth_first ( depth_limit ) :
show = False
if isinstance ( ti [ - 1 ] , ( int , long ) ) :
if isinstance ( s , basestring ) :
show = text
else :
show = sub
else :
show = attr
if show :
line = " %s %s " % ( ( " %% - %d s " % tree_width ) % repr ( list ( ti ) ) ,
( " %% - %d s " % obj_width ) % ( " " * len ( ti ) + repr ( s ) ) [ 0 : obj_width ] )
output . append ( line )
return " \n " . join ( output )
def xml ( self , indent = u " " , newl = u " \n " , depth_limit = None , depth = 0 ) :
""" Get an XML representation of the SVG.
indent string used for indenting
newl string used for newlines
If depth_limit == a number , stop recursion at that depth .
depth starting depth ( not useful for users )
print svg . xml ( )
"""
attrstr = [ ]
for n , v in self . attr . items ( ) :
if isinstance ( v , dict ) :
v = u " ; " . join ( [ u " %s : %s " % ( ni , vi ) for ni , vi in v . items ( ) ] )
elif isinstance ( v , ( list , tuple ) ) :
v = u " , " . join ( v )
attrstr . append ( u " %s = %s " % ( n , repr ( v ) ) )
attrstr = u " " . join ( attrstr )
if len ( self . sub ) == 0 :
return u " %s < %s %s /> " % ( indent * depth , self . t , attrstr )
if depth_limit is None or depth_limit > depth :
substr = [ ]
for s in self . sub :
if isinstance ( s , SVG ) :
substr . append ( s . xml ( indent , newl , depth_limit , depth + 1 ) + newl )
elif isinstance ( s , basestring ) :
substr . append ( u " %s %s %s " % ( indent * ( depth + 1 ) , s , newl ) )
else :
substr . append ( " %s %s %s " % ( indent * ( depth + 1 ) , repr ( s ) , newl ) )
substr = u " " . join ( substr )
return u " %s < %s %s > %s %s %s </ %s > " % ( indent * depth , self . t , attrstr , newl , substr , indent * depth , self . t )
else :
return u " %s < %s ( %d sub) %s /> " % ( indent * depth , self . t , len ( self . sub ) , attrstr )
def standalone_xml ( self , indent = u " " , newl = u " \n " , encoding = u " utf-8 " ) :
""" Get an XML representation of the SVG that can be saved/rendered.
indent string used for indenting
newl string used for newlines
"""
if self . t == " svg " :
top = self
else :
top = canvas ( self )
return u """ \
< ? xml version = " 1.0 " encoding = " %s " standalone = " no " ? >
< ! DOCTYPE svg PUBLIC " -//W3C//DTD SVG 1.1//EN " " http://www.w3.org/Graphics/SVG/1.1/DTD/svg11.dtd " >
""" % e ncoding + (u " " .join(top.__standalone_xml(indent, newl))) # end of return statement
def __standalone_xml ( self , indent , newl ) :
output = [ u " < %s " % self . t ]
for n , v in self . attr . items ( ) :
if isinstance ( v , dict ) :
v = u " ; " . join ( [ u " %s : %s " % ( ni , vi ) for ni , vi in v . items ( ) ] )
elif isinstance ( v , ( list , tuple ) ) :
v = u " , " . join ( v )
output . append ( u ' %s = " %s " ' % ( n , v ) )
if len ( self . sub ) == 0 :
output . append ( u " /> %s %s " % ( newl , newl ) )
return output
elif self . t == " text " or self . t == " tspan " or self . t == " style " :
output . append ( u " > " )
else :
output . append ( u " > %s %s " % ( newl , newl ) )
for s in self . sub :
if isinstance ( s , SVG ) :
output . extend ( s . __standalone_xml ( indent , newl ) )
else :
output . append ( unicode ( s ) )
if self . t == " tspan " :
output . append ( u " </ %s > " % self . t )
else :
output . append ( u " </ %s > %s %s " % ( self . t , newl , newl ) )
return output
def interpret_fileName ( self , fileName = None ) :
if fileName is None :
fileName = _default_fileName
if re . search ( " windows " , platform . system ( ) , re . I ) and not os . path . isabs ( fileName ) :
fileName = _default_directory + os . sep + fileName
return fileName
def save ( self , fileName = None , encoding = " utf-8 " , compresslevel = None ) :
""" Save to a file for viewing. Note that svg.save() overwrites the file named _default_fileName.
fileName default = None note that _default_fileName will be overwritten if
no fileName is specified . If the extension
is " .svgz " or " .gz " , the output will be gzipped
encoding default = " utf-8 " file encoding
compresslevel default = None if a number , the output will be gzipped with that
compression level ( 1 - 9 , 1 being fastest and 9 most
thorough )
"""
fileName = self . interpret_fileName ( fileName )
if compresslevel is not None or re . search ( r " \ .svgz$ " , fileName , re . I ) or re . search ( r " \ .gz$ " , fileName , re . I ) :
import gzip
if compresslevel is None :
f = gzip . GzipFile ( fileName , " w " )
else :
f = gzip . GzipFile ( fileName , " w " , compresslevel )
f = codecs . EncodedFile ( f , " utf-8 " , encoding )
f . write ( self . standalone_xml ( encoding = encoding ) )
f . close ( )
else :
f = codecs . open ( fileName , " w " , encoding = encoding )
f . write ( self . standalone_xml ( encoding = encoding ) )
f . close ( )
def inkview ( self , fileName = None , encoding = " utf-8 " ) :
""" View in " inkview " , assuming that program is available on your system.
fileName default = None note that any file named _default_fileName will be
overwritten if no fileName is specified . If the extension
is " .svgz " or " .gz " , the output will be gzipped
encoding default = " utf-8 " file encoding
"""
fileName = self . interpret_fileName ( fileName )
self . save ( fileName , encoding )
os . spawnvp ( os . P_NOWAIT , " inkview " , ( " inkview " , fileName ) )
def inkscape ( self , fileName = None , encoding = " utf-8 " ) :
""" View in " inkscape " , assuming that program is available on your system.
fileName default = None note that any file named _default_fileName will be
overwritten if no fileName is specified . If the extension
is " .svgz " or " .gz " , the output will be gzipped
encoding default = " utf-8 " file encoding
"""
fileName = self . interpret_fileName ( fileName )
self . save ( fileName , encoding )
os . spawnvp ( os . P_NOWAIT , " inkscape " , ( " inkscape " , fileName ) )
def firefox ( self , fileName = None , encoding = " utf-8 " ) :
""" View in " firefox " , assuming that program is available on your system.
fileName default = None note that any file named _default_fileName will be
overwritten if no fileName is specified . If the extension
is " .svgz " or " .gz " , the output will be gzipped
encoding default = " utf-8 " file encoding
"""
fileName = self . interpret_fileName ( fileName )
self . save ( fileName , encoding )
os . spawnvp ( os . P_NOWAIT , " firefox " , ( " firefox " , fileName ) )
######################################################################
_canvas_defaults = { " width " : " 400px " ,
" height " : " 400px " ,
" viewBox " : " 0 0 100 100 " ,
" xmlns " : " http://www.w3.org/2000/svg " ,
" xmlns:xlink " : " http://www.w3.org/1999/xlink " ,
" version " : " 1.1 " ,
" style " : { " stroke " : " black " ,
" fill " : " none " ,
" stroke-width " : " 0.5pt " ,
" stroke-linejoin " : " round " ,
" text-anchor " : " middle " ,
} ,
" font-family " : [ " Helvetica " , " Arial " , " FreeSans " , " Sans " , " sans " , " sans-serif " ] ,
}
def canvas ( * sub , * * attr ) :
""" Creates a top-level SVG object, allowing the user to control the
image size and aspect ratio .
canvas ( sub , sub , sub . . . , attribute = value )
sub optional list nested SVG elements or text / Unicode
attribute = value pairs optional keywords SVG attributes
Default attribute values :
width " 400px "
height " 400px "
viewBox " 0 0 100 100 "
xmlns " http://www.w3.org/2000/svg "
xmlns : xlink " http://www.w3.org/1999/xlink "
version " 1.1 "
style " stroke:black; fill:none; stroke-width:0.5pt; stroke-linejoin:round; text-anchor:middle "
font - family " Helvetica,Arial,FreeSans?,Sans,sans,sans-serif "
"""
attributes = dict ( _canvas_defaults )
attributes . update ( attr )
if sub is None or sub == ( ) :
return SVG ( " svg " , * * attributes )
else :
return SVG ( " svg " , * sub , * * attributes )
def canvas_outline ( * sub , * * attr ) :
""" Same as canvas(), but draws an outline around the drawable area,
so that you know how close your image is to the edges . """
svg = canvas ( * sub , * * attr )
match = re . match ( r " [, \ t]*([0-9e.+ \ -]+)[, \ t]+([0-9e.+ \ -]+)[, \ t]+([0-9e.+ \ -]+)[, \ t]+([0-9e.+ \ -]+)[, \ t]* " , svg [ " viewBox " ] )
if match is None :
raise ValueError , " canvas viewBox is incorrectly formatted "
x , y , width , height = [ float ( x ) for x in match . groups ( ) ]
svg . prepend ( SVG ( " rect " , x = x , y = y , width = width , height = height , stroke = " none " , fill = " cornsilk " ) )
svg . append ( SVG ( " rect " , x = x , y = y , width = width , height = height , stroke = " black " , fill = " none " ) )
return svg
def template ( fileName , svg , replaceme = " REPLACEME " ) :
""" Loads an SVG image from a file, replacing instances of
< REPLACEME / > with a given svg object .
fileName required name of the template SVG
svg required SVG object for replacement
replaceme default = " REPLACEME " fake SVG element to be replaced by the given object
>> > print load ( " template.svg " )
None < svg ( 2 sub ) style = u ' stroke:black; fill:none; stroke-width:0.5pt; stroke-linejoi
[ 0 ] < rect height = u ' 100 ' width = u ' 100 ' stroke = u ' none ' y = u ' 0 ' x = u ' 0 ' fill = u ' yellow '
[ 1 ] < REPLACEME / >
>> >
>> > print template ( " template.svg " , SVG ( " circle " , cx = 50 , cy = 50 , r = 30 ) )
None < svg ( 2 sub ) style = u ' stroke:black; fill:none; stroke-width:0.5pt; stroke-linejoi
[ 0 ] < rect height = u ' 100 ' width = u ' 100 ' stroke = u ' none ' y = u ' 0 ' x = u ' 0 ' fill = u ' yellow '
[ 1 ] < circle cy = 50 cx = 50 r = 30 / >
"""
output = load ( fileName )
for ti , s in output :
if isinstance ( s , SVG ) and s . t == replaceme :
output [ ti ] = svg
return output
######################################################################
def load ( fileName ) :
""" Loads an SVG image from a file. """
return load_stream ( file ( fileName ) )
def load_stream ( stream ) :
""" Loads an SVG image from a stream (can be a string or a file object). """
from xml . sax import handler , make_parser
from xml . sax . handler import feature_namespaces , feature_external_ges , feature_external_pes
class ContentHandler ( handler . ContentHandler ) :
def __init__ ( self ) :
self . stack = [ ]
self . output = None
self . all_whitespace = re . compile ( r " ^ \ s*$ " )
def startElement ( self , name , attr ) :
s = SVG ( name )
s . attr = dict ( attr . items ( ) )
if len ( self . stack ) > 0 :
last = self . stack [ - 1 ]
last . sub . append ( s )
self . stack . append ( s )
def characters ( self , ch ) :
if not isinstance ( ch , basestring ) or self . all_whitespace . match ( ch ) is None :
if len ( self . stack ) > 0 :
last = self . stack [ - 1 ]
if len ( last . sub ) > 0 and isinstance ( last . sub [ - 1 ] , basestring ) :
last . sub [ - 1 ] = last . sub [ - 1 ] + " \n " + ch
else :
last . sub . append ( ch )
def endElement ( self , name ) :
if len ( self . stack ) > 0 :
last = self . stack [ - 1 ]
if ( isinstance ( last , SVG ) and last . t == " style " and
" type " in last . attr and last . attr [ " type " ] == " text/css " and
len ( last . sub ) == 1 and isinstance ( last . sub [ 0 ] , basestring ) ) :
last . sub [ 0 ] = " <![CDATA[ \n " + last . sub [ 0 ] + " ]]> "
self . output = self . stack . pop ( )
ch = ContentHandler ( )
parser = make_parser ( )
parser . setContentHandler ( ch )
parser . setFeature ( feature_namespaces , 0 )
parser . setFeature ( feature_external_ges , 0 )
parser . parse ( stream )
return ch . output
######################################################################
def set_func_name ( f , name ) :
""" try to patch the function name string into a function object """
try :
f . func_name = name
except TypeError :
# py 2.3 raises: TypeError: readonly attribute
pass
def totrans ( expr , vars = ( " x " , " y " ) , globals = None , locals = None ) :
""" Converts to a coordinate transformation (a function that accepts
two arguments and returns two values ) .
expr required a string expression or a function
of two real or one complex value
vars default = ( " x " , " y " ) independent variable names ; a singleton
( " z " , ) is interpreted as complex
globals default = None dict of global variables
locals default = None dict of local variables
"""
if locals is None :
locals = { } # python 2.3's eval() won't accept None
if callable ( expr ) :
if expr . func_code . co_argcount == 2 :
return expr
elif expr . func_code . co_argcount == 1 :
split = lambda z : ( z . real , z . imag )
output = lambda x , y : split ( expr ( x + y * 1 j ) )
set_func_name ( output , expr . func_name )
return output
else :
raise TypeError , " must be a function of 2 or 1 variables "
if len ( vars ) == 2 :
g = math . __dict__
if globals is not None :
g . update ( globals )
output = eval ( " lambda %s , %s : ( %s ) " % ( vars [ 0 ] , vars [ 1 ] , expr ) , g , locals )
set_func_name ( output , " %s , %s -> %s " % ( vars [ 0 ] , vars [ 1 ] , expr ) )
return output
elif len ( vars ) == 1 :
g = cmath . __dict__
if globals is not None :
g . update ( globals )
output = eval ( " lambda %s : ( %s ) " % ( vars [ 0 ] , expr ) , g , locals )
split = lambda z : ( z . real , z . imag )
output2 = lambda x , y : split ( output ( x + y * 1 j ) )
set_func_name ( output2 , " %s -> %s " % ( vars [ 0 ] , expr ) )
return output2
else :
raise TypeError , " vars must have 2 or 1 elements "
def window ( xmin , xmax , ymin , ymax , x = 0 , y = 0 , width = 100 , height = 100 ,
xlogbase = None , ylogbase = None , minusInfinity = - 1000 , flipx = False , flipy = True ) :
""" Creates and returns a coordinate transformation (a function that
accepts two arguments and returns two values ) that transforms from
( xmin , ymin ) , ( xmax , ymax )
to
( x , y ) , ( x + width , y + height ) .
xlogbase , ylogbase default = None , None if a number , transform
logarithmically with given base
minusInfinity default = - 1000 what to return if
log ( 0 or negative ) is attempted
flipx default = False if true , reverse the direction of x
flipy default = True if true , reverse the direction of y
( When composing windows , be sure to set flipy = False . )
"""
if flipx :
ox1 = x + width
ox2 = x
else :
ox1 = x
ox2 = x + width
if flipy :
oy1 = y + height
oy2 = y
else :
oy1 = y
oy2 = y + height
ix1 = xmin
iy1 = ymin
ix2 = xmax
iy2 = ymax
if xlogbase is not None and ( ix1 < = 0. or ix2 < = 0. ) :
raise ValueError , " x range incompatible with log scaling: ( %g , %g ) " % ( ix1 , ix2 )
if ylogbase is not None and ( iy1 < = 0. or iy2 < = 0. ) :
raise ValueError , " y range incompatible with log scaling: ( %g , %g ) " % ( iy1 , iy2 )
def maybelog ( t , it1 , it2 , ot1 , ot2 , logbase ) :
if t < = 0. :
return minusInfinity
else :
return ot1 + 1. * ( math . log ( t , logbase ) - math . log ( it1 , logbase ) ) / ( math . log ( it2 , logbase ) - math . log ( it1 , logbase ) ) * ( ot2 - ot1 )
xlogstr , ylogstr = " " , " "
if xlogbase is None :
xfunc = lambda x : ox1 + 1. * ( x - ix1 ) / ( ix2 - ix1 ) * ( ox2 - ox1 )
else :
xfunc = lambda x : maybelog ( x , ix1 , ix2 , ox1 , ox2 , xlogbase )
xlogstr = " xlog= %g " % xlogbase
if ylogbase is None :
yfunc = lambda y : oy1 + 1. * ( y - iy1 ) / ( iy2 - iy1 ) * ( oy2 - oy1 )
else :
yfunc = lambda y : maybelog ( y , iy1 , iy2 , oy1 , oy2 , ylogbase )
ylogstr = " ylog= %g " % ylogbase
output = lambda x , y : ( xfunc ( x ) , yfunc ( y ) )
set_func_name ( output , " ( %g , %g ), ( %g , %g ) -> ( %g , %g ), ( %g , %g ) %s %s " % (
ix1 , ix2 , iy1 , iy2 , ox1 , ox2 , oy1 , oy2 , xlogstr , ylogstr ) )
return output
def rotate ( angle , cx = 0 , cy = 0 ) :
""" Creates and returns a coordinate transformation which rotates
around ( cx , cy ) by " angle " degrees . """
angle * = math . pi / 180.
return lambda x , y : ( cx + math . cos ( angle ) * ( x - cx ) - math . sin ( angle ) * ( y - cy ) , cy + math . sin ( angle ) * ( x - cx ) + math . cos ( angle ) * ( y - cy ) )
class Fig :
""" Stores graphics primitive objects and applies a single coordinate
transformation to them . To compose coordinate systems , nest Fig
objects .
Fig ( obj , obj , obj . . . , trans = function )
obj optional list a list of drawing primitives
trans default = None a coordinate transformation function
>> > fig = Fig ( Line ( 0 , 0 , 1 , 1 ) , Rect ( 0.2 , 0.2 , 0.8 , 0.8 ) , trans = " 2*x, 2*y " )
>> > print fig . SVG ( ) . xml ( )
< g >
< path d = ' M0 0L2 2 ' / >
< path d = ' M0.4 0.4L1.6 0.4ZL1.6 1.6ZL0.4 1.6ZL0.4 0.4ZZ ' / >
< / g >
>> > print Fig ( fig , trans = " x/2., y/2. " ) . SVG ( ) . xml ( )
< g >
< path d = ' M0 0L1 1 ' / >
< path d = ' M0.2 0.2L0.8 0.2ZL0.8 0.8ZL0.2 0.8ZL0.2 0.2ZZ ' / >
< / g >
"""
def __repr__ ( self ) :
if self . trans is None :
return " <Fig ( %d items)> " % len ( self . d )
elif isinstance ( self . trans , basestring ) :
return " <Fig ( %d items) x,y -> %s > " % ( len ( self . d ) , self . trans )
else :
return " <Fig ( %d items) %s > " % ( len ( self . d ) , self . trans . func_name )
def __init__ ( self , * d , * * kwds ) :
self . d = list ( d )
defaults = { " trans " : None , }
defaults . update ( kwds )
kwds = defaults
self . trans = kwds [ " trans " ] ; del kwds [ " trans " ]
if len ( kwds ) != 0 :
raise TypeError , " Fig() got unexpected keyword arguments %s " % kwds . keys ( )
def SVG ( self , trans = None ) :
""" Apply the transformation " trans " and return an SVG object.
Coordinate transformations in nested Figs will be composed .
"""
if trans is None :
trans = self . trans
if isinstance ( trans , basestring ) :
trans = totrans ( trans )
output = SVG ( " g " )
for s in self . d :
if isinstance ( s , SVG ) :
output . append ( s )
elif isinstance ( s , Fig ) :
strans = s . trans
if isinstance ( strans , basestring ) :
strans = totrans ( strans )
if trans is None :
subtrans = strans
elif strans is None :
subtrans = trans
else :
subtrans = lambda x , y : trans ( * strans ( x , y ) )
output . sub + = s . SVG ( subtrans ) . sub
elif s is None :
pass
else :
output . append ( s . SVG ( trans ) )
return output
class Plot :
""" Acts like Fig, but draws a coordinate axis. You also need to supply plot ranges.
Plot ( xmin , xmax , ymin , ymax , obj , obj , obj . . . , keyword options . . . )
xmin , xmax required minimum and maximum x values ( in the objs ' coordinates)
ymin , ymax required minimum and maximum y values ( in the objs ' coordinates)
obj optional list drawing primitives
keyword options keyword list options defined below
The following are keyword options , with their default values :
trans None transformation function
x , y 5 , 5 upper - left corner of the Plot in SVG coordinates
width , height 90 , 90 width and height of the Plot in SVG coordinates
flipx , flipy False , True flip the sign of the coordinate axis
minusInfinity - 1000 if an axis is logarithmic and an object is plotted at 0 or
a negative value , - 1000 will be used as a stand - in for NaN
atx , aty 0 , 0 the place where the coordinate axes cross
xticks - 10 request ticks according to the standard tick specification
( see help ( Ticks ) )
xminiticks True request miniticks according to the standard minitick
specification
xlabels True request tick labels according to the standard tick label
specification
xlogbase None if a number , the axis and transformation are logarithmic
with ticks at the given base ( 10 being the most common )
( same for y )
arrows None if a new identifier , create arrow markers and draw them
at the ends of the coordinate axes
text_attr { } a dictionary of attributes for label text
axis_attr { } a dictionary of attributes for the axis lines
"""
def __repr__ ( self ) :
if self . trans is None :
return " <Plot ( %d items)> " % len ( self . d )
else :
return " <Plot ( %d items) %s > " % ( len ( self . d ) , self . trans . func_name )
def __init__ ( self , xmin , xmax , ymin , ymax , * d , * * kwds ) :
self . xmin , self . xmax , self . ymin , self . ymax = xmin , xmax , ymin , ymax
self . d = list ( d )
defaults = { " trans " : None ,
" x " : 5 , " y " : 5 , " width " : 90 , " height " : 90 ,
" flipx " : False , " flipy " : True ,
" minusInfinity " : - 1000 ,
" atx " : 0 , " xticks " : - 10 , " xminiticks " : True , " xlabels " : True , " xlogbase " : None ,
" aty " : 0 , " yticks " : - 10 , " yminiticks " : True , " ylabels " : True , " ylogbase " : None ,
" arrows " : None ,
" text_attr " : { } , " axis_attr " : { } ,
}
defaults . update ( kwds )
kwds = defaults
self . trans = kwds [ " trans " ] ; del kwds [ " trans " ]
self . x = kwds [ " x " ] ; del kwds [ " x " ]
self . y = kwds [ " y " ] ; del kwds [ " y " ]
self . width = kwds [ " width " ] ; del kwds [ " width " ]
self . height = kwds [ " height " ] ; del kwds [ " height " ]
self . flipx = kwds [ " flipx " ] ; del kwds [ " flipx " ]
self . flipy = kwds [ " flipy " ] ; del kwds [ " flipy " ]
self . minusInfinity = kwds [ " minusInfinity " ] ; del kwds [ " minusInfinity " ]
self . atx = kwds [ " atx " ] ; del kwds [ " atx " ]
self . xticks = kwds [ " xticks " ] ; del kwds [ " xticks " ]
self . xminiticks = kwds [ " xminiticks " ] ; del kwds [ " xminiticks " ]
self . xlabels = kwds [ " xlabels " ] ; del kwds [ " xlabels " ]
self . xlogbase = kwds [ " xlogbase " ] ; del kwds [ " xlogbase " ]
self . aty = kwds [ " aty " ] ; del kwds [ " aty " ]
self . yticks = kwds [ " yticks " ] ; del kwds [ " yticks " ]
self . yminiticks = kwds [ " yminiticks " ] ; del kwds [ " yminiticks " ]
self . ylabels = kwds [ " ylabels " ] ; del kwds [ " ylabels " ]
self . ylogbase = kwds [ " ylogbase " ] ; del kwds [ " ylogbase " ]
self . arrows = kwds [ " arrows " ] ; del kwds [ " arrows " ]
self . text_attr = kwds [ " text_attr " ] ; del kwds [ " text_attr " ]
self . axis_attr = kwds [ " axis_attr " ] ; del kwds [ " axis_attr " ]
if len ( kwds ) != 0 :
raise TypeError , " Plot() got unexpected keyword arguments %s " % kwds . keys ( )
def SVG ( self , trans = None ) :
""" Apply the transformation " trans " and return an SVG object. """
if trans is None :
trans = self . trans
if isinstance ( trans , basestring ) :
trans = totrans ( trans )
self . last_window = window ( self . xmin , self . xmax , self . ymin , self . ymax ,
x = self . x , y = self . y , width = self . width , height = self . height ,
xlogbase = self . xlogbase , ylogbase = self . ylogbase ,
minusInfinity = self . minusInfinity , flipx = self . flipx , flipy = self . flipy )
d = ( [ Axes ( self . xmin , self . xmax , self . ymin , self . ymax , self . atx , self . aty ,
self . xticks , self . xminiticks , self . xlabels , self . xlogbase ,
self . yticks , self . yminiticks , self . ylabels , self . ylogbase ,
self . arrows , self . text_attr , * * self . axis_attr ) ]
+ self . d )
return Fig ( Fig ( * d , * * { " trans " : trans } ) ) . SVG ( self . last_window )
class Frame :
text_defaults = { " stroke " : " none " , " fill " : " black " , " font-size " : 5 , }
axis_defaults = { }
tick_length = 1.5
minitick_length = 0.75
text_xaxis_offset = 1.
text_yaxis_offset = 2.
text_xtitle_offset = 6.
text_ytitle_offset = 12.
def __repr__ ( self ) :
return " <Frame ( %d items)> " % len ( self . d )
def __init__ ( self , xmin , xmax , ymin , ymax , * d , * * kwds ) :
""" Acts like Fig, but draws a coordinate frame around the data. You also need to supply plot ranges.
Frame ( xmin , xmax , ymin , ymax , obj , obj , obj . . . , keyword options . . . )
xmin , xmax required minimum and maximum x values ( in the objs ' coordinates)
ymin , ymax required minimum and maximum y values ( in the objs ' coordinates)
obj optional list drawing primitives
keyword options keyword list options defined below
The following are keyword options , with their default values :
x , y 20 , 5 upper - left corner of the Frame in SVG coordinates
width , height 75 , 80 width and height of the Frame in SVG coordinates
flipx , flipy False , True flip the sign of the coordinate axis
minusInfinity - 1000 if an axis is logarithmic and an object is plotted at 0 or
a negative value , - 1000 will be used as a stand - in for NaN
xtitle None if a string , label the x axis
xticks - 10 request ticks according to the standard tick specification
( see help ( Ticks ) )
xminiticks True request miniticks according to the standard minitick
specification
xlabels True request tick labels according to the standard tick label
specification
xlogbase None if a number , the axis and transformation are logarithmic
with ticks at the given base ( 10 being the most common )
( same for y )
text_attr { } a dictionary of attributes for label text
axis_attr { } a dictionary of attributes for the axis lines
"""
self . xmin , self . xmax , self . ymin , self . ymax = xmin , xmax , ymin , ymax
self . d = list ( d )
defaults = { " x " : 20 , " y " : 5 , " width " : 75 , " height " : 80 ,
" flipx " : False , " flipy " : True , " minusInfinity " : - 1000 ,
" xtitle " : None , " xticks " : - 10 , " xminiticks " : True , " xlabels " : True ,
" x2labels " : None , " xlogbase " : None ,
" ytitle " : None , " yticks " : - 10 , " yminiticks " : True , " ylabels " : True ,
" y2labels " : None , " ylogbase " : None ,
" text_attr " : { } , " axis_attr " : { } ,
}
defaults . update ( kwds )
kwds = defaults
self . x = kwds [ " x " ] ; del kwds [ " x " ]
self . y = kwds [ " y " ] ; del kwds [ " y " ]
self . width = kwds [ " width " ] ; del kwds [ " width " ]
self . height = kwds [ " height " ] ; del kwds [ " height " ]
self . flipx = kwds [ " flipx " ] ; del kwds [ " flipx " ]
self . flipy = kwds [ " flipy " ] ; del kwds [ " flipy " ]
self . minusInfinity = kwds [ " minusInfinity " ] ; del kwds [ " minusInfinity " ]
self . xtitle = kwds [ " xtitle " ] ; del kwds [ " xtitle " ]
self . xticks = kwds [ " xticks " ] ; del kwds [ " xticks " ]
self . xminiticks = kwds [ " xminiticks " ] ; del kwds [ " xminiticks " ]
self . xlabels = kwds [ " xlabels " ] ; del kwds [ " xlabels " ]
self . x2labels = kwds [ " x2labels " ] ; del kwds [ " x2labels " ]
self . xlogbase = kwds [ " xlogbase " ] ; del kwds [ " xlogbase " ]
self . ytitle = kwds [ " ytitle " ] ; del kwds [ " ytitle " ]
self . yticks = kwds [ " yticks " ] ; del kwds [ " yticks " ]
self . yminiticks = kwds [ " yminiticks " ] ; del kwds [ " yminiticks " ]
self . ylabels = kwds [ " ylabels " ] ; del kwds [ " ylabels " ]
self . y2labels = kwds [ " y2labels " ] ; del kwds [ " y2labels " ]
self . ylogbase = kwds [ " ylogbase " ] ; del kwds [ " ylogbase " ]
self . text_attr = dict ( self . text_defaults )
self . text_attr . update ( kwds [ " text_attr " ] ) ; del kwds [ " text_attr " ]
self . axis_attr = dict ( self . axis_defaults )
self . axis_attr . update ( kwds [ " axis_attr " ] ) ; del kwds [ " axis_attr " ]
if len ( kwds ) != 0 :
raise TypeError , " Frame() got unexpected keyword arguments %s " % kwds . keys ( )
def SVG ( self ) :
""" Apply the window transformation and return an SVG object. """
self . last_window = window ( self . xmin , self . xmax , self . ymin , self . ymax ,
x = self . x , y = self . y , width = self . width , height = self . height ,
xlogbase = self . xlogbase , ylogbase = self . ylogbase ,
minusInfinity = self . minusInfinity , flipx = self . flipx , flipy = self . flipy )
left = YAxis ( self . ymin , self . ymax , self . xmin , self . yticks , self . yminiticks , self . ylabels , self . ylogbase ,
None , None , None , self . text_attr , * * self . axis_attr )
right = YAxis ( self . ymin , self . ymax , self . xmax , self . yticks , self . yminiticks , self . y2labels , self . ylogbase ,
None , None , None , self . text_attr , * * self . axis_attr )
bottom = XAxis ( self . xmin , self . xmax , self . ymin , self . xticks , self . xminiticks , self . xlabels , self . xlogbase ,
None , None , None , self . text_attr , * * self . axis_attr )
top = XAxis ( self . xmin , self . xmax , self . ymax , self . xticks , self . xminiticks , self . x2labels , self . xlogbase ,
None , None , None , self . text_attr , * * self . axis_attr )
left . tick_start = - self . tick_length
left . tick_end = 0
left . minitick_start = - self . minitick_length
left . minitick_end = 0.
left . text_start = self . text_yaxis_offset
right . tick_start = 0.
right . tick_end = self . tick_length
right . minitick_start = 0.
right . minitick_end = self . minitick_length
right . text_start = - self . text_yaxis_offset
right . text_attr [ " text-anchor " ] = " start "
bottom . tick_start = 0.
bottom . tick_end = self . tick_length
bottom . minitick_start = 0.
bottom . minitick_end = self . minitick_length
bottom . text_start = - self . text_xaxis_offset
top . tick_start = - self . tick_length
top . tick_end = 0.
top . minitick_start = - self . minitick_length
top . minitick_end = 0.
top . text_start = self . text_xaxis_offset
top . text_attr [ " dominant-baseline " ] = " text-after-edge "
output = Fig ( * self . d ) . SVG ( self . last_window )
output . prepend ( left . SVG ( self . last_window ) )
output . prepend ( bottom . SVG ( self . last_window ) )
output . prepend ( right . SVG ( self . last_window ) )
output . prepend ( top . SVG ( self . last_window ) )
if self . xtitle is not None :
output . append ( SVG ( " text " , self . xtitle , transform = " translate( %g , %g ) " % ( ( self . x + self . width / 2. ) , ( self . y + self . height + self . text_xtitle_offset ) ) , dominant_baseline = " text-before-edge " , * * self . text_attr ) )
if self . ytitle is not None :
output . append ( SVG ( " text " , self . ytitle , transform = " translate( %g , %g ) rotate(-90) " % ( ( self . x - self . text_ytitle_offset ) , ( self . y + self . height / 2. ) ) , * * self . text_attr ) )
return output
######################################################################
def pathtoPath ( svg ) :
""" Converts SVG( " path " , d= " ... " ) into Path(d=[...]). """
if not isinstance ( svg , SVG ) or svg . t != " path " :
raise TypeError , " Only SVG <path /> objects can be converted into Paths "
attr = dict ( svg . attr )
d = attr [ " d " ]
del attr [ " d " ]
for key in attr . keys ( ) :
if not isinstance ( key , str ) :
value = attr [ key ]
del attr [ key ]
attr [ str ( key ) ] = value
return Path ( d , * * attr )
class Path :
""" Path represents an SVG path, an arbitrary set of curves and
straight segments . Unlike SVG ( " path " , d = " ... " ) , Path stores
coordinates as a list of numbers , rather than a string , so that it is
transformable in a Fig .
Path ( d , attribute = value )
d required path data
attribute = value pairs keyword list SVG attributes
See http : / / www . w3 . org / TR / SVG / paths . html for specification of paths
from text .
Internally , Path data is a list of tuples with these definitions :
* ( " Z/z " , ) : close the current path
* ( " H/h " , x ) or ( " V/v " , y ) : a horizontal or vertical line
segment to x or y
* ( " M/m/L/l/T/t " , x , y , global ) : moveto , lineto , or smooth
quadratic curveto point ( x , y ) . If global = True , ( x , y ) should
not be transformed .
* ( " S/sQ/q " , cx , cy , cglobal , x , y , global ) : polybezier or
smooth quadratic curveto point ( x , y ) using ( cx , cy ) as a
control point . If cglobal or global = True , ( cx , cy ) or ( x , y )
should not be transformed .
* ( " C/c " , c1x , c1y , c1global , c2x , c2y , c2global , x , y , global ) :
cubic curveto point ( x , y ) using ( c1x , c1y ) and ( c2x , c2y ) as
control points . If c1global , c2global , or global = True , ( c1x , c1y ) ,
( c2x , c2y ) , or ( x , y ) should not be transformed .
* ( " A/a " , rx , ry , rglobal , x - axis - rotation , angle , large - arc - flag ,
sweep - flag , x , y , global ) : arcto point ( x , y ) using the
aforementioned parameters .
* ( " ,/. " , rx , ry , rglobal , angle , x , y , global ) : an ellipse at
point ( x , y ) with radii ( rx , ry ) . If angle is 0 , the whole
ellipse is drawn ; otherwise , a partial ellipse is drawn .
"""
defaults = { }
def __repr__ ( self ) :
return " <Path ( %d nodes) %s > " % ( len ( self . d ) , self . attr )
def __init__ ( self , d = [ ] , * * attr ) :
if isinstance ( d , basestring ) :
self . d = self . parse ( d )
else :
self . d = list ( d )
self . attr = dict ( self . defaults )
self . attr . update ( attr )
def parse_whitespace ( self , index , pathdata ) :
""" Part of Path ' s text-command parsing algorithm; used internally. """
while index < len ( pathdata ) and pathdata [ index ] in ( " " , " \t " , " \r " , " \n " , " , " ) :
index + = 1
return index , pathdata
def parse_command ( self , index , pathdata ) :
""" Part of Path ' s text-command parsing algorithm; used internally. """
index , pathdata = self . parse_whitespace ( index , pathdata )
if index > = len ( pathdata ) :
return None , index , pathdata
command = pathdata [ index ]
if " A " < = command < = " Z " or " a " < = command < = " z " :
index + = 1
return command , index , pathdata
else :
return None , index , pathdata
def parse_number ( self , index , pathdata ) :
""" Part of Path ' s text-command parsing algorithm; used internally. """
index , pathdata = self . parse_whitespace ( index , pathdata )
if index > = len ( pathdata ) :
return None , index , pathdata
first_digit = pathdata [ index ]
if " 0 " < = first_digit < = " 9 " or first_digit in ( " - " , " + " , " . " ) :
start = index
while index < len ( pathdata ) and ( " 0 " < = pathdata [ index ] < = " 9 " or pathdata [ index ] in ( " - " , " + " , " . " , " e " , " E " ) ) :
index + = 1
end = index
index = end
return float ( pathdata [ start : end ] ) , index , pathdata
else :
return None , index , pathdata
def parse_boolean ( self , index , pathdata ) :
""" Part of Path ' s text-command parsing algorithm; used internally. """
index , pathdata = self . parse_whitespace ( index , pathdata )
if index > = len ( pathdata ) :
return None , index , pathdata
first_digit = pathdata [ index ]
if first_digit in ( " 0 " , " 1 " ) :
index + = 1
return int ( first_digit ) , index , pathdata
else :
return None , index , pathdata
def parse ( self , pathdata ) :
""" Parses text-commands, converting them into a list of tuples.
Called by the constructor . """
output = [ ]
index = 0
while True :
command , index , pathdata = self . parse_command ( index , pathdata )
index , pathdata = self . parse_whitespace ( index , pathdata )
if command is None and index == len ( pathdata ) :
break # this is the normal way out of the loop
if command in ( " Z " , " z " ) :
output . append ( ( command , ) )
######################
elif command in ( " H " , " h " , " V " , " v " ) :
errstring = " Path command \" %s \" requires a number at index %d " % ( command , index )
num1 , index , pathdata = self . parse_number ( index , pathdata )
if num1 is None :
raise ValueError , errstring
while num1 is not None :
output . append ( ( command , num1 ) )
num1 , index , pathdata = self . parse_number ( index , pathdata )
######################
elif command in ( " M " , " m " , " L " , " l " , " T " , " t " ) :
errstring = " Path command \" %s \" requires an x,y pair at index %d " % ( command , index )
num1 , index , pathdata = self . parse_number ( index , pathdata )
num2 , index , pathdata = self . parse_number ( index , pathdata )
if num1 is None :
raise ValueError , errstring
while num1 is not None :
if num2 is None :
raise ValueError , errstring
output . append ( ( command , num1 , num2 , False ) )
num1 , index , pathdata = self . parse_number ( index , pathdata )
num2 , index , pathdata = self . parse_number ( index , pathdata )
######################
elif command in ( " S " , " s " , " Q " , " q " ) :
errstring = " Path command \" %s \" requires a cx,cy,x,y quadruplet at index %d " % ( command , index )
num1 , index , pathdata = self . parse_number ( index , pathdata )
num2 , index , pathdata = self . parse_number ( index , pathdata )
num3 , index , pathdata = self . parse_number ( index , pathdata )
num4 , index , pathdata = self . parse_number ( index , pathdata )
if num1 is None :
raise ValueError , errstring
while num1 is not None :
if num2 is None or num3 is None or num4 is None :
raise ValueError , errstring
output . append ( ( command , num1 , num2 , False , num3 , num4 , False ) )
num1 , index , pathdata = self . parse_number ( index , pathdata )
num2 , index , pathdata = self . parse_number ( index , pathdata )
num3 , index , pathdata = self . parse_number ( index , pathdata )
num4 , index , pathdata = self . parse_number ( index , pathdata )
######################
elif command in ( " C " , " c " ) :
errstring = " Path command \" %s \" requires a c1x,c1y,c2x,c2y,x,y sextuplet at index %d " % ( command , index )
num1 , index , pathdata = self . parse_number ( index , pathdata )
num2 , index , pathdata = self . parse_number ( index , pathdata )
num3 , index , pathdata = self . parse_number ( index , pathdata )
num4 , index , pathdata = self . parse_number ( index , pathdata )
num5 , index , pathdata = self . parse_number ( index , pathdata )
num6 , index , pathdata = self . parse_number ( index , pathdata )
if num1 is None :
raise ValueError , errstring
while num1 is not None :
if num2 is None or num3 is None or num4 is None or num5 is None or num6 is None :
raise ValueError , errstring
output . append ( ( command , num1 , num2 , False , num3 , num4 , False , num5 , num6 , False ) )
num1 , index , pathdata = self . parse_number ( index , pathdata )
num2 , index , pathdata = self . parse_number ( index , pathdata )
num3 , index , pathdata = self . parse_number ( index , pathdata )
num4 , index , pathdata = self . parse_number ( index , pathdata )
num5 , index , pathdata = self . parse_number ( index , pathdata )
num6 , index , pathdata = self . parse_number ( index , pathdata )
######################
elif command in ( " A " , " a " ) :
errstring = " Path command \" %s \" requires a rx,ry,angle,large-arc-flag,sweep-flag,x,y septuplet at index %d " % ( command , index )
num1 , index , pathdata = self . parse_number ( index , pathdata )
num2 , index , pathdata = self . parse_number ( index , pathdata )
num3 , index , pathdata = self . parse_number ( index , pathdata )
num4 , index , pathdata = self . parse_boolean ( index , pathdata )
num5 , index , pathdata = self . parse_boolean ( index , pathdata )
num6 , index , pathdata = self . parse_number ( index , pathdata )
num7 , index , pathdata = self . parse_number ( index , pathdata )
if num1 is None :
raise ValueError , errstring
while num1 is not None :
if num2 is None or num3 is None or num4 is None or num5 is None or num6 is None or num7 is None :
raise ValueError , errstring
output . append ( ( command , num1 , num2 , False , num3 , num4 , num5 , num6 , num7 , False ) )
num1 , index , pathdata = self . parse_number ( index , pathdata )
num2 , index , pathdata = self . parse_number ( index , pathdata )
num3 , index , pathdata = self . parse_number ( index , pathdata )
num4 , index , pathdata = self . parse_boolean ( index , pathdata )
num5 , index , pathdata = self . parse_boolean ( index , pathdata )
num6 , index , pathdata = self . parse_number ( index , pathdata )
num7 , index , pathdata = self . parse_number ( index , pathdata )
return output
def SVG ( self , trans = None ) :
""" Apply the transformation " trans " and return an SVG object. """
if isinstance ( trans , basestring ) :
trans = totrans ( trans )
x , y , X , Y = None , None , None , None
output = [ ]
for datum in self . d :
if not isinstance ( datum , ( tuple , list ) ) :
raise TypeError , " pathdata elements must be tuples/lists "
command = datum [ 0 ]
######################
if command in ( " Z " , " z " ) :
x , y , X , Y = None , None , None , None
output . append ( " Z " )
######################
elif command in ( " H " , " h " , " V " , " v " ) :
command , num1 = datum
if command == " H " or ( command == " h " and x is None ) :
x = num1
elif command == " h " :
x + = num1
elif command == " V " or ( command == " v " and y is None ) :
y = num1
elif command == " v " :
y + = num1
if trans is None :
X , Y = x , y
else :
X , Y = trans ( x , y )
output . append ( " L %g %g " % ( X , Y ) )
######################
elif command in ( " M " , " m " , " L " , " l " , " T " , " t " ) :
command , num1 , num2 , isglobal12 = datum
if trans is None or isglobal12 :
if command . isupper ( ) or X is None or Y is None :
X , Y = num1 , num2
else :
X + = num1
Y + = num2
x , y = X , Y
else :
if command . isupper ( ) or x is None or y is None :
x , y = num1 , num2
else :
x + = num1
y + = num2
X , Y = trans ( x , y )
COMMAND = command . capitalize ( )
output . append ( " %s %g %g " % ( COMMAND , X , Y ) )
######################
elif command in ( " S " , " s " , " Q " , " q " ) :
command , num1 , num2 , isglobal12 , num3 , num4 , isglobal34 = datum
if trans is None or isglobal12 :
if command . isupper ( ) or X is None or Y is None :
CX , CY = num1 , num2
else :
CX = X + num1
CY = Y + num2
else :
if command . isupper ( ) or x is None or y is None :
cx , cy = num1 , num2
else :
cx = x + num1
cy = y + num2
CX , CY = trans ( cx , cy )
if trans is None or isglobal34 :
if command . isupper ( ) or X is None or Y is None :
X , Y = num3 , num4
else :
X + = num3
Y + = num4
x , y = X , Y
else :
if command . isupper ( ) or x is None or y is None :
x , y = num3 , num4
else :
x + = num3
y + = num4
X , Y = trans ( x , y )
COMMAND = command . capitalize ( )
output . append ( " %s %g %g %g %g " % ( COMMAND , CX , CY , X , Y ) )
######################
elif command in ( " C " , " c " ) :
command , num1 , num2 , isglobal12 , num3 , num4 , isglobal34 , num5 , num6 , isglobal56 = datum
if trans is None or isglobal12 :
if command . isupper ( ) or X is None or Y is None :
C1X , C1Y = num1 , num2
else :
C1X = X + num1
C1Y = Y + num2
else :
if command . isupper ( ) or x is None or y is None :
c1x , c1y = num1 , num2
else :
c1x = x + num1
c1y = y + num2
C1X , C1Y = trans ( c1x , c1y )
if trans is None or isglobal34 :
if command . isupper ( ) or X is None or Y is None :
C2X , C2Y = num3 , num4
else :
C2X = X + num3
C2Y = Y + num4
else :
if command . isupper ( ) or x is None or y is None :
c2x , c2y = num3 , num4
else :
c2x = x + num3
c2y = y + num4
C2X , C2Y = trans ( c2x , c2y )
if trans is None or isglobal56 :
if command . isupper ( ) or X is None or Y is None :
X , Y = num5 , num6
else :
X + = num5
Y + = num6
x , y = X , Y
else :
if command . isupper ( ) or x is None or y is None :
x , y = num5 , num6
else :
x + = num5
y + = num6
X , Y = trans ( x , y )
COMMAND = command . capitalize ( )
output . append ( " %s %g %g %g %g %g %g " % ( COMMAND , C1X , C1Y , C2X , C2Y , X , Y ) )
######################
elif command in ( " A " , " a " ) :
command , num1 , num2 , isglobal12 , angle , large_arc_flag , sweep_flag , num3 , num4 , isglobal34 = datum
oldx , oldy = x , y
OLDX , OLDY = X , Y
if trans is None or isglobal34 :
if command . isupper ( ) or X is None or Y is None :
X , Y = num3 , num4
else :
X + = num3
Y + = num4
x , y = X , Y
else :
if command . isupper ( ) or x is None or y is None :
x , y = num3 , num4
else :
x + = num3
y + = num4
X , Y = trans ( x , y )
if x is not None and y is not None :
centerx , centery = ( x + oldx ) / 2. , ( y + oldy ) / 2.
CENTERX , CENTERY = ( X + OLDX ) / 2. , ( Y + OLDY ) / 2.
if trans is None or isglobal12 :
RX = CENTERX + num1
RY = CENTERY + num2
else :
rx = centerx + num1
ry = centery + num2
RX , RY = trans ( rx , ry )
COMMAND = command . capitalize ( )
output . append ( " %s %g %g %g %d %d %g %g " % ( COMMAND , RX - CENTERX , RY - CENTERY , angle , large_arc_flag , sweep_flag , X , Y ) )
elif command in ( " , " , " . " ) :
command , num1 , num2 , isglobal12 , angle , num3 , num4 , isglobal34 = datum
if trans is None or isglobal34 :
if command == " . " or X is None or Y is None :
X , Y = num3 , num4
else :
X + = num3
Y + = num4
x , y = None , None
else :
if command == " . " or x is None or y is None :
x , y = num3 , num4
else :
x + = num3
y + = num4
X , Y = trans ( x , y )
if trans is None or isglobal12 :
RX = X + num1
RY = Y + num2
else :
rx = x + num1
ry = y + num2
RX , RY = trans ( rx , ry )
RX , RY = RX - X , RY - Y
X1 , Y1 = X + RX * math . cos ( angle * math . pi / 180. ) , Y + RX * math . sin ( angle * math . pi / 180. )
X2 , Y2 = X + RY * math . sin ( angle * math . pi / 180. ) , Y - RY * math . cos ( angle * math . pi / 180. )
X3 , Y3 = X - RX * math . cos ( angle * math . pi / 180. ) , Y - RX * math . sin ( angle * math . pi / 180. )
X4 , Y4 = X - RY * math . sin ( angle * math . pi / 180. ) , Y + RY * math . cos ( angle * math . pi / 180. )
output . append ( " M %g %g A %g %g %g 0 0 %g %g A %g %g %g 0 0 %g %g A %g %g %g 0 0 %g %g A %g %g %g 0 0 %g %g " % (
X1 , Y1 , RX , RY , angle , X2 , Y2 , RX , RY , angle , X3 , Y3 , RX , RY , angle , X4 , Y4 , RX , RY , angle , X1 , Y1 ) )
return SVG ( " path " , d = " " . join ( output ) , * * self . attr )
######################################################################
def funcRtoC ( expr , var = " t " , globals = None , locals = None ) :
""" Converts a complex " z(t) " string to a function acceptable for Curve.
expr required string in the form " z(t) "
var default = " t " name of the independent variable
globals default = None dict of global variables used in the expression ;
you may want to use Python ' s builtin globals()
locals default = None dict of local variables
"""
if locals is None :
locals = { } # python 2.3's eval() won't accept None
g = cmath . __dict__
if globals is not None :
g . update ( globals )
output = eval ( " lambda %s : ( %s ) " % ( var , expr ) , g , locals )
split = lambda z : ( z . real , z . imag )
output2 = lambda t : split ( output ( t ) )
set_func_name ( output2 , " %s -> %s " % ( var , expr ) )
return output2
def funcRtoR2 ( expr , var = " t " , globals = None , locals = None ) :
""" Converts a " f(t), g(t) " string to a function acceptable for Curve.
expr required string in the form " f(t), g(t) "
var default = " t " name of the independent variable
globals default = None dict of global variables used in the expression ;
you may want to use Python ' s builtin globals()
locals default = None dict of local variables
"""
if locals is None :
locals = { } # python 2.3's eval() won't accept None
g = math . __dict__
if globals is not None :
g . update ( globals )
output = eval ( " lambda %s : ( %s ) " % ( var , expr ) , g , locals )
set_func_name ( output , " %s -> %s " % ( var , expr ) )
return output
def funcRtoR ( expr , var = " x " , globals = None , locals = None ) :
""" Converts a " f(x) " string to a function acceptable for Curve.
expr required string in the form " f(x) "
var default = " x " name of the independent variable
globals default = None dict of global variables used in the expression ;
you may want to use Python ' s builtin globals()
locals default = None dict of local variables
"""
if locals is None :
locals = { } # python 2.3's eval() won't accept None
g = math . __dict__
if globals is not None :
g . update ( globals )
output = eval ( " lambda %s : ( %s , %s ) " % ( var , var , expr ) , g , locals )
set_func_name ( output , " %s -> %s " % ( var , expr ) )
return output
class Curve :
""" Draws a parametric function as a path.
Curve ( f , low , high , loop , attribute = value )
f required a Python callable or string in
the form " f(t), g(t) "
low , high required left and right endpoints
loop default = False if True , connect the endpoints
attribute = value pairs keyword list SVG attributes
"""
defaults = { }
random_sampling = True
recursion_limit = 15
linearity_limit = 0.05
discontinuity_limit = 5.
def __repr__ ( self ) :
return " <Curve %s [ %s , %s ] %s > " % ( self . f , self . low , self . high , self . attr )
def __init__ ( self , f , low , high , loop = False , * * attr ) :
self . f = f
self . low = low
self . high = high
self . loop = loop
self . attr = dict ( self . defaults )
self . attr . update ( attr )
### nested class Sample
class Sample :
def __repr__ ( self ) :
t , x , y , X , Y = self . t , self . x , self . y , self . X , self . Y
if t is not None :
t = " %g " % t
if x is not None :
x = " %g " % x
if y is not None :
y = " %g " % y
if X is not None :
X = " %g " % X
if Y is not None :
Y = " %g " % Y
return " <Curve.Sample t= %s x= %s y= %s X= %s Y= %s > " % ( t , x , y , X , Y )
def __init__ ( self , t ) :
self . t = t
def link ( self , left , right ) :
self . left , self . right = left , right
def evaluate ( self , f , trans ) :
self . x , self . y = f ( self . t )
if trans is None :
self . X , self . Y = self . x , self . y
else :
self . X , self . Y = trans ( self . x , self . y )
### end Sample
### nested class Samples
class Samples :
def __repr__ ( self ) :
return " <Curve.Samples ( %d samples)> " % len ( self )
def __init__ ( self , left , right ) :
self . left , self . right = left , right
def __len__ ( self ) :
count = 0
current = self . left
while current is not None :
count + = 1
current = current . right
return count
def __iter__ ( self ) :
self . current = self . left
return self
def next ( self ) :
current = self . current
if current is None :
raise StopIteration
self . current = self . current . right
return current
### end nested class
def sample ( self , trans = None ) :
""" Adaptive-sampling algorithm that chooses the best sample points
for a parametric curve between two endpoints and detects
discontinuities . Called by SVG ( ) . """
oldrecursionlimit = sys . getrecursionlimit ( )
sys . setrecursionlimit ( self . recursion_limit + 100 )
try :
# the best way to keep all the information while sampling is to make a linked list
if not ( self . low < self . high ) :
raise ValueError , " low must be less than high "
low , high = self . Sample ( float ( self . low ) ) , self . Sample ( float ( self . high ) )
low . link ( None , high )
high . link ( low , None )
low . evaluate ( self . f , trans )
high . evaluate ( self . f , trans )
# adaptive sampling between the low and high points
self . subsample ( low , high , 0 , trans )
# Prune excess points where the curve is nearly linear
left = low
while left . right is not None :
# increment mid and right
mid = left . right
right = mid . right
if ( right is not None and
left . X is not None and left . Y is not None and
mid . X is not None and mid . Y is not None and
right . X is not None and right . Y is not None ) :
numer = left . X * ( right . Y - mid . Y ) + mid . X * ( left . Y - right . Y ) + right . X * ( mid . Y - left . Y )
denom = math . sqrt ( ( left . X - right . X ) * * 2 + ( left . Y - right . Y ) * * 2 )
if denom != 0. and abs ( numer / denom ) < self . linearity_limit :
# drop mid (the garbage collector will get it)
left . right = right
right . left = left
else :
# increment left
left = left . right
else :
left = left . right
self . last_samples = self . Samples ( low , high )
finally :
sys . setrecursionlimit ( oldrecursionlimit )
def subsample ( self , left , right , depth , trans = None ) :
""" Part of the adaptive-sampling algorithm that chooses the best
sample points . Called by sample ( ) . """
if self . random_sampling :
mid = self . Sample ( left . t + random . uniform ( 0.3 , 0.7 ) * ( right . t - left . t ) )
else :
mid = self . Sample ( left . t + 0.5 * ( right . t - left . t ) )
left . right = mid
right . left = mid
mid . link ( left , right )
mid . evaluate ( self . f , trans )
# calculate the distance of closest approach of mid to the line between left and right
numer = left . X * ( right . Y - mid . Y ) + mid . X * ( left . Y - right . Y ) + right . X * ( mid . Y - left . Y )
denom = math . sqrt ( ( left . X - right . X ) * * 2 + ( left . Y - right . Y ) * * 2 )
# if we haven't sampled enough or left fails to be close enough to right, or mid fails to be linear enough...
if ( depth < 3 or
( denom == 0 and left . t != right . t ) or
denom > self . discontinuity_limit or
( denom != 0. and abs ( numer / denom ) > self . linearity_limit ) ) :
# and we haven't sampled too many points
if depth < self . recursion_limit :
self . subsample ( left , mid , depth + 1 , trans )
self . subsample ( mid , right , depth + 1 , trans )
else :
# We've sampled many points and yet it's still not a small linear gap.
# Break the line: it's a discontinuity
mid . y = mid . Y = None
def SVG ( self , trans = None ) :
""" Apply the transformation " trans " and return an SVG object. """
return self . Path ( trans ) . SVG ( )
def Path ( self , trans = None , local = False ) :
""" Apply the transformation " trans " and return a Path object in
global coordinates . If local = True , return a Path in local coordinates
( which must be transformed again ) . """
if isinstance ( trans , basestring ) :
trans = totrans ( trans )
if isinstance ( self . f , basestring ) :
self . f = funcRtoR2 ( self . f )
self . sample ( trans )
output = [ ]
for s in self . last_samples :
if s . X is not None and s . Y is not None :
if s . left is None or s . left . Y is None :
command = " M "
else :
command = " L "
if local :
output . append ( ( command , s . x , s . y , False ) )
else :
output . append ( ( command , s . X , s . Y , True ) )
if self . loop :
output . append ( ( " Z " , ) )
return Path ( output , * * self . attr )
######################################################################
class Poly :
""" Draws a curve specified by a sequence of points. The curve may be
piecewise linear , like a polygon , or a Bezier curve .
Poly ( d , mode , loop , attribute = value )
d required list of tuples representing points
and possibly control points
mode default = " L " " lines " , " bezier " , " velocity " ,
" foreback " , " smooth " , or an abbreviation
loop default = False if True , connect the first and last
point , closing the loop
attribute = value pairs keyword list SVG attributes
The format of the tuples in d depends on the mode .
" lines " / " L " d = [ ( x , y ) , ( x , y ) , . . . ]
piecewise - linear segments joining the ( x , y ) points
" bezier " / " B " d = [ ( x , y , c1x , c1y , c2x , c2y ) , . . . ]
Bezier curve with two control points ( control points
preceed ( x , y ) , as in SVG paths ) . If ( c1x , c1y ) and
( c2x , c2y ) both equal ( x , y ) , you get a linear
interpolation ( " lines " )
" velocity " / " V " d = [ ( x , y , vx , vy ) , . . . ]
curve that passes through ( x , y ) with velocity ( vx , vy )
( one unit of arclength per unit time ) ; in other words ,
( vx , vy ) is the tangent vector at ( x , y ) . If ( vx , vy ) is
( 0 , 0 ) , you get a linear interpolation ( " lines " ) .
" foreback " / " F " d = [ ( x , y , bx , by , fx , fy ) , . . . ]
like " velocity " except that there is a left derivative
( bx , by ) and a right derivative ( fx , fy ) . If ( bx , by )
equals ( fx , fy ) ( with no minus sign ) , you get a
" velocity " curve
" smooth " / " S " d = [ ( x , y ) , ( x , y ) , . . . ]
a " velocity " interpolation with ( vx , vy ) [ i ] equal to
( ( x , y ) [ i + 1 ] - ( x , y ) [ i - 1 ] ) / 2 : the minimal derivative
"""
defaults = { }
def __repr__ ( self ) :
return " <Poly ( %d nodes) mode= %s loop= %s %s > " % (
len ( self . d ) , self . mode , repr ( self . loop ) , self . attr )
def __init__ ( self , d = [ ] , mode = " L " , loop = False , * * attr ) :
self . d = list ( d )
self . mode = mode
self . loop = loop
self . attr = dict ( self . defaults )
self . attr . update ( attr )
def SVG ( self , trans = None ) :
""" Apply the transformation " trans " and return an SVG object. """
return self . Path ( trans ) . SVG ( )
def Path ( self , trans = None , local = False ) :
""" Apply the transformation " trans " and return a Path object in
global coordinates . If local = True , return a Path in local coordinates
( which must be transformed again ) . """
if isinstance ( trans , basestring ) :
trans = totrans ( trans )
if self . mode [ 0 ] == " L " or self . mode [ 0 ] == " l " :
mode = " L "
elif self . mode [ 0 ] == " B " or self . mode [ 0 ] == " b " :
mode = " B "
elif self . mode [ 0 ] == " V " or self . mode [ 0 ] == " v " :
mode = " V "
elif self . mode [ 0 ] == " F " or self . mode [ 0 ] == " f " :
mode = " F "
elif self . mode [ 0 ] == " S " or self . mode [ 0 ] == " s " :
mode = " S "
vx , vy = [ 0. ] * len ( self . d ) , [ 0. ] * len ( self . d )
for i in xrange ( len ( self . d ) ) :
inext = ( i + 1 ) % len ( self . d )
iprev = ( i - 1 ) % len ( self . d )
vx [ i ] = ( self . d [ inext ] [ 0 ] - self . d [ iprev ] [ 0 ] ) / 2.
vy [ i ] = ( self . d [ inext ] [ 1 ] - self . d [ iprev ] [ 1 ] ) / 2.
if not self . loop and ( i == 0 or i == len ( self . d ) - 1 ) :
vx [ i ] , vy [ i ] = 0. , 0.
else :
raise ValueError , " mode must be \" lines \" , \" bezier \" , \" velocity \" , \" foreback \" , \" smooth \" , or an abbreviation "
d = [ ]
indexes = range ( len ( self . d ) )
if self . loop and len ( self . d ) > 0 :
indexes . append ( 0 )
for i in indexes :
inext = ( i + 1 ) % len ( self . d )
iprev = ( i - 1 ) % len ( self . d )
x , y = self . d [ i ] [ 0 ] , self . d [ i ] [ 1 ]
if trans is None :
X , Y = x , y
else :
X , Y = trans ( x , y )
if d == [ ] :
if local :
d . append ( ( " M " , x , y , False ) )
else :
d . append ( ( " M " , X , Y , True ) )
elif mode == " L " :
if local :
d . append ( ( " L " , x , y , False ) )
else :
d . append ( ( " L " , X , Y , True ) )
elif mode == " B " :
c1x , c1y = self . d [ i ] [ 2 ] , self . d [ i ] [ 3 ]
if trans is None :
C1X , C1Y = c1x , c1y
else :
C1X , C1Y = trans ( c1x , c1y )
c2x , c2y = self . d [ i ] [ 4 ] , self . d [ i ] [ 5 ]
if trans is None :
C2X , C2Y = c2x , c2y
else :
C2X , C2Y = trans ( c2x , c2y )
if local :
d . append ( ( " C " , c1x , c1y , False , c2x , c2y , False , x , y , False ) )
else :
d . append ( ( " C " , C1X , C1Y , True , C2X , C2Y , True , X , Y , True ) )
elif mode == " V " :
c1x , c1y = self . d [ iprev ] [ 2 ] / 3. + self . d [ iprev ] [ 0 ] , self . d [ iprev ] [ 3 ] / 3. + self . d [ iprev ] [ 1 ]
c2x , c2y = self . d [ i ] [ 2 ] / - 3. + x , self . d [ i ] [ 3 ] / - 3. + y
if trans is None :
C1X , C1Y = c1x , c1y
else :
C1X , C1Y = trans ( c1x , c1y )
if trans is None :
C2X , C2Y = c2x , c2y
else :
C2X , C2Y = trans ( c2x , c2y )
if local :
d . append ( ( " C " , c1x , c1y , False , c2x , c2y , False , x , y , False ) )
else :
d . append ( ( " C " , C1X , C1Y , True , C2X , C2Y , True , X , Y , True ) )
elif mode == " F " :
c1x , c1y = self . d [ iprev ] [ 4 ] / 3. + self . d [ iprev ] [ 0 ] , self . d [ iprev ] [ 5 ] / 3. + self . d [ iprev ] [ 1 ]
c2x , c2y = self . d [ i ] [ 2 ] / - 3. + x , self . d [ i ] [ 3 ] / - 3. + y
if trans is None :
C1X , C1Y = c1x , c1y
else :
C1X , C1Y = trans ( c1x , c1y )
if trans is None :
C2X , C2Y = c2x , c2y
else :
C2X , C2Y = trans ( c2x , c2y )
if local :
d . append ( ( " C " , c1x , c1y , False , c2x , c2y , False , x , y , False ) )
else :
d . append ( ( " C " , C1X , C1Y , True , C2X , C2Y , True , X , Y , True ) )
elif mode == " S " :
c1x , c1y = vx [ iprev ] / 3. + self . d [ iprev ] [ 0 ] , vy [ iprev ] / 3. + self . d [ iprev ] [ 1 ]
c2x , c2y = vx [ i ] / - 3. + x , vy [ i ] / - 3. + y
if trans is None :
C1X , C1Y = c1x , c1y
else :
C1X , C1Y = trans ( c1x , c1y )
if trans is None :
C2X , C2Y = c2x , c2y
else :
C2X , C2Y = trans ( c2x , c2y )
if local :
d . append ( ( " C " , c1x , c1y , False , c2x , c2y , False , x , y , False ) )
else :
d . append ( ( " C " , C1X , C1Y , True , C2X , C2Y , True , X , Y , True ) )
if self . loop and len ( self . d ) > 0 :
d . append ( ( " Z " , ) )
return Path ( d , * * self . attr )
######################################################################
class Text :
""" Draws a text string at a specified point in local coordinates.
x , y required location of the point in local coordinates
d required text / Unicode string
attribute = value pairs keyword list SVG attributes
"""
defaults = { " stroke " : " none " , " fill " : " black " , " font-size " : 5 , }
def __repr__ ( self ) :
return " <Text %s at ( %g , %g ) %s > " % ( repr ( self . d ) , self . x , self . y , self . attr )
def __init__ ( self , x , y , d , * * attr ) :
self . x = x
self . y = y
self . d = unicode ( d )
self . attr = dict ( self . defaults )
self . attr . update ( attr )
def SVG ( self , trans = None ) :
""" Apply the transformation " trans " and return an SVG object. """
if isinstance ( trans , basestring ) :
trans = totrans ( trans )
X , Y = self . x , self . y
if trans is not None :
X , Y = trans ( X , Y )
return SVG ( " text " , self . d , x = X , y = Y , * * self . attr )
class TextGlobal :
""" Draws a text string at a specified point in global coordinates.
x , y required location of the point in global coordinates
d required text / Unicode string
attribute = value pairs keyword list SVG attributes
"""
defaults = { " stroke " : " none " , " fill " : " black " , " font-size " : 5 , }
def __repr__ ( self ) :
return " <TextGlobal %s at ( %s , %s ) %s > " % ( repr ( self . d ) , str ( self . x ) , str ( self . y ) , self . attr )
def __init__ ( self , x , y , d , * * attr ) :
self . x = x
self . y = y
self . d = unicode ( d )
self . attr = dict ( self . defaults )
self . attr . update ( attr )
def SVG ( self , trans = None ) :
""" Apply the transformation " trans " and return an SVG object. """
return SVG ( " text " , self . d , x = self . x , y = self . y , * * self . attr )
######################################################################
_symbol_templates = { " dot " : SVG ( " symbol " , SVG ( " circle " , cx = 0 , cy = 0 , r = 1 , stroke = " none " , fill = " black " ) , viewBox = " 0 0 1 1 " , overflow = " visible " ) ,
" box " : SVG ( " symbol " , SVG ( " rect " , x1 = - 1 , y1 = - 1 , x2 = 1 , y2 = 1 , stroke = " none " , fill = " black " ) , viewBox = " 0 0 1 1 " , overflow = " visible " ) ,
" uptri " : SVG ( " symbol " , SVG ( " path " , d = " M -1 0.866 L 1 0.866 L 0 -0.866 Z " , stroke = " none " , fill = " black " ) , viewBox = " 0 0 1 1 " , overflow = " visible " ) ,
" downtri " : SVG ( " symbol " , SVG ( " path " , d = " M -1 -0.866 L 1 -0.866 L 0 0.866 Z " , stroke = " none " , fill = " black " ) , viewBox = " 0 0 1 1 " , overflow = " visible " ) ,
}
def make_symbol ( id , shape = " dot " , * * attr ) :
""" Creates a new instance of an SVG symbol to avoid cross-linking objects.
id required a new identifier ( string / Unicode )
shape default = " dot " the shape name from _symbol_templates
attribute = value list keyword list modify the SVG attributes of the new symbol
"""
output = copy . deepcopy ( _symbol_templates [ shape ] )
for i in output . sub :
i . attr . update ( attr_preprocess ( attr ) )
output [ " id " ] = id
return output
_circular_dot = make_symbol ( " circular_dot " )
class Dots :
""" Dots draws SVG symbols at a set of points.
d required list of ( x , y ) points
symbol default = None SVG symbol or a new identifier to
label an auto - generated symbol ;
if None , use pre - defined _circular_dot
width , height default = 1 , 1 width and height of the symbols
in SVG coordinates
attribute = value pairs keyword list SVG attributes
"""
defaults = { }
def __repr__ ( self ) :
return " <Dots ( %d nodes) %s > " % ( len ( self . d ) , self . attr )
def __init__ ( self , d = [ ] , symbol = None , width = 1. , height = 1. , * * attr ) :
self . d = list ( d )
self . width = width
self . height = height
self . attr = dict ( self . defaults )
self . attr . update ( attr )
if symbol is None :
self . symbol = _circular_dot
elif isinstance ( symbol , SVG ) :
self . symbol = symbol
else :
self . symbol = make_symbol ( symbol )
def SVG ( self , trans = None ) :
""" Apply the transformation " trans " and return an SVG object. """
if isinstance ( trans , basestring ) :
trans = totrans ( trans )
output = SVG ( " g " , SVG ( " defs " , self . symbol ) )
id = " # %s " % self . symbol [ " id " ]
for p in self . d :
x , y = p [ 0 ] , p [ 1 ]
if trans is None :
X , Y = x , y
else :
X , Y = trans ( x , y )
item = SVG ( " use " , x = X , y = Y , xlink__href = id )
if self . width is not None :
item [ " width " ] = self . width
if self . height is not None :
item [ " height " ] = self . height
output . append ( item )
return output
######################################################################
_marker_templates = { " arrow_start " : SVG ( " marker " , SVG ( " path " , d = " M 9 3.6 L 10.5 0 L 0 3.6 L 10.5 7.2 L 9 3.6 Z " ) , viewBox = " 0 0 10.5 7.2 " , refX = " 9 " , refY = " 3.6 " , markerWidth = " 10.5 " , markerHeight = " 7.2 " , markerUnits = " strokeWidth " , orient = " auto " , stroke = " none " , fill = " black " ) ,
" arrow_end " : SVG ( " marker " , SVG ( " path " , d = " M 1.5 3.6 L 0 0 L 10.5 3.6 L 0 7.2 L 1.5 3.6 Z " ) , viewBox = " 0 0 10.5 7.2 " , refX = " 1.5 " , refY = " 3.6 " , markerWidth = " 10.5 " , markerHeight = " 7.2 " , markerUnits = " strokeWidth " , orient = " auto " , stroke = " none " , fill = " black " ) ,
}
def make_marker ( id , shape , * * attr ) :
""" Creates a new instance of an SVG marker to avoid cross-linking objects.
id required a new identifier ( string / Unicode )
shape required the shape name from _marker_templates
attribute = value list keyword list modify the SVG attributes of the new marker
"""
output = copy . deepcopy ( _marker_templates [ shape ] )
for i in output . sub :
i . attr . update ( attr_preprocess ( attr ) )
output [ " id " ] = id
return output
class Line ( Curve ) :
""" Draws a line between two points.
Line ( x1 , y1 , x2 , y2 , arrow_start , arrow_end , attribute = value )
x1 , y1 required the starting point
x2 , y2 required the ending point
arrow_start default = None if an identifier string / Unicode ,
draw a new arrow object at the
beginning of the line ; if a marker ,
draw that marker instead
arrow_end default = None same for the end of the line
attribute = value pairs keyword list SVG attributes
"""
defaults = { }
def __repr__ ( self ) :
return " <Line ( %g , %g ) to ( %g , %g ) %s > " % (
self . x1 , self . y1 , self . x2 , self . y2 , self . attr )
def __init__ ( self , x1 , y1 , x2 , y2 , arrow_start = None , arrow_end = None , * * attr ) :
self . x1 , self . y1 , self . x2 , self . y2 = x1 , y1 , x2 , y2
self . arrow_start , self . arrow_end = arrow_start , arrow_end
self . attr = dict ( self . defaults )
self . attr . update ( attr )
def SVG ( self , trans = None ) :
""" Apply the transformation " trans " and return an SVG object. """
line = self . Path ( trans ) . SVG ( )
if ( ( self . arrow_start != False and self . arrow_start is not None ) or
( self . arrow_end != False and self . arrow_end is not None ) ) :
defs = SVG ( " defs " )
if self . arrow_start != False and self . arrow_start is not None :
if isinstance ( self . arrow_start , SVG ) :
defs . append ( self . arrow_start )
line . attr [ " marker-start " ] = " url(# %s ) " % self . arrow_start [ " id " ]
elif isinstance ( self . arrow_start , basestring ) :
defs . append ( make_marker ( self . arrow_start , " arrow_start " ) )
line . attr [ " marker-start " ] = " url(# %s ) " % self . arrow_start
else :
raise TypeError , " arrow_start must be False/None or an id string for the new marker "
if self . arrow_end != False and self . arrow_end is not None :
if isinstance ( self . arrow_end , SVG ) :
defs . append ( self . arrow_end )
line . attr [ " marker-end " ] = " url(# %s ) " % self . arrow_end [ " id " ]
elif isinstance ( self . arrow_end , basestring ) :
defs . append ( make_marker ( self . arrow_end , " arrow_end " ) )
line . attr [ " marker-end " ] = " url(# %s ) " % self . arrow_end
else :
raise TypeError , " arrow_end must be False/None or an id string for the new marker "
return SVG ( " g " , defs , line )
return line
def Path ( self , trans = None , local = False ) :
""" Apply the transformation " trans " and return a Path object in
global coordinates . If local = True , return a Path in local coordinates
( which must be transformed again ) . """
self . f = lambda t : ( self . x1 + t * ( self . x2 - self . x1 ) , self . y1 + t * ( self . y2 - self . y1 ) )
self . low = 0.
self . high = 1.
self . loop = False
if trans is None :
return Path ( [ ( " M " , self . x1 , self . y1 , not local ) , ( " L " , self . x2 , self . y2 , not local ) ] , * * self . attr )
else :
return Curve . Path ( self , trans , local )
class LineGlobal :
""" Draws a line between two points, one or both of which is in
global coordinates .
Line ( x1 , y1 , x2 , y2 , lcoal1 , local2 , arrow_start , arrow_end , attribute = value )
x1 , y1 required the starting point
x2 , y2 required the ending point
local1 default = False if True , interpret first point as a
local coordinate ( apply transform )
local2 default = False if True , interpret second point as a
local coordinate ( apply transform )
arrow_start default = None if an identifier string / Unicode ,
draw a new arrow object at the
beginning of the line ; if a marker ,
draw that marker instead
arrow_end default = None same for the end of the line
attribute = value pairs keyword list SVG attributes
"""
defaults = { }
def __repr__ ( self ) :
local1 , local2 = " " , " "
if self . local1 :
local1 = " L "
if self . local2 :
local2 = " L "
return " <LineGlobal %s ( %s , %s ) to %s ( %s , %s ) %s > " % (
local1 , str ( self . x1 ) , str ( self . y1 ) , local2 , str ( self . x2 ) , str ( self . y2 ) , self . attr )
def __init__ ( self , x1 , y1 , x2 , y2 , local1 = False , local2 = False , arrow_start = None , arrow_end = None , * * attr ) :
self . x1 , self . y1 , self . x2 , self . y2 = x1 , y1 , x2 , y2
self . local1 , self . local2 = local1 , local2
self . arrow_start , self . arrow_end = arrow_start , arrow_end
self . attr = dict ( self . defaults )
self . attr . update ( attr )
def SVG ( self , trans = None ) :
""" Apply the transformation " trans " and return an SVG object. """
if isinstance ( trans , basestring ) :
trans = totrans ( trans )
X1 , Y1 , X2 , Y2 = self . x1 , self . y1 , self . x2 , self . y2
if self . local1 :
X1 , Y1 = trans ( X1 , Y1 )
if self . local2 :
X2 , Y2 = trans ( X2 , Y2 )
line = SVG ( " path " , d = " M %s %s L %s %s " % ( X1 , Y1 , X2 , Y2 ) , * * self . attr )
if ( ( self . arrow_start != False and self . arrow_start is not None ) or
( self . arrow_end != False and self . arrow_end is not None ) ) :
defs = SVG ( " defs " )
if self . arrow_start != False and self . arrow_start is not None :
if isinstance ( self . arrow_start , SVG ) :
defs . append ( self . arrow_start )
line . attr [ " marker-start " ] = " url(# %s ) " % self . arrow_start [ " id " ]
elif isinstance ( self . arrow_start , basestring ) :
defs . append ( make_marker ( self . arrow_start , " arrow_start " ) )
line . attr [ " marker-start " ] = " url(# %s ) " % self . arrow_start
else :
raise TypeError , " arrow_start must be False/None or an id string for the new marker "
if self . arrow_end != False and self . arrow_end is not None :
if isinstance ( self . arrow_end , SVG ) :
defs . append ( self . arrow_end )
line . attr [ " marker-end " ] = " url(# %s ) " % self . arrow_end [ " id " ]
elif isinstance ( self . arrow_end , basestring ) :
defs . append ( make_marker ( self . arrow_end , " arrow_end " ) )
line . attr [ " marker-end " ] = " url(# %s ) " % self . arrow_end
else :
raise TypeError , " arrow_end must be False/None or an id string for the new marker "
return SVG ( " g " , defs , line )
return line
class VLine ( Line ) :
""" Draws a vertical line.
VLine ( y1 , y2 , x , attribute = value )
y1 , y2 required y range
x required x position
attribute = value pairs keyword list SVG attributes
"""
defaults = { }
def __repr__ ( self ) :
return " <VLine ( %g , %g ) at x= %s %s > " % ( self . y1 , self . y2 , self . x , self . attr )
def __init__ ( self , y1 , y2 , x , * * attr ) :
self . x = x
self . attr = dict ( self . defaults )
self . attr . update ( attr )
Line . __init__ ( self , x , y1 , x , y2 , * * self . attr )
def Path ( self , trans = None , local = False ) :
""" Apply the transformation " trans " and return a Path object in
global coordinates . If local = True , return a Path in local coordinates
( which must be transformed again ) . """
self . x1 = self . x
self . x2 = self . x
return Line . Path ( self , trans , local )
class HLine ( Line ) :
""" Draws a horizontal line.
HLine ( x1 , x2 , y , attribute = value )
x1 , x2 required x range
y required y position
attribute = value pairs keyword list SVG attributes
"""
defaults = { }
def __repr__ ( self ) :
return " <HLine ( %g , %g ) at y= %s %s > " % ( self . x1 , self . x2 , self . y , self . attr )
def __init__ ( self , x1 , x2 , y , * * attr ) :
self . y = y
self . attr = dict ( self . defaults )
self . attr . update ( attr )
Line . __init__ ( self , x1 , y , x2 , y , * * self . attr )
def Path ( self , trans = None , local = False ) :
""" Apply the transformation " trans " and return a Path object in
global coordinates . If local = True , return a Path in local coordinates
( which must be transformed again ) . """
self . y1 = self . y
self . y2 = self . y
return Line . Path ( self , trans , local )
######################################################################
class Rect ( Curve ) :
""" Draws a rectangle.
Rect ( x1 , y1 , x2 , y2 , attribute = value )
x1 , y1 required the starting point
x2 , y2 required the ending point
attribute = value pairs keyword list SVG attributes
"""
defaults = { }
def __repr__ ( self ) :
return " <Rect ( %g , %g ), ( %g , %g ) %s > " % (
self . x1 , self . y1 , self . x2 , self . y2 , self . attr )
def __init__ ( self , x1 , y1 , x2 , y2 , * * attr ) :
self . x1 , self . y1 , self . x2 , self . y2 = x1 , y1 , x2 , y2
self . attr = dict ( self . defaults )
self . attr . update ( attr )
def SVG ( self , trans = None ) :
""" Apply the transformation " trans " and return an SVG object. """
return self . Path ( trans ) . SVG ( )
def Path ( self , trans = None , local = False ) :
""" Apply the transformation " trans " and return a Path object in
global coordinates . If local = True , return a Path in local coordinates
( which must be transformed again ) . """
if trans is None :
return Path ( [ ( " M " , self . x1 , self . y1 , not local ) , ( " L " , self . x2 , self . y1 , not local ) , ( " L " , self . x2 , self . y2 , not local ) , ( " L " , self . x1 , self . y2 , not local ) , ( " Z " , ) ] , * * self . attr )
else :
self . low = 0.
self . high = 1.
self . loop = False
self . f = lambda t : ( self . x1 + t * ( self . x2 - self . x1 ) , self . y1 )
d1 = Curve . Path ( self , trans , local ) . d
self . f = lambda t : ( self . x2 , self . y1 + t * ( self . y2 - self . y1 ) )
d2 = Curve . Path ( self , trans , local ) . d
del d2 [ 0 ]
self . f = lambda t : ( self . x2 + t * ( self . x1 - self . x2 ) , self . y2 )
d3 = Curve . Path ( self , trans , local ) . d
del d3 [ 0 ]
self . f = lambda t : ( self . x1 , self . y2 + t * ( self . y1 - self . y2 ) )
d4 = Curve . Path ( self , trans , local ) . d
del d4 [ 0 ]
return Path ( d = ( d1 + d2 + d3 + d4 + [ ( " Z " , ) ] ) , * * self . attr )
######################################################################
class Ellipse ( Curve ) :
""" Draws an ellipse from a semimajor vector (ax,ay) and a semiminor
length ( b ) .
Ellipse ( x , y , ax , ay , b , attribute = value )
x , y required the center of the ellipse / circle
ax , ay required a vector indicating the length
and direction of the semimajor axis
b required the length of the semiminor axis .
If equal to sqrt ( ax2 + ay2 ) , the
ellipse is a circle
attribute = value pairs keyword list SVG attributes
( If sqrt ( ax * * 2 + ay * * 2 ) is less than b , then ( ax , ay ) is actually the
semiminor axis . )
"""
defaults = { }
def __repr__ ( self ) :
return " <Ellipse ( %g , %g ) a=( %g , %g ), b= %g %s > " % (
self . x , self . y , self . ax , self . ay , self . b , self . attr )
def __init__ ( self , x , y , ax , ay , b , * * attr ) :
self . x , self . y , self . ax , self . ay , self . b = x , y , ax , ay , b
self . attr = dict ( self . defaults )
self . attr . update ( attr )
def SVG ( self , trans = None ) :
""" Apply the transformation " trans " and return an SVG object. """
return self . Path ( trans ) . SVG ( )
def Path ( self , trans = None , local = False ) :
""" Apply the transformation " trans " and return a Path object in
global coordinates . If local = True , return a Path in local coordinates
( which must be transformed again ) . """
angle = math . atan2 ( self . ay , self . ax ) + math . pi / 2.
bx = self . b * math . cos ( angle )
by = self . b * math . sin ( angle )
self . f = lambda t : ( self . x + self . ax * math . cos ( t ) + bx * math . sin ( t ) , self . y + self . ay * math . cos ( t ) + by * math . sin ( t ) )
self . low = - math . pi
self . high = math . pi
self . loop = True
return Curve . Path ( self , trans , local )
######################################################################
def unumber ( x ) :
""" Converts numbers to a Unicode string, taking advantage of special
Unicode characters to make nice minus signs and scientific notation .
"""
output = u " %g " % x
if output [ 0 ] == u " - " :
output = u " \u2013 " + output [ 1 : ]
index = output . find ( u " e " )
if index != - 1 :
uniout = unicode ( output [ : index ] ) + u " \u00d7 10 "
saw_nonzero = False
for n in output [ index + 1 : ] :
if n == u " + " :
pass # uniout += u"\u207a"
elif n == u " - " :
uniout + = u " \u207b "
elif n == u " 0 " :
if saw_nonzero :
uniout + = u " \u2070 "
elif n == u " 1 " :
saw_nonzero = True
uniout + = u " \u00b9 "
elif n == u " 2 " :
saw_nonzero = True
uniout + = u " \u00b2 "
elif n == u " 3 " :
saw_nonzero = True
uniout + = u " \u00b3 "
elif u " 4 " < = n < = u " 9 " :
saw_nonzero = True
if saw_nonzero :
uniout + = eval ( " u \" \\ u %x \" " % ( 0x2070 + ord ( n ) - ord ( u " 0 " ) ) )
else :
uniout + = n
if uniout [ : 2 ] == u " 1 \u00d7 " :
uniout = uniout [ 2 : ]
return uniout
return output
class Ticks :
""" Superclass for all graphics primitives that draw ticks,
miniticks , and tick labels . This class only draws the ticks .
Ticks ( f , low , high , ticks , miniticks , labels , logbase , arrow_start ,
arrow_end , text_attr , attribute = value )
f required parametric function along which ticks
will be drawn ; has the same format as
the function used in Curve
low , high required range of the independent variable
ticks default = - 10 request ticks according to the standard
tick specification ( see below )
miniticks default = True request miniticks according to the
standard minitick specification ( below )
labels True request tick labels according to the
standard tick label specification ( below )
logbase default = None if a number , the axis is logarithmic with
ticks at the given base ( usually 10 )
arrow_start default = None if a new string identifier , draw an arrow
at the low - end of the axis , referenced by
that identifier ; if an SVG marker object ,
use that marker
arrow_end default = None if a new string identifier , draw an arrow
at the high - end of the axis , referenced by
that identifier ; if an SVG marker object ,
use that marker
text_attr default = { } SVG attributes for the text labels
attribute = value pairs keyword list SVG attributes for the tick marks
Standard tick specification :
* True : same as - 10 ( below ) .
* Positive number N : draw exactly N ticks , including the endpoints . To
subdivide an axis into 10 equal - sized segments , ask for 11 ticks .
* Negative number - N : draw at least N ticks . Ticks will be chosen with
" natural " values , multiples of 2 or 5.
* List of values : draw a tick mark at each value .
* Dict of value , label pairs : draw a tick mark at each value , labeling
it with the given string . This lets you say things like { 3.14159 : " pi " } .
* False or None : no ticks .
Standard minitick specification :
* True : draw miniticks with " natural " values , more closely spaced than
the ticks .
* Positive number N : draw exactly N miniticks , including the endpoints .
To subdivide an axis into 100 equal - sized segments , ask for 101 miniticks .
* Negative number - N : draw at least N miniticks .
* List of values : draw a minitick mark at each value .
* False or None : no miniticks .
Standard tick label specification :
* True : use the unumber function ( described below )
* Format string : standard format strings , e . g . " %5.2f " for 12.34
* Python callable : function that converts numbers to strings
* False or None : no labels
"""
defaults = { " stroke-width " : " 0.25pt " , }
text_defaults = { " stroke " : " none " , " fill " : " black " , " font-size " : 5 , }
tick_start = - 1.5
tick_end = 1.5
minitick_start = - 0.75
minitick_end = 0.75
text_start = 2.5
text_angle = 0.
def __repr__ ( self ) :
return " <Ticks %s from %s to %s ticks= %s labels= %s %s > " % (
self . f , self . low , self . high , str ( self . ticks ) , str ( self . labels ) , self . attr )
def __init__ ( self , f , low , high , ticks = - 10 , miniticks = True , labels = True , logbase = None ,
arrow_start = None , arrow_end = None , text_attr = { } , * * attr ) :
self . f = f
self . low = low
self . high = high
self . ticks = ticks
self . miniticks = miniticks
self . labels = labels
self . logbase = logbase
self . arrow_start = arrow_start
self . arrow_end = arrow_end
self . attr = dict ( self . defaults )
self . attr . update ( attr )
self . text_attr = dict ( self . text_defaults )
self . text_attr . update ( text_attr )
def orient_tickmark ( self , t , trans = None ) :
""" Return the position, normalized local x vector, normalized
local y vector , and angle of a tick at position t .
Normally only used internally .
"""
if isinstance ( trans , basestring ) :
trans = totrans ( trans )
if trans is None :
f = self . f
else :
f = lambda t : trans ( * self . f ( t ) )
eps = _epsilon * abs ( self . high - self . low )
X , Y = f ( t )
Xprime , Yprime = f ( t + eps )
xhatx , xhaty = ( Xprime - X ) / eps , ( Yprime - Y ) / eps
norm = math . sqrt ( xhatx * * 2 + xhaty * * 2 )
if norm != 0 :
xhatx , xhaty = xhatx / norm , xhaty / norm
else :
xhatx , xhaty = 1. , 0.
angle = math . atan2 ( xhaty , xhatx ) + math . pi / 2.
yhatx , yhaty = math . cos ( angle ) , math . sin ( angle )
return ( X , Y ) , ( xhatx , xhaty ) , ( yhatx , yhaty ) , angle
def SVG ( self , trans = None ) :
""" Apply the transformation " trans " and return an SVG object. """
if isinstance ( trans , basestring ) :
trans = totrans ( trans )
self . last_ticks , self . last_miniticks = self . interpret ( )
tickmarks = Path ( [ ] , * * self . attr )
minitickmarks = Path ( [ ] , * * self . attr )
output = SVG ( " g " )
if ( ( self . arrow_start != False and self . arrow_start is not None ) or
( self . arrow_end != False and self . arrow_end is not None ) ) :
defs = SVG ( " defs " )
if self . arrow_start != False and self . arrow_start is not None :
if isinstance ( self . arrow_start , SVG ) :
defs . append ( self . arrow_start )
elif isinstance ( self . arrow_start , basestring ) :
defs . append ( make_marker ( self . arrow_start , " arrow_start " ) )
else :
raise TypeError , " arrow_start must be False/None or an id string for the new marker "
if self . arrow_end != False and self . arrow_end is not None :
if isinstance ( self . arrow_end , SVG ) :
defs . append ( self . arrow_end )
elif isinstance ( self . arrow_end , basestring ) :
defs . append ( make_marker ( self . arrow_end , " arrow_end " ) )
else :
raise TypeError , " arrow_end must be False/None or an id string for the new marker "
output . append ( defs )
eps = _epsilon * ( self . high - self . low )
for t , label in self . last_ticks . items ( ) :
( X , Y ) , ( xhatx , xhaty ) , ( yhatx , yhaty ) , angle = self . orient_tickmark ( t , trans )
if ( ( not self . arrow_start or abs ( t - self . low ) > eps ) and
( not self . arrow_end or abs ( t - self . high ) > eps ) ) :
tickmarks . d . append ( ( " M " , X - yhatx * self . tick_start , Y - yhaty * self . tick_start , True ) )
tickmarks . d . append ( ( " L " , X - yhatx * self . tick_end , Y - yhaty * self . tick_end , True ) )
angle = ( angle - math . pi / 2. ) * 180. / math . pi + self . text_angle
########### a HACK! ############ (to be removed when Inkscape handles baselines)
if _hacks [ " inkscape-text-vertical-shift " ] :
if self . text_start > 0 :
X + = math . cos ( angle * math . pi / 180. + math . pi / 2. ) * 2.
Y + = math . sin ( angle * math . pi / 180. + math . pi / 2. ) * 2.
else :
X + = math . cos ( angle * math . pi / 180. + math . pi / 2. ) * 2. * 2.5
Y + = math . sin ( angle * math . pi / 180. + math . pi / 2. ) * 2. * 2.5
########### end hack ###########
if label != " " :
output . append ( SVG ( " text " , label , transform = " translate( %g , %g ) rotate( %g ) " %
( X - yhatx * self . text_start , Y - yhaty * self . text_start , angle ) , * * self . text_attr ) )
for t in self . last_miniticks :
skip = False
for tt in self . last_ticks . keys ( ) :
if abs ( t - tt ) < eps :
skip = True
break
if not skip :
( X , Y ) , ( xhatx , xhaty ) , ( yhatx , yhaty ) , angle = self . orient_tickmark ( t , trans )
if ( ( not self . arrow_start or abs ( t - self . low ) > eps ) and
( not self . arrow_end or abs ( t - self . high ) > eps ) ) :
minitickmarks . d . append ( ( " M " , X - yhatx * self . minitick_start , Y - yhaty * self . minitick_start , True ) )
minitickmarks . d . append ( ( " L " , X - yhatx * self . minitick_end , Y - yhaty * self . minitick_end , True ) )
output . prepend ( tickmarks . SVG ( trans ) )
output . prepend ( minitickmarks . SVG ( trans ) )
return output
def interpret ( self ) :
""" Evaluate and return optimal ticks and miniticks according to
the standard minitick specification .
Normally only used internally .
"""
if self . labels is None or self . labels == False :
format = lambda x : " "
elif self . labels == True :
format = unumber
elif isinstance ( self . labels , basestring ) :
format = lambda x : ( self . labels % x )
elif callable ( self . labels ) :
format = self . labels
else :
raise TypeError , " labels must be None/False, True, a format string, or a number->string function "
# Now for the ticks
ticks = self . ticks
# Case 1: ticks is None/False
if ticks is None or ticks == False :
return { } , [ ]
# Case 2: ticks is the number of desired ticks
elif isinstance ( ticks , ( int , long ) ) :
if ticks == True :
ticks = - 10
if self . logbase is None :
ticks = self . compute_ticks ( ticks , format )
else :
ticks = self . compute_logticks ( self . logbase , ticks , format )
# Now for the miniticks
if self . miniticks == True :
if self . logbase is None :
return ticks , self . compute_miniticks ( ticks )
else :
return ticks , self . compute_logminiticks ( self . logbase )
elif isinstance ( self . miniticks , ( int , long ) ) :
return ticks , self . regular_miniticks ( self . miniticks )
elif getattr ( self . miniticks , " __iter__ " , False ) :
return ticks , self . miniticks
elif self . miniticks == False or self . miniticks is None :
return ticks , [ ]
else :
raise TypeError , " miniticks must be None/False, True, a number of desired miniticks, or a list of numbers "
# Cases 3 & 4: ticks is iterable
elif getattr ( ticks , " __iter__ " , False ) :
# Case 3: ticks is some kind of list
if not isinstance ( ticks , dict ) :
output = { }
eps = _epsilon * ( self . high - self . low )
for x in ticks :
if format == unumber and abs ( x ) < eps :
output [ x ] = u " 0 "
else :
output [ x ] = format ( x )
ticks = output
# Case 4: ticks is a dict
else :
pass
# Now for the miniticks
if self . miniticks == True :
if self . logbase is None :
return ticks , self . compute_miniticks ( ticks )
else :
return ticks , self . compute_logminiticks ( self . logbase )
elif isinstance ( self . miniticks , ( int , long ) ) :
return ticks , self . regular_miniticks ( self . miniticks )
elif getattr ( self . miniticks , " __iter__ " , False ) :
return ticks , self . miniticks
elif self . miniticks == False or self . miniticks is None :
return ticks , [ ]
else :
raise TypeError , " miniticks must be None/False, True, a number of desired miniticks, or a list of numbers "
else :
raise TypeError , " ticks must be None/False, a number of desired ticks, a list of numbers, or a dictionary of explicit markers "
def compute_ticks ( self , N , format ) :
""" Return less than -N or exactly N optimal linear ticks.
Normally only used internally .
"""
if self . low > = self . high :
raise ValueError , " low must be less than high "
if N == 1 :
raise ValueError , " N can be 0 or >1 to specify the exact number of ticks or negative to specify a maximum "
eps = _epsilon * ( self . high - self . low )
if N > = 0 :
output = { }
x = self . low
for i in xrange ( N ) :
if format == unumber and abs ( x ) < eps :
label = u " 0 "
else :
label = format ( x )
output [ x ] = label
x + = ( self . high - self . low ) / ( N - 1. )
return output
N = - N
counter = 0
granularity = 10 * * math . ceil ( math . log10 ( max ( abs ( self . low ) , abs ( self . high ) ) ) )
lowN = math . ceil ( 1. * self . low / granularity )
highN = math . floor ( 1. * self . high / granularity )
while lowN > highN :
countermod3 = counter % 3
if countermod3 == 0 :
granularity * = 0.5
elif countermod3 == 1 :
granularity * = 0.4
else :
granularity * = 0.5
counter + = 1
lowN = math . ceil ( 1. * self . low / granularity )
highN = math . floor ( 1. * self . high / granularity )
last_granularity = granularity
last_trial = None
while True :
trial = { }
for n in range ( int ( lowN ) , int ( highN ) + 1 ) :
x = n * granularity
if format == unumber and abs ( x ) < eps :
label = u " 0 "
else :
label = format ( x )
trial [ x ] = label
if int ( highN ) + 1 - int ( lowN ) > = N :
if last_trial is None :
v1 , v2 = self . low , self . high
return { v1 : format ( v1 ) , v2 : format ( v2 ) }
else :
low_in_ticks , high_in_ticks = False , False
for t in last_trial . keys ( ) :
if 1. * abs ( t - self . low ) / last_granularity < _epsilon :
low_in_ticks = True
if 1. * abs ( t - self . high ) / last_granularity < _epsilon :
high_in_ticks = True
lowN = 1. * self . low / last_granularity
highN = 1. * self . high / last_granularity
if abs ( lowN - round ( lowN ) ) < _epsilon and not low_in_ticks :
last_trial [ self . low ] = format ( self . low )
if abs ( highN - round ( highN ) ) < _epsilon and not high_in_ticks :
last_trial [ self . high ] = format ( self . high )
return last_trial
last_granularity = granularity
last_trial = trial
countermod3 = counter % 3
if countermod3 == 0 :
granularity * = 0.5
elif countermod3 == 1 :
granularity * = 0.4
else :
granularity * = 0.5
counter + = 1
lowN = math . ceil ( 1. * self . low / granularity )
highN = math . floor ( 1. * self . high / granularity )
def regular_miniticks ( self , N ) :
""" Return exactly N linear ticks.
Normally only used internally .
"""
output = [ ]
x = self . low
for i in xrange ( N ) :
output . append ( x )
x + = ( self . high - self . low ) / ( N - 1. )
return output
def compute_miniticks ( self , original_ticks ) :
""" Return optimal linear miniticks, given a set of ticks.
Normally only used internally .
"""
if len ( original_ticks ) < 2 :
original_ticks = ticks ( self . low , self . high ) # XXX ticks is undefined!
original_ticks = original_ticks . keys ( )
original_ticks . sort ( )
if self . low > original_ticks [ 0 ] + _epsilon or self . high < original_ticks [ - 1 ] - _epsilon :
raise ValueError , " original_ticks { %g ... %g } extend beyond [ %g , %g ] " % ( original_ticks [ 0 ] , original_ticks [ - 1 ] , self . low , self . high )
granularities = [ ]
for i in range ( len ( original_ticks ) - 1 ) :
granularities . append ( original_ticks [ i + 1 ] - original_ticks [ i ] )
spacing = 10 * * ( math . ceil ( math . log10 ( min ( granularities ) ) - 1 ) )
output = [ ]
x = original_ticks [ 0 ] - math . ceil ( 1. * ( original_ticks [ 0 ] - self . low ) / spacing ) * spacing
while x < = self . high :
if x > = self . low :
already_in_ticks = False
for t in original_ticks :
if abs ( x - t ) < _epsilon * ( self . high - self . low ) :
already_in_ticks = True
if not already_in_ticks :
output . append ( x )
x + = spacing
return output
def compute_logticks ( self , base , N , format ) :
""" Return less than -N or exactly N optimal logarithmic ticks.
Normally only used internally .
"""
if self . low > = self . high :
raise ValueError , " low must be less than high "
if N == 1 :
raise ValueError , " N can be 0 or >1 to specify the exact number of ticks or negative to specify a maximum "
eps = _epsilon * ( self . high - self . low )
if N > = 0 :
output = { }
x = self . low
for i in xrange ( N ) :
if format == unumber and abs ( x ) < eps :
label = u " 0 "
else :
label = format ( x )
output [ x ] = label
x + = ( self . high - self . low ) / ( N - 1. )
return output
N = - N
lowN = math . floor ( math . log ( self . low , base ) )
highN = math . ceil ( math . log ( self . high , base ) )
output = { }
for n in range ( int ( lowN ) , int ( highN ) + 1 ) :
x = base * * n
label = format ( x )
if self . low < = x < = self . high :
output [ x ] = label
for i in range ( 1 , len ( output ) ) :
keys = output . keys ( )
keys . sort ( )
keys = keys [ : : i ]
values = map ( lambda k : output [ k ] , keys )
if len ( values ) < = N :
for k in output . keys ( ) :
if k not in keys :
output [ k ] = " "
break
if len ( output ) < = 2 :
output2 = self . compute_ticks ( N = - int ( math . ceil ( N / 2. ) ) , format = format )
lowest = min ( output2 )
for k in output :
if k < lowest :
output2 [ k ] = output [ k ]
output = output2
return output
def compute_logminiticks ( self , base ) :
""" Return optimal logarithmic miniticks, given a set of ticks.
Normally only used internally .
"""
if self . low > = self . high :
raise ValueError , " low must be less than high "
lowN = math . floor ( math . log ( self . low , base ) )
highN = math . ceil ( math . log ( self . high , base ) )
output = [ ]
num_ticks = 0
for n in range ( int ( lowN ) , int ( highN ) + 1 ) :
x = base * * n
if self . low < = x < = self . high :
num_ticks + = 1
for m in range ( 2 , int ( math . ceil ( base ) ) ) :
minix = m * x
if self . low < = minix < = self . high :
output . append ( minix )
if num_ticks < = 2 :
return [ ]
else :
return output
######################################################################
class CurveAxis ( Curve , Ticks ) :
""" Draw an axis with tick marks along a parametric curve.
CurveAxis ( f , low , high , ticks , miniticks , labels , logbase , arrow_start , arrow_end ,
text_attr , attribute = value )
f required a Python callable or string in
the form " f(t), g(t) " , just like Curve
low , high required left and right endpoints
ticks default = - 10 request ticks according to the standard
tick specification ( see help ( Ticks ) )
miniticks default = True request miniticks according to the
standard minitick specification
labels True request tick labels according to the
standard tick label specification
logbase default = None if a number , the x axis is logarithmic
with ticks at the given base ( 10 being
the most common )
arrow_start default = None if a new string identifier , draw an
arrow at the low - end of the axis ,
referenced by that identifier ; if an
SVG marker object , use that marker
arrow_end default = None if a new string identifier , draw an
arrow at the high - end of the axis ,
referenced by that identifier ; if an
SVG marker object , use that marker
text_attr default = { } SVG attributes for the text labels
attribute = value pairs keyword list SVG attributes
"""
defaults = { " stroke-width " : " 0.25pt " , }
text_defaults = { " stroke " : " none " , " fill " : " black " , " font-size " : 5 , }
def __repr__ ( self ) :
return " <CurveAxis %s [ %s , %s ] ticks= %s labels= %s %s > " % (
self . f , self . low , self . high , str ( self . ticks ) , str ( self . labels ) , self . attr )
def __init__ ( self , f , low , high , ticks = - 10 , miniticks = True , labels = True , logbase = None ,
arrow_start = None , arrow_end = None , text_attr = { } , * * attr ) :
tattr = dict ( self . text_defaults )
tattr . update ( text_attr )
Curve . __init__ ( self , f , low , high )
Ticks . __init__ ( self , f , low , high , ticks , miniticks , labels , logbase , arrow_start , arrow_end , tattr , * * attr )
def SVG ( self , trans = None ) :
""" Apply the transformation " trans " and return an SVG object. """
func = Curve . SVG ( self , trans )
ticks = Ticks . SVG ( self , trans ) # returns a <g />
if self . arrow_start != False and self . arrow_start is not None :
if isinstance ( self . arrow_start , basestring ) :
func . attr [ " marker-start " ] = " url(# %s ) " % self . arrow_start
else :
func . attr [ " marker-start " ] = " url(# %s ) " % self . arrow_start . id
if self . arrow_end != False and self . arrow_end is not None :
if isinstance ( self . arrow_end , basestring ) :
func . attr [ " marker-end " ] = " url(# %s ) " % self . arrow_end
else :
func . attr [ " marker-end " ] = " url(# %s ) " % self . arrow_end . id
ticks . append ( func )
return ticks
class LineAxis ( Line , Ticks ) :
""" Draws an axis with tick marks along a line.
LineAxis ( x1 , y1 , x2 , y2 , start , end , ticks , miniticks , labels , logbase ,
arrow_start , arrow_end , text_attr , attribute = value )
x1 , y1 required starting point
x2 , y2 required ending point
start , end default = 0 , 1 values to start and end labeling
ticks default = - 10 request ticks according to the standard
tick specification ( see help ( Ticks ) )
miniticks default = True request miniticks according to the
standard minitick specification
labels True request tick labels according to the
standard tick label specification
logbase default = None if a number , the x axis is logarithmic
with ticks at the given base ( usually 10 )
arrow_start default = None if a new string identifier , draw an arrow
at the low - end of the axis , referenced by
that identifier ; if an SVG marker object ,
use that marker
arrow_end default = None if a new string identifier , draw an arrow
at the high - end of the axis , referenced by
that identifier ; if an SVG marker object ,
use that marker
text_attr default = { } SVG attributes for the text labels
attribute = value pairs keyword list SVG attributes
"""
defaults = { " stroke-width " : " 0.25pt " , }
text_defaults = { " stroke " : " none " , " fill " : " black " , " font-size " : 5 , }
def __repr__ ( self ) :
return " <LineAxis ( %g , %g ) to ( %g , %g ) ticks= %s labels= %s %s > " % (
self . x1 , self . y1 , self . x2 , self . y2 , str ( self . ticks ) , str ( self . labels ) , self . attr )
def __init__ ( self , x1 , y1 , x2 , y2 , start = 0. , end = 1. , ticks = - 10 , miniticks = True , labels = True ,
logbase = None , arrow_start = None , arrow_end = None , exclude = None , text_attr = { } , * * attr ) :
self . start = start
self . end = end
self . exclude = exclude
tattr = dict ( self . text_defaults )
tattr . update ( text_attr )
Line . __init__ ( self , x1 , y1 , x2 , y2 , * * attr )
Ticks . __init__ ( self , None , None , None , ticks , miniticks , labels , logbase , arrow_start , arrow_end , tattr , * * attr )
def interpret ( self ) :
if self . exclude is not None and not ( isinstance ( self . exclude , ( tuple , list ) ) and len ( self . exclude ) == 2 and
isinstance ( self . exclude [ 0 ] , ( int , long , float ) ) and isinstance ( self . exclude [ 1 ] , ( int , long , float ) ) ) :
raise TypeError , " exclude must either be None or (low, high) "
ticks , miniticks = Ticks . interpret ( self )
if self . exclude is None :
return ticks , miniticks
ticks2 = { }
for loc , label in ticks . items ( ) :
if self . exclude [ 0 ] < = loc < = self . exclude [ 1 ] :
ticks2 [ loc ] = " "
else :
ticks2 [ loc ] = label
return ticks2 , miniticks
def SVG ( self , trans = None ) :
""" Apply the transformation " trans " and return an SVG object. """
line = Line . SVG ( self , trans ) # must be evaluated first, to set self.f, self.low, self.high
f01 = self . f
self . f = lambda t : f01 ( 1. * ( t - self . start ) / ( self . end - self . start ) )
self . low = self . start
self . high = self . end
if self . arrow_start != False and self . arrow_start is not None :
if isinstance ( self . arrow_start , basestring ) :
line . attr [ " marker-start " ] = " url(# %s ) " % self . arrow_start
else :
line . attr [ " marker-start " ] = " url(# %s ) " % self . arrow_start . id
if self . arrow_end != False and self . arrow_end is not None :
if isinstance ( self . arrow_end , basestring ) :
line . attr [ " marker-end " ] = " url(# %s ) " % self . arrow_end
else :
line . attr [ " marker-end " ] = " url(# %s ) " % self . arrow_end . id
ticks = Ticks . SVG ( self , trans ) # returns a <g />
ticks . append ( line )
return ticks
class XAxis ( LineAxis ) :
""" Draws an x axis with tick marks.
XAxis ( xmin , xmax , aty , ticks , miniticks , labels , logbase , arrow_start , arrow_end ,
exclude , text_attr , attribute = value )
xmin , xmax required the x range
aty default = 0 y position to draw the axis
ticks default = - 10 request ticks according to the standard
tick specification ( see help ( Ticks ) )
miniticks default = True request miniticks according to the
standard minitick specification
labels True request tick labels according to the
standard tick label specification
logbase default = None if a number , the x axis is logarithmic
with ticks at the given base ( usually 10 )
arrow_start default = None if a new string identifier , draw an arrow
at the low - end of the axis , referenced by
that identifier ; if an SVG marker object ,
use that marker
arrow_end default = None if a new string identifier , draw an arrow
at the high - end of the axis , referenced by
that identifier ; if an SVG marker object ,
use that marker
exclude default = None if a ( low , high ) pair , don ' t draw text
labels within this range
text_attr default = { } SVG attributes for the text labels
attribute = value pairs keyword list SVG attributes for all lines
The exclude option is provided for Axes to keep text from overlapping
where the axes cross . Normal users are not likely to need it .
"""
defaults = { " stroke-width " : " 0.25pt " , }
text_defaults = { " stroke " : " none " , " fill " : " black " , " font-size " : 5 , " dominant-baseline " : " text-before-edge " , }
text_start = - 1.
text_angle = 0.
def __repr__ ( self ) :
return " <XAxis ( %g , %g ) at y= %g ticks= %s labels= %s %s > " % (
self . xmin , self . xmax , self . aty , str ( self . ticks ) , str ( self . labels ) , self . attr ) # XXX self.xmin/xmax undefd!
def __init__ ( self , xmin , xmax , aty = 0 , ticks = - 10 , miniticks = True , labels = True , logbase = None ,
arrow_start = None , arrow_end = None , exclude = None , text_attr = { } , * * attr ) :
self . aty = aty
tattr = dict ( self . text_defaults )
tattr . update ( text_attr )
LineAxis . __init__ ( self , xmin , aty , xmax , aty , xmin , xmax , ticks , miniticks , labels , logbase , arrow_start , arrow_end , exclude , tattr , * * attr )
def SVG ( self , trans = None ) :
""" Apply the transformation " trans " and return an SVG object. """
self . y1 = self . aty
self . y2 = self . aty
return LineAxis . SVG ( self , trans )
class YAxis ( LineAxis ) :
""" Draws a y axis with tick marks.
YAxis ( ymin , ymax , atx , ticks , miniticks , labels , logbase , arrow_start , arrow_end ,
exclude , text_attr , attribute = value )
ymin , ymax required the y range
atx default = 0 x position to draw the axis
ticks default = - 10 request ticks according to the standard
tick specification ( see help ( Ticks ) )
miniticks default = True request miniticks according to the
standard minitick specification
labels True request tick labels according to the
standard tick label specification
logbase default = None if a number , the y axis is logarithmic
with ticks at the given base ( usually 10 )
arrow_start default = None if a new string identifier , draw an arrow
at the low - end of the axis , referenced by
that identifier ; if an SVG marker object ,
use that marker
arrow_end default = None if a new string identifier , draw an arrow
at the high - end of the axis , referenced by
that identifier ; if an SVG marker object ,
use that marker
exclude default = None if a ( low , high ) pair , don ' t draw text
labels within this range
text_attr default = { } SVG attributes for the text labels
attribute = value pairs keyword list SVG attributes for all lines
The exclude option is provided for Axes to keep text from overlapping
where the axes cross . Normal users are not likely to need it .
"""
defaults = { " stroke-width " : " 0.25pt " , }
text_defaults = { " stroke " : " none " , " fill " : " black " , " font-size " : 5 , " text-anchor " : " end " , " dominant-baseline " : " middle " , }
text_start = 2.5
text_angle = 90.
def __repr__ ( self ) :
return " <YAxis ( %g , %g ) at x= %g ticks= %s labels= %s %s > " % (
self . ymin , self . ymax , self . atx , str ( self . ticks ) , str ( self . labels ) , self . attr ) # XXX self.ymin/ymax undefd!
def __init__ ( self , ymin , ymax , atx = 0 , ticks = - 10 , miniticks = True , labels = True , logbase = None ,
arrow_start = None , arrow_end = None , exclude = None , text_attr = { } , * * attr ) :
self . atx = atx
tattr = dict ( self . text_defaults )
tattr . update ( text_attr )
LineAxis . __init__ ( self , atx , ymin , atx , ymax , ymin , ymax , ticks , miniticks , labels , logbase , arrow_start , arrow_end , exclude , tattr , * * attr )
def SVG ( self , trans = None ) :
""" Apply the transformation " trans " and return an SVG object. """
self . x1 = self . atx
self . x2 = self . atx
return LineAxis . SVG ( self , trans )
class Axes :
""" Draw a pair of intersecting x-y axes.
Axes ( xmin , xmax , ymin , ymax , atx , aty , xticks , xminiticks , xlabels , xlogbase ,
yticks , yminiticks , ylabels , ylogbase , arrows , text_attr , attribute = value )
xmin , xmax required the x range
ymin , ymax required the y range
atx , aty default = 0 , 0 point where the axes try to cross ;
if outside the range , the axes will
cross at the closest corner
xticks default = - 10 request ticks according to the standard
tick specification ( see help ( Ticks ) )
xminiticks default = True request miniticks according to the
standard minitick specification
xlabels True request tick labels according to the
standard tick label specification
xlogbase default = None if a number , the x axis is logarithmic
with ticks at the given base ( usually 10 )
yticks default = - 10 request ticks according to the standard
tick specification
yminiticks default = True request miniticks according to the
standard minitick specification
ylabels True request tick labels according to the
standard tick label specification
ylogbase default = None if a number , the y axis is logarithmic
with ticks at the given base ( usually 10 )
arrows default = None if a new string identifier , draw arrows
referenced by that identifier
text_attr default = { } SVG attributes for the text labels
attribute = value pairs keyword list SVG attributes for all lines
"""
defaults = { " stroke-width " : " 0.25pt " , }
text_defaults = { " stroke " : " none " , " fill " : " black " , " font-size " : 5 , }
def __repr__ ( self ) :
return " <Axes x=( %g , %g ) y=( %g , %g ) at ( %g , %g ) %s > " % (
self . xmin , self . xmax , self . ymin , self . ymax , self . atx , self . aty , self . attr )
def __init__ ( self , xmin , xmax , ymin , ymax , atx = 0 , aty = 0 ,
xticks = - 10 , xminiticks = True , xlabels = True , xlogbase = None ,
yticks = - 10 , yminiticks = True , ylabels = True , ylogbase = None ,
arrows = None , text_attr = { } , * * attr ) :
self . xmin , self . xmax = xmin , xmax
self . ymin , self . ymax = ymin , ymax
self . atx , self . aty = atx , aty
self . xticks , self . xminiticks , self . xlabels , self . xlogbase = xticks , xminiticks , xlabels , xlogbase
self . yticks , self . yminiticks , self . ylabels , self . ylogbase = yticks , yminiticks , ylabels , ylogbase
self . arrows = arrows
self . text_attr = dict ( self . text_defaults )
self . text_attr . update ( text_attr )
self . attr = dict ( self . defaults )
self . attr . update ( attr )
def SVG ( self , trans = None ) :
""" Apply the transformation " trans " and return an SVG object. """
atx , aty = self . atx , self . aty
if atx < self . xmin :
atx = self . xmin
if atx > self . xmax :
atx = self . xmax
if aty < self . ymin :
aty = self . ymin
if aty > self . ymax :
aty = self . ymax
xmargin = 0.1 * abs ( self . ymin - self . ymax )
xexclude = atx - xmargin , atx + xmargin
ymargin = 0.1 * abs ( self . xmin - self . xmax )
yexclude = aty - ymargin , aty + ymargin
if self . arrows is not None and self . arrows != False :
xarrow_start = self . arrows + " .xstart "
xarrow_end = self . arrows + " .xend "
yarrow_start = self . arrows + " .ystart "
yarrow_end = self . arrows + " .yend "
else :
xarrow_start = xarrow_end = yarrow_start = yarrow_end = None
xaxis = XAxis ( self . xmin , self . xmax , aty , self . xticks , self . xminiticks , self . xlabels , self . xlogbase , xarrow_start , xarrow_end , exclude = xexclude , text_attr = self . text_attr , * * self . attr ) . SVG ( trans )
yaxis = YAxis ( self . ymin , self . ymax , atx , self . yticks , self . yminiticks , self . ylabels , self . ylogbase , yarrow_start , yarrow_end , exclude = yexclude , text_attr = self . text_attr , * * self . attr ) . SVG ( trans )
return SVG ( " g " , * ( xaxis . sub + yaxis . sub ) )
######################################################################
class HGrid ( Ticks ) :
""" Draws the horizontal lines of a grid over a specified region
using the standard tick specification ( see help ( Ticks ) ) to place the
grid lines .
HGrid ( xmin , xmax , low , high , ticks , miniticks , logbase , mini_attr , attribute = value )
xmin , xmax required the x range
low , high required the y range
ticks default = - 10 request ticks according to the standard
tick specification ( see help ( Ticks ) )
miniticks default = False request miniticks according to the
standard minitick specification
logbase default = None if a number , the axis is logarithmic
with ticks at the given base ( usually 10 )
mini_attr default = { } SVG attributes for the minitick - lines
( if miniticks != False )
attribute = value pairs keyword list SVG attributes for the major tick lines
"""
defaults = { " stroke-width " : " 0.25pt " , " stroke " : " gray " , }
mini_defaults = { " stroke-width " : " 0.25pt " , " stroke " : " lightgray " , " stroke-dasharray " : " 1,1 " , }
def __repr__ ( self ) :
return " <HGrid x=( %g , %g ) %g <= y <= %g ticks= %s miniticks= %s %s > " % (
self . xmin , self . xmax , self . low , self . high , str ( self . ticks ) , str ( self . miniticks ) , self . attr )
def __init__ ( self , xmin , xmax , low , high , ticks = - 10 , miniticks = False , logbase = None , mini_attr = { } , * * attr ) :
self . xmin , self . xmax = xmin , xmax
self . mini_attr = dict ( self . mini_defaults )
self . mini_attr . update ( mini_attr )
Ticks . __init__ ( self , None , low , high , ticks , miniticks , None , logbase )
self . attr = dict ( self . defaults )
self . attr . update ( attr )
def SVG ( self , trans = None ) :
""" Apply the transformation " trans " and return an SVG object. """
self . last_ticks , self . last_miniticks = Ticks . interpret ( self )
ticksd = [ ]
for t in self . last_ticks . keys ( ) :
ticksd + = Line ( self . xmin , t , self . xmax , t ) . Path ( trans ) . d
miniticksd = [ ]
for t in self . last_miniticks :
miniticksd + = Line ( self . xmin , t , self . xmax , t ) . Path ( trans ) . d
return SVG ( " g " , Path ( d = ticksd , * * self . attr ) . SVG ( ) , Path ( d = miniticksd , * * self . mini_attr ) . SVG ( ) )
class VGrid ( Ticks ) :
""" Draws the vertical lines of a grid over a specified region
using the standard tick specification ( see help ( Ticks ) ) to place the
grid lines .
HGrid ( ymin , ymax , low , high , ticks , miniticks , logbase , mini_attr , attribute = value )
ymin , ymax required the y range
low , high required the x range
ticks default = - 10 request ticks according to the standard
tick specification ( see help ( Ticks ) )
miniticks default = False request miniticks according to the
standard minitick specification
logbase default = None if a number , the axis is logarithmic
with ticks at the given base ( usually 10 )
mini_attr default = { } SVG attributes for the minitick - lines
( if miniticks != False )
attribute = value pairs keyword list SVG attributes for the major tick lines
"""
defaults = { " stroke-width " : " 0.25pt " , " stroke " : " gray " , }
mini_defaults = { " stroke-width " : " 0.25pt " , " stroke " : " lightgray " , " stroke-dasharray " : " 1,1 " , }
def __repr__ ( self ) :
return " <VGrid y=( %g , %g ) %g <= x <= %g ticks= %s miniticks= %s %s > " % (
self . ymin , self . ymax , self . low , self . high , str ( self . ticks ) , str ( self . miniticks ) , self . attr )
def __init__ ( self , ymin , ymax , low , high , ticks = - 10 , miniticks = False , logbase = None , mini_attr = { } , * * attr ) :
self . ymin , self . ymax = ymin , ymax
self . mini_attr = dict ( self . mini_defaults )
self . mini_attr . update ( mini_attr )
Ticks . __init__ ( self , None , low , high , ticks , miniticks , None , logbase )
self . attr = dict ( self . defaults )
self . attr . update ( attr )
def SVG ( self , trans = None ) :
""" Apply the transformation " trans " and return an SVG object. """
self . last_ticks , self . last_miniticks = Ticks . interpret ( self )
ticksd = [ ]
for t in self . last_ticks . keys ( ) :
ticksd + = Line ( t , self . ymin , t , self . ymax ) . Path ( trans ) . d
miniticksd = [ ]
for t in self . last_miniticks :
miniticksd + = Line ( t , self . ymin , t , self . ymax ) . Path ( trans ) . d
return SVG ( " g " , Path ( d = ticksd , * * self . attr ) . SVG ( ) , Path ( d = miniticksd , * * self . mini_attr ) . SVG ( ) )
class Grid ( Ticks ) :
""" Draws a grid over a specified region using the standard tick
specification ( see help ( Ticks ) ) to place the grid lines .
Grid ( xmin , xmax , ymin , ymax , ticks , miniticks , logbase , mini_attr , attribute = value )
xmin , xmax required the x range
ymin , ymax required the y range
ticks default = - 10 request ticks according to the standard
tick specification ( see help ( Ticks ) )
miniticks default = False request miniticks according to the
standard minitick specification
logbase default = None if a number , the axis is logarithmic
with ticks at the given base ( usually 10 )
mini_attr default = { } SVG attributes for the minitick - lines
( if miniticks != False )
attribute = value pairs keyword list SVG attributes for the major tick lines
"""
defaults = { " stroke-width " : " 0.25pt " , " stroke " : " gray " , }
mini_defaults = { " stroke-width " : " 0.25pt " , " stroke " : " lightgray " , " stroke-dasharray " : " 1,1 " , }
def __repr__ ( self ) :
return " <Grid x=( %g , %g ) y=( %g , %g ) ticks= %s miniticks= %s %s > " % (
self . xmin , self . xmax , self . ymin , self . ymax , str ( self . ticks ) , str ( self . miniticks ) , self . attr )
def __init__ ( self , xmin , xmax , ymin , ymax , ticks = - 10 , miniticks = False , logbase = None , mini_attr = { } , * * attr ) :
self . xmin , self . xmax = xmin , xmax
self . ymin , self . ymax = ymin , ymax
self . mini_attr = dict ( self . mini_defaults )
self . mini_attr . update ( mini_attr )
Ticks . __init__ ( self , None , None , None , ticks , miniticks , None , logbase )
self . attr = dict ( self . defaults )
self . attr . update ( attr )
def SVG ( self , trans = None ) :
""" Apply the transformation " trans " and return an SVG object. """
self . low , self . high = self . xmin , self . xmax
self . last_xticks , self . last_xminiticks = Ticks . interpret ( self )
self . low , self . high = self . ymin , self . ymax
self . last_yticks , self . last_yminiticks = Ticks . interpret ( self )
ticksd = [ ]
for t in self . last_xticks . keys ( ) :
ticksd + = Line ( t , self . ymin , t , self . ymax ) . Path ( trans ) . d
for t in self . last_yticks . keys ( ) :
ticksd + = Line ( self . xmin , t , self . xmax , t ) . Path ( trans ) . d
miniticksd = [ ]
for t in self . last_xminiticks :
miniticksd + = Line ( t , self . ymin , t , self . ymax ) . Path ( trans ) . d
for t in self . last_yminiticks :
miniticksd + = Line ( self . xmin , t , self . xmax , t ) . Path ( trans ) . d
return SVG ( " g " , Path ( d = ticksd , * * self . attr ) . SVG ( ) , Path ( d = miniticksd , * * self . mini_attr ) . SVG ( ) )
######################################################################
class XErrorBars :
""" Draws x error bars at a set of points. This is usually used
before ( under ) a set of Dots at the same points .
XErrorBars ( d , attribute = value )
d required list of ( x , y , xerr . . . ) points
attribute = value pairs keyword list SVG attributes
If points in d have
* 3 elements , the third is the symmetric error bar
* 4 elements , the third and fourth are the asymmetric lower and
upper error bar . The third element should be negative ,
e . g . ( 5 , 5 , - 1 , 2 ) is a bar from 4 to 7.
* more than 4 , a tick mark is placed at each value . This lets
you nest errors from different sources , correlated and
uncorrelated , statistical and systematic , etc .
"""
defaults = { " stroke-width " : " 0.25pt " , }
def __repr__ ( self ) :
return " <XErrorBars ( %d nodes)> " % len ( self . d )
def __init__ ( self , d = [ ] , * * attr ) :
self . d = list ( d )
self . attr = dict ( self . defaults )
self . attr . update ( attr )
def SVG ( self , trans = None ) :
""" Apply the transformation " trans " and return an SVG object. """
if isinstance ( trans , basestring ) :
trans = totrans ( trans ) # only once
output = SVG ( " g " )
for p in self . d :
x , y = p [ 0 ] , p [ 1 ]
if len ( p ) == 3 :
bars = [ x - p [ 2 ] , x + p [ 2 ] ]
else :
bars = [ x + pi for pi in p [ 2 : ] ]
start , end = min ( bars ) , max ( bars )
output . append ( LineAxis ( start , y , end , y , start , end , bars , False , False , * * self . attr ) . SVG ( trans ) )
return output
class YErrorBars :
""" Draws y error bars at a set of points. This is usually used
before ( under ) a set of Dots at the same points .
YErrorBars ( d , attribute = value )
d required list of ( x , y , yerr . . . ) points
attribute = value pairs keyword list SVG attributes
If points in d have
* 3 elements , the third is the symmetric error bar
* 4 elements , the third and fourth are the asymmetric lower and
upper error bar . The third element should be negative ,
e . g . ( 5 , 5 , - 1 , 2 ) is a bar from 4 to 7.
* more than 4 , a tick mark is placed at each value . This lets
you nest errors from different sources , correlated and
uncorrelated , statistical and systematic , etc .
"""
defaults = { " stroke-width " : " 0.25pt " , }
def __repr__ ( self ) :
return " <YErrorBars ( %d nodes)> " % len ( self . d )
def __init__ ( self , d = [ ] , * * attr ) :
self . d = list ( d )
self . attr = dict ( self . defaults )
self . attr . update ( attr )
def SVG ( self , trans = None ) :
""" Apply the transformation " trans " and return an SVG object. """
if isinstance ( trans , basestring ) :
trans = totrans ( trans ) # only once
output = SVG ( " g " )
for p in self . d :
x , y = p [ 0 ] , p [ 1 ]
if len ( p ) == 3 :
bars = [ y - p [ 2 ] , y + p [ 2 ] ]
else :
bars = [ y + pi for pi in p [ 2 : ] ]
start , end = min ( bars ) , max ( bars )
output . append ( LineAxis ( x , start , x , end , start , end , bars , False , False , * * self . attr ) . SVG ( trans ) )
return output