SYMBOL INDEX (276 symbols across 38 files)

FILE: anim/aistpp.py
  function load (line 23) | def load(

FILE: anim/amass.py
  function load (line 24) | def load(

FILE: anim/animation.py
  class Animation (line 17) | class Animation:
    method __init__ (line 18) | def __init__(
    method __len__ (line 44) | def __len__(self) -> int: return len(self.trans)
    method __add__ (line 46) | def __add__(self, other: Animation) -> Animation:
    method __getitem__ (line 57) | def __getitem__(self, index: int | slice) -> Animation:
    method cat (line 79) | def cat(self, other: Animation) -> None:
    method parents (line 89) | def parents(self) -> list[int]:
    method joint_names (line 93) | def joint_names(self) -> list[str]:
    method offsets (line 97) | def offsets(self) -> np.ndarray:
    method grot (line 104) | def grot(self) -> np.ndarray:
    method crot (line 109) | def crot(self) -> np.ndarray:
    method axangs (line 124) | def axangs(self) -> np.ndarray:
    method xforms (line 129) | def xforms(self) -> np.ndarray:
    method ortho6ds (line 134) | def ortho6ds(self) -> np.ndarray:
    method sw_tws (line 138) | def sw_tws(self) -> tuple[np.ndarray, np.ndarray]:
    method set_positions_from_fk (line 184) | def set_positions_from_fk(self) -> None:
    method lpos (line 188) | def lpos(self) -> np.ndarray:
    method gpos (line 194) | def gpos(self) -> np.ndarray:
    method rtpos (line 205) | def rtpos(self) -> np.ndarray:
    method cpos (line 222) | def cpos(self) -> np.ndarray:
    method croot (line 233) | def croot(self, idx: int=None) -> tuple[np.ndarray, np.ndarray]:
    method gposvel (line 254) | def gposvel(self) -> np.ndarray:
    method cposvel (line 262) | def cposvel(self) -> np.ndarray:
    method lrotvel (line 270) | def lrotvel(self) -> np.ndarray:
    method local_transform (line 284) | def local_transform(self) -> np.ndarray:
    method global_transform (line 294) | def global_transform(self) -> np.ndarray:
    method local_dualquat (line 308) | def local_dualquat(self) -> np.ndarray:
    method global_dualquat (line 312) | def global_dualquat(self) -> np.ndarray:
    method proj_root_pos (line 318) | def proj_root_pos(self, remove_vertical: bool=False) -> np.ndarray:
    method proj_root_rot (line 334) | def proj_root_rot(self) -> np.ndarray:
    method root_direction (line 341) | def root_direction(self, remove_vertical: bool=False) -> np.ndarray:
    method future_traj_poss (line 365) | def future_traj_poss(self, frame: int, remove_vertical: bool=True, csp...
    method future_traj_dirs (line 391) | def future_traj_dirs(self, frame: int, remove_vertical: bool=True, csp...
    method clamp_future_idxs (line 417) | def clamp_future_idxs(self, offset: int) -> np.ndarray:
    method calc_foot_contact (line 429) | def calc_foot_contact(
    method mirror (line 470) | def mirror(self, dataset: str=None) -> Animation:
    method no_animation (line 504) | def no_animation(
  function mirror_rot_trans (line 516) | def mirror_rot_trans(
  function animation_mirror (line 540) | def animation_mirror(lrot, lpos, names, parents):

FILE: anim/blend.py
  function lerp (line 12) | def lerp(x, y, t):
  function quat_lerp (line 16) | def quat_lerp(x: np.ndarray, y: np.ndarray, t: float) -> np.ndarray:
  function slerp (line 21) | def slerp(x: np.ndarray, y: np.ndarray, t: float) -> np.ndarray:

FILE: anim/bvh.py
  function load (line 16) | def load(
  function load_hierarchy (line 78) | def load_hierarchy(
  function load_motion (line 141) | def load_motion(
  function save (line 185) | def save(
  function save_hierarchy (line 210) | def save_hierarchy(
  function save_motion (line 262) | def save_motion(

FILE: anim/inverse_kinematics/ccd_ik.py
  function normalize_vector (line 15) | def normalize_vector(vector: np.ndarray):
  function simple_ccd_ik (line 21) | def simple_ccd_ik(

FILE: anim/inverse_kinematics/fabrik.py
  function normalize_vector (line 15) | def normalize_vector(vector: np.ndarray):
  function backward_reaching (line 20) | def backward_reaching(
  function forward_reaching (line 54) | def forward_reaching(
  function simple_fabrik (line 85) | def simple_fabrik(

FILE: anim/inverse_kinematics/jacobi_ik.py
  function simple_jacobi_ik (line 15) | def simple_jacobi_ik(

FILE: anim/inverse_kinematics/two_bone_ik.py
  function normalize (line 15) | def normalize(tensor: np.ndarray, axis:int=-1) -> np.ndarray:
  function two_bone_ik (line 26) | def two_bone_ik(

FILE: anim/keyframe.py
  class KeyFrame (line 9) | class KeyFrame:
  class KeyFrameAnimation (line 17) | class KeyFrameAnimation:
    method __init__ (line 18) | def __init__(

FILE: anim/motion_matching/database.py
  class Database (line 10) | class Database(Animation):
    method __init__ (line 11) | def __init__(
    method __len__ (line 48) | def __len__(self) -> int: return len(self.trans)
    method __add__ (line 50) | def __add__(self, other: Database) -> Database:
    method cat (line 71) | def cat(self, other: Database) -> None:
    method gposvel (line 83) | def gposvel(self) -> np.ndarray:
    method cposvel (line 92) | def cposvel(self) -> np.ndarray:
    method lrotvel (line 101) | def lrotvel(self) -> np.ndarray:
    method clamp_future_idxs (line 117) | def clamp_future_idxs(self, offset: int) -> np.ndarray:
  class MatchingDatabase (line 128) | class MatchingDatabase:
    method __len__ (line 145) | def __len__(self):

FILE: anim/motion_matching/mm.py
  function normalize_features (line 10) | def normalize_features(features: np.ndarray, weights: np.ndarray, axis=0...
  function normalize_query (line 18) | def normalize_query(query: np.ndarray, means:np.ndarray, stds:np.ndarray...
  function calc_box_distance (line 24) | def calc_box_distance(
  function create_position_features (line 40) | def create_position_features(db: Database) -> np.ndarray:
  function create_velocity_features (line 46) | def create_velocity_features(db: Database) -> np.ndarray:
  function create_traj_features (line 53) | def create_traj_features(db: Database) -> np.ndarray:
  function create_matching_features (line 67) | def create_matching_features(
  function create_matching_aabb (line 108) | def create_matching_aabb(features: np.ndarray, size: int | None) -> tupl...
  function create_matching_database (line 129) | def create_matching_database(
  function motion_matching_search (line 187) | def motion_matching_search(
  function brute_force_search (line 220) | def brute_force_search(
  function aabb_search (line 233) | def aabb_search(
  function kd_search (line 282) | def kd_search(
  function faiss_search (line 294) | def faiss_search(

FILE: anim/pose.py
  class Pose (line 7) | class Pose:
    method __init__ (line 8) | def __init__(
    method forward_kinematics (line 18) | def forward_kinematics(self):
    method set_local_positions (line 27) | def set_local_positions(self):
    method set_global_positions (line 32) | def set_global_positions(self):
    method set_global_rotations (line 35) | def set_global_rotations(self):
    method set_gpos_and_grot (line 38) | def set_gpos_and_grot(self):

FILE: anim/skel.py
  class Joint (line 7) | class Joint:
  function axis_to_vector (line 14) | def axis_to_vector(axis: str):
  class Skel (line 31) | class Skel:
    method __init__ (line 32) | def __init__(
    method __len__ (line 60) | def __len__(self) -> int:
    method __getitem__ (line 63) | def __getitem__(self, index: int | slice | str) -> Joint | list[Joint]:
    method indices (line 67) | def indices(self) -> list[int]:
    method parents (line 71) | def parents(self) -> list[int]:
    method children (line 76) | def children(self) -> dict[int: list[int]]:
    method names (line 87) | def names(self) -> list[str]:
    method offsets (line 92) | def offsets(self) -> np.ndarray:
    method dofs (line 100) | def dofs(self) -> list[int]:
    method joint_depths (line 104) | def joint_depths(self) -> list[int]:
    method bone_lengths (line 115) | def bone_lengths(self) -> np.ndarray:
    method get_joint (line 120) | def get_joint(self, index: int | slice | str) -> Joint | list[Joint]:
    method get_index_from_jname (line 126) | def get_index_from_jname(self, jname: str) -> int:
    method get_children (line 131) | def get_children(
    method get_parent (line 155) | def get_parent(
    method from_names_parents_offsets (line 181) | def from_names_parents_offsets(

FILE: anim/smpl.py
  function load_model (line 266) | def load_model(model_path: Path, gender: str=None):
  function calc_skel_offsets (line 283) | def calc_skel_offsets(

FILE: anitaichi/animation/anim.py
  function forward_kinematics_rotations (line 6) | def forward_kinematics_rotations(
  function forward_kinematics (line 20) | def forward_kinematics(
  function ti_forward_kinematics (line 38) | def ti_forward_kinematics(
  function convert_to_vector_field (line 69) | def convert_to_vector_field(array: np.ndarray):
  class Pose (line 76) | class Pose:
    method __init__ (line 77) | def __init__(
    method local_rotations (line 88) | def local_rotations(self):
    method global_rotations (line 92) | def global_rotations(self):
    method global_positions (line 99) | def global_positions(self):
  class Animation (line 108) | class Animation:
    method __init__ (line 109) | def __init__(
    method __len__ (line 124) | def __len__(self):
    method local_rotations (line 128) | def local_rotations(self):
    method global_rotations (line 132) | def global_rotations(self):
    method global_positions (line 148) | def global_positions(self):
    method ti_fk (line 156) | def ti_fk(self):
    method to_pose (line 166) | def to_pose(self, frame_num: int) -> Pose:
    method to_poses (line 172) | def to_poses(self, frames: slice) -> list[Pose]:

FILE: anitaichi/animation/anim_loader/bvh.py
  function load (line 17) | def load(
  function load_hierarchy (line 72) | def load_hierarchy(
  function load_motion (line 133) | def load_motion(
  function save (line 178) | def save(
  function save_hierarchy (line 201) | def save_hierarchy(
  function save_motion (line 252) | def save_motion(

FILE: anitaichi/animation/skel.py
  class Joint (line 6) | class Joint:
    method __init__ (line 7) | def __init__(self, name: str, index: int, parent: int, offset: list[fl...
  class Skel (line 19) | class Skel:
    method __init__ (line 20) | def __init__(self, joints: list[Joint], skel_name: str=None):
    method __len__ (line 25) | def __len__(self):
    method offsets (line 32) | def offsets(self) -> np.ndarray: # For numpy array
    method offsets_field (line 39) | def offsets_field(self): # For Taichi field
    method parents (line 46) | def parents(self) -> list[int]:
    method parents_field (line 50) | def parents_field(self):

FILE: anitaichi/transform/quat.py
  function eye (line 8) | def eye(shape, dtype=np.float32):
  function length (line 12) | def length(x):
  function normalize (line 16) | def normalize(x, eps=1e-8):
  function abs (line 19) | def abs(x):
  function inv (line 23) | def inv(q):
  function dot (line 27) | def dot(x, y):
  function mul (line 31) | def mul(x, y):
  function inv_mul (line 41) | def inv_mul(x, y):
  function mul_inv (line 44) | def mul_inv(x, y):
  function mul_vec (line 48) | def mul_vec(q, x):
  function inv_mul_vec (line 52) | def inv_mul_vec(q, x):
  function unroll (line 55) | def unroll(x):
  function between (line 64) | def between(x, y):
  function log (line 70) | def log(x, eps=1e-5):
  function exp (line 75) | def exp(x, eps=1e-5):
  function fk (line 82) | def fk(lrot, lpos, parents):
  function fk_rot (line 91) | def fk_rot(lrot, parents):
  function fk_centric (line 100) | def fk_centric(lrot, lpos, parents):
  function ik (line 107) | def ik(grot, gpos, parents):
  function ik_rot (line 121) | def ik_rot(grot, parents):
  function fk_vel (line 126) | def fk_vel(lrot, lpos, lvel, lang, parents):
  function lerp (line 144) | def lerp(x, y, t):
  function quat_lerp (line 148) | def quat_lerp(x, y, t):
  function slerp (line 152) | def slerp(x, y, t):
  function to_axis_angle (line 175) | def to_axis_angle(x, eps=1e-5):
  function to_euler (line 192) | def to_euler(x, order='zyx'):
  function to_xform (line 211) | def to_xform(x):
  function to_xform_xy (line 228) | def to_xform_xy(x):
  function to_scaled_angle_axis (line 244) | def to_scaled_angle_axis(x, eps=1e-5):
  function from_angle_axis (line 253) | def from_angle_axis(angle, axis):
  function from_axis_angle (line 260) | def from_axis_angle(rots):
  function from_euler (line 266) | def from_euler(e, order='zyx', mode="degree"):
  function from_xform (line 281) | def from_xform(ts):
  function from_xform_xy (line 309) | def from_xform_xy(x):
  function from_scaled_angle_axis (line 323) | def from_scaled_angle_axis(x, eps=1e-5):

FILE: anitaichi/transform/ti_quat.py
  function eye (line 7) | def eye():
  function normalize (line 11) | def normalize(q):
  function inv (line 15) | def inv(q):
  function abs (line 19) | def abs(q):
  function mul (line 23) | def mul(x, y):
  function mul_vec3 (line 31) | def mul_vec3(q, v):
  function from_angle_and_axis (line 37) | def from_angle_and_axis(angle, axis):
  function from_euler (line 43) | def from_euler(euler, order):

FILE: anitaichi/transform/ti_vec.py
  function cross (line 5) | def cross(v, w):
  function dot (line 12) | def dot(v, w):
  function length (line 16) | def length(v):
  function normalize (line 20) | def normalize(v):

FILE: anitaichi/transform/vec.py
  function cross (line 6) | def cross(a, b):
  function dot (line 12) | def dot(a, b, keepdims=False):
  function length (line 15) | def length(v, keepdims=False):
  function normalize (line 18) | def normalize(v):

FILE: model/LMM/decompressor.py
  class Compressor (line 6) | class Compressor(nn.Module):
    method __call__ (line 12) | def __call__(self, x):
  class Decompressor (line 23) | class Decompressor(nn.Module):
    method __call__ (line 28) | def __call__(self, x):

FILE: model/LMM/preprocessing.py
  function preprocess_motion_data (line 10) | def preprocess_motion_data(BASEPATH: Path, cfg, save_dir: Path):

FILE: model/LMM/projector.py
  class Projector (line 6) | class Projector(nn.Module):
    method __call__ (line 12) | def __call__(self, x):

FILE: model/LMM/setting.py
  class Config (line 14) | class Config:
    method __init__ (line 15) | def __init__(self, BASEPATH: Path, setting_path: str) -> None:
    method single_gpu_setting (line 75) | def single_gpu_setting(self):
    method multi_gpus_setting (line 78) | def multi_gpus_setting(self):

FILE: model/LMM/stepper.py
  class Stepper (line 6) | class Stepper(nn.Module):
    method __call__ (line 12) | def __call__(self, x):

FILE: model/LMM/train.py
  function main (line 27) | def main(cfg: Config):
  function train_decompressor (line 48) | def train_decompressor(cfg, ckpt_dir):
  function train_stepper (line 51) | def train_stepper(cfg, ckpt_dir):
  function train_projector (line 54) | def train_projector(cfg, ckpt_dir):

FILE: test/bvh_viewer.py
  function get_frame (line 15) | def get_frame(
  function update_camera (line 23) | def update_camera(camera, anim, frame):

FILE: test/calc_matching_time.py
  function create_database (line 13) | def create_database(bvh_dir: Path, files: list, starts: list, ends: list...

FILE: test/character_controller.py
  function create_database (line 14) | def create_database(bvh_dir: Path, files: list, starts: list, ends: list...
  function main (line 23) | def main():

FILE: test/path_following.py
  function create_database (line 18) | def create_database(bvh_dir: Path, files: list, starts: list, ends: list...
  function create_query (line 27) | def create_query(db: Database, idx: int, traj_feats: np.ndarray) -> np.n...
  function create_circle_traj (line 36) | def create_circle_traj():
  function create_animation_from_idxs (line 54) | def create_animation_from_idxs(db: Database, anim_frames: list[int]) -> ...
  function main (line 76) | def main():

FILE: test/plotting.py
  function update (line 51) | def update(index: int):

FILE: test/plotting_cspace.py
  function update (line 32) | def update(index: int):

FILE: test/plotting_global.py
  function update (line 34) | def update(index: int):

FILE: util/dualquat.py
  function eye (line 6) | def eye(shape: list[int], dtype=np.float32) -> np.ndarray:
  function normalize (line 10) | def normalize(dq: np.ndarray, eps=1e-8) -> np.ndarray:
  function abs (line 15) | def abs(dq: np.ndarray) -> np.ndarray:
  function inv (line 20) | def inv(dq: np.ndarray) -> np.ndarray:
  function mul (line 25) | def mul(x: np.ndarray, y: np.ndarray) -> np.ndarray:
  function from_trans (line 30) | def from_trans(trans: np.ndarray) -> np.ndarray:
  function from_rot (line 35) | def from_rot(rot: np.ndarray) -> np.ndarray:
  function from_rot_and_trans (line 38) | def from_rot_and_trans(rot: np.ndarray, trans: np.ndarray) -> np.ndarray:
  function to_trans (line 45) | def to_trans(dq: np.ndarray) -> np.ndarray:
  function to_rot (line 48) | def to_rot(dq: np.ndarray) -> np.ndarray:
  function fk (line 51) | def fk(dq: np.ndarray, parents: list[int]) -> np.ndarray:

FILE: util/load.py
  function toml_load (line 10) | def toml_load(path: Path):
  function yaml_load (line 16) | def yaml_load(path: Path):
  function pickle_load (line 22) | def pickle_load(path: Path, encoding: str="ASCII"):
  function json_load (line 27) | def json_load(path: Path):

FILE: util/quat.py
  function _fast_cross (line 6) | def _fast_cross(a, b):
  function eye (line 13) | def eye(shape, dtype=np.float32):
  function length (line 17) | def length(x):
  function normalize (line 21) | def normalize(x, eps=1e-8):
  function abs (line 24) | def abs(x):
  function inv (line 28) | def inv(q):
  function dot (line 32) | def dot(x, y):
  function mul (line 36) | def mul(x, y):
  function inv_mul (line 46) | def inv_mul(x, y):
  function mul_inv (line 49) | def mul_inv(x, y):
  function mul_vec (line 53) | def mul_vec(q, x):
  function inv_mul_vec (line 57) | def inv_mul_vec(q, x):
  function unroll (line 60) | def unroll(x):
  function between (line 69) | def between(x, y):
  function log (line 75) | def log(x, eps=1e-5):
  function exp (line 80) | def exp(x, eps=1e-5):
  function fk (line 87) | def fk(lrot, lpos, parents):
  function fk_rot (line 96) | def fk_rot(lrot, parents):
  function fk_vel (line 104) | def fk_vel(lrot, lpos, lvel, lang, parents):
  function ik (line 122) | def ik(grot, gpos, parents):
  function ik_rot (line 136) | def ik_rot(grot, parents):
  function to_axis_angle (line 147) | def to_axis_angle(x, eps=1e-5):
  function to_euler (line 164) | def to_euler(x, order='zyx'):
  function to_xform (line 183) | def to_xform(x):
  function to_xform_xy (line 200) | def to_xform_xy(x):
  function to_scaled_angle_axis (line 216) | def to_scaled_angle_axis(x, eps=1e-5):
  function from_angle_axis (line 225) | def from_angle_axis(angle, axis):
  function from_axis_angle (line 232) | def from_axis_angle(rots, eps=1e-5):
  function from_euler (line 239) | def from_euler(e, order='zyx', mode="degree"):
  function from_xform (line 254) | def from_xform(ts):
  function from_xform_xy (line 282) | def from_xform_xy(x):
  function from_scaled_angle_axis (line 296) | def from_scaled_angle_axis(x, eps=1e-5):