writer now works properly

Author: fsalmoir

pygem/igeshandler.py

@@ -2,9 +2,10 @@
 Utilities for reading and writing different CAD files.
 """
 import numpy as np
+import sys
 import pygem.filehandler as fh
 from OCC.IGESControl import (IGESControl_Reader, IGESControl_Writer)
-from OCC.BRep import BRep_Tool
+from OCC.BRep import (BRep_Tool, BRep_Builder)
 from OCC.BRepBuilderAPI import (BRepBuilderAPI_NurbsConvert, BRepBuilderAPI_MakeWire, BRepBuilderAPI_MakeEdge, BRepBuilderAPI_MakeFace)
 from OCC.GeomConvert import geomconvert_SurfaceToBSplineSurface
 import OCC.TopoDS
@@ -94,8 +95,6 @@ def parse(self, filename):
 
 			n_faces += 1
 			faces_explorer.Next()
-			
-		print control_point_position
 
 		self._control_point_position = control_point_position
 
@@ -133,6 +132,10 @@ def write(self, mesh_points, filename):
 		n_faces = 0
 		control_point_position = self._control_point_position
 
+		compound_builder = BRep_Builder()
+		compound = OCC.TopoDS.TopoDS_Compound()
+		compound_builder.MakeCompound(compound)
+		
 		while faces_explorer.More():
 
 			# similar to the parser method
@@ -158,7 +161,7 @@ def write(self, mesh_points, filename):
 					i += 1
 
 			## construct the deformed wire for the trimmed surfaces
-			wireMaker = BRepBuilderAPI_MakeWire()
+			wire_maker = BRepBuilderAPI_MakeWire()
 			tol = ShapeFix_ShapeTolerance()
 			brep = BRepBuilderAPI_MakeFace(occ_face.GetHandle(), 1e-4).Face()
 			brep_face = BRep_Tool.Surface(brep)
@@ -172,23 +175,22 @@ def write(self, mesh_points, filename):
 				# evaluating the new edge: same (u,v) coordinates, but different (x,y,x) ones
 				edge_phis_coordinates_aux = BRepBuilderAPI_MakeEdge(edge_uv_coordinates[0], brep_face)
 				edge_phis_coordinates = edge_phis_coordinates_aux.Edge()
+				#print edge_phis_coordinates
 				tol.SetTolerance(edge_phis_coordinates, 1e-4)
-				wireMaker.Add(edge_phis_coordinates)
+				wire_maker.Add(edge_phis_coordinates)
 				edge_explorer.Next()
 
 			#grouping the edges in a wire
-			wire = wireMaker.Wire()
+			wire = wire_maker.Wire()
 
 			## trimming the surfaces
-			brep_surf = BRepBuilderAPI_MakeFace(occ_face.GetHandle(), wire).Face()
-			writer.AddShape(brep_surf)
-			
-			#print writer
-
+			brep_surf = BRepBuilderAPI_MakeFace(occ_face.GetHandle(), wire).Shape()
+			compound_builder.Add(compound, brep_surf)
 			n_faces += 1
-			faces_explorer.Next()	
+			faces_explorer.Next()
+
+		writer.AddShape(compound)
 
-		## write out the iges file
 		writer.Write(self.outfile)
 
 
@@ -212,15 +214,42 @@ def plot(self, plot_file=None, save_fig=False):
 		reader = IGESControl_Reader()
 		reader.ReadFile(plot_file)
 		reader.TransferRoots()
-		shape = reader.Shape()
+		shape = reader.Shape()		
 
 		display, start_display, add_menu, add_function_to_menu = init_display()
 		display.FitAll()
 		display.DisplayShape(shape, update=True)
 
+		def export_to_BMP(event=None):
+			display.View.Dump('./capture_bmp.bmp')
+
+
+		def export_to_PNG(event=None):
+			display.View.Dump('./capture_png.png')
+
+
+		def export_to_JPEG(event=None):
+			display.View.Dump('./capture_jpeg.jpeg')
+
+
+		def export_to_TIFF(event=None):
+			display.View.Dump('./capture_tiff.tiff')
+
+
+		def exit(event=None):
+			sys.exit()
+		
+		add_menu('screencapture')
+		add_function_to_menu('screencapture', export_to_BMP)
+		add_function_to_menu('screencapture', export_to_PNG)
+		add_function_to_menu('screencapture', export_to_JPEG)
+		add_function_to_menu('screencapture', export_to_TIFF)
+		add_function_to_menu('screencapture', exit)
+		
 		# Show the plot to the screen
 		if not save_fig:
 			start_display()
 		else:
-			display.View.Dump(plot_file.split('.')[0] + '.ppm')
+			f = display.View.View().GetObject()
+			display.View.Dump(plot_file.split('.')[0] + '.jpg')