o SiV@sZddlmZddlmZmZmZddlZddlm Z ddl m Z m Z m Z GdddZdS))partial)AnyDictOptionalN) Recording) asdict_nonullfastcopyifnonecseZdZdZdeeeefddfddZdededdffd d Z d edefd d Z deddffdd Z deeeffddZ d edefddZ d edejfddZd edefddZd efddZZS)CustomFieldMixinaX :class:`CustomFieldMixin` is intended for classes such as Cut or SupervisionSegment that support holding custom, user-defined fields. .. caution:: Due to the way inheritance and dataclasses work before Python 3.10, it is necessary to re-define ``custom`` attribute in dataclasses that inherit from this mixin. customreturnNcCs ||_dSNr )selfr rA/home/ubuntu/.local/lib/python3.10/site-packages/lhotse/custom.py__init__s zCustomFieldMixin.__init__keyvaluecsX||jvrt||dSt|ji}|dur||dn|||<|r*||_dSdS)a0 This magic function is called when the user tries to set an attribute. We use it as syntactic sugar to store custom attributes in ``self.custom`` field, so that they can be (de)serialized later. Setting a ``None`` value will remove the attribute from ``custom``. N)__dataclass_fields__super __setattr__r r pop)rrrr  __class__rrrs   zCustomFieldMixin.__setattr__namecCs^|j}|durtd|||vr|j|S|dr(|dd}t|j|Std|)a This magic function is called when the user tries to access an attribute of :class:`.MonoCut` that doesn't exist. It is used for accessing the custom attributes of cuts. We use it to look up the ``custom`` field: when it's None or empty, we'll just raise AttributeError as usual. If ``item`` is found in ``custom``, we'll return ``custom[item]``. If ``item`` starts with "load_", we'll assume the name of the relevant attribute comes after that, and that value of that field is of type :class:`~lhotse.array.Array` or :class:`~lhotse.array.TemporalArray`. We'll return its ``load`` method to call by the user. Example of attaching and reading an alignment as TemporalArray:: >>> cut = MonoCut('cut1', start=0, duration=4, channel=0) >>> cut.alignment = TemporalArray(...) >>> ali = cut.load_alignment() NzNo such attribute: load_)r AttributeError startswithr load_custom)rrr attr_namerrr __getattr__)s    zCustomFieldMixin.__getattr__csF||jvr t||jdus||jvrtd|d|j|=dS)z/Used to support ``del cut.custom_attr`` syntax.NzNo such member: '')rr __delattr__r r)rrrrrr$Ls   zCustomFieldMixin.__delattr__cCst|Sr )r)rrrrto_dictTszCustomFieldMixin.to_dictcCs.t||jdur |jnid}||j|<|S)zGReturn a copy of this object with an extra custom field assigned to it.Nr)rr copy)rrrcpyrrr with_customWs  zCustomFieldMixin.with_customcKs8ddlm}m}ddlm}|j|}t||r!|jdi|St||r3|jd|j |j d|St|t r|j|d}|durNd|vrN| d}|j|dd rb|j dd |i|S|jrs|jd||j |j d |S|j d||j |j d |St||r|jdi|Std |d |d|d)a! Load custom data as numpy array. The custom data is expected to have been stored in cuts ``custom`` field as an :class:`~lhotse.array.Array`, :class:`~lhotse.array.TemporalArray`, or :class:`~lhotse.image.image.Image` manifest. .. note:: It works with Array/Image manifests stored via attribute assignments, e.g.: ``cut.my_custom_data = Array(...)`` or ``cut = cut.attach_image('img', ...)``. :param name: name of the custom attribute. :return: a numpy array with the data. r)Array TemporalArray)Image)startduration_channel_selectorNchannel _unalignedFchannels)r1offsetr-zTo load z, the cut needs to have field z (or cut.custom['zW']) defined, and its value has to be a manifest of type Array, TemporalArray, or Image.r) lhotse.arrayr)r*lhotse.image.imager+r get isinstanceloadr,r-rr load_audio has_video load_video ValueError)rrkwargsr)r*r+rr1rrrr _sB       zCustomFieldMixin.load_customcCs|jdurdS||jvS)z Check if the Cut has a custom attribute with name ``name``. :param name: name of the custom attribute. :return: a boolean. NFrrrrrr has_customs  zCustomFieldMixin.has_customcCs$|jdus ||jvr dS|j|=|Sr rr=rrr drop_customszCustomFieldMixin.drop_custom)__name__ __module__ __qualname____doc__rrstrrrrr"r$r%r(npndarrayr boolr>r? __classcell__rrrrr s #6 r ) functoolsrtypingrrrnumpyrElhotser lhotse.utilsrrr r rrrrs