main_gh60.c

/* This file is part of ukbdc.
 *
 * ukbdc 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.
 *
 * ukbdc 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 ukbdc; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
 */

#include "platforms.h"
#include <avr/io.h>
#include <avr/pgmspace.h>
#include <avr/interrupt.h>
#include <avr/power.h>
#include <util/delay.h>
#include <stdbool.h>

#include "main.h"
#include "io.h"
#include "hid.h"
#include "rawhid.h"
#include "rawhid_protocol.h"
#include "layout.h"
#include "matrix.h"
#include "timer.h"
#include "leds.h"
#include "system.h"

uint8_t matrix[5][14] = {
	{0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13},
	{14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27},
	{28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41},
	{42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55},
	{56, 57, 58, 0, 0, 59, 0, 0, 0, 64, 60, 61, 62, 63},
};

uint8_t rows[] = {0, 1, 2, 3, 4};
uint8_t cols[] = {5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18};

volatile bool should_scan = false;

void on_key_press(uint8_t key, bool event)
{
	if (USB_is_sleeping())
		USB_wakeup();
	else
		LAYOUT_set_key_state(key, event);
}

bool was_sleeping = false;

void main_task()
{
	if (USB_is_sleeping()) {
		if (!was_sleeping) {
			LED_set_indicators(0x00);
			while (!LED_all_stable())
				;
			clock_prescale_set(clock_div_4);
		}
		was_sleeping = true;
	} else {
		if (was_sleeping) {
			clock_prescale_set(clock_div_1);
			LED_set_indicators(HID_get_leds());
		}
		was_sleeping = false;
	}
	if (should_scan) {
		should_scan = false;
		bool changed = MATRIX_scan();
		if (changed)
			HID_commit_state();
		if (HID_leds_changed())
			LED_set_indicators(HID_get_leds());
	}
}

int main(void)
{
	clock_prescale_set(clock_div_1);
	SYSTEM_init();
	TIMER_init();

	USB_init();

	/* initialize with 65 keys */
	LAYOUT_init(65);
	LAYOUT_set((struct layout*)LAYOUT_BEGIN);
	LAYOUT_set_callback(&HID_set_scancode_state);

	MATRIX_init(5, rows, 14, cols, (const uint8_t*)matrix, &on_key_press);

	HID_init();
	HID_commit_state();

	LED_init();

	SYSTEM_subscribe(USB_SOF, ANY, MAIN_handle_sof);
	int sleep_tmr = TIMER_add(32, true);
	SYSTEM_subscribe(TIMER, sleep_tmr, MAIN_sleep_timer_handler);

	SYSTEM_add_task(main_task, 0);
	SYSTEM_add_task(RAWHID_PROTOCOL_task, 0);

	SYSTEM_main_loop();
}

void MAIN_sleep_timer_handler()
{
	if (!USB_is_sleeping())
		return;
	/* make sure to scan the matrix from time to time while asleep */
	should_scan = true;
}

void MAIN_handle_sof(void *data)
{
	static uint16_t num = 0;
	if (num > 20) {
		num = 0;
		should_scan = true;
	} else {
		++num;
	}
}