ctkTransferFunctionControlPointsItem.cpp 11 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320
  1. /*=========================================================================
  2. Library: CTK
  3. Copyright (c) 2010 Kitware Inc.
  4. Licensed under the Apache License, Version 2.0 (the "License");
  5. you may not use this file except in compliance with the License.
  6. You may obtain a copy of the License at
  7. http://www.commontk.org/LICENSE
  8. Unless required by applicable law or agreed to in writing, software
  9. distributed under the License is distributed on an "AS IS" BASIS,
  10. WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  11. See the License for the specific language governing permissions and
  12. limitations under the License.
  13. =========================================================================*/
  14. /// Qt includes
  15. #include <QColor>
  16. #include <QDebug>
  17. #include <QLinearGradient>
  18. #include <QGraphicsSceneMouseEvent>
  19. #include <QPainter>
  20. #include <QtGlobal>
  21. #include <QVariant>
  22. #include <QKeyEvent>
  23. /// CTK includes
  24. #include "ctkTransferFunctionControlPointsItem.h"
  25. #include "ctkTransferFunctionScene.h"
  26. #include "ctkTransferFunctionWidget.h"
  27. //-----------------------------------------------------------------------------
  28. class ctkTransferFunctionControlPointsItemPrivate:
  29. public ctkPrivate<ctkTransferFunctionControlPointsItem>
  30. {
  31. public:
  32. ctkTransferFunctionControlPointsItemPrivate();
  33. void init();
  34. QList<QPointF> ControlPoints;
  35. QSize PointSize;
  36. int SelectedPoint;
  37. };
  38. //-----------------------------------------------------------------------------
  39. ctkTransferFunctionControlPointsItemPrivate::ctkTransferFunctionControlPointsItemPrivate()
  40. {
  41. this->PointSize = QSize(12,12);
  42. this->SelectedPoint = -1;
  43. }
  44. //-----------------------------------------------------------------------------
  45. void ctkTransferFunctionControlPointsItemPrivate::init()
  46. {
  47. CTK_P(ctkTransferFunctionControlPointsItem);
  48. p->setAcceptedMouseButtons(Qt::LeftButton);
  49. }
  50. //-----------------------------------------------------------------------------
  51. ctkTransferFunctionControlPointsItem::ctkTransferFunctionControlPointsItem(QGraphicsItem* parentGraphicsItem)
  52. :ctkTransferFunctionItem(parentGraphicsItem)
  53. {
  54. CTK_INIT_PRIVATE(ctkTransferFunctionControlPointsItem);
  55. ctk_d()->init();
  56. }
  57. //-----------------------------------------------------------------------------
  58. ctkTransferFunctionControlPointsItem::ctkTransferFunctionControlPointsItem(
  59. ctkTransferFunction* transferFunction, QGraphicsItem* parentItem)
  60. :ctkTransferFunctionItem(transferFunction, parentItem)
  61. {
  62. CTK_INIT_PRIVATE(ctkTransferFunctionControlPointsItem);
  63. ctk_d()->init();
  64. }
  65. //-----------------------------------------------------------------------------
  66. ctkTransferFunctionControlPointsItem::~ctkTransferFunctionControlPointsItem()
  67. {
  68. }
  69. //-----------------------------------------------------------------------------
  70. void ctkTransferFunctionControlPointsItem::paint(
  71. QPainter * painter, const QStyleOptionGraphicsItem * option, QWidget * widget)
  72. {
  73. CTK_D(ctkTransferFunctionControlPointsItem);
  74. int count = this->transferFunction() ? this->transferFunction()->count() : 0;
  75. if (count <= 0)
  76. {
  77. return;
  78. }
  79. ctkTransferFunctionScene* tfScene = dynamic_cast<ctkTransferFunctionScene*>(this->scene());
  80. Q_ASSERT(tfScene);
  81. const QPainterPath& curve = tfScene->curve();
  82. QPen pen(QColor(255, 255, 255, 191), 1);
  83. pen.setCosmetic(true);
  84. painter->setPen(pen);
  85. painter->drawPath(curve);
  86. d->ControlPoints = tfScene->points();
  87. painter->setBrush(QBrush(QColor(191, 191, 191, 127)));
  88. painter->save();
  89. QTransform transform = painter->transform();
  90. painter->setTransform(QTransform());
  91. foreach(const QPointF& point, d->ControlPoints)
  92. {
  93. QPointF pos = transform.map(point);
  94. painter->drawEllipse(pos.x() - d->PointSize.width() / 2,
  95. pos.y() - d->PointSize.height() / 2,
  96. d->PointSize.width(), d->PointSize.width());
  97. //points.addEllipse(point, d->PointSize.width(), d->PointSize.height());
  98. }
  99. painter->restore();
  100. }
  101. //-----------------------------------------------------------------------------
  102. void ctkTransferFunctionControlPointsItem::mousePressEvent(QGraphicsSceneMouseEvent* e)
  103. {
  104. CTK_D(ctkTransferFunctionControlPointsItem);
  105. QWidget* w = e->widget();
  106. ctkTransferFunctionWidget* view = qobject_cast<ctkTransferFunctionWidget*>(e->widget()->parentWidget());
  107. Q_ASSERT(view);
  108. // e->pos() is ok, pointArea should be in the world coordiate
  109. QRect pointViewArea(QPoint(-d->PointSize.width() / 2, -d->PointSize.height() / 2), d->PointSize);
  110. QPolygonF pointItemArea = this->mapFromScene(view->mapToScene(pointViewArea));
  111. d->SelectedPoint = -1;
  112. for(int i = 0; i < d->ControlPoints.count(); ++i)
  113. {
  114. if (pointItemArea.translated(d->ControlPoints[i]).containsPoint(e->pos(), Qt::OddEvenFill))
  115. {
  116. d->SelectedPoint = i;
  117. break;
  118. }
  119. }
  120. if (d->SelectedPoint >= 0)
  121. {
  122. return;
  123. }
  124. ctkTransferFunctionScene* tfScene = dynamic_cast<ctkTransferFunctionScene*>(this->scene());
  125. Q_ASSERT(tfScene);
  126. // convert coordinates
  127. QPointF tfPos = tfScene->mapPointFromScene(e->pos());
  128. // add point to transfer function
  129. // returns index
  130. int index = this->transferFunction()->insertControlPoint( tfPos.x());
  131. //NOT WORKING IN COMPOSITE
  132. // update value of the point
  133. if (!QSharedPointer<ctkControlPoint>(this->transferFunction()->controlPoint(index))->value().canConvert<QColor>())
  134. {
  135. this->transferFunction()->setControlPointValue( index, tfPos.y());
  136. }
  137. }
  138. //-----------------------------------------------------------------------------
  139. void ctkTransferFunctionControlPointsItem::mouseMoveEvent(QGraphicsSceneMouseEvent* e)
  140. {
  141. CTK_D(ctkTransferFunctionControlPointsItem);
  142. if (d->SelectedPoint < 0)
  143. {
  144. e->ignore();
  145. return;
  146. }
  147. ctkTransferFunctionScene* tfScene = dynamic_cast<ctkTransferFunctionScene*>(this->scene());
  148. Q_ASSERT(tfScene);
  149. QPointF newPos = tfScene->mapPointFromScene(e->pos());
  150. // Deal with borders
  151. if(d->SelectedPoint == 0 || d->SelectedPoint == this->transferFunction()->count() )
  152. {
  153. // BEHAVIOR TO BE IMPLEMENTED
  154. // int borderBehavior = tfScene->borderBehavior();
  155. // LockBorder
  156. // Create new point on border at same height
  157. // Create new point on border on top
  158. // Create new point on border on bottom
  159. qDebug() << "border" ;
  160. return;
  161. }
  162. else
  163. {
  164. // TO BE IMPLEMENTED
  165. //int movePointsBehavior = tfScene->movePointsBehavior();
  166. // initialize to BLOCK_MOVE for now
  167. int movePointsBehavior = STOP_MOVE;
  168. switch( movePointsBehavior ){
  169. case STOP_MOVE:
  170. stopPoints(newPos);
  171. break;
  172. case SWITCH_MOVE:
  173. switchPoints(newPos);
  174. break;
  175. case DRAW_MOVE:
  176. drawPoints(newPos);
  177. break;
  178. case FUSION_MOVE:
  179. fusionPoints(newPos);
  180. break;
  181. default:
  182. break;
  183. }
  184. }
  185. }
  186. //-----------------------------------------------------------------------------
  187. void ctkTransferFunctionControlPointsItem::stopPoints( QPointF iPointF )
  188. {
  189. CTK_D(ctkTransferFunctionControlPointsItem);
  190. if( this->transferFunction()->controlPoint(d->SelectedPoint - 1)->x() > iPointF.x() ||
  191. this->transferFunction()->controlPoint(d->SelectedPoint + 1)->x() < iPointF.x())
  192. {
  193. return;
  194. }
  195. updatePointPosition(iPointF);
  196. }
  197. //-----------------------------------------------------------------------------
  198. void ctkTransferFunctionControlPointsItem::switchPoints( QPointF iPointF )
  199. {
  200. CTK_D(ctkTransferFunctionControlPointsItem);
  201. // Increment or decrement selected point?
  202. // Don't need to check borders since it is done just before calling this method...
  203. if( this->transferFunction()->controlPoint(d->SelectedPoint - 1)->x() > iPointF.x() )
  204. {
  205. // NOT WORKING IF COMPOSITE TRANSFER FUNCTION
  206. double value = this->transferFunction()->value(d->SelectedPoint-1 ).toDouble();
  207. d->SelectedPoint -= 1;
  208. this->transferFunction()->setControlPointValue(d->SelectedPoint+1, value);
  209. }
  210. else if ( this->transferFunction()->controlPoint(d->SelectedPoint + 1)->x() < iPointF.x() )
  211. {
  212. double value = this->transferFunction()->value(d->SelectedPoint + 1 ).toDouble();
  213. d->SelectedPoint += 1;
  214. this->transferFunction()->setControlPointValue(d->SelectedPoint-1, value);
  215. }
  216. updatePointPosition(iPointF);
  217. }
  218. //-----------------------------------------------------------------------------
  219. void ctkTransferFunctionControlPointsItem::drawPoints( QPointF iPointF )
  220. {
  221. CTK_D(ctkTransferFunctionControlPointsItem);
  222. // Increment or decrement selected point
  223. if( this->transferFunction()->controlPoint(d->SelectedPoint - 1)->x() > iPointF.x() )
  224. {
  225. //change the selected point
  226. d->SelectedPoint = d->SelectedPoint -1;
  227. }
  228. else if ( this->transferFunction()->controlPoint(d->SelectedPoint + 1)->x() < iPointF.x() )
  229. {
  230. d->SelectedPoint = d->SelectedPoint +1;
  231. }
  232. updatePointPosition(iPointF);
  233. }
  234. //-----------------------------------------------------------------------------
  235. void ctkTransferFunctionControlPointsItem::fusionPoints( QPointF iPointF )
  236. {
  237. // TO BE IMPLEMENTED
  238. // if 2 points are to close: delete one..?
  239. }
  240. //-----------------------------------------------------------------------------
  241. void ctkTransferFunctionControlPointsItem::updatePointPosition( QPointF iPoint )
  242. {
  243. CTK_D(ctkTransferFunctionControlPointsItem);
  244. this->transferFunction()->setControlPointPos(d->SelectedPoint, iPoint.x());
  245. // TEST NOT WORKING IN COMPOSITE TRANSFER FUNCTION
  246. if ( ! QSharedPointer<ctkControlPoint>(this->transferFunction()->controlPoint(d->SelectedPoint))->value().canConvert<QColor>() )
  247. {
  248. this->transferFunction()->setControlPointValue(d->SelectedPoint, iPoint.y());
  249. }
  250. }
  251. //-----------------------------------------------------------------------------
  252. void ctkTransferFunctionControlPointsItem::mouseReleaseEvent(QGraphicsSceneMouseEvent* e)
  253. {
  254. CTK_D(ctkTransferFunctionControlPointsItem);
  255. if (d->SelectedPoint < 0)
  256. {
  257. e->ignore();
  258. return;
  259. }
  260. d->SelectedPoint = -1;
  261. }
  262. //-----------------------------------------------------------------------------
  263. void ctkTransferFunctionControlPointsItem::keyPressEvent ( QKeyEvent * keyEvent )
  264. {
  265. CTK_D(ctkTransferFunctionControlPointsItem);
  266. // if a point is selected, "d" and "D" = delete the point
  267. if (d->SelectedPoint >= 0)
  268. {
  269. if ( ! QString::compare(keyEvent->text(), "d") || ! QString::compare(keyEvent->text(), "D") )
  270. {
  271. // Get value of selected point
  272. qreal xValue = this->transferFunction()->controlPoint( d->SelectedPoint )->x();
  273. // remove point using value ( can't use point ID because it could engender
  274. // problems when multiple points selection implemented)
  275. this->transferFunction()->removeControlPoint( xValue );
  276. }
  277. }
  278. }