Spaces:
Building
Building
File size: 1,997 Bytes
d49f7bc |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 |
# Copyright (c) Meta Platforms, Inc. and affiliates.
# This source code is licensed under the MIT license found in the
# LICENSE file in the root directory of this source tree.
from __future__ import annotations
from animated_drawings.model.transform import Transform
from typing import List
class Joint(Transform):
"""
Skeletal joint used representing character poses.
"""
def __init__(self, **kwargs) -> None:
super().__init__(**kwargs)
def joint_count(self) -> int:
""" Returns 1 + the number of Joint children in this joint's kinematic chain (recursive) """
count: int = 1
for c in self.get_children():
if isinstance(c, Joint):
count += c.joint_count()
return count
def get_chain_worldspace_positions(self) -> List[float]:
""" Get xzy worldspace coordinates of all joints within the chain. """
self.update_transforms(update_ancestors=True)
return self._get_chain_worldspace_positions(self, [])
def _get_chain_worldspace_positions(self, joint: Joint, position_list: List[float]) -> List[float]:
position_list.extend(joint.get_world_position(update_ancestors=False))
for c in joint.get_children():
if not isinstance(c, Joint):
continue
self._get_chain_worldspace_positions(c, position_list)
return position_list
def get_chain_joint_names(self):
""" Traverse through joint in depth-first order and return names of joints in the order they are visited. """
joint_names: List[str] = []
return self._get_chain_joint_names(self, joint_names)
def _get_chain_joint_names(self, joint: Joint, joint_name_list: List[str]) -> List[str]:
joint_name_list.append(str(joint.name))
for c in joint.get_children():
if not isinstance(c, Joint):
continue
self._get_chain_joint_names(c, joint_name_list)
return joint_name_list
|