mirror of
https://github.com/openmv/openmv.git
synced 2025-11-04 14:49:50 +08:00
Move LCD code into module. It's now ultra fast, safe and friendly. I
tested everything too and added an example script (very simple).
This commit is contained in:
parent
e13da8b3b8
commit
94910c8ed7
@ -161,6 +161,7 @@ OBJ += $(addprefix $(BUILD)/$(OMV_DIR)/py/, \
|
||||
py_image.o \
|
||||
py_time.o \
|
||||
mlx90620.o \
|
||||
py_lcd.o \
|
||||
py_gif.o \
|
||||
py_mjpeg.o \
|
||||
)
|
||||
|
||||
@ -64,6 +64,7 @@ SRCS += $(addprefix py/, \
|
||||
py_image.c \
|
||||
py_time.c \
|
||||
mlx90620.c \
|
||||
py_lcd.c \
|
||||
py_gif.c \
|
||||
py_mjpeg.c \
|
||||
)
|
||||
|
||||
@ -59,6 +59,7 @@
|
||||
#include "py_sensor.h"
|
||||
#include "py_image.h"
|
||||
#include "mlx90620.h"
|
||||
#include "py_lcd.h"
|
||||
|
||||
int errno;
|
||||
extern char _fatfs_buf;
|
||||
@ -271,7 +272,7 @@ static void make_flash_fs()
|
||||
int main(void)
|
||||
{
|
||||
FRESULT f_res;
|
||||
int sensor_init_ret;
|
||||
int sensor_init_ret = 0;
|
||||
bool first_soft_reset = true;
|
||||
|
||||
// Stack limit should be less than real stack size, so we
|
||||
@ -316,6 +317,7 @@ soft_reset:
|
||||
pyb_usb_init0();
|
||||
sensor_init0();
|
||||
fb_alloc_init0();
|
||||
py_lcd_init0();
|
||||
|
||||
#if MICROPY_HW_ENABLE_RTC
|
||||
if (first_soft_reset) {
|
||||
|
||||
400
src/omv/py/py_lcd.c
Normal file
400
src/omv/py/py_lcd.c
Normal file
@ -0,0 +1,400 @@
|
||||
/*
|
||||
* This file is part of the OpenMV project.
|
||||
* Copyright (c) 2013/2014 Ibrahim Abdelkader <i.abdalkader@gmail.com>
|
||||
* This work is licensed under the MIT license, see the file LICENSE for details.
|
||||
*
|
||||
* LCD Python module.
|
||||
*
|
||||
*/
|
||||
#include <mp.h>
|
||||
#include <objstr.h>
|
||||
#include <spi.h>
|
||||
#include <systick.h>
|
||||
#include "imlib.h"
|
||||
#include "fb_alloc.h"
|
||||
#include "ff_wrapper.h"
|
||||
#include "py_assert.h"
|
||||
#include "py_helper.h"
|
||||
#include "py_image.h"
|
||||
|
||||
#define RST_PORT GPIOD
|
||||
#define RST_PIN GPIO_PIN_12
|
||||
#define RST_PIN_WRITE(bit) HAL_GPIO_WritePin(RST_PORT, RST_PIN, bit);
|
||||
|
||||
#define RS_PORT GPIOD
|
||||
#define RS_PIN GPIO_PIN_13
|
||||
#define RS_PIN_WRITE(bit) HAL_GPIO_WritePin(RS_PORT, RS_PIN, bit);
|
||||
|
||||
#define CS_PORT GPIOB
|
||||
#define CS_PIN GPIO_PIN_12
|
||||
#define CS_PIN_WRITE(bit) HAL_GPIO_WritePin(CS_PORT, CS_PIN, bit);
|
||||
|
||||
#define LED_PORT GPIOA
|
||||
#define LED_PIN GPIO_PIN_5
|
||||
#define LED_PIN_WRITE(bit) HAL_GPIO_WritePin(LED_PORT, LED_PIN, bit);
|
||||
|
||||
extern mp_obj_t pyb_spi_send(mp_uint_t n_args, const mp_obj_t *pos_args, mp_map_t *kw_args);
|
||||
extern mp_obj_t pyb_spi_make_new(mp_obj_t type_in, mp_uint_t n_args, mp_uint_t n_kw, const mp_obj_t *args);
|
||||
extern mp_obj_t pyb_spi_deinit(mp_obj_t self_in);
|
||||
|
||||
static mp_obj_t spi_port = NULL;
|
||||
static int width = 0;
|
||||
static int height = 0;
|
||||
static enum { LCD_NONE, LCD_SHIELD } type = LCD_NONE;
|
||||
static bool backlight_init = false;
|
||||
|
||||
// Send out 8-bit data using the SPI object.
|
||||
static void lcd_write_command_byte(uint8_t data_byte)
|
||||
{
|
||||
mp_map_t arg_map;
|
||||
arg_map.all_keys_are_qstrs = true;
|
||||
arg_map.is_fixed = true;
|
||||
arg_map.is_ordered = true;
|
||||
arg_map.used = 0;
|
||||
arg_map.alloc = 0;
|
||||
arg_map.table = NULL;
|
||||
|
||||
CS_PIN_WRITE(false);
|
||||
RS_PIN_WRITE(false); // command
|
||||
pyb_spi_send(
|
||||
2, (mp_obj_t []) {
|
||||
spi_port,
|
||||
mp_obj_new_int(data_byte)
|
||||
},
|
||||
&arg_map
|
||||
);
|
||||
CS_PIN_WRITE(true);
|
||||
}
|
||||
|
||||
// Send out 8-bit data using the SPI object.
|
||||
static void lcd_write_data_byte(uint8_t data_byte)
|
||||
{
|
||||
mp_map_t arg_map;
|
||||
arg_map.all_keys_are_qstrs = true;
|
||||
arg_map.is_fixed = true;
|
||||
arg_map.is_ordered = true;
|
||||
arg_map.used = 0;
|
||||
arg_map.alloc = 0;
|
||||
arg_map.table = NULL;
|
||||
|
||||
CS_PIN_WRITE(false);
|
||||
RS_PIN_WRITE(true); // data
|
||||
pyb_spi_send(
|
||||
2, (mp_obj_t []) {
|
||||
spi_port,
|
||||
mp_obj_new_int(data_byte)
|
||||
},
|
||||
&arg_map
|
||||
);
|
||||
CS_PIN_WRITE(true);
|
||||
}
|
||||
|
||||
// Send out 8-bit data using the SPI object.
|
||||
static void lcd_write_command(uint8_t data_byte, uint32_t len, uint8_t *dat)
|
||||
{
|
||||
lcd_write_command_byte(data_byte);
|
||||
for (uint32_t i=0; i<len; i++) lcd_write_data_byte(dat[i]);
|
||||
}
|
||||
|
||||
// Send out 8-bit data using the SPI object.
|
||||
static void lcd_write_data(uint32_t len, uint8_t *dat)
|
||||
{
|
||||
mp_obj_str_t arg_str;
|
||||
arg_str.base.type = &mp_type_bytes;
|
||||
arg_str.hash = 0;
|
||||
arg_str.len = len;
|
||||
arg_str.data = dat;
|
||||
|
||||
mp_map_t arg_map;
|
||||
arg_map.all_keys_are_qstrs = true;
|
||||
arg_map.is_fixed = true;
|
||||
arg_map.is_ordered = true;
|
||||
arg_map.used = 0;
|
||||
arg_map.alloc = 0;
|
||||
arg_map.table = NULL;
|
||||
|
||||
CS_PIN_WRITE(false);
|
||||
RS_PIN_WRITE(true); // data
|
||||
pyb_spi_send(
|
||||
2, (mp_obj_t []) {
|
||||
spi_port,
|
||||
&arg_str
|
||||
},
|
||||
&arg_map
|
||||
);
|
||||
CS_PIN_WRITE(true);
|
||||
}
|
||||
|
||||
static mp_obj_t py_lcd_deinit()
|
||||
{
|
||||
switch (type) {
|
||||
case LCD_NONE:
|
||||
return mp_const_none;
|
||||
case LCD_SHIELD:
|
||||
HAL_GPIO_DeInit(RST_PORT, RST_PIN);
|
||||
HAL_GPIO_DeInit(RS_PORT, RS_PIN);
|
||||
HAL_GPIO_DeInit(CS_PORT, CS_PIN);
|
||||
pyb_spi_deinit(spi_port);
|
||||
spi_port = NULL;
|
||||
width = 0;
|
||||
height = 0;
|
||||
type = LCD_NONE;
|
||||
if (backlight_init) {
|
||||
HAL_GPIO_DeInit(LED_PORT, LED_PIN);
|
||||
backlight_init = false;
|
||||
}
|
||||
return mp_const_none;
|
||||
}
|
||||
return mp_const_none;
|
||||
}
|
||||
|
||||
static mp_obj_t py_lcd_init(uint n_args, const mp_obj_t *args, mp_map_t *kw_args)
|
||||
{
|
||||
py_lcd_deinit();
|
||||
switch (py_helper_lookup_int(kw_args, MP_OBJ_NEW_QSTR(MP_QSTR_type), LCD_SHIELD)) {
|
||||
case LCD_NONE:
|
||||
return mp_const_none;
|
||||
case LCD_SHIELD:
|
||||
{
|
||||
GPIO_InitTypeDef GPIO_InitStructure;
|
||||
GPIO_InitStructure.Pull = GPIO_NOPULL;
|
||||
GPIO_InitStructure.Speed = GPIO_SPEED_LOW;
|
||||
GPIO_InitStructure.Mode = GPIO_MODE_OUTPUT_OD;
|
||||
|
||||
GPIO_InitStructure.Pin = CS_PIN;
|
||||
CS_PIN_WRITE(true); // Set first to prevent glitches.
|
||||
HAL_GPIO_Init(CS_PORT, &GPIO_InitStructure);
|
||||
|
||||
GPIO_InitStructure.Mode = GPIO_MODE_OUTPUT_PP;
|
||||
|
||||
GPIO_InitStructure.Pin = RST_PIN;
|
||||
RST_PIN_WRITE(true); // Set first to prevent glitches.
|
||||
HAL_GPIO_Init(RST_PORT, &GPIO_InitStructure);
|
||||
|
||||
GPIO_InitStructure.Pin = RS_PIN;
|
||||
RS_PIN_WRITE(true); // Set first to prevent glitches.
|
||||
HAL_GPIO_Init(RS_PORT, &GPIO_InitStructure);
|
||||
|
||||
spi_port = pyb_spi_make_new(NULL,
|
||||
2, // n_args
|
||||
3, // n_kw
|
||||
(mp_obj_t []) {
|
||||
MP_OBJ_NEW_SMALL_INT(2), // SPI Port
|
||||
MP_OBJ_NEW_SMALL_INT(SPI_MODE_MASTER),
|
||||
MP_OBJ_NEW_QSTR(MP_QSTR_baudrate),
|
||||
MP_OBJ_NEW_SMALL_INT(1000000000/66), // 66 ns clk period
|
||||
MP_OBJ_NEW_QSTR(MP_QSTR_polarity),
|
||||
MP_OBJ_NEW_SMALL_INT(0),
|
||||
MP_OBJ_NEW_QSTR(MP_QSTR_phase),
|
||||
MP_OBJ_NEW_SMALL_INT(0)
|
||||
}
|
||||
);
|
||||
width = 128;
|
||||
height = 160;
|
||||
type = LCD_SHIELD;
|
||||
backlight_init = false;
|
||||
|
||||
RST_PIN_WRITE(false);
|
||||
systick_sleep(100);
|
||||
RST_PIN_WRITE(true);
|
||||
systick_sleep(100);
|
||||
lcd_write_command_byte(0x11); // Sleep Exit
|
||||
systick_sleep(120);
|
||||
|
||||
// Memory Data Access Control
|
||||
lcd_write_command(0x36, 1, (uint8_t []) {0xC0});
|
||||
|
||||
// Interface Pixel Format
|
||||
lcd_write_command(0x3A, 1, (uint8_t []) {0x05});
|
||||
|
||||
// Display on
|
||||
lcd_write_command_byte(0x29);
|
||||
|
||||
return mp_const_none;
|
||||
}
|
||||
}
|
||||
return mp_const_none;
|
||||
}
|
||||
|
||||
static mp_obj_t py_lcd_width()
|
||||
{
|
||||
if (type == LCD_NONE) return mp_const_none;
|
||||
return mp_obj_new_int(width);
|
||||
}
|
||||
|
||||
static mp_obj_t py_lcd_height()
|
||||
{
|
||||
if (type == LCD_NONE) return mp_const_none;
|
||||
return mp_obj_new_int(height);
|
||||
}
|
||||
|
||||
static mp_obj_t py_lcd_type()
|
||||
{
|
||||
if (type == LCD_NONE) return mp_const_none;
|
||||
return mp_obj_new_int(type);
|
||||
}
|
||||
|
||||
static mp_obj_t py_lcd_set_backlight(mp_obj_t state_obj)
|
||||
{
|
||||
switch (type) {
|
||||
case LCD_NONE:
|
||||
return mp_const_none;
|
||||
case LCD_SHIELD:
|
||||
{
|
||||
bool bit = !!mp_obj_get_int(state_obj);
|
||||
if (!backlight_init) {
|
||||
GPIO_InitTypeDef GPIO_InitStructure;
|
||||
GPIO_InitStructure.Pull = GPIO_NOPULL;
|
||||
GPIO_InitStructure.Speed = GPIO_SPEED_LOW;
|
||||
GPIO_InitStructure.Mode = GPIO_MODE_OUTPUT_OD;
|
||||
GPIO_InitStructure.Pin = LED_PIN;
|
||||
LED_PIN_WRITE(bit); // Set first to prevent glitches.
|
||||
HAL_GPIO_Init(LED_PORT, &GPIO_InitStructure);
|
||||
backlight_init = true;
|
||||
}
|
||||
LED_PIN_WRITE(bit);
|
||||
return mp_const_none;
|
||||
}
|
||||
}
|
||||
return mp_const_none;
|
||||
}
|
||||
|
||||
static mp_obj_t py_lcd_get_backlight()
|
||||
{
|
||||
switch (type) {
|
||||
case LCD_NONE:
|
||||
return mp_const_none;
|
||||
case LCD_SHIELD:
|
||||
if (!backlight_init) {
|
||||
return mp_const_none;
|
||||
}
|
||||
return mp_obj_new_int(HAL_GPIO_ReadPin(LED_PORT, LED_PIN));
|
||||
}
|
||||
return mp_const_none;
|
||||
}
|
||||
|
||||
static mp_obj_t py_lcd_display(uint n_args, const mp_obj_t *args, mp_map_t *kw_args)
|
||||
{
|
||||
image_t *arg_img = py_image_cobj(args[0]);
|
||||
PY_ASSERT_FALSE_MSG(IM_IS_JPEG(arg_img),
|
||||
"Operation not supported on JPEG");
|
||||
|
||||
rectangle_t arg_r;
|
||||
py_helper_lookup_rectangle(kw_args, arg_img, &arg_r);
|
||||
rectangle_t rect;
|
||||
if (!rectangle_subimg(arg_img, &arg_r, &rect)) ff_no_intersection(NULL);
|
||||
|
||||
// Fit X.
|
||||
int l_pad = 0, r_pad = 0;
|
||||
if (rect.w > width) {
|
||||
int adjust = rect.w - width;
|
||||
rect.w -= adjust;
|
||||
rect.x += adjust / 2;
|
||||
} else if (rect.w < width) {
|
||||
int adjust = width - rect.w;
|
||||
l_pad = adjust / 2;
|
||||
r_pad = (adjust + 1) / 2;
|
||||
}
|
||||
|
||||
// Fit Y.
|
||||
int t_pad = 0, b_pad = 0;
|
||||
if (rect.h > height) {
|
||||
int adjust = rect.h - height;
|
||||
rect.h -= adjust;
|
||||
rect.y += adjust / 2;
|
||||
} else if (rect.h < height) {
|
||||
int adjust = height - rect.h;
|
||||
t_pad = adjust / 2;
|
||||
b_pad = (adjust + 1) / 2;
|
||||
}
|
||||
|
||||
switch (type) {
|
||||
case LCD_NONE:
|
||||
return mp_const_none;
|
||||
case LCD_SHIELD:
|
||||
lcd_write_command_byte(0x2C);
|
||||
uint8_t *zero = fb_alloc0(width*2);
|
||||
uint16_t *line = fb_alloc(width*2);
|
||||
for (int i=0; i<t_pad; i++) {
|
||||
lcd_write_data(width*2, zero);
|
||||
}
|
||||
for (int i=0; i<rect.h; i++) {
|
||||
if (l_pad) {
|
||||
lcd_write_data(l_pad*2, zero); // l_pad < width
|
||||
}
|
||||
if (IM_IS_GS(arg_img)) {
|
||||
for (int j=0; j<rect.w; j++) {
|
||||
uint8_t pixel = IM_GET_GS_PIXEL(arg_img, (rect.x + j), (rect.y + i));
|
||||
line[j] = IM_RGB565(IM_R825(pixel),IM_G826(pixel),IM_B825(pixel));
|
||||
}
|
||||
lcd_write_data(rect.w*2, (uint8_t *) line);
|
||||
} else {
|
||||
lcd_write_data(rect.w*2, (uint8_t *)
|
||||
(((uint16_t *) arg_img->pixels) +
|
||||
((rect.y + i) * arg_img->w) + rect.x));
|
||||
}
|
||||
if (r_pad) {
|
||||
lcd_write_data(r_pad*2, zero); // r_pad < width
|
||||
}
|
||||
}
|
||||
for (int i=0; i<b_pad; i++) {
|
||||
lcd_write_data(width*2, zero);
|
||||
}
|
||||
fb_free();
|
||||
fb_free();
|
||||
return mp_const_none;
|
||||
}
|
||||
return mp_const_none;
|
||||
}
|
||||
|
||||
static mp_obj_t py_lcd_clear()
|
||||
{
|
||||
switch (type) {
|
||||
case LCD_NONE:
|
||||
return mp_const_none;
|
||||
case LCD_SHIELD:
|
||||
lcd_write_command_byte(0x2C);
|
||||
uint8_t *zero = fb_alloc0(width*2);
|
||||
for (int i=0; i<height; i++) {
|
||||
lcd_write_data(width*2, zero);
|
||||
}
|
||||
fb_free();
|
||||
return mp_const_none;
|
||||
}
|
||||
return mp_const_none;
|
||||
}
|
||||
|
||||
STATIC MP_DEFINE_CONST_FUN_OBJ_KW(py_lcd_init_obj, 0, py_lcd_init);
|
||||
STATIC MP_DEFINE_CONST_FUN_OBJ_0(py_lcd_deinit_obj, py_lcd_deinit);
|
||||
STATIC MP_DEFINE_CONST_FUN_OBJ_0(py_lcd_width_obj, py_lcd_width);
|
||||
STATIC MP_DEFINE_CONST_FUN_OBJ_0(py_lcd_height_obj, py_lcd_height);
|
||||
STATIC MP_DEFINE_CONST_FUN_OBJ_0(py_lcd_type_obj, py_lcd_type);
|
||||
STATIC MP_DEFINE_CONST_FUN_OBJ_1(py_lcd_set_backlight_obj, py_lcd_set_backlight);
|
||||
STATIC MP_DEFINE_CONST_FUN_OBJ_0(py_lcd_get_backlight_obj, py_lcd_get_backlight);
|
||||
STATIC MP_DEFINE_CONST_FUN_OBJ_KW(py_lcd_display_obj, 1, py_lcd_display);
|
||||
STATIC MP_DEFINE_CONST_FUN_OBJ_0(py_lcd_clear_obj, py_lcd_clear);
|
||||
static const mp_map_elem_t globals_dict_table[] = {
|
||||
{ MP_OBJ_NEW_QSTR(MP_QSTR___name__), MP_OBJ_NEW_QSTR(MP_QSTR_lcd) },
|
||||
{ MP_OBJ_NEW_QSTR(MP_QSTR_init), (mp_obj_t)&py_lcd_init_obj },
|
||||
{ MP_OBJ_NEW_QSTR(MP_QSTR_deinit), (mp_obj_t)&py_lcd_deinit_obj },
|
||||
{ MP_OBJ_NEW_QSTR(MP_QSTR_width), (mp_obj_t)&py_lcd_width_obj },
|
||||
{ MP_OBJ_NEW_QSTR(MP_QSTR_height), (mp_obj_t)&py_lcd_height_obj },
|
||||
{ MP_OBJ_NEW_QSTR(MP_QSTR_type), (mp_obj_t)&py_lcd_type_obj },
|
||||
{ MP_OBJ_NEW_QSTR(MP_QSTR_set_backlight), (mp_obj_t)&py_lcd_set_backlight_obj },
|
||||
{ MP_OBJ_NEW_QSTR(MP_QSTR_get_backlight), (mp_obj_t)&py_lcd_get_backlight_obj },
|
||||
{ MP_OBJ_NEW_QSTR(MP_QSTR_display), (mp_obj_t)&py_lcd_display_obj },
|
||||
{ MP_OBJ_NEW_QSTR(MP_QSTR_clear), (mp_obj_t)&py_lcd_clear_obj },
|
||||
{ NULL, NULL },
|
||||
};
|
||||
STATIC MP_DEFINE_CONST_DICT(globals_dict, globals_dict_table);
|
||||
|
||||
const mp_obj_module_t lcd_module = {
|
||||
.base = { &mp_type_module },
|
||||
.name = MP_QSTR_lcd,
|
||||
.globals = (mp_obj_t)&globals_dict,
|
||||
};
|
||||
|
||||
void py_lcd_init0()
|
||||
{
|
||||
py_lcd_deinit();
|
||||
}
|
||||
12
src/omv/py/py_lcd.h
Normal file
12
src/omv/py/py_lcd.h
Normal file
@ -0,0 +1,12 @@
|
||||
/*
|
||||
* This file is part of the OpenMV project.
|
||||
* Copyright (c) 2013/2014 Ibrahim Abdelkader <i.abdalkader@gmail.com>
|
||||
* This work is licensed under the MIT license, see the file LICENSE for details.
|
||||
*
|
||||
* LCD Python module.
|
||||
*
|
||||
*/
|
||||
#ifndef __PY_LCD_H__
|
||||
#define __PY_LCD_H__
|
||||
void py_lcd_init0();
|
||||
#endif // __PY_LCD_H__
|
||||
15
usr/examples/Example1-LCD-Shield-Demo.py
Normal file
15
usr/examples/Example1-LCD-Shield-Demo.py
Normal file
@ -0,0 +1,15 @@
|
||||
# Example 1 - LCD Shield Demo
|
||||
#
|
||||
# Note: To run this example you will need a LCD Shield for your OpenMV Cam.
|
||||
#
|
||||
# The LCD Shield allows you to view your OpenMV Cam's frame buffer on the go.
|
||||
|
||||
import sensor, image, lcd
|
||||
|
||||
sensor.reset() # Initialize the camera sensor.
|
||||
sensor.set_pixformat(sensor.RGB565) # or sensor.GRAYSCALE
|
||||
sensor.set_framesize(sensor.QQVGA2) # Special 128x160 framesize for LCD Shield.
|
||||
lcd.init() # Initialize the lcd screen.
|
||||
|
||||
while(True):
|
||||
lcd.display(sensor.snapshot()) # Take a picture and display the image.
|
||||
Loading…
Reference in New Issue
Block a user