source: tspsg/src/tspmodel.cpp @ aaf2113307

0.1.3.145-beta1-symbian0.1.4.170-beta2-bb10appveyorimgbotreadme
Last change on this file since aaf2113307 was aaf2113307, checked in by Oleksii Serdiuk, 15 years ago

+ Implemented File/Save? action.
+ Added "Save Solution" and "Back to Task" buttons to Solution tab for better usability.

  • Increased maximum number of cities to 20. Solving for 18-20 cities is already takes much time, so I thought it doesn't make sense to increase more.
  • Columns and rows are now resized to contents on all platforms.
  • Property mode set to 100644
File size: 10.3 KB
RevLine 
[2f915f19f2]1/*
[430bd7f7e9]2 *  TSPSG: TSP Solver and Generator
[5354a01311]3 *  Copyright (C) 2007-2009 Lёppa <contacts[at]oleksii[dot]name>
[2f915f19f2]4 *
5 *  $Id$
6 *  $URL$
7 *
8 *  This file is part of TSPSG.
9 *
10 *  TSPSG is free software: you can redistribute it and/or modify
11 *  it under the terms of the GNU General Public License as published by
12 *  the Free Software Foundation, either version 3 of the License, or
13 *  (at your option) any later version.
14 *
15 *  TSPSG is distributed in the hope that it will be useful,
16 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
17 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
18 *  GNU General Public License for more details.
19 *
20 *  You should have received a copy of the GNU General Public License
21 *  along with TSPSG.  If not, see <http://www.gnu.org/licenses/>.
22 */
23
[bcfd415fe2]24#include "tspmodel.h"
[2f915f19f2]25
26CTSPModel::CTSPModel(QObject *parent)
[aecdf994f9]27        : QAbstractTableModel(parent), nCities(0)
[2f915f19f2]28{
[665d32434f]29        settings = new QSettings(QSettings::IniFormat,QSettings::UserScope,"TSPSG","tspsg");
[2f915f19f2]30}
31
[9b7064d770]32inline int CTSPModel::rand(int min, int max) const
[2f915f19f2]33{
34        return min + (int)(((float)qrand() / RAND_MAX) * max);
35}
36
[4c96f94558]37int CTSPModel::rowCount(const QModelIndex &) const
[2f915f19f2]38{
39        return nCities;
40}
41
[4c96f94558]42int CTSPModel::columnCount(const QModelIndex &) const
[2f915f19f2]43{
44        return nCities;
45}
46
[5354a01311]47QVariant CTSPModel::headerData(int section, Qt::Orientation orientation, int role) const
[2f915f19f2]48{
[c02b4903bd]49        if (role == Qt::DisplayRole) {
[5354a01311]50                if (orientation == Qt::Vertical)
[5587b87fac]51                        return trUtf8("City %1").arg(section + 1);
[5354a01311]52                else
53                        return trUtf8("%1").arg(section + 1);
[c02b4903bd]54        }
[2f915f19f2]55        return QVariant();
56}
57
58QVariant CTSPModel::data(const QModelIndex &index, int role) const
59{
60        if (!index.isValid())
61                return QVariant();
62        if (role == Qt::TextAlignmentRole)
63                return int(Qt::AlignCenter);
[2bc8e278b7]64        else if (role == Qt::FontRole) {
65QFont font;
66                font.setBold(true);
67                return font;
68        } else if (role == Qt::DisplayRole || role == Qt::EditRole) {
[2f915f19f2]69                if (index.row() < nCities && index.column() < nCities)
[2bc8e278b7]70                        if (table[index.row()][index.column()] == INFINITY)
71                                return trUtf8(INFSTR);
72                        else
73                                // HACK: Converting to string to prevent spinbox in edit mode
74                                return QVariant(table[index.row()][index.column()]).toString();
[2f915f19f2]75                else
76                        return QVariant();
[2bc8e278b7]77        } else if (role == Qt::UserRole)
78                return table[index.row()][index.column()];
[2f915f19f2]79        return QVariant();
80}
81
82bool CTSPModel::setData(const QModelIndex &index, const QVariant &value, int role)
83{
[2bc8e278b7]84        if (!index.isValid())
85                return false;
86        if (role == Qt::EditRole && index.row() != index.column()) {
87                if (value.toString().compare(INFSTR) == 0)
88                        table[index.row()][index.column()] = INFINITY;
89                else {
90bool ok;
91double tmp = value.toDouble(&ok);
92                        if (!ok || tmp < 0)
93                                return false;
94                        else
95                                table[index.row()][index.column()] = tmp;
96                }
97                emit dataChanged(index,index);
98                return true;
99        }
100        return false;
[2f915f19f2]101}
102
103Qt::ItemFlags CTSPModel::flags(const QModelIndex &index) const
104{
105Qt::ItemFlags flags = QAbstractItemModel::flags(index);
[2bc8e278b7]106        if (index.row() != index.column())
[2f915f19f2]107                flags |= Qt::ItemIsEditable;
108        return flags;
109}
110
[993d5af6f6]111quint16 CTSPModel::numCities() const
[2f915f19f2]112{
113        return nCities;
114}
115
116void CTSPModel::setNumCities(int n)
117{
[2bc8e278b7]118        if (n == nCities)
119                return;
120        emit layoutAboutToBeChanged();
[430bd7f7e9]121        table.resize(n);
122        for (int k = 0; k < n; k++) {
123                table[k].resize(n);
[2f915f19f2]124        }
[430bd7f7e9]125        if (n > nCities)
126                for (int k = nCities; k < n; k++)
127                        table[k][k] = INFINITY;
[2f915f19f2]128        nCities = n;
[2bc8e278b7]129        emit layoutChanged();
130}
131
[899d1b8e15]132void CTSPModel::clear()
133{
134        for (int r = 0; r < nCities; r++)
135                for (int c = 0; c < nCities; c++)
136                        if (r != c)
137                                table[r][c] = 0;
138        emit dataChanged(index(0,0),index(nCities - 1,nCities - 1));
139}
140
[430bd7f7e9]141inline bool CTSPModel::loadError(QDataStream::Status status)
[9b7064d770]142{
143QString err;
144        if (status == QDataStream::Ok)
145                return false;
146        else if (status == QDataStream::ReadPastEnd)
147                err = trUtf8("Unexpected end of file.");
148        else if (status == QDataStream::ReadCorruptData)
149                err = trUtf8("Corrupt data read. File possibly corrupted.");
150        else
151                err = trUtf8("Unknown error.");
152        QMessageBox(QMessageBox::Critical,trUtf8("Task Load"),trUtf8("Unable to load task:") + "\n" + err,QMessageBox::Ok).exec();
153        return true;
154}
155
[b424a7e320]156bool CTSPModel::loadTask(QString fname)
[993d5af6f6]157{
158QFile f(fname);
[9b7064d770]159        if (!f.open(QIODevice::ReadOnly)) {
160                QMessageBox(QMessageBox::Critical,trUtf8("Task Load"),QString(trUtf8("Unable to open task file.\nError: %1")).arg(f.errorString()),QMessageBox::Ok).exec();
[b424a7e320]161                return false;
[9b7064d770]162        }
[993d5af6f6]163QDataStream ds(&f);
164        ds.setVersion(QDataStream::Qt_4_4);
165quint32 sig;
166        ds >> sig;
[9b7064d770]167        if (loadError(ds.status()))
[b424a7e320]168                return false;
[993d5af6f6]169        ds.device()->reset();
[e96fad3079]170        if (sig == TSPT) {
[b424a7e320]171                if (!loadTSPT(&ds)) {
172                        f.close();
173                        return false;
174                }
[e96fad3079]175        } else if ((sig >> 16) == ZKT) {
[b424a7e320]176                if (!loadZKT(&ds)) {
177                        f.close();
178                        return false;
179                }
[e96fad3079]180        } else {
[9b7064d770]181                QMessageBox(QMessageBox::Critical,trUtf8("Task Load"),trUtf8("Unable to load task:") + "\n" + trUtf8("Unknown file format or file is corrupted."),QMessageBox::Ok).exec();
[b424a7e320]182                f.close();
183                return false;
184        }
[993d5af6f6]185        f.close();
[b424a7e320]186        return true;
[993d5af6f6]187}
188
[b424a7e320]189bool CTSPModel::loadTSPT(QDataStream *ds)
[993d5af6f6]190{
191        // Skipping signature
192        ds->skipRawData(sizeof(TSPT));
[9b7064d770]193        if (loadError(ds->status()))
[b424a7e320]194                return false;
[993d5af6f6]195        // File version
196quint8 version;
197        *ds >> version;
[9b7064d770]198        if (loadError(ds->status()))
[b424a7e320]199                return false;
[993d5af6f6]200        if (version > TSPT_VERSION) {
[9b7064d770]201                QMessageBox(QMessageBox::Critical,trUtf8("Task Load"),trUtf8("Unable to load task:") + "\n" + trUtf8("File version is newer than application supports.\nPlease, try to update application."),QMessageBox::Ok).exec();
[b424a7e320]202                return false;
[993d5af6f6]203        }
204        // Skipping metadata
205        ds->skipRawData(TSPT_META_SIZE);
[9b7064d770]206        if (loadError(ds->status()))
[b424a7e320]207                return false;
[aaf2113307]208        // Number of cities
[993d5af6f6]209quint16 size;
210        *ds >> size;
[9b7064d770]211        if (loadError(ds->status()))
[b424a7e320]212                return false;
[aaf2113307]213        if ((size < 3) || (size > MAX_NUM_CITIES)) {
[9b7064d770]214                QMessageBox(QMessageBox::Critical,trUtf8("Task Load"),trUtf8("Unable to load task:") + "\n" + trUtf8("Unexpected data read.\nFile is possibly corrupted."),QMessageBox::Ok).exec();
[b424a7e320]215                return false;
[9b7064d770]216        }
[aaf2113307]217        if (nCities != size) {
218                setNumCities(size);
[993d5af6f6]219                emit numCitiesChanged(size);
[aaf2113307]220        }
221        // Travel costs
[993d5af6f6]222        for (int r = 0; r < size; r++)
223                for (int c = 0; c < size; c++)
[9b7064d770]224                        if (r != c) {
[993d5af6f6]225                                *ds >> table[r][c];
[9b7064d770]226                                if (loadError(ds->status())) {
227                                        clear();
[b424a7e320]228                                        return false;
[9b7064d770]229                                }
230                        }
[993d5af6f6]231        emit dataChanged(index(0,0),index(nCities - 1,nCities - 1));
[b424a7e320]232        return true;
[993d5af6f6]233}
234
[b424a7e320]235bool CTSPModel::loadZKT(QDataStream *ds)
[993d5af6f6]236{
237        // Skipping signature
238        ds->skipRawData(sizeof(ZKT));
[9b7064d770]239        if (loadError(ds->status()))
[b424a7e320]240                return false;
[993d5af6f6]241        // File version
242quint16 version;
243        ds->readRawData(reinterpret_cast<char *>(&version),2);
[9b7064d770]244        if (loadError(ds->status()))
[b424a7e320]245                return false;
[993d5af6f6]246        if (version > ZKT_VERSION) {
[9b7064d770]247                QMessageBox(QMessageBox::Critical,trUtf8("Task Load"),trUtf8("Unable to load task:") + "\n" + trUtf8("File version is newer than application supports.\nPlease, try to update application."),QMessageBox::Ok).exec();
[b424a7e320]248                return false;
[993d5af6f6]249        }
[aaf2113307]250        // Number of cities
[993d5af6f6]251quint8 size;
252        ds->readRawData(reinterpret_cast<char *>(&size),1);
[9b7064d770]253        if (loadError(ds->status()))
[b424a7e320]254                return false;
[9b7064d770]255        if ((size < 3) || (size > 5)) {
256                QMessageBox(QMessageBox::Critical,trUtf8("Task Load"),trUtf8("Unable to load task:") + "\n" + trUtf8("Unexpected data read.\nFile is possibly corrupted."),QMessageBox::Ok).exec();
[b424a7e320]257                return false;
[9b7064d770]258        }
[aaf2113307]259        if (nCities != size) {
260                setNumCities(size);
[993d5af6f6]261                emit numCitiesChanged(size);
[aaf2113307]262        }
263        // Travel costs
[993d5af6f6]264double val;
[430bd7f7e9]265        for (int r = 0; r < 5; r++)
266                for (int c = 0; c < 5; c++)
267                        if ((r != c) && (r < size)) {
[993d5af6f6]268                                ds->readRawData(reinterpret_cast<char *>(&val),8);
[9b7064d770]269                                if (loadError(ds->status())) {
270                                        clear();
[b424a7e320]271                                        return false;
[9b7064d770]272                                }
[993d5af6f6]273                                table[r][c] = val;
[9b7064d770]274                        } else {
[993d5af6f6]275                                ds->skipRawData(8);
[9b7064d770]276                                if (loadError(ds->status())) {
277                                        clear();
[b424a7e320]278                                        return false;
[9b7064d770]279                                }
280                        }
[993d5af6f6]281        emit dataChanged(index(0,0),index(nCities - 1,nCities - 1));
[b424a7e320]282        return true;
[993d5af6f6]283}
284
[690f6939a7]285bool CTSPModel::saveTask(QString fname)
[993d5af6f6]286{
287QFile f(fname);
[9b7064d770]288        if (!f.open(QIODevice::WriteOnly)) {
289                QMessageBox(QMessageBox::Critical,trUtf8("Task Save"),QString(trUtf8("Unable to create task file.\nError: %1\nMaybe, file is read-only?")).arg(f.errorString()),QMessageBox::Ok).exec();
[690f6939a7]290                return false;
[9b7064d770]291        }
[993d5af6f6]292QDataStream ds(&f);
293        ds.setVersion(QDataStream::Qt_4_4);
[9b7064d770]294        if (f.error() != QFile::NoError) {
295                QMessageBox(QMessageBox::Critical,trUtf8("Task Save"),trUtf8("Unable to save task.\nError: %1").arg(f.errorString()),QMessageBox::Ok).exec();
296                f.close();
[690f6939a7]297                return false;
[9b7064d770]298        }
[993d5af6f6]299        // File signature
300        ds << TSPT;
[9b7064d770]301        if (f.error() != QFile::NoError) {
302                QMessageBox(QMessageBox::Critical,trUtf8("Task Save"),trUtf8("Unable to save task.\nError: %1").arg(f.errorString()),QMessageBox::Ok).exec();
303                f.close();
[690f6939a7]304                return false;
[9b7064d770]305        }
[993d5af6f6]306        // File version
307        ds << TSPT_VERSION;
[9b7064d770]308        if (f.error() != QFile::NoError) {
309                QMessageBox(QMessageBox::Critical,trUtf8("Task Save"),trUtf8("Unable to save task.\nError: %1").arg(f.errorString()),QMessageBox::Ok).exec();
310                f.close();
[690f6939a7]311                return false;
[9b7064d770]312        }
[993d5af6f6]313        // File metadata version
314        ds << TSPT_META_VERSION;
[9b7064d770]315        if (f.error() != QFile::NoError) {
316                QMessageBox(QMessageBox::Critical,trUtf8("Task Save"),trUtf8("Unable to save task.\nError: %1").arg(f.errorString()),QMessageBox::Ok).exec();
317                f.close();
[690f6939a7]318                return false;
[9b7064d770]319        }
[993d5af6f6]320        // Metadata
321        ds << OSID;
[9b7064d770]322        if (f.error() != QFile::NoError) {
323                QMessageBox(QMessageBox::Critical,trUtf8("Task Save"),trUtf8("Unable to save task.\nError: %1").arg(f.errorString()),QMessageBox::Ok).exec();
324                f.close();
[690f6939a7]325                return false;
[9b7064d770]326        }
[993d5af6f6]327        // Number of cities
328        ds << nCities;
[9b7064d770]329        if (f.error() != QFile::NoError) {
330                QMessageBox(QMessageBox::Critical,trUtf8("Task Save"),trUtf8("Unable to save task.\nError: %1").arg(f.errorString()),QMessageBox::Ok).exec();
331                f.close();
[690f6939a7]332                return false;
[9b7064d770]333        }
[993d5af6f6]334        // Costs
335        for (int r = 0; r < nCities; r++)
336                for (int c = 0; c < nCities; c++)
[9b7064d770]337                        if (r != c) {
[993d5af6f6]338                                ds << table[r][c];
[9b7064d770]339                                if (f.error() != QFile::NoError) {
340                                        QMessageBox(QMessageBox::Critical,trUtf8("Task Save"),trUtf8("Unable to save task.\nError: %1").arg(f.errorString()),QMessageBox::Ok).exec();
341                                        f.close();
[690f6939a7]342                                        return false;
[9b7064d770]343                                }
344                        }
[993d5af6f6]345        f.close();
[690f6939a7]346        return true;
[993d5af6f6]347}
348
[2bc8e278b7]349void CTSPModel::randomize()
350{
[aecdf994f9]351int randMin = settings->value("MinCost",DEF_RAND_MIN).toInt();
352int randMax = settings->value("MaxCost",DEF_RAND_MAX).toInt();
[2bc8e278b7]353        for (int r = 0; r < nCities; r++)
354                for (int c = 0; c < nCities; c++)
355                        if (r != c)
356                                table[r][c] = rand(randMin,randMax);
357        emit dataChanged(index(0,0),index(nCities - 1,nCities - 1));
[2f915f19f2]358}
Note: See TracBrowser for help on using the repository browser.