Mercurial > ldforge / file revision

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

#include <QDir>
#include <QRegExp>
#include <QFileDialog>
#include "ldDocument.h"
#include "mainwindow.h"
#include "primitives.h"
#include "ui_makeprim.h"
#include "miscallenous.h"
#include "colors.h"
#include "ldpaths.h"
#include "documentmanager.h"

QList<PrimitiveCategory*> g_PrimitiveCategories;
QList<Primitive> g_primitives;
static PrimitiveScanner* g_activeScanner = nullptr;
PrimitiveCategory* g_unmatched = nullptr;

static const QStringList g_radialNameRoots =
{
	"edge",
	"cyli",
	"disc",
	"ndis",
	"ring",
	"con"
};

PrimitiveScanner* ActivePrimitiveScanner()
{
	return g_activeScanner;
}

QString getPrimitivesCfgPath()
{
	return qApp->applicationDirPath() + DIRSLASH "prims.cfg";
}

// =============================================================================
//
void LoadPrimitives()
{
	// Try to load prims.cfg
	QFile conf(getPrimitivesCfgPath());

	if (not conf.open(QIODevice::ReadOnly))
	{
		// No prims.cfg, build it
		PrimitiveScanner::start();
	}
	else
	{
		while (not conf.atEnd())
		{
			QString line = conf.readLine();

			if (line.endsWith("\n"))
				line.chop(1);

			if (line.endsWith("\r"))
				line.chop(1);

			int space = line.indexOf(" ");

			if (space == -1)
				continue;

			Primitive info;
			info.name = line.left(space);
			info.title = line.mid(space + 1);
			g_primitives << info;
		}

		PrimitiveCategory::populateCategories();
		print("%1 primitives loaded.\n", g_primitives.size());
	}
}

// =============================================================================
//
static void GetRecursiveFilenames(QDir dir, QList<QString>& fnames)
{
	QFileInfoList flist = dir.entryInfoList(QDir::Files | QDir::Dirs | QDir::NoDotAndDotDot);

	for (const QFileInfo& info : flist)
	{
		if (info.isDir())
			GetRecursiveFilenames(QDir(info.absoluteFilePath()), fnames);
		else
			fnames << info.absoluteFilePath();
	}
}

// =============================================================================
//
PrimitiveScanner::PrimitiveScanner(QObject* parent) :
	QObject(parent),
	m_i(0)
{
	g_activeScanner = this;
	QDir dir = LDPaths::primitivesDir();
	m_baselen = dir.absolutePath().length();
	GetRecursiveFilenames(dir, m_files);
	emit starting(m_files.size());
	print("Scanning primitives...");
}

// =============================================================================
//
PrimitiveScanner::~PrimitiveScanner()
{
	g_activeScanner = nullptr;
}

// =============================================================================
//
void PrimitiveScanner::work()
{
	int j = qMin(m_i + 100, m_files.size());

	for (; m_i < j; ++m_i)
	{
		QString fname = m_files[m_i];
		QFile f(fname);

		if (not f.open(QIODevice::ReadOnly))
			continue;

		Primitive info;
		info.name = fname.mid(m_baselen + 1);  // make full path relative
		info.name.replace('/', '\\');  // use DOS backslashes, they're expected
		info.category = nullptr;
		QByteArray titledata = f.readLine();

		if (titledata != QByteArray())
			info.title = QString::fromUtf8(titledata);

		info.title = info.title.simplified();

		if (Q_LIKELY(info.title[0] == '0'))
		{
			info.title.remove(0, 1);  // remove 0
			info.title = info.title.simplified();
		}

		m_prims << info;
	}

	if (m_i == m_files.size())
	{
		// Done with primitives, now save to a config file
		QString path = getPrimitivesCfgPath();
		QFile conf(path);

		if (not conf.open(QIODevice::WriteOnly | QIODevice::Text))
			Critical(format("Couldn't write primitive list %1: %2",
				path, conf.errorString()));
		else
		{
			for (Primitive& info : m_prims)
				fprint(conf, "%1 %2\r\n", info.name, info.title);

			conf.close();
		}

		g_primitives = m_prims;
		PrimitiveCategory::populateCategories();
		print("%1 primitives scanned", g_primitives.size());
		g_activeScanner = nullptr;
		emit workDone();
		deleteLater();
	}
	else
	{
		// Defer to event loop, pick up the work later
		emit update(m_i);
		QMetaObject::invokeMethod(this, "work", Qt::QueuedConnection);
	}
}

// =============================================================================
//
void PrimitiveScanner::start()
{
	if (g_activeScanner)
		return;

	PrimitiveCategory::loadCategories();
	PrimitiveScanner* scanner = new PrimitiveScanner;
	scanner->work();
}

// =============================================================================
//
PrimitiveCategory::PrimitiveCategory(QString name, QObject* parent) :
	QObject(parent),
	m_name(name) {}

// =============================================================================
//
void PrimitiveCategory::populateCategories()
{
	loadCategories();

	for (PrimitiveCategory* cat : g_PrimitiveCategories)
		cat->prims.clear();

	for (Primitive& prim : g_primitives)
	{
		bool matched = false;
		prim.category = nullptr;

		// Go over the categories and their regexes, if and when there's a match,
		// the primitive's category is set to the category the regex beloings to.
		for (PrimitiveCategory* cat : g_PrimitiveCategories)
		{
			for (RegexEntry& entry : cat->regexes)
			{
				switch(entry.type)
				{
					case EFilenameRegex:
					{
						// f-regex, check against filename
						matched = entry.regex.exactMatch(prim.name);
					} break;

					case ETitleRegex:
					{
						// t-regex, check against title
						matched = entry.regex.exactMatch(prim.title);
					} break;
				}

				if (matched)
				{
					prim.category = cat;
					break;
				}
			}

			// Drop out if a category was decided on.
			if (prim.category)
				break;
		}

		// If there was a match, add the primitive to the category.
		// Otherwise, add it to the list of unmatched primitives.
		if (prim.category)
			prim.category->prims << prim;
		else
			g_unmatched->prims << prim;
	}

	// Sort the categories. Note that we do this here because we need the existing
	// order for regex matching.
	qSort(g_PrimitiveCategories.begin(), g_PrimitiveCategories.end(),
		[](PrimitiveCategory* const& a, PrimitiveCategory* const& b) -> bool
		{
			return a->name() < b->name();
		});
}

// =============================================================================
//
void PrimitiveCategory::loadCategories()
{
	for (PrimitiveCategory* cat : g_PrimitiveCategories)
		delete cat;

	g_PrimitiveCategories.clear();
	QString path = ":/data/primitive-categories.cfg";
	QFile f(path);

	if (not f.open(QIODevice::ReadOnly))
	{
		Critical(format(QObject::tr("Failed to open primitive categories: %1"), f.errorString()));
		return;
	}

	PrimitiveCategory* cat = nullptr;

	while (not f.atEnd())
	{
		QString line = f.readLine();
		int colon;

		if (line.endsWith("\n"))
			line.chop(1);

		if (line.length() == 0 or line[0] == '#')
			continue;

		if ((colon = line.indexOf(":")) == -1)
		{
			if (cat and cat->isValidToInclude())
				g_PrimitiveCategories << cat;

			cat = new PrimitiveCategory(line);
		}
		else if (cat)
		{
			QString cmd = line.left(colon);
			RegexType type = EFilenameRegex;

			if (cmd == "f")
				type = EFilenameRegex;
			else if (cmd == "t")
				type = ETitleRegex;
			else
			{
				print(tr("Warning: unknown command \"%1\" on line \"%2\""), cmd, line);
				continue;
			}

			QRegExp regex(line.mid(colon + 1));
			RegexEntry entry = { regex, type };
			cat->regexes << entry;
		}
		else
			print("Warning: Rules given before the first category name");
	}

	if (cat->isValidToInclude())
		g_PrimitiveCategories << cat;

	// Add a category for unmatched primitives.
	// Note: if this function is called the second time, g_unmatched has been
	// deleted at the beginning of the function and is dangling at this point.
	g_unmatched = new PrimitiveCategory(tr("Other"));
	g_PrimitiveCategories << g_unmatched;
	f.close();
}

// =============================================================================
//
bool PrimitiveCategory::isValidToInclude()
{
	if (regexes.isEmpty())
	{
		print(tr("Warning: category \"%1\" left without patterns"), name());
		deleteLater();
		return false;
	}

	return true;
}

QString PrimitiveCategory::name() const
{
	return m_name;
}

// =============================================================================
//
bool IsPrimitiveLoaderBusy()
{
	return g_activeScanner;
}

// =============================================================================
//
static double GetRadialPoint(int i, int divs, double(*func)(double))
{
	return (*func)((i * 2 * Pi) / divs);
}

// =============================================================================
//
void MakeCircle(int segs, int divs, double radius, QList<QLineF>& lines)
{
	for (int i = 0; i < segs; ++i)
	{
		double x0 = radius * GetRadialPoint(i, divs, cos),
			x1 = radius * GetRadialPoint(i + 1, divs, cos),
			z0 = radius * GetRadialPoint(i, divs, sin),
			z1 = radius * GetRadialPoint(i + 1, divs, sin);

		lines << QLineF(QPointF(x0, z0), QPointF(x1, z1));
	}
}

// =============================================================================
//
LDObjectList MakePrimitive(PrimitiveType type, int segs, int divs, int num)
{
	LDObjectList objs;
	QList<int> condLineSegs;
	QList<QLineF> circle;

	MakeCircle(segs, divs, 1, circle);

	for (int i = 0; i < segs; ++i)
	{
		double x0 = circle[i].x1(),
				   x1 = circle[i].x2(),
				   z0 = circle[i].y1(),
				   z1 = circle[i].y2();

		switch(type)
		{
			case Circle:
			{
				Vertex v0(x0, 0.0f, z0),
				  v1(x1, 0.0f, z1);

				LDLine* line(LDSpawn<LDLine>());
				line->setVertex(0, v0);
				line->setVertex(1, v1);
				line->setColor(EdgeColor);
				objs << line;
			} break;

			case Cylinder:
			case Ring:
			case Cone:
			{
				double x2, x3, z2, z3;
				double y0, y1, y2, y3;

				if (type == Cylinder)
				{
					x2 = x1;
					x3 = x0;
					z2 = z1;
					z3 = z0;

					y0 = y1 = 0.0f;
					y2 = y3 = 1.0f;
				}
				else
				{
					x2 = x1 *(num + 1);
					x3 = x0 *(num + 1);
					z2 = z1 *(num + 1);
					z3 = z0 *(num + 1);

					x0 *= num;
					x1 *= num;
					z0 *= num;
					z1 *= num;

					if (type == Ring)
						y0 = y1 = y2 = y3 = 0.0f;
					else
					{
						y0 = y1 = 1.0f;
						y2 = y3 = 0.0f;
					}
				}

				Vertex v0(x0, y0, z0),
					   v1(x1, y1, z1),
					   v2(x2, y2, z2),
					   v3(x3, y3, z3);

				LDQuad* quad(LDSpawn<LDQuad>(v0, v1, v2, v3));
				quad->setColor(MainColor);

				if (type == Cylinder)
					quad->invert();

				objs << quad;

				if (type == Cylinder or type == Cone)
					condLineSegs << i;
			} break;

			case Disc:
			case DiscNeg:
			{
				double x2, z2;

				if (type == Disc)
					x2 = z2 = 0.0f;
				else
				{
					x2 = (x0 >= 0.0f) ? 1.0f : -1.0f;
					z2 = (z0 >= 0.0f) ? 1.0f : -1.0f;
				}

				Vertex v0(x0, 0.0f, z0),
					   v1(x1, 0.0f, z1),
					   v2(x2, 0.0f, z2);

				// Disc negatives need to go the other way around, otherwise
				// they'll end up upside-down.
				LDTriangle* seg(LDSpawn<LDTriangle>());
				seg->setColor(MainColor);
				seg->setVertex(type == Disc ? 0 : 2, v0);
				seg->setVertex(1, v1);
				seg->setVertex(type == Disc ? 2 : 0, v2);
				objs << seg;
			} break;
		}
	}

	// If this is not a full circle, we need a conditional line at the other
	// end, too.
	if (segs < divs and condLineSegs.size() != 0)
		condLineSegs << segs;

	for (int i : condLineSegs)
	{
		Vertex v0(GetRadialPoint(i, divs, cos), 0.0f, GetRadialPoint(i, divs, sin)),
		  v1,
		  v2(GetRadialPoint(i + 1, divs, cos), 0.0f, GetRadialPoint(i + 1, divs, sin)),
		  v3(GetRadialPoint(i - 1, divs, cos), 0.0f, GetRadialPoint(i - 1, divs, sin));

		if (type == Cylinder)
		{
			v1 = Vertex(v0[X], 1.0f, v0[Z]);
		}
		else if (type == Cone)
		{
			v1 = Vertex(v0[X] *(num + 1), 0.0f, v0[Z] *(num + 1));
			v0.setX(v0.x() * num);
			v0.setY(1.0);
			v0.setZ(v0.z() * num);
		}

		LDCondLine* line = (LDSpawn<LDCondLine>());
		line->setColor(EdgeColor);
		line->setVertex(0, v0);
		line->setVertex(1, v1);
		line->setVertex(2, v2);
		line->setVertex(3, v3);
		objs << line;
	}

	return objs;
}

// =============================================================================
//
static QString PrimitiveTypeName(PrimitiveType type)
{
	// Not translated as primitives are in English.
	return type == Circle   ? "Circle" :
		   type == Cylinder ? "Cylinder" :
		   type == Disc     ? "Disc" :
		   type == DiscNeg  ? "Disc Negative" :
		   type == Ring     ? "Ring" : "Cone";
}

// =============================================================================
//
QString MakeRadialFileName(PrimitiveType type, int segs, int divs, int num)
{
	int numer = segs,
			denom = divs;

	// Simplify the fractional part, but the denominator must be at least 4.
	Simplify(numer, denom);

	if (denom < 4)
	{
		const int factor = 4 / denom;
		numer *= factor;
		denom *= factor;
	}

	// Compose some general information: prefix, fraction, root, ring number
	QString prefix = (divs == LowResolution) ? "" : format("%1/", divs);
	QString frac = format("%1-%2", numer, denom);
	QString root = g_radialNameRoots[type];
	QString numstr = (type == Ring or type == Cone) ? format("%1", num) : "";

	// Truncate the root if necessary(7-16rin4.dat for instance).
	// However, always keep the root at least 2 characters.
	int extra = (frac.length() + numstr.length() + root.length()) - 8;
	root.chop(qBound(0, extra, 2));

	// Stick them all together and return the result.
	return prefix + frac + root + numstr + ".dat";
}

// =============================================================================
//
LDDocument* GeneratePrimitive(PrimitiveType type, int segs, int divs, int num)
{
	// Make the description
	QString frac = QString::number((float) segs / divs);
	QString name = MakeRadialFileName(type, segs, divs, num);
	QString descr;

	// Ensure that there's decimals, even if they're 0.
	if (frac.indexOf(".") == -1)
		frac += ".0";

	if (type == Ring or type == Cone)
	{
		QString spacing =
			(num < 10) ? "  " :
			(num < 100) ? " "  : "";

		descr = format("%1 %2%3 x %4", PrimitiveTypeName(type), spacing, num, frac);
	}
	else
		descr = format("%1 %2", PrimitiveTypeName(type), frac);

	// Prepend "Hi-Res" if 48/ primitive.
	if (divs == HighResolution)
		descr.insert(0, "Hi-Res ");

	LDDocument* document = g_win->newDocument();
	document->setDefaultName(name);

	QString author = APPNAME;
	QString license = "";

	if (not config.defaultName().isEmpty())
	{
		license = PreferredLicenseText();
		author = format("%1 [%2]", config.defaultName(), config.defaultUser());
	}

	LDObjectList objs;

	objs << LDSpawn<LDComment>(descr)
		 << LDSpawn<LDComment>(format("Name: %1", name))
		 << LDSpawn<LDComment>(format("Author: %1", author))
		 << LDSpawn<LDComment>(format("!LDRAW_ORG Unofficial_%1Primitive",
									  divs == HighResolution ?  "48_" : ""))
		 << LDSpawn<LDComment>(license)
		 << LDSpawn<LDEmpty>()
		 << LDSpawn<LDBfc>(BfcStatement::CertifyCCW)
		 << LDSpawn<LDEmpty>();

	document->openForEditing();
	document->history()->setIgnoring(false);
	document->addObjects(objs);
	document->addObjects(MakePrimitive(type, segs, divs, num));
	document->addHistoryStep();
	return document;
}

// =============================================================================
//
LDDocument* GetPrimitive(PrimitiveType type, int segs, int divs, int num)
{
	QString name = MakeRadialFileName(type, segs, divs, num);
	LDDocument* f = g_win->documents()->getDocumentByName(name);

	if (f)
		return f;

	return GeneratePrimitive(type, segs, divs, num);
}

// =============================================================================
//
PrimitivePrompt::PrimitivePrompt(QWidget* parent, Qt::WindowFlags f) :
	QDialog(parent, f)
{
	ui = new Ui_MakePrimUI;
	ui->setupUi(this);
	connect(ui->cb_hires, SIGNAL(toggled(bool)), this, SLOT(hiResToggled(bool)));
}

// =============================================================================
//
PrimitivePrompt::~PrimitivePrompt()
{
	delete ui;
}

// =============================================================================
//
void PrimitivePrompt::hiResToggled(bool on)
{
	ui->sb_segs->setMaximum(on ? HighResolution : LowResolution);

	// If the current value is 16 and we switch to hi-res, default the
	// spinbox to 48.
	if (on and ui->sb_segs->value() == LowResolution)
		ui->sb_segs->setValue(HighResolution);
}