/*-----------------------------------------------------*\
| DeviceView.cpp                                        |
|                                                       |
|  OpenRGB Device view widget for Qt                    |
|                                                       |
| Adam Honse (calcprogrammer1@gmail.com)                |
\*-----------------------------------------------------*/

#include "DeviceView.h"
#include "RGBController.h"
#include <QPainter>
#include <QResizeEvent>
#include <QStyleOption>
#include <QtCore/qmath.h>
#include <QDebug>
#include <QMouseEvent>

DeviceView::DeviceView(QWidget *parent) :
    QWidget(parent),
    initSize(128,128),
    mouseDown(false)
{
    controller = NULL;
    setMouseTracking(1);

    size = width();
}

struct led_label
{
    QString label_text;
    QString label_utf8;
};

static const std::map<std::string, led_label> led_label_lookup =
{
    { "Key: A",                 { "A"     , "A",               }},
    { "Key: B",                 { "B"     , "B",               }},
    { "Key: C",                 { "C"     , "C",               }},
    { "Key: D",                 { "D"     , "D",               }},
    { "Key: E",                 { "E"     , "E",               }},
    { "Key: F",                 { "F"     , "F",               }},
    { "Key: G",                 { "G"     , "G",               }},
    { "Key: H",                 { "H"     , "H",               }},
    { "Key: I",                 { "I"     , "I",               }},
    { "Key: J",                 { "J"     , "J",               }},
    { "Key: K",                 { "K"     , "K",               }},
    { "Key: L",                 { "L"     , "L",               }},
    { "Key: M",                 { "M"     , "M",               }},
    { "Key: N",                 { "N"     , "N",               }},
    { "Key: O",                 { "O"     , "O",               }},
    { "Key: P",                 { "P"     , "P",               }},
    { "Key: Q",                 { "Q"     , "Q",               }},
    { "Key: R",                 { "R"     , "R",               }},
    { "Key: S",                 { "S"     , "S",               }},
    { "Key: T",                 { "T"     , "T",               }},
    { "Key: U",                 { "U"     , "U",               }},
    { "Key: V",                 { "V"     , "V",               }},
    { "Key: W",                 { "W"     , "W",               }},
    { "Key: X",                 { "X"     , "X",               }},
    { "Key: Y",                 { "Y"     , "Y",               }},
    { "Key: Z",                 { "Z"     , "Z",               }},
    { "Key: 0",                 { "0"     , "0",               }},
    { "Key: 1",                 { "1"     , "1",               }},
    { "Key: 2",                 { "2"     , "2",               }},
    { "Key: 3",                 { "3"     , "3",               }},
    { "Key: 4",                 { "4"     , "4",               }},
    { "Key: 5",                 { "5"     , "5",               }},
    { "Key: 6",                 { "6"     , "6",               }},
    { "Key: 7",                 { "7"     , "7",               }},
    { "Key: 8",                 { "8"     , "8",               }},
    { "Key: 9",                 { "9"     , "9",               }},
    { "Key: F1",                { "F1"    , "F1",              }},
    { "Key: F2",                { "F2"    , "F2",              }},
    { "Key: F3",                { "F3"    , "F3",              }},
    { "Key: F4",                { "F4"    , "F4",              }},
    { "Key: F5",                { "F5"    , "F5",              }},
    { "Key: F6",                { "F6"    , "F6",              }},
    { "Key: F7",                { "F7"    , "F7",              }},
    { "Key: F8",                { "F8"    , "F8",              }},
    { "Key: F9",                { "F9"    , "F9",              }},
    { "Key: F10",               { "F10"   , "F10",             }},
    { "Key: F11",               { "F11"   , "F11",             }},
    { "Key: F12",               { "F12"   , "F12",             }},
    { "Key: `",                 { "`"     , "`",               }},
    { "Key: -",                 { "-"     , "-",               }},
    { "Key: =",                 { "="     , "=",               }},
    { "Key: [",                 { "["     , "[",               }},
    { "Key: ]",                 { "]"     , "]",               }},
    { "Key: \\ (ANSI)",         { "\\"    , "\\",              }},
    { "Key: \\ (ISO)",          { "\\"    , "\\",              }},
    { "Key: ;",                 { ";"     , ";",               }},
    { "Key: '",                 { "'"     , "'",               }},
    { "Key: ,",                 { ","     , ",",               }},
    { "Key: .",                 { "."     , ".",               }},
    { "Key: /",                 { "/"     , "/",               }},
    { "Key: Escape",            { "Esc"   , "Esc",             }},
    { "Key: Print Screen",      { "Prt"   , "Prt",             }},
    { "Key: Scroll Lock",       { "Scr"   , "Scr",             }},
    { "Key: Pause/Break",       { "Brk"   , "Brk",             }},
    { "Key: Backspace",         { "Bks"   , "\xE2\x8C\xAB"     }}, // ⌫
    { "Key: Insert",            { "Ins"   , "Ins",             }},
    { "Key: Home",              { "Hom"   , "Hom",             }},
    { "Key: Page Up",           { "PUp"   , "PUp",             }},
    { "Key: Tab",               { "Tab"   , "\xE2\x86\xb9"     }}, // ⭾ ↹ ⇄ ⇥
    { "Key: Delete",            { "Del"   , "Del",             }},
    { "Key: End",               { "End"   , "End",             }},
    { "Key: Page Down",         { "PDn"   , "PDn",             }},
    { "Key: Caps Lock",         { "Cap"   , "Cap",             }},
    { "Key: Enter",             { "Ent"   , "\xE2\x86\xB5"     }}, // ↵ ⏎
    { "Key: Enter (ISO)",       { "Ent"   , "\xE2\x86\xB5"     }}, // ↵ ⏎
    { "Key: Left Shift",        { "Sft"   , "\xE2\x87\xA7"     }}, // ⇧
    { "Key: Right Shift",       { "Sft"   , "\xE2\x87\xA7"     }}, // ⇧
    { "Key: Up Arrow",          { "Up"    , "\xF0\x9F\xA1\xB9" }}, // ↑ 🡹
    { "Key: Left Control",      { "Ctl"   , "Ctl",             }},
    { "Key: Left Windows",      { "Win"   , "\xe2\x9d\x96"     }}, // ❖
    { "Key: Left Fn",           { "Fn"    , "Fn",              }},
    { "Key: Left Alt",          { "Alt"   , "Alt",             }},
    { "Key: Space",             { "Spc"   , "Spc",             }},
    { "Key: Right Alt",         { "Alt"   , "Alt",             }},
    { "Key: Right Fn",          { "Fn"    , "Fn",              }},
    { "Key: Right Windows",     { "Win"   , "\xe2\x9d\x96"     }}, // ❖
    { "Key: Menu",              { "Mnu"   , "\xE2\x98\xB0"     }}, // ▤ ☰ 𝌆 🗏
    { "Key: Right Control",     { "Ctl"   , "Ctl",             }},
    { "Key: Left Arrow",        { "Lft"   , "\xF0\x9F\xA1\xB8" }}, // ↑ 🡹
    { "Key: Down Arrow",        { "Dn"    , "\xF0\x9F\xA1\xBB" }}, // ↑ 🡹
    { "Key: Right Arrow",       { "Rgt"   , "\xF0\x9F\xA1\xBA" }}, // ↑ 🡹
    { "Key: Num Lock",          { "Num"   , "Num",             }},
    { "Key: Number Pad /",      { "/"     , "/",               }},
    { "Key: Number Pad *",      { "*"     , "*",               }},
    { "Key: Number Pad -",      { "-"     , "-",               }},
    { "Key: Number Pad +",      { "+"     , "+",               }},
    { "Key: Number Pad .",      { "."     , ".",               }},
    { "Key: Number Pad Enter",  { "Ent"   , "\xE2\x86\xB5"     }}, // ↵ ⏎
    { "Key: Number Pad 0",      { "0"     , "0",               }},
    { "Key: Number Pad 1",      { "1"     , "1",               }},
    { "Key: Number Pad 2",      { "2"     , "2",               }},
    { "Key: Number Pad 3",      { "3"     , "3",               }},
    { "Key: Number Pad 4",      { "4"     , "4",               }},
    { "Key: Number Pad 5",      { "5"     , "5",               }},
    { "Key: Number Pad 6",      { "6"     , "6",               }},
    { "Key: Number Pad 7",      { "7"     , "7",               }},
    { "Key: Number Pad 8",      { "8"     , "8",               }},
    { "Key: Number Pad 9",      { "9"     , "9",               }},
};

void DeviceView::setController(RGBController * controller_ptr)
{
    /*-----------------------------------------------------*\
    | Store the controller pointer                          |
    \*-----------------------------------------------------*/
    controller = controller_ptr;

    /*-----------------------------------------------------*\
    | Set the size of the selection flags vector            |
    \*-----------------------------------------------------*/
    selectionFlags.resize(controller->leds.size());

    /*-----------------------------------------------------*\
    | Set the size of the zone and LED position vectors     |
    \*-----------------------------------------------------*/
    zone_pos.resize(controller->zones.size());
    led_pos.resize(controller->leds.size());
    led_labels.resize(controller->leds.size());

    /*-----------------------------------------------------*\
    | Process position and size for zones                   |
    \*-----------------------------------------------------*/
    unsigned int maxWidth       = 0;
    unsigned int maxCols        = 20;
    float        totalHeight    = 0;
    float        zonePadding    = 1;    // Amount of space between zones
    float        ledPadding     = 0.1;

    for(std::size_t zone_idx = 0; zone_idx < controller->zones.size(); zone_idx++)
    {
        if((controller->zones[zone_idx].type == ZONE_TYPE_MATRIX) && (controller->zones[zone_idx].matrix_map))
        {
            totalHeight += controller->zones[zone_idx].matrix_map->height;
            zone_pos[zone_idx].matrix_w = controller->zones[zone_idx].matrix_map->width;
        }
        else
        {
            unsigned int count = controller->zones[zone_idx].leds_count;
            zone_pos[zone_idx].matrix_w = std::min(count, maxCols);
            totalHeight += count / maxCols + !!(count % maxCols); // Equivalent to ceil(float(count) / maxCols);
        }

        if(zone_pos[zone_idx].matrix_w > maxWidth)
        {
            maxWidth = zone_pos[zone_idx].matrix_w;
        }
    }

    /*-----------------------------------------------------*\
    | Add some space for zone names and padding             |
    \*-----------------------------------------------------*/
    totalHeight += controller->zones.size() * zonePadding;

    float atom      = 1.0 / maxWidth;       // Atom is the width of a single square; if the whole thing becomes too tall, we ignore it and let the view widget take care of it
    float current_y = 0;                    // We will be descending, placing each zone one atom below the previous one
    matrix_h        = totalHeight * atom;

    for(std::size_t zone_idx = 0; zone_idx < controller->zones.size(); zone_idx++)
    {
        zone_pos[zone_idx].matrix_x = (1.0 - (zone_pos[zone_idx].matrix_w * atom)) / 2;
        zone_pos[zone_idx].matrix_y = current_y + 0.5 * atom;
        zone_pos[zone_idx].matrix_w *= atom;
        zone_pos[zone_idx].matrix_h = 0.4 * atom;
        current_y                  += zonePadding * atom;

        /*-----------------------------------------------------*\
        | Now process the position and size for the LEDs        |
        \*-----------------------------------------------------*/
        if((controller->zones[zone_idx].type == ZONE_TYPE_MATRIX) && (controller->zones[zone_idx].matrix_map))
        {
            matrix_map_type * map = controller->zones[zone_idx].matrix_map;

            for(unsigned int led_x = 0; led_x < map->width; led_x++)
            {
                for(unsigned int led_y = 0; led_y < map->height; led_y++)
                {
                    unsigned int map_idx    = led_y * map->width + led_x;
                    unsigned int color_idx  = map->map[map_idx] + controller->zones[zone_idx].start_idx;

                    if(color_idx != 0xFFFFFFFF && color_idx < led_pos.size())
                    {
                        led_pos[color_idx].matrix_x = (zone_pos[zone_idx].matrix_x + led_x + ledPadding) * atom;
                        led_pos[color_idx].matrix_y = current_y + (led_y + ledPadding) * atom;
                        led_pos[color_idx].matrix_w = (1 - (2 * ledPadding)) * atom;
                        led_pos[color_idx].matrix_h = (1 - (2 * ledPadding)) * atom;

                        /*-----------------------------------------------------*\
                        | Expand large keys to fill empty spaces in matrix, if  |
                        | possible.  Large keys can fill left, down, up, or wide|
                        | Fill Left:                                            |
                        |    Tab                                                |
                        |    Caps Lock                                          |
                        |    Left Shift                                         |
                        |    Right Shift                                        |
                        |    Backspace                                          |
                        |    Number Pad 0                                       |
                        |                                                       |
                        | Fill Up or Down:                                      |
                        |    Number Pad Enter                                   |
                        |    Number Pad +                                       |
                        |                                                       |
                        | Fill Wide:                                            |
                        |    Space                                              |
                        \*-----------------------------------------------------*/
                        if(led_x < map->width - 1 && map->map[map_idx + 1] == 0xFFFFFFFF)
                        {
                            if( ( controller->leds[color_idx].name == "Key: Tab"          )
                             || ( controller->leds[color_idx].name == "Key: Caps Lock"    )
                             || ( controller->leds[color_idx].name == "Key: Left Shift"   )
                             || ( controller->leds[color_idx].name == "Key: Right Shift"  )
                             || ( controller->leds[color_idx].name == "Key: Backspace"    )
                             || ( controller->leds[color_idx].name == "Key: Number Pad 0" ) )
                            {
                                led_pos[color_idx].matrix_w += atom;
                            }
                        }
                        if( ( controller->leds[color_idx].name == "Key: Number Pad Enter" )
                         || ( controller->leds[color_idx].name == "Key: Number Pad +"     ) )
                        {
                            if(led_y < map->height - 1 && map->map[map_idx + map->width] == 0xFFFFFFFF)
                            {
                                led_pos[color_idx].matrix_h += atom;
                            }
                            /* TODO: check if there isn't another widened key above */
                            else if(led_y > 0 && map->map[map_idx - map->width] == 0xFFFFFFFF)
                            {
                                led_pos[color_idx].matrix_y -= atom;
                                led_pos[color_idx].matrix_h += atom;
                            }
                        }
                        else if(controller->leds[color_idx].name == "Key: Space")
                        {
                            for(unsigned int map_idx2 = map_idx - 1; map_idx2 > led_y * map->width && map->map[map_idx2] == 0xFFFFFFFF; --map_idx2)
                            {
                                led_pos[color_idx].matrix_x -= atom;
                                led_pos[color_idx].matrix_w += atom;
                            }
                            for(unsigned int map_idx2 = map_idx + 1; map_idx2 < (led_y + 1) * map->width && map->map[map_idx2] == 0xFFFFFFFF; ++map_idx2)
                            {
                                led_pos[color_idx].matrix_w += atom;
                            }
                        }
                    }
                }
            }

            current_y += map->height * atom;
        }
        else
        {
            for(unsigned int i = 0; i < controller->zones[zone_idx].leds_count; i++)
            {
                led_pos[i + controller->zones[zone_idx].start_idx].matrix_x = zone_pos[zone_idx].matrix_x + (i % maxCols + ledPadding) * atom;
                led_pos[i + controller->zones[zone_idx].start_idx].matrix_y = current_y + (i / maxCols + ledPadding) * atom;
                led_pos[i + controller->zones[zone_idx].start_idx].matrix_w = (1 - (2 * ledPadding)) * atom;
                led_pos[i + controller->zones[zone_idx].start_idx].matrix_h = (1 - (2 * ledPadding)) * atom;
            }

            current_y += (controller->zones[zone_idx].leds_count / maxCols + !!(controller->zones[zone_idx].leds_count % maxCols)) * atom;
        }
    }

    /*-----------------------------------------------------*\
    | Update LED labels                                     |
    \*-----------------------------------------------------*/
    for(std::size_t led_idx = 0; led_idx < controller->leds.size(); led_idx++)
    {
        std::map<std::string, led_label>::const_iterator it = led_label_lookup.find(controller->leds[led_idx].name);

        if(it != led_label_lookup.end())
        {
            led_labels[led_idx] = it->second.label_utf8;
        }
    }

    /*-----------------------------------------------------*\
    | Update cached size and offset                         |
    \*-----------------------------------------------------*/
    size     = width();
    offset_x = 0;

    if(height() < size * matrix_h)
    {
        size     = height() / matrix_h;
        offset_x = (width() - size) / 2;
    }
}

QSize DeviceView::sizeHint () const
{
    return QSize(height(),height());
}

QSize DeviceView::minimumSizeHint () const
{
    return initSize;
}

void DeviceView::mousePressEvent(QMouseEvent *event)
{
    ctrlDown    = event->modifiers().testFlag(Qt::ControlModifier);
    mouseDown   = true;
    mouseMoved  = false;

    if(ctrlDown)
    {
        previousFlags = selectionFlags;
        previousSelection = selectedLeds;
    }

    /*-----------------------------------------------------*\
    | It's okay if the size becomes negative                |
    \*-----------------------------------------------------*/
    selectionRect.setLeft(event->x());
    selectionRect.setTop(event->y());
    selectionRect.setRight(event->x());
    selectionRect.setBottom(event->y());

    updateSelection();
    update();
}

void DeviceView::mouseMoveEvent(QMouseEvent *event)
{
    lastMousePos = event->pos();
    selectionRect.setRight(event->x());
    selectionRect.setBottom(event->y());

    if(mouseDown)
    {
        mouseMoved  = true;
        ctrlDown    = event->modifiers().testFlag(Qt::ControlModifier);

        /*-----------------------------------------------------*\
        | Clear the previous selection in case ctrl is released |
        \*-----------------------------------------------------*/
        if(!ctrlDown)
        {
            previousSelection.clear();
            previousFlags.clear();
            previousFlags.resize(controller->leds.size());
        }
        updateSelection();
    }
    update();
}

void DeviceView::mouseReleaseEvent(QMouseEvent* event)
{
    selectionRect = selectionRect.normalized();
    mouseDown = false;

    /*-----------------------------------------------------*\
    | Check if the user clicked a zone name & select it     |
    \*-----------------------------------------------------*/
    if(!mouseMoved)
    {
        int size     = width();
        int offset_x = 0;

        if(height() < size * matrix_h)
        {
            size     = height() / matrix_h;
            offset_x = (width() - size) / 2;
        }

        for(std::size_t zone_idx = 0; zone_idx < controller->zones.size(); zone_idx++)
        {
            int posx = zone_pos[zone_idx].matrix_x * size + offset_x;
            int posy = zone_pos[zone_idx].matrix_y * size;
            int posw = zone_pos[zone_idx].matrix_w * size;
            int posh = zone_pos[zone_idx].matrix_h * size;

            QRect rect = {posx, posy, posw, posh};

            if(rect.contains(event->pos()))
            {
                selectZone(zone_idx, ctrlDown);
            }
        }
    }
    update();
}

void DeviceView::resizeEvent(QResizeEvent* /*event*/)
{
    size     = width();
    offset_x = 0;

    if(height() < size * matrix_h)
    {
        size     = height() / matrix_h;
        offset_x = (width() - size) / 2;
    }
    update();
}

void DeviceView::paintEvent(QPaintEvent* /* event */)
{
    QPainter painter(this);
    QFont font = painter.font();

    /*-----------------------------------------------------*\
    | If Device View is hidden, don't paint                 |
    \*-----------------------------------------------------*/
    if(isHidden())
    {
        return;
    }

    /*-----------------------------------------------------*\
    | If controller has resized, reinitialize local data    |
    \*-----------------------------------------------------*/
    if(controller->leds.size() != led_pos.size())
    {
        setController(controller);
    }

    /*-----------------------------------------------------*\
    | LED rectangles                                        |
    \*-----------------------------------------------------*/
    for(std::size_t led_idx = 0; led_idx < controller->leds.size(); led_idx++)
    {
        int posx = led_pos[led_idx].matrix_x * size + offset_x;
        int posy = led_pos[led_idx].matrix_y * size;
        int posw = led_pos[led_idx].matrix_w * size;
        int posh = led_pos[led_idx].matrix_h * size;

        QRect rect = {posx, posy, posw, posh};

        /*-----------------------------------------------------*\
        | Fill color                                            |
        \*-----------------------------------------------------*/
        QColor currentColor = QColor::fromRgb(
                    RGBGetRValue(controller->colors[led_idx]),
                    RGBGetGValue(controller->colors[led_idx]),
                    RGBGetBValue(controller->colors[led_idx]));
        painter.setBrush(currentColor);

        /*-----------------------------------------------------*\
        | Border color                                          |
        \*-----------------------------------------------------*/
        if(selectionFlags[led_idx])
        {
            painter.setPen(palette().highlight().color());
        }
        else
        {
            painter.setPen(palette().dark().color());
        }
        painter.drawRect(rect);

        /*-----------------------------------------------------*\
        | Label                                                 |
        | Set the font color so that the text is visible        |
        \*-----------------------------------------------------*/
        font.setPixelSize(posh / 2);
        painter.setFont(font);

        unsigned int luma = (unsigned int)(0.2126f * currentColor.red() + 0.7152f * currentColor.green() + 0.0722f * currentColor.blue());

        if(luma > 127)
        {
            painter.setPen(Qt::black);
        }
        else
        {
            painter.setPen(Qt::white);
        }
        painter.drawText(rect, Qt::AlignVCenter | Qt::AlignHCenter, QString(led_labels[led_idx]));
    }

    font.setPixelSize(12);
    painter.setFont(font);

    /*-----------------------------------------------------*\
    | Zone names                                            |
    \*-----------------------------------------------------*/
    for(std::size_t zone_idx = 0; zone_idx < controller->zones.size(); zone_idx++)
    {
        int posx = zone_pos[zone_idx].matrix_x * size + offset_x;
        int posy = zone_pos[zone_idx].matrix_y * size;
        int posw = zone_pos[zone_idx].matrix_w * size;
        int posh = zone_pos[zone_idx].matrix_h * size;

        QRect rect = {posx, posy, posw, posh};

        if(rect.contains(lastMousePos) && (!mouseDown || !mouseMoved))
        {
            painter.setPen(palette().highlight().color());
        }
        else
        {
            painter.setPen(palette().windowText().color());
        }
        painter.drawText(posx, posy + posh, QString(controller->zones[zone_idx].name.c_str()));
    }

    /*-----------------------------------------------------*\
    | Selection area                                        |
    \*-----------------------------------------------------*/
    if(mouseDown)
    {
        QRect rect = selectionRect.normalized();
        QColor color = palette().highlight().color();
        color.setAlpha(63);
        painter.fillRect(rect, color);
        color.setAlpha(127);
        painter.setBrush(color);
        painter.drawRect(rect);
    }
}

void DeviceView::updateSelection()
{
    selectedLeds.clear();
    selectionFlags.clear();
    selectionFlags.resize(controller->leds.size());

    QRect sel              = selectionRect.normalized();
    std::vector<led>& leds = controller->leds;

    for(std::size_t led_idx = 0; led_idx < leds.size(); ++led_idx)
    {
        /*-----------------------------------------------------*\
        | Check intersection                                    |
        \*-----------------------------------------------------*/
        int posx = led_pos[led_idx].matrix_x * size + offset_x;
        int posy = led_pos[led_idx].matrix_y * size;
        int posw = led_pos[led_idx].matrix_w * size;
        int posh = led_pos[led_idx].matrix_h * size;

        QRect rect = {posx, posy, posw, posh};

        selectionFlags[led_idx] = 0;

        if(sel.intersects(rect))
        {
            selectionFlags[led_idx] = 1;
        }
        if(ctrlDown)
        {
            selectionFlags[led_idx] ^= previousFlags[led_idx];
        }

        if(selectionFlags[led_idx])
        {
            selectedLeds.push_back(led_idx);
        }
    }
    
    update();

    /*-----------------------------------------------------*\
    | Send selection changed signal                         |
    \*-----------------------------------------------------*/
    emit selectionChanged(selectedLeds);
}

bool DeviceView::selectLed(int target)
{
    if(target < 0 || size_t(target) >= controller->leds.size())
    {
        return false;
    }

    selectedLeds.resize(1);
    selectedLeds[0] = target;
    selectionFlags.clear();
    selectionFlags.resize(controller->leds.size());
    selectionFlags[target] = 1;

    update();

    /*-----------------------------------------------------*\
    | Send selection changed signal                         |
    \*-----------------------------------------------------*/
    emit selectionChanged(selectedLeds);

    return true;
}

bool DeviceView::selectLeds(QVector<int> target)
{
    for(int item: target)
    {
        if(item < 0 || size_t(item) >= controller->leds.size())
        {
            return false;
        }
    }

    selectionFlags.clear();
    selectionFlags.resize(controller->leds.size());

    for(int item: target)
    {
        selectionFlags[item] = 1;
    }

    //selectedLeds = target;

    /*-----------------------------------------------------*\
    | Filter out duplicate items                            |
    \*-----------------------------------------------------*/
    selectedLeds.clear();

    for(int i = 0; i < selectionFlags.size(); ++i)
    {
        if(selectionFlags[i])
        {
            selectedLeds.push_back(i);
        }
    }

    update();
    
    /*-----------------------------------------------------*\
    | Send selection changed signal                         |
    \*-----------------------------------------------------*/
    emit selectionChanged(selectedLeds);

    return true;
}

bool DeviceView::selectZone(int zone, bool add)
{
    if(zone < 0 || size_t(zone) >= controller->zones.size())
    {
        return false;
    }

    if(!add)
    {
        selectedLeds.clear();
        selectionFlags.clear();
        selectionFlags.resize(controller->leds.size());
    }

    int zoneStart = controller->zones[zone].start_idx;

    for(std::size_t led_idx = 0; led_idx < controller->zones[zone].leds_count; ++led_idx)
    {
        if(!selectionFlags[zoneStart + led_idx])
        {
            selectedLeds.push_back(zoneStart + led_idx);
            selectionFlags[zoneStart + led_idx] = 1;
        }
    }

    update();
    
    /*-----------------------------------------------------*\
    | Send selection changed signal                         |
    \*-----------------------------------------------------*/
    emit selectionChanged(selectedLeds);

    return true;
}

void DeviceView::clearSelection()
{
    /*-----------------------------------------------------*\
    | Same as selecting the entire device                   |
    \*-----------------------------------------------------*/
    selectedLeds.clear();
    selectionFlags.clear();
    selectionFlags.resize(controller->leds.size());
}

void DeviceView::setSelectionColor(RGBColor color)
{
    if(selectedLeds.isEmpty())
    {
        controller->SetAllLEDs(color);
    }
    else
    {
        for(int led_idx: selectedLeds)
        {
            controller->SetLED(led_idx, color);
        }
    }
    controller->UpdateLEDs();
    update();
}