Yf!ddlmZddlZddlZddlmZddlmZddlm Z m Z m Z ddl m Z ddlmZddlmZmZdd lmZed d Gd d ZGddeZeGddZeGddZeGddZe ee eedffZe ee eeeffZGddej>Z GddeZ!y)) annotationsN) dataclass)Enum)AnyDictUnion) processors)Frame) FrameRenderer ProcessorList)SessionT)frozeneqc0eZdZUdZded<ded<ded<y)SpeedscopeFramea3 Data class to store data needed for speedscope's concept of a frame, hereafter referred to as a "speedscope frame", as opposed to a "pyinstrument frame". This type must be hashable in order to use it as a dictionary key; a dictionary will be used to track unique speedscope frames. z str | Nonenamefilez int | NonelineN__name__ __module__ __qualname____doc____annotations__a/var/www/highfloat_scraper/venv/lib/python3.12/site-packages/pyinstrument/renderers/speedscope.pyrrs   rrceZdZdZdZdZy)SpeedscopeEventTypez?Enum representing the only two types of speedscope frame eventsOCN)rrrrOPENCLOSErrrrr sI D Errc0eZdZUdZded<ded<ded<y) SpeedscopeEventz Data class to store speedscope's concept of an "event", which corresponds to opening or closing stack frames as functions or methods are entered or exited. rtypefloatatintframeNrrrrr%r%'s  I Jrr%cZeZdZUdZded<ded<ded<dZded <d Zded <d Zded <y)SpeedscopeProfilezB Data class to store speedscope's concept of a "profile". strrlist[SpeedscopeEvent]eventsr' end_value start_valueeventedr&secondsunitN)rrrrrr2r&r5rrrr,r,4s8 I !!KD#D#rr,cZeZdZUdZded<ded<ded<dZded <d Zd ed <d Zded<y )SpeedscopeFilezE Data class encoding fields in speedscope's JSON file schema r-rzlist[SpeedscopeProfile]profilesz dict[str, list[SpeedscopeFrame]]sharedz2https://www.speedscope.app/file-format-schema.jsonschemaNNoneactive_profile_index pyinstrumentexporter)rrrrrr:r<r>rrrr7r7Bs9 I%% ,,FFCF!%$%"Hc"rr7ceZdZdZddZy)SpeedscopeEncoderz` Encoder class used by json.dumps to serialize the various speedscope data classes. ct|trE|j|j|j|j |j |jdSt|trE|j|j|j|j|j|jdSt|ttfr|j }|St|t"r |j$St&j(j+||S)N)z$schemaractiveProfileIndexr>r8r9)r&rr5 startValueendValuer/) isinstancer7r:rr<r>r8r9r,r&r5r2r0r/rr%__dict__rvaluejson JSONEncoderdefault)selfods rrJzSpeedscopeEncoder.defaultZs a (88&'&<&<JJJJ((   a* +mmKK((   a/?; rc|gSt|j|j|j}||jvr"t |j|j|<|j|}t tj|j|}|g}|jD]"}|j|j|$|xj|jz c_ |jr|xj|jz c_ t tj |j|}|j#||S)a Builds up a list of speedscope events that are used to populate the "events" array in speedscope-formatted JSON. This method has two notable side effects: * it populates the self._frame_to_index dictionary that matches speedscope frames with their positions in the "shared" array of speedscope output; this dictionary will be used to write this "shared" array in the render method * it accumulates a running total of time elapsed by accumulating the self_time spent in each pyinstrument frame; this running total is used by speedscope events to construct a flame chart. )rfunction file_pathline_norUlenr%rr"rTchildrenextend render_frame absorbed_timeis_synthetic_leaftimer#append)rKr*sframe sframe_index open_event events_arraychild close_events rr_zSpeedscopeRenderer.render_frames( =I!%--P -- -+.t/C/C+DD  (++F3 $%8%=%=t?O?OQ]^ /9l ^^ :E    1 1% 8 9 :" E///  " "    * &&9&?&?AQAQS_` K(rc|j|j}tjdtj|j }d|j d|}t||j||jg}t|jDcgc]}|}}d|i}t|||} dtj| tzScc}w)Nz%Y-%m-%dT%H-%M-%SzCPU profile for z at framesz%s )cls) preprocess root_framerbstrftime localtime start_timeprogramr,r_durationiterrUr7rHdumpsr@) rKsessionr*id_r sprofile_listrd sframe_list shared_dictspeedscope_files rrenderzSpeedscopeRenderer.renders 2 2 45==!4dnnWEWEW6XY&w&7tC5A dD$5$5e$N>N O2 DHH\H\C]-^f-^ -^- (}kJ ?8IJJJ ._s# Cctjtjtjtjtj tj gS)z Default Processors for speedscope renderer; note that processors.aggregate_repeated_calls is removed because speedscope is a timeline-based format. )r remove_importlibremove_tracebackhidemerge_consecutive_self_time"remove_unnecessary_self_time_nodesremove_irrelevant_nodes*remove_first_pyinstrument_frames_processor)rKs rdefault_processorsz%SpeedscopeRenderer.default_processorssD  ' '  + +  2 2  9 9  . .  A A   r)rVr)r*z Frame | NonerNr.)rvr )rNr ) rrrroutput_file_extensionrSr_r|r __classcell__)rWs@rrPrPus&. >?BK(  rrP)" __future__rrHrb dataclassesrenumrtypingrrrr=r pyinstrument.framer pyinstrument.renderers.baser r pyinstrument.sessionr rrr%r,r7r-r)SpeedscopeFrameDictTyper'SpeedscopeEventDictTyperIr@rPrrrrs" !###$D(  $4   ! $            # #  #sE#sD.$99:sE*=uc*I$JJK1((1B} } r