main.c

   1 /* Copyright (C) 2011-2013 by Jacob Alexander
   2  *
   3  * Permission is hereby granted, free of charge, to any person obtaining a copy
   4  * of this software and associated documentation files (the "Software"), to deal
   5  * in the Software without restriction, including without limitation the rights
   6  * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
   7  * copies of the Software, and to permit persons to whom the Software is
   8  * furnished to do so, subject to the following conditions:
   9  *
  10  * The above copyright notice and this permission notice shall be included in
  11  * all copies or substantial portions of the Software.
  12  *
  13  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  14  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  15  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  16  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  17  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  18  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  19  * THE SOFTWARE.
  20  */
  21
  22 // ----- Includes -----
  23
  24 // Compiler Includes
  25 #include <Lib/MainLib.h>
  26
  27 // Project Includes
  28 #include <macro.h>
  29 #include <scan_loop.h>
  30 #include <usb_com.h>
  31
  32 #include <led.h>
  33 #include <print.h>
  34
  35
  36
  37 // ----- Defines -----
  38
  39 // Verified Keypress Defines
  40 #define USB_TRANSFER_DIVIDER 10 // 1024 == 1 Send of keypresses per second, 1 == 1 Send of keypresses per ~1 millisecond
  41
  42
  43
  44 // ----- Macros -----
  45 #if defined(_at90usb162_) || defined(_atmega32u4_) || defined(_at90usb646_) || defined(_at90usb1286_)
  46 #define CPU_PRESCALE(n) (CLKPR = 0x80, CLKPR = (n))
  47 #endif
  48
  49
  50
  51 // ----- Variables -----
  52
  53 // Timer Interrupt for flagging a send of the sampled key detection data to the USB host
  54 uint16_t sendKeypressCounter = 0;
  55
  56 // Flag generated by the timer interrupt
  57 volatile uint8_t sendKeypresses = 0;
  58
  59
  60
  61 // ----- Functions -----
  62
  63 // Initial Pin Setup, make sure they are sane
  64 inline void pinSetup(void)
  65 {
  66
  67 // AVR
  68 #if defined(_at90usb162_) || defined(_atmega32u4_) || defined(_at90usb646_) || defined(_at90usb1286_)
  69
  70         // For each pin, 0=input, 1=output
  71 #if defined(__AVR_AT90USB1286__)
  72         DDRA = 0x00;
  73 #endif
  74         DDRB = 0x00;
  75         DDRC = 0x00;
  76         DDRD = 0x00;
  77         DDRE = 0x00;
  78         DDRF = 0x00;
  79
  80
  81         // Setting pins to either high or pull-up resistor
  82 #if defined(__AVR_AT90USB1286__)
  83         PORTA = 0x00;
  84 #endif
  85         PORTB = 0x00;
  86         PORTC = 0x00;
  87         PORTD = 0x00;
  88         PORTE = 0x00;
  89         PORTF = 0x00;
  90
  91 // ARM
  92 #elif defined(_mk20dx128_)
  93         // TODO
  94 #endif
  95 }
  96
  97
  98 inline void usbTimerSetup(void)
  99 {
 100 // AVR
 101 #if defined(_at90usb162_) || defined(_atmega32u4_) || defined(_at90usb646_) || defined(_at90usb1286_)
 102
 103         // Setup with 16 MHz clock
 104         CPU_PRESCALE( 0 );
 105
 106         // Setup ISR Timer for flagging a kepress send to USB
 107         // Set to 256 * 1024 (8 bit timer with Clock/1024 prescalar) timer
 108         TCCR0A = 0x00;
 109         TCCR0B = 0x03;
 110         TIMSK0 = (1 << TOIE0);
 111
 112 // ARM
 113 #elif defined(_mk20dx128_)
 114         // TODO
 115 #endif
 116 }
 117
 118
 119 int main(void)
 120 {
 121         // Configuring Pins
 122         pinSetup();
 123         init_errorLED();
 124
 125         // Setup USB Module
 126         usb_setup();
 127
 128         // Setup ISR Timer for flagging a kepress send to USB
 129         usbTimerSetup();
 130
 131         // Main Detection Loop
 132         uint8_t ledTimer = 15; // Enable LED for a short time
 133         while ( 1 )
 134         {
 135                 // Setup the scanning module
 136                 scan_setup();
 137
 138                 while ( 1 )
 139                 {
 140                         // Acquire Key Indices
 141                         // Loop continuously until scan_loop returns 0
 142                         cli();
 143                         while ( scan_loop() );
 144                         sei();
 145
 146                         // Run Macros over Key Indices and convert to USB Keys
 147                         process_macros();
 148
 149                         // Send keypresses over USB if the ISR has signalled that it's time
 150                         if ( !sendKeypresses )
 151                                 continue;
 152
 153                         // Send USB Data
 154                         usb_send();
 155
 156                         // Clear sendKeypresses Flag
 157                         sendKeypresses = 0;
 158
 159                         // Indicate Error, if valid
 160                         errorLED( ledTimer );
 161
 162                         if ( ledTimer > 0 )
 163                                 ledTimer--;
 164                 }
 165
 166                 // Loop should never get here (indicate error)
 167                 ledTimer = 255;
 168
 169                 // HID Debug Error message
 170                 erro_print("Detection loop error, this is very bad...bug report!");
 171         }
 172 }
 173
 174
 175 // ----- Interrupts -----
 176
 177 // AVR - USB Keyboard Data Send Counter Interrupt
 178 #if defined(_at90usb162_) || defined(_atmega32u4_) || defined(_at90usb646_) || defined(_at90usb1286_)
 179
 180 ISR( TIMER0_OVF_vect )
 181 {
 182         sendKeypressCounter++;
 183         if ( sendKeypressCounter > USB_TRANSFER_DIVIDER ) {
 184                 sendKeypressCounter = 0;
 185                 sendKeypresses = 1;
 186         }
 187 }
 188
 189 // ARM - USB Keyboard Data Send Counter Interrupt
 190 #elif defined(_mk20dx128_)
 191         // TODO
 192 #endif
 193