Re-added save and restore buttons

cockatrice/src/setsmodel.cpp

@@ -97,32 +97,51 @@ void SetsModel::swapRows(int oldRow, int newRow)
     sets.insert(newRow, temp);
     endInsertRows();
 
-    for (int i = 0; i < sets.size(); i++)
-        sets[i]->setSortKey(i);
-
-    sets.sortByKey();
-
     emit dataChanged(index(0, 0), index(rowCount() - 1, columnCount() - 1));
 }
 
 void SetsModel::sort(int column, Qt::SortOrder order)
 {
-    QMap<QString, CardSet *> setMap;
+    QMap<QVariant, CardSet *> setMap;
     int numRows = rowCount();
     int row;
 
     for(row = 0; row < numRows; ++row)
-        setMap.insertMulti(index(row, column).data().toString(), sets.at(row));
+        setMap.insertMulti(index(row, column).data(), sets.at(row));
 
-    row = (order == Qt::AscendingOrder) ? 0 : numRows - 1;
     
-    foreach(CardSet * set, setMap)
-        set->setSortKey((order == Qt::AscendingOrder) ? row++ : row--);
+    QList<CardSet *> tmp = setMap.values();
+    sets.clear();
+    if(order == Qt::AscendingOrder)
+    {
+        for(row = 0; row < tmp.size(); row++) {
+            sets.append(tmp.at(row));
+        }
+    } else {
+        for(row = tmp.size() - 1; row >= 0; row--) {
+            sets.append(tmp.at(row));
+        }
+    }
 
-    sets.sortByKey();
     emit dataChanged(index(0, 0), index(numRows - 1, columnCount() - 1));
 }
 
+void SetsModel::save()
+{
+    for (int i = 0; i < sets.size(); i++)
+        sets[i]->setSortKey(i);
+
+    sets.sortByKey();
+}
+
+void SetsModel::restore(CardDatabase *db)
+{
+    sets = db->getSetList();
+    sets.sortByKey();
+
+    emit dataChanged(index(0, 0), index(rowCount() - 1, columnCount() - 1));
+}
+
 QStringList SetsModel::mimeTypes() const
 {
     return QStringList() << "application/x-cockatricecardset";

cockatrice/src/setsmodel.h

@@ -40,6 +40,8 @@ public:
     QStringList mimeTypes() const;
     void swapRows(int oldRow, int newRow);
     void sort(int column, Qt::SortOrder order = Qt::AscendingOrder);
+    void save();
+    void restore(CardDatabase *db);
 };
 
 #endif

cockatrice/src/window_sets.cpp

@@ -13,8 +13,6 @@ WndSets::WndSets(QWidget *parent)
     model = new SetsModel(db, this);
     view = new QTreeView;
     view->setModel(model);
-    view->sortByColumn(SetsModel::SortKeyCol, Qt::AscendingOrder);
-    view->setColumnHidden(SetsModel::SortKeyCol, true);
 
     view->setAlternatingRowColors(true);
     view->setUniformRowHeights(true);
@@ -36,6 +34,13 @@ WndSets::WndSets(QWidget *parent)
     view->header()->setSectionResizeMode(SetsModel::LongNameCol, QHeaderView::ResizeToContents);
 #endif
 
+    view->sortByColumn(SetsModel::SortKeyCol, Qt::AscendingOrder);
+    view->setColumnHidden(SetsModel::SortKeyCol, true);
+
+    saveButton = new QPushButton(tr("Save set ordering"));
+    connect(saveButton, SIGNAL(clicked()), this, SLOT(actSave()));
+    restoreButton = new QPushButton(tr("Restore saved set ordering"));
+    connect(restoreButton, SIGNAL(clicked()), this, SLOT(actRestore()));
     upButton = new QPushButton(tr("Move selected set up"));
     connect(upButton, SIGNAL(clicked()), this, SLOT(actUp()));
     downButton = new QPushButton(tr("Move selected set down"));
@@ -62,6 +67,9 @@ WndSets::WndSets(QWidget *parent)
     mainLayout->addWidget(topButton, 1, 1, 1, 1);
     mainLayout->addWidget(bottomButton, 2, 1, 1, 1);
 
+    mainLayout->addWidget(saveButton, 3, 0, 1, 1);
+    mainLayout->addWidget(restoreButton, 3, 1, 1, 1);
+
     QWidget *centralWidget = new QWidget;
     centralWidget->setLayout(mainLayout);
     setCentralWidget(centralWidget);
@@ -74,6 +82,16 @@ WndSets::~WndSets()
 {
 }
 
+void WndSets::actSave()
+{
+    model->save();
+}
+
+void WndSets::actRestore()
+{
+    model->restore(db);
+}
+
 void WndSets::actToggleButtons(const QItemSelection & selected, const QItemSelection &)
 {
     bool disabled = selected.empty();

cockatrice/src/window_sets.h

@@ -15,13 +15,15 @@ class WndSets : public QMainWindow {
 private:
     SetsModel *model;
     QTreeView *view;
-    QPushButton *upButton, *downButton, *bottomButton, *topButton;
+    QPushButton *saveButton, *restoreButton, *upButton, *downButton, *bottomButton, *topButton;
 public:
     WndSets(QWidget *parent = 0);
     ~WndSets();
 protected:
     void selectRow(int row);
 private slots:
+    void actSave();
+    void actRestore();
     void actUp();
     void actDown();
     void actTop();