TerrariaRGB/OpenRGB.cpp at master · T4g1/TerrariaRGB · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
#include "OpenRGB.h"

Client client("Terraria CUESDK_2015.dll client");
int lastDeviceIndex = 0;
uint32_t* originalModes;

void formatOpenRgbError(std::string message, std::string status) {
    error("OpenRGB SDK: %s: %s (error code: %d)",
        message,
        status,
        int(client.getLastSystemError()));
}

void formatOpenRgbError(std::string message, ConnectStatus status) {
    formatOpenRgbError(message, enumString(status));
}

void formatOpenRgbError(std::string message, RequestStatus status) {
    formatOpenRgbError(message, enumString(status));
}

bool connectClient() {
    ConnectStatus status = client.connect(SDK_HOST);
    if (status != ConnectStatus::Success) {
        formatOpenRgbError("Failed to connect", status);
        return false;
    }

    info("OpenRGB SDK: Connected");

    return true;
}

size_t getDeviceCount() {
    DeviceListResult result = client.requestDeviceList();
    if (result.status != RequestStatus::Success) {
        formatOpenRgbError("Unable to request devices list", result.status);
        return 0;
    }

    return result.devices.size();
}

const Device* getLastDevice() {
    return getDevice(lastDeviceIndex);
}

const Device* getDevice(int deviceIndex) {
    DeviceListResult result = client.requestDeviceList();
    if (result.status != RequestStatus::Success) {
        formatOpenRgbError("Unable to request devices list", result.status);
        return NULL;
    }

    if (deviceIndex > result.devices.size()) {
        error("There are only %d devices", result.devices.size());
        return NULL;
    }

    lastDeviceIndex = deviceIndex;

    return &result.devices[deviceIndex];
}

void setLedColor(int ledIndex, uint8_t r, uint8_t g, uint8_t b) {
    const Device * device = getLastDevice();

    client.setLEDColor(device->leds[ledIndex], Color {r, g, b});
}

void saveOriginalModes() {
    info("OpenRGB: Saving current modes...");
    int deviceCount = getDeviceCount();
    originalModes = new uint32_t[deviceCount];
    for (int i=0; i<deviceCount; i++) {
        const Device* device = getDevice(i);
        originalModes[i] = device->active_mode;

        debug("DEVICE %d: mode: %d saved", i, originalModes[i]);
    }
}

void restoreOriginalModes() {
    info("OpenRGB: Restoring original modes...");
    int deviceCount = getDeviceCount();
    for (int i=0; i<deviceCount; i++) {
        const Device* device = getDevice(i);
        uint32_t originalMode = originalModes[i];

        client.changeMode(*device, device->modes[originalModes[i]]);

        debug("DEVICE %d: mode: %d restored", i, originalMode);
    }

    delete originalModes;
}