Subversion Repositories group.electronics

#include <avr/io.h>

#include <avr/wdt.h>

#include <avr/interrupt.h>

#include <avr/pgmspace.h>

#include <util/delay.h>

#include "usbdrv.h"

#include "config.h"

#ifndef NULL

#define NULL    ((void *)0)

#endif

#define USB_GET_STATE  100

#define USB_LEDS_SET  101

/* ------------------------------------------------------------------------- */

uint8_t getKey(void);

void doButtons(uint8_t);

inline void setLeds(uint8_t);

struct {

	int16_t axis[2];

	uint8_t buttons;

} reportBuffer;

volatile struct {

	uint8_t buttons;

	uint8_t waitup;

	uint8_t timer;

} debounce;

volatile struct {

	uint8_t data;

	union {

		uint8_t led_method:1;

		uint8_t other:1;

		uint8_t reserved:6;

	};

} config;

static uchar usbReplyBuf[16];

volatile uint8_t currLeds = 0;

volatile uint8_t currKeys = 0;

volatile uint8_t tmr0_ovf = 0;

volatile uint32_t systime = 0;

/* ------------------------------------------------------------------------- */

void hadUsbReset(void) {

}

uchar	usbFunctionSetup(uchar data[8])

{

    usbRequest_t    *rq = (void *)data;

        switch (rq->bRequest ) {

		case USB_GET_STATE:

			usbReplyBuf[0] = currLeds;

			usbReplyBuf[1] = currKeys;

			usbMsgPtr = usbReplyBuf;

			return sizeof(usbReplyBuf);

		case USB_LEDS_SET:

			currLeds = rq->wValue.bytes[0];

			return 0;

        }

	return 0;

}

int main(void) {

	DIDR0 = 0x00;

  /*

  DDR : 1 = Output, 0 = Input

  PORT: 1 = Pullup for Input, otherwise set output

  PIN : Read input pin

  */

  /*

        PB0     - Output                - LED 3

        PB1     - Output                - LED 4

        PB2     - Output                - LED 5

        PB3     - Output                - LED 6

        PB4     - Output                - LED 7

        PB5     - 

        PB6     - 

        PB7     - 

  */

  DDRB          = 0B00011111;

  PORTB         = 0B00000000;

  /*

        PC0     - Output		- ButtonPad Gnd0

        PC1     - Output		- ButtonPad Gnd1

        PC2     - Input, Pullup		- ButtonPad 0

        PC3     - Input, Pullup		- ButtonPad 1

        PC4     - Input, Pullup		- ButtonPad 2

        PC5     - Input, Pullup		- ButtonPad 3

  */

  DDRC          = 0B00000011;

  PORTC         = 0B00111111;

  /*

        PD0     - 

        PD1     - 

        PD2     - 

        PD3     - 

        PD4     - 

        PD5     - Output		- LED 0

        PD6     - Output		- LED 1

        PD7     - Output		- LED 2

  */

  DDRD          = 0B11100000;

  PORTD         = 0B00000000;

	setLeds(0xff);

    usbDeviceDisconnect();

    _delay_ms(500);

    usbDeviceConnect();

    TIMSK0 = (1<<TOIE0);                    // Enable timer overflow

    TCNT0 = 0x00;                           // Set Timer0 initial value to 0

    TCCR0B = (1<< CS00) ;                   // /1 prescaler

    wdt_enable(WDTO_1S);

    usbInit();

    sei();

	config.led_method = 1;

	setLeds(currLeds);

    for(;;){

        wdt_reset();

        usbPoll();

	currKeys = getKey();

	if (config.led_method != rbi(PORTB, 5)) {

		// Reset the leds when switcing	to buttons

		if (config.led_method == 0) currLeds = 1;

		config.led_method = rbi(PORTB, 5);

	}

	if (config.led_method == 1) {

		uint8_t pressed = getKey();

		doButtons(pressed);

	}

	setLeds(currLeds);

    }

    return 0;

}

inline void setLeds(uint8_t leds) {

	PORTB &= 0xE0;

        PORTB |= (0x1F & (leds >> 3));

	PORTD &= 0x1F;

        PORTD |= (0xE0 & (leds << 5));

}

void doButtons(uint8_t pressed) {

        // Deboucing

        // When i key first goes down, wait 5 ms, check it again to see if its still down

        if (pressed && debounce.buttons == 0 && debounce.timer == 0) {

                debounce.buttons = pressed;

                debounce.timer = 5;

        }

        // The key has come up

        if (pressed != debounce.buttons) {

                debounce.buttons = 0;

                debounce.timer = 0;

                debounce.waitup = 0;

        }

        // Debounce timer is up, process our button

        if (debounce.buttons && debounce.timer == 0 && debounce.waitup != 1) {

                uint8_t i = 0;

                for (i=0; i<=7; i++) {

                        // Button pressed and the led is currently on

                        if ( rbi(debounce.buttons, i) == 1 && rbi(currLeds, i) == 1 ) {

                                if ( i == 0 && rbi(currLeds, 1) != 1)  //Dont turn off com1 if no comm2

                                        break;

                                if ( i == 1 && rbi(currLeds, 0) != 1)  //Dont turn off com2 if no comm1

                                        break;

                                cbi(currLeds, i);

                        // Button is pressed and led is currently off

                         } else if ( rbi(debounce.buttons, i) == 1 && rbi(currLeds, i) == 0 ) {

                                if ( i == 0 && rbi(currLeds, 1) == 1)  //Turn on comm2, turn off comm1

                                        cbi(currLeds,1);

                                if ( i == 1 && rbi(currLeds, 0) == 1)  //Turn on comm1, turn off comm2

                                        cbi(currLeds,0);

                                sbi(currLeds, i);

                        }

                }

                debounce.waitup = 1;

        }

}

// Gnd = PC0, PC1

// Btn = PC2, PC3, PC4, PC5

uint8_t getKey() {

        uint8_t key = 0;

        cbi(PORTC, 1);

        _delay_us(10);        // Wait for the port change

        if (rbi(PINC, 2) == 0) key = 1;

        if (rbi(PINC, 3) == 0) key = 2;

        if (rbi(PINC, 4) == 0) key = 4;

        if (rbi(PINC, 5) == 0) key = 8;

        sbi(PORTC, 1);

        cbi(PORTC, 0);

        _delay_us(10);

        if (rbi(PINC, 2) == 0) key = 16;

        if (rbi(PINC, 3) == 0) key = 32;

        if (rbi(PINC, 4) == 0) key = 64;

        if (rbi(PINC, 5) == 0) key = 128;

        sbi(PORTC, 0);

        return key;

}

ISR(TIMER0_OVF_vect) {

        tmr0_ovf++;

	// Clk/1 TCCR0B = (1<< CS00);

	//20.0Mhz, 1ms = 78ovf

	//16.5Mhz, 1ms = 64ovf

	//12.0Mhz, 1ms = 46ovf

        if (tmr0_ovf>=78) {

                systime++;

                tmr0_ovf = 0;

		if (debounce.timer != 0)

			debounce.timer--;

        }

}