#include <avr/io.h>
#include <stdint.h>
#include "usb_common.h"
#include "usb_private.h"
#include "usb_api.h"
#include "wiring.h"
void usb_serial_class::begin(long speed)
{
peek_buf = -1;
usb_init();
uint16_t begin_wait = (uint16_t)millis();
while (1) {
if (usb_configuration) {
delay(200); return;
}
if (usb_suspended) {
uint16_t begin_suspend = (uint16_t)millis();
while (usb_suspended) {
if ((uint16_t)millis() - begin_suspend > 250) {
return;
}
}
}
if ((uint16_t)millis() - begin_wait > 2500) return;
}
}
void usb_serial_class::end()
{
usb_shutdown();
delay(25);
}
int usb_serial_class::available()
{
uint8_t n=0, i, intr_state;
intr_state = SREG;
cli();
if (usb_configuration) {
UENUM = CDC_RX_ENDPOINT;
n = UEBCLX;
if (!n) {
i = UEINTX;
if (i & (1<<RXOUTI) && !(i & (1<<RWAL))) UEINTX = 0x6B;
}
}
SREG = intr_state;
if (peek_buf >= 0 && n < 255) n++;
return n;
}
int usb_serial_class::peek()
{
if (peek_buf < 0) peek_buf = read();
return peek_buf;
}
int usb_serial_class::read(void)
{
uint8_t c, intr_state;
if (peek_buf >= 0) {
c = peek_buf;
peek_buf = -1;
return c;
}
intr_state = SREG;
cli();
if (!usb_configuration) {
SREG = intr_state;
return -1;
}
UENUM = CDC_RX_ENDPOINT;
retry:
c = UEINTX;
if (!(c & (1<<RWAL))) {
if (c & (1<<RXOUTI)) {
UEINTX = 0x6B;
goto retry;
}
SREG = intr_state;
return -1;
}
c = UEDATX;
if (!(UEINTX & (1<<RWAL))) UEINTX = 0x6B;
SREG = intr_state;
return c;
}
void usb_serial_class::flush()
{
uint8_t intr_state;
if (usb_configuration) {
intr_state = SREG;
cli();
UENUM = CDC_RX_ENDPOINT;
while ((UEINTX & (1<<RWAL))) {
UEINTX = 0x6B;
}
SREG = intr_state;
}
peek_buf = -1;
}
#if 0#endif
size_t usb_serial_class::write(const uint8_t *buffer, uint16_t size)
{
uint8_t timeout, intr_state, write_size;
size_t count=0;
if (!usb_configuration) {
setWriteError();
goto end;
}
intr_state = SREG;
cli();
UENUM = CDC_TX_ENDPOINT;
if (transmit_previous_timeout) {
if (!(UEINTX & (1<<RWAL))) {
SREG = intr_state;
setWriteError();
goto end;
}
transmit_previous_timeout = 0;
}
while (size) {
timeout = UDFNUML + TRANSMIT_TIMEOUT;
while (1) {
if (UEINTX & (1<<RWAL)) break;
SREG = intr_state;
if (UDFNUML == timeout) {
transmit_previous_timeout = 1;
setWriteError();
goto end;
}
if (!usb_configuration) {
setWriteError();
goto end;
}
intr_state = SREG;
cli();
UENUM = CDC_TX_ENDPOINT;
}
write_size = CDC_TX_SIZE - UEBCLX;
if (write_size > size) write_size = size;
size -= write_size;
count += write_size;
#define ASM_COPY1(src, dest, tmp) "ld " tmp ", " src "\n\t" "st " dest ", " tmp "\n\t"
#define ASM_COPY2(src, dest, tmp) ASM_COPY1(src, dest, tmp) ASM_COPY1(src, dest, tmp)
#define ASM_COPY4(src, dest, tmp) ASM_COPY2(src, dest, tmp) ASM_COPY2(src, dest, tmp)
#define ASM_COPY8(src, dest, tmp) ASM_COPY4(src, dest, tmp) ASM_COPY4(src, dest, tmp)
#if 1
do {
uint8_t tmp;
asm volatile(
"L%=begin:" "\n\t"
"ldi r30, %4" "\n\t"
"sub r30, %3" "\n\t"
"cpi r30, %4" "\n\t"
"brsh L%=err" "\n\t"
"lsl r30" "\n\t"
"clr r31" "\n\t"
"subi r30, lo8(-(pm(L%=table)))" "\n\t"
"sbci r31, hi8(-(pm(L%=table)))" "\n\t"
"ijmp" "\n\t"
"L%=err:" "\n\t"
"rjmp L%=end" "\n\t"
"L%=table:" "\n\t"
#if (CDC_TX_SIZE == 64)
ASM_COPY8("Y+", "X", "%1")
ASM_COPY8("Y+", "X", "%1")
ASM_COPY8("Y+", "X", "%1")
ASM_COPY8("Y+", "X", "%1")
#endif
#if (CDC_TX_SIZE >= 32)
ASM_COPY8("Y+", "X", "%1")
ASM_COPY8("Y+", "X", "%1")
#endif
#if (CDC_TX_SIZE >= 16)
ASM_COPY8("Y+", "X", "%1")
#endif
ASM_COPY8("Y+", "X", "%1")
"L%=end:" "\n\t"
: "+y" (buffer), "=r" (tmp)
: "x" (&UEDATX), "r" (write_size), "M" (CDC_TX_SIZE)
: "r30", "r31"
);
} while (0);
#endif
if (!(UEINTX & (1<<RWAL))) UEINTX = 0x3A;
transmit_flush_timer = TRANSMIT_FLUSH_TIMEOUT;
}
SREG = intr_state;
end:
return count;
}
void usb_serial_class::send_now(void)
{
uint8_t intr_state;
intr_state = SREG;
cli();
if (usb_configuration && transmit_flush_timer) {
UENUM = CDC_TX_ENDPOINT;
UEINTX = 0x3A;
transmit_flush_timer = 0;
}
SREG = intr_state;
}
uint32_t usb_serial_class::baud(void)
{
return *(uint32_t *)cdc_line_coding;
}
uint8_t usb_serial_class::stopbits(void)
{
return cdc_line_coding[4];
}
uint8_t usb_serial_class::paritytype(void)
{
return cdc_line_coding[5];
}
uint8_t usb_serial_class::numbits(void)
{
return cdc_line_coding[6];
}
uint8_t usb_serial_class::dtr(void)
{
return (cdc_line_rtsdtr & USB_SERIAL_DTR) ? 1 : 0;
}
uint8_t usb_serial_class::rts(void)
{
return (cdc_line_rtsdtr & USB_SERIAL_RTS) ? 1 : 0;
}
usb_serial_class::operator bool()
{
if (usb_configuration &&
(cdc_line_rtsdtr & (USB_SERIAL_DTR | USB_SERIAL_RTS))) {
return true;
}
return false;
}
size_t usb_keyboard_class::write(uint8_t c)
{
if (c < 0x80) {
utf8_state = 0;
write_unicode(c);
} else if (c < 0xC0) {
c &= 0x3F;
if (utf8_state == 1) {
utf8_state = 0;
write_unicode(unicode_wchar | c);
} else if (utf8_state == 2) {
unicode_wchar |= ((uint16_t)c << 6);
utf8_state = 1;
}
} else if (c < 0xE0) {
unicode_wchar = (uint16_t)(c & 0x1F) << 6;
utf8_state = 1;
} else if (c < 0xF0) {
unicode_wchar = (uint16_t)(c & 0x0F) << 12;
utf8_state = 2;
} else {
utf8_state = 255;
}
return 1;
}
KEYCODE_TYPE usb_keyboard_class::unicode_to_keycode(uint16_t cpoint)
{
if (cpoint < 32) {
if (cpoint == 10) return KEY_ENTER & KEYCODE_MASK;
if (cpoint == 11) return KEY_TAB & KEYCODE_MASK;
return 0;
}
if (cpoint < 128) {
if (sizeof(KEYCODE_TYPE) == 1) {
return pgm_read_byte(keycodes_ascii + (cpoint - 0x20));
} else if (sizeof(KEYCODE_TYPE) == 2) {
return pgm_read_word(keycodes_ascii + (cpoint - 0x20));
}
return 0;
}
#ifdef ISO_8859_1_A0
if (cpoint <= 0xA0) return 0;
if (cpoint < 0x100) {
if (sizeof(KEYCODE_TYPE) == 1) {
return pgm_read_byte(keycodes_iso_8859_1 + (cpoint - 0xA0));
} else if (sizeof(KEYCODE_TYPE) == 2) {
return pgm_read_word(keycodes_iso_8859_1 + (cpoint - 0xA0));
}
return 0;
}
#endif
#ifdef KEYCODE_EXTRA00
if (cpoint == UNICODE_EXTRA00) return KEYCODE_EXTRA00 & 0x3FFF;
#endif
#ifdef KEYCODE_EXTRA01
if (cpoint == UNICODE_EXTRA01) return KEYCODE_EXTRA01 & 0x3FFF;
#endif
#ifdef KEYCODE_EXTRA02
if (cpoint == UNICODE_EXTRA02) return KEYCODE_EXTRA02 & 0x3FFF;
#endif
#ifdef KEYCODE_EXTRA03
if (cpoint == UNICODE_EXTRA03) return KEYCODE_EXTRA03 & 0x3FFF;
#endif
#ifdef KEYCODE_EXTRA04
if (cpoint == UNICODE_EXTRA04) return KEYCODE_EXTRA04 & 0x3FFF;
#endif
#ifdef KEYCODE_EXTRA05
if (cpoint == UNICODE_EXTRA05) return KEYCODE_EXTRA05 & 0x3FFF;
#endif
#ifdef KEYCODE_EXTRA06
if (cpoint == UNICODE_EXTRA06) return KEYCODE_EXTRA06 & 0x3FFF;
#endif
#ifdef KEYCODE_EXTRA07
if (cpoint == UNICODE_EXTRA07) return KEYCODE_EXTRA07 & 0x3FFF;
#endif
#ifdef KEYCODE_EXTRA08
if (cpoint == UNICODE_EXTRA08) return KEYCODE_EXTRA08 & 0x3FFF;
#endif
#ifdef KEYCODE_EXTRA09
if (cpoint == UNICODE_EXTRA09) return KEYCODE_EXTRA09 & 0x3FFF;
#endif
return 0;
}
void usb_keyboard_class::write_keycode(KEYCODE_TYPE keycode)
{
if (!keycode) return;
#ifdef DEADKEYS_MASK
KEYCODE_TYPE deadkeycode = deadkey_to_keycode(keycode);
if (deadkeycode) write_key(deadkeycode);
#endif
write_key(keycode);
}
KEYCODE_TYPE usb_keyboard_class::deadkey_to_keycode(KEYCODE_TYPE keycode)
{
#ifdef DEADKEYS_MASK
keycode &= DEADKEYS_MASK;
if (keycode == 0) return 0;
#ifdef ACUTE_ACCENT_BITS
if (keycode == ACUTE_ACCENT_BITS) return DEADKEY_ACUTE_ACCENT;
#endif
#ifdef CEDILLA_BITS
if (keycode == CEDILLA_BITS) return DEADKEY_CEDILLA;
#endif
#ifdef CIRCUMFLEX_BITS
if (keycode == CIRCUMFLEX_BITS) return DEADKEY_CIRCUMFLEX;
#endif
#ifdef DIAERESIS_BITS
if (keycode == DIAERESIS_BITS) return DEADKEY_DIAERESIS;
#endif
#ifdef GRAVE_ACCENT_BITS
if (keycode == GRAVE_ACCENT_BITS) return DEADKEY_GRAVE_ACCENT;
#endif
#ifdef TILDE_BITS
if (keycode == TILDE_BITS) return DEADKEY_TILDE;
#endif
#ifdef RING_ABOVE_BITS
if (keycode == RING_ABOVE_BITS) return DEADKEY_RING_ABOVE;
#endif
#endif return 0;
}
void usb_keyboard_class::write_key(KEYCODE_TYPE keycode)
{
keyboard_report_data[0] = keycode_to_modifier(keycode);
keyboard_report_data[1] = 0;
keyboard_report_data[2] = keycode_to_key(keycode);
keyboard_report_data[3] = 0;
keyboard_report_data[4] = 0;
keyboard_report_data[5] = 0;
keyboard_report_data[6] = 0;
keyboard_report_data[7] = 0;
send_now();
keyboard_report_data[0] = 0;
keyboard_report_data[2] = 0;
send_now();
}
uint8_t usb_keyboard_class::keycode_to_modifier(KEYCODE_TYPE keycode)
{
uint8_t modifier=0;
#ifdef SHIFT_MASK
if (keycode & SHIFT_MASK) modifier |= MODIFIERKEY_SHIFT;
#endif
#ifdef ALTGR_MASK
if (keycode & ALTGR_MASK) modifier |= MODIFIERKEY_RIGHT_ALT;
#endif
#ifdef RCTRL_MASK
if (keycode & RCTRL_MASK) modifier |= MODIFIERKEY_RIGHT_CTRL;
#endif
return modifier;
}
uint8_t usb_keyboard_class::keycode_to_key(KEYCODE_TYPE keycode)
{
uint8_t key = keycode & 0x3F;
#ifdef KEY_NON_US_100
if (key == KEY_NON_US_100) key = 100;
#endif
return key;
}
void usb_keyboard_class::set_modifier(uint16_t c)
{
keyboard_report_data[0] = (uint8_t)c;
}
void usb_keyboard_class::set_key1(uint8_t c)
{
keyboard_report_data[2] = c;
}
void usb_keyboard_class::set_key2(uint8_t c)
{
keyboard_report_data[3] = c;
}
void usb_keyboard_class::set_key3(uint8_t c)
{
keyboard_report_data[4] = c;
}
void usb_keyboard_class::set_key4(uint8_t c)
{
keyboard_report_data[5] = c;
}
void usb_keyboard_class::set_key5(uint8_t c)
{
keyboard_report_data[6] = c;
}
void usb_keyboard_class::set_key6(uint8_t c)
{
keyboard_report_data[7] = c;
}
void usb_keyboard_class::send_now(void)
{
uint8_t intr_state, timeout;
if (!usb_configuration) return;
intr_state = SREG;
cli();
UENUM = KEYBOARD_ENDPOINT;
timeout = UDFNUML + 50;
while (1) {
if (UEINTX & (1<<RWAL)) break;
SREG = intr_state;
if (!usb_configuration) return;
if (UDFNUML == timeout) return;
intr_state = SREG;
cli();
UENUM = KEYBOARD_ENDPOINT;
}
UEDATX = keyboard_report_data[0];
UEDATX = keyboard_report_data[1];
UEDATX = keyboard_report_data[2];
UEDATX = keyboard_report_data[3];
UEDATX = keyboard_report_data[4];
UEDATX = keyboard_report_data[5];
UEDATX = keyboard_report_data[6];
UEDATX = keyboard_report_data[7];
UEINTX = 0x3A;
keyboard_idle_count = 0;
SREG = intr_state;
}
void usb_keyboard_class::press(uint16_t n)
{
uint8_t key, mod, msb, modrestore=0;
msb = n >> 8;
if (msb >= 0xC2) {
if (msb <= 0xDF) {
n = (n & 0x3F) | ((uint16_t)(msb & 0x1F) << 6);
} else if (msb == 0xF0) {
presskey(n, 0);
return;
} else if (msb == 0xE0) {
presskey(0, n);
return;
} else if (msb == 0xE2) {
return;
} else if (msb >= 0xE4 && msb <= 0xE7) {
return;
} else {
return;
}
}
KEYCODE_TYPE keycode = unicode_to_keycode(n);
if (!keycode) return;
#ifdef DEADKEYS_MASK
KEYCODE_TYPE deadkeycode = deadkey_to_keycode(keycode);
if (deadkeycode) {
modrestore = keyboard_report_data[0];
if (modrestore) {
keyboard_report_data[0] = 0;
send_now();
}
mod = keycode_to_modifier(deadkeycode);
key = keycode_to_key(deadkeycode);
presskey(key, mod);
releasekey(key, mod);
}
#endif
mod = keycode_to_modifier(keycode);
key = keycode_to_key(keycode);
presskey(key, mod | modrestore);
}
void usb_keyboard_class::release(uint16_t n)
{
uint8_t key, mod, msb;
msb = n >> 8;
if (msb >= 0xC2) {
if (msb <= 0xDF) {
n = (n & 0x3F) | ((uint16_t)(msb & 0x1F) << 6);
} else if (msb == 0xF0) {
releasekey(n, 0);
return;
} else if (msb == 0xE0) {
releasekey(0, n);
return;
} else if (msb == 0xE2) {
return;
} else if (msb >= 0xE4 && msb <= 0xE7) {
return;
} else {
return;
}
}
KEYCODE_TYPE keycode = unicode_to_keycode(n);
if (!keycode) return;
mod = keycode_to_modifier(keycode);
key = keycode_to_key(keycode);
releasekey(key, mod);
}
void usb_keyboard_class::presskey(uint8_t key, uint8_t modifier)
{
bool send_required = false;
uint8_t i;
if (modifier) {
if ((keyboard_report_data[0] & modifier) != modifier) {
keyboard_report_data[0] |= modifier;
send_required = true;
}
}
if (key) {
for (i=2; i < 8; i++) {
if (keyboard_report_data[i] == key) goto end;
}
for (i=2; i < 8; i++) {
if (keyboard_report_data[i] == 0) {
keyboard_report_data[i] = key;
send_required = true;
goto end;
}
}
}
end:
if (send_required) send_now();
}
void usb_keyboard_class::releasekey(uint8_t key, uint8_t modifier)
{
bool send_required = false;
uint8_t i;
if (modifier) {
if ((keyboard_report_data[0] & modifier) != 0) {
keyboard_report_data[0] &= ~modifier;
send_required = true;
}
}
if (key) {
for (i=2; i < 8; i++) {
if (keyboard_report_data[i] == key) {
keyboard_report_data[i] = 0;
send_required = true;
}
}
}
if (send_required) send_now();
}
void usb_keyboard_class::releaseAll(void)
{
uint8_t i, anybits;
anybits = keyboard_report_data[0];
for (i=2; i < 8; i++) {
anybits |= keyboard_report_data[i];
keyboard_report_data[i] = 0;
}
if (!anybits) return;
keyboard_report_data[0] = 0;
send_now();
}
void usb_mouse_class::move(int8_t x, int8_t y, int8_t wheel, int8_t horiz)
{
uint8_t intr_state, timeout;
if (!usb_configuration) return;
if (x == -128) x = -127;
if (y == -128) y = -127;
if (wheel == -128) wheel = -127;
if (horiz == -128) horiz = -127;
intr_state = SREG;
cli();
UENUM = MOUSE_ENDPOINT;
timeout = UDFNUML + 50;
while (1) {
if (UEINTX & (1<<RWAL)) break;
SREG = intr_state;
if (!usb_configuration) return;
if (UDFNUML == timeout) return;
intr_state = SREG;
cli();
UENUM = MOUSE_ENDPOINT;
}
UEDATX = mouse_buttons;
UEDATX = x;
UEDATX = y;
UEDATX = wheel;
UEDATX = horiz;
UEINTX = 0x3A;
SREG = intr_state;
}
void usb_mouse_class::click(uint8_t b)
{
mouse_buttons = b;
move(0, 0);
mouse_buttons = 0;
move(0, 0);
}
void usb_mouse_class::scroll(int8_t wheel, int8_t horiz)
{
move(0, 0, wheel, horiz);
}
void usb_mouse_class::set_buttons(uint8_t left, uint8_t middle, uint8_t right, uint8_t back, uint8_t forward)
{
uint8_t mask=0;
if (left) mask |= 1;
if (middle) mask |= 4;
if (right) mask |= 2;
if (back) mask |= 8;
if (forward) mask |= 16;
mouse_buttons = mask;
move(0, 0);
}
void usb_mouse_class::press(uint8_t b)
{
uint8_t prev = mouse_buttons;
mouse_buttons |= (b & 7);
if (mouse_buttons != prev) move(0, 0);
}
void usb_mouse_class::release(uint8_t b)
{
uint8_t prev = mouse_buttons;
mouse_buttons &= ~(b & 7);
if (mouse_buttons != prev) move(0, 0);
}
bool usb_mouse_class::isPressed(uint8_t b)
{
return ((mouse_buttons & (b & 7)) != 0);
}
void usb_joystick_class::send_now(void)
{
uint8_t intr_state, timeout;
if (!usb_configuration) return;
intr_state = SREG;
cli();
UENUM = JOYSTICK_ENDPOINT;
timeout = UDFNUML + 50;
while (1) {
if (UEINTX & (1<<RWAL)) break;
SREG = intr_state;
if (!usb_configuration) return;
if (UDFNUML == timeout) return;
intr_state = SREG;
cli();
UENUM = JOYSTICK_ENDPOINT;
}
UEDATX = joystick_report_data[0];
UEDATX = joystick_report_data[1];
UEDATX = joystick_report_data[2];
UEDATX = joystick_report_data[3];
UEDATX = joystick_report_data[4];
UEDATX = joystick_report_data[5];
UEDATX = joystick_report_data[6];
UEDATX = joystick_report_data[7];
UEDATX = joystick_report_data[8];
UEDATX = joystick_report_data[9];
UEDATX = joystick_report_data[10];
UEDATX = joystick_report_data[11];
UEINTX = 0x3A;
SREG = intr_state;
}
usb_serial_class Serial = usb_serial_class();
usb_keyboard_class Keyboard = usb_keyboard_class();
usb_mouse_class Mouse = usb_mouse_class();
usb_joystick_class Joystick = usb_joystick_class();