33 lines
807 B
Python
33 lines
807 B
Python
import py_trees as ptree
|
|
from typing import Any
|
|
from robowaiter.behavior_lib._base.Act import Act
|
|
from robowaiter.behavior_lib._base.Behavior import Status
|
|
|
|
class Make(Act):
|
|
can_be_expanded = True
|
|
num_args = 1
|
|
valid_args = (
|
|
"Coffee",
|
|
)
|
|
|
|
def __init__(self, *args):
|
|
super().__init__(*args)
|
|
self.target_obj = self.args[0]
|
|
|
|
|
|
@classmethod
|
|
def get_info(cls,arg):
|
|
info = None
|
|
if arg == "Coffee":
|
|
info = {
|
|
"add": {f'On(Coffee,Table)'},
|
|
}
|
|
return info
|
|
|
|
|
|
def _update(self) -> ptree.common.Status:
|
|
op_type = 1
|
|
self.scene.move_task_area(op_type)
|
|
self.scene.op_task_execute(op_type)
|
|
self.scene.state["condition_set"].add("At(Coffee,Bar)")
|
|
return Status.RUNNING |