bcf - return added viewpoint in add_visinfo_handler #4453

This commit is contained in:
Andrej730
2024-08-09 15:58:46 +05:00
parent 3e7a2c1b83
commit 526a2a6a80
2 changed files with 13 additions and 5 deletions
+7 -2
View File
@@ -293,9 +293,14 @@ class TopicHandler:
self.add_visinfo_handler(vi_handler) self.add_visinfo_handler(vi_handler)
return vi_handler return vi_handler
def add_visinfo_handler(self, new_viewpoint: VisualizationInfoHandler) -> None: def add_visinfo_handler(self, new_viewpoint: VisualizationInfoHandler) -> mdl.ViewPoint:
self.viewpoints[new_viewpoint.guid + ".bcfv"] = new_viewpoint self.viewpoints[new_viewpoint.guid + ".bcfv"] = new_viewpoint
self.markup.viewpoints.append(mdl.ViewPoint(viewpoint=new_viewpoint.guid + ".bcfv", guid=new_viewpoint.guid)) viewpoint = mdl.ViewPoint(
viewpoint=new_viewpoint.guid + ".bcfv",
guid=new_viewpoint.guid,
)
self.markup.viewpoints.append(viewpoint)
return viewpoint
def __eq__(self, other: object) -> bool | NoReturn: def __eq__(self, other: object) -> bool | NoReturn:
return ( return (
+6 -3
View File
@@ -191,13 +191,16 @@ class TopicHandler:
self.add_visinfo_handler(vi_handler) self.add_visinfo_handler(vi_handler)
return vi_handler return vi_handler
def add_visinfo_handler(self, new_viewpoint: VisualizationInfoHandler) -> None: def add_visinfo_handler(self, new_viewpoint: VisualizationInfoHandler) -> mdl.ViewPoint:
self.viewpoints[new_viewpoint.guid + ".bcfv"] = new_viewpoint self.viewpoints[new_viewpoint.guid + ".bcfv"] = new_viewpoint
if self.topic.viewpoints is None: if self.topic.viewpoints is None:
self.topic.viewpoints = mdl.TopicViewpoints() self.topic.viewpoints = mdl.TopicViewpoints()
self.topic.viewpoints.view_point.append( viewpoint = mdl.ViewPoint(
mdl.ViewPoint(viewpoint=new_viewpoint.guid + ".bcfv", guid=new_viewpoint.guid) viewpoint=new_viewpoint.guid + ".bcfv",
guid=new_viewpoint.guid,
) )
self.topic.viewpoints.view_point.append(viewpoint)
return viewpoint
def __eq__(self, other: object) -> bool | NoReturn: def __eq__(self, other: object) -> bool | NoReturn:
return ( return (