4 * (c) 2020 Martin Mareš <mj@ucw.cz>
9 #include <libopencm3/cm3/cortex.h>
10 #include <libopencm3/cm3/nvic.h>
11 #include <libopencm3/cm3/systick.h>
12 #include <libopencm3/cm3/scb.h>
13 #include <libopencm3/stm32/rcc.h>
14 #include <libopencm3/stm32/desig.h>
15 #include <libopencm3/stm32/gpio.h>
16 #include <libopencm3/stm32/usart.h>
17 #include <libopencm3/stm32/i2c.h>
18 #include <libopencm3/usb/dfu.h>
19 #include <libopencm3/usb/usbd.h>
23 /*** Hardware init ***/
25 static void clock_init(void)
27 rcc_clock_setup_in_hse_8mhz_out_72mhz();
29 rcc_periph_clock_enable(RCC_GPIOA);
30 rcc_periph_clock_enable(RCC_GPIOB);
31 rcc_periph_clock_enable(RCC_GPIOC);
32 rcc_periph_clock_enable(RCC_I2C1);
33 rcc_periph_clock_enable(RCC_USART1);
34 rcc_periph_clock_enable(RCC_USB);
36 rcc_periph_reset_pulse(RST_GPIOA);
37 rcc_periph_reset_pulse(RST_GPIOB);
38 rcc_periph_reset_pulse(RST_GPIOC);
39 rcc_periph_reset_pulse(RST_I2C1);
40 rcc_periph_reset_pulse(RST_USART1);
41 rcc_periph_reset_pulse(RST_USB);
44 static void gpio_init(void)
46 // PA9 = TXD1 for debugging console
47 // PA10 = RXD1 for debugging console
48 gpio_set_mode(GPIOA, GPIO_MODE_OUTPUT_50_MHZ, GPIO_CNF_OUTPUT_ALTFN_PUSHPULL, GPIO9);
49 gpio_set_mode(GPIOA, GPIO_MODE_INPUT, GPIO_CNF_INPUT_FLOAT, GPIO10);
51 // PC13 = BluePill LED
52 gpio_set_mode(GPIOC, GPIO_MODE_OUTPUT_50_MHZ, GPIO_CNF_OUTPUT_PUSHPULL, GPIO13);
53 gpio_clear(GPIOC, GPIO13);
55 // PB7 = SDA for display controller
56 // PB6 = SCL for display controller
57 gpio_set_mode(GPIOB, GPIO_MODE_OUTPUT_50_MHZ, GPIO_CNF_OUTPUT_ALTFN_OPENDRAIN, GPIO6 | GPIO7);
59 // PB8 = SFH5110 output (5V tolerant)
60 gpio_set_mode(GPIOC, GPIO_MODE_INPUT, GPIO_CNF_INPUT_FLOAT, GPIO8);
63 static void usart_init(void)
65 usart_set_baudrate(USART1, 115200);
66 usart_set_databits(USART1, 8);
67 usart_set_stopbits(USART1, USART_STOPBITS_1);
68 usart_set_mode(USART1, USART_MODE_TX);
69 usart_set_parity(USART1, USART_PARITY_NONE);
70 usart_set_flow_control(USART1, USART_FLOWCONTROL_NONE);
75 /*** System ticks ***/
77 static volatile u32 ms_ticks;
79 void sys_tick_handler(void)
84 static void tick_init(void)
86 systick_set_frequency(1000, CPU_CLOCK_MHZ * 1000000);
87 systick_counter_enable();
88 systick_interrupt_enable();
91 static void delay_ms(uint ms)
93 u32 start_ticks = ms_ticks;
94 while (ms_ticks - start_ticks < ms)
120 static byte ctrl = 0x56;
122 static void display_update(void)
124 debug_puts("Display update\n");
128 cmds[2] = ((disp[1] & 0x88) >> 3) | ((disp[1] & 0x44) >> 1) | ((disp[1] & 0x22) << 1) | ((disp[1] & 0x11) << 3);
130 i2c_transfer7(I2C1, 0x76/2, (byte *) cmds, sizeof(cmds), NULL, 0);
132 cmds[2] = ((disp[3] & 0x88) >> 3) | ((disp[3] & 0x44) >> 1) | ((disp[3] & 0x22) << 1) | ((disp[3] & 0x11) << 3);
134 i2c_transfer7(I2C1, 0x70/2, (byte *) cmds, sizeof(cmds), NULL, 0);
136 debug_puts("Update done\n");
139 static void display_init(void)
141 debug_puts("I2C init\n");
142 i2c_peripheral_disable(I2C1);
143 i2c_set_speed(I2C1, i2c_speed_sm_100k, rcc_apb1_frequency / 1000000);
144 i2c_peripheral_enable(I2C1);
153 static void display_test(void)
161 byte cmds[] = { 0x00, mode ? 0x77 : 0x77 };
162 i2c_transfer7(I2C1, 0x70/2, (byte *) cmds, sizeof(cmds), NULL, 0);
166 byte disp[] = { 0xff, 0xff, mode ? 0xff : 0x00, mode ? 0xff : 0xff };
167 byte cmds[] = { 0x00, 0x77, 0, 0, 0, 0 };
168 cmds[2] = (disp[0] & 0xf0) | (disp[2] >> 4);
169 cmds[3] = (disp[1] & 0xf0) | (disp[3] >> 4);
170 cmds[4] = (disp[2] & 0x0f) | (disp[0] << 4);
171 cmds[5] = (disp[3] & 0x0f) | (disp[1] << 4);
172 i2c_transfer7(I2C1, 0x70/2, (byte *) cmds, sizeof(cmds), NULL, 0);
178 static const byte lcd_font[] = {
199 static usbd_device *usbd_dev;
202 STR_MANUFACTURER = 1,
207 static char usb_serial_number[13];
209 static const char *usb_strings[] = {
210 "United Computer Wizards",
215 static const struct usb_device_descriptor device = {
216 .bLength = USB_DT_DEVICE_SIZE,
217 .bDescriptorType = USB_DT_DEVICE,
219 .bDeviceClass = 0xFF,
220 .bDeviceSubClass = 0,
221 .bDeviceProtocol = 0,
222 .bMaxPacketSize0 = 64,
226 .iManufacturer = STR_MANUFACTURER,
227 .iProduct = STR_PRODUCT,
228 .iSerialNumber = STR_SERIAL,
229 .bNumConfigurations = 1,
232 static const struct usb_endpoint_descriptor endpoints[] = {{
233 // Bulk end-point for sending values to the display
234 .bLength = USB_DT_ENDPOINT_SIZE,
235 .bDescriptorType = USB_DT_ENDPOINT,
236 .bEndpointAddress = 0x01,
237 .bmAttributes = USB_ENDPOINT_ATTR_BULK,
238 .wMaxPacketSize = 64,
242 static const struct usb_interface_descriptor iface = {
243 .bLength = USB_DT_INTERFACE_SIZE,
244 .bDescriptorType = USB_DT_INTERFACE,
245 .bInterfaceNumber = 0,
246 .bAlternateSetting = 0,
248 .bInterfaceClass = 0xFF,
249 .bInterfaceSubClass = 0,
250 .bInterfaceProtocol = 0,
252 .endpoint = endpoints,
255 static const struct usb_dfu_descriptor dfu_function = {
256 .bLength = sizeof(struct usb_dfu_descriptor),
257 .bDescriptorType = DFU_FUNCTIONAL,
258 .bmAttributes = USB_DFU_CAN_DOWNLOAD | USB_DFU_WILL_DETACH,
259 .wDetachTimeout = 255,
260 .wTransferSize = 1024,
261 .bcdDFUVersion = 0x0100,
264 static const struct usb_interface_descriptor dfu_iface = {
265 .bLength = USB_DT_INTERFACE_SIZE,
266 .bDescriptorType = USB_DT_INTERFACE,
267 .bInterfaceNumber = 1,
268 .bAlternateSetting = 0,
270 .bInterfaceClass = 0xFE,
271 .bInterfaceSubClass = 1,
272 .bInterfaceProtocol = 1,
275 .extra = &dfu_function,
276 .extralen = sizeof(dfu_function),
279 static const struct usb_interface ifaces[] = {{
281 .altsetting = &iface,
284 .altsetting = &dfu_iface,
287 static const struct usb_config_descriptor config = {
288 .bLength = USB_DT_CONFIGURATION_SIZE,
289 .bDescriptorType = USB_DT_CONFIGURATION,
292 .bConfigurationValue = 1,
294 .bmAttributes = 0x80,
295 .bMaxPower = 50, // multiplied by 2 mA
299 static byte usb_configured;
300 static uint8_t usbd_control_buffer[64];
302 static void dfu_detach_complete(usbd_device *dev UNUSED, struct usb_setup_data *req UNUSED)
304 // Reset to bootloader, which implements the rest of DFU
305 debug_printf("Switching to DFU\n");
310 static enum usbd_request_return_codes dfu_control_cb(usbd_device *dev UNUSED,
311 struct usb_setup_data *req,
312 uint8_t **buf UNUSED,
313 uint16_t *len UNUSED,
314 void (**complete)(usbd_device *dev, struct usb_setup_data *req))
316 if (req->bmRequestType != 0x21 || req->bRequest != DFU_DETACH)
317 return USBD_REQ_NOTSUPP;
319 *complete = dfu_detach_complete;
320 return USBD_REQ_HANDLED;
323 static void ep01_cb(usbd_device *dev, uint8_t ep UNUSED)
325 // We received a frame from the USB host
327 uint len = usbd_ep_read_packet(dev, 0x01, buf, 8);
328 debug_printf("USB: Host sent %u bytes\n", len);
330 for (uint i=0; i<4; i++) {
332 disp[i] = lcd_font[buf[i]];
342 static void set_config_cb(usbd_device *dev, uint16_t wValue UNUSED)
344 usbd_register_control_callback(
346 USB_REQ_TYPE_CLASS | USB_REQ_TYPE_INTERFACE,
347 USB_REQ_TYPE_TYPE | USB_REQ_TYPE_RECIPIENT,
349 usbd_ep_setup(dev, 0x01, USB_ENDPOINT_ATTR_BULK, 64, ep01_cb);
353 static void reset_cb(void)
355 debug_printf("USB: Reset\n");
359 static volatile bool usb_event_pending;
361 void usb_lp_can_rx0_isr(void)
364 * We handle USB in the main loop to avoid race conditions between
365 * USB interrupts and other code. However, we need an interrupt to
366 * up the main loop from sleep.
368 * We set up only the low-priority ISR, because high-priority ISR handles
369 * only double-buffered bulk transfers and isochronous transfers.
371 nvic_disable_irq(NVIC_USB_LP_CAN_RX0_IRQ);
372 usb_event_pending = 1;
375 static void usb_init(void)
377 // Simulate USB disconnect
378 gpio_set_mode(GPIOA, GPIO_MODE_OUTPUT_50_MHZ, GPIO_CNF_OUTPUT_OPENDRAIN, GPIO11 | GPIO12);
379 gpio_clear(GPIOA, GPIO11 | GPIO12);
382 usbd_dev = usbd_init(
383 &st_usbfs_v1_usb_driver,
387 ARRAY_SIZE(usb_strings),
389 sizeof(usbd_control_buffer)
391 usbd_register_reset_callback(usbd_dev, reset_cb);
392 usbd_register_set_config_callback(usbd_dev, set_config_cb);
393 usb_event_pending = 1;
396 /*** Testing of IR receiver ***/
400 static u16 get_bit(void)
403 return !gpio_get(GPIOB, GPIO8);
406 x += !gpio_get(GPIOB, GPIO8);
407 x += !gpio_get(GPIOB, GPIO8);
408 x += !gpio_get(GPIOB, GPIO8);
409 x += !gpio_get(GPIOB, GPIO8);
414 #define MAX_SAMPLES 1024
415 u32 samples[MAX_SAMPLES];
417 static void ir_test_loop(void)
419 debug_puts("\n\n### Infrared Remote Control receiver ###\n\n");
421 systick_set_reload(0xffffff);
422 systick_counter_enable();
423 systick_interrupt_disable();
427 gpio_set(GPIOC, GPIO13);
430 debug_puts("Waiting for silence\n");
434 debug_puts("Ready...");
441 start = systick_get_value();
445 gpio_clear(GPIOC, GPIO13);
452 now = systick_get_value();
453 len = (start - now) & 0xffffff;
455 samples[nsamp++] = len;
459 } while (curr == last);
460 samples[nsamp++] = len;
461 if (nsamp >= MAX_SAMPLES)
468 for (uint i=0; i<nsamp; i++) {
469 debug_putc(i ? ' ' : '\n');
470 debug_printf("%u", (unsigned int)((samples[i] + CPU_CLOCK_MHZ - 1) / CPU_CLOCK_MHZ)); // in μs
491 desig_get_unique_id_as_dfu(usb_serial_number);
493 debug_printf("Hello, world!\n");
501 if (ms_ticks - last_blink >= 500) {
503 last_blink = ms_ticks;
507 if (usb_event_pending) {
509 usb_event_pending = 0;
510 nvic_clear_pending_irq(NVIC_USB_LP_CAN_RX0_IRQ);
511 nvic_enable_irq(NVIC_USB_LP_CAN_RX0_IRQ);
514 wait_for_interrupt();