Files
OpenRGB/Controllers/LianLiController/LianLiUniHubSLV2Controller.cpp
2023-01-25 19:05:09 +00:00

321 lines
11 KiB
C++

/*-----------------------------------------*\
| LianLiUniHubSLV2Controller.cpp |
| |
| Driver for Lian Li Uni Hub SLV2 USB |
| lighting controller |
| |
| Will Kennedy 01/17/2023 |
| Oliver P 04/26/2022 |
| Credit to Luca Lovisa for original work. |
\*-----------------------------------------*/
#include "LianLiUniHubSLV2Controller.h"
#include <string.h>
using namespace std::chrono_literals;
LianLiUniHubSLV2Controller::LianLiUniHubSLV2Controller(hid_device* dev_handle, const char* path, std::string dev_name)
{
dev = dev_handle;
location = path;
name = dev_name;
}
LianLiUniHubSLV2Controller::~LianLiUniHubSLV2Controller()
{
hid_close(dev);
}
std::string LianLiUniHubSLV2Controller::GetDeviceLocation()
{
return("HID: " + location);
}
std::string LianLiUniHubSLV2Controller::GetFirmwareVersionString()
{
wchar_t product_string[40];
int ret = hid_get_product_string(dev, product_string, 40);
if (ret != 0)
{
return ("");
}
std::wstring return_wstring = product_string;
std::string return_string(return_wstring.begin(),return_wstring.end());
return(return_string.substr(return_string.find_last_of("-")+1,4).c_str());
}
std::string LianLiUniHubSLV2Controller::GetName()
{
return(name);
}
std::string LianLiUniHubSLV2Controller::GetSerialString()
{
wchar_t serial_string[20];
int ret = hid_get_serial_number_string(dev, serial_string, 20);
if (ret != 0)
{
return ("");
}
std::wstring return_wstring = serial_string;
std::string return_string(return_wstring.begin(), return_wstring.end());
return(return_string);
}
float brightnessLimit(RGBColor color)
{
/*---------------------------------------------------------*\
| Limiter to protect LEDs |
\*---------------------------------------------------------*/
if(UNIHUB_SLV2_LED_LIMITER && (RGBGetRValue(color) + RGBGetBValue(color) + RGBGetGValue(color) > 460))
{
return 460.f / (RGBGetRValue(color) + RGBGetBValue(color) + RGBGetGValue(color));
}
return 1;
}
void LianLiUniHubSLV2Controller::SetChannelLEDs(unsigned char channel, RGBColor * colors, unsigned int num_colors, float brightness)
{
unsigned char led_data[16 * 6 * 3];
int fan_idx = 0;
int mod_led_idx;
int cur_led_idx;
if(num_colors == 0)
{
return; // Do nothing, channel isn't in use
}
for(unsigned int led_idx = 0; led_idx < num_colors; led_idx++)
{
mod_led_idx = (led_idx % 16);
if((mod_led_idx == 0) && (led_idx != 0))
{
fan_idx++;
}
float brightness_scale = brightness * brightnessLimit(colors[led_idx]);
//Determine current position of led_data array from colors array
cur_led_idx = ((mod_led_idx + (fan_idx * 16)) * 3);
led_data[cur_led_idx + 0] = RGBGetRValue(colors[led_idx]) * brightness_scale;
led_data[cur_led_idx + 1] = RGBGetBValue(colors[led_idx]) * brightness_scale;
led_data[cur_led_idx + 2] = RGBGetGValue(colors[led_idx]) * brightness_scale;
}
/*---------------------------------------------------------*\
| Send fan LED data |
\*---------------------------------------------------------*/
SendStartAction
(
channel, // Current channel
(fan_idx + 1) // Number of fans
);
SendColorData
(
channel, // Channel
(fan_idx + 1)*16,
led_data
);
SendCommitAction
(
channel, // Channel
UNIHUB_SLV2_LED_MODE_STATIC_COLOR, // Effect
UNIHUB_SLV2_LED_SPEED_000, // Speed
UNIHUB_SLV2_LED_DIRECTION_LTR, // Direction
UNIHUB_SLV2_LED_BRIGHTNESS_100 // Brightness
);
}
void LianLiUniHubSLV2Controller::SetChannelMode(unsigned char channel, const mode active_mode, unsigned int num_fans)
{
static unsigned int brightness_code[5] =
{
UNIHUB_SLV2_LED_BRIGHTNESS_000,
UNIHUB_SLV2_LED_BRIGHTNESS_025,
UNIHUB_SLV2_LED_BRIGHTNESS_050,
UNIHUB_SLV2_LED_BRIGHTNESS_075,
UNIHUB_SLV2_LED_BRIGHTNESS_100
};
static unsigned int speed_code[5] =
{
UNIHUB_SLV2_LED_SPEED_000,
UNIHUB_SLV2_LED_SPEED_025,
UNIHUB_SLV2_LED_SPEED_050,
UNIHUB_SLV2_LED_SPEED_075,
UNIHUB_SLV2_LED_SPEED_100
};
unsigned char fan_led_data[16 * 6 * 3];
int cur_led_idx;
float brightness;
/*-----------------------------------------------------*\
| Zero out buffer |
\*-----------------------------------------------------*/
memset(fan_led_data, 0x00, sizeof(fan_led_data));
std::vector<RGBColor> colors = active_mode.colors;
unsigned int num_colors = colors.size();
if(!colors.empty()) // Update led_data if there's colors
{
brightness = static_cast<float>(active_mode.brightness)/4;
if (num_colors == 6)
{
for(unsigned int i = 0; i < 6; i++)
{
float brightness_scale = brightness * brightnessLimit(colors[i]);
for(unsigned int led_idx = 0; led_idx < 16 * 3; led_idx += 3)
{
cur_led_idx = (i * 16 * 3) + led_idx;
fan_led_data[cur_led_idx + 0] = (RGBGetRValue(colors[i]) * brightness_scale);
fan_led_data[cur_led_idx + 1] = (RGBGetBValue(colors[i]) * brightness_scale);
fan_led_data[cur_led_idx + 2] = (RGBGetGValue(colors[i]) * brightness_scale);
}
}
}
else
{
colors.resize(4);
for(unsigned int i = num_colors; i < 4; i++)
{
colors[i] = 0x00;
}
// needs a 72 length array of 4 colors, even if less are defined
for(unsigned int j = 0; j < 4; j++)
{
float brightness_scale = brightness * brightnessLimit(colors[j]);
for(unsigned int i = 0; i < 6; i++)
{
cur_led_idx = (i * 12) + (j * 3);
fan_led_data[cur_led_idx + 0] = RGBGetRValue(colors[j]) * brightness_scale;
fan_led_data[cur_led_idx + 1] = RGBGetBValue(colors[j]) * brightness_scale;
fan_led_data[cur_led_idx + 2] = RGBGetGValue(colors[j]) * brightness_scale;
}
}
}
}
SendStartAction
(
channel, // Current channel
(num_fans + 1) // Number of fans
);
SendColorData
(
channel, // Channel
(num_fans + 1)*16,
fan_led_data // Data
);
SendCommitAction
(
channel, // Channel
active_mode.value, // Effect
speed_code[active_mode.speed], // Speed
active_mode.direction, // Direction
brightness_code[active_mode.brightness] // Brightness
);
}
void LianLiUniHubSLV2Controller::SendStartAction(unsigned char channel, unsigned int num_fans)
{
unsigned char usb_buf[65];
/*-----------------------------------------------------*\
| Zero out buffer |
\*-----------------------------------------------------*/
memset(usb_buf, 0x00, sizeof(usb_buf));
/*-----------------------------------------------------*\
| Set up message packet |
\*-----------------------------------------------------*/
usb_buf[0x00] = UNIHUB_SLV2_TRANSACTION_ID;
usb_buf[0x01] = 0x10;
usb_buf[0x02] = 0x60;
usb_buf[0x03] = (channel << 4) + num_fans;
/*-----------------------------------------------------*\
| Send packet |
\*-----------------------------------------------------*/
hid_write(dev, usb_buf, sizeof(usb_buf));
std::this_thread::sleep_for(5ms);
}
void LianLiUniHubSLV2Controller::SendColorData(unsigned char channel, unsigned int num_leds, unsigned char* led_data)
{
/*---------------------------------------------------------*\
| Send LED data |
\*---------------------------------------------------------*/
unsigned char usb_buf[353];
/*-----------------------------------------------------*\
| Zero out buffer |
\*-----------------------------------------------------*/
memset(usb_buf, 0x00, sizeof(usb_buf));
/*-----------------------------------------------------*\
| Set up message packet |
\*-----------------------------------------------------*/
usb_buf[0x00] = UNIHUB_SLV2_TRANSACTION_ID;
usb_buf[0x01] = 0x30 + channel; // action + channel(30 = channel 1, 31 = channel 2, etc.)
/*-----------------------------------------------------*\
| Copy in color data bytes |
\*-----------------------------------------------------*/
memcpy(&usb_buf[0x02], led_data, num_leds * 3);
/*-----------------------------------------------------*\
| Send packet |
\*-----------------------------------------------------*/
hid_write(dev, usb_buf, sizeof(usb_buf));
std::this_thread::sleep_for(5ms);
}
void LianLiUniHubSLV2Controller::SendCommitAction(unsigned char channel, unsigned char effect, unsigned char speed, unsigned int direction, unsigned int brightness)
{
unsigned char usb_buf[65];
/*-----------------------------------------------------*\
| Zero out buffer |
\*-----------------------------------------------------*/
memset(usb_buf, 0x00, sizeof(usb_buf));
/*-----------------------------------------------------*\
| Set up message packet |
\*-----------------------------------------------------*/
usb_buf[0x00] = UNIHUB_SLV2_TRANSACTION_ID;
usb_buf[0x01] = 0x10 + channel; // Channel+device (10 = channel 1, 11 = channel 2, etc.)
usb_buf[0x02] = effect; // Effect
usb_buf[0x03] = speed; // Speed, 02=0%, 01=25%, 00=50%, ff=75%, fe=100%
usb_buf[0x04] = direction; // Direction, right=00, left=01
usb_buf[0x05] = brightness; // Brightness, 0=100%, 1= 75%, 2 = 50%, 3 = 25%, 8 = 0%
/*-----------------------------------------------------*\
| Send packet |
\*-----------------------------------------------------*/
hid_write(dev, usb_buf, sizeof(usb_buf));
std::this_thread::sleep_for(5ms);
}