1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
|
# -*- encoding: utf-8 -*-
#
#
# Copyright (C) 2005-2006 André Wobst <wobsta@pyx-project.org>
#
# This file is part of PyX (https://pyx-project.org/).
#
# PyX 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.
#
# PyX 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 PyX; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA
import math
from pyx import path, unit
class _positioner:
"""interface definition of axis tick position methods
- these methods are used for the postitioning of the ticks
when painting an axis"""
# TODO: should we add a local transformation (for label text etc?)
# (this might replace tickdirection (and even tickposition?))
def vbasepath(self, v1=None, v2=None):
"""return the basepath as a path
- like basepath, but for graph coordinates"""
def vgridpath(self, v):
"""return the gridpath as a path for a given position v
in graph coordinates
- might return None when no gridpath is available"""
return None
def vtickpoint_pt(self, v):
"""return tick position (x, y) in pts for a tick at position v in graph coordinates"""
def vtickdirection(self, v):
"""return direction tuple (dx, dy) for a tick at position v in graph coordinates"""
class pathpositioner(_positioner):
"""axis tick position methods along an arbitrary path"""
def __init__(self, p, direction=1):
self.path = p
self.normpath = p.normpath()
self.arclen_pt = self.normpath.arclen_pt()
self.arclen = self.arclen_pt * unit.t_pt
self.direction = direction
def vbasepath(self, v1=None, v2=None):
if v1 is None:
if v2 is None:
return self.path
else:
return self.normpath.split(self.normpath.arclentoparam(v2 * self.arclen))[0]
else:
if v2 is None:
return self.normpath.split(self.normpath.arclentoparam(v1 * self.arclen))[1]
else:
return self.normpath.split(*self.normpath.arclentoparam([v1 * self.arclen, v2 * self.arclen]))[1]
def vtickpoint_pt(self, v):
return self.normpath.at_pt(self.normpath.arclentoparam(v * self.arclen))
def vtickdirection(self, v):
return self.normpath.rotation(self.normpath.arclentoparam(v * self.arclen)).apply_pt(0, self.direction)
class lineaxispos_pt:
"""an axispos linear along a line with a fix direction for the ticks"""
def __init__(self, x1_pt, y1_pt, x2_pt, y2_pt, fixtickdirection, vgridpath):
self.x1_pt = x1_pt
self.y1_pt = y1_pt
self.x2_pt = x2_pt
self.y2_pt = y2_pt
self.fixtickdirection = fixtickdirection
self.vgridpath = vgridpath
def vbasepath(self, v1=None, v2=None):
if v1 is None:
v1 = 0
if v2 is None:
v2 = 1
return path.line_pt((1-v1)*self.x1_pt+v1*self.x2_pt,
(1-v1)*self.y1_pt+v1*self.y2_pt,
(1-v2)*self.x1_pt+v2*self.x2_pt,
(1-v2)*self.y1_pt+v2*self.y2_pt)
def vtickpoint_pt(self, v):
return (1-v)*self.x1_pt+v*self.x2_pt, (1-v)*self.y1_pt+v*self.y2_pt
def vtickdirection(self, v):
return self.fixtickdirection
class flexlineaxispos_pt(lineaxispos_pt):
"""an axispos linear along a line with flexible direction for the ticks"""
def __init__(self, vtickpoint_pt, vtickdirection, vgridpath):
self.vtickpoint_pt = vtickpoint_pt
self.vtickdirection = vtickdirection
self.vgridpath = vgridpath
def vbasepath(self, v1=None, v2=None):
if v1 is None:
v1 = 0
if v2 is None:
v2 = 1
x1_pt, y1_pt = self.vtickpoint_pt(v1)
x2_pt, y2_pt = self.vtickpoint_pt(v2)
return path.line_pt(x1_pt, y1_pt, x2_pt, y2_pt)
|
