avr-something/hidModule.cpp at master · M3Henry/avr-something · GitHub

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
59
60
61
62
#include "hidModule.h"
#include "ctrl.h"
#include <util/delay.h>

void hid::init()
{
	io::direction::B() = ledRED;

	io::out::D() = wheelCOM | buttonCOM;
	io::direction::D() = wheelCOM | buttonCOM | ledYEL | ledGRN;
}

uint8_t hid::sampleInput()
{
	constexpr uint8_t maskC = bits(2,3,4,5);
	constexpr uint8_t maskD = wheelCOM | buttonCOM;

	const auto saveCdir = io::direction::C();
	io::direction::C() = saveCdir & ~maskC;

	const auto saveCout = io::out::C();
	io::out::C() = saveCout | maskC;
	io::out::D() = mix(io::out::D(), wheelCOM, maskD);

	_delay_us(1);
	auto buttons = (maskC & ~io::pin::C()) << 1;
	io::pin::D() = maskD;

	_delay_us(1);
	buttons |= (maskC & ~io::pin::C()) >> 2;

	io::out::C() = saveCout;
	io::direction::C() = saveCdir;

	return buttons;
}

void hid::setLeds(const uint8_t val)
{
	io::out::B() = mix(io::out::B(), val, ledRED);
	io::out::D() = mix(io::out::D(), val, ledYEL | ledGRN);
}

void hid::test()
{
	ctrl::disableJTAG();

	init();

	for (;;)
	{
		auto val = sampleInput();

		const auto r = (val & (buttonL | wheelA)) ? ledRED : 0;
		const auto y = (val & (buttonD | buttonU)) ? ledYEL : 0;
		const auto g = (val & (buttonR | wheelB)) ? ledGRN : 0;

		const auto leds = r | y | g;

		setLeds((val & buttonC) ? ~leds : leds);
	}
}