Merge pull request #288 from openscad/issue11_2

Issue11 2

1 files changed, 100 insertions, 0 deletions

diff --git a/src/export_png.cc b/src/export_png.cc
new file mode 100644
index 0000000..c6c191c
--- /dev/null
+++ b/src/export_png.cc
@@ -0,0 +1,100 @@
+#include "export.h"
+#include "printutils.h"
+#include "OffscreenView.h"
+#include "CsgInfo.h"
+#include <stdio.h>
+#include "polyset.h"
+#include "rendersettings.h"
+
+#ifdef ENABLE_CGAL
+#include "CGALRenderer.h"
+#include "CGAL_renderer.h"
+#include "cgal.h"
+
+void export_png_with_cgal(CGAL_Nef_polyhedron *root_N, Camera &cam, std::ostream &output)
+{
+	OffscreenView *glview;
+	try {
+		glview = new OffscreenView( cam.pixel_width, cam.pixel_height );
+	} catch (int error) {
+		fprintf(stderr,"Can't create OpenGL OffscreenView. Code: %i.\n", error);
+		return;
+	}
+	CGALRenderer cgalRenderer(*root_N);
+
+	BoundingBox bbox;
+	if (cgalRenderer.polyhedron) {
+		CGAL::Bbox_3 cgalbbox = cgalRenderer.polyhedron->bbox();
+		bbox = BoundingBox(
+		  Vector3d(cgalbbox.xmin(), cgalbbox.ymin(), cgalbbox.zmin()),
+		  Vector3d(cgalbbox.xmax(), cgalbbox.ymax(), cgalbbox.zmax())  );
+	}
+	else if (cgalRenderer.polyset) {
+		bbox = cgalRenderer.polyset->getBoundingBox();
+	}
+
+	if (cam.type == Camera::NONE) {
+		cam.type = Camera::VECTOR;
+		cam.center = getBoundingCenter(bbox);
+		double radius = getBoundingRadius(bbox);
+		Vector3d cameradir(1, 1, -0.5);
+		cam.eye = cam.center - radius*2*cameradir;
+	}
+
+	//std::cerr << center << "\n";
+	//std::cerr << radius << "\n";
+
+	glview->setCamera( cam );
+	glview->setRenderer(&cgalRenderer);
+	glview->paintGL();
+	glview->save(output);
+}
+
+#ifdef ENABLE_OPENCSG
+#include "OpenCSGRenderer.h"
+#include <opencsg.h>
+#endif
+
+void export_png_with_opencsg(Tree &tree, Camera &cam, std::ostream &output)
+{
+#ifdef ENABLE_OPENCSG
+  CsgInfo csgInfo = CsgInfo();
+  if ( !csgInfo.compile_chains( tree ) ) {
+		fprintf(stderr,"Couldn't initialize OpenCSG chains\n");
+		return;
+	}
+
+	try {
+		csgInfo.glview = new OffscreenView( cam.pixel_width, cam.pixel_height );
+	} catch (int error) {
+		fprintf(stderr,"Can't create OpenGL OffscreenView. Code: %i.\n", error);
+		return;
+	}
+
+	OpenCSGRenderer opencsgRenderer(csgInfo.root_chain, csgInfo.highlights_chain, csgInfo.background_chain, csgInfo.glview->shaderinfo);
+
+	if (cam.type == Camera::NONE) {
+		cam.type = Camera::VECTOR;
+		double radius = 1.0;
+		if (csgInfo.root_chain) {
+			BoundingBox bbox = csgInfo.root_chain->getBoundingBox();
+			cam.center = (bbox.min() + bbox.max()) / 2;
+			radius = (bbox.max() - bbox.min()).norm() / 2;
+		}
+		Vector3d cameradir(1, 1, -0.5);
+		cam.eye = cam.center - radius*1.8*cameradir;
+	}
+
+	csgInfo.glview->setCamera( cam );
+	csgInfo.glview->setRenderer(&opencsgRenderer);
+	OpenCSG::setContext(0);
+	OpenCSG::setOption(OpenCSG::OffscreenSetting, OpenCSG::FrameBufferObject);
+	csgInfo.glview->paintGL();
+	csgInfo.glview->save(output);
+#else
+	fprintf(stderr,"This openscad was built without OpenCSG support\n");
+#endif
+}
+
+
+#endif // ENABLE_CGAL