73 lines
2.3 KiB
Python
73 lines
2.3 KiB
Python
|
|
from EXP.exp_tools import collect_action_nodes,get_start,BTTest,goal_transfer_str,collect_cond_nodes
|
|
import copy
|
|
import random
|
|
seed = 1
|
|
random.seed(seed)
|
|
|
|
multiple_num=1
|
|
action_list = collect_action_nodes(random,multiple_num)
|
|
# for act in action_list:
|
|
# print(act.name,act.cost)
|
|
|
|
start_robowaiter = get_start()
|
|
|
|
# 计算state总数
|
|
state_num = collect_cond_nodes()
|
|
print("states num: ",state_num)
|
|
print("act num: ",len(action_list))
|
|
|
|
# goal_states = []
|
|
# with open('easy.txt', 'r') as file:
|
|
# # with open('easy_easy.txt', 'r') as file:
|
|
# for line in file:
|
|
# clean_line = line.strip()
|
|
# goal_states.append(clean_line)
|
|
# print(goal_states)
|
|
|
|
# goal_set_ls=[]
|
|
# for count, goal_str in enumerate(goal_states):
|
|
# goal = copy.deepcopy(goal_transfer_str(goal_str))
|
|
# goal_set_ls.append(goal)
|
|
# print(goal_set_ls)
|
|
|
|
# goal_states={"On_Dessert_Bar"}
|
|
# goal_states={"On_MilkDrink_Bar2"}
|
|
# goal_states={"Is_TubeLight_On"}
|
|
# goal_states = goal_set_ls
|
|
# goal_states = {'On(VacuumCup,WaterTable)'}
|
|
# goal_states = {'At(Robot,WaterTable)'}
|
|
# goal_states = {'Is(Table1,Clean)'}
|
|
|
|
# goal_states = {'On(Coffee,Bar)'}
|
|
# goal_states = {'IsClean_Table1'}
|
|
# goal_states = {'~On_Water_Bar'}
|
|
# goal_states = {'~On_Water_Table2'}
|
|
# goal_states = {'~Near_Robot_Bar'}
|
|
# goal_states = {'~Low_ACTemperature'}
|
|
# goal_states = {'~Closed_Curtain'}
|
|
# goal_states = {'~Active_TubeLight'}
|
|
# goal_states = {'~Active_HallLight'}
|
|
|
|
|
|
# goal_states = {'~On_Water_Bar & (On_Coffee_Table2 | On_Bernachon_Table2)'}
|
|
# goal_states = {'~On_Water_Bar & On_Coffee_Table2'}
|
|
# goal_states = {'On_Water_Bar'}
|
|
# goal_states = {'~On_Water_Bar'}
|
|
# goal_states = {'On_Softdrink_Bar'}
|
|
# goal_states = {'On_Coffee_Table2 | On_Bernachon_Table2'}
|
|
|
|
# goal_states = {'Low_ACTemperature'}
|
|
# goal_states = {'~On_Softdrink_Table1 & Closed_Curtain'}
|
|
goal_states = {'On_Coffee_Bar & (IsClean_Floor | ~Active_TubeLight)'}
|
|
|
|
|
|
# goal_states = {'~Closed_Curtain & On_Coffee_Bar'}
|
|
|
|
# todo: 行为树鲁棒性测试，随机生成规划问题
|
|
# # 设置生成规划问题集的超参数：文字数、解深度、迭代次数
|
|
|
|
BTTest(bt_algo_opt=True, goal_states=goal_states,action_list=action_list,start_robowaiter=start_robowaiter)
|
|
# print("\n")
|
|
# 对比
|
|
# BTTest(bt_algo_opt=False, goal_states=goal_states,action_list=action_list,start_robowaiter=start_robowaiter) |