def addComponent(self, shape, name, color):
"""Add new shape to top assembly of self.doc & return uid"""
labels = TDF_LabelSequence()
shape_tool = XCAFDoc_DocumentTool_ShapeTool(self.doc.Main())
color_tool = XCAFDoc_DocumentTool_ColorTool(self.doc.Main())
shape_tool.GetShapes(labels)
try:
rootLabel = labels.Value(1) # First label at root
except RuntimeError as e:
print(e)
return
newLabel = shape_tool.AddComponent(rootLabel, shape, True)
# Get referrred label and apply color to it
refLabel = TDF_Label() # label of referred shape
isRef = shape_tool.GetReferredShape(newLabel, refLabel)
if isRef:
color_tool.SetColor(refLabel, color, XCAFDoc_ColorGen)
self.setLabelName(newLabel, name)
logger.info('Part %s added to root label', name)
shape_tool.UpdateAssemblies()
self.doc = self.doc_linter() # This gets color to work
self.parse_doc()
entry = newLabel.EntryDumpToString()
uid = entry + '.0' # this should work OK since it is new
return uid
class StepAnalyzer():
"""A class that analyzes the structure of an OCAF document."""
def __init__(self, document=None, filename=None):
"""Supply one or the other: document or STEP filename."""
self.uid = 1
self.indent = 0
self.output = ""
self.fname = filename
if filename:
self.doc = self.read_file(filename)
elif document:
self.doc = document
self.shape_tool = XCAFDoc_DocumentTool_ShapeTool(self.doc.Main())
else:
print("Supply one or the other: document or STEP filename.")
def read_file(self, fname):
"""Read STEP file and return <TDocStd_Document>."""
# Create the application, empty document and shape_tool
doc = TDocStd_Document(TCollection_ExtendedString("STEP"))
app = XCAFApp_Application_GetApplication()
app.NewDocument(TCollection_ExtendedString("MDTV-XCAF"), doc)
self.shape_tool = XCAFDoc_DocumentTool_ShapeTool(doc.Main())
self.shape_tool.SetAutoNaming(True)
# Read file and return populated doc
step_reader = STEPCAFControl_Reader()
step_reader.SetColorMode(True)
step_reader.SetLayerMode(True)
step_reader.SetNameMode(True)
step_reader.SetMatMode(True)
status = step_reader.ReadFile(fname)
if status == IFSelect_RetDone:
step_reader.Transfer(doc)
return doc
def dump(self):
"""Return assembly structure in indented outline form.
Format of lines:
Component Name [entry] => Referred Label Name [entry]
Components are shown indented w/r/t line above."""
if self.fname:
self.output += f"Assembly structure of file: {self.fname}\n\n"
else:
self.output += "Assembly structure of doc:\n\n"
self.indent = 0
# Find root label of step doc
labels = TDF_LabelSequence()
self.shape_tool.GetShapes(labels)
nbr = labels.Length()
rootlabel = labels.Value(1) # First label at root
# Get information from root label
name = rootlabel.GetLabelName()
entry = rootlabel.EntryDumpToString()
is_assy = self.shape_tool.IsAssembly(rootlabel)
if is_assy:
# If 1st label at root holds an assembly, it is the Top Assy.
# Through this label, the entire assembly is accessible.
# There is no need to explicitly examine other labels at root.
self.output += f"{self.uid}\t[{entry}] {name}\t"
self.uid += 1
self.indent += 2
top_comps = TDF_LabelSequence() # Components of Top Assy
subchilds = False
is_assy = self.shape_tool.GetComponents(rootlabel, top_comps,
subchilds)
self.output += f"Number of labels at root = {nbr}\n"
if top_comps.Length():
self.find_components(top_comps)
return self.output
def find_components(self, comps):
"""Discover components from comps (LabelSequence) of an assembly.
Components of an assembly are, by definition, references which refer
to either a shape or another assembly. Components are essentially
'instances' of the referred shape or assembly, and carry a location
vector specifing the location of the referred shape or assembly.
"""
for j in range(comps.Length()):
c_label = comps.Value(j + 1) # component label <class 'TDF_Label'>
c_name = c_label.GetLabelName()
c_entry = c_label.EntryDumpToString()
ref_label = TDF_Label() # label of referred shape (or assembly)
is_ref = self.shape_tool.GetReferredShape(c_label, ref_label)
if is_ref: # just in case all components are not references
ref_entry = ref_label.EntryDumpToString()
ref_name = ref_label.GetLabelName()
indent = "\t" * self.indent
self.output += f"{self.uid}{indent}[{c_entry}] {c_name}"
self.output += f" => [{ref_entry}] {ref_name}\n"
self.uid += 1
if self.shape_tool.IsAssembly(ref_label):
self.indent += 1
ref_comps = TDF_LabelSequence() # Components of Assy
subchilds = False
_ = self.shape_tool.GetComponents(ref_label, ref_comps,
subchilds)
if ref_comps.Length():
self.find_components(ref_comps)
self.indent -= 1