2021-02-24 00:23:14 -05:00
/*
* 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>
2021-03-15 21:19:45 -04:00
# include "py/binary.h"
2021-02-24 00:23:14 -05:00
# include "py/obj.h"
# include "py/runtime.h"
2021-02-24 02:00:13 -05:00
//| """Collection of bitmap manipulation tools"""
//|
2021-02-24 00:23:14 -05:00
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
2021-03-16 13:20:09 -04:00
if ( point = = mp_const_none ) {
2021-02-24 00:23:14 -05:00
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
2021-03-16 13:20:09 -04:00
if ( xy_tuple = = mp_const_none ) {
2021-02-24 00:23:14 -05:00
* x = x_default ;
* y = y_default ;
2021-03-16 13:20:09 -04:00
} else if ( ! MP_OBJ_IS_OBJ ( xy_tuple ) ) {
2021-02-24 00:23:14 -05:00
mp_raise_ValueError ( translate ( " clip point must be (x,y) tuple " ) ) ;
} else {
2021-03-16 13:20:09 -04:00
mp_obj_t * items ;
2021-02-24 00:23:14 -05:00
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 ,
2021-03-16 13:20:09 -04:00
mp_obj_t clip1_tuple , int16_t * clip1_x , int16_t * clip1_y ) {
2021-02-24 00:23:14 -05:00
// 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)
2021-03-16 13:20:09 -04:00
if ( * clip0_x > * clip1_x ) {
2021-02-24 00:23:14 -05:00
int16_t temp_value = * clip0_x ; // swap values
* clip0_x = * clip1_x ;
* clip1_x = temp_value ;
}
2021-03-16 13:20:09 -04:00
if ( * clip0_y > * clip1_y ) {
2021-02-24 00:23:14 -05:00
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 ;
}
}
//|
2021-02-24 02:00:13 -05:00
//| def rotozoom(
2021-02-24 17:03:50 -05:00
//| 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],
2021-02-24 02:00:13 -05:00
//| 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).
2021-02-24 00:23:14 -05:00
//|
2021-02-24 17:03:50 -05:00
//| :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
2021-02-24 02:00:13 -05:00
//| :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"""
//| ...
2021-02-24 00:23:14 -05:00
//|
2021-03-16 13:20:09 -04:00
STATIC mp_obj_t bitmaptools_obj_rotozoom ( size_t n_args , const mp_obj_t * pos_args , mp_map_t * kw_args ) {
2021-02-24 17:03:50 -05:00
enum { ARG_dest_bitmap , ARG_source_bitmap ,
2021-03-16 13:20:09 -04:00
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 } ;
2021-02-24 00:23:14 -05:00
static const mp_arg_t allowed_args [ ] = {
{ MP_QSTR_dest_bitmap , MP_ARG_REQUIRED | MP_ARG_OBJ } ,
2021-02-24 17:03:50 -05:00
{ MP_QSTR_source_bitmap , MP_ARG_REQUIRED | MP_ARG_OBJ } ,
2021-02-24 00:23:14 -05:00
{ 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
2021-03-16 13:20:09 -04:00
{ MP_QSTR_skip_index , MP_ARG_OBJ | MP_ARG_KW_ONLY , { . u_obj = mp_const_none } } ,
2021-02-24 00:23:14 -05:00
} ;
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 ;
2021-03-16 13:20:09 -04:00
ox = validate_point ( args [ ARG_ox ] . u_obj , destination - > width / 2 ) ;
2021-02-24 00:23:14 -05:00
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 ;
2021-03-16 13:20:09 -04:00
px = validate_point ( args [ ARG_px ] . u_obj , source - > width / 2 ) ;
2021-02-24 00:23:14 -05:00
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 ,
2021-03-16 13:20:09 -04:00
args [ ARG_dest_clip1 ] . u_obj , & dest_clip1_x , & dest_clip1_y ) ;
2021-02-24 00:23:14 -05:00
// 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 ,
2021-03-16 13:20:09 -04:00
args [ ARG_source_clip1 ] . u_obj , & source_clip1_x , & source_clip1_y ) ;
2021-02-24 00:23:14 -05:00
// Confirm the angle value
2021-03-16 13:20:09 -04:00
float angle = 0.0 ;
if ( args [ ARG_angle ] . u_obj ! = mp_const_none ) {
2021-02-24 00:23:14 -05:00
angle = mp_obj_get_float ( args [ ARG_angle ] . u_obj ) ;
}
// Confirm the scale value
2021-03-16 13:20:09 -04:00
float scale = 1.0 ;
if ( args [ ARG_scale ] . u_obj ! = mp_const_none ) {
2021-02-24 00:23:14 -05:00
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
2021-03-16 13:20:09 -04:00
if ( args [ ARG_skip_index ] . u_obj = = mp_const_none ) {
2021-02-24 00:23:14 -05:00
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 ,
2021-03-16 13:20:09 -04:00
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 ) ;
2021-02-24 00:23:14 -05:00
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)
2021-03-10 12:37:27 -05:00
//|
//| 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
2021-03-11 17:18:17 -05:00
//| :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)
2021-03-10 12:37:27 -05:00
//| :param int value: Bitmap palette index that will be written into the rectangular
//| fill region in the destination bitmap"""
//| ...
//|
2021-03-16 13:20:09 -04:00
STATIC mp_obj_t bitmaptools_obj_fill_region ( size_t n_args , const mp_obj_t * pos_args , mp_map_t * kw_args ) {
2021-03-10 12:37:27 -05:00
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 ) {
2021-03-16 13:20:09 -04:00
mp_raise_ValueError ( translate ( " out of range of target " ) ) ;
}
2021-03-10 12:37:27 -05:00
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"""
//| ...
//|
2021-03-16 13:20:09 -04:00
STATIC mp_obj_t bitmaptools_obj_draw_line ( size_t n_args , const mp_obj_t * pos_args , mp_map_t * kw_args ) {
2021-03-10 12:37:27 -05:00
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 ) {
2021-03-16 13:20:09 -04:00
mp_raise_ValueError ( translate ( " out of range of target " ) ) ;
}
2021-03-10 12:37:27 -05:00
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)
2021-03-16 13:20:09 -04:00
if ( ( x1 < 0 ) | | ( x2 < 0 ) | | ( y1 < 0 ) | | ( y2 < 0 ) | |
( x1 > = destination - > width ) | | ( x2 > = destination - > width ) | |
( y1 > = destination - > height ) | | ( y2 > = destination - > height ) ) {
2021-03-10 12:37:27 -05:00
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
2021-02-24 00:23:14 -05:00
2021-03-16 14:37:04 -04:00
//| 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:
2021-03-15 21:19:45 -04:00
//| """Inserts pixels from ``data`` into the rectangle of width× height pixels with the upper left corner at ``(x,y)``
//|
2021-03-16 11:35:40 -04:00
//| The values from ``data`` are taken modulo the number of color values
2021-03-17 10:30:51 -04:00
//| avalable in the destination bitmap.
2021-03-15 21:19:45 -04:00
//|
2021-03-16 11:35:40 -04:00
//| 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.
2021-03-15 21:19:45 -04:00
//|
2021-03-16 11:35:40 -04:00
//| The coordinates affected by the blit are ``x1 <= x < x2`` and ``y1 <= y < y2``.
2021-03-15 21:19:45 -04:00
//|
2021-03-16 11:35:40 -04:00
//| ``data`` must contain at least as many elements as required. If it
//| contains excess elements, they are ignored.
2021-03-15 21:19:45 -04:00
//|
2021-03-16 11:35:40 -04:00
//| 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.
2021-03-15 21:19:45 -04:00
//|
//| :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,
2021-03-16 14:37:04 -04:00
//| set to None to copy all pixels
2021-03-15 21:19:45 -04:00
//| """
//| ...
//|
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 } } ,
2021-04-03 11:41:20 -04:00
} ;
2021-03-15 21:19:45 -04:00
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 " ) ) ;
}
2021-03-16 13:20:09 -04:00
size_t output_element_count = ( x2 - x1 ) * ( y2 - y1 ) ;
2021-03-15 21:19:45 -04:00
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 ) ;
2021-03-17 10:30:51 -04:00
//| 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:
2021-03-16 11:35:40 -04:00
//| """Reads from a binary file into a bitmap.
2021-03-14 14:27:28 -04:00
//|
2021-03-16 11:35:40 -04:00
//| 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.
2021-03-14 14:27:28 -04:00
//|
2021-03-16 11:42:46 -04:00
//| 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.
2021-03-16 11:35:40 -04:00
//|
//| This function doesn't parse image headers, but is useful to speed up loading of uncompressed image formats such as PCF glyph data.
2021-03-14 14:27:28 -04:00
//|
//| :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.
2021-03-17 10:30:51 -04:00
//| :param bool reverse_rows: Reverse the direction of the row loading (required for some bitmap images).
2021-03-14 14:27:28 -04:00
//| """
//| ...
//|
STATIC mp_obj_t bitmaptools_readinto ( size_t n_args , const mp_obj_t * pos_args , mp_map_t * kw_args ) {
2021-03-16 11:01:12 -04:00
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 } ;
2021-03-14 14:27:28 -04:00
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 } } ,
2021-03-16 11:01:12 -04:00
{ MP_QSTR_reverse_rows , MP_ARG_BOOL , { . u_bool = false } } ,
2021-03-14 14:27:28 -04:00
} ;
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 ) ;
}
2021-03-16 13:20:09 -04:00
pyb_file_obj_t * file = MP_OBJ_TO_PTR ( args [ ARG_file ] . u_obj ) ;
2021-03-14 14:27:28 -04:00
int element_size = args [ ARG_element_size ] . u_int ;
if ( element_size ! = 1 & & element_size ! = 2 & & element_size ! = 4 ) {
2021-03-16 13:20:09 -04:00
mp_raise_ValueError_varg ( translate ( " invalid element_size %d, must be, 1, 2, or 4 " ) , element_size ) ;
2021-03-14 14:27:28 -04:00
}
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 ;
2021-03-16 11:01:12 -04:00
bool reverse_rows = args [ ARG_reverse_rows ] . u_bool ;
2021-03-14 14:27:28 -04:00
2021-03-16 11:01:12 -04:00
common_hal_bitmaptools_readinto ( bitmap , file , element_size , bits_per_pixel , reverse_pixels_in_element , swap_bytes_in_element , reverse_rows ) ;
2021-03-14 14:27:28 -04:00
return mp_const_none ;
}
MP_DEFINE_CONST_FUN_OBJ_KW ( bitmaptools_readinto_obj , 0 , bitmaptools_readinto ) ;
2021-02-24 00:23:14 -05:00
STATIC const mp_rom_map_elem_t bitmaptools_module_globals_table [ ] = {
2021-03-14 14:27:28 -04:00
{ MP_ROM_QSTR ( MP_QSTR_readinto ) , MP_ROM_PTR ( & bitmaptools_readinto_obj ) } ,
2021-02-24 00:23:14 -05:00
{ MP_ROM_QSTR ( MP_QSTR_rotozoom ) , MP_ROM_PTR ( & bitmaptools_rotozoom_obj ) } ,
2021-03-15 21:19:45 -04:00
{ MP_ROM_QSTR ( MP_QSTR_arrayblit ) , MP_ROM_PTR ( & bitmaptools_arrayblit_obj ) } ,
2021-03-10 12:37:27 -05:00
{ 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 ) } ,
2021-02-24 00:23:14 -05:00
} ;
STATIC MP_DEFINE_CONST_DICT ( bitmaptools_module_globals , bitmaptools_module_globals_table ) ;
const mp_obj_module_t bitmaptools_module = {
. base = { & mp_type_module } ,
2021-03-16 13:20:09 -04:00
. globals = ( mp_obj_dict_t * ) & bitmaptools_module_globals ,
2021-02-24 00:23:14 -05:00
} ;