o
    i%                     @  sl   d Z ddlmZ ddlmZmZ ddlmZmZ ddl	m
Z
 ddlmZ g dZG dd	 d	eZe ZZd
S )z
A collection of dictionary-based wrappers around the signal operations defined in :py:class:`monai.transforms.signal.array`.

Class names are ended with 'd' to denote dictionary-based transforms.
    )annotations)HashableMapping)KeysCollectionNdarrayOrTensor)SignalFillEmpty)MapTransform)SignalFillEmptydSignalFillEmptyDSignalFillEmptyDictc                      s4   e Zd ZdZejZdd fd	d
ZdddZ  ZS )r	   aU  
    Applies the SignalFillEmptyd transform on the input. All NaN values will be replaced with the
    replacement value.

    Args:
        keys: keys of the corresponding items to model output.
        allow_missing_keys: don't raise exception if key is missing.
        replacement: The value that the NaN entries shall be mapped to.
    NF        keysr   allow_missing_keysboolc                   s   t  || t|d| _d S )N)replacement)super__init__r   signal_fill_empty)selfr   r   r   	__class__ d/home/dell461/cl/sdc2/last_ska_mid/HISourceFinder-master-l/src/monai/transforms/signal/dictionary.pyr   )   s   zSignalFillEmptyd.__init__data"Mapping[Hashable, NdarrayOrTensor]returnc                 C  s&   |  |D ]}| || ||< q|S )N)key_iteratorr   )r   r   keyr   r   r   __call__-   s   zSignalFillEmptyd.__call__)NFr   )r   r   r   r   )r   r   r   r   )	__name__
__module____qualname____doc__r   backendr   r   __classcell__r   r   r   r   r	      s
    
r	   N)r"   
__future__r   collections.abcr   r   monai.config.type_definitionsr   r   Zmonai.transforms.signal.arrayr   monai.transforms.transformr   __all__r	   r
   r   r   r   r   r   <module>   s   
