Updated

Scripts/Modeling/Edit/AlignEdge.py

@@ -0,0 +1,165 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+import maya.cmds as mc
+import maya.mel as mel
+import maya.OpenMaya as om
+import maya.OpenMayaUI as omui
+import maya.api.OpenMaya as OpenMaya
+from maya.OpenMaya import MGlobal
+import math
+
+def run():
+    mesh=mc.ls(sl=1,fl=1)
+    if len(mesh) == 1:
+        checkLongName = mc.ls(mesh[0],l=1)
+        parentNode = checkLongName[0].split('|')
+        if len(parentNode) > 2:
+            outParent = ''
+            outParent = '|'.join(parentNode[1:-1])
+            mc.parent(mesh[0],w=1)
+        cleanList = ('sampleCurv*','sampleMes*','rotationPlan*')
+        for c in cleanList:
+            if mc.objExists(c):
+                mc.delete(c)
+        gface, gHitp,cEdge,cEdgePos = getClosestEdge()
+        mc.select(cEdge)
+        checkCVList=mc.ls( mc.polyListComponentConversion (cEdge,fe=True,tv=True),flatten=True)
+        mx,my,mz = mc.pointPosition(checkCVList[0],w=1)
+        mc.polyPlane(w=1, h=1, sx=1, sy=1, ax=(0,1,0), cuv=2, ch=0, n='rotationPlane')
+        mc.polyCreateFacet( p=[(mx, my, mz),(cEdgePos[0], cEdgePos[1], cEdgePos[2]),(gHitp[0], gHitp[1], gHitp[2])] )
+        mc.rename('sampleMesh')
+        mc.select("rotationPlane.vtx[0:2]", "sampleMesh.vtx[0:2]")
+        CMD = 'snap3PointsTo3Points(0);'
+        mel.eval(CMD)
+        mc.parent(mesh[0],'rotationPlane')
+        axes = ["X", "Y", "Z"]
+        for a in axes:
+            val = mc.getAttr( mesh[0] + ".rotate" + a)
+            valTmp = ''
+            if val > 0:
+                valTmp = val + 45
+            else:
+                valTmp = val - 45
+            valNew = int (valTmp/90)
+            mc.setAttr(( mesh[0] + ".rotate" + a), (valNew*90))
+
+        mc.move(gHitp[0], gHitp[1], gHitp[2], mesh[0], rpr=True,wd=True)
+        mc.select(mesh[0])
+        mc.parent(w=1)
+        if len(parentNode) > 2:
+            mc.parent(mesh[0],outParent)
+        for c in cleanList:
+            if mc.objExists(c):
+                mc.delete(c)
+
+def getClosestEdge():
+    mayaMesh = mc.ls(sl=1,fl=1)
+    gFace = ''
+    gHitP = ''
+    gFace,gHitP = getClosestMeshHit(mayaMesh[0])
+    listF2E=mc.ls( mc.polyListComponentConversion (gFace,ff=True,te=True),flatten=True)
+    cEdge = ''
+    smallestDist = 1000000
+    cEdgePos = []
+    for l in listF2E:
+        mc.select(l)
+        mc.polyToCurve(form=2, degree=1, conformToSmoothMeshPreview=1)
+        sampleCurve = mc.ls(sl=1)
+        selectionList = om.MSelectionList()
+        selectionList.add(sampleCurve[0])
+        dagPath = om.MDagPath()
+        selectionList.getDagPath(0, dagPath)
+        omCurveOut = om.MFnNurbsCurve(dagPath)
+        pointInSpace = om.MPoint(gHitP[0],gHitP[1],gHitP[2])
+        closestPoint = om.MPoint()
+        closestPoint = omCurveOut.closestPoint(pointInSpace)
+        getDist = math.sqrt( ((closestPoint[0] - gHitP[0])**2)  + ((closestPoint[1]- gHitP[1])**2) + ((closestPoint[2] - gHitP[2])**2))
+        if getDist < smallestDist:
+            smallestDist = getDist
+            cEdge = l
+            cEdgePos = [closestPoint[0],closestPoint[1],closestPoint[2]]
+        mc.delete(sampleCurve)
+    mc.select(cEdge)
+    return(gFace,gHitP,cEdge,cEdgePos)
+
+def getClosestMeshHit(mayaMesh):
+    myShape = mc.listRelatives(mayaMesh, shapes=True,f=True)
+    checkList = screenVisPoly()
+    checkList.remove(myShape[0])
+    meshPos = mc.xform(mayaMesh,q=1, ws=1, a=1, piv=1)
+    posXXX = [meshPos[0],meshPos[1],meshPos[2]]
+    shortDistanceCheck = 10000
+    resultFace = []
+    resultCV =[]
+    resultHitPoint = []
+    for c in checkList:
+        transNode = mc.listRelatives(c, p=True)
+        getFaceDist,getFace,getHitPoint  = getClosestPointOnFace(transNode[0],posXXX)
+        #print (getCV, getFaceDist, getFace)
+        if getFaceDist < shortDistanceCheck:
+            shortDistanceCheck = getFaceDist
+            resultFace = getFace
+            resultHitPoint = getHitPoint
+    return (resultFace,resultHitPoint)
+
+def getClosestPointOnFace(mayaMesh,pos=[0,0,0]):
+    mVector = OpenMaya.MVector(pos)#using MVector type to represent position
+    selectionList = OpenMaya.MSelectionList()
+    selectionList.add(mayaMesh)
+    dPath= selectionList.getDagPath(0)
+    mMesh=OpenMaya.MFnMesh(dPath)
+    ID = mMesh.getClosestPoint(OpenMaya.MPoint(mVector),space=OpenMaya.MSpace.kWorld)[1] #getting closest face ID
+    closestPoint= mMesh.getClosestPoint(OpenMaya.MPoint(mVector),space=OpenMaya.MSpace.kWorld)[0]
+    cpx = closestPoint[0]
+    cpy = closestPoint[1]
+    cpz = closestPoint[2]
+    hitPointPosition = [cpx,cpy,cpz]
+    hitFaceName = (mayaMesh+'.f['+str(ID)+']')
+    getFaceDist = math.sqrt( ((pos[0] - cpx)**2)  + ((pos[1]- cpy)**2) + ((pos[2] - cpz)**2))
+    return (getFaceDist, hitFaceName,hitPointPosition)
+
+def screenVisPoly():
+    commonList= []
+    view = omui.M3dView.active3dView()
+    om.MGlobal.selectFromScreen(0, 0, view.portWidth(), view.portHeight(), om.MGlobal.kReplaceList)
+    objects = om.MSelectionList()
+    sel = om.MSelectionList()
+    om.MGlobal.getActiveSelectionList(objects)
+    om.MGlobal.setActiveSelectionList(sel, om.MGlobal.kReplaceList)
+    fromScreen = []
+    objects.getSelectionStrings(fromScreen)
+    shapesOnScreen = mc.listRelatives(fromScreen, shapes=True,f=True)
+    meshList = mc.ls(type='mesh',l=True)#only polygon
+    if len(meshList)>0 and shapesOnScreen is not None:
+        commonList = list(set(meshList) & set(shapesOnScreen))
+        return commonList
+    else:
+        commonList = []
+        return commonList
+
+def getPolyFaceCenter(faceName):
+    meshFaceName = faceName.split('.')[0]
+    findVtx = mc.polyInfo(faceName, fv=1)
+    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)
+    centerX = 0
+    centerY = 0
+    centerZ = 0
+    for g in getNumber:
+        x,y,z = mc.pointPosition((meshFaceName + '.vtx['+g + ']'),w=1)
+        centerX = centerX + x
+        centerY = centerY + y
+        centerZ = centerZ + z
+
+    centerX = centerX/len(getNumber)
+    centerY = centerY/len(getNumber)
+    centerZ = centerZ/len(getNumber)
+    return centerX,centerY,centerZ
+
+if __name__ == "__main__":
+    run()