qsplineseries.cpp
108 lines
| 3.8 KiB
| text/x-c
|
CppLexer
Marek Rosa
|
r295 | #include "qsplineseries.h" | ||
Marek Rosa
|
r401 | QTCOMMERCIALCHART_BEGIN_NAMESPACE | ||
Marek Rosa
|
r295 | QSplineSeries::QSplineSeries(QObject *parent) : | ||
Marek Rosa
|
r419 | QLineSeries(parent) | ||
Marek Rosa
|
r295 | { | ||
Marek Rosa
|
r431 | connect(this,SIGNAL(pointAdded(int)), this, SLOT(updateControlPoints())); | ||
connect(this,SIGNAL(pointRemoved(int)), this, SLOT(updateControlPoints())); | ||||
connect(this,SIGNAL(pointReplaced(int)), this, SLOT(updateControlPoints())); | ||||
Marek Rosa
|
r295 | } | ||
Marek Rosa
|
r305 | |||
Marek Rosa
|
r401 | void QSplineSeries::calculateControlPoints() | ||
{ | ||||
// Based on http://www.codeproject.com/Articles/31859/Draw-a-Smooth-Curve-through-a-Set-of-2D-Points-wit | ||||
// CPOL Licence | ||||
Marek Rosa
|
r419 | int n = m_x.size() - 1; | ||
Marek Rosa
|
r401 | if (n == 1) | ||
{ // Special case: Bezier curve should be a straight line. | ||||
// firstControlPoints = new Point[1]; | ||||
// 3P1 = 2P0 + P3 | ||||
Marek Rosa
|
r419 | m_controlPoints.append(QPointF((2 * m_x[0] + m_x[1]) / 3, (2 * m_y[0] + m_y[1]) / 3)); | ||
Marek Rosa
|
r401 | |||
// P2 = 2P1 P0 | ||||
Marek Rosa
|
r419 | m_controlPoints.append(QPointF(2 * m_controlPoints[0].x() - m_x[0], 2 * m_controlPoints[0].y() - m_y[0])); | ||
Marek Rosa
|
r401 | return; | ||
Marek Rosa
|
r318 | } | ||
Marek Rosa
|
r401 | // Calculate first Bezier control points | ||
// Right hand side vector | ||||
// Set of equations for P0 to Pn points. | ||||
// | ||||
// | 2 1 0 0 ... 0 0 0 ... 0 0 0 | | P1_1 | | P0 + 2 * P1 | | ||||
// | 1 4 1 0 ... 0 0 0 ... 0 0 0 | | P1_2 | | 4 * P1 + 2 * P2 | | ||||
// | 0 1 4 1 ... 0 0 0 ... 0 0 0 | | P1_3 | | 4 * P2 + 2 * P3 | | ||||
// | . . . . . . . . . . . . | | ... | | ... | | ||||
// | 0 0 0 0 ... 1 4 1 ... 0 0 0 | * | P1_i | = | 4 * P(i-1) + 2 * Pi | | ||||
// | . . . . . . . . . . . . | | ... | | ... | | ||||
// | 0 0 0 0 0 0 0 0 ... 1 4 1 | | P1_(n-1)| | 4 * P(n-2) + 2 * P(n-1) | | ||||
// | 0 0 0 0 0 0 0 0 ... 0 2 7 | | P1_n | | 8 * P(n-1) + Pn | | ||||
// | ||||
QList<qreal> rhs; | ||||
Marek Rosa
|
r419 | rhs.append(m_x[0] + 2 * m_x[1]); | ||
Marek Rosa
|
r401 | |||
// Set right hand side X values | ||||
Marek Rosa
|
r429 | for (int i = 1; i < n - 1; ++i) | ||
Marek Rosa
|
r419 | rhs.append(4 * m_x[i] + 2 * m_x[i + 1]); | ||
Marek Rosa
|
r401 | |||
Marek Rosa
|
r419 | rhs.append((8 * m_x[n - 1] + m_x[n]) / 2.0); | ||
Marek Rosa
|
r401 | // Get first control points X-values | ||
QList<qreal> x = getFirstControlPoints(rhs); | ||||
Marek Rosa
|
r419 | rhs[0] = m_y[0] + 2 * m_y[1]; | ||
Marek Rosa
|
r401 | |||
// Set right hand side Y values | ||||
Marek Rosa
|
r429 | for (int i = 1; i < n - 1; ++i) | ||
Marek Rosa
|
r419 | rhs[i] = 4 * m_y[i] + 2 * m_y[i + 1]; | ||
Marek Rosa
|
r401 | |||
Marek Rosa
|
r419 | rhs[n - 1] = (8 * m_y[n - 1] + m_y[n]) / 2.0; | ||
Marek Rosa
|
r401 | // Get first control points Y-values | ||
QList<qreal> y = getFirstControlPoints(rhs); | ||||
// Fill output arrays. | ||||
Marek Rosa
|
r429 | for (int i = 0; i < n; ++i) | ||
Marek Rosa
|
r318 | { | ||
Marek Rosa
|
r401 | // First control point | ||
m_controlPoints.append(QPointF(x[i], y[i])); | ||||
// Second control point | ||||
if (i < n - 1) | ||||
Marek Rosa
|
r419 | m_controlPoints.append(QPointF(2 * m_x[i + 1] - x[i + 1], 2 * m_y[i + 1] - y[i + 1])); | ||
Marek Rosa
|
r401 | else | ||
Marek Rosa
|
r419 | m_controlPoints.append(QPointF((m_x[n] + x[n - 1]) / 2, (m_y[n] + y[n - 1]) / 2)); | ||
Marek Rosa
|
r318 | } | ||
Marek Rosa
|
r401 | } | ||
QList<qreal> QSplineSeries::getFirstControlPoints(QList<qreal> rhs) | ||||
{ | ||||
QList<qreal> x; // Solution vector. | ||||
QList<qreal> tmp; // Temp workspace. | ||||
qreal b = 2.0; | ||||
x.append(rhs[0] / b); | ||||
tmp.append(0); | ||||
for (int i = 1; i < rhs.size(); i++) // Decomposition and forward substitution. | ||||
{ | ||||
tmp.append(1 / b); | ||||
b = (i < rhs.size() - 1 ? 4.0 : 3.5) - tmp[i]; | ||||
x.append((rhs[i] - x[i - 1]) / b); | ||||
} | ||||
for (int i = 1; i < rhs.size(); i++) | ||||
x[rhs.size() - i - 1] -= tmp[rhs.size() - i] * x[rhs.size() - i]; // Backsubstitution. | ||||
return x; | ||||
} | ||||
Marek Rosa
|
r431 | |||
void QSplineSeries::updateControlPoints() | ||||
{ | ||||
if(m_x.size() > 1) | ||||
{ | ||||
m_controlPoints.clear(); | ||||
calculateControlPoints(); | ||||
} | ||||
} | ||||
Marek Rosa
|
r401 | #include "moc_qsplineseries.cpp" | ||
QTCOMMERCIALCHART_END_NAMESPACE | ||||