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