301 lines
10 KiB
Python
301 lines
10 KiB
Python
import json
|
|
import os
|
|
import sys
|
|
|
|
import FreeCAD as App
|
|
import FreeCADGui as Gui
|
|
import Part
|
|
import Sketcher
|
|
from PySide import QtWidgets
|
|
|
|
|
|
MARKER = "FREECAD_GUI_WORKFLOW_RESULT="
|
|
|
|
|
|
def progress(phase):
|
|
print("FREECAD_GUI_WORKFLOW_PROGRESS=" + phase, file=sys.stderr, flush=True)
|
|
|
|
|
|
def emit(value, status=0):
|
|
result_file = os.environ.get("FREECAD_GUI_WORKFLOW_RESULT_FILE", "")
|
|
if result_file:
|
|
parent = os.path.dirname(result_file)
|
|
if parent:
|
|
os.makedirs(parent, exist_ok=True)
|
|
with open(result_file, "w", encoding="utf-8") as stream:
|
|
json.dump(value, stream, sort_keys=True, separators=(",", ":"))
|
|
stream.write("\n")
|
|
os._exit(status)
|
|
payload = (MARKER + json.dumps(value, sort_keys=True, separators=(",", ":")) + "\n").encode("utf-8")
|
|
offset = 0
|
|
while offset < len(payload):
|
|
offset += os.write(sys.stdout.fileno(), payload[offset:])
|
|
os._exit(status)
|
|
|
|
|
|
def flush_gui():
|
|
Gui.updateGui()
|
|
QtWidgets.QApplication.processEvents()
|
|
|
|
|
|
def active_edit_name():
|
|
gui_document = Gui.activeDocument()
|
|
in_edit = gui_document.getInEdit() if gui_document else None
|
|
obj = getattr(in_edit, "Object", None) if in_edit else None
|
|
return str(getattr(obj, "Name", ""))
|
|
|
|
|
|
def task_panel_state(phase):
|
|
main_window = Gui.getMainWindow()
|
|
focus = QtWidgets.QApplication.focusWidget()
|
|
buttons = []
|
|
for box in main_window.findChildren(QtWidgets.QDialogButtonBox):
|
|
if not box.isVisible():
|
|
continue
|
|
for button in box.buttons():
|
|
buttons.append({
|
|
"text": str(button.text()).replace("&", ""),
|
|
"objectName": str(button.objectName()),
|
|
"enabled": bool(button.isEnabled()),
|
|
"default": bool(button.isDefault()),
|
|
"role": str(box.buttonRole(button)),
|
|
})
|
|
known_button_ids = {id(button) for box in main_window.findChildren(QtWidgets.QDialogButtonBox) for button in box.buttons()}
|
|
for button in main_window.findChildren(QtWidgets.QPushButton):
|
|
if id(button) in known_button_ids or not button.isVisible():
|
|
continue
|
|
buttons.append({
|
|
"text": str(button.text()).replace("&", ""),
|
|
"objectName": str(button.objectName()),
|
|
"enabled": bool(button.isEnabled()),
|
|
"default": bool(button.isDefault()),
|
|
"role": "direct-push-button",
|
|
})
|
|
fields = []
|
|
field_types = (
|
|
QtWidgets.QLineEdit,
|
|
QtWidgets.QComboBox,
|
|
QtWidgets.QSpinBox,
|
|
QtWidgets.QDoubleSpinBox,
|
|
QtWidgets.QCheckBox,
|
|
)
|
|
seen = set()
|
|
for field_type in field_types:
|
|
for widget in main_window.findChildren(field_type):
|
|
identity = id(widget)
|
|
if identity in seen or not widget.isVisible():
|
|
continue
|
|
seen.add(identity)
|
|
fields.append({
|
|
"className": str(widget.metaObject().className()),
|
|
"objectName": str(widget.objectName()),
|
|
"enabled": bool(widget.isEnabled()),
|
|
"focus": widget is focus,
|
|
})
|
|
return {
|
|
"phase": phase,
|
|
"activeDialog": bool(Gui.Control.activeDialog()),
|
|
"inEdit": active_edit_name(),
|
|
"focus": None if focus is None else {
|
|
"className": str(focus.metaObject().className()),
|
|
"objectName": str(focus.objectName()),
|
|
},
|
|
"buttons": buttons,
|
|
"fieldCount": len(fields),
|
|
"fieldClasses": sorted({field["className"] for field in fields}),
|
|
}
|
|
|
|
|
|
def click_accept():
|
|
main_window = Gui.getMainWindow()
|
|
def finish_click(button):
|
|
button.click()
|
|
flush_gui()
|
|
if Gui.Control.activeDialog() or active_edit_name():
|
|
Gui.activeDocument().resetEdit()
|
|
flush_gui()
|
|
return True
|
|
|
|
candidates = []
|
|
for box in main_window.findChildren(QtWidgets.QDialogButtonBox):
|
|
if not box.isVisible():
|
|
continue
|
|
for button in box.buttons():
|
|
if box.buttonRole(button) == QtWidgets.QDialogButtonBox.AcceptRole and button.isEnabled():
|
|
candidates.append(button)
|
|
if candidates:
|
|
return finish_click(candidates[0])
|
|
for button in main_window.findChildren(QtWidgets.QPushButton):
|
|
text = str(button.text()).replace("&", "").strip().lower()
|
|
if button.isVisible() and button.isEnabled() and text in ("ok", "accept", "done"):
|
|
return finish_click(button)
|
|
dialog = Gui.Control.activeDialog()
|
|
if dialog and hasattr(dialog, "accept"):
|
|
dialog.accept()
|
|
flush_gui()
|
|
return True
|
|
return False
|
|
|
|
|
|
def selection_names():
|
|
return [str(item.ObjectName) for item in Gui.Selection.getSelectionEx()]
|
|
|
|
|
|
def available_transactions(document, method_name):
|
|
method = getattr(document, method_name, None)
|
|
if not callable(method):
|
|
return []
|
|
try:
|
|
return [str(value) for value in method()]
|
|
except Exception:
|
|
return []
|
|
|
|
|
|
def profile_name(feature):
|
|
value = getattr(feature, "Profile", None)
|
|
if isinstance(value, tuple) and value:
|
|
return str(getattr(value[0], "Name", ""))
|
|
return str(getattr(value, "Name", ""))
|
|
|
|
|
|
def counter(document, name):
|
|
try:
|
|
return int(getattr(document, name))
|
|
except Exception:
|
|
return None
|
|
|
|
|
|
try:
|
|
progress("activate-workbench:start")
|
|
Gui.activateWorkbench("PartDesignWorkbench")
|
|
flush_gui()
|
|
progress("activate-workbench:done")
|
|
document = App.newDocument("PadGuiSuccess")
|
|
document.UndoMode = 1
|
|
body = document.addObject("PartDesign::Body", "Body")
|
|
sketch = body.newObject("Sketcher::SketchObject", "Sketch")
|
|
points = [
|
|
App.Vector(0, 0, 0),
|
|
App.Vector(4, 0, 0),
|
|
App.Vector(4, 3, 0),
|
|
App.Vector(0, 3, 0),
|
|
]
|
|
for index in range(4):
|
|
sketch.addGeometry(Part.LineSegment(points[index], points[(index + 1) % 4]), False)
|
|
document.recompute()
|
|
Gui.activeView().setActiveObject("pdbody", body)
|
|
Gui.Selection.clearSelection()
|
|
Gui.Selection.addSelection(document.Name, sketch.Name)
|
|
flush_gui()
|
|
progress("fixture:done")
|
|
|
|
command = Gui.Command.get("PartDesign_Pad")
|
|
actions = list(command.getAction()) if command else []
|
|
before = {
|
|
"activeWorkbench": str(Gui.activeWorkbench().name()),
|
|
"commandRegistered": command is not None,
|
|
"commandActive": bool(command.isActive()) if command else False,
|
|
"actionCount": len(actions),
|
|
"actionEnabled": bool(actions[0].isEnabled()) if actions else None,
|
|
"selection": selection_names(),
|
|
"objectNames": [str(obj.Name) for obj in document.Objects],
|
|
"bodyTip": str(getattr(body.Tip, "Name", "")) if body.Tip else "",
|
|
"undoNames": available_transactions(document, "getAvailableUndoNames"),
|
|
}
|
|
|
|
progress("run-command:start")
|
|
Gui.runCommand("PartDesign_Pad", 0)
|
|
flush_gui()
|
|
progress("run-command:done")
|
|
pad = document.getObject("Pad")
|
|
opened = task_panel_state("opened")
|
|
if pad is None:
|
|
raise RuntimeError("PartDesign_Pad did not create a Pad preview object")
|
|
pad.Length = 10.0
|
|
document.recompute()
|
|
flush_gui()
|
|
progress("preview:done")
|
|
preview = {
|
|
"name": str(pad.Name),
|
|
"typeId": str(pad.TypeId),
|
|
"length": round(float(pad.Length.Value), 9),
|
|
"shapeValid": bool(pad.Shape.isValid()),
|
|
"solidCount": len(pad.Shape.Solids),
|
|
"volume": round(float(pad.Shape.Volume), 9),
|
|
"bodyTip": str(getattr(body.Tip, "Name", "")) if body.Tip else "",
|
|
"profile": profile_name(pad),
|
|
}
|
|
progress("accept:start")
|
|
acceptClicked = click_accept()
|
|
progress("accept:done")
|
|
document.recompute()
|
|
flush_gui()
|
|
closed = task_panel_state("closed")
|
|
pad = document.getObject("Pad")
|
|
after = {
|
|
"acceptClicked": acceptClicked,
|
|
"selection": selection_names(),
|
|
"objectNames": [str(obj.Name) for obj in document.Objects],
|
|
"bodyGroup": [str(obj.Name) for obj in body.Group],
|
|
"bodyTip": str(getattr(body.Tip, "Name", "")) if body.Tip else "",
|
|
"pad": {
|
|
"name": str(pad.Name),
|
|
"typeId": str(pad.TypeId),
|
|
"length": round(float(pad.Length.Value), 9),
|
|
"shapeValid": bool(pad.Shape.isValid()),
|
|
"solidCount": len(pad.Shape.Solids),
|
|
"faceCount": len(pad.Shape.Faces),
|
|
"edgeCount": len(pad.Shape.Edges),
|
|
"vertexCount": len(pad.Shape.Vertexes),
|
|
"volume": round(float(pad.Shape.Volume), 9),
|
|
"profile": profile_name(pad),
|
|
"state": [str(value) for value in pad.State],
|
|
},
|
|
"undoCount": counter(document, "UndoCount"),
|
|
"redoCount": counter(document, "RedoCount"),
|
|
"undoNames": available_transactions(document, "getAvailableUndoNames"),
|
|
"redoNames": available_transactions(document, "getAvailableRedoNames"),
|
|
}
|
|
version = App.Version()
|
|
result = {
|
|
"schemaVersion": 1,
|
|
"baselineId": "freecad-1.1.1",
|
|
"freecadVersion": ".".join(str(value) for value in version[:3]),
|
|
"gitCommit": str(version[7]) if len(version) > 7 else "",
|
|
"workflowId": "partdesign-pad-task",
|
|
"commandId": "PartDesign_Pad",
|
|
"state": "success",
|
|
"before": before,
|
|
"taskPanel": {
|
|
"opened": opened,
|
|
"preview": preview,
|
|
"closed": closed,
|
|
},
|
|
"after": after,
|
|
"success": bool(
|
|
opened["activeDialog"]
|
|
and opened["inEdit"] == "Pad"
|
|
and acceptClicked
|
|
and not closed["activeDialog"]
|
|
and not closed["inEdit"]
|
|
and after["bodyTip"] == "Pad"
|
|
and after["pad"]["shapeValid"]
|
|
and after["pad"]["solidCount"] == 1
|
|
and abs(after["pad"]["volume"] - 120.0) < 1e-7
|
|
),
|
|
}
|
|
progress("result:done")
|
|
App.closeDocument(document.Name)
|
|
emit(result, 0 if result["success"] else 2)
|
|
except Exception as error:
|
|
progress("error:" + type(error).__name__)
|
|
emit({
|
|
"schemaVersion": 1,
|
|
"baselineId": "freecad-1.1.1",
|
|
"workflowId": "partdesign-pad-task",
|
|
"commandId": "PartDesign_Pad",
|
|
"state": "success",
|
|
"success": False,
|
|
"error": {"type": type(error).__name__, "message": str(error)},
|
|
}, 2)
|