src/kcategorizedview.cpp

   1 /**
   2   * This file is part of the KDE project
   3   * Copyright (C) 2007 Rafael Fernández López <ereslibre@kde.org>
   4   *
   5   * This library is free software; you can redistribute it and/or
   6   * modify it under the terms of the GNU Library General Public
   7   * License as published by the Free Software Foundation; either
   8   * version 2 of the License, or (at your option) any later version.
   9   *
  10   * This library 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 GNU
  13   * Library General Public License for more details.
  14   *
  15   * You should have received a copy of the GNU Library General Public License
  16   * along with this library; see the file COPYING.LIB.  If not, write to
  17   * the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
  18   * Boston, MA 02110-1301, USA.
  19   */
  20
  21 #include "kcategorizedview.h"
  22 #include "kcategorizedview_p.h"
  23
  24 #include <math.h> // trunc on C99 compliant systems
  25 #include <kdefakes.h> // trunc for not C99 compliant systems
  26
  27 #include <QPainter>
  28 #include <QScrollBar>
  29 #include <QPaintEvent>
  30
  31 #include <kstyle.h>
  32
  33 #include "kcategorydrawer.h"
  34 #include "kcategorizedsortfilterproxymodel.h"
  35
  36 // By defining DOLPHIN_DRAGANDDROP the custom drag and drop implementation of
  37 // KCategorizedView is bypassed to have a consistent drag and drop look for all
  38 // views. Hopefully transparent pixmaps for drag objects will be supported in
  39 // Qt 4.4, so that this workaround can be skipped.
  40 #define DOLPHIN_DRAGANDDROP
  41
  42 KCategorizedView::Private::Private(KCategorizedView *listView)
  43     : listView(listView)
  44     , categoryDrawer(0)
  45     , biggestItemSize(QSize(0, 0))
  46     , mouseButtonPressed(false)
  47     , rightMouseButtonPressed(false)
  48     , isDragging(false)
  49     , dragLeftViewport(false)
  50     , proxyModel(0)
  51 {
  52 }
  53
  54 KCategorizedView::Private::~Private()
  55 {
  56 }
  57
  58 const QModelIndexList &KCategorizedView::Private::intersectionSet(const QRect &rect)
  59 {
  60     QModelIndex index;
  61     QRect indexVisualRect;
  62
  63     intersectedIndexes.clear();
  64
  65     int itemHeight;
  66
  67     if (listView->gridSize().isEmpty())
  68     {
  69         itemHeight = biggestItemSize.height();
  70     }
  71     else
  72     {
  73         itemHeight = listView->gridSize().height();
  74     }
  75
  76     // Lets find out where we should start
  77     int top = proxyModel->rowCount() - 1;
  78     int bottom = 0;
  79     int middle = (top + bottom) / 2;
  80     while (bottom <= top)
  81     {
  82         middle = (top + bottom) / 2;
  83
  84         index = proxyModel->index(middle, 0);
  85         indexVisualRect = visualRect(index);
  86         // We need the whole height (not only the visualRect). This will help us to update
  87         // all needed indexes correctly (ereslibre)
  88         indexVisualRect.setHeight(indexVisualRect.height() + (itemHeight - indexVisualRect.height()));
  89
  90         if (qMax(indexVisualRect.topLeft().y(),
  91                  indexVisualRect.bottomRight().y()) < qMin(rect.topLeft().y(),
  92                                                            rect.bottomRight().y()))
  93         {
  94             bottom = middle + 1;
  95         }
  96         else
  97         {
  98             top = middle - 1;
  99         }
 100     }
 101
 102     for (int i = middle; i < proxyModel->rowCount(); i++)
 103     {
 104         index = proxyModel->index(i, 0);
 105         indexVisualRect = visualRect(index);
 106
 107         if (rect.intersects(indexVisualRect))
 108             intersectedIndexes.append(index);
 109
 110         // If we passed next item, stop searching for hits
 111         if (qMax(rect.bottomRight().y(), rect.topLeft().y()) <
 112                                                   qMin(indexVisualRect.topLeft().y(),
 113                                                        indexVisualRect.bottomRight().y()))
 114             break;
 115     }
 116
 117     return intersectedIndexes;
 118 }
 119
 120 QRect KCategorizedView::Private::visualRectInViewport(const QModelIndex &index) const
 121 {
 122     if (!index.isValid())
 123         return QRect();
 124
 125     QString curCategory = elementsInfo[index.row()].category;
 126
 127     QRect retRect;
 128
 129     if (listView->layoutDirection() == Qt::LeftToRight)
 130     {
 131         retRect = QRect(listView->spacing(), listView->spacing() * 2 +
 132                         categoryDrawer->categoryHeight(listView->viewOptions()), 0, 0);
 133     }
 134     else
 135     {
 136         retRect = QRect(listView->viewport()->width() - listView->spacing(), listView->spacing() * 2 +
 137                         categoryDrawer->categoryHeight(listView->viewOptions()), 0, 0);
 138     }
 139
 140     int viewportWidth = listView->viewport()->width() - listView->spacing();
 141
 142     int itemHeight;
 143     int itemWidth;
 144
 145     if (listView->gridSize().isEmpty())
 146     {
 147         itemHeight = biggestItemSize.height();
 148         itemWidth = biggestItemSize.width();
 149     }
 150     else
 151     {
 152         itemHeight = listView->gridSize().height();
 153         itemWidth = listView->gridSize().width();
 154     }
 155
 156     int itemWidthPlusSeparation = listView->spacing() + itemWidth;
 157     int elementsPerRow = viewportWidth / itemWidthPlusSeparation;
 158     if (!elementsPerRow)
 159         elementsPerRow++;
 160
 161     int column = elementsInfo[index.row()].relativeOffsetToCategory % elementsPerRow;
 162     int row = elementsInfo[index.row()].relativeOffsetToCategory / elementsPerRow;
 163
 164     if (listView->layoutDirection() == Qt::LeftToRight)
 165     {
 166         retRect.setLeft(retRect.left() + column * listView->spacing() +
 167                         column * itemWidth);
 168     }
 169     else
 170     {
 171         retRect.setLeft(retRect.right() - column * listView->spacing() -
 172                          column * itemWidth - itemWidth);
 173
 174         retRect.setRight(retRect.right() - column * listView->spacing() -
 175                          column * itemWidth);
 176     }
 177
 178     foreach (const QString &category, categories)
 179     {
 180         if (category == curCategory)
 181             break;
 182
 183         float rows = (float) ((float) categoriesIndexes[category].count() /
 184                               (float) elementsPerRow);
 185
 186         int rowsInt = categoriesIndexes[category].count() / elementsPerRow;
 187
 188         if (rows - trunc(rows)) rowsInt++;
 189
 190         retRect.setTop(retRect.top() +
 191                        (rowsInt * itemHeight) +
 192                        categoryDrawer->categoryHeight(listView->viewOptions()) +
 193                        listView->spacing() * 2);
 194
 195         if (listView->gridSize().isEmpty())
 196         {
 197             retRect.setTop(retRect.top() +
 198                            (rowsInt * listView->spacing()));
 199         }
 200     }
 201
 202     if (listView->gridSize().isEmpty())
 203     {
 204         retRect.setTop(retRect.top() + row * listView->spacing() +
 205                        (row * itemHeight));
 206     }
 207     else
 208     {
 209         retRect.setTop(retRect.top() + (row * itemHeight));
 210     }
 211
 212     retRect.setWidth(itemWidth);
 213
 214     QModelIndex heightIndex = proxyModel->index(index.row(), 0);
 215     if (listView->gridSize().isEmpty())
 216     {
 217         retRect.setHeight(listView->sizeHintForIndex(heightIndex).height());
 218     }
 219     else
 220     {
 221         retRect.setHeight(qMin(listView->sizeHintForIndex(heightIndex).height(),
 222                                listView->gridSize().height()));
 223     }
 224
 225     return retRect;
 226 }
 227
 228 QRect KCategorizedView::Private::visualCategoryRectInViewport(const QString &category)
 229                                                                            const
 230 {
 231     QRect retRect(listView->spacing(),
 232                   listView->spacing(),
 233                   listView->viewport()->width() - listView->spacing() * 2,
 234                   0);
 235
 236     if (!proxyModel->rowCount() || !categories.contains(category))
 237         return QRect();
 238
 239     QModelIndex index = proxyModel->index(0, 0, QModelIndex());
 240
 241     int viewportWidth = listView->viewport()->width() - listView->spacing();
 242
 243     int itemHeight;
 244     int itemWidth;
 245
 246     if (listView->gridSize().isEmpty())
 247     {
 248         itemHeight = biggestItemSize.height();
 249         itemWidth = biggestItemSize.width();
 250     }
 251     else
 252     {
 253         itemHeight = listView->gridSize().height();
 254         itemWidth = listView->gridSize().width();
 255     }
 256
 257     int itemWidthPlusSeparation = listView->spacing() + itemWidth;
 258     int elementsPerRow = viewportWidth / itemWidthPlusSeparation;
 259
 260     if (!elementsPerRow)
 261         elementsPerRow++;
 262
 263     foreach (const QString &itCategory, categories)
 264     {
 265         if (itCategory == category)
 266             break;
 267
 268         float rows = (float) ((float) categoriesIndexes[itCategory].count() /
 269                               (float) elementsPerRow);
 270         int rowsInt = categoriesIndexes[itCategory].count() / elementsPerRow;
 271
 272         if (rows - trunc(rows)) rowsInt++;
 273
 274         retRect.setTop(retRect.top() +
 275                        (rowsInt * itemHeight) +
 276                        categoryDrawer->categoryHeight(listView->viewOptions()) +
 277                        listView->spacing() * 2);
 278
 279         if (listView->gridSize().isEmpty())
 280         {
 281             retRect.setTop(retRect.top() +
 282                            (rowsInt * listView->spacing()));
 283         }
 284     }
 285
 286     retRect.setHeight(categoryDrawer->categoryHeight(listView->viewOptions()));
 287
 288     return retRect;
 289 }
 290
 291 // We're sure elementsPosition doesn't contain index
 292 const QRect &KCategorizedView::Private::cacheIndex(const QModelIndex &index)
 293 {
 294     QRect rect = visualRectInViewport(index);
 295     elementsPosition[index.row()] = rect;
 296
 297     return elementsPosition[index.row()];
 298 }
 299
 300 // We're sure categoriesPosition doesn't contain category
 301 const QRect &KCategorizedView::Private::cacheCategory(const QString &category)
 302 {
 303     QRect rect = visualCategoryRectInViewport(category);
 304     categoriesPosition[category] = rect;
 305
 306     return categoriesPosition[category];
 307 }
 308
 309 const QRect &KCategorizedView::Private::cachedRectIndex(const QModelIndex &index)
 310 {
 311     if (elementsPosition.contains(index.row())) // If we have it cached
 312     {                                        // return it
 313         return elementsPosition[index.row()];
 314     }
 315     else                                     // Otherwise, cache it
 316     {                                        // and return it
 317         return cacheIndex(index);
 318     }
 319 }
 320
 321 const QRect &KCategorizedView::Private::cachedRectCategory(const QString &category)
 322 {
 323     if (categoriesPosition.contains(category)) // If we have it cached
 324     {                                                // return it
 325         return categoriesPosition[category];
 326     }
 327     else                                            // Otherwise, cache it and
 328     {                                               // return it
 329         return cacheCategory(category);
 330     }
 331 }
 332
 333 QRect KCategorizedView::Private::visualRect(const QModelIndex &index)
 334 {
 335     QRect retRect = cachedRectIndex(index);
 336     int dx = -listView->horizontalOffset();
 337     int dy = -listView->verticalOffset();
 338     retRect.adjust(dx, dy, dx, dy);
 339
 340     return retRect;
 341 }
 342
 343 QRect KCategorizedView::Private::categoryVisualRect(const QString &category)
 344 {
 345     QRect retRect = cachedRectCategory(category);
 346     int dx = -listView->horizontalOffset();
 347     int dy = -listView->verticalOffset();
 348     retRect.adjust(dx, dy, dx, dy);
 349
 350     return retRect;
 351 }
 352
 353 void KCategorizedView::Private::drawNewCategory(const QModelIndex &index,
 354                                                 int sortRole,
 355                                                 const QStyleOption &option,
 356                                                 QPainter *painter)
 357 {
 358     if (!index.isValid())
 359     {
 360         return;
 361     }
 362
 363     QStyleOption optionCopy = option;
 364     const QString category = proxyModel->data(index, KCategorizedSortFilterProxyModel::CategoryDisplayRole).toString();
 365
 366     optionCopy.state &= ~QStyle::State_Selected;
 367
 368     if ((category == hoveredCategory) && !mouseButtonPressed)
 369     {
 370         optionCopy.state |= QStyle::State_MouseOver;
 371     }
 372     else if ((category == hoveredCategory) && mouseButtonPressed)
 373     {
 374         QPoint initialPressPosition = listView->viewport()->mapFromGlobal(QCursor::pos());
 375         initialPressPosition.setY(initialPressPosition.y() + listView->verticalOffset());
 376         initialPressPosition.setX(initialPressPosition.x() + listView->horizontalOffset());
 377
 378         if (initialPressPosition == this->initialPressPosition)
 379         {
 380             optionCopy.state |= QStyle::State_Selected;
 381         }
 382     }
 383
 384     categoryDrawer->drawCategory(index,
 385                                  sortRole,
 386                                  optionCopy,
 387                                  painter);
 388 }
 389
 390
 391 void KCategorizedView::Private::updateScrollbars()
 392 {
 393     // find the last index in the last category
 394     QModelIndex lastIndex = categoriesIndexes.isEmpty() ? QModelIndex() : categoriesIndexes[categories.last()].last();
 395
 396     int lastItemBottom = cachedRectIndex(lastIndex).top() +
 397                          listView->spacing() + (listView->gridSize().isEmpty() ? 0 : listView->gridSize().height()) - listView->viewport()->height();
 398
 399     listView->horizontalScrollBar()->setRange(0, 0);
 400
 401     listView->verticalScrollBar()->setSingleStep(listView->viewport()->height() / 10);
 402     listView->verticalScrollBar()->setPageStep(listView->viewport()->height());
 403     listView->verticalScrollBar()->setRange(0, lastItemBottom);
 404 }
 405
 406 void KCategorizedView::Private::drawDraggedItems(QPainter *painter)
 407 {
 408     QStyleOptionViewItemV3 option = listView->viewOptions();
 409     option.state &= ~QStyle::State_MouseOver;
 410     foreach (const QModelIndex &index, listView->selectionModel()->selectedIndexes())
 411     {
 412         const int dx = mousePosition.x() - initialPressPosition.x() + listView->horizontalOffset();
 413         const int dy = mousePosition.y() - initialPressPosition.y() + listView->verticalOffset();
 414
 415         option.rect = visualRect(index);
 416         option.rect.adjust(dx, dy, dx, dy);
 417
 418         if (option.rect.intersects(listView->viewport()->rect()))
 419         {
 420             listView->itemDelegate(index)->paint(painter, option, index);
 421         }
 422     }
 423 }
 424
 425 void KCategorizedView::Private::layoutChanged(bool forceItemReload)
 426 {
 427     if ((listView->viewMode() == KCategorizedView::IconMode) && proxyModel &&
 428         categoryDrawer && proxyModel->isCategorizedModel() &&
 429         ((forceItemReload ||
 430           (modelSortRole != proxyModel->sortRole()) ||
 431           (modelSortColumn != proxyModel->sortColumn()) ||
 432           (modelSortOrder != proxyModel->sortOrder()) ||
 433           (modelLastRowCount != proxyModel->rowCount()) ||
 434           (modelCategorized != proxyModel->isCategorizedModel()))))
 435     {
 436         // Force the view to update all elements
 437         listView->rowsInsertedArtifficial(QModelIndex(), 0, proxyModel->rowCount() - 1);
 438
 439         if (!forceItemReload)
 440         {
 441             modelSortRole = proxyModel->sortRole();
 442             modelSortColumn = proxyModel->sortColumn();
 443             modelSortOrder = proxyModel->sortOrder();
 444             modelLastRowCount = proxyModel->rowCount();
 445             modelCategorized = proxyModel->isCategorizedModel();
 446         }
 447     }
 448     else if ((listView->viewMode() == KCategorizedView::IconMode) && proxyModel &&
 449              categoryDrawer && proxyModel->isCategorizedModel())
 450     {
 451         updateScrollbars();
 452     }
 453 }
 454
 455 void KCategorizedView::Private::drawDraggedItems()
 456 {
 457     QRect rectToUpdate;
 458     QRect currentRect;
 459     foreach (const QModelIndex &index, listView->selectionModel()->selectedIndexes())
 460     {
 461         int dx = mousePosition.x() - initialPressPosition.x() + listView->horizontalOffset();
 462         int dy = mousePosition.y() - initialPressPosition.y() + listView->verticalOffset();
 463
 464         currentRect = visualRect(index);
 465         currentRect.adjust(dx, dy, dx, dy);
 466
 467         if (currentRect.intersects(listView->viewport()->rect()))
 468         {
 469             rectToUpdate = rectToUpdate.united(currentRect);
 470         }
 471     }
 472
 473     listView->viewport()->update(lastDraggedItemsRect.united(rectToUpdate));
 474
 475     lastDraggedItemsRect = rectToUpdate;
 476 }
 477
 478
 479 //==============================================================================
 480
 481
 482 KCategorizedView::KCategorizedView(QWidget *parent)
 483     : QListView(parent)
 484     , d(new Private(this))
 485 {
 486 }
 487
 488 KCategorizedView::~KCategorizedView()
 489 {
 490     delete d;
 491 }
 492
 493 void KCategorizedView::setGridSize(const QSize &size)
 494 {
 495     QListView::setGridSize(size);
 496
 497     d->layoutChanged(true);
 498 }
 499
 500 void KCategorizedView::setModel(QAbstractItemModel *model)
 501 {
 502     d->lastSelection = QItemSelection();
 503     d->forcedSelectionPosition = 0;
 504     d->elementsInfo.clear();
 505     d->elementsPosition.clear();
 506     d->categoriesIndexes.clear();
 507     d->categoriesPosition.clear();
 508     d->categories.clear();
 509     d->intersectedIndexes.clear();
 510     d->modelIndexList.clear();
 511     d->hovered = QModelIndex();
 512     d->mouseButtonPressed = false;
 513     d->rightMouseButtonPressed = false;
 514
 515     if (d->proxyModel)
 516     {
 517         QObject::disconnect(d->proxyModel,
 518                             SIGNAL(layoutChanged()),
 519                             this, SLOT(slotLayoutChanged()));
 520
 521         QObject::disconnect(d->proxyModel,
 522                             SIGNAL(dataChanged(QModelIndex,QModelIndex)),
 523                             this, SLOT(slotLayoutChanged()));
 524
 525         QObject::disconnect(d->proxyModel,
 526                             SIGNAL(rowsRemoved(QModelIndex,int,int)),
 527                             this, SLOT(rowsRemoved(QModelIndex,int,int)));
 528     }
 529
 530     QListView::setModel(model);
 531
 532     d->proxyModel = dynamic_cast<KCategorizedSortFilterProxyModel*>(model);
 533
 534     if (d->proxyModel)
 535     {
 536         d->modelSortRole = d->proxyModel->sortRole();
 537         d->modelSortColumn = d->proxyModel->sortColumn();
 538         d->modelSortOrder = d->proxyModel->sortOrder();
 539         d->modelLastRowCount = d->proxyModel->rowCount();
 540         d->modelCategorized = d->proxyModel->isCategorizedModel();
 541
 542         QObject::connect(d->proxyModel,
 543                          SIGNAL(layoutChanged()),
 544                          this, SLOT(slotLayoutChanged()));
 545
 546         QObject::connect(d->proxyModel,
 547                          SIGNAL(dataChanged(QModelIndex,QModelIndex)),
 548                          this, SLOT(slotLayoutChanged()));
 549
 550         QObject::connect(d->proxyModel,
 551                          SIGNAL(rowsRemoved(QModelIndex,int,int)),
 552                          this, SLOT(rowsRemoved(QModelIndex,int,int)));
 553
 554         if (d->proxyModel->rowCount())
 555         {
 556             d->layoutChanged(true);
 557         }
 558     }
 559     else
 560     {
 561         d->modelCategorized = false;
 562     }
 563 }
 564
 565 QRect KCategorizedView::visualRect(const QModelIndex &index) const
 566 {
 567     if ((viewMode() != KCategorizedView::IconMode) || !d->proxyModel ||
 568         !d->categoryDrawer || !d->proxyModel->isCategorizedModel())
 569     {
 570         return QListView::visualRect(index);
 571     }
 572
 573     if (!qobject_cast<const QSortFilterProxyModel*>(index.model()))
 574     {
 575         return d->visualRect(d->proxyModel->mapFromSource(index));
 576     }
 577
 578     return d->visualRect(index);
 579 }
 580
 581 KCategoryDrawer *KCategorizedView::categoryDrawer() const
 582 {
 583     return d->categoryDrawer;
 584 }
 585
 586 void KCategorizedView::setCategoryDrawer(KCategoryDrawer *categoryDrawer)
 587 {
 588     d->lastSelection = QItemSelection();
 589     d->forcedSelectionPosition = 0;
 590     d->elementsInfo.clear();
 591     d->elementsPosition.clear();
 592     d->categoriesIndexes.clear();
 593     d->categoriesPosition.clear();
 594     d->categories.clear();
 595     d->intersectedIndexes.clear();
 596     d->modelIndexList.clear();
 597     d->hovered = QModelIndex();
 598     d->mouseButtonPressed = false;
 599     d->rightMouseButtonPressed = false;
 600
 601     if (!categoryDrawer && d->proxyModel)
 602     {
 603         QObject::disconnect(d->proxyModel,
 604                             SIGNAL(layoutChanged()),
 605                             this, SLOT(slotLayoutChanged()));
 606
 607         QObject::disconnect(d->proxyModel,
 608                             SIGNAL(dataChanged(QModelIndex,QModelIndex)),
 609                             this, SLOT(slotLayoutChanged()));
 610
 611         QObject::disconnect(d->proxyModel,
 612                             SIGNAL(rowsRemoved(QModelIndex,int,int)),
 613                             this, SLOT(rowsRemoved(QModelIndex,int,int)));
 614     }
 615     else if (categoryDrawer && d->proxyModel)
 616     {
 617         QObject::connect(d->proxyModel,
 618                          SIGNAL(layoutChanged()),
 619                          this, SLOT(slotLayoutChanged()));
 620
 621         QObject::connect(d->proxyModel,
 622                          SIGNAL(dataChanged(QModelIndex,QModelIndex)),
 623                          this, SLOT(slotLayoutChanged()));
 624
 625         QObject::connect(d->proxyModel,
 626                          SIGNAL(rowsRemoved(QModelIndex,int,int)),
 627                          this, SLOT(rowsRemoved(QModelIndex,int,int)));
 628     }
 629
 630     d->categoryDrawer = categoryDrawer;
 631
 632     if (categoryDrawer)
 633     {
 634         if (d->proxyModel)
 635         {
 636             if (d->proxyModel->rowCount())
 637             {
 638                 d->layoutChanged(true);
 639             }
 640         }
 641     }
 642     else
 643     {
 644         updateGeometries();
 645     }
 646 }
 647
 648 QModelIndex KCategorizedView::indexAt(const QPoint &point) const
 649 {
 650     if ((viewMode() != KCategorizedView::IconMode) || !d->proxyModel ||
 651         !d->categoryDrawer || !d->proxyModel->isCategorizedModel())
 652     {
 653         return QListView::indexAt(point);
 654     }
 655
 656     QModelIndex index;
 657
 658     QModelIndexList item = d->intersectionSet(QRect(point, point));
 659
 660     if (item.count() == 1)
 661     {
 662         index = item[0];
 663     }
 664
 665     d->hovered = index;
 666
 667     return index;
 668 }
 669
 670 void KCategorizedView::reset()
 671 {
 672     QListView::reset();
 673
 674     d->lastSelection = QItemSelection();
 675     d->forcedSelectionPosition = 0;
 676     d->elementsInfo.clear();
 677     d->elementsPosition.clear();
 678     d->categoriesIndexes.clear();
 679     d->categoriesPosition.clear();
 680     d->categories.clear();
 681     d->intersectedIndexes.clear();
 682     d->modelIndexList.clear();
 683     d->hovered = QModelIndex();
 684     d->biggestItemSize = QSize(0, 0);
 685     d->mouseButtonPressed = false;
 686     d->rightMouseButtonPressed = false;
 687 }
 688
 689 void KCategorizedView::paintEvent(QPaintEvent *event)
 690 {
 691     if ((viewMode() != KCategorizedView::IconMode) || !d->proxyModel ||
 692         !d->categoryDrawer || !d->proxyModel->isCategorizedModel())
 693     {
 694         QListView::paintEvent(event);
 695         return;
 696     }
 697
 698     QStyleOptionViewItemV3 option = viewOptions();
 699     option.widget = this;
 700     if (wordWrap())
 701     {
 702         option.features |= QStyleOptionViewItemV2::WrapText;
 703     }
 704
 705     QPainter painter(viewport());
 706     QRect area = event->rect();
 707     const bool focus = (hasFocus() || viewport()->hasFocus()) &&
 708                         currentIndex().isValid();
 709     const QStyle::State state = option.state;
 710     const bool enabled = (state & QStyle::State_Enabled) != 0;
 711
 712     painter.save();
 713
 714     QModelIndexList dirtyIndexes = d->intersectionSet(area);
 715     foreach (const QModelIndex &index, dirtyIndexes)
 716     {
 717         option.state = state;
 718         option.rect = visualRect(index);
 719
 720         if (selectionModel() && selectionModel()->isSelected(index))
 721         {
 722             option.state |= QStyle::State_Selected;
 723         }
 724
 725         if (enabled)
 726         {
 727             QPalette::ColorGroup cg;
 728             if ((d->proxyModel->flags(index) & Qt::ItemIsEnabled) == 0)
 729             {
 730                 option.state &= ~QStyle::State_Enabled;
 731                 cg = QPalette::Disabled;
 732             }
 733             else
 734             {
 735                 cg = QPalette::Normal;
 736             }
 737             option.palette.setCurrentColorGroup(cg);
 738         }
 739
 740         if (focus && currentIndex() == index)
 741         {
 742             option.state |= QStyle::State_HasFocus;
 743             if (this->state() == EditingState)
 744                 option.state |= QStyle::State_Editing;
 745         }
 746
 747         if ((index == d->hovered) && !d->mouseButtonPressed && (this->state() != QAbstractItemView::DraggingState))
 748             option.state |= QStyle::State_MouseOver;
 749         else
 750             option.state &= ~QStyle::State_MouseOver;
 751
 752         itemDelegate(index)->paint(&painter, option, index);
 753     }
 754
 755     // Redraw categories
 756     QStyleOptionViewItem otherOption;
 757     bool intersectedInThePast = false;
 758     foreach (const QString &category, d->categories)
 759     {
 760         otherOption = option;
 761         otherOption.rect = d->categoryVisualRect(category);
 762         otherOption.state &= ~QStyle::State_MouseOver;
 763
 764         if (otherOption.rect.intersects(area))
 765         {
 766             intersectedInThePast = true;
 767
 768             QModelIndex indexToDraw = d->proxyModel->index(d->categoriesIndexes[category][0].row(), d->proxyModel->sortColumn());
 769
 770             d->drawNewCategory(indexToDraw,
 771                                d->proxyModel->sortRole(), otherOption, &painter);
 772         }
 773         else if (intersectedInThePast)
 774         {
 775             break; // the visible area has been finished, we don't need to keep asking, the rest won't intersect
 776                    // this is doable because we know that categories are correctly ordered on the list
 777         }
 778     }
 779
 780     if (d->mouseButtonPressed && !d->isDragging)
 781     {
 782         QPoint start, end, initialPressPosition;
 783
 784         initialPressPosition = d->initialPressPosition;
 785
 786         initialPressPosition.setY(initialPressPosition.y() - verticalOffset());
 787         initialPressPosition.setX(initialPressPosition.x() - horizontalOffset());
 788
 789         if (d->initialPressPosition.x() > d->mousePosition.x() ||
 790             d->initialPressPosition.y() > d->mousePosition.y())
 791         {
 792             start = d->mousePosition;
 793             end = initialPressPosition;
 794         }
 795         else
 796         {
 797             start = initialPressPosition;
 798             end = d->mousePosition;
 799         }
 800
 801         QStyleOptionRubberBand yetAnotherOption;
 802         yetAnotherOption.initFrom(this);
 803         yetAnotherOption.shape = QRubberBand::Rectangle;
 804         yetAnotherOption.opaque = false;
 805         yetAnotherOption.rect = QRect(start, end).intersected(viewport()->rect().adjusted(-16, -16, 16, 16));
 806         painter.save();
 807         style()->drawControl(QStyle::CE_RubberBand, &yetAnotherOption, &painter);
 808         painter.restore();
 809     }
 810
 811     if (d->isDragging && !d->dragLeftViewport)
 812     {
 813         painter.setOpacity(0.5);
 814         d->drawDraggedItems(&painter);
 815     }
 816
 817     painter.restore();
 818 }
 819
 820 void KCategorizedView::resizeEvent(QResizeEvent *event)
 821 {
 822     QListView::resizeEvent(event);
 823
 824     // Clear the items positions cache
 825     d->elementsPosition.clear();
 826     d->categoriesPosition.clear();
 827     d->forcedSelectionPosition = 0;
 828
 829     if ((viewMode() != KCategorizedView::IconMode) || !d->proxyModel ||
 830         !d->categoryDrawer || !d->proxyModel->isCategorizedModel())
 831     {
 832         return;
 833     }
 834
 835     d->updateScrollbars();
 836 }
 837
 838 void KCategorizedView::setSelection(const QRect &rect,
 839                                     QItemSelectionModel::SelectionFlags flags)
 840 {
 841     if ((viewMode() != KCategorizedView::IconMode) || !d->proxyModel ||
 842         !d->categoryDrawer || !d->proxyModel->isCategorizedModel())
 843     {
 844         QListView::setSelection(rect, flags);
 845         return;
 846     }
 847
 848     if (!flags)
 849         return;
 850
 851     if (flags & QItemSelectionModel::Clear)
 852     {
 853         selectionModel()->clear();
 854         d->lastSelection.clear();
 855     }
 856
 857     QModelIndexList dirtyIndexes = d->intersectionSet(rect);
 858
 859     // no items affected, just leave
 860     if (!dirtyIndexes.count())
 861     {
 862         selectionModel()->select(d->lastSelection, QItemSelectionModel::SelectCurrent);
 863
 864         return;
 865     }
 866
 867     QModelIndex topLeft;
 868     QModelIndex bottomRight;
 869
 870     if (d->mouseButtonPressed || d->rightMouseButtonPressed) // selection with click + drag
 871     {
 872         QItemSelection selection;
 873
 874         QModelIndex prev = dirtyIndexes[0];
 875         QModelIndex first = prev;
 876         foreach (const QModelIndex &index, dirtyIndexes)
 877         {
 878             // we have a different interval. non-contiguous items
 879             if ((index.row() - prev.row()) > 1) {
 880                 selection << QItemSelectionRange(first, prev);
 881
 882                 first = index;
 883             }
 884
 885             prev = index;
 886         }
 887
 888         selection << QItemSelectionRange(first, prev);
 889
 890         if (flags & QItemSelectionModel::Current)
 891         {
 892             if (rect.topLeft() == rect.bottomRight())
 893             {
 894                 selectionModel()->setCurrentIndex(indexAt(rect.topLeft()), QItemSelectionModel::NoUpdate);
 895             }
 896
 897             selection.merge(d->lastSelection, flags);
 898         }
 899         else
 900         {
 901             selection.merge(selectionModel()->selection(), flags);
 902
 903             selectionModel()->select(selection, QItemSelectionModel::SelectCurrent);
 904
 905             return;
 906         }
 907
 908         selectionModel()->select(selection, flags);
 909     }
 910     else // selection with click + keyboard keys
 911     {
 912         QModelIndex topLeftIndex = indexAt(QPoint(rect.topLeft().x(),
 913                                                   rect.topLeft().y()));
 914         QModelIndex bottomRightIndex = indexAt(QPoint(rect.bottomRight().x(),
 915                                                       rect.bottomRight().y()));
 916
 917         // keyboard selection comes "upside down". Let's normalize it
 918         if (topLeftIndex.row() > bottomRightIndex.row())
 919         {
 920             QModelIndex auxIndex = topLeftIndex;
 921             topLeftIndex = bottomRightIndex;
 922             bottomRightIndex = auxIndex;
 923         }
 924
 925         int viewportWidth = viewport()->width() - spacing();
 926         int itemWidth;
 927
 928         if (gridSize().isEmpty())
 929         {
 930             itemWidth = d->biggestItemSize.width();
 931         }
 932         else
 933         {
 934             itemWidth = gridSize().width();
 935         }
 936
 937         int itemWidthPlusSeparation = spacing() + itemWidth;
 938         int elementsPerRow = viewportWidth / itemWidthPlusSeparation;
 939         if (!elementsPerRow)
 940             elementsPerRow++;
 941
 942         QModelIndexList theoricDirty(dirtyIndexes);
 943         dirtyIndexes.clear();
 944         int first = model()->rowCount();
 945         int last = 0;
 946
 947         foreach (const QModelIndex &index, theoricDirty)
 948         {
 949             if ((index.row() < first) &&
 950                 ((((topLeftIndex.row() / elementsPerRow) == (index.row() / elementsPerRow)) &&
 951                   ((topLeftIndex.row() % elementsPerRow) <= (index.row() % elementsPerRow))) ||
 952                  (topLeftIndex.row() / elementsPerRow) != (index.row() / elementsPerRow)))
 953             {
 954                 first = index.row();
 955                 topLeft = index;
 956             }
 957
 958             if ((index.row() > last) &&
 959                 ((((bottomRightIndex.row() / elementsPerRow) == (index.row() / elementsPerRow)) &&
 960                   ((bottomRightIndex.row() % elementsPerRow) >= (index.row() % elementsPerRow))) ||
 961                  (bottomRightIndex.row() / elementsPerRow) != (index.row() / elementsPerRow)))
 962             {
 963                 last = index.row();
 964                 bottomRight = index;
 965             }
 966         }
 967
 968         for (int i = first; i <= last; i++)
 969         {
 970             dirtyIndexes << model()->index(i, theoricDirty[0].column(), theoricDirty[0].parent());
 971         }
 972
 973         QItemSelection selection(topLeft, bottomRight);
 974
 975         selectionModel()->select(selection, flags);
 976     }
 977 }
 978
 979 void KCategorizedView::mouseMoveEvent(QMouseEvent *event)
 980 {
 981     QListView::mouseMoveEvent(event);
 982
 983     if ((viewMode() != KCategorizedView::IconMode) || !d->proxyModel ||
 984         !d->categoryDrawer || !d->proxyModel->isCategorizedModel())
 985     {
 986         return;
 987     }
 988
 989     const QString previousHoveredCategory = d->hoveredCategory;
 990
 991     d->mousePosition = event->pos();
 992     d->hoveredCategory = QString();
 993
 994     // Redraw categories
 995     foreach (const QString &category, d->categories)
 996     {
 997         if (d->categoryVisualRect(category).intersects(QRect(event->pos(), event->pos())))
 998         {
 999             d->hoveredCategory = category;
1000             viewport()->update(d->categoryVisualRect(category));
1001         }
1002         else if ((category == previousHoveredCategory) &&
1003                  (!d->categoryVisualRect(previousHoveredCategory).intersects(QRect(event->pos(), event->pos()))))
1004         {
1005             viewport()->update(d->categoryVisualRect(category));
1006         }
1007     }
1008
1009     QRect rect;
1010     if (d->mouseButtonPressed && !d->isDragging)
1011     {
1012         QPoint start, end, initialPressPosition;
1013
1014         initialPressPosition = d->initialPressPosition;
1015
1016         initialPressPosition.setY(initialPressPosition.y() - verticalOffset());
1017         initialPressPosition.setX(initialPressPosition.x() - horizontalOffset());
1018
1019         if (d->initialPressPosition.x() > d->mousePosition.x() ||
1020             d->initialPressPosition.y() > d->mousePosition.y())
1021         {
1022             start = d->mousePosition;
1023             end = initialPressPosition;
1024         }
1025         else
1026         {
1027             start = initialPressPosition;
1028             end = d->mousePosition;
1029         }
1030
1031         rect = QRect(start, end).intersected(viewport()->rect().adjusted(-16, -16, 16, 16));
1032     }
1033 }
1034
1035 void KCategorizedView::mousePressEvent(QMouseEvent *event)
1036 {
1037     d->dragLeftViewport = false;
1038
1039     if (event->button() == Qt::LeftButton)
1040     {
1041         d->mouseButtonPressed = true;
1042
1043         d->initialPressPosition = event->pos();
1044         d->initialPressPosition.setY(d->initialPressPosition.y() +
1045                                                               verticalOffset());
1046         d->initialPressPosition.setX(d->initialPressPosition.x() +
1047                                                             horizontalOffset());
1048     }
1049     else if (event->button() == Qt::RightButton)
1050     {
1051         d->rightMouseButtonPressed = true;
1052     }
1053
1054     QListView::mousePressEvent(event);
1055
1056     d->lastSelection = selectionModel()->selection();
1057
1058     viewport()->update(d->categoryVisualRect(d->hoveredCategory));
1059 }
1060
1061 void KCategorizedView::mouseReleaseEvent(QMouseEvent *event)
1062 {
1063     d->mouseButtonPressed = false;
1064     d->rightMouseButtonPressed = false;
1065
1066     QListView::mouseReleaseEvent(event);
1067
1068     if ((viewMode() != KCategorizedView::IconMode) || !d->proxyModel ||
1069         !d->categoryDrawer || !d->proxyModel->isCategorizedModel())
1070     {
1071         return;
1072     }
1073
1074     QPoint initialPressPosition = viewport()->mapFromGlobal(QCursor::pos());
1075     initialPressPosition.setY(initialPressPosition.y() + verticalOffset());
1076     initialPressPosition.setX(initialPressPosition.x() + horizontalOffset());
1077
1078     if (initialPressPosition == d->initialPressPosition)
1079     {
1080         foreach(const QString &category, d->categories)
1081         {
1082             if (d->categoryVisualRect(category).contains(event->pos()))
1083             {
1084                 QItemSelection selection;
1085                 QModelIndexList indexList = d->categoriesIndexes[category];
1086
1087                 foreach (const QModelIndex &index, indexList)
1088                 {
1089                     QModelIndex selectIndex = index.model()->index(index.row(), 0);
1090
1091                     selection << QItemSelectionRange(selectIndex);
1092                 }
1093
1094                 selectionModel()->select(selection, QItemSelectionModel::Select);
1095
1096                 break;
1097             }
1098         }
1099     }
1100
1101     if (d->hovered.isValid())
1102         viewport()->update(visualRect(d->hovered));
1103     else if (!d->hoveredCategory.isEmpty())
1104         viewport()->update(d->categoryVisualRect(d->hoveredCategory));
1105 }
1106
1107 void KCategorizedView::leaveEvent(QEvent *event)
1108 {
1109     d->hovered = QModelIndex();
1110     d->hoveredCategory = QString();
1111
1112     QListView::leaveEvent(event);
1113 }
1114
1115 void KCategorizedView::startDrag(Qt::DropActions supportedActions)
1116 {
1117     // FIXME: QAbstractItemView does far better here since it sets the
1118     //        pixmap of selected icons to the dragging cursor, but it sets a non
1119     //        ARGB window so it is no transparent. Use QAbstractItemView when
1120     //        this is fixed on Qt.
1121     // QAbstractItemView::startDrag(supportedActions);
1122 #if !defined(DOLPHIN_DRAGANDDROP)
1123     QListView::startDrag(supportedActions);
1124 #endif
1125
1126     d->isDragging = false;
1127     d->mouseButtonPressed = false;
1128     d->rightMouseButtonPressed = false;
1129
1130     viewport()->update(d->lastDraggedItemsRect);
1131 }
1132
1133 void KCategorizedView::dragMoveEvent(QDragMoveEvent *event)
1134 {
1135     d->mousePosition = event->pos();
1136
1137     if (d->mouseButtonPressed)
1138     {
1139         d->isDragging = true;
1140     }
1141     else
1142     {
1143         d->isDragging = false;
1144     }
1145
1146     d->dragLeftViewport = false;
1147
1148     if ((viewMode() != KCategorizedView::IconMode) || !d->proxyModel ||
1149         !d->categoryDrawer || !d->proxyModel->isCategorizedModel())
1150     {
1151 #if defined(DOLPHIN_DRAGANDDROP)
1152         QAbstractItemView::dragMoveEvent(event);
1153 #else
1154         QListView::dragMoveEvent(event);
1155 #endif
1156         return;
1157     }
1158
1159     d->drawDraggedItems();
1160 }
1161
1162 void KCategorizedView::dragLeaveEvent(QDragLeaveEvent *event)
1163 {
1164     d->dragLeftViewport = true;
1165
1166 #if defined(DOLPHIN_DRAGANDDROP)
1167     QAbstractItemView::dragLeaveEvent(event);
1168 #else
1169     QListView::dragLeaveEvent(event);
1170 #endif
1171 }
1172
1173 void KCategorizedView::dropEvent(QDropEvent *event)
1174 {
1175 #if defined(DOLPHIN_DRAGANDDROP)
1176     QAbstractItemView::dropEvent(event);
1177 #else
1178     QListView::dropEvent(event);
1179 #endif
1180 }
1181
1182 QModelIndex KCategorizedView::moveCursor(CursorAction cursorAction,
1183                                          Qt::KeyboardModifiers modifiers)
1184 {
1185     if ((viewMode() != KCategorizedView::IconMode) ||
1186          !d->proxyModel ||
1187          !d->categoryDrawer ||
1188           d->categories.isEmpty() ||
1189          !d->proxyModel->isCategorizedModel()
1190        )
1191     {
1192         return QListView::moveCursor(cursorAction, modifiers);
1193     }
1194
1195     QModelIndex current = selectionModel()->currentIndex();
1196
1197     if (!current.isValid())
1198     {
1199         current = model()->index(0, 0, QModelIndex());
1200         selectionModel()->select(current, QItemSelectionModel::NoUpdate);
1201         d->forcedSelectionPosition = 0;
1202
1203         return current;
1204     }
1205
1206     int viewportWidth = viewport()->width() - spacing();
1207     int itemWidth;
1208
1209     if (gridSize().isEmpty())
1210     {
1211         itemWidth = d->biggestItemSize.width();
1212     }
1213     else
1214     {
1215         itemWidth = gridSize().width();
1216     }
1217
1218     int itemWidthPlusSeparation = spacing() + itemWidth;
1219     int elementsPerRow = viewportWidth / itemWidthPlusSeparation;
1220     if (!elementsPerRow)
1221         elementsPerRow++;
1222
1223     QString lastCategory = d->categories.first();
1224     QString theCategory = d->categories.first();
1225     QString afterCategory = d->categories.first();
1226
1227     bool hasToBreak = false;
1228     foreach (const QString &category, d->categories)
1229     {
1230         if (hasToBreak)
1231         {
1232             afterCategory = category;
1233
1234             break;
1235         }
1236
1237         if (category == d->elementsInfo[current.row()].category)
1238         {
1239             theCategory = category;
1240
1241             hasToBreak = true;
1242         }
1243
1244         if (!hasToBreak)
1245         {
1246             lastCategory = category;
1247         }
1248     }
1249
1250     switch (cursorAction)
1251     {
1252         case QAbstractItemView::MoveUp: {
1253             if (d->elementsInfo[current.row()].relativeOffsetToCategory >= elementsPerRow)
1254             {
1255                 int indexToMove = current.row();
1256                 indexToMove -= qMin(((d->elementsInfo[current.row()].relativeOffsetToCategory) + d->forcedSelectionPosition), elementsPerRow - d->forcedSelectionPosition + (d->elementsInfo[current.row()].relativeOffsetToCategory % elementsPerRow));
1257
1258                 return d->proxyModel->index(indexToMove, 0);
1259             }
1260             else
1261             {
1262                 int lastCategoryLastRow = (d->categoriesIndexes[lastCategory].count() - 1) % elementsPerRow;
1263                 int indexToMove = current.row() - d->elementsInfo[current.row()].relativeOffsetToCategory;
1264
1265                 if (d->forcedSelectionPosition >= lastCategoryLastRow)
1266                 {
1267                     indexToMove -= 1;
1268                 }
1269                 else
1270                 {
1271                     indexToMove -= qMin((lastCategoryLastRow - d->forcedSelectionPosition + 1), d->forcedSelectionPosition + elementsPerRow + 1);
1272                 }
1273
1274                 return d->proxyModel->index(indexToMove, 0);
1275             }
1276         }
1277
1278         case QAbstractItemView::MoveDown: {
1279             if (d->elementsInfo[current.row()].relativeOffsetToCategory < (d->categoriesIndexes[theCategory].count() - 1 - ((d->categoriesIndexes[theCategory].count() - 1) % elementsPerRow)))
1280             {
1281                 int indexToMove = current.row();
1282                 indexToMove += qMin(elementsPerRow, d->categoriesIndexes[theCategory].count() - 1 - d->elementsInfo[current.row()].relativeOffsetToCategory);
1283
1284                 return d->proxyModel->index(indexToMove, 0);
1285             }
1286             else
1287             {
1288                 int afterCategoryLastRow = qMin(elementsPerRow, d->categoriesIndexes[afterCategory].count());
1289                 int indexToMove = current.row() + (d->categoriesIndexes[theCategory].count() - d->elementsInfo[current.row()].relativeOffsetToCategory);
1290
1291                 if (d->forcedSelectionPosition >= afterCategoryLastRow)
1292                 {
1293                     indexToMove += afterCategoryLastRow - 1;
1294                 }
1295                 else
1296                 {
1297                     indexToMove += qMin(d->forcedSelectionPosition, elementsPerRow);
1298                 }
1299
1300                 return d->proxyModel->index(indexToMove, 0);
1301             }
1302         }
1303
1304         case QAbstractItemView::MoveLeft:
1305             if (layoutDirection() == Qt::RightToLeft)
1306             {
1307                 if (!(d->elementsInfo[current.row() + 1].relativeOffsetToCategory % elementsPerRow))
1308                     return current;
1309
1310                 d->forcedSelectionPosition = d->elementsInfo[current.row() + 1].relativeOffsetToCategory % elementsPerRow;
1311
1312 #if 0 //follow qt view behavior. lateral movements won't change visual row
1313                 if (d->forcedSelectionPosition < 0)
1314                     d->forcedSelectionPosition = (d->categoriesIndexes[theCategory].count() - 1) % elementsPerRow;
1315 #endif
1316
1317                 return d->proxyModel->index(current.row() + 1, 0);
1318             }
1319
1320             if (!(d->elementsInfo[current.row()].relativeOffsetToCategory % elementsPerRow))
1321                 return current;
1322
1323             d->forcedSelectionPosition = d->elementsInfo[current.row() - 1].relativeOffsetToCategory % elementsPerRow;
1324
1325 #if 0 //follow qt view behavior. lateral movements won't change visual row
1326             if (d->forcedSelectionPosition < 0)
1327                 d->forcedSelectionPosition = (d->categoriesIndexes[theCategory].count() - 1) % elementsPerRow;
1328 #endif
1329
1330             return d->proxyModel->index(current.row() - 1, 0);
1331
1332         case QAbstractItemView::MoveRight:
1333             if (layoutDirection() == Qt::RightToLeft)
1334             {
1335                 if (!(d->elementsInfo[current.row()].relativeOffsetToCategory % elementsPerRow))
1336                     return current;
1337
1338                 d->forcedSelectionPosition = d->elementsInfo[current.row() - 1].relativeOffsetToCategory % elementsPerRow;
1339
1340 #if 0 //follow qt view behavior. lateral movements won't change visual row
1341                 if (d->forcedSelectionPosition < 0)
1342                     d->forcedSelectionPosition = (d->categoriesIndexes[theCategory].count() - 1) % elementsPerRow;
1343 #endif
1344
1345                 return d->proxyModel->index(current.row() - 1, 0);
1346             }
1347
1348             if (!(d->elementsInfo[current.row() + 1].relativeOffsetToCategory % elementsPerRow))
1349                 return current;
1350
1351             d->forcedSelectionPosition = d->elementsInfo[current.row() + 1].relativeOffsetToCategory % elementsPerRow;
1352
1353 #if 0 //follow qt view behavior. lateral movements won't change visual row
1354             if (d->forcedSelectionPosition < 0)
1355                 d->forcedSelectionPosition = (d->categoriesIndexes[theCategory].count() - 1) % elementsPerRow;
1356 #endif
1357
1358             return d->proxyModel->index(current.row() + 1, 0);
1359
1360         default:
1361             break;
1362     }
1363
1364     return QListView::moveCursor(cursorAction, modifiers);
1365 }
1366
1367 void KCategorizedView::rowsInserted(const QModelIndex &parent,
1368                                     int start,
1369                                     int end)
1370 {
1371     QListView::rowsInserted(parent, start, end);
1372
1373     if ((viewMode() != KCategorizedView::IconMode) || !d->proxyModel ||
1374         !d->categoryDrawer || !d->proxyModel->isCategorizedModel())
1375     {
1376         d->forcedSelectionPosition = 0;
1377         d->elementsInfo.clear();
1378         d->elementsPosition.clear();
1379         d->categoriesIndexes.clear();
1380         d->categoriesPosition.clear();
1381         d->categories.clear();
1382         d->intersectedIndexes.clear();
1383         d->modelIndexList.clear();
1384         d->hovered = QModelIndex();
1385         d->biggestItemSize = QSize(0, 0);
1386         d->mouseButtonPressed = false;
1387         d->rightMouseButtonPressed = false;
1388
1389         return;
1390     }
1391
1392     rowsInsertedArtifficial(parent, start, end);
1393 }
1394
1395 void KCategorizedView::rowsInsertedArtifficial(const QModelIndex &parent,
1396                                                int start,
1397                                                int end)
1398 {
1399     Q_UNUSED(parent);
1400
1401     d->forcedSelectionPosition = 0;
1402     d->elementsInfo.clear();
1403     d->elementsPosition.clear();
1404     d->categoriesIndexes.clear();
1405     d->categoriesPosition.clear();
1406     d->categories.clear();
1407     d->intersectedIndexes.clear();
1408     d->modelIndexList.clear();
1409     d->hovered = QModelIndex();
1410     d->biggestItemSize = QSize(0, 0);
1411     d->mouseButtonPressed = false;
1412     d->rightMouseButtonPressed = false;
1413
1414     if (start > end || end < 0 || start < 0 || !d->proxyModel->rowCount())
1415     {
1416         return;
1417     }
1418
1419     // Add all elements mapped to the source model and explore categories
1420     QString prevCategory = d->proxyModel->data(d->proxyModel->index(0, d->proxyModel->sortColumn()), KCategorizedSortFilterProxyModel::CategoryDisplayRole).toString();
1421     QString lastCategory = prevCategory;
1422     QModelIndexList modelIndexList;
1423     struct Private::ElementInfo elementInfo;
1424     int offset = -1;
1425     for (int k = 0; k < d->proxyModel->rowCount(); ++k)
1426     {
1427         QModelIndex index = d->proxyModel->index(k, d->proxyModel->sortColumn());
1428         QModelIndex indexSize = d->proxyModel->index(k, 0);
1429
1430         d->biggestItemSize = QSize(qMax(sizeHintForIndex(indexSize).width(),
1431                                         d->biggestItemSize.width()),
1432                                    qMax(sizeHintForIndex(indexSize).height(),
1433                                         d->biggestItemSize.height()));
1434
1435         d->modelIndexList << index;
1436
1437         lastCategory = d->proxyModel->data(index, KCategorizedSortFilterProxyModel::CategoryDisplayRole).toString();
1438
1439         elementInfo.category = lastCategory;
1440
1441         if (prevCategory != lastCategory)
1442         {
1443             offset = 0;
1444             d->categoriesIndexes.insert(prevCategory, modelIndexList);
1445             d->categories << prevCategory;
1446             modelIndexList.clear();
1447         }
1448         else
1449         {
1450             offset++;
1451         }
1452
1453         elementInfo.relativeOffsetToCategory = offset;
1454
1455         modelIndexList << index;
1456         prevCategory = lastCategory;
1457
1458         d->elementsInfo.insert(index.row(), elementInfo);
1459     }
1460
1461     d->categoriesIndexes.insert(prevCategory, modelIndexList);
1462     d->categories << prevCategory;
1463
1464     d->updateScrollbars();
1465
1466     // FIXME: We need to safely save the last selection. This is on my TODO
1467     // list (ereslibre).
1468     selectionModel()->clear();
1469 }
1470
1471 void KCategorizedView::rowsRemoved(const QModelIndex &parent,
1472                                    int start,
1473                                    int end)
1474 {
1475     if ((viewMode() == KCategorizedView::IconMode) && d->proxyModel &&
1476         d->categoryDrawer && d->proxyModel->isCategorizedModel())
1477     {
1478         // Force the view to update all elements
1479         rowsInsertedArtifficial(QModelIndex(), 0, d->proxyModel->rowCount() - 1);
1480     }
1481 }
1482
1483 void KCategorizedView::updateGeometries()
1484 {
1485     if ((viewMode() != KCategorizedView::IconMode) || !d->proxyModel ||
1486         !d->categoryDrawer || !d->proxyModel->isCategorizedModel())
1487     {
1488         QListView::updateGeometries();
1489         return;
1490     }
1491
1492     // Avoid QListView::updateGeometries(), since it will try to set another
1493     // range to our scroll bars, what we don't want (ereslibre)
1494     QAbstractItemView::updateGeometries();
1495 }
1496
1497 void KCategorizedView::slotLayoutChanged()
1498 {
1499     d->layoutChanged();
1500 }
1501
1502 void KCategorizedView::currentChanged(const QModelIndex &current,
1503                                       const QModelIndex &previous)
1504 {
1505     // We need to update the forcedSelectionPosition property in order to correctly
1506     // navigate after with keyboard using up & down keys
1507
1508     int viewportWidth = viewport()->width() - spacing();
1509
1510     int itemHeight;
1511     int itemWidth;
1512
1513     if (gridSize().isEmpty())
1514     {
1515         itemHeight = d->biggestItemSize.height();
1516         itemWidth = d->biggestItemSize.width();
1517     }
1518     else
1519     {
1520         itemHeight = gridSize().height();
1521         itemWidth = gridSize().width();
1522     }
1523
1524     int itemWidthPlusSeparation = spacing() + itemWidth;
1525     int elementsPerRow = viewportWidth / itemWidthPlusSeparation;
1526     if (!elementsPerRow)
1527         elementsPerRow++;
1528
1529     d->forcedSelectionPosition = d->elementsInfo[current.row()].relativeOffsetToCategory % elementsPerRow;
1530
1531     QListView::currentChanged(current, previous);
1532 }
1533
1534 #include "kcategorizedview.moc"