o Sij@sddlmZmZddlmZddlmZmZmZm Z da iida de ddfd d Z de fd d Zd edefddZGdddZdS)) lru_cachewraps)Lock)AnyCallableDictOptionalF)cached noncachedenabledreturnNcCsBt|tsJ|at||stdD]}|qdSdS)Nr ) isinstanceboolLHOTSE_CACHING_ENABLED AudioCacheenableLHOTSE_CACHED_METHOD_REGISTRYvalues cache_clear)r methodrB/home/ubuntu/.local/lib/python3.10/site-packages/lhotse/caching.pyset_caching_enableds  rcCstSN)rrrrris_caching_enabledsrrcsb|jtdvrtdd|td<tdd|td<t|dtffdd }|S) a4 Least-recently-used cache decorator. It enhances Python's built-in ``lru_cache`` with a dynamic lookup of whether to apply the cached, or noncached variant of the decorated function. To disable/enable caching globally in Lhotse, call:: >>> from lhotse import set_caching_enabled >>> set_caching_enabled(True) # enable >>> set_caching_enabled(False) # disable Currently it hard-codes the cache size at 512 items (regardless of the array size). Check :meth:`functools.lru_cache` for more details. r zMethod 'z' is already cached. We don't support caching different methods which have the same __qualname__ attribute (i.e., class name + method name).r i)maxsizer cs.tr td}ntd}||i|S)Nr r )rr)argskwargsmnamerrwrapperBs z"dynamic_lru_cache..wrapper) __qualname__r ValueErrorrrr)rr!rrrdynamic_lru_cache"s    r$c@seZdZUdZdZeed<dZeed<dZ eed<iZ e e e fed<eZeed <edd d Zed efddZede d ee fddZede de fddZed efddZedddZdS)ra Cache of 'bytes' objects with audio data. It is used to cache the "command" type audio inputs. By default it is disabled, to enable call `set_caching_enabled(True)` or `AudioCache.enable()`. The cache size is limited to max 100 elements and 500MB of audio. A global dict `__cache_dict` (static member variable of class AudioCache) is holding the wavs as 'bytes' arrays. The key is the 'source' identifier (i.e. the command for loading the data). Thread-safety is ensured by a threading.Lock guard. F_AudioCache__enabledgeͽAmax_cache_memorydmax_cache_elements_AudioCache__cache_dict_AudioCache__lockTcCs||_|s |dSdSr)r%_AudioCache__clear_cache)clsr rrrris zAudioCache.enabler cCs|jSr)r%r,rrrr oszAudioCache.enabledkeycCsZ|jsdS|j||jvr|j|WdS WddS1s&wYdS)zo Test if 'key' is in the chache. If yes return the bytes array, otherwise return None. N)r%r*r))r,r.rrr try_cachess $zAudioCache.try_cachevaluecCs|jsdSt||jkrdS|jJt|j|jkr-|jtt|jt|j|jkst|t |jkrN|jtt|jt|t |jks8||j|<WddS1s^wYdS)z Add the new (key,value) pair to cache. Possibly free some elements before adding the new pair. The oldest elements are removed first. N) r%lenr&r*r)r(popnextiterr_AudioCache__cache_memory)r,r.r0rrr add_to_caches "zAudioCache.add_to_cachecCs(d}|jD] \}}|t|7}q|S)zn Return size of AudioCache values in bytes. (internal, not to be called from outside) r)r)itemsr1)r,ansr.r0rrr__cache_memoryszAudioCache.__cache_memoryNcCs6|j|jWddS1swYdS)z3 Clear the cache, remove the data. N)r*r)clearr-rrr __clear_caches "zAudioCache.__clear_cache)T)r N)__name__ __module__r"__doc__r%r__annotations__r&intr(r)rstrbytesrr* classmethodrr rr/r6r5r+rrrrrPs&      r) functoolsrr threadingrtypingrrrrrrrrrr$rrrrrs  .