plasma: Support pins given as machine.Pin obj.

Author: Gadgetoid

micropython/modules/plasma/plasma.cpp

@@ -64,7 +64,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 +78,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 +269,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 +282,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;