Class StructFieldSerializerVisitor

python/pyfory/struct.py:873–973

class StructFieldSerializerVisitor(TypeVisitor):
    def __init__(
        self,
        type_resolver,
    ):
        self.type_resolver = type_resolver

    def visit_array(self, field_name, elem_type, carrier, types_path=None):
        type_id = _array_type_id(elem_type, carrier)
        from pyfory.serializer import (
            ForyArrayFieldSerializer,
            Numpy1DArraySerializer,
            PyArraySerializer,
            fory_array_wrapper_type,
            typecode_dict,
            typeid_code,
        )

        if carrier == "array":
            wrapper_type = fory_array_wrapper_type(type_id)
            return ForyArrayFieldSerializer(self.type_resolver, wrapper_type, type_id, field_name)
        if carrier == "pyarray":
            typecode = typeid_code.get(type_id)
            if typecode is None:
                raise TypeError(f"pyfory.PyArray does not support array type id {type_id}")
            _itemsize, ftype, _type_id = typecode_dict[typecode]
            return PyArraySerializer(self.type_resolver, ftype, type_id)
        if carrier == "ndarray":
            for dtype, (_itemsize, _format, ftype, dtype_type_id) in Numpy1DArraySerializer.dtypes_dict.items():
                if dtype_type_id == type_id:
                    if ENABLE_FORY_CYTHON_SERIALIZATION:
                        from pyfory.serialization import Numpy1DArraySerializer as Numpy1DArrayRuntimeSerializer

                        return Numpy1DArrayRuntimeSerializer(self.type_resolver, ftype, dtype)
                    return Numpy1DArraySerializer(self.type_resolver, ftype, dtype)
            raise TypeError(f"pyfory.NDArray does not support array type id {type_id}")
        raise TypeError(f"Unknown array carrier {carrier!r}")

    def visit_list(self, field_name, elem_type, types_path=None):
        from pyfory.serializer import ListSerializer  # Local import
        from pyfory.type_util import unwrap_ref

        # Infer type recursively for type such as List[Dict[str, str]]
        elem_type, elem_ref_override = unwrap_ref(elem_type)
        elem_serializer = infer_field("item", elem_type, self, types_path=types_path)
        return ListSerializer(self.type_resolver, list, elem_serializer, elem_ref_override)

    def visit_set(self, field_name, elem_type, types_path=None):
        from pyfory.serializer import SetSerializer  # Local import
        from pyfory.type_util import unwrap_ref

        # Infer type recursively for type such as Set[Dict[str, str]]
        elem_type, elem_ref_override = unwrap_ref(elem_type)
        elem_serializer = infer_field("item", elem_type, self, types_path=types_path)
        return SetSerializer(self.type_resolver, set, elem_serializer, elem_ref_override)

    def visit_tuple(self, field_name, elem_types, types_path=None):
        from pyfory.serializer import TupleSerializer  # Local import