Remove colors from scheme list on delete/backspace press

nyalldawson · nyalldawson · commit 05e16291eaa3 · 2014-09-15T21:40:50.000+10:00
diff --git a/python/gui/qgscolorschemelist.sip b/python/gui/qgscolorschemelist.sip
@@ -170,5 +170,9 @@ class QgsColorSchemeList: QTreeView
      */
     void copyColors();
 
+  protected:
+
+    void keyPressEvent( QKeyEvent* event );
+
 };
 
diff --git a/src/gui/qgscolorschemelist.cpp b/src/gui/qgscolorschemelist.cpp
@@ -21,6 +21,7 @@
 #include <QColorDialog>
 #include <QMimeData>
 #include <QClipboard>
+#include <QKeyEvent>
 
 //For model testing
 //#include "modeltest.h"
@@ -131,6 +132,31 @@ void QgsColorSchemeList::copyColors()
   QApplication::clipboard()->setMimeData( mimeData );
 }
 
+void QgsColorSchemeList::keyPressEvent( QKeyEvent *event )
+{
+  //listen out for delete/backspace presses and remove selected colors
+  if (( event->key() == Qt::Key_Backspace || event->key() == Qt::Key_Delete ) )
+  {
+    QList<int> rows;
+    foreach ( const QModelIndex &index, selectedIndexes() )
+    {
+      rows << index.row();
+    }
+    //remove duplicates
+    QList<int> rowsToRemove =  QList<int>::fromSet( rows.toSet() );
+
+    //remove rows in descending order
+    qSort( rowsToRemove.begin(), rowsToRemove.end(), qGreater<int>() );
+    foreach ( const int row, rowsToRemove )
+    {
+      mModel->removeRow( row );
+    }
+    return;
+  }
+
+  QTreeView::keyPressEvent( event );
+}
+
 bool QgsColorSchemeList::importColorsFromGpl( QFile &file )
 {
   QgsNamedColorList importedColors;
diff --git a/src/gui/qgscolorschemelist.h b/src/gui/qgscolorschemelist.h
@@ -211,6 +211,10 @@ class GUI_EXPORT QgsColorSchemeList: public QTreeView
      */
     void copyColors();
 
+  protected:
+
+    void keyPressEvent( QKeyEvent* event );
+
   private:
     QgsColorScheme* mScheme;
     QgsColorSchemeModel* mModel;
