#include "mplutils.h"
#include "py_converters.h"
#include "_backend_agg.h"
typedef struct
{
PyObject_HEAD
RendererAgg *x;
Py_ssize_t shape[3];
Py_ssize_t strides[3];
Py_ssize_t suboffsets[3];
} PyRendererAgg;
typedef struct
{
PyObject_HEAD
BufferRegion *x;
Py_ssize_t shape[3];
Py_ssize_t strides[3];
Py_ssize_t suboffsets[3];
} PyBufferRegion;
/**********************************************************************
* BufferRegion
* */
static PyObject *PyBufferRegion_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
{
PyBufferRegion *self;
self = (PyBufferRegion *)type->tp_alloc(type, 0);
self->x = NULL;
return (PyObject *)self;
}
static void PyBufferRegion_dealloc(PyBufferRegion *self)
{
delete self->x;
Py_TYPE(self)->tp_free((PyObject *)self);
}
static PyObject *PyBufferRegion_to_string(PyBufferRegion *self, PyObject *args, PyObject *kwds)
{
return PyBytes_FromStringAndSize((const char *)self->x->get_data(),
self->x->get_height() * self->x->get_stride());
}
/* TODO: This doesn't seem to be used internally. Remove? */
static PyObject *PyBufferRegion_set_x(PyBufferRegion *self, PyObject *args, PyObject *kwds)
{
int x;
if (!PyArg_ParseTuple(args, "i:set_x", &x)) {
return NULL;
}
self->x->get_rect().x1 = x;
Py_RETURN_NONE;
}
static PyObject *PyBufferRegion_set_y(PyBufferRegion *self, PyObject *args, PyObject *kwds)
{
int y;
if (!PyArg_ParseTuple(args, "i:set_y", &y)) {
return NULL;
}
self->x->get_rect().y1 = y;
Py_RETURN_NONE;
}
static PyObject *PyBufferRegion_get_extents(PyBufferRegion *self, PyObject *args, PyObject *kwds)
{
agg::rect_i rect = self->x->get_rect();
return Py_BuildValue("IIII", rect.x1, rect.y1, rect.x2, rect.y2);
}
static PyObject *PyBufferRegion_to_string_argb(PyBufferRegion *self, PyObject *args, PyObject *kwds)
{
PyObject *bufobj;
uint8_t *buf;
bufobj = PyBytes_FromStringAndSize(NULL, self->x->get_height() * self->x->get_stride());
buf = (uint8_t *)PyBytes_AS_STRING(bufobj);
CALL_CPP_CLEANUP("to_string_argb", (self->x->to_string_argb(buf)), Py_DECREF(bufobj));
return bufobj;
}
int PyBufferRegion_get_buffer(PyBufferRegion *self, Py_buffer *buf, int flags)
{
Py_INCREF(self);
buf->obj = (PyObject *)self;
buf->buf = self->x->get_data();
buf->len = self->x->get_width() * self->x->get_height() * 4;
buf->readonly = 0;
buf->format = (char *)"B";
buf->ndim = 3;
self->shape[0] = self->x->get_height();
self->shape[1] = self->x->get_width();
self->shape[2] = 4;
buf->shape = self->shape;
self->strides[0] = self->x->get_width() * 4;
self->strides[1] = 4;
self->strides[2] = 1;
buf->strides = self->strides;
buf->suboffsets = NULL;
buf->itemsize = 1;
buf->internal = NULL;
return 1;
}
static PyTypeObject PyBufferRegionType;
static PyTypeObject *PyBufferRegion_init_type(PyObject *m, PyTypeObject *type)
{
static PyMethodDef methods[] = {
{ "to_string", (PyCFunction)PyBufferRegion_to_string, METH_NOARGS, NULL },
{ "to_string_argb", (PyCFunction)PyBufferRegion_to_string_argb, METH_NOARGS, NULL },
{ "set_x", (PyCFunction)PyBufferRegion_set_x, METH_VARARGS, NULL },
{ "set_y", (PyCFunction)PyBufferRegion_set_y, METH_VARARGS, NULL },
{ "get_extents", (PyCFunction)PyBufferRegion_get_extents, METH_NOARGS, NULL },
{ NULL }
};
static PyBufferProcs buffer_procs;
memset(&buffer_procs, 0, sizeof(PyBufferProcs));
buffer_procs.bf_getbuffer = (getbufferproc)PyBufferRegion_get_buffer;
memset(type, 0, sizeof(PyTypeObject));
type->tp_name = "matplotlib.backends._backend_agg.BufferRegion";
type->tp_basicsize = sizeof(PyBufferRegion);
type->tp_dealloc = (destructor)PyBufferRegion_dealloc;
type->tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE | Py_TPFLAGS_HAVE_NEWBUFFER;
type->tp_methods = methods;
type->tp_new = PyBufferRegion_new;
type->tp_as_buffer = &buffer_procs;
if (PyType_Ready(type) < 0) {
return NULL;
}
/* Don't need to add to module, since you can't create buffer
regions directly from Python */
return type;
}
/**********************************************************************
* RendererAgg
* */
static PyObject *PyRendererAgg_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
{
PyRendererAgg *self;
self = (PyRendererAgg *)type->tp_alloc(type, 0);
self->x = NULL;
return (PyObject *)self;
}
static int PyRendererAgg_init(PyRendererAgg *self, PyObject *args, PyObject *kwds)
{
unsigned int width;
unsigned int height;
double dpi;
int debug = 0;
if (!PyArg_ParseTuple(args, "IId|i:RendererAgg", &width, &height, &dpi, &debug)) {
return -1;
}
if (dpi <= 0.0) {
PyErr_SetString(PyExc_ValueError, "dpi must be positive");
return -1;
}
if (width >= 1 << 16 || height >= 1 << 16) {
PyErr_Format(
PyExc_ValueError,
"Image size of %dx%d pixels is too large. "
"It must be less than 2^16 in each direction.",
width, height);
return -1;
}
CALL_CPP_INIT("RendererAgg", self->x = new RendererAgg(width, height, dpi))
return 0;
}
static void PyRendererAgg_dealloc(PyRendererAgg *self)
{
delete self->x;
Py_TYPE(self)->tp_free((PyObject *)self);
}
static PyObject *PyRendererAgg_draw_path(PyRendererAgg *self, PyObject *args, PyObject *kwds)
{
GCAgg gc;
py::PathIterator path;
agg::trans_affine trans;
PyObject *faceobj = NULL;
agg::rgba face;
if (!PyArg_ParseTuple(args,
"O&O&O&|O:draw_path",
&convert_gcagg,
&gc,
&convert_path,
&path,
&convert_trans_affine,
&trans,
&faceobj)) {
return NULL;
}
if (!convert_face(faceobj, gc, &face)) {
return NULL;
}
CALL_CPP("draw_path", (self->x->draw_path(gc, path, trans, face)));
Py_RETURN_NONE;
}
static PyObject *PyRendererAgg_draw_text_image(PyRendererAgg *self, PyObject *args, PyObject *kwds)
{
numpy::array_view<agg::int8u, 2> image;
double x;
double y;
double angle;
GCAgg gc;
if (!PyArg_ParseTuple(args,
"O&dddO&:draw_text_image",
&image.converter_contiguous,
&image,
&x,
&y,
&angle,
&convert_gcagg,
&gc)) {
return NULL;
}
CALL_CPP("draw_text_image", (self->x->draw_text_image(gc, image, x, y, angle)));
Py_RETURN_NONE;
}
PyObject *PyRendererAgg_draw_markers(PyRendererAgg *self, PyObject *args, PyObject *kwds)
{
GCAgg gc;
py::PathIterator marker_path;
agg::trans_affine marker_path_trans;
py::PathIterator path;
agg::trans_affine trans;
PyObject *faceobj = NULL;
agg::rgba face;
if (!PyArg_ParseTuple(args,
"O&O&O&O&O&|O:draw_markers",
&convert_gcagg,
&gc,
&convert_path,
&marker_path,
&convert_trans_affine,
&marker_path_trans,
&convert_path,
&path,
&convert_trans_affine,
&trans,
&faceobj)) {
return NULL;
}
if (!convert_face(faceobj, gc, &face)) {
return NULL;
}
CALL_CPP("draw_markers",
(self->x->draw_markers(gc, marker_path, marker_path_trans, path, trans, face)));
Py_RETURN_NONE;
}
static PyObject *PyRendererAgg_draw_image(PyRendererAgg *self, PyObject *args, PyObject *kwds)
{
GCAgg gc;
double x;
double y;
numpy::array_view<agg::int8u, 3> image;
if (!PyArg_ParseTuple(args,
"O&ddO&:draw_image",
&convert_gcagg,
&gc,
&x,
&y,
&image.converter_contiguous,
&image)) {
return NULL;
}
x = mpl_round(x);
y = mpl_round(y);
gc.alpha = 1.0;
CALL_CPP("draw_image", (self->x->draw_image(gc, x, y, image)));
Py_RETURN_NONE;
}
static PyObject *
PyRendererAgg_draw_path_collection(PyRendererAgg *self, PyObject *args, PyObject *kwds)
{
GCAgg gc;
agg::trans_affine master_transform;
PyObject *pathobj;
numpy::array_view<const double, 3> transforms;
numpy::array_view<const double, 2> offsets;
agg::trans_affine offset_trans;
numpy::array_view<const double, 2> facecolors;
numpy::array_view<const double, 2> edgecolors;
numpy::array_view<const double, 1> linewidths;
DashesVector dashes;
numpy::array_view<const uint8_t, 1> antialiaseds;
PyObject *ignored;
e_offset_position offset_position;
if (!PyArg_ParseTuple(args,
"O&O&OO&O&O&O&O&O&O&O&OO&:draw_path_collection",
&convert_gcagg,
&gc,
&convert_trans_affine,
&master_transform,
&pathobj,
&convert_transforms,
&transforms,
&convert_points,
&offsets,
&convert_trans_affine,
&offset_trans,
&convert_colors,
&facecolors,
&convert_colors,
&edgecolors,
&linewidths.converter,
&linewidths,
&convert_dashes_vector,
&dashes,
&antialiaseds.converter,
&antialiaseds,
&ignored,
&convert_offset_position,
&offset_position)) {
return NULL;
}
try
{
py::PathGenerator path(pathobj);
CALL_CPP("draw_path_collection",
(self->x->draw_path_collection(gc,
master_transform,
path,
transforms,
offsets,
offset_trans,
facecolors,
edgecolors,
linewidths,
dashes,
antialiaseds,
offset_position)));
}
catch (const py::exception &)
{
return NULL;
}
Py_RETURN_NONE;
}
static PyObject *PyRendererAgg_draw_quad_mesh(PyRendererAgg *self, PyObject *args, PyObject *kwds)
{
GCAgg gc;
agg::trans_affine master_transform;
unsigned int mesh_width;
unsigned int mesh_height;
numpy::array_view<const double, 3> coordinates;
numpy::array_view<const double, 2> offsets;
agg::trans_affine offset_trans;
numpy::array_view<const double, 2> facecolors;
int antialiased;
numpy::array_view<const double, 2> edgecolors;
if (!PyArg_ParseTuple(args,
"O&O&IIO&O&O&O&iO&:draw_quad_mesh",
&convert_gcagg,
&gc,
&convert_trans_affine,
&master_transform,
&mesh_width,
&mesh_height,
&coordinates.converter,
&coordinates,
&convert_points,
&offsets,
&convert_trans_affine,
&offset_trans,
&convert_colors,
&facecolors,
&antialiased,
&convert_colors,
&edgecolors)) {
return NULL;
}
CALL_CPP("draw_quad_mesh",
(self->x->draw_quad_mesh(gc,
master_transform,
mesh_width,
mesh_height,
coordinates,
offsets,
offset_trans,
facecolors,
antialiased,
edgecolors)));
Py_RETURN_NONE;
}
static PyObject *
PyRendererAgg_draw_gouraud_triangle(PyRendererAgg *self, PyObject *args, PyObject *kwds)
{
GCAgg gc;
numpy::array_view<const double, 2> points;
numpy::array_view<const double, 2> colors;
agg::trans_affine trans;
if (!PyArg_ParseTuple(args,
"O&O&O&O&|O:draw_gouraud_triangle",
&convert_gcagg,
&gc,
&points.converter,
&points,
&colors.converter,
&colors,
&convert_trans_affine,
&trans)) {
return NULL;
}
if (points.dim(0) != 3 || points.dim(1) != 2) {
PyErr_Format(PyExc_ValueError,
"points must be a 3x2 array, got %" NPY_INTP_FMT "x%" NPY_INTP_FMT,
points.dim(0), points.dim(1));
return NULL;
}
if (colors.dim(0) != 3 || colors.dim(1) != 4) {
PyErr_Format(PyExc_ValueError,
"colors must be a 3x4 array, got %" NPY_INTP_FMT "x%" NPY_INTP_FMT,
colors.dim(0), colors.dim(1));
return NULL;
}
CALL_CPP("draw_gouraud_triangle", (self->x->draw_gouraud_triangle(gc, points, colors, trans)));
Py_RETURN_NONE;
}
static PyObject *
PyRendererAgg_draw_gouraud_triangles(PyRendererAgg *self, PyObject *args, PyObject *kwds)
{
GCAgg gc;
numpy::array_view<const double, 3> points;
numpy::array_view<const double, 3> colors;
agg::trans_affine trans;
if (!PyArg_ParseTuple(args,
"O&O&O&O&|O:draw_gouraud_triangles",
&convert_gcagg,
&gc,
&points.converter,
&points,
&colors.converter,
&colors,
&convert_trans_affine,
&trans)) {
return NULL;
}
if (points.size() != 0 && (points.dim(1) != 3 || points.dim(2) != 2)) {
PyErr_Format(PyExc_ValueError,
"points must be a Nx3x2 array, got %" NPY_INTP_FMT "x%" NPY_INTP_FMT "x%" NPY_INTP_FMT,
points.dim(0), points.dim(1), points.dim(2));
return NULL;
}
if (colors.size() != 0 && (colors.dim(1) != 3 || colors.dim(2) != 4)) {
PyErr_Format(PyExc_ValueError,
"colors must be a Nx3x4 array, got %" NPY_INTP_FMT "x%" NPY_INTP_FMT "x%" NPY_INTP_FMT,
colors.dim(0), colors.dim(1), colors.dim(2));
return NULL;
}
if (points.size() != colors.size()) {
PyErr_Format(PyExc_ValueError,
"points and colors arrays must be the same length, got %" NPY_INTP_FMT " and %" NPY_INTP_FMT,
points.dim(0), colors.dim(0));
return NULL;
}
CALL_CPP("draw_gouraud_triangles", self->x->draw_gouraud_triangles(gc, points, colors, trans));
Py_RETURN_NONE;
}
static PyObject *PyRendererAgg_tostring_rgb(PyRendererAgg *self, PyObject *args, PyObject *kwds)
{
PyObject *buffobj = NULL;
buffobj = PyBytes_FromStringAndSize(NULL, self->x->get_width() * self->x->get_height() * 3);
if (buffobj == NULL) {
return NULL;
}
CALL_CPP_CLEANUP("tostring_rgb",
(self->x->tostring_rgb((uint8_t *)PyBytes_AS_STRING(buffobj))),
Py_DECREF(buffobj));
return buffobj;
}
static PyObject *PyRendererAgg_tostring_argb(PyRendererAgg *self, PyObject *args, PyObject *kwds)
{
PyObject *buffobj = NULL;
buffobj = PyBytes_FromStringAndSize(NULL, self->x->get_width() * self->x->get_height() * 4);
if (buffobj == NULL) {
return NULL;
}
CALL_CPP_CLEANUP("tostring_argb",
(self->x->tostring_argb((uint8_t *)PyBytes_AS_STRING(buffobj))),
Py_DECREF(buffobj));
return buffobj;
}
static PyObject *PyRendererAgg_tostring_bgra(PyRendererAgg *self, PyObject *args, PyObject *kwds)
{
PyObject *buffobj = NULL;
buffobj = PyBytes_FromStringAndSize(NULL, self->x->get_width() * self->x->get_height() * 4);
if (buffobj == NULL) {
return NULL;
}
CALL_CPP_CLEANUP("to_string_bgra",
(self->x->tostring_bgra((uint8_t *)PyBytes_AS_STRING(buffobj))),
Py_DECREF(buffobj));
return buffobj;
}
static PyObject *
PyRendererAgg_get_content_extents(PyRendererAgg *self, PyObject *args, PyObject *kwds)
{
agg::rect_i extents;
CALL_CPP("get_content_extents", (extents = self->x->get_content_extents()));
return Py_BuildValue(
"iiii", extents.x1, extents.y1, extents.x2 - extents.x1, extents.y2 - extents.y1);
}
static PyObject *PyRendererAgg_buffer_rgba(PyRendererAgg *self, PyObject *args, PyObject *kwds)
{
#if PY3K
return PyBytes_FromStringAndSize((const char *)self->x->pixBuffer,
self->x->get_width() * self->x->get_height() * 4);
#else
return PyBuffer_FromReadWriteMemory(self->x->pixBuffer,
self->x->get_width() * self->x->get_height() * 4);
#endif
}
int PyRendererAgg_get_buffer(PyRendererAgg *self, Py_buffer *buf, int flags)
{
Py_INCREF(self);
buf->obj = (PyObject *)self;
buf->buf = self->x->pixBuffer;
buf->len = self->x->get_width() * self->x->get_height() * 4;
buf->readonly = 0;
buf->format = (char *)"B";
buf->ndim = 3;
self->shape[0] = self->x->get_height();
self->shape[1] = self->x->get_width();
self->shape[2] = 4;
buf->shape = self->shape;
self->strides[0] = self->x->get_width() * 4;
self->strides[1] = 4;
self->strides[2] = 1;
buf->strides = self->strides;
buf->suboffsets = NULL;
buf->itemsize = 1;
buf->internal = NULL;
return 1;
}
static PyObject *PyRendererAgg_clear(PyRendererAgg *self, PyObject *args, PyObject *kwds)
{
CALL_CPP("clear", self->x->clear());
Py_RETURN_NONE;
}
static PyObject *PyRendererAgg_copy_from_bbox(PyRendererAgg *self, PyObject *args, PyObject *kwds)
{
agg::rect_d bbox;
BufferRegion *reg;
PyObject *regobj;
if (!PyArg_ParseTuple(args, "O&:copy_from_bbox", &convert_rect, &bbox)) {
return 0;
}
CALL_CPP("copy_from_bbox", (reg = self->x->copy_from_bbox(bbox)));
regobj = PyBufferRegion_new(&PyBufferRegionType, NULL, NULL);
((PyBufferRegion *)regobj)->x = reg;
return regobj;
}
static PyObject *PyRendererAgg_restore_region(PyRendererAgg *self, PyObject *args, PyObject *kwds)
{
PyBufferRegion *regobj;
int xx1 = 0, yy1 = 0, xx2 = 0, yy2 = 0, x = 0, y = 0;
if (!PyArg_ParseTuple(args,
"O!|iiiiii:restore_region",
&PyBufferRegionType,
®obj,
&xx1,
&yy1,
&xx2,
&yy2,
&x,
&y)) {
return 0;
}
if (PySequence_Size(args) == 1) {
CALL_CPP("restore_region", (self->x->restore_region(*(regobj->x))));
} else {
CALL_CPP("restore_region", self->x->restore_region(*(regobj->x), xx1, yy1, xx2, yy2, x, y));
}
Py_RETURN_NONE;
}
PyTypeObject PyRendererAggType;
static PyTypeObject *PyRendererAgg_init_type(PyObject *m, PyTypeObject *type)
{
static PyMethodDef methods[] = {
{"draw_path", (PyCFunction)PyRendererAgg_draw_path, METH_VARARGS, NULL},
{"draw_markers", (PyCFunction)PyRendererAgg_draw_markers, METH_VARARGS, NULL},
{"draw_text_image", (PyCFunction)PyRendererAgg_draw_text_image, METH_VARARGS, NULL},
{"draw_image", (PyCFunction)PyRendererAgg_draw_image, METH_VARARGS, NULL},
{"draw_path_collection", (PyCFunction)PyRendererAgg_draw_path_collection, METH_VARARGS, NULL},
{"draw_quad_mesh", (PyCFunction)PyRendererAgg_draw_quad_mesh, METH_VARARGS, NULL},
{"draw_gouraud_triangle", (PyCFunction)PyRendererAgg_draw_gouraud_triangle, METH_VARARGS, NULL},
{"draw_gouraud_triangles", (PyCFunction)PyRendererAgg_draw_gouraud_triangles, METH_VARARGS, NULL},
{"tostring_rgb", (PyCFunction)PyRendererAgg_tostring_rgb, METH_NOARGS, NULL},
{"tostring_argb", (PyCFunction)PyRendererAgg_tostring_argb, METH_NOARGS, NULL},
{"tostring_bgra", (PyCFunction)PyRendererAgg_tostring_bgra, METH_NOARGS, NULL},
{"get_content_extents", (PyCFunction)PyRendererAgg_get_content_extents, METH_NOARGS, NULL},
{"buffer_rgba", (PyCFunction)PyRendererAgg_buffer_rgba, METH_NOARGS, NULL},
{"clear", (PyCFunction)PyRendererAgg_clear, METH_NOARGS, NULL},
{"copy_from_bbox", (PyCFunction)PyRendererAgg_copy_from_bbox, METH_VARARGS, NULL},
{"restore_region", (PyCFunction)PyRendererAgg_restore_region, METH_VARARGS, NULL},
{NULL}
};
static PyBufferProcs buffer_procs;
memset(&buffer_procs, 0, sizeof(PyBufferProcs));
buffer_procs.bf_getbuffer = (getbufferproc)PyRendererAgg_get_buffer;
memset(type, 0, sizeof(PyTypeObject));
type->tp_name = "matplotlib.backends._backend_agg.RendererAgg";
type->tp_basicsize = sizeof(PyRendererAgg);
type->tp_dealloc = (destructor)PyRendererAgg_dealloc;
type->tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE | Py_TPFLAGS_HAVE_NEWBUFFER;
type->tp_methods = methods;
type->tp_init = (initproc)PyRendererAgg_init;
type->tp_new = PyRendererAgg_new;
type->tp_as_buffer = &buffer_procs;
if (PyType_Ready(type) < 0) {
return NULL;
}
if (PyModule_AddObject(m, "RendererAgg", (PyObject *)type)) {
return NULL;
}
return type;
}
extern "C" {
#if PY3K
static struct PyModuleDef moduledef = {
PyModuleDef_HEAD_INIT,
"_backend_agg",
NULL,
0,
NULL,
NULL,
NULL,
NULL,
NULL
};
#define INITERROR return NULL
PyMODINIT_FUNC PyInit__backend_agg(void)
#else
#define INITERROR return
PyMODINIT_FUNC init_backend_agg(void)
#endif
{
PyObject *m;
#if PY3K
m = PyModule_Create(&moduledef);
#else
m = Py_InitModule3("_backend_agg", NULL, NULL);
#endif
if (m == NULL) {
INITERROR;
}
import_array();
if (!PyRendererAgg_init_type(m, &PyRendererAggType)) {
INITERROR;
}
if (!PyBufferRegion_init_type(m, &PyBufferRegionType)) {
INITERROR;
}
#if PY3K
return m;
#endif
}
} // extern "C"