# add a radar sensor robot_id = robot_params['id'] sim.add_sensor(robot_id, sensor_params) readings = sim.read_sensors() # Initialize player model for control input p = Player_vi(all_params, sim.get_occupied()) # Initialize View if GUI_ON: rstate = sim.get_state() bstate = p.get_bstate() v = View() v.update_all(rstate, bstate) # Main simulation loop rstate = None iters = 0 start_time = time.time() for i in range(MAX_STEPS): iters += 1 reading = sim.read_sensors()[robot_id] p.update_belief(reading) p_act = p.get_action(reading) #p_act = action_seq[i] sim.add_action(p_act, robot_id) sim.update_turn()
#from model.world_model.gen_maze import gen_terrain_dfs #from model.world_model.Maze import Maze from model.Simulation import Simulation from view.render2d import View #import matplotlib.pyplot as plt #import numpy as np sim_params = {'world_type':'maze', 'world_length':100, 'world_width':100, 'robot_length':2, 'robot_stochastic':False, 'radar_radius':8,} robot_id = 1 v = View() sim = Simulation(sim_params) state = sim.get_state() v.update_rstate(state) sim.add_robot('car',robot_id) state = sim.get_state() v.update_rstate(state) sim.add_sensor(robot_id, 'radar') readings = sim.read_sensors() print readings raw_input('pause : press any key to finish...')