summaryrefslogtreecommitdiffstats
path: root/statisticsdialog.cpp
blob: 30141c2dc14289a1f8950458789467102ced816c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
/*
  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
  2 of the License, or (at your option) any later version.
*/

#include <QTabWidget>
#include <QVBoxLayout>
#include <QHBoxLayout>
#include <QSqlQuery>
#include <QVariant>
#include <QPushButton>
#include <QLocale>
#include <QFile>
#include <QWebView>

#include "statisticsdialog.h"

StatisticsDialog::StatisticsDialog(QWidget *parent, Qt::WindowFlags f) : QDialog(parent, f){
	//html skeleton
	QFile htmlFile(":/statistics.html");
	htmlFile.open(QIODevice::ReadOnly);
	QString html = htmlFile.readAll();

	//genre distribution
	QWebView *genreView = new QWebView;
	QString genreHtml = html;
	genreHtml.replace("{title}", tr("Genre distribution"));
	genreHtml.replace("{desc}", tr("Genres"));
	QList<QList<QVariant> > data = queryData("SELECT genres.tgenrename, COUNT(seriesparts_genremap.iseriesparts_id) FROM genres, seriesparts_genremap WHERE genres.igenres_id = seriesparts_genremap.igenres_id GROUP BY genres.tgenrename ORDER BY count DESC;");
	QString tableData = table(data, max(data));
	genreHtml.replace("{table}", tableData);
	genreView->setHtml(genreHtml);

	//quality distribution
	QWebView *qualityView = new QWebView;
	QString qualityHtml = html;
	qualityHtml.replace("{title}", tr("Quality distribution"));
	qualityHtml.replace("{desc}", tr("Quality"));
	data = queryData("SELECT files.siquality, COUNT(seriesparts.iseriesparts_id) FROM files, seriesparts WHERE files.iseriespart_id = seriesparts.iseriesparts_id AND files.siquality IS NOT NULL GROUP BY files.siquality ORDER by siquality desc;");
	tableData = table(data, max(data));
	qualityHtml.replace("{table}", tableData);
	qualityView->setHtml(qualityHtml);

	//top 20 actors
	QWebView *topActorView = new QWebView;
	QString topActorHtml = html;
	topActorHtml.replace("{title}", tr("Top 20 actors"));
	topActorHtml.replace("{desc}", tr("Actor"));
	data = queryData("SELECT actors.tactorname, COUNT(seriesparts_actormap.iseriesparts_id) FROM actors, seriesparts_actormap WHERE actors.iactors_id = seriesparts_actormap.iactors_id GROUP BY actors.tactorname ORDER BY count desc LIMIT 20");
	tableData = table(data, max(data));
	topActorHtml.replace("{table}", tableData);
	topActorView->setHtml(topActorHtml);

	//general statistics
	QWebView *generalView = new QWebView;
	QString generalHtml = html;
	generalHtml.remove(QRegExp("<thead>.*</thead>"));
	tableData = generalStatistics();
	generalHtml.replace("{title}", tr("General statistics"));
	generalHtml.replace("{table}", tableData);
	generalView->setHtml(generalHtml);

	//tab
	QTabWidget *tab = new QTabWidget;
	tab->addTab(genreView, tr("Genres"));
	tab->addTab(qualityView, tr("Quality"));
	tab->addTab(topActorView, tr("Top 20 actors"));
	tab->addTab(generalView, tr("General statistics"));

	//close button
	QPushButton *close = new QPushButton(tr("Close"));
	connect(close, SIGNAL(clicked()), this, SLOT(accept()));
	QHBoxLayout *buttonLayout = new QHBoxLayout;
	buttonLayout->setAlignment(Qt::AlignCenter);
	buttonLayout->addWidget(close);

	//the dialog
	QVBoxLayout *mainLayout = new QVBoxLayout;
	mainLayout->addWidget(tab);
	mainLayout->addLayout(buttonLayout);
	setLayout(mainLayout);
	setWindowTitle(tr("SheMov - Statistics"));
}

QString StatisticsDialog::table(const QList<QList<QVariant> > &data, int max) const {
	QString table;
	QLocale loc;
	foreach(QList<QVariant> l, data){
		table.append("<tr>");
		int percent = l.at(1).toInt() * 100 / max - 1;
		percent = percent < 1 ? 1 : percent;
		table.append(QString("<td class=\"descr\">%1</td>").arg(l.at(0).toString()));
		table.append(QString("<td class=\"bar\"><div style=\"width: %1%\" class=\"bar\"></div></td>").arg(percent));
		table.append(QString("<td class=\"total\">%1</td>").arg(loc.toString(l.at(1).toInt())));
		table.append("</tr>");
	}
	return table;
}

QList<QList<QVariant> > StatisticsDialog::queryData(const QString &query) const {
	QSqlDatabase db = QSqlDatabase::database("treedb");
	QSqlQuery q(query, db);
	QList<QList<QVariant> > retval;
	while(q.next()){
		QList<QVariant> data;
		data << q.value(0) << q.value(1);
		retval << data;
	}
	return retval;
}

int StatisticsDialog::max(const QList<QList<QVariant> > &data) const{
	int retval = 0;
	foreach(QList<QVariant> l, data){
		int value = l.at(1).toInt();
		if(value > retval){
			retval = value;
		}
	}
	return retval;
}

QString StatisticsDialog::generalStatistics() const{
	//get database
	QSqlDatabase db = QSqlDatabase::database("treedb");

	//series count
	QSqlQuery q1("SELECT COUNT(*) FROM series", db);
	q1.first();
	int numSeries = q1.value(0).toInt();

	//movies count
	QSqlQuery q2("SELECT COUNT(*) FROM seriesparts", db);
	q2.first();
	int numMovies = q2.value(0).toInt();

	//files count
	QSqlQuery q3("SELECT COUNT(*) FROM files", db);
	q3.first();
	int numFiles = q3.value(0).toInt();

	//movie files
	QSqlQuery q4("SELECT COUNT(*) FROM files WHERE sifiletype = 1", db);
	q4.first();
	int numMovieFiles = q4.value(0).toInt();

	//cover files
	QSqlQuery q5("SELECT COUNT(*) FROM files WHERE sifiletype != 1", db);
	q5.first();
	int numCoverFiles = q5.value(0).toInt();

	//total archive size
	QSqlQuery q6("SELECT ROUND(SUM(bisize) / 1024 / 1024 / 1024, 2) FROM files", db);
	q6.first();
	float totalArchiveSize = q6.value(0).toFloat();

	//local file size
	QSqlQuery q7("SELECT ROUND(SUM(bisize) / 1024 / 1024 / 1024, 2) FROM files WHERE idvd = -1", db);
	q7.first();
	float localFileSize = q7.value(0).toFloat();

	//archived file size
	QSqlQuery q8("SELECT ROUND(SUM(bisize) / 1024 / 1024 / 1024, 2) FROM files WHERE idvd != -1", db);
	q8.first();
	float archivedFileSize = q8.value(0).toFloat();

	//actor count
	QSqlQuery q9("SELECT COUNT(*) FROM actors", db);
	q9.first();
	int actorCount = q9.value(0).toInt();

	//genre count
	QSqlQuery q10("SELECT COUNT(*) FROM genres", db);
	q10.first();
	int genreCount = q10.value(0).toInt();

	//build the tabel
	QString retval;
	QLocale loc;
	retval.append(QString("<tr><td>Number of series</td><td style=\"padding-left: 30px\">%1</td></tr>").arg(numSeries));
	retval.append(QString("<tr><td>Number of movies</td><td style=\"padding-left: 30px\">%1</td></tr>").arg(numMovies));
	retval.append(QString("<tr><td>Number of files</td><td style=\"padding-left: 30px\">%1</td></tr>").arg(numFiles));
	retval.append(QString("<tr><td>Number of movie files</td><td style=\"padding-left: 30px\">%1</td></tr>").arg(numMovieFiles));
	retval.append(QString("<tr><td>Number of cover files</td><td style=\"padding-left: 30px\">%1</td></tr>").arg(numCoverFiles));
	retval.append(QString("<tr><td>Total archive size</td><td style=\"padding-left: 30px\">%1 GB</td></tr>").arg(loc.toString(totalArchiveSize)));
	retval.append(QString("<tr><td>Local files size</td><td style=\"padding-left: 30px\">%1 GB</td></tr>").arg(loc.toString(localFileSize)));
	retval.append(QString("<tr><td>Archived files size</td><td style=\"padding-left: 30px\">%1 GB</td></tr>").arg(loc.toString(archivedFileSize)));
	retval.append(QString("<tr><td>Number of actors</td><td style=\"padding-left: 30px\">%1</td></tr>").arg(actorCount));
	retval.append(QString("<tr><td>Number of genres</td><td style=\"padding-left: 30px\">%1</td></tr>").arg(genreCount));
	return retval;
}