]> git.sesse.net Git - kdenlive/blob - src/onmonitoritems/rotoscoping/bpointitem.cpp
rotoscoping: allow adding new points to the spline
[kdenlive] / src / onmonitoritems / rotoscoping / bpointitem.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 "bpointitem.h"
20 #include "splineitem.h"
21
22 #include <QPainter>
23 #include <QGraphicsSceneMouseEvent>
24 #include <QCursor>
25
26
27 BPointItem::BPointItem(BPoint point, QGraphicsItem* parent) :
28         QAbstractGraphicsShapeItem(parent),
29         m_selection(-1)
30 {
31     setFlags(QGraphicsItem::ItemIsMovable | QGraphicsItem::ItemIsSelectable);
32
33     QPen framepen(Qt::SolidLine);
34     framepen.setColor(Qt::yellow);
35     setPen(framepen);
36     setBrush(Qt::NoBrush);
37     setAcceptHoverEvents(true);
38
39     setPos(point.p);
40     m_point.h1 = mapFromScene(point.h1);
41     m_point.p = mapFromScene(point.p);
42     m_point.h2 = mapFromScene(point.h2);
43     m_point.handlesLinked = false;
44 }
45
46 BPoint BPointItem::getPoint()
47 {
48     return BPoint(mapToScene(m_point.h1), mapToScene(m_point.p), mapToScene(m_point.h2));
49 }
50
51 void BPointItem::setPoint(BPoint point)
52 {
53     setPos(point.p);
54     prepareGeometryChange();
55     m_point.h1 = mapFromScene(point.h1);
56     m_point.p = mapFromScene(point.p);
57     m_point.h2 = mapFromScene(point.h2);
58 }
59
60 int BPointItem::type() const
61 {
62     return Type;
63 }
64
65 QRectF BPointItem::boundingRect() const
66 {
67     QPolygonF p = QPolygonF() << m_point.h1 << m_point.p << m_point.h2;
68     return p.boundingRect().adjusted(-6, -6, 6, 6);
69 }
70
71 void BPointItem::paint(QPainter* painter, const QStyleOptionGraphicsItem* option, QWidget* widget)
72 {
73     Q_UNUSED(option);
74     Q_UNUSED(widget);
75
76     painter->setPen(QPen(Qt::yellow, 1, Qt::SolidLine));
77     painter->setBrush(QBrush(isSelected() ? Qt::red : Qt::yellow));
78     painter->setRenderHint(QPainter::Antialiasing);
79
80     double handleSize = 6 / painter->matrix().m11();
81     double handleSizeHalf = handleSize / 2;
82
83     QPolygonF handle = QPolygonF() << QPointF(0, -handleSizeHalf) << QPointF(handleSizeHalf, 0) << QPointF(0, handleSizeHalf) << QPointF(-handleSizeHalf, 0);
84
85     painter->drawLine(m_point.h1, m_point.p);
86     painter->drawLine(m_point.p, m_point.h2);
87
88 #if QT_VERSION >= 0x040600
89     painter->drawConvexPolygon(handle.translated(m_point.h1.x(), m_point.h1.y()));
90     painter->drawConvexPolygon(handle.translated(m_point.h2.x(), m_point.h2.y()));
91 #else
92     tmp = handle;
93     tmp.translate(m_point.h1.x(), m_point.h1.y());
94     p.drawConvexPolygon(tmp);
95     tmp.translate(m_point.h2.x(), m_point.h2.y());
96     p.drawConvexPolygon(tmp);
97 #endif
98
99     painter->drawEllipse(QRectF(m_point.p.x() - handleSizeHalf,
100                                 m_point.p.y() - handleSizeHalf, handleSize, handleSize));
101 }
102
103 int BPointItem::getSelection(QPointF pos)
104 {
105     QPainterPath mouseArea;
106     mouseArea.addRect(pos.x() - 6, pos.y() - 6, 12, 12);
107
108     if (mouseArea.contains(m_point.h1))
109         return 0;
110     else if (mouseArea.contains(m_point.p))
111         return 1;
112     else if (mouseArea.contains(m_point.h2))
113         return 2;
114
115     return -1;
116 }
117
118 void BPointItem::mousePressEvent(QGraphicsSceneMouseEvent* event)
119 {
120     m_selection = getSelection(event->pos());
121
122     if (m_selection < 0) {
123         event->ignore();
124         setSelected(false);
125     } else {
126         setSelected(true);
127     }
128 }
129
130 void BPointItem::mouseMoveEvent(QGraphicsSceneMouseEvent* event)
131 {
132     prepareGeometryChange();
133     switch (m_selection) {
134     case 0:
135         m_point.setH1(event->pos());
136         break;
137     case 1:
138         m_point.setP(event->pos());
139         break;
140     case 2:
141         m_point.setH2(event->pos());
142         break;
143     }
144
145     if (parentItem()) {
146         SplineItem *parent = qgraphicsitem_cast<SplineItem*>(parentItem());
147         if (parent)
148             parent->updateSpline();
149     }
150 }
151
152 void BPointItem::mouseReleaseEvent(QGraphicsSceneMouseEvent* event)
153 {
154 QGraphicsItem::mouseReleaseEvent(event);
155 }
156
157 void BPointItem::hoverMoveEvent(QGraphicsSceneHoverEvent* event)
158 {
159     if (getSelection(event->pos()) < 0)
160         unsetCursor();
161     else
162         setCursor(QCursor(Qt::PointingHandCursor));
163 }