Zxc200611/pocketpy
1
0
Fork 0
mirror of https://github.com/pocketpy/pocketpy synced 2025-12-06 18:20:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

up

Browse Source
This commit is contained in:
blueloveTH 2025-11-23 02:19:55 +08:00
parent 4827e6dea3
commit 70bb5222f9
3 changed files with 84 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
include/pocketpy/vmath.h
View File

@ -42,4 +42,5 @@ typedef union c11_color32 {
unsigned char a;
};
unsigned char data[4];
uint32_t u32;
} c11_color32;

3
include/typings/array2d.pyi
View File

@ -1,5 +1,5 @@
from typing import Callable, Literal, overload, Iterator, Self
from vmath import vec2i
from vmath import vec2i, color32
Neighborhood = Literal['Moore', 'von Neumann']
@ -35,6 +35,7 @@ class array2d_like[T]:
"""
def render(self) -> str: ...
def render_with_color(self, fg: array2d_like[color32 | None], bg: array2d_like[color32 | None]) -> str: ...
def all(self: array2d_like[bool]) -> bool: ...
def any(self: array2d_like[bool]) -> bool: ...

78
src/modules/array2d.c
View File

@ -131,7 +131,10 @@ static bool array2d_like_render(int argc, py_Ref argv) {
for(int j = 0; j < self->n_rows; j++) {
for(int i = 0; i < self->n_cols; i++) {
py_Ref item = self->f_get(self, i, j);
if(!py_str(item)) return false;
if(!py_str(item)) {
c11_sbuf__dtor(&buf);
return false;
}
c11_sbuf__write_sv(&buf, py_tosv(py_retval()));
}
if(j < self->n_rows - 1) c11_sbuf__write_char(&buf, '\n');
@ -140,6 +143,78 @@ static bool array2d_like_render(int argc, py_Ref argv) {
return true;
}
static bool array2d_like_render_with_color(int argc, py_Ref argv) {
PY_CHECK_ARGC(3);
c11_sbuf buf;
c11_sbuf__ctor(&buf);
c11_array2d_like* self = py_touserdata(argv);
if(!py_checkinstance(py_arg(1), tp_array2d_like)) return false;
if(!py_checkinstance(py_arg(2), tp_array2d_like)) return false;
c11_array2d_like* fg_colors = py_touserdata(py_arg(1));
c11_array2d_like* bg_colors = py_touserdata(py_arg(2));
c11_color32 curr_fg, curr_bg;
curr_fg.u32 = 0;
curr_bg.u32 = 0;
for(int j = 0; j < self->n_rows; j++) {
for(int i = 0; i < self->n_cols; i++) {
py_Ref item = self->f_get(self, i, j);
if(!py_str(item)) {
c11_sbuf__dtor(&buf);
return false;
}
py_Ref fg_item = fg_colors->f_get(fg_colors, i, j);
py_Ref bg_item = bg_colors->f_get(bg_colors, i, j);
c11_color32 new_fg, new_bg;
if(py_isnone(fg_item)) {
new_fg.u32 = 0;
} else {
if(!py_checktype(fg_item, tp_color32)) {
c11_sbuf__dtor(&buf);
return false;
}
new_fg = py_tocolor32(fg_item);
}
if(py_isnone(bg_item)) {
new_bg.u32 = 0;
} else {
if(!py_checktype(bg_item, tp_color32)) {
c11_sbuf__dtor(&buf);
return false;
}
new_bg = py_tocolor32(bg_item);
}
if(curr_fg.u32 != new_fg.u32 || curr_bg.u32 != new_bg.u32) {
if(curr_fg.u32 != 0 || curr_bg.u32 != 0) c11_sbuf__write_cstr(&buf, "\x1b[0m");
curr_fg = new_fg;
curr_bg = new_bg;
if(curr_fg.u32 != 0)
pk_sprintf(&buf, "\x1b[38;2;%d;%d;%dm", curr_fg.r, curr_fg.g, curr_fg.b);
if(curr_bg.u32 != 0)
pk_sprintf(&buf, "\x1b[48;2;%d;%d;%dm", curr_bg.r, curr_bg.g, curr_bg.b);
}
c11_sbuf__write_sv(&buf, py_tosv(py_retval()));
}
if(j < self->n_rows - 1) {
if(curr_fg.u32 != 0 || curr_bg.u32 != 0) {
curr_fg.u32 = 0;
curr_bg.u32 = 0;
c11_sbuf__write_cstr(&buf, "\x1b[0m");
} else {
c11_sbuf__write_char(&buf, '\n');
}
}
}
c11_sbuf__py_submit(&buf, py_retval());
return true;
}
static bool array2d_like_all(int argc, py_Ref argv) {
PY_CHECK_ARGC(1);
c11_array2d_like* self = py_touserdata(argv);
@ -754,6 +829,7 @@ static void register_array2d_like(py_Ref mod) {
py_bindmethod(type, "index", array2d_like_index);
py_bindmethod(type, "render", array2d_like_render);
py_bindmethod(type, "render_with_color", array2d_like_render_with_color);
py_bindmethod(type, "all", array2d_like_all);
py_bindmethod(type, "any", array2d_like_any);