Index _ | A | B | C | D | E | F | G | H | I | K | L | M | N | O | P | R | S | T | U | V | W | X | Y | Z _ __call__() (torchdrivesim.simulator.AgentTypeFunctor method) __getitem__() (torchdrivesim.mesh.BaseMesh method) (torchdrivesim.simulator.SimulatorInterface method) __post_init__() (torchdrivesim.map.Stopline method) (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) _filename_defaults() (in module torchdrivesim.map) _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) _resource_path (in module torchdrivesim) _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) actor_id (torchdrivesim.map.Stopline attribute) actor_states (torchdrivesim.traffic_lights.TrafficLightGroupState attribute) ActorStates (in module torchdrivesim.traffic_lights) 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_type (torchdrivesim.map.Stopline attribute) 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) assert_pytorch3d_available() (in module torchdrivesim) 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.cv2.CV2RendererConfig attribute) (torchdrivesim.rendering.CV2RendererConfig 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) Cameras (class in torchdrivesim.rendering.base) categories (torchdrivesim.mesh.BirdviewMesh attribute) center (torchdrivesim.map.MapConfig attribute) (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) collect_all_current_light_states() (torchdrivesim.traffic_lights.TrafficLightController 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.pytorch3d) 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) current_light_state_tensor_from_controller() (in module torchdrivesim.traffic_lights) current_state (torchdrivesim.traffic_lights.TrafficLightController property) (torchdrivesim.traffic_lights.TrafficLightStateMachine property) current_state_with_name (torchdrivesim.traffic_lights.TrafficLightController property) CV2Renderer (class in torchdrivesim.rendering) (class in torchdrivesim.rendering.cv2) CV2RendererConfig (class in torchdrivesim.rendering) (class in torchdrivesim.rendering.cv2) 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) download_iai_map() (in module torchdrivesim.map) DummyRenderer (class in torchdrivesim.rendering) (class in torchdrivesim.rendering.base) DummyRendererConfig (class in torchdrivesim.rendering) (class in torchdrivesim.rendering.base) duration (torchdrivesim.traffic_lights.TrafficLightGroupState attribute) (torchdrivesim.traffic_lights.TrafficLightStateMachine property) 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) find_map_config() (in module torchdrivesim.map) find_wrong_way_stoplines() (in module torchdrivesim.map) 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) from_json() (torchdrivesim.traffic_lights.TrafficLightController class method) (torchdrivesim.traffic_lights.TrafficLightStateMachine class 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_camera_center() (torchdrivesim.rendering.base.Cameras method) get_color() (torchdrivesim.rendering.base.BirdviewRenderer method) (torchdrivesim.rendering.BirdviewRenderer method) get_current_actor_states() (torchdrivesim.traffic_lights.TrafficLightStateMachine 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_number_of_light_groups() (torchdrivesim.traffic_lights.TrafficLightController 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_view_to_proj_transform() (torchdrivesim.rendering.base.Cameras 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_waypoints_state() (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) get_world_to_view_transform() (torchdrivesim.rendering.base.Cameras method) glctx_sessions (in module torchdrivesim.rendering.nvdiffrast) green (torchdrivesim.traffic_lights.TrafficLightState attribute) 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) iai_location_name (torchdrivesim.map.MapConfig attribute) 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) (in module torchdrivesim.rendering.pytorch3d) 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_inclusion_tolerance (torchdrivesim.simulator.TorchDriveConfig attribute) lanelet_map (torchdrivesim.map.MapConfig property) lanelet_map_origin (torchdrivesim.map.MapConfig attribute) lanelet_map_to_lane_mesh() (in module torchdrivesim.lanelet2) lanelet_orientation_loss() (in module torchdrivesim.infractions) lanelet_path (torchdrivesim.map.MapConfig attribute) LANELET_TAGS_TO_EXCLUDE (in module torchdrivesim.infractions) LaneletError left_handed_coordinates (torchdrivesim.map.MapConfig attribute) (torchdrivesim.rendering.base.RendererConfig attribute) (torchdrivesim.rendering.RendererConfig attribute) (torchdrivesim.simulator.TorchDriveConfig attribute) length (torchdrivesim.map.Stopline 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) load_map_config() (in module torchdrivesim.map) 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) (in module torchdrivesim.traffic_lights) 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) MapConfig (class in torchdrivesim.map) max_minibatch_size (torchdrivesim.rendering.nvdiffrast.NvdiffrastRendererConfig attribute) (torchdrivesim.rendering.NvdiffrastRendererConfig attribute) merge() (torchdrivesim.mesh.BaseMesh method) merge_dicts() (in module torchdrivesim.utils) mesh_path (torchdrivesim.map.MapConfig attribute) 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.map torchdrivesim.mesh torchdrivesim.rendering torchdrivesim.rendering.base torchdrivesim.rendering.cv2 torchdrivesim.rendering.nvdiffrast torchdrivesim.rendering.pytorch3d torchdrivesim.simulator torchdrivesim.traffic_controls torchdrivesim.traffic_lights torchdrivesim.utils N name (torchdrivesim.map.MapConfig attribute) next_state (torchdrivesim.traffic_lights.TrafficLightGroupState attribute) nograd (torchdrivesim.simulator.CollisionMetric attribute) nograd_pytorch3d (torchdrivesim.simulator.CollisionMetric attribute) none (torchdrivesim.traffic_lights.TrafficLightState 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) note (torchdrivesim.map.MapConfig attribute) 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) orientation (torchdrivesim.map.Stopline 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) point_to_mesh_distance_pt() (in module torchdrivesim.infractions) precision_rounding() (in module torchdrivesim._iou_utils) project_world_to_clip_space() (torchdrivesim.rendering.base.Cameras method) pytorch3d() (torchdrivesim.mesh.AttributeMesh method) (torchdrivesim.mesh.BaseMesh method) (torchdrivesim.mesh.BirdviewMesh method) Pytorch3DNotFound 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) red (torchdrivesim.traffic_lights.TrafficLightState attribute) 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.cv2.CV2Renderer method) (torchdrivesim.rendering.CV2Renderer 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) reset() (torchdrivesim.traffic_lights.TrafficLightController method) (torchdrivesim.traffic_lights.TrafficLightStateMachine method) Resolution (in module torchdrivesim.utils) resolve_paths_to_absolute() (in module torchdrivesim.map) reverse_transform_points_screen() (torchdrivesim.rendering.base.Cameras method) RGBMesh (class in torchdrivesim.mesh) road_mesh (torchdrivesim.map.MapConfig property) 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) sequence_number (torchdrivesim.traffic_lights.TrafficLightGroupState attribute) 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) set_to() (torchdrivesim.traffic_lights.TrafficLightController method) (torchdrivesim.traffic_lights.TrafficLightStateMachine 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_per_machine (torchdrivesim.traffic_lights.TrafficLightController property) state_size (torchdrivesim.kinematic.KinematicModel attribute) states (torchdrivesim.traffic_lights.TrafficLightStateMachine property) 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) Stopline (class in torchdrivesim.map) stoplines (torchdrivesim.map.MapConfig property) stoplines_path (torchdrivesim.map.MapConfig attribute) StopSignControl (class in torchdrivesim.traffic_controls) store_map_config() (in module torchdrivesim.map) T TeleportingKinematicModel (class in torchdrivesim.kinematic) tensor_color() (in module torchdrivesim.mesh) TensorPerAgentType (in module torchdrivesim.goals) (in module torchdrivesim.simulator) tick() (torchdrivesim.traffic_lights.TrafficLightController method) (torchdrivesim.traffic_lights.TrafficLightStateMachine method) time_remaining (torchdrivesim.traffic_lights.TrafficLightController property) (torchdrivesim.traffic_lights.TrafficLightStateMachine property) 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) to_json() (torchdrivesim.traffic_lights.TrafficLightController method) (torchdrivesim.traffic_lights.TrafficLightStateMachine 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.map module torchdrivesim.mesh module torchdrivesim.rendering module torchdrivesim.rendering.base module torchdrivesim.rendering.cv2 module torchdrivesim.rendering.nvdiffrast module torchdrivesim.rendering.pytorch3d module torchdrivesim.simulator module torchdrivesim.traffic_controls module torchdrivesim.traffic_lights module torchdrivesim.utils module total_replay_time (torchdrivesim.traffic_controls.BaseTrafficControl property) traffic_controls_from_map_config() (in module torchdrivesim.map) traffic_light_controller (torchdrivesim.map.MapConfig property) traffic_light_controller_path (torchdrivesim.map.MapConfig attribute) TrafficLightControl (class in torchdrivesim.traffic_controls) TrafficLightController (class in torchdrivesim.traffic_lights) TrafficLightGroupState (class in torchdrivesim.traffic_lights) TrafficLightState (class in torchdrivesim.traffic_lights) TrafficLightStateMachine (class in torchdrivesim.traffic_lights) TrajectoryVisualizationWrapper (class in torchdrivesim.simulator) transform() (torchdrivesim.rendering.base.BirdviewRenderer method) (torchdrivesim.rendering.BirdviewRenderer method) transform_points_screen() (torchdrivesim.rendering.base.Cameras method) trim() (torchdrivesim.mesh.AttributeMesh method) (torchdrivesim.mesh.BaseMesh method) (torchdrivesim.mesh.BirdviewMesh method) trim_mesh_before_rendering (torchdrivesim.rendering.cv2.CV2RendererConfig attribute) (torchdrivesim.rendering.CV2RendererConfig attribute) U unflattening() (torchdrivesim.kinematic.KinematicBicycle method) (torchdrivesim.kinematic.KinematicModel method) unify() (torchdrivesim.mesh.BirdviewMesh class method) unpack_attributes() (in module torchdrivesim.behavior.iai) unpack_state() (torchdrivesim.kinematic.KinematicModel static method) unpickle() (torchdrivesim.mesh.AttributeMesh class method) (torchdrivesim.mesh.BaseMesh class method) (torchdrivesim.mesh.BirdviewMesh class method) update_current_state_and_time() (torchdrivesim.traffic_lights.TrafficLightController 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 waypoint_removal_threshold (torchdrivesim.simulator.TorchDriveConfig attribute) WaypointGoal (class in torchdrivesim.goals) width (torchdrivesim.map.Stopline attribute) wrong_way_angle_threshold (torchdrivesim.simulator.TorchDriveConfig attribute) X x (torchdrivesim.map.Stopline attribute) Y y (torchdrivesim.map.Stopline attribute) yellow (torchdrivesim.traffic_lights.TrafficLightState attribute) YieldControl (class in torchdrivesim.traffic_controls) Z zs (torchdrivesim.mesh.BirdviewMesh attribute)