/*******************************************************

 Demo Program for HIDAPI

 Alan Ott

 Signal 11 Software

 2010-07-20

 Copyright 2010, All Rights Reserved

 This contents of this file may be used by anyone

 for any reason without any conditions and may be

 used as a starting point for your own applications

 which use HIDAPI.

********************************************************/

#include <fx.h>

#include "hidapi.h"

#include "mac_support.h"

#include <string.h>

#include <stdlib.h>

#include <limits.h>

#ifdef _WIN32

	// Thanks Microsoft, but I know how to use strncpy().

	#pragma warning(disable:4996)

#endif

class MainWindow : public FXMainWindow {

	FXDECLARE(MainWindow)

public:

	enum {

		ID_FIRST = FXMainWindow::ID_LAST,

		ID_CONNECT,

		ID_DISCONNECT,

		ID_RESCAN,

		ID_SEND_OUTPUT_REPORT,

		ID_SEND_FEATURE_REPORT,

		ID_GET_FEATURE_REPORT,

		ID_CLEAR,

		ID_TIMER,

		ID_MAC_TIMER,

		ID_LAST,

	};

private:

	FXList *device_list;

	FXButton *connect_button;

	FXButton *disconnect_button;

	FXButton *rescan_button;

	FXButton *output_button;

	FXLabel *connected_label;

	FXTextField *output_text;

	FXTextField *output_len;

	FXButton *feature_button;

	FXButton *get_feature_button;

	FXTextField *feature_text;

	FXTextField *feature_len;

	FXTextField *get_feature_text;

	FXText *input_text;

	FXFont *title_font;

	struct hid_device_info *devices;

	hid_device *connected_device;

	size_t getDataFromTextField(FXTextField *tf, char *buf, size_t len);

	int getLengthFromTextField(FXTextField *tf);

protected:

	MainWindow() {};

public:

	MainWindow(FXApp *a);

	~MainWindow();

	virtual void create();

	long onConnect(FXObject *sender, FXSelector sel, void *ptr);

	long onDisconnect(FXObject *sender, FXSelector sel, void *ptr);

	long onRescan(FXObject *sender, FXSelector sel, void *ptr);

	long onSendOutputReport(FXObject *sender, FXSelector sel, void *ptr);

	long onSendFeatureReport(FXObject *sender, FXSelector sel, void *ptr);

	long onGetFeatureReport(FXObject *sender, FXSelector sel, void *ptr);

	long onClear(FXObject *sender, FXSelector sel, void *ptr);

	long onTimeout(FXObject *sender, FXSelector sel, void *ptr);

	long onMacTimeout(FXObject *sender, FXSelector sel, void *ptr);

};

// FOX 1.7 changes the timeouts to all be nanoseconds.

// Fox 1.6 had all timeouts as milliseconds.

#if (FOX_MINOR >= 7)

	const int timeout_scalar = 1000*1000;

#else

	const int timeout_scalar = 1;

#endif

FXMainWindow *g_main_window;

FXDEFMAP(MainWindow) MainWindowMap [] = {

	FXMAPFUNC(SEL_COMMAND, MainWindow::ID_CONNECT, MainWindow::onConnect ),

	FXMAPFUNC(SEL_COMMAND, MainWindow::ID_DISCONNECT, MainWindow::onDisconnect ),

	FXMAPFUNC(SEL_COMMAND, MainWindow::ID_RESCAN, MainWindow::onRescan ),

	FXMAPFUNC(SEL_COMMAND, MainWindow::ID_SEND_OUTPUT_REPORT, MainWindow::onSendOutputReport ),

	FXMAPFUNC(SEL_COMMAND, MainWindow::ID_SEND_FEATURE_REPORT, MainWindow::onSendFeatureReport ),

	FXMAPFUNC(SEL_COMMAND, MainWindow::ID_GET_FEATURE_REPORT, MainWindow::onGetFeatureReport ),

	FXMAPFUNC(SEL_COMMAND, MainWindow::ID_CLEAR, MainWindow::onClear ),

	FXMAPFUNC(SEL_TIMEOUT, MainWindow::ID_TIMER, MainWindow::onTimeout ),

	FXMAPFUNC(SEL_TIMEOUT, MainWindow::ID_MAC_TIMER, MainWindow::onMacTimeout ),

};

FXIMPLEMENT(MainWindow, FXMainWindow, MainWindowMap, ARRAYNUMBER(MainWindowMap));

MainWindow::MainWindow(FXApp *app)

	: FXMainWindow(app, "HIDAPI Test Application", NULL, NULL, DECOR_ALL, 200,100, 425,700)

{

	devices = NULL;

	connected_device = NULL;

	FXVerticalFrame *vf = new FXVerticalFrame(this, LAYOUT_FILL_Y|LAYOUT_FILL_X);

	FXLabel *label = new FXLabel(vf, "HIDAPI Test Tool");

	title_font = new FXFont(getApp(), "Arial", 14, FXFont::Bold);

	label->setFont(title_font);

	new FXLabel(vf,

		"Select a device and press Connect.", NULL, JUSTIFY_LEFT);

	new FXLabel(vf,

		"Output data bytes can be entered in the Output section, \n"

		"separated by space, comma or brackets. Data starting with 0x\n"

		"is treated as hex. Data beginning with a 0 is treated as \n"

		"octal. All other data is treated as decimal.", NULL, JUSTIFY_LEFT);

	new FXLabel(vf,

		"Data received from the device appears in the Input section.",

		NULL, JUSTIFY_LEFT);

	new FXLabel(vf,

		"Optionally, a report length may be specified. Extra bytes are\n"

		"padded with zeros. If no length is specified, the length is \n"

		"inferred from the data.",

		NULL, JUSTIFY_LEFT);

	new FXLabel(vf, "");

	// Device List and Connect/Disconnect buttons

	FXHorizontalFrame *hf = new FXHorizontalFrame(vf, LAYOUT_FILL_X);

	//device_list = new FXList(new FXHorizontalFrame(hf,FRAME_SUNKEN|FRAME_THICK, 0,0,0,0, 0,0,0,0), NULL, 0, LISTBOX_NORMAL|LAYOUT_FILL_X|LAYOUT_FILL_Y|LAYOUT_FIX_WIDTH|LAYOUT_FIX_HEIGHT, 0,0,300,200);

	device_list = new FXList(new FXHorizontalFrame(hf,FRAME_SUNKEN|FRAME_THICK|LAYOUT_FILL_X|LAYOUT_FILL_Y, 0,0,0,0, 0,0,0,0), NULL, 0, LISTBOX_NORMAL|LAYOUT_FILL_X|LAYOUT_FILL_Y, 0,0,300,200);

	FXVerticalFrame *buttonVF = new FXVerticalFrame(hf);

	connect_button = new FXButton(buttonVF, "Connect", NULL, this, ID_CONNECT, BUTTON_NORMAL|LAYOUT_FILL_X);

	disconnect_button = new FXButton(buttonVF, "Disconnect", NULL, this, ID_DISCONNECT, BUTTON_NORMAL|LAYOUT_FILL_X);

	disconnect_button->disable();

	rescan_button = new FXButton(buttonVF, "Re-Scan devices", NULL, this, ID_RESCAN, BUTTON_NORMAL|LAYOUT_FILL_X);

	new FXHorizontalFrame(buttonVF, 0, 0,0,0,0, 0,0,50,0);

	connected_label = new FXLabel(vf, "Disconnected");

	new FXHorizontalFrame(vf);

	// Output Group Box

	FXGroupBox *gb = new FXGroupBox(vf, "Output", FRAME_GROOVE|LAYOUT_FILL_X);

	FXMatrix *matrix = new FXMatrix(gb, 3, MATRIX_BY_COLUMNS|LAYOUT_FILL_X);

	new FXLabel(matrix, "Data");

	new FXLabel(matrix, "Length");

	new FXLabel(matrix, "");

	//hf = new FXHorizontalFrame(gb, LAYOUT_FILL_X);

	output_text = new FXTextField(matrix, 30, NULL, 0, TEXTFIELD_NORMAL|LAYOUT_FILL_X|LAYOUT_FILL_COLUMN);

	output_text->setText("1 0x81 0");

	output_len = new FXTextField(matrix, 5, NULL, 0, TEXTFIELD_NORMAL|LAYOUT_FILL_X|LAYOUT_FILL_COLUMN);

	output_button = new FXButton(matrix, "Send Output Report", NULL, this, ID_SEND_OUTPUT_REPORT, BUTTON_NORMAL|LAYOUT_FILL_X);

	output_button->disable();

	//new FXHorizontalFrame(matrix, LAYOUT_FILL_X);

	//hf = new FXHorizontalFrame(gb, LAYOUT_FILL_X);

	feature_text = new FXTextField(matrix, 30, NULL, 0, TEXTFIELD_NORMAL|LAYOUT_FILL_X|LAYOUT_FILL_COLUMN);

	feature_len = new FXTextField(matrix, 5, NULL, 0, TEXTFIELD_NORMAL|LAYOUT_FILL_X|LAYOUT_FILL_COLUMN);

	feature_button = new FXButton(matrix, "Send Feature Report", NULL, this, ID_SEND_FEATURE_REPORT, BUTTON_NORMAL|LAYOUT_FILL_X);

	feature_button->disable();

	get_feature_text = new FXTextField(matrix, 30, NULL, 0, TEXTFIELD_NORMAL|LAYOUT_FILL_X|LAYOUT_FILL_COLUMN);

	new FXWindow(matrix);

	get_feature_button = new FXButton(matrix, "Get Feature Report", NULL, this, ID_GET_FEATURE_REPORT, BUTTON_NORMAL|LAYOUT_FILL_X);

	get_feature_button->disable();

	// Input Group Box

	gb = new FXGroupBox(vf, "Input", FRAME_GROOVE|LAYOUT_FILL_X|LAYOUT_FILL_Y);

	FXVerticalFrame *innerVF = new FXVerticalFrame(gb, LAYOUT_FILL_X|LAYOUT_FILL_Y);

	input_text = new FXText(new FXHorizontalFrame(innerVF,LAYOUT_FILL_X|LAYOUT_FILL_Y|FRAME_SUNKEN|FRAME_THICK, 0,0,0,0, 0,0,0,0), NULL, 0, LAYOUT_FILL_X|LAYOUT_FILL_Y);

	input_text->setEditable(false);

	new FXButton(innerVF, "Clear", NULL, this, ID_CLEAR, BUTTON_NORMAL|LAYOUT_RIGHT);

}

MainWindow::~MainWindow()

{

	if (connected_device)

		hid_close(connected_device);

	hid_exit();

	delete title_font;

}

void

MainWindow::create()

{

	FXMainWindow::create();

	show();

	onRescan(NULL, 0, NULL);

#ifdef __APPLE__

	init_apple_message_system();

#endif

	getApp()->addTimeout(this, ID_MAC_TIMER,

		50 * timeout_scalar /*50ms*/);

}

long

MainWindow::onConnect(FXObject *sender, FXSelector sel, void *ptr)

{

	if (connected_device != NULL)

		return 1;

	FXint cur_item = device_list->getCurrentItem();

	if (cur_item < 0)

		return -1;

	FXListItem *item = device_list->getItem(cur_item);

	if (!item)

		return -1;

	struct hid_device_info *device_info = (struct hid_device_info*) item->getData();

	if (!device_info)

		return -1;

	connected_device =  hid_open_path(device_info->path);

	if (!connected_device) {

		FXMessageBox::error(this, MBOX_OK, "Device Error", "Unable To Connect to Device");

		return -1;

	}

	hid_set_nonblocking(connected_device, 1);

	getApp()->addTimeout(this, ID_TIMER,

		5 * timeout_scalar /*5ms*/);

	FXString s;

	s.format("Connected to: %04hx:%04hx -", device_info->vendor_id, device_info->product_id);

	s += FXString(" ") + device_info->manufacturer_string;

	s += FXString(" ") + device_info->product_string;

	connected_label->setText(s);

	output_button->enable();

	feature_button->enable();

	get_feature_button->enable();

	connect_button->disable();

	disconnect_button->enable();

	input_text->setText("");

	return 1;

}

long

MainWindow::onDisconnect(FXObject *sender, FXSelector sel, void *ptr)

{

	hid_close(connected_device);

	connected_device = NULL;

	connected_label->setText("Disconnected");

	output_button->disable();

	feature_button->disable();

	get_feature_button->disable();

	connect_button->enable();

	disconnect_button->disable();

	getApp()->removeTimeout(this, ID_TIMER);

	return 1;

}

long

MainWindow::onRescan(FXObject *sender, FXSelector sel, void *ptr)

{

	struct hid_device_info *cur_dev;

	device_list->clearItems();

	// List the Devices

	hid_free_enumeration(devices);

	devices = hid_enumerate(0x0, 0x0);

	cur_dev = devices;	

	while (cur_dev) {

		// Add it to the List Box.

		FXString s;

		FXString usage_str;

		s.format("%04hx:%04hx -", cur_dev->vendor_id, cur_dev->product_id);

		s += FXString(" ") + cur_dev->manufacturer_string;

		s += FXString(" ") + cur_dev->product_string;

		usage_str.format(" (usage: %04hx:%04hx) ", cur_dev->usage_page, cur_dev->usage);

		s += usage_str;

		FXListItem *li = new FXListItem(s, NULL, cur_dev);

		device_list->appendItem(li);

		cur_dev = cur_dev->next;

	}

	if (device_list->getNumItems() == 0)

		device_list->appendItem("*** No Devices Connected ***");

	else {

		device_list->selectItem(0);

	}

	return 1;

}

size_t

MainWindow::getDataFromTextField(FXTextField *tf, char *buf, size_t len)

{

	const char *delim = " ,{}\t\r\n";

	FXString data = tf->getText();

	const FXchar *d = data.text();

	size_t i = 0;

	// Copy the string from the GUI.

	size_t sz = strlen(d);

	char *str = (char*) malloc(sz+1);

	strcpy(str, d);

	// For each token in the string, parse and store in buf[].

	char *token = strtok(str, delim);

	while (token) {

		char *endptr;

		long int val = strtol(token, &endptr, 0);

		buf[i++] = val;

		token = strtok(NULL, delim);

	}

	free(str);

	return i;

}

/* getLengthFromTextField()

   Returns length:

	 0: empty text field

	>0: valid length

	-1: invalid length */

int

MainWindow::getLengthFromTextField(FXTextField *tf)

{

	long int len;

	FXString str = tf->getText();

	size_t sz = str.length();

	if (sz > 0) {

		char *endptr;

		len = strtol(str.text(), &endptr, 0);

		if (endptr != str.text() && *endptr == '\0') {

			if (len <= 0) {

				FXMessageBox::error(this, MBOX_OK, "Invalid length", "Enter a length greater than zero.");

				return -1;

			}

			return len;

		}

		else

			return -1;

	}

	return 0;

}

long

MainWindow::onSendOutputReport(FXObject *sender, FXSelector sel, void *ptr)

{

	char buf[256];

	size_t data_len, len;

	int textfield_len;

	memset(buf, 0x0, sizeof(buf));

	textfield_len = getLengthFromTextField(output_len);

	data_len = getDataFromTextField(output_text, buf, sizeof(buf));

	if (textfield_len < 0) {

		FXMessageBox::error(this, MBOX_OK, "Invalid length", "Length field is invalid. Please enter a number in hex, octal, or decimal.");

		return 1;

	}

	if (textfield_len > sizeof(buf)) {

		FXMessageBox::error(this, MBOX_OK, "Invalid length", "Length field is too long.");

		return 1;

	}

	len = (textfield_len)? textfield_len: data_len;

	int res = hid_write(connected_device, (const unsigned char*)buf, len);

	if (res < 0) {

		FXMessageBox::error(this, MBOX_OK, "Error Writing", "Could not write to device. Error reported was: %ls", hid_error(connected_device));

	}

	return 1;

}

long

MainWindow::onSendFeatureReport(FXObject *sender, FXSelector sel, void *ptr)

{

	char buf[256];

	size_t data_len, len;

	int textfield_len;

	memset(buf, 0x0, sizeof(buf));

	textfield_len = getLengthFromTextField(feature_len);

	data_len = getDataFromTextField(feature_text, buf, sizeof(buf));

	if (textfield_len < 0) {

		FXMessageBox::error(this, MBOX_OK, "Invalid length", "Length field is invalid. Please enter a number in hex, octal, or decimal.");

		return 1;

	}

	if (textfield_len > sizeof(buf)) {

		FXMessageBox::error(this, MBOX_OK, "Invalid length", "Length field is too long.");

		return 1;

	}

	len = (textfield_len)? textfield_len: data_len;

	int res = hid_send_feature_report(connected_device, (const unsigned char*)buf, len); 

	if (res < 0) {

		FXMessageBox::error(this, MBOX_OK, "Error Writing", "Could not send feature report to device. Error reported was: %ls", hid_error(connected_device));

	}

	return 1;

}

long

MainWindow::onGetFeatureReport(FXObject *sender, FXSelector sel, void *ptr)

{

	char buf[256];

	size_t len;

	memset(buf, 0x0, sizeof(buf));

	len = getDataFromTextField(get_feature_text, buf, sizeof(buf));

	if (len != 1) {

		FXMessageBox::error(this, MBOX_OK, "Too many numbers", "Enter only a single report number in the text field");

	}

	int res = hid_get_feature_report(connected_device, (unsigned char*)buf, sizeof(buf));

	if (res < 0) {

		FXMessageBox::error(this, MBOX_OK, "Error Getting Report", "Could not get feature report from device. Error reported was: %ls", hid_error(connected_device));

	}

	if (res > 0) {

		FXString s;

		s.format("Returned Feature Report. %d bytes:\n", res);

		for (int i = 0; i < res; i++) {

			FXString t;

			t.format("%02hhx ", buf[i]);

			s += t;

			if ((i+1) % 4 == 0)

				s += " ";

			if ((i+1) % 16 == 0)

				s += "\n";

		}

		s += "\n";

		input_text->appendText(s);

		input_text->setBottomLine(INT_MAX);

	}

	return 1;

}

long

MainWindow::onClear(FXObject *sender, FXSelector sel, void *ptr)

{

	input_text->setText("");

	return 1;

}

long

MainWindow::onTimeout(FXObject *sender, FXSelector sel, void *ptr)

{

	unsigned char buf[256];

	int res = hid_read(connected_device, buf, sizeof(buf));

	if (res > 0) {

		FXString s;

		s.format("Received %d bytes:\n", res);

		for (int i = 0; i < res; i++) {

			FXString t;

			t.format("%02hhx ", buf[i]);

			s += t;

			if ((i+1) % 4 == 0)

				s += " ";

			if ((i+1) % 16 == 0)

				s += "\n";

		}

		s += "\n";

		input_text->appendText(s);

		input_text->setBottomLine(INT_MAX);

	}

	if (res < 0) {

		input_text->appendText("hid_read() returned error\n");

		input_text->setBottomLine(INT_MAX);

	}

	getApp()->addTimeout(this, ID_TIMER,

		5 * timeout_scalar /*5ms*/);

	return 1;

}

long

MainWindow::onMacTimeout(FXObject *sender, FXSelector sel, void *ptr)

{

#ifdef __APPLE__

	check_apple_events();

	getApp()->addTimeout(this, ID_MAC_TIMER,

		50 * timeout_scalar /*50ms*/);

#endif

	return 1;

}

int main(int argc, char **argv)

{

	FXApp app("HIDAPI Test Application", "Signal 11 Software");

	app.init(argc, argv);

	g_main_window = new MainWindow(&app);

	app.create();

	app.run();

	return 0;

}