/*---------------------------------------------------------*\
|  Processing Code for NZXT Hue+                            |
|                                                           |
|  Adam Honse (calcprogrammer1@gmail.com), 12/11/2016       |
\*---------------------------------------------------------*/

#include "HuePlusController.h"

#include <fstream>
#include <iostream>
#include <string>


#ifdef WIN32
#include <Windows.h>
#else
#include <unistd.h>

static void Sleep(unsigned int milliseconds)
{
    usleep(1000 * milliseconds);
}
#endif

HuePlusController::HuePlusController()
{

}

HuePlusController::~HuePlusController()
{
}

void HuePlusController::Initialize(char* port_name)
{
    strcpy(led_string, port_name);
    
    serialport = new serial_port(port_name, HUE_PLUS_BAUD);

    channel_leds[HUE_PLUS_CHANNEL_1_IDX] = GetStripsOnChannel(HUE_PLUS_CHANNEL_1) * 10;
    channel_leds[HUE_PLUS_CHANNEL_2_IDX] = GetStripsOnChannel(HUE_PLUS_CHANNEL_2) * 10;
}

char* HuePlusController::GetLEDString()
{
    return(led_string);
}

unsigned int HuePlusController::GetStripsOnChannel(unsigned int channel)
{
    unsigned int ret_val = 0;

    if (serialport != NULL)
    {
        unsigned char *serial_buf;
        serial_buf = new unsigned char[5];

        serial_buf[0] = 0x8D;
        serial_buf[1] = channel;

        serialport->serial_flush_rx();
        serialport->serial_write((char *)serial_buf, 2);
        serialport->serial_flush_tx();

        Sleep(50);

        int bytes_read = serialport->serial_read((char *)serial_buf, 5);

        if(bytes_read == 5)
        {
            ret_val = serial_buf[4];
        }

        delete[] serial_buf;
    }

    return(ret_val);
}

void HuePlusController::SetChannelLEDs(unsigned int channel, std::vector<RGBColor> colors)
{
    if (serialport != NULL)
    {
        unsigned char *serial_buf;

        serial_buf = new unsigned char[HUE_PLUS_PACKET_SIZE];

        serial_buf[0] = 0x4B;
        serial_buf[1] = channel;
        serial_buf[2] = HUE_PLUS_MODE_FIXED;				
        serial_buf[3] = 0x00; 
        serial_buf[4] = 0x00;

        for (int i = 5; i < HUE_PLUS_PACKET_SIZE; i++)
        {
            serial_buf[i] = 0x00;
        }

        for (unsigned int idx = 0; idx < (colors.size() * 3); idx += 3)
        {
            int pixel_idx = idx / 3;
            RGBColor color = colors[pixel_idx];
            serial_buf[idx + 5] = RGBGetGValue(color);
            serial_buf[idx + 6] = RGBGetRValue(color);
            serial_buf[idx + 7] = RGBGetBValue(color);
        }

        serialport->serial_write((char *)serial_buf, HUE_PLUS_PACKET_SIZE);
        serialport->serial_flush_tx();

        delete[] serial_buf;
    }

}