flatland.envs package

Subpackages

• flatland.envs.step_utils package
  • Submodules
    • flatland.envs.step_utils.action_preprocessing module
      • preprocess_action_when_waiting()
      • preprocess_moving_action()
      • preprocess_raw_action()
      • process_do_nothing()
      • process_illegal_action()
      • process_left_right()
    • flatland.envs.step_utils.action_saver module
      • ActionSaver
    • flatland.envs.step_utils.env_utils module
      • AgentTransitionData
      • apply_action_independent()
      • state_position_sync_check()
    • flatland.envs.step_utils.malfunction_handler module
      • MalfunctionHandler
      • get_number_of_steps_to_break()
    • flatland.envs.step_utils.speed_counter module
      • SpeedCounter
    • flatland.envs.step_utils.state_machine module
      • TrainStateMachine
    • flatland.envs.step_utils.states module
      • StateTransitionSignals
      • TrainState
    • flatland.envs.step_utils.transition_utils module
      • check_action()
      • check_action_on_agent()
      • check_bounds()
      • check_valid_action()
  • Module contents

Submodules

• flatland.envs.agent_chains module
  • ChainTestEnv
    • ChainTestEnv.addAgent()
    • ChainTestEnv.addAgentToRow()
    • ChainTestEnv.create_test_chain()
    • ChainTestEnv.nextRow()
  • MotionCheck
    • MotionCheck.addAgent()
    • MotionCheck.block_preds()
    • MotionCheck.check_motion()
    • MotionCheck.find_conflicts()
    • MotionCheck.find_same_dest()
    • MotionCheck.find_stop_preds()
    • MotionCheck.find_stops()
    • MotionCheck.find_stops2()
    • MotionCheck.find_swaps()
    • MotionCheck.find_swaps2()
    • MotionCheck.get_G_reversed()
    • MotionCheck.reset_G_reversed()
  • create_test_agents()
  • create_test_agents2()
  • main()
  • render()
  • test_agent_following()
• flatland.envs.agent_utils module
  • Agent
    • Agent.action_saver
    • Agent.arrival_time
    • Agent.direction
    • Agent.earliest_departure
    • Agent.handle
    • Agent.initial_direction
    • Agent.initial_position
    • Agent.latest_arrival
    • Agent.malfunction_handler
    • Agent.moving
    • Agent.old_direction
    • Agent.old_position
    • Agent.position
    • Agent.speed_counter
    • Agent.state_machine
    • Agent.target
    • Agent.waypoints
    • Agent.waypoints_earliest_departure
    • Agent.waypoints_latest_arrival
  • EnvAgent
    • EnvAgent.apply_timetable()
    • EnvAgent.from_line()
    • EnvAgent.get_current_delay()
    • EnvAgent.get_shortest_path()
    • EnvAgent.get_time_remaining_until_latest_arrival()
    • EnvAgent.get_travel_time_on_shortest_path()
    • EnvAgent.load_legacy_static_agent()
    • EnvAgent.malfunction_data
    • EnvAgent.reset()
    • EnvAgent.speed_data
    • EnvAgent.state
    • EnvAgent.to_agent()
  • load_env_agent()
• flatland.envs.distance_map module
  • DistanceMap
    • DistanceMap.get()
    • DistanceMap.reset()
    • DistanceMap.set()
• flatland.envs.fast_methods module
  • fast_argmax()
  • fast_clip()
  • fast_count_nonzero()
  • fast_delete()
  • fast_isclose()
  • fast_position_equal()
  • fast_where()
• flatland.envs.flatland_envs_rail_env_utils module
• flatland.envs.grid4_generators_utils module
  • align_cell_to_city()
  • connect_rail_in_grid_map()
  • connect_straight_line_in_grid_map()
  • fix_inner_nodes()
• flatland.envs.line_generators module
  • BaseLineGen
    • BaseLineGen.generate()
  • SparseLineGen
    • SparseLineGen.decide_orientation()
    • SparseLineGen.generate()
  • line_from_file()
  • sparse_line_generator()
  • speed_initialization_helper()
• flatland.envs.malfunction_generators module
  • FileMalfunctionGen
  • Malfunction
    • Malfunction.num_broken_steps
  • MalfunctionParameters
    • MalfunctionParameters.malfunction_rate
    • MalfunctionParameters.max_duration
    • MalfunctionParameters.min_duration
  • MalfunctionProcessData
    • MalfunctionProcessData.malfunction_rate
    • MalfunctionProcessData.max_duration
    • MalfunctionProcessData.min_duration
  • NoMalfunctionGen
  • ParamMalfunctionGen
    • ParamMalfunctionGen.generate()
    • ParamMalfunctionGen.generate_rand_numbers()
    • ParamMalfunctionGen.get_process_data()
  • malfunction_from_file()
  • malfunction_from_params()
  • no_malfunction_generator()
  • single_malfunction_generator()
• flatland.envs.observations module
  • GlobalObsForRailEnv
    • GlobalObsForRailEnv.get()
    • GlobalObsForRailEnv.reset()
    • GlobalObsForRailEnv.set_env()
  • LocalObsForRailEnv
    • LocalObsForRailEnv.field_of_view()
    • LocalObsForRailEnv.get()
    • LocalObsForRailEnv.reset()
  • Node
    • Node.childs
    • Node.dist_min_to_target
    • Node.dist_other_agent_encountered
    • Node.dist_other_target_encountered
    • Node.dist_own_target_encountered
    • Node.dist_potential_conflict
    • Node.dist_to_next_branch
    • Node.dist_unusable_switch
    • Node.num_agents_malfunctioning
    • Node.num_agents_opposite_direction
    • Node.num_agents_ready_to_depart
    • Node.num_agents_same_direction
    • Node.speed_min_fractional
  • TreeObsForRailEnv
    • TreeObsForRailEnv.get()
    • TreeObsForRailEnv.get_many()
    • TreeObsForRailEnv.print_node_features()
    • TreeObsForRailEnv.print_subtree()
    • TreeObsForRailEnv.reset()
    • TreeObsForRailEnv.set_env()
    • TreeObsForRailEnv.tree_explored_actions_char
    • TreeObsForRailEnv.util_print_obs_subtree()
• flatland.envs.persistence module
  • RailEnvPersister
    • RailEnvPersister.deprecated_get_agent_state_msg()
    • RailEnvPersister.deprecated_get_full_state_dist_msg()
    • RailEnvPersister.deprecated_get_full_state_msg()
    • RailEnvPersister.deprecated_set_full_state_dist_msg()
    • RailEnvPersister.deprecated_set_full_state_msg()
    • RailEnvPersister.get_full_state()
    • RailEnvPersister.load()
    • RailEnvPersister.load_env_dict()
    • RailEnvPersister.load_new()
    • RailEnvPersister.load_resource()
    • RailEnvPersister.save()
    • RailEnvPersister.save_episode()
    • RailEnvPersister.set_full_state()
• flatland.envs.predictions module
  • DummyPredictorForRailEnv
    • DummyPredictorForRailEnv.get()
  • ShortestPathPredictorForRailEnv
    • ShortestPathPredictorForRailEnv.get()
• flatland.envs.rail_env module
  • RailEnv
    • RailEnv.action_required()
    • RailEnv.add_agent()
    • RailEnv.clear_rewards_dict()
    • RailEnv.close()
    • RailEnv.end_of_episode_update()
    • RailEnv.generate_state_transition_signals()
    • RailEnv.get_agent_handles()
    • RailEnv.get_info_dict()
    • RailEnv.get_num_agents()
    • RailEnv.get_valid_directions_on_grid()
    • RailEnv.handle_done_state()
    • RailEnv.initialize_renderer()
    • RailEnv.preprocess_action()
    • RailEnv.record_timestep()
    • RailEnv.render()
    • RailEnv.reset()
    • RailEnv.reset_agents()
    • RailEnv.save()
    • RailEnv.step()
    • RailEnv.update_renderer()
    • RailEnv.update_step_rewards()
• flatland.envs.rail_env_action module
  • RailEnvActions
    • RailEnvActions.DO_NOTHING
    • RailEnvActions.MOVE_FORWARD
    • RailEnvActions.MOVE_LEFT
    • RailEnvActions.MOVE_RIGHT
    • RailEnvActions.STOP_MOVING
    • RailEnvActions.is_action_valid()
    • RailEnvActions.is_moving_action()
    • RailEnvActions.to_char()
  • RailEnvGridPos
    • RailEnvGridPos.c
    • RailEnvGridPos.r
  • RailEnvNextAction
    • RailEnvNextAction.action
    • RailEnvNextAction.next_direction
    • RailEnvNextAction.next_position
• flatland.envs.rail_env_shortest_paths module
  • get_action_for_move()
  • get_k_shortest_paths()
  • get_new_position_for_action()
  • get_shortest_paths()
  • get_valid_move_actions_()
  • visualize_distance_map()
• flatland.envs.rail_env_utils module
  • env_generator()
  • load_flatland_environment_from_file()
• flatland.envs.rail_generators module
  • EmptyRailGen
    • EmptyRailGen.generate()
  • RailFromGridGen
    • RailFromGridGen.generate()
  • RailGen
    • RailGen.generate()
  • RailGeneratorProduct
  • SparseRailGen
    • SparseRailGen.argsort()
    • SparseRailGen.generate()
    • SparseRailGen.get_closest_neighbour_for_direction()
  • empty_rail_generator()
  • rail_from_file()
  • rail_from_grid_transition_map()
  • sparse_rail_generator()
• flatland.envs.rail_trainrun_data_structures module
  • TrainrunWaypoint
    • TrainrunWaypoint.scheduled_at
    • TrainrunWaypoint.waypoint
  • Waypoint
    • Waypoint.direction
    • Waypoint.position
• flatland.envs.rewards module
  • Rewards
    • Rewards.alpha
    • Rewards.beta
    • Rewards.cancellation_factor
    • Rewards.cancellation_time_buffer
    • Rewards.end_of_episode_reward()
    • Rewards.epsilon
    • Rewards.global_reward
    • Rewards.intermediate_early_departure_penalty_factor
    • Rewards.intermediate_late_arrival_penalty_factor
    • Rewards.intermediate_not_served_penalty
    • Rewards.invalid_action_penalty
    • Rewards.start_penalty
    • Rewards.step_penalty
    • Rewards.step_reward()
    • Rewards.stop_penalty
• flatland.envs.schedule_generators module
• flatland.envs.timetable_generators module
  • len_handle_none()
  • timetable_generator()
  • ttgen_flatland2()
• flatland.envs.timetable_utils module
  • Line
    • Line.agent_directions
    • Line.agent_positions
    • Line.agent_speeds
    • Line.agent_targets
  • Timetable
    • Timetable.earliest_departures
    • Timetable.latest_arrivals
    • Timetable.max_episode_steps

Module contents