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
|
// SPDX-License-Identifier: GPL-2.0
#include "legend.h"
#include "statscolors.h"
#include <QFontMetrics>
#include <QGraphicsSceneMouseEvent>
#include <QGraphicsWidget>
#include <QPen>
static const double legendBorderSize = 2.0;
static const double legendBoxBorderSize = 1.0;
static const double legendBoxScale = 0.8; // 1.0: text-height of the used font
static const double legendInternalBorderSize = 2.0;
static const QColor legendColor(0x00, 0x8e, 0xcc, 192); // Note: fourth argument is opacity
static const QColor legendBorderColor(Qt::black);
Legend::Legend(QGraphicsWidget *chart, const std::vector<QString> &names) :
QGraphicsRectItem(chart), chart(chart), displayedItems(0), width(0.0), height(0.0)
{
setZValue(30.0); // On top of everything else.
entries.reserve(names.size());
int idx = 0;
for (const QString &name: names)
entries.emplace_back(name, idx++, (int)names.size(), this);
// Calculate the height and width of the elements
if (!entries.empty()) {
QFontMetrics fm(entries[0].text->font());
fontHeight = fm.height();
for (Entry &e: entries)
e.width = fontHeight + 2.0 * legendBoxBorderSize +
fm.size(Qt::TextSingleLine, e.text->text()).width();
} else {
fontHeight = 0.0;
}
setPen(QPen(legendBorderColor, legendBorderSize));
setBrush(QBrush(legendColor));
resize(); // Draw initial legend
}
Legend::Entry::Entry(const QString &name, int idx, int numBins, QGraphicsItem *parent) :
rect(new QGraphicsRectItem(parent)),
text(new QGraphicsSimpleTextItem(name, parent))
{
rect->setZValue(30.0);
rect->setPen(QPen(legendBorderColor, legendBoxBorderSize));
rect->setBrush(QBrush(binColor(idx, numBins)));
text->setZValue(30.0);
text->setBrush(QBrush(darkLabelColor));
}
void Legend::hide()
{
for (Entry &e: entries) {
e.rect->hide();
e.text->hide();
}
QGraphicsRectItem::hide();
}
void Legend::resize()
{
if (entries.empty())
return hide();
QSizeF size = chart->size();
// Silly heuristics: make the legend at most half as high and half as wide as the chart.
// Not sure if that makes sense - this might need some optimization.
int maxRows = static_cast<int>(size.height() / 2.0 - 2.0 * legendInternalBorderSize) / fontHeight;
if (maxRows <= 0)
return hide();
int numColumns = ((int)entries.size() - 1) / maxRows + 1;
int numRows = ((int)entries.size() - 1) / numColumns + 1;
double x = legendInternalBorderSize;
displayedItems = 0;
for (int col = 0; col < numColumns; ++col) {
double y = legendInternalBorderSize;
double nextX = x;
for (int row = 0; row < numRows; ++row) {
int idx = col * numRows + row;
if (idx >= (int)entries.size())
break;
entries[idx].pos = QPointF(x, y);
nextX = std::max(nextX, x + entries[idx].width);
y += fontHeight;
++displayedItems;
}
x += nextX;
width = nextX;
if (width >= size.width() / 2.0) // More than half the chart-width -> give up
break;
}
width += legendInternalBorderSize;
height = 2 * legendInternalBorderSize + numRows * fontHeight;
updatePosition();
}
void Legend::updatePosition()
{
if (displayedItems <= 0)
return hide();
// For now, place the legend in the top right corner.
QPointF pos(chart->size().width() - width - 10.0, 10.0);
setRect(QRectF(pos, QSizeF(width, height)));
for (int i = 0; i < displayedItems; ++i) {
QPointF itemPos = pos + entries[i].pos;
QRectF rect(itemPos, QSizeF(fontHeight, fontHeight));
// Decrease box size by legendBoxScale factor
double delta = fontHeight * (1.0 - legendBoxScale) / 2.0;
rect = rect.adjusted(delta, delta, -delta, -delta);
entries[i].rect->setRect(rect);
itemPos.rx() += fontHeight + 2.0 * legendBoxBorderSize;
entries[i].text->setPos(itemPos);
entries[i].rect->show();
entries[i].text->show();
}
for (int i = displayedItems; i < (int)entries.size(); ++i) {
entries[i].rect->hide();
entries[i].text->hide();
}
show();
}
|
