def test_write_step_file(self) -> None:
"""Exports a colored box into a STEP file"""
### initialisation
doc = TDocStd_Document(TCollection_ExtendedString("pythonocc-doc"))
self.assertTrue(doc is not None)
# Get root assembly
shape_tool = XCAFDoc_DocumentTool.ShapeTool(doc.Main())
colors = XCAFDoc_DocumentTool.ColorTool(doc.Main())
### create the shape to export
test_shape = BRepPrimAPI_MakeBox(100.0, 100.0, 100.0).Shape()
### add shape
shp_label = shape_tool.AddShape(test_shape)
### set a color for this shape
r = 1.0
g = b = 0.5
red_color = Quantity_Color(r, g, b,
Quantity_TypeOfColor.Quantity_TOC_RGB)
colors.SetColor(shp_label, red_color, XCAFDoc_ColorGen)
# write file
WS = XSControl_WorkSession()
writer = STEPCAFControl_Writer(WS, False)
writer.Transfer(doc, STEPControl_AsIs)
status = writer.Write("./test_io/test_ocaf_generated.stp")
self.assertTrue(status)
self.assertTrue(os.path.isfile("./test_io/test_ocaf_generated.stp"))
def test_read_step_file(self) -> None:
"""Reads the previous step file"""
# create an handle to a document
doc = TDocStd_Document(TCollection_ExtendedString("pythonocc-doc"))
# Get root assembly
shape_tool = XCAFDoc_DocumentTool.ShapeTool(doc.Main())
l_colors = XCAFDoc_DocumentTool.ColorTool(doc.Main())
step_reader = STEPCAFControl_Reader()
step_reader.SetColorMode(True)
step_reader.SetLayerMode(True)
step_reader.SetNameMode(True)
step_reader.SetMatMode(True)
status = step_reader.ReadFile("./test_io/test_ocaf.stp")
if status == IFSelect_RetDone:
step_reader.Transfer(doc)
labels = TDF_LabelSequence()
color_labels = TDF_LabelSequence()
shape_tool.GetFreeShapes(labels)
self.assertEqual(labels.Length(), 1)
sub_shapes_labels = TDF_LabelSequence()
self.assertFalse(shape_tool.IsAssembly(labels.Value(1)))
shape_tool.GetSubShapes(labels.Value(1), sub_shapes_labels)
self.assertEqual(sub_shapes_labels.Length(), 0)
l_colors.GetColors(color_labels)
self.assertEqual(color_labels.Length(), 1)
label_shp = labels.Value(1)
a_shape = shape_tool.GetShape(label_shp)
self.assertFalse(a_shape.IsNull())
def __init__(self, filename, layer_name='layer-00'):
logger.info("StepOcafExporter instantiated with "
"filename : %s" % filename)
check_exporter_filename(filename, step_extensions)
check_overwrite(filename)
self.filename = filename
self.h_doc = h_doc = Handle_TDocStd_Document()
# logger.info("Empty Doc?", h_doc.IsNull())
if h_doc.IsNull():
logger.info("Empty Doc?")
# Create the application
app = _XCAFApp.XCAFApp_Application_GetApplication().GetObject()
app.NewDocument(TCollection_ExtendedString("MDTV-CAF"), h_doc)
# Get root assembly
doc = h_doc.GetObject()
h_shape_tool = XCAFDoc_DocumentTool().ShapeTool(doc.Main())
l_colors = XCAFDoc_DocumentTool().ColorTool(doc.Main())
l_layers = XCAFDoc_DocumentTool().LayerTool(doc.Main())
_ = TDF_LabelSequence()
_ = TDF_LabelSequence()
# TopoDS_Shape aShape;
self.shape_tool = h_shape_tool.GetObject()
self.top_label = self.shape_tool.NewShape()
self.colors = l_colors.GetObject()
self.layers = l_layers.GetObject()
self.current_color = Quantity_Color(Quantity_NOC_RED)
self.current_layer = self.layers.AddLayer(
TCollection_ExtendedString(layer_name))
self.layer_names = {}
def read_step(self, fn):
"""
Read and translate a STEP file.
:param str fn: The filename.
:return: The shapes label.
:rtype: afem.exchange.xde.Label.
:raise RuntimeError: If the file cannot be read.
"""
reader = STEPCAFControl_Reader()
reader.SetNameMode(True)
reader.SetColorMode(True)
status = reader.Perform(fn, self._doc)
if not status:
raise RuntimeError("Error reading STEP file.")
self._shape = Shape.wrap(reader.Reader().OneShape())
label = XCAFDoc_DocumentTool.ShapesLabel_(self._doc.Main())
return XdeLabel(label)
def read_file(self):
r"""Read file"""
logger.info("Reading STEP file")
h_doc = Handle_TDocStd_Document()
# Create the application
app = _XCAFApp.XCAFApp_Application_GetApplication().GetObject()
app.NewDocument(TCollection_ExtendedString("MDTV-CAF"), h_doc)
# Get root assembly
doc = h_doc.GetObject()
h_shape_tool = XCAFDoc_DocumentTool().ShapeTool(doc.Main())
color_tool = XCAFDoc_DocumentTool().ColorTool(doc.Main())
layer_tool = XCAFDoc_DocumentTool().LayerTool(doc.Main())
_ = XCAFDoc_DocumentTool().MaterialTool(doc.Main())
step_reader = STEPCAFControl_Reader()
step_reader.SetColorMode(True)
step_reader.SetLayerMode(True)
step_reader.SetNameMode(True)
step_reader.SetMatMode(True)
status = step_reader.ReadFile(str(self.filename))
if status == IFSelect_RetDone:
logger.info("Transfer doc to STEPCAFControl_Reader")
step_reader.Transfer(doc.GetHandle())
labels = TDF_LabelSequence()
_ = TDF_LabelSequence()
# TopoDS_Shape a_shape;
_ = h_shape_tool.GetObject()
h_shape_tool.GetObject().GetFreeShapes(labels)
logger.info('Number of shapes at root :%i' % labels.Length())
# for i in range(labels.Length()):
# a_shape = h_shape_tool.GetObject().GetShape(labels.Value(i+1))
# logger.debug("%i - type : %s" % (i, a_shape.ShapeType()))
# sub_shapes_labels = TDF_LabelSequence()
# print("Is Assembly?", shape_tool.IsAssembly(labels.Value(i + 1)))
# # sub_shapes = shape_tool.getsubshapes(labels.Value(i+1),
# sub_shapes_labels)
#
# sub_shapes = shape_tool.FindSubShape(labels.Value(i + 1),
# a_shape, labels.Value(i + 1))
# print('Number of subshapes in the assembly : %i' %
# sub_shapes_labels.Length())
#
# color_tool.GetObject().GetColors(color_labels)
# logger.info('Number of colors : %i' % color_labels.Length())
for i in range(labels.Length()):
# print i
label = labels.Value(i + 1)
logger.debug("Label : %s" % label)
a_shape = h_shape_tool.GetObject().GetShape(labels.Value(i + 1))
# string_seq = TColStd_HSequenceOfExtendedString()
# string_seq is an TColStd_HSequenceOfExtendedString
string_seq = layer_tool.GetObject().GetLayers(a_shape)
color = Quantity_Color()
_ = color_tool.GetObject().GetColor(a_shape, XCAFDoc_ColorSurf,
color)
logger.info("The shape type is : %i" % a_shape.ShapeType())
if a_shape.ShapeType() == TopAbs_COMPOUND:
logger.info("The shape type is TopAbs_COMPOUND")
topo = Topo(a_shape)
logger.info("Nb of compounds : %i" % topo.number_of_compounds)
logger.info("Nb of solids : %i" % topo.number_of_solids)
logger.info("Nb of shells : %i" % topo.number_of_shells)
for solid in topo.solids:
logger.info("Adding solid to the shapes list")
self._shapes.append(solid)
elif a_shape.ShapeType() == TopAbs_SOLID:
logger.info("The shape type is TopAbs_SOLID")
self._shapes.append(a_shape)
self._colors.append(color)
self._layers.append(string_seq)
return True
def shapes_label(self):
"""
:return: The shapes label of the document.
:rtype: afem.exchange.xde.XdeLabel
"""
return XdeLabel(XCAFDoc_DocumentTool.ShapesLabel_(self._doc.Main()))
def _init_tool(self):
self._tool = XCAFDoc_DocumentTool.ShapeTool(self._doc.Main())