hofee/nbv_sim
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add new service

Browse Source
This commit is contained in:
0nhc 2024-10-13 01:04:35 -05:00
parent 82815304bd
commit 60e327b8fb
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
scripts/bt_sim_node.py
View File

@ -73,7 +73,7 @@ class BtSimNode:
SwitchController,
self.switch_controller,
)
rospy.Service("get_target_seg_id", Trigger, self.get_target_seg_id)
rospy.Service("get_target_seg_id", TargetID, self.get_target_seg_id)
def seed(self, req):
self.sim.seed(req.seed)
@ -88,10 +88,10 @@ class BtSimNode:
self.activate_plugins()
return ResetResponse(to_bbox_msg(bbox))
def seed(self, req):
self.sim.seed(req.seed)
rospy.loginfo(f"Seeded the rng with {req.seed}.")
return SeedResponse()
def get_target_seg_id(self, req):
response = TargetIDResponse()
response.id = self.sim.select_target()
return response
def switch_controller(self, req):
for controller in req.stop_controllers: