openscad/tests/opencsgtest.cc

259 lines
6.8 KiB
C++

#include <GL/glew.h>
#include "openscad.h"
#include "builtin.h"
#include "context.h"
#include "node.h"
#include "module.h"
#include "polyset.h"
#include "Tree.h"
#include "CSGTermEvaluator.h"
#include "CGALEvaluator.h"
#include "PolySetCGALEvaluator.h"
#include "OpenCSGRenderer.h"
#include "ThrownTogetherRenderer.h"
#include "csgterm.h"
#include "OffscreenView.h"
#include <QApplication>
#include <QFile>
#include <QDir>
#include <QSet>
#include <QTimer>
using std::cerr;
using std::cout;
QString commandline_commands;
QString librarydir;
QSet<QString> dependencies;
const char *make_command = NULL;
//#define DEBUG
void handle_dep(QString filename)
{
if (filename.startsWith("/"))
dependencies.insert(filename);
else
dependencies.insert(QDir::currentPath() + QString("/") + filename);
if (!QFile(filename).exists() && make_command) {
char buffer[4096];
snprintf(buffer, 4096, "%s '%s'", make_command, filename.replace("'", "'\\''").toUtf8().data());
system(buffer); // FIXME: Handle error
}
}
// static void renderfunc(void *vp)
// {
// glClearColor(1.0, 0.0, 0.0, 0.0);
// glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT | GL_STENCIL_BUFFER_BIT);
// }
struct CsgInfo
{
CSGTerm *root_norm_term; // Normalized CSG products
class CSGChain *root_chain;
std::vector<CSGTerm*> highlight_terms;
CSGChain *highlights_chain;
std::vector<CSGTerm*> background_terms;
CSGChain *background_chain;
OffscreenView *glview;
};
int main(int argc, char *argv[])
{
if (argc != 2) {
fprintf(stderr, "Usage: %s <file.scad>\n", argv[0]);
exit(1);
}
const char *filename = argv[1];
initialize_builtin_functions();
initialize_builtin_modules();
QApplication app(argc, argv);
QDir original_path = QDir::current();
QString currentdir = QDir::currentPath();
QDir libdir(QApplication::instance()->applicationDirPath());
#ifdef Q_WS_MAC
libdir.cd("../Resources"); // Libraries can be bundled
if (!libdir.exists("libraries")) libdir.cd("../../..");
#elif defined(Q_OS_UNIX)
if (libdir.cd("../share/openscad/libraries")) {
librarydir = libdir.path();
} else
if (libdir.cd("../../share/openscad/libraries")) {
librarydir = libdir.path();
} else
if (libdir.cd("../../libraries")) {
librarydir = libdir.path();
} else
#endif
if (libdir.cd("libraries")) {
librarydir = libdir.path();
}
Context root_ctx;
root_ctx.functions_p = &builtin_functions;
root_ctx.modules_p = &builtin_modules;
root_ctx.set_variable("$fn", Value(0.0));
root_ctx.set_variable("$fs", Value(1.0));
root_ctx.set_variable("$fa", Value(12.0));
root_ctx.set_variable("$t", Value(0.0));
Value zero3;
zero3.type = Value::VECTOR;
zero3.append(new Value(0.0));
zero3.append(new Value(0.0));
zero3.append(new Value(0.0));
root_ctx.set_variable("$vpt", zero3);
root_ctx.set_variable("$vpr", zero3);
AbstractModule *root_module;
ModuleInstantiation root_inst;
AbstractNode *root_node;
QFileInfo fileInfo(filename);
handle_dep(filename);
FILE *fp = fopen(filename, "rt");
if (!fp) {
fprintf(stderr, "Can't open input file `%s'!\n", filename);
exit(1);
} else {
QString text;
char buffer[513];
int ret;
while ((ret = fread(buffer, 1, 512, fp)) > 0) {
buffer[ret] = 0;
text += buffer;
}
fclose(fp);
root_module = parse((text+commandline_commands).toAscii().data(), fileInfo.absolutePath().toLocal8Bit(), false);
if (!root_module) {
exit(1);
}
}
QDir::setCurrent(fileInfo.absolutePath());
AbstractNode::resetIndexCounter();
root_node = root_module->evaluate(&root_ctx, &root_inst);
Tree tree(root_node);
CsgInfo csgInfo;
QHash<std::string, CGAL_Nef_polyhedron> cache;
CGALEvaluator cgalevaluator(cache, tree);
PolySetCGALEvaluator psevaluator(cgalevaluator);
CSGTermEvaluator evaluator(tree);
CSGTerm *root_raw_term = evaluator.evaluateCSGTerm(*root_node,
csgInfo.highlight_terms,
csgInfo.background_terms);
if (!root_raw_term) {
cerr << "Error: CSG generation failed! (no top level object found)\n";
return 1;
}
csgInfo.root_norm_term = root_raw_term->link();
// CSG normalization
while (1) {
CSGTerm *n = csgInfo.root_norm_term->normalize();
csgInfo.root_norm_term->unlink();
if (csgInfo.root_norm_term == n)
break;
csgInfo.root_norm_term = n;
}
assert(csgInfo.root_norm_term);
csgInfo.root_chain = new CSGChain();
csgInfo.root_chain->import(csgInfo.root_norm_term);
if (csgInfo.highlight_terms.size() > 0) {
cerr << "Compiling highlights (" << csgInfo.highlight_terms.size() << " CSG Trees)...\n";
csgInfo.highlights_chain = new CSGChain();
for (unsigned int i = 0; i < csgInfo.highlight_terms.size(); i++) {
while (1) {
CSGTerm *n = csgInfo.highlight_terms[i]->normalize();
csgInfo.highlight_terms[i]->unlink();
if (csgInfo.highlight_terms[i] == n)
break;
csgInfo.highlight_terms[i] = n;
}
csgInfo.highlights_chain->import(csgInfo.highlight_terms[i]);
}
}
if (csgInfo.background_terms.size() > 0) {
cerr << "Compiling background (" << csgInfo.background_terms.size() << " CSG Trees)...\n";
csgInfo.background_chain = new CSGChain();
for (unsigned int i = 0; i < csgInfo.background_terms.size(); i++) {
while (1) {
CSGTerm *n = csgInfo.background_terms[i]->normalize();
csgInfo.background_terms[i]->unlink();
if (csgInfo.background_terms[i] == n)
break;
csgInfo.background_terms[i] = n;
}
csgInfo.background_chain->import(csgInfo.background_terms[i]);
}
}
QDir::setCurrent(original_path.absolutePath());
csgInfo.glview = new OffscreenView(512,512);
BoundingBox bbox = csgInfo.root_chain->getBoundingBox();
Vector3d center = (bbox.min() + bbox.max()) / 2;
double radius = (bbox.max() - bbox.min()).norm() / 2;
Vector3d cameradir(1, 1, -0.5);
Vector3d camerapos = center - radius*1.5*cameradir;
csgInfo.glview->setCamera(camerapos, center);
glewInit();
#ifdef DEBUG
cout << "GLEW version " << glewGetString(GLEW_VERSION) << "\n";
cout << (const char *)glGetString(GL_RENDERER) << "(" << (const char *)glGetString(GL_VENDOR) << ")\n"
<< "OpenGL version " << (const char *)glGetString(GL_VERSION) << "\n";
cout << "Extensions: " << (const char *)glGetString(GL_EXTENSIONS) << "\n";
if (GLEW_ARB_framebuffer_object) {
cout << "ARB_FBO supported\n";
}
if (GLEW_EXT_framebuffer_object) {
cout << "EXT_FBO supported\n";
}
if (GLEW_EXT_packed_depth_stencil) {
cout << "EXT_packed_depth_stencil\n";
}
#endif
OpenCSGRenderer opencsgRenderer(csgInfo.root_chain, csgInfo.highlights_chain, csgInfo.background_chain, csgInfo.glview->shaderinfo);
ThrownTogetherRenderer thrownTogetherRenderer(csgInfo.root_chain, csgInfo.highlights_chain, csgInfo.background_chain);
// csgInfo.glview->setRenderer(&thrownTogetherRenderer);
csgInfo.glview->setRenderer(&opencsgRenderer);
csgInfo.glview->paintGL();
csgInfo.glview->save("/dev/stdout");
destroy_builtin_functions();
destroy_builtin_modules();
return 0;
}