#include "widget.h"
#include <QGridLayout>
#include <QPushButton>
#include <QLabel>

#include <QSqlQuery>
#include <qscatterseries.h>
#include <qchartview.h>
#include <qchartaxis.h>
#include <barchartseries.h>
#include <qbarcategory.h>
#include <qbarset.h>
#include <QListWidget>

QTCOMMERCIALCHART_USE_NAMESPACE

Widget::Widget(QWidget *parent)
    : QWidget(parent)
{    
    setGeometry(100, 100, 800, 600);

    // right panel layout
    countrieslist = new QListWidget;
    countrieslist->setSelectionMode(QAbstractItemView::MultiSelection);

    yearslist = new QListWidget;
    yearslist->setSelectionMode(QAbstractItemView::ExtendedSelection);
    for (int i = 1990; i < 2011; i++)
        yearslist->addItem(QString("%1").arg(i));

    QPushButton* refreshButton = new QPushButton(tr("Refresh"));
    connect(refreshButton, SIGNAL(clicked()), this, SLOT(refreshChart()));

    QVBoxLayout* rightPanelLayout = new QVBoxLayout;
    rightPanelLayout->addWidget(countrieslist);
    rightPanelLayout->addWidget(yearslist);
    rightPanelLayout->addWidget(refreshButton);
    rightPanelLayout->setStretch(0, 1);
    rightPanelLayout->setStretch(1, 0);

    // main layout
    chartArea = new QChartView(this);
    chartArea->setChartTitle("GDP by country");
    QGridLayout* mainLayout = new QGridLayout;
    mainLayout->addWidget(chartArea, 0, 0);
    mainLayout->addLayout(rightPanelLayout, 0, 1);
    mainLayout->setColumnStretch(0,1);
    setLayout(mainLayout);

    // connect to the database
    db = QSqlDatabase::addDatabase("QSQLITE");
    db.setDatabaseName("gdpData");
    if(!db.open())
    {
        qDebug() << "could not open database. SQLite db file missing (?)";
        return;
    }

    // get the list of all countires and regions.
    QSqlQuery query;
    query.exec("SELECT DISTINCT country FROM gdp2");

    // add the countries to the country filter
    while (query.next()) {
        countrieslist->addItem(query.value(0).toString());
    }

    // hide axis X labels
    QChartAxis newAxis = chartArea->defaultAxisX();
    newAxis.setLabelsVisible(false);
//    newAxis.setLabelsOrientation(QChartAxis::LabelsOrientationSlide);
    chartArea->setDefaultAxisX(newAxis);
}

Widget::~Widget()
{
    //
    db.close();
}

void Widget::refreshChart()
{
    // selected countries items list is not sorted. copy the values to QStringlist and sort them.
    QStringList selectedCountriesStrings;
    QList<QListWidgetItem*> selectedCountriesItems = countrieslist->selectedItems();
    for (int i = 0; i < selectedCountriesItems.size(); i++)
        selectedCountriesStrings.append(selectedCountriesItems[i]->text());
    selectedCountriesStrings.sort();

    // use the sorted selected coutries list to initialize BarCategory
    QBarCategory* category = new QBarCategory;
    for (int i = 0; i < selectedCountriesStrings.size(); i++)
        *category << selectedCountriesStrings[i];
    BarChartSeries* series0 = new BarChartSeries(category);

    // prepare the selected counries SQL query
    QString countriesQuery = "country IN (";
    for (int i = 0; i < selectedCountriesStrings.size(); i++)
    {
        countriesQuery.append("'" + selectedCountriesStrings[i] + "'");
        if ( i < selectedCountriesStrings.size() - 1)
            countriesQuery.append(",");
        else
            countriesQuery.append(")");
    }

    QSqlQuery query;
    // selected years items list is not sorted. copy the values to QList<Integer> and sort them.
    QList<int> selectedYearsInts;
    QList<QListWidgetItem*> selectedYearsItems = yearslist->selectedItems();
    for (int i = 0; i < selectedYearsItems.size(); i++)
        selectedYearsInts.append(selectedYearsItems[i]->text().toInt());
    qSort(selectedYearsInts);

    // perform a query for each selected year
    for (int i = 0; i < selectedYearsInts.size(); i++)
    {
        query.exec("SELECT country,gdpvalue FROM gdp2 where year=" + QString("%1").arg(selectedYearsInts[i]) + " AND " + countriesQuery);
        QBarSet* barSet = new QBarSet;
//        while (query.next()) {
//            qDebug() << query.value(0).toString() << " : " << query.value(1).toString();
//        }
        query.first();

        // the data for some of the coutries for some years might be missing.
        // QBarChart needs bars to have same size
        for (int k = 0; k < selectedCountriesStrings.size(); k++)
        {
            if (selectedCountriesStrings[k] == query.value(0).toString())
            {
                *barSet << query.value(1).toReal();
                qDebug() << query.value(0).toString() << query.value(1).toReal() << " : " << QString("%1").arg(selectedYearsInts[i]);
                query.next();
            }
            else
            {
                // data missing, put 0
                *barSet << 0.0f;
                qDebug() << "Putting 0 for Bosnia" << " : " << QString("%1").arg(selectedYearsInts[i]);
            }
        }
        series0->addBarSet(barSet);
    }

    // add the serie to the chart
    chartArea->addSeries(series0);
}