106 lines
5.1 KiB
C++
106 lines
5.1 KiB
C++
#include "frmfinancial.h"
|
|
#include "ui_frmfinancial.h"
|
|
#include "qdebug.h"
|
|
|
|
frmFinancial::frmFinancial(QWidget *parent) : QWidget(parent), ui(new Ui::frmFinancial)
|
|
{
|
|
ui->setupUi(this);
|
|
this->initForm();
|
|
}
|
|
|
|
frmFinancial::~frmFinancial()
|
|
{
|
|
delete ui;
|
|
}
|
|
|
|
void frmFinancial::initForm()
|
|
{
|
|
ui->customPlot->legend->setVisible(true);
|
|
|
|
// generate two sets of random walk data (one for candlestick and one for ohlc chart):
|
|
int n = 500;
|
|
QVector<double> time(n), value1(n), value2(n);
|
|
QDateTime start(QDate(2022, 1, 1), QTime(0, 0));
|
|
start.setTimeSpec(Qt::UTC);
|
|
double startTime = start.toMSecsSinceEpoch() / 1000.0;
|
|
double binSize = 3600 * 24; // bin data in 1 day intervals
|
|
time[0] = startTime;
|
|
value1[0] = 60;
|
|
value2[0] = 20;
|
|
std::srand(9);
|
|
for (int i = 1; i < n; ++i) {
|
|
time[i] = startTime + 3600 * i;
|
|
value1[i] = value1[i - 1] + (std::rand() / (double)RAND_MAX - 0.5) * 10;
|
|
value2[i] = value2[i - 1] + (std::rand() / (double)RAND_MAX - 0.5) * 3;
|
|
}
|
|
|
|
// create candlestick chart:
|
|
QCPFinancial *candlesticks = new QCPFinancial(ui->customPlot->xAxis, ui->customPlot->yAxis);
|
|
candlesticks->setName("Candlestick");
|
|
candlesticks->setChartStyle(QCPFinancial::csCandlestick);
|
|
candlesticks->data()->set(QCPFinancial::timeSeriesToOhlc(time, value1, binSize, startTime));
|
|
candlesticks->setWidth(binSize * 0.9);
|
|
candlesticks->setTwoColored(true);
|
|
candlesticks->setBrushPositive(QColor(245, 245, 245));
|
|
candlesticks->setBrushNegative(QColor(40, 40, 40));
|
|
candlesticks->setPenPositive(QPen(QColor(0, 0, 0)));
|
|
candlesticks->setPenNegative(QPen(QColor(0, 0, 0)));
|
|
|
|
// create ohlc chart:
|
|
QCPFinancial *ohlc = new QCPFinancial(ui->customPlot->xAxis, ui->customPlot->yAxis);
|
|
ohlc->setName("OHLC");
|
|
ohlc->setChartStyle(QCPFinancial::csOhlc);
|
|
ohlc->data()->set(QCPFinancial::timeSeriesToOhlc(time, value2, binSize / 3.0, startTime)); // divide binSize by 3 just to make the ohlc bars a bit denser
|
|
ohlc->setWidth(binSize * 0.2);
|
|
ohlc->setTwoColored(true);
|
|
|
|
// create bottom axis rect for volume bar chart:
|
|
QCPAxisRect *volumeAxisRect = new QCPAxisRect(ui->customPlot);
|
|
ui->customPlot->plotLayout()->addElement(1, 0, volumeAxisRect);
|
|
volumeAxisRect->setMaximumSize(QSize(QWIDGETSIZE_MAX, 100));
|
|
volumeAxisRect->axis(QCPAxis::atBottom)->setLayer("axes");
|
|
volumeAxisRect->axis(QCPAxis::atBottom)->grid()->setLayer("grid");
|
|
// bring bottom and main axis rect closer together:
|
|
ui->customPlot->plotLayout()->setRowSpacing(0);
|
|
volumeAxisRect->setAutoMargins(QCP::msLeft | QCP::msRight | QCP::msBottom);
|
|
volumeAxisRect->setMargins(QMargins(0, 0, 0, 0));
|
|
// create two bar plottables, for positive (green) and negative (red) volume bars:
|
|
ui->customPlot->setAutoAddPlottableToLegend(false);
|
|
QCPBars *volumePos = new QCPBars(volumeAxisRect->axis(QCPAxis::atBottom), volumeAxisRect->axis(QCPAxis::atLeft));
|
|
QCPBars *volumeNeg = new QCPBars(volumeAxisRect->axis(QCPAxis::atBottom), volumeAxisRect->axis(QCPAxis::atLeft));
|
|
for (int i = 0; i < n / 5; ++i) {
|
|
int v = std::rand() % 20000 + std::rand() % 20000 + std::rand() % 20000 - 10000 * 3;
|
|
(v < 0 ? volumeNeg : volumePos)->addData(startTime + 3600 * 5.0 * i, qAbs(v)); // add data to either volumeNeg or volumePos, depending on sign of v
|
|
}
|
|
volumePos->setWidth(3600 * 4);
|
|
volumePos->setPen(Qt::NoPen);
|
|
volumePos->setBrush(QColor(100, 180, 110));
|
|
volumeNeg->setWidth(3600 * 4);
|
|
volumeNeg->setPen(Qt::NoPen);
|
|
volumeNeg->setBrush(QColor(180, 90, 90));
|
|
|
|
// interconnect x axis ranges of main and bottom axis rects:
|
|
connect(ui->customPlot->xAxis, SIGNAL(rangeChanged(QCPRange)), volumeAxisRect->axis(QCPAxis::atBottom), SLOT(setRange(QCPRange)));
|
|
connect(volumeAxisRect->axis(QCPAxis::atBottom), SIGNAL(rangeChanged(QCPRange)), ui->customPlot->xAxis, SLOT(setRange(QCPRange)));
|
|
// configure axes of both main and bottom axis rect:
|
|
QSharedPointer<QCPAxisTickerDateTime> dateTimeTicker(new QCPAxisTickerDateTime);
|
|
dateTimeTicker->setDateTimeSpec(Qt::UTC);
|
|
dateTimeTicker->setDateTimeFormat("dd. MMMM");
|
|
volumeAxisRect->axis(QCPAxis::atBottom)->setTicker(dateTimeTicker);
|
|
volumeAxisRect->axis(QCPAxis::atBottom)->setTickLabelRotation(15);
|
|
ui->customPlot->xAxis->setBasePen(Qt::NoPen);
|
|
ui->customPlot->xAxis->setTickLabels(false);
|
|
ui->customPlot->xAxis->setTicks(false); // only want vertical grid in main axis rect, so hide xAxis backbone, ticks, and labels
|
|
ui->customPlot->xAxis->setTicker(dateTimeTicker);
|
|
ui->customPlot->rescaleAxes();
|
|
ui->customPlot->xAxis->scaleRange(1.025, ui->customPlot->xAxis->range().center());
|
|
ui->customPlot->yAxis->scaleRange(1.1, ui->customPlot->yAxis->range().center());
|
|
|
|
// make axis rects' left side line up:
|
|
QCPMarginGroup *group = new QCPMarginGroup(ui->customPlot);
|
|
ui->customPlot->axisRect()->setMarginGroup(QCP::msLeft | QCP::msRight, group);
|
|
volumeAxisRect->setMarginGroup(QCP::msLeft | QCP::msRight, group);
|
|
|
|
ui->customPlot->setInteractions(QCP::iRangeDrag | QCP::iRangeZoom);
|
|
}
|