FreeCAD workbench to create assembly handbooks
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

144 lines
5.9 KiB

import FreeCADGui as Gui
import FreeCAD as App
class AHB_ParseStep:
def GetResources(self):
return {"MenuText": "[dev] parse STEP",
"ToolTip": "Reload informations from the original STEP file",
"Pixmap": ""
}
def IsActive(self):
return True
def Activated(self):
import step_parser
import os
import importlib
importlib.reload(step_parser)
print("Reloading STEP file metadata...")
doc = App.activeDocument()
filename: str = doc.FileName
if filename is None:
raise BaseException("You must save your FreeCAD document before parsing the corresponding STEP file")
filename = os.path.splitext(filename)[0] + ".step"
step_info = step_parser.parse(filename)
objects = doc.Objects
override_names = False
feature_idx = 0
total_features = 0
obj: App.DocumentObject
for obj in objects:
if obj.TypeId == 'Part::Feature':
total_features = total_features + 1
object_idx = -1
if override_names:
object_idx = feature_idx + 1
else:
fixed_name = step_parser.process_name(obj.Label)[0]
for idx in range(len(step_info.objects)):
if step_info.objects[idx].name == fixed_name:
object_idx = idx
break
if 0 <= object_idx < len(step_info.objects):
object_info = step_info.objects[object_idx]
print("phase 1 " + obj.Label + " => " + object_info.name)
obj.Label = object_info.name
if object_info.layer is not None:
if "Base_Reference" not in obj.PropertiesList:
obj.addProperty("App::PropertyString", "Base_Reference", "Base")
obj.Base_Reference = object_info.layer.reference
if "Base_Layer" not in obj.PropertiesList:
obj.addProperty("App::PropertyString", "Base_Layer", "Base")
obj.Base_Layer = object_info.layer.name
feature_idx = feature_idx + 1
step_info2 = step_parser.parse(filename, 'SOLIDS')
print("STEP names phase 2 (" + str(len(step_info2.objects)) + ")")
feature_idx = 0
# don't know why, but these objects are not ordered like the others
special_objects = {
'10_01_003 F_st7758_1': 'COMPOUND004', # manivelle droite
'CHO04': 'COMPOUND1257', # guidon
'CHO40': 'COMPOUND1258', # garde-boue droit
'CHO41': 'COMPOUND1259', # garde-boue gauche
'Pochette laterale': 'COMPOUND1260', # pochette gauche
'Pochette laterale_1': 'COMPOUND1261', # pochette droite
}
object_idx = -1
for obj in objects:
if obj.TypeId == 'Part::Feature':
if object_idx >= 0:
object_idx = object_idx + 1
explicit_object_idx = -1
for s in special_objects.items():
if obj.Label == s[1]:
for idx, info in enumerate(step_info2.objects):
if info.name == s[0]:
explicit_object_idx = idx
if explicit_object_idx >= 0:
object_idx = explicit_object_idx
else:
while True:
if object_idx < 0 or object_idx >= len(step_info2.objects):
break
object_info = step_info2.objects[object_idx]
skip = False
for s in special_objects.items():
if s[0] == object_info.name:
skip = True
if skip:
object_idx = object_idx + 1
else:
break
if 0 <= object_idx < len(step_info2.objects):
object_info = step_info2.objects[object_idx]
if obj.Label.startswith("COMPOUND"):
print("phase 2 " + obj.Label + " => " + object_info.name + " (ref="+object_info.layer.reference+", layer="+object_info.layer.name+")")
if "Base_OriginalLabel" not in obj.PropertiesList:
obj.addProperty("App::PropertyString", "Base_OriginalLabel", "Base")
obj.Base_OriginalLabel = obj.Label
obj.Label = object_info.name
if object_info.layer is not None:
final_ref = step_parser.process_name(object_info.layer.reference)[1]
if object_info.name == final_ref or object_info.name.startswith(final_ref + "_"):
obj.Label = object_info.layer.reference
else:
final_ref = object_info.layer.reference
if "Base_Reference" not in obj.PropertiesList:
obj.addProperty("App::PropertyString", "Base_Reference", "Base")
obj.Base_Reference = final_ref
if "Base_Layer" not in obj.PropertiesList:
obj.addProperty("App::PropertyString", "Base_Layer", "Base")
obj.Base_Layer = object_info.layer.name
feature_idx = feature_idx + 1
print("STEP names import finished")
from ahb_command import AHB_CommandWrapper
AHB_CommandWrapper.addGuiCommand('AHB_parse_step', AHB_ParseStep())