mirrors
/
pimoroni-pico
mirror of https://github.com/pimoroni/pimoroni-pico.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

clamg-format - new code

This commit is contained in:
Graeme Winter 2021-04-16 06:13:36 +01:00
parent 4a0523ef1e
commit d5e9dd001e
1 changed files with 80 additions and 83 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

163
micropython/modules/pico_scroll/pico_scroll.cpp
View File

@ -41,136 +41,133 @@ mp_obj_t picoscroll_update() {
}
mp_obj_t picoscroll_set_pixel(mp_obj_t x_obj, mp_obj_t y_obj, mp_obj_t v_obj) {
if(scroll != nullptr) {
if (scroll != nullptr) {
int x = mp_obj_get_int(x_obj);
int y = mp_obj_get_int(y_obj);
int val = mp_obj_get_int(v_obj);
if(x < 0 || x >= PicoScroll::WIDTH || y < 0 || y >= PicoScroll::HEIGHT)
if (x < 0 || x >= PicoScroll::WIDTH || y < 0 || y >= PicoScroll::HEIGHT)
mp_raise_ValueError("x or y out of range.");
else {
if(val < 0 || val > 255)
if (val < 0 || val > 255)
mp_raise_ValueError("val out of range. Expected 0 to 255");
else
scroll->set_pixel(x, y, val);
}
}
else
} else
mp_raise_msg(&mp_type_RuntimeError, NOT_INITIALISED_MSG);
return mp_const_none;
}
mp_obj_t picoscroll_show_text(mp_obj_t text_obj, mp_obj_t brightness_obj, mp_obj_t offset_obj) {
if(scroll != nullptr) {
mp_obj_t picoscroll_show_text(mp_obj_t text_obj, mp_obj_t brightness_obj,
mp_obj_t offset_obj) {
if (scroll != nullptr) {
mp_buffer_info_t bufinfo;
unsigned char * buffer;
int text_len, bfr_len;
mp_get_buffer_raise(text_obj, &bufinfo, MP_BUFFER_READ);
unsigned char * values = (unsigned char *) bufinfo.buf;
unsigned char *buffer;
int text_len, bfr_len;
mp_get_buffer_raise(text_obj, &bufinfo, MP_BUFFER_READ);
unsigned char *values = (unsigned char *)bufinfo.buf;
int brightness = mp_obj_get_int(brightness_obj);
int offset = mp_obj_get_int(offset_obj);
text_len = bufinfo.len;
bfr_len = 6 * text_len;
int width = PicoScroll::WIDTH;
int height = PicoScroll::HEIGHT;
// clear the scroll, so only need to write visible bytes
scroll->clear();
text_len = bufinfo.len;
bfr_len = 6 * text_len;
if ((offset < -width) || (offset > bfr_len)) {
return mp_const_none;
}
int width = PicoScroll::WIDTH;
int height = PicoScroll::HEIGHT;
// allocate buffer, render text, free buffer #TODO probably can do
// without the buffer here
buffer = (unsigned char *) m_malloc (sizeof(unsigned char) * bfr_len);
render(values, text_len, buffer, bfr_len);
for (int x = 0; x < width; x++) {
int k = offset + x;
if ((k >= 0) && (k < bfr_len)) {
unsigned char col = buffer[k];
for (int y = 0; y < height; y++) {
int val = brightness * ((col >> y) & 1);
scroll->set_pixel(x, y, val);
}
}
}
m_free(buffer);
// clear the scroll, so only need to write visible bytes
scroll->clear();
if ((offset < -width) || (offset > bfr_len)) {
return mp_const_none;
}
// allocate buffer, render text, free buffer #TODO probably can do
// without the buffer here
buffer = (unsigned char *)m_malloc(sizeof(unsigned char) * bfr_len);
render(values, text_len, buffer, bfr_len);
for (int x = 0; x < width; x++) {
int k = offset + x;
if ((k >= 0) && (k < bfr_len)) {
unsigned char col = buffer[k];
for (int y = 0; y < height; y++) {
int val = brightness * ((col >> y) & 1);
scroll->set_pixel(x, y, val);
}
}
}
m_free(buffer);
} else {
mp_raise_msg(&mp_type_RuntimeError, NOT_INITIALISED_MSG);
}
return mp_const_none;
}
mp_obj_t picoscroll_set_pixels(mp_obj_t image_obj) {
if(scroll != nullptr) {
if (scroll != nullptr) {
mp_buffer_info_t bufinfo;
mp_get_buffer_raise(image_obj, &bufinfo, MP_BUFFER_RW);
mp_get_buffer_raise(image_obj, &bufinfo, MP_BUFFER_RW);
if (bufinfo.len < (PicoScroll::WIDTH * PicoScroll::HEIGHT)) {
mp_raise_msg(&mp_type_IndexError, BUFFER_TOO_SMALL_MSG);
}
if (bufinfo.len < (PicoScroll::WIDTH * PicoScroll::HEIGHT)) {
mp_raise_msg(&mp_type_IndexError, BUFFER_TOO_SMALL_MSG);
}
unsigned char * values = (unsigned char *) bufinfo.buf;
unsigned char *values = (unsigned char *)bufinfo.buf;
for (int y = 0; y < PicoScroll::HEIGHT; y++) {
for (int x = 0; x < PicoScroll::WIDTH; x++) {
int val = values[y * PicoScroll::WIDTH + x];
for (int x = 0; x < PicoScroll::WIDTH; x++) {
int val = values[y * PicoScroll::WIDTH + x];
scroll->set_pixel(x, y, val);
}
}
}
}
else
} else
mp_raise_msg(&mp_type_RuntimeError, NOT_INITIALISED_MSG);
return mp_const_none;
}
mp_obj_t picoscroll_show_bitmap_1d(mp_obj_t bitmap_obj, mp_obj_t brightness_obj, mp_obj_t offset_obj) {
if(scroll != nullptr) {
mp_obj_t picoscroll_show_bitmap_1d(mp_obj_t bitmap_obj, mp_obj_t brightness_obj,
mp_obj_t offset_obj) {
if (scroll != nullptr) {
mp_buffer_info_t bufinfo;
mp_get_buffer_raise(bitmap_obj, &bufinfo, MP_BUFFER_RW);
mp_get_buffer_raise(bitmap_obj, &bufinfo, MP_BUFFER_RW);
int offset = mp_obj_get_int(offset_obj);
int brightness = mp_obj_get_int(brightness_obj);
int length = bufinfo.len;
int width = PicoScroll::WIDTH;
int height = PicoScroll::HEIGHT;
int length = bufinfo.len;
int width = PicoScroll::WIDTH;
int height = PicoScroll::HEIGHT;
// this obviously shouldn't happen as the scroll is 17x7 pixels
// would fall off end of byte if this the case
if (height > 8) {
mp_raise_msg(&mp_type_RuntimeError, INCORRECT_SIZE_MSG);
}
// this obviously shouldn't happen as the scroll is 17x7 pixels
// would fall off end of byte if this the case
if (height > 8) {
mp_raise_msg(&mp_type_RuntimeError, INCORRECT_SIZE_MSG);
}
unsigned char * values = (unsigned char *) bufinfo.buf;
unsigned char *values = (unsigned char *)bufinfo.buf;
// clear the scroll, so only need to write visible bytes
scroll->clear();
// clear the scroll, so only need to write visible bytes
scroll->clear();
if ((offset < -width) || (offset > length)) {
return mp_const_none;
}
if ((offset < -width) || (offset > length)) {
return mp_const_none;
}
for (int x = 0; x < width; x++) {
int k = offset + x;
if ((k >= 0) && (k < length)) {
unsigned char col = values[k];
for (int y = 0; y < height; y++) {
int val = brightness * ((col >> y) & 1);
scroll->set_pixel(x, y, val);
}
}
}
for (int x = 0; x < width; x++) {
int k = offset + x;
if ((k >= 0) && (k < length)) {
unsigned char col = values[k];
for (int y = 0; y < height; y++) {
int val = brightness * ((col >> y) & 1);
scroll->set_pixel(x, y, val);
}
}
}
} else {
mp_raise_msg(&mp_type_RuntimeError, NOT_INITIALISED_MSG);
}
@ -179,7 +176,7 @@ mp_obj_t picoscroll_show_bitmap_1d(mp_obj_t bitmap_obj, mp_obj_t brightness_obj,
}
mp_obj_t picoscroll_clear() {
if(scroll != nullptr)
if (scroll != nullptr)
scroll->clear();
else
mp_raise_msg(&mp_type_RuntimeError, NOT_INITIALISED_MSG);