test/functional/test_arrow.py

   1 #!/usr/bin/env python
   2 import pyx
   3 from pyx import *
   4 from pyx.path import *
   5 import math
   6
   7 def testarrow(c):
   8     c.stroke(path(moveto(10,20),
   9                 curveto(12,16,14,15,12,19),
  10                 rcurveto(-3,2,3,3,-2,4)),
  11              [deco.barrow.small, deco.earrow.normal,
  12               deco.arrow(pos=0.5),
  13               deco.arrow(pos=0.7, reversed=1)])
  14
  15     c.stroke(path(arc(8,15,4,10,70)), [deco.barrow.small, deco.earrow.normal])
  16     c.stroke(path(arc(8,15,3,10,70)), [deco.barrow.small, deco.earrow.normal])
  17     c.stroke(path(arc(8,15,2,10,70)), [deco.barrow.small, deco.earrow.normal])
  18     c.stroke(path(arc(8,15,1,10,70)), [deco.barrow.small, deco.earrow.normal])
  19     c.stroke(path(arc(8,15,0.5,10,70)), [deco.barrow.small, deco.earrow.normal])
  20
  21     base = 2
  22
  23     pal = color.lineargradient_rgb(color.rgb.red, color.rgb.blue)
  24
  25     c.stroke(path(moveto(5,10), rlineto(5,0)),
  26            [#deco.barrow(size=base/math.sqrt(8)*unit.t_pt, constriction=1),
  27             deco.earrow.SMall,
  28             deco.colorgradient(pal),
  29             deco.text("start", arclenfrombegin=0, angle=90)])
  30     c.stroke(path(moveto(5,10.5), rlineto(5,0)),
  31            [deco.barrow(size=base/math.sqrt(4)*unit.t_pt, constriction=1),
  32             deco.earrow.Small,
  33             deco.colorgradient(pal),
  34             deco.text("start+1", arclenfrombegin=1, angle=90)])
  35     c.stroke(path(moveto(5,11), rlineto(5,0)),
  36            [deco.barrow(size=base/math.sqrt(2)*unit.t_pt, constriction=1),
  37             deco.earrow.small,
  38             deco.colorgradient(pal),
  39             deco.text("center", angle=90)])
  40     c.stroke(path(moveto(5,11.5), rlineto(5,0)),
  41            [deco.barrow(size=base/math.sqrt(1)*unit.t_pt, constriction=1),
  42             deco.earrow.normal,
  43             deco.colorgradient(pal),
  44             deco.text("end-1", arclenfromend=1, angle=90)])
  45     c.stroke(path(moveto(5,12), rlineto(5,0)),
  46            [deco.barrow(size=base*math.sqrt(2)*unit.t_pt, constriction=1),
  47             deco.earrow.large,
  48             deco.text("end", arclenfromend=0, angle=90)])
  49     c.stroke(path(moveto(5,12.5), rlineto(5,0)),
  50            [deco.barrow(size=base*math.sqrt(4)*unit.t_pt, constriction=1),
  51             deco.earrow.Large])
  52     c.stroke(path(moveto(5,13), rlineto(5,0)),
  53            [deco.barrow(size=base*math.sqrt(8)*unit.t_pt, constriction=1),
  54             deco.earrow.LArge])
  55     c.stroke(path(moveto(5,13.5), rlineto(5,0)),
  56            [deco.barrow(size=base*math.sqrt(16)*unit.t_pt, constriction=1),
  57             deco.earrow.LARge])
  58
  59     lt = style.linewidth.THick
  60
  61     c.stroke(path(moveto(11,10), rlineto(5,0)),
  62            [lt,
  63             deco.barrow(size=base/math.sqrt(8)*unit.t_pt, constriction=1),
  64             deco.earrow.SMall,
  65             deco.colorgradient(pal, [style.linewidth.THIN])])
  66     c.stroke(path(moveto(11,10.5), rlineto(5,0)),
  67            [lt,
  68             deco.barrow(size=base/math.sqrt(4)*unit.t_pt, constriction=1),
  69             deco.earrow.Small])
  70     c.stroke(path(moveto(11,11), rlineto(5,0)),
  71            [lt,
  72             deco.barrow(size=base/math.sqrt(2)*unit.t_pt, constriction=1),
  73             deco.earrow.small])
  74     c.stroke(path(moveto(11,11.5), rlineto(5,0)),
  75            [lt,
  76             deco.barrow(size=base/math.sqrt(1)*unit.t_pt, constriction=1),
  77             deco.earrow.normal])
  78     c.stroke(path(moveto(11,12), rlineto(5,0)),
  79            [lt,
  80             deco.barrow(size=base*math.sqrt(2)*unit.t_pt, constriction=1),
  81             deco.earrow.large])
  82     c.stroke(path(moveto(11,12.5), rlineto(5,0)),
  83            [lt,
  84             deco.barrow(size=base*math.sqrt(4)*unit.t_pt, constriction=1),
  85             deco.earrow.Large(attrs=[deco.stroked([style.linestyle.dashed]), deco.filled.clear])])
  86     c.stroke(path(moveto(11,13), rlineto(5,0)),
  87            [lt,
  88             deco.barrow(size=base*math.sqrt(8)*unit.t_pt, constriction=1),
  89             deco.earrow.LArge(attrs=[style.linestyle.dashed, color.rgb.green])])
  90     c.stroke(path(moveto(11,13.5), rlineto(5,0)),
  91            [lt,
  92             deco.barrow(size=base*math.sqrt(16)*unit.t_pt, constriction=1),
  93             deco.earrow.LARge(attrs=[color.rgb.red,
  94                                      deco.stroked([style.linejoin.round]),
  95                                      deco.filled([color.rgb.blue])])])
  96
  97 def linehatched(c):
  98     p = rect(0, 0, 5, 0.5).transformed(trafo.rotate(60))
  99     c.draw(p, [deco.linehatched45.normal])
 100     c.fill(p, [trafo.translate(2,0), deco.linehatched(0.1, 60, cross=1)]) # should not be filled, but hatched
 101     c.stroke(p, [trafo.translate(4,0), deco.linehatched90.normal])
 102     c.draw(p, [trafo.translate(6,0), deco.linehatched(0.1, 30, strokestyles=[color.rgb.red, style.linewidth.normal], cross=1)])
 103
 104 def testbrace(c):
 105     p = line(0, 0, 5, 0).transformed(trafo.rotate(60))
 106     c.stroke(p, [deco.brace(dist=-0.1, stretch=1.1), deco.text(r"hallo", textdist=20*unit.x_pt, relangle=-90), trafo.translate(8,0)])
 107
 108     tr = trafo.translate(12, 0)
 109     c.draw(line(0,0,0,4), [deco.leftbrace, tr])
 110     c.draw(line(4,0,4,4), [deco.rightbrace, tr])
 111     c.draw(line(0,0,4,0), [deco.belowbrace, tr])
 112     c.stroke(line(0,4,4,4), [deco.abovebrace, tr, style.linewidth.THin])
 113
 114     tr = trafo.translate(12.5, 0.5)
 115     c.stroke(line(0,0,3,0), [deco.straightbrace, tr, color.rgb.red])
 116     c.draw(line(3,0,3,3), [deco.straightbrace(fillattrs=[color.rgb.blue]), tr])
 117     c.draw(line(3,3,0,3), [deco.straightbrace, tr])
 118     c.draw(line(0,3,0,0), [deco.straightbrace, tr])
 119
 120 c = canvas.canvas()
 121 testarrow(c)
 122 linehatched(c)
 123 testbrace(c)
 124 c.writeEPSfile("test_arrow", page_paperformat=document.paperformat.A4, page_rotated=0, page_fittosize=1)
 125 c.writePDFfile("test_arrow", page_paperformat=document.paperformat.A4)
 126 c.writeSVGfile("test_arrow", page_paperformat=document.paperformat.A4)
 127