src-avr: Add ADC routines to telescience
[fw/altos] / src-avr / ao_usb_avr.c
index fbae08bac8623e66d3c043bd6343e7ed0a51be3c..74bdea23ef8d5eb630dde4c59d390eaf276672ec 100644 (file)
 #include "ao.h"
 #include "ao_usb.h"
 
-//#define printf(format, args...)
+#define USB_DEBUG 0
+
+#if USB_DEBUG
+#define debug(format, args...) printf(format, ## args)
+#else
+#define debug(format, args...)
+#endif
 
 struct ao_task __xdata ao_usb_task;
 
@@ -42,12 +48,12 @@ static __xdata uint8_t *__xdata ao_usb_ep0_out_data;
 static __xdata uint8_t ao_usb_in_flushed;
 static __xdata uint8_t ao_usb_running;
 static __xdata uint8_t ao_usb_configuration;
+static __xdata uint8_t ueienx_0;
 
 void
 ao_usb_set_address(uint8_t address)
 {
        UDADDR = (0 << ADDEN) | address;
-       ao_usb_running = 1;
        ao_usb_addr_pending = 1;
 }
 
@@ -60,14 +66,14 @@ static void
 ao_usb_dump_ep(uint8_t ep)
 {
        UENUM = ep;
-       printf ("EP %d: UECONX %02x UECFG0X %02x UECFG1X %02x UEIENX %02x UESTA0X %02x UESTA1X %02X\n",
+       debug ("EP %d: UECONX %02x UECFG0X %02x UECFG1X %02x UEIENX %02x UESTA0X %02x UESTA1X %02X\n",
                ep, UECONX, UECFG0X, UECFG1X, UEIENX, UESTA0X, UESTA1X);
 }
 
 static void
 ao_usb_set_ep0(void)
 {
-       printf ("set_ep0\n");
+       debug ("set_ep0\n");
        /* Set the CONTROL max packet size, single buffered */
        UENUM = 0;
        UECONX = (1 << EPEN);                                   /* Enable */
@@ -79,8 +85,8 @@ ao_usb_set_ep0(void)
                   (0 << EPBK0) |                               /* Single bank */
                   (1 << ALLOC));
 
-       UEIENX = ((1 << RXSTPE) |                               /* Enable SETUP interrupt */
-                 (1 << RXOUTE));                               /* Enable OUT interrupt */
+       ueienx_0 = ((1 << RXSTPE) |                             /* Enable SETUP interrupt */
+                   (1 << RXOUTE));                             /* Enable OUT interrupt */
 
 //     ao_usb_dump_ep(0);
        ao_usb_addr_pending = 0;
@@ -100,7 +106,9 @@ ao_usb_set_configuration(void)
                   (1 << EPBK0) |                               /* Double bank */
                   (1 << ALLOC));                               /* Allocate */
 
+#if 0
        UEIENX = ((1 << TXINE));                                /* Enable IN complete interrupt */
+#endif
 
        ao_usb_dump_ep(AO_USB_IN_EP);
 
@@ -118,6 +126,7 @@ ao_usb_set_configuration(void)
        UEIENX = ((1 << RXOUTE));                               /* Enable OUT complete interrupt */
 
        ao_usb_dump_ep(AO_USB_OUT_EP);
+       ao_usb_running = 1;
 }
 
 ISR(USB_GEN_vect)
@@ -156,10 +165,8 @@ ao_usb_ep0_set_in_pending(uint8_t in_pending)
 {
        ao_usb_ep0_in_pending = in_pending;
 
-       if (in_pending) {
-               UENUM = 0;
-               UEIENX = ((1 << RXSTPE) | (1 << RXOUTE) | (1 << TXINE));        /* Enable IN interrupt */
-       }
+       if (in_pending)
+               ueienx_0 = ((1 << RXSTPE) | (1 << RXOUTE) | (1 << TXINE));      /* Enable IN interrupt */
 }
 
 /* Send an IN data packet */
@@ -171,7 +178,7 @@ ao_usb_ep0_flush(void)
        cli();
        UENUM = 0;
        if (!(UEINTX & (1 << TXINI))) {
-               printf("EP0 not accepting IN data\n");
+               debug("EP0 not accepting IN data\n");
                ao_usb_ep0_set_in_pending(1);
        } else {
                this_len = ao_usb_ep0_in_len;
@@ -186,13 +193,13 @@ ao_usb_ep0_flush(void)
                else
                        ao_usb_ep0_set_in_pending(1);
 
-               printf ("Flush EP0 len %d:", this_len);
+               debug ("Flush EP0 len %d:", this_len);
                while (this_len--) {
                        uint8_t c = *ao_usb_ep0_in_data++;
-                       printf(" %02x", c);
+                       debug(" %02x", c);
                        UEDATX = c;
                }
-               printf ("\n");
+               debug ("\n");
 
                /* Clear the TXINI bit to send the packet */
                UEINTX &= ~(1 << TXINI);
@@ -208,17 +215,17 @@ ao_usb_ep0_fill(uint8_t len, uint8_t ack)
                len = ao_usb_ep0_out_len;
        ao_usb_ep0_out_len -= len;
 
-//     printf ("EP0 UEINTX %02x UEBCLX %d UEBCHX %d\n",
+//     debug ("EP0 UEINTX %02x UEBCLX %d UEBCHX %d\n",
 //             UEINTX, UEBCLX, UEBCHX);
        /* Pull all of the data out of the packet */
-       printf ("Fill EP0 len %d:", len);
+       debug ("Fill EP0 len %d:", len);
        UENUM = 0;
        while (len--) {
                uint8_t c = UEDATX;
                *ao_usb_ep0_out_data++ = c;
-               printf (" %02x", c);
+               debug (" %02x", c);
        }
-       printf ("\n");
+       debug ("\n");
 
        /* ACK the packet */
        UEINTX &= ~ack;
@@ -238,7 +245,7 @@ ao_usb_ep0_setup(void)
        ao_usb_ep0_out_len = 8;
        ao_usb_ep0_fill(8, (1 << RXSTPI) | (1 << RXOUTI) | (1 << TXINI));
        if (ao_usb_ep0_out_len != 0) {
-               printf ("invalid setup packet length\n");
+               debug ("invalid setup packet length\n");
                return;
        }
 
@@ -266,31 +273,31 @@ ao_usb_ep0_setup(void)
        ao_usb_ep0_in_len = 0;
        switch(ao_usb_setup.dir_type_recip & AO_USB_SETUP_TYPE_MASK) {
        case AO_USB_TYPE_STANDARD:
-               printf ("Standard setup packet\n");
+               debug ("Standard setup packet\n");
                switch(ao_usb_setup.dir_type_recip & AO_USB_SETUP_RECIP_MASK) {
                case AO_USB_RECIP_DEVICE:
-                       printf ("Device setup packet\n");
+                       debug ("Device setup packet\n");
                        switch(ao_usb_setup.request) {
                        case AO_USB_REQ_GET_STATUS:
-                               printf ("get status\n");
+                               debug ("get status\n");
                                ao_usb_ep0_queue_byte(0);
                                ao_usb_ep0_queue_byte(0);
                                break;
                        case AO_USB_REQ_SET_ADDRESS:
-                               printf ("set address %d\n", ao_usb_setup.value);
+                               debug ("set address %d\n", ao_usb_setup.value);
                                ao_usb_set_address(ao_usb_setup.value);
                                break;
                        case AO_USB_REQ_GET_DESCRIPTOR:
-                               printf ("get descriptor %d\n", ao_usb_setup.value);
+                               debug ("get descriptor %d\n", ao_usb_setup.value);
                                ao_usb_get_descriptor(ao_usb_setup.value);
                                break;
                        case AO_USB_REQ_GET_CONFIGURATION:
-                               printf ("get configuration %d\n", ao_usb_configuration);
+                               debug ("get configuration %d\n", ao_usb_configuration);
                                ao_usb_ep0_queue_byte(ao_usb_configuration);
                                break;
                        case AO_USB_REQ_SET_CONFIGURATION:
                                ao_usb_configuration = ao_usb_setup.value;
-                               printf ("set configuration %d\n", ao_usb_configuration);
+                               debug ("set configuration %d\n", ao_usb_configuration);
                                ao_usb_set_configuration();
                                break;
                        }
@@ -299,7 +306,7 @@ ao_usb_ep0_setup(void)
 #ifndef AVR
                        #pragma disable_warning 110
 #endif
-                       printf ("Interface setup packet\n");
+                       debug ("Interface setup packet\n");
                        switch(ao_usb_setup.request) {
                        case AO_USB_REQ_GET_STATUS:
                                ao_usb_ep0_queue_byte(0);
@@ -313,7 +320,7 @@ ao_usb_ep0_setup(void)
                        }
                        break;
                case AO_USB_RECIP_ENDPOINT:
-                       printf ("Endpoint setup packet\n");
+                       debug ("Endpoint setup packet\n");
                        switch(ao_usb_setup.request) {
                        case AO_USB_REQ_GET_STATUS:
                                ao_usb_ep0_queue_byte(0);
@@ -324,15 +331,15 @@ ao_usb_ep0_setup(void)
                }
                break;
        case AO_USB_TYPE_CLASS:
-               printf ("Class setup packet\n");
+               debug ("Class setup packet\n");
                switch (ao_usb_setup.request) {
                case SET_LINE_CODING:
-                       printf ("set line coding\n");
+                       debug ("set line coding\n");
                        ao_usb_ep0_out_len = 7;
                        ao_usb_ep0_out_data = (__xdata uint8_t *) &ao_usb_line_coding;
                        break;
                case GET_LINE_CODING:
-                       printf ("get line coding\n");
+                       debug ("get line coding\n");
                        ao_usb_ep0_in_len = 7;
                        ao_usb_ep0_in_data = (uint8_t *) &ao_usb_line_coding;
                        break;
@@ -344,7 +351,7 @@ ao_usb_ep0_setup(void)
        if (ao_usb_ep0_state != AO_USB_EP0_DATA_OUT) {
                if (ao_usb_setup.length < ao_usb_ep0_in_len)
                        ao_usb_ep0_in_len = ao_usb_setup.length;
-               printf ("Start ep0 in delivery %d\n", ao_usb_ep0_in_len);
+               debug ("Start ep0 in delivery %d\n", ao_usb_ep0_in_len);
                ao_usb_ep0_set_in_pending(1);
        }
 }
@@ -355,21 +362,21 @@ ao_usb_ep0(void)
 {
        uint8_t intx, udint;
 
-       printf ("usb task started\n");
+       debug ("usb task started\n");
        ao_usb_ep0_state = AO_USB_EP0_IDLE;
        for (;;) {
                cli();
                for (;;) {
                        udint = UDINT;
                        UDINT = 0;
-//                     printf ("UDINT %02x\n", udint);
+//                     debug ("UDINT %02x\n", udint);
                        if (udint & (1 << EORSTI)) {
                                ao_usb_configuration = 0;
                                ao_usb_set_ep0();
                        }
                        UENUM = 0;
                        intx = UEINTX;
-//                     printf ("UEINTX %02x\n", intx);
+//                     debug ("UEINTX %02x\n", intx);
                        if (intx & ((1 << RXSTPI) | (1 << RXOUTI)))
                                break;
                        if ((intx & (1 << TXINI))) {
@@ -381,14 +388,16 @@ ao_usb_ep0(void)
                                                UDADDR |= (1 << ADDEN);
                                                ao_usb_addr_pending = 0;
                                        }
-                                       UEIENX = ((1 << RXSTPE) | (1 << RXOUTE));       /* Disable IN interrupt */
+                                       ueienx_0 = ((1 << RXSTPE) | (1 << RXOUTE));     /* Disable IN interrupt */
                                }
                        }
-//                     printf ("usb task sleeping...\n");
+//                     debug ("usb task sleeping...\n");
+                       UENUM = 0;
+                       UEIENX = ueienx_0;
                        ao_sleep(&ao_usb_task);
                }
                sei();
-//             printf ("UEINTX for ep0 is %02x\n", intx);
+//             debug ("UEINTX for ep0 is %02x\n", intx);
                if (intx & (1 << RXSTPI)) {
                        ao_usb_ep0_setup();
                }
@@ -397,7 +406,7 @@ ao_usb_ep0(void)
                        ao_usb_ep0_set_in_pending(1);
                }
                if (intx & (1 << TXINI) && ao_usb_ep0_in_pending) {
-                       printf ("continue sending ep0 IN data\n");
+                       debug ("continue sending ep0 IN data\n");
                        ao_usb_ep0_flush();
                }
        }
@@ -413,15 +422,18 @@ ao_usb_in_wait(void)
                if (UEINTX & (1 << RWAL))
                        break;
 
+               cli();
                /* Wait for an IN buffer to be ready */
                for (;;) {
                        UENUM = AO_USB_IN_EP;
                        if ((UEINTX & (1 << TXINI)))
                                break;
+                       UEIENX = (1 << TXINE);
                        ao_sleep(&ao_usb_in_flushed);
                }
                /* Ack the interrupt */
                UEINTX &= ~(1 << TXINI);
+               sei();
        }
 }
 
@@ -459,8 +471,6 @@ ao_usb_putchar(char c) __critical __reentrant
        if (!ao_usb_running)
                return;
 
-       ao_usb_in_flushed = 0;
-
        ao_usb_in_wait();
 
        /* Queue a byte */
@@ -470,6 +480,7 @@ ao_usb_putchar(char c) __critical __reentrant
        /* Send the packet when full */
        if ((UEINTX & (1 << RWAL)) == 0)
                ao_usb_in_send();
+       ao_usb_in_flushed = 0;
 }
 
 static char
@@ -478,24 +489,30 @@ _ao_usb_pollchar(void)
        char c;
        uint8_t intx;
 
+       if (!ao_usb_running)
+               return AO_READ_AGAIN;
+
        for (;;) {
                UENUM = AO_USB_OUT_EP;
                intx = UEINTX;
-               printf("usb_pollchar UEINTX %02d\n", intx);
+               debug("usb_pollchar UEINTX %02d\n", intx);
                if (intx & (1 << RWAL))
                        break;
 
                if (intx & (1 << FIFOCON)) {
                        /* Ack the last packet */
-                       UEINTX &= ~(1 << FIFOCON);
+                       UEINTX = (uint8_t) ~(1 << FIFOCON);
                }
 
                /* Check to see if a packet has arrived */
-               if ((intx & (1 << RXOUTI)) == 0)
+               if ((intx & (1 << RXOUTI)) == 0) {
+                       UENUM = AO_USB_OUT_EP;
+                       UEIENX = (1 << RXOUTE);
                        return AO_READ_AGAIN;
+               }
 
                /* Ack the interrupt */
-               UEINTX &= ~(1 << RXOUTI);
+               UEINTX = ~(1 << RXOUTI);
        }
 
        /* Pull a character out of the fifo */
@@ -519,32 +536,55 @@ ao_usb_getchar(void) __critical
        char    c;
 
        cli();
-       while ((c = ao_usb_pollchar()) == AO_READ_AGAIN)
+       while ((c = _ao_usb_pollchar()) == AO_READ_AGAIN)
                ao_sleep(&ao_stdin_ready);
        sei();
        return c;
 }
 
+uint16_t       control_count;
+uint16_t       in_count;
+uint16_t       out_count;
+
 /* Endpoint interrupt */
 ISR(USB_COM_vect)
 {
+       uint8_t old_num = UENUM;
        uint8_t i = UEINT;
 
+#ifdef AO_LED_RED
        ao_led_toggle(AO_LED_RED);
+#endif
        UEINT = 0;
-       if (i & (1 << 0))
+       if (i & (1 << 0)) {
+               UENUM = 0;
+               UEIENX = 0;
                ao_wakeup(&ao_usb_task);
-       if (i & (1 << AO_USB_IN_EP))
+               ++control_count;
+       }
+       if (i & (1 << AO_USB_IN_EP)) {
+               UENUM = AO_USB_IN_EP;
+               UEIENX = 0;
                ao_wakeup(&ao_usb_in_flushed);
-       if (i & (1 << AO_USB_OUT_EP))
+               in_count++;
+       }
+       if (i & (1 << AO_USB_OUT_EP)) {
+               UENUM = AO_USB_OUT_EP;
+               UEIENX = 0;
                ao_wakeup(&ao_stdin_ready);
+               ++out_count;
+       }
+       UENUM = old_num;
 }
 
 #if AVR_VCC_5V
 #define AO_PAD_REGULATOR_INIT  (1 << UVREGE)   /* Turn on pad regulator */
 #endif
 #if AVR_VCC_3V3
-#define AO_PAD_REGULATOR_INIT  0               /* Turn off pad regulator */
+/* TeleScience V0.1 has a hardware bug -- UVcc is hooked up, but UCap is not
+ * Make this work by running power through UVcc to the USB system
+ */
+#define AO_PAD_REGULATOR_INIT  (1 << UVREGE)   /* Turn off pad regulator */
 #endif
 
 #if AVR_CLOCK == 16000000UL
@@ -598,11 +638,14 @@ ao_usb_enable(void)
 
        ao_usb_configuration = 0;
 
-       printf ("ao_usb_enable\n");
+       debug ("ao_usb_enable\n");
 
+       debug ("UHWCON %02x USBCON %02x PLLCSR %02x UDIEN %02x\n",
+              UHWCON, USBCON, PLLCSR, UDIEN);
        UDCON = (0 << DETACH);  /* Clear the DETACH bit to plug into the bus */
 }
 
+#if USB_DEBUG
 struct ao_task __xdata ao_usb_echo_task;
 
 static void
@@ -616,14 +659,30 @@ ao_usb_echo(void)
                ao_usb_flush();
        }
 }
+#endif
+
+static void
+ao_usb_irq(void)
+{
+       printf ("control: %d out: %d in: %d\n",
+               control_count, out_count, in_count);
+}
+
+__code struct ao_cmds ao_usb_cmds[] = {
+       { ao_usb_irq, "i\0Show USB interrupt counts" },
+       { 0, NULL }
+};
 
 void
 ao_usb_init(void)
 {
        ao_usb_enable();
 
-       printf ("ao_usb_init\n");
+       debug ("ao_usb_init\n");
        ao_add_task(&ao_usb_task, ao_usb_ep0, "usb");
+#if USB_DEBUG
        ao_add_task(&ao_usb_echo_task, ao_usb_echo, "usb echo");
-//     ao_add_stdio(ao_usb_pollchar, ao_usb_putchar, ao_usb_flush);
+#endif
+       ao_cmd_register(&ao_usb_cmds[0]);
+       ao_add_stdio(ao_usb_pollchar, ao_usb_putchar, ao_usb_flush);
 }