def test_iges_exporter_happy_path(box_shape): r"""Happy path""" filename = path_from_file(__file__, "./models_out/box.IgS") exporter = IgesExporter(filename) exporter.add_shape(box_shape) exporter.write_file() assert os.path.isfile(filename)
def test_iges_exporter_happy_path_shape_subclass(box_shape): r"""Happy path with a subclass of TopoDS_Shape""" filename = path_from_file(__file__, "./models_out/box.igs") exporter = IgesExporter(filename) solid = shape_to_topology(box_shape) assert isinstance(solid, TopoDS.TopoDS_Solid) exporter.add_shape(solid) exporter.write_file() assert os.path.isfile(filename)
def shape_to_file(shape, pth, filename, format='iges'): """write a Shape to a .iges .brep .stl or .step file""" from OCCDataExchange.brep import BrepExporter from OCCDataExchange.iges import IgesExporter from OCCDataExchange.step import StepExporter from OCCDataExchange.stl import StlExporter _pth = os.path.join(pth, filename) assert not os.path.isdir(_pth), 'wrong path, filename' _file = str("%s.%s" % (_pth, format)) _formats = ['iges', 'igs', 'step', 'stp', 'brep', 'stl'] assert format in _formats, '%s is not a readable format, should be one of %s ' % (format, _formats) if format in ['iges', 'igs']: writer = IgesExporter(_file) writer.add_shape(shape) writer.write_file() return _file elif format in ['step', 'stp']: writer = StepExporter(_file) writer.add_shape(shape) writer.write_file() return _file elif format == 'brep': writer = BrepExporter(_file) writer.set_shape(shape) writer.write_file() return _file elif format == 'stl': writer = StlExporter(_file) writer.set_shape(shape) writer.write_file() return _file else: raise ValueError('format should be one of [iges,igs], [step,stp], brep, stl\ngot %s' % (format))
def test_iges_exporter_overwrite(box_shape): r"""Happy path with a subclass of TopoDS_Shape""" filename = path_from_file(__file__, "./models_out/box.igs") exporter = IgesExporter(filename) solid = shape_to_topology(box_shape) assert isinstance(solid, TopoDS.TopoDS_Solid) exporter.add_shape(solid) exporter.write_file() assert os.path.isfile(filename) # read the written box.igs importer = IgesImporter(filename) topo_compound = Topo(importer.compound) assert topo_compound.number_of_faces() == 6 assert topo_compound.number_of_edges() == 24 # add a sphere and write again with same exporter sphere = BRepPrimAPI.BRepPrimAPI_MakeSphere(10) exporter.add_shape(sphere.Shape()) exporter.write_file() # this creates a file with a box and a sphere # check that the file contains the box and the sphere importer = IgesImporter(filename) topo_compound = Topo(importer.compound) assert topo_compound.number_of_faces() == 7 # 6 from box + 1 from sphere # create a new exporter and overwrite with a box only filename = path_from_file(__file__, "./models_out/box.igs") exporter = IgesExporter(filename) solid = shape_to_topology(box_shape) exporter.add_shape(solid) exporter.write_file() assert os.path.isfile(filename) # check the file only contains a box importer = IgesImporter(filename) topo_compound = Topo(importer.compound) assert topo_compound.number_of_faces() == 6 # 6 from box
def test_iges_exporter_wrong_format(box_shape): r"""Format is not 5.1 or 5.3""" filename = path_from_file(__file__, "./models_out/box.igs") with pytest.raises(ValueError): IgesExporter(filename, format="48.3")
def test_iges_exporter_adding_not_a_shape(box_shape): r"""Adding something to the exporter that is not a TopoDS_Shape or a subclass""" filename = path_from_file(__file__, "./models_out/box.igs") exporter = IgesExporter(filename) with pytest.raises(ValueError): exporter.add_shape(gp.gp_Pnt(1, 1, 1))
def test_iges_exporter_wrong_extension(box_shape): r"""Trying to write a step file with the IgesExporter""" filename = path_from_file(__file__, "./models_out/box.step") with pytest.raises(AssertionError): IgesExporter(filename)
def test_iges_exporter_wrong_filename(box_shape): r"""Trying to write to a non-existent directory""" filename = path_from_file(__file__, "./nonexistent/box.igs") with pytest.raises(AssertionError): IgesExporter(filename)
def shape_to_file(shape, pth, filename, format='iges'): """write a Shape to a .iges .brep .stl or .step file""" from OCCDataExchange.brep import BrepExporter from OCCDataExchange.iges import IgesExporter from OCCDataExchange.step import StepExporter from OCCDataExchange.stl import StlExporter _pth = os.path.join(pth, filename) assert not os.path.isdir(_pth), 'wrong path, filename' _file = str("%s.%s" % (_pth, format)) _formats = ['iges', 'igs', 'step', 'stp', 'brep', 'stl'] assert format in _formats, '%s is not a readable format, should be one of %s ' % ( format, _formats) if format in ['iges', 'igs']: writer = IgesExporter(_file) writer.add_shape(shape) writer.write_file() return _file elif format in ['step', 'stp']: writer = StepExporter(_file) writer.add_shape(shape) writer.write_file() return _file elif format == 'brep': writer = BrepExporter(_file) writer.set_shape(shape) writer.write_file() return _file elif format == 'stl': writer = StlExporter(_file) writer.set_shape(shape) writer.write_file() return _file else: raise ValueError( 'format should be one of [iges,igs], [step,stp], brep, stl\ngot %s' % (format))
#!/usr/bin/env python # coding: utf-8 r"""Exporting multiple shapes to IGES""" import logging from OCC import BRepPrimAPI from OCCDataExchange.iges import IgesExporter from OCCDataExchange import path_from_file logging.basicConfig(level=logging.DEBUG, format='%(asctime)s :: %(levelname)6s :: %(module)20s :: %(lineno)3d :: %(message)s') # First create a simple shape to export box_shape = BRepPrimAPI.BRepPrimAPI_MakeBox(50, 50, 50).Shape() sphere_shape = BRepPrimAPI.BRepPrimAPI_MakeSphere(20).Shape() # Export to IGES filename = path_from_file(__file__, "./models_out/result_export_multi.iges") my_iges_exporter = IgesExporter(filename, format="5.3") my_iges_exporter.add_shape(box_shape) my_iges_exporter.add_shape(sphere_shape) my_iges_exporter.write_file()
#!/usr/bin/env python # coding: utf-8 r"""Exporting multiple shapes to IGES""" import logging from OCC import BRepPrimAPI from OCCDataExchange.iges import IgesExporter from OCCDataExchange.utils import path_from_file logging.basicConfig( level=logging.DEBUG, format= '%(asctime)s :: %(levelname)6s :: %(module)20s :: %(lineno)3d :: %(message)s' ) # First create a simple shape to export box_shape = BRepPrimAPI.BRepPrimAPI_MakeBox(50, 50, 50).Shape() sphere_shape = BRepPrimAPI.BRepPrimAPI_MakeSphere(20).Shape() # Export to IGES filename = path_from_file(__file__, "./models_out/result_export_multi.iges") my_iges_exporter = IgesExporter(filename, format="5.3") my_iges_exporter.add_shape(box_shape) my_iges_exporter.add_shape(sphere_shape) my_iges_exporter.write_file()