def paint(self, canvas, data, axis, axispos):
if self.breaklinesattrs is not None:
breaklinesdist_pt = unit.topt(self.breaklinesdist)
breaklineslength_pt = unit.topt(self.breaklineslength)
breaklinesextent_pt = (0.5*breaklinesdist_pt*math.fabs(self.cos) +
0.5*breaklineslength_pt*math.fabs(self.sin))
if canvas.extent_pt < breaklinesextent_pt:
canvas.extent_pt = breaklinesextent_pt
for v in [data.subaxes[name].vminover for name in data.names[1:]]:
# use a tangent of the basepath (this is independent of the tickdirection)
p = axispos.vbasepath(v, None).normpath()
breakline = p.tangent(0, length=self.breaklineslength)
widthline = p.tangent(0, length=self.breaklinesdist).transformed(trafomodule.rotate(self.breaklinesangle+90, *breakline.atbegin()))
# XXX Uiiii
tocenter = map(lambda x: 0.5*(x[0]-x[1]), zip(breakline.atbegin(), breakline.atend()))
towidth = map(lambda x: 0.5*(x[0]-x[1]), zip(widthline.atbegin(), widthline.atend()))
breakline = breakline.transformed(trafomodule.translate(*tocenter).rotated(self.breaklinesangle, *breakline.atbegin()))
breakline1 = breakline.transformed(trafomodule.translate(*towidth))
breakline2 = breakline.transformed(trafomodule.translate(-towidth[0], -towidth[1]))
canvas.fill(path.path(path.moveto_pt(*breakline1.atbegin_pt()),
path.lineto_pt(*breakline1.atend_pt()),
path.lineto_pt(*breakline2.atend_pt()),
path.lineto_pt(*breakline2.atbegin_pt()),
path.closepath()), [color.gray.white])
canvas.stroke(breakline1, self.defaultbreaklinesattrs + self.breaklinesattrs)
canvas.stroke(breakline2, self.defaultbreaklinesattrs + self.breaklinesattrs)
_title.paint(self, canvas, data, axis, axispos)
def draw_dot(self, canvas, linewidth=0.3, color=None, dot=P(0, 0)):
if color is None:
color = self.color
p = pyx.path.path(pyx.path.moveto(*dot),
pyx.path.rlineto(*PP(0.1, -45)))
canvas.stroke(p.transformed(trafo.translate(*self.head)), [
style.linecap.round,
style.linewidth(linewidth), style.linejoin.round, color
])
def draw_extra(self, canvas, linewidth=0.3, color=None):
if color is None:
color = self.color
if self.drawn_extra == False:
for p in self.paths_extra:
canvas.stroke(p.transformed(trafo.translate(*self.head)), [
style.linecap.round,
style.linewidth(linewidth), style.linejoin.round, color
])
self.drawn_extra = True
def draw(self, canvas, linewidth=0.3, color=None):
if color is None:
color = self.color
for p in self.paths:
tmp = p.transformed(trafo.translate(*self.head))
canvas.stroke(tmp, [
style.linecap.round,
style.linewidth(linewidth), style.linejoin.round, color
])
self.drawn = True
def paint(self, canvas, data, axis, positioner):
namepos = []
for name in data.names:
subaxis = data.subaxes[name]
v = subaxis.vmin + self.namepos * (subaxis.vmax - subaxis.vmin)
x, y = positioner.vtickpoint_pt(v)
dx, dy = positioner.vtickdirection(v)
namepos.append((v, x, y, dx, dy))
nameboxes = []
if self.nameattrs is not None:
for (v, x, y, dx, dy), name in zip(namepos, data.names):
nameattrs = self.defaultnameattrs + self.nameattrs
if self.namedirection is not None:
nameattrs.append(
self.namedirection.trafo(tick.temp_dx, tick.temp_dy))
nameboxes.append(
canvas.texrunner.text_pt(x, y, str(name), nameattrs))
labeldist_pt = canvas.extent_pt + unit.topt(self.namedist)
if len(namepos) > 1:
equaldirection = 1
for np in namepos[1:]:
if np[3] != namepos[0][3] or np[4] != namepos[0][4]:
equaldirection = 0
else:
equaldirection = 0
if equaldirection and ((not namepos[0][3] and self.namevequalize) or
(not namepos[0][4] and self.namehequalize)):
box.linealignequal_pt(nameboxes, labeldist_pt, -namepos[0][3],
-namepos[0][4])
else:
for namebox, np in zip(nameboxes, namepos):
namebox.linealign_pt(labeldist_pt, -np[3], -np[4])
if self.basepathattrs is not None:
p = positioner.vbasepath()
if p is not None:
canvas.stroke(p,
self.defaultbasepathattrs + self.basepathattrs)
if (self.tickattrs is not None
and (self.innerticklength is not None
or self.outerticklength is not None)):
if self.innerticklength is not None:
innerticklength_pt = unit.topt(self.innerticklength)
if canvas.extent_pt < -innerticklength_pt:
canvas.extent_pt = -innerticklength_pt
elif self.outerticklength is not None:
innerticklength_pt = 0
if self.outerticklength is not None:
outerticklength_pt = unit.topt(self.outerticklength)
if canvas.extent_pt < outerticklength_pt:
canvas.extent_pt = outerticklength_pt
elif innerticklength_pt is not None:
outerticklength_pt = 0
for v in [data.subaxes[name].vminover
for name in data.names] + [1]:
x, y = positioner.vtickpoint_pt(v)
dx, dy = positioner.vtickdirection(v)
x1 = x + dx * innerticklength_pt
y1 = y + dy * innerticklength_pt
x2 = x - dx * outerticklength_pt
y2 = y - dy * outerticklength_pt
canvas.stroke(path.line_pt(x1, y1, x2, y2),
self.defaulttickattrs + self.tickattrs)
for (v, x, y, dx, dy), namebox in zip(namepos, nameboxes):
newextent_pt = namebox.extent_pt(dx, dy) + labeldist_pt
if canvas.extent_pt < newextent_pt:
canvas.extent_pt = newextent_pt
for namebox in nameboxes:
canvas.insert(namebox)
_title.paint(self, canvas, data, axis, positioner)
def paint(self, canvas, data, axis, axispos):
for t in data.ticks:
t.temp_v = axis.convert(data, t)
t.temp_x_pt, t.temp_y_pt = axispos.vtickpoint_pt(t.temp_v)
t.temp_dx, t.temp_dy = axispos.vtickdirection(t.temp_v)
maxticklevel, maxlabellevel = tick.maxlevels(data.ticks)
labeldist_pt = unit.topt(self.labeldist)
# create & align t.temp_labelbox
for t in data.ticks:
if t.labellevel is not None:
labelattrs = attr.selectattrs(self.labelattrs, t.labellevel,
maxlabellevel)
if labelattrs is not None:
labelattrs = self.defaultlabelattrs + labelattrs
if self.labeldirection is not None:
labelattrs.append(
self.labeldirection.trafo(t.temp_dx, t.temp_dy))
if t.labelattrs is not None:
labelattrs.extend(t.labelattrs)
t.temp_labelbox = canvas.texrunner.text_pt(
t.temp_x_pt, t.temp_y_pt, t.label, labelattrs)
if len(data.ticks) > 1:
equaldirection = 1
for t in data.ticks[1:]:
if t.temp_dx != data.ticks[
0].temp_dx or t.temp_dy != data.ticks[0].temp_dy:
equaldirection = 0
else:
equaldirection = 0
if equaldirection and (
(not data.ticks[0].temp_dx and self.labelvequalize) or
(not data.ticks[0].temp_dy and self.labelhequalize)):
if self.labelattrs is not None:
box.linealignequal_pt([
t.temp_labelbox
for t in data.ticks if t.labellevel is not None
], labeldist_pt, -data.ticks[0].temp_dx,
-data.ticks[0].temp_dy)
else:
for t in data.ticks:
if t.labellevel is not None and self.labelattrs is not None:
t.temp_labelbox.linealign_pt(labeldist_pt, -t.temp_dx,
-t.temp_dy)
for t in data.ticks:
if t.ticklevel is not None and self.tickattrs is not None:
tickattrs = attr.selectattrs(
self.defaulttickattrs + self.tickattrs, t.ticklevel,
maxticklevel)
if tickattrs is not None:
innerticklength = attr.selectattr(self.innerticklength,
t.ticklevel,
maxticklevel)
outerticklength = attr.selectattr(self.outerticklength,
t.ticklevel,
maxticklevel)
if innerticklength is not None or outerticklength is not None:
if innerticklength is None:
innerticklength = 0
if outerticklength is None:
outerticklength = 0
innerticklength_pt = unit.topt(innerticklength)
outerticklength_pt = unit.topt(outerticklength)
x1 = t.temp_x_pt + t.temp_dx * innerticklength_pt
y1 = t.temp_y_pt + t.temp_dy * innerticklength_pt
x2 = t.temp_x_pt - t.temp_dx * outerticklength_pt
y2 = t.temp_y_pt - t.temp_dy * outerticklength_pt
canvas.stroke(path.line_pt(x1, y1, x2, y2), tickattrs)
if outerticklength_pt > canvas.extent_pt:
canvas.extent_pt = outerticklength_pt
if -innerticklength_pt > canvas.extent_pt:
canvas.extent_pt = -innerticklength_pt
if self.gridattrs is not None:
gridattrs = attr.selectattrs(
self.defaultgridattrs + self.gridattrs, t.ticklevel,
maxticklevel)
if gridattrs is not None:
canvas.stroke(axispos.vgridpath(t.temp_v), gridattrs)
if t.labellevel is not None and self.labelattrs is not None:
canvas.insert(t.temp_labelbox)
canvas.labels.append(t.temp_labelbox)
extent_pt = t.temp_labelbox.extent_pt(t.temp_dx,
t.temp_dy) + labeldist_pt
if extent_pt > canvas.extent_pt:
canvas.extent_pt = extent_pt
if self.labelattrs is None:
canvas.labels = None
if self.basepathattrs is not None:
canvas.stroke(axispos.vbasepath(),
self.defaultbasepathattrs + self.basepathattrs)
# for t in data.ticks:
# del t.temp_v # we've inserted those temporary variables ... and do not care any longer about them
# del t.temp_x_pt
# del t.temp_y_pt
# del t.temp_dx
# del t.temp_dy
# if t.labellevel is not None and self.labelattrs is not None:
# del t.temp_labelbox
_title.paint(self, canvas, data, axis, axispos)
def paint(self, canvas, data, axis, positioner):
namepos = []
for name in data.names:
subaxis = data.subaxes[name]
v = subaxis.vmin + self.namepos * (subaxis.vmax - subaxis.vmin)
x, y = positioner.vtickpoint_pt(v)
dx, dy = positioner.vtickdirection(v)
namepos.append((v, x, y, dx, dy))
nameboxes = []
if self.nameattrs is not None:
for (v, x, y, dx, dy), name in zip(namepos, data.names):
nameattrs = self.defaultnameattrs + self.nameattrs
if self.namedirection is not None:
nameattrs.append(self.namedirection.trafo(dx, dy))
nameboxes.append(canvas.texrunner.text_pt(x, y, str(name), nameattrs))
labeldist_pt = canvas.extent_pt + unit.topt(self.namedist)
if len(namepos) > 1:
equaldirection = 1
for np in namepos[1:]:
if np[3] != namepos[0][3] or np[4] != namepos[0][4]:
equaldirection = 0
else:
equaldirection = 0
if equaldirection and ((not namepos[0][3] and self.namevequalize) or
(not namepos[0][4] and self.namehequalize)):
box.linealignequal_pt(nameboxes, labeldist_pt, -namepos[0][3], -namepos[0][4])
else:
for namebox, np in zip(nameboxes, namepos):
namebox.linealign_pt(labeldist_pt, -np[3], -np[4])
if self.basepathattrs is not None:
p = positioner.vbasepath()
if p is not None:
canvas.stroke(p, self.defaultbasepathattrs + self.basepathattrs)
if ( self.tickattrs is not None and
(self.innerticklength is not None or self.outerticklength is not None) ):
if self.innerticklength is not None:
innerticklength_pt = unit.topt(self.innerticklength)
if canvas.extent_pt < -innerticklength_pt:
canvas.extent_pt = -innerticklength_pt
elif self.outerticklength is not None:
innerticklength_pt = 0
if self.outerticklength is not None:
outerticklength_pt = unit.topt(self.outerticklength)
if canvas.extent_pt < outerticklength_pt:
canvas.extent_pt = outerticklength_pt
elif innerticklength_pt is not None:
outerticklength_pt = 0
for v in [data.subaxes[name].vminover for name in data.names] + [1]:
x, y = positioner.vtickpoint_pt(v)
dx, dy = positioner.vtickdirection(v)
x1 = x + dx * innerticklength_pt
y1 = y + dy * innerticklength_pt
x2 = x - dx * outerticklength_pt
y2 = y - dy * outerticklength_pt
canvas.stroke(path.line_pt(x1, y1, x2, y2), self.defaulttickattrs + self.tickattrs)
for (v, x, y, dx, dy), namebox in zip(namepos, nameboxes):
newextent_pt = namebox.extent_pt(dx, dy) + labeldist_pt
if canvas.extent_pt < newextent_pt:
canvas.extent_pt = newextent_pt
for namebox in nameboxes:
canvas.insert(namebox)
_title.paint(self, canvas, data, axis, positioner)
def paint(self, canvas, data, axis, axispos):
for t in data.ticks:
t.temp_v = axis.convert(data, t)
t.temp_x_pt, t.temp_y_pt = axispos.vtickpoint_pt(t.temp_v)
t.temp_dx, t.temp_dy = axispos.vtickdirection(t.temp_v)
maxticklevel, maxlabellevel = tick.maxlevels(data.ticks)
labeldist_pt = unit.topt(self.labeldist)
# create & align t.temp_labelbox
for t in data.ticks:
if t.labellevel is not None:
labelattrs = attr.selectattrs(self.labelattrs, t.labellevel, maxlabellevel)
if labelattrs is not None:
labelattrs = self.defaultlabelattrs + labelattrs
if self.labeldirection is not None:
labelattrs.append(self.labeldirection.trafo(t.temp_dx, t.temp_dy))
if t.labelattrs is not None:
labelattrs.extend(t.labelattrs)
t.temp_labelbox = canvas.texrunner.text_pt(t.temp_x_pt, t.temp_y_pt, t.label, labelattrs)
if len(data.ticks) > 1:
equaldirection = 1
for t in data.ticks[1:]:
if t.temp_dx != data.ticks[0].temp_dx or t.temp_dy != data.ticks[0].temp_dy:
equaldirection = 0
else:
equaldirection = 0
if equaldirection and ((not data.ticks[0].temp_dx and self.labelvequalize) or
(not data.ticks[0].temp_dy and self.labelhequalize)):
if self.labelattrs is not None:
box.linealignequal_pt([t.temp_labelbox for t in data.ticks if t.labellevel is not None],
labeldist_pt, -data.ticks[0].temp_dx, -data.ticks[0].temp_dy)
else:
for t in data.ticks:
if t.labellevel is not None and self.labelattrs is not None:
t.temp_labelbox.linealign_pt(labeldist_pt, -t.temp_dx, -t.temp_dy)
for t in data.ticks:
if t.ticklevel is not None and self.tickattrs is not None:
tickattrs = attr.selectattrs(self.defaulttickattrs + self.tickattrs, t.ticklevel, maxticklevel)
if tickattrs is not None:
innerticklength = attr.selectattr(self.innerticklength, t.ticklevel, maxticklevel)
outerticklength = attr.selectattr(self.outerticklength, t.ticklevel, maxticklevel)
if innerticklength is not None or outerticklength is not None:
if innerticklength is None:
innerticklength = 0
if outerticklength is None:
outerticklength = 0
innerticklength_pt = unit.topt(innerticklength)
outerticklength_pt = unit.topt(outerticklength)
x1 = t.temp_x_pt + t.temp_dx * innerticklength_pt
y1 = t.temp_y_pt + t.temp_dy * innerticklength_pt
x2 = t.temp_x_pt - t.temp_dx * outerticklength_pt
y2 = t.temp_y_pt - t.temp_dy * outerticklength_pt
canvas.stroke(path.line_pt(x1, y1, x2, y2), tickattrs)
if outerticklength_pt > canvas.extent_pt:
canvas.extent_pt = outerticklength_pt
if -innerticklength_pt > canvas.extent_pt:
canvas.extent_pt = -innerticklength_pt
if self.gridattrs is not None:
gridattrs = attr.selectattrs(self.defaultgridattrs + self.gridattrs, t.ticklevel, maxticklevel)
if gridattrs is not None:
canvas.stroke(axispos.vgridpath(t.temp_v), gridattrs)
if t.labellevel is not None and self.labelattrs is not None:
canvas.insert(t.temp_labelbox)
canvas.labels.append(t.temp_labelbox)
extent_pt = t.temp_labelbox.extent_pt(t.temp_dx, t.temp_dy) + labeldist_pt
if extent_pt > canvas.extent_pt:
canvas.extent_pt = extent_pt
if self.labelattrs is None:
canvas.labels = None
if self.basepathattrs is not None:
canvas.stroke(axispos.vbasepath(), self.defaultbasepathattrs + self.basepathattrs)
# for t in data.ticks:
# del t.temp_v # we've inserted those temporary variables ... and do not care any longer about them
# del t.temp_x_pt
# del t.temp_y_pt
# del t.temp_dx
# del t.temp_dy
# if t.labellevel is not None and self.labelattrs is not None:
# del t.temp_labelbox
_title.paint(self, canvas, data, axis, axispos)
