git.sesse.net Git - kdenlive/blob - 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     setAcceptHoverEvents(true);
  34
  35     setPos(point.p);
  36     m_point.h1 = mapFromScene(point.h1);
  37     m_point.p = mapFromScene(point.p);
  38     m_point.h2 = mapFromScene(point.h2);
  39     m_point.handlesLinked = false;
  40 }
  41
  42 BPoint BPointItem::getPoint()
  43 {
  44     return BPoint(mapToScene(m_point.h1), mapToScene(m_point.p), mapToScene(m_point.h2));
  45 }
  46
  47 void BPointItem::setPoint(BPoint point)
  48 {
  49     setPos(point.p);
  50     prepareGeometryChange();
  51     m_point.h1 = mapFromScene(point.h1);
  52     m_point.p = mapFromScene(point.p);
  53     m_point.h2 = mapFromScene(point.h2);
  54 }
  55
  56 int BPointItem::type() const
  57 {
  58     return Type;
  59 }
  60
  61 QRectF BPointItem::boundingRect() const
  62 {
  63     QPolygonF p = QPolygonF() << m_point.h1 << m_point.p << m_point.h2;
  64     return p.boundingRect().adjusted(-6, -6, 6, 6);
  65 }
  66
  67 void BPointItem::paint(QPainter* painter, const QStyleOptionGraphicsItem* option, QWidget* widget)
  68 {
  69     Q_UNUSED(option);
  70     Q_UNUSED(widget);
  71
  72     if (isEnabled()) {
  73         painter->setPen(QPen(Qt::yellow, 1, Qt::SolidLine));
  74         painter->setBrush(QBrush(isSelected() ? Qt::red : Qt::yellow));
  75     } else {
  76         painter->setPen(QPen(Qt::gray, 1, Qt::SolidLine));
  77         painter->setBrush(QBrush(Qt::gray));
  78     }
  79     painter->setRenderHint(QPainter::Antialiasing);
  80
  81     double handleSize = 6 / painter->matrix().m11();
  82     double handleSizeHalf = handleSize / 2;
  83
  84     QPolygonF handle = QPolygonF() << QPointF(0, -handleSizeHalf) << QPointF(handleSizeHalf, 0) << QPointF(0, handleSizeHalf) << QPointF(-handleSizeHalf, 0);
  85
  86     painter->drawLine(m_point.h1, m_point.p);
  87     painter->drawLine(m_point.p, m_point.h2);
  88
  89 #if QT_VERSION >= 0x040600
  90     painter->drawConvexPolygon(handle.translated(m_point.h1.x(), m_point.h1.y()));
  91     painter->drawConvexPolygon(handle.translated(m_point.h2.x(), m_point.h2.y()));
  92 #else
  93     tmp = handle;
  94     tmp.translate(m_point.h1.x(), m_point.h1.y());
  95     p.drawConvexPolygon(tmp);
  96     tmp.translate(m_point.h2.x(), m_point.h2.y());
  97     p.drawConvexPolygon(tmp);
  98 #endif
  99
 100     painter->drawEllipse(QRectF(m_point.p.x() - handleSizeHalf,
 101                                 m_point.p.y() - handleSizeHalf, handleSize, handleSize));
 102 }
 103
 104 int BPointItem::getSelection(QPointF pos)
 105 {
 106     QRectF mouseArea(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         if (event->button() == Qt::RightButton && m_selection == 1) {
 127             SplineItem *parent = qgraphicsitem_cast<SplineItem*>(parentItem());
 128             if (parent) {
 129                 parent->removeChild(this);
 130                 return;
 131             }
 132         }
 133         setSelected(true);
 134     }
 135 }
 136
 137 void BPointItem::mouseMoveEvent(QGraphicsSceneMouseEvent* event)
 138 {
 139     prepareGeometryChange();
 140     switch (m_selection) {
 141     case 0:
 142         m_point.setH1(event->pos());
 143         break;
 144     case 1:
 145         m_point.setP(event->pos());
 146         break;
 147     case 2:
 148         m_point.setH2(event->pos());
 149         break;
 150     }
 151
 152     if (parentItem()) {
 153         SplineItem *parent = qgraphicsitem_cast<SplineItem*>(parentItem());
 154         if (parent)
 155             parent->updateSpline(true);
 156     }
 157 }
 158
 159 void BPointItem::mouseReleaseEvent(QGraphicsSceneMouseEvent* event)
 160 {
 161     if (parentItem()) {
 162         SplineItem *parent = qgraphicsitem_cast<SplineItem*>(parentItem());
 163         if (parent)
 164             parent->updateSpline(false);
 165     }
 166     QGraphicsItem::mouseReleaseEvent(event);
 167 }
 168
 169 void BPointItem::hoverMoveEvent(QGraphicsSceneHoverEvent* event)
 170 {
 171     if (getSelection(event->pos()) < 0)
 172         unsetCursor();
 173     else
 174         setCursor(QCursor(Qt::PointingHandCursor));
 175 }