mirror of
https://github.com/aurora-opensource/xviz.git
synced 2026-01-25 14:16:45 +00:00
3cd43ae3af
* Fix Protobuf writer to add PBE1 prefix to file * Update protobuf & fix directory writer * Fix FutureInstance builder and add tests
30 lines
791 B
Python
30 lines
791 B
Python
from xviz_avs.builder.base_builder import XVIZBaseBuilder, CATEGORY, PRIMITIVE_TYPES
|
|
from xviz_avs.v2.core_pb2 import Link
|
|
|
|
class XVIZLinkBuilder(XVIZBaseBuilder):
|
|
def __init__(self, metadata, logger=None):
|
|
super().__init__(None, metadata, logger)
|
|
self._links = None
|
|
self._target_stream = None
|
|
|
|
def parent(self, target_stream):
|
|
self._target_stream = target_stream
|
|
|
|
def _flush(self):
|
|
if not self._links:
|
|
self._links = {}
|
|
|
|
data = Link()
|
|
if self._target_stream:
|
|
data.target_pose = self._target_stream
|
|
self._links[self._stream_id] = data
|
|
|
|
def reset(self):
|
|
super().reset()
|
|
|
|
def get_data(self):
|
|
if self._stream_id:
|
|
self._flush()
|
|
|
|
return self._links
|