openscad/dxfdim.cc

/*
 *  OpenSCAD (www.openscad.at)
 *  Copyright (C) 2009  Clifford Wolf <clifford@clifford.at>
 *
 *  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 2 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, write to the Free Software
 *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 *
 */

#include "openscad.h"
#include "printutils.h"

#include <sys/types.h>
#include <sys/stat.h>
#include <unistd.h>

QHash<QString,Value> dxf_dim_cache;
QHash<QString,Value> dxf_cross_cache;

Value builtin_dxf_dim(const QVector<QString> &argnames, const QVector<Value> &args)
{
	QString filename;
	QString layername;
	QString name;
	double xorigin = 0;
	double yorigin = 0;
	double scale = 1;

	for (int i = 0; i < argnames.count() && i < args.count(); i++) {
		if (argnames[i] == "file")
			filename = args[i].text;
		if (argnames[i] == "layer")
			layername = args[i].text;
		if (argnames[i] == "origin")
			args[i].getv2(xorigin, yorigin);
		if (argnames[i] == "scale")
			args[i].getnum(scale);
		if (argnames[i] == "name")
			name = args[i].text;
	}

	struct stat st;
	memset(&st, 0, sizeof(struct stat));
	stat(filename.toAscii().data(), &st);

	QString key = filename + "|" + layername + "|" + name + "|" + QString::number(xorigin) + "|" + QString::number(yorigin) +
			"|" + QString::number(scale) + "|" + QString::number(st.st_mtime) + "|" + QString::number(st.st_size);

	if (dxf_dim_cache.contains(key))
		return dxf_dim_cache[key];

	DxfData dxf(36, 0, 0, filename, layername, xorigin, yorigin, scale);

	for (int i = 0; i < dxf.dims.count(); i++)
	{
		if (!name.isNull() && dxf.dims[i].name != name)
			continue;

		DxfData::Dim *d = &dxf.dims[i];
		int type = d->type & 7;

		if (type == 0) {
			// Rotated, horizontal or vertical
			double x = d->coords[4][0] - d->coords[3][0];
			double y = d->coords[4][1] - d->coords[3][1];
			double angle = d->angle;
			double distance_projected_on_line = fabs(x * cos(angle*M_PI/180) + y * sin(angle*M_PI/180));
			return dxf_dim_cache[key] = Value(distance_projected_on_line);
		}
		if (type == 1) {
			// Aligned
		}
		if (type == 2) {
			// Angular
			double a1 = atan2(d->coords[0][0] - d->coords[5][0], d->coords[0][1] - d->coords[5][1]);
			double a2 = atan2(d->coords[4][0] - d->coords[3][0], d->coords[4][1] - d->coords[3][1]);
			return dxf_dim_cache[key] = Value(fabs(a1 - a2) * 180 / M_PI);
		}
		if (type == 3) {
			// Diameter
		}
		if (type == 4) {
			// Radius
		}
		if (type == 5) {
			// Angular 3 Point
		}

		PRINTA("WARNING: Dimension `%1' in `%2', layer `%3' has unsupported type!", name, filename, layername);
		return Value();
	}

	PRINTA("WARNING: Can't find dimension `%1' in `%2', layer `%3'!", name, filename, layername);

	return Value();
}

Value builtin_dxf_cross(const QVector<QString> &argnames, const QVector<Value> &args)
{
	QString filename;
	QString layername;
	double xorigin = 0;
	double yorigin = 0;
	double scale = 1;

	for (int i = 0; i < argnames.count() && i < args.count(); i++) {
		if (argnames[i] == "file")
			filename = args[i].text;
		if (argnames[i] == "layer")
			layername = args[i].text;
		if (argnames[i] == "origin")
			args[i].getv2(xorigin, yorigin);
		if (argnames[i] == "scale")
			args[i].getnum(scale);
	}

	struct stat st;
	memset(&st, 0, sizeof(struct stat));
	stat(filename.toAscii().data(), &st);

	QString key = filename + "|" + layername + "|" + QString::number(xorigin) + "|" + QString::number(yorigin) +
			"|" + QString::number(scale) + "|" + QString::number(st.st_mtime) + "|" + QString::number(st.st_size);

	if (dxf_cross_cache.contains(key))
		return dxf_cross_cache[key];

	DxfData dxf(36, 0, 0, filename, layername, xorigin, yorigin, scale);

	double coords[4][2];

	for (int i = 0, j = 0; i < dxf.paths.count(); i++) {
		if (dxf.paths[i].points.count() != 2)
			continue;
		coords[j][0] = dxf.paths[i].points[0]->x;
		coords[j++][1] = dxf.paths[i].points[0]->y;
		coords[j][0] = dxf.paths[i].points[1]->x;
		coords[j++][1] = dxf.paths[i].points[1]->y;

		if (j == 4) {
			double x1 = coords[0][0], y1 = coords[0][1];
			double x2 = coords[1][0], y2 = coords[1][1];
			double x3 = coords[2][0], y3 = coords[2][1];
			double x4 = coords[3][0], y4 = coords[3][1];
			double dem = (y4 - y3)*(x2 - x1) - (x4 - x3)*(y2 - y1);
			if (dem == 0)
				break;
			double ua = ((x4 - x3)*(y1 - y3) - (y4 - y3)*(x1 - x3)) / dem;
			// double ub = ((x2 - x1)*(y1 - y3) - (y2 - y1)*(x1 - x3)) / dem;
			double x = x1 + ua*(x2 - x1);
			double y = y1 + ua*(y2 - y1);
			Value ret;
			ret.type = Value::VECTOR;
			ret.vec.append(new Value(x));
			ret.vec.append(new Value(y));
			return dxf_cross_cache[key] = ret;
		}
	}

	PRINTA("WARNING: Can't find cross in `%1', layer `%2'!", filename, layername);

	return Value();
}

void initialize_builtin_dxf_dim()
{
	builtin_functions["dxf_dim"] = new BuiltinFunction(&builtin_dxf_dim);
	builtin_functions["dxf_cross"] = new BuiltinFunction(&builtin_dxf_cross);
}