Updated
This commit is contained in:
254
Scripts/Modeling/Edit/PolyFold.py
Normal file
254
Scripts/Modeling/Edit/PolyFold.py
Normal file
@ -0,0 +1,254 @@
|
||||
#!/usr/bin/env python
|
||||
# -*- coding: utf-8 -*-
|
||||
|
||||
import maya.cmds as cmds
|
||||
import math
|
||||
import re
|
||||
import maya.mel as mel
|
||||
|
||||
def updateSNA(new):
|
||||
cmds.intField('snapAngleV', e=1, v = new )
|
||||
|
||||
def polyFoldDoit():
|
||||
global snapStepping
|
||||
global pfGeo
|
||||
pfGeo = []
|
||||
getSV = cmds.intField('snapAngleV', q=1, v =1 )
|
||||
snapStepping = getSV
|
||||
checCurrentkHUD = cmds.headsUpDisplay(lh=1)
|
||||
if checCurrentkHUD is not None:
|
||||
for t in checCurrentkHUD:
|
||||
cmds.headsUpDisplay(t, rem=1)
|
||||
global ctx
|
||||
ctx = 'plCtx'
|
||||
# Delete dragger context if it already exists
|
||||
if cmds.draggerContext(ctx, exists=True):
|
||||
cmds.deleteUI(ctx)
|
||||
# Create dragger context and set it to the active tool
|
||||
cmds.draggerContext(ctx, pressCommand = pfPress, rc = pfOff, dragCommand = pfDrag, name=ctx, cursor='crossHair',undoMode='step')
|
||||
cmds.setToolTo(ctx)
|
||||
|
||||
def pfOff():
|
||||
global pfGeo
|
||||
cmds.delete(pfGeo[0], ch=1)
|
||||
cmds.headsUpDisplay( 'HUDunFoldStep',rem=True)
|
||||
cleanList = ('rotOffset','rotBase','rotCC')
|
||||
for c in cleanList:
|
||||
if cmds.objExists(c):
|
||||
cmds.delete(c)
|
||||
cmds.setToolTo('selectSuperContext')
|
||||
|
||||
def currentStep():
|
||||
global viewPortCount
|
||||
return viewPortCount
|
||||
|
||||
def pfPress():
|
||||
global ctx
|
||||
global screenX,screenY
|
||||
global lockCount
|
||||
global storeCount
|
||||
global viewPortCount
|
||||
global pfGeo
|
||||
viewPortCount = 0
|
||||
lockCount = 50
|
||||
storeCount = 0
|
||||
vpX, vpY, _ = cmds.draggerContext(ctx, query=True, anchorPoint=True)
|
||||
screenX = vpX
|
||||
screenY = vpY
|
||||
lockX = vpX
|
||||
cmds.headsUpDisplay( 'HUDunFoldStep', section=1, block=0, blockSize='large', label='Fold Angle', labelFontSize='large', command=currentStep, atr=1)
|
||||
selFace = []
|
||||
selFace = cmds.filterExpand(expand=True ,sm=34)
|
||||
if selFace:
|
||||
pfGeo = cmds.ls(hl=1)
|
||||
cmds.delete(pfGeo[0], ch=1)
|
||||
cleanList = ('rotOffset','rotBase','rotCC')
|
||||
for c in cleanList:
|
||||
if cmds.objExists(c):
|
||||
cmds.delete(c)
|
||||
selGeo = cmds.ls(hl=1)
|
||||
cmds.ConvertSelectionToEdgePerimeter()
|
||||
cmds.polySelectConstraint(mode =2, type = 0x8000 ,where =2)
|
||||
cmds.polySelectConstraint(m =0,w=0)
|
||||
getEdges = cmds.ls(sl=1,fl=1)
|
||||
midEdge = getEdges[int((len(getEdges)/2))]
|
||||
cmds.select(midEdge)
|
||||
cmds.polySelectConstraint(type=0x8000, propagate=4)
|
||||
cmds.polySelectConstraint(m =0,w=0,propagate=0)
|
||||
contEdges = cmds.ls(sl=1,fl=1)
|
||||
commonList = list(set(getEdges) - ( set(getEdges) - set(contEdges) ) )
|
||||
cmds.select(commonList)
|
||||
listVtx = vtxLoopOrder(commonList)
|
||||
pA = cmds.pointPosition(listVtx[0], w =1)
|
||||
pB = cmds.pointPosition(listVtx[-1], w =1)
|
||||
cmds.group( em=True, name='rotBase')
|
||||
cmds.setAttr('rotBase.translateX',pA[0])
|
||||
cmds.setAttr('rotBase.translateY',pA[1])
|
||||
cmds.setAttr('rotBase.translateZ',pA[2])
|
||||
cmds.group( em=True, name='rotAim')
|
||||
cmds.setAttr('rotAim.translateX',pB[0])
|
||||
cmds.setAttr('rotAim.translateY',pB[1])
|
||||
cmds.setAttr('rotAim.translateZ',pB[2])
|
||||
consNode = cmds.aimConstraint(('rotAim'),('rotBase'),offset=[0,0,0], weight=1, aimVector=[1,0,0], upVector=[0,1,0], worldUpType='scene')
|
||||
cmds.select(selFace)
|
||||
cmds.CreateCluster()
|
||||
checkClusterName = cmds.ls(sl=1,fl=1,typ='transform')
|
||||
cmds.rename(checkClusterName, 'rotCC')
|
||||
newNode = cmds.duplicate('rotBase',rr=1)
|
||||
cmds.rename(newNode[0],'rotOffset')
|
||||
cmds.parent('rotOffset', 'rotBase')
|
||||
cmds.delete('rotBase',constraints=1)
|
||||
cmds.setAttr('rotOffset.rotateX',0)
|
||||
cmds.setAttr('rotOffset.rotateY',0)
|
||||
cmds.setAttr('rotOffset.rotateZ',0)
|
||||
cmds.parent('rotCC','rotOffset')
|
||||
cmds.select(selFace)
|
||||
cmd = 'doMenuComponentSelection("' + pfGeo[0] +'", "facet");'
|
||||
mel.eval(cmd)
|
||||
cmds.delete('rotAim')
|
||||
cmds.setAttr('rotBase.visibility',0)
|
||||
|
||||
def pfDrag():
|
||||
global screenX,screenY
|
||||
global viewPortCount
|
||||
global lockCount
|
||||
global storeCount
|
||||
global snapStepping
|
||||
modifiers = cmds.getModifiers()
|
||||
vpX, vpY, _ = cmds.draggerContext(ctx, query=True, dragPoint=True)
|
||||
stepping = snapStepping
|
||||
if(modifiers == 1):
|
||||
if screenX > vpX:
|
||||
lockCount = lockCount - 1
|
||||
else:
|
||||
lockCount = lockCount + 1
|
||||
screenX = vpX
|
||||
|
||||
if lockCount < -360:
|
||||
lockCount = 360
|
||||
elif lockCount > 360:
|
||||
lockCount = 360
|
||||
getX = int(lockCount / stepping)*stepping
|
||||
if storeCount != getX:
|
||||
storeCount = getX
|
||||
cmds.setAttr('rotOffset.rotateX',storeCount)
|
||||
viewPortCount = storeCount
|
||||
elif(modifiers == 4):
|
||||
if screenX > vpX:
|
||||
lockCount = lockCount - 5
|
||||
else:
|
||||
lockCount = lockCount + 5
|
||||
screenX = vpX
|
||||
|
||||
if lockCount < -360:
|
||||
lockCount = 360
|
||||
elif lockCount > 360:
|
||||
lockCount = 360
|
||||
getX = int(lockCount / stepping)*stepping
|
||||
if storeCount != getX:
|
||||
storeCount = getX
|
||||
cmds.setAttr('rotOffset.rotateX',storeCount)
|
||||
viewPortCount = storeCount
|
||||
else:
|
||||
if screenX > vpX:
|
||||
lockCount = lockCount -1
|
||||
else:
|
||||
lockCount = lockCount + 1
|
||||
screenX = vpX
|
||||
if lockCount < -360:
|
||||
lockCount = 360
|
||||
elif lockCount > 360:
|
||||
lockCount = 360
|
||||
cmds.setAttr('rotOffset.rotateX',lockCount)
|
||||
viewPortCount = lockCount
|
||||
|
||||
cmds.refresh(f=True)
|
||||
|
||||
def vtxLoopOrder(edgelist):
|
||||
selEdges = edgelist
|
||||
#print(selEdges)
|
||||
shapeNode = cmds.listRelatives(selEdges[0], fullPath=True , parent=True )
|
||||
transformNode = cmds.listRelatives(shapeNode[0], fullPath=True , parent=True )
|
||||
edgeNumberList = []
|
||||
for a in selEdges:
|
||||
checkNumber = ((a.split('.')[1]).split('\n')[0]).split(' ')
|
||||
for c in checkNumber:
|
||||
findNumber = ''.join([n for n in c.split('|')[-1] if n.isdigit()])
|
||||
if findNumber:
|
||||
edgeNumberList.append(findNumber)
|
||||
getNumber = []
|
||||
for s in selEdges:
|
||||
evlist = cmds.polyInfo(s,ev=True)
|
||||
checkNumber = ((evlist[0].split(':')[1]).split('\n')[0]).split(' ')
|
||||
for c in checkNumber:
|
||||
findNumber = ''.join([n for n in c.split('|')[-1] if n.isdigit()])
|
||||
if findNumber:
|
||||
getNumber.append(findNumber)
|
||||
dup = set([x for x in getNumber if getNumber.count(x) > 1])
|
||||
getHeadTail = list(set(getNumber) - dup)
|
||||
vftOrder = []
|
||||
vftOrder.append(getHeadTail[0])
|
||||
count = 0
|
||||
while len(dup)> 0 and count < 100:
|
||||
checkVtx = transformNode[0]+'.vtx['+ vftOrder[-1] + ']'
|
||||
velist = cmds.polyInfo(checkVtx,ve=True)
|
||||
getNumber = []
|
||||
checkNumber = ((velist[0].split(':')[1]).split('\n')[0]).split(' ')
|
||||
for c in checkNumber:
|
||||
findNumber = ''.join([n for n in c.split('|')[-1] if n.isdigit()])
|
||||
if findNumber:
|
||||
getNumber.append(findNumber)
|
||||
findNextEdge = []
|
||||
for g in getNumber:
|
||||
if g in edgeNumberList:
|
||||
findNextEdge = g
|
||||
edgeNumberList.remove(findNextEdge)
|
||||
checkVtx = transformNode[0]+'.e['+ findNextEdge + ']'
|
||||
findVtx = cmds.polyInfo(checkVtx,ev=True)
|
||||
getNumber = []
|
||||
checkNumber = ((findVtx[0].split(':')[1]).split('\n')[0]).split(' ')
|
||||
for c in checkNumber:
|
||||
findNumber = ''.join([n for n in c.split('|')[-1] if n.isdigit()])
|
||||
if findNumber:
|
||||
getNumber.append(findNumber)
|
||||
gotNextVtx = []
|
||||
for g in getNumber:
|
||||
if g in dup:
|
||||
gotNextVtx = g
|
||||
dup.remove(gotNextVtx)
|
||||
vftOrder.append(gotNextVtx)
|
||||
count += 1
|
||||
vftOrder.append(getHeadTail[1])
|
||||
finalList = []
|
||||
for v in vftOrder:
|
||||
finalList.append(transformNode[0]+'.vtx['+ v + ']' )
|
||||
return finalList
|
||||
|
||||
def run():
|
||||
if cmds.window("polyFoldDoitUI", exists = True):
|
||||
cmds.deleteUI("polyFoldDoitUI")
|
||||
polyFoldDoitUI = cmds.window("polyFoldDoitUI",title = "Poly Fold", w=340)
|
||||
cmds.frameLayout(labelVisible= False)
|
||||
cmds.text(l ='')
|
||||
cmds.rowColumnLayout(nc=15 ,cw=[(1,5),(2,80),(3,40),(4,5),(5,20),(6,5),(7,20),(8,5),(9,20),(10,5),(11,20),(12,5),(13,20),(14,5),(15,50)])
|
||||
cmds.text(l ='')
|
||||
cmds.text(l ='Snap Angle:')
|
||||
cmds.intField('snapAngleV', v = 15 )
|
||||
cmds.text(l ='')
|
||||
cmds.button( l= '5', c= lambda x: updateSNA(5))
|
||||
cmds.text(l ='')
|
||||
cmds.button( l= '10', c= lambda x: updateSNA(10))
|
||||
cmds.text(l ='')
|
||||
cmds.button( l= '15', c= lambda x: updateSNA(15))
|
||||
cmds.text(l ='')
|
||||
cmds.button( l= '30', c= lambda x: updateSNA(30))
|
||||
cmds.text(l ='')
|
||||
cmds.button( l= '45', c= lambda x: updateSNA(45))
|
||||
cmds.text(l ='')
|
||||
cmds.iconTextButton(style='textOnly', l='Fold', rpt=1,bgc = [0.2,0.2,0.2], c=polyFoldDoit)
|
||||
cmds.setParent( '..' )
|
||||
cmds.text(l ='')
|
||||
cmds.showWindow(polyFoldDoitUI)
|
||||
|
||||
if __name__ == "__main__":
|
||||
run()
|
||||
Reference in New Issue
Block a user