123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250 |
- /*=========================================================================
- Library: CTK
-
- Copyright (c) 2010 Kitware Inc.
- Licensed under the Apache License, Version 2.0 (the "License");
- you may not use this file except in compliance with the License.
- You may obtain a copy of the License at
- http://www.commontk.org/LICENSE
- Unless required by applicable law or agreed to in writing, software
- distributed under the License is distributed on an "AS IS" BASIS,
- WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- See the License for the specific language governing permissions and
- limitations under the License.
-
- =========================================================================*/
- /// Qt includes
- #include <QColor>
- #include <QDebug>
- #include <QLinearGradient>
- #include <QGraphicsSceneMouseEvent>
- #include <QPainter>
- #include <QtGlobal>
- #include <QVariant>
- /// CTK includes
- #include "ctkTransferFunction.h"
- #include "ctkTransferFunctionControlPointsItem.h"
- //-----------------------------------------------------------------------------
- class ctkTransferFunctionControlPointsItemPrivate:
- public ctkPrivate<ctkTransferFunctionControlPointsItem>
- {
- public:
- ctkTransferFunctionControlPointsItemPrivate();
- void init();
- QList<QPointF> ControlPoints;
- QSizeF PointSize;
- int SelectedPoint;
- };
- //-----------------------------------------------------------------------------
- ctkTransferFunctionControlPointsItemPrivate::ctkTransferFunctionControlPointsItemPrivate()
- {
- this->PointSize = QSizeF(10.,10.);
- this->SelectedPoint = -1;
- }
- //-----------------------------------------------------------------------------
- void ctkTransferFunctionControlPointsItemPrivate::init()
- {
- CTK_P(ctkTransferFunctionControlPointsItem);
- p->setAcceptedMouseButtons(Qt::LeftButton);
- }
- //-----------------------------------------------------------------------------
- ctkTransferFunctionControlPointsItem::ctkTransferFunctionControlPointsItem(QGraphicsItem* parentGraphicsItem)
- :ctkTransferFunctionItem(parentGraphicsItem)
- {
- CTK_INIT_PRIVATE(ctkTransferFunctionControlPointsItem);
- ctk_d()->init();
- }
- //-----------------------------------------------------------------------------
- ctkTransferFunctionControlPointsItem::ctkTransferFunctionControlPointsItem(
- ctkTransferFunction* transferFunction, QGraphicsItem* parentItem)
- :ctkTransferFunctionItem(transferFunction, parentItem)
- {
- CTK_INIT_PRIVATE(ctkTransferFunctionControlPointsItem);
- ctk_d()->init();
- }
- //-----------------------------------------------------------------------------
- ctkTransferFunctionControlPointsItem::~ctkTransferFunctionControlPointsItem()
- {
- }
- //-----------------------------------------------------------------------------
- void ctkTransferFunctionControlPointsItem::paint(
- QPainter * painter, const QStyleOptionGraphicsItem * option, QWidget * widget)
- {
- CTK_D(ctkTransferFunctionControlPointsItem);
- int count = this->transferFunction() ? this->transferFunction()->count() : 0;
- if (count <= 0)
- {
- return;
- }
- qreal rangeX[2];
- this->transferFunction()->range(rangeX);
- qreal rangeXDiff = this->rect().width() / (rangeX[1] - rangeX[0]);
- qreal rangeXOffSet = rangeX[0];
- QVariant rangeY[2];
- rangeY[0] = this->transferFunction()->minValue();
- rangeY[1] = this->transferFunction()->maxValue();
- qreal rangeYDiff = this->rect().height();
- qreal rangeYOffSet;
- if (rangeY[0].canConvert<qreal>())
- {
- if (rangeY[1].toReal() == rangeY[0].toReal())
- {
- rangeYDiff /= rangeY[0].toReal();
- rangeYOffSet = 0.;
- }
- else
- {
- rangeYDiff /= rangeY[1].toReal() - rangeY[0].toReal();
- rangeYOffSet = rangeY[0].toReal();
- }
- }
- else if (rangeY[0].canConvert<QColor>())
- {
- if ( rangeY[1].value<QColor>().alphaF() == rangeY[0].value<QColor>().alphaF())
- {
- rangeYDiff /= rangeY[0].value<QColor>().alphaF();
- rangeYOffSet = 0.;
- }
- else
- {
- rangeYDiff /= rangeY[1].value<QColor>().alphaF() - rangeY[0].value<QColor>().alphaF();
- rangeYOffSet = rangeY[0].value<QColor>().alphaF();
- }
- }
- else
- {
- Q_ASSERT(rangeY[0].canConvert<qreal>() ||
- rangeY[0].canConvert<QColor>());
- }
- ctkControlPoint* startCP = this->transferFunction()->controlPoint(0);
- ctkControlPoint* endCP = 0;
- qreal start = 0;
- qreal end = 0;
- QPainterPath path;
- QPointF startPos(startCP->x() - rangeXOffSet, this->y(startCP->value()) - rangeYOffSet);
- startPos.rx() *= rangeXDiff;
- startPos.setY(this->rect().height()
- - startPos.y() * rangeYDiff);
-
- d->ControlPoints.clear();
- d->ControlPoints << startPos;
- path.moveTo(startPos);
- for(int i = 1; i < count; ++i)
- {
- endCP = this->transferFunction()->controlPoint(i);
- if (dynamic_cast<ctkNonLinearControlPoint*>(startCP))
- {
- QList<ctkPoint> points = this->nonLinearPoints(startCP, endCP);
- int j;
- for (j = 1; j < points.count(); ++j)
- {
- path.lineTo(
- QPointF((points[j].X - rangeXOffSet) * rangeXDiff, this->rect().height() -
- (this->y(points[j].Value) - rangeYOffSet) * rangeYDiff));
- }
- j = points.count() - 1;
- d->ControlPoints << QPointF((points[j].X - rangeXOffSet) * rangeXDiff, this->rect().height() -
- (this->y(points[j].Value)- rangeYOffSet) * rangeYDiff );
- }
- else //dynamic_cast<ctkBezierControlPoint*>(startCP))
- {
- QList<ctkPoint> points = this->bezierParams(startCP, endCP);
- QList<ctkPoint>::iterator it = points.begin();
- QList<QPointF> bezierPoints;
- foreach(const ctkPoint& p, points)
- {
- bezierPoints <<
- QPointF((p.X - rangeXOffSet)* rangeXDiff ,
- this->rect().height() - (this->y(p.Value) - rangeYOffSet)* rangeYDiff);
- }
- path.cubicTo(bezierPoints[1], bezierPoints[2], bezierPoints[3]);
- d->ControlPoints << bezierPoints[3];
- }
- //qDebug() << i << points[0] << points[1] << points[2] << points[3];
- delete startCP;
- startCP = endCP;
- }
- if (startCP)
- {
- delete startCP;
- }
- painter->setRenderHint(QPainter::Antialiasing);
- painter->setPen(QPen(QColor(255, 255, 255, 191), 1));
- painter->drawPath(path);
- QPainterPath points;
- foreach(const QPointF& point, d->ControlPoints)
- {
- points.addEllipse(point, d->PointSize.width(), d->PointSize.height());
- }
- painter->setBrush(QBrush(QColor(191, 191, 191, 127)));
- painter->drawPath(points);
- }
- //-----------------------------------------------------------------------------
- void ctkTransferFunctionControlPointsItem::mousePressEvent(QGraphicsSceneMouseEvent* e)
- {
- CTK_D(ctkTransferFunctionControlPointsItem);
- QRectF pointArea(QPointF(0,0), d->PointSize*2.);
- d->SelectedPoint = -1;
- for(int i = 0; i < d->ControlPoints.count(); ++i)
- {
- pointArea.moveCenter(d->ControlPoints[i]);
- if (pointArea.contains(e->pos()))
- {
- d->SelectedPoint = i;
- break;
- }
- }
- if (d->SelectedPoint < 0)
- {
- e->ignore();
- }
- }
- //-----------------------------------------------------------------------------
- void ctkTransferFunctionControlPointsItem::mouseMoveEvent(QGraphicsSceneMouseEvent* e)
- {
- CTK_D(ctkTransferFunctionControlPointsItem);
- if (d->SelectedPoint < 0)
- {
- e->ignore();
- return;
- }
- qreal range[2];
- this->transferFunction()->range(range);
- qreal newPos = range[0] + e->pos().x() / (this->rect().width() / (range[1] - range[0]));
- newPos = qBound(range[0], newPos, range[1]);
- this->transferFunction()->setControlPointPos(d->SelectedPoint, newPos);
- //this->transferFunction()->setControlPointValue(d->SelectedPoint, e->y());
- }
- //-----------------------------------------------------------------------------
- void ctkTransferFunctionControlPointsItem::mouseReleaseEvent(QGraphicsSceneMouseEvent* e)
- {
- CTK_D(ctkTransferFunctionControlPointsItem);
- if (d->SelectedPoint < 0)
- {
- e->ignore();
- return;
- }
- d->SelectedPoint = -1;
- }
|