Esempio n. 1
0
    def applyUvToWorldRatioToSelection(self, targetRatio=None):
        oSel = pm.ls(sl=True)

        pm.mel.eval("ConvertSelectionToFaces;")

        sel = pm.ls(sl=True)

        if targetRatio == None:
            targetRatio = input()
        currentRatio = self.calcUvToWorldAreaRatio()
        amountToScaleBeforeSqrt = targetRatio / currentRatio
        amountToScale = math.sqrt(amountToScaleBeforeSqrt)

        ## Calc a bounding box in uv space, and it's center, for a scale pivot
        bb = pm.polyEvaluate(sel, boundingBoxComponent2d=True)
        bbUMin = bb[0][0]
        bbUMax = bb[0][1]
        bbVMin = bb[1][0]
        bbVMax = bb[1][1]
        bbUCenter = (bbUMax + bbUMin) * 0.5
        bbVCenter = (bbVMax + bbVMin) * 0.5

        pm.polyEditUV(sel, pu=bbUCenter, pv=bbVCenter, su=amountToScale, sv=amountToScale)

        pm.select(oSel)
Esempio n. 2
0
    def applyUvToWorldRatioToSelection(self, targetRatio=None):
        oSel = pm.ls(sl=True)

        pm.mel.eval("ConvertSelectionToFaces;")

        sel = pm.ls(sl=True)

        if targetRatio == None:
            targetRatio = input()
        currentRatio = self.calcUvToWorldAreaRatio()
        amountToScaleBeforeSqrt = targetRatio / currentRatio
        amountToScale = math.sqrt(amountToScaleBeforeSqrt)

        ## Calc a bounding box in uv space, and it's center, for a scale pivot
        bb = pm.polyEvaluate(sel, boundingBoxComponent2d=True)
        bbUMin = bb[0][0]
        bbUMax = bb[0][1]
        bbVMin = bb[1][0]
        bbVMax = bb[1][1]
        bbUCenter = (bbUMax + bbUMin) * 0.5
        bbVCenter = (bbVMax + bbVMin) * 0.5

        pm.polyEditUV(sel,
                      pu=bbUCenter,
                      pv=bbVCenter,
                      su=amountToScale,
                      sv=amountToScale)

        pm.select(oSel)
Esempio n. 3
0
    def unfold3dOnlySelected(self):

        origSel = pm.ls(selection=True)

        pm.mel.eval("PolySelectConvert 4;")  ##"ConvertSelectionToUVs;" )
        origUvs = pm.ls(selection=True)
        pm.mel.eval("InvertSelection;")

        selCount = len(pm.ls(selection=True))

        ## In case all uvs are selected, the inverted selection will be zero!
        if selCount == 0:
            pm.select(origSel)
            pm.mel.eval("ConvertSelectionToUVs;")
            pm.mel.eval(
                "Unfold3D -unfold -iterations 1 -p 1 -borderintersection 1 -triangleflip 1;"
            )
            pm.select(origSel)
        else:
            ## Beware!  there's a huge potential bug  in this commands behavior
            ##   the way it is now working here is to get u and v  values, as back to back entries added into a list
            ##   so the list is twice as long as the number of uvs   eg.   v = coords[i*2  +1]
            uvsToRestore = pm.ls(selection=True, flatten=True)
            coords = pm.polyEditUV(uvsToRestore,
                                   query=True,
                                   uValue=True,
                                   vValue=True)

            pm.select(origSel)
            pm.mel.eval("ConvertSelectionToUVs;")
            pm.mel.eval(
                "Unfold3D -unfold -iterations 1 -p 1 -borderintersection 1 -triangleflip 1;"
            )

            #print( uvsToRestore )
            #print len(coordsU)
            #print len( uvsToRestore )

            #print( coords )

            for i, uv in enumerate(uvsToRestore):
                #print( coords )
                pm.polyEditUV(uv,
                              relative=False,
                              uValue=coords[i * 2],
                              vValue=coords[i * 2 + 1])

            #cmds.Unfold3D( cmds.ls(sl=True), u=True, p=True, ite=1, bi=True, tf=True, ms=1024, rs=2 )

            pm.select(origSel)
Esempio n. 4
0
def sd_randomize_uvs(rand=0.3):
    """
    After selecting uv's in the uv editor, this script will move
    around uv's randomly according to the object with which they belong.
    :param rand: The distance to be moved randomized between -rand and rand.
    :return: None
    """
    # Check if rand is a float or integer.
    if isinstance(rand, basestring):
        raise ValueError('please input a float or integer value')

    # define selection.
    o_sel = pm.ls(sl=True, flatten=True)

    # Check if there are any objects in the selection that are not MeshUVs.
    non_mesh_uv_objects = [o for o in o_sel if not isinstance(o, pm.MeshUV)]
    if non_mesh_uv_objects:
        raise TypeError('please only select uv points')

    # List the objects that the UVs belong to.
    u_list = [o.name().split('.')[0] for o in o_sel]

    # Create a Dictionary containing the objects and the UV vertices selected.
    uv_coord_dict = {
        o: [
            u.index() for u in o_sel if o == u.name().split('.')[0]
        ] for o in u_list
    }

    # Create random numbers relating to the objects selected.
    random_nums = {
        o: [
            random.uniform(-rand, rand),
            random.uniform(-rand, rand)
        ] for o in u_list
    }

    # Cycle through the dictionary moving the object's contained UV vertices.
    for uv in uv_coord_dict:
        for vtx in uv_coord_dict[uv]:
            pm.polyEditUV(
                '{}.map[{}]'.format(uv, vtx),
                u=random_nums[uv][0],
                v=random_nums[uv][1],
                relative=True
            )

    return None
Esempio n. 5
0
    def unfold3dOnlySelected(self):

        origSel = pm.ls(selection=True)

        pm.mel.eval("PolySelectConvert 4;")  ##"ConvertSelectionToUVs;" )
        origUvs = pm.ls(selection=True)
        pm.mel.eval("InvertSelection;")

        selCount = len(pm.ls(selection=True))

        ## In case all uvs are selected, the inverted selection will be zero!
        if selCount == 0:
            pm.select(origSel)
            pm.mel.eval("ConvertSelectionToUVs;")
            pm.mel.eval("Unfold3D -unfold -iterations 1 -p 1 -borderintersection 1 -triangleflip 1;")
            pm.select(origSel)
        else:
            ## Beware!  there's a huge potential bug  in this commands behavior
            ##   the way it is now working here is to get u and v  values, as back to back entries added into a list
            ##   so the list is twice as long as the number of uvs   eg.   v = coords[i*2  +1]
            uvsToRestore = pm.ls(selection=True, flatten=True)
            coords = pm.polyEditUV(uvsToRestore, query=True, uValue=True, vValue=True)

            pm.select(origSel)
            pm.mel.eval("ConvertSelectionToUVs;")
            pm.mel.eval("Unfold3D -unfold -iterations 1 -p 1 -borderintersection 1 -triangleflip 1;")

            # print( uvsToRestore )
            # print len(coordsU)
            # print len( uvsToRestore )

            # print( coords )

            for i, uv in enumerate(uvsToRestore):
                # print( coords )
                pm.polyEditUV(uv, relative=False, uValue=coords[i * 2], vValue=coords[i * 2 + 1])

            # cmds.Unfold3D( cmds.ls(sl=True), u=True, p=True, ite=1, bi=True, tf=True, ms=1024, rs=2 )

            pm.select(origSel)
 def scale_uvs(self, s=None, u=None, v=None, pu=0, pv=0):
     if not s is None:
         u = s
         v = s
     pm.polyEditUV(su=u, sv=v, pu=pu, pv=pv)
 def move_uvs(self, u, v):
     pm.polyEditUV(u=u, v=v)
Esempio n. 8
0
    def bake( self, modelsPath="", highSuffix="", xnormalExe="", xmlFiles="", cageSuffix="", typeSuffixes="", cageOffset=None ):
        
        try:
            cmds.progressWindow( endProgress=True )
        except:
            pass  #print( "..." )
        cmds.progressWindow(isInterruptable=1)

        
        if highSuffix=="":
            highSuffix="_hi"
        
        if cageSuffix=="":
            cageSuffix= "_cage"
        
        if cageOffset==None:
            cageOffset=2.0
        
        if typeSuffixes=="":
            typeSuffixes=['normals','heights','vcols', 'occlusion', 'curvature', 'cavity']
        typeOfTypeSuffixes = type(typeSuffixes)
        if typeOfTypeSuffixes==type( "a_string" ) or typeOfTypeSuffixes==type( u"a_unicode_string" ):
            typeSuffixes = typeSuffixes.split(",")

        
        xnormal_exe = xnormalExe.replace("\\","/")
        folder_with_high_models = modelsPath.replace("\\","/")
        xml_files = xmlFiles.split(',')

        objExt = ".obj"
        
        ## Set pythons current working directory
        os.chdir( folder_with_high_models )
        
        ## Get the selected objects, store as both an original selection
        ## and a copy of the original selection, objs
        osel = pm.ls(selection=True)
        objs = osel[:]
        
        scriptIsCanceled = False
        
        cancelCheckSleepTimeStartup = 0.5
        cancelCheckSleepTime = 0.01
        
        for obj in objs:
            time.sleep( cancelCheckSleepTimeStartup )
            if cmds.progressWindow(query=1, isCancelled=1):
                print( "Script canceled by user,   eg. escape key." )
                scriptIsCanceled = True
                break
            try: ## try each obj
                ## Select the object and set up some basic variables,
                ## such as the name and the python cwd
                pm.select( obj )                
                obj_name = obj.name()
                cwd = os.getcwd()
                cwdUni = unicode(cwd)
                
                ## Triangulate before export
                print( obj )
                tri_op = pm.polyTriangulate( obj )
                
                ## Export lo,  triangulated
                export_target_including_file_and_path = os.getcwd().replace("\\","/") + "/" + obj_name + objExt
                pm.select( obj )
                pm.exportSelected(
                    export_target_including_file_and_path,sh = 1, pr = 1, typ = "OBJexport", es = 1, force=True
                )
                
                print( obj )
                
                try:
                    cageOffsetCustom = obj.mmmmBakerCageOffset.get()
                except:
                    cageOffsetCustom = cageOffset
                    print( "no custom cage offset found, using default" )
                
                trans_op = pm.polyMoveVertex( obj, localTranslateZ=cageOffsetCustom )
                
                ## Export Cage
                export_target_including_file_and_path = \
                    os.getcwd().replace("\\","/") + "/" + obj_name + cageSuffix + objExt
                pm.select( obj )                    
                pm.exportSelected( export_target_including_file_and_path,
                    sh = 1, pr = 1, typ = "OBJexport", es = 1, force=True
                )
                
                ## Cleanup cage offset and triangulation
                pm.delete( trans_op )
                pm.delete( tri_op )
                
                
                            #if obj_name.endswith( lo_suffix ):
                            #    high_name = obj_name.replace( lo_suffix, highSuffix )
                            #else:
                high_name = obj_name + highSuffix
                    

                for i, xml_file in enumerate(xml_files):
                    time.sleep( cancelCheckSleepTime )                
                    if cmds.progressWindow(query=1, isCancelled=1):
                        scriptIsCanceled=True
                        print( "Script canceled by user,   eg. escape key." )
                        break
                    
                    pm.select( obj )
                    pm.mel.eval(  "ConvertSelectionToUVs;" )
                    uvs = pm.ls(selection=True, flatten=True)
                    coUV =  pm.polyEditUV( uvs, query=True )
                    coU, coV = coUV[::2], coUV[1::2]
                    pm.select(obj)
                    
                    
                    """"
                    for c in coU:
                        if c >= 0.0 and c < 1.0:
                            udim1001=True  
                        if c >= 1.0 and c < 2.0:
                            udim1002=True    
                        if c >= 2.0 and c < 3.0:
                            udim1003=True    
                        if c >= 3.0 and c < 4.0:
                            udim1004=True
                    """

                            
                    dictOfUsedUvRanges = {}
                    for cIndex in xrange( len(coU) ):
                        tu = coU[cIndex]
                        tv = coV[cIndex]
                        tuInt = math.floor( tu )
                        tvInt = math.floor( tv )
                        
                        dictOfUsedUvRanges[ tuInt,tvInt ] = int( 1000 + (tuInt+1) + (tvInt*10) )
                        
                    #print(   "length of dictionary:")
                    #print(   len(  dictOfUsedUvRanges.keys()  )    )
                    #break
                    
                    
                    for keyRange, udim   in dictOfUsedUvRanges.items():
                        uInKey = keyRange[0]
                        vInKey = keyRange[1]
                        uOffset = uInKey * -1.0
                        vOffset = vInKey * -1.0
                        if uOffset==0.0:
                            uOffset=0.0  #in case of negative 0.0, f*****g hell!
                        if vOffset==0.0:
                            vOffset=0.0  #in case of negative 0.0, f*****g hell!
                            
                        
                        
                        # do something similar for vOffset.
                        ## edit xml export with uOffset and uOffset here!
                        
                    
                        """"
                        if udim1001==True: uOffsets.append( 0.0 )
                        if udim1002==True: uOffsets.append( -1.0 )
                        if udim1003==True: uOffsets.append( -2.0 )
                        if udim1004==True: uOffsets.append( -3.0 )
                        

                        
                        
                        for uOffset in uOffsets:
                        """
                        
                        time.sleep( cancelCheckSleepTime )
                        if cmds.progressWindow(query=1, isCancelled=1):
                            scriptIsCanceled=True
                            print( "Script canceled by user,   eg. escape key." )
                            break
                        xml_file_abs = os.path.join( cwdUni, xml_file )
                        print( xml_file_abs )
                        xml_file_modified = u'tmp_mmmmBaker.xml'
                        xml_file_modified_abs = os.path.join( cwdUni, xml_file_modified )
                        
                        xml_in_mem = ''
                        with open( xml_file_abs, 'r' ) as fh:
                            xml_in_mem = fh.read()
                        
                        xml_in_mem = xml_in_mem.replace(
                             'UOffset="0.', 'UOffset="' + str(uOffset)
                        )
                        xml_in_mem = xml_in_mem.replace(
                             'VOffset="0.', 'VOffset="' + str(vOffset)
                        )
                        
                        uOffsetAsInt = int( -1 * uOffset)
                        vOffsetAsInt = int( -1 * uOffset)
                        
                        
                        folder_with_high_models__no_trailing_slash = \
                            folder_with_high_models
                        if folder_with_high_models__no_trailing_slash.endswith("/"):
                            folder_with_high_models__no_trailing_slash =\
                                folder_with_high_models__no_trailing_slash[:-1]
                        xml_in_mem = xml_in_mem.replace(
                             "C:\\Users\\Public\\mmmmBaker",
                             folder_with_high_models.replace("/","\\")
                        )                        
                        
                        root = XmlTree.fromstring(xml_in_mem)
                        
                        ## Print the entire xml tree for reference'
                        #print( XmlTree.tostring( root ) )
                        
                        ## find used udims on objects
                        ## find u offset and v offset for low objects
                        ## set up a list to run a for loop for to
                        ## handle each udim found
                        
                        ## find low object
                        ## find cage object
                        ## find use cage option
                        #
                        ## find hi objects
                        ## find output

                        with open( xml_file_modified_abs, 'w' ) as fh:
                            fh.write( xml_in_mem )
                    
                        print( obj )
                        ##print( xml_in_mem )
                        print( uOffsetAsInt, vOffsetAsInt )    
                    
                    
                        try:
                            lo_to_copy_for_replacement = obj_name + objExt
                            cage_to_copy_for_replacement = obj_name + cageSuffix + objExt
                            hi_to_copy_for_replacement = obj_name + highSuffix + objExt
                                
                            shutil.copy2(lo_to_copy_for_replacement, "replaceme_lo" + objExt)
                            shutil.copy2(hi_to_copy_for_replacement, "replaceme_hi" + objExt)
                            shutil.copy2(cage_to_copy_for_replacement, "replaceme_cage" + objExt)

                            print( "xnormal subprocess starting...")
                            print(    xnormal_exe + " " + str(xml_file_modified)    )
                            subprocess.check_call( [ xnormal_exe, str(xml_file_modified)  ] )
                            print( "xnormal subprocess complete!  Continuing...")
                            for typeSuffix in typeSuffixes:
                                try:
                                    baked_name = "replaceme_" + typeSuffix + ".exr"
                                    proper_name = obj_name + "_" + "xml"+str(i).zfill(2) + "_" + "udim" + str( udim ) + "_" + typeSuffix + ".exr"
                                    shutil.copy2( baked_name, proper_name )
                                except:
                                    print(    traceback.format_exc()    )
                            
                        except:
                            print( traceback.format_exc() )

            except: ## except each obj
                print( traceback.format_exc() )
                
        pm.select( osel )
        
        ## Do something here to attempt to combine/composite textures together.
        ## Until then, just tell the user to use the photoshop action
        ## which should be provided with MmmmTools
        ## Add some explaination for how to use the action
        
        cmds.progressWindow( endProgress=True )
        if not scriptIsCanceled:
            print( "Baking complete!")
            print( "Baked (or at least attempted to bake) " + str( len(objs) ) + " objects." )
Esempio n. 9
0
    def bake(self,
             modelsPath="",
             highSuffix="",
             xnormalExe="",
             xmlFiles="",
             cageSuffix="",
             typeSuffixes="",
             cageOffset=None):

        try:
            cmds.progressWindow(endProgress=True)
        except:
            pass  #print( "..." )
        cmds.progressWindow(isInterruptable=1)

        if highSuffix == "":
            highSuffix = "_hi"

        if cageSuffix == "":
            cageSuffix = "_cage"

        if cageOffset == None:
            cageOffset = 2.0

        if typeSuffixes == "":
            typeSuffixes = [
                'normals', 'heights', 'vcols', 'occlusion', 'curvature',
                'cavity'
            ]
        typeOfTypeSuffixes = type(typeSuffixes)
        if typeOfTypeSuffixes == type(
                "a_string") or typeOfTypeSuffixes == type(u"a_unicode_string"):
            typeSuffixes = typeSuffixes.split(",")

        xnormal_exe = xnormalExe.replace("\\", "/")
        folder_with_high_models = modelsPath.replace("\\", "/")
        xml_files = xmlFiles.split(',')

        objExt = ".obj"

        ## Set pythons current working directory
        os.chdir(folder_with_high_models)

        ## Get the selected objects, store as both an original selection
        ## and a copy of the original selection, objs
        osel = pm.ls(selection=True)
        objs = osel[:]

        scriptIsCanceled = False

        cancelCheckSleepTimeStartup = 0.5
        cancelCheckSleepTime = 0.01

        for obj in objs:
            time.sleep(cancelCheckSleepTimeStartup)
            if cmds.progressWindow(query=1, isCancelled=1):
                print("Script canceled by user,   eg. escape key.")
                scriptIsCanceled = True
                break
            try:  ## try each obj
                ## Select the object and set up some basic variables,
                ## such as the name and the python cwd
                pm.select(obj)
                obj_name = obj.name()
                cwd = os.getcwd()
                cwdUni = unicode(cwd)

                ## Triangulate before export
                print(obj)
                tri_op = pm.polyTriangulate(obj)

                ## Export lo,  triangulated
                export_target_including_file_and_path = os.getcwd().replace(
                    "\\", "/") + "/" + obj_name + objExt
                pm.select(obj)
                pm.exportSelected(export_target_including_file_and_path,
                                  sh=1,
                                  pr=1,
                                  typ="OBJexport",
                                  es=1,
                                  force=True)

                print(obj)

                try:
                    cageOffsetCustom = obj.mmmmBakerCageOffset.get()
                except:
                    cageOffsetCustom = cageOffset
                    print("no custom cage offset found, using default")

                trans_op = pm.polyMoveVertex(obj,
                                             localTranslateZ=cageOffsetCustom)

                ## Export Cage
                export_target_including_file_and_path = \
                    os.getcwd().replace("\\","/") + "/" + obj_name + cageSuffix + objExt
                pm.select(obj)
                pm.exportSelected(export_target_including_file_and_path,
                                  sh=1,
                                  pr=1,
                                  typ="OBJexport",
                                  es=1,
                                  force=True)

                ## Cleanup cage offset and triangulation
                pm.delete(trans_op)
                pm.delete(tri_op)

                #if obj_name.endswith( lo_suffix ):
                #    high_name = obj_name.replace( lo_suffix, highSuffix )
                #else:
                high_name = obj_name + highSuffix

                for i, xml_file in enumerate(xml_files):
                    time.sleep(cancelCheckSleepTime)
                    if cmds.progressWindow(query=1, isCancelled=1):
                        scriptIsCanceled = True
                        print("Script canceled by user,   eg. escape key.")
                        break

                    pm.select(obj)
                    pm.mel.eval("ConvertSelectionToUVs;")
                    uvs = pm.ls(selection=True, flatten=True)
                    coUV = pm.polyEditUV(uvs, query=True)
                    coU, coV = coUV[::2], coUV[1::2]
                    pm.select(obj)
                    """"
                    for c in coU:
                        if c >= 0.0 and c < 1.0:
                            udim1001=True  
                        if c >= 1.0 and c < 2.0:
                            udim1002=True    
                        if c >= 2.0 and c < 3.0:
                            udim1003=True    
                        if c >= 3.0 and c < 4.0:
                            udim1004=True
                    """

                    dictOfUsedUvRanges = {}
                    for cIndex in xrange(len(coU)):
                        tu = coU[cIndex]
                        tv = coV[cIndex]
                        tuInt = math.floor(tu)
                        tvInt = math.floor(tv)

                        dictOfUsedUvRanges[tuInt,
                                           tvInt] = int(1000 + (tuInt + 1) +
                                                        (tvInt * 10))

                    #print(   "length of dictionary:")
                    #print(   len(  dictOfUsedUvRanges.keys()  )    )
                    #break

                    for keyRange, udim in dictOfUsedUvRanges.items():
                        uInKey = keyRange[0]
                        vInKey = keyRange[1]
                        uOffset = uInKey * -1.0
                        vOffset = vInKey * -1.0
                        if uOffset == 0.0:
                            uOffset = 0.0  #in case of negative 0.0, f*****g hell!
                        if vOffset == 0.0:
                            vOffset = 0.0  #in case of negative 0.0, f*****g hell!

                        # do something similar for vOffset.
                        ## edit xml export with uOffset and uOffset here!
                        """"
                        if udim1001==True: uOffsets.append( 0.0 )
                        if udim1002==True: uOffsets.append( -1.0 )
                        if udim1003==True: uOffsets.append( -2.0 )
                        if udim1004==True: uOffsets.append( -3.0 )
                        

                        
                        
                        for uOffset in uOffsets:
                        """

                        time.sleep(cancelCheckSleepTime)
                        if cmds.progressWindow(query=1, isCancelled=1):
                            scriptIsCanceled = True
                            print("Script canceled by user,   eg. escape key.")
                            break
                        xml_file_abs = os.path.join(cwdUni, xml_file)
                        print(xml_file_abs)
                        xml_file_modified = u'tmp_mmmmBaker.xml'
                        xml_file_modified_abs = os.path.join(
                            cwdUni, xml_file_modified)

                        xml_in_mem = ''
                        with open(xml_file_abs, 'r') as fh:
                            xml_in_mem = fh.read()

                        xml_in_mem = xml_in_mem.replace(
                            'UOffset="0.', 'UOffset="' + str(uOffset))
                        xml_in_mem = xml_in_mem.replace(
                            'VOffset="0.', 'VOffset="' + str(vOffset))

                        uOffsetAsInt = int(-1 * uOffset)
                        vOffsetAsInt = int(-1 * uOffset)


                        folder_with_high_models__no_trailing_slash = \
                            folder_with_high_models
                        if folder_with_high_models__no_trailing_slash.endswith(
                                "/"):
                            folder_with_high_models__no_trailing_slash =\
                                folder_with_high_models__no_trailing_slash[:-1]
                        xml_in_mem = xml_in_mem.replace(
                            "C:\\Users\\Public\\mmmmBaker",
                            folder_with_high_models.replace("/", "\\"))

                        root = XmlTree.fromstring(xml_in_mem)

                        ## Print the entire xml tree for reference'
                        #print( XmlTree.tostring( root ) )

                        ## find used udims on objects
                        ## find u offset and v offset for low objects
                        ## set up a list to run a for loop for to
                        ## handle each udim found

                        ## find low object
                        ## find cage object
                        ## find use cage option
                        #
                        ## find hi objects
                        ## find output

                        with open(xml_file_modified_abs, 'w') as fh:
                            fh.write(xml_in_mem)

                        print(obj)
                        ##print( xml_in_mem )
                        print(uOffsetAsInt, vOffsetAsInt)

                        try:
                            lo_to_copy_for_replacement = obj_name + objExt
                            cage_to_copy_for_replacement = obj_name + cageSuffix + objExt
                            hi_to_copy_for_replacement = obj_name + highSuffix + objExt

                            shutil.copy2(lo_to_copy_for_replacement,
                                         "replaceme_lo" + objExt)
                            shutil.copy2(hi_to_copy_for_replacement,
                                         "replaceme_hi" + objExt)
                            shutil.copy2(cage_to_copy_for_replacement,
                                         "replaceme_cage" + objExt)

                            print("xnormal subprocess starting...")
                            print(xnormal_exe + " " + str(xml_file_modified))
                            subprocess.check_call(
                                [xnormal_exe,
                                 str(xml_file_modified)])
                            print(
                                "xnormal subprocess complete!  Continuing...")
                            for typeSuffix in typeSuffixes:
                                try:
                                    baked_name = "replaceme_" + typeSuffix + ".exr"
                                    proper_name = obj_name + "_" + "xml" + str(
                                        i).zfill(2) + "_" + "udim" + str(
                                            udim) + "_" + typeSuffix + ".exr"
                                    shutil.copy2(baked_name, proper_name)
                                except:
                                    print(traceback.format_exc())

                        except:
                            print(traceback.format_exc())

            except:  ## except each obj
                print(traceback.format_exc())

        pm.select(osel)

        ## Do something here to attempt to combine/composite textures together.
        ## Until then, just tell the user to use the photoshop action
        ## which should be provided with MmmmTools
        ## Add some explaination for how to use the action

        cmds.progressWindow(endProgress=True)
        if not scriptIsCanceled:
            print("Baking complete!")
            print("Baked (or at least attempted to bake) " + str(len(objs)) +
                  " objects.")
def create_mesh( xml_path, name, locator=None ) :
	pm.select( None )
	tree = et.parse( xml_path )
	root_keys = tree.getroot().keys()
	if( not 'imagePath' in root_keys ) :
		pm.error( 'The XML file does not appear to be a Texture Packer XML' )
	texture_size = ( tree.getroot().get( 'width' ), tree.getroot().get( 'height' ) )

	sprite = tree.find( ".//*[@n='%s']" % ( name ) )
	material = create_material( xml_path )

	if( not len( material.shadingGroups() ) ) :
		pm.error( 'Material %s is not connected to a Shading Group. Aborting.' )

	plane_name = __get_filename_noext( name ) + '_G'
	if( locator ) :
		plane_name = '%s_%s' % ( locator.name(), plane_name )
	
	try : 
		plane_trans = pm.PyNode( plane_name )		
	except :
		# v = pm.datatypes.Vector( float(sprite.get('w')) / 100.0, float(sprite.get('h')) / 100.0 )
		# v = v.normal()
		w_scale = float(sprite.get('w')) / 100.0
		h_scale = float(sprite.get('h')) / 100.0
		w_h = w_scale / h_scale
		h_w = h_scale / w_scale
		if( w_h > h_w ) :
			wp = 1.0 / w_h
			# print wp
			w_scale = 1.0
			h_scale = 1.0 / w_h
			# print 'w_h', w_scale, h_scale
		else :
			hp = 1.0 / h_w
			# print hp
			h_scale = 1.0
			w_scale = 1.0 / h_w
			# print w_h, h_w
			# print 'h_w', w_scale, h_scale
		plane_trans = pm.polyPlane(
			name=plane_name,
			sw=1, sh=1,			
			w=w_scale, h=h_scale
		)[0]
	plane_shape = plane_trans.getShape()

	if( not plane_trans.hasAttr( THU_MFT_SPRITE_ATTR ) ) :
		plane_trans.addAttr( THU_MFT_SPRITE_ATTR, dt='string' )
	plane_trans.setAttr( THU_MFT_SPRITE_ATTR, name.replace( '.png', '' ) )

	pm.sets( material.shadingGroups()[0], edit=True, forceElement=plane_trans )

	sx = ( float( sprite.get( 'x' ) ) / float( texture_size[0] ) )
	sy = 1 - ( float( sprite.get( 'y' ) ) / float( texture_size[1] ) )
	sw = ( float( sprite.get( 'w' ) ) / float( texture_size[0] ) )
	sh = ( float( sprite.get( 'h' ) ) / float( texture_size[1] ) )

	uv_positions = (
		( sx, sy - sh ),		
		( sx + sw, sy - sh ),
		( sx, sy ),
		( sx + sw, sy )
	)

	for uv, uv_position in zip( plane_shape.uvs, uv_positions ) :
		pm.polyEditUV( uv, r=False, u=uv_position[0], v=uv_position[1] )

	if( locator ) :
		# print locator
		plane_trans.setParent( locator )
		plane_trans.setTranslation( ( 0, 0, 0 ) )
		plane_trans.setRotation( ( 0, 0, 0 ) )
		plane_trans.setScale( ( 1, 1, 1 ) )		
		locator_bounds = locator.getBoundingBox()
		plane_bounds = plane_trans.getBoundingBox()
		if( plane_bounds.width() > plane_bounds.height() ) :
			pass
		else :
			s = locator_bounds.height() / plane_bounds.height()
			# plane_trans.setScale( (s, s, s) )
			
		

	pm.select( plane_trans )
	return plane_trans
Esempio n. 11
0
    def createRivet(self):
        """
        Create one rivet for each selected vertex
        """
        selObjs = pm.ls(selection=True)
        if not selObjs:
            ## Nothing is selected, so print out warning and raise error        
            print( self.emsgPleaseSelect )
            return
            #raise Exception, self.emsgPleaseSelect
        self.originalSel = selObjs  #### Store this step of of the selection in case we want it again later

        uvs = []
        
                
        uvs.extend( pm.polyListComponentConversion( selObjs, tuv=True )  ) #### Change this to the full version
              
        #uvs.extend( [uv for uv in selObjs if '.uv' in uv] )  #### This is a very good list comprehension, perhaps expand it though
            ## the extend method of a list adds all items from another given list to the list

        uvsFromNurbs = []
        for i in selObjs:
            if pm.objectType( i ) == 'nurbsSurface' :
                uvs.append( i )

        
        ##  select our new, smaller/filtered uvs       
        pm.select( uvs )
        uvs = pm.ls( flatten = True, selection = True )  ## The flatten command returns each component individually in the list, rather than using strings that specify ranges of selections.  It takes more RAM but is often much more suitable to work with.

        ## Create a group so that we can organize the rivets  -  **** Note that this should be improved with the MmmmTools upcoming unique naming system
        if not pm.objExists('_follicle_grp'):    #### This line and the next should eventually be improved to not use a hardcoded name
            group = pm.group( em = True, w=True, n='_follicle_grp' )    #### **** Totally recreate this line, use a variable name or at least a unique one
        else:
            group = pm.PyNode('_follicle_grp')
            
        rivets = []
        
        pm.select( selObjs )

        failCount = 0

        ## Give an error msg if the user didn't use the script on a compatible selection
        if not uvs:
            failCount += 1
            print( self.emsgPleaseSelect )

        ## Everything is good, proceed to investigate and create rivets
        for uv in uvs:
                ## The commented out print line are simple useful for debugging
                print pm.objectType( uv )                
                
                objShapeName, index = tuple( uv.split( '.', 1 )  )
                obj = pm.PyNode( objShapeName )
                loc = pm.createNode( 'locator' )
                tr = getT( loc ) 
                #print( "Transform was: " + tr )
                hair = pm.createNode( 'follicle', parent = tr )
                pm.parent( tr, group )  ####  This line sucks because it's using a f*****g stupid name again
                rivets.append( tr )
                
                ## Poly mesh handler
                if pm.objectType( obj ) == 'mesh':
                    obj.outMesh >> hair.inputMesh
                    uvPos = pm.polyEditUV( uv, query=True )
                ## Nurbs surface handler
                elif pm.objectType( obj ) == 'nurbsSurface':
                    obj.local >> hair.inputSurface
                    ## The index is a messy string, so we need to pull uv data out of it
                    uvTuple = ( index.strip('.uv[]').split('][')  )
                    ## We need to create the tuple as floats, because we got it as strings
                    uvPos = (    float(uvTuple[0]), float(uvTuple[1])     )
                    #uvPos = ( uvTuple[0], uv[1] )#index.strip('.uv[]').split('][')

                    ## Handle conditions where the uvs aren't normalized, this may not be required often
                    maxU = float(  obj.maxValueU.get()     )
                    maxV = float(  obj.maxValueV.get()     )
                    uvPos = (    uvPos[0]/maxU,  uvPos[1]/maxV   )
                ## Handle other cases, where this script can't do anything useful
                else:
                    print( obj + ' with uv: ' + uv + \
                        '   was incompatible, it much be either polygons or Nurbs' )
                    failCount += 1                        
                    continue
                    
                u, v = uvPos                    

                ## Make the hair follow the model, both by parenting, and by keeping its translate and rotate matched
                obj.worldMatrix >> hair.inputWorldMatrix
                hair.outTranslate >> tr.translate
                hair.outRotate >> tr.rotate   ## Note:  The hair has no outScale
                
                ## Set the u and v parameters of the hair so that it sticks to the correct place on the model
                hair.parameterU.set( u )
                hair.parameterV.set( v )
                
                ## Put the rivet into a group so we can select it afterwards
                self.lastRivetsCreated.append( loc )
           
        ## Select all the new rivets we created
        pm.select( self.lastRivetsCreated, replace=True )
        if failCount:
            print( str(failCount) + """ rivets failed to be created. Most likely because the selection was not correct. Try selecting vertices on a nurbs surface or a polygon mesh and running the script again. """)
        else:
            print( self.msgSucess )
        return       
Esempio n. 12
0
def create_mesh(xml_path, name, locator=None):
    pm.select(None)
    tree = et.parse(xml_path)
    root_keys = tree.getroot().keys()
    if (not 'imagePath' in root_keys):
        pm.error('The XML file does not appear to be a Texture Packer XML')
    texture_size = (tree.getroot().get('width'), tree.getroot().get('height'))

    sprite = tree.find(".//*[@n='%s']" % (name))
    material = create_material(xml_path)

    if (not len(material.shadingGroups())):
        pm.error('Material %s is not connected to a Shading Group. Aborting.')

    plane_name = __get_filename_noext(name) + '_G'
    if (locator):
        plane_name = '%s_%s' % (locator.name(), plane_name)

    try:
        plane_trans = pm.PyNode(plane_name)
    except:
        # v = pm.datatypes.Vector( float(sprite.get('w')) / 100.0, float(sprite.get('h')) / 100.0 )
        # v = v.normal()
        w_scale = float(sprite.get('w')) / 100.0
        h_scale = float(sprite.get('h')) / 100.0
        w_h = w_scale / h_scale
        h_w = h_scale / w_scale
        if (w_h > h_w):
            wp = 1.0 / w_h
            # print wp
            w_scale = 1.0
            h_scale = 1.0 / w_h
            # print 'w_h', w_scale, h_scale
        else:
            hp = 1.0 / h_w
            # print hp
            h_scale = 1.0
            w_scale = 1.0 / h_w
            # print w_h, h_w
            # print 'h_w', w_scale, h_scale
        plane_trans = pm.polyPlane(name=plane_name,
                                   sw=1,
                                   sh=1,
                                   w=w_scale,
                                   h=h_scale)[0]
    plane_shape = plane_trans.getShape()

    if (not plane_trans.hasAttr(THU_MFT_SPRITE_ATTR)):
        plane_trans.addAttr(THU_MFT_SPRITE_ATTR, dt='string')
    plane_trans.setAttr(THU_MFT_SPRITE_ATTR, name.replace('.png', ''))

    pm.sets(material.shadingGroups()[0], edit=True, forceElement=plane_trans)

    sx = (float(sprite.get('x')) / float(texture_size[0]))
    sy = 1 - (float(sprite.get('y')) / float(texture_size[1]))
    sw = (float(sprite.get('w')) / float(texture_size[0]))
    sh = (float(sprite.get('h')) / float(texture_size[1]))

    uv_positions = ((sx, sy - sh), (sx + sw, sy - sh), (sx, sy), (sx + sw, sy))

    for uv, uv_position in zip(plane_shape.uvs, uv_positions):
        pm.polyEditUV(uv, r=False, u=uv_position[0], v=uv_position[1])

    if (locator):
        # print locator
        plane_trans.setParent(locator)
        plane_trans.setTranslation((0, 0, 0))
        plane_trans.setRotation((0, 0, 0))
        plane_trans.setScale((1, 1, 1))
        locator_bounds = locator.getBoundingBox()
        plane_bounds = plane_trans.getBoundingBox()
        if (plane_bounds.width() > plane_bounds.height()):
            pass
        else:
            s = locator_bounds.height() / plane_bounds.height()
            # plane_trans.setScale( (s, s, s) )

    pm.select(plane_trans)
    return plane_trans
 def scale_uvs( self, s=None, u=None, v=None, pu=0,pv=0 ):
     if not s is None:
         u = s
         v = s
     pm.polyEditUV( su=u, sv=v, pu=pu, pv=pv )
 def move_uvs( self, u, v ):
     pm.polyEditUV( u=u, v=v )