MCPcopy Create free account

hub / github.com/LightwheelAI/AutoDataGen / functions

Functions359 in github.com/LightwheelAI/AutoDataGen

↓ 42 callersFunctionas_torch
View sim buffers as ``torch.Tensor`` (no-op if already a tensor).
source/autosim/autosim/utils/data_util.py:12
↓ 34 callersFunction_log
(message: str = "")
examples/grasp_authoring/grasp_editor.py:120
↓ 33 callersMethodinfo
(self, msg, *args, **kwargs)
source/autosim/autosim/core/logger.py:32
↓ 29 callersMethod_to_curobo_device
Convert tensor to cuRobo device for isolated device management.
source/autosim/autosim/capabilities/motion_planning/curobo/curobo_planner.py:169
↓ 28 callersMethodwarning
(self, msg, *args, **kwargs)
source/autosim/autosim/core/logger.py:35
↓ 23 callersMethoddebug
(self, msg, *args, **kwargs)
source/autosim/autosim/core/logger.py:29
↓ 23 callersMethodget
Get a skill from the registry.
source/autosim/autosim/core/registration.py:233
↓ 22 callersMethodto
Move all tensors to device
source/autosim/autosim/core/types.py:153
↓ 16 callersFunctionconvert_quat
Converts quaternion from one convention to another. The convention to convert TO is specified as an optional argument. If to == 'xyzw', then
source/autosim/autosim/utils/data_util.py:31
↓ 12 callersFunctioncoerce_backend_data
(backend_utils, device, data, dtype="float32")
examples/grasp_authoring/grasp_editor_helper.py:561
↓ 8 callersFunctiondraw_line
Draws a single line between two points.
source/autosim/autosim/utils/debug_util.py:224
↓ 7 callersFunction_required_value
(cfg: dict, key: str, cfg_path: str)
examples/grasp_authoring/grasp_editor_helper.py:76
↓ 6 callersMethod_build_activate_joint_state
Extract the planner's active joint state from full simulation joint state. cuRobo typically plans over a subset of "active joints" (`self._pl
source/autosim/autosim/skills/base_skill.py:127
↓ 6 callersMethod_build_world_state
Build the world state.
source/autosim/autosim/core/pipeline.py:247
↓ 6 callersFunctionas_numpy
(data)
examples/grasp_authoring/grasp_editor_helper.py:579
↓ 6 callersFunctioncoerce_articulation_data
(self, data, dtype="float32")
examples/grasp_authoring/grasp_editor_helper.py:709
↓ 5 callersFunction_as_pose7
(pose7: Sequence[float] | torch.Tensor, *, device: torch.device, dtype: torch.dtype)
examples/visualization/curobo_world_obstacles.py:53
↓ 5 callersFunction_compose_robot_to_world
Compose pose in robot root frame to world frame.
examples/visualization/curobo_world_obstacles.py:129
↓ 5 callersFunction_draw_oriented_box
(*, pose_w: _Pose7, half_dims_xyz: torch.Tensor, color, thickness: float, z_lift: float)
examples/visualization/curobo_world_obstacles.py:88
↓ 5 callersFunction_is_obstacle_enabled
Check if an obstacle is enabled in the collision checker. Args: collision_checker: cuRobo collision checker instance obstacle_nam
examples/visualization/curobo_world_obstacles.py:139
↓ 5 callersFunction_required_str
(cfg: dict, key: str, cfg_path: str)
examples/grasp_authoring/grasp_editor_helper.py:89
↓ 5 callersMethodget_link_poses
Get the poses of specific links in the robot root frame. Returns all links if link_names is None.
source/autosim/autosim/capabilities/motion_planning/curobo/curobo_planner.py:934
↓ 5 callersMethodinitialize
Initialize the AutoSim pipeline.
source/autosim/autosim/core/pipeline.py:61
↓ 5 callersFunctionmake_pipeline
Create a pipeline instance from the registry. Args: id: The registered pipeline ID. **cfg_overrides: Optional keyword arguments t
source/autosim/autosim/core/registration.py:122
↓ 5 callersMethodreset_env
Reset the environment.
source/autosim/autosim/core/pipeline.py:136
↓ 5 callersMethodstep
Execute one step of the skill. Args: state: The current state of the world. Returns: The output of the skill
source/autosim/autosim/core/skill.py:100
↓ 4 callersFunction_footprint_from_local_points
( prim: Usd.Prim, points_local: np.ndarray, xform_cache: UsdGeom.XformCache, world_mat: Gf.Mat
source/autosim/autosim/capabilities/navigation/occupancy_map.py:538
↓ 4 callersFunction_get_ee_pose_world
Return EE pose in world frame as [x, y, z, qx, qy, qz, qw] via cuRobo FK.
examples/visualization/curobo_collision_spheres.py:138
↓ 4 callersFunction_is_descendant_or_self
(path: Sdf.Path, root_path: Sdf.Path)
examples/grasp_authoring/grasp_editor_helper.py:955
↓ 4 callersMethodreset
Reset the skill.
source/autosim/autosim/core/skill.py:112
↓ 4 callersMethodset_target_object
Set the current target object for selective collision checking. Args: target_object: Name of the target object, or None to disabl
source/autosim/autosim/capabilities/motion_planning/curobo/curobo_planner.py:568
↓ 3 callersMethod__init__
(self, extra_cfg: LiftSkillExtraCfg)
source/autosim/autosim/skills/relative_reach.py:198
↓ 3 callersFunction_build_curobo_q
Build a joint position tensor in cuRobo's joint order from Isaac Lab state. Isaac Lab and cuRobo use different joint orderings. We look up each c
examples/visualization/curobo_collision_spheres.py:78
↓ 3 callersFunction_cylinder_like_footprint_poly_xy
( prim: Usd.Prim, radius: float, xform_cache: UsdGeom.XformCache, world_mat: Gf.Matrix4d | Non
source/autosim/autosim/capabilities/navigation/occupancy_map.py:559
↓ 3 callersFunction_is_joint_prim
(prim)
examples/grasp_authoring/grasp_editor_helper.py:946
↓ 3 callersMethod_is_valid_grid_pos
Check if grid position is within bounds (internal numpy representation)
source/autosim/autosim/capabilities/navigation/a_star/a_star_planner.py:109
↓ 3 callersFunction_make_bbox_cache
Create a BBoxCache that covers all USD purposes. Collision-only meshes (e.g. ``*/Collisions/*`` under Isaac Lab assets) are typically authored
source/autosim/autosim/capabilities/navigation/occupancy_map.py:148
↓ 3 callersFunction_normalize_quat_xyzw
(quat: tuple[float, float, float, float])
examples/grasp_authoring/grasp_editor.py:475
↓ 3 callersFunction_parse_axis_vector
Parse a '+x'/'-z'/'+x+y' axis string into a normalized direction vector.
source/autosim/autosim/skills/rotate.py:33
↓ 3 callersMethod_refine_curobo_world_collision
Refine cuRobo world collision state before planning. This method synchronizes all dynamic collision geometry: - Rigid object poses (i
source/autosim/autosim/capabilities/motion_planning/curobo/curobo_planner.py:578
↓ 3 callersFunction_required_mapping
(cfg: dict, key: str, cfg_path: str)
examples/grasp_authoring/grasp_editor_helper.py:82
↓ 3 callersFunction_vec_add
(a: tuple[float, ...], b: tuple[float, ...])
examples/grasp_authoring/grasp_editor.py:490
↓ 3 callersFunctionclear_debug_drawing
Clears all debug drawings.
source/autosim/autosim/utils/debug_util.py:234
↓ 3 callersMethodcreate
Create a skill instance from the registry, extra_cfg will overwrite the default value in the skill configuration.
source/autosim/autosim/core/registration.py:241
↓ 3 callersFunctionensure_physics_sim_view
()
examples/grasp_authoring/grasp_editor_helper.py:552
↓ 3 callersMethoderror
(self, msg, *args, **kwargs)
source/autosim/autosim/core/logger.py:38
↓ 3 callersMethodget_ee_pose
Get the end-effector pose of the robot.
source/autosim/autosim/capabilities/motion_planning/curobo/curobo_planner.py:924
↓ 3 callersMethodplan
Plan the skill. Args: state: The current state of the world. goal: The goal of the skill. Returns:
source/autosim/autosim/core/skill.py:66
↓ 3 callersMethodplan_motion
Plan a trajectory to one of K candidate target poses from a shared start joint state. Always uses cuRobo's batch planning under the
source/autosim/autosim/capabilities/motion_planning/curobo/curobo_planner.py:678
↓ 3 callersMethodregister_apply_method
Register an apply method for a specific skill.
source/autosim/autosim/core/action_adapter.py:60
↓ 3 callersFunctionto_numpy
(data)
examples/grasp_authoring/grasp_editor_helper.py:568
↓ 2 callersFunction_add_internal_reference
(references, prim_path: Sdf.Path)
examples/grasp_authoring/grasp_editor_helper.py:1098
↓ 2 callersFunction_apply_articulation_root
(prim)
examples/grasp_authoring/grasp_editor_helper.py:982
↓ 2 callersFunction_apply_live_poses
Update `pipeline._env_extra_info.object_reach_target_poses` from JSON.
examples/visualization/reach_target_pose.py:103
↓ 2 callersFunction_apply_pending_grasp_editor_selection
(ui_builder)
examples/grasp_authoring/grasp_editor_helper.py:510
↓ 2 callersMethod_check_skill_extra_cfg
modify the extra configuration of the skills.
source/autosim/autosim/core/pipeline.py:193
↓ 2 callersMethod_collect_primitives_by_prefix
Collect all primitive names from cuRobo world model that start with given prefix. Args: prefix: Prefix to filter primitive names
source/autosim/autosim/capabilities/motion_planning/curobo/curobo_planner.py:304
↓ 2 callersMethod_compute_goal_from_offset
Compute reach goal by transforming object-frame offsets into robot root frame. Args: env: The Isaac Lab environment.
source/autosim/autosim/skills/reach.py:222
↓ 2 callersMethod_compute_relative_extra_target_poses
Project cached or current primary-frame offsets onto the target primary pose.
source/autosim/autosim/skills/reach.py:82
↓ 2 callersMethod_compute_relative_offsets_in_primary
Compute the current rigid offset from the primary link frame to each extra link.
source/autosim/autosim/skills/reach.py:115
↓ 2 callersFunction_configure_grasp_editor_physics_sync
Use USD transform syncing instead of PhysX Fabric for Grasp Editor. Grasp Editor reads and displays ordinary USD prim transforms. With Fabric ena
examples/grasp_authoring/grasp_editor_helper.py:231
↓ 2 callersFunction_converted_grasp_begin_marker
()
examples/grasp_authoring/grasp_editor.py:519
↓ 2 callersFunction_converted_grasp_end_marker
()
examples/grasp_authoring/grasp_editor.py:523
↓ 2 callersFunction_cross
(a: tuple[float, float, float], b: tuple[float, float, float])
examples/grasp_authoring/grasp_editor.py:482
↓ 2 callersFunction_draw_ee_frame
Draw EE frame axes using debug lines (R=X, G=Y, B=Z).
examples/visualization/curobo_collision_spheres.py:165
↓ 2 callersFunction_draw_spheres
Draw collision spheres as three orthogonal wireframe circles.
examples/visualization/curobo_collision_spheres.py:214
↓ 2 callersFunction_flattened_prototype_reference_targets
(prim)
examples/grasp_authoring/grasp_editor_helper.py:1087
↓ 2 callersFunction_footprint_poly_xy_from_world_points
XY convex hull of world-frame points; requires at least 3 distinct vertices.
source/autosim/autosim/capabilities/navigation/occupancy_map.py:483
↓ 2 callersMethod_get_current_primary_and_extra_link_poses
Get current primary and extra link poses in robot root frame.
source/autosim/autosim/skills/reach.py:62
↓ 2 callersFunction_get_spheres_world
Return (positions, radii) for all active collision spheres in world frame.
examples/visualization/curobo_collision_spheres.py:100
↓ 2 callersFunction_load_creator
(creator: str | PipelineCreator | ConfigCreator)
source/autosim/autosim/core/registration.py:114
↓ 2 callersFunction_prim_local_to_world_matrix
( prim: Usd.Prim, xform_cache: UsdGeom.XformCache, world_mat: Gf.Matrix4d | None )
source/autosim/autosim/capabilities/navigation/occupancy_map.py:476
↓ 2 callersFunction_prim_world_matrix_from_body_frame
World transform: T_world_prim = T_world_body(scene) @ T_body_prim(usd_fixed).
source/autosim/autosim/capabilities/navigation/occupancy_map.py:391
↓ 2 callersFunction_rasterize_prim_paths
Rasterize prims; return count successfully marked occupied. ``world_mat_for_path`` maps prim path → override world matrix (scene group). Missing
source/autosim/autosim/capabilities/navigation/occupancy_map.py:645
↓ 2 callersFunction_reference_list_items
(reference_list_op)
examples/grasp_authoring/grasp_editor_helper.py:1077
↓ 2 callersFunction_relative_to_any_link
(path: Sdf.Path, link_paths: list[Sdf.Path])
examples/grasp_authoring/grasp_editor_helper.py:959
↓ 2 callersFunction_snapshot_object_poses_w
(*, env, object_names: list[str])
examples/visualization/reach_target_pose.py:126
↓ 2 callersFunction_targets_are_inside_links
(targets: list[Sdf.Path], link_paths: list[Sdf.Path])
examples/grasp_authoring/grasp_editor_helper.py:996
↓ 2 callersFunction_to_pose_numpy
(pos_w: torch.Tensor, quat_w: torch.Tensor)
source/autosim/autosim/capabilities/navigation/occupancy_map.py:415
↓ 2 callersFunction_transform_points_local_to_world
Transform Nx3 local points to world frame (USD row-vector convention).
source/autosim/autosim/capabilities/navigation/occupancy_map.py:468
↓ 2 callersFunction_update_visualization
(pipeline, env_id, color: tuple = (0.2, 0.9, 0.2, 0.8), ee_scale: float = 0.1)
examples/visualization/curobo_collision_spheres.py:226
↓ 2 callersFunction_vec_scale
(vector: tuple[float, ...], scale: float)
examples/grasp_authoring/grasp_editor.py:494
↓ 2 callersFunction_world_aabb_or_none
Return ``(min_xyz, max_xyz)`` of a prim's world-space AABB, or None if it is empty/invalid. USD returns an empty ``Gf.Range3d`` (``min > max``, c
source/autosim/autosim/capabilities/navigation/occupancy_map.py:170
↓ 2 callersMethod_world_to_grid
Convert world coordinates to grid coordinates (row, col) Args: pos: [x, y] in world frame Returns: grid_pos:
source/autosim/autosim/capabilities/navigation/a_star/a_star_planner.py:79
↓ 2 callersFunctionactivate_profile_gripper_dofs
(self)
examples/grasp_authoring/grasp_editor_helper.py:863
↓ 2 callersMethodapply
Apply the skill output to the environment. Args: skill: The skill instance. skill_output: The output of the
source/autosim/autosim/core/action_adapter.py:75
↓ 2 callersFunctionas_backend_data
(view, data, dtype="float32")
examples/grasp_authoring/grasp_editor_helper.py:584
↓ 2 callersFunctioncoerce_articulation_action
(articulation_view, control_actions)
examples/grasp_authoring/grasp_editor_helper.py:712
↓ 2 callersFunctioncreate_marker
(marker_name: str)
source/autosim/autosim/utils/debug_util.py:16
↓ 2 callersFunctioncross
(o: np.ndarray, a: np.ndarray, b: np.ndarray)
source/autosim/autosim/capabilities/navigation/occupancy_map.py:205
↓ 2 callersMethodextract_goal_from_info
Extract the goal from the skill information. Args: skill_info: The skill information. env: The environment.
source/autosim/autosim/core/skill.py:50
↓ 2 callersMethodget_cfg
Get the configuration of the skill.
source/autosim/autosim/core/skill.py:44
↓ 2 callersFunctionget_physics_sim_view
()
examples/grasp_authoring/grasp_editor_helper.py:545
↓ 2 callersMethodlist_skills
List all registered skill names.
source/autosim/autosim/core/registration.py:248
↓ 2 callersFunctionpatch_grasp_editor_runtime
(gripper_profile: GripperProfile)
examples/grasp_authoring/grasp_editor_helper.py:126
↓ 2 callersMethodregister
Register a skill in the registry.
source/autosim/autosim/core/registration.py:217
↓ 2 callersMethodshould_skip_apply
Check if the skill should be skipped for applying the action. Args: skill: The skill instance.
source/autosim/autosim/core/action_adapter.py:65
↓ 2 callersMethodvisualize_debug_target_pose
Visualize the debug target pose.
source/autosim/autosim/skills/base_skill.py:121
↓ 2 callersFunctionvisualize_marker
(marker_name: str, pose: torch.Tensor)
source/autosim/autosim/utils/debug_util.py:26
↓ 2 callersFunctionvisualize_reach_target_poses
Visualize all reach target poses from env_extra_info as frame markers. Creates markers for: - ``env_extra_info.object_reach_target_poses`` un
source/autosim/autosim/utils/debug_util.py:55
next →1–100 of 359, ranked by callers