Merge pull request #37 from bollwyvl/repr-png

etjones · etjones · commit f8dac5fdca9d · 2015-08-11T17:45:25.000-07:00
adding _repr_png_ for ipython
diff --git a/solid/solidpython.py b/solid/solidpython.py
@@ -11,6 +11,8 @@
 
 import os, sys, re
 import inspect
+import subprocess
+import tempfile
 
 openscad_builtins = [
     # 2D primitives
@@ -602,6 +604,34 @@ def __mul__(self, x):
         '''
         return intersection()(self, x)
 
+    def _repr_png_(self):
+        '''
+        Allow rich clients such as the IPython Notebook, to display the current
+        OpenSCAD rendering of this object.
+        '''
+        png_data = None
+        tmp = tempfile.NamedTemporaryFile(suffix=".scad", delete=False)
+        tmp_png = tempfile.NamedTemporaryFile(suffix=".png", delete=False)
+        try:
+            scad_text = scad_render(self).encode("utf-8")
+            tmp.write(scad_text)
+            tmp.close()
+            tmp_png.close()
+            subprocess.Popen([
+                "openscad",
+                "--preview",
+                "-o", tmp_png.name,
+                tmp.name
+            ]).communicate()
+
+            with open(tmp_png.name, "rb") as png:
+                png_data = png.read()
+        finally:
+            os.unlink(tmp.name)
+            os.unlink(tmp_png.name)
+
+        return png_data
+
 
 class IncludedOpenSCADObject(OpenSCADObject):
 
