pimoroni-pico/micropython/modules/pico_scroll/pico_scroll.cpp

101 lines
2.5 KiB
C++

#include "hardware/spi.h"
#include "hardware/sync.h"
#include "pico/binary_info.h"
#include "../../../libraries/pico_scroll/pico_scroll.hpp"
using namespace pimoroni;
PicoScroll *scroll = nullptr;
extern "C" {
#include "pico_scroll.h"
#define NOT_INITIALISED_MSG "Cannot call this function, as picoscroll is not initialised. Call picoscroll.init() first."
mp_obj_t picoscroll_init() {
if(scroll == nullptr)
scroll = new PicoScroll();
scroll->init();
return mp_const_none;
}
mp_obj_t picoscroll_get_width() {
return mp_obj_new_int(PicoScroll::WIDTH);
}
mp_obj_t picoscroll_get_height() {
return mp_obj_new_int(PicoScroll::HEIGHT);
}
mp_obj_t picoscroll_update() {
if(scroll != nullptr)
scroll->update();
else
mp_raise_msg(&mp_type_RuntimeError, NOT_INITIALISED_MSG);
return mp_const_none;
}
mp_obj_t picoscroll_set_pixel(mp_obj_t x_obj, mp_obj_t y_obj, mp_obj_t v_obj) {
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)
mp_raise_ValueError("x or y out of range.");
else {
if(val < 0 || val > 255)
mp_raise_ValueError("val out of range. Expected 0 to 255");
else
scroll->set_pixel(x, y, val);
}
}
else
mp_raise_msg(&mp_type_RuntimeError, NOT_INITIALISED_MSG);
return mp_const_none;
}
mp_obj_t picoscroll_clear() {
if(scroll != nullptr)
scroll->clear();
else
mp_raise_msg(&mp_type_RuntimeError, NOT_INITIALISED_MSG);
return mp_const_none;
}
mp_obj_t picoscroll_is_pressed(mp_obj_t button_obj) {
bool buttonPressed = false;
if(scroll != nullptr) {
int buttonID = mp_obj_get_int(button_obj);
switch(buttonID) {
case 0:
buttonPressed = scroll->is_pressed(PicoScroll::A);
break;
case 1:
buttonPressed = scroll->is_pressed(PicoScroll::B);
break;
case 2:
buttonPressed = scroll->is_pressed(PicoScroll::X);
break;
case 3:
buttonPressed = scroll->is_pressed(PicoScroll::Y);
break;
default:
mp_raise_ValueError("button not valid. Expected 0 to 3");
break;
}
}
else
mp_raise_msg(&mp_type_RuntimeError, NOT_INITIALISED_MSG);
return buttonPressed ? mp_const_true : mp_const_false;
}
}