test-display2/main.c

   1 /*
   2  *      Workshop Clock
   3  *
   4  *      (c) 2020-2023 Martin Mareš <mj@ucw.cz>
   5  */
   6
   7 #include "util.h"
   8
   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/stm32/timer.h>
  19 #include <libopencm3/usb/dfu.h>
  20 #include <libopencm3/usb/usbd.h>
  21
  22 #include <string.h>
  23
  24 static void ep82_send(u32 key_code);
  25
  26 /*** Hardware init ***/
  27
  28 static void clock_init(void)
  29 {
  30         rcc_clock_setup_pll(&rcc_hse_configs[RCC_CLOCK_HSE8_72MHZ]);
  31
  32         rcc_periph_clock_enable(RCC_GPIOA);
  33         rcc_periph_clock_enable(RCC_GPIOB);
  34         rcc_periph_clock_enable(RCC_GPIOC);
  35         rcc_periph_clock_enable(RCC_I2C1);
  36         rcc_periph_clock_enable(RCC_USART1);
  37         rcc_periph_clock_enable(RCC_USB);
  38         rcc_periph_clock_enable(RCC_TIM1);
  39
  40         rcc_periph_reset_pulse(RST_GPIOA);
  41         rcc_periph_reset_pulse(RST_GPIOB);
  42         rcc_periph_reset_pulse(RST_GPIOC);
  43         rcc_periph_reset_pulse(RST_I2C1);
  44         rcc_periph_reset_pulse(RST_USART1);
  45         rcc_periph_reset_pulse(RST_USB);
  46         rcc_periph_reset_pulse(RST_TIM1);
  47 }
  48
  49 static void gpio_init(void)
  50 {
  51         // PA9 = TXD1 for debugging console
  52         // PA10 = RXD1 for debugging console
  53         gpio_set_mode(GPIOA, GPIO_MODE_OUTPUT_50_MHZ, GPIO_CNF_OUTPUT_ALTFN_PUSHPULL, GPIO9);
  54         gpio_set_mode(GPIOA, GPIO_MODE_INPUT, GPIO_CNF_INPUT_FLOAT, GPIO10);
  55
  56         // PC13 = BluePill LED
  57         gpio_set_mode(GPIOC, GPIO_MODE_OUTPUT_50_MHZ, GPIO_CNF_OUTPUT_PUSHPULL, GPIO13);
  58         gpio_clear(GPIOC, GPIO13);
  59
  60         // PB7 = SDA for display controller
  61         // PB6 = SCL for display controller
  62         gpio_set_mode(GPIOB, GPIO_MODE_OUTPUT_50_MHZ, GPIO_CNF_OUTPUT_ALTFN_OPENDRAIN, GPIO6 | GPIO7);
  63
  64         // PA8 = SFH5110 output (5V tolerant) connected to TIM1_CH1
  65         gpio_set_mode(GPIOC, GPIO_MODE_INPUT, GPIO_CNF_INPUT_FLOAT, GPIO8);
  66 }
  67
  68 static void usart_init(void)
  69 {
  70         usart_set_baudrate(USART1, 115200);
  71         usart_set_databits(USART1, 8);
  72         usart_set_stopbits(USART1, USART_STOPBITS_1);
  73         usart_set_mode(USART1, USART_MODE_TX);
  74         usart_set_parity(USART1, USART_PARITY_NONE);
  75         usart_set_flow_control(USART1, USART_FLOWCONTROL_NONE);
  76
  77         usart_enable(USART1);
  78 }
  79
  80 /*** System ticks ***/
  81
  82 static volatile u32 ms_ticks;
  83
  84 void sys_tick_handler(void)
  85 {
  86         ms_ticks++;
  87 }
  88
  89 static void tick_init(void)
  90 {
  91         systick_set_frequency(1000, CPU_CLOCK_MHZ * 1000000);
  92         systick_counter_enable();
  93         systick_interrupt_enable();
  94 }
  95
  96 static void delay_ms(uint ms)
  97 {
  98         u32 start_ticks = ms_ticks;
  99         while (ms_ticks - start_ticks < ms)
 100                 ;
 101 }
 102
 103 /*** Display ***/
 104
 105 /*
 106  *      Display digits:
 107  *
 108  *                 ---- 40 ----
 109  *               |              |
 110  *               |              |
 111  *              20              80
 112  *               |              |
 113  *               |              |
 114  *                 ---- 10 ----
 115  *               |              |
 116  *               |              |
 117  *              08              02
 118  *               |              |
 119  *               |              |
 120  *                 ---- 04 ----
 121  *                                      (01)
 122  */
 123
 124 static byte disp[4];
 125 static byte ctrl = 0x56;
 126
 127 static const byte disp_font[] = {
 128         [0] = 0xee,
 129         [1] = 0x82,
 130         [2] = 0xdc,
 131         [3] = 0xd6,
 132         [4] = 0xb2,
 133         [5] = 0x76,
 134         [6] = 0x7e,
 135         [7] = 0xc2,
 136         [8] = 0xfe,
 137         [9] = 0xf6,
 138         [10] = 0xea,
 139         [11] = 0x3e,
 140         [12] = 0x6c,
 141         [13] = 0x9e,
 142         [14] = 0x7c,
 143         [15] = 0x78,
 144 };
 145
 146 static void display_update(void)
 147 {
 148         // debug_puts("Display update\n");
 149
 150         byte cmds[4];
 151         cmds[0] = 0;
 152         cmds[1] = ctrl;
 153         cmds[2] = ((disp[1] & 0x88) >> 3) | ((disp[1] & 0x44) >> 1) | ((disp[1] & 0x22) << 1) | ((disp[1] & 0x11) << 3);
 154         cmds[3] = disp[0];
 155         i2c_transfer7(I2C1, 0x76/2, (byte *) cmds, sizeof(cmds), NULL, 0);
 156
 157         cmds[2] = ((disp[3] & 0x88) >> 3) | ((disp[3] & 0x44) >> 1) | ((disp[3] & 0x22) << 1) | ((disp[3] & 0x11) << 3);
 158         cmds[3] = disp[2];
 159         i2c_transfer7(I2C1, 0x70/2, (byte *) cmds, sizeof(cmds), NULL, 0);
 160
 161         // debug_puts("Update done\n");
 162 }
 163
 164 static void display_init(void)
 165 {
 166         debug_puts("I2C init\n");
 167         i2c_peripheral_disable(I2C1);
 168         i2c_set_speed(I2C1, i2c_speed_sm_100k, rcc_apb1_frequency / 1000000);
 169         i2c_peripheral_enable(I2C1);
 170
 171         disp[0] = 0x10;
 172         disp[1] = 0x10;
 173         disp[2] = 0x10;
 174         disp[3] = 0x10;
 175         display_update();
 176 }
 177
 178 /*** Infrared Remote Control ***/
 179
 180 static void ir_init(void)
 181 {
 182         debug_puts("IR init\n");
 183
 184         // TIM1 will measure pulses and spaces between them with 1μs resolution
 185         timer_set_prescaler(TIM1, 71);          // 72 MHz / 72 = 1 MHz
 186         timer_set_mode(TIM1, TIM_CR1_CKD_CK_INT, TIM_CR1_CMS_EDGE, TIM_CR1_DIR_UP);
 187         timer_set_period(TIM1, 65535);
 188         timer_update_on_overflow(TIM1);
 189
 190         // IC1 will trigger on TI1 (TIM1_CH1) falling edge
 191         timer_ic_set_input(TIM1, TIM_IC1, TIM_IC_IN_TI1);
 192         // timer_ic_set_filter(TIM1, TIM_IC1, TIM_IC_OFF);
 193         timer_set_oc_polarity_low(TIM1, TIM_OC1);       // OC functions affect IC, too
 194
 195         // IC2 will trigger on TI1 (TIM1_CH1) rising edge
 196         timer_ic_set_input(TIM1, TIM_IC2, TIM_IC_IN_TI1);
 197         timer_set_oc_polarity_high(TIM1, TIM_OC2);
 198
 199         // OC3 will trigger on a break longer than 50 ms
 200         timer_set_oc_mode(TIM1, TIM_OC3, TIM_OCM_ACTIVE);
 201         timer_set_oc_value(TIM1, TIM_OC3, 30000);
 202
 203         // Program slave controller to reset the timer on IC1
 204         timer_slave_set_trigger(TIM1, TIM_SMCR_TS_TI1FP1);
 205         timer_slave_set_mode(TIM1, TIM_SMCR_SMS_RM);
 206
 207         // Request interrupts
 208         timer_enable_irq(TIM1, TIM_DIER_CC1IE | TIM_DIER_CC2IE | TIM_DIER_CC3IE);
 209         nvic_enable_irq(NVIC_TIM1_CC_IRQ);
 210
 211         // Enable ICs and OCs
 212         timer_enable_oc_output(TIM1, TIM_OC1);
 213         timer_enable_oc_output(TIM1, TIM_OC2);
 214         timer_enable_oc_output(TIM1, TIM_OC3);
 215
 216         timer_enable_counter(TIM1);
 217 }
 218
 219 // Circular queue of pulse durations
 220 #define IR_MAX_PULSES 32
 221 static u32 ir_pulses[IR_MAX_PULSES];    // Top 16 bits = mark, bottom 16 bits = space
 222 static u16 ir_last_pulse;
 223 static uint ir_pulses_rx, ir_pulses_tx;
 224
 225 #define IR_INF 0xffff
 226
 227 #define IR_MARK(x) (uint)((x) >> 16)
 228 #define IR_SPACE(x) (uint)((x) & 0xffff)
 229
 230 static inline bool between(uint x, uint min, uint max)
 231 {
 232         return x >= min && x <= max;
 233 }
 234
 235 static void ir_record_pulse(uint mark, uint space)
 236 {
 237         uint i = ir_pulses_tx;
 238         ir_pulses_tx = (i + 1) % IR_MAX_PULSES;
 239         if (ir_pulses_tx != ir_pulses_rx) {
 240                 ir_pulses[i] = (mark << 16) | space;
 241         } else {
 242                 // Overflow detected
 243                 ir_pulses[i] = (IR_INF << 16) | IR_INF;
 244         }
 245 }
 246
 247 void tim1_cc_isr(void)
 248 {
 249         if (TIM_SR(TIM1) & TIM_SR_CC1IF) {
 250                 TIM_SR(TIM1) &= ~TIM_SR_CC1IF;
 251                 u16 now = TIM_CCR1(TIM1);
 252                 if (ir_last_pulse) {
 253                         ir_record_pulse(ir_last_pulse, now - ir_last_pulse);
 254                         ir_last_pulse = 0;
 255                 }
 256         }
 257         if (TIM_SR(TIM1) & TIM_SR_CC2IF) {
 258                 TIM_SR(TIM1) &= ~TIM_SR_CC2IF;
 259                 ir_last_pulse = TIM_CCR2(TIM1);
 260         }
 261         if (TIM_SR(TIM1) & TIM_SR_CC3IF) {
 262                 TIM_SR(TIM1) &= ~TIM_SR_CC3IF;
 263                 if (ir_last_pulse) {
 264                         ir_record_pulse(ir_last_pulse, IR_INF);
 265                         ir_last_pulse = 0;
 266                 }
 267         }
 268 }
 269
 270 static u32 ir_get_pulse(void)
 271 {
 272         u32 out = 0;
 273
 274         cm_disable_interrupts();
 275         if (ir_pulses_rx != ir_pulses_tx) {
 276                 out = ir_pulses[ir_pulses_rx];
 277                 ir_pulses_rx = (ir_pulses_rx + 1) % IR_MAX_PULSES;
 278         }
 279         cm_enable_interrupts();
 280         return out;
 281 }
 282
 283 // Decoder for Onkyo RC-748S
 284
 285 static void ir_decode(void)
 286 {
 287         u32 pulse = ir_get_pulse();
 288         if (!pulse)
 289                 return;
 290
 291         uint mark = IR_MARK(pulse);
 292         uint space = IR_SPACE(pulse);
 293
 294 #ifdef IR_TEST
 295         debug_printf("IR: %d %d\n", mark, space);
 296         return;
 297 #endif
 298
 299         static u16 ir_bits;
 300         static u32 ir_code;
 301 #define IR_ERR 0xff
 302
 303         // debug_printf("IR(%d): %d %d\n", ir_bits, mark, space);
 304
 305         if (space == IR_INF) {
 306                 ir_bits = 0;
 307         } else if (ir_bits == IR_ERR) {
 308                 // Error state
 309         } else if (ir_bits == 0) {
 310                 // Start?
 311                 if (between(mark, 8900, 9200)) {
 312                         if (between(space, 4200, 4600)) {
 313                                 ir_bits = 1;
 314                                 ir_code = 0;
 315                         } else if (between(space, 2000, 2300)) {
 316                                 debug_printf("IR: => REP\n");
 317                                 ir_bits = IR_ERR;
 318                         }
 319                 }
 320         } else {
 321                 if (between(mark, 500, 700)) {
 322                         ir_bits++;
 323                         if (between(space, 400, 700)) {
 324                                 // 0
 325                         } else if (between(space, 1500, 1800)) {
 326                                 // 1
 327                                 ir_code |= 1U << (33 - ir_bits);
 328                         } else {
 329                                 ir_bits = IR_ERR;
 330                         }
 331                         if (ir_bits == 33) {
 332                                 debug_printf("IR: => %08x\n", (uint)ir_code);
 333                                 disp[3] ^= 0x01;
 334                                 display_update();
 335                                 ir_bits = IR_ERR;
 336                                 ep82_send(ir_code);
 337                         }
 338                 } else {
 339                         ir_bits = IR_ERR;
 340                 }
 341         }
 342 }
 343
 344 /*** USB ***/
 345
 346 static usbd_device *usbd_dev;
 347
 348 enum usb_string {
 349         STR_MANUFACTURER = 1,
 350         STR_PRODUCT,
 351         STR_SERIAL,
 352 };
 353
 354 static char usb_serial_number[13];
 355
 356 static const char *usb_strings[] = {
 357         "United Computer Wizards",
 358         "Workshop Clock",
 359         usb_serial_number,
 360 };
 361
 362 static const struct usb_device_descriptor device = {
 363         .bLength = USB_DT_DEVICE_SIZE,
 364         .bDescriptorType = USB_DT_DEVICE,
 365         .bcdUSB = 0x0200,
 366         .bDeviceClass = 0xFF,
 367         .bDeviceSubClass = 0,
 368         .bDeviceProtocol = 0,
 369         .bMaxPacketSize0 = 64,
 370         .idVendor = 0x4242,
 371         .idProduct = 0x0007,
 372         .bcdDevice = 0x0000,
 373         .iManufacturer = STR_MANUFACTURER,
 374         .iProduct = STR_PRODUCT,
 375         .iSerialNumber = STR_SERIAL,
 376         .bNumConfigurations = 1,
 377 };
 378
 379 static const struct usb_endpoint_descriptor endpoints[] = {{
 380         // Bulk end-point for sending values to the display
 381         .bLength = USB_DT_ENDPOINT_SIZE,
 382         .bDescriptorType = USB_DT_ENDPOINT,
 383         .bEndpointAddress = 0x01,
 384         .bmAttributes = USB_ENDPOINT_ATTR_BULK,
 385         .wMaxPacketSize = 64,
 386         .bInterval = 1,
 387 }, {
 388         // Bulk end-point for receiving remote control keys
 389         .bLength = USB_DT_ENDPOINT_SIZE,
 390         .bDescriptorType = USB_DT_ENDPOINT,
 391         .bEndpointAddress = 0x82,
 392         .bmAttributes = USB_ENDPOINT_ATTR_BULK,
 393         .wMaxPacketSize = 4,
 394         .bInterval = 1,
 395 }};
 396
 397 static const struct usb_interface_descriptor iface = {
 398         .bLength = USB_DT_INTERFACE_SIZE,
 399         .bDescriptorType = USB_DT_INTERFACE,
 400         .bInterfaceNumber = 0,
 401         .bAlternateSetting = 0,
 402         .bNumEndpoints = 2,
 403         .bInterfaceClass = 0xFF,
 404         .bInterfaceSubClass = 0,
 405         .bInterfaceProtocol = 0,
 406         .iInterface = 0,
 407         .endpoint = endpoints,
 408 };
 409
 410 static const struct usb_dfu_descriptor dfu_function = {
 411         .bLength = sizeof(struct usb_dfu_descriptor),
 412         .bDescriptorType = DFU_FUNCTIONAL,
 413         .bmAttributes = USB_DFU_CAN_DOWNLOAD | USB_DFU_WILL_DETACH,
 414         .wDetachTimeout = 255,
 415         .wTransferSize = 1024,
 416         .bcdDFUVersion = 0x0100,
 417 };
 418
 419 static const struct usb_interface_descriptor dfu_iface = {
 420         .bLength = USB_DT_INTERFACE_SIZE,
 421         .bDescriptorType = USB_DT_INTERFACE,
 422         .bInterfaceNumber = 1,
 423         .bAlternateSetting = 0,
 424         .bNumEndpoints = 0,
 425         .bInterfaceClass = 0xFE,
 426         .bInterfaceSubClass = 1,
 427         .bInterfaceProtocol = 1,
 428         .iInterface = 0,
 429
 430         .extra = &dfu_function,
 431         .extralen = sizeof(dfu_function),
 432 };
 433
 434 static const struct usb_interface ifaces[] = {{
 435         .num_altsetting = 1,
 436         .altsetting = &iface,
 437 }, {
 438         .num_altsetting = 1,
 439         .altsetting = &dfu_iface,
 440 }};
 441
 442 static const struct usb_config_descriptor config = {
 443         .bLength = USB_DT_CONFIGURATION_SIZE,
 444         .bDescriptorType = USB_DT_CONFIGURATION,
 445         .wTotalLength = 0,
 446         .bNumInterfaces = 2,
 447         .bConfigurationValue = 1,
 448         .iConfiguration = 0,
 449         .bmAttributes = 0x80,
 450         .bMaxPower = 50,        // multiplied by 2 mA
 451         .interface = ifaces,
 452 };
 453
 454 static bool usb_configured;
 455 static uint8_t usbd_control_buffer[64];
 456
 457 static bool usb_tx_in_flight;
 458 static byte ep82_tx_buffer[4];
 459
 460 static byte disp_alive;
 461
 462 static void dfu_detach_complete(usbd_device *dev UNUSED, struct usb_setup_data *req UNUSED)
 463 {
 464         // Reset to bootloader, which implements the rest of DFU
 465         debug_printf("Switching to DFU\n");
 466         debug_flush();
 467         scb_reset_core();
 468 }
 469
 470 static enum usbd_request_return_codes dfu_control_cb(usbd_device *dev UNUSED,
 471         struct usb_setup_data *req,
 472         uint8_t **buf UNUSED,
 473         uint16_t *len UNUSED,
 474         void (**complete)(usbd_device *dev, struct usb_setup_data *req))
 475 {
 476         if (req->bmRequestType != 0x21 || req->bRequest != DFU_DETACH)
 477                 return USBD_REQ_NOTSUPP;
 478
 479         *complete = dfu_detach_complete;
 480         return USBD_REQ_HANDLED;
 481 }
 482
 483 static void ep01_cb(usbd_device *dev, uint8_t ep UNUSED)
 484 {
 485         // We received a frame from the USB host
 486         byte buf[8];
 487         uint len = usbd_ep_read_packet(dev, 0x01, buf, 8);
 488         debug_printf("USB: Host sent %u bytes\n", len);
 489         if (len >= 5) {
 490                 for (uint i=0; i<4; i++) {
 491                         disp[i] &= 0x01;
 492                         if (buf[i] < 16)
 493                                 disp[i] |= disp_font[buf[i]];
 494                 }
 495                 disp[1] &= 0xfe;
 496                 if (buf[4])
 497                         disp[1] |= 0x01;
 498                 display_update();
 499                 disp_alive = 10;
 500         }
 501 }
 502
 503 static void ep82_send(u32 key_code)
 504 {
 505         if (usb_tx_in_flight) {
 506                 debug_printf("USB: Send overrun!\n");
 507                 return;
 508         }
 509
 510         debug_printf("USB: Sending key to host\n");
 511         put_u32_be(ep82_tx_buffer, key_code);
 512         usbd_ep_write_packet(usbd_dev, 0x82, ep82_tx_buffer, 4);
 513         usb_tx_in_flight = true;
 514 }
 515
 516 static void ep82_cb(usbd_device *dev UNUSED, uint8_t ep UNUSED)
 517 {
 518         // We completed sending a frame to the USB host
 519         usb_tx_in_flight = false;
 520         debug_printf("USB: Key sending complete\n");
 521 }
 522
 523 static void set_config_cb(usbd_device *dev, uint16_t wValue UNUSED)
 524 {
 525         usbd_register_control_callback(
 526                 dev,
 527                 USB_REQ_TYPE_CLASS | USB_REQ_TYPE_INTERFACE,
 528                 USB_REQ_TYPE_TYPE | USB_REQ_TYPE_RECIPIENT,
 529                 dfu_control_cb);
 530         usbd_ep_setup(dev, 0x01, USB_ENDPOINT_ATTR_BULK, 64, ep01_cb);
 531         usbd_ep_setup(dev, 0x82, USB_ENDPOINT_ATTR_BULK, 4, ep82_cb);
 532         usb_configured = true;
 533 }
 534
 535 static void reset_cb(void)
 536 {
 537         debug_printf("USB: Reset\n");
 538         usb_configured = false;
 539 }
 540
 541 static volatile bool usb_event_pending;
 542
 543 void usb_lp_can_rx0_isr(void)
 544 {
 545         /*
 546          *  We handle USB in the main loop to avoid race conditions between
 547          *  USB interrupts and other code. However, we need an interrupt to
 548          *  up the main loop from sleep.
 549          *
 550          *  We set up only the low-priority ISR, because high-priority ISR handles
 551          *  only double-buffered bulk transfers and isochronous transfers.
 552          */
 553         nvic_disable_irq(NVIC_USB_LP_CAN_RX0_IRQ);
 554         usb_event_pending = 1;
 555 }
 556
 557 static void usb_init(void)
 558 {
 559         // Simulate USB disconnect
 560         gpio_set_mode(GPIOA, GPIO_MODE_OUTPUT_50_MHZ, GPIO_CNF_OUTPUT_OPENDRAIN, GPIO11 | GPIO12);
 561         gpio_clear(GPIOA, GPIO11 | GPIO12);
 562         delay_ms(100);
 563
 564         usbd_dev = usbd_init(
 565                 &st_usbfs_v1_usb_driver,
 566                 &device,
 567                 &config,
 568                 usb_strings,
 569                 ARRAY_SIZE(usb_strings),
 570                 usbd_control_buffer,
 571                 sizeof(usbd_control_buffer)
 572         );
 573         usbd_register_reset_callback(usbd_dev, reset_cb);
 574         usbd_register_set_config_callback(usbd_dev, set_config_cb);
 575         usb_event_pending = 1;
 576 }
 577
 578 /*** Main ***/
 579
 580 int main(void)
 581 {
 582         clock_init();
 583         gpio_init();
 584         usart_init();
 585
 586         tick_init();
 587         desig_get_unique_id_as_dfu(usb_serial_number);
 588
 589         debug_printf("Hello, world!\n");
 590
 591         usb_init();
 592         display_init();
 593         ir_init();
 594
 595         u32 last_blink = 0;
 596
 597         for (;;) {
 598                 if (ms_ticks - last_blink >= 500) {
 599                         debug_led_toggle();
 600                         last_blink = ms_ticks;
 601                         if (disp_alive) {
 602                                 if (!--disp_alive) {
 603                                         disp[0] = (disp[0] & 0x01) | 0x10;
 604                                         disp[1] = (disp[1] & 0x01) | 0x10;
 605                                         disp[2] = (disp[2] & 0x01) | 0x10;
 606                                         disp[3] = (disp[3] & 0x01) | 0x10;
 607                                 }
 608                         }
 609                         disp[0] ^= 0x01;
 610                         display_update();
 611                 }
 612
 613                 ir_decode();
 614
 615                 if (usb_event_pending) {
 616                         usbd_poll(usbd_dev);
 617                         usb_event_pending = 0;
 618                         nvic_clear_pending_irq(NVIC_USB_LP_CAN_RX0_IRQ);
 619                         nvic_enable_irq(NVIC_USB_LP_CAN_RX0_IRQ);
 620                 }
 621
 622                 wait_for_interrupt();
 623         }
 624
 625         return 0;
 626 }