yamf/tools/polyline/polyline.cpp

   1 /***************************************************************************
   2  *   Copyright (C) 2007 by Jorge Cuadrado                                  *
   3  *   kuadrosxx@gmail.com                                                     *
   4  *                                                                         *
   5  *   This program 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  *   This program 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 this program; if not, write to the                         *
  17  *   Free Software Foundation, Inc.,                                       *
  18  *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
  19  ***************************************************************************/
  20
  21 #include "polyline.h"
  22
  23 #include <drawing/photogram.h>
  24 #include <drawing/brushmanager.h>
  25 #include <drawing/paintarea.h>
  26
  27 #include <dgraphics/patheditor.h>
  28 #include <dgui/action.h>
  29 #include <dgui/iconloader.h>
  30 #include <dcore/debug.h>
  31
  32 #include <selection/private/contoureditor.h>
  33 #include <model/command/editnodesitem.h>
  34
  35 #include <model/frame.h>
  36 #include <item/path.h>
  37
  38
  39 #include <QPointF>
  40 #include <QIcon>
  41 #include <QKeySequence>
  42 #include <QGraphicsPathItem>
  43 #include <QPainterPath>
  44 #include <QMatrix>
  45 #include <QGraphicsLineItem>
  46 #include <QGraphicsView>
  47
  48 #include <QGraphicsSceneMouseEvent>
  49
  50 namespace YAMF {
  51 namespace Drawing {
  52 namespace Tool {
  53
  54 struct PolyLine::Private
  55 {
  56         bool begin, moved;
  57         QPointF center;
  58         QPointF right;
  59         Drawing::Tool::Private::ContourEditor *editor;
  60         QPainterPath path;
  61         QPainterPath lastPart;
  62
  63         YAMF::Item::Path *item;
  64         PaintArea * paintArea;
  65         DGui::Action *action;
  66
  67         QGraphicsLineItem *line1, *line2;
  68
  69         QGraphicsPathItem *line3;
  70
  71
  72
  73 };
  74
  75 PolyLine::PolyLine(QObject *parent): AbstractTool(parent), d(new Private)
  76 {
  77         d->begin = false;
  78
  79         d->editor = 0;
  80         d->item = 0;
  81         d->paintArea = 0;
  82         d->line1 = new QGraphicsLineItem(0,0,0,0);
  83         d->line1->setPen ( QPen(Qt::red) );
  84         d->line2 = new QGraphicsLineItem(0,0,0,0);
  85         d->line2->setPen ( QPen(Qt::green) );
  86
  87         d->line3 = new QGraphicsPathItem();
  88
  89         d->action = new DGui::Action( DGui::IconLoader::self()->load("draw-path.svg"), tr("Poly line"), this);
  90         d->action->setShortcut( QKeySequence(tr("")) );
  91 }
  92
  93
  94 PolyLine::~PolyLine()
  95 {
  96         delete d;
  97 }
  98
  99 void PolyLine::init(Photogram *photogram)
 100 {
 101         endItem();
 102         //FIXME
 103         foreach(QGraphicsView *view,  photogram->views() )
 104         {
 105                 view->setDragMode ( QGraphicsView::NoDrag );
 106
 107                 Q_CHECK_PTR(view->scene());
 108                 if ( QGraphicsScene *sscene = qobject_cast<QGraphicsScene *>(view->scene()) )
 109                 {
 110                         foreach(QGraphicsItem *item, sscene->items() )
 111                         {
 112                                 item->setFlag(QGraphicsItem::ItemIsSelectable, false);
 113                                 item->setFlag(QGraphicsItem::ItemIsMovable, false);
 114                         }
 115                         sscene->addItem( d->line1 );
 116                         sscene->addItem( d->line2 );
 117                 }
 118         }
 119 }
 120
 121 QString PolyLine::id() const
 122 {
 123         return tr("PolyLine");
 124 }
 125
 126 void PolyLine::press(const QGraphicsSceneMouseEvent *input, PaintArea *const paintArea)
 127 {
 128         d->paintArea = paintArea;
 129         QGraphicsScene *scene = paintArea->scene();
 130         if(scene->items().contains(d->line3))
 131         {
 132                 scene->removeItem(d->line3);
 133         }
 134
 135         BrushManager *brushManager = paintArea->brushManager();
 136
 137         scene->clearSelection();
 138
 139         if(!d->item)
 140         {
 141                 d->path = QPainterPath();
 142                 d->path.moveTo(input->scenePos());
 143
 144                 d->item = new YAMF::Item::Path();
 145
 146                 d->editor = new Drawing::Tool::Private::ContourEditor(d->item, scene);
 147                 connect(d->editor, SIGNAL(nodeClicked()), this, SLOT(nodeChanged()));
 148                 scene->addItem( d->item );
 149
 150
 151                 d->begin = true;
 152                 d->moved = false;
 153         }
 154         else
 155         {
 156                 d->begin = false;
 157
 158                 if(!d->moved)
 159                 {
 160                         d->path.cubicTo(d->center, d->center, input->scenePos());
 161                 }
 162                 else
 163                 {
 164                         d->moved = false;
 165                         d->path.cubicTo(d->center, d->right, input->scenePos());
 166                 }
 167
 168         }
 169         d->center = input->scenePos();
 170
 171         d->item->setPen( brushManager->pen() );
 172 }
 173
 174 void PolyLine::move(const QGraphicsSceneMouseEvent *input, PaintArea *const paintArea)
 175 {
 176         QGraphicsScene *scene = paintArea->scene();
 177         if(d->item)
 178         {
 179                 if(input->buttons() == Qt::NoButton)
 180                 {
 181                         QPainterPath lastPart;
 182
 183                         lastPart.moveTo(d->center);
 184
 185                         if(!d->moved)
 186                         {
 187 //                              lastPart.lineTo(input->scenePos());
 188                                 lastPart.cubicTo(d->center, d->center, input->scenePos());
 189                         }
 190                         else
 191                         {
 192                                 lastPart.cubicTo(d->center, d->right, input->scenePos());
 193                         }
 194
 195                         d->line3->setPath(lastPart);
 196                         if(!scene->items().contains(d->line3))
 197                         {
 198                                 scene->addItem(d->line3);
 199                         }
 200
 201                         if(!scene->items().contains(d->line1))
 202                         {
 203                                 scene->addItem( d->line1 );
 204                         }
 205                         if(!scene->items().contains(d->line2))
 206                         {
 207                                 scene->addItem( d->line2 );
 208                         }
 209                 }
 210                 else if(d->editor && d->editor->isSelected())
 211                 {
 212                         d->path = d->item->path();
 213                 }
 214                 else if(input->buttons() == Qt::LeftButton)
 215                 {
 216                         paintArea->setDragMode(QGraphicsView::NoDrag);
 217
 218                         if(d->begin)
 219                         {
 220                                 d->moved = true;
 221                                 d->right = input->scenePos();
 222
 223                                 d->line2->setLine(QLineF(d->right, d->center));
 224                                 if(!scene->items().contains(d->line2))
 225                                 {
 226                                         scene->addItem( d->line2 );
 227                                 }
 228                         }
 229                         else
 230                         {
 231
 232                                 int index = d->path.elementCount()-3;
 233
 234                                 if( d->path.elementAt(index).type == QPainterPath::CurveToElement )
 235                                 {
 236                                         if(!d->moved)
 237                                         {
 238                                                 QPointF point = d->path.elementAt(index+1);
 239                                                 d->path.setElementPositionAt(index, point.x(), point.y() );
 240                                         }
 241
 242                                         QPointF mirror = d->center - ( input->scenePos() - d->center);
 243
 244                                         d->path.setElementPositionAt(index+1, mirror.x(), mirror.y());
 245
 246                                         d->line1->setLine(QLineF(mirror, d->center));
 247                                         d->line2->setLine(QLineF(d->right, d->center));
 248
 249
 250                                         if(!scene->items().contains(d->line1))
 251                                         {
 252                                                 scene->addItem( d->line1 );
 253                                         }
 254                                         if(!scene->items().contains(d->line2))
 255                                         {
 256                                                 scene->addItem( d->line2 );
 257                                         }
 258                                 }
 259
 260                                 d->moved = true;
 261                                 d->right = input->scenePos();
 262                                 d->item->setPath(d->path);
 263
 264                         }
 265
 266                 }
 267         }
 268 }
 269
 270 void PolyLine::release(const QGraphicsSceneMouseEvent *input, PaintArea *const paintArea)
 271 {
 272         Q_UNUSED(input);
 273
 274         d->editor->expandAllNodes ();
 275
 276
 277         if(d->begin)
 278         {
 279                 paintArea->currentFrame()->addItem(d->item);
 280         }
 281         else if(!d->editor->isSelected())
 282         {
 283                 int position  = paintArea->currentFrame()->visualIndexOf(d->item);
 284
 285                 if(position != -1)
 286                 {
 287                         QPainterPath path = d->item->path();
 288                         d->editor->restore();
 289                         paintArea->addCommand(new Command::EditNodesItem(d->item, d->path));
 290                         d->editor->save();
 291                 }
 292         }
 293 }
 294
 295 void PolyLine::doubleClick(const QGraphicsSceneMouseEvent *input, PaintArea *const paintArea)
 296 {
 297         Q_UNUSED(input);
 298         Q_UNUSED(paintArea);
 299         endItem();
 300
 301 }
 302
 303 DGui::Action *PolyLine::action() const
 304 {
 305         return d->action;
 306 }
 307
 308 int PolyLine::type() const
 309 {
 310         return AbstractTool::Brush;
 311 }
 312
 313 void PolyLine::aboutToChangeTool()
 314 {
 315         endItem();
 316 }
 317
 318 void PolyLine::photogramChanged(Photogram *const photogram)
 319 {
 320         Q_UNUSED(photogram);
 321         if(d->item)
 322         {
 323 //      FIXME:
 324 //              d->path = d->item->path();
 325 //              d->editor->setItem(d->item);
 326 //              d->editor->show();
 327 //              d->editor->expandAllNodes();
 328                 int size = d->item->path().elementCount();
 329                 if(size > 0)
 330                 {
 331                         d->center = d->item->path().elementAt(size -1);
 332                 }
 333         }
 334 }
 335
 336 void PolyLine::endItem()
 337 {
 338         if(d->item)
 339         {
 340                 d->path = QPainterPath();
 341                 d->item = 0;
 342                 delete d->editor;
 343                 d->editor = 0;
 344         }
 345 }
 346
 347 void PolyLine::nodeChanged()
 348 {
 349
 350         if(d->paintArea)
 351         {
 352                 int position  = d->paintArea->currentFrame()->visualIndexOf(d->item);
 353
 354                 if(position != -1)
 355                 {
 356
 357                         int size = d->item->path().elementCount();
 358                         if(size > 0)
 359                         {
 360                                 d->center = d->item->path().elementAt(size -1);
 361
 362                         }
 363                         d->path = d->item->path();
 364
 365                         d->editor->restore();
 366                         d->paintArea->addCommand(new Command::EditNodesItem(d->item, d->path));
 367                         d->editor->save();
 368                 }
 369         }
 370
 371
 372
 373 }
 374
 375 }
 376 }
 377 }