Context Navigation

source: tspsg-svn/trunk/src/tspmodel.cpp @ 42

Last change on this file since 42 was 42, checked in by laleppa, 15 years ago

+ Finished solving algorithm (needs thorough testing).
+ Solution can be saved to HTML or OpenDocument? format.
+ Added VERSIONINFO resource for windows builds.

Updated translations to have unified terminology everywhere.

NB: This will be the first public alpha build.

Property svn:keywords set to Id URL

File size: 9.9 KB

Rev	Line
[14]	1	/*
[42]	2	* TSPSG: TSP Solver and Generator
[17]	3	* Copyright (C) 2007-2009 Lёppa <contacts[at]oleksii[dot]name>
[14]	4	*
	5	* $Id: tspmodel.cpp 42 2009-07-31 18:23:07Z laleppa $
	6	* $URL: https://tspsg.svn.sourceforge.net/svnroot/tspsg/trunk/src/tspmodel.cpp $
	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
	24	#include "tspmodel.h"
	25
	26	CTSPModel::CTSPModel(QObject *parent)
[21]	27	: QAbstractTableModel(parent), nCities(0)
[14]	28	{
[23]	29	settings = new QSettings(QSettings::IniFormat,QSettings::UserScope,"TSPSG","tspsg");
[14]	30	}
	31
[35]	32	inline int CTSPModel::rand(int min, int max) const
[14]	33	{
	34	return min + (int)(((float)qrand() / RAND_MAX) * max);
	35	}
	36
[36]	37	int CTSPModel::rowCount(const QModelIndex &) const
[14]	38	{
	39	return nCities;
	40	}
	41
[36]	42	int CTSPModel::columnCount(const QModelIndex &) const
[14]	43	{
	44	return nCities;
	45	}
	46
[17]	47	QVariant CTSPModel::headerData(int section, Qt::Orientation orientation, int role) const
[14]	48	{
[34]	49	if (role == Qt::DisplayRole) {
[17]	50	if (orientation == Qt::Vertical)
[27]	51	return trUtf8("City %1").arg(section + 1);
[17]	52	else
	53	return trUtf8("%1").arg(section + 1);
[34]	54	}
[14]	55	return QVariant();
	56	}
	57
	58	QVariant 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);
[15]	64	else if (role == Qt::FontRole) {
	65	QFont font;
	66	font.setBold(true);
	67	return font;
	68	} else if (role == Qt::DisplayRole \|\| role == Qt::EditRole) {
[14]	69	if (index.row() < nCities && index.column() < nCities)
[15]	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();
[14]	75	else
	76	return QVariant();
[15]	77	} else if (role == Qt::UserRole)
	78	return table[index.row()][index.column()];
[14]	79	return QVariant();
	80	}
	81
	82	bool CTSPModel::setData(const QModelIndex &index, const QVariant &value, int role)
	83	{
[15]	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 {
	90	bool ok;
	91	double 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;
[14]	101	}
	102
	103	Qt::ItemFlags CTSPModel::flags(const QModelIndex &index) const
	104	{
	105	Qt::ItemFlags flags = QAbstractItemModel::flags(index);
[15]	106	if (index.row() != index.column())
[14]	107	flags \|= Qt::ItemIsEditable;
	108	return flags;
	109	}
	110
[31]	111	quint16 CTSPModel::numCities() const
[14]	112	{
	113	return nCities;
	114	}
	115
	116	void CTSPModel::setNumCities(int n)
	117	{
[15]	118	if (n == nCities)
	119	return;
	120	emit layoutAboutToBeChanged();
[42]	121	table.resize(n);
	122	for (int k = 0; k < n; k++) {
	123	table[k].resize(n);
[14]	124	}
[42]	125	if (n > nCities)
	126	for (int k = nCities; k < n; k++)
	127	table[k][k] = INFINITY;
[14]	128	nCities = n;
[15]	129	emit layoutChanged();
[14]	130	}
	131
[29]	132	void 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
[42]	141	inline bool CTSPModel::loadError(QDataStream::Status status)
[35]	142	{
	143	QString 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
[31]	156	void CTSPModel::loadTask(QString fname)
	157	{
	158	QFile f(fname);
[35]	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();
	161	return;
	162	}
[31]	163	QDataStream ds(&f);
	164	ds.setVersion(QDataStream::Qt_4_4);
	165	quint32 sig;
	166	ds >> sig;
[35]	167	if (loadError(ds.status()))
	168	return;
[31]	169	ds.device()->reset();
	170	if (sig == TSPT)
	171	loadTSPT(&ds);
	172	else if ((sig >> 16) == ZKT)
	173	loadZKT(&ds);
	174	else
[35]	175	QMessageBox(QMessageBox::Critical,trUtf8("Task Load"),trUtf8("Unable to load task:") + "\n" + trUtf8("Unknown file format or file is corrupted."),QMessageBox::Ok).exec();
[31]	176	f.close();
	177	}
	178
	179	void CTSPModel::loadTSPT(QDataStream *ds)
	180	{
	181	// Skipping signature
	182	ds->skipRawData(sizeof(TSPT));
[35]	183	if (loadError(ds->status()))
	184	return;
[31]	185	// File version
	186	quint8 version;
	187	*ds >> version;
[35]	188	if (loadError(ds->status()))
	189	return;
[31]	190	if (version > TSPT_VERSION) {
[35]	191	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();
[31]	192	return;
	193	}
	194	// Skipping metadata
	195	ds->skipRawData(TSPT_META_SIZE);
[35]	196	if (loadError(ds->status()))
	197	return;
[31]	198	// Cities number
	199	quint16 size;
	200	*ds >> size;
[35]	201	if (loadError(ds->status()))
	202	return;
	203	if (size < 3) {
	204	QMessageBox(QMessageBox::Critical,trUtf8("Task Load"),trUtf8("Unable to load task:") + "\n" + trUtf8("Unexpected data read.\nFile is possibly corrupted."),QMessageBox::Ok).exec();
	205	return;
	206	}
[31]	207	if (nCities != size)
	208	emit numCitiesChanged(size);
	209	// Costs
	210	for (int r = 0; r < size; r++)
	211	for (int c = 0; c < size; c++)
[35]	212	if (r != c) {
[31]	213	*ds >> table[r][c];
[35]	214	if (loadError(ds->status())) {
	215	clear();
	216	return;
	217	}
	218	}
[31]	219	emit dataChanged(index(0,0),index(nCities - 1,nCities - 1));
	220	}
	221
	222	void CTSPModel::loadZKT(QDataStream *ds)
	223	{
	224	// Skipping signature
	225	ds->skipRawData(sizeof(ZKT));
[35]	226	if (loadError(ds->status()))
	227	return;
[31]	228	// File version
	229	quint16 version;
	230	ds->readRawData(reinterpret_cast<char *>(&version),2);
[35]	231	if (loadError(ds->status()))
	232	return;
[31]	233	if (version > ZKT_VERSION) {
[35]	234	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();
[31]	235	return;
	236	}
	237	// Cities number
	238	quint8 size;
	239	ds->readRawData(reinterpret_cast<char *>(&size),1);
[35]	240	if (loadError(ds->status()))
	241	return;
	242	if ((size < 3) \|\| (size > 5)) {
	243	QMessageBox(QMessageBox::Critical,trUtf8("Task Load"),trUtf8("Unable to load task:") + "\n" + trUtf8("Unexpected data read.\nFile is possibly corrupted."),QMessageBox::Ok).exec();
	244	return;
	245	}
[31]	246	if (nCities != size)
	247	emit numCitiesChanged(size);
	248	// Costs
	249	double val;
[42]	250	for (int r = 0; r < 5; r++)
	251	for (int c = 0; c < 5; c++)
	252	if ((r != c) && (r < size)) {
[31]	253	ds->readRawData(reinterpret_cast<char *>(&val),8);
[35]	254	if (loadError(ds->status())) {
	255	clear();
	256	return;
	257	}
[31]	258	table[r][c] = val;
[35]	259	} else {
[31]	260	ds->skipRawData(8);
[35]	261	if (loadError(ds->status())) {
	262	clear();
	263	return;
	264	}
	265	}
[31]	266	emit dataChanged(index(0,0),index(nCities - 1,nCities - 1));
	267	}
	268
[37]	269	bool CTSPModel::saveTask(QString fname)
[31]	270	{
	271	QFile f(fname);
[35]	272	if (!f.open(QIODevice::WriteOnly)) {
	273	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();
[37]	274	return false;
[35]	275	}
[31]	276	QDataStream ds(&f);
	277	ds.setVersion(QDataStream::Qt_4_4);
[35]	278	if (f.error() != QFile::NoError) {
	279	QMessageBox(QMessageBox::Critical,trUtf8("Task Save"),trUtf8("Unable to save task.\nError: %1").arg(f.errorString()),QMessageBox::Ok).exec();
	280	f.close();
[37]	281	return false;
[35]	282	}
[31]	283	// File signature
	284	ds << TSPT;
[35]	285	if (f.error() != QFile::NoError) {
	286	QMessageBox(QMessageBox::Critical,trUtf8("Task Save"),trUtf8("Unable to save task.\nError: %1").arg(f.errorString()),QMessageBox::Ok).exec();
	287	f.close();
[37]	288	return false;
[35]	289	}
[31]	290	// File version
	291	ds << TSPT_VERSION;
[35]	292	if (f.error() != QFile::NoError) {
	293	QMessageBox(QMessageBox::Critical,trUtf8("Task Save"),trUtf8("Unable to save task.\nError: %1").arg(f.errorString()),QMessageBox::Ok).exec();
	294	f.close();
[37]	295	return false;
[35]	296	}
[31]	297	// File metadata version
	298	ds << TSPT_META_VERSION;
[35]	299	if (f.error() != QFile::NoError) {
	300	QMessageBox(QMessageBox::Critical,trUtf8("Task Save"),trUtf8("Unable to save task.\nError: %1").arg(f.errorString()),QMessageBox::Ok).exec();
	301	f.close();
[37]	302	return false;
[35]	303	}
[31]	304	// Metadata
	305	ds << OSID;
[35]	306	if (f.error() != QFile::NoError) {
	307	QMessageBox(QMessageBox::Critical,trUtf8("Task Save"),trUtf8("Unable to save task.\nError: %1").arg(f.errorString()),QMessageBox::Ok).exec();
	308	f.close();
[37]	309	return false;
[35]	310	}
[31]	311	// Number of cities
	312	ds << nCities;
[35]	313	if (f.error() != QFile::NoError) {
	314	QMessageBox(QMessageBox::Critical,trUtf8("Task Save"),trUtf8("Unable to save task.\nError: %1").arg(f.errorString()),QMessageBox::Ok).exec();
	315	f.close();
[37]	316	return false;
[35]	317	}
[31]	318	// Costs
	319	for (int r = 0; r < nCities; r++)
	320	for (int c = 0; c < nCities; c++)
[35]	321	if (r != c) {
[31]	322	ds << table[r][c];
[35]	323	if (f.error() != QFile::NoError) {
	324	QMessageBox(QMessageBox::Critical,trUtf8("Task Save"),trUtf8("Unable to save task.\nError: %1").arg(f.errorString()),QMessageBox::Ok).exec();
	325	f.close();
[37]	326	return false;
[35]	327	}
	328	}
[31]	329	f.close();
[37]	330	return true;
[31]	331	}
	332
[15]	333	void CTSPModel::randomize()
	334	{
[21]	335	int randMin = settings->value("MinCost",DEF_RAND_MIN).toInt();
	336	int randMax = settings->value("MaxCost",DEF_RAND_MAX).toInt();
[15]	337	for (int r = 0; r < nCities; r++)
	338	for (int c = 0; c < nCities; c++)
	339	if (r != c)
	340	table[r][c] = rand(randMin,randMax);
	341	emit dataChanged(index(0,0),index(nCities - 1,nCities - 1));
	342	}

Note: See TracBrowser for help on using the repository browser.

Download in other formats: