Source code for nanome.api.ui.messages

from nanome._internal import serializer_fields as serializer_fields
from nanome._internal.enum_utils import IntEnum
from . import serializers


[docs]class ButtonCallback(serializer_fields.TypeSerializer):
[docs]    def version(self):
        return 2

[docs]    def name(self):
        return "ButtonCallback"

[docs]    def serialize(self, version, value, context):
        id = value[0]
        state = value[1]
        if (version == 0):
            plugin_mask = (context._plugin_id << 24) & 0x7FFFFFFF
            id |= plugin_mask
        context.write_int(id)
        if version >= 2:
            context.write_bool(state)

[docs]    def deserialize(self, version, context):
        content_id = context.read_int()
        if (version == 0):
            id_mask = 0x00FFFFFF
            content_id &= id_mask
        state = False
        if version >= 2:
            state = context.read_bool()
        return (content_id, state)


[docs]class DropdownCallback(serializer_fields.TypeSerializer):
[docs]    def version(self):
        return 0

[docs]    def name(self):
        return "DropdownCallback"

[docs]    def serialize(self, version, value, context):
        # value is a tuple containing the image ID and the item index.
        context.write_int(value[0])
        context.write_int(value[1])

[docs]    def deserialize(self, version, context):
        id = context.read_int()
        item_index = context.read_int()
        return id, item_index


[docs]class GetMenuTransform(serializer_fields.TypeSerializer):
[docs]    def version(self):
        return 0

[docs]    def name(self):
        return "GetMenuTransform"

[docs]    def serialize(self, version, value, context):
        context.write_byte(value)

[docs]    def deserialize(self, version, context):
        return None


[docs]class GetMenuTransformResponse(serializer_fields.TypeSerializer):

    def __init__(self):
        self.pos = serializer_fields.UnityPositionField()
        self.rot = serializer_fields.UnityRotationField()
        self.vec3 = serializer_fields.Vector3Field()

[docs]    def version(self):
        return 0

[docs]    def name(self):
        return "GetMenuTransformResponse"

[docs]    def serialize(self, version, value, context):
        pass

[docs]    def deserialize(self, version, context):
        menu_position = context.read_using_serializer(self.pos)
        menu_rotation = context.read_using_serializer(self.rot)
        menu_scale = context.read_using_serializer(self.vec3)

        result = (menu_position, menu_rotation, menu_scale)
        return result


[docs]class ImageCallback(serializer_fields.TypeSerializer):
[docs]    def version(self):
        return 1

[docs]    def name(self):
        return "ImageCallback"

[docs]    def serialize(self, version, value, context):
        # value is a tuple containing the image ID, the x coordinate and the y coordinate.
        if (version == 0):
            plugin_mask = (context._plugin_id << 24) & 0x7FFFFFFF
            value = (value[0] | plugin_mask, value[1], value[2])
        context.write_int(value[0])
        context.write_float(value[1])
        context.write_float(value[2])

[docs]    def deserialize(self, version, context):
        id = context.read_int()
        if (version == 0):
            id_mask = 0x00FFFFFF
            id &= id_mask
        x = context.read_float()
        y = context.read_float()
        return id, x, y


[docs]class MenuCallback(serializer_fields.TypeSerializer):
[docs]    def version(self):
        return 1

[docs]    def name(self):
        return "MenuCallback"

[docs]    def serialize(self, version, value, context):
        if version >= 1:
            context.write_byte(value[0])
        context.write_bool(value[1])

[docs]    def deserialize(self, version, context):
        if version >= 1:
            index = context.read_byte()
        else:
            index = 0
        value = context.read_bool()
        return (index, value)


[docs]class SetMenuTransform(serializer_fields.TypeSerializer):

    def __init__(self):
        self.pos = serializer_fields.UnityPositionField()
        self.rot = serializer_fields.UnityRotationField()
        self.vec3 = serializer_fields.Vector3Field()

[docs]    def version(self):
        return 0

[docs]    def name(self):
        return "SetMenuTransform"

[docs]    def serialize(self, version, value, data):
        data.write_byte(value[0])
        data.write_using_serializer(self.pos, value[1])
        data.write_using_serializer(self.rot, value[2])
        data.write_using_serializer(self.vec3, value[3])

[docs]    def deserialize(self, version, data):
        return None


[docs]class SliderCallback(serializer_fields.TypeSerializer):
[docs]    def version(self):
        return 1

[docs]    def name(self):
        return "SliderCallback"

[docs]    def serialize(self, version, value, context):
        if (version == 0):
            plugin_mask = (context._plugin_id << 24) & 0x7FFFFFFF
            value[0] |= plugin_mask
        context.write_int(value[0])
        context.write_float(value[1])

[docs]    def deserialize(self, version, context):
        content_id = context.read_int()
        if (version == 0):
            id_mask = 0x00FFFFFF
            content_id &= id_mask
        result = (content_id, context.read_float())
        return result


[docs]class TextInputCallback(serializer_fields.TypeSerializer):

    def __init__(self):
        self.__tuple = serializer_fields.TupleField(serializer_fields.IntField(), serializer_fields.StringField())

[docs]    def version(self):
        return 1

[docs]    def name(self):
        return "TextInputCallback"

[docs]    def serialize(self, version, value, context):
        if (version == 0):
            plugin_mask = (context._plugin_id << 24) & 0x7FFFFFFF
            value[0] |= plugin_mask
        context.write_using_serializer(self.__tuple, value)

[docs]    def deserialize(self, version, context):
        tup = context.read_using_serializer(self.__tuple)
        if (version == 0):
            id_mask = 0x00FFFFFF
            tup = (tup[0] & id_mask, tup[1])
        return tup


[docs]class UIHook(serializer_fields.TypeSerializer):
[docs]    class Type(IntEnum):
        button_hover = 0
        image_pressed = 1
        image_held = 2
        image_released = 3

[docs]    def version(self):
        return 0

[docs]    def name(self):
        return "UIHook"

[docs]    def serialize(self, version, value, context):
        context.write_byte(value[0])
        context.write_int(value[1])

[docs]    def deserialize(self, version, context):
        raise NotImplementedError


[docs]class UpdateContent(serializer_fields.TypeSerializer):

    def __init__(self):
        self._array = serializer_fields.ArrayField()
        self._content = serializers.UIBaseSerializer()
        self._array.set_type(self._content)

[docs]    def version(self):
        return 1

[docs]    def name(self):
        return "SendUIContent"

[docs]    def serialize(self, version, value, context):
        if version == 0:
            context.write_using_serializer(self._content, value[0])
        else:
            context.write_using_serializer(self._array, value)

[docs]    def deserialize(self, version, context):
        return None


[docs]class UpdateMenu(serializer_fields.TypeSerializer):

    def __init__(self):
        self.menu = serializers.MenuSerializer()
        self.array = serializer_fields.ArrayField()
        self.layout = serializers.LayoutNodeSerializer()
        self.content = serializers.UIBaseSerializer()

[docs]    def version(self):
        return 2

[docs]    def name(self):
        return "UpdateMenu"

[docs]    def serialize(self, version, value, context):
        (menu, shallow) = value
        if version >= 1:
            context.write_byte(menu.index)
        if version >= 2:
            context.write_bool(shallow)

        context.write_using_serializer(self.menu, menu)
        nodes = []
        content = []
        if not shallow:
            nodes = menu._get_all_nodes()
            content = menu._get_all_content()
        self.array.set_type(self.layout)
        context.write_using_serializer(self.array, nodes)
        self.array.set_type(self.content)
        context.write_using_serializer(self.array, content)

[docs]    def deserialize(self, version, context):
        return None


[docs]class UpdateNode(serializer_fields.TypeSerializer):

    def __init__(self):
        self._array = serializer_fields.ArrayField()
        self._node_serializer = serializers.LayoutNodeSerializerDeep()
        self._array.set_type(self._node_serializer)

[docs]    def version(self):
        return 1

[docs]    def name(self):
        return "SendLayoutNode"

[docs]    def serialize(self, version, value, context):
        if version == 0:
            context.write_using_serializer(self._node_serializer, value[0])
        else:
            context.write_using_serializer(self._array, value)

[docs]    def deserialize(self, version, context):
        return None


[docs]class SendNotification(serializer_fields.TypeSerializer):

    def __init__(self):
        self.string = serializer_fields.StringField()

[docs]    def version(self):
        return 0

[docs]    def name(self):
        return "SendNotification"

[docs]    def serialize(self, version, value, context):
        context.write_uint(value[0])
        context.write_using_serializer(self.string, value[1])

[docs]    def deserialize(self, version, context):
        raise NotImplementedError