diff options
Diffstat (limited to 'stmhal/usb.c')
| -rw-r--r-- | stmhal/usb.c | 240 |
1 files changed, 240 insertions, 0 deletions
diff --git a/stmhal/usb.c b/stmhal/usb.c new file mode 100644 index 000000000..8cfb0e7df --- /dev/null +++ b/stmhal/usb.c @@ -0,0 +1,240 @@ +#include <string.h> + +/* +#include "usb_core.h" +#include "usbd_cdc_core.h" +#include "usbd_pyb_core.h" +#include "usbd_usr.h" +*/ +#include "usbd_core.h" +#include "usbd_desc.h" +#include "usbd_cdc.h" +#include "usbd_cdc_interface.h" + +#include "misc.h" +#include "mpconfig.h" +#include "qstr.h" +#include "obj.h" +#include "pendsv.h" +#include "usb.h" + +#ifdef USE_DEVICE_MODE +//extern CDC_IF_Prop_TypeDef VCP_fops; +USBD_HandleTypeDef hUSBDDevice; +#endif + +static int dev_is_enabled = 0; +uint32_t APP_dev_is_connected = 0; /* used by usbd_cdc_vcp */ +static char rx_buf[64]; +static int rx_buf_in; +static int rx_buf_out; +static int interrupt_char = VCP_CHAR_NONE; +mp_obj_t mp_const_vcp_interrupt = MP_OBJ_NULL; + +void pyb_usb_dev_init(int usb_dev_type) { +#ifdef USE_DEVICE_MODE + if (!dev_is_enabled) { + // only init USB once in the device's power-lifetime + switch (usb_dev_type) { + case PYB_USB_DEV_VCP_MSC: + // XXX USBD_CDC_Init (called by one of these functions below) uses malloc, + // so the memory is invalid after a soft reset (which resets the GC). + USBD_Init(&hUSBDDevice, &VCP_Desc, 0); + USBD_RegisterClass(&hUSBDDevice, &USBD_CDC); + USBD_CDC_RegisterInterface(&hUSBDDevice, &USBD_CDC_fops); + USBD_Start(&hUSBDDevice); + //USBD_Init(&USB_OTG_Core, USB_OTG_FS_CORE_ID, &USR_desc, &USBD_PYB_cb, &USR_cb); + break; + + case PYB_USB_DEV_HID: + //USBD_Init(&USB_OTG_Core, USB_OTG_FS_CORE_ID, &USR_desc, &USBD_PYB_HID_cb, &USR_cb); + // TODO + break; + } + } + rx_buf_in = 0; + rx_buf_out = 0; + interrupt_char = VCP_CHAR_NONE; + dev_is_enabled = 1; + + // create an exception object for interrupting by VCP + mp_const_vcp_interrupt = mp_obj_new_exception_msg(&mp_type_OSError, "VCPInterrupt"); +#endif +} + +bool usb_vcp_is_enabled(void) { + return dev_is_enabled; +} + +bool usb_vcp_is_connected(void) { + return APP_dev_is_connected; +} + +void usb_vcp_set_interrupt_char(int c) { + if (dev_is_enabled) { + interrupt_char = c; + } +} + +void usb_vcp_receive(const char *buf, uint32_t len) { + if (dev_is_enabled) { + for (int i = 0; i < len; i++) { + + // catch special interrupt character + if (buf[i] == interrupt_char) { + // raise exception when interrupts are finished + mp_obj_exception_clear_traceback(mp_const_vcp_interrupt); + pendsv_nlr_jump(mp_const_vcp_interrupt); + interrupt_char = VCP_CHAR_NONE; + continue; + } + + rx_buf[rx_buf_in++] = buf[i]; + if (rx_buf_in >= sizeof(rx_buf)) { + rx_buf_in = 0; + } + if (rx_buf_in == rx_buf_out) { + rx_buf_out = rx_buf_in + 1; + if (rx_buf_out >= sizeof(rx_buf)) { + rx_buf_out = 0; + } + } + } + } +} + +int usb_vcp_rx_any(void) { + if (rx_buf_in >= rx_buf_out) { + return rx_buf_in - rx_buf_out; + } else { + return rx_buf_in + sizeof(rx_buf) - rx_buf_out; + } +} + +char usb_vcp_rx_get(void) { + while (rx_buf_out == rx_buf_in) { + } + char c = rx_buf[rx_buf_out]; + rx_buf_out += 1; + if (rx_buf_out >= sizeof(rx_buf)) { + rx_buf_out = 0; + } + return c; +} + +void usb_vcp_send_str(const char *str) { + usb_vcp_send_strn(str, strlen(str)); +} + +void usb_vcp_send_strn(const char *str, int len) { +#ifdef USE_DEVICE_MODE + if (dev_is_enabled) { + #if 0 + USBD_CDC_fops.pIf_DataTx((const uint8_t*)str, len); + #endif + } +#endif +} + +#include "usbd_conf.h" + +/* These are external variables imported from CDC core to be used for IN + transfer management. */ +#ifdef USE_DEVICE_MODE +extern uint8_t UserRxBuffer[];/* Received Data over USB are stored in this buffer */ +extern uint8_t UserTxBuffer[];/* Received Data over UART (CDC interface) are stored in this buffer */ +extern uint32_t BuffLength; +extern uint32_t UserTxBufPtrIn;/* Increment this pointer or roll it back to + start address when data are received over USART */ +extern uint32_t UserTxBufPtrOut; /* Increment this pointer or roll it back to + start address when data are sent over USB */ +#endif + +void usb_vcp_send_strn_cooked(const char *str, int len) { +#ifdef USE_DEVICE_MODE + #if 0 + for (const char *top = str + len; str < top; str++) { + if (*str == '\n') { + APP_Rx_Buffer[APP_Rx_ptr_in] = '\r'; + APP_Rx_ptr_in = (APP_Rx_ptr_in + 1) & (APP_RX_DATA_SIZE - 1); + } + APP_Rx_Buffer[APP_Rx_ptr_in] = *str; + APP_Rx_ptr_in = (APP_Rx_ptr_in + 1) & (APP_RX_DATA_SIZE - 1); + } + #endif +#endif +} + +void usb_hid_send_report(uint8_t *buf) { +#ifdef USE_DEVICE_MODE + #if 0 + USBD_HID_SendReport(&USB_OTG_Core, buf, 4); + #endif +#endif +} + +/******************************************************************************/ +// code for experimental USB OTG support + +#ifdef USE_HOST_MODE + +#include "led.h" +#include "usbh_core.h" +#include "usbh_usr.h" +#include "usbh_hid_core.h" +#include "usbh_hid_keybd.h" +#include "usbh_hid_mouse.h" + +__ALIGN_BEGIN USBH_HOST USB_Host __ALIGN_END ; + +static int host_is_enabled = 0; + +void pyb_usb_host_init(void) { + if (!host_is_enabled) { + // only init USBH once in the device's power-lifetime + /* Init Host Library */ + USBH_Init(&USB_OTG_Core, USB_OTG_FS_CORE_ID, &USB_Host, &HID_cb, &USR_Callbacks); + } + host_is_enabled = 1; +} + +void pyb_usb_host_process(void) { + USBH_Process(&USB_OTG_Core, &USB_Host); +} + +uint8_t usb_keyboard_key = 0; + +// TODO this is an ugly hack to get key presses +uint pyb_usb_host_get_keyboard(void) { + uint key = usb_keyboard_key; + usb_keyboard_key = 0; + return key; +} + +void USR_MOUSE_Init(void) { + led_state(4, 1); + USB_OTG_BSP_mDelay(100); + led_state(4, 0); +} + +void USR_MOUSE_ProcessData(HID_MOUSE_Data_TypeDef *data) { + led_state(4, 1); + USB_OTG_BSP_mDelay(50); + led_state(4, 0); +} + +void USR_KEYBRD_Init(void) { + led_state(4, 1); + USB_OTG_BSP_mDelay(100); + led_state(4, 0); +} + +void USR_KEYBRD_ProcessData(uint8_t pbuf) { + led_state(4, 1); + USB_OTG_BSP_mDelay(50); + led_state(4, 0); + //lcd_print_strn((char*)&pbuf, 1); + usb_keyboard_key = pbuf; +} + +#endif // USE_HOST_MODE |