Raise an error if co-ordinates are incorrectly ordered

This commit is contained in:
Andrew Murray 2023-03-01 11:22:35 +11:00
parent 6e9c0ae5a0
commit b84c29a035
6 changed files with 66 additions and 24 deletions

Binary file not shown.

Before

Width:  |  Height:  |  Size: 21 KiB

After

Width:  |  Height:  |  Size: 21 KiB

Binary file not shown.

Before

Width:  |  Height:  |  Size: 20 KiB

After

Width:  |  Height:  |  Size: 20 KiB

View File

@ -355,7 +355,13 @@ def ellipse_various_sizes_helper(filled):
for w in ellipse_sizes: for w in ellipse_sizes:
y = 1 y = 1
for h in ellipse_sizes: for h in ellipse_sizes:
border = [x, y, x + w - 1, y + h - 1] x1 = x + w
if w:
x1 -= 1
y1 = y + h
if h:
y1 -= 1
border = [x, y, x1, y1]
if filled: if filled:
draw.ellipse(border, fill="white") draw.ellipse(border, fill="white")
else: else:
@ -932,9 +938,6 @@ def test_square():
img, draw = create_base_image_draw((10, 10)) img, draw = create_base_image_draw((10, 10))
draw.rectangle((2, 2, 7, 7), BLACK) draw.rectangle((2, 2, 7, 7), BLACK)
assert_image_equal_tofile(img, expected, "square as normal rectangle failed") assert_image_equal_tofile(img, expected, "square as normal rectangle failed")
img, draw = create_base_image_draw((10, 10))
draw.rectangle((7, 7, 2, 2), BLACK)
assert_image_equal_tofile(img, expected, "square as inverted rectangle failed")
def test_triangle_right(): def test_triangle_right():
@ -1499,3 +1502,20 @@ def test_polygon2():
draw.polygon([(18, 30), (19, 31), (18, 30), (85, 30), (60, 72)], "red") draw.polygon([(18, 30), (19, 31), (18, 30), (85, 30), (60, 72)], "red")
expected = "Tests/images/imagedraw_outline_polygon_RGB.png" expected = "Tests/images/imagedraw_outline_polygon_RGB.png"
assert_image_similar_tofile(im, expected, 1) assert_image_similar_tofile(im, expected, 1)
def test_incorrectly_ordered_coordinates():
im = Image.new("RGB", (W, H))
draw = ImageDraw.Draw(im)
with pytest.raises(ValueError):
draw.arc((1, 1, 0, 0), 10, 260)
with pytest.raises(ValueError):
draw.chord((1, 1, 0, 0), 10, 260)
with pytest.raises(ValueError):
draw.ellipse((1, 1, 0, 0))
with pytest.raises(ValueError):
draw.pieslice((1, 1, 0, 0), 10, 260)
with pytest.raises(ValueError):
draw.rectangle((1, 1, 0, 0))
with pytest.raises(ValueError):
draw.rounded_rectangle((1, 1, 0, 0))

View File

@ -303,6 +303,12 @@ class ImageDraw:
(x0, y0), (x1, y1) = xy (x0, y0), (x1, y1) = xy
else: else:
x0, y0, x1, y1 = xy x0, y0, x1, y1 = xy
if x1 < x0 or y1 < y0:
msg = (
"x1 must be greater than or equal to x0,"
" and y1 must be greater than or equal to y0"
)
raise ValueError(msg)
if corners is None: if corners is None:
corners = (True, True, True, True) corners = (True, True, True, True)

View File

@ -251,6 +251,8 @@ PyImaging_GetBuffer(PyObject *buffer, Py_buffer *view) {
static const char *must_be_sequence = "argument must be a sequence"; static const char *must_be_sequence = "argument must be a sequence";
static const char *must_be_two_coordinates = static const char *must_be_two_coordinates =
"coordinate list must contain exactly 2 coordinates"; "coordinate list must contain exactly 2 coordinates";
static const char *incorrectly_ordered_coordinates =
"x1 must be greater than or equal to x0, and y1 must be greater than or equal to y0";
static const char *wrong_mode = "unrecognized image mode"; static const char *wrong_mode = "unrecognized image mode";
static const char *wrong_raw_mode = "unrecognized raw mode"; static const char *wrong_raw_mode = "unrecognized raw mode";
static const char *outside_image = "image index out of range"; static const char *outside_image = "image index out of range";
@ -2805,6 +2807,11 @@ _draw_arc(ImagingDrawObject *self, PyObject *args) {
free(xy); free(xy);
return NULL; return NULL;
} }
if (xy[2] < xy[0] || xy[3] < xy[1]) {
PyErr_SetString(PyExc_ValueError, incorrectly_ordered_coordinates);
free(xy);
return NULL;
}
n = ImagingDrawArc( n = ImagingDrawArc(
self->image->image, self->image->image,
@ -2886,6 +2893,11 @@ _draw_chord(ImagingDrawObject *self, PyObject *args) {
free(xy); free(xy);
return NULL; return NULL;
} }
if (xy[2] < xy[0] || xy[3] < xy[1]) {
PyErr_SetString(PyExc_ValueError, incorrectly_ordered_coordinates);
free(xy);
return NULL;
}
n = ImagingDrawChord( n = ImagingDrawChord(
self->image->image, self->image->image,
@ -2932,6 +2944,11 @@ _draw_ellipse(ImagingDrawObject *self, PyObject *args) {
free(xy); free(xy);
return NULL; return NULL;
} }
if (xy[2] < xy[0] || xy[3] < xy[1]) {
PyErr_SetString(PyExc_ValueError, incorrectly_ordered_coordinates);
free(xy);
return NULL;
}
n = ImagingDrawEllipse( n = ImagingDrawEllipse(
self->image->image, self->image->image,
@ -3101,6 +3118,11 @@ _draw_pieslice(ImagingDrawObject *self, PyObject *args) {
free(xy); free(xy);
return NULL; return NULL;
} }
if (xy[2] < xy[0] || xy[3] < xy[1]) {
PyErr_SetString(PyExc_ValueError, incorrectly_ordered_coordinates);
free(xy);
return NULL;
}
n = ImagingDrawPieslice( n = ImagingDrawPieslice(
self->image->image, self->image->image,
@ -3197,6 +3219,11 @@ _draw_rectangle(ImagingDrawObject *self, PyObject *args) {
free(xy); free(xy);
return NULL; return NULL;
} }
if (xy[2] < xy[0] || xy[3] < xy[1]) {
PyErr_SetString(PyExc_ValueError, incorrectly_ordered_coordinates);
free(xy);
return NULL;
}
n = ImagingDrawRectangle( n = ImagingDrawRectangle(
self->image->image, self->image->image,

View File

@ -85,25 +85,22 @@ point32(Imaging im, int x, int y, int ink) {
static inline void static inline void
point32rgba(Imaging im, int x, int y, int ink) { point32rgba(Imaging im, int x, int y, int ink) {
unsigned int tmp1; unsigned int tmp;
if (x >= 0 && x < im->xsize && y >= 0 && y < im->ysize) { if (x >= 0 && x < im->xsize && y >= 0 && y < im->ysize) {
UINT8 *out = (UINT8 *)im->image[y] + x * 4; UINT8 *out = (UINT8 *)im->image[y] + x * 4;
UINT8 *in = (UINT8 *)&ink; UINT8 *in = (UINT8 *)&ink;
out[0] = BLEND(in[3], out[0], in[0], tmp1); out[0] = BLEND(in[3], out[0], in[0], tmp);
out[1] = BLEND(in[3], out[1], in[1], tmp1); out[1] = BLEND(in[3], out[1], in[1], tmp);
out[2] = BLEND(in[3], out[2], in[2], tmp1); out[2] = BLEND(in[3], out[2], in[2], tmp);
} }
} }
static inline void static inline void
hline8(Imaging im, int x0, int y0, int x1, int ink) { hline8(Imaging im, int x0, int y0, int x1, int ink) {
int tmp, pixelwidth; int pixelwidth;
if (y0 >= 0 && y0 < im->ysize) { if (y0 >= 0 && y0 < im->ysize) {
if (x0 > x1) {
tmp = x0, x0 = x1, x1 = tmp;
}
if (x0 < 0) { if (x0 < 0) {
x0 = 0; x0 = 0;
} else if (x0 >= im->xsize) { } else if (x0 >= im->xsize) {
@ -126,13 +123,9 @@ hline8(Imaging im, int x0, int y0, int x1, int ink) {
static inline void static inline void
hline32(Imaging im, int x0, int y0, int x1, int ink) { hline32(Imaging im, int x0, int y0, int x1, int ink) {
int tmp;
INT32 *p; INT32 *p;
if (y0 >= 0 && y0 < im->ysize) { if (y0 >= 0 && y0 < im->ysize) {
if (x0 > x1) {
tmp = x0, x0 = x1, x1 = tmp;
}
if (x0 < 0) { if (x0 < 0) {
x0 = 0; x0 = 0;
} else if (x0 >= im->xsize) { } else if (x0 >= im->xsize) {
@ -152,13 +145,9 @@ hline32(Imaging im, int x0, int y0, int x1, int ink) {
static inline void static inline void
hline32rgba(Imaging im, int x0, int y0, int x1, int ink) { hline32rgba(Imaging im, int x0, int y0, int x1, int ink) {
int tmp; unsigned int tmp;
unsigned int tmp1;
if (y0 >= 0 && y0 < im->ysize) { if (y0 >= 0 && y0 < im->ysize) {
if (x0 > x1) {
tmp = x0, x0 = x1, x1 = tmp;
}
if (x0 < 0) { if (x0 < 0) {
x0 = 0; x0 = 0;
} else if (x0 >= im->xsize) { } else if (x0 >= im->xsize) {
@ -173,9 +162,9 @@ hline32rgba(Imaging im, int x0, int y0, int x1, int ink) {
UINT8 *out = (UINT8 *)im->image[y0] + x0 * 4; UINT8 *out = (UINT8 *)im->image[y0] + x0 * 4;
UINT8 *in = (UINT8 *)&ink; UINT8 *in = (UINT8 *)&ink;
while (x0 <= x1) { while (x0 <= x1) {
out[0] = BLEND(in[3], out[0], in[0], tmp1); out[0] = BLEND(in[3], out[0], in[0], tmp);
out[1] = BLEND(in[3], out[1], in[1], tmp1); out[1] = BLEND(in[3], out[1], in[1], tmp);
out[2] = BLEND(in[3], out[2], in[2], tmp1); out[2] = BLEND(in[3], out[2], in[2], tmp);
x0++; x0++;
out += 4; out += 4;
} }