|
|
/****************************************************************************
|
|
|
**
|
|
|
** 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 "widget.h"
|
|
|
|
|
|
#include <QChart>
|
|
|
#include <QScatterSeries>
|
|
|
#include <QAxis>
|
|
|
#include <QBarSet>
|
|
|
#include <QBarSeries>
|
|
|
#include <QLegend>
|
|
|
|
|
|
#include <QGridLayout>
|
|
|
#include <QPushButton>
|
|
|
#include <QLabel>
|
|
|
#include <QListWidget>
|
|
|
#include <QPrinter>
|
|
|
#include <QPrintDialog>
|
|
|
#include <QRadioButton>
|
|
|
#include <QStringList>
|
|
|
#include <QSqlQuery>
|
|
|
#include <QDebug>
|
|
|
|
|
|
QTCOMMERCIALCHART_USE_NAMESPACE
|
|
|
|
|
|
Widget::Widget(QWidget *parent)
|
|
|
: QWidget(parent)
|
|
|
{
|
|
|
setGeometry(100, 100, 1000, 600);
|
|
|
|
|
|
// right panel layout
|
|
|
barChartRadioButton = new QRadioButton(tr("Bar chart"));
|
|
|
barChartRadioButton->setChecked(true);
|
|
|
scatterChartRadioButton = new QRadioButton(tr("Scatter chart"));
|
|
|
scatterChartRadioButton->setChecked(false);
|
|
|
countrieslist = new QListWidget;
|
|
|
countrieslist->setSelectionMode(QAbstractItemView::MultiSelection);
|
|
|
|
|
|
//list of years widget
|
|
|
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()));
|
|
|
|
|
|
QPushButton* printButton = new QPushButton(tr("Print to pdf"));
|
|
|
connect(printButton, SIGNAL(clicked()), this, SLOT(printChart()));
|
|
|
|
|
|
QVBoxLayout* rightPanelLayout = new QVBoxLayout;
|
|
|
rightPanelLayout->addWidget(barChartRadioButton);
|
|
|
rightPanelLayout->addWidget(scatterChartRadioButton);
|
|
|
rightPanelLayout->addWidget(countrieslist);
|
|
|
rightPanelLayout->addWidget(yearslist);
|
|
|
rightPanelLayout->addWidget(refreshButton);
|
|
|
rightPanelLayout->addWidget(printButton);
|
|
|
rightPanelLayout->setStretch(0, 1);
|
|
|
rightPanelLayout->setStretch(1, 0);
|
|
|
|
|
|
QChart *chart = new QChart();
|
|
|
chart->setTitle("GDP by country");
|
|
|
chart->legend()->setVisible(true);
|
|
|
|
|
|
// main layout
|
|
|
chartView = new QChartView(chart);
|
|
|
QGridLayout* mainLayout = new QGridLayout;
|
|
|
mainLayout->addWidget(chartView, 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());
|
|
|
}
|
|
|
}
|
|
|
|
|
|
Widget::~Widget()
|
|
|
{
|
|
|
//
|
|
|
db.close();
|
|
|
}
|
|
|
|
|
|
/*!
|
|
|
refreshes the chart
|
|
|
*/
|
|
|
void Widget::refreshChart()
|
|
|
{
|
|
|
chartView->chart()->removeAllSeries();
|
|
|
|
|
|
// 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();
|
|
|
|
|
|
QSqlQuery query;
|
|
|
// selected years items list is not sorted. copy the values to QList<int> 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.begin(), selectedYearsInts.end());
|
|
|
|
|
|
if (barChartRadioButton->isChecked())
|
|
|
{
|
|
|
// use the sorted selected coutries list to initialize BarCategory
|
|
|
QStringList category;
|
|
|
for (int i = 0; i < selectedCountriesStrings.size(); i++)
|
|
|
category << selectedCountriesStrings[i];
|
|
|
QBarSeries* series0 = new QBarSeries();
|
|
|
series0->setCategories(category);
|
|
|
// series0 = new QBarSeries(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(")");
|
|
|
}
|
|
|
|
|
|
// 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(QString::number(selectedYearsInts[i]));
|
|
|
|
|
|
// 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 the missing data" << " : " << QString("%1").arg(selectedYearsInts[i]);
|
|
|
}
|
|
|
}
|
|
|
series0->append(barSet);
|
|
|
}
|
|
|
// add the serie to the chart
|
|
|
chartView->chart()->addSeries(series0);
|
|
|
}
|
|
|
else if (scatterChartRadioButton->isChecked())
|
|
|
{
|
|
|
QString yearsQuery = "year IN (";
|
|
|
for (int i = 0; i < selectedYearsInts.size(); i++)
|
|
|
{
|
|
|
yearsQuery.append("'" + QString("%1").arg(selectedYearsInts[i]) + "'");
|
|
|
if ( i < selectedYearsInts.size() - 1)
|
|
|
yearsQuery.append(",");
|
|
|
else
|
|
|
yearsQuery.append(")");
|
|
|
}
|
|
|
|
|
|
// perform a query for each selected country
|
|
|
for (int i = 0; i < selectedCountriesStrings.size(); i++)
|
|
|
{
|
|
|
query.exec("SELECT year,gdpvalue FROM gdp2 where country='" + selectedCountriesStrings[i] + "' AND " + yearsQuery);
|
|
|
query.first();
|
|
|
|
|
|
QScatterSeries* series = new QScatterSeries;
|
|
|
series->setName(selectedCountriesStrings[i]);
|
|
|
// the data for some of the coutries for some years might be missing.
|
|
|
for (int k = 0; k < selectedYearsInts.size(); k++)
|
|
|
{
|
|
|
if (selectedYearsInts[k] == query.value(0).toInt())
|
|
|
{
|
|
|
*series << QPointF(query.value(0).toInt() , query.value(1).toReal());
|
|
|
qDebug() << query.value(0).toString() << query.value(1).toReal() << " : " << QString("%1").arg(selectedYearsInts[k]);
|
|
|
query.next();
|
|
|
}
|
|
|
else
|
|
|
{
|
|
|
// data missing, put 0
|
|
|
*series << QPointF(selectedYearsInts[k] , 0.0f);
|
|
|
qDebug() << "Putting 0 for the missing data" << " : " << QString("%1").arg(selectedYearsInts[i]) << " " << query.value(0).toInt();
|
|
|
}
|
|
|
}
|
|
|
// chartArea->axisX()->setRange(selectedYearsInts[selectedYearsInts.size() - 1] + 1, selectedYearsInts[0] - 1);
|
|
|
chartView->chart()->addSeries(series);
|
|
|
}
|
|
|
chartView->chart()->axisX()->setRange(selectedYearsInts[selectedYearsInts.size() - 1] - 1, selectedYearsInts[0] + 1);
|
|
|
}
|
|
|
}
|
|
|
|
|
|
void Widget::printChart()
|
|
|
{
|
|
|
QPrinter printer;
|
|
|
// QPrinter printer(QPrinter::HighResolution);
|
|
|
printer.setOutputFormat(QPrinter::PdfFormat);
|
|
|
printer.setOrientation(QPrinter::Landscape);
|
|
|
printer.setOutputFileName("print.pdf");
|
|
|
|
|
|
QPainter painter;
|
|
|
painter.begin(&printer);
|
|
|
chartView->render(&painter);
|
|
|
}
|
|
|
|
