mirror of
https://github.com/PhotonVision/photonvision
synced 2026-06-26 01:51:40 +00:00
105 lines
4.3 KiB
Django/Jinja
105 lines
4.3 KiB
Django/Jinja
###############################################################################
|
|
## Copyright (C) Photon Vision.
|
|
###############################################################################
|
|
## This program is free software: you can redistribute it and/or modify
|
|
## it under the terms of the GNU General Public License as published by
|
|
## the Free Software Foundation, either version 3 of the License, or
|
|
## (at your option) any later version.
|
|
##
|
|
## This program is distributed in the hope that it will be useful,
|
|
## but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
## GNU General Public License for more details.
|
|
##
|
|
## You should have received a copy of the GNU General Public License
|
|
## along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
###############################################################################
|
|
|
|
###############################################################################
|
|
## THIS FILE WAS AUTO-GENERATED BY ./photon-serde/generate_messages.py.
|
|
## --> DO NOT MODIFY <--
|
|
###############################################################################
|
|
|
|
from typing import TYPE_CHECKING
|
|
|
|
from ..packet import Packet
|
|
from ..targeting import * # noqa
|
|
|
|
if TYPE_CHECKING:
|
|
{%- set ns = namespace(types=[]) -%}
|
|
{%- for field in fields|unique(attribute="type")-%}
|
|
{%- set _ = ns.types.append(field.type) -%}
|
|
{%- endfor -%}
|
|
{% set _ = ns.types.append(name) -%}
|
|
{%- for type in ns.types|sort%}
|
|
{%- if not type | is_shimmed and not type | is_intrinsic %}
|
|
from ..targeting import {{ type }} # noqa
|
|
{%- endif %}
|
|
{%- endfor%}
|
|
|
|
|
|
class {{ name }}Serde:
|
|
# Message definition md5sum. See photon_packet.adoc for details
|
|
MESSAGE_VERSION = "{{ message_hash }}"
|
|
MESSAGE_FORMAT = "{{ message_fmt }}"
|
|
|
|
@staticmethod
|
|
def pack(value: '{{ name }}' ) -> 'Packet':
|
|
ret = Packet()
|
|
{% for field in fields -%}
|
|
{%- if field.type | is_shimmed %}
|
|
ret.{{ get_message_by_name(field.type).python_encode_shim}}(value.{{ field.name }})
|
|
{%- elif field.optional == True %}
|
|
# {{ field.name }} is optional! it better not be a VLA too
|
|
ret.encodeOptional(value.{{ field.name }}, {{ field.type }}.photonStruct)
|
|
{%- elif field.vla == True and not field.type | is_intrinsic %}
|
|
# {{ field.name }} is a custom VLA!
|
|
ret.encodeList(value.{{ field.name }}, {{ field.type }}.photonStruct)
|
|
{%- elif field.vla == True and field.type | is_intrinsic %}
|
|
# {{ field.name }} is a custom VLA!
|
|
ret.encode{{ type_map[field.type].java_type.title() }}List(value.{{ field.name }})
|
|
{%- elif field.type | is_intrinsic %}
|
|
# {{ field.name }} is of intrinsic type {{ field.type }}
|
|
ret.{{ type_map[field.type].java_encode_shim }}(value.{{field.name}})
|
|
{%- else %}
|
|
# {{ field.name }} is of non-intrinsic type {{ field.type }}
|
|
ret.encodeBytes({{ field.type }}.photonStruct.pack(value.{{field.name}}).getData())
|
|
{%- endif %}
|
|
{%- if not loop.last %}
|
|
{% endif -%}
|
|
{% endfor%}
|
|
return ret
|
|
|
|
|
|
@staticmethod
|
|
def unpack(packet: 'Packet') -> '{{ name }}':
|
|
ret = {{ name }}()
|
|
{% for field in fields -%}
|
|
{%- if field.type | is_shimmed %}
|
|
ret.{{ field.name }} = {{ get_message_by_name(field.type).python_decode_shim }}()
|
|
{%- elif field.optional == True %}
|
|
# {{ field.name }} is optional! it better not be a VLA too
|
|
ret.{{ field.name }} = packet.decodeOptional({{ field.type }}.photonStruct)
|
|
{%- elif field.vla == True and not field.type | is_intrinsic %}
|
|
# {{ field.name }} is a custom VLA!
|
|
ret.{{ field.name }} = packet.decodeList({{ field.type }}.photonStruct)
|
|
{%- elif field.vla == True and field.type | is_intrinsic %}
|
|
# {{ field.name }} is a custom VLA!
|
|
ret.{{ field.name }} = packet.decode{{ type_map[field.type].java_type.title() }}List()
|
|
{%- elif field.type | is_intrinsic %}
|
|
# {{ field.name }} is of intrinsic type {{ field.type }}
|
|
ret.{{field.name}} = packet.{{ type_map[field.type].java_decode_method }}()
|
|
{%- else %}
|
|
# {{ field.name }} is of non-intrinsic type {{ field.type }}
|
|
ret.{{field.name}} = {{ field.type }}.photonStruct.unpack(packet)
|
|
{%- endif %}
|
|
{%- if not loop.last %}
|
|
{% endif -%}
|
|
{% endfor%}
|
|
|
|
return ret
|
|
|
|
|
|
# Hack ourselves into the base class
|
|
{{ name }}.photonStruct = {{ name }}Serde()
|