コード例 #1
0
 def get_bone_child(self, _bone):
     if Is.posebone(_bone):
         bone = _bone.bone
         if _bone.child:
             child = _bone.child.bone
         else:
             # No bone connected, so try to find a substitute
             child = self.get_closest_child(bone)
     else:
         bone = _bone
         child = None
         if len(bone.children) == 1:
             child = bone.children[0]
         elif bone.children:
             for bone_child in bone.children:
                 if bone_child.use_connect:
                     if (child is not None):
                         child_dist = Get.distance(child.tail, bone.tail)
                         active_dist = Get.distance(bone_child.tail,
                                                    bone.tail)
                         if (child_dist == active_dist):
                             # both children tails same distance from bone
                             # Just sort alphabetically
                             child = sorted(
                                 (b.name, b)
                                 for b in [child, bone_child])[0][1]
                             continue
                         elif (child_dist < active_dist):
                             # Previous is closes
                             continue
                     child = bone_child
             if child is None:
                 # No bone connected, so try to find a substitute
                 child = self.get_closest_child(bone)
     return (bone, child)
コード例 #2
0
    def get_closest_child(bone):
        if len(bone.children) == 1:
            return bone.children[0]
        elif bone.children:
            # Multiple children, None connected; search by distance
            children = dict()
            for bone_child in bone.children:
                dist = Get.distance(bone_child.head, bone.tail)
                if dist not in children:
                    children[dist] = list()
                children[dist].append(bone_child)

            closest = children[sorted(children)[0]]
            if len(closest) == 1:
                return closest[0]
            else:
                # Multiple children same distance from bone; check tail distanxce
                children = dict()
                for bone_child in closest:
                    dist = Get.distance(bone_child.tail, bone.tail)
                    if dist not in children:
                        children[dist] = list()
                    children[dist].append(bone_child)
                closest = children[sorted(children)[0]]
                if len(closest) == 1:
                    return closest[0]
                else:
                    # All children tails same distance from bone
                    # Just sort alphabetically
                    return sorted((b.name, b) for b in closest)[0][1]