Changeset 15 in tspsg-svn for trunk/src/tspmodel.cpp

Timestamp:

Nov 5, 2007, 1:32:40 AM (17 years ago)

Author:

laleppa

Message:

Finished converting to Qt's model/view architecture

File:

• trunk/src/tspmodel.cpp (modified) (7 diffs)

trunk/src/tspmodel.cpp

@@ -1 +1 @@
 /*
  *  TSPSG - TSP Solver and Generator
- *  Copyright (C) 2007 L¸ppa <lacontacts[at]gmail[dot]com>
+ *  Copyright (C) 2007 Lёppa <lacontacts[at]gmail[dot]com>
  *
  *  $Id$
@@ -26 +26 @@
 
 CTSPModel::CTSPModel(QObject *parent)
-        : QAbstractTableModel(parent), randMin(1), randMax(10)
+        : QAbstractTableModel(parent), randMin(1), randMax(10), nCities(0)
 {
 }
@@ -48 +48 @@
 {
         if (role == Qt::DisplayRole)
-                return trUtf8("Ãîðîä %1").arg(section + 1);
+                return trUtf8("Город %1").arg(section + 1);
         return QVariant();
 }
@@ -58 +58 @@
         if (role == Qt::TextAlignmentRole)
                 return int(Qt::AlignCenter);
-        else if (role == Qt::DisplayRole) {
-                if (index.row() == index.column())
-                        return trUtf8("---");
+        else if (role == Qt::FontRole) {
+QFont font;
+                font.setBold(true);
+                return font;
+        } else if (role == Qt::DisplayRole || role == Qt::EditRole) {
                 if (index.row() < nCities && index.column() < nCities)
-                        return table.at(index.row())->at(index.column());
+                        if (table[index.row()][index.column()] == INFINITY)
+                                return trUtf8(INFSTR);
+                        else
+                                // HACK: Converting to string to prevent spinbox in edit mode
+                                return QVariant(table[index.row()][index.column()]).toString();
                 else
                         return QVariant();
-        }
+        } else if (role == Qt::UserRole)
+                return table[index.row()][index.column()];
         return QVariant();
 }
@@ -71 +78 @@
 bool CTSPModel::setData(const QModelIndex &index, const QVariant &value, int role)
 {
-        return true;
+        if (!index.isValid())
+                return false;
+        if (role == Qt::EditRole && index.row() != index.column()) {
+                if (value.toString().compare(INFSTR) == 0)
+                        table[index.row()][index.column()] = INFINITY;
+                else {
+bool ok;
+double tmp = value.toDouble(&ok);
+                        if (!ok || tmp < 0)
+                                return false;
+                        else
+                                table[index.row()][index.column()] = tmp;
+                }
+                emit dataChanged(index,index);
+                return true;
+        }
+        return false;
 }
 
@@ -77 +100 @@
 {
 Qt::ItemFlags flags = QAbstractItemModel::flags(index);
-        if (index.row() == index.column())
+        if (index.row() != index.column())
                 flags |= Qt::ItemIsEditable;
         return flags;
@@ -89 +112 @@
 void CTSPModel::setNumCities(int n)
 {
+        if (n == nCities)
+                return;
+        emit layoutAboutToBeChanged();
         if (n > nCities) {
-                foreach(QList<double> *row,table) {
-                        for (int k = nCities; k < n; k++)
-                                row->append(rand(randMin,randMax));
+                for (int r = 0; r < nCities; r++) {
+                        for (int c = nCities; c < n; c++)
+                                if (r == c)
+                                        table[r][c] = INFINITY;
+                                else
+                                        table[r][c] = rand(randMin,randMax);
                 }
-                for (int k = nCities; k < n; k++) {
-QList<double> *row = new QList<double>[n];
-                        foreach(double cell,*row)
-                                cell = rand(randMin,randMax);
-                        table.append(row);
+                for (int r = nCities; r < n; r++) {
+                        for (int c = 0; c < n; c++)
+                                if (r == c)
+                                        table[r][c] = INFINITY;
+                                else
+                                        table[r][c] = rand(randMin,randMax);
                 }
-        } else if (n < nCities) {
-                // TODO: Shrinking table
         }
         nCities = n;
+        emit layoutChanged();
 }
 
+void CTSPModel::randomize()
+{
+        for (int r = 0; r < nCities; r++)
+                for (int c = 0; c < nCities; c++)
+                        if (r != c)
+                                table[r][c] = rand(randMin,randMax);
+        emit dataChanged(index(0,0),index(nCities - 1,nCities - 1));
+}
+