117 lines
		
	
	
		
			3.3 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			117 lines
		
	
	
		
			3.3 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
/*
 | 
						|
Bluefruit Protocol for TMK firmware
 | 
						|
Author: Benjamin Gould, 2013
 | 
						|
Based on code Copyright 2011 Jun Wako <wakojun@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 <stdint.h>
 | 
						|
#include <avr/interrupt.h>
 | 
						|
#include <avr/wdt.h>
 | 
						|
#include <avr/sleep.h>
 | 
						|
#include <util/delay.h>
 | 
						|
#include "serial.h"
 | 
						|
#include "keyboard.h"
 | 
						|
#include "usb.h"
 | 
						|
#include "host.h"
 | 
						|
#include "timer.h"
 | 
						|
#include "print.h"
 | 
						|
#include "debug.h"
 | 
						|
#include "sendchar.h"
 | 
						|
#include "suspend.h"
 | 
						|
#include "bluefruit.h"
 | 
						|
#include "pjrc.h"
 | 
						|
 | 
						|
#define CPU_PRESCALE(n)    (CLKPR = 0x80, CLKPR = (n))
 | 
						|
 | 
						|
#define HOST_DRIVER_NOT_SET     0
 | 
						|
#define BLUEFRUIT_HOST_DRIVER   1
 | 
						|
#define PJRC_HOST_DRIVER        2
 | 
						|
 | 
						|
int main(void)
 | 
						|
{   
 | 
						|
 | 
						|
    CPU_PRESCALE(0);
 | 
						|
 | 
						|
    DDRD  = _BV(PD5);
 | 
						|
    DDRB  = _BV(PB0);
 | 
						|
    
 | 
						|
    PORTD = _BV(PD5);
 | 
						|
    PORTB = _BV(PB0);
 | 
						|
 | 
						|
    print_set_sendchar(sendchar);
 | 
						|
 | 
						|
    usb_init();
 | 
						|
    _delay_ms(2000);
 | 
						|
    // while (!usb_configured()) /* wait */
 | 
						|
 | 
						|
    dprintf("Initializing keyboard...\n");
 | 
						|
    keyboard_init();
 | 
						|
    
 | 
						|
    // This implementation is pretty simplistic... if the USB connection
 | 
						|
    // is not configured, choose the Bluefruit, otherwise use USB
 | 
						|
    // Definitely would prefer to have this driven by an input pin and make
 | 
						|
    // it switch dynamically - BCG
 | 
						|
    if (!usb_configured()) {
 | 
						|
    
 | 
						|
        // Send power to Bluefruit... Adafruit says it takes 27 mA, I think
 | 
						|
        // the pins should provide 40 mA, but just in case I switch the 
 | 
						|
        // Bluefruit using a transistor - BCG
 | 
						|
        DDRB   = _BV(PB6);
 | 
						|
        PORTB |= _BV(PB6);
 | 
						|
    
 | 
						|
        dprintf("Setting host driver to bluefruit...\n");
 | 
						|
        host_set_driver(bluefruit_driver());
 | 
						|
 | 
						|
        dprintf("Initializing serial...\n");
 | 
						|
        serial_init();
 | 
						|
        
 | 
						|
        // wait an extra second for the PC's operating system
 | 
						|
        // to load drivers and do whatever it does to actually
 | 
						|
        // be ready for input
 | 
						|
        _delay_ms(1000);
 | 
						|
        PORTD = ~_BV(PD5);
 | 
						|
        dprintf("Starting main loop");
 | 
						|
        while (1) {
 | 
						|
            keyboard_task();
 | 
						|
        }
 | 
						|
 | 
						|
    } else {
 | 
						|
 | 
						|
        // I'm not smart enough to get this done with LUFA - BCG
 | 
						|
        dprintf("Setting host driver to PJRC...\n");
 | 
						|
        host_set_driver(pjrc_driver());
 | 
						|
#ifdef SLEEP_LED_ENABLE
 | 
						|
    sleep_led_init();
 | 
						|
#endif
 | 
						|
        // wait an extra second for the PC's operating system
 | 
						|
        // to load drivers and do whatever it does to actually
 | 
						|
        // be ready for input
 | 
						|
        _delay_ms(1000);
 | 
						|
        PORTB = ~_BV(PB0);
 | 
						|
        dprintf("Starting main loop");
 | 
						|
        while (1) {
 | 
						|
            while (suspend) {
 | 
						|
                suspend_power_down();
 | 
						|
                if (remote_wakeup && suspend_wakeup_condition()) {
 | 
						|
                    usb_remote_wakeup();
 | 
						|
                }
 | 
						|
            }
 | 
						|
            keyboard_task(); 
 | 
						|
        }
 | 
						|
    }
 | 
						|
 | 
						|
}
 |