def build_all():
networks = libSerialization.getNetworksByClass('RigRoot')
for network in networks:
rigroot = libSerialization.import_network(network)
if rigroot.build():
pymel.delete(network)
libSerialization.export_network(rigroot)
def _actionRebuild(self):
for qItem in self.treeWidget.selectedItems():
rig = qItem.rig
if rig.isBuilt():
rig.unbuild()
rig.build()
#pymel.delete(rig._network) # TODO: AUTOMATIC UPDATE
libSerialization.export_network(rig)
def _actionUnbuild(self):
for qItem in self.treeWidget.selectedItems():
rig = qItem.rig
if rig.isBuilt():
rig.unbuild()
else:
pymel.warning("Can't unbuild {0}, already unbuilt.".format(rig))
#pymel.delete(rig._network) # TODO: AUTOMATIC UPDATE
libSerialization.export_network(rig)
def _actionAddPart(self, _cls):
part = _cls(_input=pymel.selected())
self.root.append(part)
net = libSerialization.export_network(self.root) # Export part and only part
pymel.select(net)
self.updateData()
self.updateUi()
def unbuild_all():
networks = libSerialization.getNetworksByClass('RigRoot')
for network in networks:
rigroot = libSerialization.import_network(network)
rigroot.unbuild()
pymel.delete(network)
# Write changes to scene
network = libSerialization.export_network(rigroot)
pymel.select(network)
def test_RigCtrl(self):
from classRigCtrl import RigCtrl
from omtk.libs import libSerialization
log.info("test_RigCtrl")
foo = RigCtrl()
foo.build()
pymel.setKeyframe(foo.node)
foo.unbuild()
network = libSerialization.export_network(foo)
pymel.select(network)
foo.build()
def from_selection():
import pymel.core as pymel
da = DigitalAsset(pymel.selected())
print da.__repr__()
libSerialization.export_network(da)
return da
def detect(*args, **kwargs):
"""
Fully automatic routine that create rig elements by analysing the joints structure.
This is only meant as a quick way to get started and is in no way production ready.
It is recommended that the 't-pose' or 45 angle 't-pose' is respected on the character before running this routine.
"""
jnts = pymel.ls(type='joint')
# Validate the joints hyerarchy since it is mandatory to autorig.
roots = []
for jnt in jnts:
root = jnt.root()
if root and root not in roots:
roots.append(root)
if len(roots) > 1:
log.error("There are more than one joint root in the scene. Please clean up.")
return None
root = next(iter(roots), None)
if not root:
log.error("Found no joint root.")
return None
# Get the rig heights and radius
height = 0
radius = 0
for jnt in jnts:
pos = jnt.getTranslation(space='world')
pos_x = pos.x
h = pos.y
pos_z = pos.z
r = pow(pow(pos_x, 2) + pow(pos_z, 2), 0.5)
if h > height:
height = h
if r > radius:
radius = r
MINIMUM_HEIGHT=0.01
if height < MINIMUM_HEIGHT:
log.error("Skeletton height is too small. Expected more than {0}".format(MINIMUM_HEIGHT))
return None
MINIMUM_RADIUS = 0.01
if radius < MINIMUM_RADIUS:
log.error("Skeletton radius is too small. Expected more than {0}".format(MINIMUM_RADIUS))
return None
#
# Configure Rig
#
rig = classRigRoot.RigRoot()
def get_arms(jnts):
chains = []
for jnt in jnts:
arm_jnts = get_arm(jnt)
if arm_jnts:
chains.append(arm_jnts)
return chains
# Detect hands?
def get_arm(jnt):
children = jnt.getChildren()
# Hand have a minimum of three fingers or not fingers at all
# todo: More robust!
if len(children) < 2 and len(children) != 0:
return False
# At least two parents (upperarm and forearm)
forearm = jnt.getParent()
upperarm = forearm.getParent() if isinstance(forearm, pymel.PyNode) else None
if not upperarm or not forearm:
return False
# Arms don't point upward, don't confuse ourself with the spine
upperarm_dir = get_direction(upperarm, forearm)
MAX_DIRECTION_Y = 0.75
if abs(upperarm_dir.y) > MAX_DIRECTION_Y:
return False
arm_jnts = [upperarm, forearm, jnt]
log.debug("Found Arm using {0}".format(arm_jnts))
return arm_jnts
def get_legs(jnts):
chains = []
for jnt in jnts:
leg_jnts = get_leg(jnt)
if leg_jnts:
chains.append(leg_jnts)
return chains
def get_direction(jnt_inn, jnt_out):
val = jnt_out.getTranslation(space='world') - jnt_inn.getTranslation(space='world')
val.normalize()
return val
def get_leg(jnt):
# The tip of the leg never have childrens
if len(jnt.getChildren()) > 0:
return False
# A leg have 5 joints from with the first two point to the ground
parents = []
parent = jnt
while parent:
parent = parent.getParent()
parents.append(parent)
if len(parents) < 5:
return False
thigh = parents[3]
calf = parents[2]
foot = parents[1]
toe = parents[0]
# Validate thigh direction
DIR_MINIMUM = -0.5
thigh_dir = get_direction(thigh, calf)
if thigh_dir.y >= DIR_MINIMUM:
return False
# Validate calf direction
calf_dir = get_direction(calf, foot)
if calf_dir.y >= DIR_MINIMUM:
return False
#print jnt, parents[3], parents[2],
leg_jnts = [thigh, calf, foot, toe, jnt]
log.debug("Found Leg using {0}".format(leg_jnts))
return leg_jnts
'''
def get_spines(jnts):
chains = []
for jnt in jnts:
spine_jnts = get_spine(jnt)
if spine_jnts:
chains.append(spine_jnts)
return chains
'''
log.debug("Detected rig layout:")
log.debug("\tHeight: {0}".format(height))
log.debug("\tRadius: {0}".format(radius))
# Detect legs
from rigLeg import Leg
legs_jnts = get_legs(jnts)
for leg_jnts in legs_jnts:
for leg_jnt in leg_jnts:
jnts.remove(leg_jnt)
rig.append(Leg(leg_jnts))
rig.append(Twistbone([leg_jnts[0], leg_jnts[1]]))
rig.append(Twistbone([leg_jnts[1], leg_jnts[2]]))
print len(jnts)
# Detect arms
from rigArm import Arm
arms_jnts = get_arms(jnts)
for arm_jnts in arms_jnts:
for arm_jnt in arm_jnts:
jnts.remove(arm_jnt)
rig.append(Arm(arm_jnts))
rig.append(Twistbone([arm_jnts[0], arm_jnts[1]]))
rig.append(Twistbone([arm_jnts[1], arm_jnts[2]]))
print len(jnts)
rig.build()
libSerialization.export_network(rig)