85 lines
2.3 KiB
Python
85 lines
2.3 KiB
Python
import os
|
|
import pickle
|
|
import time
|
|
import random
|
|
import math
|
|
|
|
import matplotlib.pyplot as plt
|
|
import numpy as np
|
|
|
|
from robowaiter.scene import scene
|
|
# from navigate import Navigator
|
|
from robowaiter.algos.navigate.navigate import Navigator
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
# def navigate_test(scene):
|
|
|
|
file_name = 'map_5.pkl'
|
|
if os.path.exists(file_name):
|
|
with open(file_name, 'rb') as file:
|
|
map = pickle.load(file)
|
|
|
|
scene.init_world(1, 11)
|
|
scene = scene.Scene(sceneID=0)
|
|
|
|
navigator = Navigator(scene=scene, area_range=[-350, 600, -400, 1450], map=map)
|
|
|
|
'''场景1: 无行人环境 robot到达指定目标'''
|
|
# goal = np.array((-100, 700))
|
|
|
|
'''场景2: 静止行人环境 robot到达指定目标'''
|
|
# scene.clean_walker()
|
|
# scene.add_walker(50, 300, 0)
|
|
# scene.add_walker(-50, 500, 0)
|
|
# goal = np.array((-100, 700))
|
|
|
|
'''场景3: 移动行人环境 robot到达指定目标'''
|
|
scene.walk_to(100, 0, -90, dis_limit=10)
|
|
scene.clean_walker()
|
|
scene.add_walkers([[50, 300],[-50, 500],[0, 700]])
|
|
# scene.add_walker(50, 300, 0)
|
|
# scene.add_walker(-50, 500, 0)
|
|
# scene.add_walker(0, 700, 0)
|
|
scene.control_walker([scene.walker_control_generator(walkerID=0, autowalk=False, speed=50, X=-50, Y=600, Yaw=0)])
|
|
scene.control_walker([scene.walker_control_generator(walkerID=1, autowalk=False, speed=50, X=100, Y=150, Yaw=0)])
|
|
scene.control_walker([scene.walker_control_generator(walkerID=2, autowalk=False, speed=50, X=0, Y=0, Yaw=0)])
|
|
|
|
# goal = (-100, 700)
|
|
# goal = (-300)
|
|
# goal = (340.0, 900.0)
|
|
|
|
# goal = (240.0, 1000.0)
|
|
# goal = (340.0, 900.0)
|
|
goal = (240.0, 1160.0)
|
|
|
|
'''场景4: 行人自由移动 robot到达指定目标'''
|
|
# # TODO: autowalk=True仿真器会闪退 ???
|
|
# scene.clean_walker()
|
|
# scene.add_walker(50, 300, 0)
|
|
# scene.add_walker(-50, 500, 0)
|
|
# scene.control_walker([scene.walker_control_generator(walkerID=0, autowalk=True, speed=20, X=0, Y=0, Yaw=0)])
|
|
# scene.control_walker([scene.walker_control_generator(walkerID=1, autowalk=True, speed=20, X=0, Y=0, Yaw=0)])
|
|
# time.sleep(5)
|
|
# goal = np.array((-100, 700))
|
|
|
|
navigator.navigate(goal, animation=False)
|
|
|
|
scene.clean_walker()
|
|
print(scene.status.collision) # TODO: 不显示碰撞信息 ???
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|