Yf$dZddlmZddlZddlmZmZmZmZddl m Z m Z m Z ddl mZmZedee dffZeeefZddZdd Zdd Zdd Z d dd Z dd Z d ddZ ddZy)z Processors are functions that take a Frame object, and mutate the tree to perform some task. They can mutate the tree in-place, but also can change the root frame, they should always be called like:: frame = processor(frame, options=...) ) annotationsN)AnyCallableDictUnion)SELF_TIME_FRAME_IDENTIFIERFrame FrameGroup)combine_framesdelete_frame_from_tree.c|y|jD]8}t|||jsd|jvs,t|d:|S)zR Removes ``|jD]}t|||jj dd|S)aH Converts a timeline into a time-aggregate summary. Adds together calls along the same call stack, so that repeated calls appear as the same frame. Removes time-linearity - frames are sorted according to total time spent. Useful for outputs that display a summary of execution (e.g. text and html outputs) N)intorc|jSN)time)cs rz*aggregate_repeated_calls..Zs qvvrT)keyreverse)r identifierr aggregate_repeated_calls _childrensort)rrchildren_by_identifierraggregate_frames rr'r';s }/1=   5 54U5E5EFO 5 78= "5#3#3 4=9 89  OO-t< Lrc<|y|jd|jddfd d fd |jD]U}|js:|r2tfd|jDrt |}||t ||W|S) aO Groups frames that should be hidden into :class:`FrameGroup` objects, according to ``hide_regex`` and ``show_regex`` in the options dict, as applied to the file path of the source code of the frame. If both match, 'show' has precedence. Options: ``hide_regex`` regular expression, which if matches the file path, hides the frame in a frame group. ``show_regex`` regular expression, which if matches the file path, ensures the frame is not hidden Single frames are not grouped, there must be at least two frames in a group. N hide_regex show_regexc|jxsd}duxrtj|}duxrtj|}|ry|ry|j S)NFT)rrematchis_application_code)rframe_file_path should_show should_hider-r.s rshould_be_hiddenz8group_library_frames_processor..should_be_hiddenxsb///R!-X288J3X !-X288J3X   ,,,,rcn|j||jD]}|s ||yr ) add_framer)rgroupradd_frames_to_groupr7s rr;z;group_library_frames_processor..add_frames_to_groups4 ^^ 2E&#E51 2rc3.K|] }|ywr ).0ccr7s r z1group_library_frames_processor..s+ZR,>+Z(Zu%E u -&ug>? Lrc|yd}|jD]L}|jtk(r5|r0|xj|jz c_|j H|}Kd}N|r|jD]}t ||d|S)z2 Combines consecutive 'self time' frames. NT)r recursive)rr&rr!remove_from_parentmerge_consecutive_self_time)rrrFprevious_self_time_framers rrHrHs  }# ,   9 9'(--;-((*,1('+ $ ,^^ PE 'w$ O P Lrc|yt|jdk(r:|jdjtk(rt |jdd|jD]}t |||S)z When a frame has only one child, and that is a self-time frame, remove that node and move the time to parent, since it's unnecessary - it clutters the output and offers no additional information. Nrnothingrr)lenrr&rr "remove_unnecessary_self_time_nodesrs rrNrNsm } 5>>aENN1$5$@$@D^$^u~~a0yIC*5'BC Lrc|y||j}|dkrd}|jdd}|jD]$}|j|z }||kst|d&|jD]}t ||||S) z Remove nodes that represent less than e.g. 1% of the output. Options: ``filter_threshold`` sets the minimum duration of a frame to be included in the output. Default: 0.01. NrggE6filter_thresholdg{Gz?rLr)r total_time)r!rBrr remove_irrelevant_nodes)rrrQrPrproportion_of_totals rrRrRs }ZZ  ?J{{#5t<B#jj:5 !1 1 "5y A B Ow:NO Lrc|ydd}dd}dd}|}||s|S|jd}||s|S|jd}||s|S||r|jd}||r|j|S)z The first few frames when using the command line are the __main__ of pyinstrument, the eval, and the 'runpy' module. I want to remove that from the output. Nc|jduxr:tjd|jxrt|jdkDS)Nz.*pyinstrument[/\\]__main__.pyr)rr1r2rMrrs ris_initial_pyinstrument_framezQremove_first_pyinstrument_frames_processor..is_initial_pyinstrument_framesA OO4 ' (:EOOL (ENN#a' rc|jdkDxr8|jduxr(d|jvxrt|jdkDS)N皙?zr)proportion_of_parentrrMrrVs r is_exec_framezAremove_first_pyinstrument_frames_processor..is_exec_framesO  & & , (t+ (eoo- (ENN#a'  rc|jdkDxrZ|jduxrJtjd|jxsd|jvxrt |j dkDS)NrYz .*runpy.pyzr)rZrr1r2rMrrVs ris_runpy_framezBremove_first_pyinstrument_frames_processor..is_runpy_framesg  & & , (t+ (-9`=MQVQ`Q`=` (ENN#a'  rrrA)rrG)rrrWr[r]results r*remove_first_pyinstrument_frames_processorr_s }   F ( 0 __Q F   __Q F & !  #    Mr)r Frame | NonerProcessorOptionsreturnr`)T)rr`rrarFboolrbr`r )rr`rrarQz float | Nonerbr`)__doc__ __future__rr1typingrrrrpyinstrument.framerr r pyinstrument.frame_opsr r ProcessorTypestrrarrr'rDrHrNrRr_r=rrrks# --LLI eE4K001 S>  $!H6tGK "2?C< "2*PT "2@LF; ;"2;;r