Exemplo n.º 1
0
 def clampInfluences(self, mesh, maxInf, debug=0, force=False):
     '''
     Sets max influences on skincluster of mesh / cutting off smallest ones
     '''
     skinClust = self.findRelatedSkinCluster(mesh)
 
     lockedInfluences = self.checkLockedInfluences(skinClust)
     doit = True
     if lockedInfluences:
         if force:
             self.unlockLockedInfluences(skinClust)
             cmds.warning('Locked influences were unlocked on skinCluster')
         else:
             doit = False
     
     if doit:
         verts = self.checkMaxSkinInfluences(mesh, maxInf)
         
         print 'pruneVertWeights>> Pruning', len(verts), 'vertices'
         
         for v in verts:
             infs = cmds.skinPercent(skinClust, (mesh + ".vtx[" + str(v) + "]"), q=1, v=1)
             active = []
             for inf in infs:
                 if inf > 0.0: active.append(inf)
             active = list(reversed(sorted(active)))
             if debug: print 'Clamping vertex', v, 'to', active[maxInf]
             cmds.skinPercent(skinClust, (mesh + ".vtx[" + str(v) + "]"), pruneWeights=(active[maxInf]*1.001))
     else:
         cmds.warning('Cannot clamp influences due to locked weights on skinCluster')
Exemplo n.º 2
0
    def doIt(self, argList):
        # get only the first object from argument list
        try:
            obj = misc.getArgObj(self.syntax(), argList)[0]
        except:
            cmds.warning("No object selected!")
            return
        if (cmds.objectType(obj) != 'transform'):
            cmds.error("Object is not of type transform!")
            return

        # get and cast array of MPoint-Objects to list structure (needed for ConvexHull)
        pSet = list([p.x, p.y, p.z] for p in misc.getPoints(obj))
        # execute delaunay triangulation algorithm on point set of current object
        hull = Delaunay(pSet)

        # delaunay-triangulated tetrahedra are not ordered --> make sure implicit normal points outwards of each tetrahedra
        for tetra in hull.points[hull.simplices]:
            center = map(operator.add, map(operator.add, tetra[0], tetra[1]), map(operator.add, tetra[2], tetra[3]))
            center = map(operator.div, center, [4.0]*3)
            # check if tetrahedron is oriented in wrong direction (normals should point away from center)
            v1 = map(operator.sub, tetra[0], tetra[1])
            v2 = map(operator.sub, tetra[0], tetra[2])
            # check dot product: if smaller 0, swap vertices 0 and 2
            if (np.dot(np.cross(v1, v2), map(operator.sub, tetra[0], center) ) > 0):
                # swap 0 and 2
                tetra[[0,2]] = tetra[[2,0]]
            self.appendToResult(str(tetra.flatten().tolist())[1:-1])
Exemplo n.º 3
0
def printCurvePoints():
    """
    Use: For printing to the console or write out to a file, the points of a new custom curve.
    This can be added to the curve Dictionary for quickly building more curves for all rigging tools using the curve Dictionary.
    @param toFile: if you would like to output the points to a text file in your home folder or not
    @type toFile: Boolean
    """
    curvePoints = []
    curSel = cmds.ls(sl = True)
    if not curSel:
        cmds.warning('Nothing selected for export.')
        pass
    else:
        for nurbsCurve in curSel:
            getDegree = cmds.getAttr(nurbsCurve + '.degree')
            getSpans = cmds.getAttr(nurbsCurve + '.spans')
            numPoints = int(getDegree) + int(getSpans)
            for x in range (numPoints):
                getWS = cmds.xform(nurbsCurve + '.cv[%s]' % x, query = True, t = True, ws = True)
                tempVar = []
                for p in getWS:
                    tempVar.append("{0:.2f}".format(p))
                curvePoints.append(tempVar)
                tempVar = []
            numKnots = numPoints + getDegree -1

        sys.stdout.write('cmds.curve(name = "' + str(curSel[0]) + '", d = ' + str(getDegree) + ', p = ' + str(curvePoints) + ')')
Exemplo n.º 4
0
def mtt_log(msg, add_tag=None, msg_type=None, verbose=True):
    """ Format output message '[TAG][add_tag] Message content'

    :param msg: (string) message content
    :param add_tag: (string or list) add extra tag to output
    :param msg_type: define message type. Accept : None, 'warning', 'error'
    :param verbose: (bool) enable headsUpMessage output
    """
    # define tags
    tag_str = '[MTT]'

    # append custom tags
    if add_tag:
        if not isinstance(add_tag, list):
            add_tag = [add_tag]

        for tag in add_tag:
            tag_str += '[%s]' % tag.upper()

    # output message the right way
    if msg_type == 'warning':
        cmds.warning('%s %s' % (tag_str, msg))
    elif msg_type == 'error':
        cmds.error('%s %s' % (tag_str, msg))
    else:
        print '%s %s\n' % (tag_str, msg),

    if verbose and MTTSettings.value('showHeadsUp'):
        cmds.headsUpMessage(msg)
Exemplo n.º 5
0
    def _get_recommended_pivot_bank(self, geometries, tm_ref, tm_ref_dir, pos_toes, direction=1):
        """
        Determine recommended position using ray-cast from the toes.
        TODO: If the ray-case fail, use a specified default value.
        return: The recommended position as a world pymel.datatypes.Vector
        """
        # Sanity check, ensure that at least one point is in the bounds of geometries.
        # This can prevent rays from being fired from outside a geometry.
        # TODO: Make it more robust.
        filtered_geometries = []
        for geometry in geometries:
            xmin, ymin, zmin, xmax, ymax, zmax = cmds.exactWorldBoundingBox(geometry.__melobject__())
            bound = pymel.datatypes.BoundingBox((xmin, ymin, zmin), (xmax, ymax, zmax))
            if bound.contains(pos_toes):
                filtered_geometries.append(geometry)

        dir = pymel.datatypes.Point(direction, 0, 0) * tm_ref_dir
        pos = libRigging.ray_cast_nearest(pos_toes, dir, filtered_geometries)
        if not pos:
            cmds.warning("Can't automatically solve FootRoll bank inn pivot.")
            pos = pos_toes

        pos.y = 0

        return pos
def buildOceanNurbsPreviewPlane(xRes = 20, zRes = 20, size = 8, oceanShader = '', boatName = ''):
    """
    Creates NURBS plane that moves along with ocean so we can intersect this with our boatName to create the emitter curves
    """
    debug(None, method = 'buildOceanNurbsPreviewPlane', message = 'Building NURBS Intersection plane now', verbose = False)
    if cmds.objExists('%s_NurbsIntersect_geo' % boatName):
        cmds.warning('Skipping %s_NurbsIntersect. Already exists in the scene!' % boatName)
        debug(None, method = 'buildOceanNurbsPreviewPlane', message = 'Skipping %s_NurbsIntersect. Already exists in the scene!' % boatName, verbose = False)
    else:
        NURBSPlane = cmds.nurbsPlane( width=1,lengthRatio = 1,degree = 1, patchesU=xRes, patchesV=zRes, axis = (0,1,0) , constructionHistory = 0, name = '%s_NurbsIntersect_geo' % boatName)
        debug(None, method = 'buildOceanNurbsPreviewPlane', message = 'Built NURBSPlane: %s' % NURBSPlane, verbose = False)

        ## Now set the attrs of the NURBSPlane
        cmds.setAttr ( "%s.rotate" % NURBSPlane[0], lock= True)
        cmds.setAttr ("%s.scaleY" % NURBSPlane[0], lock= True)
        cmds.setAttr ("%s.scaleX" % NURBSPlane[0], size)
        cmds.setAttr ("%s.scaleZ" % NURBSPlane[0], size)
        cmds.setAttr ( "%s.translateY" % NURBSPlane[0] , lock= True)
        #cmds.setAttr (NURBSPlane[0] + ".visibility",0)
        cmds.setAttr ("%s.primaryVisibility" % NURBSPlane[0],0)
        cmds.setAttr ("%s.visibleInReflections" % NURBSPlane[0],0)
        cmds.setAttr ("%s.visibleInRefractions" % NURBSPlane[0],0)
        debug(None, method = 'buildOceanNurbsPreviewPlane', message = 'All Attrs set correctly', verbose = False)
        ## Cleanup the build
        _cleanupNURBSPlane(NURBSPlane, boatName)
        debug(None, method = 'buildOceanNurbsPreviewPlane', message = 'Cleanup Performed Successfully', verbose = False)
        ## Build the expression
        _buildNURBSPlaneExpression(NURBSPlane, xRes, zRes, oceanShader)
        debug(None, method = 'buildOceanNurbsPreviewPlane', message = 'Expression Built Successfully', verbose = False)
Exemplo n.º 7
0
def keyToOn():
    controller = mc.ls(sl=True)[0]

    channelBox = mel.eval('global string $gChannelBoxName; $temp=$gChannelBoxName;')	#fetch maya's main channelbox
    attrs = mc.channelBox(channelBox, q=True, sma=True)

    if attrs:
        channel = controller + "." + attrs[0]
        constraint = mc.listConnections(channel, type="constraint")[0]
        currentTime = mc.currentTime( query=True )

        # save rest pose on constraint
        # mc.currentTime(currentTime-1)
        constraintChannels = mc.listConnections(constraint, type="animCurve")
        mc.parentConstraint(controller, constraint, edit=True, maintainOffset=True)
        for i in constraintChannels:
            value = mc.getAttr(mc.listConnections(i, plugs=True)[0])
            setKeyframe(mc.listConnections(i, plugs=True)[0], currentTime, value)

        # set key on 1 for controller
        setKeyframe(channel, currentTime-1, value=0)
        setKeyframe(channel, currentTime, value=1)

        # mc.currentTime( currentTime )
    else:
        mc.warning("Select constrain channel!")
Exemplo n.º 8
0
def copy_user_attr(selparentshapes, seltargetshapes, copyUV=True):
    listattrvalue = {}
    listdatatype = {}
    userattr = cmds.listAttr(selparentshapes, ud=True)
    if copyUV and cmds.nodeType(seltargetshapes) == 'mesh' and cmds.nodeType(selparentshapes) == 'mesh':
            cmds.transferAttributes(selparentshapes, seltargetshapes, transferUVs=1, transferColors=2, searchMethod=3, sampleSpace=5, flipUVs=False)
    if userattr:
        for attr in userattr:
            nodetype = cmds.nodeType(selparentshapes)
            checkrendershape = cmds.getAttr(selparentshapes + '.intermediateObject')
            if checkrendershape != 1 or nodetype != 'mesh':
                key = attr
                value = cmds.getAttr("%s.%s" % (selparentshapes, key))
                data = cmds.getAttr("%s.%s" % (selparentshapes, key), typ=True)
                listattrvalue[key] = value
                listdatatype[key] = data
    checkrendershape = cmds.getAttr(seltargetshapes + '.intermediateObject')
    if checkrendershape != 1:
        for key in listattrvalue:
            if not cmds.attributeQuery(key, node=seltargetshapes, ex=True):
                if listdatatype[key] == 'string':
                    cmds.addAttr(seltargetshapes, longName=key, dataType=listdatatype[key])
                    cmds.setAttr("%s.%s" % (seltargetshapes, key), listattrvalue[key], type=listdatatype[key])
                else:
                    cmds.addAttr(seltargetshapes, longName=key)
                    cmds.setAttr("%s.%s" % (seltargetshapes, key), listattrvalue[key])
            else:
                cmds.warning('Attribute ' + key + ' already on ' + seltargetshapes)
                if cmds.getAttr("%s.%s" % (seltargetshapes, key), se=True):
                    if listdatatype[key] == 'string':
                        cmds.setAttr("%s.%s" % (seltargetshapes, key), listattrvalue[key], type=listdatatype[key])
                    else:
                        cmds.setAttr("%s.%s" % (seltargetshapes, key), listattrvalue[key])
Exemplo n.º 9
0
 def saveFileToPublish(self):
     self.currentFilePath = cmds.file(q=1, sn=1)
     if self.currentFilePath:
         if '/work/' in self.currentFilePath:
             self.baseName = os.path.basename(self.currentFilePath)
             self.publishPath = self.currentFilePath.split(self.baseName)[0].replace('/work/', '/publish/')
             self.publishFilePath = os.path.join(self.publishPath, self.baseName)
             if os.listdir(self.publishPath):
                 fileName = sorted(os.listdir(self.publishPath))[-1].replace('.ma', '.mb')
                 newPublishFilePath = os.path.join(self.publishPath, fileName)
                 curVersion = os.path.splitext(fileName)[0].split('.v')[-1]
                 newVersion = int(curVersion) + 1
                 newVersionStr = '.v%03d' % newVersion
                 self.publishFilePath = newPublishFilePath.replace('.v%03d' % int(curVersion), newVersionStr)
                 cmds.file(f=1, save=1)
                 cmds.file(rename=self.publishFilePath)
                 cmds.file(f=1, save=1, type = 'mayaBinary')
                 cmds.file(self.currentFilePath, f=1, open=1)
                 return self.publishFilePath
             else:
                 fileName = self.baseName.replace('.ma', '.mb')
                 newPublishFilePath = os.path.join(self.publishPath, fileName)
                 self.publishFilePath = newPublishFilePath
                 cmds.file(f=1, save=1)
                 cmds.file(rename=self.publishFilePath)
                 cmds.file(f=1, save=1, type = 'mayaBinary')
                 cmds.file(self.currentFilePath, f=1, open=1)
                 return self.publishFilePath
         else:
             cmds.confirmDialog(m="File is not in work directory.. Kindly save your file in work directory and run this script", b="Ok")
     else:
         cmds.warning('Cannot find proper file. Please save our file and proceed....')
Exemplo n.º 10
0
    def doIt(self, argList):
        # get objects from argument list (size >= 2)
        try:
            obj = misc.getArgObj(self.syntax(), argList)
            if (len(obj) < 2):
                cmds.error("Select at least 2 objects!")
                return
            if (cmds.objectType(obj[0]) != 'transform' or cmds.objectType(obj[1]) != 'transform'):
                cmds.error("Object is not of type transform!")
                return
        except:
            cmds.warning("No objects selected or only one object given!")
            return

        argData = om.MArgParser(self.syntax(), argList)

        # read all arguments and set default values
        keepCD = argData.flagArgumentBool('keepConvexDecomposition', 0) if (argData.isFlagSet('keepConvexDecomposition')) else True
        nvol = argData.flagArgumentDouble('normVolume', 0) if (
            argData.isFlagSet('normVolume')) else 100.0
        norm = argData.flagArgumentBool('norm', 0) if (
            argData.isFlagSet('norm')) else True
        s_file = os.path.abspath(argData.flagArgumentString('saveFile', 0)) if (argData.isFlagSet('saveFile')) else ""
        save = False
        if s_file != "":
            o_file = open(s_file, 'w')
            o_file.write("i0, i1, name0, name1, its_volume, dice\n")
            save = True

        # get all flags for vhacd over static parsing method
        vhacd_par = vhacd.vhacd.readArgs(argData)
        if (vhacd_par is None):
            cmds.error("V-HACD: one or more arguments are invalid!")
            return

        #norm the volume
        if norm:
            for o in obj:
                v = cmds.getVolume(o)
                scale_factor = (nvol/ v) ** (1./3)
                # cmds.xform(o, scale=[scale_factor, scale_factor, scale_factor])
                cmds.scale(scale_factor, scale_factor, scale_factor, o, relative=True)
                cmds.makeIdentity(o, apply=True)

        intersection_matrix = np.matrix(str(cmds.intersection(obj, kcd=keepCD, matlabOutput=True, **vhacd_par)))
        dice_matrix = np.matrix(intersection_matrix)

        for i in range(0, intersection_matrix.shape[0]):
            for j in range(i, intersection_matrix.shape[1]):
                its_volume = intersection_matrix[i, j]
                dice_matrix[i, j] = (its_volume*2)/(intersection_matrix[i, i]+intersection_matrix[j, j])
                if save:
                    o_file.write(",".join((str(i), str(j), obj[i], obj[j], str(its_volume), str(dice_matrix[i, j]))))
                    o_file.write("\n")
                    o_file.flush()

        if save:
            o_file.close()

        self.setResult(str(dice_matrix))
Exemplo n.º 11
0
    def get_num_cvs(self, curve):
        '''
        Desc:
        Get cvs lenght from a curve

        Parameter:
        curve = curve to get cvs positin list from
        coords_space = the coordinat space, can be "world" (default), or "local"

        Return:
        list with positions
        '''

        # If curve is nod define or not correct release error
        if curve:
            # Get curve shape
            curve_shape = KstMaya.get_shape_node(curve)

            # Get degree
            degree = cmds.getAttr(curve_shape+".degree")

            # Get spans
            spans = cmds.getAttr(curve_shape+".spans")

            # Calulating ncvs with formula spans+degree
            ncvs = spans+degree

            # Return the list
            return ncvs
        else:
            cmds.warning("Curve %s,  is not defined, or is not a curve, double check!" % curve)
            return None
Exemplo n.º 12
0
def smarterActivateSet(*args):
    set = GetSetOptions()
    # print set.options, '_________'
    opt = str(len(set.options) - 1)
    if len(set.sel) != 0:
        if len(set.sel) > 1:  # multi object selection
            activateSet(set.sel)
            cmds.warning('-- multiple objects set as character set --')
            return set.sel
        if set.current:
            if set.current in set.options:
                idx = set.options.index(set.current)
                if idx != len(set.options) - 1:
                    activateSet(set.options[idx + 1])
                    message('set --   ' + set.options[idx + 1] + '     -- ' + str(idx + 1) + ' of ' + opt, maya=True)
                else:
                    activateSet(set.options[0])
                    message('set --   ' + set.options[0] + '     -- ' + str(0) + ' of ' + opt, maya=True)
            else:
                activateSet(set.options[1])
                message('set --   ' + set.options[1] + '     -- ' + str(1) + ' of ' + opt, maya=True)
        else:
            activateSet(set.options[1])
            message('set --   ' + set.options[1] + '     -- ' + str(1) + ' of ' + opt, maya=True)
    else:
        activateSet(set.options[0])
Exemplo n.º 13
0
def publish():
	# Import Modules
	import shutil

	# Get SceneName and Root
	fullName = cmds.file(sceneName=True, q=True)
	paths = fullName.split("/")

	taskName = paths[-2].split("_")[2]
	assetCode = paths[-2].split("_")[1]
	assetName = Assets.getFullName(assetCode)

	outFolder =  "/".join(paths[:-1]) + "/" + assetCode + "_" + taskName + "_OUT"
	outName = assetName + "_" + taskName

	cmds.file( save=True, type='mayaAscii' )					# Save File
	shutil.copy2(fullName, outFolder + "/" + outName + ".ma")	# Copy File to MASTER
	cmds.warning("[Kroentlied Pipeline] Published !")

	# Copy File to BackUp
	oldFolder = outFolder + "/" + assetCode + "_" + taskName + "_OUT_OLD"
	backup = VersionControl.getLatest(oldFolder, 1)

	if not backup:	# No Backup found yet
	    backup = outName + "_BackUp_v001.ma"

	shutil.copy2(fullName, oldFolder + "/" + backup)
	print "[Kroentlied Pipeline] PublishBackup: " + backup
	return
def setPlanes(*args):
    """sets clipping planes for cameras based on float fields in UI. Depending on radio button, it will either do all camera or only selected"""

    all = cmds.radioButtonGrp("camRBG", q=True, sl=True)
    far = cmds.floatFieldGrp("farFFG", q=True, v1=True)
    near = cmds.floatFieldGrp("nearFFG", q=True, v1=True)

    cams = []
    if all==1:
        cams.extend(cmds.ls(type="camera"))
    elif all==2:
        transf = cmds.ls(sl=True, type="transform")
        for each in transf:
            shape = cmds.listRelatives(each, s=True)
            if shape:
                if cmds.objectType(shape) == "camera":
                    cams.extend(shape)
    #for each, set shape.farClipPlane 100000
    if cams:
        print cams
        for cam in cams:
            try:
                cmds.setAttr("%s.farClipPlane"%cam, far)
                cmds.setAttr("%s.nearClipPlane"%cam, near)

            except:
                cmds.warning("Couldn't change the farClipPlane of %s"%cam)
Exemplo n.º 15
0
 def offsetCurveDragCmd(self, *args ):
     
     if not self._dragStart:
         
         selCurves = cmds.ls( sl=1 )
     
         self._targetCurves = []
         self._keepTargetCurveValues = []
         for curve in selCurves:
             if cmds.attributeQuery( 'centerRate' , node = curve, ex=1 ):
                 self._targetCurves.append( curve )
                 self._keepTargetCurveValues.append( cmds.getAttr( curve+'.centerRate' ) )
         
         if not self._targetCurves:
             cmds.warning( "Select VolumeCurves" )
             return None
         
         self._dragStart = True
         cmds.undoInfo( swf=0 )
         
     multRate = cmds.floatSliderGrp( self._offsetCurve, q=1, v=1 )
         
     for i in range( len( self._targetCurves ) ):
         targetCurve = self._targetCurves[i]
         value       = self._keepTargetCurveValues[i]
         
         cmds.setAttr( targetCurve+'.centerRate', value*multRate )
Exemplo n.º 16
0
def getFaceCam():
    
    meta=generic.Meta()
    
    sel=cmds.ls(selection=True)
    
    faceCam=''
    
    if len(sel)<1:
        # warning if nothing is selected
        cmds.warning('Nothing is selected! Select a control first.')
        
        return
    else:
        node=sel[0]
        
        # warning if selection is not a control type
        if meta.getData(node)['type']!='control':
            cmds.warning('Selection is not a control! Select a control first.')
            
            return
        else:
            # getting root node
            root=meta.upStream(node,'root')
            
            # getting face camera from all cameras
            cameras=meta.downStream(root, 'camera',allNodes=True)
            for camera in cameras:
                if meta.getData(camera)['component']=='face':
                    faceCam=cmds.listConnections('%s.message' % camera,type='transform')[0]
    
    return faceCam
Exemplo n.º 17
0
def TappInstall_browse(*args):
    repoPath=cmds.fileDialog2(dialogStyle=1,fileMode=3)
    
    if repoPath:
        repoPath=repoPath[0].replace('\\','/')
        
        check=False
        #checking all subdirectories
        for name in os.listdir(repoPath):
            
            #confirm that this is the Tapp directory
            if name=='Tapp':
                check=True
        
        if check:
            #create the text file that contains the Tapp directory path
            path=cmds.internalVar(upd=True)+'Tapp.yml'
            
            f=open(path,'w')
            data='{launchWindowAtStartup: False, repositoryPath: \''+repoPath+'\'}'
            f.write(data)
            f.close()
    
            #run setup
            sys.path.append(repoPath)
            
            cmds.evalDeferred('import Tapp')
            
            #delete ui
            cmds.deleteUI('TappInstall_UI')
            
        else:
            cmds.warning('Selected directory is not the \'Tapp\' directory. Please try again')
Exemplo n.º 18
0
    def doIt(self, argList):
        # get only the first object from argument list
        try:
            obj = misc.getArgObj(self.syntax(), argList)[0]
        except:
            cmds.warning("No object selected!")
            return
        if (cmds.objectType(obj) != 'transform'):
            cmds.error("Object is not of type transform!")
            return

        # get and cast array of MPoint-Objects to list structure (needed for ConvexHull)
        pSet = list([p.x, p.y, p.z] for p in misc.getPoints(obj))
        # execute convex hull algorithm on point set of current object
        hull = ConvexHull(pSet)
        # get empty mesh object
        mesh = om2.MFnMesh()
        # add each polygon (triangle) in incremental process to mesh
        center = cmds.centerPoint(obj)
        for i,tri in enumerate(hull.points[hull.simplices]):
            # triangle vertices are not ordered --> make sure all normals point away from object's center
            v1 = map(operator.sub, tri[0], tri[1])
            v2 = map(operator.sub, tri[0], tri[2])
            # cross product of v1 and v2 is the current face normal; if dot product with vector from origin is > 0 use vertex order 0,1,2 and if dot product < 0 reverse order (all triangles are defined clockwise when looking in normal direction -- counter-clockwise when looking onto front of face)
            if (np.dot(np.cross(v1, v2), map(operator.sub, tri[0], center) ) > 0 ):
                mesh.addPolygon( ( om2.MPoint(tri[0]), om2.MPoint(tri[1]), om2.MPoint(tri[2]) ) )
            else:
                mesh.addPolygon( ( om2.MPoint(tri[0]), om2.MPoint(tri[2]), om2.MPoint(tri[1]) ) )
        # get transform node of shapeNode and rename it to match object's name
        transformNode = cmds.listRelatives(mesh.name(), p = 1)
        transformNode = cmds.rename(transformNode, obj + "_ch")
        self.setResult(transformNode)
Exemplo n.º 19
0
def _getFrames( frameRange ):
    fr = _getTextFieldValue( frameRange )
    frames = []
    
    if not re.search('^\d+(\s*[-,]?\s*\d+)*$', fr):
        _invalidFrameRange()
    
    fr = fr.replace(' ', '')
    frameRanges = fr.split(',')
    
    for f in frameRanges: 
        if re.search('^[0-9]+[-][0-9]+$', f):
            start, end = f.split('-')
            
            if int(start) > int(end):
                cmds.warning( 'Start of range is greater than end. Will not create any frames.' )
            
            for i in range(int(start), (int(end)+1)):
                frames.append(i)
        elif re.search('^[0-9]+$', f):
            frames.append( re.search('^[0-9]+$', f).group() )
        else:
            _invalidFrameRange()
    
    if len(frames) == 0:
        _invalidFrameRange()
    
    return frames
Exemplo n.º 20
0
 def buttonReplacePhClicked(self):
     itemList = self.ui.placeholderList.selectedItems()
     selection = cmds.ls(sl=True, tr=True)
     if len(itemList) > 0 and len(selection) > 0:
         replacePHs(str(itemList[0].text()))
     else:
         cmds.warning("Please select a placeholder from the list.")
	def reverseSelected(self,*args):
		selected = cmds.ls( sl=True )
		reverseSelected = cmds.ls( sl=True )

		parentResult = cmds.listRelatives( selected[0], parent=True )

		if parentResult is not None:
			parent = parentResult[0]
			children = cmds.listRelatives( parent, children=True, pa=True )

			reverseSelected.reverse()
			reverseIndex = 0

			numberOfSelected = len(selected)

			for item in children:
				if reverseIndex < numberOfSelected and item == selected[reverseIndex]:
					cmds.reorder( reverseSelected[reverseIndex], back=True )
					reverseIndex = reverseIndex+1
				else:
					cmds.reorder( item, back=True )

			cmds.select( reverseSelected )
		else:
			cmds.warning( 'Selected objects are parented to the World, they need to be children of a transform' )
Exemplo n.º 22
0
def symmetry():

    #getting nodes
    sel = cmds.ls(selection=True)
    if len(sel) == 1:
        cmds.warning('Select the edited mesh first, then the original mesh last!')
        return
    if len(sel) > 2:
        cmds.warning('Select the edited mesh first, then the original mesh last!')
        return

    edit = sel[0]
    original = sel[1]

    #creating symmetry mesh
    mirror = mirrorBlendshape([edit], original)[0]
    copy = cmds.duplicate(original)[0]
    bldShp = cmds.blendShape(edit, mirror, copy)[0]
    cmds.setAttr('%s.%s' % (bldShp, edit), 1)
    cmds.setAttr('%s.%s' % (bldShp, mirror), 1)
    symmetry = cmds.duplicate(copy)[0]
    symmetry = cmds.rename(symmetry, edit + '_symmetry')

    cmds.delete(copy, mirror)

    return symmetry
Exemplo n.º 23
0
 def cb_drawLines(self, x=None) :
     if None in [ self.markerset, self.prefix ] : 
         m.warning("No markerset or prefix")
         return
     
     print "drawing lines for " + str(self.prefix)
     self.markerset.drawLines( self.prefix )
Exemplo n.º 24
0
Arquivo: mdc.py Projeto: ivanzotov/mdc
def prepare(setname="cache_set", groupname="cache_group", prefix="cache_"):
    selection = cmds.ls(sl=True, l=True)
    
    if len(selection) == 0:
        cmds.warning("Please select objects!")
        return
        
    cmds.duplicate()
    dup = cmds.ls(sl=True, l=True)
    
    if cmds.objExists(groupname):
        cmds.parent(dup, groupname)
    else:
        cmds.group(w=True, n=groupname)
    
    cmds.select(groupname)
    dup = get_shapes()
    add_to_set_of_shapes(setname, dup)
    
    i = 0
    for obj in dup:
        cmds.blendShape([selection[i], obj], w=(0, 1),o="world", foc=True)
        objects = []
        objects.append(obj)
        addattr(objects, prefix+str(i))
        i=i+1
Exemplo n.º 25
0
Arquivo: gui.py Projeto: Bumpybox/Tapp
 def on_getObjectRegion_pressed(self):
     
     sel=cmds.ls(selection=True)
     
     #checking selection for meshes
     check=False
     for node in sel:
         
         shape=cmds.listRelatives(node,shapes=True)
         if shape:
             
             if cmds.nodeType(shape)=='mesh':
                 
                 check=True
     
     if check:
     
         regions=utils.getMeshAnimation()
         
         regionNode=self.getSelectedRegionNode()
         
         for r in regions:
             
             utils.setRegionNode(regionNode, r)
     
     else:
         
         cmds.warning('Nothing selected. Please select one or more meshes!')
Exemplo n.º 26
0
 def window(dismiss):
     if dismiss == "load":                 
         self.loadData()
     else:                 
         cmds.warning("If you want to load later, go to aTools menu/Animation Crash Recovery option box")
     
     self.saveBackup()           
Exemplo n.º 27
0
def rootsOnPath(path='X:/_image/_projects/SI/HOL/_Assets/ScareCrow/scenes/Anim_Asset/Path_v01.ma', reverse=False):
    allLocs = []
    roots = rootList()
    if not nsPaths():
        cn.uiEnable(controls='modelPanel')
        roots = nsHollow(roots)
        for root in roots:
            nsA = nsPaths()
            cmds.file(path, r=True, type="mayaAscii", gl=True, loadReferenceDepth="all", mergeNamespacesOnClash=False, namespace="Path_v01")
            nsB = nsPaths()
            ns = list(set(nsB) - set(nsA))[0]
            # segments off
            segmentsOff(ns=ns)
            cmds.select(root)
            cmds.select(ns + ':path', add=True)
            # pose the path
            pathPose(root=root, ns=ns)
            # attach
            allLocs.append(attach(up=ns + ':up', reverse=reverse))
            # break
        cn.uiEnable(controls='modelPanel')
        for locs in allLocs:
            cmds.select(locs, add=True)
    else:
        cmds.warning('--  remove all path rigs before starting  --')
Exemplo n.º 28
0
    def doIt(self, argList):
        # get only the first object from argument list
        try:
            obj = misc.getArgObj(self.syntax(), argList)[0]
        except:
            cmds.warning("No object selected!")
            return
        if (cmds.objectType(obj) != 'transform'):
            cmds.error("Object is not of type transform!")
            return
        # parse arguments and get flags
        argData = om.MArgParser (self.syntax(), argList)
        axisOrder = argData.flagArgumentString('axisOrder', 0) if (argData.isFlagSet('axisOrder')) else 'yzx'
        fast = argData.flagArgumentBool('fast', 0) if (argData.isFlagSet('fast')) else False
        # get eigenvectors as matrix (column-wise), reshape matrix and append extra row/column
        eig = np.matrix(cmds.eigenvector(obj, ao = axisOrder, f = fast))
        eig.shape = (3,3)
        eig = np.append(eig, [[0,0,0]], axis = 0)
        eig = np.append(eig.transpose(), [[0,0,0,1]], axis = 0).transpose()

        # return 4x4 matrix as 16 float values
        util = om.MScriptUtil()
        util.createFromList(eig.getA1().tolist(), eig.size)

        self.setResult(om.MDoubleArray(util.asDoublePtr(), eig.size))
Exemplo n.º 29
0
 def loadSel(self,tF,loadNum):
     sels = cmds.ls(sl = True)
     if len(sels) == loadNum:
         sel = sels[0]
         cmds.textField(tF,e = True,text = str(sel))
     else:
         cmds.warning ('place Load --' + str(loadNum)  + ' --objects')
Exemplo n.º 30
0
def getNodeType(node=cmds.ls(sl=True, long=True)):
    if node:
        nodeShape = cmds.listRelatives(node[0], shapes=True, f=True)[0]
        return cmds.nodeType(nodeShape)
    else:
        cmds.warning('Please select a node')
        return None
Exemplo n.º 31
0
    def set_attr_enhanced(self, obj, attribute, new_value):
        ''' 
        Set attribute for the provided object using different methods depending on its type
        
                Parameters:
                    obj (string): Name of the node/object.
                    attribute (string): Name of the attribute to set. E.g. ".cacheFile"
                    new_value (string): New value to update
        '''

        #print(obj + ' ' + attribute  + ' ' + new_value) # Debugging

        if cmds.objExists(obj):
            obj_type = cmds.objectType(obj) or ''
        else:
            obj_type = ''

        complex_types = ['cacheFile', 'reference']

        if obj_type not in complex_types:
            cmds.setAttr(obj + attribute, new_value, type='string')
        else:
            if obj_type == 'cacheFile':
                format_path = os.path.splitext(new_value)[0].replace("\\", "/")
                file_name = format_path.split('/')[-1]
                format_path_no_file = format_path[::-1].split("/", 1)[-1][::-1]

                try:
                    if os.path.isfile(format_path_no_file + '/' +
                                      file_name.replace('.xml', '') + '.xml'):
                        cmds.setAttr(obj + '.cachePath',
                                     format_path_no_file,
                                     type='string')
                        cmds.setAttr(obj + '.cacheName',
                                     file_name,
                                     type='string')
                        return True
                    else:
                        return False
                except:
                    return False

            if obj_type == 'reference':
                not_skipped = True
                try:
                    cmds.referenceQuery(obj, isLoaded=True)
                except:
                    not_skipped = False

                if not_skipped:
                    if os.path.isfile(new_value):
                        try:
                            cmds.file(new_value, loadReference=obj)
                        except:
                            return False
                    else:
                        cmds.warning(
                            'Provided reference path : "' + new_value +
                            '" doesn\'t lead to a valid file. Previous path was retained.'
                        )
                else:
                    cmds.warning('Reference file inaccessible.')
Exemplo n.º 32
0
 def _call_pick_attribute(self):
     plugs = list_channelbox_highlited_plugs()
     if not plugs:
         return cmds.warning('No plug selected in channelbox')
     self._attribute.setText(plugs[-1])
     self.update_wedging_tabs_states()
Exemplo n.º 33
0
def warningFunc(key, *args):
    message = MESSAGE[key]
    title = TITLE[key]
    cmds.confirmDialog(t=title, m=message, b='OK')
    cmds.warning(message)
    logger.info(message)
Exemplo n.º 34
0
def get_anim_curves_from_objects(nodes):
    """ Gets the animation curves connected to nodes.
    
    :param nodes: List with MFnDependencyNode
    :type nodes: list of om.MFnDependencyNode
    :return: Tuple of curves and plugs
    :rtype: (list of om.MFnDependencyNode, list of om.MPlug)
    """

    curves = []
    plugs = []
    channelbox_attr = get_channelbox_attributes()

    animlayers.cache.reset(
    )  # always reset cache before querying for animation layers!
    has_anim_layers = animlayers.has_anim_layers()

    if has_anim_layers and animlayers.all_layers_locked():
        cmds.warning('All animation layers are locked!')

    # get curves
    for node in nodes:
        # get all attributes
        attr_count = node.attributeCount()
        for index in range(attr_count):
            attr = node.attribute(index)
            plug = node.findPlug(attr, True)

            if plug.isLocked or not plug.isKeyable:
                continue

            connections = plug.connectedTo(True, False)

            # if the attribute has a connection
            if connections:
                conn_node = connections[0].node()

                api = conn_node.apiType()
                if api in ANIM_CURVE_TYPES:
                    # filter out attributes not selected in channelbox
                    if channelbox_attr:
                        attr_name = om.MFnAttribute(attr).shortName
                        if attr_name not in channelbox_attr:
                            continue

                    # add the node if it matches one of the types we want
                    curves.append(om.MFnDependencyNode(conn_node))
                    plugs.append(plug)

                # find curve in animation layer
                elif has_anim_layers and api in animlayers.BLEND_NODE_TYPES:
                    # filter out attributes not selected in channelbox
                    if channelbox_attr:
                        attr_name = om.MFnAttribute(attr).shortName
                        if attr_name not in channelbox_attr:
                            continue

                    # for testing purposes
                    # print('Attribute: %s' % plug)

                    # benchmark_start = time.clock()
                    best_layer = animlayers.get_best_layer(plug)
                    if not best_layer:
                        continue

                    # for testing purposes
                    # try:
                    #     print('-> Best layer is %s' % (om.MFnDependencyNode(best_layer).name()))
                    # except Exception as e:
                    #     pass

                    curve_node = animlayers.get_anim_curve(plug, best_layer)
                    # animlayers.cache.benchmark += time.clock() - benchmark_start
                    if curve_node:
                        curves.append(om.MFnDependencyNode(curve_node))
                        plugs.append(plug)

    # sys.stdout.write('# Retrieved %d curves in %.4f sec\n' % (len(curve_list), animlayers.cache.benchmark))
    return curves, plugs
Exemplo n.º 35
0
def readCoreData(pathToUVXML='', parentGrp=''):

    tree = ET.parse(pathToUVXML)
    root = tree.getroot()

    #debug(None, method = 'care_archive_readXML.readCoreData', message = '{0:<10}{1}'.format('ASSETNAME: ',  root.tag), verbose = False)
    if not cmds.objExists(root.tag):
        cmds.group(n=root.tag, em=True)

    ## CHeck for this assets core archive group
    if parentGrp:
        assetCoreArchive = '|%s|%s|geo_hrc|%s_CORE_ARCHIVES_hrc' % (
            parentGrp, root.tag, root.tag)
    else:
        assetCoreArchive = '%s|geo_hrc|%s_CORE_ARCHIVES_hrc' % (root.tag,
                                                                root.tag)

    coreArchiveGroupName = '%s_CORE_ARCHIVES_hrc' % root.tag

    if cmds.objExists(assetCoreArchive):
        try:
            cmds.delete(assetCoreArchive)
        except RuntimeError:
            pass

    ## Now build a fresh group
    cmds.group(n=coreArchiveGroupName, em=True)
    cmds.parent(coreArchiveGroupName, '%s|%s|geo_hrc' % (parentGrp, root.tag))

    ## Now process the rest of the xml
    geoToProcess = root.getchildren()
    if not geoToProcess:
        cmds.warning('THERE IS NO DATA WRITTEN TO FILE: %s' % pathToUVXML)
    else:
        ## Process each geos data now
        for eachGeo in geoToProcess:
            coreData = eachGeo.getchildren()
            path = coreData[1].attrib['value'].replace('_iColoni_',
                                                       ':').replace(
                                                           '_iFwdSlashi_', '/')
            translateData = coreData[2].attrib['value'].split(' ')
            rotateData = coreData[3].attrib['value'].split(' ')
            scaleData = coreData[4].attrib['value'].split(' ')

            ## Get the parent group
            parent = '%s_%s' % (root.tag, coreData[0].attrib['value'])
            if not cmds.objExists('%s|%s' % (assetCoreArchive, parent)):
                cmds.group(n=parent, em=True)
                cmds.parent('|%s' % parent, assetCoreArchive)

            ## Now process the GEO
            geoName = '%s_%s' % (root.tag, eachGeo.tag.replace(
                '_iPipei_', '|').split('|')[-1])
            longGeoName = '%s|%s|%s' % (assetCoreArchive, parent, geoName)

            if not cmds.objExists(longGeoName):
                try:
                    cmds.group(n=geoName, em=True)
                    cmds.parent('|%s' % geoName,
                                '%s|%s' % (assetCoreArchive, parent))
                except:
                    cmds.warning(
                        'Failed to create group for core archive, missing assets parent group!'
                    )

            cmds.setAttr('%s.translate' % longGeoName, float(translateData[0]),
                         float(translateData[1]), float(translateData[2]))
            cmds.setAttr('%s.rotate' % longGeoName, float(rotateData[0]),
                         float(rotateData[1]), float(rotateData[2]))
            cmds.setAttr('%s.scale' % longGeoName, float(scaleData[0]),
                         float(scaleData[1]), float(scaleData[2]))
            cmds.setAttr('%s.scale' % longGeoName, float(scaleData[0]),
                         float(scaleData[1]), float(scaleData[2]))
            if not cmds.objExists('%s.mcAssArchive' % longGeoName):
                cmds.addAttr(longGeoName, ln='mcAssArchive', dt='string')
            cmds.setAttr('%s.mcAssArchive' % longGeoName, path, type='string')
Exemplo n.º 36
0
    def update(self, container, representation):

        import os
        from maya import cmds

        node = container["objectName"]

        path = api.get_representation_path(representation)

        # Get reference node from container members
        members = cmds.sets(node, query=True, nodesOnly=True)
        reference_node = self._get_reference_node(members)

        file_type = {
            "ma": "mayaAscii",
            "mb": "mayaBinary",
            "abc": "Alembic"
        }.get(representation["name"])

        assert file_type, "Unsupported representation: %s" % representation

        assert os.path.exists(path), "%s does not exist." % path

        # Need to save alembic settings and reapply, cause referencing resets
        # them to incoming data.
        alembic_attrs = ["speed", "offset", "cycleType"]
        alembic_data = {}
        if representation["name"] == "abc":
            alembic_nodes = cmds.ls("{}:*".format(members[0].split(":")[0]),
                                    type="AlembicNode")
            if alembic_nodes:
                for attr in alembic_attrs:
                    node_attr = "{}.{}".format(alembic_nodes[0], attr)
                    alembic_data[attr] = cmds.getAttr(node_attr)
            else:
                cmds.warning("No alembic nodes found in {}".format(
                    cmds.ls("{}:*".format(members[0].split(":")[0]))))

        try:
            content = cmds.file(path,
                                loadReference=reference_node,
                                type=file_type,
                                returnNewNodes=True)
        except RuntimeError as exc:
            # When changing a reference to a file that has load errors the
            # command will raise an error even if the file is still loaded
            # correctly (e.g. when raising errors on Arnold attributes)
            # When the file is loaded and has content, we consider it's fine.
            if not cmds.referenceQuery(reference_node, isLoaded=True):
                raise

            content = cmds.referenceQuery(reference_node,
                                          nodes=True,
                                          dagPath=True)
            if not content:
                raise

            self.log.warning("Ignoring file read error:\n%s", exc)

        # Reapply alembic settings.
        if representation["name"] == "abc":
            alembic_nodes = cmds.ls("{}:*".format(members[0].split(":")[0]),
                                    type="AlembicNode")
            if alembic_nodes:
                for attr, value in alembic_data.items():
                    cmds.setAttr("{}.{}".format(alembic_nodes[0], attr), value)

        # Fix PLN-40 for older containers created with Avalon that had the
        # `.verticesOnlySet` set to True.
        if cmds.getAttr("{}.verticesOnlySet".format(node)):
            self.log.info("Setting %s.verticesOnlySet to False", node)
            cmds.setAttr("{}.verticesOnlySet".format(node), False)

        # Add new nodes of the reference to the container
        cmds.sets(content, forceElement=node)

        # Remove any placeHolderList attribute entries from the set that
        # are remaining from nodes being removed from the referenced file.
        members = cmds.sets(node, query=True)
        invalid = [x for x in members if ".placeHolderList" in x]
        if invalid:
            cmds.sets(invalid, remove=node)

        # Update metadata
        cmds.setAttr("{}.representation".format(node),
                     str(representation["_id"]),
                     type="string")
Exemplo n.º 37
0
def exportFile(outputPath, topNode):
    mc.warning('export start')
    mc.select(topNode)
    mc.file(outputPath, typ='mayaAscii', f=True, es=True, pr=True)
    mc.warning('export end')
Exemplo n.º 38
0
    def create(self):

        import ui_menuEditor

        try:
            self.clearMenu()
        except:
            cmds.warning("Clear Mesh Error")
        try:
            self.clearPopupCa()
        except:
            cmds.warning("Clear Popup(Ctrl + Alt) Error")
        try:
            self.clearPopupSa()
        except:
            cmds.warning("Clear Popup(Shift + Alt) Error")

        if cmds.menu(self.mainMenu, q=1, ex=1):
            cmds.deleteUI(self.mainMenu)
        cmds.menu(self.mainMenu, p='MayaWindow', l=self.label, to=1)

        cmds.menuItem(l='UI - List Menu Paths',
                      p=self.mainMenu,
                      c=loadMenuPathEditor,
                      image=os.path.dirname(__file__) + '/icons/list.png')
        cmds.menuItem(l='UI - Menu Editor',
                      p=self.mainMenu,
                      c=ui_menuEditor.show,
                      image='')
        cmds.menuItem(d=1, dl="Main Menu", p=self.mainMenu)

        for menuPath in getMenuPaths(Menu_Global.menuListFile):
            menuChecked = self.isMenuChecked(menuPath)
            dirname = Menu_Global.getMenuLabel(menuPath)
            menuName = cmds.menuItem(l=dirname,
                                     p=self.mainMenu,
                                     cb=menuChecked)
            cmds.menuItem(menuName,
                          e=1,
                          c=partial(self.selectMenu, menuName, menuPath))
            if menuChecked: self.selectMenu(menuName, menuPath)

        cmds.menuItem(d=1,
                      dl="Popup Menu( Shift + alt + RMB )",
                      p=self.mainMenu)
        popupShiftAlt = cmds.menuItem(l="Popup Menu( Shift + alt + RMB )",
                                      p=self.mainMenu,
                                      sm=1)
        cmds.radioMenuItemCollection(p=popupShiftAlt)
        cmds.menuItem(l="Clear", p=popupShiftAlt, rb=1, c=self.clearPopupSa)
        for menuPath in getMenuPaths(Menu_Global.popupListFile_sa):
            popupChecked = self.isPopupChecked(menuPath, 'sa')
            dirname = Menu_Global.getMenuLabel(menuPath)
            popupName = cmds.menuItem(l=dirname,
                                      p=popupShiftAlt,
                                      rb=popupChecked)
            cmds.menuItem(popupName,
                          e=1,
                          c=partial(self.selectPopup, menuPath, 'sa'),
                          p=popupShiftAlt)
            if popupChecked: self.selectPopup(menuPath, 'sa')

        cmds.menuItem(d=1,
                      dl="Popup Menu( Ctrl + alt + RMB )",
                      p=self.mainMenu)
        popupCtrlAlt = cmds.menuItem(l="Popup Menu( Shift + alt + RMB )",
                                     p=self.mainMenu,
                                     sm=1)
        cmds.radioMenuItemCollection(p=popupCtrlAlt)
        cmds.menuItem(l="Clear", p=popupCtrlAlt, rb=1, c=self.clearPopupCa)
        for menuPath in getMenuPaths(Menu_Global.popupListFile_ca):
            popupChecked = self.isPopupChecked(menuPath, 'ca')
            dirname = Menu_Global.getMenuLabel(menuPath)
            popupName = cmds.menuItem(l=dirname,
                                      p=popupCtrlAlt,
                                      rb=popupChecked)
            cmds.menuItem(popupName,
                          e=1,
                          c=partial(self.selectPopup, menuPath, 'ca'))
            if popupChecked: self.selectPopup(menuPath, 'ca')

        cmds.menuItem(d=1, dl="", p=self.mainMenu)
        cmds.menuItem(
            l="Get Icon",
            p=self.mainMenu,
            c=
            "import webbrowser;url = 'http://www.flaticon.com/';webbrowser.open_new_tab( url )",
            image=os.path.dirname(__file__) + '/icons/search.png')
        cmds.menuItem(d=1, dl="", p=self.mainMenu)
        cmds.menuItem(l="Reload Modules", c=reloadModules, p=self.mainMenu)
        cmds.menuItem(d=1, dl="", p=self.mainMenu)
        cmds.menuItem(l="Reload Menu",
                      c=loadMenu,
                      p=self.mainMenu,
                      image=os.path.dirname(__file__) + '/icons/reload.png')
Exemplo n.º 39
0
def import_createHairSystemByCurveAttr():
    
    beforeHairSystem = set( cmds.ls( exactType='hairSystem', l=True) )
    selectedObj = cmds.ls(sl=True, l=True)
    curves = cmds.ls( selectedObj, l=True, exactType='nurbsCurve' )
    temp = cmds.listRelatives( selectedObj, ad=True, f=True, type='nurbsCurve')
    if temp:
        temp = [c for c in temp if not cmds.getAttr( c+".intermediateObject") ]
        curves.extend( temp )
    curves = tuple( set(curves) )
    curvesList = {}
    curves = [c for c in curves if cmds.attributeQuery( 'hairData', node=c, exists=True)]
    for c in curves:
        try:
            hairData = json.loads( cmds.getAttr(c+'.hairData') )
            hairName = hairData[ 'hairSystem']
            if not curvesList.has_key( hairName ):
                curvesList[hairName] = []
        except:
            print cmds.warning( '%s.hairData value loaded failed!'%(c)  )
            continue
        curvesList[hairName].append( c )
    
    numHairSystem = len(curvesList.keys())
    if not numHairSystem:
        return
    
    gMainProgressBar = mel.eval('$tmp = $gMainProgressBar')
    cmds.progressBar( gMainProgressBar, edit=True, beginProgress=True, isInterruptable=True,
                    status='Creating Hair System ...', maxValue=numHairSystem  )
    
    for hairName, curves in curvesList.iteritems():
        cmds.progressBar(gMainProgressBar, edit=True, step=1 )
        if not curves:
            continue
        cmds.select( curves, r=True)
        mel.eval( 'makeCurvesDynamic 2 { "0", "0", "0", "0", "1"}' )
        
    cmds.progressBar(gMainProgressBar, edit=True, endProgress=True)
    
        
    #Transfer attributes from curve to hairSystem and follicle attributes
    objs = cmds.ls(l=True, exactType='hairSystem')
    #folAttrs = ('clumpWidthMult',  'densityMult', 'curlMult', 'clumpTwistOffset', 'colorBlend', 'colorR', 'colorG', 'colorB')
    
    for hair in objs:
        fols = cmds.listConnections( hair+'.inputHair', d=False, shapes=True)
        c = cmds.listConnections( fols[0]+'.startPosition', d=False, shapes=True)[0]
        
        hairData = json.loads( cmds.getAttr( c+'.hairData') )
        hairSysName = hairData.get( 'hairSystem', None)
        if hairSysName and not cmds.attributeQuery('originalName', node=hair, exists=True):
            cmds.addAttr( hair, ln='originalName', dt='string' )
            cmds.setAttr( hair+'.originalName', hairSysName, type='string')
            
            cmds.setAttr( hair+'.active', False)
            cmds.setAttr( hair+'.simulationMethod', 1)
            
        for fol in fols:
            c = cmds.listConnections( fol+'.startPosition', d=False, shapes=True)[0]
            hairData = json.loads( cmds.getAttr( c+'.hairData') )
            hairData.pop( 'hairSystem', None)
            if hairData and not cmds.attributeQuery('oriAttrs', node=fol, exists=True):
                cmds.addAttr(fol, ln='oriAttrs', dt='string' )
                cmds.setAttr( fol+'.oriAttrs', json.dumps(hairData), type='string' )
                
    afterHairSystem = set(  cmds.ls(exactType='hairSystem', l=True)  )
    
    newHairSystems = tuple( afterHairSystem.difference( beforeHairSystem )    )
    
    hairSysGrp = cmds.group( newHairSystems, name='hairsystemGrp_#')
    newHairSystems = cmds.listRelatives(hairSysGrp, ad=True, type='hairSystem', f=True)
    pfxHairs = []
    for hair in newHairSystems:
        pfxHairs.extend( cmds.listConnections( hair+'.outputRenderHairs', s=False) )    
    cmds.group( pfxHairs, name='pfxHairGrp_#' ) 
    
    return newHairSystems
Exemplo n.º 40
0
    def refresh_table(self, is_repair_attempt=False, is_search_replace=False):
        ''' 
        Main Refresh Function
        
                Parameters:
                    is_repair_attempt=False (bool): Is attempting to auto repair paths? (Called by the Auto Path Repair Button)
                    is_search_replace=False (bool): Is it doing a search and replace? (Called by the Search and Replace Button)
        
        '''

        common_locations = []  # Locations where files were found
        is_search_dir_valid = False
        if is_repair_attempt:
            search_dir = self.filepath_le.text()
            if os.path.isdir(search_dir):
                is_search_dir_valid = True
            else:
                cmds.warning(
                    'The search directory doesn\'t exist. Please select a valid path and try again.'
                )

        self.set_cell_changed_connection_enabled(
            False)  # So it doesn't update it unecessarly

        self.table_wdg.setRowCount(0)  # Remove all rows

        # Used to detect installed plugins
        node_types = cmds.ls(nodeTypes=True)

        # Common Nodes
        file_nodes = cmds.ls(type='file')
        path_nodes = file_nodes

        # Available Types
        available_node_types = ['audio', 'cacheFile', 'AlembicNode', 'gpuCache','BifMeshImportNode',\
                           'RedshiftProxyMesh','RedshiftVolumeShape','RedshiftNormalMap','RedshiftDomeLight','RedshiftIESLight', \
                           'MASH_Audio','aiPhotometricLight','aiStandIn','aiVolume', 'imagePlane']

        # Add Types for Loaded Plugins
        path_node_types = []
        for obj_type in available_node_types:
            if obj_type in node_types:
                path_node_types.append(obj_type)

        # Add Extra Nodes to Path Nodes
        for node_type in path_node_types:
            try:
                nodes_list = cmds.ls(type=node_type)
                path_nodes += nodes_list
            except:
                pass

        # Add References
        refs = cmds.ls(rf=True)
        path_nodes += refs

        # Populate Table
        for i in range(len(path_nodes)):

            ################ Start Directory Search ################
            if is_repair_attempt and is_search_dir_valid:
                try:
                    file_items = self.get_path_items(
                        path_nodes[i]
                    )  # (path, is_path_valid, node_type_string, icon, node_attr)
                    progress_bar_name = 'Searching'
                    query_path = file_items[0]
                    initial_result = os.path.exists(query_path)
                    query_path = query_path.replace(
                        '\\', '/')  # Format it - The main Query
                    desired_file = query_path.split('/')[
                        -1]  # Extract file name (short_name)
                    accept_dir = False
                    is_udim_file = False
                    is_image_sequence = False

                    # Check if using UDIMs or Image Sequences
                    if file_items[2] == 'File':
                        try:
                            uv_tiling_mode = cmds.getAttr(
                                path_nodes[i] +
                                '.uvTilingMode')  # Is it using UDIM?
                            use_frame_extension = cmds.getAttr(
                                path_nodes[i] + '.useFrameExtension'
                            )  # Is it an image sequence?
                            is_image_sequence = use_frame_extension
                            if uv_tiling_mode != 0:
                                udim_file_pattern = maya.app.general.fileTexturePathResolver.getFilePatternString(
                                    query_path, use_frame_extension,
                                    uv_tiling_mode)
                                query_path = udim_file_pattern  #.replace('<UDIM>','1001') Handled later using regex
                                is_udim_file = True
                        except:
                            pass

                    # Handle desired folder (instead of file)
                    if file_items[2] == 'Bifrost Cache':
                        if query_path.endswith('/'):
                            desired_file = query_path.split('/')[-2]
                        else:
                            desired_file = query_path.split('/')[-1]
                        accept_dir = True

                    is_found = False
                    if (initial_result != True) and (
                            len(common_locations) != 0
                    ):  # If common locations are available try them first
                        for loc in common_locations:
                            formatted_path = loc.replace("\\", "/")
                            formatted_path = formatted_path[::-1]
                            formatted_path = formatted_path.split("/", 1)[-1]
                            formatted_path = formatted_path[::-1]
                            common_path_result = os.path.exists(
                                formatted_path + "/" + desired_file)
                            if common_path_result == True:
                                resolved_path = (formatted_path + "/" +
                                                 desired_file).replace(
                                                     '/', '\\')
                                #print(path_nodes[i] + ' found using known location.') # Debugging
                                self.set_attr_enhanced(path_nodes[i],
                                                       file_items[4],
                                                       resolved_path)
                                is_found = True

                    # Full Search/Walk
                    if (initial_result != True) and (is_found == False):
                        search_count = 0  # How many folders to look into (walk) for the progress bar
                        for path in os.walk(
                                search_dir
                        ):  # generates the file names in a directory tree by walking the tree either top-b or b-top
                            search_count += 1
                        resolved_path = query_path

                        self.make_progress_bar(
                            progress_bar_name, search_count
                        )  # make_progress_bar(name, maxVal) - Max value is the number of folders
                        for path, dirs, files in os.walk(
                                search_dir
                        ):  # root_dir_path, sub_dirs, files in os.walk(my_dir)
                            self.move_progress_bar(progress_bar_name, 1)
                            path = path.replace('/', '\\')

                            # Handle Files
                            if desired_file in files:
                                resolved_path = (path + '\\' +
                                                 desired_file).replace(
                                                     '/', '\\')
                                common_locations.append(resolved_path)
                                is_found = True

                            # Handle Folders (instead of files)
                            if accept_dir and desired_file in dirs:
                                resolved_path = (path + '\\' +
                                                 desired_file).replace(
                                                     '/', '\\')
                                common_locations.append(resolved_path)
                                is_found = True

                            # Handle UDIMs
                            if is_udim_file and is_found == False:
                                file_name = os.path.splitext(
                                    desired_file)[0].replace('<UDIM>', '')
                                extension = os.path.splitext(desired_file)[1]

                                pattern = re.compile(file_name +
                                                     '\\d\\d\\d\\d' +
                                                     extension)

                                first_found_file = ''

                                if any(pattern.match(line) for line in files):
                                    lines_to_log = [
                                        line for line in files
                                        if pattern.match(line)
                                    ]
                                    first_found_file = lines_to_log[0]

                                if first_found_file != '':
                                    resolved_path = (path + '\\' +
                                                     first_found_file).replace(
                                                         '/', '\\')
                                    if os.path.exists(resolved_path):
                                        common_locations.append(resolved_path)
                                        is_found = True

                            # Handle Image sequences
                            if is_image_sequence and is_found == False:
                                file_name = os.path.splitext(
                                    desired_file)[0].replace('<f>',
                                                             '').replace(
                                                                 '<F>', '')
                                extension = os.path.splitext(desired_file)[1]

                                pattern = re.compile(file_name + '\\d+' +
                                                     extension)

                                first_found_file = ''

                                if any(pattern.match(line) for line in files):
                                    lines_to_log = [
                                        line for line in files
                                        if pattern.match(line)
                                    ]
                                    first_found_file = lines_to_log[0]

                                if first_found_file != '':
                                    resolved_path = (path + '\\' +
                                                     first_found_file).replace(
                                                         '/', '\\')
                                    if os.path.exists(resolved_path):
                                        common_locations.append(resolved_path)
                                        is_found = True
                        if is_found:
                            #print(path_nodes[i] + ' has a valid path.') # Debugging
                            self.set_attr_enhanced(path_nodes[i],
                                                   file_items[4],
                                                   resolved_path)
                    self.kill_progress_window(
                        progress_bar_name)  # Kill progress bar
                except:
                    self.kill_progress_window(progress_bar_name)
            ################ End Directory Search ################

            # Search and Replace
            if is_search_replace:
                try:
                    file_items = self.get_path_items(path_nodes[i])
                    old_path = file_items[0]
                    new_path = old_path.replace(self.search_string,
                                                self.replace_string)
                    self.set_attr_enhanced(
                        path_nodes[i], file_items[4], new_path
                    )  #(path, is_path_valid, node_type_string, icon, node_attr)
                except:
                    pass

            # Refresh Table
            file_items = self.get_path_items(path_nodes[i])
            self.table_wdg.insertRow(i)

            self.table_wdg.setFocusPolicy(QtCore.Qt.NoFocus)  # No highlight

            self.insert_item(i, 1, path_nodes[i], None, path_nodes[i])

            if file_items:  # (path, is_path_valid, node_type_string, icon, node_attr)
                if file_items[1]:
                    self.insert_item(i,
                                     2,
                                     file_items[2],
                                     None,
                                     cmds.objectType(path_nodes[i]),
                                     icon_path=file_items[3],
                                     editable=False)
                    self.insert_icon(i, 0, ':confirm.png')
                else:
                    self.insert_item(i,
                                     2,
                                     file_items[2],
                                     None,
                                     cmds.objectType(path_nodes[i]),
                                     icon_path=file_items[3],
                                     editable=False)
                    self.insert_icon(i, 0, ':error.png')

                self.insert_item(i, 3, file_items[0], file_items[4],
                                 file_items[0])

        self.set_cell_changed_connection_enabled(True)
Exemplo n.º 41
0
#TO-DO----------------create UI. . . (this is already done, btw)
#TO-DO----------------options for showing values captured, for copying only selected attr. .. for multiple keys? ?
#TO-DO----------------copy and store? or copy from the object at that frame (other object?
import maya.cmds as cmds

sel = cmds.ls(sl=True, type="transform")
if (len(sel)) > 1:
    base = sel[0]

    objs = sel[1:]

    #get transforms
    pos = cmds.getAttr("%s.translate" % base)
    rot = cmds.getAttr("%s.rotate" % base)
    scale = cmds.getAttr("%s.scale" % base)

    #put transforms
    for obj in objs:
        cmds.setAttr("%s.translate" % obj, pos[0][0], pos[0][1], pos[0][2])
        cmds.setAttr("%s.rotate" % obj, rot[0][0], rot[0][1], rot[0][2])
        cmds.setAttr("%s.scale" % obj, scale[0][1], scale[0][1], scale[0][2])

else:
    cmds.warning("Must select more than one object!")
Exemplo n.º 42
0
def alembic_import_hairAndShave( **kwargs):
    '''{'path':'Pipeline Cache/Alembic/alembic_import_hairAndShave( )ONLYSE',
'tip':'恢复shave和hair',
'icon':'$ICONDIR/alembic.png',
'html':'http://10.99.17.61/hq_maya/alembicPipeline',
'usage':"""
#根据选择物体的hairData和shaveData以及messenger物体的attributePresets和shaveHairInitialState属性来恢复hair和shave
#使用方法:    1.将保存了attributePresets和shaveHairInitialState属性的locator的名字赋给messenger
#        2.选择要恢复hair和shave的物体的组。
messenger = 'F2_HanJXN_SC03_AN_mb_INOF'
$fun( messenger=messenger)""",
}
'''
    selectedObj =  kwargs.get( 'objects', cmds.ls(sl=True, l=True))
    if not selectedObj:
        raise IOError( "Select objects to create hair and shave!")
    
    #得到属性的json对象
    hairAndShave = []
    messenger = kwargs.get( 'messenger', None)
    if messenger:
        if not cmds.objExists( messenger):
            raise IOError( '%s is not exists!'%(messenger) )
        if not cmds.attributeQuery( 'attributePresets', node=messenger, exists=True):
            raise IOError( '%s.attributePresets is not exists!'%(messenger) )
        
        if not cmds.attributeQuery( 'shaveHairInitialState', node=messenger, exists=True):
            shaveInitialState = None
        else:
            shaveInitialState = cmds.getAttr( messenger+'.shaveHairInitialState' )
            shaveInitialState = json.loads(shaveInitialState)
            #raise IOError( '%s.shaveHairInitialState is not exists!'%(messenger) )
        jsonPreset = cmds.getAttr(  messenger+'.attributePresets'  )
    else:
        jsonPreset = kwargs.get( 'jsonPreset', None)
    
    keyFromAttr = kwargs.get( 'keyFromAttr', 'originalName')   
    
        
    #创建hairSystem
    hairSystems = import_createHairSystemByCurveAttr()    
    if hairSystems:        
        hairAndShave.extend( hairSystems )   
    
    
    #检查shave插件
    loadedShave = False
    if not cmds.pluginInfo( 'shaveNode.mll', query=True, loaded=True):
        try: cmds.loadPlugin( 'shaveNode.mll' )
        except: pass    
    if cmds.pluginInfo( 'shaveNode.mll', query=True, loaded=True):
        loadedShave=True
    
    #创建shaveHair    
    if loadedShave:
        cmds.select( selectedObj, r=True)
        shaveHairs = import_createShaveHairByObjectsAttr()
        if shaveHairs:
            hairAndShave.extend( shaveHairs )    
    
    
    if hairAndShave:
        #载入shaveHair和hairSystem的预设,从json对象中
        qm.loadJsonAttrsPreset( objects=hairAndShave, jsonPreset=jsonPreset, keyFromAttr=keyFromAttr )
        
        #创建并载入hairSystem的brush节点,并从jsonPrest中载入预设
        if hairSystems:
            brushNodes = []
            for hair in hairSystems:
                if cmds.attributeQuery('brushNode', node=hair, exists=True):                   
                    
                    pfxHair = cmds.listConnections( hair+'.outputRenderHairs', s=False, shapes=True)
                    brushNode = cmds.listConnections( pfxHair[0]+'.brush', d=False, shapes=True)                    
                    if not brushNode:
                        cmds.select( hair, r=True)
                        mel.eval('assignBrushToHairSystem;')
                        brushNode = cmds.listConnections( pfxHair[0]+'.brush', d=False, shapes=True)              
                    
                    qm.delAttr( brushNode, ['originalName',] )
                    cmds.addAttr( brushNode[0], ln='originalName', dt='string' )
                    cmds.setAttr( brushNode[0]+'.originalName', cmds.getAttr( hair+'.brushNode' ),
                                  type='string')
                    brushNodes.append( brushNode[0] )
                    
            #Load brush attributes from jsonPreset
            if brushNodes:
                qm.loadJsonAttrsPreset( objects=brushNodes, jsonPreset=jsonPreset, keyFromAttr=keyFromAttr )
                    
        
        #Connect texture or shader to shaveHair
        for shave in hairAndShave:
            if cmds.attributeQuery( 'shaderData', node=shave, exists=True):
                shaderData = json.loads( cmds.getAttr( shave+'.shaderData')    )
                for fromAttr, toAttr in shaderData:
                    try:
                        cmds.connectAttr( fromAttr, shave+'.'+toAttr, f=True)
                    except:
                        pass

    #Initialize shaveHiar initial state
    if loadedShave and shaveInitialState and shaveHairs:
        gMainProgressBar = mel.eval('$tmp = $gMainProgressBar')
        cmds.progressBar( gMainProgressBar, edit=True, beginProgress=True, isInterruptable=True,
                status='Initialize state of shaveHair....', maxValue=len(shaveHairs)  )
        
         
        startFrame = shaveInitialState.get('frame', None)
        if not startFrame:
            raise IOError( "没有得到shavehair的初始状态所在的帧")
        cmds.currentTime( startFrame, e=True)
        #allShaveHair = cmds.ls( exactType='shaveHair', l=True )
        allShaveHair = shaveHairs
        shaveHairInitialData = {}
        for shave in allShaveHair:
            if cmds.attributeQuery( 'originalName', node=shave, exists=True):
                oriName = cmds.getAttr( shave+'.originalName' )
                shaveHairInitialData[oriName] = ([], shave)
        
        for shave, curveDataList in shaveInitialState.iteritems():
            cmds.progressBar(gMainProgressBar, edit=True, step=1 )
            if shaveHairInitialData.has_key(shave):
                for curveData in curveDataList:
                    newCurve = cmds.curve(degree=curveData[0], p=curveData[1:])
                    shaveHairInitialData[shave][0].append( newCurve )
        cmds.progressBar(gMainProgressBar, edit=True, endProgress=True)
        
        for shave, objects in shaveHairInitialData.iteritems():
            if objects[0]:
                cmds.select(objects[0], r=True)
                cmds.select(objects[1], add=True )
                cmds.shaveCombFromCurves()
            else:
                cmds.warning( "%s skiped!"%(shave)   )
            cmds.delete( objects[0] )
        
    return hairAndShave
Exemplo n.º 43
0
    def shaveReName(self):
        noShaveCacheFiles = []
        allOnlyShaveShapes = []
        PROJECT_PATH = mm.eval('getenv "OCTV_PROJECTS"')
        OCT_DRIVE = r'\\octvision.com\cg'
        allShaveShapes = mc.ls(type='shaveHair')
        for eachShape in allShaveShapes:
            allOnlyShaveShapes.append(eachShape.split("|")[-1])

        allOnlyShaveShapes = list(set(allOnlyShaveShapes))
        if len(allShaveShapes) > len(allOnlyShaveShapes):
            mc.warning(u"毛发Shave的shapes节点有重名的,将导致同名的Shave使用同一个缓存!")
        myshaveGlobals = "shaveGlobals"

        allOnlyShaveShapes = list(set(allOnlyShaveShapes))
        myshaveGlobals = "shaveGlobals"
        if allShaveShapes and myshaveGlobals:
            shavePath = mc.getAttr("%s.tmpDir" % myshaveGlobals)
            if shavePath:
                if shavePath.find('${OCTV_PROJECTS}') >= 0:
                    shavePath = shavePath.replace('${OCTV_PROJECTS}',
                                                  PROJECT_PATH)
                elif shavePath.find('z:') >= 0:
                    shavePath = shavePath.replace('z:', OCT_DRIVE)
                elif shavePath.find('Z:') >= 0:
                    shavePath = shavePath.replace('Z:', OCT_DRIVE)
                if not os.path.isdir(shavePath):
                    proPath = mc.workspace(q=True, rd=True)
                    shavePath = os.path.normpath(
                        os.path.join(proPath, shavePath))
                if shavePath.find('z:') >= 0:
                    shavePath = shavePath.replace('z:', OCT_DRIVE)
                elif shavePath.find('Z:') >= 0:
                    shavePath = shavePath.replace('Z:', OCT_DRIVE)
                if os.path.isdir(shavePath):
                    allSahveDirs = os.listdir(shavePath)
                    if allSahveDirs:
                        shaveNames = []
                        for eachDir in allSahveDirs:
                            shaveNames.append(eachDir.split(".")[0])
                        shaveNames = list(set(shaveNames))
                        for eachShape in allShaveShapes:
                            if mc.reference(eachShape, isNodeReferenced=True):
                                mc.confirmDialog(message=u"请先导入参考的文件!")
                                return
                            myName = ""
                            if ":" in eachShape.split("|")[-1]:
                                shaveNames = eachShape.replace(":", "_")
                                myName = mc.rename(eachShape, shaveNames)
                                eachTran = mc.listRelatives(myName,
                                                            f=True,
                                                            p=True)[0]
                                pathTran = eachTran.split("|")
                                if ":" in pathTran[-1]:
                                    tranName = pathTran[-1].replace(":", "_")
                                    myName = mc.rename(pathTran[-1], tranName)
                            else:
                                noShaveCacheFiles.append(myshaveGlobals)

        if not noShaveCacheFiles:
            mc.confirmDialog(message=u"修改shave的名字成功!")
            return
Exemplo n.º 44
0
def output_createHiarOutputCurve(hairSystemObjs = None, hasPfxHair=True, createMarked=False):
    if not hairSystemObjs:
        if not cmds.ls(sl=True):
            cmds.warning( 'Select a hairSystem node!' )
            return None
        hairSystemObjs = cmds.ls(sl=True, l=True, exactType='hairSystem')
    if not hairSystemObjs:
        hairSystemObjs = cmds.listRelatives(  cmds.ls(sl=True, l=True), shapes=True, type='hairSystem' )
    
    if not hairSystemObjs:
        return None
    
    
    if isinstance(hairSystemObjs, (str,unicode) ):
        hairSystemObjs = [hairSystemObjs]
    
    
    #hairSystemObjs, hasPfxHair, createMarked = cmds.ls(exactType='hairSystem', l=True), False, False
    resultCruveGrps = []
    allHairData = {}
    numFollicle = 0
    
    for hair in hairSystemObjs: 
        pfxHair = cmds.listConnections( hair+'.outputRenderHairs', s=False, shapes=True)
        if hasPfxHair and not pfxHair:
            continue
        
        
        allHairData[hair] = {}
        follicles = cmds.listConnections( hair+'.inputHair', d=False, shapes=True)
        if not follicles:
            continue
        
        #Connected attributes infomations to attributes of materials
        materialsInfo = []
        mayaShader = cmds.listConnections( hair, d=False, p=True, c=True, type='shadingDependNode', shapes=True)
        aiHairs = cmds.listConnections( hair, d=False, p=True, c=True, type='aiHair', shapes=True )
        if mayaShader:
            materialsInfo.extend(  
                                    [  (mayaShader[i+1],mayaShader[i].split('.', 1)[-1] ) for i in range(0, len(mayaShader), 2)     ]
                                    )
        if aiHairs:
            materialsInfo.extend(     
                                    [   (aiHairs[i+1],aiHairs[i].split('.', 1)[-1] ) for i in range(0, len(aiHairs), 2)     ]
                                    )
        if materialsInfo:
            
            qm.delAttr( hair, ['shaderData', 'brushAttrs'])
            cmds.addAttr( hair, ln='shaderData', dt='string')
            cmds.setAttr( hair+'.shaderData', json.dumps(materialsInfo), type='string' )
            
        
        
        if not createMarked:
            tempFollicles = []
            for fol in follicles:
                markedCurves = cmds.listConnections( fol+'.outCurve', s=False, shapes=True)
                if not markedCurves:
                    tempFollicles.append( fol )
                    continue
                
                if not [c for c in markedCurves if cmds.attributeQuery('hairData', node=c, exists=True)  ]:
                    tempFollicles.append( fol )
                    continue
                
            if not tempFollicles:
                continue
            follicles = tempFollicles
        
        allHairData[hair] = follicles 
        numFollicle = numFollicle+len( follicles )
        
    if not numFollicle:
        return []
    gMainProgressBar = mel.eval('$tmp = $gMainProgressBar')
    cmds.progressBar( gMainProgressBar, edit=True, beginProgress=True, isInterruptable=True,
                    status='Creating curve...', maxValue=numFollicle )
        
    curNum = 1
    folAttrs = {'clumpWidthMult':1,
                'densityMult':    1,
                'curlMult':    1,
                'clumpTwistOffset':    0,
                'colorBlend':    0,
                'colorR':   0,
                'colorG':   0,
                'colorB':   0,
                }
    
    for hair, follicles in allHairData.iteritems():
        outCurvesGrp = cmds.group( em=True, name='hairSystem_OutputCurves_#')
        for fol in follicles:
            if cmds.progressBar(gMainProgressBar, query=True, isCancelled=True ):
                break
            
            newCurve = cmds.createNode( "nurbsCurve")
            hairValue = {}
            hairValue['hairSystem'] = hair
            for attr, v in folAttrs.iteritems():
                curV = cmds.getAttr( fol+'.'+attr )
                if curV!=v:
                    hairValue[attr] = curV
            jsonStr = json.dumps( hairValue )
                
            cmds.addAttr( newCurve, ln='hairData', dt='string' )            
            cmds.setAttr( newCurve+'.hairData', jsonStr, type='string')
            
            curvePapa = cmds.listRelatives( newCurve, p=True, pa=True )[0]
            cmds.connectAttr( fol+'.outCurve', newCurve+'.create')
            if not cmds.listConnections( fol+'.currentPosition', d=False):
                hairSysAttr = cmds.listConnections( fol+'.outHair', p=True, s=False, shapes=True, type='hairSystem', exactType=True)[0]
                pIndex = re.match( r'.+\[(\d+)\]$', hairSysAttr).groups()[0]
                cmds.connectAttr( hair+'.outputHair[%s]'%(pIndex), fol+'.currentPosition')
                
            cmds.parent( curvePapa, outCurvesGrp, r=True  )                
            cmds.progressBar(gMainProgressBar, edit=True, step=1, status='Creating curve %s/%s'%(curNum, numFollicle) )
         
            curNum = curNum+1
        resultCruveGrps.append( outCurvesGrp )
        
    cmds.progressBar(gMainProgressBar, edit=True, endProgress=True)
    return resultCruveGrps
Exemplo n.º 45
0
    def build(self):
        '''
        '''
        super(Arm, self).build()
        # get the values from the user. Node attributes in the graph.
        clavicleCtrl = self.getAttributeByName('clavicleCtrl').getValue()
        swingCtrl = self.getAttributeByName('swingCtrl').getValue()
        MirrorSwing = self.getAttributeByName('MirrorSwing').getValue()
        swingNul, swingMirrorOrt, swingOrt, swingCtrl = control.create(name=swingCtrl, 
                                                controlType="square",
                                                hierarchy=['nul','mirror_ort','ort'])
        clavicleNul, clavicleOrt, clavicleCtrl = control.create(name=clavicleCtrl, 
                                                                controlType="square",
                                                                hierarchy=['nul','ort'])

        # position the clavicle in the correct location
        clavicleJointMatrix = mc.xform(self._clavicleJoint, q=True, ws=True, matrix=True)
        mc.xform(clavicleNul, ws=True, matrix=clavicleJointMatrix)
        
        # get the aim direction
        aimDistance = mc.getAttr("{}.t".format(self.jointList[1]))[0]
        aimAttr, aimVector = self._getDistanceVector(aimDistance)
        sidePos = mc.xform(self.jointList[1], q=True, ws=True, t=True)

        # set the Ort node to be mirrored from the other side
        if sidePos[0] < 0:            
            mc.setAttr("{}.r{}".format(clavicleOrt, aimAttr.strip("-")), 180)
            mc.setAttr("{}.s{}".format(clavicleOrt, aimAttr.strip("-")), -1)

            # mirror the swing also if it is required by the build
            if MirrorSwing:
                mc.setAttr("{}.r{}".format(swingMirrorOrt, aimAttr.strip("-")), 180)
                mc.setAttr("{}.s{}".format(swingMirrorOrt, aimAttr.strip("-")), -1)

        # move the shoulderSwing control to the correct location.
        shoulderCtrlMatrix = mc.xform(self._fkControls[0], q=True, ws=True, matrix=True)
        mc.xform(swingNul, ws=True, matrix=shoulderCtrlMatrix)

        # Hookup clavicle connect nul, the direct connection for the rotate allow keeps the auto
        # clav from causint a double rotation on the shoulder
        clavicleConnect = mc.duplicate(clavicleCtrl, po=1, n=clavicleCtrl+'_connect')[0]
        mc.parent(clavicleConnect, clavicleNul)
        mc.connectAttr(clavicleCtrl+'.r', clavicleConnect+'.r')
        mc.connectAttr(clavicleCtrl+'.s', clavicleConnect+'.s')

        # This allows the translates to come through with auto clav
        clavicleConnectTranslate = mc.duplicate(swingNul, po=1, n=clavicleCtrl+'_connect_trans')[0]
        mc.parent(clavicleConnectTranslate, clavicleCtrl)
        mc.pointConstraint(clavicleConnectTranslate, clavicleConnect)

        # getting rid of the point constraint for now. Not sure we need it.
        #mc.pointConstraint(clavicleCtrl, self._clavicleJoint)
        mc.orientConstraint(clavicleCtrl, self._clavicleJoint)

        # parent the shoulderSwing control to the clavicle control.
        mc.parent((self._fkControls[0], self._stretchTargetJointList[0]), swingCtrl)
        mc.parent(swingNul, clavicleConnect)
        
        # Connect to passed anchor
        #
        anchor = self.getAttributeByName('anchor').getValue()
        if mc.objExists(anchor):
            mc.parent(clavicleNul, anchor) 
        else:
            # parent the clavicle to the group of this part.
            mc.parent(clavicleNul, self.name)
            mc.warning('Anchor object [ {} ] does not exist.'.format(anchor)) 


        # set the rotate order for the shoulder control
        mc.setAttr("{}.rotateOrder".format(self._fkControls[0]), 2)
        # set the rotate order for the swing control
        mc.setAttr("{}.rotateOrder".format(swingCtrl), 2)
        #self._fkControls.extend([clavicleCtrl,swingCtrl])

        # PSD driver - transform that picks up the auto clav and anim control rotation
        clavicleDriverPar = mc.duplicate(clavicleConnect, po=1, n=clavicleCtrl+'_driver_par')[0]
        clavicleDriver = mc.duplicate(clavicleConnect, po=1, n=clavicleCtrl+'_driver')[0]
Exemplo n.º 46
0
 def __load_plugin(self):
     if not cmds.pluginInfo('push_deformer.py', l=True, q=True):
         if cmds.loadPlugin('push_deformer.py') == None:
             cmds.warning('Failed to load plugin.')
             return False
     return True
Exemplo n.º 47
0
def publish_fbx_rig_file(versionUp=True, origScene=None, *args):
    """
    requires an EXPORT_JNT_Grp group with one root for each export rig named: 'name_Root_Jnt'
    requires a GEO group with one folder for each export rig named: 'name_Geo_Grp'
    names should correspond ("fish_Root_Jnt", "fish_Geo_Grp")
    """
    # all happens in current:
    if not origScene:
        cmds.warning("assetPublish.publish_fbx_rig_file: You haven't passed in a scene path!")
        return(False)

    pp = uf.PathParser(origScene)

    geoGrp = cmds.ls("GEO")
    jntGrp = cmds.ls("EXPORT_JNT_Grp")

    # check for geo grps
    if not geoGrp or len(geoGrp)>1:
        cmds.warning("AssetPublish.publish_fbx_rig_file:You either have no grp called 'GEO', or too many objects called 'GEO'.\n fbx export aborted!")
        return(False)
    geos = child_match_check(geoGrp, "*_Geo_Grp")
    if not geos:
        return(False)

    # check for jnt grps
    if not jntGrp or len(jntGrp)>1:
        cmds.warning("AssetPublish.publish_fbx_rig_file:You either have no grp called 'EXPORT_JNT_Grp', or too many objects called 'EXPORT_JNT_Grp'.\n fbx export aborted!")
        return(False)    
    roots = child_match_check(jntGrp, "*_Root_Jnt")
    if not roots:
        return(False)

    # check correspondence
    correspond = check_correspondence(geos, roots)
    if not correspond:
        return(False)

    pubFbxPath = uf.fix_path(os.path.join(pp.phasePath, "Publish/FBX/"))
    tokens = pp.fileName.split("_")
    tokens[-2] = "Publish"
    pubFileName = "_".join(tokens)[:-3] + ".fbx"
    pubFilePath = uf.fix_path(os.path.join(pubFbxPath, pubFileName))

# check if there's any animation in the file (time based), abort if there is
# check for references, etc. . 

    # delete constraints
    cmds.delete(cmds.ls(type="constraint"))
    cmds.select(cl=True)
    # move jnts and geo into world parent
    for root in roots:
        basename = root.split("_Root_Jnt")[0]
        geo = "{0}_Geo_Grp".format(basename)
        cmds.parent([geo, root], w=True)
        # create filename
        tokens[-2] = basename
        pubFileName = "_".join(tokens)[:-3] + ".fbx"
        pubFilePath = uf.fix_path(os.path.join(pubFbxPath, pubFileName))
        cmds.select([root, geo], r=True)

        # if this exists, should we overwrite?
        if os.path.isfile(pubFilePath):
            overwrite = cmds.confirmDialog(title="Overwrite Confirmation", message = "A publish FBX already exists for this file.\nShould we overwrite?", button = ("Overwrite", "Cancel"), defaultButton = "Overwrite", cancelButton = "Cancel", dismissString = "Cancel")

            if overwrite == "Cancel":
                print "Publish skipped for FBX file (.fbx) called {0}".format(pubFilePath)
                return(True) 

        mel.eval('FBXLoadExportPresetFile -f "{0}";'.format(preset))
        mel.eval('FBXExport -f "{0}" -s'.format(pubFilePath))

    return(True)
Exemplo n.º 48
0
def mayaToolsInstall_Cancel(*args):

    cmds.deleteUI("mayaToolsInstall_UI")

    cmds.warning("Maya Tools will not be setup")
Exemplo n.º 49
0
def applyAnimData(animData, pasteInPlace=True, onlySelectedNodes=False, showProgress=None, status=None): 
    
    if animData:   
        
        status      = "aTools - Applying animation data..." if not status else status
        objects     = animData["objects"]
                        
        if not onlySelectedNodes:
            #print "objects1", objects
            if len(objects) > 0: objects = [loopObj for loopObj in objects if loopObj is not None and cmds.objExists(loopObj)]
            #print "objects2", objects
            if len(objects) > 0: cmds.select(objects)        
        else:
            objects      = getObjsSel()
        
        if not objects: 
            cmds.warning("No objects to apply.")
            return
                
        cmds.refresh(suspend=True)
        if showProgress: utilMod.startProgressBar(status)
        
        if pasteInPlace:
            currKey  = cmds.currentTime(query=True) 
            for aData in animData["animData"]:
                allKeys         = []
                keys            = aData["keyframeData"]
                for n, key in enumerate(keys):
                    timeChange      = aData["keyframeData"][n][0]                        
                    allKeys.append(timeChange)
            
            firstKey = 0    
            if allKeys: 
                firstKey    = min(allKeys)
                lastKey     = max(allKeys)
                cutIn       = currKey+firstKey
                cuOut       = lastKey+firstKey
            
        else: 
            cutIn   = -49999
            cuOut   = 50000
            
   
        objsAttrs       = [loopItem["objAttr"] for loopItem in animData["animData"]]        
        existObjsAttrs  = [loopObjAttr for loopObjAttr in objsAttrs if cmds.objExists(loopObjAttr)]
                
        createDummyKey(existObjsAttrs)
        cmds.cutKey(existObjsAttrs, time=(cutIn, cuOut), clear=True)
        
        if showProgress:
            totalSteps      = 0            
            firstStep       = 0
            thisStep        = 0
            estimatedTime   = None
            startChrono     = None            
            
            for loopObjAttr in existObjsAttrs:  
                index           = objsAttrs.index(loopObjAttr)
                aData           = animData["animData"][index]
                keys            = aData["keyframeData"] 
                totalSteps      = totalSteps + len(keys)
        
        
        for loopObjAttr in existObjsAttrs: 
            
            index           = objsAttrs.index(loopObjAttr)
            aData           = animData["animData"][index]
            weighted        = aData["curveData"][0]
            infinity        = aData["curveData"][1]
            keys            = aData["keyframeData"] 
             
            
            for n, key in enumerate(keys):
                
                if showProgress: 
                    if cmds.progressBar(G.progBar, query=True, isCancelled=True ): 
                        refresh() 
                        utilMod.setProgressBar(endProgress=True)
                        return
                    startChrono = utilMod.chronoStart(startChrono, firstStep, thisStep, totalSteps, estimatedTime, status)   
                
                #read values
                timeChange      = aData["keyframeData"][n][0]
                valueChange     = aData["keyframeData"][n][1]
                breakdown       = aData["keyframeData"][n][2]
                inTangentType   = aData["tangentData"][n][0]
                outTangentType  = aData["tangentData"][n][1]
                ix              = aData["tangentData"][n][2]
                iy              = aData["tangentData"][n][3]
                ox              = aData["tangentData"][n][4]
                oy              = aData["tangentData"][n][5]
                lock            = aData["tangentData"][n][6]
                weightLock      = aData["tangentData"][n][7]  
                
                if pasteInPlace: timeChange  = timeChange-firstKey+currKey                      
                
                time            = (timeChange,timeChange)                        
                
                # create key
                cmds.setKeyframe(loopObjAttr, time=time, value=valueChange, noResolve=True)
                
                if n == 0:
                    cmds.keyTangent(loopObjAttr, weightedTangents=weighted)
                    cmds.setInfinity(loopObjAttr, edit=True, preInfinite=infinity[0], postInfinite=infinity[1])
                
                if breakdown: cmds.keyframe(loopObjAttr, edit=True, time=time, breakdown=True)
                cmds.keyTangent(loopObjAttr,  time=time, ix=ix, iy=iy, ox=ox, oy=oy, lock=lock)
                if weighted: cmds.keyTangent(loopObjAttr, time=time, weightLock=weightLock)
                cmds.keyTangent(loopObjAttr,  time=time, inTangentType=inTangentType, outTangentType=outTangentType)
        
                if showProgress: estimatedTime = utilMod.chronoEnd(startChrono, firstStep, thisStep, totalSteps)
                thisStep += 1
        
        deleteDummyKey(existObjsAttrs)
            
        if showProgress: 
            refresh()
            utilMod.setProgressBar(endProgress=True)
Exemplo n.º 50
0
    def build(self):
        '''
        '''
        super(ArmOld, self).build()
        clavicleCtrl = self.getAttributeByName('clavicleCtrl').getValue()
        swingCtrl = self.getAttributeByName('swingCtrl').getValue()
        swingCtrlHierarchy = control.create(name=swingCtrl, 
                                                controlType="square",
                                                hierarchy=['nul','ort'])
        clavicleCtrlHierarchy = control.create(name=clavicleCtrl, 
                                                controlType="square",
                                                hierarchy=['nul','ort'])

        

        '''
        clavicle = 'clavicle_l'
        child = 'shoulderSwing_l_nul'

        aimTarget = mc.duplicate(clavicle, po=1, n=clavicle + '_aim_target')[0]
        aim = mc.duplicate(clavicle, po=1, n=clavicle + '_aim')[0]

        mc.delete(mc.pointConstraint(child, aimTarget))
        mc.parent(aimTarget, clavicle)
        mc.setAttr(aimTarget + '.tx', 1)

        mc.aimConstraint(aimTarget, aim, offset=[0, 0, 0],
                         weight=1, aimVector=[1, 0, 0],
                         worldUpType="none",
                         upVector=[0, 0, 0])

        mc.pointConstraint(aimTarget, child)
        '''

        clavicleCtrl = clavicleCtrlHierarchy[-1]
        clavicleNul = clavicleCtrlHierarchy[0]
        swingCtrl = swingCtrlHierarchy[-1]
        swingNul = swingCtrlHierarchy[0]
        clavicleJointMatrix = mc.xform(self._clavicleJoint, q=True, ws=True, matrix=True)
        mc.xform(clavicleNul, ws=True, matrix=clavicleJointMatrix)

        # move the shoulderSwing control to the correct location.
        shoulderCtrlMatrix = mc.xform(self._fkControls[0], q=True, ws=True, matrix=True)
        mc.xform(swingNul, ws=True, matrix=shoulderCtrlMatrix)

        # Hookup clavicle connect nul, the direct connection for the rotate allow keeps the auto
        # clav from causint a double rotation on the shoulder
        clavicleConnect = mc.duplicate(clavicleCtrl, po=1, n=clavicleCtrl+'_connect')[0]
        mc.parent(clavicleConnect, clavicleNul)
        mc.connectAttr(clavicleCtrl+'.r', clavicleConnect+'.r')
        mc.connectAttr(clavicleCtrl+'.s', clavicleConnect+'.s')
        # PSD driver - transform that picks up the auto clav and anim control rotation
        clavicleDriver = mc.duplicate(clavicleConnect, po=1, n=clavicleCtrl+'_driver')[0]
        mc.orientConstraint(clavicleCtrl, clavicleDriver)

        # This allows the translates to come through with auto clav
        clavicleConnectTranslate = mc.duplicate(swingNul, po=1, n=clavicleCtrl+'_connect_trans')[0]
        mc.parent(clavicleConnectTranslate, clavicleCtrl)
        mc.pointConstraint(clavicleConnectTranslate, clavicleConnect)


        mc.pointConstraint(clavicleCtrl, self._clavicleJoint)
        mc.orientConstraint(clavicleCtrl, self._clavicleJoint)

        # parent the shoulderSwing control to the clavicle control.
        mc.parent(("{}_nul".format(self._fkControls[0]), self._stretchTargetJointList[0]), swingCtrl)
        mc.parent(swingNul, clavicleConnect)

        # parent constrain the shoulder ik joint to the clavicle joint.
        mc.parentConstraint(self._clavicleJoint, self.ikfkSystem.getIkJointList()[0], mo=True)

        # parent the clavicle to the group of this part.
        mc.parent(clavicleNul, self.name)
        
        # Connect to passed anchor
        #
        anchor = self.getAttributeByName('anchor').getValue()
        if mc.objExists(anchor):
            mc.parent(clavicleNul, anchor) 
        else:
            mc.warning('Anchor object [ {} ] does not exist.'.format(anchor)) 


        # set the rotate order for the shoulder control
        mc.setAttr("{}.rotateOrder".format(self._fkControls[0]), 2)
        # set the rotate order for the swing control
        mc.setAttr("{}.rotateOrder".format(swingCtrl), 2)
Exemplo n.º 51
0
        cmds.refresh(su=True)

        cmds.bakeResults(sNull, t=(iIn, iOut), simulation=True)
        cmds.delete('TempConst')

        cmds.refresh(su=False)

        ApplyConst(sNull, s)

    cmds.select(cl=True)
    restoreLayout('Custom_AnimWithCam')

elif K == 13:  # Ctl + Alt + Shift # Special Plot
    if not oSel:
        cmds.warning(' Please Select an obj first.')
    else:
        oCamera = ''
        oPanel = cmds.getPanel(wf=True)

        if 'modelPanel' in oPanel:
            oCamera = cmds.modelEditor(oPanel, q=True,
                                       camera=True).split('Shape')[0]
            if not cmds.objExists(oCamera):
                cmds.warning('View not selected!!')

        print 'Camera : ', oCamera
        if oCamera:
            aPlotList = []
            # Create Plot Nulls
            for o in oSel:
Exemplo n.º 52
0
def publish_fbx_anim_file(versionUp=True, origScene=None, *args):
    # THIS IS FOR ANIM EXPORTING

    if not origScene:
        cmds.warning("assetPublish.publish_fbx_anim_file: You haven't passed in a scene path!")
        return(False)
# assuming references
    refs = cmds.file(q=True, r=True)
    if not refs:
        cmds.warning("There are no references in this scene. . .")
        return(False)
    if len(refs) > 1:
        cmds.warning("There are too many references in this scene. . .")
        return(False)
# below would all be under a for loop for each reference in the stages? 
    pp = uf.PathParser(origScene)

# assuming a namespace
    geoGrp = cmds.ls("*:GEO")
    jntGrp = cmds.ls("*:EXPORT_JNT_Grp")

    # check for geo grps
    if not geoGrp or len(geoGrp)>1:
        cmds.warning("AssetPublish.publish_fbx_anim_file:You either have no grp called 'GEO' -IN A NAMESPACE-, or too many objects called 'GEO'.\n fbx export aborted!")
        return(False)
    geos = child_match_check(geoGrp[0], "*_Geo_Grp")
    if not geos:
        return(False)

    # check for jnt grps
    if not jntGrp or len(jntGrp)>1:
        cmds.warning("AssetPublish.publish_fbx_anim_file:You either have no grp called 'EXPORT_JNT_Grp' -IN A NAMESPACE-, or too many objects called 'EXPORT_JNT_Grp'.\n fbx export aborted!")
        return(False)    
    roots = child_match_check(jntGrp[0], "*_Root_Jnt")
    if not roots:
        return(False)

    cmds.file(refs[0], ir=True)

    # check correspondence of geo and root jnts
    correspond = check_correspondence(geos, roots)
    if not correspond:
        return(False)

    pubFbxPath = uf.fix_path(os.path.join(pp.phasePath, "Publish/FBX/"))
    tokens = pp.fileName.split("_")
    tokens[-2] = "Publish"
    pubFileName = "_".join(tokens)[:-3] + ".fbx"
    pubFilePath = uf.fix_path(os.path.join(pubFbxPath, pubFileName))

    # bake joints
    for r in roots:
        # get child roots if joints
        allD = cmds.listRelatives(r, allDescendents=True)
        jnts = [x for x in allD if cmds.objectType(x, isa="joint")]
        # function to bake selected
        for j in jnts:
            attr=["t","r","s"]
            co=["x","y","z"]
            attrLs=[]
            for at in attr:
                for c in co:
                    attrLs.append("%s.%s%s"%(j,at,c))
            for x in attrLs:
                try:
                    mc.setAttr(x, k=1)
                except:
                    pass

    namespace = cmds.file(refs[0], q=True, ns=True)
    uf.remove_namespaces()

    # delete constraints
    cmds.delete(cmds.ls(type="constraint"))
    cmds.select(cl=True)
    # move jnts and geo into world parent
    for root in roots:
        basename = root.split("_Root_Jnt")[0]
        geo = "{0}_Geo_Grp".format(basename)
        cmds.parent([geo, root], w=True)
        # create filename
        tokens[-2] = basename
        pubFileName = "_".join(tokens)[:-3] + ".fbx"
        pubFilePath = uf.fix_path(os.path.join(pubFbxPath, pubFileName))
        cmds.select([root, geo], r=True)

        # if this exists, should we overwrite?
        if os.path.isfile(pubFilePath):
            overwrite = cmds.confirmDialog(title="Overwrite Confirmation", message = "A publish FBX already exists for this file.\nShould we overwrite?", button = ("Overwrite", "Cancel"), defaultButton = "Overwrite", cancelButton = "Cancel", dismissString = "Cancel")

            if overwrite == "Cancel":
                print "Publish skipped for FBX file (.fbx) called {0}".format(pubFilePath)
                return(True) 

        mel.eval('FBXLoadExportPresetFile -f "{0}";'.format(preset))
        mel.eval('FBXExport -f "{0}" -s'.format(pubFilePath))

    return(True)    


    # loop through namespaces/references:
    #   import reference
    #   delete namespace
    #   loop through joint grps, geo grps:
    #       bake jnts
    #       clean up joints shit

    #       "CloudStage_Tree_main_Rig1_A_v0001.fbx"
    #           stage     asset ns    subgrp  version
        
    #       check for version folder, create it if it doesn't exist
    pass
Exemplo n.º 53
0
def flexi_loft(add_ik=0, bind=0, loft=0, cluster=0, stretchy=0, name=""):
    raw_curves = mc.ls(sl=1)
    curves = []
    ik_handle_list = []
    ik_spline_crv_list = []
    joint_list = []
    joint_chain_dict = {}
    grp_list = []

    # Creating brand new, clean curves out of selection
    for j, n in enumerate(raw_curves, 1):
        new_curve = mc.duplicate(n, n="{}_Loft_Crv{}".format(name, j))
        try:
            mc.parent(new_curve, w=1)
        except:
            pass
        mc.makeIdentity(new_curve, apply=True, t=1, r=1, s=1, n=0)
        mc.delete(new_curve, ch=1)
        mc.xform(new_curve, cp=1)
        curves.append(new_curve[0])

    # Joint chain and ikSpline section
    for n, n in enumerate(curves):
        crv_joint_list = joint_on_curve_cvs(path_crv=n)
        parent_in_order(sel=crv_joint_list)
        mc.joint(crv_joint_list, e=1, oj="xyz", secondaryAxisOrient="yup")
        joint_chain_dict.update({n: crv_joint_list})
        joint_list.append(crv_joint_list[0])
        mc.select(cl=1)

        if add_ik == 1:
            # Creating ikSpline - rename curve - append to grouping lists
            ik_handle_data = mc.ikHandle(sj=crv_joint_list[0],
                                         ee=crv_joint_list[-1],
                                         sol="ikSplineSolver",
                                         n=n + "_ikHandle")
            ik_crv = mc.rename(ik_handle_data[2],
                               "{}_ikSpline_Crv{}".format(name, n + 1))
            ik_handle_list.append(ik_handle_data[0])
            ik_spline_crv_list.append(ik_crv)

            if stretchy == 1:
                make_stretchy_spline(crv_joint_list, ik_crv, new_curve[0])

            if n == 0:
                # If it is the first loop - create groups - else - parent new items
                ik_grp = mc.group(ik_handle_data[0], n=name + "_ikHandle_Grp")
                spline_grp = mc.group(ik_crv, n=name + "_ikSpline_Crv_Grp")
                grp_list.append(ik_grp)
                grp_list.append(spline_grp)
            else:
                mc.parent(ik_handle_data[0], ik_grp)
                mc.parent(ik_crv, spline_grp)

        if bind == 1:
            mc.skinCluster(crv_joint_list, n)

    if loft == 1:
        loft_data = mc.loft(curves,
                            ch=1,
                            u=1,
                            c=0,
                            ar=0,
                            d=3,
                            ss=10,
                            rn=0,
                            po=1,
                            rsn=1)
        loft_srf = mc.rename(loft_data[0], name + "_LoftSrf_Geo")
        loft_grp = mc.group(loft_srf, n=name + "_LoftSrf_Geo_Grp")
        grp_list.append(loft_grp)

    if cluster == 1 and add_ik == 1:
        # Creates clusters holding the same cv on each ikSpline curve
        # Calculating the number of Cv's for loop
        curve_deg = mc.getAttr(ik_spline_crv_list[0] + ".degree")
        curve_spa = mc.getAttr(ik_spline_crv_list[0] + ".spans")
        # CV's = degrees + spans
        cv_count = curve_deg + curve_spa
        cls_list = []

        for n in range(cv_count):
            mc.select(cl=1)
            for j in ik_spline_crv_list:
                mc.select("{}.cv[{}]".format(j, n), add=1)
            cluster = mc.cluster(n="{}_Csl{}".format(name, n))
            cls_list.append(cluster[1])

        cluster_grp = mc.group(cls_list, n=name + "_Cls_Grp")
        grp_list.append(cluster_grp)

    else:
        mc.warning("addIk is off")

    curves_grp = mc.group(curves, n="{}_Loft_Crv_Grp".format(name))
    joint_grp = mc.group(joint_list, n=name + "_Jnt_Grp")
    grp_list.append(curves_grp)
    grp_list.append(joint_grp)
    sys_grp = mc.group(grp_list, n=name + "_Sys_Grp")

    return sys_grp
Exemplo n.º 54
0
    def Add_Crv_Pick_Fun(self):
        if len(cmds.ls(sl=True)) > 0:
            Selection = cmds.ls(sl=True,l=1)[0] 
            SelectionShape = cmds.listRelatives(Selection)[0]
            SelectionType = cmds.nodeType( SelectionShape )
            if SelectionType == "nurbsCurve":
                self.Add_Crv_LE.setText(Selection)
                try :
                    self.Add_Crv_Get.clicked.disconnect()
                except:
                    pass
                self.Add_Crv_Get.clicked.connect(partial(self.Select_OBJ_Fun,cmds.ls(Selection))
            else:
                cmds.warning(u"请选择nurbsCurve进行获取")
                cmds.headsUpMessage(u"请选择Curve进行获取")
        else :
            self.Add_Crv_LE.setText("")
        
        if self.Add_Crv_LE.text() != "":
            self.Add_Crv_Label.setVisible(False)
            self.Add_Crv_Get.setVisible(True)
        else:
            self.Add_Crv_Label.setVisible(True)
            self.Add_Crv_Get.setVisible(False)

    def Add_Loc_Pick_Fun(self):
        if len(cmds.ls(sl=True)) > 0 :
            Selection = cmds.ls(sl=True,l=1)[0] 
            SelectionShape = cmds.listRelatives(Selection)[0]
            SelectionType = cmds.nodeType( SelectionShape )
            if SelectionType == "locator":
                self.Add_Loc_LE.setText(Selection)
                try :
                    self.Add_Loc_Get.clicked.disconnect()
                except:
                    pass
                self.Add_Loc_Get.clicked.connect(partial(self.Select_OBJ_Fun,Selection))
            else:
                cmds.warning(u"请选择Locator进行获取")
                cmds.headsUpMessage(u"请选择Locator进行获取")
        else :
            self.Add_Loc_LE.setText("")
        
        if self.Add_Loc_LE.text() != "":
            self.Add_Loc_Label.setVisible(False)
            self.Add_Loc_Get.setVisible(True)
        else:
            self.Add_Loc_Label.setVisible(True)
            self.Add_Loc_Get.setVisible(False)

    def Cam_Input_Toggle_Fn(self):
        if self.Cam_Input_Toggle_Check:
            self.Cam_Input_Toggle_Check = False
            self.Cam_Input_Toggle_Anim.setDirection(QAbstractAnimation.Forward)
            self.Cam_Input_Toggle_Anim.start()
            self.Cam_Input_Toggle.setText(u"▼输入设置")
            self.Cam_Input_Toggle.setStyleSheet('font:normal')
        else:
            self.Cam_Input_Toggle_Check = True
            self.Cam_Input_Toggle_Anim.setDirection(QAbstractAnimation.Backward)
            self.Cam_Input_Toggle_Anim.start()
            self.Cam_Input_Toggle.setText(u"■输入设置")
            self.Cam_Input_Toggle.setStyleSheet('font:bold')

    def Cam_Output_Toggle_Fn(self):
        if self.Cam_Output_Toggle_Check:
            self.Cam_Output_Toggle_Check = False
            self.Cam_Output_Toggle_Anim.setDirection(QAbstractAnimation.Forward)
            self.Cam_Output_Toggle_Anim.start()
            self.Cam_Output_Toggle.setText(u"▼输出设置")
            self.Cam_Output_Toggle.setStyleSheet('font:normal')
        else:
            self.Cam_Output_Toggle_Check = True
            self.Cam_Output_Toggle_Anim.setDirection(QAbstractAnimation.Backward)
            self.Cam_Output_Toggle_Anim.start()
            self.Cam_Output_Toggle.setText(u"■输出设置")
            self.Cam_Output_Toggle.setStyleSheet('font:bold')

    def Select_OBJ_Fun(self,selectTarget):
        if selectTarget != "":
            cmds.select(selectTarget)
Exemplo n.º 55
0
    def processSHDTemplate(self, tk, templateFile = '', assetDict = {}, selected = False):
        """
        Used to fetch most recent publishes
        @param tk : tank instance
        @param templateFile: the tank template file specified in the shot_step.yml
        #param assetDict: dict in format assetName, assetParent
        @type templateFile: template
        """
        debug(self.app, method = 'processTemplates', message = 'assetDict:    %s' % assetDict, verbose = False)

        myFinalAssetDict = {}
        
        ## Now fetch all the SRF template paths from shotgun
        getTemplatePaths = tk.paths_from_template(templateFile, {'Step' : 'SRF'})
        #debug(self.app, method = 'processTemplates', message = 'getTemplatePaths: %s' % getTemplatePaths, verbose = False)
        
        ## Now look for each assets template path:        
        for key, var in assetDict.items():
            debug(self.app, method = 'processTemplates', message = 'Processing asset %s now' % key, verbose = False)
            versions = []
            
            for eachPath in getTemplatePaths:
                ### IN scene we get CHAR_Zip_hrc
                ### then we break this back to the publish names that are NOT allowed to have _'s
                ### The publish file names are then CHARZip
                ### BUT a really intelligent Co-Ord decided to make ad asset called CHAR_ZIP_RustryPropeller... that broke the code looking for CharZip as now we have more than one 
                ### So we have to be able to;
                ## Accomodate the following... Char_Zip_hrc  Char_Zip_hrc1 Char_Zip_RustryPropellor_hrc
                
                #'I:\\bubblebathbay\\assets\\CharacterA\\CHAR_Zip_RustyPropeller\\SRF\\publish\\xml\\CHARZip.v001.xml'
                #'I:\\bubblebathbay\\assets\\CharacterA\\CHAR_Zip_RustyPropeller\\SRF\\publish\\xml\\CHARZipRustyPropeller.v001.xml'
                splitPathToAssetName = eachPath.split('\\')[4]
                #debug(self.app, method = 'processTemplates', message = 'splitPathToAssetName: %s' % splitPathToAssetName, verbose = False)
                if key.lower() == splitPathToAssetName.lower():
                    versions.append(eachPath)
            
           # debug(self.app, method = 'processTemplates', message = 'versions.... %s' % versions, verbose = False)

            ## Now if versions has stuff in it..
            if versions:
                if selected:
                    myFinalAssetDict[key] = [max(versions), var[0], var[1]]
                else:
                    myFinalAssetDict[key] = [max(versions), var]
            else:
                debug(self.app, method = 'processTemplates', message = '%s does not have an xml file published.' % key, verbose = False)
                pass
                
        debug(self.app, method = 'processTemplates', message = 'myFinalAssetDict:   %s' % myFinalAssetDict, verbose = False)
        # {u'CHAR_Sydney': ['I:\\bubblebathbay\\assets\\CharacterA\\CHAR_Sydney\\SRF\\publish\\xml\\CHARSydney.v056.xml', u'|ABC_ANIM_CACHES_hrc']}

        allNodes = {}
        for key, var in myFinalAssetDict.items():
            XMLPath = var[0].replace(os.path.sep, "/")
            root = xml.parse(XMLPath).getroot()

            nodes = [each for each in root.getchildren()[0] if each.tag == 'Nodes'][0]
            shadingEngines = [each for each in root.getchildren()[0] if each.tag == 'ShadingEngine'][0]

            for each in nodes.getchildren():
                allNodes.setdefault(each.tag, [])
                allNodes[each.tag].append(key)

            for each in shadingEngines.getchildren():
                sgName = each.attrib['value'].split(',')[0]
                allNodes.setdefault(sgName, [])
                allNodes[sgName].append(key)

        ## Create .txt for modelers as reference
        rebuildShader = 'Continue anyway...'
        shadingDupeCheck = [dupe for dupe in allNodes.itervalues() if len(dupe) > 1]
        if shadingDupeCheck:
            import datetime
            currentDateTime = datetime.datetime.now().strftime("%Y%m%d.%H%M")
            _filePath = 'C:/Temp/shadingNodeNameClash.%s.txt' % currentDateTime
            _file = open(_filePath, 'w')
            for k, v in allNodes.iteritems():
                if len(v) > 1:
                    _file.writelines('%s:\r\n' % k)
                    for dupe in v:
                        _file.writelines('\t%s\r\n' % dupe)
                    _file.writelines('\r\n')
            _file.close()

            ## Give warning to the artist that there's shading node name clashes
            dupeString = 'Shading node name clashes found across multiple assets. Please be advised that this process probably won\'t rebuild 100% as SRF file due to name clashes, a republish of unique shading node name is needed! A Reference .txt of what\'s clashing was compiled at below path.\r\n\r\n'
            #dupeString += _filePath
            #rebuildShader = cmds.confirmDialog(title = 'SHADER REBUILD', message = dupeString, button = ['Continue anyway...', 'Cancel'], defaultButton = 'Continue anyway...', cancelButton = 'Cancel', dismissString = 'Cancel')
            cmds.warning(dupeString)
        #if rebuildShader == 'Continue anyway...':
        for key, var in myFinalAssetDict.items():
            debug(self.app, method = 'processTemplates', message = 'key %s ...' % key, verbose = False)
            debug(self.app, method = 'processTemplates', message = 'var[0] %s ...' % var[0], verbose = False)
            debug(self.app, method = 'processTemplates', message = 'var[1] %s ...' % var[1], verbose = False)

            if os.path.isfile(var[0].replace(os.path.sep, "/")):
                debug(self.app, method = 'processTemplates', message = 'Creating shaders for %s now with path %s ...' % (key, var[0].replace(os.path.sep, "/")), verbose = False)
                debug(self.app, method = 'processTemplates', message = 'parentGrp: %s' % var[1], verbose = False)

                if selected:
                    shd.createAll(XMLPath = var[0].replace(os.path.sep, "/"), parentGrp = var[1], Namespace = '', Root = 'MaterialNodes', selected = True, selectedOrigHrcName = var[2])
                    debug(self.app, method = 'processTemplates', message = 'Connecting shaders for %s now with path %s ...' % (key, var[0].replace(os.path.sep, "/")), verbose = False)
                    versionNumber = var[0].replace(os.path.sep, "/").split('.')[-2].split('v')[-1]
                    shd.connectAll(XMLPath = var[0].replace(os.path.sep, "/"), parentGrp= var[1], Namespace = '', Root = 'MaterialNodes', selected = True, selectedOrigHrcName = var[2], xmlVersionNumber = versionNumber)
                    # ## Now add the version tag to the selected _hrc asset group
                    # versionNumber = var[0].replace(os.path.sep, "/").split('.')[-2].split('v')[-1]
                    # debug(self.app, method = 'processTemplates', message = 'JAMES SHD VERS NUMBER!!!! %s' % versionNumber, verbose = False)
                    # self._addVersionTag(var[2], versionNumber)
                else:
                    shd.createAll(XMLPath = var[0].replace(os.path.sep, "/"), parentGrp = var[1], Namespace = '', Root = 'MaterialNodes')
                    debug(self.app, method = 'processTemplates', message = 'Connecting shaders for %s now with path %s ...' % (key, var[0].replace(os.path.sep, "/")), verbose = False)
                    versionNumber = var[0].replace(os.path.sep, "/").split('.')[-2].split('v')[-1]
                    shd.connectAll(XMLPath = var[0].replace(os.path.sep, "/"), parentGrp= var[1], Namespace = '', Root = 'MaterialNodes', xmlVersionNumber = versionNumber)
            else:
                debug(self.app, method = 'processTemplates', message = 'FAILED: No valid published xml file found for %s ...' % (key, var[0].replace(os.path.sep, "/")), verbose = False)
                pass
Exemplo n.º 56
0
def uv_from_element(element):
    """Return the UV coordinate of given 'element'

    Supports components, meshes, nurbs.

    """

    supported = ["mesh", "nurbsSurface"]

    uv = [0.5, 0.5]

    if "." not in element:
        type = cmds.nodeType(element)
        if type == "transform":
            geometry_shape = cmds.listRelatives(element, shapes=True)

            if len(geometry_shape) >= 1:
                geometry_shape = geometry_shape[0]
            else:
                return

        elif type in supported:
            geometry_shape = element

        else:
            cmds.error("Could not do what you wanted..")
            return
    else:
        # If it is indeed a component - get the current Mesh
        try:
            parent = element.split(".", 1)[0]

            # Maya is funny in that when the transform of the shape
            # of the component elemen has children, the name returned
            # by that elementection is the shape. Otherwise, it is
            # the transform. So lets see what type we're dealing with here.
            if cmds.nodeType(parent) in supported:
                geometry_shape = parent
            else:
                geometry_shape = cmds.listRelatives(parent, shapes=1)[0]

            if not geometry_shape:
                cmds.error("Skipping %s: Could not find shape." % element)
                return

            if len(cmds.ls(geometry_shape)) > 1:
                cmds.warning("Multiple shapes with identical "
                             "names found. This might not work")

        except TypeError as e:
            cmds.warning("Skipping %s: Didn't find a shape "
                         "for component elementection. %s" % (element, e))
            return

        try:
            type = cmds.nodeType(geometry_shape)

            if type == "nurbsSurface":
                # If a surfacePoint is elementected on a nurbs surface
                root, u, v = element.rsplit("[", 2)
                uv = [float(u[:-1]), float(v[:-1])]

            if type == "mesh":
                # -----------
                # Average the U and V values
                # ===========
                uvs = cmds.polyListComponentConversion(element, toUV=1)
                if not uvs:
                    cmds.warning("Couldn't derive any UV's from "
                                 "component, reverting to default U and V")
                    raise TypeError

                # Flatten list of Uv's as sometimes it returns
                # neighbors like this [2:3] instead of [2], [3]
                flattened = []

                for uv in uvs:
                    flattened.extend(cmds.ls(uv, flatten=True))

                uvs = flattened

                sumU = 0
                sumV = 0
                for uv in uvs:
                    try:
                        u, v = cmds.polyEditUV(uv, query=True)
                    except:
                        cmds.warning("Couldn't find any UV coordinated, "
                                     "reverting to default U and V")
                        raise TypeError

                    sumU += u
                    sumV += v

                averagedU = sumU / len(uvs)
                averagedV = sumV / len(uvs)

                uv = [averagedU, averagedV]
        except TypeError:
            pass

    return uv
Exemplo n.º 57
0
    def fetchShadersForSelected(self, tk):
        """
        Function to handle fetching the shaders for selected _hrc groups only.
        """
        inprogressBar = pbui.ProgressBarUI(title = 'Building Shaders For Selected Assets:')
        inprogressBar.show()
        inprogressBar.updateProgress(percent = 5, doingWhat = 'Processing scene info...')      
        if self.mcLoaded:
            ## ASSSIGN DEFAULT LAMBERT AND CLEAN THE HYERPSHADE!
            for each in cmds.ls(sl = True):
                try:
                    cmds.sets(each, e = True , forceElement = 'initialShadingGroup')
                except:
                    cmds.warning('FAILED to set initial Shading group for %s' % each)
                    pass
            [cmds.lockNode(cp, lock = True) for cp in cmds.ls(type = 'core_renderpass')] ## Lock all the core_renderpasses before deleting unused to preserve...
            mel.eval("MLdeleteUnused();")
            
            scene_path = '%s' % os.path.abspath(cmds.file(query=True, sn= True))
            debug(self.app, method = 'fetchShadersForSelected', message = 'scene_path... %s' % scene_path, verbose = False)
            
            ## Build an entity type to get some values from.
            entity = self.app.context.entity                                                                                    ## returns {'type': 'Shot', 'name': 'ep100_sh010', 'id': 1166}
            debug(self.app, method = 'fetchShadersForSelected', message = 'entity... %s' % entity, verbose = False)
                        
            ## Filter for the matching ID for the shot
            sg_filters = [["id", "is", entity["id"]]]
            debug(self.app, method = 'fetchShadersForSelected', message = 'sg_filters... %s' % sg_filters, verbose = False)
            
            ## Build an entity type to get some values from.
            sg_entity_type = self.app.context.entity["type"]                                                                   ## returns Shot
            debug(self.app, method = 'fetchShadersForSelected', message = 'sg_entity_type...\n%s' % sg_entity_type, verbose = False)
            
            ## DATA
            ## NOTES SO HERE WE DON'T NEED TO CALL THE ASSETS FIELD FROM SHOTGUN
            ## WE CAN JUST GRAB THE LATEST PUBLISH FILE FROM EACH OF THE TEMPLATE STEPS
            inprogressBar.updateProgress(percent = 10, doingWhat = 'Processing scene info...')
            shadersTemplate = tk.templates[self.app.get_setting('maya_asset_SHD_XML_template')]
            debug(self.app, method = 'fetchShadersForSelected', message = 'shadersTemplate...\n%s' % shadersTemplate, verbose = False)
            
            texturesTemplate = tk.templates[self.app.get_setting('maya_asset_textures_template')]
            debug(self.app, method = 'fetchShadersForSelected', message = 'texturesTemplate...\n%s' % texturesTemplate, verbose = False)
    
            ## Now get a list of assets in the scene
            inprogressBar.updateProgress(percent = 15, doingWhat = 'Processing assets...')
            inprogressBar.updateProgress(percent = 20, doingWhat = 'Processing xml...')
            assetDict = {} ## key: shotgunName  var: inSceneName
            for grp in cmds.ls(sl= True):
                if cmds.ls(grp, dag=True, type="mesh"):
                    getParent = cmds.listRelatives(grp, parent = True)
                    if getParent:
                        assetDict[grp.split('_hrc')[0]] = [cmds.listRelatives(grp, parent = True)[0], grp]
                    else:
                        assetDict[grp.split('_hrc')[0]] = ['', grp] ##make the parentGroup nothing so it paths to a root asset in the scene correctly

            debug(self.app, method = 'fetchShadersForSelected', message = 'Assets... %s' % assetDict, verbose = False)

            ## Now process XML
            debug(self.app, method = 'fetchShadersForSelected', message = 'Processing template... %s' % shadersTemplate, verbose = False)
            self.processSHDTemplate(tk = tk, templateFile = shadersTemplate, assetDict = assetDict, selected = True)
            
            self.finalBuildStuff(True, inprogressBar)

        else:
            inprogressBar.close()
            cmds.warning("NO MENTAL CORE FOUND!")
            pass
Exemplo n.º 58
0
def MCR_FrozenTransforms(xmlDefaults, *args):
    gradeValue = {
        'Aplus': int(xmlDefaults.find('gradeValue').find('Aplus').text),
        'F': int(xmlDefaults.find('gradeValue').find('F').text),
    }

    development = True
    if development:
        print 'MCR Frozen Transforms running'

    gMainProgressBar = maya.mel.eval('$tmp = $gMainProgressBar')

    cmds.progressBar(gMainProgressBar,
                     edit=True,
                     beginProgress=True,
                     isInterruptable=True,
                     status='Checking Transforms',
                     maxValue=100)
    step = 100 / 7
    cmds.progressBar(gMainProgressBar, edit=True, step=step)
    utils = utilities.utilities()
    cmds.progressBar(gMainProgressBar, edit=True, step=step)
    rootNodes = utils.masterGroupTest()
    cmds.progressBar(gMainProgressBar, edit=True, step=step)
    collectedNodes = utils.nodeCollector(rootNodes)
    cmds.progressBar(gMainProgressBar, edit=True, step=step)
    sortedNodes = utils.sortNodeList(collectedNodes, 'transform')
    cmds.progressBar(gMainProgressBar, edit=True, step=step)
    hasTransforms = utils.frozenTransforms(sortedNodes)

    cleanedNodes = []
    if hasTransforms:
        for node in hasTransforms:
            # print ('node: {}'.format(node))
            try:
                if cmds.listRelatives(node):
                    # print('listRelatives(node): {}'.format(cmds.listRelatives(node)))
                    f_type = cmds.nodeType(
                        cmds.listRelatives(node, path=True)[0])
                    if f_type not in ('imagePlane', 'camera'):
                        cleanedNodes.append(node)
            except RuntimeError:
                cmds.warning('RuntimeError Occurred - Adding node anyway...')
                cmds.warning('Problematic node: {}'.format(node))
                cleanedNodes.append(node)

    redFlagNodes = []

    for node in cleanedNodes:
        # print('checking node: {}'.format(node))
        # print('Relatives: {}'.format(cmds.listRelatives(node)))
        if cmds.listRelatives(node) != None:
            if len(cmds.listRelatives(node)) <= 2:
                # print('{} has <= 1 relatives'.format(node))
                # print('Relatives: {}'.format(cmds.listRelatives(node)))
                for relative in cmds.listRelatives(node, path=True):
                    if cmds.nodeType(relative) not in ('transform', 'shape'):
                        # print ('Red Flag!: {}'.format(node))
                        if node not in redFlagNodes:
                            redFlagNodes.append(node)

    if len(redFlagNodes):
        # print('red flag nodes!!!\n{}'.format(redFlagNodes))
        for node in redFlagNodes:
            # print('removing node: {}'.format(node))
            cleanedNodes.remove(node)

    if development:
        print 'cleanedNodes: {}'.format(cleanedNodes)

    cmds.progressBar(gMainProgressBar, edit=True, step=step)
    print "Frozen Transforms Automation Successful!"
    if cleanedNodes:
        # if development:
        # 	print cleanedNodes
        tempDict = {
            'grade_value': gradeValue['F'],
            'comment_text': '',
            'default_comments_text': 'Object has non frozen transforms.',
            'example_comments_text': ''
        }
    else:
        tempDict = {
            'grade_value': gradeValue['Aplus'],
            'comment_text': '',
            'default_comments_text': 'Transforms Frozen! Good job!',
            'example_comments_text': ''
        }

    cmds.progressBar(gMainProgressBar, edit=True, step=step)

    cmds.progressBar(gMainProgressBar, edit=True, endProgress=True)
    return tempDict
Exemplo n.º 59
0
    def assignLODs(self, *args):
        """
        Creates LOD group based off triangle count of meshes selected

        Warnings:
            'Select more than one mesh to generate LODs'
            'Multiple lodGroups are parented to selected Meshes'
        """
        if not self._hasSelection():
            return
        selection = cmds.ls(sl=True, l=True)
        #Checks if the user has selected at least 2 meshes
        if len(selection) < 2:
            cmds.warning('Select more than one mesh to generate LODs')
            return

        #Checks for LOD groups
        lodGroups = []
        lodMeshes = []
        collisionGroups = []
        for selected in selection:
            selectedShort = selected.split('|')[-1]
            collisionGroup = '|SM_%s|%s_Collision' % (selectedShort,
                                                      selectedShort)
            if cmds.objExists('|SM_%s|%s_Collision' %
                              (selectedShort, selectedShort)):
                collisionGroups.append(collisionGroup)
            parent = cmds.listRelatives(selected, f=True, p=True)
            if parent:
                cleanParent = parent[0][:parent[0].rfind('|')]
                if cleanParent:
                    if cmds.nodeType(cleanParent) == 'lodGroup':
                        if parent not in lodGroups:
                            lodGroups.append(cleanParent)
        if len(lodGroups) > 1:
            cmds.warning('Multiple LOD groups are parented to selected Meshes')
            return
        elif len(collisionGroups) > 1:
            cmds.warning(
                'Multiple collision groups are parented to selected Meshes')
            return
        elif len(lodGroups) == 1:
            cfd = cmds.confirmDialog(title='Confirm',
                                     message='Are trying to add LODs to %s?' %
                                     lodGroups[0].split('|')[1],
                                     button=['Yes', 'No'],
                                     defaultButton='Yes',
                                     cancelButton='No',
                                     dismissString='No')
            if cfd == 'No':
                return
            else:
                #Creates list of lodgroup relatives
                relatives = cmds.listRelatives('|%s' % lodGroups[0], f=True)
                for relative in relatives:
                    lod = str(cmds.listRelatives(relative, f=True))
                    for char in [' ', '[', ']']:
                        lod = lod.replace(char, "")
                    lod = lod.split(',')
                    for mesh in lod:
                        lodMeshes.append(unicode(mesh[2:-1]))
                #Removes duplicates
                for lodMesh in lodMeshes:
                    for selected in selection:
                        if selected in lodMesh:
                            selection.remove(selected)
                #need to research u'n' is in list when item under |lodGroup|lod_#|u'n' when mesh is deleted
                while 'n' in lodMeshes:
                    lodMeshes.remove('n')
        elif len(collisionGroups) == 1:
            cfd = cmds.confirmDialog(title='Confirm',
                                     message='Are trying to add LODs to %s?' %
                                     collisionGroups[0].split('|')[1],
                                     button=['Yes', 'No'],
                                     defaultButton='Yes',
                                     cancelButton='No',
                                     dismissString='No')
            if cfd == 'No':
                cmds.warning(
                    'Remove any selections under %s before trying to assign LODs'
                )
                return
            else:
                pass

        lodMeshes = lodMeshes + selection
        #Gets triangle count of mesh
        meshesInfo = []
        for lodMesh in lodMeshes:
            cmds.select(lodMesh)
            meshInfo = {}
            meshInfo['object'] = lodMesh
            meshInfo['triangles'] = cmds.polyEvaluate(t=True)
            meshesInfo.append(meshInfo)
        cmds.select(cl=True)

        #Sorts array by triangle count from high to low
        meshesInfo = sorted(meshesInfo,
                            key=operator.itemgetter('triangles'),
                            reverse=True)

        #Finds renderMeshName, lodGroup name, location in WS
        if lodGroups:
            renderMeshName = lodGroups[0].split('|')[1].replace('SM_', '')
        elif collisionGroups:
            renderMeshName = collisionGroups[0].split('|')[1].replace(
                'SM_', '')
        else:
            renderMeshName = self._checkRenderMeshName(
                meshesInfo[0]['object'].split('|')[-1])  #shortname
        lodGroup = renderMeshName + '_LOD'
        cmds.select(meshesInfo[0]['object'])

        #Renames and moves meshes to 0,0,0 in world space
        if cmds.listRelatives(meshesInfo[0]['object'], p=True):
            meshesInfo[0]['object'] = cmds.parent(meshesInfo[0]['object'],
                                                  w=True)[0]
        meshesInfo[0]['object'] = cmds.rename(meshesInfo[0]['object'],
                                              renderMeshName)
        for i, meshInfo in enumerate(meshesInfo[1:]):
            newName = 'LOD_%s_%02d' % (renderMeshName, i + 1)
            mesh = cmds.rename(meshInfo['object'], newName)
            if cmds.listRelatives(mesh, p=True):
                cmds.parent(mesh, w=True)
            meshInfo['object'] = newName

        #Deletes old LOD group
        if cmds.objExists('|SM_%s|%s' % (renderMeshName, lodGroup)):
            cmds.delete('|SM_%s|%s' % (renderMeshName, lodGroup))

        #Creates lodGroup based on triangle count
        for meshInfo in meshesInfo:
            cmds.select('|%s' % meshInfo['object'], add=True)
        cmds.xform(ws=True, t=(0, 0, 0))
        cmds.LevelOfDetailGroup()
        lodGroup = cmds.rename(lodGroup)
        for i in range(len(meshesInfo)):
            cmds.setAttr('%s.displayLevel[%d]' % (lodGroup, i), 1)
        renderMeshGroup = self.createMainGroup(renderMeshName)
        cmds.parent(lodGroup, renderMeshGroup)
        cmds.select(cl=True)
Exemplo n.º 60
0
    def fetchAllShaders(self, tk):        
        """
        Function to handle fetching the shaders
        """
        inprogressBar = pbui.ProgressBarUI(title = 'Building Shaders for All Assets:')
        inprogressBar.show()
        if self.mcLoaded:
            inprogressBar.updateProgress(percent = 5, doingWhat = 'Processing scene info...')      
            scene_path = '%s' % os.path.abspath(cmds.file(query=True, sn= True))
            debug(self.app, method = 'fetchAllShaders', message = 'scene_path... %s' % scene_path, verbose = False)
            
            ## Build an entity type to get some values from.
            entity = self.app.context.entity                                                                                    ## returns {'type': 'Shot', 'name': 'ep100_sh010', 'id': 1166}
            debug(self.app, method = 'fetchAllShaders', message = 'entity... %s' % entity, verbose = False)
                        
            ## Filter for the matching ID for the shot
            sg_filters = [["id", "is", entity["id"]]]
            debug(self.app, method = 'fetchAllShaders', message = 'sg_filters... %s' % sg_filters, verbose = False)
            
            ## Build an entity type to get some values from.
            sg_entity_type = self.app.context.entity["type"]                                                                   ## returns Shot
            debug(self.app, method = 'fetchAllShaders', message = 'sg_entity_type...\n%s' % sg_entity_type, verbose = False)
            
            ## DATA
            ## NOTES SO HERE WE DON'T NEED TO CALL THE ASSETS FIELD FROM SHOTGUN
            ## WE CAN JUST GRAB THE LATEST PUBLISH FILE FROM EACH OF THE TEMPLATE STEPS
            inprogressBar.updateProgress(percent = 10, doingWhat = 'Processing scene info...')
            shadersTemplate = tk.templates[self.app.get_setting('maya_asset_SHD_XML_template')]
            debug(self.app, method = 'fetchAllShaders', message = 'shadersTemplate...\n%s' % shadersTemplate, verbose = False)
            
            texturesTemplate = tk.templates[self.app.get_setting('maya_asset_textures_template')]
            debug(self.app, method = 'fetchAllShaders', message = 'texturesTemplate...\n%s' % texturesTemplate, verbose = False)

            ## Now get a list of assets in the scene
            inprogressBar.updateProgress(percent = 15, doingWhat = 'Processing assets...')
            inprogressBar.updateProgress(percent = 20, doingWhat = 'Processing UV and SHD XML files...')
            assetDict = {} ## key: shotgunName  var: inSceneName
            dupAssets = {}
            for parentGrp in cmds.ls(assemblies = True, long = True):
                if cmds.ls(parentGrp, dag=True, type="mesh"):
                    for each in cmds.listRelatives(parentGrp, children = True):
                        ## Check for duplicate or base assets
                        if not cmds.objExists('%s.dupAsset' % each):
                            assetDict[each.split('_hrc')[0]] = parentGrp
                            #{assetName: parentGrpName}
                        else: # handle the duplicate naming
                            origAssetName = each.split('_hrc')[0]
                            dupAssets[each] = [origAssetName, parentGrp]
                            #{duplicateGrName : [origAssetName, parentGrpName]}
                            debug(self.app, method = 'fetchAllShaders', message = 'DUPLICATE FOUND... origAssetName: %s' % origAssetName, verbose = False)

            debug(self.app, method = 'fetchAllShaders', message = 'Assets... %s' % assetDict, verbose = False)
            debug(self.app, method = 'fetchAllShaders', message = 'Duplicate Assets... %s' % dupAssets, verbose = False)
                                      
            ## Now process SHD XML
            debug(self.app, method = 'fetchAllShaders', message = 'Processing template... %s' % shadersTemplate, verbose = False)
            self.processSHDTemplate(tk = tk, templateFile = shadersTemplate, assetDict = assetDict, selected = False)
            
            self.finalBuildStuff(False, inprogressBar)
        else:
            inprogressBar.close()
            cmds.warning("NO MENTAL CORE FOUND!")