1 files changed, 97 insertions, 0 deletions
diff --git a/src/CsgInfo.h b/src/CsgInfo.h
new file mode 100644
index 0000000..0413bec
--- /dev/null
+++ b/src/CsgInfo.h
@@ -0,0 +1,97 @@
+#ifndef __CSGINFO_H__
+#define __CSGINFO_H__
+
+#include "OffscreenView.h"
+#include "csgterm.h"
+#include "Tree.h"
+#include "CGALEvaluator.h"
+#include "CSGTermEvaluator.h"
+#include "csgtermnormalizer.h"
+#include "rendersettings.h"
+#include "printutils.h"
+
+class CsgInfo
+{
+public:
+	CsgInfo()
+	{
+		root_chain = NULL;
+		highlights_chain = NULL;
+		background_chain = NULL;
+		glview = NULL;
+		progress_function = NULL;
+		normalizelimit = RenderSettings::inst()->openCSGTermLimit;
+	}
+	OffscreenView *glview;
+	shared_ptr<CSGTerm> root_norm_term;    // Normalized CSG products
+	class CSGChain *root_chain;
+	std::vector<shared_ptr<CSGTerm> > highlight_terms;
+	CSGChain *highlights_chain;
+	std::vector<shared_ptr<CSGTerm> > background_terms;
+	CSGChain *background_chain;
+
+	int normalizelimit;
+
+	void (*progress_function)(void);
+	void set_progress_function( void (*funcname)(void) )
+	{
+		progress_function = funcname;
+	}
+	void call_progress_function()
+	{
+		if (progress_function) progress_function();
+	}
+
+	bool prep_chains( const Tree &tree )
+	{
+		const AbstractNode *root_node = tree.root();
+		CGALEvaluator cgalevaluator(tree);
+		CSGTermEvaluator evaluator(tree, &cgalevaluator.psevaluator);
+		boost::shared_ptr<CSGTerm> root_raw_term = evaluator.evaluateCSGTerm( *root_node, this->highlight_terms, this->background_terms );
+
+		if (!root_raw_term) {
+			fprintf(stderr, "Error: CSG generation failed! (no top level object found)\n");
+			call_progress_function();
+			return false;
+		}
+
+		PRINT("Compiling design (CSG Products normalization)...");
+		call_progress_function();
+		CSGTermNormalizer normalizer( normalizelimit );
+		this->root_norm_term = normalizer.normalize(root_raw_term);
+		if (this->root_norm_term) {
+			this->root_chain = new CSGChain();
+			this->root_chain->import(this->root_norm_term);
+			fprintf(stderr, "Normalized CSG tree has %d elements\n", int(this->root_chain->polysets.size()));
+		}
+		else {
+			this->root_chain = NULL;
+			fprintf(stderr, "WARNING: CSG normalization resulted in an empty tree\n");
+			call_progress_function();
+		}
+
+		if (this->highlight_terms.size() > 0) {
+			std::cerr << "Compiling highlights (" << this->highlight_terms.size() << " CSG Trees)...\n";
+			call_progress_function();
+			this->highlights_chain = new CSGChain();
+			for (unsigned int i = 0; i < this->highlight_terms.size(); i++) {
+				this->highlight_terms[i] = normalizer.normalize(this->highlight_terms[i]);
+				this->highlights_chain->import(this->highlight_terms[i]);
+			}
+		}
+
+		if (this->background_terms.size() > 0) {
+			std::cerr << "Compiling background (%d CSG Trees)..." << this->background_terms.size() << " CSG Trees)...\n";
+			call_progress_function();
+			this->background_chain = new CSGChain();
+			for (unsigned int i = 0; i < this->background_terms.size(); i++) {
+				this->background_terms[i] = normalizer.normalize(this->background_terms[i]);
+				this->background_chain->import(this->background_terms[i]);
+			}
+		}
+		return true;
+	}
+};
+
+#endif
+