def setupMannequinUI(): """Sets up the side panel UI for the Mannequin plugin.""" currentContext = cmds.currentCtx() influenceObjectsStr = cmds.mannequinContext(currentContext, q=True, io=True) ioTokens = influenceObjectsStr.split(" ") ioDagPaths = ioTokens[::2] ioAvailableStyles = ioTokens[1::2] mannequinDockPtr = MQtUtil.findLayout("mannequinPaletteDock") mannequinDock = wrapInstance(long(mannequinDockPtr), QWidget) mannequinDock.setMinimumWidth(300) mannequinLayoutPtr = MQtUtil.findLayout("mannequinPaletteLayout") mannequinLayout = wrapInstance(long(mannequinLayoutPtr), QWidget) mannequinSearchPtr = MQtUtil.findControl("mannequinSearchField") mannequinSearch = wrapInstance(long(mannequinSearchPtr), QLineEdit) uiFile = QFile(os.path.join(os.path.dirname(__file__), "mannequin.ui")) uiFile.open(QFile.ReadOnly) gui = mannequinToolPanel.loader.load(uiFile, parentWidget=mannequinLayout) uiFile.close() selList = om.MSelectionList() for obj in ioDagPaths: selList.add(obj) joints = [] for i in range(selList.length()): dagPath = selList.getDagPath(i) dependNode = selList.getDependNode(i) styles = ioAvailableStyles[i] joints.append(JointInfo(dagPath, dependNode, styles)) # Alphabetize joints by full DAG path; should sort slightly better. joints = sorted(joints, key=lambda j: j.dagPath.fullPathName()) prefixTrim = commonPrefix(joints) mannequinToolPanel.reset(mannequinLayout, gui, mannequinSearch, prefixTrim) jointDisplays = organizeJoints(joints) for jointDisplay in jointDisplays: mannequinToolPanel.layoutJointGroup(jointDisplay) mannequinToolPanel.finishLayout()
def convert(mguiName): """ From the name of a Maya UI element of any type to pointer, and wrap the pointer into a python QObject :param mguiName: Maya UI Name :type mguiName: string :return: QWidget or subclass instance :rtype: QtWidgets.QWidget Thanks to Nathan Horne """ # Get pointer ptr = MQtUtil.findControl(mguiName) if ptr is None: ptr = MQtUtil.findLayout(mguiName) if ptr is None: ptr = MQtUtil.findMenuItem(mguiName) if ptr is None: return None # Find corresponding class qObj = QtCompat.wrapInstance(long(ptr), QtCore.QObject) metaObj = qObj.metaObject() cls = metaObj.className() superCls = metaObj.superClass().className() if hasattr(QtWidgets, cls): base = getattr(QtWidgets, cls) elif hasattr(QtWidgets, superCls): base = getattr(QtWidgets, superCls) else: base = QtWidgets.QWidget return QtCompat.wrapInstance(long(ptr), base)
def convert(mguiName): """ From the name of a Maya UI element of any type to pointer, and wrap the pointer into a python QObject :param mguiName: Maya UI Name :type mguiName: string :return: QWidget or subclass instance :rtype: QtGui.QWidget Thanks to Nathan Horne """ # Get pointer ptr = MQtUtil.findControl(mguiName) if ptr is None: ptr = MQtUtil.findLayout(mguiName) if ptr is None: ptr = MQtUtil.findMenuItem(mguiName) if ptr is None: return None # Find corresponding class qObj = wrapInstance(long(ptr), QtCore.QObject) metaObj = qObj.metaObject() cls = metaObj.className() superCls = metaObj.superClass().className() if hasattr(QtGui, cls): base = getattr(QtGui, cls) elif hasattr(QtGui, superCls): base = getattr(QtGui, superCls) else: base = QtGui.QWidget return wrapInstance(long(ptr), base)
def __init__(self, name='scriptEditor', path='', **kwargs): parentName = cmds.setParent(q=True) p = MQtUtil.findLayout(parentName) lytWd = wrapInstance(long(p), QtWidgets.QWidget) widget = ScriptEditorWidget(name, textChangeHandler=self.textChanged, **kwargs) lytWd.layout().addWidget(widget) self.__widget_ref = _weakref(widget) self.editor = parentName + '|' + name super(ScriptEditor, self).__init__(path or self.editor)
def get_q_object(element_name): """ :param element_name: str(<maya_element_name>) :return: QObject """ pointer = MQtUtil.findControl(element_name) if not pointer: pointer = MQtUtil.findLayout(element_name) if not pointer: pointer = MQtUtil.findMenuItem(element_name) return shiboken.wrapInstance(pointer, QtCore.QObject)
def getChildWin(title, widgType=QtWidgets.QWidget): for w in getMayaMainWindow().children(): if w.isWidgetType() and w.windowTitle() == title: return w else: ptr = MQtUtil.findControl(title) if not ptr: ptr = MQtUtil.findWindow(title) if not ptr: ptr = MQtUtil.findLayout(title) if not ptr: return return shiboken2.wrapInstance(long(ptr), widgType)
def _find_widget_ptr(widget): ptr = (MQtUtil.findControl(widget) or MQtUtil.findLayout(widget) or MQtUtil.findMenuItem(widget)) return ptr