| 1 | /**************************************************************************** |
| 2 | ** |
| 3 | ** Copyright (C) 2016 The Qt Company Ltd. |
| 4 | ** Contact: https://www.qt.io/licensing/ |
| 5 | ** |
| 6 | ** This file is part of the Qt Charts module of the Qt Toolkit. |
| 7 | ** |
| 8 | ** $QT_BEGIN_LICENSE:GPL$ |
| 9 | ** Commercial License Usage |
| 10 | ** Licensees holding valid commercial Qt licenses may use this file in |
| 11 | ** accordance with the commercial license agreement provided with the |
| 12 | ** Software or, alternatively, in accordance with the terms contained in |
| 13 | ** a written agreement between you and The Qt Company. For licensing terms |
| 14 | ** and conditions see https://www.qt.io/terms-conditions. For further |
| 15 | ** information use the contact form at https://www.qt.io/contact-us. |
| 16 | ** |
| 17 | ** GNU General Public License Usage |
| 18 | ** Alternatively, this file may be used under the terms of the GNU |
| 19 | ** General Public License version 3 or (at your option) any later version |
| 20 | ** approved by the KDE Free Qt Foundation. The licenses are as published by |
| 21 | ** the Free Software Foundation and appearing in the file LICENSE.GPL3 |
| 22 | ** included in the packaging of this file. Please review the following |
| 23 | ** information to ensure the GNU General Public License requirements will |
| 24 | ** be met: https://www.gnu.org/licenses/gpl-3.0.html. |
| 25 | ** |
| 26 | ** $QT_END_LICENSE$ |
| 27 | ** |
| 28 | ****************************************************************************/ |
| 29 | |
| 30 | #include <QtCharts/QBarCategoryAxis> |
| 31 | #include <QtCharts/QCandlestickSeries> |
| 32 | #include <QtCharts/QCandlestickSet> |
| 33 | #include <QtCharts/QDateTimeAxis> |
| 34 | #include <QtCharts/QHCandlestickModelMapper> |
| 35 | #include <QtCharts/QValueAxis> |
| 36 | #include <QtCore/QDateTime> |
| 37 | #include <QtCore/QDebug> |
| 38 | #include <QtCore/QRandomGenerator> |
| 39 | #include <QtWidgets/QCheckBox> |
| 40 | #include <QtWidgets/QComboBox> |
| 41 | #include <QtWidgets/QDoubleSpinBox> |
| 42 | #include <QtWidgets/QGridLayout> |
| 43 | #include <QtWidgets/QHeaderView> |
| 44 | #include <QtWidgets/QLabel> |
| 45 | #include <QtWidgets/QPushButton> |
| 46 | #include <QtWidgets/QTableView> |
| 47 | #include "customtablemodel.h" |
| 48 | #include "mainwidget.h" |
| 49 | |
| 50 | QT_CHARTS_USE_NAMESPACE |
| 51 | |
| 52 | MainWidget::MainWidget(QWidget *parent) |
| 53 | : QWidget(parent), |
| 54 | m_chart(new QChart()), |
| 55 | m_chartView(new QChartView(m_chart, this)), |
| 56 | m_axisX(nullptr), |
| 57 | m_axisY(nullptr), |
| 58 | m_maximumColumnWidth(-1.0), |
| 59 | m_minimumColumnWidth(5.0), |
| 60 | m_bodyOutlineVisible(true), |
| 61 | m_capsVisible(false), |
| 62 | m_bodyWidth(0.5), |
| 63 | m_capsWidth(0.5), |
| 64 | m_customIncreasingColor(false), |
| 65 | m_customDecreasingColor(false), |
| 66 | m_hModelMapper(new QHCandlestickModelMapper(this)) |
| 67 | { |
| 68 | m_chartView->setRenderHint(hint: QPainter::Antialiasing, enabled: false); |
| 69 | |
| 70 | m_hModelMapper->setModel(new CustomTableModel(this)); |
| 71 | m_hModelMapper->setTimestampColumn(0); |
| 72 | m_hModelMapper->setOpenColumn(1); |
| 73 | m_hModelMapper->setHighColumn(2); |
| 74 | m_hModelMapper->setLowColumn(3); |
| 75 | m_hModelMapper->setCloseColumn(4); |
| 76 | |
| 77 | QGridLayout *mainLayout = new QGridLayout(); |
| 78 | mainLayout->addLayout(createSeriesControlsLayout(), row: 0, column: 0); |
| 79 | mainLayout->addLayout(createSetsControlsLayout(), row: 1, column: 0); |
| 80 | mainLayout->addLayout(createCandlestickControlsLayout(), row: 2, column: 0); |
| 81 | mainLayout->addLayout(createMiscellaneousControlsLayout(), row: 3, column: 0); |
| 82 | mainLayout->addWidget(m_chartView, row: 0, column: 1, rowSpan: mainLayout->rowCount() + 1, columnSpan: 1); |
| 83 | mainLayout->addLayout(createModelMapperControlsLayout(), row: 0, column: 2, rowSpan: mainLayout->rowCount(), columnSpan: 1); |
| 84 | setLayout(mainLayout); |
| 85 | |
| 86 | addSeries(); |
| 87 | } |
| 88 | |
| 89 | MainWidget::~MainWidget() |
| 90 | { |
| 91 | } |
| 92 | |
| 93 | QGridLayout *MainWidget::createSeriesControlsLayout() |
| 94 | { |
| 95 | QGridLayout *layout = new QGridLayout(); |
| 96 | int row = 0; |
| 97 | |
| 98 | layout->addWidget(new QLabel(QStringLiteral("Series controls:" )), row, column: 0, Qt::AlignLeft); |
| 99 | |
| 100 | QPushButton *addSeriesButton = new QPushButton(QStringLiteral("Add a series" )); |
| 101 | connect(sender: addSeriesButton, SIGNAL(clicked(bool)), receiver: this, SLOT(addSeries())); |
| 102 | layout->addWidget(addSeriesButton, row: row++, column: 1, Qt::AlignLeft); |
| 103 | |
| 104 | QPushButton *removeSeriesButton = new QPushButton(QStringLiteral("Remove a series" )); |
| 105 | connect(sender: removeSeriesButton, SIGNAL(clicked(bool)), receiver: this, SLOT(removeSeries())); |
| 106 | layout->addWidget(removeSeriesButton, row: row++, column: 1, Qt::AlignLeft); |
| 107 | |
| 108 | QPushButton *removeAllSeriesButton = new QPushButton(QStringLiteral("Remove all series" )); |
| 109 | connect(sender: removeAllSeriesButton, SIGNAL(clicked(bool)), receiver: this, SLOT(removeAllSeries())); |
| 110 | layout->addWidget(removeAllSeriesButton, row: row++, column: 1, Qt::AlignLeft); |
| 111 | |
| 112 | return layout; |
| 113 | } |
| 114 | |
| 115 | QGridLayout *MainWidget::createSetsControlsLayout() |
| 116 | { |
| 117 | QGridLayout *layout = new QGridLayout(); |
| 118 | int row = 0; |
| 119 | |
| 120 | layout->addWidget(new QLabel(QStringLiteral("Sets controls:" )), row, column: 0, Qt::AlignLeft); |
| 121 | |
| 122 | QPushButton *addSetButton = new QPushButton(QStringLiteral("Add a set" )); |
| 123 | connect(sender: addSetButton, SIGNAL(clicked(bool)), receiver: this, SLOT(addSet())); |
| 124 | layout->addWidget(addSetButton, row: row++, column: 1, Qt::AlignLeft); |
| 125 | |
| 126 | QPushButton *insertSetButton = new QPushButton(QStringLiteral("Insert a set" )); |
| 127 | connect(sender: insertSetButton, SIGNAL(clicked(bool)), receiver: this, SLOT(insertSet())); |
| 128 | layout->addWidget(insertSetButton, row: row++, column: 1, Qt::AlignLeft); |
| 129 | |
| 130 | QPushButton *removeSetButton = new QPushButton(QStringLiteral("Remove a set" )); |
| 131 | connect(sender: removeSetButton, SIGNAL(clicked(bool)), receiver: this, SLOT(removeSet())); |
| 132 | layout->addWidget(removeSetButton, row: row++, column: 1, Qt::AlignLeft); |
| 133 | |
| 134 | QPushButton *removeAllSetsButton = new QPushButton(QStringLiteral("Remove all sets" )); |
| 135 | connect(sender: removeAllSetsButton, SIGNAL(clicked(bool)), receiver: this, SLOT(removeAllSets())); |
| 136 | layout->addWidget(removeAllSetsButton, row: row++, column: 1, Qt::AlignLeft); |
| 137 | |
| 138 | return layout; |
| 139 | } |
| 140 | |
| 141 | QGridLayout *MainWidget::createCandlestickControlsLayout() |
| 142 | { |
| 143 | QGridLayout *layout = new QGridLayout(); |
| 144 | int row = 0; |
| 145 | |
| 146 | layout->addWidget(new QLabel(QStringLiteral("Maximum column width:" )), row, column: 0, Qt::AlignLeft); |
| 147 | QDoubleSpinBox *maximumColumnWidthSpinBox = new QDoubleSpinBox(); |
| 148 | maximumColumnWidthSpinBox->setRange(min: -1.0, max: 1024.0); |
| 149 | maximumColumnWidthSpinBox->setDecimals(0); |
| 150 | maximumColumnWidthSpinBox->setValue(m_maximumColumnWidth); |
| 151 | maximumColumnWidthSpinBox->setSingleStep(1.0); |
| 152 | connect(sender: maximumColumnWidthSpinBox, SIGNAL(valueChanged(double)), |
| 153 | receiver: this, SLOT(changeMaximumColumnWidth(double))); |
| 154 | layout->addWidget(maximumColumnWidthSpinBox, row: row++, column: 1, Qt::AlignLeft); |
| 155 | |
| 156 | layout->addWidget(new QLabel(QStringLiteral("Minimum column width:" )), row, column: 0, Qt::AlignLeft); |
| 157 | QDoubleSpinBox *minimumColumnWidthSpinBox = new QDoubleSpinBox(); |
| 158 | minimumColumnWidthSpinBox->setRange(min: -1.0, max: 1024.0); |
| 159 | minimumColumnWidthSpinBox->setDecimals(0); |
| 160 | minimumColumnWidthSpinBox->setValue(m_minimumColumnWidth); |
| 161 | minimumColumnWidthSpinBox->setSingleStep(1.0); |
| 162 | connect(sender: minimumColumnWidthSpinBox, SIGNAL(valueChanged(double)), |
| 163 | receiver: this, SLOT(changeMinimumColumnWidth(double))); |
| 164 | layout->addWidget(minimumColumnWidthSpinBox, row: row++, column: 1, Qt::AlignLeft); |
| 165 | |
| 166 | QCheckBox *bodyOutlineVisible = new QCheckBox(QStringLiteral("Body outline visible" )); |
| 167 | connect(sender: bodyOutlineVisible, SIGNAL(toggled(bool)), receiver: this, SLOT(bodyOutlineVisibleToggled(bool))); |
| 168 | bodyOutlineVisible->setChecked(m_bodyOutlineVisible); |
| 169 | layout->addWidget(bodyOutlineVisible, row: row++, column: 0, Qt::AlignLeft); |
| 170 | |
| 171 | QCheckBox *capsVisible = new QCheckBox(QStringLiteral("Caps visible" )); |
| 172 | connect(sender: capsVisible, SIGNAL(toggled(bool)), receiver: this, SLOT(capsVisibleToggled(bool))); |
| 173 | capsVisible->setChecked(m_capsVisible); |
| 174 | layout->addWidget(capsVisible, row: row++, column: 0, Qt::AlignLeft); |
| 175 | |
| 176 | layout->addWidget(new QLabel(QStringLiteral("Candlestick body width:" )), row, column: 0, Qt::AlignLeft); |
| 177 | QDoubleSpinBox *bodyWidthSpinBox = new QDoubleSpinBox(); |
| 178 | bodyWidthSpinBox->setRange(min: -1.0, max: 2.0); |
| 179 | bodyWidthSpinBox->setValue(m_bodyWidth); |
| 180 | bodyWidthSpinBox->setSingleStep(0.1); |
| 181 | connect(sender: bodyWidthSpinBox, SIGNAL(valueChanged(double)), receiver: this, SLOT(changeBodyWidth(double))); |
| 182 | layout->addWidget(bodyWidthSpinBox, row: row++, column: 1, Qt::AlignLeft); |
| 183 | |
| 184 | layout->addWidget(new QLabel(QStringLiteral("Candlestick caps width:" )), row, column: 0, Qt::AlignLeft); |
| 185 | QDoubleSpinBox *capsWidthSpinBox = new QDoubleSpinBox(); |
| 186 | capsWidthSpinBox->setRange(min: -1.0, max: 2.0); |
| 187 | capsWidthSpinBox->setValue(m_capsWidth); |
| 188 | capsWidthSpinBox->setSingleStep(0.1); |
| 189 | connect(sender: capsWidthSpinBox, SIGNAL(valueChanged(double)), receiver: this, SLOT(changeCapsWidth(double))); |
| 190 | layout->addWidget(capsWidthSpinBox, row: row++, column: 1, Qt::AlignLeft); |
| 191 | |
| 192 | QCheckBox *increasingColor = new QCheckBox(QStringLiteral("Custom increasing color (only S1)" )); |
| 193 | connect(sender: increasingColor, SIGNAL(toggled(bool)), receiver: this, SLOT(customIncreasingColorToggled(bool))); |
| 194 | increasingColor->setChecked(m_customIncreasingColor); |
| 195 | layout->addWidget(increasingColor, row: row++, column: 0, rowSpan: 1, columnSpan: 2, Qt::AlignLeft); |
| 196 | |
| 197 | QCheckBox *decreasingColor = new QCheckBox(QStringLiteral("Custom decreasing color (only S1)" )); |
| 198 | connect(sender: decreasingColor, SIGNAL(toggled(bool)), receiver: this, SLOT(customDecreasingColorToggled(bool))); |
| 199 | decreasingColor->setChecked(m_customDecreasingColor); |
| 200 | layout->addWidget(decreasingColor, row: row++, column: 0, rowSpan: 1, columnSpan: 2, Qt::AlignLeft); |
| 201 | |
| 202 | return layout; |
| 203 | } |
| 204 | |
| 205 | QGridLayout *MainWidget::createMiscellaneousControlsLayout() |
| 206 | { |
| 207 | QGridLayout *layout = new QGridLayout(); |
| 208 | int row = 0; |
| 209 | |
| 210 | layout->addWidget(new QLabel(QStringLiteral("Miscellaneous:" )), row, column: 0, Qt::AlignLeft); |
| 211 | |
| 212 | QCheckBox *antialiasingCheckBox = new QCheckBox(QStringLiteral("Antialiasing" )); |
| 213 | connect(sender: antialiasingCheckBox, SIGNAL(toggled(bool)), receiver: this, SLOT(antialiasingToggled(bool))); |
| 214 | antialiasingCheckBox->setChecked(false); |
| 215 | layout->addWidget(antialiasingCheckBox, row: row++, column: 1, Qt::AlignLeft); |
| 216 | |
| 217 | QCheckBox *animationCheckBox = new QCheckBox(QStringLiteral("Animation" )); |
| 218 | connect(sender: animationCheckBox, SIGNAL(toggled(bool)), receiver: this, SLOT(animationToggled(bool))); |
| 219 | animationCheckBox->setChecked(false); |
| 220 | layout->addWidget(animationCheckBox, row: row++, column: 1, Qt::AlignLeft); |
| 221 | |
| 222 | QCheckBox *legendCheckBox = new QCheckBox(QStringLiteral("Legend" )); |
| 223 | connect(sender: legendCheckBox, SIGNAL(toggled(bool)), receiver: this, SLOT(legendToggled(bool))); |
| 224 | legendCheckBox->setChecked(true); |
| 225 | layout->addWidget(legendCheckBox, row: row++, column: 1, Qt::AlignLeft); |
| 226 | |
| 227 | QCheckBox *titleCheckBox = new QCheckBox(QStringLiteral("Title" )); |
| 228 | connect(sender: titleCheckBox, SIGNAL(toggled(bool)), receiver: this, SLOT(titleToggled(bool))); |
| 229 | titleCheckBox->setChecked(true); |
| 230 | layout->addWidget(titleCheckBox, row: row++, column: 1, Qt::AlignLeft); |
| 231 | |
| 232 | layout->addWidget(new QLabel(QStringLiteral("Chart theme:" )), row, column: 0, Qt::AlignLeft); |
| 233 | QComboBox *chartThemeComboBox = new QComboBox(); |
| 234 | chartThemeComboBox->addItem(QStringLiteral("Light" )); |
| 235 | chartThemeComboBox->addItem(QStringLiteral("Blue Cerulean" )); |
| 236 | chartThemeComboBox->addItem(QStringLiteral("Dark" )); |
| 237 | chartThemeComboBox->addItem(QStringLiteral("Brown Sand" )); |
| 238 | chartThemeComboBox->addItem(QStringLiteral("Blue Ncs" )); |
| 239 | chartThemeComboBox->addItem(QStringLiteral("High Contrast" )); |
| 240 | chartThemeComboBox->addItem(QStringLiteral("Blue Icy" )); |
| 241 | chartThemeComboBox->addItem(QStringLiteral("Qt" )); |
| 242 | connect(sender: chartThemeComboBox,SIGNAL(currentIndexChanged(int)),receiver: this,SLOT(changeChartTheme(int))); |
| 243 | layout->addWidget(chartThemeComboBox, row: row++, column: 1, Qt::AlignLeft); |
| 244 | |
| 245 | layout->addWidget(new QLabel(QStringLiteral("Axis X:" )), row, column: 0, Qt::AlignLeft); |
| 246 | QComboBox *axisXComboBox = new QComboBox(); |
| 247 | axisXComboBox->addItem(QStringLiteral("BarCategory" )); |
| 248 | axisXComboBox->addItem(QStringLiteral("DateTime" )); |
| 249 | axisXComboBox->addItem(QStringLiteral("Value" )); |
| 250 | connect(sender: axisXComboBox, SIGNAL(currentIndexChanged(int)), receiver: this, SLOT(changeAxisX(int))); |
| 251 | layout->addWidget(axisXComboBox, row: row++, column: 1, Qt::AlignLeft); |
| 252 | |
| 253 | return layout; |
| 254 | } |
| 255 | |
| 256 | QGridLayout *MainWidget::createModelMapperControlsLayout() |
| 257 | { |
| 258 | QGridLayout *layout = new QGridLayout(); |
| 259 | int row = 0; |
| 260 | |
| 261 | layout->addWidget(new QLabel(QStringLiteral("First series:" )), row, column: 0, Qt::AlignLeft); |
| 262 | |
| 263 | QPushButton *attachModelMapperButton = new QPushButton(QStringLiteral("Attach model mapper" )); |
| 264 | connect(sender: attachModelMapperButton, SIGNAL(clicked(bool)), receiver: this, SLOT(attachModelMapper())); |
| 265 | layout->addWidget(attachModelMapperButton, row: row++, column: 1, Qt::AlignLeft); |
| 266 | |
| 267 | QPushButton *detachModelMappeButton = new QPushButton(QStringLiteral("Detach model mapper" )); |
| 268 | connect(sender: detachModelMappeButton, SIGNAL(clicked(bool)), receiver: this, SLOT(detachModelMapper())); |
| 269 | layout->addWidget(detachModelMappeButton, row: row++, column: 1, Qt::AlignLeft); |
| 270 | |
| 271 | QTableView *tableView = new QTableView(); |
| 272 | tableView->setMinimumSize(minw: 320, minh: 480); |
| 273 | tableView->setMaximumSize(maxw: 320, maxh: 480); |
| 274 | tableView->horizontalHeader()->setSectionResizeMode(QHeaderView::Stretch); |
| 275 | tableView->verticalHeader()->setSectionResizeMode(QHeaderView::Stretch); |
| 276 | Q_ASSERT_X(m_hModelMapper->model(), Q_FUNC_INFO, "Model is not initialized" ); |
| 277 | tableView->setModel(m_hModelMapper->model()); |
| 278 | layout->addWidget(tableView, row: row++, column: 0, rowSpan: 1, columnSpan: 2, Qt::AlignLeft); |
| 279 | |
| 280 | layout->addItem(item: new QSpacerItem(1, 1, QSizePolicy::Fixed, QSizePolicy::Expanding), row: row++, column: 0); |
| 281 | |
| 282 | return layout; |
| 283 | } |
| 284 | |
| 285 | qreal MainWidget::randomValue(int min, int max) const |
| 286 | { |
| 287 | if (min > max) |
| 288 | qSwap(value1&: min, value2&: max); |
| 289 | return QRandomGenerator::global()->bounded(lowest: min, highest: max); |
| 290 | } |
| 291 | |
| 292 | QCandlestickSet *MainWidget::randomSet(qreal timestamp) |
| 293 | { |
| 294 | QCandlestickSet *set = new QCandlestickSet(timestamp); |
| 295 | set->setOpen(randomValue(min: 4, max: 11)); |
| 296 | set->setHigh(randomValue(min: 12, max: 15)); |
| 297 | set->setLow(randomValue(min: 0, max: 3)); |
| 298 | set->setClose(randomValue(min: 4, max: 11)); |
| 299 | |
| 300 | return set; |
| 301 | } |
| 302 | |
| 303 | void MainWidget::updateAxes() |
| 304 | { |
| 305 | if (m_chart->axes().isEmpty()) |
| 306 | m_chart->createDefaultAxes(); |
| 307 | |
| 308 | QCandlestickSeries *series; |
| 309 | if (!m_chart->series().isEmpty()) |
| 310 | series = qobject_cast<QCandlestickSeries *>(object: m_chart->series().at(i: 0)); |
| 311 | else |
| 312 | series = nullptr; |
| 313 | |
| 314 | m_axisX = m_chart->axes(orientation: Qt::Horizontal).first(); |
| 315 | if (series && !series->sets().isEmpty()) { |
| 316 | if (m_axisX->type() == QAbstractAxis::AxisTypeBarCategory) { |
| 317 | QBarCategoryAxis *axisX = qobject_cast<QBarCategoryAxis *>(object: m_axisX); |
| 318 | QStringList categories; |
| 319 | for (int i = 0; i < series->sets().count(); ++i) |
| 320 | categories.append(t: QString::number(i)); |
| 321 | axisX->setCategories(categories); |
| 322 | } else { // QAbstractAxis::AxisTypeDateTime || QAbstractAxis::AxisTypeValue |
| 323 | qreal msInMonth = 31.0 * 24.0 * 60.0 * 60.0 * 1000.0; |
| 324 | qreal min = series->sets().first()->timestamp() - msInMonth; |
| 325 | qreal max = series->sets().last()->timestamp() + msInMonth; |
| 326 | QDateTime minDateTime = QDateTime::fromMSecsSinceEpoch(msecs: min); |
| 327 | QDateTime maxDateTime = QDateTime::fromMSecsSinceEpoch(msecs: max); |
| 328 | |
| 329 | if (m_axisX->type() == QAbstractAxis::AxisTypeDateTime) |
| 330 | m_axisX->setRange(min: minDateTime, max: maxDateTime); |
| 331 | else |
| 332 | m_axisX->setRange(min, max); |
| 333 | } |
| 334 | } |
| 335 | |
| 336 | m_axisY = m_chart->axes(orientation: Qt::Vertical).first(); |
| 337 | m_axisY->setMax(15); |
| 338 | m_axisY->setMin(0); |
| 339 | } |
| 340 | |
| 341 | void MainWidget::addSeries() |
| 342 | { |
| 343 | if (m_chart->series().count() > 9) { |
| 344 | qDebug() << "Maximum series count is 10" ; |
| 345 | return; |
| 346 | } |
| 347 | |
| 348 | QCandlestickSeries *series = new QCandlestickSeries(); |
| 349 | series->setName(QStringLiteral("S%1" ).arg(a: m_chart->series().count() + 1)); |
| 350 | series->setMaximumColumnWidth(m_maximumColumnWidth); |
| 351 | series->setMinimumColumnWidth(m_minimumColumnWidth); |
| 352 | series->setBodyOutlineVisible(m_bodyOutlineVisible); |
| 353 | series->setBodyWidth(m_bodyWidth); |
| 354 | series->setCapsVisible(m_capsVisible); |
| 355 | series->setCapsWidth(m_capsWidth); |
| 356 | |
| 357 | if (m_chart->series().isEmpty()) { |
| 358 | if (m_customIncreasingColor) |
| 359 | series->setIncreasingColor(QColor(Qt::green)); |
| 360 | if (m_customDecreasingColor) |
| 361 | series->setDecreasingColor(QColor(Qt::red)); |
| 362 | |
| 363 | for (int month = 1; month <= 12; ++month) { |
| 364 | QDateTime dateTime; |
| 365 | dateTime.setDate(QDate(QDateTime::currentDateTime().date().year(), month, 1)); |
| 366 | dateTime.setTime(QTime(12, 34, 56, 789)); |
| 367 | |
| 368 | QCandlestickSet *set = randomSet(timestamp: dateTime.toMSecsSinceEpoch()); |
| 369 | series->append(set); |
| 370 | } |
| 371 | } else { |
| 372 | QCandlestickSeries *s = qobject_cast<QCandlestickSeries *>(object: m_chart->series().at(i: 0)); |
| 373 | for (int i = 0; i < s->sets().count(); ++i) { |
| 374 | QCandlestickSet *set = randomSet(timestamp: s->sets().at(i)->timestamp()); |
| 375 | series->append(set); |
| 376 | } |
| 377 | } |
| 378 | |
| 379 | m_chart->addSeries(series); |
| 380 | |
| 381 | updateAxes(); |
| 382 | if (!series->attachedAxes().contains(t: m_axisX)) |
| 383 | series->attachAxis(axis: m_axisX); |
| 384 | if (!series->attachedAxes().contains(t: m_axisY)) |
| 385 | series->attachAxis(axis: m_axisY); |
| 386 | } |
| 387 | |
| 388 | void MainWidget::removeSeries() |
| 389 | { |
| 390 | if (m_chart->series().isEmpty()) { |
| 391 | qDebug() << "Create a series first" ; |
| 392 | return; |
| 393 | } |
| 394 | |
| 395 | if (m_chart->series().count() == 1) |
| 396 | detachModelMapper(); |
| 397 | |
| 398 | QCandlestickSeries *series = qobject_cast<QCandlestickSeries *>(object: m_chart->series().last()); |
| 399 | m_chart->removeSeries(series); |
| 400 | delete series; |
| 401 | series = nullptr; |
| 402 | } |
| 403 | |
| 404 | void MainWidget::removeAllSeries() |
| 405 | { |
| 406 | if (m_chart->series().isEmpty()) { |
| 407 | qDebug() << "Create a series first" ; |
| 408 | return; |
| 409 | } |
| 410 | |
| 411 | detachModelMapper(); |
| 412 | |
| 413 | m_chart->removeAllSeries(); |
| 414 | } |
| 415 | |
| 416 | void MainWidget::addSet() |
| 417 | { |
| 418 | if (m_chart->series().isEmpty()) { |
| 419 | qDebug() << "Create a series first" ; |
| 420 | return; |
| 421 | } |
| 422 | |
| 423 | foreach (QAbstractSeries *s, m_chart->series()) { |
| 424 | QCandlestickSeries *series = qobject_cast<QCandlestickSeries *>(object: s); |
| 425 | |
| 426 | QDateTime dateTime; |
| 427 | if (series->count()) { |
| 428 | dateTime.setMSecsSinceEpoch(series->sets().last()->timestamp()); |
| 429 | dateTime = dateTime.addMonths(months: 1); |
| 430 | } else { |
| 431 | dateTime.setDate(QDate(QDateTime::currentDateTime().date().year(), 1, 1)); |
| 432 | dateTime.setTime(QTime(12, 34, 56, 789)); |
| 433 | } |
| 434 | |
| 435 | QCandlestickSet *set = randomSet(timestamp: dateTime.toMSecsSinceEpoch()); |
| 436 | series->append(set); |
| 437 | } |
| 438 | |
| 439 | updateAxes(); |
| 440 | } |
| 441 | |
| 442 | void MainWidget::insertSet() |
| 443 | { |
| 444 | if (m_chart->series().isEmpty()) { |
| 445 | qDebug() << "Create a series first" ; |
| 446 | return; |
| 447 | } |
| 448 | |
| 449 | foreach (QAbstractSeries *s, m_chart->series()) { |
| 450 | QCandlestickSeries *series = qobject_cast<QCandlestickSeries *>(object: s); |
| 451 | |
| 452 | QDateTime dateTime; |
| 453 | if (series->count()) { |
| 454 | dateTime.setMSecsSinceEpoch(series->sets().first()->timestamp()); |
| 455 | dateTime = dateTime.addMonths(months: -1); |
| 456 | } else { |
| 457 | dateTime.setDate(QDate(QDateTime::currentDateTime().date().year(), 1, 1)); |
| 458 | dateTime.setTime(QTime(12, 34, 56, 789)); |
| 459 | } |
| 460 | |
| 461 | QCandlestickSet *set = randomSet(timestamp: dateTime.toMSecsSinceEpoch()); |
| 462 | series->insert(index: 0, set); |
| 463 | } |
| 464 | |
| 465 | updateAxes(); |
| 466 | } |
| 467 | |
| 468 | void MainWidget::removeSet() |
| 469 | { |
| 470 | if (m_chart->series().isEmpty()) { |
| 471 | qDebug() << "Create a series first" ; |
| 472 | return; |
| 473 | } |
| 474 | |
| 475 | foreach (QAbstractSeries *s, m_chart->series()) { |
| 476 | QCandlestickSeries *series = qobject_cast<QCandlestickSeries *>(object: s); |
| 477 | if (series->sets().isEmpty()) |
| 478 | qDebug() << "Create a set first" ; |
| 479 | else |
| 480 | series->remove(set: series->sets().last()); |
| 481 | } |
| 482 | |
| 483 | updateAxes(); |
| 484 | } |
| 485 | |
| 486 | void MainWidget::removeAllSets() |
| 487 | { |
| 488 | if (m_chart->series().isEmpty()) { |
| 489 | qDebug() << "Create a series first" ; |
| 490 | return; |
| 491 | } |
| 492 | |
| 493 | foreach (QAbstractSeries *s, m_chart->series()) { |
| 494 | QCandlestickSeries *series = qobject_cast<QCandlestickSeries *>(object: s); |
| 495 | if (series->sets().isEmpty()) |
| 496 | qDebug() << "Create a set first" ; |
| 497 | else |
| 498 | series->clear(); |
| 499 | } |
| 500 | |
| 501 | updateAxes(); |
| 502 | } |
| 503 | |
| 504 | void MainWidget::changeMaximumColumnWidth(double width) |
| 505 | { |
| 506 | m_maximumColumnWidth = width; |
| 507 | foreach (QAbstractSeries *s, m_chart->series()) { |
| 508 | QCandlestickSeries *series = qobject_cast<QCandlestickSeries *>(object: s); |
| 509 | series->setMaximumColumnWidth(m_maximumColumnWidth); |
| 510 | } |
| 511 | } |
| 512 | |
| 513 | void MainWidget::changeMinimumColumnWidth(double width) |
| 514 | { |
| 515 | m_minimumColumnWidth = width; |
| 516 | foreach (QAbstractSeries *s, m_chart->series()) { |
| 517 | QCandlestickSeries *series = qobject_cast<QCandlestickSeries *>(object: s); |
| 518 | series->setMinimumColumnWidth(m_minimumColumnWidth); |
| 519 | } |
| 520 | } |
| 521 | |
| 522 | void MainWidget::bodyOutlineVisibleToggled(bool visible) |
| 523 | { |
| 524 | m_bodyOutlineVisible = visible; |
| 525 | foreach (QAbstractSeries *s, m_chart->series()) { |
| 526 | QCandlestickSeries *series = qobject_cast<QCandlestickSeries *>(object: s); |
| 527 | series->setBodyOutlineVisible(m_bodyOutlineVisible); |
| 528 | } |
| 529 | } |
| 530 | |
| 531 | void MainWidget::capsVisibleToggled(bool visible) |
| 532 | { |
| 533 | m_capsVisible = visible; |
| 534 | foreach (QAbstractSeries *s, m_chart->series()) { |
| 535 | QCandlestickSeries *series = qobject_cast<QCandlestickSeries *>(object: s); |
| 536 | series->setCapsVisible(m_capsVisible); |
| 537 | } |
| 538 | } |
| 539 | |
| 540 | void MainWidget::changeBodyWidth(double width) |
| 541 | { |
| 542 | m_bodyWidth = width; |
| 543 | foreach (QAbstractSeries *s, m_chart->series()) { |
| 544 | QCandlestickSeries *series = qobject_cast<QCandlestickSeries *>(object: s); |
| 545 | series->setBodyWidth(m_bodyWidth); |
| 546 | } |
| 547 | } |
| 548 | |
| 549 | void MainWidget::changeCapsWidth(double width) |
| 550 | { |
| 551 | m_capsWidth = width; |
| 552 | foreach (QAbstractSeries *s, m_chart->series()) { |
| 553 | QCandlestickSeries *series = qobject_cast<QCandlestickSeries *>(object: s); |
| 554 | series->setCapsWidth(m_capsWidth); |
| 555 | } |
| 556 | } |
| 557 | |
| 558 | void MainWidget::customIncreasingColorToggled(bool custom) |
| 559 | { |
| 560 | m_customIncreasingColor = custom; |
| 561 | |
| 562 | if (m_chart->series().isEmpty()) |
| 563 | return; |
| 564 | |
| 565 | QCandlestickSeries *series = qobject_cast<QCandlestickSeries *>(object: m_chart->series().at(i: 0)); |
| 566 | if (series) { |
| 567 | QColor color = m_customIncreasingColor ? QColor(Qt::green) : QColor(); |
| 568 | series->setIncreasingColor(color); |
| 569 | } |
| 570 | } |
| 571 | |
| 572 | void MainWidget::customDecreasingColorToggled(bool custom) |
| 573 | { |
| 574 | m_customDecreasingColor = custom; |
| 575 | |
| 576 | if (m_chart->series().isEmpty()) |
| 577 | return; |
| 578 | |
| 579 | QCandlestickSeries *series = qobject_cast<QCandlestickSeries *>(object: m_chart->series().at(i: 0)); |
| 580 | if (series) { |
| 581 | QColor color = m_customDecreasingColor ? QColor(Qt::red) : QColor(); |
| 582 | series->setDecreasingColor(color); |
| 583 | } |
| 584 | } |
| 585 | |
| 586 | void MainWidget::antialiasingToggled(bool enabled) |
| 587 | { |
| 588 | m_chartView->setRenderHint(hint: QPainter::Antialiasing, enabled); |
| 589 | } |
| 590 | |
| 591 | void MainWidget::animationToggled(bool enabled) |
| 592 | { |
| 593 | if (enabled) |
| 594 | m_chart->setAnimationOptions(QChart::SeriesAnimations); |
| 595 | else |
| 596 | m_chart->setAnimationOptions(QChart::NoAnimation); |
| 597 | } |
| 598 | |
| 599 | void MainWidget::legendToggled(bool visible) |
| 600 | { |
| 601 | m_chart->legend()->setVisible(visible); |
| 602 | if (visible) |
| 603 | m_chart->legend()->setAlignment(Qt::AlignBottom); |
| 604 | } |
| 605 | |
| 606 | void MainWidget::titleToggled(bool visible) |
| 607 | { |
| 608 | if (visible) |
| 609 | m_chart->setTitle(QStringLiteral("Candlestick Chart" )); |
| 610 | else |
| 611 | m_chart->setTitle(QString()); |
| 612 | } |
| 613 | |
| 614 | void MainWidget::changeChartTheme(int themeIndex) |
| 615 | { |
| 616 | if (themeIndex < QChart::ChartThemeLight || themeIndex > QChart::ChartThemeQt) { |
| 617 | qDebug() << "Invalid chart theme index:" << themeIndex; |
| 618 | return; |
| 619 | } |
| 620 | |
| 621 | m_chart->setTheme((QChart::ChartTheme)(themeIndex)); |
| 622 | } |
| 623 | |
| 624 | void MainWidget::changeAxisX(int axisXIndex) |
| 625 | { |
| 626 | if (m_axisX) { |
| 627 | m_chart->removeAxis(axis: m_axisX); |
| 628 | delete m_axisX; |
| 629 | } |
| 630 | |
| 631 | switch (axisXIndex) { |
| 632 | case 0: |
| 633 | m_axisX = new QBarCategoryAxis(); |
| 634 | break; |
| 635 | case 1: |
| 636 | m_axisX = new QDateTimeAxis(); |
| 637 | break; |
| 638 | case 2: |
| 639 | m_axisX = new QValueAxis(); |
| 640 | break; |
| 641 | default: |
| 642 | qDebug() << "Invalid axis x index:" << axisXIndex; |
| 643 | return; |
| 644 | } |
| 645 | |
| 646 | m_chart->addAxis(axis: m_axisX, alignment: Qt::AlignBottom); |
| 647 | |
| 648 | updateAxes(); |
| 649 | |
| 650 | foreach (QAbstractSeries *series, m_chart->series()) |
| 651 | series->attachAxis(axis: m_axisX); |
| 652 | } |
| 653 | |
| 654 | void MainWidget::attachModelMapper() |
| 655 | { |
| 656 | if (m_hModelMapper->series()) { |
| 657 | qDebug() << "Model mapper is already attached" ; |
| 658 | return; |
| 659 | } |
| 660 | |
| 661 | if (m_chart->series().isEmpty()) |
| 662 | addSeries(); |
| 663 | |
| 664 | QCandlestickSeries *series = qobject_cast<QCandlestickSeries *>(object: m_chart->series().at(i: 0)); |
| 665 | Q_ASSERT(series); |
| 666 | series->setName(QStringLiteral("SWMM" )); // Series With Model Mapper |
| 667 | |
| 668 | CustomTableModel *model = qobject_cast<CustomTableModel *>(object: m_hModelMapper->model()); |
| 669 | foreach (QCandlestickSet *set, series->sets()) |
| 670 | model->addRow(set); |
| 671 | |
| 672 | m_hModelMapper->setFirstSetRow(0); |
| 673 | m_hModelMapper->setLastSetRow(model->rowCount() - 1); |
| 674 | m_hModelMapper->setSeries(series); |
| 675 | } |
| 676 | |
| 677 | void MainWidget::detachModelMapper() |
| 678 | { |
| 679 | if (!m_hModelMapper->series()) |
| 680 | return; |
| 681 | |
| 682 | QCandlestickSeries *series = qobject_cast<QCandlestickSeries *>(object: m_hModelMapper->series()); |
| 683 | Q_ASSERT(series); |
| 684 | series->setName(QStringLiteral("S1" )); |
| 685 | |
| 686 | m_hModelMapper->setSeries(nullptr); |
| 687 | m_hModelMapper->setFirstSetRow(-1); |
| 688 | m_hModelMapper->setLastSetRow(-1); |
| 689 | |
| 690 | CustomTableModel *model = qobject_cast<CustomTableModel *>(object: m_hModelMapper->model()); |
| 691 | model->clearRows(); |
| 692 | } |
| 693 | |