KissCount/src/view/StatsPanel.cpp

362 lines
10 KiB
C++

/*
Copyright 2010-2012 Grégory Soutadé
This file is part of KissCount.
KissCount 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.
KissCount 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 KissCount. If not, see <http://www.gnu.org/licenses/>.
*/
#include <QMessageBox>
#include <QLabel>
#include <QtCharts/QLineSeries>
#include <QtCharts/QCategoryAxis>
#include <QtCharts/QDateTimeAxis>
#include "StatsPanel.hpp"
using namespace QtCharts;
StatsPanel::StatsPanel(KissCount* kiss, wxUI *parent, bool lowResolution) :
KissPanel(kiss, parent, lowResolution), _chartView(0)
{
QHBoxLayout *hbox = new QHBoxLayout();
QHBoxLayout *hbox1 = new QHBoxLayout();
QVBoxLayout *vbox = new QVBoxLayout();
_hbox2 = new QHBoxLayout();
_vbox2 = new QVBoxLayout();
_vbox3 = new QVBoxLayout();
int i, nbCategories;
User* user = _kiss->GetUser();
std::vector<Account>::iterator accountIt;
std::vector<Category>::iterator categoryIt;
std::map<int, std::vector<int> > operations;
std::map<int, std::vector<int> >::iterator it;
_icons[KissPanel::LOW_RES_ICON] = STATS_LOW_ICON;
_icons[KissPanel::HIGH_RES_ICON] = STATS_ICON;
setLayout(vbox);
_monthFrom = new QComboBox(parent);
for(i=0; i<12; i++)
_monthFrom->addItem(wxUI::months[i]);
_yearFrom = new QComboBox(parent);
_monthTo = new QComboBox(parent);
for(i=0; i<12; i++)
_monthTo->addItem(wxUI::months[i]);
_yearTo = new QComboBox(parent);
operations = _kiss->GetAllOperations();
for(i=0, it = operations.begin(); it != operations.end(); it++, i++)
{
_yearFrom->addItem(QString::number(it->first));
_yearTo->addItem(QString::number(it->first));
}
if (i) i--;
_yearFrom->setCurrentIndex(i);
_yearTo->setCurrentIndex(i);
_monthFrom->setCurrentIndex(0);
_monthTo->setCurrentIndex(11);
QLabel* label = new QLabel(_("From"));
label->setAlignment(Qt::AlignRight);
hbox->addWidget(label, 1);
hbox->addWidget(_monthFrom, 2);
hbox->addWidget(_yearFrom, 2);
hbox->addStretch(1);
label = new QLabel(_("To"));
label->setAlignment(Qt::AlignRight);
hbox->addWidget(label, 1);
hbox->addWidget(_monthTo, 2);
hbox->addWidget(_yearTo, 2);
hbox->addStretch(1);
connect(_yearFrom, SIGNAL(currentIndexChanged(int)), this, SLOT(OnRangeChange(int)));
connect(_monthFrom, SIGNAL(currentIndexChanged(int)), this, SLOT(OnRangeChange(int)));
connect(_yearTo, SIGNAL(currentIndexChanged(int)), this, SLOT(OnRangeChange(int)));
connect(_monthTo, SIGNAL(currentIndexChanged(int)), this, SLOT(OnRangeChange(int)));
_account = new QListWidget(this);
for(i=0, accountIt = user->_accounts.begin(); accountIt != user->_accounts.end(); accountIt++, i++)
{
_account->addItem(accountIt->name);
}
_account->setSizePolicy(QSizePolicy::Minimum, QSizePolicy::Minimum);
_account->setSizeAdjustPolicy(QAbstractScrollArea::AdjustToContents);
connect(_account, SIGNAL(itemClicked(QListWidgetItem*)), this, SLOT(OnAccountChange(QListWidgetItem*)));
_categories = new QString[user->GetCategoriesNumber()] ;
for(i=0, categoryIt = user->_categories.begin();
categoryIt != user->_categories.end();
categoryIt++, i++)
{
_categoriesIndexes[categoryIt->id] = i;
_categories[i] = _(categoryIt->name.toStdString().c_str()) ;
}
DEFAULT_FONT(font);
_costRepartitionBanner = new CostRepartitionBanner(_kiss, this, _categories);
hbox1->addStretch(1);
hbox1->addWidget(_account, 2);
hbox1->addStretch(1);
_vbox3->addLayout(hbox1, 1);
nbCategories = user->GetCategoriesNumber();
_categoriesValues = new int[nbCategories];
for(i=0; i<nbCategories; i++)
_categoriesValues[i] = 0.0;
_vbox3->addWidget(_costRepartitionBanner, 3);
_vbox2->addLayout(hbox);
_hbox2->addLayout(_vbox2);
vbox->addLayout(_hbox2);
_hbox2->addLayout(_vbox3);
_hbox2->setStretch(0, 4);
_hbox2->setStretch(1, 1);
OnRangeChange(0);
layout();
}
KissPanel* StatsPanel::CreatePanel()
{
return new StatsPanel(_kiss, _wxUI, _lowResolution);
}
QString StatsPanel::GetToolTip()
{
return _("Statistics");
}
void StatsPanel::UpdateStats(int monthFrom, int yearFrom, int monthTo, int yearTo)
{
std::map<int, std::map<int, std::map<int, int> > > accountAmounts;
std::map<int, int> categories;
std::map<int, int> tags;
std::map<int, std::vector<int> > operations;
std::vector<Account>::iterator accountIt;
std::map<int, int>::iterator categoriesIt;
std::map<int, std::map<int, int> >::iterator accountYearIt;
int total;
int account, size, i, a, b, nbDays;
QString value, v;
User* user = _kiss->GetUser();
QDate date;
bool failed;
QLineSeries * series = new QLineSeries();
QLineSeries * series0 = series;
QDateTimeAxis *axisX = new QDateTimeAxis;
QDateTime xValue;
if (_chartView)
{
_vbox2->removeWidget(_chartView);
delete _chartView;
}
_plot = new QChart();
_plot->createDefaultAxes();
if (monthFrom == monthTo && yearFrom == yearTo)
{
date.setDate(yearFrom, monthFrom+1, 1);
nbDays = date.daysInMonth();
_kiss->GetMonthStats(monthFrom, yearFrom, nbDays, &operations, &categories, &tags);
// Line on 0 all over the years
for (a=0; a<nbDays; a++)
{
date = QDate(yearFrom, monthFrom+1, a+1);
xValue.setDate(date);
series->append(xValue.toMSecsSinceEpoch(), 0.0);
}
series->setName(_("0 line"));
series->setColor(Qt::red);
axisX->setFormat("dd");
_plot->addSeries(series);
i=0;
for (account = 0, i = 0, accountIt = user->_accounts.begin(); accountIt != user->_accounts.end();
account++, accountIt++)
{
if (_account->item(account)->checkState() != Qt::Checked)
continue;
series = new QLineSeries();
for (a=0; a<nbDays; a++)
{
date = QDate(yearFrom, monthFrom+1, a+1);
xValue.setDate(date);
series->append(xValue.toMSecsSinceEpoch(),
(double) operations[accountIt->id][a] / 100);
}
series->setName(user->GetAccountName(accountIt->id));
_plot->addSeries(series);
}
}
else
{
_kiss->GetStats(monthFrom, yearFrom, monthTo, yearTo, &accountAmounts, &categories, &tags);
// Line on 0 all over the years
nbDays = ((yearTo - yearFrom) + 1) * 12;
for (a=0; a<(nbDays/12); a++)
{
for(b=0; b<12; b++)
{
date = QDate(yearFrom+a, monthFrom+b+1, 1);
xValue.setDate(date);
series->append(xValue.toMSecsSinceEpoch(), 0.0);
}
}
axisX->setLabelsAngle(-60);
axisX->setLineVisible(true);
axisX->setFormat("MM yyyy");
series->setName(_("0 line"));
series->setColor(Qt::red);
_plot->addSeries(series);
for (account = 0, i = 0, accountIt = user->_accounts.begin(); accountIt != user->_accounts.end();
account++, accountIt++, i++)
{
if (_account->item(account)->checkState() != Qt::Checked)
{
i-- ;
continue;
}
series = new QLineSeries();
for(a = 0, accountYearIt = accountAmounts[accountIt->id].begin();
accountYearIt != accountAmounts[accountIt->id].end();
accountYearIt++, a++)
{
for(b = 0; b<=12; b++)
{
date = QDate(yearFrom+a, b+1, 1);
xValue.setDate(date);
if (!accountAmounts[accountIt->id][accountYearIt->first].count(b))
{
/*
If previously failed, continue to avoid to set
account to 0 (only for display)
*/
if (!b || failed) continue;
/*
Compute cur month value (if there are operations)
as next month value
*/
*series << QPointF(xValue.toMSecsSinceEpoch(),
(double) (accountAmounts[accountIt->id][accountYearIt->first][b-1]
+ _kiss->CalcAccountAmount(accountIt->id, b-1, accountYearIt->first, 0)) / 100);
failed = true;
}
else
{
*series << QPointF(xValue.toMSecsSinceEpoch(),
(double) accountAmounts[accountIt->id][accountYearIt->first][b] / 100);
failed = false;
}
size++;
}
}
series->setName(user->GetAccountName(accountIt->id));
_plot->addSeries(series);
}
}
// yAxis->setTitleText (_("Amount"));
QLegend* legend = _plot->legend();
legend->setAlignment(Qt::AlignRight);
_plot->createDefaultAxes();
_plot->setAxisX(axisX, series0);
_chartView = new QChartView(_plot);
_chartView->setRenderHint(QPainter::Antialiasing);
_vbox2->addWidget(_chartView);
total = 0.0;
for(i=0, categoriesIt = categories.begin(); categoriesIt != categories.end(); categoriesIt++, i++)
{
_categoriesValues[_categoriesIndexes[categoriesIt->first]] = categoriesIt->second;
total += categoriesIt->second;
}
_costRepartitionBanner->UpdateCosts(_categoriesValues, total);
layout();
}
void StatsPanel::OnShow()
{
_wxUI->setWindowTitle(_kiss->GetUser()->_name + _(" - ") + _("Statistics"));
}
void StatsPanel::OnRangeChange(int index)
{
int monthFrom, monthTo, yearFrom, yearTo;
QListWidgetItem* item;
std::vector<Account>::iterator accountIt;
int i;
User* user = _kiss->GetUser();
monthFrom = _monthFrom->currentIndex();
yearFrom = _yearFrom->currentText().toInt();
monthTo = _monthTo->currentIndex();
yearTo = _yearTo->currentText().toInt();
if (yearTo < yearFrom ||
(yearFrom == yearTo && monthFrom > monthTo))
{
QMessageBox::critical(0, _("Error"), _("Invalide date range"));
return;
}
for(i=0, accountIt = user->_accounts.begin(); accountIt != user->_accounts.end(); accountIt++, i++)
{
item = _account->item(i);
if (accountIt->hidden || !accountIt->validAt(monthFrom, yearFrom, monthTo, yearTo))
item->setCheckState (Qt::Unchecked);
else
item->setCheckState (Qt::Checked);
}
_costRepartitionBanner->Reset();
UpdateStats(monthFrom, yearFrom, monthTo, yearTo);
}
void StatsPanel::OnAccountChange(QListWidgetItem* item)
{
OnRangeChange(0);
}