154 lines
4.7 KiB
Python
154 lines
4.7 KiB
Python
import os
|
|
import math
|
|
import ldmt_function.ldmt_loadUIFile
|
|
import importlib
|
|
#importlib.reload(ldmt_function.ldmt_loadUIFile)
|
|
from ldmt_function.ldmt_loadUIFile import get_maya_window, load_ui_file
|
|
import maya.OpenMayaUI as omui
|
|
from ldmt_core import ldmt_cmds as ld
|
|
from functools import partial
|
|
import maya.cmds as cmds
|
|
import maya.mel as mel
|
|
|
|
try:
|
|
from Qt.QtCore import *
|
|
from Qt.QtGui import *
|
|
from Qt.QtWidgets import *
|
|
# from Qt.QtUiTools import *
|
|
from Qt import __version__
|
|
from Qt.QtCompat import wrapInstance
|
|
except ImportError:
|
|
from Qt.QtCore import *
|
|
from Qt.QtGui import *
|
|
# from Qt.QtUiTools import *
|
|
from Qt import __version__
|
|
from Qt.QtCompat import wrapInstance
|
|
|
|
import maya.OpenMaya as om
|
|
import maya.api.OpenMaya as om2
|
|
import random
|
|
import ast
|
|
|
|
LDMTPATH = ld.getPath('LDMT')
|
|
ldmt_uifile = LDMTPATH + '/ldmt_ui/ldmt_clothTransfer.ui'
|
|
# ldmt_list_form, ldmt_list_base = load_ui_type(ldmt_uifile)
|
|
ldmt_window_name = 'ldmt_clothTransfer'
|
|
ldmt_button_name = 'btn_'+ldmt_window_name.split('_')[1]
|
|
|
|
'''
|
|
#Functions
|
|
'''
|
|
|
|
'''
|
|
#UI
|
|
'''
|
|
class ldmt_cls(QDialog):
|
|
def __init__(self, parent = get_maya_window()):
|
|
super(ldmt_cls, self).__init__(parent)
|
|
self.window_name = ldmt_window_name
|
|
self.setWindowTitle(ldmt_window_name)
|
|
self.setWindowFlags(Qt.Window)
|
|
# self.setupUi(self)
|
|
# self.move(QCursor.pos() + QPoint(20,20))
|
|
self.set_ui()
|
|
# update status bar so it's not only show in help line window.
|
|
self.setupBtn()
|
|
# self.statusbar.showMessage(ld.tag())
|
|
# self.installStartBar()
|
|
|
|
def set_ui(self):
|
|
main_layout = QVBoxLayout(self)
|
|
|
|
self.ui = load_ui_file(ldmt_uifile)
|
|
|
|
main_layout.addWidget(self.ui)
|
|
|
|
def setupBtn(self):
|
|
self.ui.btn_setCloth.clicked.connect(self.setCloth)
|
|
self.ui.btn_setOrigin.clicked.connect(self.setOrigin)
|
|
self.ui.btn_setTarget.clicked.connect(self.setTarget)
|
|
self.ui.btn_selectCloth.clicked.connect(self.selectCloth)
|
|
self.ui.btn_selectOrigin.clicked.connect(self.selectOrigin)
|
|
self.ui.btn_selectTarget.clicked.connect(self.selectTarget)
|
|
|
|
self.ui.btn_transfer.clicked.connect(self.transfer)
|
|
|
|
def setCloth(self):
|
|
sel = cmds.ls(sl=1,o=1)
|
|
sel = sel[0]
|
|
if sel != []:
|
|
self.ui.btn_selectCloth.setText(str(sel))
|
|
else:
|
|
self.ui.btn_selectCloth.setText('...')
|
|
def setOrigin(self):
|
|
sel = cmds.ls(sl=1,o=1)
|
|
sel = sel[0]
|
|
if sel != []:
|
|
self.ui.btn_selectOrigin.setText(str(sel))
|
|
else:
|
|
self.ui.btn_selectOrigin.setText('...')
|
|
def setTarget(self):
|
|
sel = cmds.ls(sl=1,o=1)
|
|
sel = sel[0]
|
|
if sel != []:
|
|
self.ui.btn_selectTarget.setText(str(sel))
|
|
else:
|
|
self.ui.btn_selectTarget.setText('...')
|
|
def selectSetObjects(self,sel):
|
|
if sel != '...':
|
|
cmds.select(sel,r=1)
|
|
else:
|
|
ld.msg('Nothing selected')
|
|
def selectCloth(self):
|
|
sel = self.ui.btn_selectCloth.text()
|
|
self.selectSetObjects(sel)
|
|
def selectOrigin(self):
|
|
sel = self.ui.btn_selectOrigin.text()
|
|
self.selectSetObjects(sel)
|
|
def selectTarget(self):
|
|
sel = self.ui.btn_selectTarget.text()
|
|
self.selectSetObjects(sel)
|
|
def transfer(self):
|
|
cloth = self.ui.btn_selectCloth.text()
|
|
origin = self.ui.btn_selectOrigin.text()
|
|
target = self.ui.btn_selectTarget.text()
|
|
cmds.makeIdentity(target,apply=True, t=1, r=1, s=1, n=0)
|
|
cmds.select(cloth,r=1)
|
|
cmds.select(origin,add=1)
|
|
mel.eval('DeleteHistory;')
|
|
mel.eval('CreateWrap;')
|
|
blendObjs = [target,origin]
|
|
blendName = cmds.blendShape(blendObjs,o='world',n='clothTransfer#')
|
|
cmds.blendShape(blendName,e=1,w=(0,1))
|
|
|
|
def installStartBar(self):
|
|
allQWidgets = self.findChildren(QWidget)
|
|
for i in allQWidgets:
|
|
i.installEventFilter(self)
|
|
|
|
def eventFilter(self, obj, event ):
|
|
'''Connect signals on mouse over'''
|
|
if event.type() == QEvent.Enter:
|
|
self.oldMessage = ld.tag()
|
|
self.statusbar.showMessage(' '+obj.statusTip(),0)
|
|
elif event.type() == QEvent.Leave:
|
|
self.statusbar.showMessage(' '+self.oldMessage, 0)
|
|
pass
|
|
event.accept()
|
|
return False
|
|
# def closeEvent(self,event):
|
|
# ld.turnToolBtnOff(self,ldmt_button_name)
|
|
# cmds.deleteUI(ldmt_window_name)
|
|
|
|
def ldmt_show():
|
|
|
|
if cmds.window(ldmt_window_name, exists=True):
|
|
cmds.deleteUI(ldmt_window_name)
|
|
|
|
app = QApplication.instance()
|
|
dialog = ldmt_cls(parent=app.activeWindow())
|
|
dialog.show()
|
|
app.exec_()
|
|
|
|
if __name__ == '__main__':
|
|
ldmt_show() |