qmk_firmware/keyboards/bioi/main.c

388 lines
8.8 KiB
C

/*
Copyright 2019 Basic I/O Instruments(Scott Wei) <scot.wei@gmail.com>
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 2 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
#include <avr/pgmspace.h>
#include <util/delay.h>
#include "report.h"
#include "host.h"
#include "host_driver.h"
#include "keyboard.h"
#include "action.h"
#include "led.h"
#include "sendchar.h"
#include "debug.h"
#include "print.h"
#ifdef SLEEP_LED_ENABLE
#include "sleep_led.h"
#endif
#include "suspend.h"
#include "usb_descriptor.h"
#include "lufa.h"
#include "quantum.h"
#include <util/atomic.h>
#ifdef NKRO_ENABLE
#include "keycode_config.h"
extern keymap_config_t keymap_config;
#endif
#ifdef AUDIO_ENABLE
#include <audio.h>
#endif
#ifdef BLUETOOTH_ENABLE
#ifdef BLUETOOTH_BLUEFRUIT_LE
#include "bluefruit_le.h"
#else
#include "bluetooth.h"
#endif
#endif
#ifdef VIRTSER_ENABLE
#include "virtser.h"
#endif
#if defined(RGBLIGHT_ENABLE)
#include "rgblight.h"
#endif
#ifdef MIDI_ENABLE
#include "qmk_midi.h"
#endif
#ifdef RAW_ENABLE
#include "raw_hid.h"
#endif
#include "ble.h"
#include "usart.h"
#include <avr/power.h>
#include <avr/sleep.h>
bool force_usb = false; //Reserved for FORCE USB Mode function.
bool force_ble = false; //Reserved for FORCE USB Mode function.
bool usb_connected = false;
bool ble_enabled = false;
uint32_t kb_idle_timer = 0;
bool usb_state_sent = false;
uint8_t USB_DeviceLastState = 0;
#ifdef RAW_ENABLE
/** \brief Raw HID Task
*
* FIXME: Needs doc
*/
static void raw_hid_task(void)
{
// Create a temporary buffer to hold the read in data from the host
uint8_t data[RAW_EPSIZE];
bool data_read = false;
// Device must be connected and configured for the task to run
if (USB_DeviceState != DEVICE_STATE_Configured)
return;
Endpoint_SelectEndpoint(RAW_OUT_EPNUM);
// Check to see if a packet has been sent from the host
if (Endpoint_IsOUTReceived())
{
// Check to see if the packet contains data
if (Endpoint_IsReadWriteAllowed())
{
/* Read data */
Endpoint_Read_Stream_LE(data, sizeof(data), NULL);
data_read = true;
}
// Finalize the stream transfer to receive the last packet
Endpoint_ClearOUT();
if (data_read)
{
raw_hid_receive(data, sizeof(data));
}
}
}
#endif
static void setup_mcu(void)
{
/* Disable watchdog if enabled by bootloader/fuses */
MCUSR &= ~(1 << WDRF);
wdt_disable();
CLKPR = (1 << CLKPCE);
CLKPR = (0 << CLKPS3) | (0 << CLKPS2) | (0 << CLKPS1) | (0 << CLKPS0);
}
static void setup_usb(void)
{
// Leonardo needs. Without this USB device is not recognized.
USB_Disable();
USB_Init();
// for Console_Task
USB_Device_EnableSOFEvents();
print_set_sendchar(sendchar);
}
void power_saving(void)
{
power_adc_disable();
power_usart0_disable();
power_spi_disable();
power_twi_disable();
USBCON |= (1 << FRZCLK); // Freeze the USB Clock
PLLCSR &= ~(1 << PLLE); // Disable the USB Clock (PPL)
USBCON &= ~(1 << USBE);
}
void power_recover(void)
{
USBCON |= (1 << USBE);
PLLCSR |= (1 << PLLE); // Resume the USB Clock (PPL)
USBCON &= ~(1 << FRZCLK); // Resume the USB Clock
power_adc_enable();
power_usart0_enable();
power_spi_enable();
power_twi_enable();
}
void ble_task_init(void)
{
kb_idle_timer = timer_read32(); //Mark current time, reserved for further usage;
}
void ble_task(void)
{
if (USB_DeviceLastState != USB_DeviceState)
{
usb_state_sent = false;
#ifdef BLE_DEBUG
send_str(PSTR("USB State Changed\r\n"));
if (USB_DeviceState == DEVICE_STATE_Unattached)
{
send_str(PSTR("USB State Unattached\r\n"));
}
#endif
if (USB_DeviceState == DEVICE_STATE_Powered)
{
#ifdef BLE_DEBUG
send_str(PSTR("USB State Powered\r\n"));
#endif
power_recover();
host_set_driver(&null_driver);
}
#ifdef BLE_DEBUG
if ((USB_DeviceState == DEVICE_STATE_Default))
{
send_str(PSTR("USB State Default\r\n"));
}
if ((USB_DeviceState == DEVICE_STATE_Addressed))
{
send_str(PSTR("USB State Addressed\r\n"));
}
if (USB_DeviceState == DEVICE_STATE_Configured)
{
send_str(PSTR("USB State Configured\r\n"));
}
if (USB_DeviceState > DEVICE_STATE_Unattached)
{
}
else
{
//
}
#endif
}
else
{
#ifdef BLE_DEBUG
if (!usb_state_sent)
{
if (USB_DeviceState == DEVICE_STATE_Unattached)
{
send_str(PSTR("USB State Stopped at Unattached\r\n"));
}
if (USB_DeviceState == DEVICE_STATE_Powered)
{
send_str(PSTR("USB State Stopped at Powered\r\n"));
}
if ((USB_DeviceState == DEVICE_STATE_Default))
{
send_str(PSTR("USB State Stopped at Default\r\n"));
}
if ((USB_DeviceState == DEVICE_STATE_Addressed))
{
send_str(PSTR("USB State Stopped at Addressed\r\n"));
}
if (USB_DeviceState == DEVICE_STATE_Configured)
{
send_str(PSTR("USB State Stopped at Configured\r\n"));
}
}
#endif
if (USB_DeviceState == DEVICE_STATE_Unattached)
{
if (host_get_driver() && host_get_driver() != &bluefruit_driver)
{
#ifdef BLE_DEBUG
send_str(PSTR("USB State stopped at Unattached\r\n"));
#endif
ble_task_init();
force_usb = 0;
usb_connected = 0;
//Reinit USB to prepare for next connection.
USB_Init();
USB_Detach();
USB_Attach();
#ifdef BLE_DEBUG
send_str(PSTR("Loading &bluefruit_driver\r\n"));
#endif
host_set_driver(&bluefruit_driver);
clear_keyboard();
power_saving();
}
else
{
//Do nothing if USB is unattached and the driver is &bluefruit_driver
}
}
if (USB_DeviceState == DEVICE_STATE_Configured)
{
if (host_get_driver() && host_get_driver() != &lufa_driver)
{
#ifdef BLE_DEBUG
send_str(PSTR("USB State stopped at Configured\r\n"));
#endif
power_recover();
usb_connected = 1;
ble_enabled = 0;
#ifdef BLE_DEBUG
send_str(PSTR("Loading &lufa_driver\r\n"));
#endif
host_set_driver(&lufa_driver);
clear_keyboard();
}
else
{
//Do nothing if the driver is &lufa_driver
}
}
usb_state_sent = true;
}
USB_DeviceLastState = USB_DeviceState;
}
// Use a custom main() function because the task logic is different from the common one.
int main(void)
{
#ifdef MIDI_ENABLE
setup_midi();
#endif
setup_mcu();
keyboard_setup();
setup_usb();
sei();
#if defined(BLUETOOTH_RN42)
serial_init();
#endif
/* wait for USB startup to get ready for debug output */
uint8_t timeout = 255; // timeout when USB is not available(Bluetooth)
while (timeout-- && USB_DeviceState != DEVICE_STATE_Configured)
{
wait_ms(4);
#if defined(INTERRUPT_CONTROL_ENDPOINT)
;
#else
USB_USBTask();
#endif
}
print("\nUSB init\n");
keyboard_init();
host_set_driver(&lufa_driver);
backlight_disable();
//host_set_driver(&lufa_driver);
print("Keyboard initialized.\n");
//Init Hardware UART
usart_init();
#ifdef BLE_DEBUG
send_str(PSTR("Keyboard has been setup up\r\n"));
if (usb_connected)
{
send_str(PSTR("usb_connected=1\r\n"));
}
else
{
send_str(PSTR("usb_connected=0\r\n"));
}
#endif
#ifdef SLEEP_LED_ENABLE
sleep_led_init();
#endif
#ifdef VIRTSER_ENABLE
virtser_init();
#endif
while (1)
{
ble_task();
keyboard_task();
#ifdef RAW_ENABLE
raw_hid_task();
#endif
#if defined(RGBLIGHT_ENABLE)
rgblight_task();
#endif
#if !defined(INTERRUPT_CONTROL_ENDPOINT)
USB_USBTask();
#endif
}
}