o 1&i@sdZddlmZddlZddlZddlZddlmZmZzddl Z dZ Wn e y/dZ Ynwe Z dd d Z dd d ZGdddZdS)a Memory monitoring utilities for measuring memory usage. Example usage: tracker = MemoryTracker("my_function") with tracker.monitor(): my_function() # Access data: tracker.rss_delta, tracker.duration, etc. # Get formatted string: tracker.get_summary() ) annotationsN)DictOptionalTFreturn Optional[int]cCs.trzt}|jWStyYdSwdS)z Get current available system memory in bytes. Used for memory threshold checking in parallel test execution. Returns: int or None: Available memory in bytes, or None if unavailable N) _HAS_PSUTILpsutilvirtual_memory available Exception)sys_memr AC:\wamp64\www\opt\env\Lib\site-packages\numba/misc/memoryutils.pyget_available_memorys  rDict[str, Optional[int]]c Cs~i}tr-ztt}|}|j|d<t}|j|d<Wn tj tj fy,Ynwd|vr5d|d<d|vr=d|d<|S)a^ Get memory usage information needed for monitoring. Returns only RSS and available memory which are the fields actually used by the MemoryTracker. Returns: dict: Memory usage information including: - rss: Current process RSS (physical memory currently used) - available: Available system memory rssr N) rrProcessosgetpid memory_inforr r Z NoSuchProcessZ AccessDenied)rprocessZmem_infor r r rget_memory_usage/s   rc@steZdZUdZded<ded<ded<ded<d ed <d ed <ded <d ed<dddZejddZdddZ dS) MemoryTrackerz A simple memory monitor that tracks RSS delta and timing. Stores monitoring data in instance attributes for later access. Each instance is typically used for monitoring a single operation. intpidstrnamez float | None start_timeend_timezDict[str, int | None] | None start_memory end_memorydurationz int | None rss_deltacCs8t|_||_d|_d|_d|_d|_d|_d|_ dS)z6Initialize a MemoryTracker with empty monitoring data.N) rrrrrrrr r!r")selfrr r r__init__ds  zMemoryTracker.__init__c cst|_t|_z2|VWt|_t|_|j|j|_|jdd}|jdd}|r8|r8|||_dSd|_dSt|_t|_|j|j|_|jdd}|jdd}|re|re|||_wd|_w)a Context manager to monitor memory usage during function execution. Records start/end memory usage and timing, calculates RSS delta, and stores all data in instance attributes. Args: name (str): Name/identifier for the function or operation being monitored Yields: self: The MemoryTracker instance for accessing stored data rrN) timerrrrr r!getr")r# start_rssend_rssr r rmonitoros6   zMemoryTracker.monitorrc Cs|jdus |jdurtd|jd}ddd}tdt|j}|jdd }|jdd }d |jd |j d |d |j ddd||d||d||j ddd||g}d |S)a Return a formatted summary of the memory monitoring data. Formats the stored monitoring data into a human-readable string containing name, PID, RSS delta, available memory, duration, and start time. Returns: str: Formatted summary string with monitoring results Note: Should be called after monitor() context has completed to ensure all data is available. Nz$Memory monitoring data not availabler FcSsz|durdS|dkr dSd}|r|dkrdnd}t|}dD]}|d kr0||d d |S|d }q||d d S) z&Convert bytes to human readable formatNzN/Arz0 B-+)BZKBMBGBg@z.2f z TB)abs)Z bytes_val show_signsignunitr r r format_bytess z/MemoryTracker.get_summary..format_bytesz%H:%M:%SrrzName: zPID: zStart: z Duration: z.3fsz Start RSS: z End RSS: z RSS delta: T)r2zAvail memory: z | )F) rr ValueErrorr&r%strftime localtimerrrr!r"join)r#Zcurrent_availabler5Zstart_tsr'r(bufr r r get_summarys"        zMemoryTracker.get_summaryN)rr)rr) __name__ __module__ __qualname____doc____annotations__r$ contextlibcontextmanagerr)r<r r r rrTs    r)rr)rr)r@ __future__rrrBr%typingrrrr ImportErrorZ IS_SUPPORTEDrrrr r r rs     %