QMKhost/src/main.cpp

#include <stdio.h>
#include <cstdint>
#include <vector>
#include <stdexcept>
// For reading FIFO
#include <fcntl.h>
#include <unistd.h>
// For sleep
#include <chrono>
#include <thread>

// Local files
#include "utility/bitmap.hpp"
#include "utility/buffer.hpp"
#include "signal_processing/fft.hpp"
#include "ergodox.hpp"
#include "commands.h"


// TODO:
// stereo support (and maybe different bitrates?)
// Optimization: don't copy filename in buffer?
// understand consumption rate
// understand BIN2HZ
// understand values and sizes (DFT_TOTAL, DFT_NONZERO, etc)
//   note that wave and spectrum have different sizes
//
// MPD interface
// beat detection


// How many keys in a column * resolution per key.
// this MUST fit inside a uint8_t.
const uint8_t kb_resolution = 5 * 50;

// How many resolution steps to skip at the top and bottom.
const size_t bottom_skip = 25;
const size_t top_skip = 50;


const size_t width = 10;
const size_t height = bottom_skip + kb_resolution + top_skip;


int main(int argc, char *argv[]) {

	// buffer size for waveform:
	// (44100 / fps * 10), make 10 bigger for slower scrolling
	//
	// Double both buffer sizes if stereo

	// FFT generator
	FFT_Visualizer fft = FFT_Visualizer(
		width, height,
		100, 5000
	);

	// Audio buffer
	Buffer buf = Buffer(
		"/tmp/mpd.fifo",
		44100 / 2, // Keep 500ms of data in buffer
		fft.compute_buffer_output_size()
	);

	// HID interface wrapper
	Ergodox Dox = Ergodox::init(
		0x3297,
		0x4976,
		0x61,
		0xFF60
	);

	// Write buffer
	uint8_t hid_buf[Dox.packet_size];

	wprintf(L"Trying to connect...\n");
	while (!Dox.is_connected()) {
		Dox.try_connect();
		std::this_thread::sleep_for(std::chrono::milliseconds(500));
	}


	while (1) {

		if (Dox.is_connected()) {
			if (Dox.get_animation_mode() == 0x02) {
				buf.update();
				fft.update(buf);

				hid_buf[0] = CMD_ANIM_DATA_fft;
				for (size_t i = 0; i < 10; i++) {
					// Get height from fft, apply bottom_skip
					ssize_t h = fft.get_output()[i] - bottom_skip;

					// Enforce max and min
					// max implicitly enforces top_skip
					h = h>kb_resolution ? kb_resolution : h;
					h = h<0 ? 0 : h;

					hid_buf[i + 1] = h;
				}
			}

			Dox.write(CMD_ANIM_DATA, hid_buf, Dox.packet_size);

			// Dox.write might detect that we've been disconnected,
			// and Dox.read will fail if we are.
			// This check prevents it from doing that, and instead jumps to reconnect.
			if (!Dox.is_connected()) { continue; }

			// Read a packet if there is a packet to read
			if (Dox.read()) {
				uint8_t cmd = Dox.read_buf[0];

				switch(cmd) {
					case CMD_SEND_STATE:
						break;
				}
			}
		} else {
			wprintf(L"Trying to connect...\n");
			while (!Dox.is_connected()) {
				Dox.try_connect();
				std::this_thread::sleep_for(std::chrono::milliseconds(500));
			}
		}
	}

	return 0;
}