Skip to content

Commit 2421702

Browse files
committed
untested use nicknames for mapping to cam
1 parent b59ea07 commit 2421702

File tree

2 files changed

+6
-5
lines changed

2 files changed

+6
-5
lines changed

PiSideCode/init.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -265,6 +265,7 @@ def log_performance(walleye_data):
265265
# for i in range(len(poses)):
266266
# if poses[i][0].X() < 2000: # TODO what is this doing here?
267267
walleye_data.robot_publisher.udp_pose_publish(
268+
[walleye_data.cam_nicknames[identifier] for identifier in images.keys()],
268269
[pose[0] for pose in poses],
269270
[pose[1] for pose in poses],
270271
ambig,
@@ -276,7 +277,7 @@ def log_performance(walleye_data):
276277

277278
# Tag servoing mode
278279
elif len(tag_corners) > 0:
279-
walleye_data.robot_publisher.udp_tag_publish(tags, tag_corners, list(img_time.values()))
280+
walleye_data.robot_publisher.udp_tag_publish([walleye_data.cam_nicknames[identifier] for identifier in images.keys()], tags, tag_corners, list(img_time.values()))
280281

281282
# Update video stream for web interface
282283
if walleye_data.should_update_web_stream:

PiSideCode/publisher/network_table_publisher.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,7 @@ def get_time(self) -> int:
6060
def set_table(self, name: str):
6161
self.table = self.inst.getTable(name)
6262

63-
def udp_pose_publish(self, pose1, pose2, ambig, timestamps, tags, tag_corners):
63+
def udp_pose_publish(self, names, pose1, pose2, ambig, timestamps, tags, tag_corners):
6464
data_dict = {}
6565

6666
for i in range(len(pose1)):
@@ -79,7 +79,7 @@ def udp_pose_publish(self, pose1, pose2, ambig, timestamps, tags, tag_corners):
7979
"Tags": tags[i].tolist(),
8080
"TagCorners": tag_corners[i].tolist(),
8181
}
82-
data_dict[self.name + str(i)] = camDict
82+
data_dict[names[i]] = camDict
8383
data_str = json.dumps(data_dict)
8484

8585
try:
@@ -88,7 +88,7 @@ def udp_pose_publish(self, pose1, pose2, ambig, timestamps, tags, tag_corners):
8888
# NetworkIO.logger.error("Failed to publish pose in UDP: ", exc_info=e)
8989
pass
9090

91-
def udp_tag_publish(self, tags, tag_corners, timestamps):
91+
def udp_tag_publish(self, names, tags, tag_corners, timestamps):
9292
data_dict = {}
9393

9494
# Loop through each camera
@@ -99,7 +99,7 @@ def udp_tag_publish(self, tags, tag_corners, timestamps):
9999
self.update_num[i] += 1
100100

101101
# Data for camera i
102-
data_dict[self.name + str(i)] = {
102+
data_dict[names[i]] = {
103103
"Mode": 1,
104104
"Update": (self.update_num[i]),
105105
"Tags": tags[i].tolist(),

0 commit comments

Comments
 (0)