API tools faq
paste
Advertisement
bradon

RandomTransform

bradon
Mar 18th, 2013
39
0
Never
Add comment
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 17.78 KB | None | 0 0
raw download clone embed print report
  1. #===============================================================================
  2. # Randomly Transform an object or selected objects
  3. # author: Bradon Webb
  4. # [email protected]
  5. # 03/18/13
  6. #
  7. #
  8. # Run script for UI
  9. #
  10. #
  11. # ==============================================================================
  12.  
  13.  
  14. import maya.cmds as mc
  15. import random
  16.  
  17. def randomTransform(selectedNodes=None, minTranslation=0, maxTranslation=10, minRotation=0, maxRotation=360, minScale=.01, maxScale=1,
  18.                                axes=['X','Y','Z'], translateRandom=True, rotateRandom=True, scaleRandom=True, scaleUniform=True):
  19.     '''
  20.    @summary:
  21.        randomly transforms an input object(s) OR if none it will randomly transform the selected objects
  22.    @arguments:
  23.        selectedNodes = list of nodes or will use a selection if none specified
  24.        minTranslation = float for minimum translate
  25.        maxTranslation = float for maximum translate
  26.        minRotation = float for minimum translate
  27.        maxRotation = float for maximum translate
  28.        minScale = float for minimum translate
  29.        maxScale = float for maximum translate
  30.        axes = list of axis to do randomizations on    
  31.        translateRandom = Boolean to do translation channels
  32.        rotateRandom = Boolean to do Rotation channels    
  33.        scaleRandom = Boolean to do Scale channels
  34.        scaleUniform = Boolean to keep Scale values uniform
  35.    '''
  36.  
  37.     # If no input mehses specified, then store selected objects in a list    
  38.     if not selectedNodes:
  39.         selectedNodes = mc.ls(selection=True, long=True, type='transform')
  40.    
  41.     # If no objects are selected error informatively
  42.     if not selectedNodes:
  43.         raise RuntimeError, 'please select an object to randomly transform'
  44.                  
  45.     # For each selected mesh, and each specified axis, loop with random values for translation rotation and scale                  
  46.     for selectedNode in selectedNodes:
  47.         # Random Scale for Uniform Scaling
  48.         randomScaleValue = random.uniform(minScale, maxScale)
  49.        
  50.         if selectedNode and axes:
  51.             for axis in axes:
  52.                 if translateRandom:
  53.                     randomTranslateValue = random.uniform(minTranslation, maxTranslation)
  54.                     mc.setAttr('{selectedNode}.translate{axis}'.format(selectedNode=selectedNode, axis=axis.upper()), randomTranslateValue)
  55.                
  56.                 if rotateRandom:
  57.                     randomRotationValue = random.uniform(minRotation, maxRotation)
  58.                     mc.setAttr('{selectedNode}.rotate{axis}'.format(selectedNode=selectedNode, axis=axis.upper()), randomRotationValue)
  59.                
  60.                 if scaleRandom:
  61.                     if not scaleUniform:      
  62.                         randomScaleValue = random.uniform(minScale, maxScale)
  63.                     mc.setAttr('{selectedNode}.scale{axis}'.format(selectedNode=selectedNode, axis=axis.upper()), randomScaleValue)                    
  64.  
  65. # Begin UI -----------------------------------------------------------
  66.  
  67. def loadHelp(*args):
  68.     pass
  69.  
  70. def randomTransformUI():
  71.    
  72.     def getUIDetails(*args):
  73.  
  74.         # check the axis Checkbox
  75.         Xaxis = mc.checkBox(XaxisCheckBox, query=True, value=True)
  76.         Yaxis = mc.checkBox(YaxisCheckBox, query=True, value=True)
  77.         Zaxis = mc.checkBox(ZaxisCheckBox, query=True, value=True)
  78.        
  79.         # deifnine axis variable to populate
  80.         axes = []
  81.        
  82.         # Add values to the axis list
  83.         if Xaxis:
  84.             axes.append('X')
  85.         if Yaxis:
  86.             axes.append('Y')
  87.         if Zaxis:
  88.             axes.append('Z')
  89.            
  90.         # if user has not specified an axis warn user    
  91.         if not axes:
  92.             raise RuntimeError, 'Please select at least one axis'
  93.  
  94.         # get UI detials
  95.         translateRandom = mc.checkBox(translateCheckBox, query=True, value=True)
  96.         rotateRandom = mc.checkBox(rotateCheckBox, query=True, value=True)
  97.         scaleRandom = mc.checkBox(scaleCheckBox, query=True, value=True)
  98.         scaleUniform = mc.checkBox(uniformScaleCheckBox, query=True, value=True)
  99.        
  100.         minTranslation = mc.floatField(translateMinFloat, query=True, value=True)
  101.         maxTranslation = mc.floatField(translateMaxFloat, query=True, value=True)
  102.        
  103.         minRotation = mc.floatField(rotateMinFloat, query=True, value=True)
  104.         maxRotation = mc.floatField(rotateMaxFloat, query=True, value=True)
  105.        
  106.         minScale = mc.floatField(scaleMinFloat, query=True, value=True)
  107.         maxScale = mc.floatField(scaleMaxFloat, query=True, value=True)
  108.        
  109.        
  110.         #-- Debug
  111.         '''        
  112.        print axes
  113.        print translateRandom
  114.        print rotateRandom
  115.        print scaleRandom
  116.        print scaleUniform
  117.        print minTranslation
  118.        print maxTranslation
  119.        print minRotation
  120.        print maxRotation
  121.        print minScale
  122.        print maxScale
  123.        '''
  124.        
  125.         # run function with args
  126.         randomTransform(selectedNodes=None,
  127.                         minTranslation=minTranslation,
  128.                         maxTranslation=maxTranslation,
  129.                         minRotation=minRotation,
  130.                         maxRotation=maxRotation,
  131.                         minScale=minScale,
  132.                         maxScale=maxScale,
  133.                         axes=axes,
  134.                         translateRandom=translateRandom,
  135.                         rotateRandom=rotateRandom,
  136.                         scaleRandom=scaleRandom,
  137.                         scaleUniform=scaleUniform,)
  138.        
  139.         #End Get Details
  140.        
  141.     def updateUI(*args):
  142.         # Check the state of the checkBoxes
  143.         translateCheckBoxState = mc.checkBox(translateCheckBox, query=True, value=True)
  144.         rotateCheckBoxState = mc.checkBox(rotateCheckBox, query=True, value=True)
  145.         scaleCheckBoxState = mc.checkBox(scaleCheckBox, query=True, value=True)
  146.        
  147.         # update the enable state on or off depending on the check box state
  148.         mc.floatField(translateMinFloat, edit=True, enable=translateCheckBoxState)
  149.         mc.floatField(translateMaxFloat, edit=True, enable=translateCheckBoxState)
  150.        
  151.         mc.floatField(rotateMinFloat, edit=True, enable=rotateCheckBoxState)
  152.         mc.floatField(rotateMaxFloat, edit=True, enable=rotateCheckBoxState)
  153.        
  154.         mc.floatField(scaleMinFloat, edit=True, enable=scaleCheckBoxState)
  155.         mc.floatField(scaleMaxFloat, edit=True, enable=scaleCheckBoxState)
  156.        
  157.        
  158.        
  159.        
  160.     # Define new window name
  161.     newWindow = 'randomTransform'    
  162.    
  163.     # Check to see if the Window exists, Delete Window
  164.     if mc.window(newWindow, query=True, exists=True):
  165.         mc.deleteUI(newWindow)
  166.    
  167.     # Define window attributes
  168.     mc.window(newWindow, title='randomTransform', sizeable=True)
  169.    
  170.     # Create menu bar with help    
  171.     menuBarLayout = mc.menuBarLayout()
  172.     helpMenu = mc.menu( label='help', tearOff=False )
  173.     howToHelp = mc.menuItem( label='How to use', command=loadHelp)
  174.    
  175.     # Main Layout of all menu is a form
  176.     mainFormLayout = mc.formLayout()
  177.    
  178.     menuSeparator = mc.separator(style='in')
  179.    
  180.     axisLabel = mc.text('Axis')
  181.    
  182.     XaxisCheckBox = mc.checkBox(label='X')
  183.     YaxisCheckBox = mc.checkBox(label='Y')
  184.     ZaxisCheckBox = mc.checkBox(label='Z')
  185.    
  186.     axisSeparator = mc.separator(style='in')
  187.    
  188.     translateLabel = mc.text('Translate')
  189.     translateCheckBox = mc.checkBox(label='', onCommand=updateUI, offCommand=updateUI)
  190.  
  191.     translateMinMaxLabel = mc.text(label='Min/Max')
  192.     translateMinFloat = mc.floatField(annotation='Translation Min', enable=False)
  193.     translateMaxFloat = mc.floatField(annotation='Translation Max', enable=False)
  194.    
  195.     translateSeparator = mc.separator(style='in')
  196.    
  197.     rotateLabel = mc.text('Rotate')
  198.     rotateCheckBox = mc.checkBox(label='', onCommand=updateUI, offCommand=updateUI)
  199.  
  200.     rotateMinMaxLabel = mc.text(label='Min/Max')
  201.     rotateMinFloat = mc.floatField(annotation='Rotate Min', enable=False)
  202.     rotateMaxFloat = mc.floatField(annotation='Rotate Max', enable=False)
  203.    
  204.     rotateSeparator = mc.separator(style='in')
  205.    
  206.     scaleLabel = mc.text('Scale')
  207.     scaleCheckBox = mc.checkBox(label='', onCommand=updateUI, offCommand=updateUI)
  208.    
  209.     uniformScaleLabel = mc.text('Uniform')
  210.     uniformScaleCheckBox = mc.checkBox(label='')
  211.  
  212.     scaleMinMaxLabel = mc.text(label='Min/Max')
  213.     scaleMinFloat = mc.floatField(annotation='Scale Min', enable=False)
  214.     scaleMaxFloat = mc.floatField(annotation='Scale Max', enable=False)
  215.    
  216.     scaleSeparator = mc.separator(style='in')
  217.    
  218.     randomizeButton = mc.button(label='Randomize Transforms', command=getUIDetails)
  219.    
  220.    
  221.     # Position the controls
  222.     mc.formLayout(mainFormLayout, edit=True, # attach to the form
  223.                                            attachForm=[(menuSeparator, 'left', 3),
  224.                                                        (menuSeparator, 'right', 3),
  225.                                                        (menuSeparator, 'top', 3),
  226.  
  227.                                                        (axisSeparator, 'left', 3),
  228.                                                        (axisSeparator, 'right', 3),
  229.                                                        
  230.                                                        (translateMinMaxLabel, 'left', 10),
  231.                                                        (translateMaxFloat, 'right', 10),
  232.                                                        
  233.                                                        (translateSeparator, 'left', 3),
  234.                                                        (translateSeparator, 'right', 3),
  235.                                                        
  236.                                                        (rotateMinMaxLabel, 'left', 10),
  237.                                                        (rotateMaxFloat, 'right', 10),
  238.                                                    
  239.                                                        (rotateSeparator, 'left', 3),
  240.                                                        (rotateSeparator, 'right', 3),
  241.                                                        
  242.                                                        (scaleMinMaxLabel, 'left', 10),
  243.                                                        (scaleMaxFloat, 'right', 10),
  244.                                                        
  245.                                                        (scaleSeparator, 'left', 3),
  246.                                                        (scaleSeparator, 'right', 3),                                                      
  247.                                                        
  248.                                                        (randomizeButton, 'left', 3),
  249.                                                        (randomizeButton, 'right', 3),
  250.                                                        (randomizeButton, 'bottom', 3),
  251.                                                                                                              
  252.                                            ],
  253.  
  254.                                            attachControl=[(XaxisCheckBox, 'top', 3, menuSeparator),
  255.                                                           (YaxisCheckBox, 'top', 3, menuSeparator),
  256.                                                           (ZaxisCheckBox, 'top', 3, menuSeparator),
  257.                                                           (axisLabel, 'top', 3, menuSeparator),
  258.                                                                                                                    
  259.                                                           (XaxisCheckBox, 'left', 3, axisLabel),
  260.                                                           (YaxisCheckBox, 'left', 3, XaxisCheckBox),
  261.                                                           (ZaxisCheckBox, 'left', 3, YaxisCheckBox),
  262.                                                          
  263.                                                           #--translate controls
  264.                                                           (axisSeparator, 'top', 3, XaxisCheckBox),
  265.                                                          
  266.                                                           (translateLabel, 'top', 10, axisSeparator),
  267.                                                           (translateCheckBox, 'top', 10, axisSeparator),
  268.                                                           (translateCheckBox, 'left', 3, translateLabel),
  269.                                                          
  270.                                                           (translateMinMaxLabel, 'top', 10, translateLabel),
  271.                                                           (translateMinFloat, 'top', 10, translateLabel),
  272.                                                           (translateMaxFloat, 'top', 10, translateLabel),                                                          
  273.                                                           (translateMinFloat, 'left', 3, translateMinMaxLabel),
  274.                                                          
  275.                                                           #--rotate controls
  276.                                                           (translateSeparator, 'top', 10, translateMinFloat),
  277.                                                          
  278.                                                           (rotateLabel, 'top', 10, translateSeparator),
  279.                                                           (rotateCheckBox, 'top', 10, translateSeparator),
  280.                                                           (rotateCheckBox, 'left', 3, rotateLabel),
  281.                                                          
  282.                                                           (rotateMinMaxLabel, 'top', 10, rotateLabel),
  283.                                                           (rotateMinFloat, 'top', 10, rotateLabel),
  284.                                                           (rotateMaxFloat, 'top', 10, rotateLabel),                                                          
  285.                                                           (rotateMinFloat, 'left', 3, rotateMinMaxLabel),
  286.                                                          
  287.                                                           #--scale controls
  288.                                                           (rotateSeparator, 'top', 10, rotateMinFloat),
  289.                                                          
  290.                                                           (scaleLabel, 'top', 10, rotateSeparator),
  291.                                                           (scaleCheckBox, 'top', 10, rotateSeparator),
  292.                                                           (scaleCheckBox, 'left', 3, scaleLabel),
  293.                                                          
  294.                                                           (uniformScaleLabel, 'top', 3, scaleLabel),
  295.                                                           (uniformScaleCheckBox, 'top', 3, scaleLabel),
  296.                                                           (uniformScaleCheckBox, 'left', 3, uniformScaleLabel),
  297.                                                            
  298.                                                           (scaleMinMaxLabel, 'top', 10, uniformScaleLabel),
  299.                                                           (scaleMinFloat, 'top', 10, uniformScaleLabel),
  300.                                                           (scaleMaxFloat, 'top', 10, uniformScaleLabel),                                                          
  301.                                                           (scaleMinFloat, 'left', 3, scaleMinMaxLabel),
  302.                                                          
  303.                                                           (scaleSeparator, 'top', 10, scaleMinFloat),
  304.                                                          
  305.                                            ],
  306.                                            
  307.                                            attachPosition=[(translateLabel, 'right', 2, 45),
  308.                                                            (axisLabel, 'right', 2, 45),
  309.  
  310.                                                            (translateMinFloat, 'right', 2, 50),
  311.                                                            (translateMaxFloat, 'left', 2, 50),
  312.                                                            
  313.                                                            (rotateLabel, 'right', 2, 45),
  314.                                                            
  315.                                                            (rotateMinFloat, 'right', 2, 50),
  316.                                                            (rotateMaxFloat, 'left', 2, 50),
  317.                                                            
  318.                                                            (scaleLabel, 'right', 2, 45),
  319.                                                            (uniformScaleLabel, 'right', 2, 45),
  320.                                                            
  321.                                                            (scaleMinFloat, 'right', 2, 50),
  322.                                                            (scaleMaxFloat, 'left', 2, 50),
  323.                                                            
  324.                                                            (translateSeparator, 'bottom', 2, 25),
  325.                                                            (rotateSeparator, 'bottom', 2, 50),
  326.                                                            (scaleSeparator, 'bottom', 2, 80),
  327.                                            ]
  328.                   )
  329.  
  330.     mc.showWindow(newWindow)
  331.    
  332. randomTransformUI()
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement
Public Pastes
Advertisement
We use cookies for various purposes including analytics. By continuing to use Pastebin, you agree to our use of cookies as described in the Cookies Policy.  OK, I Understand
Not a member of Pastebin yet?
Sign Up, it unlocks many cool features!