import vs

## Flexsort, by LordAardvark
# Helper functions
def GetType( var ):
    return type(var).__name__

def GetChildren( root ):
    children = []

    for child in root.children:
        children += [child]

        if child.children:
            children += GetChildren(child)

    return children

# Globals
shot = sfm.GetCurrentShot()
animSet = sfm.GetCurrentAnimationSet()
gameModel = animSet.gameModel
rootGroup = animSet.GetRootControlGroup()

controlGroups = GetChildren(animSet.rootControlGroup)
addFlexes = []

backwardsAssoc = {}

#######################################################
## Add your flexes here at your leisure.
##
## The general syntax is as such:
## addFlexes.append( [ "Group1.Group2.Group3", vs.Color(r,g,b,a), [ "flex1", "flex2", "flex3" ] ] )
##
## The group hierarchy will be created as necessary, split along dots.
##
## Note color is only for the last group.
##
## Is it valid to have a group with no flexes, but only a color, as such:
## addFlexes.append( [ "Group1.Group2.Group3", vs.Color(r,g,b,a) ] )

# Generic facepose
addFlexes.append( [ "Flexes", vs.Color(0,128,255,255)] )
addFlexes.append( [ "Flexes.Face", vs.Color(255,128,32,255)] )
addFlexes.append( [ "Flexes.Face.Eyes", vs.Color(255,128,32,255)] )
addFlexes.append( [ "Flexes.Face.Upper Face", vs.Color(255,128,32,255)] )
addFlexes.append( [ "Flexes.Face.Upper Face.Eyelids", vs.Color(32,196,196,255), [ "EyelidBottomRaise", "EyelidBottomLower", "EyelidTopLower", "EyelidTopRaise" ] ] )
addFlexes.append( [ "Flexes.Face.Upper Face.Eyebrows", vs.Color(32,196,196,255), [ "EyebrowRaise", "EyebrowLower", "InnerCloser", "InnerLower", "InnerRaise", "OuterLower", "OuterRaise" ] ] )
addFlexes.append( [ "Flexes.Face.Mid Face", vs.Color(255, 128, 32,255), [ "CheekPuff", "CheekRaise", "NoseFlare", "NoseWrinkler" ] ] )
addFlexes.append( [ "Flexes.Face.Lower Face", vs.Color(255, 128, 32,255) ] )
addFlexes.append( [ "Flexes.Face.Lower Face.Jaw", vs.Color(32, 196, 196,255), [ "JawDrop", "JawRecede", "JawRise" ] ] )
addFlexes.append( [ "Flexes.Face.Lower Face.LipShape", vs.Color(32, 196, 196,255), [ "LipsDepressorBig", "LipsDepressorSmall", "LipsForwardSmall", "LipsForwardBig", "LipsForwardFull", "LipsNarrowFull", "LipsNarrowBig", "LipsNarrowSmall", "SmileBigISwear", "SmileKawaii", "LipsWiden", "LipCornerIn", "LipCornerOut", "LipCornerRaise", "LipCornerLow" ] ] )
addFlexes.append( [ "Flexes.Face.Lower Face.Lips", vs.Color(32, 196, 196,255), [ "LipBottomCurlIn", "LipBottomCurlOut", "LipBottomOut", "LipBottomLower", "LipBottomRaise", "LipTopCurlIn", "LipTopCurlOut", "LipTopOut", "LipTopRaise", "LipTopLower"] ] )

# Daz bodies
addFlexes.append( [ "Flexes.Genitals", vs.Color(255,128,32,255), [ "ClosedPussy", "PussyOpen", "PussyOpenPlus", "ButtOpen", "AnusOpen", "AnusOpenPlus" ] ] )

## End what end-users should be touching
#######################################################