plasma: Support pins given as machine.Pin obj.

Author: Gadgetoid

micropython/modules/plasma/plasma.cpp

@@ -9,6 +9,7 @@ using namespace plasma;
 extern "C" {
 #include "plasma.h"
 #include "py/builtin.h"
+#include "machine_pin.h"
 
 typedef struct _mp_obj_float_t {
     mp_obj_base_t base;
@@ -64,7 +65,7 @@ mp_obj_t PlasmaWS2812_make_new(const mp_obj_type_t *type, size_t n_args, size_t
         { MP_QSTR_num_leds, MP_ARG_REQUIRED | MP_ARG_INT },
         { MP_QSTR_pio, MP_ARG_REQUIRED | MP_ARG_INT },
         { MP_QSTR_sm, MP_ARG_REQUIRED | MP_ARG_INT },
-        { MP_QSTR_dat, MP_ARG_REQUIRED | MP_ARG_INT },
+        { MP_QSTR_dat, MP_ARG_REQUIRED | MP_ARG_OBJ },
         { MP_QSTR_freq, MP_ARG_INT, {.u_int = WS2812::DEFAULT_SERIAL_FREQ} },
         { MP_QSTR_buffer, MP_ARG_OBJ, {.u_obj = nullptr} },
         { MP_QSTR_rgbw, MP_ARG_BOOL, {.u_bool = false} },
@@ -78,7 +79,7 @@ mp_obj_t PlasmaWS2812_make_new(const mp_obj_type_t *type, size_t n_args, size_t
     int num_leds = args[ARG_num_leds].u_int;
     PIO pio = args[ARG_pio].u_int == 0 ? pio0 : pio1;
     int sm = args[ARG_sm].u_int;
-    int dat = args[ARG_dat].u_int;
+    int dat = mp_hal_get_pin_obj(args[ARG_dat].u_obj);
     int freq = args[ARG_freq].u_int;
     bool rgbw = args[ARG_rgbw].u_bool;
     WS2812::COLOR_ORDER color_order = (WS2812::COLOR_ORDER)args[ARG_color_order].u_int;
@@ -269,8 +270,8 @@ mp_obj_t PlasmaAPA102_make_new(const mp_obj_type_t *type, size_t n_args, size_t
         { MP_QSTR_num_leds, MP_ARG_REQUIRED | MP_ARG_INT },
         { MP_QSTR_pio, MP_ARG_REQUIRED | MP_ARG_INT },
         { MP_QSTR_sm, MP_ARG_REQUIRED | MP_ARG_INT },
-        { MP_QSTR_dat, MP_ARG_REQUIRED | MP_ARG_INT },
-        { MP_QSTR_clk, MP_ARG_REQUIRED | MP_ARG_INT },
+        { MP_QSTR_dat, MP_ARG_REQUIRED | MP_ARG_OBJ },
+        { MP_QSTR_clk, MP_ARG_REQUIRED | MP_ARG_OBJ },
         { MP_QSTR_freq, MP_ARG_INT, {.u_int = APA102::DEFAULT_SERIAL_FREQ} },
         { MP_QSTR_buffer, MP_ARG_OBJ, {.u_obj = nullptr} },
     };
@@ -282,8 +283,8 @@ mp_obj_t PlasmaAPA102_make_new(const mp_obj_type_t *type, size_t n_args, size_t
     int num_leds = args[ARG_num_leds].u_int;
     PIO pio = args[ARG_pio].u_int == 0 ? pio0 : pio1;
     int sm = args[ARG_sm].u_int;
-    int dat = args[ARG_dat].u_int;
-    int clk = args[ARG_clk].u_int;
+    int dat = mp_hal_get_pin_obj(args[ARG_dat].u_obj);
+    int clk = mp_hal_get_pin_obj(args[ARG_clk].u_obj);
     int freq = args[ARG_freq].u_int;
 
     APA102::RGB *buffer = nullptr;