#include <stdio.h>
#include <stdlib.h>
#include <stdbool.h>
#include <libusb-1.0/libusb.h>
#include "printx.h"
#include "usb.h"

#define VENDOR_ID			0xfccf
#define PRODUCT_ID			0xa001
#define ENDPOINT_OUT		0x01
#define	ENDPOINT_IN			0x82
#define	MAX_SIZE_OUT		64
#define HEIGHT				240
#define WIDTH				320
#define	STATUS_QUERY_LENGTH	11

libusb_device_handle*	screenHandle;
unsigned char			data[MAX_SIZE_OUT];
unsigned short			dataPointer = 0;

void blackScreen(void)
{
	unsigned char data[] = {0x81, 0x00, 0x00};
	int transfered;
	libusb_bulk_transfer(screenHandle, ENDPOINT_OUT, data, 3, &transfered, 0);
}

void fillDataWithHeader(void)
{
	data[dataPointer] = 0x82; // HEADER
	dataPointer++;
	data[dataPointer] = 0x00; // XPOS
	dataPointer++;
	data[dataPointer] = 0x00; // XPOS
	dataPointer++;
	data[dataPointer] = 0x00; // YPOS
	dataPointer++;
	data[dataPointer] = 0x00; // YPOS
	dataPointer++;
	data[dataPointer] = 0x40; // Width 0x40
	dataPointer++;
	data[dataPointer] = 0x01; // Width 0x01
	dataPointer++;
	data[dataPointer] = 0xF0; // Height 
	dataPointer++;
	data[dataPointer] = 0x00; // Height
	dataPointer++;
	data[dataPointer] = 0x00; // Copy
	dataPointer++;
}

void fillDataWithSubHeader(void)
{
	data[dataPointer] = 0x02; // HEADER
	dataPointer++;
}

void sendUSBData(int i)
{
	//printf("Sending data %d\n", i);
	int transfered;
	libusb_bulk_transfer(screenHandle, ENDPOINT_OUT, data, dataPointer, &transfered, 0);
	dataPointer = 0;
	fillDataWithSubHeader();
}

void displayPicture(char* filename)
{
	FILE* file = NULL;
	file = fopen(filename, "r");
	dataPointer = 0;

	if(file == NULL)
	{
		printf("Unable to open the picture\n");
		return;
	}
	else
		printf("Picture successfully opened\n");

	fillDataWithHeader();
	for(int i=0; i<HEIGHT*WIDTH*2; i++)
	{
		if(feof(file))
		{
			printf("File corrupted\n");
			return;
		}
		data[dataPointer] = getc(file);
		dataPointer++;
		if(dataPointer == MAX_SIZE_OUT)
			sendUSBData(i);
	}

	if(dataPointer != 0)
		sendUSBData(-1);

	printf("Screen updated\n");
}

bool initUSB(void)
{
	int				ret;

	ret = libusb_init(NULL);
	if(ret != 0)
	{
		printf("Error while initializing libusb, return : %d\n", ret);
		return false;
	}

	struct			libusb_config_descriptor* dConfig = NULL;
	libusb_device**	list = NULL;
	ssize_t			cnt = libusb_get_device_list(NULL, &list);

	struct libusb_device_descriptor dDevice;


	printf("Starting lsusb things\n");

	if(cnt < 0)
	{
		printf("Unable to get USB device list\n");
		return false;
	}

	printf("%d devices detected\n", (int)cnt);
	printf("List of compatible devices detected\n");

	for (int i = 0; i < cnt; i++)
	{
		libusb_device *device = list[i];
		ret = libusb_get_device_descriptor(device, &dDevice);
		if(VENDOR_ID == dDevice.idVendor && PRODUCT_ID == dDevice.idProduct)
		{
			printf("Bus %03d Device %03d: ID %04x:%04x\n",
					libusb_get_bus_number(device),
					libusb_get_device_address(device), dDevice.idVendor,
					dDevice.idProduct);
			ret = libusb_open(device, &screenHandle);
			if(ret != 0)
			{
				printf("Unable to open this device, error %d\n", ret);
				return false;
			}
		}
	}

	libusb_free_device_list(list, 1);

	ret = libusb_get_config_descriptor(libusb_get_device(screenHandle), 0, &dConfig);
	if(ret!=0)
	{
		printf("Descriptor for this device unavailable\n");
		return false;
	}
	else
	{
		for(int j=0; j<dConfig->bNumInterfaces; j++)
		{
			if(libusb_kernel_driver_active(screenHandle, j) && (libusb_detach_kernel_driver(screenHandle, j) != 0))
			{
				printf("Unable to detach this device\n");
				return false;
			}
		}
		ret = libusb_set_configuration(screenHandle, dConfig->bConfigurationValue);
		if(ret != 0)
		{
			printf("Configuration unavailable, error %d\n", ret);
			return false;
		}
		for(int j=0; j<dConfig->bNumInterfaces; j++)
			if(libusb_claim_interface(screenHandle, j) != 0)
			{
				printf("Device not claimed\n");
				return false;
			}
			else
				printf("Interface %d ready\n", j);
	}
	
	libusb_free_config_descriptor(dConfig);

	displayPicture("img/home.rgb");

	return true;
}

bool screenTouched(int* x, int* y)
{
	unsigned char	data[STATUS_QUERY_LENGTH];
	int				transfered;
	libusb_interrupt_transfer(screenHandle, ENDPOINT_IN, data, STATUS_QUERY_LENGTH, &transfered, 0);
	if(transfered != STATUS_QUERY_LENGTH)
	{
		printx(ERROR, UI, "Status recieved is %d bytes long, and expected %d\n", transfered, STATUS_QUERY_LENGTH);
		return false;
	}

	*x = data[3] + (256 * (int)data[4]);
	*y = data[7];

	printx(DEBUG, UI, "Screen %s at X %d Y %d\n", data[2] == 1 ? "touched" : "untouched", *x, *y);

	return data[2] == 1;
}