/**
 ******************************************************************************
 *
 * @file       fieldtreeitem.h
 * @author     The OpenPilot Team, http://www.openpilot.org Copyright (C) 2010.
 * @addtogroup GCSPlugins GCS Plugins
 * @{
 * @addtogroup UAVObjectBrowserPlugin UAVObject Browser Plugin
 * @{
 * @brief The UAVObject Browser gadget plugin
 *****************************************************************************/
/*
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
 * or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
 * for more details.
 *
 * You should have received a copy of the GNU General Public License along
 * with this program; if not, write to the Free Software Foundation, Inc.,
 * 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
 */

#ifndef FIELDTREEITEM_H
#define FIELDTREEITEM_H

#include "treeitem.h"
#include <QtCore/QStringList>
#include <QWidget>
#include <QSpinBox>
#include <QDoubleSpinBox>
#include <qscispinbox/QScienceSpinBox.h>
#include <QComboBox>
#include <limits>

#define QINT8MIN   std::numeric_limits<qint8>::min()
#define QINT8MAX   std::numeric_limits<qint8>::max()
#define QUINTMIN   std::numeric_limits<quint8>::min()
#define QUINT8MAX  std::numeric_limits<quint8>::max()
#define QINT16MIN  std::numeric_limits<qint16>::min()
#define QINT16MAX  std::numeric_limits<qint16>::max()
#define QUINT16MAX std::numeric_limits<quint16>::max()
#define QINT32MIN  std::numeric_limits<qint32>::min()
#define QINT32MAX  std::numeric_limits<qint32>::max()
#define QUINT32MAX std::numeric_limits<qint32>::max()

class FieldTreeItem : public TreeItem {
    Q_OBJECT
public:

    FieldTreeItem(int index, const QList<QVariant> &data, UAVObjectField *field, TreeItem *parent = 0) :
        TreeItem(data, parent), m_index(index), m_field(field)
    {}

    FieldTreeItem(int index, const QVariant &data, UAVObjectField *field, TreeItem *parent = 0) :
        TreeItem(data, parent), m_index(index), m_field(field)
    {}

    bool isEditable()
    {
        return true;
    }

    virtual QWidget *createEditor(QWidget *parent)   = 0;
    virtual QVariant getEditorValue(QWidget *editor) = 0;
    virtual void setEditorValue(QWidget *editor, QVariant value) = 0;
    virtual void apply() {}
    virtual bool isKnown()
    {
        return parent()->isKnown();
    }


protected:
    int m_index;
    UAVObjectField *m_field;
};

class EnumFieldTreeItem : public FieldTreeItem {
    Q_OBJECT
public:
    EnumFieldTreeItem(UAVObjectField *field, int index, const QList<QVariant> &data, TreeItem *parent = 0) :
        FieldTreeItem(index, data, field, parent), m_enumOptions(field->getOptions())
    {}

    EnumFieldTreeItem(UAVObjectField *field, int index, const QVariant &data, TreeItem *parent = 0) :
        FieldTreeItem(index, data, field, parent), m_enumOptions(field->getOptions())
    {}

    void setData(QVariant value, int column)
    {
        QStringList options = m_field->getOptions();
        QVariant tmpValue   = m_field->getValue(m_index);
        int tmpValIndex     = options.indexOf(tmpValue.toString());

        setChanged(tmpValIndex != value);
        TreeItem::setData(value, column);
    }

    QString enumOptions(int index)
    {
        if ((index < 0) || (index >= m_enumOptions.length())) {
            return QString("Invalid Value (") + QString().setNum(index) + QString(")");
        }
        return m_enumOptions.at(index);
    }

    void apply()
    {
        int value = data().toInt();
        QStringList options = m_field->getOptions();

        m_field->setValue(options[value], m_index);
        setChanged(false);
    }

    void update()
    {
        QStringList options = m_field->getOptions();
        QVariant value = m_field->getValue(m_index);
        int valIndex   = options.indexOf(value.toString());

        if (data() != valIndex || changed()) {
            TreeItem::setData(valIndex);
            setHighlight(true);
        }
    }

    QWidget *createEditor(QWidget *parent)
    {
        QComboBox *editor = new QComboBox(parent);

        // Setting ClickFocus lets the ComboBox stay open on Mac OSX.
        editor->setFocusPolicy(Qt::ClickFocus);
        foreach(QString option, m_enumOptions) {
            editor->addItem(option);
        }
        return editor;
    }

    QVariant getEditorValue(QWidget *editor)
    {
        QComboBox *comboBox = static_cast<QComboBox *>(editor);

        return comboBox->currentIndex();
    }

    void setEditorValue(QWidget *editor, QVariant value)
    {
        QComboBox *comboBox = static_cast<QComboBox *>(editor);

        comboBox->setCurrentIndex(value.toInt());
    }

private:
    QStringList m_enumOptions;
};

class IntFieldTreeItem : public FieldTreeItem {
    Q_OBJECT
public:
    IntFieldTreeItem(UAVObjectField *field, int index, const QList<QVariant> &data, TreeItem *parent = 0) :
        FieldTreeItem(index, data, field, parent)
    {
        setMinMaxValues();
    }

    IntFieldTreeItem(UAVObjectField *field, int index, const QVariant &data, TreeItem *parent = 0) :
        FieldTreeItem(index, data, field, parent)
    {
        setMinMaxValues();
    }

    void setMinMaxValues()
    {
        switch (m_field->getType()) {
        case UAVObjectField::INT8:
            m_minValue = QINT8MIN;
            m_maxValue = QINT8MAX;
            break;
        case UAVObjectField::INT16:
            m_minValue = QINT16MIN;
            m_maxValue = QINT16MAX;
            break;
        case UAVObjectField::INT32:
            m_minValue = QINT32MIN;
            m_maxValue = QINT32MAX;
            break;
        case UAVObjectField::UINT8:
            m_minValue = QUINTMIN;
            m_maxValue = QUINT8MAX;
            break;
        case UAVObjectField::UINT16:
            m_minValue = QUINTMIN;
            m_maxValue = QUINT16MAX;
            break;
        case UAVObjectField::UINT32:
            m_minValue = QUINTMIN;
            m_maxValue = QUINT32MAX;
            break;
        default:
            Q_ASSERT(false);
            break;
        }
    }

    QWidget *createEditor(QWidget *parent)
    {
        QSpinBox *editor = new QSpinBox(parent);

        editor->setMinimum(m_minValue);
        editor->setMaximum(m_maxValue);
        return editor;
    }

    QVariant getEditorValue(QWidget *editor)
    {
        QSpinBox *spinBox = static_cast<QSpinBox *>(editor);

        spinBox->interpretText();
        return spinBox->value();
    }

    void setEditorValue(QWidget *editor, QVariant value)
    {
        QSpinBox *spinBox = static_cast<QSpinBox *>(editor);

        spinBox->setValue(value.toInt());
    }

    void setData(QVariant value, int column)
    {
        setChanged(m_field->getValue(m_index) != value);
        TreeItem::setData(value, column);
    }

    void apply()
    {
        m_field->setValue(data().toInt(), m_index);
        setChanged(false);
    }

    void update()
    {
        int value = m_field->getValue(m_index).toInt();

        if (data() != value || changed()) {
            TreeItem::setData(value);
            setHighlight(true);
        }
    }

private:
    int m_minValue;
    int m_maxValue;
};

class FloatFieldTreeItem : public FieldTreeItem {
    Q_OBJECT
public:
    FloatFieldTreeItem(UAVObjectField *field, int index, const QList<QVariant> &data, bool scientific = false, TreeItem *parent = 0) :
        FieldTreeItem(index, data, field, parent), m_useScientificNotation(scientific) {}

    FloatFieldTreeItem(UAVObjectField *field, int index, const QVariant &data, bool scientific = false, TreeItem *parent = 0) :
        FieldTreeItem(index, data, field, parent), m_useScientificNotation(scientific) {}

    void setData(QVariant value, int column)
    {
        setChanged(m_field->getValue(m_index) != value);
        TreeItem::setData(value, column);
    }

    void apply()
    {
        m_field->setValue(data().toDouble(), m_index);
        setChanged(false);
    }

    void update()
    {
        double value = m_field->getValue(m_index).toDouble();

        if (data() != value || changed()) {
            TreeItem::setData(value);
            setHighlight(true);
        }
    }

    QWidget *createEditor(QWidget *parent)
    {
        if (m_useScientificNotation) {
            QScienceSpinBox *editor = new QScienceSpinBox(parent);
            editor->setDecimals(6);
            editor->setMinimum(-std::numeric_limits<float>::max());
            editor->setMaximum(std::numeric_limits<float>::max());
            return editor;
        } else {
            QDoubleSpinBox *editor = new QDoubleSpinBox(parent);
            editor->setDecimals(8);
            editor->setMinimum(-std::numeric_limits<float>::max());
            editor->setMaximum(std::numeric_limits<float>::max());
            return editor;
        }
    }

    QVariant getEditorValue(QWidget *editor)
    {
        if (m_useScientificNotation) {
            QScienceSpinBox *spinBox = static_cast<QScienceSpinBox *>(editor);
            spinBox->interpretText();
            return spinBox->value();
        } else {
            QDoubleSpinBox *spinBox = static_cast<QDoubleSpinBox *>(editor);
            spinBox->interpretText();
            return spinBox->value();
        }
    }

    void setEditorValue(QWidget *editor, QVariant value)
    {
        if (m_useScientificNotation) {
            QScienceSpinBox *spinBox = static_cast<QScienceSpinBox *>(editor);
            spinBox->setValue(value.toDouble());
        } else {
            QDoubleSpinBox *spinBox = static_cast<QDoubleSpinBox *>(editor);
            spinBox->setValue(value.toDouble());
        }
    }

private:
    bool m_useScientificNotation;
};

class HexFieldTreeItem : public FieldTreeItem {
    Q_OBJECT
public:
    HexFieldTreeItem(UAVObjectField *field, int index, const QList<QVariant> &data, TreeItem *parent = 0) :
        FieldTreeItem(index, data, field, parent)
    {}

    HexFieldTreeItem(UAVObjectField *field, int index, const QVariant &data, TreeItem *parent = 0) :
        FieldTreeItem(index, data, field, parent)
    {}

    QWidget *createEditor(QWidget *parent)
    {
        QLineEdit *lineEdit = new QLineEdit(parent);

        lineEdit->setInputMask(QString(TreeItem::maxHexStringLength(m_field->getType()), 'H'));

        return lineEdit;
    }

    QVariant getEditorValue(QWidget *editor)
    {
        QLineEdit *lineEdit = static_cast<QLineEdit *>(editor);

        return lineEdit->text();
    }

    void setEditorValue(QWidget *editor, QVariant value)
    {
        QLineEdit *lineEdit = static_cast<QLineEdit *>(editor);

        lineEdit->setText(value.toString());
    }

    void setData(QVariant value, int column)
    {
        setChanged(m_field->getValue(m_index) != toUInt(value));
        TreeItem::setData(value, column);
    }

    void apply()
    {
        m_field->setValue(toUInt(data()), m_index);
        setChanged(false);
    }

    void update()
    {
        QVariant value = toHexString(m_field->getValue(m_index));

        if (data() != value || changed()) {
            TreeItem::setData(value);
            setHighlight(true);
        }
    }

private:
    QVariant toHexString(QVariant value)
    {
        QString str;
        bool ok;

        return str.setNum(value.toUInt(&ok), 16).toUpper();
    }

    QVariant toUInt(QVariant str)
    {
        bool ok;

        return str.toString().toUInt(&ok, 16);
    }
};

class CharFieldTreeItem : public FieldTreeItem {
    Q_OBJECT
public:
    CharFieldTreeItem(UAVObjectField *field, int index, const QList<QVariant> &data, TreeItem *parent = 0) :
        FieldTreeItem(index, data, field, parent)
    {}

    CharFieldTreeItem(UAVObjectField *field, int index, const QVariant &data, TreeItem *parent = 0) :
        FieldTreeItem(index, data, field, parent)
    {}

    QWidget *createEditor(QWidget *parent)
    {
        QLineEdit *lineEdit = new QLineEdit(parent);

        lineEdit->setInputMask(QString(1, 'N'));

        return lineEdit;
    }

    QVariant getEditorValue(QWidget *editor)
    {
        QLineEdit *lineEdit = static_cast<QLineEdit *>(editor);

        return lineEdit->text();
    }

    void setEditorValue(QWidget *editor, QVariant value)
    {
        QLineEdit *lineEdit = static_cast<QLineEdit *>(editor);

        lineEdit->setText(value.toString());
    }

    void setData(QVariant value, int column)
    {
        setChanged(m_field->getValue(m_index) != toUInt(value));
        TreeItem::setData(value, column);
    }

    void apply()
    {
        m_field->setValue(toUInt(data()), m_index);
        setChanged(false);
    }

    void update()
    {
        QVariant value = toChar(m_field->getValue(m_index));

        if (data() != value || changed()) {
            TreeItem::setData(value);
            setHighlight(true);
        }
    }

private:
    QVariant toChar(QVariant value)
    {
        return value.toChar();
    }

    QVariant toUInt(QVariant str)
    {
        return QVariant(str.toString().at(0).toLatin1());
    }
};

#endif // FIELDTREEITEM_H