Mercurial > ldforge / file revision

/*
 *  LDForge: LDraw parts authoring CAD
 *  Copyright (C) 2013 - 2018 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 <cmath>
#include <QString>
#include <QObject>
#include <QStringList>
#include <QMetaType>
#include <QVector3D>
#include <functional>
#include "macros.h"

class LDObject;
class QFile;
class QTextStream;
class Matrix;
class LDDocument;

using int8 = qint8;
using int16 = qint16;
using int32 = qint32;
using int64 = qint64;
using uint8 = quint8;
using uint16 = quint16;
using uint32 = quint32;
using uint64 = quint64;
using LDObjectList = QList<LDObject*>;

template<typename T, typename R>
using Pair = std::pair<T, R>;

enum Axis
{
	X,
	Y,
	Z
};

//
// Derivative of QVector3D: this class is used for the vertices.
//
class Vertex : public QVector3D
{
public:
	using ApplyFunction = std::function<void(Axis, double&)>;
	using ApplyConstFunction = std::function<void(Axis, double)>;

	Vertex();
	Vertex(const QVector3D& a);
	Vertex(qreal xpos, qreal ypos, qreal zpos);

	void	apply(ApplyFunction func);
	void	apply(ApplyConstFunction func) const;
	QString	toString(bool mangled = false) const;
	void	transform(const Matrix& matr, const Vertex& pos);
	void	setCoordinate(Axis ax, qreal value);

	Vertex&	operator+= (const Vertex& other);
	Vertex	operator+(const Vertex& other) const;
	Vertex&	operator*= (qreal scalar);
	Vertex	operator*(qreal scalar) const;
	bool	operator<(const Vertex& other) const;
	double	operator[](Axis ax) const;
};

inline Vertex operator*(qreal scalar, const Vertex& vertex)
{
	return vertex * scalar;
}

Q_DECLARE_METATYPE(Vertex)

/*
 * A mathematical 3 x 3 matrix
 */
class Matrix
{
public:
	Matrix();
	Matrix(std::initializer_list<double> values);

	double determinant() const;
	QString toString() const;
	void zero();
	Matrix& operator=(const Matrix& other);
	Matrix operator*(const Matrix& other) const;
	double& operator[](int idx);
	const double& operator[](int idx) const;
	double& operator()(int row, int column);
	const double& operator()(int row, int column) const;
	bool operator==(const Matrix& other) const;

private:
	double coefficients[9];
};

inline double& Matrix::operator[](int idx)
{
	return coefficients[idx];
}

inline const double& Matrix::operator[](int idx) const
{
	return coefficients[idx];
}

inline double& Matrix::operator()(int row, int column)
{
	return coefficients[row * 3 + column];
}

inline const double& Matrix::operator()(int row, int column) const
{
	return coefficients[row * 3 + column];
}


//
// Defines a bounding box that encompasses a given set of objects.
// vertex0 is the minimum vertex, vertex1 is the maximum vertex.
//
class LDBoundingBox
{
public:
	LDBoundingBox();

	void calcObject(LDObject* obj);
	void calcVertex(const Vertex& vertex);
	Vertex center() const;
	bool isEmpty() const;
	double longestMeasurement() const;
	void reset(); 
	const Vertex& vertex0() const;
	const Vertex& vertex1() const;

	LDBoundingBox& operator<<(LDObject* obj);
	LDBoundingBox& operator<<(const Vertex& v);

private:
	bool m_isEmpty;
	Vertex m_vertex0;
	Vertex m_vertex1;
};

static const double Pi = 3.14159265358979323846;


// =============================================================================
// Plural expression
template<typename T>
static inline const char* plural(T n)
{
	return (n != 1) ? "s" : "";
}

template<typename T>
bool isZero(T a)
{
	return qFuzzyCompare(a + 1.0, 1.0);
}

template<typename T>
bool isInteger(T a)
{
	return (qAbs(a - floor(a)) < 0.00001) or(qAbs(a - ceil(a)) < 0.00001);
}

template<typename T>
void removeDuplicates(T& a)
{
	std::sort(a.begin(), a.end());
	a.erase(std::unique(a.begin(), a.end()), a.end());
}

//
// Returns true if first arg is equal to any of the other args
//
template<typename T, typename Arg, typename... Args>
bool isOneOf(T const& a, Arg const& arg, Args const&... args)
{
	if (a == arg)
		return true;

	return isOneOf(a, args...);
}

template<typename T>
bool isOneOf(T const&)
{
	return false;
}