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_Pnt(1, 1, 1))
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_Solid)
exporter.add_shape(solid)
exporter.write_file()
assert os.path.isfile(filename)
#!/usr/bin/env python
# coding: utf-8
r"""Exporting multiple shapes to IGES"""
import logging
import OCC.Core.BRepPrimAPI
from aocxchange.iges import IgesExporter
from corelib.core.files 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 = OCC.BRepPrimAPI.BRepPrimAPI_MakeBox(50, 50, 50).Shape()
sphere_shape = OCC.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()
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_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_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(IgesUnknownFormatException):
IgesExporter(filename, format_="48.3")
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(IncompatibleFileFormatException):
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(DirectoryNotFoundException):
IgesExporter(filename)
def handle_cad_file_save_as(shapes):
r"""Handle the logic of cad file save as
Parameters
----------
shapes : iterable of OCC.TopoDS.TopoDS_Shape
Returns
-------
tuple
path, type (step, iges ...), extra_info(format, schema or None)
"""
# assert len(shapes) > 0
if len(shapes) == 0:
msg = "shapes list does not contain any shape"
raise ValueError(msg)
with SaveAsCadDialog() as save_as_cad_dialog:
if save_as_cad_dialog.ShowModal() == wx.ID_OK:
path = str(save_as_cad_dialog.GetPath())
extension = extract_file_extension(path)
if extension.lower() in step_extensions:
type_ = "step"
with StepSchemaDialog() as step_schema_dialog:
if step_schema_dialog.ShowModal() == wx.ID_OK:
extra_info = schema = str(step_schema_dialog.selection.GetValue())
exporter = StepExporter(path, schema=schema)
for shape in shapes:
exporter.add_shape(shape)
exporter.write_file()
elif extension.lower() in iges_extensions:
type_ = "iges"
with IgesFormatDialog() as iges_format_dialog:
if iges_format_dialog.ShowModal() == wx.ID_OK:
extra_info = format_ = str(iges_format_dialog.selection.GetValue())
exporter = IgesExporter(path, format_=format_)
for shape in shapes:
exporter.add_shape(shape)
exporter.write_file()
elif extension.lower() in stl_extensions:
type_ = "stl"
with StlFormatDialog() as stl_format_dialog:
if stl_format_dialog.ShowModal() == wx.ID_OK:
extra_info = ascii_mode = True if stl_format_dialog.selection.GetValue() == "ASCII" else False
exporter = StlExporter(path, ascii_mode=ascii_mode)
# TODO : warning message if len(shapes) > 1
exporter.set_shape(shapes[0])
exporter.write_file()
elif extension.lower() in brep_extensions:
type_ = "brep"
extra_info = None
exporter = BrepExporter(path)
# TODO : warning message if len(shapes) > 1
exporter.set_shape(shapes[0])
exporter.write_file()
else:
msg = "File extension indicates a file type " \
"that is not supported"
raise ValueError(msg)
return path, type_, extra_info
else:
return None, None, None