#include <iostream>
#include <iomanip>

#include <fcntl.h>
#include <unistd.h>
#include <sys/ioctl.h>
#include <linux/hidraw.h>
#include <cstdlib>
#include <cstdint>

using std::cout, std::cerr, std::endl, std::hex, std::setw, std::setfill, std::ios;

struct __attribute__ ((packed)) lamp_array_attributes_report {
    const uint8_t report_id = 1;
    uint16_t lamp_count;
    uint32_t bounding_box_width_in_micrometers;
    uint32_t bounding_box_height_in_micrometers;
    uint32_t bounding_box_depth_in_micrometers;
    uint32_t lamp_array_kind;
    uint32_t min_update_interval_in_microseconds;
};

struct __attribute__ ((packed)) lamp_attributes_request_report {
    const uint8_t report_id = 2;
    uint16_t lamp_id;
};

struct __attribute__ ((packed)) lamp_attributes_response_report {
    const uint8_t report_id = 3;
    uint16_t lamp_id;
    uint32_t position_x_in_micrometers;
    uint32_t position_y_in_micrometers;
    uint32_t position_z_in_micrometers;
    uint32_t update_latency_in_microseconds;
    uint32_t lamp_purpose;
    uint8_t red_level_count;
    uint8_t green_level_count;
    uint8_t blue_level_count;
    uint8_t intensity_level_count;
    uint8_t is_programmable;
    uint8_t input_binding;
};

struct __attribute__ ((packed)) lamp_multi_update_report {
    const uint8_t report_id = 4;
    uint8_t lamp_count;
    uint8_t lamp_update_flags;
    uint16_t lamp_id[8];
    struct {
        uint8_t red_update_channel;
        uint8_t green_update_channel;
        uint8_t blue_update_channel;
        uint8_t intensity_update_channel;
    } update_channels[8];
};

struct __attribute__ ((packed)) lamp_range_update_report {
    const uint8_t report_id = 5;
    uint8_t lamp_update_flags;
    uint16_t lamp_id_start;
    uint16_t lamp_id_end;
    uint8_t red_update_channel;
    uint8_t green_update_channel;
    uint8_t blue_update_channel;
    uint8_t intensity_update_channel;
};

struct __attribute__ ((packed)) lamp_array_control_report {
    const uint8_t report_id = 6;
    uint8_t autonomous_mode;
};

int main(int argc, char *argv[])
{
    if (argc < 2) {
        cout << "Usage: " << argv[0] << "<hidraw_path>" << endl;
        return EXIT_SUCCESS;
    }

    int hidraw = open(argv[1], O_WRONLY|O_NONBLOCK);
    if (hidraw < 0) {
        cerr << "main: open(\"" << argv[1] << "\", O_WRONLY|O_NONBLOCK) failed." << endl;
        return hidraw;
    }

    int result = 0;

    struct lamp_range_update_report lamp_range_update_report;
    lamp_range_update_report.intensity_update_channel = 0xff;
    lamp_range_update_report.lamp_update_flags = 0x01;
    lamp_range_update_report.lamp_id_start = 0;
    lamp_range_update_report.lamp_id_end = 101;

    while (true) {
        lamp_range_update_report.red_update_channel = 0xff;
        lamp_range_update_report.green_update_channel = 0;
        lamp_range_update_report.blue_update_channel = 0;
        result = ioctl(hidraw, HIDIOCSFEATURE(sizeof(struct lamp_range_update_report)), &lamp_range_update_report);
        if (result < 0) {
            cerr << "main: ioctl(hidraw, HIDIOCSFEATURE(sizeof(struct lamp_range_update_report)), lamp_range_update_report) failed." << endl;
            close(hidraw);
            return result;
        }

        lamp_range_update_report.red_update_channel = 0;
        lamp_range_update_report.green_update_channel = 0xff;
        lamp_range_update_report.blue_update_channel = 0;
        result = ioctl(hidraw, HIDIOCSFEATURE(sizeof(struct lamp_range_update_report)), &lamp_range_update_report);
        if (result < 0) {
            cerr << "main: ioctl(hidraw, HIDIOCSFEATURE(sizeof(struct lamp_range_update_report)), lamp_range_update_report) failed." << endl;
            close(hidraw);
            return result;
        }

        lamp_range_update_report.red_update_channel = 0;
        lamp_range_update_report.green_update_channel = 0;
        lamp_range_update_report.blue_update_channel = 0xff;
        result = ioctl(hidraw, HIDIOCSFEATURE(sizeof(struct lamp_range_update_report)), &lamp_range_update_report);
        if (result < 0) {
            cerr << "main: ioctl(hidraw, HIDIOCSFEATURE(sizeof(struct lamp_range_update_report)), lamp_range_update_report) failed." << endl;
            close(hidraw);
            return result;
        }
    }

    close(hidraw);

    return EXIT_SUCCESS;
}