103 lines
4.4 KiB
Python
103 lines
4.4 KiB
Python
# -*- coding: utf-8 -*-
|
|
# ***************************************************************************
|
|
# * Copyright (c) 2019 sliptonic <shopinthewoods@gmail.com> *
|
|
# * *
|
|
# * This program is free software; you can redistribute it and/or modify *
|
|
# * it under the terms of the GNU Lesser General Public License (LGPL) *
|
|
# * as published by the Free Software Foundation; either version 2 of *
|
|
# * the License, or (at your option) any later version. *
|
|
# * for detail see the LICENCE text file. *
|
|
# * *
|
|
# * This program is distributed in the hope that it will be useful, *
|
|
# * but WITHOUT ANY WARRANTY; without even the implied warranty of *
|
|
# * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
|
|
# * GNU Library General Public License for more details. *
|
|
# * *
|
|
# * You should have received a copy of the GNU Library General Public *
|
|
# * License along with this program; if not, write to the Free Software *
|
|
# * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 *
|
|
# * USA *
|
|
# * *
|
|
# ***************************************************************************
|
|
|
|
import FreeCADGui
|
|
import FreeCAD
|
|
import Path
|
|
import Path.Main.Gui.JobCmd as PathJobCmd
|
|
import Path.Tool.Controller as PathToolController
|
|
import PathGui
|
|
import PathScripts.PathUtils as PathUtils
|
|
from PySide import QtGui
|
|
|
|
if False:
|
|
Path.Log.setLevel(Path.Log.Level.DEBUG, Path.Log.thisModule())
|
|
Path.Log.trackModule(Path.Log.thisModule())
|
|
else:
|
|
Path.Log.setLevel(Path.Log.Level.INFO, Path.Log.thisModule())
|
|
|
|
translate = FreeCAD.Qt.translate
|
|
|
|
|
|
class PathUtilsUserInput(object):
|
|
def selectedToolController(self):
|
|
tc = None
|
|
# check if a user has selected a tool controller in the tree.
|
|
# Return the first one and remove all from selection
|
|
for sel in FreeCADGui.Selection.getSelectionEx():
|
|
if hasattr(sel.Object, "Proxy"):
|
|
if isinstance(sel.Object.Proxy, PathToolController.ToolController):
|
|
if tc is None:
|
|
tc = sel.Object
|
|
FreeCADGui.Selection.removeSelection(sel.Object)
|
|
return tc
|
|
|
|
def chooseToolController(self, controllers):
|
|
form = FreeCADGui.PySideUic.loadUi(":/panels/DlgTCChooser.ui")
|
|
mylist = [i.Label for i in controllers]
|
|
form.uiToolController.addItems(mylist)
|
|
r = form.exec_()
|
|
if not r:
|
|
return None
|
|
return [i for i in controllers if i.Label == form.uiToolController.currentText()][0]
|
|
|
|
def chooseJob(self, jobs):
|
|
job = None
|
|
selected = FreeCADGui.Selection.getSelection()
|
|
if 1 == len(selected) and selected[0] in jobs:
|
|
job = selected[0]
|
|
else:
|
|
modelSelected = []
|
|
for job in jobs:
|
|
if all([o in job.Model.Group for o in selected]):
|
|
modelSelected.append(job)
|
|
if 1 == len(modelSelected):
|
|
job = modelSelected[0]
|
|
else:
|
|
modelObjectSelected = []
|
|
for job in jobs:
|
|
if all([o in job.Proxy.baseObjects(job) for o in selected]):
|
|
modelObjectSelected.append(job)
|
|
if 1 == len(modelObjectSelected):
|
|
job = modelObjectSelected[0]
|
|
else:
|
|
if modelObjectSelected:
|
|
mylist = [j.Label for j in modelObjectSelected]
|
|
else:
|
|
mylist = [j.Label for j in jobs]
|
|
|
|
jobname, result = QtGui.QInputDialog.getItem(
|
|
None, translate("Path", "Choose a CAM Job"), None, mylist
|
|
)
|
|
|
|
if result is False:
|
|
return None
|
|
else:
|
|
job = [j for j in jobs if j.Label == jobname][0]
|
|
return job
|
|
|
|
def createJob(self):
|
|
return PathJobCmd.CommandJobCreate().Activated()
|
|
|
|
|
|
PathUtils.UserInput = PathUtilsUserInput()
|