def exportCAF(assy: AssemblyProtocol, path: str) -> bool:
"""
Export an assembly to a OCAF xml file (internal OCCT format).
"""
folder, fname = os.path.split(path)
name, ext = os.path.splitext(fname)
ext = ext[1:] if ext[0] == "." else ext
_, doc = toCAF(assy)
app = XCAFApp_Application.GetApplication_s()
store = XmlDrivers_DocumentStorageDriver(
TCollection_ExtendedString("Copyright: Open Cascade, 2001-2002"))
ret = XmlDrivers_DocumentRetrievalDriver()
app.DefineFormat(
TCollection_AsciiString("XmlOcaf"),
TCollection_AsciiString("Xml XCAF Document"),
TCollection_AsciiString(ext),
ret,
store,
)
doc.SetRequestedFolder(TCollection_ExtendedString(folder))
doc.SetRequestedName(TCollection_ExtendedString(name))
status = app.SaveAs(doc, TCollection_ExtendedString(path))
app.Close(doc)
return status == PCDM_StoreStatus.PCDM_SS_OK
def toCAF(assy: AssemblyProtocol,
coloredSTEP: bool = False) -> Tuple[TDF_Label, TDocStd_Document]:
# prepare a doc
app = XCAFApp_Application.GetApplication_s()
doc = TDocStd_Document(TCollection_ExtendedString("XmlOcaf"))
app.InitDocument(doc)
tool = XCAFDoc_DocumentTool.ShapeTool_s(doc.Main())
tool.SetAutoNaming_s(False)
ctool = XCAFDoc_DocumentTool.ColorTool_s(doc.Main())
# add root
top = tool.NewShape()
TDataStd_Name.Set_s(top, TCollection_ExtendedString("CQ assembly"))
# add leafs and subassemblies
subassys: Dict[str, Tuple[TDF_Label, Location]] = {}
for k, v in assy.traverse():
# leaf part
lab = tool.NewShape()
tool.SetShape(lab, Compound.makeCompound(v.shapes).wrapped)
setName(lab, f"{k}_part", tool)
# assy part
subassy = tool.NewShape()
tool.AddComponent(subassy, lab, TopLoc_Location())
setName(subassy, k, tool)
# handle colors - this logic is needed for proper STEP export
color = v.color
tmp = v
if coloredSTEP:
while not color and tmp.parent:
tmp = tmp.parent
color = tmp.color
if color:
setColor(lab, color, ctool)
else:
if color:
setColor(subassy, color, ctool)
subassys[k] = (subassy, v.loc)
for ch in v.children:
tool.AddComponent(subassy, subassys[ch.name][0],
subassys[ch.name][1].wrapped)
tool.AddComponent(top, subassys[assy.name][0], assy.loc.wrapped)
tool.UpdateAssemblies()
return top, doc
def load(self, filename):
if not os.path.exists(filename):
raise FileNotFoundError(filename)
print("Reading STEP file ...", flush=True, end="")
time.sleep(0.01) # ensure output is shown
fmt = TCollection_ExtendedString("CadQuery-XCAF")
doc = TDocStd_Document(fmt)
self.shape_tool = XCAFDoc_DocumentTool.ShapeTool_s(doc.Main())
self.color_tool = XCAFDoc_DocumentTool.ColorTool_s(doc.Main())
reader = STEPCAFControl_Reader()
reader.SetNameMode(True)
reader.SetColorMode(True)
reader.SetLayerMode(True)
reader.ReadFile(filename)
reader.Transfer(doc)
print(" done")
time.sleep(0.01) # ensure output is shown
root_labels = TDF_LabelSequence()
self.shape_tool.GetFreeShapes(root_labels)
result = []
for i in range(root_labels.Length()):
label = root_labels.Value(i + 1)
if self.shape_tool.IsAssembly_s(label):
result.append({
"name": self.get_name(label),
"shape": self.get_shape(label),
"shapes": self.get_subshapes(label),
"color": self.get_color(label),
"loc": self.get_location(label),
})
else:
raise RuntimeError("ERROR: Shapes are not supported")
self.assembly = result
def setName(l: TDF_Label, name: str, tool):
TDataStd_Name.Set_s(l, TCollection_ExtendedString(name))