luckyrobots 0.1.68__py3-none-any.whl → 0.1.70__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (57) hide show
  1. luckyrobots/__init__.py +23 -12
  2. luckyrobots/client.py +800 -0
  3. luckyrobots/config/robots.yaml +231 -71
  4. luckyrobots/engine/__init__.py +23 -0
  5. luckyrobots/{utils → engine}/check_updates.py +108 -48
  6. luckyrobots/{utils → engine}/download.py +61 -39
  7. luckyrobots/engine/manager.py +427 -0
  8. luckyrobots/grpc/__init__.py +6 -0
  9. luckyrobots/grpc/generated/__init__.py +18 -0
  10. luckyrobots/grpc/generated/agent_pb2.py +61 -0
  11. luckyrobots/grpc/generated/agent_pb2_grpc.py +255 -0
  12. luckyrobots/grpc/generated/camera_pb2.py +45 -0
  13. luckyrobots/grpc/generated/camera_pb2_grpc.py +155 -0
  14. luckyrobots/grpc/generated/common_pb2.py +39 -0
  15. luckyrobots/grpc/generated/common_pb2_grpc.py +27 -0
  16. luckyrobots/grpc/generated/media_pb2.py +35 -0
  17. luckyrobots/grpc/generated/media_pb2_grpc.py +27 -0
  18. luckyrobots/grpc/generated/mujoco_pb2.py +47 -0
  19. luckyrobots/grpc/generated/mujoco_pb2_grpc.py +248 -0
  20. luckyrobots/grpc/generated/scene_pb2.py +54 -0
  21. luckyrobots/grpc/generated/scene_pb2_grpc.py +248 -0
  22. luckyrobots/grpc/generated/telemetry_pb2.py +43 -0
  23. luckyrobots/grpc/generated/telemetry_pb2_grpc.py +154 -0
  24. luckyrobots/grpc/generated/viewport_pb2.py +48 -0
  25. luckyrobots/grpc/generated/viewport_pb2_grpc.py +155 -0
  26. luckyrobots/grpc/proto/agent.proto +152 -0
  27. luckyrobots/grpc/proto/camera.proto +41 -0
  28. luckyrobots/grpc/proto/common.proto +36 -0
  29. luckyrobots/grpc/proto/hazel_rpc.proto +32 -0
  30. luckyrobots/grpc/proto/media.proto +26 -0
  31. luckyrobots/grpc/proto/mujoco.proto +64 -0
  32. luckyrobots/grpc/proto/scene.proto +70 -0
  33. luckyrobots/grpc/proto/telemetry.proto +43 -0
  34. luckyrobots/grpc/proto/viewport.proto +45 -0
  35. luckyrobots/luckyrobots.py +212 -0
  36. luckyrobots/models/__init__.py +13 -0
  37. luckyrobots/models/camera.py +97 -0
  38. luckyrobots/models/observation.py +135 -0
  39. luckyrobots/{utils/helpers.py → utils.py} +75 -40
  40. luckyrobots-0.1.70.dist-info/METADATA +262 -0
  41. luckyrobots-0.1.70.dist-info/RECORD +44 -0
  42. {luckyrobots-0.1.68.dist-info → luckyrobots-0.1.70.dist-info}/WHEEL +1 -1
  43. luckyrobots/core/luckyrobots.py +0 -628
  44. luckyrobots/core/manager.py +0 -236
  45. luckyrobots/core/models.py +0 -68
  46. luckyrobots/core/node.py +0 -273
  47. luckyrobots/message/__init__.py +0 -18
  48. luckyrobots/message/pubsub.py +0 -145
  49. luckyrobots/message/srv/client.py +0 -81
  50. luckyrobots/message/srv/service.py +0 -135
  51. luckyrobots/message/srv/types.py +0 -83
  52. luckyrobots/message/transporter.py +0 -427
  53. luckyrobots/utils/event_loop.py +0 -94
  54. luckyrobots/utils/sim_manager.py +0 -413
  55. luckyrobots-0.1.68.dist-info/METADATA +0 -253
  56. luckyrobots-0.1.68.dist-info/RECORD +0 -24
  57. {luckyrobots-0.1.68.dist-info → luckyrobots-0.1.70.dist-info}/licenses/LICENSE +0 -0
@@ -0,0 +1,248 @@
1
+ # Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
2
+ """Client and server classes corresponding to protobuf-defined services."""
3
+ import grpc
4
+ import warnings
5
+
6
+ from . import mujoco_pb2 as mujoco__pb2
7
+
8
+ GRPC_GENERATED_VERSION = "1.76.0"
9
+ GRPC_VERSION = grpc.__version__
10
+ _version_not_supported = False
11
+
12
+ try:
13
+ from grpc._utilities import first_version_is_lower
14
+
15
+ _version_not_supported = first_version_is_lower(
16
+ GRPC_VERSION, GRPC_GENERATED_VERSION
17
+ )
18
+ except ImportError:
19
+ _version_not_supported = True
20
+
21
+ if _version_not_supported:
22
+ raise RuntimeError(
23
+ f"The grpc package installed is at version {GRPC_VERSION},"
24
+ + " but the generated code in mujoco_pb2_grpc.py depends on"
25
+ + f" grpcio>={GRPC_GENERATED_VERSION}."
26
+ + f" Please upgrade your grpc module to grpcio>={GRPC_GENERATED_VERSION}"
27
+ + f" or downgrade your generated code using grpcio-tools<={GRPC_VERSION}."
28
+ )
29
+
30
+
31
+ class MujocoServiceStub(object):
32
+ """MuJoCo state + control IO."""
33
+
34
+ def __init__(self, channel):
35
+ """Constructor.
36
+
37
+ Args:
38
+ channel: A grpc.Channel.
39
+ """
40
+ self.GetJointState = channel.unary_unary(
41
+ "/hazel.rpc.v1.MujocoService/GetJointState",
42
+ request_serializer=mujoco__pb2.GetJointStateRequest.SerializeToString,
43
+ response_deserializer=mujoco__pb2.GetJointStateResponse.FromString,
44
+ _registered_method=True,
45
+ )
46
+ self.SendControl = channel.unary_unary(
47
+ "/hazel.rpc.v1.MujocoService/SendControl",
48
+ request_serializer=mujoco__pb2.SendControlRequest.SerializeToString,
49
+ response_deserializer=mujoco__pb2.SendControlResponse.FromString,
50
+ _registered_method=True,
51
+ )
52
+ self.GetMujocoInfo = channel.unary_unary(
53
+ "/hazel.rpc.v1.MujocoService/GetMujocoInfo",
54
+ request_serializer=mujoco__pb2.GetMujocoInfoRequest.SerializeToString,
55
+ response_deserializer=mujoco__pb2.GetMujocoInfoResponse.FromString,
56
+ _registered_method=True,
57
+ )
58
+ self.StreamJointState = channel.unary_stream(
59
+ "/hazel.rpc.v1.MujocoService/StreamJointState",
60
+ request_serializer=mujoco__pb2.GetJointStateRequest.SerializeToString,
61
+ response_deserializer=mujoco__pb2.GetJointStateResponse.FromString,
62
+ _registered_method=True,
63
+ )
64
+
65
+
66
+ class MujocoServiceServicer(object):
67
+ """MuJoCo state + control IO."""
68
+
69
+ def GetJointState(self, request, context):
70
+ """Missing associated documentation comment in .proto file."""
71
+ context.set_code(grpc.StatusCode.UNIMPLEMENTED)
72
+ context.set_details("Method not implemented!")
73
+ raise NotImplementedError("Method not implemented!")
74
+
75
+ def SendControl(self, request, context):
76
+ """Missing associated documentation comment in .proto file."""
77
+ context.set_code(grpc.StatusCode.UNIMPLEMENTED)
78
+ context.set_details("Method not implemented!")
79
+ raise NotImplementedError("Method not implemented!")
80
+
81
+ def GetMujocoInfo(self, request, context):
82
+ """Missing associated documentation comment in .proto file."""
83
+ context.set_code(grpc.StatusCode.UNIMPLEMENTED)
84
+ context.set_details("Method not implemented!")
85
+ raise NotImplementedError("Method not implemented!")
86
+
87
+ def StreamJointState(self, request, context):
88
+ """Streaming variant for continuous state observation"""
89
+ context.set_code(grpc.StatusCode.UNIMPLEMENTED)
90
+ context.set_details("Method not implemented!")
91
+ raise NotImplementedError("Method not implemented!")
92
+
93
+
94
+ def add_MujocoServiceServicer_to_server(servicer, server):
95
+ rpc_method_handlers = {
96
+ "GetJointState": grpc.unary_unary_rpc_method_handler(
97
+ servicer.GetJointState,
98
+ request_deserializer=mujoco__pb2.GetJointStateRequest.FromString,
99
+ response_serializer=mujoco__pb2.GetJointStateResponse.SerializeToString,
100
+ ),
101
+ "SendControl": grpc.unary_unary_rpc_method_handler(
102
+ servicer.SendControl,
103
+ request_deserializer=mujoco__pb2.SendControlRequest.FromString,
104
+ response_serializer=mujoco__pb2.SendControlResponse.SerializeToString,
105
+ ),
106
+ "GetMujocoInfo": grpc.unary_unary_rpc_method_handler(
107
+ servicer.GetMujocoInfo,
108
+ request_deserializer=mujoco__pb2.GetMujocoInfoRequest.FromString,
109
+ response_serializer=mujoco__pb2.GetMujocoInfoResponse.SerializeToString,
110
+ ),
111
+ "StreamJointState": grpc.unary_stream_rpc_method_handler(
112
+ servicer.StreamJointState,
113
+ request_deserializer=mujoco__pb2.GetJointStateRequest.FromString,
114
+ response_serializer=mujoco__pb2.GetJointStateResponse.SerializeToString,
115
+ ),
116
+ }
117
+ generic_handler = grpc.method_handlers_generic_handler(
118
+ "hazel.rpc.v1.MujocoService", rpc_method_handlers
119
+ )
120
+ server.add_generic_rpc_handlers((generic_handler,))
121
+ server.add_registered_method_handlers(
122
+ "hazel.rpc.v1.MujocoService", rpc_method_handlers
123
+ )
124
+
125
+
126
+ # This class is part of an EXPERIMENTAL API.
127
+ class MujocoService(object):
128
+ """MuJoCo state + control IO."""
129
+
130
+ @staticmethod
131
+ def GetJointState(
132
+ request,
133
+ target,
134
+ options=(),
135
+ channel_credentials=None,
136
+ call_credentials=None,
137
+ insecure=False,
138
+ compression=None,
139
+ wait_for_ready=None,
140
+ timeout=None,
141
+ metadata=None,
142
+ ):
143
+ return grpc.experimental.unary_unary(
144
+ request,
145
+ target,
146
+ "/hazel.rpc.v1.MujocoService/GetJointState",
147
+ mujoco__pb2.GetJointStateRequest.SerializeToString,
148
+ mujoco__pb2.GetJointStateResponse.FromString,
149
+ options,
150
+ channel_credentials,
151
+ insecure,
152
+ call_credentials,
153
+ compression,
154
+ wait_for_ready,
155
+ timeout,
156
+ metadata,
157
+ _registered_method=True,
158
+ )
159
+
160
+ @staticmethod
161
+ def SendControl(
162
+ request,
163
+ target,
164
+ options=(),
165
+ channel_credentials=None,
166
+ call_credentials=None,
167
+ insecure=False,
168
+ compression=None,
169
+ wait_for_ready=None,
170
+ timeout=None,
171
+ metadata=None,
172
+ ):
173
+ return grpc.experimental.unary_unary(
174
+ request,
175
+ target,
176
+ "/hazel.rpc.v1.MujocoService/SendControl",
177
+ mujoco__pb2.SendControlRequest.SerializeToString,
178
+ mujoco__pb2.SendControlResponse.FromString,
179
+ options,
180
+ channel_credentials,
181
+ insecure,
182
+ call_credentials,
183
+ compression,
184
+ wait_for_ready,
185
+ timeout,
186
+ metadata,
187
+ _registered_method=True,
188
+ )
189
+
190
+ @staticmethod
191
+ def GetMujocoInfo(
192
+ request,
193
+ target,
194
+ options=(),
195
+ channel_credentials=None,
196
+ call_credentials=None,
197
+ insecure=False,
198
+ compression=None,
199
+ wait_for_ready=None,
200
+ timeout=None,
201
+ metadata=None,
202
+ ):
203
+ return grpc.experimental.unary_unary(
204
+ request,
205
+ target,
206
+ "/hazel.rpc.v1.MujocoService/GetMujocoInfo",
207
+ mujoco__pb2.GetMujocoInfoRequest.SerializeToString,
208
+ mujoco__pb2.GetMujocoInfoResponse.FromString,
209
+ options,
210
+ channel_credentials,
211
+ insecure,
212
+ call_credentials,
213
+ compression,
214
+ wait_for_ready,
215
+ timeout,
216
+ metadata,
217
+ _registered_method=True,
218
+ )
219
+
220
+ @staticmethod
221
+ def StreamJointState(
222
+ request,
223
+ target,
224
+ options=(),
225
+ channel_credentials=None,
226
+ call_credentials=None,
227
+ insecure=False,
228
+ compression=None,
229
+ wait_for_ready=None,
230
+ timeout=None,
231
+ metadata=None,
232
+ ):
233
+ return grpc.experimental.unary_stream(
234
+ request,
235
+ target,
236
+ "/hazel.rpc.v1.MujocoService/StreamJointState",
237
+ mujoco__pb2.GetJointStateRequest.SerializeToString,
238
+ mujoco__pb2.GetJointStateResponse.FromString,
239
+ options,
240
+ channel_credentials,
241
+ insecure,
242
+ call_credentials,
243
+ compression,
244
+ wait_for_ready,
245
+ timeout,
246
+ metadata,
247
+ _registered_method=True,
248
+ )
@@ -0,0 +1,54 @@
1
+ # -*- coding: utf-8 -*-
2
+ # Generated by the protocol buffer compiler. DO NOT EDIT!
3
+ # NO CHECKED-IN PROTOBUF GENCODE
4
+ # source: scene.proto
5
+ # Protobuf Python Version: 6.31.1
6
+ """Generated protocol buffer code."""
7
+ from google.protobuf import descriptor as _descriptor
8
+ from google.protobuf import descriptor_pool as _descriptor_pool
9
+ from google.protobuf import runtime_version as _runtime_version
10
+ from google.protobuf import symbol_database as _symbol_database
11
+ from google.protobuf.internal import builder as _builder
12
+
13
+ _runtime_version.ValidateProtobufRuntimeVersion(
14
+ _runtime_version.Domain.PUBLIC, 6, 31, 1, "", "scene.proto"
15
+ )
16
+ # @@protoc_insertion_point(imports)
17
+
18
+ _sym_db = _symbol_database.Default()
19
+
20
+
21
+ from . import common_pb2 as common__pb2
22
+
23
+
24
+ DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
25
+ b'\n\x0bscene.proto\x12\x0chazel.rpc.v1\x1a\x0c\x63ommon.proto"~\n\nEntityInfo\x12"\n\x02id\x18\x01 \x01(\x0b\x32\x16.hazel.rpc.v1.EntityId\x12\x0c\n\x04name\x18\x02 \x01(\t\x12*\n\ttransform\x18\x03 \x01(\x0b\x32\x17.hazel.rpc.v1.Transform\x12\x12\n\ncomponents\x18\x04 \x03(\t"\x15\n\x13GetSceneInfoRequest"T\n\x14GetSceneInfoResponse\x12\x12\n\nscene_name\x18\x01 \x01(\t\x12\x12\n\nscene_path\x18\x02 \x01(\t\x12\x14\n\x0c\x65ntity_count\x18\x03 \x01(\r"M\n\x13ListEntitiesRequest\x12\x1a\n\x12include_transforms\x18\x01 \x01(\x08\x12\x1a\n\x12include_components\x18\x02 \x01(\x08"B\n\x14ListEntitiesResponse\x12*\n\x08\x65ntities\x18\x01 \x03(\x0b\x32\x18.hazel.rpc.v1.EntityInfo"V\n\x10GetEntityRequest\x12$\n\x02id\x18\x01 \x01(\x0b\x32\x16.hazel.rpc.v1.EntityIdH\x00\x12\x0e\n\x04name\x18\x02 \x01(\tH\x00\x42\x0c\n\nidentifier"L\n\x11GetEntityResponse\x12\r\n\x05\x66ound\x18\x01 \x01(\x08\x12(\n\x06\x65ntity\x18\x02 \x01(\x0b\x32\x18.hazel.rpc.v1.EntityInfo"k\n\x19SetEntityTransformRequest\x12"\n\x02id\x18\x01 \x01(\x0b\x32\x16.hazel.rpc.v1.EntityId\x12*\n\ttransform\x18\x02 \x01(\x0b\x32\x17.hazel.rpc.v1.Transform">\n\x1aSetEntityTransformResponse\x12\x0f\n\x07success\x18\x01 \x01(\x08\x12\x0f\n\x07message\x18\x02 \x01(\t2\xf3\x02\n\x0cSceneService\x12U\n\x0cGetSceneInfo\x12!.hazel.rpc.v1.GetSceneInfoRequest\x1a".hazel.rpc.v1.GetSceneInfoResponse\x12U\n\x0cListEntities\x12!.hazel.rpc.v1.ListEntitiesRequest\x1a".hazel.rpc.v1.ListEntitiesResponse\x12L\n\tGetEntity\x12\x1e.hazel.rpc.v1.GetEntityRequest\x1a\x1f.hazel.rpc.v1.GetEntityResponse\x12g\n\x12SetEntityTransform\x12\'.hazel.rpc.v1.SetEntityTransformRequest\x1a(.hazel.rpc.v1.SetEntityTransformResponseB\x03\xf8\x01\x01\x62\x06proto3'
26
+ )
27
+
28
+ _globals = globals()
29
+ _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
30
+ _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, "scene_pb2", _globals)
31
+ if not _descriptor._USE_C_DESCRIPTORS:
32
+ _globals["DESCRIPTOR"]._loaded_options = None
33
+ _globals["DESCRIPTOR"]._serialized_options = b"\370\001\001"
34
+ _globals["_ENTITYINFO"]._serialized_start = 43
35
+ _globals["_ENTITYINFO"]._serialized_end = 169
36
+ _globals["_GETSCENEINFOREQUEST"]._serialized_start = 171
37
+ _globals["_GETSCENEINFOREQUEST"]._serialized_end = 192
38
+ _globals["_GETSCENEINFORESPONSE"]._serialized_start = 194
39
+ _globals["_GETSCENEINFORESPONSE"]._serialized_end = 278
40
+ _globals["_LISTENTITIESREQUEST"]._serialized_start = 280
41
+ _globals["_LISTENTITIESREQUEST"]._serialized_end = 357
42
+ _globals["_LISTENTITIESRESPONSE"]._serialized_start = 359
43
+ _globals["_LISTENTITIESRESPONSE"]._serialized_end = 425
44
+ _globals["_GETENTITYREQUEST"]._serialized_start = 427
45
+ _globals["_GETENTITYREQUEST"]._serialized_end = 513
46
+ _globals["_GETENTITYRESPONSE"]._serialized_start = 515
47
+ _globals["_GETENTITYRESPONSE"]._serialized_end = 591
48
+ _globals["_SETENTITYTRANSFORMREQUEST"]._serialized_start = 593
49
+ _globals["_SETENTITYTRANSFORMREQUEST"]._serialized_end = 700
50
+ _globals["_SETENTITYTRANSFORMRESPONSE"]._serialized_start = 702
51
+ _globals["_SETENTITYTRANSFORMRESPONSE"]._serialized_end = 764
52
+ _globals["_SCENESERVICE"]._serialized_start = 767
53
+ _globals["_SCENESERVICE"]._serialized_end = 1138
54
+ # @@protoc_insertion_point(module_scope)
@@ -0,0 +1,248 @@
1
+ # Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
2
+ """Client and server classes corresponding to protobuf-defined services."""
3
+ import grpc
4
+ import warnings
5
+
6
+ from . import scene_pb2 as scene__pb2
7
+
8
+ GRPC_GENERATED_VERSION = "1.76.0"
9
+ GRPC_VERSION = grpc.__version__
10
+ _version_not_supported = False
11
+
12
+ try:
13
+ from grpc._utilities import first_version_is_lower
14
+
15
+ _version_not_supported = first_version_is_lower(
16
+ GRPC_VERSION, GRPC_GENERATED_VERSION
17
+ )
18
+ except ImportError:
19
+ _version_not_supported = True
20
+
21
+ if _version_not_supported:
22
+ raise RuntimeError(
23
+ f"The grpc package installed is at version {GRPC_VERSION},"
24
+ + " but the generated code in scene_pb2_grpc.py depends on"
25
+ + f" grpcio>={GRPC_GENERATED_VERSION}."
26
+ + f" Please upgrade your grpc module to grpcio>={GRPC_GENERATED_VERSION}"
27
+ + f" or downgrade your generated code using grpcio-tools<={GRPC_VERSION}."
28
+ )
29
+
30
+
31
+ class SceneServiceStub(object):
32
+ """Scene inspection + basic editing."""
33
+
34
+ def __init__(self, channel):
35
+ """Constructor.
36
+
37
+ Args:
38
+ channel: A grpc.Channel.
39
+ """
40
+ self.GetSceneInfo = channel.unary_unary(
41
+ "/hazel.rpc.v1.SceneService/GetSceneInfo",
42
+ request_serializer=scene__pb2.GetSceneInfoRequest.SerializeToString,
43
+ response_deserializer=scene__pb2.GetSceneInfoResponse.FromString,
44
+ _registered_method=True,
45
+ )
46
+ self.ListEntities = channel.unary_unary(
47
+ "/hazel.rpc.v1.SceneService/ListEntities",
48
+ request_serializer=scene__pb2.ListEntitiesRequest.SerializeToString,
49
+ response_deserializer=scene__pb2.ListEntitiesResponse.FromString,
50
+ _registered_method=True,
51
+ )
52
+ self.GetEntity = channel.unary_unary(
53
+ "/hazel.rpc.v1.SceneService/GetEntity",
54
+ request_serializer=scene__pb2.GetEntityRequest.SerializeToString,
55
+ response_deserializer=scene__pb2.GetEntityResponse.FromString,
56
+ _registered_method=True,
57
+ )
58
+ self.SetEntityTransform = channel.unary_unary(
59
+ "/hazel.rpc.v1.SceneService/SetEntityTransform",
60
+ request_serializer=scene__pb2.SetEntityTransformRequest.SerializeToString,
61
+ response_deserializer=scene__pb2.SetEntityTransformResponse.FromString,
62
+ _registered_method=True,
63
+ )
64
+
65
+
66
+ class SceneServiceServicer(object):
67
+ """Scene inspection + basic editing."""
68
+
69
+ def GetSceneInfo(self, request, context):
70
+ """Missing associated documentation comment in .proto file."""
71
+ context.set_code(grpc.StatusCode.UNIMPLEMENTED)
72
+ context.set_details("Method not implemented!")
73
+ raise NotImplementedError("Method not implemented!")
74
+
75
+ def ListEntities(self, request, context):
76
+ """Missing associated documentation comment in .proto file."""
77
+ context.set_code(grpc.StatusCode.UNIMPLEMENTED)
78
+ context.set_details("Method not implemented!")
79
+ raise NotImplementedError("Method not implemented!")
80
+
81
+ def GetEntity(self, request, context):
82
+ """Missing associated documentation comment in .proto file."""
83
+ context.set_code(grpc.StatusCode.UNIMPLEMENTED)
84
+ context.set_details("Method not implemented!")
85
+ raise NotImplementedError("Method not implemented!")
86
+
87
+ def SetEntityTransform(self, request, context):
88
+ """Missing associated documentation comment in .proto file."""
89
+ context.set_code(grpc.StatusCode.UNIMPLEMENTED)
90
+ context.set_details("Method not implemented!")
91
+ raise NotImplementedError("Method not implemented!")
92
+
93
+
94
+ def add_SceneServiceServicer_to_server(servicer, server):
95
+ rpc_method_handlers = {
96
+ "GetSceneInfo": grpc.unary_unary_rpc_method_handler(
97
+ servicer.GetSceneInfo,
98
+ request_deserializer=scene__pb2.GetSceneInfoRequest.FromString,
99
+ response_serializer=scene__pb2.GetSceneInfoResponse.SerializeToString,
100
+ ),
101
+ "ListEntities": grpc.unary_unary_rpc_method_handler(
102
+ servicer.ListEntities,
103
+ request_deserializer=scene__pb2.ListEntitiesRequest.FromString,
104
+ response_serializer=scene__pb2.ListEntitiesResponse.SerializeToString,
105
+ ),
106
+ "GetEntity": grpc.unary_unary_rpc_method_handler(
107
+ servicer.GetEntity,
108
+ request_deserializer=scene__pb2.GetEntityRequest.FromString,
109
+ response_serializer=scene__pb2.GetEntityResponse.SerializeToString,
110
+ ),
111
+ "SetEntityTransform": grpc.unary_unary_rpc_method_handler(
112
+ servicer.SetEntityTransform,
113
+ request_deserializer=scene__pb2.SetEntityTransformRequest.FromString,
114
+ response_serializer=scene__pb2.SetEntityTransformResponse.SerializeToString,
115
+ ),
116
+ }
117
+ generic_handler = grpc.method_handlers_generic_handler(
118
+ "hazel.rpc.v1.SceneService", rpc_method_handlers
119
+ )
120
+ server.add_generic_rpc_handlers((generic_handler,))
121
+ server.add_registered_method_handlers(
122
+ "hazel.rpc.v1.SceneService", rpc_method_handlers
123
+ )
124
+
125
+
126
+ # This class is part of an EXPERIMENTAL API.
127
+ class SceneService(object):
128
+ """Scene inspection + basic editing."""
129
+
130
+ @staticmethod
131
+ def GetSceneInfo(
132
+ request,
133
+ target,
134
+ options=(),
135
+ channel_credentials=None,
136
+ call_credentials=None,
137
+ insecure=False,
138
+ compression=None,
139
+ wait_for_ready=None,
140
+ timeout=None,
141
+ metadata=None,
142
+ ):
143
+ return grpc.experimental.unary_unary(
144
+ request,
145
+ target,
146
+ "/hazel.rpc.v1.SceneService/GetSceneInfo",
147
+ scene__pb2.GetSceneInfoRequest.SerializeToString,
148
+ scene__pb2.GetSceneInfoResponse.FromString,
149
+ options,
150
+ channel_credentials,
151
+ insecure,
152
+ call_credentials,
153
+ compression,
154
+ wait_for_ready,
155
+ timeout,
156
+ metadata,
157
+ _registered_method=True,
158
+ )
159
+
160
+ @staticmethod
161
+ def ListEntities(
162
+ request,
163
+ target,
164
+ options=(),
165
+ channel_credentials=None,
166
+ call_credentials=None,
167
+ insecure=False,
168
+ compression=None,
169
+ wait_for_ready=None,
170
+ timeout=None,
171
+ metadata=None,
172
+ ):
173
+ return grpc.experimental.unary_unary(
174
+ request,
175
+ target,
176
+ "/hazel.rpc.v1.SceneService/ListEntities",
177
+ scene__pb2.ListEntitiesRequest.SerializeToString,
178
+ scene__pb2.ListEntitiesResponse.FromString,
179
+ options,
180
+ channel_credentials,
181
+ insecure,
182
+ call_credentials,
183
+ compression,
184
+ wait_for_ready,
185
+ timeout,
186
+ metadata,
187
+ _registered_method=True,
188
+ )
189
+
190
+ @staticmethod
191
+ def GetEntity(
192
+ request,
193
+ target,
194
+ options=(),
195
+ channel_credentials=None,
196
+ call_credentials=None,
197
+ insecure=False,
198
+ compression=None,
199
+ wait_for_ready=None,
200
+ timeout=None,
201
+ metadata=None,
202
+ ):
203
+ return grpc.experimental.unary_unary(
204
+ request,
205
+ target,
206
+ "/hazel.rpc.v1.SceneService/GetEntity",
207
+ scene__pb2.GetEntityRequest.SerializeToString,
208
+ scene__pb2.GetEntityResponse.FromString,
209
+ options,
210
+ channel_credentials,
211
+ insecure,
212
+ call_credentials,
213
+ compression,
214
+ wait_for_ready,
215
+ timeout,
216
+ metadata,
217
+ _registered_method=True,
218
+ )
219
+
220
+ @staticmethod
221
+ def SetEntityTransform(
222
+ request,
223
+ target,
224
+ options=(),
225
+ channel_credentials=None,
226
+ call_credentials=None,
227
+ insecure=False,
228
+ compression=None,
229
+ wait_for_ready=None,
230
+ timeout=None,
231
+ metadata=None,
232
+ ):
233
+ return grpc.experimental.unary_unary(
234
+ request,
235
+ target,
236
+ "/hazel.rpc.v1.SceneService/SetEntityTransform",
237
+ scene__pb2.SetEntityTransformRequest.SerializeToString,
238
+ scene__pb2.SetEntityTransformResponse.FromString,
239
+ options,
240
+ channel_credentials,
241
+ insecure,
242
+ call_credentials,
243
+ compression,
244
+ wait_for_ready,
245
+ timeout,
246
+ metadata,
247
+ _registered_method=True,
248
+ )
@@ -0,0 +1,43 @@
1
+ # -*- coding: utf-8 -*-
2
+ # Generated by the protocol buffer compiler. DO NOT EDIT!
3
+ # NO CHECKED-IN PROTOBUF GENCODE
4
+ # source: telemetry.proto
5
+ # Protobuf Python Version: 6.31.1
6
+ """Generated protocol buffer code."""
7
+ from google.protobuf import descriptor as _descriptor
8
+ from google.protobuf import descriptor_pool as _descriptor_pool
9
+ from google.protobuf import runtime_version as _runtime_version
10
+ from google.protobuf import symbol_database as _symbol_database
11
+ from google.protobuf.internal import builder as _builder
12
+
13
+ _runtime_version.ValidateProtobufRuntimeVersion(
14
+ _runtime_version.Domain.PUBLIC, 6, 31, 1, "", "telemetry.proto"
15
+ )
16
+ # @@protoc_insertion_point(imports)
17
+
18
+ _sym_db = _symbol_database.Default()
19
+
20
+
21
+ DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
22
+ b'\n\x0ftelemetry.proto\x12\x0chazel.rpc.v1"Z\n\x0fTelemetrySchema\x12\x19\n\x11observation_names\x18\x01 \x03(\t\x12\x14\n\x0c\x61\x63tion_names\x18\x02 \x03(\t\x12\n\n\x02nq\x18\x03 \x01(\r\x12\n\n\x02nu\x18\x04 \x01(\r"\x1b\n\x19GetTelemetrySchemaRequest"K\n\x1aGetTelemetrySchemaResponse\x12-\n\x06schema\x18\x01 \x01(\x0b\x32\x1d.hazel.rpc.v1.TelemetrySchema",\n\x16StreamTelemetryRequest\x12\x12\n\ntarget_fps\x18\x01 \x01(\r"\x7f\n\x0eTelemetryFrame\x12\x14\n\x0ctimestamp_ms\x18\x01 \x01(\x04\x12\x14\n\x0c\x66rame_number\x18\x02 \x01(\r\x12\x12\n\ntask_index\x18\x03 \x01(\r\x12\x18\n\x10observation_qpos\x18\x04 \x03(\x02\x12\x13\n\x0b\x61\x63tion_ctrl\x18\x05 \x03(\x02\x32\xd4\x01\n\x10TelemetryService\x12g\n\x12GetTelemetrySchema\x12\'.hazel.rpc.v1.GetTelemetrySchemaRequest\x1a(.hazel.rpc.v1.GetTelemetrySchemaResponse\x12W\n\x0fStreamTelemetry\x12$.hazel.rpc.v1.StreamTelemetryRequest\x1a\x1c.hazel.rpc.v1.TelemetryFrame0\x01\x42\x03\xf8\x01\x01\x62\x06proto3'
23
+ )
24
+
25
+ _globals = globals()
26
+ _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
27
+ _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, "telemetry_pb2", _globals)
28
+ if not _descriptor._USE_C_DESCRIPTORS:
29
+ _globals["DESCRIPTOR"]._loaded_options = None
30
+ _globals["DESCRIPTOR"]._serialized_options = b"\370\001\001"
31
+ _globals["_TELEMETRYSCHEMA"]._serialized_start = 33
32
+ _globals["_TELEMETRYSCHEMA"]._serialized_end = 123
33
+ _globals["_GETTELEMETRYSCHEMAREQUEST"]._serialized_start = 125
34
+ _globals["_GETTELEMETRYSCHEMAREQUEST"]._serialized_end = 152
35
+ _globals["_GETTELEMETRYSCHEMARESPONSE"]._serialized_start = 154
36
+ _globals["_GETTELEMETRYSCHEMARESPONSE"]._serialized_end = 229
37
+ _globals["_STREAMTELEMETRYREQUEST"]._serialized_start = 231
38
+ _globals["_STREAMTELEMETRYREQUEST"]._serialized_end = 275
39
+ _globals["_TELEMETRYFRAME"]._serialized_start = 277
40
+ _globals["_TELEMETRYFRAME"]._serialized_end = 404
41
+ _globals["_TELEMETRYSERVICE"]._serialized_start = 407
42
+ _globals["_TELEMETRYSERVICE"]._serialized_end = 619
43
+ # @@protoc_insertion_point(module_scope)