/*
  Q Light Controller Plus
  qlcchannel.cpp

  Copyright (C) Heikki Junnila
                Massimo Callegari

  Licensed under the Apache License, Version 2.0 (the "License");
  you may not use this file except in compliance with the License.
  You may obtain a copy of the License at

      http://www.apache.org/licenses/LICENSE-2.0.txt

  Unless required by applicable law or agreed to in writing, software
  distributed under the License is distributed on an "AS IS" BASIS,
  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  See the License for the specific language governing permissions and
  limitations under the License.
*/

#include <QXmlStreamReader>
#include <QStringList>
#include <QMetaEnum>
#include <QPainter>
#include <iostream>
#include <QString>
#include <QDebug>
#include <QFile>

#include "qlcchannel.h"
#include "qlccapability.h"

#define KXMLQLCChannelGroupIntensity   QString("Intensity")
#define KXMLQLCChannelGroupColour      QString("Colour")
#define KXMLQLCChannelGroupGobo        QString("Gobo")
#define KXMLQLCChannelGroupPrism       QString("Prism")
#define KXMLQLCChannelGroupShutter     QString("Shutter")
#define KXMLQLCChannelGroupBeam        QString("Beam")
#define KXMLQLCChannelGroupSpeed       QString("Speed")
#define KXMLQLCChannelGroupEffect      QString("Effect")
#define KXMLQLCChannelGroupPan         QString("Pan")
#define KXMLQLCChannelGroupTilt        QString("Tilt")
#define KXMLQLCChannelGroupMaintenance QString("Maintenance")
#define KXMLQLCChannelGroupNothing     QString("Nothing")

#define KXMLQLCChannelColourGeneric    QString("Generic")
#define KXMLQLCChannelColourRed        QString("Red")
#define KXMLQLCChannelColourGreen      QString("Green")
#define KXMLQLCChannelColourBlue       QString("Blue")
#define KXMLQLCChannelColourCyan       QString("Cyan")
#define KXMLQLCChannelColourMagenta    QString("Magenta")
#define KXMLQLCChannelColourYellow     QString("Yellow")
#define KXMLQLCChannelColourAmber      QString("Amber")
#define KXMLQLCChannelColourWhite      QString("White")
#define KXMLQLCChannelColourUV         QString("UV")
#define KXMLQLCChannelColourLime       QString("Lime")
#define KXMLQLCChannelColourIndigo     QString("Indigo")

QLCChannel::QLCChannel(QObject *parent)
    : QObject(parent)
    , m_preset(Custom)
    , m_group(Intensity)
    , m_defaultValue(0)
    , m_controlByte(MSB)
    , m_colour(NoColour)
{
}

QLCChannel *QLCChannel::createCopy()
{
    QLCChannel *copy = new QLCChannel();
    copy->setPreset(this->preset());
    if (this->preset() != Custom)
    {
        copy->setName(this->name());
        copy->addPresetCapability();
    }
    else
    {
        copy->setGroup(this->group());
        copy->setControlByte(this->controlByte());
        copy->setColour(this->colour());
        copy->setName(this->name());

        QListIterator<QLCCapability*> it(this->capabilities());
        while (it.hasNext() == true)
            copy->addCapability(it.next()->createCopy());
    }
    copy->setDefaultValue(this->defaultValue());

    return copy;
}

QLCChannel::~QLCChannel()
{
    while (m_capabilities.isEmpty() == false)
        delete m_capabilities.takeFirst();
}

QLCChannel& QLCChannel::operator=(const QLCChannel& channel)
{
    if (this != &channel)
    {
        QListIterator<QLCCapability*> it(channel.m_capabilities);

        m_name = channel.m_name;
        m_preset = channel.m_preset;
        m_group = channel.m_group;
        m_defaultValue = channel.m_defaultValue;
        m_controlByte = channel.m_controlByte;
        m_colour = channel.m_colour;

        /* Clear old capabilities */
        while (m_capabilities.isEmpty() == false)
            delete m_capabilities.takeFirst();

        /* Copy new capabilities from the other channel */
        while (it.hasNext() == true)
            m_capabilities.append(it.next()->createCopy());
    }

    return *this;
}

quint32 QLCChannel::invalid()
{
    return UINT_MAX;
}

/*********************************************************************
 * Presets
 *********************************************************************/

QString QLCChannel::presetToString(QLCChannel::Preset preset)
{
    int index = staticMetaObject.indexOfEnumerator("Preset");
    return staticMetaObject.enumerator(index).valueToKey(preset);
}

QLCChannel::Preset QLCChannel::stringToPreset(const QString &preset)
{
    int index = staticMetaObject.indexOfEnumerator("Preset");
    return Preset(staticMetaObject.enumerator(index).keyToValue(preset.toStdString().c_str()));
}

QLCChannel::Preset QLCChannel::preset() const
{
    return m_preset;
}

/* please see
https://github.com/mcallegari/qlcplus/wiki/Fixture-definition-presets
when changing this function */
void QLCChannel::setPreset(QLCChannel::Preset preset)
{
    if (preset == m_preset)
        return;

    m_preset = preset;

    if (preset == Custom)
        return;

    Group grp = Intensity;
    PrimaryColour col = NoColour;
    ControlByte cb = MSB;
    QString prname;

    switch (preset)
    {
        case IntensityMasterDimmer:
            prname = "Master dimmer";
        break;
        case IntensityMasterDimmerFine:
            prname = "Master dimmer fine";
            cb = LSB;
        break;
        case IntensityDimmer:
            prname = "Dimmer";
        break;
        case IntensityDimmerFine:
            prname = "Dimmer fine";
            cb = LSB;
        break;
        case IntensityRed:
            prname = KXMLQLCChannelColourRed;
            col = Red;
        break;
        case IntensityRedFine:
            prname = KXMLQLCChannelColourRed + " fine";
            col = Red;
            cb = LSB;
        break;
        case IntensityGreen:
            prname = KXMLQLCChannelColourGreen;
            col = Green;
        break;
        case IntensityGreenFine:
            prname = KXMLQLCChannelColourGreen + " fine";
            col = Green;
            cb = LSB;
        break;
        case IntensityBlue:
            prname = KXMLQLCChannelColourBlue;
            col = Blue;
        break;
        case IntensityBlueFine:
            prname = KXMLQLCChannelColourBlue + " fine";
            col = Blue;
            cb = LSB;
        break;
        case IntensityCyan:
            prname = KXMLQLCChannelColourCyan;
            col = Cyan;
        break;
        case IntensityCyanFine:
            prname = KXMLQLCChannelColourCyan + " fine";
            col = Cyan;
            cb = LSB;
        break;
        case IntensityMagenta:
            prname = KXMLQLCChannelColourMagenta;
            col = Magenta;
        break;
        case IntensityMagentaFine:
            prname = KXMLQLCChannelColourMagenta + " fine";
            col = Magenta;
            cb = LSB;
        break;
        case IntensityYellow:
            prname = KXMLQLCChannelColourYellow;
            col = Yellow;
        break;
        case IntensityYellowFine:
            prname = KXMLQLCChannelColourYellow + " fine";
            col = Yellow;
            cb = LSB;
        break;
        case IntensityAmber:
            prname = KXMLQLCChannelColourAmber;
            col = Amber;
        break;
        case IntensityAmberFine:
            prname = KXMLQLCChannelColourAmber + " fine";
            col = Amber;
            cb = LSB;
        break;
        case IntensityWhite:
            prname = KXMLQLCChannelColourWhite;
            col = White;
        break;
        case IntensityWhiteFine:
            prname = KXMLQLCChannelColourWhite + " fine";
            col = White;
            cb = LSB;
        break;
        case IntensityUV:
            prname = KXMLQLCChannelColourUV;
            col = UV;
        break;
        case IntensityUVFine:
            prname = KXMLQLCChannelColourUV + " fine";
            col = UV;
            cb = LSB;
        break;
        case IntensityIndigo:
            prname = KXMLQLCChannelColourIndigo;
            col = Indigo;
        break;
        case IntensityIndigoFine:
            prname = KXMLQLCChannelColourIndigo + " fine";
            col = Indigo;
            cb = LSB;
        break;
        case IntensityLime:
            prname = KXMLQLCChannelColourLime;
            col = Lime;
        break;
        case IntensityLimeFine:
            prname = KXMLQLCChannelColourLime + " fine";
            col = Lime;
            cb = LSB;
        break;
        case IntensityHue:
            prname = "Hue";
            //col = Hue;
        break;
        case IntensityHueFine:
            prname = "Hue fine";
            //col = Hue;
            cb = LSB;
        break;
        case IntensitySaturation:
            prname = "Saturation";
            //col = Saturation;
        break;
        case IntensitySaturationFine:
            prname = "Saturation fine";
            //col = Saturation;
            cb = LSB;
        break;
        case IntensityLightness:
            prname = "Lightness";
            //col = Lightness;
        break;
        case IntensityLightnessFine:
            prname = "Lightness fine";
            //col = Lightness;
            cb = LSB;
        break;
        case IntensityValue:
            prname = "Value";
            //col = Value;
        break;
        case IntensityValueFine:
            prname = "Value fine";
            //col = Value;
            cb = LSB;
        break;
        case PositionPan:
            grp = Pan;
            prname = KXMLQLCChannelGroupPan;
        break;
        case PositionPanFine:
            grp = Pan;
            prname = KXMLQLCChannelGroupPan + " fine";
            cb = LSB;
        break;
        case PositionTilt:
            grp = Tilt;
            prname = KXMLQLCChannelGroupTilt;
        break;
        case PositionTiltFine:
            grp = Tilt;
            prname = KXMLQLCChannelGroupTilt + " fine";
            cb = LSB;
        break;
        case PositionXAxis:
            grp = Pan;
            prname = "X Axis";
        break;
        case PositionYAxis:
            grp = Tilt;
            prname = "Y Axis";
        break;
        case SpeedPanSlowFast:
        case SpeedPanFastSlow:
            grp = Speed;
            prname = "Pan speed";
        break;
        case SpeedTiltSlowFast:
        case SpeedTiltFastSlow:
            grp = Speed;
            prname = "Tilt speed";
        break;
        case SpeedPanTiltSlowFast:
        case SpeedPanTiltFastSlow:
            grp = Speed;
            prname = "Pan/Tilt speed";
        break;
        case ColorMacro:
            grp = Colour;
            prname = "Color macro";
        break;
        case ColorWheel:
            grp = Colour;
            prname = "Color wheel";
        break;
        case ColorWheelFine:
            grp = Colour;
            prname = "Color wheel fine";
            cb = LSB;
        break;
        case ColorRGBMixer:
            grp = Colour;
            prname = "RGB mixer";
        break;
        case ColorCTOMixer:
            grp = Colour;
            prname = "CTO mixer";
        break;
        case ColorCTCMixer:
            grp = Colour;
            prname = "CTC mixer";
        break;
        case ColorCTBMixer:
            grp = Colour;
            prname = "CTB mixer";
        break;
        case GoboWheel:
            grp = Gobo;
            prname = "Gobo wheel";
        break;
        case GoboWheelFine:
            grp = Gobo;
            prname = "Gobo wheel fine";
            cb = LSB;
        break;
        case GoboIndex:
            grp = Gobo;
            prname = "Gobo index";
        break;
        case GoboIndexFine:
            grp = Gobo;
            prname = "Gobo index fine";
            cb = LSB;
        break;
        case ShutterStrobeSlowFast:
        case ShutterStrobeFastSlow:
            grp = Shutter;
            prname = "Strobe";
        break;
        case ShutterIrisMinToMax:
        case ShutterIrisMaxToMin:
            grp = Shutter;
            prname = "Iris";
        break;
        case ShutterIrisFine:
            grp = Shutter;
            prname = "Iris fine";
            cb = LSB;
        break;
        case BeamFocusNearFar:
        case BeamFocusFarNear:
            grp = Beam;
            prname = "Focus";
        break;
        case BeamFocusFine:
            grp = Beam;
            prname = "Focus fine";
            cb = LSB;
        break;
        case BeamZoomSmallBig:
        case BeamZoomBigSmall:
            grp = Beam;
            prname = "Zoom";
        break;
        case BeamZoomFine:
            grp = Beam;
            prname = "Zoom fine";
            cb = LSB;
        break;
        case PrismRotationFastSlow:
        case PrismRotationSlowFast:
            grp = Prism;
            prname = "Prism rotation";
        break;
        case NoFunction:
            grp = Nothing;
            prname = "No function";
        break;
        default:
        break;
    }

    if (name().isEmpty())
        setName(prname);
    setGroup(grp);
    setColour(col);
    setControlByte(cb);
}

/* please see
https://github.com/mcallegari/qlcplus/wiki/Fixture-definition-presets
when changing this function */
QLCCapability *QLCChannel::addPresetCapability()
{
    QLCCapability *cap = new QLCCapability();
    switch (m_preset)
    {
        case IntensityMasterDimmer:
        case IntensityDimmer:
            cap->setName(name() + " (0 - 100%)");
        break;
        case IntensityRed:
        case IntensityGreen:
        case IntensityBlue:
        case IntensityCyan:
        case IntensityMagenta:
        case IntensityYellow:
        case IntensityAmber:
        case IntensityWhite:
        case IntensityUV:
        case IntensityIndigo:
        case IntensityLime:
        case IntensityHue:
        case IntensitySaturation:
        case IntensityLightness:
        case IntensityValue:
            cap->setName(name() + " intensity (0 - 100%)");
        break;
        case IntensityMasterDimmerFine:
        case IntensityDimmerFine:
        case IntensityRedFine:
        case IntensityGreenFine:
        case IntensityBlueFine:
        case IntensityCyanFine:
        case IntensityMagentaFine:
        case IntensityYellowFine:
        case IntensityAmberFine:
        case IntensityWhiteFine:
        case IntensityUVFine:
        case IntensityIndigoFine:
        case IntensityLimeFine:
        case IntensityHueFine:
        case IntensitySaturationFine:
        case IntensityLightnessFine:
        case IntensityValueFine:
        case PositionPan:
        case PositionPanFine:
        case PositionTilt:
        case PositionTiltFine:
        case PositionXAxis:
        case PositionYAxis:
        case ColorWheelFine:
        case ColorRGBMixer:
        case ColorCTOMixer:
        case ColorCTCMixer:
        case ColorCTBMixer:
        case GoboWheelFine:
        case GoboIndexFine:
        case ShutterIrisFine:
        case BeamFocusFine:
        case BeamZoomFine:
        case NoFunction:
            cap->setName(name());
        break;
        case SpeedPanSlowFast:
            cap->setName("Pan (Slow to fast)");
        break;
        case SpeedPanFastSlow:
            cap->setName("Pan (Fast to slow)");
        break;
        case SpeedTiltSlowFast:
            cap->setName("Tilt (Slow to fast)");
        break;
        case SpeedTiltFastSlow:
            cap->setName("Tilt (Fast to slow)");
        break;
        case SpeedPanTiltSlowFast:
            cap->setName("Pan and tilt (Slow to fast)");
        break;
        case SpeedPanTiltFastSlow:
            cap->setName("Pan and tilt (Fast to slow)");
        break;
        case ColorMacro:
        case ColorWheel:
        case GoboWheel:
        case GoboIndex:
            cap->setName(name() + " presets");
        break;
        case ShutterStrobeSlowFast:
            cap->setName("Strobe (Slow to fast)");
        break;
        case ShutterStrobeFastSlow:
            cap->setName("Strobe (Fast to slow)");
        break;
        case ShutterIrisMinToMax:
            cap->setName("Iris (Minimum to maximum)");
        break;
        case ShutterIrisMaxToMin:
            cap->setName("Iris (Maximum to minimum)");
        break;
        case BeamFocusNearFar:
            cap->setName("Beam (Near to far)");
        break;
        case BeamFocusFarNear:
            cap->setName("Beam (Far to near)");
        break;
        case BeamZoomSmallBig:
            cap->setName("Zoom (Small to big)");
        break;
        case BeamZoomBigSmall:
            cap->setName("Zoom (Big to small)");
        break;
        case PrismRotationSlowFast:
            cap->setName("Prism rotation (Slow to fast)");
        break;
        case PrismRotationFastSlow:
            cap->setName("Prism rotation (Fast to slow)");
        break;
        default:
        break;
    }

    addCapability(cap);

    return cap;
}

/*****************************************************************************
 * Groups
 *****************************************************************************/

QStringList QLCChannel::groupList()
{
    QStringList list;

    // Keep this list in alphabetical order because it's used only in UI
    list.append(KXMLQLCChannelGroupBeam);
    list.append(KXMLQLCChannelGroupColour);
    list.append(KXMLQLCChannelGroupEffect);
    list.append(KXMLQLCChannelGroupGobo);
    list.append(KXMLQLCChannelGroupIntensity);
    list.append(KXMLQLCChannelGroupMaintenance);
    list.append(KXMLQLCChannelGroupNothing);
    list.append(KXMLQLCChannelGroupPan);
    list.append(KXMLQLCChannelGroupPrism);
    list.append(KXMLQLCChannelGroupShutter);
    list.append(KXMLQLCChannelGroupSpeed);
    list.append(KXMLQLCChannelGroupTilt);

    return list;
}

QString QLCChannel::groupToString(Group grp)
{
    switch (grp)
    {
    case Intensity:
        return KXMLQLCChannelGroupIntensity;
    case Colour:
        return KXMLQLCChannelGroupColour;
    case Gobo:
        return KXMLQLCChannelGroupGobo;
    case Prism:
        return KXMLQLCChannelGroupPrism;
    case Shutter:
        return KXMLQLCChannelGroupShutter;
    case Beam:
        return KXMLQLCChannelGroupBeam;
    case Speed:
        return KXMLQLCChannelGroupSpeed;
    case Effect:
        return KXMLQLCChannelGroupEffect;
    case Pan:
        return KXMLQLCChannelGroupPan;
    case Tilt:
        return KXMLQLCChannelGroupTilt;
    case Maintenance:
        return KXMLQLCChannelGroupMaintenance;
    default:
        return KXMLQLCChannelGroupNothing;
    }
}

QLCChannel::Group QLCChannel::stringToGroup(const QString& str)
{
    if (str == KXMLQLCChannelGroupIntensity)
        return Intensity;
    else if (str == KXMLQLCChannelGroupColour)
        return Colour;
    else if (str == KXMLQLCChannelGroupGobo)
        return Gobo;
    else if (str == KXMLQLCChannelGroupPrism)
        return Prism;
    else if (str == KXMLQLCChannelGroupShutter)
        return Shutter;
    else if (str == KXMLQLCChannelGroupBeam)
        return Beam;
    else if (str == KXMLQLCChannelGroupSpeed)
        return Speed;
    else if (str == KXMLQLCChannelGroupEffect)
        return Effect;
    else if (str == KXMLQLCChannelGroupPan)
        return Pan;
    else if (str == KXMLQLCChannelGroupTilt)
        return Tilt;
    else if (str == KXMLQLCChannelGroupMaintenance)
        return Maintenance;
    else if (str == KXMLQLCChannelGroupNothing)
        return Nothing;
    else
        return NoGroup;
}

void QLCChannel::setGroup(Group grp)
{
    m_group = grp;
}

QLCChannel::Group QLCChannel::group() const
{
    return m_group;
}

QPixmap QLCChannel::drawIntensity(QColor color, QString str) const
{
    QPixmap pm(32, 32);
    QPainter painter(&pm);
    painter.setRenderHint(QPainter::Antialiasing);

    /*QFont tfont = QApplication::font();
    tfont.setBold(true);
    tfont.setPixelSize(14);
    painter.setFont(tfont);*/

    pm.fill(color);
    if (str == "B")
        painter.setPen(Qt::white);
    painter.drawText(0, 0, 32, 32, Qt::AlignHCenter|Qt::AlignVCenter, str);

    return pm;
}

QIcon QLCChannel::getIntensityIcon() const
{
    QPixmap pm(32, 32);

    if (m_colour == QLCChannel::Red)
        pm = drawIntensity(Qt::red, "R");
    else if (m_colour == QLCChannel::Green)
        pm = drawIntensity(Qt::green, "G");
    else if (m_colour == QLCChannel::Blue)
        pm = drawIntensity(Qt::blue, "B");
    else if (m_colour == QLCChannel::Cyan)
        pm = drawIntensity(Qt::cyan, "C");
    else if (m_colour == QLCChannel::Magenta)
        pm = drawIntensity(Qt::magenta, "M");
    else if (m_colour == QLCChannel::Yellow)
        pm = drawIntensity(Qt::yellow, "Y");
    else if (m_colour == QLCChannel::Amber)
        pm = drawIntensity(QColor(0xFFFF7E00), "A");
    else if (m_colour == QLCChannel::White)
        pm = drawIntensity(Qt::white, "W");
    else if (m_colour == QLCChannel::UV)
        pm = drawIntensity(QColor(0xFF9400D3), "UV");
    else if (m_colour == QLCChannel::Lime)
        pm = drawIntensity(QColor(0xFFADFF2F), "L");
    else if (m_colour == QLCChannel::Indigo)
        pm = drawIntensity(QColor(0xFF4B0082), "I");
    else
    {
        // None of the primary colours matched and since this is an
        // intensity channel, it must be controlling a plain dimmer OSLT.
        return QIcon(":/intensity.png");
    }

    return QIcon(pm);
}

QString QLCChannel::getIntensityColorCode(bool svg) const
{
    QString prefix = svg ? "qrc" : "";
    QString ext = svg ? "svg" : "png";

    if (m_colour == QLCChannel::Red)
        return svg ? QString("qrc:/red.svg") : QString("#FF0000");
    else if (m_colour == QLCChannel::Green)
        return svg ? QString("qrc:/green.svg") : QString("#00FF00");
    else if (m_colour == QLCChannel::Blue)
        return svg ? QString("qrc:/blue.svg") : QString("#0000FF");
    else if (m_colour == QLCChannel::Cyan)
        return svg ? QString("qrc:/cyan.svg") : QString("#00FFFF");
    else if (m_colour == QLCChannel::Magenta)
        return svg ? QString("qrc:/magenta.svg") : QString("#FF00FF");
    else if (m_colour == QLCChannel::Yellow)
        return svg ? QString("qrc:/yellow.svg") : QString("#FFFF00");
    else if (m_colour == QLCChannel::Amber)
        return svg ? QString("qrc:/amber.svg") : QString("#FF7E00");
    else if (m_colour == QLCChannel::White)
        return svg ? QString("qrc:/white.svg") : QString("#FFFFFF");
    else if (m_colour == QLCChannel::UV)
        return svg ? QString("qrc:/uv.svg") : QString("#9400D3");
    else if (m_colour == QLCChannel::Lime)
        return svg ? QString("qrc:/lime.svg") : QString("#ADFF2F");
    else if (m_colour == QLCChannel::Indigo)
        return svg ? QString("qrc:/indigo.svg") : QString("#4B0082");
    else
    {
        // None of the primary colours matched and since this is an
        // intensity channel, it must be controlling a plain dimmer OSLT.
        return QString("%1:/intensity.%2").arg(prefix).arg(ext);
    }

    return QString("%1:/intensity.%2").arg(prefix).arg(ext);
}

QIcon QLCChannel::getIcon() const
{
    if (group() == Intensity)
        return getIntensityIcon();
    else
        return QIcon(getIconNameFromGroup(group()));
}

QString QLCChannel::getIconNameFromGroup(QLCChannel::Group grp, bool svg) const
{
    QString prefix = svg ? "qrc" : "";
    QString ext = svg ? "svg" : "png";

    switch(grp)
    {
        case Pan: return QString("%1:/pan.%2").arg(prefix).arg(ext); break;
        case Tilt: return QString("%1:/tilt.%2").arg(prefix).arg(ext); break;
        case Colour: return QString("%1:/colorwheel.%2").arg(prefix).arg(ext); break;
        case Effect: return QString("%1:/star.%2").arg(prefix).arg(ext); break;
        case Gobo: return QString("%1:/gobo.%2").arg(prefix).arg(ext); break;
        case Shutter: return QString("%1:/shutter.%2").arg(prefix).arg(ext); break;
        case Speed: return QString("%1:/speed.%2").arg(prefix).arg(ext); break;
        case Prism: return QString("%1:/prism.%2").arg(prefix).arg(ext); break;
        case Maintenance: return QString("%1:/configure.%2").arg(prefix).arg(ext); break;
        case Intensity: return getIntensityColorCode(svg); break;
        case Beam: return QString("%1:/beam.%2").arg(prefix).arg(ext); break;
        case Nothing: return QString("%1:/uncheck.%2").arg(prefix).arg(ext); break;
        default:
        break;
    }

    return QString("%1:/intensity.%2").arg(prefix).arg(ext);
}

/*****************************************************************************
 * Properties
 *****************************************************************************/

QString QLCChannel::name() const
{
    return m_name;
}

void QLCChannel::setName(const QString &name)
{
    m_name = name;
}

uchar QLCChannel::defaultValue() const
{
    return m_defaultValue;
}

void QLCChannel::setDefaultValue(uchar value)
{
    m_defaultValue = value;
}

void QLCChannel::setControlByte(ControlByte byte)
{
    m_controlByte = byte;
}

QLCChannel::ControlByte QLCChannel::controlByte() const
{
    return m_controlByte;
}

/*****************************************************************************
 * Colours
 *****************************************************************************/

QStringList QLCChannel::colourList()
{
    QStringList list;
    //list << KXMLQLCChannelColourGeneric;
    list << KXMLQLCChannelColourRed;
    list << KXMLQLCChannelColourGreen;
    list << KXMLQLCChannelColourBlue;
    list << KXMLQLCChannelColourCyan;
    list << KXMLQLCChannelColourMagenta;
    list << KXMLQLCChannelColourYellow;
    list << KXMLQLCChannelColourAmber;
    list << KXMLQLCChannelColourWhite;
    list << KXMLQLCChannelColourUV;
    list << KXMLQLCChannelColourLime;
    list << KXMLQLCChannelColourIndigo;
    return list;
}

QString QLCChannel::colourToString(PrimaryColour colour)
{
    switch (colour)
    {
    case Red:
        return KXMLQLCChannelColourRed;
    case Green:
        return KXMLQLCChannelColourGreen;
    case Blue:
        return KXMLQLCChannelColourBlue;
    case Cyan:
        return KXMLQLCChannelColourCyan;
    case Magenta:
        return KXMLQLCChannelColourMagenta;
    case Yellow:
        return KXMLQLCChannelColourYellow;
    case Amber:
        return KXMLQLCChannelColourAmber;
    case White:
        return KXMLQLCChannelColourWhite;
    case UV:
        return KXMLQLCChannelColourUV;
    case Lime:
        return KXMLQLCChannelColourLime;
    case Indigo:
        return KXMLQLCChannelColourIndigo;
    case NoColour:
    default:
        return KXMLQLCChannelColourGeneric;
    }
}

QLCChannel::PrimaryColour QLCChannel::stringToColour(const QString& str)
{
    if (str == KXMLQLCChannelColourRed)
        return Red;
    else if (str == KXMLQLCChannelColourGreen)
        return Green;
    else if (str == KXMLQLCChannelColourBlue)
        return Blue;
    else if (str == KXMLQLCChannelColourCyan)
        return Cyan;
    else if (str == KXMLQLCChannelColourMagenta)
        return Magenta;
    else if (str == KXMLQLCChannelColourYellow)
        return Yellow;
    else if (str == KXMLQLCChannelColourAmber)
        return Amber;
    else if (str == KXMLQLCChannelColourWhite)
        return White;
    else if (str == KXMLQLCChannelColourUV)
        return UV;
    else if (str == KXMLQLCChannelColourLime)
        return Lime;
    else if (str == KXMLQLCChannelColourIndigo)
        return Indigo;
    else
        return NoColour;
}

void QLCChannel::setColour(QLCChannel::PrimaryColour colour)
{
    m_colour = colour;
}

QLCChannel::PrimaryColour QLCChannel::colour() const
{
    return m_colour;
}

/*****************************************************************************
 * Capabilities
 *****************************************************************************/

const QList <QLCCapability*> QLCChannel::capabilities() const
{
    return m_capabilities;
}

QLCCapability* QLCChannel::searchCapability(uchar value) const
{
    QListIterator <QLCCapability*> it(m_capabilities);
    while (it.hasNext() == true)
    {
        QLCCapability* capability = it.next();
        if (capability->min() <= value && capability->max() >= value)
            return capability;
    }

    return NULL;
}

QLCCapability* QLCChannel::searchCapability(const QString& name,
        bool exactMatch) const
{
    QListIterator <QLCCapability*> it(m_capabilities);
    while (it.hasNext() == true)
    {
        QLCCapability* capability = it.next();
        if (exactMatch == true && capability->name() == name)
            return capability;
        else if (exactMatch == false &&
                 capability->name().contains(name) == true)
            return capability;
    }

    return NULL;
}

bool QLCChannel::addCapability(QLCCapability* cap)
{
    Q_ASSERT(cap != NULL);

    /* Check for overlapping values */
    foreach (QLCCapability* another, m_capabilities)
    {
        if (another->overlaps(cap) == true)
            return false;
    }

    m_capabilities.append(cap);
    return true;
}

bool QLCChannel::setCapabilityRange(QLCCapability* cap, uchar min, uchar max)
{
    Q_ASSERT(cap != NULL);

    uchar prevMin = cap->min();
    cap->setMin(min);
    uchar prevMax = cap->max();
    cap->setMax(max);

    /* Check for overlapping values */
    foreach (QLCCapability* another, m_capabilities)
    {
        if (another == cap)
            continue;

        if (another->overlaps(cap) == true)
        {
            cap->setMin(prevMin);
            cap->setMax(prevMax);
            return false;
        }
    }

    return true;
}

bool QLCChannel::removeCapability(QLCCapability* cap)
{
    Q_ASSERT(cap != NULL);

    QMutableListIterator <QLCCapability*> it(m_capabilities);
    while (it.hasNext() == true)
    {
        if (it.next() == cap)
        {
            it.remove();
            delete cap;
            return true;
        }
    }

    return false;
}

static bool capsort(const QLCCapability* cap1, const QLCCapability* cap2)
{
    return (*cap1) < (*cap2);
}

void QLCChannel::sortCapabilities()
{
    std::sort(m_capabilities.begin(), m_capabilities.end(), capsort);
}

/*****************************************************************************
 * File operations
 *****************************************************************************/

bool QLCChannel::saveXML(QXmlStreamWriter *doc) const
{
    Q_ASSERT(doc != NULL);

    /* Channel entry */
    doc->writeStartElement(KXMLQLCChannel);
    doc->writeAttribute(KXMLQLCChannelName, m_name);

    if (m_defaultValue)
        doc->writeAttribute(KXMLQLCChannelDefault, QString::number(m_defaultValue));

    if (m_preset != Custom)
    {
        doc->writeAttribute(KXMLQLCChannelPreset, presetToString(m_preset));
        doc->writeEndElement();
        return true;
    }

    /* Group */
    doc->writeStartElement(KXMLQLCChannelGroup);
    /* Group control byte */
    doc->writeAttribute(KXMLQLCChannelGroupByte, QString::number(controlByte()));
    /* Group name */
    doc->writeCharacters(groupToString(m_group));
    doc->writeEndElement();

    /* Colour */
    if (m_colour != NoColour)
        doc->writeTextElement(KXMLQLCChannelColour, QLCChannel::colourToString(colour()));

    /* Capabilities */
    QListIterator <QLCCapability*> it(m_capabilities);
    while (it.hasNext() == true)
        it.next()->saveXML(doc);

    doc->writeEndElement();
    return true;
}

bool QLCChannel::loadXML(QXmlStreamReader &doc)
{
    if (doc.name() != KXMLQLCChannel)
    {
        qWarning() << "Channel node not found.";
        return false;
    }

    QXmlStreamAttributes attrs = doc.attributes();

    /* Get channel name */
    QString str = attrs.value(KXMLQLCChannelName).toString();
    if (str.isEmpty() == true)
        return false;
    setName(str);

    if (attrs.hasAttribute(KXMLQLCChannelDefault))
    {
        str = attrs.value(KXMLQLCChannelDefault).toString();
        setDefaultValue(uchar(str.toInt()));
    }

    if (attrs.hasAttribute(KXMLQLCChannelPreset))
    {
        str = attrs.value(KXMLQLCChannelPreset).toString();
        setPreset(stringToPreset(str));
        addPresetCapability();
    }

    /* Subtags */
    while (doc.readNextStartElement())
    {
        if (doc.name() == KXMLQLCCapability)
        {
            /* Create a new capability and attempt to load it */
            QLCCapability* cap = new QLCCapability();
            if (cap->loadXML(doc) == true)
            {
                /* Loading succeeded */
                if (addCapability(cap) == false)
                {
                    /* Value overlaps with existing value */
                    delete cap;
                }
            }
            else
            {
                /* Loading failed */
                delete cap;
                doc.skipCurrentElement();
            }
        }
        else if (doc.name() == KXMLQLCChannelGroup)
        {
            str = doc.attributes().value(KXMLQLCChannelGroupByte).toString();
            setControlByte(ControlByte(str.toInt()));
            setGroup(stringToGroup(doc.readElementText()));
        }
        else if (doc.name() == KXMLQLCChannelColour)
        {
            setColour(stringToColour(doc.readElementText()));
        }
        else
        {
            qWarning() << Q_FUNC_INFO << "Unknown Channel tag: " << doc.name();
            doc.skipCurrentElement();
        }
    }

    return true;
}
