Index _ | A | B | C | D | E | F | G | H | I | K | L | M | N | O | P | R | S | T | U | V | W | Y | Z _ __call__() (torchdrivesim.simulator.AgentTypeFunctor method) __getitem__() (torchdrivesim.mesh.BaseMesh method) (torchdrivesim.simulator.SimulatorInterface method) __post_init__() (torchdrivesim.mesh.BaseMesh method) __version__ (in module torchdrivesim) _advance_state() (torchdrivesim.goals.WaypointGoal method) _agent_waypoint_overlap() (torchdrivesim.goals.WaypointGoal method) _attrs_fill (torchdrivesim.mesh.AttributeMesh attribute) _cat_fill (torchdrivesim.mesh.BirdviewMesh attribute) _compute_collision_of_multi_agents() (torchdrivesim.simulator.HomogeneousWrapper method) (torchdrivesim.simulator.NPCWrapper method) (torchdrivesim.simulator.SelectiveWrapper method) (torchdrivesim.simulator.Simulator method) (torchdrivesim.simulator.SimulatorInterface method) (torchdrivesim.simulator.SimulatorWrapper method) _compute_collision_of_single_agent() (torchdrivesim.simulator.Simulator method) (torchdrivesim.simulator.SimulatorInterface method) (torchdrivesim.simulator.SimulatorWrapper method) _default_allowed_states() (torchdrivesim.traffic_controls.BaseTrafficControl class method) (torchdrivesim.traffic_controls.TrafficLightControl class method) _default_mask() (torchdrivesim.goals.WaypointGoal method) (torchdrivesim.traffic_controls.BaseTrafficControl method) _default_replay_states() (torchdrivesim.traffic_controls.BaseTrafficControl method) _default_state() (torchdrivesim.goals.WaypointGoal method) (torchdrivesim.traffic_controls.BaseTrafficControl method) _deserialize_tensors() (torchdrivesim.mesh.AttributeMesh class method) (torchdrivesim.mesh.BaseMesh class method) (torchdrivesim.mesh.BirdviewMesh class method) _extend_tensor() (torchdrivesim.simulator.SelectiveWrapper method) _faces_fill (torchdrivesim.mesh.BaseMesh attribute) _get_npc_action() (torchdrivesim.simulator.NPCWrapper method) _get_npc_predictions() (torchdrivesim.behavior.iai.IAIWrapper method) _npc_teleport_to() (torchdrivesim.behavior.iai.IAIWrapper method) (torchdrivesim.behavior.replay.ReplayWrapper method) (torchdrivesim.simulator.NPCWrapper method) _restrict_tensor() (torchdrivesim.simulator.SelectiveWrapper method) _trim_and_return_verts_and_faces() (torchdrivesim.mesh.BaseMesh method) _update_mask() (torchdrivesim.goals.WaypointGoal method) _update_npc_present_mask() (torchdrivesim.behavior.replay.ReplayWrapper method) (torchdrivesim.simulator.NPCWrapper method) _validate_input() (torchdrivesim.mesh.AttributeMesh method) (torchdrivesim.mesh.BaseMesh method) (torchdrivesim.mesh.BirdviewMesh method) (torchdrivesim.mesh.RGBMesh method) _verts_fill (torchdrivesim.mesh.BaseMesh attribute) A across_agent_types() (torchdrivesim.simulator.SimulatorInterface method) action_size (torchdrivesim.kinematic.KinematicBicycle attribute) (torchdrivesim.kinematic.KinematicModel attribute) (torchdrivesim.simulator.HomogeneousWrapper property) (torchdrivesim.simulator.Simulator property) (torchdrivesim.simulator.SimulatorInterface property) (torchdrivesim.simulator.SimulatorWrapper property) add_static_meshes() (torchdrivesim.rendering.base.BirdviewRenderer method) (torchdrivesim.rendering.BirdviewRenderer method) agent_concat() (torchdrivesim.simulator.HomogeneousWrapper method) agent_count (torchdrivesim.simulator.SimulatorInterface property) agent_functor (torchdrivesim.simulator.HomogeneousWrapper property) (torchdrivesim.simulator.Simulator property) (torchdrivesim.simulator.SimulatorInterface property) (torchdrivesim.simulator.SimulatorWrapper property) agent_split() (torchdrivesim.simulator.HomogeneousWrapper method) agent_types (torchdrivesim.simulator.HomogeneousWrapper property) (torchdrivesim.simulator.Simulator property) (torchdrivesim.simulator.SimulatorInterface property) (torchdrivesim.simulator.SimulatorWrapper property) AgentTypeFunctor (class in torchdrivesim.simulator) antialias (torchdrivesim.rendering.nvdiffrast.NvdiffrastRendererConfig attribute) (torchdrivesim.rendering.NvdiffrastRendererConfig attribute) assert_equal() (in module torchdrivesim.utils) attr_dim (torchdrivesim.mesh.AttributeMesh property) AttributeMesh (class in torchdrivesim.mesh) attrs (torchdrivesim.mesh.AttributeMesh attribute) B backend (torchdrivesim.rendering.base.DummyRendererConfig attribute) (torchdrivesim.rendering.base.RendererConfig attribute) (torchdrivesim.rendering.DummyRendererConfig attribute) (torchdrivesim.rendering.nvdiffrast.NvdiffrastRendererConfig attribute) (torchdrivesim.rendering.NvdiffrastRendererConfig attribute) (torchdrivesim.rendering.pytorch3d.Pytorch3DRendererConfig attribute) (torchdrivesim.rendering.Pytorch3DRendererConfig attribute) (torchdrivesim.rendering.RendererConfig attribute) BadMeshFormat BaseMesh (class in torchdrivesim.mesh) BaseTrafficControl (class in torchdrivesim.traffic_controls) batch_size (torchdrivesim.mesh.BaseMesh property) (torchdrivesim.simulator.Simulator property) (torchdrivesim.simulator.SimulatorInterface property) (torchdrivesim.simulator.SimulatorWrapper property) bbox2discs() (in module torchdrivesim.infractions) BicycleByDisplacement (class in torchdrivesim.kinematic) BicycleByOrientedDisplacement (class in torchdrivesim.kinematic) BicycleNoReversing (class in torchdrivesim.kinematic) BirdviewMesh (class in torchdrivesim.mesh) BirdviewRecordingWrapper (class in torchdrivesim.simulator) BirdviewRenderer (class in torchdrivesim.rendering) (class in torchdrivesim.rendering.base) BoundedRegionWrapper (class in torchdrivesim.simulator) box1_in_box2() (in module torchdrivesim._iou_utils) box2corners_th() (in module torchdrivesim._iou_utils) box2corners_with_rear_factor() (in module torchdrivesim._iou_utils) box_in_box_th() (in module torchdrivesim._iou_utils) box_intersection_th() (in module torchdrivesim._iou_utils) build_vertices() (in module torchdrivesim._iou_utils) build_verts_faces_from_bounding_box() (torchdrivesim.rendering.base.BirdviewRenderer method) (torchdrivesim.rendering.BirdviewRenderer method) C calculate_area() (in module torchdrivesim._iou_utils) categories (torchdrivesim.mesh.BirdviewMesh attribute) center (torchdrivesim.mesh.BaseMesh property) clone() (torchdrivesim.mesh.BaseMesh method) collate() (torchdrivesim.mesh.AttributeMesh class method) (torchdrivesim.mesh.BaseMesh class method) (torchdrivesim.mesh.BirdviewMesh class method) collision_detection_with_discs() (in module torchdrivesim.infractions) collision_metric (torchdrivesim.simulator.TorchDriveConfig attribute) CollisionMetric (class in torchdrivesim.simulator) Color (in module torchdrivesim.mesh) colors (torchdrivesim.mesh.BirdviewMesh attribute) compute_agent_collisions_metric() (in module torchdrivesim.infractions) compute_agent_collisions_metric_pytorch3d() (in module torchdrivesim.infractions) compute_collision() (torchdrivesim.simulator.SimulatorInterface method) compute_offroad() (torchdrivesim.simulator.HomogeneousWrapper method) (torchdrivesim.simulator.NPCWrapper method) (torchdrivesim.simulator.Simulator method) (torchdrivesim.simulator.SimulatorInterface method) (torchdrivesim.simulator.SimulatorWrapper method) compute_state() (torchdrivesim.traffic_controls.BaseTrafficControl method) compute_traffic_lights_violations() (torchdrivesim.simulator.SimulatorInterface method) compute_violation() (torchdrivesim.traffic_controls.BaseTrafficControl method) (torchdrivesim.traffic_controls.TrafficLightControl method) compute_wrong_way() (torchdrivesim.simulator.HomogeneousWrapper method) (torchdrivesim.simulator.NPCWrapper method) (torchdrivesim.simulator.Simulator method) (torchdrivesim.simulator.SimulatorInterface method) (torchdrivesim.simulator.SimulatorWrapper method) concat() (torchdrivesim.mesh.AttributeMesh class method) (torchdrivesim.mesh.BaseMesh class method) (torchdrivesim.mesh.BirdviewMesh class method) construct_cameras() (torchdrivesim.rendering.base.BirdviewRenderer method) (torchdrivesim.rendering.BirdviewRenderer method) construct_pytorch3d_cameras() (in module torchdrivesim.rendering.base) copy() (torchdrivesim.behavior.iai.IAIWrapper method) (torchdrivesim.behavior.replay.ReplayWrapper method) (torchdrivesim.goals.WaypointGoal method) (torchdrivesim.kinematic.BicycleByDisplacement method) (torchdrivesim.kinematic.KinematicBicycle method) (torchdrivesim.kinematic.KinematicModel method) (torchdrivesim.kinematic.SimpleKinematicModel method) (torchdrivesim.rendering.base.BirdviewRenderer method) (torchdrivesim.rendering.BirdviewRenderer method) (torchdrivesim.simulator.BirdviewRecordingWrapper method) (torchdrivesim.simulator.BoundedRegionWrapper method) (torchdrivesim.simulator.NPCWrapper method) (torchdrivesim.simulator.RecordingWrapper method) (torchdrivesim.simulator.SelectiveWrapper method) (torchdrivesim.simulator.Simulator method) (torchdrivesim.simulator.SimulatorInterface method) (torchdrivesim.simulator.SimulatorWrapper method) (torchdrivesim.traffic_controls.BaseTrafficControl method) D denormalize_action() (torchdrivesim.kinematic.KinematicBicycle method) (torchdrivesim.kinematic.KinematicModel method) (torchdrivesim.kinematic.SimpleKinematicModel method) deserialize() (torchdrivesim.mesh.BaseMesh class method) device (torchdrivesim.mesh.BaseMesh property) DictAgentTypeFunctor (class in torchdrivesim.simulator) differentiable_rendering (torchdrivesim.rendering.pytorch3d.Pytorch3DRendererConfig attribute) (torchdrivesim.rendering.Pytorch3DRendererConfig attribute) dim (torchdrivesim.mesh.BaseMesh property) discs (torchdrivesim.simulator.CollisionMetric attribute) DummyRenderer (class in torchdrivesim.rendering) (class in torchdrivesim.rendering.base) DummyRendererConfig (class in torchdrivesim.rendering) (class in torchdrivesim.rendering.base) E empty() (torchdrivesim.mesh.AttributeMesh class method) (torchdrivesim.mesh.BaseMesh class method) (torchdrivesim.mesh.BirdviewMesh class method) EPSILON (in module torchdrivesim._iou_utils) expand() (torchdrivesim.mesh.AttributeMesh method) (torchdrivesim.mesh.BaseMesh method) (torchdrivesim.mesh.BirdviewMesh method) (torchdrivesim.rendering.base.BirdviewRenderer method) (torchdrivesim.rendering.BirdviewRenderer method) extend() (torchdrivesim.behavior.replay.ReplayWrapper method) (torchdrivesim.goals.WaypointGoal method) (torchdrivesim.simulator.BirdviewRecordingWrapper method) (torchdrivesim.simulator.BoundedRegionWrapper method) (torchdrivesim.simulator.NoReentryBoundedRegionWrapper method) (torchdrivesim.simulator.SelectiveWrapper method) (torchdrivesim.simulator.Simulator method) (torchdrivesim.simulator.SimulatorInterface method) (torchdrivesim.simulator.SimulatorWrapper method) (torchdrivesim.traffic_controls.BaseTrafficControl method) extend_tensor() (torchdrivesim.simulator.NPCWrapper static method) F faces (torchdrivesim.mesh.BaseMesh attribute) faces_count (torchdrivesim.mesh.BaseMesh property) fill_attr() (torchdrivesim.mesh.BirdviewMesh method) find_direction() (in module torchdrivesim.lanelet2) find_lanelet_directions() (in module torchdrivesim.lanelet2) fit_action() (torchdrivesim.kinematic.BicycleByDisplacement method) (torchdrivesim.kinematic.BicycleByOrientedDisplacement method) (torchdrivesim.kinematic.KinematicBicycle method) (torchdrivesim.kinematic.KinematicModel method) (torchdrivesim.kinematic.OrientedKinematicModel method) (torchdrivesim.kinematic.SimpleKinematicModel method) (torchdrivesim.kinematic.TeleportingKinematicModel method) (torchdrivesim.simulator.HomogeneousWrapper method) (torchdrivesim.simulator.NPCWrapper method) (torchdrivesim.simulator.SelectiveWrapper method) (torchdrivesim.simulator.Simulator method) (torchdrivesim.simulator.SimulatorInterface method) (torchdrivesim.simulator.SimulatorWrapper method) flattening() (torchdrivesim.kinematic.KinematicBicycle method) (torchdrivesim.kinematic.KinematicModel method) fmap() (torchdrivesim.simulator.AgentTypeFunctor method) (torchdrivesim.simulator.DictAgentTypeFunctor method) (torchdrivesim.simulator.SingletonAgentTypeFunctor method) forward() (torchdrivesim.rendering.pytorch3d.Shader2D method) G generate_annulus_polygon_mesh() (in module torchdrivesim.mesh) generate_disc_mesh() (in module torchdrivesim.mesh) generate_trajectory_mesh() (in module torchdrivesim.mesh) get_agent_size() (torchdrivesim.simulator.HomogeneousWrapper method) (torchdrivesim.simulator.NPCWrapper method) (torchdrivesim.simulator.SelectiveWrapper method) (torchdrivesim.simulator.Simulator method) (torchdrivesim.simulator.SimulatorInterface method) (torchdrivesim.simulator.SimulatorWrapper method) get_agent_type() (torchdrivesim.simulator.HomogeneousWrapper method) (torchdrivesim.simulator.NPCWrapper method) (torchdrivesim.simulator.SelectiveWrapper method) (torchdrivesim.simulator.Simulator method) (torchdrivesim.simulator.SimulatorInterface method) (torchdrivesim.simulator.SimulatorWrapper method) get_all_agents_absolute() (torchdrivesim.simulator.HomogeneousWrapper method) (torchdrivesim.simulator.SelectiveWrapper method) (torchdrivesim.simulator.Simulator method) (torchdrivesim.simulator.SimulatorInterface method) (torchdrivesim.simulator.SimulatorWrapper method) get_all_agents_relative() (torchdrivesim.simulator.HomogeneousWrapper method) (torchdrivesim.simulator.NPCWrapper method) (torchdrivesim.simulator.SelectiveWrapper method) (torchdrivesim.simulator.Simulator method) (torchdrivesim.simulator.SimulatorInterface method) (torchdrivesim.simulator.SimulatorWrapper method) get_all_intersections() (in module torchdrivesim.infractions) get_birdviews() (torchdrivesim.simulator.BirdviewRecordingWrapper method) get_color() (torchdrivesim.rendering.base.BirdviewRenderer method) (torchdrivesim.rendering.BirdviewRenderer method) get_default_color_map() (in module torchdrivesim.rendering.base) get_default_rendering_levels() (in module torchdrivesim.rendering.base) get_exposed_agents() (torchdrivesim.simulator.SelectiveWrapper method) get_glctx_session() (in module torchdrivesim.rendering.nvdiffrast) get_innermost_simulator() (torchdrivesim.simulator.Simulator method) (torchdrivesim.simulator.SimulatorInterface method) (torchdrivesim.simulator.SimulatorWrapper method) get_masks() (torchdrivesim.goals.WaypointGoal method) get_params() (torchdrivesim.kinematic.KinematicBicycle method) (torchdrivesim.kinematic.KinematicModel method) get_present_mask() (torchdrivesim.simulator.BoundedRegionWrapper method) (torchdrivesim.simulator.HomogeneousWrapper method) (torchdrivesim.simulator.NoReentryBoundedRegionWrapper method) (torchdrivesim.simulator.NPCWrapper method) (torchdrivesim.simulator.SelectiveWrapper method) (torchdrivesim.simulator.Simulator method) (torchdrivesim.simulator.SimulatorInterface method) (torchdrivesim.simulator.SimulatorWrapper method) get_records() (torchdrivesim.simulator.RecordingWrapper method) get_state() (torchdrivesim.kinematic.KinematicModel method) (torchdrivesim.simulator.HomogeneousWrapper method) (torchdrivesim.simulator.NPCWrapper method) (torchdrivesim.simulator.SelectiveWrapper method) (torchdrivesim.simulator.Simulator method) (torchdrivesim.simulator.SimulatorInterface method) (torchdrivesim.simulator.SimulatorWrapper method) get_traffic_controls() (torchdrivesim.simulator.SimulatorInterface method) get_waypoints() (torchdrivesim.goals.WaypointGoal method) (torchdrivesim.simulator.HomogeneousWrapper method) (torchdrivesim.simulator.NPCWrapper method) (torchdrivesim.simulator.SelectiveWrapper method) (torchdrivesim.simulator.Simulator method) (torchdrivesim.simulator.SimulatorInterface method) (torchdrivesim.simulator.SimulatorWrapper method) get_waypoints_mask() (torchdrivesim.simulator.HomogeneousWrapper method) (torchdrivesim.simulator.NPCWrapper method) (torchdrivesim.simulator.SelectiveWrapper method) (torchdrivesim.simulator.Simulator method) (torchdrivesim.simulator.SimulatorInterface method) (torchdrivesim.simulator.SimulatorWrapper method) get_world_center() (torchdrivesim.simulator.Simulator method) (torchdrivesim.simulator.SimulatorInterface method) (torchdrivesim.simulator.SimulatorWrapper method) glctx_sessions (in module torchdrivesim.rendering.nvdiffrast) H hard (torchdrivesim.rendering.pytorch3d.RenderingBlend attribute) heuristic_initialize() (in module torchdrivesim.behavior.heuristic) highlight_ego_vehicle (torchdrivesim.rendering.base.RendererConfig attribute) (torchdrivesim.rendering.RendererConfig attribute) HomogeneousWrapper (class in torchdrivesim.simulator) I iai_drive() (in module torchdrivesim.behavior.iai) iai_initialize() (in module torchdrivesim.behavior.iai) IAIWrapper (class in torchdrivesim.behavior.iai) InitializationFailedError interaction_replay() (in module torchdrivesim.behavior.replay) IntPerAgentType (in module torchdrivesim.simulator) iou (torchdrivesim.simulator.CollisionMetric attribute) iou_differentiable() (in module torchdrivesim.infractions) iou_differentiable_fast() (in module torchdrivesim._iou_utils) iou_non_differentiable() (in module torchdrivesim._iou_utils) is_available (in module torchdrivesim.lanelet2) (in module torchdrivesim.rendering.nvdiffrast) is_exposed() (torchdrivesim.simulator.SelectiveWrapper method) is_inside_polygon() (in module torchdrivesim.utils) is_warmed_up() (torchdrivesim.simulator.BoundedRegionWrapper method) isin() (in module torchdrivesim.utils) K KinematicBicycle (class in torchdrivesim.kinematic) KinematicModel (class in torchdrivesim.kinematic) L Lanelet2NotFound lanelet_map_to_lane_mesh() (in module torchdrivesim.lanelet2) lanelet_orientation_loss() (in module torchdrivesim.infractions) LANELET_TAGS_TO_EXCLUDE (in module torchdrivesim.infractions) LaneletError left_handed_coordinates (torchdrivesim.rendering.base.RendererConfig attribute) (torchdrivesim.rendering.RendererConfig attribute) (torchdrivesim.simulator.TorchDriveConfig attribute) line_segments_to_mesh() (in module torchdrivesim.lanelet2) load() (torchdrivesim.mesh.AttributeMesh class method) (torchdrivesim.mesh.BaseMesh class method) load_lanelet_map() (in module torchdrivesim.lanelet2) logger (in module torchdrivesim.infractions) (in module torchdrivesim.kinematic) (in module torchdrivesim.mesh) (in module torchdrivesim.rendering) (in module torchdrivesim.rendering.base) (in module torchdrivesim.rendering.nvdiffrast) (in module torchdrivesim.simulator) M make_actor_mesh() (torchdrivesim.rendering.base.BirdviewRenderer method) (torchdrivesim.rendering.BirdviewRenderer method) make_direction_mesh() (torchdrivesim.rendering.base.BirdviewRenderer method) (torchdrivesim.rendering.BirdviewRenderer method) make_renderer() (torchdrivesim.rendering.pytorch3d.Pytorch3DRenderer class method) (torchdrivesim.rendering.Pytorch3DRenderer class method) make_traffic_controls_mesh() (torchdrivesim.rendering.base.BirdviewRenderer method) (torchdrivesim.rendering.BirdviewRenderer method) make_waypoint_mesh() (torchdrivesim.rendering.base.BirdviewRenderer method) (torchdrivesim.rendering.BirdviewRenderer method) map_param() (torchdrivesim.kinematic.KinematicBicycle method) (torchdrivesim.kinematic.KinematicModel method) max_minibatch_size (torchdrivesim.rendering.nvdiffrast.NvdiffrastRendererConfig attribute) (torchdrivesim.rendering.NvdiffrastRendererConfig attribute) merge() (torchdrivesim.mesh.BaseMesh method) merge_dicts() (in module torchdrivesim.utils) module torchdrivesim torchdrivesim._iou_utils torchdrivesim.behavior torchdrivesim.behavior.common torchdrivesim.behavior.heuristic torchdrivesim.behavior.iai torchdrivesim.behavior.replay torchdrivesim.goals torchdrivesim.infractions torchdrivesim.kinematic torchdrivesim.lanelet2 torchdrivesim.mesh torchdrivesim.rendering torchdrivesim.rendering.base torchdrivesim.rendering.nvdiffrast torchdrivesim.rendering.pytorch3d torchdrivesim.simulator torchdrivesim.traffic_controls torchdrivesim.utils N nograd (torchdrivesim.simulator.CollisionMetric attribute) nograd_pytorch3d (torchdrivesim.simulator.CollisionMetric attribute) NoReentryBoundedRegionWrapper (class in torchdrivesim.simulator) normalize_action() (torchdrivesim.kinematic.KinematicBicycle method) (torchdrivesim.kinematic.KinematicModel method) (torchdrivesim.kinematic.SimpleKinematicModel method) normalize_angle() (in module torchdrivesim.utils) NPCWrapper (class in torchdrivesim.simulator) num_categories (torchdrivesim.mesh.BirdviewMesh property) NvdiffrastNotFound NvdiffrastRenderer (class in torchdrivesim.rendering) (class in torchdrivesim.rendering.nvdiffrast) NvdiffrastRendererConfig (class in torchdrivesim.rendering) (class in torchdrivesim.rendering.nvdiffrast) O offroad_infraction_loss() (in module torchdrivesim.infractions) offroad_threshold (torchdrivesim.simulator.TorchDriveConfig attribute) offset() (torchdrivesim.mesh.BaseMesh method) opengl (torchdrivesim.rendering.nvdiffrast.NvdiffrastRendererConfig attribute) (torchdrivesim.rendering.NvdiffrastRendererConfig attribute) oriented_box_intersection_2d() (in module torchdrivesim._iou_utils) OrientedKinematicModel (class in torchdrivesim.kinematic) P pack_state() (torchdrivesim.kinematic.KinematicModel static method) pick_random_point_and_orientation() (in module torchdrivesim.lanelet2) pickle() (torchdrivesim.mesh.BaseMesh method) point_mesh_face_distance() (in module torchdrivesim.infractions) precision_rounding() (in module torchdrivesim._iou_utils) pytorch3d() (torchdrivesim.mesh.AttributeMesh method) (torchdrivesim.mesh.BaseMesh method) (torchdrivesim.mesh.BirdviewMesh method) Pytorch3DRenderer (class in torchdrivesim.rendering) (class in torchdrivesim.rendering.pytorch3d) Pytorch3DRendererConfig (class in torchdrivesim.rendering) (class in torchdrivesim.rendering.pytorch3d) R record() (torchdrivesim.simulator.RecordingWrapper method) RecordingWrapper (class in torchdrivesim.simulator) rectangle_vertices() (in module torchdrivesim.infractions) relative() (in module torchdrivesim.utils) render() (torchdrivesim.simulator.HomogeneousWrapper method) (torchdrivesim.simulator.NPCWrapper method) (torchdrivesim.simulator.SelectiveWrapper method) (torchdrivesim.simulator.Simulator method) (torchdrivesim.simulator.SimulatorInterface method) (torchdrivesim.simulator.SimulatorWrapper method) render_agent_direction (torchdrivesim.rendering.base.RendererConfig attribute) (torchdrivesim.rendering.RendererConfig attribute) render_egocentric() (torchdrivesim.simulator.SimulatorInterface method) render_frame() (torchdrivesim.rendering.base.BirdviewRenderer method) (torchdrivesim.rendering.BirdviewRenderer method) render_mesh() (torchdrivesim.rendering.base.BirdviewRenderer method) (torchdrivesim.rendering.base.DummyRenderer method) (torchdrivesim.rendering.BirdviewRenderer method) (torchdrivesim.rendering.DummyRenderer method) (torchdrivesim.rendering.nvdiffrast.NvdiffrastRenderer method) (torchdrivesim.rendering.NvdiffrastRenderer method) (torchdrivesim.rendering.pytorch3d.Pytorch3DRenderer method) (torchdrivesim.rendering.Pytorch3DRenderer method) render_static_meshes() (torchdrivesim.rendering.base.BirdviewRenderer method) (torchdrivesim.rendering.BirdviewRenderer method) renderer (torchdrivesim.simulator.TorchDriveConfig attribute) renderer_from_config() (in module torchdrivesim.rendering) RendererConfig (class in torchdrivesim.rendering) (class in torchdrivesim.rendering.base) rendering_mesh() (in module torchdrivesim.mesh) RenderingBlend (class in torchdrivesim.rendering.pytorch3d) ReplayWrapper (class in torchdrivesim.behavior.replay) Resolution (in module torchdrivesim.utils) RGBMesh (class in torchdrivesim.mesh) road_mesh_from_lanelet_map() (in module torchdrivesim.lanelet2) rotate() (in module torchdrivesim.utils) rotation_matrix() (in module torchdrivesim.utils) S save() (torchdrivesim.mesh.BaseMesh method) save_gif() (torchdrivesim.simulator.BirdviewRecordingWrapper method) save_png() (torchdrivesim.simulator.BirdviewRecordingWrapper method) select_batch_elements() (torchdrivesim.behavior.iai.IAIWrapper method) (torchdrivesim.behavior.replay.ReplayWrapper method) (torchdrivesim.goals.WaypointGoal method) (torchdrivesim.mesh.AttributeMesh method) (torchdrivesim.mesh.BaseMesh method) (torchdrivesim.mesh.BirdviewMesh method) (torchdrivesim.rendering.base.BirdviewRenderer method) (torchdrivesim.rendering.BirdviewRenderer method) (torchdrivesim.simulator.BirdviewRecordingWrapper method) (torchdrivesim.simulator.BoundedRegionWrapper method) (torchdrivesim.simulator.NoReentryBoundedRegionWrapper method) (torchdrivesim.simulator.SelectiveWrapper method) (torchdrivesim.simulator.Simulator method) (torchdrivesim.simulator.SimulatorInterface method) (torchdrivesim.simulator.SimulatorWrapper method) (torchdrivesim.traffic_controls.BaseTrafficControl method) SelectiveWrapper (class in torchdrivesim.simulator) separate_by_category() (torchdrivesim.mesh.BirdviewMesh method) serialize() (torchdrivesim.mesh.AttributeMesh method) (torchdrivesim.mesh.BaseMesh method) (torchdrivesim.mesh.BirdviewMesh method) set_attr() (torchdrivesim.mesh.AttributeMesh class method) set_color() (torchdrivesim.mesh.RGBMesh class method) set_params() (torchdrivesim.kinematic.KinematicBicycle method) (torchdrivesim.kinematic.KinematicModel method) set_properties() (torchdrivesim.mesh.BirdviewMesh class method) set_state() (torchdrivesim.kinematic.KinematicModel method) (torchdrivesim.simulator.HomogeneousWrapper method) (torchdrivesim.simulator.NPCWrapper method) (torchdrivesim.simulator.SelectiveWrapper method) (torchdrivesim.simulator.Simulator method) (torchdrivesim.simulator.SimulatorInterface method) (torchdrivesim.simulator.SimulatorWrapper method) (torchdrivesim.traffic_controls.BaseTrafficControl method) Shader2D (class in torchdrivesim.rendering.pytorch3d) sigmoid (torchdrivesim.rendering.pytorch3d.RenderingBlend attribute) SimpleKinematicModel (class in torchdrivesim.kinematic) Simulator (class in torchdrivesim.simulator) SimulatorInterface (class in torchdrivesim.simulator) SimulatorWrapper (class in torchdrivesim.simulator) single_agent_rendering (torchdrivesim.simulator.TorchDriveConfig attribute) SingletonAgentTypeFunctor (class in torchdrivesim.simulator) soft (torchdrivesim.rendering.pytorch3d.RenderingBlend attribute) sort_indices() (in module torchdrivesim._iou_utils) state_size (torchdrivesim.kinematic.KinematicModel attribute) step() (torchdrivesim.behavior.iai.IAIWrapper method) (torchdrivesim.behavior.replay.ReplayWrapper method) (torchdrivesim.goals.WaypointGoal method) (torchdrivesim.kinematic.BicycleByDisplacement method) (torchdrivesim.kinematic.BicycleNoReversing method) (torchdrivesim.kinematic.KinematicBicycle method) (torchdrivesim.kinematic.KinematicModel method) (torchdrivesim.kinematic.OrientedKinematicModel method) (torchdrivesim.kinematic.SimpleKinematicModel method) (torchdrivesim.kinematic.TeleportingKinematicModel method) (torchdrivesim.simulator.HomogeneousWrapper method) (torchdrivesim.simulator.NPCWrapper method) (torchdrivesim.simulator.RecordingWrapper method) (torchdrivesim.simulator.SelectiveWrapper method) (torchdrivesim.simulator.Simulator method) (torchdrivesim.simulator.SimulatorInterface method) (torchdrivesim.simulator.SimulatorWrapper method) (torchdrivesim.traffic_controls.BaseTrafficControl method) step_from_xy() (torchdrivesim.kinematic.BicycleByDisplacement method) (torchdrivesim.kinematic.BicycleByOrientedDisplacement method) StopSignControl (class in torchdrivesim.traffic_controls) T TeleportingKinematicModel (class in torchdrivesim.kinematic) tensor_color() (in module torchdrivesim.mesh) TensorPerAgentType (in module torchdrivesim.goals) (in module torchdrivesim.simulator) to() (torchdrivesim.behavior.iai.IAIWrapper method) (torchdrivesim.behavior.replay.ReplayWrapper method) (torchdrivesim.goals.WaypointGoal method) (torchdrivesim.kinematic.KinematicModel method) (torchdrivesim.mesh.AttributeMesh method) (torchdrivesim.mesh.BaseMesh method) (torchdrivesim.mesh.BirdviewMesh method) (torchdrivesim.rendering.base.BirdviewRenderer method) (torchdrivesim.rendering.BirdviewRenderer method) (torchdrivesim.simulator.BirdviewRecordingWrapper method) (torchdrivesim.simulator.BoundedRegionWrapper method) (torchdrivesim.simulator.NoReentryBoundedRegionWrapper method) (torchdrivesim.simulator.NPCWrapper method) (torchdrivesim.simulator.SelectiveWrapper method) (torchdrivesim.simulator.Simulator method) (torchdrivesim.simulator.SimulatorInterface method) (torchdrivesim.simulator.SimulatorWrapper method) (torchdrivesim.traffic_controls.BaseTrafficControl method) to_device() (torchdrivesim.simulator.AgentTypeFunctor method) TorchDriveConfig (class in torchdrivesim.simulator) torchdrivesim module torchdrivesim._iou_utils module torchdrivesim.behavior module torchdrivesim.behavior.common module torchdrivesim.behavior.heuristic module torchdrivesim.behavior.iai module torchdrivesim.behavior.replay module torchdrivesim.goals module torchdrivesim.infractions module torchdrivesim.kinematic module torchdrivesim.lanelet2 module torchdrivesim.mesh module torchdrivesim.rendering module torchdrivesim.rendering.base module torchdrivesim.rendering.nvdiffrast module torchdrivesim.rendering.pytorch3d module torchdrivesim.simulator module torchdrivesim.traffic_controls module torchdrivesim.utils module total_replay_time (torchdrivesim.traffic_controls.BaseTrafficControl property) TrafficLightControl (class in torchdrivesim.traffic_controls) TrajectoryVisualizationWrapper (class in torchdrivesim.simulator) transform() (torchdrivesim.rendering.base.BirdviewRenderer method) (torchdrivesim.rendering.BirdviewRenderer method) trim() (torchdrivesim.mesh.AttributeMesh method) (torchdrivesim.mesh.BaseMesh method) (torchdrivesim.mesh.BirdviewMesh method) U unflattening() (torchdrivesim.kinematic.KinematicBicycle method) (torchdrivesim.kinematic.KinematicModel method) unify() (torchdrivesim.mesh.BirdviewMesh class method) unpack_state() (torchdrivesim.kinematic.KinematicModel static method) unpickle() (torchdrivesim.mesh.AttributeMesh class method) (torchdrivesim.mesh.BaseMesh class method) (torchdrivesim.mesh.BirdviewMesh class method) update_exposed_agents() (torchdrivesim.simulator.BoundedRegionWrapper method) (torchdrivesim.simulator.NoReentryBoundedRegionWrapper method) (torchdrivesim.simulator.SelectiveWrapper method) update_present_mask() (torchdrivesim.simulator.HomogeneousWrapper method) (torchdrivesim.simulator.NoReentryBoundedRegionWrapper method) (torchdrivesim.simulator.NPCWrapper method) (torchdrivesim.simulator.SelectiveWrapper method) (torchdrivesim.simulator.Simulator method) (torchdrivesim.simulator.SimulatorInterface method) (torchdrivesim.simulator.SimulatorWrapper method) V validate_agent_count() (torchdrivesim.simulator.NPCWrapper method) (torchdrivesim.simulator.Simulator method) validate_agent_types() (torchdrivesim.behavior.replay.ReplayWrapper method) (torchdrivesim.simulator.Simulator method) validate_tensor_shapes() (torchdrivesim.behavior.replay.ReplayWrapper method) (torchdrivesim.simulator.Simulator method) vert_category (torchdrivesim.mesh.BirdviewMesh attribute) verts (torchdrivesim.mesh.BaseMesh attribute) verts_count (torchdrivesim.mesh.BaseMesh property) violation_rear_factor (torchdrivesim.traffic_controls.TrafficLightControl attribute) W WaypointGoal (class in torchdrivesim.goals) Y YieldControl (class in torchdrivesim.traffic_controls) Z zs (torchdrivesim.mesh.BirdviewMesh attribute)