Browse Source

tests (wip)

pull/1/head
Youen Toupin 3 years ago
parent
commit
5170022144
  1. 3
      InitGui.py
  2. 13
      ahb_cmd_export_csv.py
  3. 96
      ahb_cmd_parse_step.py

3
InitGui.py

@ -69,6 +69,9 @@ class AssemblyHandbookWorkbench(Gui.Workbench):
toolbox.append("AHB_animate") toolbox.append("AHB_animate")
if self.dev: if self.dev:
self.importModule('ahb_cmd_parse_step')
toolbox.append("AHB_parse_step")
self.importModule('ahb_cmd_reload') self.importModule('ahb_cmd_reload')
toolbox.append("AHB_reload") toolbox.append("AHB_reload")

13
ahb_cmd_export_csv.py

@ -34,12 +34,15 @@ class AHB_ExportCsv:
obj: App.DocumentObject obj: App.DocumentObject
for obj in doc.Objects: for obj in doc.Objects:
if obj.TypeId == 'Part::Feature': if obj.TypeId == 'Part::Feature':
parts.append((readProp(obj, 'Base_Layer', '').replace(',', '-') + "," + readProp(obj, 'Base_Reference', '').replace(',', '-'), "=\"" + obj.Label.replace(',', '-') + "\"")) parts.append((readProp(obj, 'Base_Layer', '').replace(',', '-') + "," + readProp(obj, 'Base_Reference', '').replace(',', '-'), obj.Label.replace(',', '-')))
parts.sort() parts.sort()
export_layer_name = False
with open(csvFileName, 'w', encoding='utf8') as csvFile: with open(csvFileName, 'w', encoding='utf8') as csvFile:
csvFile.write("Group, Reference, Name, Quantity\n") if export_layer_name:
csvFile.write("Group,")
csvFile.write("Reference, First object name, Quantity\n")
quantity = 0 quantity = 0
prevPart = parts[0] prevPart = parts[0]
@ -48,7 +51,11 @@ class AHB_ExportCsv:
if part[0] == prevPart[0]: if part[0] == prevPart[0]:
quantity += 1 quantity += 1
else: else:
csvFile.write(prevPart[0] + "," + prevPart[1] + "," + str(quantity) + "\n") if export_layer_name:
csvFile.write(prevPart[0] + ",")
else:
csvFile.write(prevPart[0].split(',')[1] + ",")
csvFile.write(prevPart[1] + "," + str(quantity) + "\n")
quantity = 1 quantity = 1
prevPart = part prevPart = part

96
ahb_cmd_parse_step.py

@ -0,0 +1,96 @@
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:
for idx in range(len(step_info.objects)):
if step_info.objects[idx].name == obj.Label:
object_idx = idx
break
if 0 <= object_idx < len(step_info.objects):
object_info = step_info.objects[object_idx]
print(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
for obj in objects:
if obj.TypeId == 'Part::Feature':
#print(obj.Label)
if obj.Label.startswith("COMPOUND"):
feature_from_end = total_features - feature_idx
object_idx = len(step_info2.objects) - feature_from_end
#print(obj.Label + " object_idx=" + str(object_idx))
if 0 <= object_idx < len(step_info2.objects):
object_info = step_info2.objects[object_idx]
print(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
print("STEP names import finished")
from ahb_command import AHB_CommandWrapper
AHB_CommandWrapper.addGuiCommand('AHB_parse_step', AHB_ParseStep())
Loading…
Cancel
Save