def test_stl_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.stl")
exporter = StlExporter(filename)
with pytest.raises(ValueError):
exporter.set_shape(gp_Pnt(1, 1, 1))
def _generate_cad(output_folder, py_geometry_file, output_format):
if output_format not in ["step", "stl", "html"]:
raise ValueError
spec = importlib.util.spec_from_file_location(py_geometry_file, py_geometry_file)
py_geometry_module = importlib.util.module_from_spec(spec)
spec.loader.exec_module(py_geometry_module)
# py_geometry_module = imp.load_source(py_geometry_file, py_geometry_file)
shape = py_geometry_module.__shape__
part_id = splitext(basename(py_geometry_file))[0]
part_id = str(part_id) # Keeps the OCC STEP Writer happy !
if output_format == "step":
# part.to_step(join(output_folder, "%s.stp" % part_id))
exporter = StepExporter(filename=join(output_folder, "%s.stp" % part_id))
exporter.add_shape(shape)
exporter.write_file()
elif output_format == "stl":
# part.to_stl(join(output_folder, "%s.stl" % part_id))
exporter = StlExporter(filename=join(output_folder, "%s.stl" % part_id))
exporter.set_shape(shape)
exporter.write_file()
# elif output_format == "html":
# part.to_html(join(output_folder, "%s.html" % part_id))
else:
msg = "Unknown export format"
logger.error(msg)
raise ValueError(msg)
def test_stl_exporter_happy_path(box_shape):
r"""Happy path"""
filename = path_from_file(__file__, "./models_out/box.sTl")
exporter = StlExporter(filename)
exporter.set_shape(box_shape)
exporter.write_file()
assert os.path.isfile(filename)
def test_stl_exporter_happy_path_shape_subclass(box_shape):
r"""Happy path with a subclass of TopoDS_Shape"""
filename = path_from_file(__file__, "./models_out/box.stl")
exporter = StlExporter(filename)
solid = shape_to_topology(box_shape)
assert isinstance(solid, TopoDS_Solid)
exporter.set_shape(solid)
exporter.write_file()
assert os.path.isfile(filename)
def shape_to_stl(shape_, stl_file_, scale, ascii_mode_, factor_,
use_min_dim_):
r"""Write a single shape to an STL file
Parameters
----------
shape_
stl_file_
ascii_mode_
factor_
use_min_dim_
"""
exporter = StlExporter(filename=stl_file_, ascii_mode=ascii_mode_)
shape_ = scale_uniform(shape_, (0, 0, 0), scale, False)
# Must mesh ! Otherwise the exporter does not write anything!
mesh(shape_, factor=factor_, use_min_dim=use_min_dim_)
exporter.set_shape(shape_)
exporter.write_file()
def test_stl_exporter_overwrite(box_shape):
r"""Happy path with a subclass of TopoDS_Shape"""
filename = path_from_file(__file__, "./models_out/box.stl")
exporter = StlExporter(filename)
solid = shape_to_topology(box_shape)
assert isinstance(solid, TopoDS_Solid)
exporter.set_shape(solid)
exporter.write_file()
assert os.path.isfile(filename)
# read the written box.stl
importer = StlImporter(filename)
topo = Topo(importer.shape)
assert topo.number_of_shells == 1
# set a sphere and write again with same exporter
sphere = BRepPrimAPI_MakeSphere(10)
exporter.set_shape(sphere.Shape())
# this creates a file with a sphere only, this is STL specific
exporter.write_file()
# check that the file contains the sphere only
importer = StlImporter(filename)
topo = Topo(importer.shape)
assert topo.number_of_shells == 1
# create a new exporter and overwrite with a box only
filename = path_from_file(__file__, "./models_out/box.stl")
exporter = StlExporter(filename)
solid = shape_to_topology(box_shape)
exporter.set_shape(solid)
exporter.write_file()
assert os.path.isfile(filename)
# check the file only contains a box
importer = StlImporter(filename)
topo = Topo(importer.shape)
assert topo.number_of_shells == 1
def test_stl_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):
StlExporter(filename)
def test_stl_exporter_wrong_filename(box_shape):
r"""Trying to write to a non-existent directory"""
filename = path_from_file(__file__, "./nonexistent/box.stl")
with pytest.raises(DirectoryNotFoundException):
StlExporter(filename)
#!/usr/bin/env python
# coding: utf-8
r"""Exporting a shape to STL"""
from __future__ import print_function
import logging
from OCC.Core.BRepPrimAPI import BRepPrimAPI_MakeBox
from aocxchange.stl import StlExporter
# from corelib.core.files import path_from_file
from corelibpy 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
my_box_shape = BRepPrimAPI_MakeBox(50, 50, 50).Shape()
# Export to STL. If ASCIIMode is set to False, then binary format is used.
filename = path_from_file(__file__, "./models_out/result_export.stl")
my_stl_exporter = StlExporter(filename, ascii_mode=True)
my_stl_exporter.set_shape(my_box_shape)
my_stl_exporter.write_file()
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
