feat: add disparity messages

This commit is contained in:
2024-01-22 19:19:41 +01:00
parent 0b0eba0c40
commit 7123d890b6
4 changed files with 352 additions and 214 deletions

View File

@ -14,7 +14,7 @@ _sym_db = _symbol_database.Default()
from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x13\x65vents/events.proto\x12\x0erobocar.events\x1a\x1fgoogle/protobuf/timestamp.proto\"T\n\x08\x46rameRef\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\n\n\x02id\x18\x02 \x01(\t\x12.\n\ncreated_at\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\"C\n\x0c\x46rameMessage\x12$\n\x02id\x18\x01 \x01(\x0b\x32\x18.robocar.events.FrameRef\x12\r\n\x05\x66rame\x18\x02 \x01(\x0c\"d\n\x0fSteeringMessage\x12\x10\n\x08steering\x18\x01 \x01(\x02\x12\x12\n\nconfidence\x18\x02 \x01(\x02\x12+\n\tframe_ref\x18\x03 \x01(\x0b\x32\x18.robocar.events.FrameRef\"d\n\x0fThrottleMessage\x12\x10\n\x08throttle\x18\x01 \x01(\x02\x12\x12\n\nconfidence\x18\x02 \x01(\x02\x12+\n\tframe_ref\x18\x03 \x01(\x0b\x32\x18.robocar.events.FrameRef\"\x82\x01\n\x10SpeedZoneMessage\x12-\n\nspeed_zone\x18\x01 \x01(\x0e\x32\x19.robocar.events.SpeedZone\x12\x12\n\nconfidence\x18\x02 \x01(\x02\x12+\n\tframe_ref\x18\x03 \x01(\x0b\x32\x18.robocar.events.FrameRef\"A\n\x10\x44riveModeMessage\x12-\n\ndrive_mode\x18\x01 \x01(\x0e\x32\x19.robocar.events.DriveMode\"f\n\x0eObjectsMessage\x12\'\n\x07objects\x18\x01 \x03(\x0b\x32\x16.robocar.events.Object\x12+\n\tframe_ref\x18\x02 \x01(\x0b\x32\x18.robocar.events.FrameRef\"\x80\x01\n\x06Object\x12(\n\x04type\x18\x01 \x01(\x0e\x32\x1a.robocar.events.TypeObject\x12\x0c\n\x04left\x18\x02 \x01(\x02\x12\x0b\n\x03top\x18\x03 \x01(\x02\x12\r\n\x05right\x18\x04 \x01(\x02\x12\x0e\n\x06\x62ottom\x18\x05 \x01(\x02\x12\x12\n\nconfidence\x18\x06 \x01(\x02\"&\n\x13SwitchRecordMessage\x12\x0f\n\x07\x65nabled\x18\x01 \x01(\x08\"\x8c\x01\n\x0bRoadMessage\x12&\n\x07\x63ontour\x18\x01 \x03(\x0b\x32\x15.robocar.events.Point\x12(\n\x07\x65llipse\x18\x02 \x01(\x0b\x32\x17.robocar.events.Ellipse\x12+\n\tframe_ref\x18\x03 \x01(\x0b\x32\x18.robocar.events.FrameRef\"\x1d\n\x05Point\x12\t\n\x01x\x18\x01 \x01(\x05\x12\t\n\x01y\x18\x02 \x01(\x05\"r\n\x07\x45llipse\x12%\n\x06\x63\x65nter\x18\x01 \x01(\x0b\x32\x15.robocar.events.Point\x12\r\n\x05width\x18\x02 \x01(\x05\x12\x0e\n\x06height\x18\x03 \x01(\x05\x12\r\n\x05\x61ngle\x18\x04 \x01(\x02\x12\x12\n\nconfidence\x18\x05 \x01(\x02\"\xf5\x01\n\rRecordMessage\x12+\n\x05\x66rame\x18\x01 \x01(\x0b\x32\x1c.robocar.events.FrameMessage\x12\x31\n\x08steering\x18\x02 \x01(\x0b\x32\x1f.robocar.events.SteeringMessage\x12;\n\x12\x61utopilot_steering\x18\x04 \x01(\x0b\x32\x1f.robocar.events.SteeringMessage\x12\x34\n\ndrive_mode\x18\x05 \x01(\x0b\x32 .robocar.events.DriveModeMessage\x12\x11\n\trecordSet\x18\x03 \x01(\t*8\n\tSpeedZone\x12\x0b\n\x07UNKNOWN\x10\x00\x12\x08\n\x04SLOW\x10\x01\x12\n\n\x06NORMAL\x10\x02\x12\x08\n\x04\x46\x41ST\x10\x03*:\n\tDriveMode\x12\x0b\n\x07INVALID\x10\x00\x12\x08\n\x04USER\x10\x01\x12\t\n\x05PILOT\x10\x02\x12\x0b\n\x07\x43OPILOT\x10\x03*2\n\nTypeObject\x12\x07\n\x03\x41NY\x10\x00\x12\x07\n\x03\x43\x41R\x10\x01\x12\x08\n\x04\x42UMP\x10\x02\x12\x08\n\x04PLOT\x10\x03\x42\nZ\x08./eventsb\x06proto3')
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x13\x65vents/events.proto\x12\x0erobocar.events\x1a\x1fgoogle/protobuf/timestamp.proto\"T\n\x08\x46rameRef\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\n\n\x02id\x18\x02 \x01(\t\x12.\n\ncreated_at\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\"C\n\x0c\x46rameMessage\x12$\n\x02id\x18\x01 \x01(\x0b\x32\x18.robocar.events.FrameRef\x12\r\n\x05\x66rame\x18\x02 \x01(\x0c\"\x8a\x01\n\x10\x44isparityMessage\x12+\n\tframe_ref\x18\x01 \x01(\x0b\x32\x18.robocar.events.FrameRef\x12\x11\n\tdisparity\x18\x02 \x01(\x0c\x12\x1e\n\x16\x66ocal_length_in_pixels\x18\x03 \x01(\x01\x12\x16\n\x0e\x62\x61seline_in_mm\x18\x04 \x01(\x01\"d\n\x0fSteeringMessage\x12\x10\n\x08steering\x18\x01 \x01(\x02\x12\x12\n\nconfidence\x18\x02 \x01(\x02\x12+\n\tframe_ref\x18\x03 \x01(\x0b\x32\x18.robocar.events.FrameRef\"d\n\x0fThrottleMessage\x12\x10\n\x08throttle\x18\x01 \x01(\x02\x12\x12\n\nconfidence\x18\x02 \x01(\x02\x12+\n\tframe_ref\x18\x03 \x01(\x0b\x32\x18.robocar.events.FrameRef\"\x82\x01\n\x10SpeedZoneMessage\x12-\n\nspeed_zone\x18\x01 \x01(\x0e\x32\x19.robocar.events.SpeedZone\x12\x12\n\nconfidence\x18\x02 \x01(\x02\x12+\n\tframe_ref\x18\x03 \x01(\x0b\x32\x18.robocar.events.FrameRef\"A\n\x10\x44riveModeMessage\x12-\n\ndrive_mode\x18\x01 \x01(\x0e\x32\x19.robocar.events.DriveMode\"f\n\x0eObjectsMessage\x12\'\n\x07objects\x18\x01 \x03(\x0b\x32\x16.robocar.events.Object\x12+\n\tframe_ref\x18\x02 \x01(\x0b\x32\x18.robocar.events.FrameRef\"\x80\x01\n\x06Object\x12(\n\x04type\x18\x01 \x01(\x0e\x32\x1a.robocar.events.TypeObject\x12\x0c\n\x04left\x18\x02 \x01(\x02\x12\x0b\n\x03top\x18\x03 \x01(\x02\x12\r\n\x05right\x18\x04 \x01(\x02\x12\x0e\n\x06\x62ottom\x18\x05 \x01(\x02\x12\x12\n\nconfidence\x18\x06 \x01(\x02\"&\n\x13SwitchRecordMessage\x12\x0f\n\x07\x65nabled\x18\x01 \x01(\x08\"\x8c\x01\n\x0bRoadMessage\x12&\n\x07\x63ontour\x18\x01 \x03(\x0b\x32\x15.robocar.events.Point\x12(\n\x07\x65llipse\x18\x02 \x01(\x0b\x32\x17.robocar.events.Ellipse\x12+\n\tframe_ref\x18\x03 \x01(\x0b\x32\x18.robocar.events.FrameRef\"\x1d\n\x05Point\x12\t\n\x01x\x18\x01 \x01(\x05\x12\t\n\x01y\x18\x02 \x01(\x05\"r\n\x07\x45llipse\x12%\n\x06\x63\x65nter\x18\x01 \x01(\x0b\x32\x15.robocar.events.Point\x12\r\n\x05width\x18\x02 \x01(\x05\x12\x0e\n\x06height\x18\x03 \x01(\x05\x12\r\n\x05\x61ngle\x18\x04 \x01(\x02\x12\x12\n\nconfidence\x18\x05 \x01(\x02\"\xf5\x01\n\rRecordMessage\x12+\n\x05\x66rame\x18\x01 \x01(\x0b\x32\x1c.robocar.events.FrameMessage\x12\x31\n\x08steering\x18\x02 \x01(\x0b\x32\x1f.robocar.events.SteeringMessage\x12;\n\x12\x61utopilot_steering\x18\x04 \x01(\x0b\x32\x1f.robocar.events.SteeringMessage\x12\x34\n\ndrive_mode\x18\x05 \x01(\x0b\x32 .robocar.events.DriveModeMessage\x12\x11\n\trecordSet\x18\x03 \x01(\t*8\n\tSpeedZone\x12\x0b\n\x07UNKNOWN\x10\x00\x12\x08\n\x04SLOW\x10\x01\x12\n\n\x06NORMAL\x10\x02\x12\x08\n\x04\x46\x41ST\x10\x03*:\n\tDriveMode\x12\x0b\n\x07INVALID\x10\x00\x12\x08\n\x04USER\x10\x01\x12\t\n\x05PILOT\x10\x02\x12\x0b\n\x07\x43OPILOT\x10\x03*2\n\nTypeObject\x12\x07\n\x03\x41NY\x10\x00\x12\x07\n\x03\x43\x41R\x10\x01\x12\x08\n\x04\x42UMP\x10\x02\x12\x08\n\x04PLOT\x10\x03\x42\nZ\x08./eventsb\x06proto3')
_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())
_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'events.events_pb2', globals())
@ -22,36 +22,38 @@ if _descriptor._USE_C_DESCRIPTORS == False:
DESCRIPTOR._options = None
DESCRIPTOR._serialized_options = b'Z\010./events'
_SPEEDZONE._serialized_start=1444
_SPEEDZONE._serialized_end=1500
_DRIVEMODE._serialized_start=1502
_DRIVEMODE._serialized_end=1560
_TYPEOBJECT._serialized_start=1562
_TYPEOBJECT._serialized_end=1612
_SPEEDZONE._serialized_start=1585
_SPEEDZONE._serialized_end=1641
_DRIVEMODE._serialized_start=1643
_DRIVEMODE._serialized_end=1701
_TYPEOBJECT._serialized_start=1703
_TYPEOBJECT._serialized_end=1753
_FRAMEREF._serialized_start=72
_FRAMEREF._serialized_end=156
_FRAMEMESSAGE._serialized_start=158
_FRAMEMESSAGE._serialized_end=225
_STEERINGMESSAGE._serialized_start=227
_STEERINGMESSAGE._serialized_end=327
_THROTTLEMESSAGE._serialized_start=329
_THROTTLEMESSAGE._serialized_end=429
_SPEEDZONEMESSAGE._serialized_start=432
_SPEEDZONEMESSAGE._serialized_end=562
_DRIVEMODEMESSAGE._serialized_start=564
_DRIVEMODEMESSAGE._serialized_end=629
_OBJECTSMESSAGE._serialized_start=631
_OBJECTSMESSAGE._serialized_end=733
_OBJECT._serialized_start=736
_OBJECT._serialized_end=864
_SWITCHRECORDMESSAGE._serialized_start=866
_SWITCHRECORDMESSAGE._serialized_end=904
_ROADMESSAGE._serialized_start=907
_ROADMESSAGE._serialized_end=1047
_POINT._serialized_start=1049
_POINT._serialized_end=1078
_ELLIPSE._serialized_start=1080
_ELLIPSE._serialized_end=1194
_RECORDMESSAGE._serialized_start=1197
_RECORDMESSAGE._serialized_end=1442
_DISPARITYMESSAGE._serialized_start=228
_DISPARITYMESSAGE._serialized_end=366
_STEERINGMESSAGE._serialized_start=368
_STEERINGMESSAGE._serialized_end=468
_THROTTLEMESSAGE._serialized_start=470
_THROTTLEMESSAGE._serialized_end=570
_SPEEDZONEMESSAGE._serialized_start=573
_SPEEDZONEMESSAGE._serialized_end=703
_DRIVEMODEMESSAGE._serialized_start=705
_DRIVEMODEMESSAGE._serialized_end=770
_OBJECTSMESSAGE._serialized_start=772
_OBJECTSMESSAGE._serialized_end=874
_OBJECT._serialized_start=877
_OBJECT._serialized_end=1005
_SWITCHRECORDMESSAGE._serialized_start=1007
_SWITCHRECORDMESSAGE._serialized_end=1045
_ROADMESSAGE._serialized_start=1048
_ROADMESSAGE._serialized_end=1188
_POINT._serialized_start=1190
_POINT._serialized_end=1219
_ELLIPSE._serialized_start=1221
_ELLIPSE._serialized_end=1335
_RECORDMESSAGE._serialized_start=1338
_RECORDMESSAGE._serialized_end=1583
# @@protoc_insertion_point(module_scope)

View File

@ -47,12 +47,14 @@ class _DriveModeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._Enum
INVALID: _DriveMode.ValueType # 0
USER: _DriveMode.ValueType # 1
PILOT: _DriveMode.ValueType # 2
COPILOT: _DriveMode.ValueType # 3
class DriveMode(_DriveMode, metaclass=_DriveModeEnumTypeWrapper): ...
INVALID: DriveMode.ValueType # 0
USER: DriveMode.ValueType # 1
PILOT: DriveMode.ValueType # 2
COPILOT: DriveMode.ValueType # 3
global___DriveMode = DriveMode
class _TypeObject:
@ -117,6 +119,33 @@ class FrameMessage(google.protobuf.message.Message):
global___FrameMessage = FrameMessage
@typing_extensions.final
class DisparityMessage(google.protobuf.message.Message):
DESCRIPTOR: google.protobuf.descriptor.Descriptor
FRAME_REF_FIELD_NUMBER: builtins.int
DISPARITY_FIELD_NUMBER: builtins.int
FOCAL_LENGTH_IN_PIXELS_FIELD_NUMBER: builtins.int
BASELINE_IN_MM_FIELD_NUMBER: builtins.int
@property
def frame_ref(self) -> global___FrameRef: ...
disparity: builtins.bytes
focal_length_in_pixels: builtins.float
baseline_in_mm: builtins.float
"""the distance between two mono cameras"""
def __init__(
self,
*,
frame_ref: global___FrameRef | None = ...,
disparity: builtins.bytes = ...,
focal_length_in_pixels: builtins.float = ...,
baseline_in_mm: builtins.float = ...,
) -> None: ...
def HasField(self, field_name: typing_extensions.Literal["frame_ref", b"frame_ref"]) -> builtins.bool: ...
def ClearField(self, field_name: typing_extensions.Literal["baseline_in_mm", b"baseline_in_mm", "disparity", b"disparity", "focal_length_in_pixels", b"focal_length_in_pixels", "frame_ref", b"frame_ref"]) -> None: ...
global___DisparityMessage = DisparityMessage
@typing_extensions.final
class SteeringMessage(google.protobuf.message.Message):
DESCRIPTOR: google.protobuf.descriptor.Descriptor