git.sesse.net Git - kdenlive/blob - src/onmonitoritems/rotoscoping/splineitem.cpp

   1 /***************************************************************************
   2  *   Copyright (C) 2011 by Till Theato (root@ttill.de)                     *
   3  *   This file is part of Kdenlive (www.kdenlive.org).                     *
   4  *                                                                         *
   5  *   Kdenlive is free software: you can redistribute it and/or modify      *
   6  *   it under the terms of the GNU General Public License as published by  *
   7  *   the Free Software Foundation, either version 2 of the License, or     *
   8  *   (at your option) any later version.                                   *
   9  *                                                                         *
  10  *   Kdenlive is distributed in the hope that it will be useful,           *
  11  *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
  12  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
  13  *   GNU General Public License for more details.                          *
  14  *                                                                         *
  15  *   You should have received a copy of the GNU General Public License     *
  16  *   along with Kdenlive.  If not, see <http://www.gnu.org/licenses/>.     *
  17  ***************************************************************************/
  18
  19 #include "splineitem.h"
  20 #include "bpointitem.h"
  21 #include "nearestpoint.h"
  22
  23 #include <QGraphicsScene>
  24 #include <QCursor>
  25 #include <QGraphicsSceneMouseEvent>
  26
  27
  28 inline QPointF closestPointInRect(QPointF point, QRectF rect)
  29 {
  30     QPointF closest;
  31     rect = rect.normalized();
  32     closest.setX(qBound<qreal>(rect.left(), point.x(), rect.right()));
  33     closest.setY(qBound<qreal>(rect.top(), point.y(), rect.bottom()));
  34     return closest;
  35 }
  36
  37 void deCasteljau(BPoint *p1, BPoint *p2, BPoint *res, double t)
  38 {
  39     QPointF ab, bc, cd;
  40
  41     ab = QLineF(p1->p, p1->h2).pointAt(t);
  42     bc = QLineF(p1->h2, p2->h1).pointAt(t);
  43     cd = QLineF(p2->h1, p2->p).pointAt(t);
  44
  45     res->h1 = QLineF(ab, bc).pointAt(t);
  46     res->h2 = QLineF(bc, cd).pointAt(t);
  47     res->p = QLineF(res->h1, res->h2).pointAt(t);
  48
  49     p1->h2 = ab;
  50     p2->h1 = cd;
  51 }
  52
  53
  54 SplineItem::SplineItem(const QList< BPoint >& points, QGraphicsItem* parent, QGraphicsScene *scene) :
  55     QGraphicsPathItem(parent, scene)
  56 {
  57     QPen framepen(Qt::SolidLine);
  58     framepen.setColor(Qt::yellow);
  59     setPen(framepen);
  60     setBrush(Qt::NoBrush);
  61     setAcceptHoverEvents(true);
  62
  63     if (points.isEmpty())
  64         return;
  65
  66     QPainterPath path(points.at(0).p);
  67     int j;
  68     for (int i = 0; i < points.count(); ++i) {
  69         new BPointItem(points.at(i), this);
  70         j = (i + 1) % points.count();
  71         path.cubicTo(points.at(i).h2, points.at(j).h1, points.at(j).p);
  72     }
  73     setPath(path);
  74 }
  75
  76 int SplineItem::type() const
  77 {
  78     return Type;
  79 }
  80
  81 void SplineItem::updateSpline()
  82 {
  83     QPainterPath path(qgraphicsitem_cast<BPointItem *>(childItems().at(0))->getPoint().p);
  84
  85     BPoint p1, p2;
  86     int j;
  87     for (int i = 0; i < childItems().count(); ++i) {
  88         j = (i + 1) % childItems().count();
  89         p1 = qgraphicsitem_cast<BPointItem *>(childItems().at(i))->getPoint();
  90         p2 = qgraphicsitem_cast<BPointItem *>(childItems().at(j))->getPoint();
  91         path.cubicTo(p1.h2, p2.h1, p2.p);
  92     }
  93     setPath(path);
  94
  95     emit changed();
  96 }
  97
  98 QList <BPoint> SplineItem::getPoints()
  99 {
 100     QList <BPoint> points;
 101     foreach (QGraphicsItem *child, childItems())
 102         points << qgraphicsitem_cast<BPointItem *>(child)->getPoint();
 103     return points;
 104 }
 105
 106 void SplineItem::mousePressEvent(QGraphicsSceneMouseEvent* event)
 107 {
 108     QGraphicsItem::mousePressEvent(event);
 109
 110     if (!event->isAccepted()) {
 111         QRectF r(event->scenePos() - QPointF(6, 6), QSizeF(12, 12));
 112         if (path().intersects(r) && !path().contains(r)) {
 113             double t = 0;
 114             BPointItem *i1, *i2;
 115             BPoint p, p1, p2;
 116             int ix = getClosestPointOnCurve(event->scenePos(), &t);
 117             i1 = qgraphicsitem_cast<BPointItem *>(childItems().at(ix));
 118             i2 = qgraphicsitem_cast<BPointItem *>(childItems().at((ix + 1) % childItems().count()));
 119             p1 = i1->getPoint();
 120             p2 = i2->getPoint();
 121
 122             deCasteljau(&p1, &p2, &p, t);
 123
 124             i1->setPoint(p1);
 125             i2->setPoint(p2);
 126
 127 #if QT_VERSION >= 0x040600
 128             BPointItem *i = new BPointItem(p, this);
 129             i->stackBefore(i2);
 130 #else
 131             QList <BPoint> points;
 132             BPointItem *item;
 133             while (childItems().count()) {
 134                 item = qgraphicsitem_cast<BPointItem *>(childItems().takeFirst());
 135                 points.append(item->getPoint());
 136                 delete item;
 137             }
 138             int j = 0;
 139             for ( ; j < points.count(); ++j) {
 140                 if (j == ix + 1)
 141                     new BPointItem(p, this);
 142                 new BPointItem(points.at(j), this);
 143             }
 144             if (j == ix + 1)
 145                 new BPointItem(p, this);
 146 #endif
 147
 148         }
 149     }
 150 }
 151
 152 void SplineItem::mouseMoveEvent(QGraphicsSceneMouseEvent* event)
 153 {
 154     QGraphicsItem::mouseMoveEvent(event);
 155 }
 156
 157 void SplineItem::mouseReleaseEvent(QGraphicsSceneMouseEvent* event)
 158 {
 159     QGraphicsItem::mouseReleaseEvent(event);
 160 }
 161
 162 void SplineItem::hoverMoveEvent(QGraphicsSceneHoverEvent* event)
 163 {
 164     QGraphicsItem::hoverMoveEvent(event);
 165
 166     QRectF r(event->scenePos() - QPointF(6, 6), QSizeF(12, 12));
 167     if (path().intersects(r) && !path().contains(r))
 168         setCursor(QCursor(Qt::PointingHandCursor));
 169     else
 170         unsetCursor();
 171 }
 172
 173 int SplineItem::getClosestPointOnCurve(QPointF point, double *tFinal)
 174 {
 175     // TODO: proper minDiff
 176     qreal diff = 10000, param = 0;
 177     BPoint p1, p2;
 178     int curveSegment = 0, j;
 179     for (int i = 0; i < childItems().count(); ++i) {
 180         j = (i + 1) % childItems().count();
 181         p1 = qgraphicsitem_cast<BPointItem *>(childItems().at(i))->getPoint();
 182         p2 = qgraphicsitem_cast<BPointItem *>(childItems().at(j))->getPoint();
 183         QPolygonF bounding = QPolygonF() << p1.p << p1.h2 << p2.h1 << p2.p;
 184         QPointF cl = closestPointInRect(point, bounding.boundingRect());
 185         qreal d = (point - cl).manhattanLength();
 186
 187         if (d > diff)
 188             continue;
 189
 190         Point2 b[4], p;
 191         double t;
 192
 193         b[0].x = p1.p.x();
 194         b[0].y = p1.p.y();
 195         b[1].x = p1.h2.x();
 196         b[1].y = p1.h2.y();
 197         b[2].x = p2.h1.x();
 198         b[2].y = p2.h1.y();
 199         b[3].x = p2.p.x();
 200         b[3].y = p2.p.y();
 201
 202         p.x = point.x();
 203         p.y = point.y();
 204
 205         Point2 n = NearestPointOnCurve(p, b, &t);
 206         cl.setX(n.x);
 207         cl.setY(n.y);
 208
 209         d = (point - cl).manhattanLength();
 210         if (d < diff) {
 211             diff = d;
 212             param = t;
 213             curveSegment = i;
 214         }
 215     }
 216
 217     *tFinal = param;
 218     return curveSegment;
 219 }
 220
 221 #include "splineitem.moc"