|
|
|
@ -1,6 +1,12 @@
|
|
|
|
|
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", |
|
|
|
@ -38,6 +44,9 @@ class AHB_ParseStep:
|
|
|
|
|
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: |
|
|
|
@ -119,14 +128,38 @@ class AHB_ParseStep:
|
|
|
|
|
obj.addProperty("App::PropertyString", "Base_OriginalLabel", "Base") |
|
|
|
|
obj.Base_OriginalLabel = obj.Label |
|
|
|
|
|
|
|
|
|
obj.Label = object_info.name |
|
|
|
|
|
|
|
|
|
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 + "_"): |
|
|
|
|
obj.Label = object_info.layer.reference |
|
|
|
|
new_label = object_info.layer.reference + "_" + final_ref |
|
|
|
|
else: |
|
|
|
|
final_ref = object_info.layer.reference |
|
|
|
|
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") |
|
|
|
@ -134,11 +167,20 @@ class AHB_ParseStep:
|
|
|
|
|
|
|
|
|
|
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()) |
|
|
|
|