BH1745 MicroPython bindings & example

pull/29/head
Phil Howard 2021-07-12 15:26:41 +01:00
rodzic ea6acf2207
commit b80b714c4f
15 zmienionych plików z 378 dodań i 11 usunięć

Wyświetl plik

@ -2,11 +2,11 @@
#include <algorithm>
namespace pimoroni {
int BH1745::init() {
bool BH1745::init() {
reset();
if (this->get_chip_id() != BH1745_CHIP_ID || this->get_manufacturer() != BH1745_MANUFACTURER) {
return 1;
return false;
}
reset();
@ -20,11 +20,15 @@ namespace pimoroni {
sleep_ms(320);
return 0;
return true;
}
i2c_inst_t* BH1745::get_i2c() const {
return i2c->get_i2c();
I2C* BH1745::get_i2c() const {
return i2c;
}
int BH1745::get_address() const {
return address;
}
uint8_t BH1745::get_chip_id() {

Wyświetl plik

@ -38,9 +38,10 @@ namespace pimoroni {
BH1745(i2c_inst_t *i2c, uint8_t addr, uint8_t sda, uint8_t scl, uint interrupt) :
BH1745(new I2C(sda, scl), address) {};
int init();
bool init();
i2c_inst_t* get_i2c() const;
I2C* get_i2c() const;
int get_address() const;
uint8_t get_chip_id();
uint8_t get_manufacturer();

Wyświetl plik

@ -1,10 +1,10 @@
add_executable(
breakout_bh1745
breakout_bh1745_demo
demo.cpp
)
# Pull in pico libraries that we need
target_link_libraries(breakout_bh1745 pico_stdlib hardware_spi bh1745)
target_link_libraries(breakout_bh1745_demo pico_stdlib hardware_spi bh1745)
# create map/bin/hex file etc.
pico_add_extra_outputs(breakout_bh1745)
pico_add_extra_outputs(breakout_bh1745_demo)

Wyświetl plik

@ -13,7 +13,7 @@ BH1745 bh1745(&i2c, BH1745::DEFAULT_I2C_ADDRESS);
int main() {
stdio_init_all();
if (bh1745.init() == 1) {
if (!bh1745.init()) {
printf("Failed to set up sensor\n");
return 1;
}

Wyświetl plik

@ -14,6 +14,7 @@ add_subdirectory(breakout_trackball)
add_subdirectory(breakout_sgp30)
add_subdirectory(breakout_as7262)
add_subdirectory(breakout_msa301)
add_subdirectory(breakout_bh1745)
add_subdirectory(pico_graphics)
add_subdirectory(pico_display)
add_subdirectory(pico_unicorn)

Wyświetl plik

@ -0,0 +1 @@
include(breakout_bh1745.cmake)

Wyświetl plik

@ -0,0 +1,11 @@
set(LIB_NAME breakout_bh1745)
add_library(${LIB_NAME} INTERFACE)
target_sources(${LIB_NAME} INTERFACE
${CMAKE_CURRENT_LIST_DIR}/${LIB_NAME}.cpp
)
target_include_directories(${LIB_NAME} INTERFACE ${CMAKE_CURRENT_LIST_DIR})
# Pull in pico libraries that we need
target_link_libraries(${LIB_NAME} INTERFACE pico_stdlib bh1745 pimoroni_i2c)

Wyświetl plik

@ -0,0 +1,5 @@
#include "breakout_bh1745.hpp"
namespace pimoroni {
}

Wyświetl plik

@ -0,0 +1,8 @@
#pragma once
#include "drivers/bh1745/bh1745.hpp"
namespace pimoroni {
typedef BH1745 BreakoutBH1745;
}

Wyświetl plik

@ -0,0 +1,21 @@
import time
import pimoroni_i2c
import breakout_bh1745
PINS_BREAKOUT_GARDEN = {"sda": 4, "scl": 5}
PINS_PICO_EXPLORER = {"sda": 20, "scl": 21}
I2C = pimoroni_i2c.PimoroniI2C(**PINS_BREAKOUT_GARDEN)
bh1745 = breakout_bh1745.BreakoutBH1745(I2C)
bh1745.leds(True)
while True:
rgbc_raw = bh1745.rgbc_raw()
rgb_clamped = bh1745.rgbc_clamped()
rgb_scaled = bh1745.rgbc_scaled()
print("Raw: {}, {}, {}, {}".format(*rgbc_raw))
print("Clamped: {}, {}, {}, {}".format(*rgb_clamped))
print("Scaled: #{:02x}{:02x}{:02x}".format(*rgb_scaled))
time.sleep(5)

Wyświetl plik

@ -0,0 +1,60 @@
#include "breakout_bh1745.h"
////////////////////////////////////////////////////////////////////////////////////////////////////
// BreakoutBH1745 Class
////////////////////////////////////////////////////////////////////////////////////////////////////
/***** Methods *****/
MP_DEFINE_CONST_FUN_OBJ_1(BreakoutBH1745_chip_id_obj, BreakoutBH1745_chip_id);
MP_DEFINE_CONST_FUN_OBJ_1(BreakoutBH1745_manufacturer_id_obj, BreakoutBH1745_manufacturer_id);
MP_DEFINE_CONST_FUN_OBJ_1(BreakoutBH1745_rgbc_raw_obj, BreakoutBH1745_rgbc_raw);
MP_DEFINE_CONST_FUN_OBJ_1(BreakoutBH1745_rgbc_clamped_obj, BreakoutBH1745_rgbc_clamped);
MP_DEFINE_CONST_FUN_OBJ_1(BreakoutBH1745_rgbc_scaled_obj, BreakoutBH1745_rgbc_scaled);
MP_DEFINE_CONST_FUN_OBJ_KW(BreakoutBH1745_leds_obj, 2, BreakoutBH1745_leds);
MP_DEFINE_CONST_FUN_OBJ_KW(BreakoutBH1745_threshold_obj, 3, BreakoutBH1745_threshold);
MP_DEFINE_CONST_FUN_OBJ_KW(BreakoutBH1745_measurement_time_ms_obj, 2, BreakoutBH1745_measurement_time_ms);
/***** Binding of Methods *****/
STATIC const mp_rom_map_elem_t BreakoutBH1745_locals_dict_table[] = {
{ MP_ROM_QSTR(MP_QSTR_chip_id), MP_ROM_PTR(&BreakoutBH1745_chip_id_obj) },
{ MP_ROM_QSTR(MP_QSTR_manufacturer_id), MP_ROM_PTR(&BreakoutBH1745_manufacturer_id_obj) },
{ MP_ROM_QSTR(MP_QSTR_rgbc_raw), MP_ROM_PTR(&BreakoutBH1745_rgbc_raw_obj) },
{ MP_ROM_QSTR(MP_QSTR_rgbc_clamped), MP_ROM_PTR(&BreakoutBH1745_rgbc_clamped_obj) },
{ MP_ROM_QSTR(MP_QSTR_rgbc_scaled), MP_ROM_PTR(&BreakoutBH1745_rgbc_scaled_obj) },
{ MP_ROM_QSTR(MP_QSTR_threshold), MP_ROM_PTR(&BreakoutBH1745_threshold_obj) },
{ MP_ROM_QSTR(MP_QSTR_measurement_time_ms), MP_ROM_PTR(&BreakoutBH1745_measurement_time_ms_obj) },
{ MP_ROM_QSTR(MP_QSTR_leds), MP_ROM_PTR(&BreakoutBH1745_leds_obj) },
};
STATIC MP_DEFINE_CONST_DICT(BreakoutBH1745_locals_dict, BreakoutBH1745_locals_dict_table);
/***** Class Definition *****/
const mp_obj_type_t breakout_bh1745_BreakoutBH1745_type = {
{ &mp_type_type },
.name = MP_QSTR_breakout_bh1745,
.print = BreakoutBH1745_print,
.make_new = BreakoutBH1745_make_new,
.locals_dict = (mp_obj_dict_t*)&BreakoutBH1745_locals_dict,
};
////////////////////////////////////////////////////////////////////////////////////////////////////
// breakout_bh1745 Module
////////////////////////////////////////////////////////////////////////////////////////////////////
/***** Globals Table *****/
STATIC const mp_map_elem_t breakout_bh1745_globals_table[] = {
{ MP_OBJ_NEW_QSTR(MP_QSTR___name__), MP_OBJ_NEW_QSTR(MP_QSTR_breakout_bh1745) },
{ MP_OBJ_NEW_QSTR(MP_QSTR_BreakoutBH1745), (mp_obj_t)&breakout_bh1745_BreakoutBH1745_type },
};
STATIC MP_DEFINE_CONST_DICT(mp_module_breakout_bh1745_globals, breakout_bh1745_globals_table);
/***** Module Definition *****/
const mp_obj_module_t breakout_bh1745_user_cmodule = {
.base = { &mp_type_module },
.globals = (mp_obj_dict_t*)&mp_module_breakout_bh1745_globals,
};
////////////////////////////////////////////////////////////////////////////////////////////////////
MP_REGISTER_MODULE(MP_QSTR_breakout_bh1745, breakout_bh1745_user_cmodule, MODULE_BREAKOUT_LTR559_ENABLED);
////////////////////////////////////////////////////////////////////////////////////////////////////
////////////////////////////////////////////////////////////////////////////////////////////////////

Wyświetl plik

@ -0,0 +1,208 @@
#include "libraries/breakout_bh1745/breakout_bh1745.hpp"
#include <cstdio>
#define MP_OBJ_TO_PTR2(o, t) ((t *)(uintptr_t)(o))
// SDA/SCL on even/odd pins, I2C0/I2C1 on even/odd pairs of pins.
#define IS_VALID_SCL(i2c, pin) (((pin) & 1) == 1 && (((pin) & 2) >> 1) == (i2c))
#define IS_VALID_SDA(i2c, pin) (((pin) & 1) == 0 && (((pin) & 2) >> 1) == (i2c))
using namespace pimoroni;
extern "C" {
#include "breakout_bh1745.h"
#include "pimoroni_i2c.h"
/***** I2C Struct *****/
typedef struct _PimoroniI2C_obj_t {
mp_obj_base_t base;
I2C *i2c;
} _PimoroniI2C_obj_t;
/***** Variables Struct *****/
typedef struct _breakout_bh1745_BreakoutBH1745_obj_t {
mp_obj_base_t base;
BreakoutBH1745 *breakout;
} breakout_bh1745_BreakoutBH1745_obj_t;
/***** Print *****/
void BreakoutBH1745_print(const mp_print_t *print, mp_obj_t self_in, mp_print_kind_t kind) {
(void)kind; //Unused input parameter
breakout_bh1745_BreakoutBH1745_obj_t *self = MP_OBJ_TO_PTR2(self_in, breakout_bh1745_BreakoutBH1745_obj_t);
BreakoutBH1745* breakout = self->breakout;
mp_print_str(print, "BreakoutBH1745(");
mp_print_str(print, "i2c = ");
mp_obj_print_helper(print, mp_obj_new_int((breakout->get_i2c()->get_i2c() == i2c0) ? 0 : 1), PRINT_REPR);
mp_print_str(print, ", address = 0x");
char buf[3];
sprintf(buf, "%02X", breakout->get_address());
mp_print_str(print, buf);
mp_print_str(print, ", sda = ");
mp_obj_print_helper(print, mp_obj_new_int(breakout->get_i2c()->get_sda()), PRINT_REPR);
mp_print_str(print, ", scl = ");
mp_obj_print_helper(print, mp_obj_new_int(breakout->get_i2c()->get_scl()), PRINT_REPR);
mp_print_str(print, ")");
}
/***** Constructor *****/
mp_obj_t BreakoutBH1745_make_new(const mp_obj_type_t *type, size_t n_args, size_t n_kw, const mp_obj_t *all_args) {
breakout_bh1745_BreakoutBH1745_obj_t *self = nullptr;
enum { ARG_i2c, ARG_address };
static const mp_arg_t allowed_args[] = {
{ MP_QSTR_i2c, MP_ARG_OBJ, {.u_obj = nullptr} },
{ MP_QSTR_address, MP_ARG_INT, {.u_int = BH1745::DEFAULT_I2C_ADDRESS} },
};
// Parse args.
mp_arg_val_t args[MP_ARRAY_SIZE(allowed_args)];
mp_arg_parse_all_kw_array(n_args, n_kw, all_args, MP_ARRAY_SIZE(allowed_args), allowed_args, args);
if(!MP_OBJ_IS_TYPE(args[ARG_i2c].u_obj, &PimoroniI2C_type)) {
mp_raise_ValueError(MP_ERROR_TEXT("BreakoutBH1745: Bad i2C object"));
return mp_const_none;
}
_PimoroniI2C_obj_t *i2c = (_PimoroniI2C_obj_t *)MP_OBJ_TO_PTR(args[ARG_i2c].u_obj);
self = m_new_obj(breakout_bh1745_BreakoutBH1745_obj_t);
self->base.type = &breakout_bh1745_BreakoutBH1745_type;
self->breakout = new BreakoutBH1745(i2c->i2c, args[ARG_address].u_int);
if(!self->breakout->init()) {
mp_raise_msg(&mp_type_RuntimeError, "BreakoutBH1745: breakout not found when initialising");
}
return MP_OBJ_FROM_PTR(self);
}
mp_obj_t BreakoutBH1745_chip_id(mp_obj_t self_in) {
breakout_bh1745_BreakoutBH1745_obj_t *self = MP_OBJ_TO_PTR2(self_in, breakout_bh1745_BreakoutBH1745_obj_t);
return mp_obj_new_int(self->breakout->get_chip_id());
}
mp_obj_t BreakoutBH1745_manufacturer_id(mp_obj_t self_in) {
breakout_bh1745_BreakoutBH1745_obj_t *self = MP_OBJ_TO_PTR2(self_in, breakout_bh1745_BreakoutBH1745_obj_t);
return mp_obj_new_int(self->breakout->get_manufacturer());
}
mp_obj_t BreakoutBH1745_rgbc_raw(mp_obj_t self_in) {
breakout_bh1745_BreakoutBH1745_obj_t *self = MP_OBJ_TO_PTR2(self_in, breakout_bh1745_BreakoutBH1745_obj_t);
const rgbc_t reading = self->breakout->get_rgbc_raw();
mp_obj_t tuple[4];
tuple[RGBC_R] = mp_obj_new_int(reading.r);
tuple[RGBC_G] = mp_obj_new_int(reading.g);
tuple[RGBC_B] = mp_obj_new_int(reading.b);
tuple[RGBC_C] = mp_obj_new_int(reading.c);
return mp_obj_new_tuple(4, tuple);
}
mp_obj_t BreakoutBH1745_rgbc_clamped(mp_obj_t self_in) {
breakout_bh1745_BreakoutBH1745_obj_t *self = MP_OBJ_TO_PTR2(self_in, breakout_bh1745_BreakoutBH1745_obj_t);
const rgbc_t reading = self->breakout->get_rgb_clamped();
mp_obj_t tuple[4];
tuple[RGBC_R] = mp_obj_new_int(reading.r);
tuple[RGBC_G] = mp_obj_new_int(reading.g);
tuple[RGBC_B] = mp_obj_new_int(reading.b);
tuple[RGBC_C] = mp_obj_new_int(reading.c);
return mp_obj_new_tuple(4, tuple);
}
mp_obj_t BreakoutBH1745_rgbc_scaled(mp_obj_t self_in) {
breakout_bh1745_BreakoutBH1745_obj_t *self = MP_OBJ_TO_PTR2(self_in, breakout_bh1745_BreakoutBH1745_obj_t);
const rgbc_t reading = self->breakout->get_rgb_scaled();
mp_obj_t tuple[4];
tuple[RGBC_R] = mp_obj_new_int(reading.r);
tuple[RGBC_G] = mp_obj_new_int(reading.g);
tuple[RGBC_B] = mp_obj_new_int(reading.b);
tuple[RGBC_C] = mp_obj_new_int(reading.c);
return mp_obj_new_tuple(4, tuple);
}
mp_obj_t BreakoutBH1745_threshold(size_t n_args, const mp_obj_t *pos_args, mp_map_t *kw_args) {
enum { ARG_self, ARG_lower, ARG_upper };
static const mp_arg_t allowed_args[] = {
{ MP_QSTR_, MP_ARG_REQUIRED | MP_ARG_OBJ },
{ MP_QSTR_lower, MP_ARG_REQUIRED | MP_ARG_INT },
{ MP_QSTR_upper, 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);
breakout_bh1745_BreakoutBH1745_obj_t *self = MP_OBJ_TO_PTR2(args[ARG_self].u_obj, breakout_bh1745_BreakoutBH1745_obj_t);
int lower = args[ARG_lower].u_int;
int upper = args[ARG_upper].u_int;
if(lower < 0 || lower > 65535) {
mp_raise_ValueError("lower out of range. Expected 0 to 65535");
} else if(upper < 0 || upper > 65535) {
mp_raise_ValueError("upper out of range. Expected 0 to 65535");
} else {
self->breakout->set_threshold_low(lower);
self->breakout->set_threshold_high(upper);
}
return mp_const_none;
}
mp_obj_t BreakoutBH1745_measurement_time_ms(size_t n_args, const mp_obj_t *pos_args, mp_map_t *kw_args) {
enum { ARG_self, ARG_time };
static const mp_arg_t allowed_args[] = {
{ MP_QSTR_, MP_ARG_REQUIRED | MP_ARG_OBJ },
{ MP_QSTR_time, 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);
breakout_bh1745_BreakoutBH1745_obj_t *self = MP_OBJ_TO_PTR2(args[ARG_self].u_obj, breakout_bh1745_BreakoutBH1745_obj_t);
int measurement_time = args[ARG_time].u_int;
if(measurement_time < 0 || measurement_time > 65535) {
mp_raise_ValueError("Time out of range. Expected 0 to 65535");
} else {
self->breakout->set_measurement_time_ms(measurement_time);
}
return mp_const_none;
}
mp_obj_t BreakoutBH1745_leds(size_t n_args, const mp_obj_t *pos_args, mp_map_t *kw_args) {
enum { ARG_self, ARG_led_state };
static const mp_arg_t allowed_args[] = {
{ MP_QSTR_, MP_ARG_REQUIRED | MP_ARG_OBJ },
{ MP_QSTR_led_state, MP_ARG_REQUIRED | MP_ARG_BOOL },
};
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);
breakout_bh1745_BreakoutBH1745_obj_t *self = MP_OBJ_TO_PTR2(args[ARG_self].u_obj, breakout_bh1745_BreakoutBH1745_obj_t);
bool state = args[ARG_led_state].u_bool;
self->breakout->set_leds(state);
return mp_const_none;
}
}

Wyświetl plik

@ -0,0 +1,26 @@
// Include MicroPython API.
#include "py/runtime.h"
#include "py/objstr.h"
/***** Constants *****/
enum {
RGBC_R,
RGBC_G,
RGBC_B,
RGBC_C
};
/***** Extern of Class Definition *****/
extern const mp_obj_type_t breakout_bh1745_BreakoutBH1745_type;
/***** Extern of Class Methods *****/
extern void BreakoutBH1745_print(const mp_print_t *print, mp_obj_t self_in, mp_print_kind_t kind);
extern mp_obj_t BreakoutBH1745_make_new(const mp_obj_type_t *type, size_t n_args, size_t n_kw, const mp_obj_t *all_args);
extern mp_obj_t BreakoutBH1745_chip_id(mp_obj_t self_in);
extern mp_obj_t BreakoutBH1745_manufacturer_id(mp_obj_t self_in);
extern mp_obj_t BreakoutBH1745_threshold(size_t n_args, const mp_obj_t *pos_args, mp_map_t *kw_args);
extern mp_obj_t BreakoutBH1745_measurement_time_ms(size_t n_args, const mp_obj_t *pos_args, mp_map_t *kw_args);
extern mp_obj_t BreakoutBH1745_rgbc_raw(mp_obj_t self_in);
extern mp_obj_t BreakoutBH1745_rgbc_clamped(mp_obj_t self_in);
extern mp_obj_t BreakoutBH1745_rgbc_scaled(mp_obj_t self_in);
extern mp_obj_t BreakoutBH1745_leds(size_t n_args, const mp_obj_t *pos_args, mp_map_t *kw_args);

Wyświetl plik

@ -0,0 +1,20 @@
set(MOD_NAME breakout_bh1745)
string(TOUPPER ${MOD_NAME} MOD_NAME_UPPER)
add_library(usermod_${MOD_NAME} INTERFACE)
target_sources(usermod_${MOD_NAME} INTERFACE
${CMAKE_CURRENT_LIST_DIR}/${MOD_NAME}.c
${CMAKE_CURRENT_LIST_DIR}/${MOD_NAME}.cpp
${CMAKE_CURRENT_LIST_DIR}/../../../libraries/${MOD_NAME}/${MOD_NAME}.cpp
${CMAKE_CURRENT_LIST_DIR}/../../../drivers/bh1745/bh1745.cpp
)
target_include_directories(usermod_${MOD_NAME} INTERFACE
${CMAKE_CURRENT_LIST_DIR}
)
target_compile_definitions(usermod_${MOD_NAME} INTERFACE
-DMODULE_${MOD_NAME_UPPER}_ENABLED=1
)
target_link_libraries(usermod INTERFACE usermod_${MOD_NAME})

Wyświetl plik

@ -21,6 +21,7 @@ include(breakout_rtc/micropython)
include(breakout_trackball/micropython)
include(breakout_sgp30/micropython)
include(breakout_colourlcd240x240/micropython)
include(breakout_bh1745/micropython)
include(pico_scroll/micropython)
include(pico_rgb_keypad/micropython)