/****************************************************************************
**
** Copyright (C) 2012 Digia Plc
** All rights reserved.
** For any questions to Digia, please use contact form at http://qt.digia.com
**
** This file is part of the Qt Commercial Charts Add-on.
**
** $QT_BEGIN_LICENSE$
** Licensees holding valid Qt Commercial licenses may use this file in
** accordance with the Qt Commercial License Agreement provided with the
** Software or, alternatively, in accordance with the terms contained in
** a written agreement between you and Digia.
**
** If you have questions regarding the use of this file, please use
** contact form at http://qt.digia.com
** $QT_END_LICENSE$
**
****************************************************************************/

#include "tablewidget.h"
#include <QGridLayout>
#include <QTableView>
#include <QChart>
#include <QStyledItemDelegate>
#include <QLineSeries>
#include <QSplineSeries>
#include <QScatterSeries>
#include <QXYModelMapper>
#include "customtablemodel.h"
#include <QPieSeries>
#include <QPieModelMapper>
#include <QPieSlice>
#include <QAreaSeries>
#include <QBarSeries>
#include <QGroupedBarSeries>
#include <QBarSet>
#include <QBarModelMapper>
#include <QPushButton>
#include <QRadioButton>
#include <QLabel>
#include <QSpinBox>
#include <QTime>
#include <QHeaderView>

TableWidget::TableWidget(QWidget *parent)
    : QWidget(parent)
    //      specialPie(0)
{
    setGeometry(1900, 100, 1000, 600);
    qsrand(QTime(0,0,0).secsTo(QTime::currentTime()));
    // create simple model for storing data
    // user's table data model
    m_model = new CustomTableModel;
    m_tableView = new QTableView;
    m_tableView->setModel(m_model);
    //    m_tableView->setMinimumHeight(300);
    m_tableView->horizontalHeader()->setResizeMode(QHeaderView::Stretch);
    m_tableView->verticalHeader()->setResizeMode(QHeaderView::Stretch);

    m_chart = new QChart;
    m_chart->legend()->setVisible(true);
    m_chart->setAnimationOptions(QChart::SeriesAnimations);
    m_chartView = new QChartView(m_chart);
    m_chartView->setRenderHint(QPainter::Antialiasing);
    m_chartView->setMinimumSize(640, 480);

    // add, remove data buttons
    QPushButton* addRowAboveButton = new QPushButton("Add row above");
    connect(addRowAboveButton, SIGNAL(clicked()), this, SLOT(addRowAbove()));

    QPushButton* addRowBelowButton = new QPushButton("Add row below");
    connect(addRowBelowButton, SIGNAL(clicked()), this, SLOT(addRowBelow()));

    QPushButton* removeRowButton = new QPushButton("Remove row");
    connect(removeRowButton, SIGNAL(clicked()), this, SLOT(removeRow()));

    QPushButton* addColumnRightButton = new QPushButton("Add column to the right");
    connect(addColumnRightButton, SIGNAL(clicked()), this, SLOT(addColumnRight()));

    QPushButton* removeColumnButton = new QPushButton("Remove column");
    connect(removeColumnButton, SIGNAL(clicked()), this, SLOT(removeColumn()));

    QPushButton* specialPieButton = new QPushButton("Test pie");
    connect(specialPieButton, SIGNAL(clicked()), this, SLOT(testPie()));


    //    QLabel *spinBoxLabel = new QLabel("Rows affected:");

    // spin box for setting number of affected items (add, remove)
    m_linesCountSpinBox = new QSpinBox;
    m_linesCountSpinBox->setRange(1, 10);
    m_linesCountSpinBox->setValue(1);

    // buttons layout
    QVBoxLayout* buttonsLayout = new QVBoxLayout;
    //    buttonsLayout->addWidget(spinBoxLabel);
    //    buttonsLayout->addWidget(m_linesCountSpinBox);
    //    buttonsLayout->addWidget(addRowAboveButton);
    buttonsLayout->addWidget(addRowBelowButton);
    buttonsLayout->addWidget(removeRowButton);
    //    buttonsLayout->addWidget(addColumnRightButton);
    //    buttonsLayout->addWidget(removeColumnButton);
        buttonsLayout->addWidget(specialPieButton);
    buttonsLayout->addStretch();

    // chart type radio buttons
    m_lineRadioButton = new QRadioButton("Line");
    m_splineRadioButton = new QRadioButton("Spline");
    m_scatterRadioButton = new QRadioButton("Scatter");
    m_pieRadioButton = new QRadioButton("Pie");
    m_areaRadioButton = new QRadioButton("Area");
    m_barRadioButton = new QRadioButton("Bar");

    connect(m_lineRadioButton, SIGNAL(toggled(bool)), this, SLOT(updateChartType(bool)));
    connect(m_splineRadioButton, SIGNAL(toggled(bool)), this, SLOT(updateChartType(bool)));
    connect(m_scatterRadioButton, SIGNAL(toggled(bool)), this, SLOT(updateChartType(bool)));
    connect(m_pieRadioButton, SIGNAL(toggled(bool)), this, SLOT(updateChartType(bool)));
    connect(m_areaRadioButton, SIGNAL(toggled(bool)), this, SLOT(updateChartType(bool)));
    connect(m_barRadioButton, SIGNAL(toggled(bool)), this, SLOT(updateChartType(bool)));
    m_lineRadioButton->setChecked(true);

    // radio buttons layout
    QVBoxLayout* radioLayout = new QVBoxLayout;
    radioLayout->addWidget(m_lineRadioButton);
        radioLayout->addWidget(m_splineRadioButton);
    //    radioLayout->addWidget(m_scatterRadioButton);
    radioLayout->addWidget(m_pieRadioButton);
    //    radioLayout->addWidget(m_areaRadioButton);
    radioLayout->addWidget(m_barRadioButton);
    radioLayout->addStretch();

    // create main layout
    QGridLayout* mainLayout = new QGridLayout;
    mainLayout->addLayout(buttonsLayout, 2, 0);
        mainLayout->addLayout(radioLayout, 3, 0);
    mainLayout->addWidget(m_tableView, 1, 0);
    mainLayout->addWidget(m_chartView, 1, 1, 2, 1);
    setLayout(mainLayout);
    m_lineRadioButton->setFocus();
}

void TableWidget::addRowAbove()
{
    m_model->insertRows(m_tableView->currentIndex().row(), m_linesCountSpinBox->value());

}

void TableWidget::addRowBelow()
{
    m_model->insertRows(m_tableView->currentIndex().row() + 1, m_linesCountSpinBox->value());

}

void TableWidget::removeRow()
{
    m_model->removeRows(m_tableView->currentIndex().row(), qMin(m_model->rowCount() - m_tableView->currentIndex().row(), m_linesCountSpinBox->value()));
}

void TableWidget::addColumnRight()
{
    m_model->insertColumns(m_tableView->currentIndex().column() + 1, m_linesCountSpinBox->value());
}

void TableWidget::removeColumn()
{
    m_model->removeColumns(m_tableView->currentIndex().column(), qMin(m_model->columnCount() - m_tableView->currentIndex().column(), m_linesCountSpinBox->value()));
}

void TableWidget::updateChartType(bool toggle)
{
    // this if is needed, so that the function is only called once.
    // For the radioButton that was enabled.
    if (toggle) {
        //        specialPie = 0;
        m_chart->removeAllSeries();
        m_chart->axisX()->setNiceNumbersEnabled(false);
        m_chart->axisY()->setNiceNumbersEnabled(false);

        // renable axes of the chart (pie hides them)
        // x axis
        QAxis *axis = m_chart->axisX();
        axis->setAxisVisible(true);
        axis->setGridLineVisible(true);
        axis->setLabelsVisible(true);

        // y axis
        axis = m_chart->axisY();
        axis->setAxisVisible(true);
        axis->setGridLineVisible(true);
        axis->setLabelsVisible(true);

        m_model->clearMapping();

        QString seriesColorHex = "#000000";
        QPen pen;
        pen.setWidth(2);

        if (m_lineRadioButton->isChecked())
        {
            m_chart->setAnimationOptions(QChart::NoAnimation);

            // series 1
            m_series = new QLineSeries;
            m_series->setModel(m_model);

            QXYModelMapper *mapper = new QXYModelMapper;
            mapper->setMapX(0);
            mapper->setMapY(1);
            mapper->setFirst(3);
            mapper->setCount(4);
            m_series->setModelMapper(mapper);
            //            m_series->setModelMapping(0,1, Qt::Vertical);
            //            m_series->setModelMappingRange(3, 4);
            m_chart->addSeries(m_series);
            seriesColorHex = "#" + QString::number(m_series->pen().color().rgb(), 16).right(6).toUpper();
            m_model->addMapping(seriesColorHex, QRect(0, 3, 2, 4));

            // series 2
            m_series = new QLineSeries;
            m_series->setModel(m_model);

            mapper = new QXYModelMapper;
            mapper->setMapX(3);
            mapper->setMapY(4);
            //            mapper->setFirst(3);
            //            mapper->setCount(4);
            m_series->setModelMapper(mapper);
            //            m_series->setModelMapping(2,3, Qt::Vertical);
            m_chart->addSeries(m_series);
            seriesColorHex = "#" + QString::number(m_series->pen().color().rgb(), 16).right(6).toUpper();
            m_model->addMapping(seriesColorHex, QRect(3, 0, 2, 1000));

            // series 3
            m_series = new QLineSeries;
            m_series->setModel(m_model);

            mapper = new QXYModelMapper;
            mapper->setMapX(5);
            mapper->setMapY(6);
            mapper->setFirst(2);
            mapper->setCount(-1);
            m_series->setModelMapper(mapper);
            //            m_series->setModelMapping(4,5, Qt::Vertical);
            //            m_series->setModelMappingRange(2, -1);
            m_chart->addSeries(m_series);
            seriesColorHex = "#" + QString::number(m_series->pen().color().rgb(), 16).right(6).toUpper();
            m_model->addMapping(seriesColorHex, QRect(5, 2, 2, 1000));
        }
        else if (m_splineRadioButton->isChecked())
        {
            m_chart->setAnimationOptions(QChart::NoAnimation);

            // series 1
            m_series = new QSplineSeries;
            m_series->setModel(m_model);

            QXYModelMapper *mapper = new QXYModelMapper;
            mapper->setMapX(0);
            mapper->setMapY(1);
            mapper->setFirst(0);
            mapper->setCount(-1);

            m_series->setModelMapper(mapper);

            m_chart->addSeries(m_series);
            seriesColorHex = "#" + QString::number(m_series->pen().color().rgb(), 16).right(6).toUpper();
            m_model->addMapping(seriesColorHex, QRect(0, 0, 2, 1000));

            // series 2
            m_series = new QSplineSeries;
            m_series->setModel(m_model);

            mapper = new QXYModelMapper;
            mapper->setMapX(2);
            mapper->setMapY(3);
            mapper->setFirst(2);
            mapper->setCount(4);

            m_series->setModelMapper(mapper);

            m_chart->addSeries(m_series);
            seriesColorHex = "#" + QString::number(m_series->pen().color().rgb(), 16).right(6).toUpper();
            m_model->addMapping(seriesColorHex, QRect(2, 2, 2, 4));

            // series 3
            m_series = new QSplineSeries;
            m_series->setModel(m_model);

            mapper = new QXYModelMapper;
            mapper->setMapX(4);
            mapper->setMapY(5);
            mapper->setFirst(2);
            mapper->setCount(-1);

            m_series->setModelMapper(mapper);

            m_chart->addSeries(m_series);
            seriesColorHex = "#" + QString::number(m_series->pen().color().rgb(), 16).right(6).toUpper();
            m_model->addMapping(seriesColorHex, QRect(4, 2, 2, 1000));
        }
        //        else if (m_scatterRadioButton->isChecked())
        //        {
        //            m_chart->setAnimationOptions(QChart::NoAnimation);

        //            // series 1
        //            m_series = new QScatterSeries;
        //            m_series->setModel(m_model);
        //            m_series->setModelMapping(0,1, Qt::Vertical);
        //            //            m_series->setModelMappingRange(2, 0);
        //            //        series->setModelMapping(0,1, Qt::Horizontal);
        //            m_chart->addSeries(m_series);

        //            seriesColorHex = "#" + QString::number(m_series->brush().color().rgb(), 16).right(6).toUpper();
        //            m_model->addMapping(seriesColorHex, QRect(0, 2, 2, 1000));

        //            // series 2
        //            m_series = new QScatterSeries;
        //            m_series->setModel(m_model);
        //            m_series->setModelMapping(2,3, Qt::Vertical);
        //            //            m_series->setModelMappingRange(1, 6);
        //            //        series->setModelMapping(2,3, Qt::Horizontal);
        //            m_chart->addSeries(m_series);

        //            seriesColorHex = "#" + QString::number(m_series->brush().color().rgb(), 16).right(6).toUpper();
        //            m_model->addMapping(seriesColorHex, QRect(2, 1, 2, 6));

        //            // series 3
        //            m_series = new QScatterSeries;
        //            m_series->setModel(m_model);
        //            m_series->setModelMapping(4,5, Qt::Vertical);
        //            //        series->setModelMapping(4,5, Qt::Horizontal);
        //            m_chart->addSeries(m_series);
        //            seriesColorHex = "#" + QString::number(m_series->brush().color().rgb(), 16).right(6).toUpper();
        //            m_model->addMapping(seriesColorHex, QRect(4, 0, 2, 1000));
        //        }
                else if (m_pieRadioButton->isChecked())
                {
                    m_chart->setAnimationOptions(QChart::SeriesAnimations);

                    // pie 1
                    QPieSeries* pieSeries = new QPieSeries;
                    pieSeries->setModel(m_model);

                    QPieModelMapper *mapper = new QPieModelMapper;
                    mapper->setMapValues(1);
                    mapper->setMapLabels(1);
                    mapper->setFirst(2);
                    mapper->setCount(5);
                    pieSeries->setModelMapper(mapper);

                    pieSeries->setLabelsVisible(true);
                    pieSeries->setPieSize(0.75);
        //            pieSeries->setHorizontalPosition(0.2);
        //            pieSeries->setVerticalPosition(0.3);

                    m_chart->addSeries(pieSeries);
                    seriesColorHex = "#" + QString::number(pieSeries->slices().at(pieSeries->slices().count()/2)->brush().color().rgb(), 16).right(6).toUpper();
                    m_model->addMapping(seriesColorHex, QRect(0, 2, 2, 5));

        //            // pie 2
        //            pieSeries = new QPieSeries;
        //            pieSeries->setModel(m_model);

        //            pieSeries->setModelMapping(1,1, Qt::Vertical);
        //            pieSeries->setModelMappingRange(2, -1);
        //            pieSeries->setLabelsVisible(true);
        //            pieSeries->setPieSize(0.35);
        //            pieSeries->setHorizontalPosition(0.8);
        //            pieSeries->setVerticalPosition(0.3);
        //            m_chart->addSeries(pieSeries);
        //            seriesColorHex = "#" + QString::number(pieSeries->slices().at(pieSeries->slices().count()/2)->brush().color().rgb(), 16).right(6).toUpper();
        //            m_model->addMapping(seriesColorHex, QRect(1, 2, 1, 1000));

        //            // pie 3
        //            pieSeries = new QPieSeries;
        //            pieSeries->setModel(m_model);
        //            pieSeries->setModelMapping(2,2, Qt::Vertical);
        //            pieSeries->setLabelsVisible(true);
        //            pieSeries->setPieSize(0.35);
        //            pieSeries->setHorizontalPosition(0.5);
        //            pieSeries->setVerticalPosition(0.75);
        //            m_chart->addSeries(pieSeries);
        //            seriesColorHex = "#" + QString::number(pieSeries->slices().at(pieSeries->slices().count()/2)->brush().color().rgb(), 16).right(6).toUpper();
        //            m_model->addMapping(seriesColorHex, QRect(2, 0, 1, 1000));

                    //            // special pie
                    //            specialPie = new QPieSeries;
                    //            specialPie->append(17, "1");
                    //            specialPie->append(45, "2");
                    //            specialPie->append(77, "3");
                    //            specialPie->append(37, "4");
                    //            specialPie->append(27, "5");
                    //            specialPie->append(47, "6");
                    //            specialPie->setPieSize(0.35);
                    //            specialPie->setHorizontalPosition(0.8);
                    //            specialPie->setVerticalPosition(0.75);
                    //            specialPie->setLabelsVisible(true);
                    //            m_chart->addSeries(specialPie);
                }
        //        else if (m_areaRadioButton->isChecked())
        //        {
        //            m_chart->setAnimationOptions(QChart::NoAnimation);

        //            QLineSeries* upperLineSeries = new QLineSeries;
        //            upperLineSeries->setModel(m_model);
        //            upperLineSeries->setModelMapping(0, 1, Qt::Vertical);
        //            //            upperLineSeries->setModelMappingRange(1, 5);
        //            QLineSeries* lowerLineSeries = new QLineSeries;
        //            lowerLineSeries->setModel(m_model);
        //            lowerLineSeries->setModelMapping(2, 3, Qt::Vertical);
        //            QAreaSeries* areaSeries = new QAreaSeries(upperLineSeries, lowerLineSeries);
        //            m_chart->addSeries(areaSeries);
        //            seriesColorHex = "#" + QString::number(areaSeries->brush().color().rgb(), 16).right(6).toUpper();
        //            m_model->addMapping(seriesColorHex, QRect(0, 1, 2, 5));
        //            m_model->addMapping(seriesColorHex, QRect(2, 0, 2, 1000));
        //        }
                else if (m_barRadioButton->isChecked())
                {
                    m_chart->setAnimationOptions(QChart::SeriesAnimations);

                    QGroupedBarSeries* barSeries = new QGroupedBarSeries();
                    barSeries->setCategories(QStringList());
                    barSeries->setModel(m_model);
                    //            barSeries->setModelMappingRange(2, 5);
//                    barSeries->setModelMapping(5, 2, 4, Qt::Vertical);

                    QBarModelMapper *mapper = new QBarModelMapper;
                    mapper->setMapCategories(5);
                    mapper->setMapBarBottom(2);
                    mapper->setMapBarTop(4);
                    barSeries->setModelMapper(mapper);
                    m_chart->addSeries(barSeries);
                    QList<QBarSet*> barsets = barSeries->barSets();
                    for (int i = 0; i < barsets.count(); i++) {
                        seriesColorHex = "#" + QString::number(barsets.at(i)->brush().color().rgb(), 16).right(6).toUpper();
                        m_model->addMapping(seriesColorHex, QRect(2 + i, 0, 1, 1000));
                    }
                }


                if (!m_barRadioButton->isChecked()) {
                    m_chart->axisX()->setRange(0, 500);
                    m_chart->axisY()->setRange(0, 220);
                }
        m_chart->legend()->setVisible(true);

        // repaint table view colors
        m_tableView->repaint();
        m_tableView->setFocus();
    }
}

void TableWidget::testPie()
{
    m_series->modelMapper()->setMapX(4);
//    m_tableView->setColumnWidth(10, 250);
    //    if (specialPie) {
    //        specialPie->remove(specialPie->slices().at(2));
    //        //    specialPie->insert(4, new QPieSlice(45, "Hello"));//specialPie->slices.at(2));
    //        specialPie->append(4, "heloo");
    //    }
}

TableWidget::~TableWidget()
{

}