Mercurial > ldforge / file revision

/*
 *  LDForge: LDraw parts authoring CAD
 *  Copyright (C) 2013 - 2017 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 "main.h"

/*
 * Models an LDraw library.
 */
class Library
{
public:
	class SubdirectoryNotFoundError : public std::exception {};

	enum FileType
	{
		Part,
		Subpart,
		Primitive,
		HighResolutionPrimitive,
		LowResolutionPrimitive,
	};

	Library(const QDir& path, const QString& name, bool frozen);

	bool frozen() const;
	QDir path() const;
	QString name() const;
	QDir subdirectory(FileType type) const;
	static QString subdirectoryName(FileType type);

private:
	QDir _path;
	QString _name;
	bool _frozen;
};

/*
 * Models a collection of libraries. A library is a folder containing directories named "p" and
 * "parts" that contains part and subpart files.
 */
class LibraryCollection : public QObject
{
	Q_OBJECT
	using Super = QVector<Library>;

public:
	struct FileSpec
	{
		QString name;
		Library::FileType type;
	};

	void addLibrary(QDir dir, QString name, bool frozen);
	QString find(QString relativePath) const;
	QString find(const QString& name, Library::FileType type) const;
	QString find(const FileSpec& spec) const;
	std::vector<Library>::iterator begin();
	std::vector<Library>::iterator end();

	static const QStringList directoryNames;

private:
	std::vector<Library> _libraries;
};

#include <QAbstractTableModel>
class LibraryCollectionTableModel : public QAbstractTableModel
{
public:
	LibraryCollectionTableModel(LibraryCollection& collection, QObject* parent = nullptr);

private:
	LibraryCollection& _collection;
};