aboutsummaryrefslogtreecommitdiffstats
path: root/stats/pieseries.h
blob: 242cfa781e2f10179f4bda5e8c04190d7c10634c (plain) (blame)
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
// SPDX-License-Identifier: GPL-2.0
// A pie chart series, which displays percentual information.
#ifndef PIE_SERIES_H
#define PIE_SERIES_H

#include "statsseries.h"

#include <memory>
#include <vector>
#include <QString>

class InformationBox;
class QGraphicsEllipseItem;
class QGraphicsSimpleTextItem;

class PieSeries : public StatsSeries {
public:
	// The pie series is initialized with (name, count) pairs.
	// If keepOrder is false, bins will be sorted by size, otherwise the sorting
	// of the shown bins will be retained. Small bins are omitted for clarity.
	PieSeries(QtCharts::QChart *chart, StatsAxis *xAxis, StatsAxis *yAxis, const QString &categoryName,
		  const std::vector<std::pair<QString, int>> &data, bool keepOrder, bool labels);
	~PieSeries();

	void updatePositions() override;
	bool hover(QPointF pos) override;
	void unhighlight() override;

	std::vector<QString> binNames();

private:
	// Get item under mouse pointer, or -1 if none
	int getItemUnderMouse(const QPointF &f) const;

	QString categoryName;
	std::vector<QString> makeInfo(int idx) const;

	struct Item {
		std::unique_ptr<QGraphicsEllipseItem> item;
		std::unique_ptr<QGraphicsSimpleTextItem> innerLabel, outerLabel;
		QString name;
		double angleTo; // In fraction of total
		int count;
		QPointF innerLabelPos, outerLabelPos; // With respect to a (-1, -1)-(1, 1) rectangle.
		Item(QtCharts::QChart *chart, const QString &name, int from, int count, int totalCount,
		     int bin_nr, int numBins, bool labels);
		void updatePositions(const QRectF &rect, const QPointF &center, double radius);
		void highlight(int bin_nr, bool highlight, int numBins);
	};
	std::vector<Item> items;
	int totalCount;

	// Entries in the "other" group. If empty, there is no "other" group.
	struct OtherItem {
		QString name;
		int count;
	};
	std::vector<OtherItem> other;

	std::unique_ptr<InformationBox> information;
	QPointF center; // center of drawing area
	double radius; // radius of pie
	int highlighted;
};

#endif