RoboWaiter/robowaiter/behavior_tree/ptml/test/ptml_test.py

25 lines
738 B
Python
Raw Normal View History

# from robowaiter.scene.scene import Scene
# from robowaiter.behavior_tree.ptml.ptmlCompiler import load
import os
from robowaiter import Robot, task_map
2023-11-09 16:07:02 +08:00
from robowaiter.utils.bt.draw import render_dot_tree
if __name__ == '__main__':
TASK_NAME = 'OT'
# create robot
project_path = "../../../"
2024-01-10 11:16:12 +08:00
ptml_path = os.path.join(project_path, 'behavior_tree/ptml/llm_test/Default.ptml')
behavior_lib_path = os.path.join(project_path, 'behavior_lib')
robot = Robot(ptml_path, behavior_lib_path)
# create task
task = task_map[TASK_NAME](robot)
2024-01-10 11:16:12 +08:00
render_dot_tree(robot.bt.root,name="llm_test")
# build and tick
# scene.BT = ptree.trees.BehaviourTree(scene.BT)
# todo: tick this bt
print(robot.bt)