8000 Generate path strings in C++ for PDF and PS · shiliang/matplotlib@1f0e4ad · GitHub
[go: up one dir, main page]
Skip to content

Commit 1f0e4ad

Browse files
mdboommdboom
committed
Generate path strings in C++ for PDF and PS
1 parent ed0ce1a commit 1f0e4ad

File tree

6 files changed

+188
-109
lines changed

6 files changed

+188
-109
lines changed

‎lib/matplotlib/backends/backend_pdf.py

Lines changed: 17 additions & 27 deletions
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,6 @@
3737
from matplotlib.backends.backend_mixed import MixedModeRenderer
3838
from matplotlib.cbook import Bunch, is_string_like, \
3939
get_realpath_and_stat, is_writable_file_like, maxdict
40-
from matplotlib.mlab import quad2cubic
4140
from matplotlib.figure import Figure
4241
from matplotlib.font_manager import findfont, is_opentype_cff_font
4342
from matplotlib.afm import AFM
@@ -48,6 +47,7 @@
4847
from matplotlib.mathtext import MathTextParser
4948
from matplotlib.transforms import Affine2D, BboxBase
5049
from matplotlib.path import Path
50+
from matplotlib import _path
5151
from matplotlib import ttconv
5252

5353
# Overview
@@ -287,6 +287,17 @@ def __repr__(self):
287287
def pdfRepr(self):
288288
return self.op
289289

290+
291+
class Verbatim(object):
292+
"""Store verbatim PDF command content for later inclusion in the
293+
stream."""
294+
def __init__(self, x):
295+
self._x = x
296+
297+
def pdfRepr(self):
298+
return self._x
299+
300+
290301
# PDF operators (not an exhaustive list)
291302
_pdfops = dict(
292303
close_fill_stroke=b'b', fill_stroke=b'B', fill=b'f', closepath=b'h',
@@ -1388,32 +1399,11 @@ def writePathCollectionTemplates(self):
13881399

13891400
@staticmethod
13901401
def pathOperations(path, transform, clip=None, simplify=None, sketch=None):
1391-
cmds = []
1392-
last_points = None
1393-
for points, code in path.iter_segments(transform, clip=clip,
1394-
simplify=simplify,
1395-
sketch=sketch):
1396-
if code == Path.MOVETO:
1397-
# This is allowed anywhere in the path
1398-
cmds.extend(points)
1399- cmds.append(Op.moveto)
1400-
elif code == Path.CLOSEPOLY:
1401-
cmds.append(Op.closepath)
1402-
elif last_points is None:
1403-
# The other operations require a previous point
1404-
raise ValueError('Path lacks initial MOVETO')
1405-
elif code == Path.LINETO:
1406-
cmds.extend(points)
1407-
cmds.append(Op.lineto)
1408-
elif code == Path.CURVE3:
1409-
points = quad2cubic(*(list(last_points[-2:]) + list(points)))
1410-
cmds.extend(points[2:])
1411-
cmds.append(Op.curveto)
1412-
elif code == Path.CURVE4:
1413-
cmds.extend(points)
1414-
cmds.append(Op.curveto)
1415-
last_points = points
1416-
return cmds
1402+
return [Verbatim(_path.convert_to_string(
1403+
path, transform, clip, simplify, sketch,
1404+
6,
1405+
[Op.moveto.op, Op.lineto.op, b'', Op.curveto.op, Op.closepath.op],
1406+
True))]
14171407

14181408
def writePath(self, path, transform, clip=False, sketch=None):
14191409
if clip:

‎lib/matplotlib/backends/backend_ps.py

Lines changed: 4 additions & 22 deletions
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,6 @@ def _fn_name(): return sys._getframe(1).f_code.co_name
2626

2727
from matplotlib.cbook import is_string_like, get_realpath_and_stat, \
2828
is_writable_file_like, maxdict, file_requires_unicode
29-
from matplotlib.mlab import quad2cubic
3029
from matplotlib.figure import Figure
3130

3231
from matplotlib.font_manager import findfont, is_opentype_cff_font
@@ -36,6 +35,7 @@ def _fn_name(): return sys._getframe(1).f_code.co_name
3635
from matplotlib._mathtext_data import uni2type1
3736
from matplotlib.text import Text
3837
from matplotlib.path import Path
38+
from matplotlib import _path
3939
from matplotlib.transforms import Affine2D
4040

4141
from matplotlib.backends.backend_mixed import MixedModeRenderer
@@ -531,27 +531,9 @@ def _convert_path(self, path, transform, clip=False, simplify=None):
531531
self.height * 72.0)
532532
else:
533533
clip = None
534-
for points, code in path.iter_segments(transform, clip=clip,
535-
simplify=simplify):
536-
if code == Path.MOVETO:
537-
ps.append("%g %g m" % tuple(points))
538-
elif code == Path.CLOSEPOLY:
539-
ps.append("cl")
540-
elif last_points is None:
541-
# The other operations require a previous point
542-
raise ValueError('Path lacks initial MOVETO')
543-
elif code == Path.LINETO:
544-
ps.append("%g %g l" % tuple(points))
545-
elif code == Path.CURVE3:
546-
points = quad2cubic(*(list(last_points[-2:]) + list(points)))
547-
ps.append("%g %g %g %g %g %g c" %
548-
tuple(points[2:]))
549-
elif code == Path.CURVE4:
550-
ps.append("%g %g %g %g %g %g c" % tuple(points))
551-
last_points = points
552-
553-
ps = "\n".join(ps)
554-
return ps
534+
return _path.convert_to_string(
535+
path, transform, clip, simplify, None,
536+
6, [b'm', b'l', b'', b'c', b'cl'], True).decode('ascii')
555537

556538
def _get_clip_path(self, clippath, clippath_transform):
557539
key = (clippath, id(clippath_transform))

‎lib/matplotlib/backends/backend_svg.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -538,7 +538,9 @@ def _convert_path(self, path, transform=None, clip=None, simplify=None):
538538
clip = (0.0, 0.0, self.width, self.height)
539539
else:
540540
clip = None
541-
return _path.convert_to_svg(path, transform, clip, simplify, 6)
541+
return _path.convert_to_string(
542+
path, transform, clip, simplify, None, 6,
543+
[b'M', b'L', b'Q', b'C', b'z'], False).decode('ascii')
542544

543545
def draw_path(self, gc, path, transform, rgbFace=None):
544546
trans_and_flip = self._make_flip_transform(transform)

‎lib/matplotlib/mlab.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4007,6 +4007,8 @@ def quad2cubic(q0x, q0y, q1x, q1y, q2x, q2y):
40074007
points of a quadratic curve, and the output is a tuple of *x* and
40084008
*y* coordinates of the four control points of the cubic curve.
40094009
"""
4010+
# TODO: Candidate for deprecation -- no longer used internally
4011+
40104012
# c0x, c0y = q0x, q0y
40114013
c1x, c1y = q0x + 2./3. * (q1x - q0x), q0y + 2./3. * (q1y - q0y)
40124014
c2x, c2y = c1x + 1./3. * (q2x - q0x), c1y + 1./3. * (q2y - q0y)

‎src/_path.h

Lines changed: 123 additions & 51 deletions
Original file line numberDiff line numberDiff line change
@@ -939,79 +939,151 @@ void cleanup_path(PathIterator &path,
939939
}
940940
}
941941

942+
void quad2cubic(double x0, double y0,
943+
double x1, double y1,
944+
double x2, double y2,
945+
double *outx, double *outy)
946+
{
947+
948+
outx[0] = x0 + 2./3. * (x1 - x0);
949+
outy[0] = y0 + 2./3. * (y1 - y0);
950+
outx[1] = outx[0] + 1./3. * (x2 - x0);
951+
outy[1] = outy[0] + 1./3. * (y2 - y0);
952+
outx[2] = x2;
953+
outy[2] = y2;
954+
}
955+
956+
char *__append_to_string(char *p, char *buffer, size_t buffersize,
957+
const char *content)
958+
{
959+
for (const char *i = content; *i; ++i) {
960+
if (p < buffer || p - buffer >= (int)buffersize) {
961+
return p;
962+
}
963+
964+
*p++ = *i;
965+
}
966+
967+
return p;
968+
}
969+
942970
template <class PathIterator>
943-
void convert_to_svg(PathIterator &path,
944-
agg::trans_affine &trans,
945-
agg::rect_d &clip_rect,
946-
bool simplify,
947-
int precision,
948-
char *buffer,
949-
size_t *buffersize)
971+
void __convert_to_string(PathIterator &path,
972+
int precision,
973+
char **codes,
974+
bool postfix,
975+
char *buffer,
976+
size_t *buffersize)
950977
{
951978
#if PY_VERSION_HEX < 0x02070000
952979
char format[64];
953980
snprintf(format, 64, "%s.%dg", "%", precision);
954981
#endif
955982

956-
typedef agg::conv_transform<py::PathIterator> transformed_path_t;
957-
typedef PathNanRemover<transformed_path_t> nan_removal_t;
958-
typedef PathClipper<nan_removal_t> clipped_t;
959-
typedef PathSimplifier<clipped_t> simplify_t;
960-
961-
bool do_clip = (clip_rect.x1 < clip_rect.x2 && clip_rect.y1 < clip_rect.y2);
962-
963-
transformed_path_t tpath(path, trans);
964-
nan_removal_t nan_removed(tpath, true, path.has_curves());
965-
clipped_t clipped(nan_removed, do_clip, clip_rect);
966-
simplify_t simplified(clipped, simplify, path.simplify_threshold());
967-
968983
char *p = buffer;
984+
double x[3];
985+
double y[3];
986+
double last_x = 0.0;
987+
double last_y = 0.0;
969988

970-
const char codes[] = { 'M', 'L', 'Q', 'C' };
971-
const int waits[] = { 1, 1, 2, 3 };
972-
973-
int wait = 0;
989+
const int sizes[] = { 1, 1, 2, 3 };
990+
int size = 0;
974991
unsigned code;
975-
double x = 0, y = 0;
976-
while ((code = simplified.vertex(&x, &y)) != agg::path_cmd_stop) {
977-
if (wait == 0) {
978-
*p++ = '\n';
979992

980-
if (code == 0x4f) {
981-
*p++ = 'z';
982-
*p++ = '\n';
983-
continue;
993+
while ((code = path.vertex(&x[0], &y[0])) != agg::path_cmd_stop) {
994+
if (code == 0x4f) {
995+
p = __append_to_string(p, buffer, *buffersize, codes[4]);
996+
} else {
997+
size = sizes[code - 1];
998+
999+
for (int i = 1; i < size; ++i) {
1000+
path.vertex(&x[i], &y[i]);
9841001
}
9851002

986-
*p++ = codes[code - 1];
987-
wait = waits[code - 1];
988-
} else {
989-
*p++ = ' ';
990-
}
1003+
/* For formats that don't support quad curves, convert to
1004+
cubic curves */
1005+
if (codes[code - 1][0] == '\0') {
1006+
quad2cubic(last_x, last_y, x[0], y[0], x[1], y[1], x, y);
1007+
code++;
1008+
size = 3;
1009+
}
1010+
1011+
if (!postfix) {
1012+
p = __append_to_string(p, buffer, *buffersize, codes[code - 1]);
1013+
p = __append_to_string(p, buffer, *buffersize, " ");
1014+
}
9911015

1016+
for (int i = 0; i < size; ++i) {
9921017
#if PY_VERSION_HEX >= 0x02070000
993-
char *str;
994-
str = PyOS_double_to_string(x, 'g', precision, 0, NULL);
995-
p += snprintf(p, *buffersize - (p - buffer), "%s", str);
996-
PyMem_Free(str);
997-
*p++ = ' ';
998-
str = PyOS_double_to_string(y, 'g', precision, 0, NULL);
999-
p += snprintf(p, *buffersize - (p - buffer), "%s", str);
1000-
PyMem_Free(str);
1018+
char *str;
1019+
str = PyOS_double_to_string(x[i], 'g', precision, 0, NULL);
1020+
p = __append_to_string(p, buffer, *buffersize, str);
1021+
PyMem_Free(str);
1022+
p = __append_to_string(p, buffer, *buffersize, " ");
1023+
str = PyOS_double_to_string(y[i], 'g', precision, 0, NULL);
1024+
p = __append_to_string(p, buffer, *buffersize, str);
1025+
PyMem_Free(str);
1026+
p = __append_to_string(p, buffer, *buffersize, " ");
10011027
#else
1002-
char str[64];
1003-
PyOS_ascii_formatd(str, 64, format, x);
1004-
p += snprintf(p, *buffersize - (p - buffer), "%s", str);
1005-
*p++ = ' ';
1006-
PyOS_ascii_formatd(str, 64, format, y);
1007-
p += snprintf(p, *buffersize - (p - buffer), "%s", str);
1028+
char str[64];
1029+
PyOS_ascii_formatd(str, 64, format, x[i]);
1030+
p = __append_to_string(p, buffer, *buffersize, str);
1031+
p = __append_to_string(p, buffer, *buffersize, " ");
1032+
PyOS_ascii_formatd(str, 64, format, y[i]);
1033+
p = __append_to_string(p, buffer, *buffersize, str);
1034+
p = __append_to_string(p, buffer, *buffersize, " ");
10081035
#endif
1036+
}
1037+
1038+
if (postfix) {
1039+
p = __append_to_string(p, buffer, *buffersize, codes[code - 1]);
1040+
}
1041+
1042+
last_x = x[size - 1];
1043+
last_y = y[size - 1];
1044+
}
10091045

1010-
--wait;
1046+
p = __append_to_string(p, buffer, *buffersize, "\n");
10111047
}
10121048

10131049
*p = '\0';
10141050
*buffersize = p - buffer;
10151051
}
10161052

1053+
template <class PathIterator>
1054+
void convert_to_string(PathIterator &path,
1055+
agg::trans_affine &trans,
1056+
agg::rect_d &clip_rect,
1057+
bool simplify,
1058+
SketchParams sketch_params,
1059+
int precision,
1060+
char **codes,
1061+
bool postfix,
1062+
char *buffer,
1063+
size_t *buffersize)
1064+
{
1065+
typedef agg::conv_transform<py::PathIterator> transformed_path_t;
1066+
typedef PathNanRemover<transformed_path_t> nan_removal_t;
1067+
typedef PathClipper<nan_removal_t> clipped_t;
1068+
typedef PathSimplifier<clipped_t> simplify_t;
1069+
typedef agg::conv_curve<simplify_t> curve_t;
1070+
typedef Sketch<curve_t> sketch_t;
1071+
1072+
bool do_clip = (clip_rect.x1 < clip_rect.x2 && clip_rect.y1 < clip_rect.y2);
1073+
1074+
transformed_path_t tpath(path, trans);
1075+
nan_removal_t nan_removed(tpath, true, path.has_curves());
1076+
clipped_t clipped(nan_removed, do_clip, clip_rect);
1077+
simplify_t simplified(clipped, simplify, path.simplify_threshold());
1078+
1079+
if (sketch_params.scale == 0.0) {
1080+
__convert_to_string(simplified, precision, codes, postfix, buffer, buffersize);
1081+
} else {
1082+
curve_t curve(simplified);
1083+
sketch_t sketch(curve, sketch_params.scale, sketch_params.length, sketch_params.randomness);
1084+
__convert_to_string(sketch, precision, codes, postfix, buffer, buffersize);
1085+
}
1086+
1087+
}
1088+
10171089
#endif

0 commit comments

Comments
 (0)
0