367 lines
13 KiB
C
367 lines
13 KiB
C
/*
|
|
* This file is part of the Micro Python project, http://micropython.org/
|
|
*
|
|
* The MIT License (MIT)
|
|
*
|
|
* Copyright (c) 2016 Glenn Ruben Bakke
|
|
*
|
|
* 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 <stdint.h>
|
|
#include <string.h>
|
|
#include <stdbool.h>
|
|
|
|
#include "py/nlr.h"
|
|
#include "py/obj.h"
|
|
#include "py/runtime.h"
|
|
|
|
#include "lcd_mono_fb.h"
|
|
#include "font_petme128_8x8.h"
|
|
|
|
#if MICROPY_PY_LCD_MONO_FB
|
|
|
|
#define LCD_BLACK 0
|
|
#define LCD_WHITE 1
|
|
|
|
typedef struct {
|
|
mp_obj_base_t base;
|
|
fb_byte_t * fb_bytes;
|
|
fb_byte_t * fb_old;
|
|
fb_byte_t * fb_dirty;
|
|
uint16_t height;
|
|
uint16_t width;
|
|
mp_uint_t bytes_stride;
|
|
mp_uint_t dirty_stride;
|
|
mp_obj_t line_update_cb;
|
|
mp_uint_t bg_color;
|
|
mp_uint_t fg_color;
|
|
mp_uint_t font_size;
|
|
} mp_obj_framebuf_t;
|
|
|
|
STATIC void lcd_enable_pixel(mp_obj_framebuf_t * p_framebuffer, uint16_t x, uint16_t y) {
|
|
uint16_t column = (x / 8);
|
|
uint16_t line = y;
|
|
uint8_t bit_pos = x % 8;
|
|
|
|
p_framebuffer->fb_bytes[line * (p_framebuffer->bytes_stride) + column].byte |= (1 << bit_pos);
|
|
p_framebuffer->fb_dirty[y / 8].byte |= (uint8_t)(0x1 << y % 8);
|
|
}
|
|
|
|
STATIC void lcd_disable_pixel(mp_obj_framebuf_t * p_framebuffer, uint16_t x, uint16_t y) {
|
|
uint16_t column = (x / 8);
|
|
uint16_t line = y;
|
|
uint8_t bit_pos = x % 8;
|
|
|
|
p_framebuffer->fb_bytes[line * (p_framebuffer->bytes_stride) + column].byte &= ~(1 << bit_pos);
|
|
p_framebuffer->fb_dirty[y/8].byte |= (uint8_t)(0x1 << y % 8);
|
|
}
|
|
|
|
STATIC void lcd_init(mp_obj_framebuf_t * p_framebuffer) {
|
|
p_framebuffer->fg_color = LCD_BLACK;
|
|
p_framebuffer->bg_color = LCD_WHITE;
|
|
|
|
memset(p_framebuffer->fb_bytes, 0x00, p_framebuffer->bytes_stride * p_framebuffer->height);
|
|
memset(p_framebuffer->fb_dirty, 0x00, p_framebuffer->dirty_stride);
|
|
}
|
|
|
|
STATIC void lcd_fg_color_set(mp_obj_framebuf_t * p_framebuffer, uint16_t color) {
|
|
p_framebuffer->fg_color = (color == 0) ? LCD_BLACK : LCD_WHITE;
|
|
}
|
|
|
|
#if 0
|
|
STATIC uint16_t lcd_fg_color_get(mp_obj_framebuf_t * p_framebuffer) {
|
|
return p_framebuffer->fg_color;
|
|
}
|
|
#endif
|
|
|
|
STATIC void lcd_bg_color_set(mp_obj_framebuf_t * p_framebuffer, uint16_t color) {
|
|
p_framebuffer->bg_color = (color == 0) ? LCD_BLACK : LCD_WHITE;
|
|
}
|
|
|
|
#if 0
|
|
STATIC uint16_t lcd_bg_color_get(mp_obj_framebuf_t * p_framebuffer) {
|
|
return p_framebuffer->bg_color;
|
|
}
|
|
#endif
|
|
|
|
STATIC void lcd_clear_screen(mp_obj_framebuf_t * p_framebuffer) {
|
|
if (p_framebuffer->bg_color == LCD_BLACK) {
|
|
memset(p_framebuffer->fb_bytes, 0x00, p_framebuffer->bytes_stride * p_framebuffer->height);
|
|
} else {
|
|
memset(p_framebuffer->fb_bytes, 0xFF, p_framebuffer->bytes_stride * p_framebuffer->height);
|
|
}
|
|
memset(p_framebuffer->fb_dirty, 0xFF, p_framebuffer->dirty_stride);
|
|
}
|
|
|
|
STATIC void lcd_print_char(mp_obj_framebuf_t * p_framebuffer, uint16_t x, uint16_t y, char ch) {
|
|
uint16_t col = x;
|
|
for (uint8_t i = 0; i < 8; i++) {
|
|
|
|
uint16_t current_col = col + (i * p_framebuffer->font_size);
|
|
|
|
for (uint8_t y_pos = 0; y_pos < 8; y_pos++) {
|
|
if ((((uint8_t)font_petme128_8x8[((ch - 32) * 8) + i]) >> y_pos) & 0x01) {
|
|
for (uint8_t s_w = 0; s_w < p_framebuffer->font_size; s_w++) {
|
|
for (uint8_t s_h = 0; s_h < p_framebuffer->font_size; s_h++) {
|
|
if (p_framebuffer->fg_color < LCD_WHITE) {
|
|
lcd_disable_pixel(p_framebuffer,
|
|
current_col + s_w,
|
|
y + (y_pos * p_framebuffer->font_size) + s_h);
|
|
} else {
|
|
lcd_enable_pixel(p_framebuffer,
|
|
current_col + s_w,
|
|
y + (y_pos * p_framebuffer->font_size) + s_h);
|
|
}
|
|
}
|
|
}
|
|
} else {
|
|
for (uint8_t s_w = 0; s_w < p_framebuffer->font_size; s_w++) {
|
|
for (uint8_t s_h = 0; s_h < p_framebuffer->font_size; s_h++) {
|
|
if (p_framebuffer->bg_color < LCD_WHITE) {
|
|
lcd_disable_pixel(p_framebuffer,
|
|
current_col + s_w,
|
|
y + (y_pos * p_framebuffer->font_size) + s_h);
|
|
} else {
|
|
lcd_enable_pixel(p_framebuffer,
|
|
current_col + s_w,
|
|
y + (y_pos * p_framebuffer->font_size) + s_h);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
#if 0
|
|
STATIC void lcd_font_size_set(mp_obj_framebuf_t * p_framebuffer, uint8_t size) {
|
|
m_font_size = size;
|
|
}
|
|
|
|
STATIC uint8_t lcd_font_size_get(mp_obj_framebuf_t * p_framebuffer) {
|
|
return m_font_size;
|
|
}
|
|
#endif
|
|
|
|
STATIC void lcd_print_string(mp_obj_framebuf_t * p_framebuffer, uint16_t x, uint16_t y, const char * p_str) {
|
|
uint16_t str_len = strlen(p_str);
|
|
for (uint16_t i = 0; i < str_len; i++) {
|
|
lcd_print_char(p_framebuffer, x + (i * 8 * p_framebuffer->font_size), y, p_str[i]);
|
|
}
|
|
}
|
|
|
|
STATIC void lcd_pixel_draw(mp_obj_framebuf_t * p_framebuffer, uint16_t x, uint16_t y, uint16_t color) {
|
|
if (color < LCD_WHITE) {
|
|
lcd_disable_pixel(p_framebuffer, x, y);
|
|
} else {
|
|
lcd_enable_pixel(p_framebuffer, x, y);
|
|
}
|
|
}
|
|
|
|
STATIC void lcd_update(mp_obj_framebuf_t * p_framebuffer, bool refresh) {
|
|
for (uint16_t i = 0; i < p_framebuffer->dirty_stride; i++) {
|
|
if (p_framebuffer->fb_dirty[i].byte != 0 || refresh) {
|
|
for (uint16_t b = 0; b < 8; b++) {
|
|
if ((((p_framebuffer->fb_dirty[i].byte >> b) & 0x01) == 1) || refresh) {
|
|
uint16_t line_num = (i * 8) + b;
|
|
mp_obj_t args[4];
|
|
mp_uint_t num_of_args = 3;
|
|
args[0] = p_framebuffer;
|
|
args[1] = MP_OBJ_NEW_SMALL_INT(line_num);
|
|
|
|
if (refresh == false) {
|
|
args[2] = mp_obj_new_bytearray_by_ref(p_framebuffer->bytes_stride,
|
|
&p_framebuffer->fb_bytes[line_num * p_framebuffer->bytes_stride]);
|
|
} else {
|
|
args[2] = mp_const_none;
|
|
}
|
|
|
|
if (p_framebuffer->fb_old != NULL) {
|
|
args[3] = mp_obj_new_bytearray_by_ref(p_framebuffer->bytes_stride,
|
|
&p_framebuffer->fb_bytes[line_num * p_framebuffer->bytes_stride]);
|
|
num_of_args = 4;
|
|
}
|
|
|
|
mp_call_function_n_kw(p_framebuffer->line_update_cb, num_of_args, 0, args);
|
|
|
|
// update old buffer
|
|
if (p_framebuffer->fb_old != NULL) {
|
|
memcpy(&p_framebuffer->fb_old[line_num * p_framebuffer->bytes_stride],
|
|
&p_framebuffer->fb_bytes[line_num * p_framebuffer->bytes_stride],
|
|
p_framebuffer->dirty_stride);
|
|
}
|
|
}
|
|
}
|
|
|
|
if (refresh == false) {
|
|
p_framebuffer->fb_dirty[i].byte = 0x00;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
STATIC mp_obj_t lcd_mono_fb_make_new(const mp_obj_type_t *type, size_t n_args, size_t n_kw, const mp_obj_t *args) {
|
|
mp_arg_check_num(n_args, n_kw, 3, 4, false);
|
|
|
|
mp_obj_framebuf_t *o = m_new_obj(mp_obj_framebuf_t);
|
|
o->base.type = type;
|
|
|
|
o->line_update_cb = args[0];
|
|
|
|
o->width = mp_obj_get_int(args[1]);
|
|
o->height = mp_obj_get_int(args[2]);
|
|
|
|
o->bytes_stride = o->width / 8;
|
|
o->dirty_stride = o->height / 8;
|
|
|
|
o->fb_bytes = m_new(fb_byte_t, (o->bytes_stride) * o->height);
|
|
o->fb_dirty = m_new(fb_byte_t, o->dirty_stride);
|
|
|
|
// default to not use double buffer
|
|
o->fb_old = NULL;
|
|
|
|
o->font_size = 1;
|
|
|
|
if (n_args >= 4) {
|
|
if (mp_obj_is_true(args[3])) {
|
|
o->fb_old = m_new(fb_byte_t, (o->bytes_stride) * o->height);
|
|
}
|
|
}
|
|
|
|
lcd_init(o);
|
|
|
|
return MP_OBJ_FROM_PTR(o);
|
|
}
|
|
|
|
STATIC mp_obj_t lcd_mono_fb_fill(mp_obj_t self_in, mp_obj_t col_in) {
|
|
mp_obj_framebuf_t *self = MP_OBJ_TO_PTR(self_in);
|
|
mp_int_t col = mp_obj_get_int(col_in);
|
|
|
|
lcd_bg_color_set(self, col);
|
|
lcd_fg_color_set(self, !col);
|
|
|
|
lcd_clear_screen(self);
|
|
|
|
return mp_const_none;
|
|
}
|
|
STATIC MP_DEFINE_CONST_FUN_OBJ_2(lcd_mono_fb_fill_obj, lcd_mono_fb_fill);
|
|
|
|
STATIC mp_obj_t lcd_mono_fb_pixel(size_t n_args, const mp_obj_t *args) {
|
|
mp_obj_framebuf_t *self = MP_OBJ_TO_PTR(args[0]);
|
|
mp_int_t x = mp_obj_get_int(args[1]);
|
|
mp_int_t y = mp_obj_get_int(args[2]);
|
|
mp_int_t color = mp_obj_get_int(args[3]);
|
|
|
|
lcd_pixel_draw(self, x, y, color);
|
|
|
|
return mp_const_none;
|
|
}
|
|
STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(lcd_mono_fb_pixel_obj, 3, 4, lcd_mono_fb_pixel);
|
|
|
|
STATIC mp_obj_t lcd_mono_fb_scroll(mp_obj_t self_in, mp_obj_t xstep_in, mp_obj_t ystep_in) {
|
|
#if 0
|
|
mp_obj_framebuf_t *self = MP_OBJ_TO_PTR(self_in);
|
|
mp_int_t xstep = mp_obj_get_int(xstep_in);
|
|
mp_int_t ystep = mp_obj_get_int(ystep_in);
|
|
#endif
|
|
return mp_const_none;
|
|
}
|
|
STATIC MP_DEFINE_CONST_FUN_OBJ_3(lcd_mono_fb_scroll_obj, lcd_mono_fb_scroll);
|
|
|
|
STATIC mp_obj_t lcd_mono_fb_text(size_t n_args, const mp_obj_t *args) {
|
|
// extract arguments
|
|
mp_obj_framebuf_t *self = MP_OBJ_TO_PTR(args[0]);
|
|
const char *str = mp_obj_str_get_str(args[1]);
|
|
mp_int_t x0 = mp_obj_get_int(args[2]);
|
|
mp_int_t y0 = mp_obj_get_int(args[3]);
|
|
|
|
lcd_print_string(self, x0, y0, str);
|
|
|
|
return mp_const_none;
|
|
}
|
|
STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(lcd_mono_fb_text_obj, 4, 5, lcd_mono_fb_text);
|
|
|
|
|
|
STATIC mp_obj_t lcd_mono_fb_show(mp_obj_t self_in) {
|
|
mp_obj_framebuf_t *self = MP_OBJ_TO_PTR(self_in);
|
|
|
|
lcd_update(self, false);
|
|
|
|
return mp_const_none;
|
|
}
|
|
STATIC MP_DEFINE_CONST_FUN_OBJ_1(lcd_mono_fb_show_obj, lcd_mono_fb_show);
|
|
|
|
STATIC mp_obj_t lcd_mono_fb_refresh(mp_obj_t self_in) {
|
|
mp_obj_framebuf_t *self = MP_OBJ_TO_PTR(self_in);
|
|
|
|
lcd_update(self, true);
|
|
|
|
return mp_const_none;
|
|
}
|
|
STATIC MP_DEFINE_CONST_FUN_OBJ_1(lcd_mono_fb_refresh_obj, lcd_mono_fb_refresh);
|
|
|
|
|
|
STATIC mp_obj_t lcd_mono_fb_del(mp_obj_t self_in) {
|
|
mp_obj_framebuf_t *self = MP_OBJ_TO_PTR(self_in);
|
|
|
|
m_free(self->fb_bytes);
|
|
m_free(self->fb_dirty);
|
|
|
|
if (self->fb_old != NULL) {
|
|
// free double buffer if used
|
|
m_free(self->fb_old);
|
|
}
|
|
|
|
return mp_const_none;
|
|
}
|
|
STATIC MP_DEFINE_CONST_FUN_OBJ_1(lcd_mono_fb_del_obj, lcd_mono_fb_del);
|
|
|
|
STATIC const mp_rom_map_elem_t lcd_mono_fb_locals_dict_table[] = {
|
|
{ MP_ROM_QSTR(MP_QSTR___del__), MP_ROM_PTR(&lcd_mono_fb_del_obj) },
|
|
{ MP_ROM_QSTR(MP_QSTR_fill), MP_ROM_PTR(&lcd_mono_fb_fill_obj) },
|
|
{ MP_ROM_QSTR(MP_QSTR_pixel), MP_ROM_PTR(&lcd_mono_fb_pixel_obj) },
|
|
{ MP_ROM_QSTR(MP_QSTR_scroll), MP_ROM_PTR(&lcd_mono_fb_scroll_obj) },
|
|
{ MP_ROM_QSTR(MP_QSTR_text), MP_ROM_PTR(&lcd_mono_fb_text_obj) },
|
|
{ MP_ROM_QSTR(MP_QSTR_show), MP_ROM_PTR(&lcd_mono_fb_show_obj) },
|
|
{ MP_ROM_QSTR(MP_QSTR_refresh), MP_ROM_PTR(&lcd_mono_fb_refresh_obj) },
|
|
};
|
|
STATIC MP_DEFINE_CONST_DICT(lcd_mono_fb_locals_dict, lcd_mono_fb_locals_dict_table);
|
|
|
|
STATIC const mp_obj_type_t mp_type_lcd_mono_fb = {
|
|
{ &mp_type_type },
|
|
.name = MP_QSTR_MonoFB,
|
|
.make_new = lcd_mono_fb_make_new,
|
|
.locals_dict = (mp_obj_t)&lcd_mono_fb_locals_dict,
|
|
};
|
|
|
|
STATIC const mp_rom_map_elem_t lcd_mono_fb_module_globals_table[] = {
|
|
{ MP_ROM_QSTR(MP_QSTR___name__), MP_ROM_QSTR(MP_QSTR_lcd_mono_fb) },
|
|
{ MP_ROM_QSTR(MP_QSTR_MonoFB), MP_ROM_PTR(&mp_type_lcd_mono_fb) },
|
|
};
|
|
|
|
STATIC MP_DEFINE_CONST_DICT(lcd_mono_fb_module_globals, lcd_mono_fb_module_globals_table);
|
|
|
|
const mp_obj_module_t mp_module_lcd_mono_fb = {
|
|
.base = { &mp_type_module },
|
|
.globals = (mp_obj_dict_t*)&lcd_mono_fb_module_globals,
|
|
};
|
|
|
|
#endif // MICROPY_PY_LCD_MONO_FB
|