forked from youen/assembly_handbook
Youen
2 years ago
3 changed files with 20 additions and 208 deletions
@ -1,191 +0,0 @@
|
||||
import FreeCADGui as Gui |
||||
import FreeCAD as App |
||||
|
||||
import re |
||||
_object_reference_re = re.compile("^(.*?_)?((CHO|ELE|ROU|TSM|QIN|ACC|DIV|TXT|FRN|T|M|L|R)[0-9]+)(_[0-9]*)?$") |
||||
_object_reference_clean_re = re.compile("^(.*?)(_st[0-9]+)?$") |
||||
_bad_reference_re = re.compile("^[0-9]*$") |
||||
_has_number_re = re.compile("_[0-9]+$") |
||||
|
||||
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': |
||||
if "Base_OriginalLabel" in obj.PropertiesList: |
||||
obj.Label = obj.Base_OriginalLabel |
||||
|
||||
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': 'COMPOUND005', # manivelle droite |
||||
'Batterie': 'COMPOUND058', # batterie basse |
||||
'Batterie_1': 'COMPOUND059', # batterie haute |
||||
'Axa compactline 35-E 6v-12v dc': 'COMPOUND1272', # phare droit |
||||
'Axa compactline 35-E 6v-12v dc_1': 'COMPOUND1273', # phare gauche |
||||
'Potence_CHO44': 'COMPOUND732', # potence |
||||
'CHO04': 'COMPOUND1290', # guidon |
||||
'CHO40': 'COMPOUND1291', # garde-boue droit |
||||
'CHO41': 'COMPOUND1292', # garde-boue gauche |
||||
'Pochette laterale': 'COMPOUND1294', # pochette gauche |
||||
'Pochette laterale_1': 'COMPOUND1293', # 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 |
||||
|
||||
final_ref = self.parse_vhelio_reference(object_info.name) |
||||
if object_info.layer is not None: |
||||
if final_ref is None: |
||||
final_ref = self.parse_vhelio_reference(step_parser.process_name(object_info.layer.reference)[1]) |
||||
|
||||
if final_ref is None: |
||||
final_ref = step_parser.process_name(object_info.layer.reference)[1] |
||||
if object_info.name != final_ref and not object_info.name.startswith(final_ref + "_"): |
||||
final_ref = None |
||||
|
||||
if final_ref is None: |
||||
final_ref = step_parser.process_name(object_info.name)[1] |
||||
|
||||
final_ref = _object_reference_clean_re.search(final_ref).group(1) |
||||
|
||||
if _bad_reference_re.search(final_ref) is not None: |
||||
final_ref = "inconnu" |
||||
if object_info.layer is not None and _bad_reference_re.search(object_info.layer.reference) is None: |
||||
final_ref = object_info.layer.reference |
||||
|
||||
if object_info.name == final_ref or object_info.name.startswith(final_ref + "_"): |
||||
new_label = object_info.layer.reference + "_" + final_ref |
||||
else: |
||||
if _bad_reference_re.search(object_info.name) is None or final_ref == "inconnu": |
||||
new_label = object_info.name |
||||
else: |
||||
new_label = final_ref |
||||
|
||||
if _has_number_re.search(new_label) is None: |
||||
new_label = new_label + "_01" |
||||
|
||||
obj.Label = new_label |
||||
|
||||
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") |
||||
if object_info.layer is None: |
||||
obj.Base_Layer = "" |
||||
else: |
||||
obj.Base_Layer = object_info.layer.name |
||||
|
||||
feature_idx = feature_idx + 1 |
||||
|
||||
print("STEP names import finished") |
||||
|
||||
def parse_vhelio_reference(self, str): |
||||
object_reference_match = _object_reference_re.search(str) |
||||
if object_reference_match is not None: |
||||
return object_reference_match.group(2) |
||||
return None |
||||
|
||||
from ahb_command import AHB_CommandWrapper |
||||
AHB_CommandWrapper.addGuiCommand('AHB_parse_step', AHB_ParseStep()) |
Loading…
Reference in new issue