| 1 | // Copyright (C) 2016 The Qt Company Ltd. |
| 2 | // SPDX-License-Identifier: LicenseRef-Qt-Commercial OR GPL-3.0-only |
| 3 | |
| 4 | #include <private/stackedbarchartitem_p.h> |
| 5 | #include <private/bar_p.h> |
| 6 | #include <private/qbarset_p.h> |
| 7 | #include <private/qabstractbarseries_p.h> |
| 8 | #include <QtCharts/QBarSet> |
| 9 | |
| 10 | QT_BEGIN_NAMESPACE |
| 11 | |
| 12 | StackedBarChartItem::StackedBarChartItem(QAbstractBarSeries *series, QGraphicsItem* item) : |
| 13 | AbstractBarChartItem(series, item) |
| 14 | { |
| 15 | m_orientation = Qt::Vertical; |
| 16 | connect(sender: series, SIGNAL(labelsPositionChanged(QAbstractBarSeries::LabelsPosition)), |
| 17 | receiver: this, SLOT(handleLabelsPositionChanged())); |
| 18 | connect(sender: series, SIGNAL(labelsFormatChanged(QString)), receiver: this, SLOT(positionLabels())); |
| 19 | } |
| 20 | |
| 21 | void StackedBarChartItem::initializeLayout(int set, int category, |
| 22 | int layoutIndex, bool resetAnimation) |
| 23 | { |
| 24 | Q_UNUSED(set); |
| 25 | Q_UNUSED(resetAnimation); |
| 26 | |
| 27 | QRectF rect; |
| 28 | |
| 29 | if (set > 0) { |
| 30 | const QBarSet *barSet = m_series->barSets().at(i: set); |
| 31 | const qreal value = barSet->at(index: category); |
| 32 | int checkIndex = set; |
| 33 | bool found = false; |
| 34 | // Negative values stack to negative side and positive values to positive side, so we need |
| 35 | // to find the previous set that stacks to the same side |
| 36 | while (checkIndex > 0 && !found) { |
| 37 | checkIndex--; |
| 38 | QBarSet *checkSet = m_series->barSets().at(i: checkIndex); |
| 39 | const qreal checkValue = checkSet->at(index: category); |
| 40 | if ((value < 0.0) == (checkValue < 0.0)) { |
| 41 | Bar *checkBar = m_indexForBarMap.value(key: checkSet).value(key: category); |
| 42 | rect = m_layout.at(i: checkBar->layoutIndex()); |
| 43 | found = true; |
| 44 | break; |
| 45 | } |
| 46 | } |
| 47 | // If we didn't find a previous set to the same direction, just stack next to the first set |
| 48 | if (!found) { |
| 49 | QBarSet * = m_series->barSets().at(i: 0); |
| 50 | Bar *firstBar = m_indexForBarMap.value(key: firsSet).value(key: category); |
| 51 | rect = m_layout.at(i: firstBar->layoutIndex()); |
| 52 | } |
| 53 | if (value < 0) |
| 54 | rect.setTop(rect.bottom()); |
| 55 | else |
| 56 | rect.setBottom(rect.top()); |
| 57 | } else { |
| 58 | QPointF topLeft; |
| 59 | QPointF bottomRight; |
| 60 | const qreal barWidth = m_series->d_func()->barWidth() * m_seriesWidth; |
| 61 | if (domain()->type() == AbstractDomain::XLogYDomain |
| 62 | || domain()->type() == AbstractDomain::LogXLogYDomain) { |
| 63 | topLeft = topLeftPoint(category, barWidth, value: domain()->minY()); |
| 64 | bottomRight = bottomRightPoint(category, barWidth, value: domain()->minY()); |
| 65 | } else { |
| 66 | topLeft = topLeftPoint(category, barWidth, value: 0.0); |
| 67 | bottomRight = bottomRightPoint(category, barWidth, value: 0.0); |
| 68 | } |
| 69 | |
| 70 | if (m_validData) { |
| 71 | rect.setTopLeft(topLeft); |
| 72 | rect.setBottomRight(bottomRight); |
| 73 | } |
| 74 | } |
| 75 | m_layout[layoutIndex] = rect.normalized(); |
| 76 | } |
| 77 | |
| 78 | QPointF StackedBarChartItem::topLeftPoint(int category, qreal barWidth, qreal value) |
| 79 | { |
| 80 | return domain()->calculateGeometryPoint( |
| 81 | point: QPointF(m_seriesPosAdjustment + category - (barWidth / 2), value), ok&: m_validData); |
| 82 | } |
| 83 | |
| 84 | QPointF StackedBarChartItem::bottomRightPoint(int category, qreal barWidth, qreal value) |
| 85 | { |
| 86 | return domain()->calculateGeometryPoint( |
| 87 | point: QPointF(m_seriesPosAdjustment + category + (barWidth / 2), value), ok&: m_validData); |
| 88 | } |
| 89 | |
| 90 | QList<QRectF> StackedBarChartItem::calculateLayout() |
| 91 | { |
| 92 | QList<QRectF> layout; |
| 93 | layout.resize(size: m_layout.size()); |
| 94 | |
| 95 | const int setCount = m_series->count(); |
| 96 | const qreal barWidth = m_series->d_func()->barWidth() * m_seriesWidth; |
| 97 | |
| 98 | QList<qreal> positiveSums(m_categoryCount, 0.0); |
| 99 | QList<qreal> negativeSums(m_categoryCount, 0.0); |
| 100 | |
| 101 | for (int set = 0; set < setCount; set++) { |
| 102 | QBarSet *barSet = m_series->barSets().at(i: set); |
| 103 | const QList<Bar *> bars = m_barMap.value(key: barSet); |
| 104 | for (int i = 0; i < m_categoryCount; i++) { |
| 105 | Bar *bar = bars.at(i); |
| 106 | const int category = bar->index(); |
| 107 | qreal &positiveSum = positiveSums[category - m_firstCategory]; |
| 108 | qreal &negativeSum = negativeSums[category - m_firstCategory]; |
| 109 | qreal value = barSet->at(index: category); |
| 110 | QRectF rect; |
| 111 | QPointF topLeft; |
| 112 | QPointF bottomRight; |
| 113 | if (value < 0.0) { |
| 114 | topLeft = topLeftPoint(category, barWidth, value: value + negativeSum); |
| 115 | if (domain()->type() == AbstractDomain::XLogYDomain |
| 116 | || domain()->type() == AbstractDomain::LogXLogYDomain) { |
| 117 | bottomRight = bottomRightPoint(category, barWidth, |
| 118 | value: set ? negativeSum : domain()->minY()); |
| 119 | } else { |
| 120 | bottomRight = bottomRightPoint(category, barWidth, value: set ? negativeSum : 0.0); |
| 121 | } |
| 122 | negativeSum += value; |
| 123 | } else { |
| 124 | topLeft = topLeftPoint(category, barWidth, value: value + positiveSum); |
| 125 | if (domain()->type() == AbstractDomain::XLogYDomain |
| 126 | || domain()->type() == AbstractDomain::LogXLogYDomain) { |
| 127 | bottomRight = bottomRightPoint(category, barWidth, |
| 128 | value: set ? positiveSum : domain()->minY()); |
| 129 | } else { |
| 130 | bottomRight = bottomRightPoint(category, barWidth, value: set ? positiveSum : 0.0); |
| 131 | } |
| 132 | positiveSum += value; |
| 133 | } |
| 134 | |
| 135 | rect.setTopLeft(topLeft); |
| 136 | rect.setBottomRight(bottomRight); |
| 137 | rect = rect.normalized(); |
| 138 | layout[bar->layoutIndex()] = rect; |
| 139 | |
| 140 | // If animating, we need to reinitialize ~zero size bars with non-zero values |
| 141 | // so the bar growth animation starts at correct spot. We shouldn't reset if rect |
| 142 | // is already at correct position vertically, so we check for that. |
| 143 | if (m_animation && value != 0.0) { |
| 144 | const QRectF &checkRect = m_layout.at(i: bar->layoutIndex()); |
| 145 | if (checkRect.isEmpty() && |
| 146 | ((value < 0.0 && !qFuzzyCompare(p1: checkRect.top(), p2: rect.top())) |
| 147 | || (value > 0.0 && !qFuzzyCompare(p1: checkRect.bottom(), p2: rect.bottom())))) { |
| 148 | initializeLayout(set, category, layoutIndex: bar->layoutIndex(), resetAnimation: true); |
| 149 | } |
| 150 | } |
| 151 | } |
| 152 | } |
| 153 | return layout; |
| 154 | } |
| 155 | |
| 156 | void StackedBarChartItem::handleLabelsPositionChanged() |
| 157 | { |
| 158 | positionLabels(); |
| 159 | } |
| 160 | |
| 161 | void StackedBarChartItem::positionLabels() |
| 162 | { |
| 163 | positionLabelsVertical(); |
| 164 | } |
| 165 | |
| 166 | QT_END_NAMESPACE |
| 167 | |
| 168 | #include "moc_stackedbarchartitem_p.cpp" |
| 169 | |