From aa9c561e9d246beb0ba934c036142f3557a9c088 Mon Sep 17 00:00:00 2001 From: Ivan Malison Date: Sat, 10 Jan 2026 07:42:04 -0800 Subject: [PATCH] style: Apply black formatting to protobuf generated files Co-Authored-By: Claude Opus 4.5 --- rbproto/python/shot_pb2.py | 63 +++++++++--------- rbproto/python/shot_pb2.pyi | 126 +++++++++++++++++++++++++++++++----- 2 files changed, 142 insertions(+), 47 deletions(-) diff --git a/rbproto/python/shot_pb2.py b/rbproto/python/shot_pb2.py index 1d6fbd2..a216881 100644 --- a/rbproto/python/shot_pb2.py +++ b/rbproto/python/shot_pb2.py @@ -6,44 +6,45 @@ from google.protobuf import descriptor as _descriptor from google.protobuf import descriptor_pool as _descriptor_pool from google.protobuf import symbol_database as _symbol_database from google.protobuf.internal import builder as _builder + # @@protoc_insertion_point(imports) _sym_db = _symbol_database.Default() - - -DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\nshot.proto\x12\x0fserialized_shot\"?\n\x03\x42ox\x12\x0c\n\x04left\x18\x01 \x01(\x02\x12\x0b\n\x03top\x18\x02 \x01(\x02\x12\r\n\x05width\x18\x03 \x01(\x02\x12\x0e\n\x06height\x18\x04 \x01(\x02\"\x1d\n\x05Point\x12\t\n\x01x\x18\x01 \x01(\x02\x12\t\n\x01y\x18\x02 \x01(\x02\"\x7f\n\rBallDetection\x12.\n\x0eplane_position\x18\x01 \x01(\x0b\x32\x16.serialized_shot.Point\x12(\n\nannotation\x18\x02 \x01(\x0b\x32\x14.serialized_shot.Box\x12\x14\n\x0cinterpolated\x18\x03 \x01(\x08\"T\n\x10RLEBallDetection\x12\x31\n\tdetection\x18\x01 \x01(\x0b\x32\x1e.serialized_shot.BallDetection\x12\r\n\x05\x63ount\x18\x02 \x01(\r\"L\n\x13RLEDetectionHistory\x12\x35\n\ndetections\x18\x01 \x03(\x0b\x32!.serialized_shot.RLEBallDetection\"F\n\x10\x44\x65tectionHistory\x12\x32\n\ndetections\x18\x01 \x03(\x0b\x32\x1e.serialized_shot.BallDetection\"\xfc\x01\n\rCollisionInfo\x12\x0e\n\x06source\x18\x01 \x01(\r\x12M\n\x10\x62\x61ll_identifiers\x18\x02 \x03(\x0b\x32\x33.serialized_shot.CollisionInfo.BallIdentifiersEntry\x12\x17\n\x0fwall_identifier\x18\x03 \x01(\r\x12\x13\n\x0b\x66rame_index\x18\x04 \x01(\r\x12\x0e\n\x06static\x18\x05 \x01(\x08\x1aN\n\x14\x42\x61llIdentifiersEntry\x12\x0b\n\x03key\x18\x01 \x01(\r\x12%\n\x05value\x18\x02 \x01(\x0b\x32\x16.serialized_shot.Point:\x02\x38\x01\"\xcc\x02\n\x04Path\x12\x13\n\x0bstart_frame\x18\x01 \x01(\r\x12\x11\n\tend_frame\x18\x02 \x01(\r\x12\x37\n\ndetections\x18\x03 \x01(\x0b\x32!.serialized_shot.DetectionHistoryH\x00\x12>\n\x0erle_detections\x18\x04 \x01(\x0b\x32$.serialized_shot.RLEDetectionHistoryH\x00\x12\x15\n\x0bnot_present\x18\x05 \x01(\x08H\x00\x12\x11\n\tis_static\x18\x06 \x01(\x08\x12\x32\n\nstart_info\x18\x07 \x01(\x0b\x32\x1e.serialized_shot.CollisionInfo\x12\x30\n\x08\x65nd_info\x18\x08 \x01(\x0b\x32\x1e.serialized_shot.CollisionInfoB\x13\n\x11\x64\x65tection_history\"R\n\x11IdentifierHistory\x12\x17\n\x0f\x62\x61ll_identifier\x18\x01 \x01(\r\x12$\n\x05paths\x18\x02 \x03(\x0b\x32\x15.serialized_shot.Path\"\xf4\x01\n\x12KeyBallIdentifiers\x12\x10\n\x08\x63ue_ball\x18\x01 \x01(\r\x12\x13\n\x0bobject_ball\x18\x02 \x01(\r\x12\x13\n\x0btarget_ball\x18\x03 \x01(\r\x12\x18\n\x10\x63ontact_sequence\x18\x04 \x03(\r\x12\x18\n\x0b\x63ue_ball_id\x18\x05 \x01(\rH\x00\x88\x01\x01\x12\x1b\n\x0eobject_ball_id\x18\x06 \x01(\rH\x01\x88\x01\x01\x12\x1b\n\x0etarget_ball_id\x18\x07 \x01(\rH\x02\x88\x01\x01\x42\x0e\n\x0c_cue_ball_idB\x11\n\x0f_object_ball_idB\x11\n\x0f_target_ball_id\"\xa8\x01\n\x04Shot\x12@\n\x14identifier_histories\x18\x03 \x03(\x0b\x32\".serialized_shot.IdentifierHistory\x12\x36\n\tkey_balls\x18\x04 \x01(\x0b\x32#.serialized_shot.KeyBallIdentifiers\x12\x13\n\x0bstart_index\x18\x05 \x01(\r\x12\x11\n\tend_index\x18\x06 \x01(\rb\x06proto3') +DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile( + b'\n\nshot.proto\x12\x0fserialized_shot"?\n\x03\x42ox\x12\x0c\n\x04left\x18\x01 \x01(\x02\x12\x0b\n\x03top\x18\x02 \x01(\x02\x12\r\n\x05width\x18\x03 \x01(\x02\x12\x0e\n\x06height\x18\x04 \x01(\x02"\x1d\n\x05Point\x12\t\n\x01x\x18\x01 \x01(\x02\x12\t\n\x01y\x18\x02 \x01(\x02"\x7f\n\rBallDetection\x12.\n\x0eplane_position\x18\x01 \x01(\x0b\x32\x16.serialized_shot.Point\x12(\n\nannotation\x18\x02 \x01(\x0b\x32\x14.serialized_shot.Box\x12\x14\n\x0cinterpolated\x18\x03 \x01(\x08"T\n\x10RLEBallDetection\x12\x31\n\tdetection\x18\x01 \x01(\x0b\x32\x1e.serialized_shot.BallDetection\x12\r\n\x05\x63ount\x18\x02 \x01(\r"L\n\x13RLEDetectionHistory\x12\x35\n\ndetections\x18\x01 \x03(\x0b\x32!.serialized_shot.RLEBallDetection"F\n\x10\x44\x65tectionHistory\x12\x32\n\ndetections\x18\x01 \x03(\x0b\x32\x1e.serialized_shot.BallDetection"\xfc\x01\n\rCollisionInfo\x12\x0e\n\x06source\x18\x01 \x01(\r\x12M\n\x10\x62\x61ll_identifiers\x18\x02 \x03(\x0b\x32\x33.serialized_shot.CollisionInfo.BallIdentifiersEntry\x12\x17\n\x0fwall_identifier\x18\x03 \x01(\r\x12\x13\n\x0b\x66rame_index\x18\x04 \x01(\r\x12\x0e\n\x06static\x18\x05 \x01(\x08\x1aN\n\x14\x42\x61llIdentifiersEntry\x12\x0b\n\x03key\x18\x01 \x01(\r\x12%\n\x05value\x18\x02 \x01(\x0b\x32\x16.serialized_shot.Point:\x02\x38\x01"\xcc\x02\n\x04Path\x12\x13\n\x0bstart_frame\x18\x01 \x01(\r\x12\x11\n\tend_frame\x18\x02 \x01(\r\x12\x37\n\ndetections\x18\x03 \x01(\x0b\x32!.serialized_shot.DetectionHistoryH\x00\x12>\n\x0erle_detections\x18\x04 \x01(\x0b\x32$.serialized_shot.RLEDetectionHistoryH\x00\x12\x15\n\x0bnot_present\x18\x05 \x01(\x08H\x00\x12\x11\n\tis_static\x18\x06 \x01(\x08\x12\x32\n\nstart_info\x18\x07 \x01(\x0b\x32\x1e.serialized_shot.CollisionInfo\x12\x30\n\x08\x65nd_info\x18\x08 \x01(\x0b\x32\x1e.serialized_shot.CollisionInfoB\x13\n\x11\x64\x65tection_history"R\n\x11IdentifierHistory\x12\x17\n\x0f\x62\x61ll_identifier\x18\x01 \x01(\r\x12$\n\x05paths\x18\x02 \x03(\x0b\x32\x15.serialized_shot.Path"\xf4\x01\n\x12KeyBallIdentifiers\x12\x10\n\x08\x63ue_ball\x18\x01 \x01(\r\x12\x13\n\x0bobject_ball\x18\x02 \x01(\r\x12\x13\n\x0btarget_ball\x18\x03 \x01(\r\x12\x18\n\x10\x63ontact_sequence\x18\x04 \x03(\r\x12\x18\n\x0b\x63ue_ball_id\x18\x05 \x01(\rH\x00\x88\x01\x01\x12\x1b\n\x0eobject_ball_id\x18\x06 \x01(\rH\x01\x88\x01\x01\x12\x1b\n\x0etarget_ball_id\x18\x07 \x01(\rH\x02\x88\x01\x01\x42\x0e\n\x0c_cue_ball_idB\x11\n\x0f_object_ball_idB\x11\n\x0f_target_ball_id"\xa8\x01\n\x04Shot\x12@\n\x14identifier_histories\x18\x03 \x03(\x0b\x32".serialized_shot.IdentifierHistory\x12\x36\n\tkey_balls\x18\x04 \x01(\x0b\x32#.serialized_shot.KeyBallIdentifiers\x12\x13\n\x0bstart_index\x18\x05 \x01(\r\x12\x11\n\tend_index\x18\x06 \x01(\rb\x06proto3' +) _globals = globals() _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals) -_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'shot_pb2', _globals) +_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, "shot_pb2", _globals) if _descriptor._USE_C_DESCRIPTORS == False: - DESCRIPTOR._options = None - _globals['_COLLISIONINFO_BALLIDENTIFIERSENTRY']._options = None - _globals['_COLLISIONINFO_BALLIDENTIFIERSENTRY']._serialized_options = b'8\001' - _globals['_BOX']._serialized_start=31 - _globals['_BOX']._serialized_end=94 - _globals['_POINT']._serialized_start=96 - _globals['_POINT']._serialized_end=125 - _globals['_BALLDETECTION']._serialized_start=127 - _globals['_BALLDETECTION']._serialized_end=254 - _globals['_RLEBALLDETECTION']._serialized_start=256 - _globals['_RLEBALLDETECTION']._serialized_end=340 - _globals['_RLEDETECTIONHISTORY']._serialized_start=342 - _globals['_RLEDETECTIONHISTORY']._serialized_end=418 - _globals['_DETECTIONHISTORY']._serialized_start=420 - _globals['_DETECTIONHISTORY']._serialized_end=490 - _globals['_COLLISIONINFO']._serialized_start=493 - _globals['_COLLISIONINFO']._serialized_end=745 - _globals['_COLLISIONINFO_BALLIDENTIFIERSENTRY']._serialized_start=667 - _globals['_COLLISIONINFO_BALLIDENTIFIERSENTRY']._serialized_end=745 - _globals['_PATH']._serialized_start=748 - _globals['_PATH']._serialized_end=1080 - _globals['_IDENTIFIERHISTORY']._serialized_start=1082 - _globals['_IDENTIFIERHISTORY']._serialized_end=1164 - _globals['_KEYBALLIDENTIFIERS']._serialized_start=1167 - _globals['_KEYBALLIDENTIFIERS']._serialized_end=1411 - _globals['_SHOT']._serialized_start=1414 - _globals['_SHOT']._serialized_end=1582 + DESCRIPTOR._options = None + _globals["_COLLISIONINFO_BALLIDENTIFIERSENTRY"]._options = None + _globals["_COLLISIONINFO_BALLIDENTIFIERSENTRY"]._serialized_options = b"8\001" + _globals["_BOX"]._serialized_start = 31 + _globals["_BOX"]._serialized_end = 94 + _globals["_POINT"]._serialized_start = 96 + _globals["_POINT"]._serialized_end = 125 + _globals["_BALLDETECTION"]._serialized_start = 127 + _globals["_BALLDETECTION"]._serialized_end = 254 + _globals["_RLEBALLDETECTION"]._serialized_start = 256 + _globals["_RLEBALLDETECTION"]._serialized_end = 340 + _globals["_RLEDETECTIONHISTORY"]._serialized_start = 342 + _globals["_RLEDETECTIONHISTORY"]._serialized_end = 418 + _globals["_DETECTIONHISTORY"]._serialized_start = 420 + _globals["_DETECTIONHISTORY"]._serialized_end = 490 + _globals["_COLLISIONINFO"]._serialized_start = 493 + _globals["_COLLISIONINFO"]._serialized_end = 745 + _globals["_COLLISIONINFO_BALLIDENTIFIERSENTRY"]._serialized_start = 667 + _globals["_COLLISIONINFO_BALLIDENTIFIERSENTRY"]._serialized_end = 745 + _globals["_PATH"]._serialized_start = 748 + _globals["_PATH"]._serialized_end = 1080 + _globals["_IDENTIFIERHISTORY"]._serialized_start = 1082 + _globals["_IDENTIFIERHISTORY"]._serialized_end = 1164 + _globals["_KEYBALLIDENTIFIERS"]._serialized_start = 1167 + _globals["_KEYBALLIDENTIFIERS"]._serialized_end = 1411 + _globals["_SHOT"]._serialized_start = 1414 + _globals["_SHOT"]._serialized_end = 1582 # @@protoc_insertion_point(module_scope) diff --git a/rbproto/python/shot_pb2.pyi b/rbproto/python/shot_pb2.pyi index b2a4f31..b5ec083 100644 --- a/rbproto/python/shot_pb2.pyi +++ b/rbproto/python/shot_pb2.pyi @@ -1,7 +1,13 @@ from google.protobuf.internal import containers as _containers from google.protobuf import descriptor as _descriptor from google.protobuf import message as _message -from typing import ClassVar as _ClassVar, Iterable as _Iterable, Mapping as _Mapping, Optional as _Optional, Union as _Union +from typing import ( + ClassVar as _ClassVar, + Iterable as _Iterable, + Mapping as _Mapping, + Optional as _Optional, + Union as _Union, +) DESCRIPTOR: _descriptor.FileDescriptor @@ -15,7 +21,13 @@ class Box(_message.Message): top: float width: float height: float - def __init__(self, left: _Optional[float] = ..., top: _Optional[float] = ..., width: _Optional[float] = ..., height: _Optional[float] = ...) -> None: ... + def __init__( + self, + left: _Optional[float] = ..., + top: _Optional[float] = ..., + width: _Optional[float] = ..., + height: _Optional[float] = ..., + ) -> None: ... class Point(_message.Message): __slots__ = ["x", "y"] @@ -23,7 +35,9 @@ class Point(_message.Message): Y_FIELD_NUMBER: _ClassVar[int] x: float y: float - def __init__(self, x: _Optional[float] = ..., y: _Optional[float] = ...) -> None: ... + def __init__( + self, x: _Optional[float] = ..., y: _Optional[float] = ... + ) -> None: ... class BallDetection(_message.Message): __slots__ = ["plane_position", "annotation", "interpolated"] @@ -33,7 +47,12 @@ class BallDetection(_message.Message): plane_position: Point annotation: Box interpolated: bool - def __init__(self, plane_position: _Optional[_Union[Point, _Mapping]] = ..., annotation: _Optional[_Union[Box, _Mapping]] = ..., interpolated: bool = ...) -> None: ... + def __init__( + self, + plane_position: _Optional[_Union[Point, _Mapping]] = ..., + annotation: _Optional[_Union[Box, _Mapping]] = ..., + interpolated: bool = ..., + ) -> None: ... class RLEBallDetection(_message.Message): __slots__ = ["detection", "count"] @@ -41,29 +60,49 @@ class RLEBallDetection(_message.Message): COUNT_FIELD_NUMBER: _ClassVar[int] detection: BallDetection count: int - def __init__(self, detection: _Optional[_Union[BallDetection, _Mapping]] = ..., count: _Optional[int] = ...) -> None: ... + def __init__( + self, + detection: _Optional[_Union[BallDetection, _Mapping]] = ..., + count: _Optional[int] = ..., + ) -> None: ... class RLEDetectionHistory(_message.Message): __slots__ = ["detections"] DETECTIONS_FIELD_NUMBER: _ClassVar[int] detections: _containers.RepeatedCompositeFieldContainer[RLEBallDetection] - def __init__(self, detections: _Optional[_Iterable[_Union[RLEBallDetection, _Mapping]]] = ...) -> None: ... + def __init__( + self, detections: _Optional[_Iterable[_Union[RLEBallDetection, _Mapping]]] = ... + ) -> None: ... class DetectionHistory(_message.Message): __slots__ = ["detections"] DETECTIONS_FIELD_NUMBER: _ClassVar[int] detections: _containers.RepeatedCompositeFieldContainer[BallDetection] - def __init__(self, detections: _Optional[_Iterable[_Union[BallDetection, _Mapping]]] = ...) -> None: ... + def __init__( + self, detections: _Optional[_Iterable[_Union[BallDetection, _Mapping]]] = ... + ) -> None: ... class CollisionInfo(_message.Message): - __slots__ = ["source", "ball_identifiers", "wall_identifier", "frame_index", "static"] + __slots__ = [ + "source", + "ball_identifiers", + "wall_identifier", + "frame_index", + "static", + ] + class BallIdentifiersEntry(_message.Message): __slots__ = ["key", "value"] KEY_FIELD_NUMBER: _ClassVar[int] VALUE_FIELD_NUMBER: _ClassVar[int] key: int value: Point - def __init__(self, key: _Optional[int] = ..., value: _Optional[_Union[Point, _Mapping]] = ...) -> None: ... + def __init__( + self, + key: _Optional[int] = ..., + value: _Optional[_Union[Point, _Mapping]] = ..., + ) -> None: ... + SOURCE_FIELD_NUMBER: _ClassVar[int] BALL_IDENTIFIERS_FIELD_NUMBER: _ClassVar[int] WALL_IDENTIFIER_FIELD_NUMBER: _ClassVar[int] @@ -74,10 +113,26 @@ class CollisionInfo(_message.Message): wall_identifier: int frame_index: int static: bool - def __init__(self, source: _Optional[int] = ..., ball_identifiers: _Optional[_Mapping[int, Point]] = ..., wall_identifier: _Optional[int] = ..., frame_index: _Optional[int] = ..., static: bool = ...) -> None: ... + def __init__( + self, + source: _Optional[int] = ..., + ball_identifiers: _Optional[_Mapping[int, Point]] = ..., + wall_identifier: _Optional[int] = ..., + frame_index: _Optional[int] = ..., + static: bool = ..., + ) -> None: ... class Path(_message.Message): - __slots__ = ["start_frame", "end_frame", "detections", "rle_detections", "not_present", "is_static", "start_info", "end_info"] + __slots__ = [ + "start_frame", + "end_frame", + "detections", + "rle_detections", + "not_present", + "is_static", + "start_info", + "end_info", + ] START_FRAME_FIELD_NUMBER: _ClassVar[int] END_FRAME_FIELD_NUMBER: _ClassVar[int] DETECTIONS_FIELD_NUMBER: _ClassVar[int] @@ -94,7 +149,17 @@ class Path(_message.Message): is_static: bool start_info: CollisionInfo end_info: CollisionInfo - def __init__(self, start_frame: _Optional[int] = ..., end_frame: _Optional[int] = ..., detections: _Optional[_Union[DetectionHistory, _Mapping]] = ..., rle_detections: _Optional[_Union[RLEDetectionHistory, _Mapping]] = ..., not_present: bool = ..., is_static: bool = ..., start_info: _Optional[_Union[CollisionInfo, _Mapping]] = ..., end_info: _Optional[_Union[CollisionInfo, _Mapping]] = ...) -> None: ... + def __init__( + self, + start_frame: _Optional[int] = ..., + end_frame: _Optional[int] = ..., + detections: _Optional[_Union[DetectionHistory, _Mapping]] = ..., + rle_detections: _Optional[_Union[RLEDetectionHistory, _Mapping]] = ..., + not_present: bool = ..., + is_static: bool = ..., + start_info: _Optional[_Union[CollisionInfo, _Mapping]] = ..., + end_info: _Optional[_Union[CollisionInfo, _Mapping]] = ..., + ) -> None: ... class IdentifierHistory(_message.Message): __slots__ = ["ball_identifier", "paths"] @@ -102,10 +167,22 @@ class IdentifierHistory(_message.Message): PATHS_FIELD_NUMBER: _ClassVar[int] ball_identifier: int paths: _containers.RepeatedCompositeFieldContainer[Path] - def __init__(self, ball_identifier: _Optional[int] = ..., paths: _Optional[_Iterable[_Union[Path, _Mapping]]] = ...) -> None: ... + def __init__( + self, + ball_identifier: _Optional[int] = ..., + paths: _Optional[_Iterable[_Union[Path, _Mapping]]] = ..., + ) -> None: ... class KeyBallIdentifiers(_message.Message): - __slots__ = ["cue_ball", "object_ball", "target_ball", "contact_sequence", "cue_ball_id", "object_ball_id", "target_ball_id"] + __slots__ = [ + "cue_ball", + "object_ball", + "target_ball", + "contact_sequence", + "cue_ball_id", + "object_ball_id", + "target_ball_id", + ] CUE_BALL_FIELD_NUMBER: _ClassVar[int] OBJECT_BALL_FIELD_NUMBER: _ClassVar[int] TARGET_BALL_FIELD_NUMBER: _ClassVar[int] @@ -120,7 +197,16 @@ class KeyBallIdentifiers(_message.Message): cue_ball_id: int object_ball_id: int target_ball_id: int - def __init__(self, cue_ball: _Optional[int] = ..., object_ball: _Optional[int] = ..., target_ball: _Optional[int] = ..., contact_sequence: _Optional[_Iterable[int]] = ..., cue_ball_id: _Optional[int] = ..., object_ball_id: _Optional[int] = ..., target_ball_id: _Optional[int] = ...) -> None: ... + def __init__( + self, + cue_ball: _Optional[int] = ..., + object_ball: _Optional[int] = ..., + target_ball: _Optional[int] = ..., + contact_sequence: _Optional[_Iterable[int]] = ..., + cue_ball_id: _Optional[int] = ..., + object_ball_id: _Optional[int] = ..., + target_ball_id: _Optional[int] = ..., + ) -> None: ... class Shot(_message.Message): __slots__ = ["identifier_histories", "key_balls", "start_index", "end_index"] @@ -132,4 +218,12 @@ class Shot(_message.Message): key_balls: KeyBallIdentifiers start_index: int end_index: int - def __init__(self, identifier_histories: _Optional[_Iterable[_Union[IdentifierHistory, _Mapping]]] = ..., key_balls: _Optional[_Union[KeyBallIdentifiers, _Mapping]] = ..., start_index: _Optional[int] = ..., end_index: _Optional[int] = ...) -> None: ... + def __init__( + self, + identifier_histories: _Optional[ + _Iterable[_Union[IdentifierHistory, _Mapping]] + ] = ..., + key_balls: _Optional[_Union[KeyBallIdentifiers, _Mapping]] = ..., + start_index: _Optional[int] = ..., + end_index: _Optional[int] = ..., + ) -> None: ...