assembly_handbook/ahb_cmd_parse_step.py

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())