531 lines
24 KiB
C
531 lines
24 KiB
C
/*
|
||
* This file is part of the Micro Python project, http://micropython.org/
|
||
*
|
||
* The MIT License (MIT)
|
||
*
|
||
* Copyright (c) 2021 Kevin Matocha
|
||
*
|
||
* Permission is hereby granted, free of charge, to any person obtaining a copy
|
||
* of this software and associated documentation files (the "Software"), to deal
|
||
* in the Software without restriction, including without limitation the rights
|
||
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||
* copies of the Software, and to permit persons to whom the Software is
|
||
* furnished to do so, subject to the following conditions:
|
||
*
|
||
* The above copyright notice and this permission notice shall be included in
|
||
* all copies or substantial portions of the Software.
|
||
*
|
||
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
||
* THE SOFTWARE.
|
||
*/
|
||
|
||
#include "shared-bindings/displayio/Bitmap.h"
|
||
#include "shared-bindings/bitmaptools/__init__.h"
|
||
|
||
#include <stdint.h>
|
||
|
||
#include "py/binary.h"
|
||
#include "py/obj.h"
|
||
#include "py/runtime.h"
|
||
|
||
//| """Collection of bitmap manipulation tools"""
|
||
//|
|
||
|
||
STATIC int16_t validate_point(mp_obj_t point, int16_t default_value) {
|
||
// Checks if point is None and returns default_value, otherwise decodes integer value
|
||
if (point == mp_const_none) {
|
||
return default_value;
|
||
}
|
||
return mp_obj_get_int(point);
|
||
}
|
||
|
||
STATIC void extract_tuple(mp_obj_t xy_tuple, int16_t *x, int16_t *y, int16_t x_default, int16_t y_default) {
|
||
// Helper function for rotozoom
|
||
// Extract x,y values from a tuple or default if None
|
||
if (xy_tuple == mp_const_none) {
|
||
*x = x_default;
|
||
*y = y_default;
|
||
} else if (!mp_obj_is_obj(xy_tuple)) {
|
||
mp_raise_ValueError(translate("clip point must be (x,y) tuple"));
|
||
} else {
|
||
mp_obj_t *items;
|
||
mp_obj_get_array_fixed_n(xy_tuple, 2, &items);
|
||
*x = mp_obj_get_int(items[0]);
|
||
*y = mp_obj_get_int(items[1]);
|
||
}
|
||
}
|
||
|
||
STATIC void validate_clip_region(displayio_bitmap_t *bitmap, mp_obj_t clip0_tuple, int16_t *clip0_x, int16_t *clip0_y,
|
||
mp_obj_t clip1_tuple, int16_t *clip1_x, int16_t *clip1_y) {
|
||
// Helper function for rotozoom
|
||
// 1. Extract the clip x,y points from the two clip tuples
|
||
// 2. Rearrange values such that clip0_ < clip1_
|
||
// 3. Constrain the clip points to within the bitmap
|
||
|
||
extract_tuple(clip0_tuple, clip0_x, clip0_y, 0, 0);
|
||
extract_tuple(clip1_tuple, clip1_x, clip1_y, bitmap->width, bitmap->height);
|
||
|
||
// Ensure the value for clip0 is less than clip1 (for both x and y)
|
||
if (*clip0_x > *clip1_x) {
|
||
int16_t temp_value = *clip0_x; // swap values
|
||
*clip0_x = *clip1_x;
|
||
*clip1_x = temp_value;
|
||
}
|
||
if (*clip0_y > *clip1_y) {
|
||
int16_t temp_value = *clip0_y; // swap values
|
||
*clip0_y = *clip1_y;
|
||
*clip1_y = temp_value;
|
||
}
|
||
|
||
// Constrain the clip window to within the bitmap boundaries
|
||
if (*clip0_x < 0) {
|
||
*clip0_x = 0;
|
||
}
|
||
if (*clip0_y < 0) {
|
||
*clip0_y = 0;
|
||
}
|
||
if (*clip0_x > bitmap->width) {
|
||
*clip0_x = bitmap->width;
|
||
}
|
||
if (*clip0_y > bitmap->height) {
|
||
*clip0_y = bitmap->height;
|
||
}
|
||
if (*clip1_x < 0) {
|
||
*clip1_x = 0;
|
||
}
|
||
if (*clip1_y < 0) {
|
||
*clip1_y = 0;
|
||
}
|
||
if (*clip1_x > bitmap->width) {
|
||
*clip1_x = bitmap->width;
|
||
}
|
||
if (*clip1_y > bitmap->height) {
|
||
*clip1_y = bitmap->height;
|
||
}
|
||
|
||
}
|
||
|
||
//|
|
||
//| def rotozoom(
|
||
//| dest_bitmap: displayio.Bitmap, source_bitmap: displayio.Bitmap,
|
||
//| *,
|
||
//| ox: int, oy: int, dest_clip0: Tuple[int, int], dest_clip1: Tuple[int, int],
|
||
//| px: int, py: int, source_clip0: Tuple[int, int], source_clip1: Tuple[int, int],
|
||
//| angle: float, scale: float, skip_index: int) -> None:
|
||
//| """Inserts the source bitmap region into the destination bitmap with rotation
|
||
//| (angle), scale and clipping (both on source and destination bitmaps).
|
||
//|
|
||
//| :param bitmap dest_bitmap: Destination bitmap that will be copied into
|
||
//| :param bitmap source_bitmap: Source bitmap that contains the graphical region to be copied
|
||
//| :param int ox: Horizontal pixel location in destination bitmap where source bitmap
|
||
//| point (px,py) is placed
|
||
//| :param int oy: Vertical pixel location in destination bitmap where source bitmap
|
||
//| point (px,py) is placed
|
||
//| :param Tuple[int,int] dest_clip0: First corner of rectangular destination clipping
|
||
//| region that constrains region of writing into destination bitmap
|
||
//| :param Tuple[int,int] dest_clip1: Second corner of rectangular destination clipping
|
||
//| region that constrains region of writing into destination bitmap
|
||
//| :param int px: Horizontal pixel location in source bitmap that is placed into the
|
||
//| destination bitmap at (ox,oy)
|
||
//| :param int py: Vertical pixel location in source bitmap that is placed into the
|
||
//| destination bitmap at (ox,oy)
|
||
//| :param Tuple[int,int] source_clip0: First corner of rectangular source clipping
|
||
//| region that constrains region of reading from the source bitmap
|
||
//| :param Tuple[int,int] source_clip1: Second corner of rectangular source clipping
|
||
//| region that constrains region of reading from the source bitmap
|
||
//| :param float angle: Angle of rotation, in radians (positive is clockwise direction)
|
||
//| :param float scale: Scaling factor
|
||
//| :param int skip_index: Bitmap palette index in the source that will not be copied,
|
||
//| set to None to copy all pixels"""
|
||
//| ...
|
||
//|
|
||
STATIC mp_obj_t bitmaptools_obj_rotozoom(size_t n_args, const mp_obj_t *pos_args, mp_map_t *kw_args) {
|
||
enum {ARG_dest_bitmap, ARG_source_bitmap,
|
||
ARG_ox, ARG_oy, ARG_dest_clip0, ARG_dest_clip1,
|
||
ARG_px, ARG_py, ARG_source_clip0, ARG_source_clip1,
|
||
ARG_angle, ARG_scale, ARG_skip_index};
|
||
|
||
static const mp_arg_t allowed_args[] = {
|
||
{MP_QSTR_dest_bitmap, MP_ARG_REQUIRED | MP_ARG_OBJ},
|
||
{MP_QSTR_source_bitmap, MP_ARG_REQUIRED | MP_ARG_OBJ},
|
||
|
||
{MP_QSTR_ox, MP_ARG_KW_ONLY | MP_ARG_OBJ, {.u_obj = mp_const_none} }, // None convert to destination->width / 2
|
||
{MP_QSTR_oy, MP_ARG_KW_ONLY | MP_ARG_OBJ, {.u_obj = mp_const_none} }, // None convert to destination->height / 2
|
||
{MP_QSTR_dest_clip0, MP_ARG_KW_ONLY | MP_ARG_OBJ, {.u_obj = mp_const_none} },
|
||
{MP_QSTR_dest_clip1, MP_ARG_KW_ONLY | MP_ARG_OBJ, {.u_obj = mp_const_none} },
|
||
|
||
{MP_QSTR_px, MP_ARG_KW_ONLY | MP_ARG_OBJ, {.u_obj = mp_const_none} }, // None convert to source->width / 2
|
||
{MP_QSTR_py, MP_ARG_KW_ONLY | MP_ARG_OBJ, {.u_obj = mp_const_none} }, // None convert to source->height / 2
|
||
{MP_QSTR_source_clip0, MP_ARG_KW_ONLY | MP_ARG_OBJ, {.u_obj = mp_const_none} },
|
||
{MP_QSTR_source_clip1, MP_ARG_KW_ONLY | MP_ARG_OBJ, {.u_obj = mp_const_none} },
|
||
|
||
{MP_QSTR_angle, MP_ARG_KW_ONLY | MP_ARG_OBJ, {.u_obj = mp_const_none} }, // None convert to 0.0
|
||
{MP_QSTR_scale, MP_ARG_KW_ONLY | MP_ARG_OBJ, {.u_obj = mp_const_none} }, // None convert to 1.0
|
||
{MP_QSTR_skip_index, MP_ARG_OBJ | MP_ARG_KW_ONLY, {.u_obj = mp_const_none} },
|
||
};
|
||
|
||
mp_arg_val_t args[MP_ARRAY_SIZE(allowed_args)];
|
||
mp_arg_parse_all(n_args, pos_args, kw_args, MP_ARRAY_SIZE(allowed_args), allowed_args, args);
|
||
|
||
displayio_bitmap_t *destination = MP_OBJ_TO_PTR(args[ARG_dest_bitmap].u_obj); // the destination bitmap
|
||
|
||
displayio_bitmap_t *source = MP_OBJ_TO_PTR(args[ARG_source_bitmap].u_obj); // the source bitmap
|
||
|
||
// ensure that the destination bitmap has at least as many `bits_per_value` as the source
|
||
if (destination->bits_per_value < source->bits_per_value) {
|
||
mp_raise_ValueError(translate("source palette too large"));
|
||
}
|
||
|
||
// Confirm the destination location target (ox,oy); if None, default to bitmap midpoint
|
||
int16_t ox, oy;
|
||
ox = validate_point(args[ARG_ox].u_obj, destination->width / 2);
|
||
oy = validate_point(args[ARG_oy].u_obj, destination->height / 2);
|
||
|
||
// Confirm the source location target (px,py); if None, default to bitmap midpoint
|
||
int16_t px, py;
|
||
px = validate_point(args[ARG_px].u_obj, source->width / 2);
|
||
py = validate_point(args[ARG_py].u_obj, source->height / 2);
|
||
|
||
// Validate the clipping regions for the destination bitmap
|
||
int16_t dest_clip0_x, dest_clip0_y, dest_clip1_x, dest_clip1_y;
|
||
|
||
validate_clip_region(destination, args[ARG_dest_clip0].u_obj, &dest_clip0_x, &dest_clip0_y,
|
||
args[ARG_dest_clip1].u_obj, &dest_clip1_x, &dest_clip1_y);
|
||
|
||
// Validate the clipping regions for the source bitmap
|
||
int16_t source_clip0_x, source_clip0_y, source_clip1_x, source_clip1_y;
|
||
|
||
validate_clip_region(source, args[ARG_source_clip0].u_obj, &source_clip0_x, &source_clip0_y,
|
||
args[ARG_source_clip1].u_obj, &source_clip1_x, &source_clip1_y);
|
||
|
||
// Confirm the angle value
|
||
float angle = 0.0;
|
||
if (args[ARG_angle].u_obj != mp_const_none) {
|
||
angle = mp_obj_get_float(args[ARG_angle].u_obj);
|
||
}
|
||
|
||
// Confirm the scale value
|
||
float scale = 1.0;
|
||
if (args[ARG_scale].u_obj != mp_const_none) {
|
||
scale = mp_obj_get_float(args[ARG_scale].u_obj);
|
||
}
|
||
if (scale < 0) { // ensure scale >= 0
|
||
scale = 1.0;
|
||
}
|
||
|
||
uint32_t skip_index;
|
||
bool skip_index_none; // Flag whether input skip_value was None
|
||
if (args[ARG_skip_index].u_obj == mp_const_none) {
|
||
skip_index = 0;
|
||
skip_index_none = true;
|
||
} else {
|
||
skip_index = mp_obj_get_int(args[ARG_skip_index].u_obj);
|
||
skip_index_none = false;
|
||
}
|
||
|
||
common_hal_bitmaptools_rotozoom(destination, ox, oy,
|
||
dest_clip0_x, dest_clip0_y,
|
||
dest_clip1_x, dest_clip1_y,
|
||
source, px, py,
|
||
source_clip0_x, source_clip0_y,
|
||
source_clip1_x, source_clip1_y,
|
||
angle,
|
||
scale,
|
||
skip_index, skip_index_none);
|
||
|
||
return mp_const_none;
|
||
}
|
||
|
||
MP_DEFINE_CONST_FUN_OBJ_KW(bitmaptools_rotozoom_obj, 0, bitmaptools_obj_rotozoom);
|
||
// requires at least 2 arguments (destination bitmap and source bitmap)
|
||
|
||
//|
|
||
//| def fill_region(
|
||
//| dest_bitmap: displayio.Bitmap,
|
||
//| x1: int, y1: int,
|
||
//| x2: int, y2: int,
|
||
//| value: int) -> None:
|
||
//| """Draws the color value into the destination bitmap within the
|
||
//| rectangular region bounded by (x1,y1) and (x2,y2), exclusive.
|
||
//|
|
||
//| :param bitmap dest_bitmap: Destination bitmap that will be written into
|
||
//| :param int x1: x-pixel position of the first corner of the rectangular fill region
|
||
//| :param int y1: y-pixel position of the first corner of the rectangular fill region
|
||
//| :param int x2: x-pixel position of the second corner of the rectangular fill region (exclusive)
|
||
//| :param int y2: y-pixel position of the second corner of the rectangular fill region (exclusive)
|
||
//| :param int value: Bitmap palette index that will be written into the rectangular
|
||
//| fill region in the destination bitmap"""
|
||
//| ...
|
||
//|
|
||
STATIC mp_obj_t bitmaptools_obj_fill_region(size_t n_args, const mp_obj_t *pos_args, mp_map_t *kw_args) {
|
||
enum {ARG_dest_bitmap, ARG_x1, ARG_y1, ARG_x2, ARG_y2, ARG_value};
|
||
|
||
static const mp_arg_t allowed_args[] = {
|
||
{MP_QSTR_dest_bitmap, MP_ARG_REQUIRED | MP_ARG_OBJ},
|
||
{MP_QSTR_x1, MP_ARG_REQUIRED | MP_ARG_INT},
|
||
{MP_QSTR_y1, MP_ARG_REQUIRED | MP_ARG_INT},
|
||
{MP_QSTR_x2, MP_ARG_REQUIRED | MP_ARG_INT},
|
||
{MP_QSTR_y2, MP_ARG_REQUIRED | MP_ARG_INT},
|
||
{MP_QSTR_value, MP_ARG_REQUIRED | MP_ARG_INT},
|
||
};
|
||
mp_arg_val_t args[MP_ARRAY_SIZE(allowed_args)];
|
||
mp_arg_parse_all(n_args, pos_args, kw_args, MP_ARRAY_SIZE(allowed_args), allowed_args, args);
|
||
|
||
displayio_bitmap_t *destination = MP_OBJ_TO_PTR(args[ARG_dest_bitmap].u_obj); // the destination bitmap
|
||
|
||
uint32_t value, color_depth;
|
||
value = args[ARG_value].u_int;
|
||
color_depth = (1 << destination->bits_per_value);
|
||
if (color_depth <= value) {
|
||
mp_raise_ValueError(translate("out of range of target"));
|
||
}
|
||
|
||
int16_t x1 = args[ARG_x1].u_int;
|
||
int16_t y1 = args[ARG_y1].u_int;
|
||
int16_t x2 = args[ARG_x2].u_int;
|
||
int16_t y2 = args[ARG_y2].u_int;
|
||
|
||
common_hal_bitmaptools_fill_region(destination, x1, y1, x2, y2, value);
|
||
|
||
return mp_const_none;
|
||
}
|
||
|
||
MP_DEFINE_CONST_FUN_OBJ_KW(bitmaptools_fill_region_obj, 0, bitmaptools_obj_fill_region);
|
||
// requires all 6 arguments
|
||
|
||
//|
|
||
//| def draw_line(
|
||
//| dest_bitmap: displayio.Bitmap,
|
||
//| x1: int, y1: int,
|
||
//| x2: int, y2: int,
|
||
//| value: int) -> None:
|
||
//| """Draws a line into a bitmap specified two endpoints (x1,y1) and (x2,y2).
|
||
//|
|
||
//| :param bitmap dest_bitmap: Destination bitmap that will be written into
|
||
//| :param int x1: x-pixel position of the line's first endpoint
|
||
//| :param int y1: y-pixel position of the line's first endpoint
|
||
//| :param int x2: x-pixel position of the line's second endpoint
|
||
//| :param int y2: y-pixel position of the line's second endpoint
|
||
//| :param int value: Bitmap palette index that will be written into the
|
||
//| line in the destination bitmap"""
|
||
//| ...
|
||
//|
|
||
STATIC mp_obj_t bitmaptools_obj_draw_line(size_t n_args, const mp_obj_t *pos_args, mp_map_t *kw_args) {
|
||
enum {ARG_dest_bitmap, ARG_x1, ARG_y1, ARG_x2, ARG_y2, ARG_value};
|
||
|
||
static const mp_arg_t allowed_args[] = {
|
||
{MP_QSTR_dest_bitmap, MP_ARG_REQUIRED | MP_ARG_OBJ},
|
||
{MP_QSTR_x1, MP_ARG_REQUIRED | MP_ARG_INT},
|
||
{MP_QSTR_y1, MP_ARG_REQUIRED | MP_ARG_INT},
|
||
{MP_QSTR_x2, MP_ARG_REQUIRED | MP_ARG_INT},
|
||
{MP_QSTR_y2, MP_ARG_REQUIRED | MP_ARG_INT},
|
||
{MP_QSTR_value, MP_ARG_REQUIRED | MP_ARG_INT},
|
||
};
|
||
mp_arg_val_t args[MP_ARRAY_SIZE(allowed_args)];
|
||
mp_arg_parse_all(n_args, pos_args, kw_args, MP_ARRAY_SIZE(allowed_args), allowed_args, args);
|
||
|
||
displayio_bitmap_t *destination = MP_OBJ_TO_PTR(args[ARG_dest_bitmap].u_obj); // the destination bitmap
|
||
|
||
uint32_t value, color_depth;
|
||
value = args[ARG_value].u_int;
|
||
color_depth = (1 << destination->bits_per_value);
|
||
if (color_depth <= value) {
|
||
mp_raise_ValueError(translate("out of range of target"));
|
||
}
|
||
|
||
int16_t x1 = args[ARG_x1].u_int;
|
||
int16_t y1 = args[ARG_y1].u_int;
|
||
int16_t x2 = args[ARG_x2].u_int;
|
||
int16_t y2 = args[ARG_y2].u_int;
|
||
|
||
// verify points are within the bitmap boundary (inclusive)
|
||
if ((x1 < 0) || (x2 < 0) || (y1 < 0) || (y2 < 0) ||
|
||
(x1 >= destination->width) || (x2 >= destination->width) ||
|
||
(y1 >= destination->height) || (y2 >= destination->height)) {
|
||
mp_raise_ValueError(translate("out of range of target"));
|
||
}
|
||
|
||
common_hal_bitmaptools_draw_line(destination, x1, y1, x2, y2, value);
|
||
|
||
return mp_const_none;
|
||
}
|
||
|
||
MP_DEFINE_CONST_FUN_OBJ_KW(bitmaptools_draw_line_obj, 0, bitmaptools_obj_draw_line);
|
||
// requires all 6 arguments
|
||
|
||
//| def arrayblit(bitmap: displayio.Bitmap, data: ReadableBuffer, x1: int=0, y1: int=0, x2: Optional[int]=None, y2: Optional[int]=None, skip_index:Optional[int]=None) -> None:
|
||
//| """Inserts pixels from ``data`` into the rectangle of width×height pixels with the upper left corner at ``(x,y)``
|
||
//|
|
||
//| The values from ``data`` are taken modulo the number of color values
|
||
//| avalable in the destination bitmap.
|
||
//|
|
||
//| If x1 or y1 are not specified, they are taken as 0. If x2 or y2
|
||
//| are not specified, or are given as -1, they are taken as the width
|
||
//| and height of the image.
|
||
//|
|
||
//| The coordinates affected by the blit are ``x1 <= x < x2`` and ``y1 <= y < y2``.
|
||
//|
|
||
//| ``data`` must contain at least as many elements as required. If it
|
||
//| contains excess elements, they are ignored.
|
||
//|
|
||
//| The blit takes place by rows, so the first elements of ``data`` go
|
||
//| to the first row, the next elements to the next row, and so on.
|
||
//|
|
||
//| :param displayio.Bitmap bitmap: A writable bitmap
|
||
//| :param ReadableBuffer data: Buffer containing the source pixel values
|
||
//| :param int x1: The left corner of the area to blit into (inclusive)
|
||
//| :param int y1: The top corner of the area to blit into (inclusive)
|
||
//| :param int x2: The right of the area to blit into (exclusive)
|
||
//| :param int y2: The bottom corner of the area to blit into (exclusive)
|
||
//| :param int skip_index: Bitmap palette index in the source that will not be copied,
|
||
//| set to None to copy all pixels
|
||
//| """
|
||
//| ...
|
||
//|
|
||
STATIC mp_obj_t bitmaptools_arrayblit(size_t n_args, const mp_obj_t *pos_args, mp_map_t *kw_args) {
|
||
enum { ARG_bitmap, ARG_data, ARG_x1, ARG_y1, ARG_x2, ARG_y2, ARG_skip_index };
|
||
static const mp_arg_t allowed_args[] = {
|
||
{ MP_QSTR_bitmap, MP_ARG_REQUIRED | MP_ARG_OBJ },
|
||
{ MP_QSTR_data, MP_ARG_REQUIRED | MP_ARG_OBJ },
|
||
{ MP_QSTR_x1, MP_ARG_INT, {.u_int = 0} },
|
||
{ MP_QSTR_y1, MP_ARG_INT, {.u_int = 0} },
|
||
{ MP_QSTR_x2, MP_ARG_INT, {.u_int = -1} },
|
||
{ MP_QSTR_y2, MP_ARG_INT, {.u_int = -1} },
|
||
{ MP_QSTR_skip_index, MP_ARG_OBJ, {.u_obj = mp_const_none } },
|
||
};
|
||
mp_arg_val_t args[MP_ARRAY_SIZE(allowed_args)];
|
||
mp_arg_parse_all(n_args, pos_args, kw_args, MP_ARRAY_SIZE(allowed_args), allowed_args, args);
|
||
|
||
if (!mp_obj_is_type(args[ARG_bitmap].u_obj, &displayio_bitmap_type)) {
|
||
mp_raise_TypeError(NULL);
|
||
}
|
||
displayio_bitmap_t *bitmap = MP_OBJ_TO_PTR(args[ARG_bitmap].u_obj);
|
||
|
||
mp_buffer_info_t bufinfo;
|
||
mp_get_buffer_raise(args[ARG_data].u_obj, &bufinfo, MP_BUFFER_READ);
|
||
|
||
int x1 = args[ARG_x1].u_int;
|
||
int y1 = args[ARG_y1].u_int;
|
||
int x2 = args[ARG_x2].u_int == -1 ? bitmap->width : args[ARG_x2].u_int;
|
||
int y2 = args[ARG_y2].u_int == -1 ? bitmap->height : args[ARG_y2].u_int;
|
||
|
||
if ((x1 < 0) || (y1 < 0) || (x1 > x2) || (y1 > y2) || (x2 > bitmap->width) || (y2 > bitmap->height)) {
|
||
mp_raise_IndexError(translate("pixel coordinates out of bounds"));
|
||
}
|
||
|
||
size_t output_element_count = (x2 - x1) * (y2 - y1);
|
||
size_t element_size = mp_binary_get_size('@', bufinfo.typecode, NULL);
|
||
size_t input_element_count = bufinfo.len / element_size;
|
||
|
||
bool skip_specified = args[ARG_skip_index].u_obj != mp_const_none;
|
||
uint32_t skip_index = skip_specified ? mp_obj_get_int(args[ARG_skip_index].u_obj) : 0;
|
||
if (input_element_count < output_element_count) {
|
||
mp_raise_IndexError(translate("index out of range"));
|
||
}
|
||
|
||
common_hal_bitmaptools_arrayblit(bitmap, bufinfo.buf, element_size, x1, y1, x2, y2, skip_specified, skip_index);
|
||
|
||
return mp_const_none;
|
||
}
|
||
MP_DEFINE_CONST_FUN_OBJ_KW(bitmaptools_arrayblit_obj, 0, bitmaptools_arrayblit);
|
||
|
||
|
||
//| def readinto(bitmap: displayio.Bitmap, file: typing.BinaryIO, bits_per_pixel: int, element_size: int = 1, reverse_pixels_in_element: bool = False, swap_bytes_in_element: bool = False, reverse_rows: bool = False) -> None:
|
||
//| """Reads from a binary file into a bitmap.
|
||
//|
|
||
//| The file must be positioned so that it consists of ``bitmap.height`` rows of pixel data, where each row is the smallest multiple of ``element_size`` bytes that can hold ``bitmap.width`` pixels.
|
||
//|
|
||
//| The bytes in an element can be optionally swapped, and the pixels in an element can be reversed. Also, the
|
||
//| row loading direction can be reversed, which may be requires for loading certain bitmap files.
|
||
//|
|
||
//| This function doesn't parse image headers, but is useful to speed up loading of uncompressed image formats such as PCF glyph data.
|
||
//|
|
||
//| :param displayio.Bitmap bitmap: A writable bitmap
|
||
//| :param typing.BinaryIO file: A file opened in binary mode
|
||
//| :param int bits_per_pixel: Number of bits per pixel. Values 1, 2, 4, 8, 16, 24, and 32 are supported;
|
||
//| :param int element_size: Number of bytes per element. Values of 1, 2, and 4 are supported, except that 24 ``bits_per_pixel`` requires 1 byte per element.
|
||
//| :param bool reverse_pixels_in_element: If set, the first pixel in a word is taken from the Most Signficant Bits; otherwise, it is taken from the Least Significant Bits.
|
||
//| :param bool swap_bytes_in_element: If the ``element_size`` is not 1, then reverse the byte order of each element read.
|
||
//| :param bool reverse_rows: Reverse the direction of the row loading (required for some bitmap images).
|
||
//| """
|
||
//| ...
|
||
//|
|
||
|
||
STATIC mp_obj_t bitmaptools_readinto(size_t n_args, const mp_obj_t *pos_args, mp_map_t *kw_args) {
|
||
enum { ARG_bitmap, ARG_file, ARG_bits_per_pixel, ARG_element_size, ARG_reverse_pixels_in_element, ARG_swap_bytes_in_element, ARG_reverse_rows };
|
||
static const mp_arg_t allowed_args[] = {
|
||
{ MP_QSTR_bitmap, MP_ARG_REQUIRED | MP_ARG_OBJ },
|
||
{ MP_QSTR_file, MP_ARG_REQUIRED | MP_ARG_OBJ },
|
||
{ MP_QSTR_bits_per_pixel, MP_ARG_REQUIRED | MP_ARG_INT },
|
||
{ MP_QSTR_element_size, MP_ARG_INT, { .u_int = 1 } },
|
||
{ MP_QSTR_reverse_pixels_in_element, MP_ARG_BOOL, { .u_bool = false } },
|
||
{ MP_QSTR_swap_bytes_in_element, MP_ARG_BOOL, { .u_bool = false } },
|
||
{ MP_QSTR_reverse_rows, MP_ARG_BOOL, { .u_bool = false } },
|
||
};
|
||
|
||
mp_arg_val_t args[MP_ARRAY_SIZE(allowed_args)];
|
||
mp_arg_parse_all(n_args, pos_args, kw_args, MP_ARRAY_SIZE(allowed_args), allowed_args, args);
|
||
|
||
if (!mp_obj_is_type(args[ARG_bitmap].u_obj, &displayio_bitmap_type)) {
|
||
mp_raise_TypeError(NULL);
|
||
}
|
||
displayio_bitmap_t *bitmap = MP_OBJ_TO_PTR(args[ARG_bitmap].u_obj);
|
||
|
||
if (!mp_obj_is_type(args[ARG_file].u_obj, &mp_type_fileio)) {
|
||
mp_raise_TypeError(NULL);
|
||
}
|
||
pyb_file_obj_t *file = MP_OBJ_TO_PTR(args[ARG_file].u_obj);
|
||
|
||
int element_size = args[ARG_element_size].u_int;
|
||
if (element_size != 1 && element_size != 2 && element_size != 4) {
|
||
mp_raise_ValueError_varg(translate("invalid element_size %d, must be, 1, 2, or 4"), element_size);
|
||
}
|
||
|
||
int bits_per_pixel = args[ARG_bits_per_pixel].u_int;
|
||
switch (bits_per_pixel) {
|
||
case 24:
|
||
if (element_size != 1) {
|
||
mp_raise_ValueError_varg(translate("invalid element size %d for bits_per_pixel %d\n"), element_size, bits_per_pixel);
|
||
}
|
||
break;
|
||
case 1:
|
||
case 2:
|
||
case 4:
|
||
case 8:
|
||
case 16:
|
||
case 32:
|
||
break;
|
||
default:
|
||
mp_raise_ValueError_varg(translate("invalid bits_per_pixel %d, must be, 1, 4, 8, 16, 24, or 32"), bits_per_pixel);
|
||
}
|
||
|
||
bool reverse_pixels_in_element = args[ARG_reverse_pixels_in_element].u_bool;
|
||
bool swap_bytes_in_element = args[ARG_swap_bytes_in_element].u_bool;
|
||
bool reverse_rows = args[ARG_reverse_rows].u_bool;
|
||
|
||
common_hal_bitmaptools_readinto(bitmap, file, element_size, bits_per_pixel, reverse_pixels_in_element, swap_bytes_in_element, reverse_rows);
|
||
|
||
return mp_const_none;
|
||
}
|
||
|
||
MP_DEFINE_CONST_FUN_OBJ_KW(bitmaptools_readinto_obj, 0, bitmaptools_readinto);
|
||
|
||
STATIC const mp_rom_map_elem_t bitmaptools_module_globals_table[] = {
|
||
{ MP_ROM_QSTR(MP_QSTR_readinto), MP_ROM_PTR(&bitmaptools_readinto_obj) },
|
||
{ MP_ROM_QSTR(MP_QSTR_rotozoom), MP_ROM_PTR(&bitmaptools_rotozoom_obj) },
|
||
{ MP_ROM_QSTR(MP_QSTR_arrayblit), MP_ROM_PTR(&bitmaptools_arrayblit_obj) },
|
||
{ MP_ROM_QSTR(MP_QSTR_fill_region), MP_ROM_PTR(&bitmaptools_fill_region_obj) },
|
||
{ MP_ROM_QSTR(MP_QSTR_draw_line), MP_ROM_PTR(&bitmaptools_draw_line_obj) },
|
||
};
|
||
STATIC MP_DEFINE_CONST_DICT(bitmaptools_module_globals, bitmaptools_module_globals_table);
|
||
|
||
const mp_obj_module_t bitmaptools_module = {
|
||
.base = {&mp_type_module },
|
||
.globals = (mp_obj_dict_t *)&bitmaptools_module_globals,
|
||
};
|