/*---------------------------------------------------------*\
| DRGBController.h                                          |
|                                                           |
|   Driver for DRGBmods                                     |
|                                                           |
|   Zhi Yan                                     25 Jun 2024 |
|                                                           |
|   This file is part of the OpenRGB project                |
|   SPDX-License-Identifier: GPL-2.0-only                   |
\*---------------------------------------------------------*/

#pragma once

#include <chrono>
#include <vector>
#include <hidapi.h>
#include "RGBController.h"

#define DRGB_V4_ONE_PACKAGE_SIZE    316
#define DRGB_V4_PACKAGE_SIZE        340
#define DRGB_V3_PACKAGE_SIZE        21
#define DRGB_V2_PACKAGE_SIZE        20

class DRGBController
{
public:
    DRGBController(hid_device* dev_handle, const char* path,  unsigned short pid);
    std::string     GetLocationString();
    std::string     GetSerialString();
    unsigned short  GetDevicePID();
    void            SetChannelLEDs(unsigned char channel, RGBColor * colors, unsigned int num_colors);
    void            SendPacket(unsigned char* colors,unsigned int buf_packets ,unsigned int LEDtotal);
    void            SendPacketFS(unsigned char* colors,unsigned int buf_packets ,bool Array);
private:
    hid_device*             dev;
    std::string             location;
    unsigned short          device_pid;
};