2@jdZddlmZddlZddlZddlZddlmZmZ ddl Z dZ n #e $rdZ YnwxYwe Z dd Z dd ZGd d ZdS)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]cjtr+ tj}|jS#t$rYnwxYwdS)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_mems GC:\Projects\Historical\venv\Lib\site-packages\numba/misc/memoryutils.pyget_available_memoryrsO +--G$ $    D  4s # 00Dict[str, Optional[int]]cBi}tr tjtj}|}|j|d<tj}|j|d<n!#tj tj f$rYnwxYwd|vrd|d<d|vrd|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) r r Processosgetpid memory_inforr r NoSuchProcess AccessDenied)rprocessmem_infors rget_memory_usager/sK  nRY[[11G**,,H!)K +--G'.'8K $ $$f&9:    D  K! E+%%#' K sA A,,B  B ceZdZUdZded<ded<ded<ded<d ed <d ed <ded <d ed<ddZejdZddZ 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_deltactj|_||_d|_d|_d|_d|_d|_d|_ dS)z6Initialize a MemoryTracker with empty monitoring data.N) rrr r"r#r$r%r&r'r()selfr"s r__init__zMemoryTracker.__init__dsE9;;    c#Ktj|_t|_ |Vtj|_t|_|j|jz |_|jdd}|jdd}|r|r||z nd|_dS#tj|_t|_|j|jz |_|jdd}|jdd}|r|r||z nd|_wxYw)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) timer#rr%r$r&r'getr()r* start_rssend_rsss rmonitorzMemoryTracker.monitorosL )++,.. ?JJJ!IKKDM.00DO MDO;DM)--eQ77Io))%33G!*>/6>w22<= NNN!IKKDM.00DO MDO;DM)--eQ77Io))%33G!*>/6>w22<= N ? ? ? ?s B::BErc "|j|jtd|jd}dd}t jdt j|j}|jdd}|jdd}d |jd |j d |d |j d dd||d||d||j dd||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 Fc|dS|dkrdSd}|r |dkrdnd}t|}dD]}|d kr ||d d |cS|d z}||d d S) z&Convert bytes to human readable formatNzN/Arz0 B-+)BKBMBGBg@z.2f z TB)abs) bytes_val show_signsignunits r format_bytesz/MemoryTracker.get_summary..format_bytess uA~~uD 5'!mmssII/ $ $v%%":I:::D:::::V# .I.... .r,z%H:%M:%SrrzName: zPID: zStart: z Duration: z.3fsz Start RSS: z End RSS: z RSS delta: T)r?zAvail memory: z | )F) r%r& ValueErrorr/r.strftime localtimer#r"r r'r(join)r*current_availablerBstart_tsr0r1bufs r get_summaryzMemoryTracker.get_summarysJ   $(?CDD D O// << / / / /$=T^DO-L-LMM%))%33 /%%eQ// !TY DH   h - - - - - 3,,y11 3 3 / W-- / / H,,t~FFF H H >\\*;<< > >  zz#r,N)r"r!)rr!) __name__ __module__ __qualname____doc____annotations__r+ contextlibcontextmanagerr2rKr,rrrTs  HHH III....,,,,    ???@444444r,r)rr)rr)rO __future__rrrQr.typingrrr r ImportError IS_SUPPORTEDrrrrSr,rrXs  #"""""  !!!!!!!!MMMKKKKK &""""Jpppppppppps %//