Mercurial > ldforge2 / file revision

/*
 *  LDForge: LDraw parts authoring CAD
 *  Copyright (C) 2013 - 2020 Teemu Piippo
 *
 *  This program is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation, either version 3 of the License, or
 *  (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#pragma once
#include <QDir>
#include <QAbstractTableModel>
#include "src/basics.h"
#include "src/colors.h"

class QSettings;

struct Library
{
	enum Role
	{
		OfficialLibrary,
		UnofficialLibrary,
		WorkingLibrary,
	} role;
	QDir path;
	static QString libraryRoleName(const Role role);
	static bool isValidRole(const Role role);
	constexpr static const Role allRoles[] = {OfficialLibrary, UnofficialLibrary, WorkingLibrary};
};

bool operator==(const Library& one, const Library& other);

Q_DECLARE_METATYPE(Library)
Q_DECLARE_METATYPE(Library::Role)
QDataStream &operator<<(QDataStream&, const Library&);
QDataStream &operator>>(QDataStream&, Library&);
using Libraries = QVector<Library>;

class LibrariesModel : public QAbstractTableModel
{
	Q_OBJECT
public:
	LibrariesModel(QObject* parent = nullptr);
	auto begin() const { return this->libraries.begin(); }
	auto end() const { return this->libraries.end(); }
	QString findFile(QString fileName) const;
	void addLibrary(const Library& library);
	void removeLibrary(const qsizetype libraryIndex);
	const Library& library(qsizetype libraryIndex) const;
	void setLibraryPath(qsizetype libraryIndex, const QDir& path);
	void setLibraryRole(qsizetype libraryIndex, const Library::Role role);
	void restoreFromSettings();
	void storeToSettings();
	qsizetype count() const;
	void moveLibrary(const qsizetype libraryFromIndex, const qsizetype libraryToIndex);
	// Definitions for QAbstractTableModel
	Qt::ItemFlags flags(const QModelIndex& index) const override;
	QVariant data(const QModelIndex& index, int role = Qt::DisplayRole) const override;
	QVariant headerData(
		int section,
		Qt::Orientation orientation,
		int role = Qt::DisplayRole) const override;
	int rowCount(const QModelIndex&) const override;
	int columnCount(const QModelIndex&) const override;
	bool isValidIndex(const qsizetype libraryIndex) const;
	ColorTable loadColorTable(QTextStream& errors) const;
private:
	enum Column
	{
		RoleColumn,
		PathColumn
	};
	void signalLibraryChange(const qsizetype library);
	Libraries libraries;
};